用C語言編寫Windows服務程式的五個步驟

標籤：函數定義   輸出   成功   結果   box   its   fopen   一定要注意了   nts   

   Windows 服務被設計用於需要在後台啟動並執行應用程式以及實現沒有使用者互動的任務。為了學習這種控制台應用程式的基礎知識，C（不是C++）是最佳選擇。本文將建立並實現一個簡單的服務程式，其功能是查詢系統中可用實體記憶體數量，然後將結果寫入一個文字檔。最後，你可以用所學知識編寫自己的Windows 服務。

當初他寫第一個NT 服務時，他到 MSDN 上找例子。在那裡他找到了一篇 Nigel Thompson 寫的文章：“Creating a Simple Win32 Service in C++”，這篇文章附帶一個 C++ 例子。雖然這篇文章很好地解釋了服務的開發過程，但是，他仍然感覺缺少他需要的重要訊息。他想理解通過什麼架構，調用什麼函數，以及何時調用，但 C++ 在這方面沒有讓他輕鬆多少。物件導向的方法固然方便，但由於用類對底層 Win32 函數調用進行了封裝，它不利於學習服務程式的基本知識。這就是為什麼他覺得 C 更加適合於編寫初級服務程式或者實現簡單背景工作的服務。在你對服務程式有了充分透徹的理解之後，用 C++ 編寫才能遊刃有餘。當他離開原來的工作崗位，不得不向另一個人轉移他的知識的時候，利用他用 C 所寫的例子就非常容易解釋 NT 服務之所以然。

服務是一個運行在後台並實現勿需使用者互動的任務的控制台程式。Windows NT/2000/XP 作業系統提供為服務程式提供專門的支援。人們可以用服務控制台來配置安裝好的服務程式，也就是 Windows 2000/XP 控制台|管理工具中的“服務”（或在“開始”|“運行”對話方塊中輸入 services.msc /s——譯者注）。可以將服務配置成作業系統啟動時自動啟動，這樣你就不必每次再重啟系統後還要手動啟動服務。

本文將首先解釋如何建立一個定期查詢可用實體記憶體並將結果寫入某個文字檔的服務。然後指導你完成組建，安裝和實現服務的整個過程。

第一步：主函數和全域定義
　　
首先，包含所需的標頭檔。例子要調用 Win32 函數（windows.h）和磁碟檔案寫入（stdio.h）：

　　#include <windows.h>
#include <stdio.h>
　　
接著，定義兩個常量：

　　#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 函數

void ServiceMain(int argc, char** argv) 
{
BOOL bRet;

bRet = TRUE;
ServiceStatus.dwServiceType     =SERVICE_WIN32; 
ServiceStatus.dwCurrentState     =SERVICE_START_PENDING; 
ServiceStatus.dwControlsAccepted   =SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
ServiceStatus.dwWin32ExitCode    = 0; 
ServiceStatus.dwServiceSpecificExitCode = 0; 
ServiceStatus.dwCheckPoint     = 0;
ServiceStatus.dwWaitHint     = 0; 
hStatus = RegisterServiceCtrlHandler("SERVICENAME", (LPHANDLER_FUNCTION)ControlHandler); 
if (hStatus == (SERVICE_STATUS_HANDLE)0) 
{
   //登陸失敗
   return;
}

// service狀態情報更新
ServiceStatus.dwCurrentState = SERVICE_RUNNING; 
SetServiceStatus (hStatus, &ServiceStatus);

while (ServiceStatus.dwCurrentState == SERVICE_RUNNING)
   {
      int result = startFunc();
      if (result)
      {
         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
         ServiceStatus.dwWin32ExitCode = -1; 
         SetServiceStatus(hStatus, &ServiceStatus);
         return;
      }
   }

return;
}

第三步：建立自己的startFunc()函數----願意寫點什麼就寫點什麼吧。：）

第四步：安裝和佈建服務
　　
程式編好了，將之編譯成 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” 選項和服務名。此服務將被標記為刪除，下次西通重啟後，該服務將被完全移除

注意：service 服務是XP系統運行在C:\Windows\System32\下的。

      2000系統是運行在C\Winnt\system\下的。

所以程式中有關路徑的地方一定要注意了。

除了這兩個系統，其他系統還沒有測試過，不過可以自己測試一下，產生的目錄在C:\memstatus.txt中，代碼如下：

 

#include <windows.h>
#include <stdio.h>

#define SLEEP_TIME 3000
#define LOGFILE "C:\\memstatus.txt"

int WriteToLog(char* );

SERVICE_STATUS ServiceStatus; 
SERVICE_STATUS_HANDLE hStatus;

void ServiceMain(int argc, char** argv); 
void ControlHandler(DWORD request); 
//int InitService();

int 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); 
return 0;
}

void ServiceMain(int argc, char** argv) 
{ 
//   int error;

   ServiceStatus.dwServiceType =    SERVICE_WIN32; 
   ServiceStatus.dwCurrentState =    SERVICE_START_PENDING; 
   ServiceStatus.dwControlsAccepted   =     SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
   ServiceStatus.dwWin32ExitCode = 0; 
   ServiceStatus.dwServiceSpecificExitCode = 0; 
   ServiceStatus.dwCheckPoint = 0; 
   ServiceStatus.dwWaitHint = 0;

   hStatus = RegisterServiceCtrlHandler(
      "MemoryStatus", 
      (LPHANDLER_FUNCTION)ControlHandler); 
   if (hStatus == (SERVICE_STATUS_HANDLE)0) 
   { 
      return; 
   }

   ServiceStatus.dwCurrentState = SERVICE_RUNNING; 
   SetServiceStatus (hStatus, &ServiceStatus);

   char memory[256];

   while (ServiceStatus.dwCurrentState == SERVICE_RUNNING)
   {

   GetCurrentDirectory(256,memory);

      int result = WriteToLog(memory);
      if (result)
      {
         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
         ServiceStatus.dwWin32ExitCode = -1; 
         SetServiceStatus(hStatus, &ServiceStatus);
         return;
      }
      Sleep(SLEEP_TIME);
   }
   return; 
}

void ControlHandler(DWORD request) 
{ 
   switch(request) 
   { 
      case SERVICE_CONTROL_STOP: 
         WriteToLog("Monitoring stopped.");

         ServiceStatus.dwWin32ExitCode = 0; 
         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
         SetServiceStatus (hStatus, &ServiceStatus);
         return;

      case SERVICE_CONTROL_SHUTDOWN: 
         WriteToLog("Monitoring stopped.");

         ServiceStatus.dwWin32ExitCode = 0; 
         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
         SetServiceStatus (hStatus, &ServiceStatus);
         return; 
        
      default:
         break;
    }

    SetServiceStatus (hStatus, &ServiceStatus);

    return; 
}

int WriteToLog(char* str)
{
    FILE* log;
    log = fopen(LOGFILE, "a+");
    if (log == NULL)
    return -1;
    fprintf(log, "%s\n", str);
    fclose(log);
    return 0;
}

////////////////下面看他的代碼吧////////////////

//main.cpp

 

#include <windows.h>
void WINAPI ServiceMain(DWORD , char** ); 
void ControlHandler(DWORD ); 
int IntelligentStart();
void txtinput();
void read();
SERVICE_STATUS ServiceStatus; 
SERVICE_STATUS_HANDLE hStatus;
int WINAPI WinMain( HINSTANCE hInstance, HINSTANCE hPreInst, LPSTR pchCmdLine, int iCmdShow )
{
    SERVICE_TABLE_ENTRY ServiceTable[2];
    ServiceTable[0].lpServiceName = "IntelligentStart";
    ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain;
    
    ServiceTable[1].lpServiceName = NULL;
    ServiceTable[1].lpServiceProc = NULL;

    StartServiceCtrlDispatcher(ServiceTable); 
return 0;
   
}

void WINAPI ServiceMain(DWORD ac, char **av) 
{

   ServiceStatus.dwServiceType      = SERVICE_WIN32; 
   ServiceStatus.dwCurrentState     = SERVICE_START_PENDING; 
   ServiceStatus.dwControlsAccepted    =   SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
   ServiceStatus.dwWin32ExitCode     = 0; 
   ServiceStatus.dwServiceSpecificExitCode = 0; 
   ServiceStatus.dwCheckPoint      = 0; 
   ServiceStatus.dwWaitHint      = 0;

   hStatus = RegisterServiceCtrlHandler("IntelligentStart", (LPHANDLER_FUNCTION)ControlHandler); 
   if (hStatus == (SERVICE_STATUS_HANDLE)0) 
   { 
      // Registering Control Handler failed
      return; 
   }

   //Report the running status to SCM. 
   ServiceStatus.dwCurrentState = SERVICE_RUNNING; 
   SetServiceStatus (hStatus, &ServiceStatus);

   // The worker loop of a service
   while (ServiceStatus.dwCurrentState == SERVICE_RUNNING)
   {
      int result = IntelligentStart();
      if (result)
      {
         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
         ServiceStatus.dwWin32ExitCode = -1; 
         SetServiceStatus(hStatus, &ServiceStatus);
         return;
      }
   }
   return; 
}

void ControlHandler(DWORD request) 
{ 
   switch(request) 
   { 
      case SERVICE_CONTROL_STOP: 
         ServiceStatus.dwWin32ExitCode = 0; 
         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
         SetServiceStatus (hStatus, &ServiceStatus);
         return;

      case SERVICE_CONTROL_SHUTDOWN: 
         ServiceStatus.dwWin32ExitCode = 0; 
         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
         SetServiceStatus (hStatus, &ServiceStatus);
         return; 
        
      default:
         break;
    }

    // Report current status
    SetServiceStatus (hStatus, &ServiceStatus);

    return; 
}

int IntelligentStart()
{

txtinput();
return 0;
}

write.cpp

#include <stdio.h>
#include <conio.h>
void txtinput()
{
FILE *fp;
char st[20] = "test new.\n";
fp=fopen("C:\\string.txt","at+");

fputs(st,fp);

fclose(fp);

return ;
}

 http://blog.csdn.net/w616589292/article/details/24265739

用C語言編寫Windows服務程式的五個步驟