C/C++实现linux和Windows查看系统、进程、服务的内存和CPU使用

#include "CheckProcessCpu.h" #include <string.h> #include <stdlib.h> #ifdef WIN32 #include <conio.h> #include <TlHelp32.h> #include <process.h> #define MY_PROCESS_ERROR(Condition) do{ if (!(Condition)) goto Exit0; } while (false) static DWORD g_sdwTickCountOld = 0; // ??һ?ε?tick???? static LARGE_INTEGER g_slgProcessTimeOld; // ??????????һ?ε?ʱ??ռ?? static DWORD g_sdwProcessorCoreNum = 0; // ???????????? static HANDLE g_shExitEvent = NULL; // ?߳??˳????? int GetProcessCpuPercent(const HANDLE hProcess, const DWORD dwElepsedTime) { int nProcCpuPercent = 0; BOOL bRetCode = FALSE; FILETIME CreateTime, ExitTime, KernelTime,UserTime; LARGE_INTEGER lgKernelTime; LARGE_INTEGER lgUserTime; LARGE_INTEGER lgCurTime; bRetCode = GetProcessTimes(hProcess, &CreateTime, &ExitTime, &KernelTime, &UserTime); if (bRetCode) { lgKernelTime.HighPart = KernelTime.dwHighDateTime; lgKernelTime.LowPart = KernelTime.dwLowDateTime; lgUserTime.HighPart = UserTime.dwHighDateTime; lgUserTime.LowPart = UserTime.dwLowDateTime; lgCurTime.QuadPart = (lgKernelTime.QuadPart + lgUserTime.QuadPart) / 10000; nProcCpuPercent = (int)((lgCurTime.QuadPart - g_slgProcessTimeOld.QuadPart) * 100 / dwElepsedTime); g_slgProcessTimeOld = lgCurTime; nProcCpuPercent = nProcCpuPercent / g_sdwProcessorCoreNum; } else { nProcCpuPercent = -1; } return nProcCpuPercent; } unsigned int BeforeProcIDCpuPercent(DWORD ProcessID) { HRESULT hr = E_FAIL; HANDLE hProcess = NULL; DWORD dwProcessId = ProcessID; DWORD dwRetVal = 0; DWORD dwCurrentTickCount = 0; DWORD dwElapsedTime = 0; int nProcessCpuPercent = 0; //TARGET_PROCESS *pTargetProcess = (TARGET_PROCESS *)pArg; hProcess = OpenProcess(PROCESS_QUERY_INFORMATION , FALSE, dwProcessId); MY_PROCESS_ERROR(hProcess); dwRetVal = WaitForSingleObject(g_shExitEvent, 200); dwCurrentTickCount = GetTickCount(); dwElapsedTime = dwCurrentTickCount - g_sdwTickCountOld; g_sdwTickCountOld = dwCurrentTickCount; nProcessCpuPercent = GetProcessCpuPercent(hProcess, dwElapsedTime); dwRetVal = WaitForSingleObject(g_shExitEvent, 200); if (WAIT_OBJECT_0 == dwRetVal || WAIT_FAILED == dwRetVal ) { goto Exit0; } dwCurrentTickCount = GetTickCount(); dwElapsedTime = dwCurrentTickCount - g_sdwTickCountOld; g_sdwTickCountOld = dwCurrentTickCount; nProcessCpuPercent = GetProcessCpuPercent(hProcess, dwElapsedTime); return nProcessCpuPercent; Exit0: if (hProcess) { CloseHandle(hProcess); hProcess = NULL; } return -1; } unsigned int ProcIDCpuPercent(DWORD ProcessID) { SYSTEM_INFO sysInfo = { 0 }; GetSystemInfo(&sysInfo); g_sdwProcessorCoreNum = sysInfo.dwNumberOfProcessors; g_shExitEvent = CreateEvent(NULL, TRUE, FALSE, NULL); if (!g_shExitEvent) { CloseHandle(g_shExitEvent); g_shExitEvent = NULL; return -1; } return BeforeProcIDCpuPercent(ProcessID); } HRESULT GetAllProcCpuPercent(const TCHAR* cpszExeFileName, unsigned int *proc_ret,unsigned int *proc_nums) { HRESULT hr = E_FAIL; DWORD dwPID; PROCESSENTRY32 pe = { 0 }; HANDLE hSnapshot = NULL; if (NULL == cpszExeFileName) { hr = HRESULT_FROM_WIN32(ERROR_BAD_ARGUMENTS); goto Exit0; } pe.dwSize = sizeof(PROCESSENTRY32); hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); if (INVALID_HANDLE_VALUE == hSnapshot) { hr = HRESULT_FROM_WIN32(GetLastError()); goto Exit0; } if (FALSE == Process32First(hSnapshot, &pe)) { hr = HRESULT_FROM_WIN32(ERROR_NO_MORE_FILES); goto Exit0; } hr = S_FALSE; do { if (0 == strcmp(cpszExeFileName, pe.szExeFile)) { dwPID = pe.th32ProcessID; (*proc_nums) ++; (*proc_ret) +=ProcIDCpuPercent(dwPID); } }while(Process32Next(hSnapshot, &pe)); hr = S_OK; Exit0: if(hSnapshot) { CloseHandle(hSnapshot); hSnapshot = NULL; } return hr; } void CheckProcessCpu(const char *process_name,unsigned int *res,unsigned int *proc_nums) { *res =0; *proc_nums =0; GetAllProcCpuPercent(process_name,res,proc_nums); } #else #ifndef PROC_DIRECTORY #define PROC_DIRECTORY "/proc/" #endif static int IsNumeric(const char* ccharptr_CharacterList) { for ( ; *ccharptr_CharacterList; ccharptr_CharacterList++) if (*ccharptr_CharacterList < '0' || *ccharptr_CharacterList > '9') return 0; // false return 1; // true } void CheckProcessCpu(const char *process_name,unsigned int *res,unsigned int *proc_nums) { *res = 0; *proc_nums = 0; DIR *dir_proc = NULL; char status_path[100] ="\0"; char proc_name_flag[100] = "\0"; char proc_name[64]= "\0"; char proc_pid_flag[100] = "\0"; unsigned int proc_pid =0; struct dirent* de_DirEntity = NULL ; dir_proc = opendir(PROC_DIRECTORY) ; if (dir_proc == NULL) { perror("Couldn't open the " PROC_DIRECTORY " directory") ; return ; } // ???ݽ??????ƣ????????н???ID????????ռ??CPU???? while ( (de_DirEntity = readdir(dir_proc)) ) { if (de_DirEntity->d_type == DT_DIR) { if (IsNumeric(de_DirEntity->d_name)){ strcpy(status_path, PROC_DIRECTORY) ; strcat(status_path, de_DirEntity->d_name) ; strcat(status_path, "/status") ; FILE * fp = fopen(status_path,"rt"); if (fp) { fgets(proc_name_flag,sizeof(proc_name),fp); sscanf(proc_name_flag,"%*s%s",proc_name); if (!strcmp(proc_name,process_name)) { fgets(proc_pid_flag,sizeof(proc_pid_flag),fp); fgets(proc_pid_flag,sizeof(proc_pid_flag),fp); fclose(fp); sscanf(proc_pid_flag,"%*s%d",&proc_pid); (*proc_nums)++; (*res) += get_process_cpu_useage(proc_pid); } } } } } closedir(dir_proc); } #endif