转载自:
一直感觉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 函数负责把程序主线程连接到服务控制管理程序。具体描述如下:
- BOOL StartServiceCtrlDispatcher(const LPSERVICE_TABLE_ENTRY lpServiceTable);
lpServiceStartTable 指向 SERVICE_TABLE_ENTRY 结构类型的数组,他包含了调用进程所提供的每个服务的入口函数和字符串名。表中的最后一个元素必须为 NULL,指明入口表结束。
SERVICE_TABLE_ENTRY 结构具体描述如下:
- typedef struct _SERVICE_TABLE_ENTRY {
- LPTSTR lpServiceName;
- LPSERVICE_MAIN_FUNCTION lpServiceProc;
- } SERVICE_TABLE_ENTRY, *LPSERVICE_TABLE_ENTRY;
lpServiceName 是一个以 NULL 结尾的字符串,标识服务名。如果是 SERVICE_WIN32_OWN_PROCESS 类型的服务,这个字符串会被忽略。
lpServiceProc 指向服务入口点函数。
-
- #include "stdafx.h"
- #include "Windows.h"
- #define SZAPPNAME "serverSample" //服务程序名
- #define SZSERVICENAME "serviceSample" //标识服务的内部名
-
-
- bool bDebugServer=false;
- SERVICE_STATUS ssStatus;
- SERVICE_STATUS_HANDLE sshStatusHandle;
- DWORD dwErr=0;
- TCHAR szErr[256];
-
-
- void WINAPI Service_Main(DWORD dwArgc, LPTSTR *lpszArgv);
- void WINAPI Service_Ctrl(DWORD dwCtrlCode);
- void installService();
- void removeService();
- void debugService(int argc,char** argv);
- bool ReportStatusToSCMgr(DWORD dwCurrentState,DWORD dwWin32ExitCode,DWORD dwWaitHint);
- void AddToMessageLog(LPTSTR lpszMsg);
-
- int _tmain(int argc, _TCHAR* argv[])
- {
- SERVICE_TABLE_ENTRY dispatchTable[]=
- {
- {TEXT(SZSERVICENAME),(LPSERVICE_MAIN_FUNCTION)Service_Main},
- { NULL,NULL}
- };
- if((argc>1)&&((*argv[1]=='-')||(argv[1]=="/")))
- {
- if(_stricmp("install",argv[1]+1)==0)
- {
- installService();
- }
- else if(_stricmp("remove",argv[1]+1)==0)
- {
- removeService();
- }
- else if(_stricmp("debug",argv[1]+1)==0)
- {
- bDebugServer=true;
- debugService(argc,argv);
- }
- else
- {
-
- printf("%s - install to install the service /n",SZAPPNAME);
- printf("%s - remove to remove the service /n",SZAPPNAME);
- printf("%s - debug to debug the service /n",SZAPPNAME);
- printf("/n StartServiceCtrlDispatcher being called./n");
- printf("This may take several seconds.Please wait./n");
- if(!StartServiceCtrlDispatcher(dispatchTable))
- AddToMessageLog(TEXT("StartServiceCtrlDispatcher failed."));
- else
- AddToMessageLog(TEXT("StartServiceCtrlDispatcher OK."));
- }
- exit(0);
- }
- return 0;
- }
2、服务入口点函数
服务入口点函数 service_main 首先调用系统函数 RegisterServiceCtrlHandler 注册服务控制处理函数 service_ctrl,然后调用 ReportStatusToSCMgr 函数,它通过系统函数 SetServiceStatus 更新服务的状态,然后调用特定的服务初始化入口函数 ServiceStart 完成具体的初始化工作。
-
- void ServiceStart(DWORD dwArgc,LPTSTR* lpszArgv);
- void WINAPI Service_Main(DWORD dwArgc, LPTSTR *lpszArgv)
- {
-
- sshStatusHandle=RegisterServiceCtrlHandler(TEXT(SZSERVICENAME),Service_Ctrl);
-
- if(!sshStatusHandle)
- {
- goto cleanup;
- return;
- }
-
- ssStatus.dwServiceType=SERVICE_WIN32_OWN_PROCESS;
- ssStatus.dwServiceSpecificExitCode=0;
-
- if(!ReportStatusToSCMgr(
- SERVICE_START_PENDING,
- NO_ERROR,
- 3000))
- goto cleanup;
- ServiceStart(dwArgc,lpszArgv);
- return;
- cleanup:
-
- if(sshStatusHandle)
- (void)ReportStatusToSCMgr(SERVICE_STOPPED,dwErr,0);
- }
3、控制处理程序函数
函数 Service_Ctrl 是服务的控制处理程序函数,由主函数线程的控制分发程序引用。在处理控制请求码时,应该在确定的时间间隔内更新服务状态检查点,避免发生服务不能响应的错误。
-
- void WINAPI Service_Ctrl(DWORD dwCtrlCode)
- {
-
- switch(dwCtrlCode)
- {
-
- case SERVICE_CONTROL_STOP:
- ReportStatusToSCMgr(SERVICE_STOP_PENDING,NO_ERROR,500);
- ServiceStop();
- return;
-
- case SERVICE_CONTROL_PAUSE:
- ReportStatusToSCMgr(SERVICE_STOP_PENDING,NO_ERROR,500);
- ServicePause();
- ssStatus.dwCurrentState=SERVICE_PAUSED;
- return;
-
- case SERVICE_CONTROL_CONTINUE:
- ReportStatusToSCMgr(SERVICE_STOP_PENDING,NO_ERROR,500);
- ServiceContinue();
- ssStatus.dwCurrentState=SERVICE_RUNNING;
- return;
-
- case SERVICE_CONTROL_INTERROGATE:
- break;
-
- default:
- break;
- }
- ReportStatusToSCMgr(ssStatus.dwCurrentState,NO_ERROR,0);
- }
除了系统定义的五种控制码外(还有一种是:SERVICE_CONTROL_SHUTDOWN),用户还可自定义控制码,其取值范围是128-255。用户可以通过控制面板中的服务项向特定服务程序的控制处理函数发送控制码,程序员可以调用系统函数 ControlService 直接向服务程序的控制处理函数发送控制码。其函数原型如下:
- BOOL ControlService(
- SC_HANDLE hService,
- DWORD dwControl,
- LPSERVICE_STATUS lpServiceStatus
- );
-
-
-
4、安装服务程序 每个已安装服务程序在 HKEY_LOCAL_MACHINE/SYSTE/CurrentControlSet/Services 下都有一个服务名的关键字,程序员可以调用系统函数 CreateService 安装服务程序,并指定服务类型,服务名等。这个函数创建一个服务对象,并将其增加到相关的服务控制管理器数据库中。下面是函数原型:
- SC_HANDLE CreateService(
- SC_HANDLE hSCManager,
- LPCTSTR lpServiceName,
- LPCTSTR lpDisplayName,
- DWORD dwDesiredAccess,
- DWORD dwServiceType,
- DWORD dwStartType,
- DWORD dwErrorControl,
- LPCTSTR lpBinaryPathName,
- LPCTSTR lpLoadOrderGroup,
- LPDWORD lpdwTagId,
- LPCTSTR lpDependencies,
- LPCTSTR lpServiceStartName,
- LPCTSTR lpPassword
- );
对于一个已安装的服务程序,可以调用系统函数 OpenService 来获取服务程序的句柄。下面是其函数原型:
- SC_HANDLE OpenService(
- SC_HANDLE hSCManager,
- LPCTSTR lpServiceName,
- DWORD dwDesiredAccess
- );
-
-
-
用CreateService 或OpenService 打开的服务程序句柄使用完毕后必须用CloseServiceHandle 关闭。OpenSCManager打开的服务管理数据库句柄也必须用它来关闭。
-
- void installService()
- {
- SC_HANDLE schService;
- SC_HANDLE schSCManager;
- TCHAR szPath[512];
-
- if(GetModuleFileName(NULL,szPath,512)==0)
- {
- _tprintf(TEXT("Unable to install %s - %s /n"),
- TEXT(SZAPPNAME),
- GetLastError());
- return;
- }
-
- schSCManager=OpenSCManager(
- NULL,
- NULL,
- SC_MANAGER_ALL_ACCESS
- );
- if(schSCManager)
- {
-
- schService=CreateService(
- schSCManager,
- TEXT(SZSERVICENAME),
- TEXT(SZAPPNAME),
- SERVICE_ALL_ACCESS,
- SERVICE_WIN32_OWN_PROCESS,
- SERVICE_DEMAND_START,
- SERVICE_ERROR_NORMAL,
- szPath,
- NULL,
- NULL,
- NULL,
- NULL,
- NULL);
- if(schService)
- {
- _tprintf(TEXT("%s installed. /n"),TEXT(SZAPPNAME));
- CloseServiceHandle(schService);
- }
- else
- {
- _tprintf(TEXT("CreateService failed - %s /n"),GetLastError());
- }
- CloseServiceHandle(schSCManager);
- }
- else
- _tprintf(TEXT("OpenSCManager failed - %s /n"),GetLastError());
- }
5、停止和删除已安装的服务程序
-
- void removeService()
- {
- SC_HANDLE schService;
- SC_HANDLE schSCManager;
-
- schSCManager=OpenSCManager(
- NULL,
- NULL,
- SC_MANAGER_ALL_ACCESS
- );
- if(schSCManager)
- {
-
- schService=OpenService(
- schSCManager,
- TEXT(SZSERVICENAME),
- SERVICE_ALL_ACCESS
- );
- if(schService)
- {
-
- if(ControlService(
- schService,
- SERVICE_CONTROL_STOP,
- &ssStatus
- ))
- {
- _tprintf(TEXT("Stopping %s."),TEXT(SZAPPNAME));
- Sleep(1000);
-
-
-
- while(QueryServiceStatus(schService,&ssStatus))
- {
- if(SERVICE_STOP_PENDING==ssStatus.dwCurrentState)
- {
- _tprintf(TEXT("."));
- Sleep(1000);
- }
- else
- break;
- }
- if(SERVICE_STOPPED==ssStatus.dwCurrentState)
- _tprintf(TEXT("/n %s stopped. /n"),TEXT(SZAPPNAME));
- else
- _tprintf(TEXT("/n %s failed to stopp. /n"),TEXT(SZAPPNAME));
- }
-
- if(DeleteService(schService))
- _tprintf(TEXT("%s removed. /n"),TEXT(SZAPPNAME));
- else
- _tprintf(TEXT("DeleteService failed - %s. /n"), GetLastError());
- CloseServiceHandle(schService);
- }
- else
- _tprintf(TEXT("OpenService failed - %s /n"),GetLastError());
- CloseServiceHandle(schSCManager);
- }
- else
- _tprintf(TEXT("OpenSCManager failed - %s /n"),GetLastError());
- }
在编译程序的时候,我们会发觉ServiceStop();ServicePause();ServiceContinue();等三个函数没有具体实现,这对于理解此文的人来说应该不难编写,在此我可以给点文档内的参考:声明SetTheServiceStatus()函数,
-
- void SetTheServiceStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode,
- DWORD dwCheckPoint, DWORD dwWaitHint)
- {
- SERVICE_STATUS ss;
-
-
-
-
- if (dwCurrentState == SERVICE_START_PENDING)
- ss.dwControlsAccepted = 0;
- else
- ss.dwControlsAccepted =
- SERVICE_ACCEPT_STOP|SERVICE_ACCEPT_SHUTDOWN;
-
-
-
-
- ss.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
- ss.dwServiceSpecificExitCode = 0;
- ss.dwCurrentState = dwCurrentState;
- ss.dwWin32ExitCode = dwWin32ExitCode;
- ss.dwCheckPoint = dwCheckPoint;
- ss.dwWaitHint = dwWaitHint;
-
-
- if (!SetServiceStatus(ssh, &ss))
- ErrorStopService(TEXT("SetServiceStatus"));
- }
然后用如下的方式来调用函数来实现源程序中缺少的功能:
- SetTheServiceStatus(SERVICE_STOPPED, GetLastError(), 0, 0);
***
本文转自walker snapshot博客51CTO博客,原文链接http://blog.51cto.com/walkerqt/1188534如需转载请自行联系原作者
RQSLT