WinAPI 编写基本服务程序

 

1. #include <Windows.h> 
2. #include <stdio.h> 
3. #include <WinBase.h> 
4.  
5. #define SLEEP_TIME 5000 
6. #define LOGFILE "D:\\memorylog.txt" 
7.  
8. SERVICE_STATUS ServiceStatus; 
9. SERVICE_STATUS_HANDLE hStatus; 
10.  
11. VOID WINAPI Handler(DWORD fdwControl); 
12. VOID WINAPI ServiceMain(DWORD dwArgc,LPTSTR lpszArgv); 
13. INT InitService(VOID); 
14. INT WriteToLog(char* str); 
15.  
16. VOID main(VOID){ 
17.     SERVICE_TABLE_ENTRY ServiceTable[]={{"Pnig0srv",(LPSERVICE_MAIN_FUNCTION)ServiceMain},{NULL,NULL}}; 
18.     if(!StartServiceCtrlDispatcher(ServiceTable)){ 
19.         printf("Start Service failed.(%d)\n",GetLastError()); 
20.     } 
21. } 
22.  
23. VOID WINAPI ServiceMain(DWORD dwArgc,LPTSTR lpszArgv){ 
24.     int error; 
25.     ServiceStatus.dwServiceType = SERVICE_WIN32; 
26.     ServiceStatus.dwCurrentState = SERVICE_START_PENDING; 
27.     ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP|SERVICE_ACCEPT_SHUTDOWN; 
28.     ServiceStatus.dwWin32ExitCode = 0; 
29.     ServiceStatus.dwServiceSpecificExitCode = 0; 
30.     ServiceStatus.dwCheckPoint = 0; 
31.     ServiceStatus.dwWaitHint = 0; 
32.  
33.     hStatus = RegisterServiceCtrlHandler("Pnig0srv",(LPHANDLER_FUNCTION)Handler); 
34.  
35.     if(hStatus == (SERVICE_STATUS_HANDLE)0){ 
36.         return; 
37.     } 
38.     error = InitService(); 
39.     if(error){ 
40.         ServiceStatus.dwCurrentState = SERVICE_STOP; 
41.         ServiceStatus.dwWin32ExitCode = -1; 
42.         SetServiceStatus(hStatus,&ServiceStatus); 
43.         return; 
44.     } 
45.     ServiceStatus.dwCurrentState = SERVICE_RUNNING; 
46.     SetServiceStatus(hStatus,&ServiceStatus); 
47.  
48.     MEMORYSTATUS memory; 
49.     while(ServiceStatus.dwCurrentState == SERVICE_RUNNING){ 
50.         char buffer[16]; 
51.         GlobalMemoryStatus(&memory); 
52.         sprintf(buffer, "%d", memory.dwAvailPhys); 
53.         int result = WriteToLog(buffer); 
54.         if (result) 
55.         { 
56.             ServiceStatus.dwCurrentState =  
57.                 SERVICE_STOPPED;  
58.             ServiceStatus.dwWin32ExitCode      = -1;  
59.             SetServiceStatus(hStatus,  
60.                 &ServiceStatus); 
61.             return; 
62.         } 
63.         Sleep(SLEEP_TIME); 
64.     } 
65.     return; 
66. } 
67.  
68. VOID WINAPI Handler(DWORD fdwControl){ 
69.     switch(fdwControl)  
70.     {  
71.     case SERVICE_CONTROL_STOP:  
72.         WriteToLog("Monitoring stopped."); 
73.  
74.         ServiceStatus.dwWin32ExitCode = 0;  
75.         ServiceStatus.dwCurrentState = SERVICE_STOPPED;  
76.         SetServiceStatus (hStatus, &ServiceStatus); 
77.         return;  
78.  
79.     case SERVICE_CONTROL_SHUTDOWN:  
80.         WriteToLog("Monitoring stopped."); 
81.  
82.         ServiceStatus.dwWin32ExitCode = 0;  
83.         ServiceStatus.dwCurrentState = SERVICE_STOPPED;  
84.         SetServiceStatus (hStatus, &ServiceStatus); 
85.         return;  
86.  
87.     default: 
88.         break; 
89.     }  
90.  
91.     SetServiceStatus (hStatus, &ServiceStatus); 
92.  
93.     return;  
94. } 
95.  
96. INT WriteToLog(char* str) 
97. { 
98.     FILE* log; 
99.     log = fopen(LOGFILE, "a+"); 
100.     if (log == NULL) 
101.         return -1; 
102.     fprintf(log, "%s\n", str); 
103.     fclose(log); 
104.     return 0; 
105. } 
106.  
107. INT InitService(){ 
108.     int result; 
109.     result = WriteToLog("Monitoring started."); 
110.     return(result);  
111. } 

 

本文转hackfreer51CTO博客，原文链接：http://blog.51cto.com/pnig0s1992/674402，如需转载请自行联系原作者