C++实现 物理内存监控(支持VC6.0 和 VS2005等)

说明:

• 1.  监控 系统 和 单个进程 物理内存使用情况;
• 2. 适用于 VC6.0 和 VS2005 等.

---

文件 "MemoryUsageStatus.h"

// 版权所有：(C)2018，Fjianhong
// 文件名称：MemoryUsageStatus.h
// 编码作者：Fjianhong
// 完成日期：2018-08-05
// 内容摘要：监控物理内存使用情况,声明文件
#pragma once

// 系统物理内存数据(来自系统定义)
struct MEMORYSTATUSEX
{ 
	DWORD     dwLength;
	DWORD     dwMemoryLoad;
	DWORDLONG ullTotalPhys;
	DWORDLONG ullAvailPhys;
	DWORDLONG ullTotalPageFile;
	DWORDLONG ullAvailPageFile;
	DWORDLONG ullTotalVirtual;
	DWORDLONG ullAvailVirtual;
	DWORDLONG ullAvailExtendedVirtual;
};

// 物理内存使用状态
class CMemoryUsageStatus
{
public:
	// 获取系统当前 物理内存可使用大小(KB)
	static DWORDLONG GetSystemCurMemoryAvailSize();

	// 获取系统 物理内存总大小(KB)
	static DWORDLONG GetSystemMemoryTotalSize();

	// 获取系统当前 物理内存使用率(百分比)
	static double GetSystemCurMemoryLoadPercentage();

	// 获取进程当前 内存使用大小(KB)
	static DWORDLONG GetProcessCurMemoryUsageSize(const PCTSTR & sProcessName);

private:
	// 获取系统当前物理内存使用状态
	static BOOL GetCurMemoryUsageStatus(MEMORYSTATUSEX & CurMemoryStatus);

	// 获取进程句柄
	static HANDLE GetProcessHandleByName(const PCTSTR & sProcessName);
};

文件 "MemoryUsageStatus.cpp"

// 版权所有：(C)2018，Fjianhong
// 文件名称：MemoryUsageStatus.cpp
// 编码作者：Fjianhong
// 完成日期：2018-08-05
// 内容摘要：监控物理内存使用情况,实现文件

#include "StdAfx.h"
#include "MemoryUsageStatus.h"
#include "tlhelp32.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

// 进程物理内存数据(来自系统定义)
struct PROCESS_MEMORY_COUNTERS
{
	DWORD dwLength;
	DWORD PageFaultCount;
	SIZE_T PeakWorkingSetSize;
	SIZE_T WorkingSetSize;
	SIZE_T QuotaPeakPagedPoolUsage;
	SIZE_T QuotaPagedPoolUsage;
	SIZE_T QuotaPeakNonPagedPoolUsage;
	SIZE_T QuotaNonPagedPoolUsage;
	SIZE_T PagefileUsage;
	SIZE_T PeakPagefileUsage;
};

const int nByteToKB = 1024;

typedef BOOL (CALLBACK* GlobalMemoryStatusEx)(MEMORYSTATUSEX* pMem);
typedef BOOL (CALLBACK* GlobalProcessMemoryStatusEx)(HANDLE hProcess, PROCESS_MEMORY_COUNTERS *pPsmemCounters, DWORD dwLength);

static HMODULE s_KernelDll = ::LoadLibrary("Kernel32.dll");
static HMODULE s_PsapiDll = ::LoadLibrary("Psapi.dll");
static GlobalMemoryStatusEx s_pGlobalMemoryStatusEx = NULL;
static GlobalProcessMemoryStatusEx s_GlobalProcessMemoryStatusEx = NULL;

//获取系统当前 物理内存可使用大小(KB)
DWORDLONG CMemoryUsageStatus::GetSystemCurMemoryAvailSize()
{
	MEMORYSTATUSEX CurMemoryStatus;
	BOOL bMemStatus = GetCurMemoryUsageStatus(CurMemoryStatus);
	if (!bMemStatus)
	{
		return 0;
	}

	DWORDLONG dAvailSize = CurMemoryStatus.ullAvailPhys;
	if (dAvailSize != 0)
	{
		dAvailSize /= nByteToKB;
	}

	return dAvailSize;
}

//获取进程当前 物理内存总大小(KB)
DWORDLONG CMemoryUsageStatus::GetSystemMemoryTotalSize()
{
	MEMORYSTATUSEX CurMemoryStatus;
	BOOL bMemStatus = GetCurMemoryUsageStatus(CurMemoryStatus);
	if (!bMemStatus)
	{
		return 0;
	}

	DWORDLONG dTotalPhys = CurMemoryStatus.ullTotalPhys;
	if (dTotalPhys != 0)
	{
		dTotalPhys /= nByteToKB;
	}

	return dTotalPhys;
}

//获取系统当前 物理内存使用率(百分比)
double CMemoryUsageStatus::GetSystemCurMemoryLoadPercentage()
{
	MEMORYSTATUSEX CurMemoryStatus;
	BOOL bMemStatus = GetCurMemoryUsageStatus(CurMemoryStatus);
	if (!bMemStatus)
	{
		return 0.0;
	}

	return (double)CurMemoryStatus.dwMemoryLoad;
}

//获取系统当前 物理内存使用状态
BOOL CMemoryUsageStatus::GetCurMemoryUsageStatus(MEMORYSTATUSEX &CurMemoryStatus)
{
	if (s_KernelDll == NULL)
	{
		return FALSE;
	}
	
	//获取物理内存函数地址
	s_pGlobalMemoryStatusEx = (GlobalMemoryStatusEx)::GetProcAddress(s_KernelDll, "GlobalMemoryStatusEx");
	if (s_pGlobalMemoryStatusEx == NULL)
	{
		return FALSE;
	}
	
	CurMemoryStatus.dwLength = sizeof(CurMemoryStatus);
	BOOL bMemory = s_pGlobalMemoryStatusEx(&CurMemoryStatus);
	if (!bMemory)
	{
		return FALSE;
	}
	
	return TRUE;
}

// 获取进程当前 内存使用大小(KB)
DWORDLONG CMemoryUsageStatus::GetProcessCurMemoryUsageSize(const PCTSTR & sProcessName)
{
	if ((sProcessName == NULL) || (s_PsapiDll == NULL))
	{
		return 0;	
	}
	
	//通过进程名称获取进程句柄
	HANDLE hProcess = GetProcessHandleByName(sProcessName);
	if(hProcess == NULL)
	{
		return 0;
	}

	//获取进程物理内存函数地址
	s_GlobalProcessMemoryStatusEx = (GlobalProcessMemoryStatusEx)::GetProcAddress(s_PsapiDll, "GetProcessMemoryInfo");
	if(s_GlobalProcessMemoryStatusEx == NULL)
	{
		return 0;
	}

	PROCESS_MEMORY_COUNTERS ProMemInfo;
	ProMemInfo.dwLength = sizeof(ProMemInfo);
	BOOL bProMem = s_GlobalProcessMemoryStatusEx(hProcess, &ProMemInfo, sizeof(ProMemInfo));
	if (!bProMem)
	{
		return 0;
	}

	DWORDLONG dProcessMem = ProMemInfo.WorkingSetSize;
	if (dProcessMem != 0)
	{
		dProcessMem /= nByteToKB;
	}
	
	return dProcessMem;
}

// 获取进程句柄
HANDLE CMemoryUsageStatus::GetProcessHandleByName(const PCTSTR & sProcessName)
{
	if (sProcessName == NULL)
	{
		return NULL;
	}
	
	CString strProcessName(sProcessName);
	CString strTemp = strProcessName.Right(4);
	if (strTemp != ".exe")
	{
		strProcessName += ".exe";
	}

	PROCESSENTRY32 ProEntry;
	ProEntry.dwSize = sizeof(PROCESSENTRY32);	
	HANDLE hSnapshot = ::CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
	if (!Process32First(hSnapshot, &ProEntry))
	{
		return NULL;
	}

	DWORD nProID(0);
	while (TRUE)
	{
		ProEntry.dwSize = sizeof(PROCESSENTRY32);
		if (!Process32Next(hSnapshot, &ProEntry))
		{
			break;
		}
		
		if (lstrcmpi(ProEntry.szExeFile, strProcessName) == 0)
		{
			nProID = ProEntry.th32ProcessID;
			break;
		}
	}
	CloseHandle(hSnapshot);
	hSnapshot = NULL;

	HANDLE hProcess(NULL);
	if (nProID != 0)
	{
		hProcess = ::OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, nProID);
	}

	return hProcess;
}

文件 "TestMemoryUsageStatus.h"

#include "MemoryUsageStatus.h"

void Test()
{
        // 获取系统当前 物理内存已使用率(百分比)
	double dPercentage = CMemoryUsageStatus::GetSystemCurMemoryLoadPercentage();

	// 获取系统当前 物理内存可使用大小(KB)
	DWORDLONG dMemoryAvailSize = CMemoryUsageStatus::GetSystemCurMemoryAvailSize();

	//获取QQ内存使用大小(KB),X86默认单个进程使用内存不能超过2G(20480KB)
	DWORDLONG szProcessUsageMem = CMemoryUsageStatus::GetProcessCurMemoryUsageSize("QQ.exe");
}