您的位置:首页 > 其它

依据名称检测进程是否存在

2010-11-16 14:00 225 查看
借鉴于网上。

#include <psapi.h >

#pragma comment(lib,"psapi.lib")

DWORD ProcessToPID(const char *InputProcessName);
int AddPrivilege(const char *Name);
const char DESTPROC[19] = "client.exe";
const char DESTPROCDebug[19] = "clientd.exe";

部分函数:

void CWatingDlg::OnTimer(UINT nIDEvent)
{
static int Value = 1;
m_Wait+= Value ;
if (m_Wait > 3100)
{
//KillTimer(nIDEvent);
Value = 0;
}
else
{
if (m_Wait%50 == 0 )
{
m_Pro_Waiting.GetRange(m_low,m_up);
m_Pro_Waiting.SetRange(0, m_up*2.0);
m_Wait *=2.0;
}
}
m_Pro_Waiting.SetPos(m_Wait);
static int nTime = 0;
DWORD pid = ProcessToPID(DESTPROC);
DWORD pidDebug = ProcessToPID(DESTPROCDebug);
nTime++;
if (nTime % 10 == 0)
{
if (pid != 0 || pidDebug != 0)
{
return;
}
else
{
exit(1);
}
}
if (nTime > 1200)
{
exit(1);
}
}

DWORD ProcessToPID(const char *InputProcessName)
{
DWORD aProcesses[1024], cbNeeded, cProcesses;
unsigned int i;
HANDLE hProcess = NULL;
HMODULE hMod = NULL;
char szProcessName[MAX_PATH] = "UnknownProcess ";

AddPrivilege(SE_DEBUG_NAME);

// 计算目前有多少进程, aProcesses[]用来存放有效的进程PIDs
if (!EnumProcesses( aProcesses, sizeof(aProcesses), &cbNeeded ) )
{
return 0;
}

cProcesses = cbNeeded / sizeof(DWORD);
// 按有效的PID遍历所有的进程
for ( i = 0; i < cProcesses; i++ )
{
// 打开特定PID的进程
hProcess = OpenProcess( PROCESS_QUERY_INFORMATION |
PROCESS_VM_READ,
FALSE, aProcesses[i]);
// 取得特定PID的进程名
if ( hProcess )
{
if(EnumProcessModules( hProcess, &hMod, sizeof(hMod), &cbNeeded) )
{
GetModuleBaseName( hProcess, hMod,
szProcessName, sizeof(szProcessName) );
//将取得的进程名与输入的进程名比较,如相同则返回进程PID
if(!stricmp(szProcessName, InputProcessName))
{
CloseHandle( hProcess );
return aProcesses[i];
}
}
}//end of if ( hProcess )
}//end of for
//没有找到相应的进程名,返回0
CloseHandle( hProcess );
return 0;
}

int AddPrivilege(const char *Name)
{
HANDLE hToken;
TOKEN_PRIVILEGES tp;
LUID Luid;

if (!OpenProcessToken(GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY,
&hToken))
{
return 1;
}

if (!LookupPrivilegeValue(NULL,Name,&Luid))
{
return 1;
}

tp.PrivilegeCount = 1;
tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
tp.Privileges[0].Luid = Luid;

if (!AdjustTokenPrivileges(hToken,
0,
&tp,
sizeof(TOKEN_PRIVILEGES),
NULL,
NULL))
{
return 1;
}

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: