信息來源:
http://hi.baidu.com/wwwanq/文章作者:fictiony
當我們在做一些管理平臺類的程序(比如Windows的任務管理器)時,往往需要限制程序只能打開一個實例。解決這個問題的大致思路很簡單,無非是在程序打開的時候判斷一下是否有與自己相同的進程開著,如果有,則關閉自身,否則正常運行。
但是,問題就出在如何判別是否有一個與自己相同的進程開著上面。我在網上搜索了一下相關的文章,發現對于這個問題的解決不外乎以下幾種方式:
1、在進程初始化時使用::CreateMutex創建一個互斥對象,通過檢測互斥對象是否已存在來確定該程序是否已經運行。
該方式的確可以很容易的實現判別程序實例是否已存在,只需要在InitInstance方法開頭添加以下語句:
m_hUnique = ::CreateMutex(NULL, FALSE, UNIQUE_ID);
if (GetLastError() == ERROR_ALREADY_EXISTS) return FALSE;
UNIQUE_ID為具有唯一性的字符串,一般可以用VC++為主程序頭文件自動生成的包含標識宏(就是.h文件頂上的那一長串宏定義),當然,也可以用工具自己手動生成,隨君所好了^^。要注意的是別忘了在ExitInstance方法中用 CloseHandle(m_hUnique) 將該互斥對象關閉。但這種方式存在一個很大的問題,就是很難獲取已打開程序實例的主窗口句柄。而我們絕大多數時候,都需要將那個程序實例的主窗口激活。為了獲取主窗口句柄,就需要再用到后面提到的其他方法。
2、遍歷所有已經打開的進程主窗口,比較窗口標題,如果找到滿足條件的標題,則表示程序已經運行,并激活該窗口。
這種方式雖然可以找到程序的主窗口,但問題明顯:A.如果窗口標題經常變化怎么辦(比如標題中會帶有打開文檔的文件名)?B.如果其他程序的主窗口標題恰好與該程序的相同怎么辦?
第一個問題可以通過寫注冊表或者寫INI文件的方式來解決。即當主窗口標題改變時,將新標題寫入注冊表或者INI文件。不過這種解決方式也忒麻煩了吧-_-|| 第二個問題就麻煩了,至少我還沒有找到好的解決方案。如果非要說一個,那我提議你“想盡辦法”“不擇手段”的將窗口標題設的和別的程序絕對不同。不過估計搞定了這步,你半條命也快沒了。
3、用::SetProp給主窗口添加一個具有唯一性的屬性值,以便在進程初始化的時候可以通過遍歷所有窗口的該屬性來判斷。
添加屬性值的代碼一般可以放在InitInstance方法的最后,如下:
::SetProp(m_pMainWnd->m_hWnd, "UNIQUE_ID", (HANDLE)UNIQUE_ID);
UNIQUE_ID是一個具有唯一性的整數值(為什么不能用字符串?因為字符串的比較需要將字符串讀取出來,而這兒只能記錄字符串地址,在別的程序里這個地址無意義,所以無法讀出這個字符串)。這種方式僅有的問題就出在如何確定該整數值是具有唯一性的。我們后面提出的解決方法,就是在這種方法的基礎上發展出來的。
在總結了上述幾種方式的利弊之后,我發現,只需要為程序建立一個具有唯一性的整數值,一方面可以通過這個值是否存在來判斷程序是否已經運行(::CreateMutex其實也是類似的概念),另一方面可以通過將這個值賦給主窗口,以便能夠找到已打開的程序實例的主窗口句柄。于是,ATOM量便派上用場了(ATOM變量類型等同于WORD,因而是一個整數值)。
ATOM量本質上就是散列表的鍵標識符,其對應鍵值為一個字符串。每個程序都有自己的ATOM量表,同時Windows也有一個全局的ATOM表。我們要用的方法就是,為程序創建一個全局的ATOM量,通過這個量是否存在來判斷程序是否已經運行,并通過將這個量作為屬性值添加到主窗口來標識這個主窗口。具體過程如下:
1、給主程序App類添加一個ATOM類型的成員變量:m_aAppId,作為程序ID。
2、在InitInstance方法開頭添加以下代碼(UNIQUE_ID是具有唯一性的字符串宏):
m_aAppId = ::GlobalFindAtom(UNIQUE_ID); //查找程序ID是否存在
if (m_aAppId) //程序ID存在,激活已打開的程序實例的主窗口
{
HWND hWnd = ::GetWindow(::GetForegroundWindow(), GW_HWNDFIRST);
for (; hWnd; hWnd = ::GetWindow(hWnd, GW_HWNDNEXT))
{
if ((ATOM)::GetProp(hWnd, "APP_ID") == m_aAppId)
{
if (::IsIconic(hWnd)) ::ShowWindow(hWnd, SW_RESTORE); //還原最小化的窗口
::SetForegroundWindow(hWnd); //激活窗口
m_aAppId = 0; //賦值0是為了防止ExitInstance中將找到的ATOM量刪除
break;
}
}
return FALSE;
}
else //程序ID不存在,創建程序ID
{
m_aAppId = ::GlobalAddAtom(APP_ID);
}
3、在InitInstance方法最后為主窗口添加標識屬性:
::SetProp(m_pMainWnd->m_hWnd, "APP_ID", (HANDLE)m_aAppId);
4、在ExitInstance方法中添加下面代碼以刪除程序ID:
if (m_aAppId) ::GlobalDeleteAtom(m_aAppId);
心得:該方法所用到的ATOM量是一個應用廣泛的技術,如::CreateMutex、::SetProp等API函數都間接用到了ATOM量。利用它,我們可以做很多需要用到唯一性驗證的事情。