用 C 语言编写 Windows 服务程序的五个步骤(代码)

  1#include <windows.h>
  2#include <stdio.h>
  3
  4#define SLEEP_TIME 5000
  5#define LOGFILE "C:\\MyServices\\memstatus.txt"
  6
  7/**/////////////////////////////////////////////////////////////
  8// Declare several global variables to share 
  9// their values across multiple functions of your program.
 10/**/////////////////////////////////////////////////////////////
 11SERVICE_STATUS          ServiceStatus; 
 12SERVICE_STATUS_HANDLE   hStatus; 
 13
 14/**/////////////////////////////////////////////////////////////
 15// Make the forward definitions of functions prototypes.
 16//
 17/**/////////////////////////////////////////////////////////////
 18void  ServiceMain(int argc, char** argv); 
 19void  ControlHandler(DWORD request); 
 20int InitService();
 21
 22int WriteToLog(char* str)
 23{
 24   FILE* log;
 25   log = fopen(LOGFILE, "a+");
 26   if (log == NULL){
 27       OutputDebugString("Log file open failed.");
 28      return -1;
 29   }
 30   fprintf(log, "%s\n", str);
 31   fclose(log);
 32   return 0;
 33}
 34
 35// Service initialization
 36int InitService() 
 37{ 
 38    OutputDebugString("Monitoring started.");
 39    int result;
 40    result = WriteToLog("Monitoring started.");
 41    return(result); 
 42}
 43
 44// Control Handler
 45void ControlHandler(DWORD request) 
 46{ 
 47   switch(request) 
 48   { 
 49      case SERVICE_CONTROL_STOP: 
 50         OutputDebugString("Monitoring stopped.");
 51         WriteToLog("Monitoring stopped.");
 52
 53         ServiceStatus.dwWin32ExitCode = 0; 
 54         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
 55         SetServiceStatus (hStatus, &ServiceStatus);
 56         return; 
 57 
 58      case SERVICE_CONTROL_SHUTDOWN: 
 59         OutputDebugString("Monitoring stopped.");
 60         WriteToLog("Monitoring stopped.");
 61
 62         ServiceStatus.dwWin32ExitCode = 0; 
 63         ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
 64         SetServiceStatus (hStatus, &ServiceStatus);
 65         return; 
 66        
 67      default:
 68         break;
 69    } 
 70 
 71    // Report current status
 72    SetServiceStatus (hStatus, &ServiceStatus);
 73 
 74    return; 
 75}
 76
 77void ServiceMain(int argc, char** argv) 
 78{ 
 79   int error; 
 80 
 81   ServiceStatus.dwServiceType = 
 82      SERVICE_WIN32; 
 83   ServiceStatus.dwCurrentState = 
 84      SERVICE_START_PENDING; 
 85   ServiceStatus.dwControlsAccepted   =  
 86      SERVICE_ACCEPT_STOP | 
 87      SERVICE_ACCEPT_SHUTDOWN;
 88   ServiceStatus.dwWin32ExitCode = 0; 
 89   ServiceStatus.dwServiceSpecificExitCode = 0; 
 90   ServiceStatus.dwCheckPoint = 0; 
 91   ServiceStatus.dwWaitHint = 0; 
 92 
 93   hStatus = RegisterServiceCtrlHandler(
 94      "MemoryStatus", 
 95      (LPHANDLER_FUNCTION)ControlHandler); 
 96   if (hStatus == (SERVICE_STATUS_HANDLE)0) 
 97   { 
 98      // Registering Control Handler failed
 99      return; 
   }  

   // Initialize Service 
   error = InitService(); 
   if (error) 
   {
      // Initialization failed
      ServiceStatus.dwCurrentState = 
         SERVICE_STOPPED; 
      ServiceStatus.dwWin32ExitCode = -1; 
      SetServiceStatus(hStatus, &ServiceStatus); 
      return; 
   } 
   // We report the running status to SCM. 
   ServiceStatus.dwCurrentState = 
      SERVICE_RUNNING; 
   SetServiceStatus (hStatus, &ServiceStatus);
 
   MEMORYSTATUS memory;
   // The worker loop of a service
   while (ServiceStatus.dwCurrentState == 
          SERVICE_RUNNING)
   {
      char buffer[16];
      GlobalMemoryStatus(&memory);
      sprintf(buffer, "%d", memory.dwAvailPhys);

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

void main(int argc, char* argv[])
{ 
   SERVICE_TABLE_ENTRY ServiceTable[2];
   ServiceTable[0].lpServiceName = "MemoryStatus";
   ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain;

   ServiceTable[1].lpServiceName = NULL;
   ServiceTable[1].lpServiceProc = NULL;
   // Start the control dispatcher thread for our service
   StartServiceCtrlDispatcher(ServiceTable);
}