在C++中计算相同的运行进程的总数

Question

我正在寻找一种方法来检测具有相同进程名称的正在运行的进程的数量。

例如，我跑了三次记事本。

NOTEPAD.EXE NOTEPAD.EXE NOTEPAD.EXE

那么它将返回3.

目前，我有这些代码来检测正在运行的进程，但不包括其正在运行的进程数量。

#include <iostream> 
#include <windows.h> 
#include <tlhelp32.h> 
#include <tchar.h> 

bool IsProcessRunning(const char *ProcessName); 

int main() 
{ 
    char *notepadRunning = (IsProcessRunning("notepad.exe")) ? "Yes" : "No"; 
    std::cout << "Is Notepad running? " << notepadRunning; 
    return 0; 
} 

bool IsProcessRunning(const char *ProcessName) 
{ 
    PROCESSENTRY32 pe32 = { sizeof(PROCESSENTRY32) }; 
    HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); 

    if(Process32First(hSnapshot, &pe32)) 
    { 
     do 
     { 
     if(_tcsicmp(pe32.szExeFile, ProcessName) == 0) 
     { 
      CloseHandle(hSnapshot); 
      return true; 
     } 
     } while(Process32Next(hSnapshot, &pe32)); 
    } 

    CloseHandle(hSnapshot); 
    return false; 
}

任何形式的帮助，将不胜感激:)

感谢。

Answer 1

您使用的是正确的API，即CreateToolhelp32Snapshot,Process32First和Process32Next。正如你所做的那样，你应该使用struct PROCESSENTRY32中的szExeFile成员。

当您找到目前的匹配时，您正在从您的功能返回。相反，你应该增加一个计数器而不是返回。并返回一个int，而不是一个布尔值。另外请务必不要做CloseHandle（hSnapshot）;直到功能结束后，你有数。

还要确保在枚举之前先获得SeDebugPrivilege的特权，这样您将获得所有会话和用户的所有流程。

为了获得特权，所以你得到的所有会话：

acquirePrivilegeByName(SE_DEBUG_NAME);// SeDebugPrivilege 

凡acquirePrivilegeByName被定义为：

BOOL acquirePrivilegeByName(
          const TCHAR  *szPrivilegeName) 
{ 
    HANDLE   htoken; 
    TOKEN_PRIVILEGES tkp; 
    DWORD   dwerr; 

    //---------------- adjust process token privileges to grant privilege 
    if (szPrivilegeName == NULL) 
    { 
     SetLastError(ERROR_INVALID_PARAMETER); 
     return FALSE; 
    } 

    if (!LookupPrivilegeValue(NULL, szPrivilegeName, &(tkp.Privileges[0].Luid))) 
     return FALSE; 

    tkp.PrivilegeCount = 1; 
    tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; 

    if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &htoken)) 
     return FALSE; 

    if (!AdjustTokenPrivileges(htoken, FALSE, &tkp, 0, NULL, NULL) || 
     GetLastError() != ERROR_SUCCESS) // may equal ERROR_NOT_ALL_ASSIGNED 
    { 
     dwerr = GetLastError(); 
     CloseHandle(htoken); 
     SetLastError(dwerr); 
     return FALSE; 
    } 

    CloseHandle(htoken); 
    SetLastError(ERROR_SUCCESS); 

    return TRUE; 
} //acquirePrivilegeByName() 

Answer 2

您可能在访问权限方面存在问题。特别是在Vista上，我不认为你可以枚举所有进程，除非你使用提升的特权运行。

您也可以尝试使用EnumProcesses API。有一个complete example on how to enumerate all processes。