當前位置:編程學習大全網 - 編程語言 - 用C寫的windows服務程序 怎麽響應WM

用C寫的windows服務程序 怎麽響應WM

第壹步:主函數和全局定義

首先,包含所需的頭文件。例子要調用 Win32 函數(windows.h)和磁盤文件寫入(stdio.h):

#include

#include

接著,定義兩個常量:

#define SLEEP_TIME 5000

#define LOGFILE "C:\\MyServices\\memstatus.txt"

SLEEP_TIME 指定兩次連續查詢可用內存之間的毫秒間隔。在第二步中編寫服務工作循環的時候要使用該常量。

LOGFILE 定義日誌文件的路徑,妳將會用 WriteToLog 函數將內存查詢的結果輸出到該文件,WriteToLog 函數定義如下:

int WriteToLog(char* str)

{

FILE* log;

log = fopen(LOGFILE, "a+");

if (log == NULL)

return -1;

fprintf(log, "%s\n", str);

fclose(log);

return 0;

}

聲明幾個全局變量,以便在程序的多個函數之間***享它們值。此外,做壹個函數的前向定義:

SERVICE_STATUS ServiceStatus;

SERVICE_STATUS_HANDLE hStatus;

void ServiceMain(int argc, char** argv);

void ControlHandler(DWORD request);

int InitService();

現在,準備工作已經就緒,妳可以開始編碼了。服務程序控制臺程序的壹個子集。因此,開始妳可以定義壹個 main 函數,它是程序的入口點。對於服務程序來說,main 的代碼令人驚訝地簡短,因為它只創建分派表並啟動控制分派機。

void main()

{

SERVICE_TABLE_ENTRY ServiceTable[2];

ServiceTable[0].lpServiceName = "MemoryStatus";

ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain;

ServiceTable[1].lpServiceName = NULL;

ServiceTable[1].lpServiceProc = NULL;

// 啟動服務的控制分派機線程

StartServiceCtrlDispatcher(ServiceTable);

}

壹個程序可能包含若幹個服務。每壹個服務都必須列於專門的分派表中(為此該程序定義了壹個 ServiceTable 結構數組)。這個表中的每壹項都要在 SERVICE_TABLE_ENTRY 結構之中。它有兩個域:

lpServiceName: 指向表示服務名稱字符串的指針;當定義了多個服務時,那麽這個域必須指定;

lpServiceProc: 指向服務主函數的指針(服務入口點);

分派表的最後壹項必須是服務名和服務主函數域的 NULL 指針,文本例子程序中只宿主壹個服務,所以服務名的定義是可選的。

服務控制管理器(SCM:Services Control Manager)是壹個管理系統所有服務的進程。當 SCM 啟動某個服務時,它等待某個進程的主線程來調用 StartServiceCtrlDispatcher 函數。將分派表傳遞給 StartServiceCtrlDispatcher。這將把調用進程的主線程轉換為控制分派器。該分派器啟動壹個新線程,該線程運行分派表中每個服務的 ServiceMain 函數(本文例子中只有壹個服務)分派器還監視程序中所有服務的執行情況。然後分派器將控制請求從 SCM 傳給服務。

註意:如果 StartServiceCtrlDispatcher 函數30秒沒有被調用,便會報錯,為了避免這種情況,我們必須在 ServiceMain 函數中(參見本文例子)或在非主函數的單獨線程中初始化服務分派表。本文所描述的服務不需要防範這樣的情況。

分派表中所有的服務執行完之後(例如,用戶通過“服務”控制面板程序停止它們),或者發生錯誤時。StartServiceCtrlDispatcher 調用返回。然後主進程終止。

第二步:ServiceMain 函數

Listing 1 展示了 ServiceMain 的代碼。該函數是服務的入口點。它運行在壹個單獨的線程當中,這個線程是由控制分派器創建的。ServiceMain 應該盡可能早早為服務註冊控制處理器。這要通過調用 RegisterServiceCtrlHadler 函數來實現。妳要將兩個參數傳遞給此函數:服務名和指向 ControlHandlerfunction 的指針。

它指示控制分派器調用 ControlHandler 函數處理 SCM 控制請求。註冊完控制處理器之後,獲得狀態句柄(hStatus)。通過調用 SetServiceStatus 函數,用 hStatus 向 SCM 報告服務的狀態。

Listing 1 展示了如何指定服務特征和其當前狀態來初始化 ServiceStatus 結構,ServiceStatus 結構的每個域都有其用途:

dwServiceType:指示服務類型,創建 Win32 服務。賦值 SERVICE_WIN32;

dwCurrentState:指定服務的當前狀態。因為服務的初始化在這裏沒有完成,所以這裏的狀態為 SERVICE_START_PENDING;

dwControlsAccepted:這個域通知 SCM 服務接受哪個域。本文例子是允許 STOP 和 SHUTDOWN 請求。處理控制請求將在第三步討論;

dwWin32ExitCode 和 dwServiceSpecificExitCode:這兩個域在妳終止服務並報告退出細節時很有用。初始化服務時並不退出,因此,它們的值為 0;

dwCheckPoint 和 dwWaitHint:這兩個域表示初始化某個服務進程時要30秒以上。本文例子服務的初始化過程很短,所以這兩個域的值都為 0。

調用 SetServiceStatus 函數向 SCM 報告服務的狀態時。要提供 hStatus 句柄和 ServiceStatus 結構。註意 ServiceStatus 壹個全局變量,所以妳可以跨多個函數使用它。ServiceMain 函數中,妳給結構的幾個域賦值,它們在服務運行的整個過程中都保持不變,比如:dwServiceType。

在報告了服務狀態之後,妳可以調用 InitService 函數來完成初始化。這個函數只是添加壹個說明性字符串到日誌文件。如下面代碼所示:

// 服務初始化

int InitService()

{

int result;

result = WriteToLog("Monitoring started.");

return(result);

}

在 ServiceMain 中,檢查 InitService 函數的返回值。如果初始化有錯(因為有可能寫日誌文件失敗),則將服務狀態置為終止並退出 ServiceMain:

error = InitService();

if (error)

{

// 初始化失敗,終止服務

ServiceStatus.dwCurrentState = SERVICE_STOPPED;

ServiceStatus.dwWin32ExitCode = -1;

SetServiceStatus(hStatus, &ServiceStatus);

// 退出 ServiceMain

return;

}

如果初始化成功,則向 SCM 報告狀態:

// 向 SCM 報告運行狀態

ServiceStatus.dwCurrentState = SERVICE_RUNNING;

SetServiceStatus (hStatus, &ServiceStatus);

接著,啟動工作循環。每五秒鐘查詢壹個可用物理內存並將結果寫入日誌文件。

如 Listing 1 所示,循環壹直到服務的狀態為 SERVICE_RUNNING 或日誌文件寫入出錯為止。狀態可能在 ControlHandler 函數響應 SCM 控制請求時修改。

第三步:處理控制請求

在第二步中,妳用 ServiceMain 函數註冊了控制處理器函數。控制處理器與處理各種 Windows 消息的窗口回調函數非常類似。它檢查 SCM 發送了什麽請求並采取相應行動。

每次妳調用 SetServiceStatus 函數的時候,必須指定服務接收 STOP 和 SHUTDOWN 請求。Listing 2 示範了如何在 ControlHandler 函數中處理它們。

STOP 請求是 SCM 終止服務的時候發送的。例如,如果用戶在“服務”控制面板中手動終止服務。SHUTDOWN 請求是關閉機器時,由 SCM 發送給所有運行中服務的請求。兩種情況的處理方式相同:

寫日誌文件,監視停止;

向 SCM 報告 SERVICE_STOPPED 狀態;

由於 ServiceStatus 結構對於整個程序而言為全局量,ServiceStatus 中的工作循環在當前狀態改變或服務終止後停止。其它的控制請求如:PAUSE 和 CONTINUE 在本文的例子沒有處理。

控制處理器函數必須報告服務狀態,即便 SCM 每次發送控制請求的時候狀態保持相同。因此,不管響應什麽請求,都要調用 SetServiceStatus。

第四步:安裝和配置服務

程序編好了,將之編譯成 exe 文件。本文例子創建的文件叫 MemoryStatus.exe,將它拷貝到 C:\MyServices 文件夾。為了在機器上安裝這個服務,需要用 SC.EXE 可執行文件,它是 Win32 Platform SDK 中附帶的壹個工具。(譯者註:Visaul Studio .NET 2003 IDE 環境中也有這個工具,具體存放位置在:C:\Program Files\Microsoft Visual Studio .NET 2003\Common7\Tools\Bin\winnt)。使用這個實用工具可以安裝和移除服務。其它控制操作將通過服務控制面板來完成。以下是用命令行安裝 MemoryStatus 服務的方法:

sc create MemoryStatus binpath= c:\MyServices\MemoryStatus.exe

發出此創建命令。指定服務名和二進制文件的路徑(註意 binpath= 和路徑之間的那個空格)。安裝成功後,便可以用服務控制面板來控制這個服務。用控制面板的工具欄啟動和終止這個服務。

 MemoryStatus 的啟動類型是手動,也就是說根據需要來啟動這個服務。右鍵單擊該服務,然後選擇上下文菜單中的“屬性”菜單項,此時顯示該服務的屬性窗口。在這裏可以修改啟動類型以及其它設置。妳還可以從“常規”標簽中啟動/停止服務。以下是從系統中移除服務的方法:

sc delete MemoryStatus

指定 “delete” 選項和服務名。此服務將被標記為刪除,下次西通重啟後,該服務將被完全移除。

第五步:測試服務

從服務控制面板啟動 MemoryStatus 服務。如果初始化不出錯,表示啟動成功。過壹會兒將服務停止。檢查壹下 C:\MyServices 文件夾中 memstatus.txt 文件的服務輸出。在我的機器上輸出是這樣的:

Monitoring started.

273469440

273379328

273133568

273084416

Monitoring stopped.

為了測試 MemoryStatus 服務在出錯情況下的行為,可以將 memstatus.txt 文件設置成只讀。這樣壹來,服務應該無法啟動。

去掉只讀屬性,啟動服務,在將文件設成只讀。服務將停止執行,因為此時日誌文件寫入失敗。如果妳更新服務控制面板的內容,會發現服務狀態是已經停止。

  • 上一篇:為什麽很多人都認為數學專業比計算機專業更有前途?
  • 下一篇:怎麽樣查看本地已經開啟的端口?出來的內容是什麽意思?
  • copyright 2024編程學習大全網