WTSEnumerateProcesses WTSFreeMemory函数在win7\vista下内存泄漏的问题

最近头都搞大了,不管什么内存泄漏工具都检查不出来程序有问题。后来决定自己动手测试,找到的严重问题如下:

在win7、vista及以上系统中,调用WTSEnumerateProcesses函数后,使用传统的WTSFreeMemory来释放内存无用,此函数在WINXP以上系统中屁用都没有!!

后来在MDSN在线查找相关函数,发现了WTSFreeMemoryEx,使用此函数来释放内存就OK!

VC6中没有定义此函数,高版本应该是有的。所以如果用VC6的话,就得自己定义这个函数了。

代码如下:

//需定义参数的一个类型
  typedef enum  { 
  WTSTypeProcessInfoLevel0,
   WTSTypeProcessInfoLevel1,
   WTSTypeSessionInfoLevel1
 } WTS_TYPE_CLASS;


//定义这个函数

 typedef BOOL (WINAPI *myWTSFreeMemoryEx)(
    WTS_TYPE_CLASS WTSTypeClass,
    PVOID pMemory,
    ULONG NumberOfEntries

//获取函数地址
 HMODULE hWtsApi32 = LoadLibrary("wtsapi32.dll"); 
 myWTSFreeMemoryEx pWTSFreeMemoryEx=(myWTSFreeMemoryEx)GetProcAddress(hWtsApi32,"WTSFreeMemoryExA");


//调用示例

  WTS_PROCESS_INFO *pWtspi;
  DWORD dwCount;
  
  if(!WTSEnumerateProcesses(0, 0, 1, &pWtspi,&dwCount))  
   return 0;

 WTSFreeMemory(pWtspi);
if (pWTSFreeMemoryEx != NULL)  //直接这么判断就好了,如果未取得这个函数就代表是老系统,老系统也不存在这个问题
 pWTSFreeMemoryEx(WTSTypeProcessInfoLevel0,pWtspi,dwCount);



此问题有朋友在国外论坛提问过,但无人回答,如需转载请留原链接吧!

#include <windows.h> #include <QString> #include <QMessageBox> int main(int argc, char *argv[]) { HANDLE mutex = ::CreateMutex(Q_NULLPTR, true, (LPCWSTR)qApp->applicationName().toStdWString().c_str()); if (GetLastError() == ERROR_ALREADY_EXISTS) { QMessageBox waringBox(QMessageBox::Warning, QStringLiteral("警告"), QStringLiteral("当前程序已运行,不能再次运行!")); waringBox.setButtonText(QMessageBox::Ok, QStringLiteral("确定")); waringBox.setStandardButtons(QMessageBox::Ok); waringBox.exec(); ::CloseHandle(mutex); return 0; } // 获取当前登录用户的会话ID DWORD sessionId; ProcessIdToSessionId(GetCurrentProcessId(), &sessionId); // 遍历所有用户会话,关闭其他用户会话中的该程序进程 DWORD sessionCount; if (ProcessIdToSessionId(GetCurrentProcessId(), &sessionId) && WTSQuerySessionInformationW(WTS_CURRENT_SERVER_HANDLE, WTS_CURRENT_SESSION, WTS_INFO_CLASS::WTS_SESSION, (LPWSTR*)&sessionCount, &sessionId)) { WTS_SESSION_INFO* pSessionInfo = NULL; if (WTSEnumerateSessions(WTS_CURRENT_SERVER_HANDLE, 0, 1, &pSessionInfo, &sessionCount)) { for (DWORD i = 0; i < sessionCount; i++) { DWORD processCount; if (WTSEnumerateProcesses(pSessionInfo[i].hServer, pSessionInfo[i].SessionId, 1, (PWTS_PROCESS_INFO*)&processCount)) { WTS_PROCESS_INFO* pProcessInfo = NULL; if (WTSEnumerateProcesses(pSessionInfo[i].hServer, pSessionInfo[i].SessionId, 1, &pProcessInfo, &processCount)) { for (DWORD j = 0; j < processCount; j++) { if (pProcessInfo[j].SessionId != sessionId && wcsstr(pProcessInfo[j].pProcessName, qApp->applicationName().toStdWString().c_str())) { HANDLE hProcess = ::OpenProcess(PROCESS_TERMINATE, FALSE, pProcessInfo[j].ProcessId); if (hProcess) { ::TerminateProcess(hProcess, 0); ::CloseHandle(hProcess); } } } ::WTSFreeMemory(pProcessInfo); } } } ::WTSFreeMemory(pSessionInfo); } } // 正常执行程序 QApplication a(argc, argv); MainWindow w; w.show(); return a.exec(); } 上面代码会报出 error: 'WTS_CURRENT_SERVER_HANDLE' was not declared in this scope错误,请修复后给我完整代码
05-30
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值