yehao's Blog

防止C++程序重复运行的几种方法

转自http://hi.baidu.com/dreamyguy/blog/item/aacc6f44086afe45500ffef9.html

有时候,为了某些要求,我们希望程序实例只运行一次。而在VB6中,我们可以很轻易的根据App.hPreInstance来判断程序是否已经运行。但是在C++中,这一切就变得不是那么容易。

虽然WinMain函数有hPreInstance参数来指示,但是那是在Win16位的前提下,到了32Bit时代,那个参数已经完全成为摆设。

而本文正好探讨了如何防止C++程序重复运行的方法。

PS:因为本人使用MFC,所以为了方便,所有代码均为以MFC为基础。大家可以根据自己需要更改

  1.查找窗体

对于存在GUI窗体(CUI暂不讨论)的程序来说,最容易想到的就是利用FindWindow,以标题作为参数进行查找主窗体,然后使其关闭即可。

通常,我们能写出如下代码:

// Find Window by Caption
// Add this code in InitInstance function of class 
// you have derived from the CWinApp class

HWND hWnd 
= ::FindWindow(NULL, "MFCDialog");

if (hWnd)
{
AfxMessageBox(
"Has been running");
return FALSE;
}

 

以上的代码可以简单的起到防止重复启动的效果,但是局限性很大。

首先,由于在FindWindow中要指定窗体的标题,如果窗体的标题在程序运行中是不断变化的,那么就给搜索带来了一定难度。

而且,如果其他程序也恰好是用相同的标题的话……- -#。当然,你可以通过在FindWindow中指定类名来减少错误。但是如果你看过我前面写的文章的话,你就会发现,MFC注册窗口类并不是那么随意,而是经过N次阴谋筹划之后……

看来这方法的局限性的确很大- -#

  2.额外窗体存储

此方法来源于对上面一种方法的补充,因为通过搜索MFC的窗体类比较困难,而且准确度不一定高。所以,我想到了使用额外窗体存储(Extra Widnow Memory)的方法

PS:关于什么是额外窗体存储,请自行google或MSDN或查看我曾经写的The Analyses Of Windows Runnning Principle

如果你使用SDK进行开发,可以在创建窗体时填充这一属性,然后用GetWindowLong获取。

而由于我使用MFC,所以我更关注如何在MFC中使用这一属性。

一般来说,我们可以使用SetWindowLong对额外窗体存储进行填充,然后用GetWindowLong获取,最后配合FindWindow来检验程序是否重复运行。

// Add this code in InitDialog function
// and you can specify any number you want
BOOL bRet = ::SetWindowLong(GetSafeHwnd(), GWL_USERDATA, 256);

// Add this code in InitInstance function
// Find Window by using extra memory

HWND hWnd 
= FindWindow(NULL, "MFCDialog");

if (hWnd)
{
BOOL bRet 
= ::GetWindowLong(hWnd, GWL_USERDATA);

if (256 == bRet)  // compare
    {
AfxMessageBox(
"Has been running");
return FALSE;
}
}

 

 3.全局原子

你可以使用GlobalAddAtom将某个特定的字符串添加到全局原子列表(Global Atom Table),然后在程序运行时检查该字符串即可。

但是这个方法有一个致命的弱点,程序退出时,Windows不会自动为你删除添加到列表中的Atom,而是需要你自己使用GlobalDeleteAtom进行删除。

这就意味着,如果你的程序意外的退出了,没有删除添加的Atom,那么,你的程序将无法运行。

所以,这并不是一个好方法。


  4.枚举进程

这或许是一个毕竟正常,或者说相对稳定的方法。

我们可以使用CreateToolhelp32Snapshot或者EnumProcess来枚举当前的进程,然后检查是否已经运行。如果担心存在同名的进程,还可以检查路径。

//   关闭一个进程  
static   bool   CloseApplication(CString   sAppToClose,   CString   sAppToUpdate,    
CString   sFilename,   bool   bAskUser   /*   =   true   */)  
{  
//   Declare   variables  
HANDLE   hProcessSnap   =   NULL;  
PROCESSENTRY32   pe32   =   {NULL};  
CString   sCompare,   sTemp;  

       //sProcessFileName是你的进程名  
CString   sProcesFilename   =   m_pPath->ExtractFilename(sFilename);  
CLanguage   *   pLanguage   =   CLanguage::Instance();  

//调整统一大小写格式,方便比较、  
#if   (_MFC_VER   <   0x0700)  
sProcesFilename.MakeLower();  
#else  
CString   sLowerTemp;  
sLowerTemp   =   sProcesFilename;  
sProcesFilename   =   sLowerTemp.MakeLower();  
#endif  

hProcessSnap   =   CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,   0);  

pe32.dwSize   =   sizeof(PROCESSENTRY32);  
//   系统进程快照  
if (Process32First(hProcessSnap,   &pe32))  
{  
do  
{  
sCompare.Format("%s",   pe32.szExeFile);  

#if   (_MFC_VER   <   0x0700)  
sCompare.MakeLower();  
#else  
CString   sLowerTemp;  
sLowerTemp   =   sCompare;  
sCompare   =   sLowerTemp.MakeLower();  
#endif  


//   Check   if   we   found   the   right   process  
if(sCompare   ==   sProcesFilename)  
{  
//   If   we   are   not   in   hidden   mode  
if(bAskUser)  
{  
//Set   up   confirmation   text  
sTemp.Format(pLanguage->GetString(IDS_CONFIRMATION_CLOSEAPPLICATION_EXPLANATION),  
sAppToClose,   sAppToUpdate,   sAppToClose);  

//Ask   confirmation  
CConfirmationDlg   dlgConfirm(pLanguage->GetString(IDS_CONFIRMATION_CLOSEAPPLICATION_TITLE),  
pLanguage->GetString(IDS_CONFIRMATION_TITLE),   sTemp,  
pLanguage->GetString(IDS_GENERAL_YES),   pLanguage->GetString(IDS_GENERAL_NO));  
if(dlgConfirm.DoModal()   ==   IDNO)  
{  
return   false;  
}  
}  

//   Get   handle   to   process  
HANDLE   hProcess   =   OpenProcess(PROCESS_ALL_ACCESS,   FALSE,   pe32.th32ProcessID);  

//   Exit   process  
DWORD   exCode;  
GetExitCodeProcess(hProcess,   &exCode);  
TerminateProcess(hProcess,   exCode);  

//   Wait   until   process   is   finished  
if   (WaitForSingleObject(hProcess,   30000)   ==   WAIT_TIMEOUT)  
return   false;  
}  
}   while   (Process32Next(hProcessSnap,   &pe32));  
}  
else  
{  
return   false;  
}  

return   true;  
}

PS:在Vista下使用EnumProcess时,要注意权限问题,OpenProcess增加了一个新的权限常数,仅限Vista。如果不增加这个参数,很多进程是无法被枚举出来的(不过MS不印象我们自己的进程- -#)。


  5.互斥对象

使用互斥对象来防止程序重复运行是一个很常用的做法,而且M$也推荐使用这种方法。和上面的几种方法相比,需要写的代码少,而且效率比较高。所谓方便易用~

一般我们会使用CreateMutex来创建互斥体,当第二次创建相同的互斥体时,这个API会返回前一个互斥体的Handle,而GetLastError则会返回ERROR_ALREADY_EXITS

// Mutex Object 
// Add this code in InistInstance function
  
HANDLE hMutex 
= NULL;
TCHAR 
*lpszName = "TestMutex";

hMutex 
= ::CreateMutex(NULL, FALSE, lpszName);
DWORD dwRet 
= ::GetLastError();

if (hMutex)
{
if (ERROR_ALREADY_EXISTS == dwRet)
{
AfxMessageBox(
"Has been running");
CloseHandle(hMutex);  
// should be closed
      return FALSE;
}
}
else
{
AfxMessageBox(
"Create Mutex Error");
}

// Add this code in Destruction function
  
::CloseHandle(hMutex);

使用互斥体时要注意几个问题:
在CreateMutex之后马上GetLastError,GetLastError是一个很复杂的API,任何牵涉到GetLastError的操作在执行之后,都会覆盖先前的值。

把正常的CloseHandle写到窗体的析构函数或者程序对象的析构函数里。不要在CreateMutex之后立刻CloseHandle,否则互斥对象会被清空。

这也是我当初所犯的错误,(不知道网上那么多错误的代码是不是经过Debug的囧),当互斥对象的最后一个Handle被Close之后,互斥对象将被删除。如果程序在退出时没有清空互斥对象,Windows将会执行这一操作。当然,把次操作交给OS不是一个好习惯。

详情请看MSDN的引用:
引用:
Use the CloseHandle function to close the handle. The system closes the handle automatically when the process terminates. The mutex object is destroyed when its last handle has been closed.
转自:http://bbs.cfan.com.cn/viewthread.php?tid=793295
http://topic.csdn.net/t/20060424/09/4708145.html
http://student.csdn.net/space.php?uid=110004&do=thread&id=3278

posted on 2011-05-03 18:28 厚积薄发 阅读(677) 评论(0)  编辑 收藏 引用 所属分类: Windows编程


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


导航

<2024年11月>
272829303112
3456789
10111213141516
17181920212223
24252627282930
1234567

统计

常用链接

留言簿

随笔分类

文章分类

文章档案

搜索

最新评论