GetProcessMemoryInfo失败,错误6句柄无效

时间:2018-06-25 04:47:33

标签: c++ windows winapi process

我正在尝试获取使用C ++在系统中运行的进程的列表。我使用Windows API中可用的功能(例如OpenProcess和CreateToolhelp32Snapshot)来完成它。

问题是该代码对于大多数进程都可以正常工作,但是不能单独获取关键系统进程的内存信息,我还提供了seDebugPrivilege并仅以管理员身份运行该程序。

我得到类似

的输出

File Name PID Memory Used

winlogon.exe 1248 2432 KB

fontdrvhost.exe 1308

WARNING: GetProcessMemoryInfo failed with error 6 (The handle is invalid)

dwm.exe 1384

WARNING: GetProcessMemoryInfo failed with error 6 (The handle is invalid)

svchost.exe 1448 4744 KB

我的代码是:

BOOL GetProcessList( )
{
    HANDLE hProcessSnap;
    HANDLE hProcess;
    HANDLE hToken;
    PROCESSENTRY32 pe32;
    DWORD dwPriorityClass;
    PROCESS_MEMORY_COUNTERS pmc;

  hProcessSnap = CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 );
  if( hProcessSnap == INVALID_HANDLE_VALUE )
  {
      char err[]="CreateToolhelp32Snapshot (of processes)";
    printError( err );
    return( FALSE );
  }

  pe32.dwSize = sizeof( PROCESSENTRY32 );

  if( !Process32First( hProcessSnap, &pe32 ) )
  {
    char err[]="Process32First";
    printError( err ); // show cause of failure
    CloseHandle( hProcessSnap );          // clean the snapshot object
    return( FALSE );
  }
  if(!OpenThreadToken(GetCurrentThread(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, FALSE, &hToken))
    {
        if (GetLastError() == ERROR_NO_TOKEN)
        {
            if (!ImpersonateSelf(SecurityImpersonation))
            return FALSE;

            if(!OpenThreadToken(GetCurrentThread(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, FALSE, &hToken)){
                char err[]="OpenThreadToken";
                printError( err );
                return FALSE;
            }
         }
        else
            return FALSE;
     }
    SetPrivilege(hToken, SE_DEBUG_NAME, FALSE);
  do
  {
    printf( "%-25s", pe32.szExeFile );
    dwPriorityClass = 0;
    SIZE_T  dwMin, dwMax;
    hProcess = OpenProcess( PROCESS_QUERY_LIMITED_INFORMATION , FALSE, pe32.th32ProcessID );
    printf("%-10d", pe32.th32ProcessID );
    if(GetProcessMemoryInfo( hProcess, (PROCESS_MEMORY_COUNTERS *)&pmc, sizeof(pmc)))
    {

        printf( "%-10d KB", pmc.PagefileUsage/1024); 
    } else{
        char err[]="GetProcessMemoryInfo";
        printError( err );
    }
        printf("\n");
    CloseHandle(hProcess);
  }while( Process32Next( hProcessSnap, &pe32 ) );
  CloseHandle( hProcessSnap );
  return( TRUE );
}

1 个答案:

答案 0 :(得分:2)

要使此代码生效,必须进行以下更改:

SetPrivilege(hToken, SE_DEBUG_NAME, FALSE)

收件人:

SetPrivilege(hToken, SE_DEBUG_NAME, TRUE)

正如其他人所评论的那样,要使此代码正确运行,必须执行以下操作:

  • 检查并报告所有可能的错误情况
  • 完成后释放所有资源,在这种情况下,释放hToken(通过CloseHandle()),并在需要时调用revertToSelf()

除此之外,这是一个不错的努力,因此请竖起大拇指。

链接到SetPrivilege(),以防将来的访客需要它:

https://msdn.microsoft.com/en-us/library/windows/desktop/aa446619.aspx

黑客很开心。