一.概述
一些重要的程序,必須讓它一直跑著;而且還要時時關(guān)心它的狀態(tài)——不能讓它出現(xiàn)死鎖現(xiàn)象。當(dāng)然,如果一個主程序會出現(xiàn)死鎖,肯定是設(shè)計或者編程上的失誤。我們首要做的事是,把這個Bug揪出來。但如果時間緊迫,這個Bug又“飄忽不定”,那么,我們還是先寫一個“看門狗”,暫時應(yīng)一下急吧。
“看門狗”的需求描述:“看門狗”的運行不出現(xiàn)界面窗口,具有一定的隱蔽性;定時判斷目標(biāo)進(jìn)程是否運行在當(dāng)前系統(tǒng)中,如果沒有則啟動目標(biāo)進(jìn)程;判斷目標(biāo)進(jìn)程是否“沒有響應(yīng)”,如果是則終止目標(biāo)進(jìn)程;如果目標(biāo)進(jìn)程“沒有響應(yīng)”的次數(shù)超過一定的數(shù)量,則將計算機系統(tǒng)重啟。
二.預(yù)備知識
首先要介紹兩個主要的函數(shù),能夠判斷目標(biāo)進(jìn)程是否“沒有響應(yīng)”。在User32.dll中(沒有文檔公開),Win2k/NT下的IsHungAppWindow和Win9X下的IsHungThread;前者是以一個窗口句柄作為參數(shù),后者是以線程ID作為參數(shù)。我們可以通過VC開發(fā)工具的Depends查到這兩個函數(shù)。
要使用這兩個函數(shù),我們必須先動態(tài)導(dǎo)入,如下:
if (m_hUser32 == NULL)
{
??? m_hUser32 = GetModuleHandle("USER32.DLL");
}
if (m_hUser32)
{
??? m_IsHungNT? = (HUNG_FUNNT) GetProcAddress(m_hUser32, "IsHungAppWindow");
??? m_IsHung9X? = (HUNG_FUN9X) GetProcAddress(m_hUser32, "IsHungThread");
}
另外,還有如下知識點:
1.??? 如何讓窗口隱藏(當(dāng)然通過Windows任務(wù)管理器還是可以看到的)
在框架窗口類的PreCreateWindow中修改窗口風(fēng)格,如下:
BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
??? if( !CFrameWnd::PreCreateWindow(cs) )
??????? return FALSE;
??? // TODO: Modify the Window class or styles here by modifying
??? //? the CREATESTRUCT cs
??? cs.dwExStyle |= WS_EX_TOOLWINDOW;? // Make invisible in taskbar
??? cs.style????? = WS_POPUP;????????? // Hide the main window
??? return TRUE;
}
2.??? 如何讓“看門狗”只運行一個進(jìn)程
使用互斥量。在CWatchDogApp::InitInstance()中,執(zhí)行如下代碼:
bool CWatchDogApp::IsUniqueCopyInProc()
{
??? m_Mutex = CreateMutex(NULL, TRUE, "System Watch Dog");
??? if (GetLastError() == ERROR_ALREADY_EXISTS)
??? {
??????? return false;
// 本文轉(zhuǎn)自 C++Builder 研究 -?http://www.ccrun.com/article.asp?i=546&d=b7r01j
??? }
??? return true;
}
該函數(shù)如果返回false,說明已經(jīng)有一個WatchDog進(jìn)程在運行了,當(dāng)前進(jìn)程就沒有必要再執(zhí)行下去了。在InitInstance如下處理:
if (!IsUniqueCopyInProc())
return FALSE;
3.??? 如何判斷當(dāng)前操作系統(tǒng)類型
bool CWatchDogApp::IsWinNT()
{?
??? OSVERSIONINFO OSVersionInfo;
??? OSVersionInfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
??? GetVersionEx(&OSVersionInfo);?
??? if (OSVersionInfo.dwPlatformId == VER_PLATFORM_WIN32_NT)
??? {
??????? return true;
??? }
??? return false;
}
4.??? 如何自動重啟計算機
在Win9x和Win2k/NT下,重啟計算機的處理略有不同:
if (theApp.IsWinNT())
{
??? // 在Win NT/2000下賦予關(guān)閉系統(tǒng)的權(quán)限
??? static HANDLE hToken;
??? static TOKEN_PRIVILEGES tp;
??? static LUID luid;
::OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY, &hToken ) ;
??? ::LookupPrivilegeValue( NULL, SE_SHUTDOWN_NAME, &luid );
??? tp.PrivilegeCount?????????? = 1;
??? tp.Privileges[0].Luid?????? = luid;
??? tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
??? ::AdjustTokenPrivileges( hToken, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), NULL, NULL );
??? return ::ExitWindowsEx(EWX_REBOOT | EWX_FORCE, 0);
}
else
{
??? return ::ExitWindowsEx(EWX_REBOOT | EWX_FORCE, 0);
}
5.??? 如何啟動、結(jié)束其他進(jìn)程
啟動進(jìn)程用CreateProcess,終止進(jìn)程用TerminateProcess。參考代碼如下:
bool CWatchDogView::RunTheSysProc()
{
??? char??? szPath[MAX_PATH];
??? GetModuleFileName(NULL, szPath, MAX_PATH);
??? CString strPath = szPath;
??? strPath = strPath.Left(strPath.ReverseFind('\')) + "\HungDemo.exe";
??? STARTUPINFO??????????? StartInfo;
??? PROCESS_INFORMATION??? procStruct;
??? memset(&StartInfo,0,sizeof(STARTUPINFO));
??? StartInfo.cb = sizeof(STARTUPINFO);
??? if (!::CreateProcess(
??????? (LPCTSTR) strPath,
??????? NULL,
??????? NULL,
??????? NULL,
??????? FALSE,
??????? NORMAL_PRIORITY_CLASS,
??????? NULL,
??????? NULL,
??????? &StartInfo,
??????? &procStruct))
??????? return false;
??? return true;
}
需要提醒的是,TerminateProcess是在萬不得已的情況下使用的,它不會進(jìn)入進(jìn)程使用的DLL的入口點通知“脫離”(Detaching)狀態(tài)。有時候,這樣做是很危險的(DLL內(nèi)部的全局?jǐn)?shù)據(jù)可能受影響較大)。
6.定時檢測子進(jìn)程的運行情況代碼如下:
void CWatchDogView::OnTimer(UINT nIDEvent)?
{
?// Get the target window handle
?HWND??? hTarget = ::FindWindow(NULL, "抄表數(shù)據(jù)庫維護");
?DWORD?? dwProcessID, dwThreadID;
?dwThreadID = ::GetWindowThreadProcessId(hTarget, &dwProcessID);
?
?// The Target window found, the verify the whether responding
?if (hTarget)??
?{
??//這里我處理得簡單了一點,源程序還加了重新啟動機器等處理
??}
?else
?{???????
??//如果沒有檢測到所監(jiān)視的子進(jìn)程在運行,則重新啟動子進(jìn)程(也就是自己想要被監(jiān)視的程序)
??RunTheSysProc();
?????????????????
?}?
?CView::OnTimer(nIDEvent);
}
7.??? 如何讓W(xué)in2k/NT自動登錄
修改注冊表。在HKEY_LOCAL_MACHINE目錄下的SoftwareMicrosoftWindows NT CurrentVersionWinLogon下的AutoAdminLogon(字符串型)設(shè)置成1,并在DefaultUserName設(shè)置默認(rèn)登錄用戶,DefaultPassword設(shè)置默認(rèn)用戶的密碼。
8.??? 如何讓W(xué)in2k/NT登錄成功后直接執(zhí)行你的程序(而不是默認(rèn)的文件瀏覽器)
修改注冊表。在注冊表HKEY_LOCAL_MACHINESOFTWAREMicrosoftWindows NT CurrentVersionWinlogonShell的值從原先的explorer.exe修改為自己程序的絕對路徑。
三.功能演示(Win2k/NT下)
友情提醒:開始演示之前,請先將你目前的工作保存。運行“看門狗”WatchDog;同時使用Ctrl+Alt+Del打開“Windows任務(wù)管理器”。稍候片刻,可以看到目標(biāo)程序HungDemo會被啟動(這個程序模擬了“沒有響應(yīng)”)。然后,WatchDog發(fā)現(xiàn)這個程序“沒有響應(yīng)”,則把它殺掉,然后重新啟動一個新的HungDemo進(jìn)程。如此的處理重復(fù)六次以后,系統(tǒng)會自動重啟。
以上是我從網(wǎng)上摘下來的一篇關(guān)于軟件看門狗的文章,我只是在原文的基礎(chǔ)上加以修改,以便于像我這樣的初學(xué)者更容易理解。對于像我樣的新手來說,其中的很多東西我還是不怎么懂,像GetModuleHandle(),GetProcAddress()之類的函數(shù),我不明白深層次的意思,我也不知道怎么修改注冊表。
????? 其實這些功能我都沒有用,我只用到了讓程序在出錯的情況下能自動重啟。沒有讓機器重啟,因為我覺得如果讓機器重啟的話,可能還會遇到很多問題,比如開機密碼之類的。其實作者講述的已經(jīng)挺明白的了,按照他的講述加上自己閱讀源程序,相信大家都能把它修改成自己想要的程序。其實,我覺得,要實現(xiàn)看門狗這個功能,最主要的是理解CreateProcess((LPCTSTR)strPath,NULL,NULL,NULL,FALSE,NORMAL_PRIORITY_CLASS,NULL,NULL,&StartInfo,&procStruct))函數(shù)和::FindWindow(NULL, "自己的程序名稱")這兩個函數(shù)。其中,CreateProcess函數(shù)主要是創(chuàng)建一個子進(jìn)程,就是被看門狗監(jiān)視的那個程序,修改程序時,只要把HungDemo.exe改成自己程序的exe就行了。而::FindWindow函數(shù)主要是用來得到子進(jìn)程的窗口句柄,以便用來判斷被監(jiān)視的程序是否已經(jīng)退出。





