后台服务程序开发模式

    转载自:http://blog.csdn.net/ashao1981/article/details/22359

 

    一直感觉VC++太复杂了,但昨天看了汪蒲阳编著的因特网应用编程,其中写到后台服务程序的编写,论述的非常详细,而且逻辑清晰,看了之后感觉明白不少,故拿来与需要之人共享,并更正了原程序的一些错误,补充了一些材料。另外还有一种用C++编写后台服务程序的思路(不算.NET上服务程序开发模型),以后整理好了再发上来。

    在2000/XP等基于NT 的操作系统中,有一个服务管理器,它管理的后台进程被称为 service。

    服务是一种应用程序类型,它在后台运行,与 UNIX 后台应用程序类似。服务应用程序通常可以在本地和通过网络为用户提供一些功能,例如客户端/服务器应用程序、Web服务器、数据库服务器以及其他基于服务器的应用程序。

    后台服务程序是在后台悄悄运行的。我们通过将自己的程序登记为服务,可以使自己的程序不出现在任务管理器中,并且随系统启动而最先运行,随系统关闭而最后停止。

    服务控制管理器是一个RPC 服务器,它显露了一组应用编程接口,程序员可以方便的编写程序来配置服务和控制远程服务器中服务程序。

    服务程序通常编写成控制台类型的应用程序,总的来说,一个遵守服务控制管理程序接口要求的程序包含下面三个函数:

1、服务程序主函数(main):调用系统函数 StartServiceCtrlDispatcher 连接程序主线程到服务控制管理程序。

2、服务入口点函数(ServiceMain):执行服务初始化任务,同时执行多个服务的服务进程有多个服务入口函数。

3、控制服务处理程序函数(Handler):在服务程序收到控制请求时由控制分发线程引用。(此处是Service_Ctrl)。

    另外在系统运行此服务之前需要安装登记服务程序:installService 函数。删除服务程序则需要先删除服务安装登记:removeService 函数。 

    服务类型:

类型

说明

SERVICE_FILE_SYSTEM_DRIVER=2

文件系统驱动服务。

SERVICE_KERNEL_DRIVER=1

驱动服务。

SERVICE_WIN32_OWN_PROCESS=16

独占一个进程的服务。

SERVICE_WIN32_SHARE_PROCESS=32

与其他服务共享一个进程的服务。

新建WIN32控制台程序, 其源文件名为service.cpp 。我用的开发工具是VC++.NET。

 1、服务程序主函数

    服务控制管理程序启动服务程序后,等待服务程序主函数调用系统函数StartServiceCtrlDispatcher。一个SERVICE_WIN32_OWN_PROCESS 类型的服务应该立即调用 StartServiceCtrlDispatcher 函数,可以在服务启动后让服务入口点函数完成初始化工作。对于 SERVICE_WIN32_OWN_PROCESS 类型的服务和程序中所有服务共同的初始化工作可以在主函数中完成,但不要超过30秒。否则必须建立另外的线程完成这些共同的初始化工作,从而保证服务程序主函数能及时地调用 StartServiceCtrlDispatcher 函数。

    主函数处理了三中命令行参数:- install,- remove,- debug,分别用于安装,删除和调试服务程序。如果不带参数运行,则认为是服务控制管理出现启动该服务程序。参数不正确则给出提示信息。

    StartServiceCtrlDispatcher 函数负责把程序主线程连接到服务控制管理程序。具体描述如下:


  1. BOOL StartServiceCtrlDispatcher(const LPSERVICE_TABLE_ENTRY lpServiceTable); 

    lpServiceStartTable 指向 SERVICE_TABLE_ENTRY 结构类型的数组,他包含了调用进程所提供的每个服务的入口函数和字符串名。表中的最后一个元素必须为 NULL,指明入口表结束。

    SERVICE_TABLE_ENTRY 结构具体描述如下:


  1. typedef struct _SERVICE_TABLE_ENTRY { 
  2.     LPTSTR lpServiceName;  
  3.     LPSERVICE_MAIN_FUNCTION lpServiceProc; 
  4. } SERVICE_TABLE_ENTRY, *LPSERVICE_TABLE_ENTRY; 

    lpServiceName 是一个以 NULL 结尾的字符串,标识服务名。如果是 SERVICE_WIN32_OWN_PROCESS 类型的服务,这个字符串会被忽略。

    lpServiceProc 指向服务入口点函数。


  1. //服务程序主函数。 
  2. #include "stdafx.h" 
  3. #include "Windows.h" 
  4. #define SZAPPNAME      "serverSample"     //服务程序名 
  5. #define SZSERVICENAME  "serviceSample"    //标识服务的内部名 
  6.   
  7. //内部变量 
  8. bool                   bDebugServer=false
  9. SERVICE_STATUS              ssStatus; 
  10. SERVICE_STATUS_HANDLE  sshStatusHandle; 
  11. DWORD                       dwErr=0; 
  12. TCHAR                       szErr[256]; 
  13.   
  14. //下面的函数由程序实现 
  15. void  WINAPI  Service_Main(DWORD dwArgc, LPTSTR *lpszArgv); 
  16. void  WINAPI  Service_Ctrl(DWORD dwCtrlCode); 
  17. void installService(); 
  18. void removeService(); 
  19. void debugService(int argc,char** argv); 
  20. bool ReportStatusToSCMgr(DWORD dwCurrentState,DWORD dwWin32ExitCode,DWORD dwWaitHint); 
  21. void AddToMessageLog(LPTSTR lpszMsg); 
  22.   
  23. int _tmain(int argc, _TCHAR* argv[]) 
  24. {    
  25.      SERVICE_TABLE_ENTRY dispatchTable[]= 
  26.      { 
  27.          {TEXT(SZSERVICENAME),(LPSERVICE_MAIN_FUNCTION)Service_Main}, 
  28.          { NULL,NULL} 
  29.      }; 
  30.      if((argc>1)&&((*argv[1]=='-')||(argv[1]=="/"))) 
  31.      { 
  32.          if(_stricmp("install",argv[1]+1)==0) 
  33.          { 
  34.               installService(); 
  35.          } 
  36.          else if(_stricmp("remove",argv[1]+1)==0) 
  37.          { 
  38.               removeService(); 
  39.          } 
  40.          else if(_stricmp("debug",argv[1]+1)==0) 
  41.          { 
  42.               bDebugServer=true
  43.               debugService(argc,argv); 
  44.          } 
  45.          else 
  46.          {        //如果未能和上面的如何参数匹配,则可能是服务控制管理程序来启动该程序。立即调用 
  47.                    //StartServiceCtrlDispatcher 函数。 
  48.               printf("%s - install to install the service /n",SZAPPNAME); 
  49.               printf("%s - remove to remove the service /n",SZAPPNAME); 
  50.               printf("%s - debug to debug the service /n",SZAPPNAME); 
  51.               printf("/n StartServiceCtrlDispatcher being called./n"); 
  52.               printf("This may take several seconds.Please wait./n"); 
  53.               if(!StartServiceCtrlDispatcher(dispatchTable)) 
  54.                    AddToMessageLog(TEXT("StartServiceCtrlDispatcher failed.")); 
  55.               else 
  56.                    AddToMessageLog(TEXT("StartServiceCtrlDispatcher OK.")); 
  57.          } 
  58.          exit(0); 
  59.      } 
  60.      return 0; 

 2、服务入口点函数

    服务入口点函数 service_main 首先调用系统函数 RegisterServiceCtrlHandler 注册服务控制处理函数 service_ctrl,然后调用 ReportStatusToSCMgr 函数,它通过系统函数 SetServiceStatus 更新服务的状态,然后调用特定的服务初始化入口函数 ServiceStart 完成具体的初始化工作。


  1. //服务入口点函数 
  2. void ServiceStart(DWORD dwArgc,LPTSTR* lpszArgv);//具体服务的初始化入口函数 
  3. void  WINAPI  Service_Main(DWORD dwArgc, LPTSTR *lpszArgv) 
  4.      //注册服务控制处理函数 
  5.      sshStatusHandle=RegisterServiceCtrlHandler(TEXT(SZSERVICENAME),Service_Ctrl); 
  6.      //如果注册失败 
  7.      if(!sshStatusHandle) 
  8.      { 
  9.          goto cleanup; 
  10.          return
  11.      } 
  12.      //初始化 SERVICE_STATUS 结构中的成员 
  13.      ssStatus.dwServiceType=SERVICE_WIN32_OWN_PROCESS; 
  14.      ssStatus.dwServiceSpecificExitCode=0; 
  15.      //更新服务状态 
  16.      if(!ReportStatusToSCMgr( 
  17.          SERVICE_START_PENDING,//服务状态,The service is starting. 
  18.          NO_ERROR,            //退出码         
  19.          3000))                   //等待时间 
  20.          goto cleanup;        //更新服务状态失败则转向 cleanup 
  21.      ServiceStart(dwArgc,lpszArgv); 
  22.      return
  23. cleanup: 
  24.      //把服务状态更新为 SERVICE_STOPPED,并退出。 
  25.      if(sshStatusHandle) 
  26.          (void)ReportStatusToSCMgr(SERVICE_STOPPED,dwErr,0); 

3、控制处理程序函数

    函数 Service_Ctrl 是服务的控制处理程序函数,由主函数线程的控制分发程序引用。在处理控制请求码时,应该在确定的时间间隔内更新服务状态检查点,避免发生服务不能响应的错误。


  1. //控制处理程序函数 
  2. void WINAPI Service_Ctrl(DWORD dwCtrlCode) 
  3.      //处理控制请求码 
  4.      switch(dwCtrlCode) 
  5.      { 
  6.          //先更新服务状态为 SERVICDE_STOP_PENDING,再停止服务。 
  7.      case SERVICE_CONTROL_STOP: 
  8.          ReportStatusToSCMgr(SERVICE_STOP_PENDING,NO_ERROR,500); 
  9.          ServiceStop();     //由具体的服务程序实现 
  10.          return
  11.          //暂停服务 
  12.      case SERVICE_CONTROL_PAUSE: 
  13.          ReportStatusToSCMgr(SERVICE_STOP_PENDING,NO_ERROR,500); 
  14.          ServicePause();    //由具体的服务程序实现 
  15.          ssStatus.dwCurrentState=SERVICE_PAUSED; 
  16.          return
  17.          //继续服务 
  18.      case SERVICE_CONTROL_CONTINUE: 
  19.          ReportStatusToSCMgr(SERVICE_STOP_PENDING,NO_ERROR,500); 
  20.          ServiceContinue(); //由具体的服务程序实现 
  21.          ssStatus.dwCurrentState=SERVICE_RUNNING; 
  22.          return
  23.          //更新服务状态 
  24.      case SERVICE_CONTROL_INTERROGATE: 
  25.          break
  26.          //无效控制码 
  27.      default
  28.          break
  29.      } 
  30.      ReportStatusToSCMgr(ssStatus.dwCurrentState,NO_ERROR,0); 

    除了系统定义的五种控制码外(还有一种是:SERVICE_CONTROL_SHUTDOWN),用户还可自定义控制码,其取值范围是128-255。用户可以通过控制面板中的服务项向特定服务程序的控制处理函数发送控制码,程序员可以调用系统函数 ControlService 直接向服务程序的控制处理函数发送控制码。其函数原型如下:


  1. BOOL ControlService(  
  2.     SC_HANDLE hService,  
  3.     DWORD dwControl,  
  4.     LPSERVICE_STATUS lpServiceStatus  
  5. );  
  6. //hService :函数 OpenService or CreateService 返回的服务程序句柄。  
  7. //dwControl :控制码,不能是SERVICE_CONTROL_SHUTDOWN。  
  8. //lpServiceStatus :返回最后收到的服务状态信息。  
 4、安装服务程序

    每个已安装服务程序在 HKEY_LOCAL_MACHINE/SYSTE/CurrentControlSet/Services 下都有一个服务名的关键字,程序员可以调用系统函数 CreateService 安装服务程序,并指定服务类型,服务名等。这个函数创建一个服务对象,并将其增加到相关的服务控制管理器数据库中。下面是函数原型:


  1. SC_HANDLE CreateService( 
  2.     SC_HANDLE hSCManager, //服务控制管理程序维护的登记数据库的句柄,由系统函数OpenSCManager 返回 
  3.     LPCTSTR lpServiceName, //以NULL 结尾的服务名,用于创建登记数据库中的关键字 
  4.     LPCTSTR lpDisplayName, //以NULL 结尾的服务名,用于用户界面标识服务 
  5.     DWORD dwDesiredAccess, //指定服务返回类型 
  6.     DWORD dwServiceType, //指定服务类型 
  7.     DWORD dwStartType, //指定何时启动服务 
  8.     DWORD dwErrorControl, //指定服务启动失败的严重程度 
  9.     LPCTSTR lpBinaryPathName, //指定服务程序二进制文件的路径 
  10.     LPCTSTR lpLoadOrderGroup, //指定顺序装入的服务组名 
  11.     LPDWORD lpdwTagId, //忽略,NULL 
  12.     LPCTSTR lpDependencies, //指定启动该服务前必须先启动的服务或服务组 
  13.     LPCTSTR lpServiceStartName, //以NULL 结尾的字符串,指定服务帐号。如是NULL,则表示使用LocalSystem 帐号 
  14.     LPCTSTR lpPassword //以NULL 结尾的字符串,指定对应的口令。为NULL表示无口令。但使用LocalSystem时填NULL 
  15. ); 

    对于一个已安装的服务程序,可以调用系统函数 OpenService 来获取服务程序的句柄。下面是其函数原型:


  1. SC_HANDLE OpenService(  
  2.     SC_HANDLE hSCManager,  
  3.     LPCTSTR lpServiceName,  
  4.     DWORD dwDesiredAccess  
  5. );  
  6. //hSCManager :服务控制管理程序微服的登记数据库的句柄。由函数 OpenSCManager function 返回 这个句柄。  
  7. //lpServiceName :将要打开的以NULL 结尾的服务程序的名字,和CreateService 中的lpServiceName 相对应。  
  8. //dwDesiredAccess :指定服务的访问类型。服务响应请求时,首先检查访问类型。  

用CreateService 或OpenService 打开的服务程序句柄使用完毕后必须用CloseServiceHandle 关闭。OpenSCManager打开的服务管理数据库句柄也必须用它来关闭。


  1. //安装服务程序 
  2. void installService() 
  3.      SC_HANDLE schService; 
  4.      SC_HANDLE schSCManager; 
  5.      TCHAR szPath[512]; 
  6.      //得到程序磁盘文件的路径 
  7.      if(GetModuleFileName(NULL,szPath,512)==0) 
  8.      { 
  9.          _tprintf(TEXT("Unable to install %s - %s /n"), 
  10.               TEXT(SZAPPNAME), 
  11.          GetLastError());//@1获取调用函数返回的最后错误码 
  12.          return
  13.      } 
  14.      //打开服务管理数据库 
  15.      schSCManager=OpenSCManager( 
  16.                             NULL,    //本地计算机 
  17.                             NULL,    //默认的数据库 
  18.                             SC_MANAGER_ALL_ACCESS  //要求所有的访问权 
  19.                             ); 
  20. if(schSCManager) 
  21.      { 
  22.          //登记服务程序 
  23.          schService=CreateService( 
  24.               schSCManager,                    //服务管理数据库句柄 
  25.               TEXT(SZSERVICENAME),             //服务名 
  26.               TEXT(SZAPPNAME),       //用于显示服务的标识 
  27.               SERVICE_ALL_ACCESS,              //响应所有的访问请求 
  28.               SERVICE_WIN32_OWN_PROCESS,       //服务类型 
  29.               SERVICE_DEMAND_START,            //启动类型 
  30.               SERVICE_ERROR_NORMAL,            //错误控制类型 
  31.               szPath,                              //服务程序磁盘文件的路径 
  32.               NULL,                                //服务不属于任何组 
  33.               NULL,                                //没有tag标识符 
  34.               NULL,              //启动服务所依赖的服务或服务组,这里仅仅是一个空字符串 
  35.               NULL,                                //LocalSystem 帐号 
  36.               NULL); 
  37.          if(schService) 
  38.          { 
  39.               _tprintf(TEXT("%s installed. /n"),TEXT(SZAPPNAME)); 
  40.               CloseServiceHandle(schService); 
  41.          } 
  42.          else 
  43.          { 
  44.               _tprintf(TEXT("CreateService failed - %s /n"),GetLastError()); 
  45.          } 
  46.          CloseServiceHandle(schSCManager); 
  47.      } 
  48.      else 
  49.          _tprintf(TEXT("OpenSCManager failed - %s /n"),GetLastError()); 

5、停止和删除已安装的服务程序


  1. //停止和删除已安装的服务程序 
  2. void removeService() 
  3.      SC_HANDLE schService; 
  4.      SC_HANDLE schSCManager; 
  5.      //打开服务管理数据库 
  6.      schSCManager=OpenSCManager( 
  7.                        NULL,    //本地计算机 
  8.                        NULL,    //默认的数据库 
  9.                        SC_MANAGER_ALL_ACCESS  //要求所有的访问权 
  10.                        ); 
  11.      if(schSCManager) 
  12.      { 
  13.          //获取服务程序句柄 
  14.          schService=OpenService( 
  15.               schSCManager,          //服务管理数据库句柄 
  16.               TEXT(SZSERVICENAME),   //服务名 
  17.               SERVICE_ALL_ACCESS     //响应所有的访问请求 
  18.               ); 
  19.          if(schService) 
  20.          { 
  21.               //试图停止服务 
  22.               if(ControlService( 
  23.                    schService,                 //服务程序句柄 
  24.                    SERVICE_CONTROL_STOP,  //停止服务请求码 
  25.                    &ssStatus              //接收最后的服务状态信息 
  26.                    )) 
  27.               { 
  28.                    _tprintf(TEXT("Stopping %s."),TEXT(SZAPPNAME)); 
  29.                    Sleep(1000); 
  30.   
  31.                    //等待服务停止 
  32.                    // 
  33.                    while(QueryServiceStatus(schService,&ssStatus)) 
  34.                    { 
  35.                        if(SERVICE_STOP_PENDING==ssStatus.dwCurrentState) 
  36.                        { 
  37.                             _tprintf(TEXT(".")); 
  38.                             Sleep(1000); 
  39.                        } 
  40.                        else 
  41.                             break
  42.                    } 
  43.                    if(SERVICE_STOPPED==ssStatus.dwCurrentState) 
  44.                        _tprintf(TEXT("/n %s stopped. /n"),TEXT(SZAPPNAME)); 
  45.                    else 
  46.                        _tprintf(TEXT("/n %s failed to stopp. /n"),TEXT(SZAPPNAME)); 
  47.               } 
  48.               //删除已安装的服务程序安装 
  49.               if(DeleteService(schService)) 
  50.                    _tprintf(TEXT("%s removed. /n"),TEXT(SZAPPNAME)); 
  51.               else 
  52. _tprintf(TEXT("DeleteService failed - %s. /n"), GetLastError()); 
  53.               CloseServiceHandle(schService); 
  54.          } 
  55.          else 
  56.               _tprintf(TEXT("OpenService failed - %s /n"),GetLastError()); 
  57.          CloseServiceHandle(schSCManager); 
  58.      } 
  59.      else 
  60.          _tprintf(TEXT("OpenSCManager failed - %s /n"),GetLastError()); 

    在编译程序的时候,我们会发觉ServiceStop();ServicePause();ServiceContinue();等三个函数没有具体实现,这对于理解此文的人来说应该不难编写,在此我可以给点文档内的参考:声明SetTheServiceStatus()函数,


  1. //  SetTheServiceStatus -   This just wraps up SetServiceStatus. 
  2. void SetTheServiceStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, 
  3.                         DWORD dwCheckPoint,   DWORD dwWaitHint) 
  4.     SERVICE_STATUS ss;  // Current status of the service. 
  5.  
  6.     //  
  7.     // Disable control requests until the service is started. 
  8.     //  
  9.     if (dwCurrentState == SERVICE_START_PENDING) 
  10.      ss.dwControlsAccepted = 0; 
  11.     else 
  12.      ss.dwControlsAccepted = 
  13.                    SERVICE_ACCEPT_STOP|SERVICE_ACCEPT_SHUTDOWN; 
  14.                    // Other flags include SERVICE_ACCEPT_PAUSE_CONTINUE 
  15.                    // and SERVICE_ACCEPT_SHUTDOWN. 
  16.  
  17.     // Initialize ss structure. 
  18.     ss.dwServiceType             = SERVICE_WIN32_OWN_PROCESS; 
  19.     ss.dwServiceSpecificExitCode = 0; 
  20.     ss.dwCurrentState            = dwCurrentState; 
  21.     ss.dwWin32ExitCode           = dwWin32ExitCode; 
  22.     ss.dwCheckPoint              = dwCheckPoint; 
  23.     ss.dwWaitHint                = dwWaitHint; 
  24.  
  25.     // Send status of the service to the Service Controller. 
  26.     if (!SetServiceStatus(ssh, &ss)) 
  27.      ErrorStopService(TEXT("SetServiceStatus")); 

    然后用如下的方式来调用函数来实现源程序中缺少的功能: 


  1. SetTheServiceStatus(SERVICE_STOPPED, GetLastError(), 0, 0);// Stop the service. 

 

***

本文转自walker snapshot博客51CTO博客,原文链接http://blog.51cto.com/walkerqt/1188534如需转载请自行联系原作者


RQSLT

上一篇:Apache与Tomcat 区别联系


下一篇:Elasticsearch聚合 之 Histogram 直方图聚合