Service程序

Service程序
2010年12月13日
　　用ATL非常简单，ATL会帮你生成一个框架。
　　以下是用SDK的：
　　编写NT的Service程序
　　任何OS都需要一个方法使当系统起动时就能运行一个或一些进程。在UNIX下那些程序由Daemons处理。如UNIX下的mail，cron，finger等等，在DOS下有TSR处理。在3.1下面，把程序放在startup组内也行。在NT下则由所谓Service来处理。在NT里当你启动后，至少有十五个service已经在运行了，其中最简单的莫过于Messenger Service了。当启动时，Messenger建立一个mailslot并且监视着它。当mailslot的队列里出现message时，它就会向NT报告。时间表
　　service稍微复杂些，当有人用at命令时，时间表service隔一会儿查一下任务表，如果时间到就执行。FTP服务器是最好的高级Service的例子。
　　在以下几种情况下应该考虑把程序写成service：
　　* 程序是用来控制某种硬件设备的。
　　* 程序需要不停地处理来访信息并返回的。(如BBS)
　　* 程序要从某个数据源周期性地拿数据。
　　* 一切有关RPC调用的程序。
　　理解NT service程序的原理
　　打开control panel里的Service可以看见诸如Clipbook Server, Computer Browser等已经在里面了。新写的Service也要被加进这个表。要被加进这个表，在注册表内要加到HKEY_LOCAL_MACHINE\System\CurrentControlSet\Services下面。每个Service必须响应一些标准的事件：
　　* Start：开始Service程序
　　* Stop：停止service程序
　　* Pause：暂停
　　* Continue：继续暂停的service
　　所有的service程序被一个叫Service Control Manager(SCM)管理着。 但个的Service程序是一个EXE程序，但必须符合SCM的接口。否则这个service程序不会工作。
　　SCM的规定接口：
　　1) EXE程序必须有main或者WinMain函数。这个函数必须立即调用StartServiceCtrlDispatcher函数。调用后，EXE便被注册到SCM里并给SCM一个指向ServiceMain()的指针，当Service程序被start时就知道哪里去调用了。当然ServiceMain()的名字可以另起一个。但main函数注册完ServiceMain到SCM后就要return。正因为这是SCM管理调用的，所以一个Service程序在DOS命令行执行不了。
　　2) 在起动service时，SCM要调用ServiceMain函数。ServiceMain函数有几件事要负责。主要的一个就是立即调用RegisterServiceCtrlHandler函数，该函数则会去注册Handler函数。RegisterServiceHandler会返回一个Handle，当服务程序需要送消息给SCM时就通过这个Handle。ServiceMain也起动了一个线程，这个现程则做这个Service程序本来应该做的事。当这个线程启动后，ServiceMain其实就开始等待有事件发生。ServiceMain返回时也就是服务程序要终止的时候。当ServiceMain返回了，服务程序也终止了。
　　3)Handler函数有个switch来分别处理从SCM收到的控制请求。缺省情况下SCM发送以下常数：
　　* SERVICE_CONTROL_STOP：让服务程序停止。
　　* SERVICE_CONTROL_PAUSE：让服务程序暂停。
　　* SERVICE_CONTROL_CONTINUE：让服务程序继续。
　　* SERVICE_CONTROL_INTERROGATE：让服务程序立即报告状态。
　　* SERVICE_CONTROL_SHUTDOWN：让服务程序Shutdown。
　　它也可以发送自定义常数(数字在128至255之间)。
　　当建立一个程序，里面有Main，ServiceMain和Handler，有一个线程做服务时，就是个完整的服务程序了。
　　一个最简单的服务程序实例：
　　理论前三讲里都讲的差不多了，这里是个最小最简单的服务程序例子
　　这个服务程序的作用是每过两秒种喇叭鸣一下。你也可以调节鸣叫的间隔时间。这个例子虽小，但它响应每个SCM控制信号。正因为如此，我想这是个很好的例子。
　　编译后，在Tools目录下用Install程序安装该服务程序：
　　..\tools\install beepserv "Beep Service " c:\....\beepserv.exe (注意一定要打对路径名)
　　别忘了得用NT管理员的身份login才能安装服务程序。想要删除用：
　　..\tools\remove beepserv
　　这里的源程序提供Makefile和Beepserve.mak你可以自己从IDE环境中或命令行编译。源程序beepserv.cpp如下：
　　// beepserv.cpp
　　// V星系的朋友们，have fun!
　　#include
　　#include
　　#include
　　#include
　　// Global variables
　　// 服务程序的名字
　　char *SERVICE_NAME = "BeepService ";
　　// 让ServiceMain等一下再结束用的事件
　　HANDLE terminateEvent = NULL;
　　// 和SCM通讯用的Handle，由RegisterServiceCtrlHandler
　　// 建立
　　SERVICE_STATUS_HANDLE serviceStatusHandle;
　　// 鸣叫的时间间隔。单位毫秒。
　　int beepDelay = 2000;
　　// 记录当前状态的标志。
　　BOOL pauseService = FALSE;
　　BOOL runningService = FALSE;
　　// 干实际工作的工作线程。
　　HANDLE threadHandle = 0;
　　void ErrorHandler(char *s, DWORD err)
　　{
　　cout CreateThread(0, 0,
　　(LPTHREAD_START_ROUTINE) ServiceThread,
　　0, 0, &id);
　　if (threadHandle==0)
　　return FALSE;
　　else
　　{
　　runningService = TRUE;
　　return TRUE;
　　}
　　}
　　// 恢复中断的服务
　　VOID ResumeService()
　　{
　　pauseService=FALSE;
　　ResumeThread(threadHandle);
　　}
　　// 暂停服务
　　VOID PauseService()
　　{
　　pauseService = TRUE;
　　SuspendThread(threadHandle);
　　}
　　// 让ServiceMan结束也就终止了服务程序
　　VOID StopService()
　　{
　　runningService=FALSE;
　　SetEvent(terminateEvent);
　　}
　　// 用SetServiceStatus更新服务程序的状态
　　BOOL SendStatusToSCM (DWORD dwCurrentState,
　　DWORD dwWin32ExitCode,
　　DWORD dwServiceSpecificExitCode,
　　DWORD dwCheckPoint,
　　DWORD dwWaitHint)
　　{
　　BOOL success;
　　SERVICE_STATUS serviceStatus;
　　// 填入SERVICE_STATUS
　　serviceStatus.dwServiceType =
　　SERVICE_WIN32_OWN_PROCESS;
　　serviceStatus.dwCurrentState = dwCurrentState;
　　if (dwCurrentState == SERVICE_START_PENDING)
　　serviceStatus.dwControlsAccepted = 0;
　　else
　　serviceStatus.dwControlsAccepted =
　　SERVICE_ACCEPT_STOP |
　　SERVICE_ACCEPT_PAUSE_CONTINUE |
　　SERVICE_ACCEPT_SHUTDOWN;
　　if (dwServiceSpecificExitCode == 0)
　　serviceStatus.dwWin32ExitCode =
　　dwWin32ExitCode;
　　else
　　serviceStatus.dwWin32ExitCode =
　　ERROR_SERVICE_SPECIFIC_ERROR;
　　serviceStatus.dwServiceSpecificExitCode =
　　dwServiceSpecificExitCode;
　　serviceStatus.dwCheckPoint = dwCheckPoint;
　　serviceStatus.dwWaitHint = dwWaitHint;
　　// 传送状态给SCM
　　success = SetServiceStatus (serviceStatusHandle,
　　&serviceStatus);
　　if (!success)
　　StopService();
　　return success;
　　}
　　// 把从service control manager受到的时间分而治之
　　VOID ServiceCtrlHandler (DWORD controlCode)
　　{
　　DWORD currentState = 0;
　　BOOL success;
　　switch(controlCode)
　　{
　　// 停止 the service
　　case SERVICE_CONTROL_STOP:
　　currentState = SERVICE_STOP_PENDING;
　　success = SendStatusToSCM(
　　SERVICE_STOP_PENDING,
　　NO_ERROR, 0, 1, 5000);
　　// Stop the service
　　StopService();
　　return;
　　// 暂停service
　　case SERVICE_CONTROL_PAUSE:
　　if (runningService && !pauseService)
　　{
　　success = SendStatusToSCM(
　　SERVICE_PAUSE_PENDING,
　　NO_ERROR, 0, 1, 1000);
　　PauseService();
　　currentState = SERVICE_PAUSED;
　　}
　　break;
　　// 恢复
　　case SERVICE_CONTROL_CONTINUE:
　　if (runningService && pauseService)
　　{
　　success = SendStatusToSCM(
　　SERVICE_CONTINUE_PENDING,
　　NO_ERROR, 0, 1, 1000);
　　ResumeService();
　　currentState = SERVICE_RUNNING;
　　}
　　break;
　　// 更新当前状态
　　case SERVICE_CONTROL_INTERROGATE:
　　// it will fall to bottom and send status
　　break;
　　// 终止时什么也不用做。你可以做些扫尾工作，但
　　// 记住必须非常快做完！
　　case SERVICE_CONTROL_SHUTDOWN:
　　return;
　　default:
　　break;
　　}
　　SendStatusToSCM(currentState, NO_ERROR,
　　0, 0, 0);
　　}
　　// 错误处理。
　　VOID terminate(DWORD error)
　　{
　　// 如果terminateEvent发生, 关闭它.
　　if (terminateEvent)
　　CloseHandle(terminateEvent);
　　// 发个message给scm
　　if (serviceStatusHandle)
　　SendStatusToSCM(SERVICE_STOPPED, error,
　　0, 0, 0);
　　// 如果线程在，杀了它
　　if (threadHandle)
　　CloseHandle(threadHandle);
　　}
　　// 开始服务程序时SCM会调用下面的serviceMain函数。
　　// ServiceMain一返回。服务程序就结束了。
　　VOID ServiceMain(DWORD argc, LPTSTR *argv)
　　{
　　BOOL success;
　　// 立即调用Registration函数
　　serviceStatusHandle =
　　RegisterServiceCtrlHandler(
　　SERVICE_NAME,
　　(LPHANDLER_FUNCTION) ServiceCtrlHandler);
　　if (!serviceStatusHandle)
　　{
　　terminate(GetLastError());
　　return;
　　}
　　success = SendStatusToSCM(
　　SERVICE_START_PENDING,
　　NO_ERROR, 0, 1, 5000);
　　if (!success)
　　{
　　terminate(GetLastError());
　　return;
　　}
　　// 制造终止事件。
　　terminateEvent = CreateEvent (0, TRUE, FALSE,
　　0);
　　if (!terminateEvent)
　　{
　　terminate(GetLastError());
　　return;
　　}
　　success = SendStatusToSCM(
　　SERVICE_START_PENDING,
　　NO_ERROR, 0, 2, 1000);
　　if (!success)
　　{
　　terminate(GetLastError());
　　return;
　　}
　　if (argc == 2)
　　{
　　int temp = atoi(argv[1]);
　　if (temp < 1000)
　　beepDelay = 2000;
　　else
　　beepDelay = temp;
　　}
　　success = SendStatusToSCM(
　　SERVICE_START_PENDING,
　　NO_ERROR, 0, 3, 5000);
　　if (!success)
　　{
　　terminate(GetLastError());
　　return;
　　}
　　// 开始服务程序。
　　success = InitService();
　　if (!success)
　　{
　　terminate(GetLastError());
　　return;
　　}
　　success = SendStatusToSCM(
　　SERVICE_RUNNING,
　　NO_ERROR, 0, 0, 0);
　　if (!success)
　　{
　　terminate(GetLastError());
　　return;
　　}
　　// 等待终止信号。
　　WaitForSingleObject (terminateEvent, INFINITE);
　　terminate(0);
　　}
　　VOID main(VOID)
　　{
　　SERVICE_TABLE_ENTRY serviceTable[] =
　　{
　　{ SERVICE_NAME,
　　(LPSERVICE_MAIN_FUNCTION) ServiceMain},
　　{ NULL, NULL }
　　};
　　BOOL success;
　　// 向SCM注册。
　　success =
　　StartServiceCtrlDispatcher(serviceTable);
　　if (!success)
　　ErrorHandler( "In StartServiceCtrlDispatcher ",
　　GetLastError());
　　}