如何寫一個windows服務？參考win sys program 13章，補充一些書中遺漏的注意點

1. 建立windows console應用程式。vs內建的windows service模板建立出來的項目看不懂。

2. _tmain函數這樣寫：

Code: Select all

/*  Main routine that starts the service control dispatcher */
VOID _tmain (int argc, LPTSTR argv[])
{
   SERVICE_TABLE_ENTRY DispatchTable[] =
   {
      { ServiceName,            ServiceMain   },
      { NULL,                  NULL }
   };

   StartServiceCtrlDispatcher (DispatchTable);
   return;
}

關鍵就是ServiceMain函數在這裡定義了。

3. 然後就是ServiceMain函數：

Code: Select all

/*   ServiceMain entry point, called when the service is created by
   the main program.  */
void WINAPI ServiceMain (DWORD argc, LPTSTR argv[])
{
   HANDLE hFile = NULL;
   DWORD Context = 1;
   size_t bytes_need_write = 0;
   DWORD bytes_written = 0;
   TCHAR write_buffer[255];
   SYSTEMTIME cur_time;
   DWORD n;
   LARGE_INTEGER file_size;

   // init logger
   logger_set_root_directory(TEXT("C:\\"));

   hServStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
   hServStatus.dwCurrentState = SERVICE_START_PENDING;
   hServStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
   hServStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
   hServStatus.dwServiceSpecificExitCode = 0;
   hServStatus.dwCheckPoint = 0;
   // in 2000ms, we should increment dwCheckPoint or set new status
   // while current status is a PENDING start/continue/stop state
   hServStatus.dwWaitHint = 5000;

   /* Warning. Older VC++ version do not have RegisterServiceCtrlHandlerEx
    * defined. You can use RegisterServiceCtrlHandler just as well */
#ifdef RegisterServiceCtrlHandlerEx
   hSStat = RegisterServiceCtrlHandlerEx(ServiceName, ServerCtrlHandlerEx, &Context);
#else
   hSStat = RegisterServiceCtrlHandler(ServiceName, ServerCtrlHandler);
#endif

   if (hSStat == NULL) {
      GET_ERROR_CODE(n);
      UTILS_RIF_WITH_LOG(FALSE, LOG_LEVEL_ERROR, __SDFILE__, __LINE__,
         TEXT("Register service ctrl handle failed. Reason: %s\n"),
         utils_format_error_string(n));
   }

   // PENDING start
   if (!SetServiceStatus (hSStat, &hServStatus)) {
      GET_ERROR_CODE(n);
      UTILS_RIF_WITH_LOG(FALSE, LOG_LEVEL_ERROR, __SDFILE__, __LINE__,
         TEXT("Set service status failed. Reason: %s\n"),
         utils_format_error_string(n));
   }

   // Open the file and log current time
   hFile = CreateFile(TEXT("C:\\testserv.txt"), GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
   if (hFile == INVALID_HANDLE_VALUE) {
      GET_ERROR_CODE(n);
      UTILS_RIF_WITH_LOG(FALSE, LOG_LEVEL_ERROR, __SDFILE__, __LINE__,
         TEXT("Create testserv file failed. Reason: %s\n"),
         utils_format_error_string(n));
   }

   // Seek to the end of file
   UTILS_RETURN_IF_FAIL(GetFileSizeEx(hFile, &file_size));
   UTILS_RETURN_IF_FAIL(SetFilePointerEx(hFile, file_size, NULL, FILE_BEGIN));

   GetLocalTime(&cur_time);
   StringCchPrintf(write_buffer, _countof(write_buffer), TEXT("TestServ started, write time: %d-%d-%d %d:%d:%d\n"),
      cur_time.wYear, cur_time.wMonth, cur_time.wDay, cur_time.wHour, cur_time.wMinute, cur_time.wSecond);
   StringCchLength(write_buffer, _countof(write_buffer), &bytes_need_write);
   WriteFile(hFile, write_buffer, bytes_need_write * sizeof(TCHAR), &bytes_written, NULL);

   hServStatus.dwCheckPoint = 0;
   hServStatus.dwWin32ExitCode = NO_ERROR;
   hServStatus.dwServiceSpecificExitCode = 0;
   hServStatus.dwCurrentState = SERVICE_RUNNING;
   if (!SetServiceStatus(hSStat, &hServStatus)) {
      GET_ERROR_CODE(n);
      UTILS_RIF_WITH_LOG(FALSE, LOG_LEVEL_ERROR, __SDFILE__, __LINE__,
         TEXT("Set service status failed. Reason: %s\n"),
         utils_format_error_string(n));
   }

   // block here until service stop
   while (!service_terminate) {
      Sleep(1000);
   }

   // stop service
   GetLocalTime(&cur_time);
   StringCchPrintf(write_buffer, _countof(write_buffer), TEXT("TestServ stopped, write time: %d-%d-%d %d:%d:%d\n"),
      cur_time.wYear, cur_time.wMonth, cur_time.wDay, cur_time.wHour, cur_time.wMinute, cur_time.wSecond);
   StringCchLength(write_buffer, _countof(write_buffer), &bytes_need_write);
   WriteFile(hFile, write_buffer, bytes_need_write * sizeof(TCHAR), &bytes_written, NULL);
   CloseHandle(hFile);

   hServStatus.dwCheckPoint = 0;
   hServStatus.dwWin32ExitCode = NO_ERROR;
   hServStatus.dwServiceSpecificExitCode = 0;
   hServStatus.dwCurrentState = SERVICE_STOPPED;
   if (!SetServiceStatus(hSStat, &hServStatus)) {
      GET_ERROR_CODE(n);
      UTILS_RIF_WITH_LOG(FALSE, LOG_LEVEL_ERROR, __SDFILE__, __LINE__,
         TEXT("Set service status failed. Reason: %s\n"),
         utils_format_error_string(n));
   }
   
   return;
}

有幾個關鍵點：

A. hServStatus是一個SERVICE_STRUCTURE，這個structure非常重要，定義了很多關鍵的資料。主要是：
(1) dwControlAccepted，不支援PAUSE和CONTINUE就不要寫上。

(2) dwWin32ExitCode這個member容易混淆。如果當前service的狀態是PENDING（start pending/pause pending/continue pending/stop pending），那麼可以設定為ERROR_SERVICE_SPECIFIC_ERROR， 這表示pending的時候如果出錯了，讓windows SCM去check dwServiceSpecificExitCode member來得到出錯碼。但是如果狀態已經成功轉換到了RUNNING/STOPPED，那麼，要設定這個dwWin32ExitCode為 NO_ERROR，此時windows SCM認為service狀態轉換沒有錯誤發生，dwServiceSpecificExitCode的值被ignore。如果 dwWin32ExitCode還為ERROR_SERVICE_SPECIFIC_ERROR的話，windows SCM就會認為service狀態轉換出錯，然後把dwServiceSpecificExitCode作為出錯碼取出來報告。這就是為什麼上面代碼 中，RUNNING和STOPPED狀態轉換完成的時候，要設定這兩個member的原因。

(3) dwCheckPoint，也是用於pending狀態的。pending狀態的時候，windows SCM要知道service當前是否alive，依靠的就是周期性的check這個dwCheckPoint member。所以，我們的service在pending狀態的時候，要經常使用SetServiceStatus來更新這個 dwCheckPoint，否則windows SCM會認為service down，從而強制stop service。此外，當service完成pending狀態切換到新狀態後，記得重設dwCheckPoint為0.

(4) dwWaitHint，這個member定義一個時間（單位毫秒），在service處於pending狀態時，如果過了這裡定義的時間，windows SCM發現service的dwCheckPoint沒有更新，或者service沒有切換到新的非pending狀態，那麼，windows SCM就會強制stop service。

B. RegisterServiceCtrlHandlerEx函數。用來註冊SCM動作的響應函數。也就是stop/continue這些命令的響應。例子：

Code: Select all

/*   Control Handler Function */
#ifdef RegisterServiceCtrlHandlerEx
DWORD WINAPI ServerCtrlHandlerEx( DWORD dwControl, DWORD dwEventType,
                       LPVOID lpEventData, LPVOID lpContext)
#else
DWORD WINAPI ServerCtrlHandler( DWORD dwControl)
#endif
// requested control code
{
   DWORD n;

   switch (dwControl) {
   case SERVICE_CONTROL_SHUTDOWN:
   case SERVICE_CONTROL_STOP:
      service_terminate = TRUE;
      hServStatus.dwCheckPoint++;
      hServStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
      hServStatus.dwServiceSpecificExitCode = 0;
      hServStatus.dwCurrentState = SERVICE_STOP_PENDING;
      if (!SetServiceStatus(hSStat, &hServStatus)) {
         GET_ERROR_CODE(n);
         UTILS_RVIF_WITH_LOG(FALSE, NO_ERROR, LOG_LEVEL_ERROR, __SDFILE__, __LINE__,
            TEXT("Set service status failed. Reason: %s\n"),
            utils_format_error_string(n));
      }
      return NO_ERROR;
   case SERVICE_CONTROL_PAUSE:
      break;
   case SERVICE_CONTROL_CONTINUE:
      break;
   case SERVICE_CONTROL_INTERROGATE:
      return NO_ERROR;
   default:
      if (dwControl > 127 && dwControl < 256) /* User Defined */
      break;
   }
   return ERROR_CALL_NOT_IMPLEMENTED;
}

(1) 對於沒有handle的control命令，return ERROR_CALL_NOT_IMPLEMENTED，對於handle的，返回NO_ERROR。例外是對於 SERVICE_CONTROL_INTERROGATE，也要返回NO_ERROR，即使我們沒有handle這個control。更具體的資訊查看 MSDN。

(2) SERVICE_CONTROL_SHUTDOWN是系統關機的時候SCM發送給service的。和SERVICE_CONTROL_STOP一樣，可以在這裡設定一個pending狀態，在ServiceMain中真正stop了，設定STOPPED狀態。

4. service本身的關鍵代碼就是這些，下面是使用SCM如何建立一個service：

Code: Select all

#define SVCNAME TEXT("EricTestService")

int _tmain(int argc, PTSTR argv[], PTSTR env[])
{
   DWORD errcode = 0;
   SC_HANDLE hManager = NULL;
   SC_HANDLE hService = NULL;

   setlocale(LC_ALL, "CHS");

   // Open SCManager
   hManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
   if (NULL == hManager) {
      errcode = GetLastError();
      _tprintf(TEXT("Open service manager failed. Reason: %s\n"), utils_format_error_string(errcode));
      goto failed;
   }

   // create service
   hService = CreateService(
        hManager,                  // SCM database
        SVCNAME,                   // name of service
        SVCNAME,                   // service name to display
        SERVICE_ALL_ACCESS,        // desired access
        SERVICE_WIN32_OWN_PROCESS, // service type
        SERVICE_AUTO_START,      // start type
        SERVICE_ERROR_NORMAL,      // error control type
      TEXT("C:\\ScheduleDownload\\Practise\\TestServ\\Debug\\TestServ.exe"),  // path to service's binary
        NULL,                      // no load ordering group
        NULL,                      // no tag identifier
        NULL,                      // no dependencies
        NULL,                      // LocalSystem account
        NULL);                     // no password

   if (NULL == hService) {
      errcode = GetLastError();
      _tprintf(TEXT("Create service failed. Reason: %s\n"), utils_format_error_string(errcode));
      CloseServiceHandle(hManager);
      goto failed;
   }

   CloseServiceHandle(hService);
   CloseServiceHandle(hManager);

   return 0;

failed:
   // wait key
   _getch();
   return 1;
}

這 就比較簡單了，使用OpenSCManager開啟SCM，使用CreateService建立一個service，此時在windows的服務中就能看 到了。start type中，設定為SERVICE_AUTO_START就會在開機時啟動（不需要使用者登入就會啟動的，已經測試過了）。其他的函數比如 StartService/DeleteService/QueryService...看看MSDN就OK了。