WaitForSingleObject 的應用
你還再為怎麼判斷一個程式是否結束而苦惱嗎?或許你在想可以用一個迴圈每隔幾秒來判斷一次這個進程是否結束,但是這樣會有問題,如果間隔時間太短就會很占CPU 資源,間隔時間太長又不能及時知道程式已經結束,或許有人馬上就想到WaitForSingleObject 函數,不錯Win32
API 提供的這個函數能很好的解決這個問題,但是在使用WaitForSingleObject 函數的過程中很容易忽略一些問題。小弟不才,在這裡略說一二。
先看一下函數的原形
DWORD WaitForSingleObject (HANDLE hHandle, DWORD dwMilliseconds);
參數hHandle 是同步對象的控制代碼。
參數dwMilliseconds 是以毫秒為單位的逾時間隔,如果該參數為0 ,那麼函數就測試同步對象的狀態並立即返回,如果該參數為INFINITE ,則逾時間隔是無限的。
表12.1 WaitForSingleObject 的傳回值
WaitForSingleObject 函數能使線程阻塞其自身執行的等待函數。只有在作為其參數的同步對象產生訊號時才會返回。或在超過規定的等待時間後,不管有無訊號,函數也都會返回。在等待函數未返回時,線程處於等待狀態,此時線程只消耗很少的CPU 時間。
1 、等待自己建立的進程
怎麼使用WaitForSingleObject 函數呢,假設我們現在啟動系統的記事本程式,然後等待記事本結束,代碼可以這樣寫:
CString strCommand="";
STARTUPINFO si = {sizeof(si)};
PROCESS_INFORMATION pi;
CHAR cWindowsDir[MAX_PATH];
CHAR cCommand[MAX_PATH];
GetWindowsDirectory(cWindowsDir,MAX_PATH); // 擷取系統目錄
strCommand= (CString)cWindowsDir+"//NotePad.exe";
strcpy(cCommand,strCommand); // 得到記事本的程式路徑
BOOL bRet = CreateProcess(NULL,cCommand,NULL,NULL,FALSE,0,NULL,NULL,&si,&pi);//啟動記事本程式
if (bRet)
{
if (!WaitForSingleObject(pi.hProcess,INFINITE))// 等待記事本程式關閉
{
AfxMessageBox(" 記事本已經關閉!");
}
}
2 、等待已經存在的進程
現在我們知道記事本什麼時候關閉了,那麼關閉的時候我們就可以做我們想做的事情。但是一般的程式都不是通過CreateProcess 建立的,要擷取已經在運行狀態的程式怎麼辦呢?沒錯,要想辦法得到運行程式的進程ID ,這就要藉助OpenProcess 函數,OpenProcess 函數用來開啟一個已經存在的進程對象,原形如下:
HANDLE OpenProcess ( DWORD dwDesiredAccess ,
BOOL bInheritHandle ,
DWORD dwProcessId
)
dwDesiredAccess 參數,進程的操作許可權。
bInheritHandle 控制代碼是否繼承
dwProcessId 進程 ID
傳回值:如果調用成功,函數返回一個該進程的控制代碼,如果失敗返回 NULL 。
怎麼擷取進程ID 呢,需要用到另外幾個函數CreateToolHelp32Snapshot 、Process32First、Process32Next ,這幾個函數都是在TLHELP32.H 中,用來一起完成對系統進程的枚舉(就像工作管理員把進程都枚舉出來一樣)
HANDLE WINAPI CreateToolhelp32Snapshot (
DWORD dwFlags ,
DWORD th32ProcessID
);
dwFlags 參數是一個標誌,一般使用TH32CS_SNAPPROCESS ,
th32ProcessID 參數是進程ID ,如果第一個參數是TH32CS_SNAPPROCESS ,這個參數就會被忽略。
傳回值,如果調用成功,返回一個Snapshot 的控制代碼,失敗返回INVALID_HANDLE_VALUE ;
BOOL WINAPI Process32First (
HANDLE hSnapshot,
LPPROCESSENTRY32 lppe
);
hSnapshot 參數 需要賦值為通過CreateToolHelp32Snapshot 的控制代碼
lppe 參數 是進程的入口資訊,需要定義一個PROCESSENTRY32 類型的變數來擷取這個值。
Process32Next 和 Process32First 函數的參數意義一致,這三個函數一起就可以不斷對進程進行枚舉,直到Process32Next 的傳回值為FALSE 。
那麼可以這樣實現等待已經在啟動並執行進程對象
PROCESSENTRY32 pe32;
HANDLE hSnaphot;
HANDLE hApp;
DWORD dProcess;
hSnaphot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0); // 擷取進程快照
Process32First(hSnaphot, &pe32); // 指向第一個進程
do
{
if (lstrcmpi(pe32.szExeFile,_T("NotePad.exe"))==0) // 尋找進程名稱為 NotePad.exe (記事本的程式)
{
dProcess = pe32.th32ProcessID;
break;
}
}while(Process32Next(hSnaphot, &pe32)); // 不斷迴圈直到取不到進程
hApp= OpenProcess(
PROCESS_VM_OPERATION|SYNCHRONIZE,FALSE,dProcess);
// 根據進程 ID 擷取程式的控制代碼
if (!WaitForSingleObject(hApp,INFINITE))// 等待記事本程式關閉
{
AfxMessageBox(" 記事本已經關閉!");
}
到現在,我們不但知道了怎麼使用WaitForSingleObject 的等待函數,還知道了GetWindowsDirectory、CreateProcess 、OpenProcess 、CreateToolHelp32Snapshot 、Process32First 、Process32Next等函數的用法,這些函數一起實現了我們想要的功能,而且還在其他方面起著重要的作用。