在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。