Windows服务监听程序

目录

Part One:函数组件实现 (1)

Part two:安装/卸载/调用服务主程序，Main函数入口 (9)

Part Three:头文件，平台/浏览器使用版本宏定义 (10)

Part Four:头文件，函数组件声明 (10)

Part One:函数组件实现

TCHAR g_szServieName[] = TEXT("HookService");//服务名称

TCHAR g_szServieDescription[] = TEXT("The Service Of HookProgram");//服务描述

SERVICE_STATUS g_svrStatus;//服务状态

SERVICE_STATUS_HANDLE g_hStatus;

DWORD g_errCode = 0;

HANDLE g_eventWait = NULL;

DWORD waitID = 0; // 监控Token插拔的线程的ID

HANDLE waitHD = NULL; // 监控Token插拔的线程的句柄

//P11_CONTEXT g_p11Ctx; // PKCS#11库的全局上下文

//*************************************************************************************************** *************************

//编译生成可执行程序的绝对路径函数

BOOL FindExePath(TCHAR* Path)

{

if ( !GetModuleFileName(NULL, Path, MAX_PATH) ) {

printf_s("Get ExeFile Failed. error code = %d\n", GetLastError());

return FALSE;

}

return TRUE;

}

DWORD WINAPI waitThreadProc(LPVOID lpParameter) {

waitID=0;

return 0;

}

//启动服务

int StartMoniter() {

//InitP11Context(NULL/*&g_p11Ctx*/, ReadStartupDir, TRUE);

// 创建Key插拔监视线程，监key的状态（插拔），保存该线程句柄

waitHD = CreateThread(

NULL, // SECURITY_ATTRIBUTES结构指针，可以为NULL

0, // 初始的栈大小，此处为，使用缺省大小

(LPTHREAD_START_ROUTINE)waitThreadProc, // 线程函数和线程启动地址

NULL,//&g_p11Ctx, // 将p11库的函数列表指针作为参数传递

CREATE_SUSPENDED, // 创建标志为CREATE_SUSPENDED，不立即运行

&waitID // 线程标识由waitID返回

);

// 启动创建好的key监控线程

ResumeThread(waitHD);//激活线程

return 0;

}

BOOL Run() {

SERVICE_TABLE_ENTRY DispatchTable[] = {

{ g_szServieName, (LPSERVICE_MAIN_FUNCTION)ServiceMain },

{ NULL, NULL } // 最后一个元素必须为{NULL, NULL}

};

// 将服务进程的主线程连接到服务控制管理器SCM

return StartServiceCtrlDispatcher(DispatchTable);

}

BOOL InstallService()

{

TCHAR FilePath[MAX_PATH+1];

ZeroMemory(FilePath,MAX_PATH);

// 首先获取服务可执行程序的绝对路径

FindExePath(FilePath);

// 建立与服务控制管理器（SCM）的连接

SC_HANDLE hSCManager;

hSCManager = OpenSCManager(

NULL, // 目标主机名，此处为NULL，本机

NULL, // 数据库名，缺省为SERVICES_ACTIVE_DATABASE（参数NULL）

SC_MANAGER_ALL_ACCESS // 访问权限

);

if ( hSCManager == NULL ) {

printf_s("OpenSCManager failed, error code = %d\n", GetLastError());

return FALSE;

}

// 创建一个服务对象并将其增添到服务控制管理器（SCM）的数据库中

SC_HANDLE hService;

hService = CreateService(

hSCManager, // 连接句柄，代表一个SCM数据库

g_szServieName, // 服务名

g_szServieName, // 服务显示名

SERVICE_ALL_ACCESS, // 服务访问权限

// 服务类型：

// SERVICE_WIN32_OWN_PROCESS表示独立进程

// SERVICE_INTERACTIVE_PROCESS表示服务能与桌面交互

SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS,

SERVICE_AUTO_START, // 启动选项，此处为自动启动

SERVICE_ERROR_NORMAL, // 错误处理

FilePath, // 服务可执行文件的路径

NULL, // 服务组，为NULL则表示此服务不属任何组NULL, // 服务组中的唯一标签

NULL, // 服务依赖项

NULL, // 服务启动帐户

NULL // 服务启动帐户的口令

);

if ( hService == NULL ) {

CloseServiceHandle(hSCManager);

printf_s("create service failed, error code = %d\n", GetLastError());

return FALSE;

}

// 更改刚创建的服务的描述字符串

SERVICE_DESCRIPTION sd = {0};

sd.lpDescription = g_szServieDescription;

ChangeServiceConfig2(hService, SERVICE_CONFIG_DESCRIPTION, &sd);

// 清理并返回

CloseServiceHandle( hService );

CloseServiceHandle( hSCManager );

return TRUE;

}

BOOL UnInstallService()

{

// 打开SCM数据库

SC_HANDLE hSCManager;

hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);

if ( hSCManager == NULL ) {

printf_s("OpenSCManager failed, error code = %d\n", GetLastError());

return FALSE;

}

// 打开指定名称的服务

SC_HANDLE hService;

hService = OpenService(

hSCManager, // SCM数据库连接

g_szServieName, // 服务名

SERVICE_ALL_ACCESS // 访问权限

);

if ( hService == NULL ) {

printf_s("OpenService failed, error code = %d\n", GetLastError());

return FALSE;

}

// 首先停止服务

SERVICE_STATUS ss = {0};

if ( !ControlService(

hService, // 打开的服务

SERVICE_CONTROL_STOP, // 控制码

&ss // 控制的返回状态

) ) {

printf_s("ControlService failed, error code = %d\n", GetLastError());

}

// 删除服务

if ( !DeleteService(hService) ) {

printf_s("DeleteService failed, error code = %d\n", GetLastError());

return FALSE;

}

else printf_s("Service deleted successfully\n");

// 清理并返回

CloseServiceHandle(hService);

CloseServiceHandle(hSCManager);

return TRUE;

}

VOID WINAPI ServiceMain(DWORD argc, LPTSTR* argv) {

// 首先填充SERVICE_STATUS结构

g_svrStatus.dwServiceType = SERVICE_WIN32; // 服务类型

g_svrStatus.dwCurrentState = SERVICE_START_PENDING; // 服务状态为待启动

g_svrStatus.dwControlsAccepted = //

SERVICE_ACCEPT_STOP|SERVICE_ACCEPT_SHUTDOWN; // 允许的控制g_svrStatus.dwWin32ExitCode = //

ERROR_SERVICE_SPECIFIC_ERROR; // 用来报告启动或停止错误的错误码

g_svrStatus.dwServiceSpecificExitCode = 0; // 服务特定的错误码，

dwWin32ExitCode为

/////////////////////////////////////////////////////// ERROR_SERVICE_SPECIFIC_ERROR被忽略g_svrStatus.dwCheckPoint = 1; // 执行长之间操作时报告状态的周期g_svrStatus.dwWaitHint = 2000; // 等待启动、停止等操作的时间限制，单位为毫秒

// 注册服务状态改变时的回调函数

DWORD dwWait;

g_hStatus = RegisterServiceCtrlHandlerEx(

g_szServieName, // 服务名称

ServiceCtrlHandlerEx, // 回调函数的地址

&dwWait // 传递给回调函数的用户自定义数据的地址

);

if ( g_hStatus == SERVICE_STATUS_HANDLE(NULL) ) {

printf_s("RegisterServiceCtrlHandler failed, error code = %d\n", GetLastError());

return;

}

// 更新SCM中的状态信息，若失败向SCM报告

if ( !SetServiceStatus(g_hStatus, &g_svrStatus) ) {

printf_s("SetServiceStatus pending failed, error code = %d\n", GetLastError());

ReportStatusToSCMgr(SERVICE_STOPPED, g_errCode, 0);

return;

}

// 创建一个事件对象，若失败向SCM报告

g_eventWait = CreateEvent(

NULL, // 指向SECURITY_ATTRIBUTES结构的指针，此处为NULL，此句柄不能被子进程继承

TRUE, // 参数TRU表示创建一个手工重置的事件对象，该对象需调用ResetEvent函数清除

FALSE, // FALSE表示新建的事件对象没有信号（signalel）

NULL // 事件对象的名称，为NULL表示匿名对象

);

if ( g_eventWait == NULL ) {

printf_s("CreateEvent failed, error code = %d\n", GetLastError());

ReportStatusToSCMgr(SERVICE_STOPPED, g_errCode, 0);

return;

}

// 启动服务主体函数

StartMoniter();

// 报告服务启动成功

if ( !ReportStatusToSCMgr(SERVICE_RUNNING, NO_ERROR, 2) ) {

printf_s("ReportStatusToSCMgr failed, error code = %d\n", GetLastError());

ReportStatusToSCMgr(SERVICE_STOPPED, g_errCode, 0);

return;

}

// 等待事件对象上的信号

dwWait = WaitForSingleObject(

g_eventWait, // 等待的对象是前面建立的事件对象

INFINITE // 没有期限

);

// 事件上有信号发生，清理并退出服务主函数

if ( g_eventWait != NULL ) CloseHandle(g_eventWait);

if ( g_hStatus != NULL ) ReportStatusToSCMgr( SERVICE_STOPPED, g_errCode, 0);

}

DWORD WINAPI ServiceCtrlHandlerEx(DWORD control, DWORD eventType, LPVOID lpEnentData, LPVOID lpContext) {

DWORD dwState = SERVICE_RUNNING;

switch (control) {

case SERVICE_CONTROL_PAUSE: // 控制码为暂停服务

if ( g_svrStatus.dwCurrentState == SERVICE_RUNNING ) { // 服务运行

//PauseService();

dwState = SERVICE_PAUSED;

}

break;

case SERVICE_CONTROL_CONTINUE: // 控制玛为服务继续

if (g_svrStatus.dwCurrentState == SERVICE_PAUSED ) {

//ResumeService();

dwState = SERVICE_RUNNING;

}

break;

case SERVICE_CONTROL_SHUTDOWN: // 控制码为主机关机

case SERVICE_CONTROL_STOP : // 控制码为停止服务

dwState = SERVICE_STOP_PENDING;

(VOID)ReportStatusToSCMgr(SERVICE_STOP_PENDING, NO_ERROR, 1); // 报告新状态

SetEvent(g_eventWait); // 在等待事件上产生信号

return 0;

case SERVICE_CONTROL_INTERROGATE: // 控制码为质询

break;

default :

break;

}

(VOID)ReportStatusToSCMgr(dwState, NO_ERROR, -1);

return 0;

}

BOOL ReportStatusToSCMgr(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwCheckPoint) {

BOOL fResult;

if ( dwCurrentState == SERVICE_START_PENDING ) g_svrStatus.dwControlsAccepted = 0;

else g_svrStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;

if ( dwCurrentState >= 0 ) g_svrStatus.dwCurrentState = dwCurrentState;

g_svrStatus.dwWin32ExitCode = dwWin32ExitCode;

if ( dwCheckPoint < 0 ) g_svrStatus.dwCheckPoint++;

else g_svrStatus.dwCheckPoint = dwCheckPoint;

// 设置服务状态

if ( !(fResult = SetServiceStatus(g_hStatus, &g_svrStatus)) )

StopService(TEXT("SetServiceStatus"));

return fResult != 0;

}

void StopService(LPTSTR lpszMsg) {

TCHAR chMsg[256];

HANDLE hEventSource;

LPTSTR lpszStrings[2];

g_errCode = GetLastError();

// 取得服务上的事件日志并生成日志文本

hEventSource = RegisterEventSource(NULL, g_szServieName);

_stprintf_s(chMsg, 256, TEXT("%s error:%d"), g_szServieName, g_errCode);

lpszStrings[0] = chMsg;

lpszStrings[1] = lpszMsg;

// 在指定的事件日志中加入记录

if ( hEventSource != NULL ) {

ReportEvent(

hEventSource, // 事件日志的句柄

EVENTLOG_ERROR_TYPE, // 写入日志的事件类型，此处为错误

0, // 事件分类，于事件源相关，可以为任意值

0, // 事件标识

NULL, // 用户的安全识别

2, // lpStrings参数中字符串的个数

0, // 事件相关数据的大小，字节为单位

(LPCTSTR*)lpszStrings, // lpStrings，用来拼装服务消息的字符串数组

NULL // 事件相关的二进制数据指针

);

// 关闭事件日志写入句柄

(VOID)DeregisterEventSource(hEventSource);

}

if ( waitHD ) TerminateThread(waitHD, 0);

// ClearP11Context(&g_p11Ctx);

// 在全局事件上产生一个信号

SetEvent(g_eventWait);

}

***************************************************************************************************** ***************************************************************************************************** ***************************************************************************************************** ***************************************************************************************************** ***************************************************************************************************** ***************************************************************************************************** *****************************************************************************************************

Part two:安装/卸载/调用服务主程序，Main函数入口

#define CountPra_zero 0

#define CountPra_one 1

#define CountPra_two 2

void ShowUseFun(int argc, char** argv)

{

printf("\tService Install&Uninstall Info:\n");

printf("\tInstall Service:");

printf("\t %s --install\n",argv[CountPra_zero]);

printf("\tUnInstall Service:");

printf("\t %s --uninstall\n",argv[CountPra_zero]);

}

int main(int argc,char* argv[])

{

if(CountPra_two==argc){

if(strcmp(argv[CountPra_one],"--install")==CountPra_zero)

InstallService();

else if(strcmp(argv[CountPra_one],"--uninstall")==CountPra_zero)

UnInstallService();

else{

printf("Input Error!\n");

ShowUseFun(argc, argv);

}

}

else if(CountPra_one==argc)

Run();

//ShowUseFun(argc, argv);

else

ShowUseFun(argc, argv);

return 0;

}

Part Three:头文件，平台/浏览器使用版本宏定义

#pragma once

// 以下宏定义要求的最低平台。要求的最低平台

// 是具有运行应用程序所需功能的Windows、Internet Explorer 等产品的

// 最早版本。通过在指定版本及更低版本的平台上启用所有可用的功能，宏可以

// 正常工作。

// 如果必须要针对低于以下指定版本的平台，请修改下列定义。

// 有关不同平台对应值的最新信息，请参考MSDN。

#ifndef WINVER // 指定要求的最低平台是Windows Vista。

#define WINVER 0x0600 // 将此值更改为相应的值，以适用于Windows 的其他版本。

#endif

#ifndef _WIN32_WINNT // 指定要求的最低平台是Windows Vista。

#define _WIN32_WINNT 0x0600 // 将此值更改为相应的值，以适用于Windows 的其他版本。#endif

#ifndef _WIN32_WINDOWS // 指定要求的最低平台是Windows 98。

#define _WIN32_WINDOWS 0x0410 // 将此值更改为适当的值，以适用于Windows Me 或更高版本。#endif

#ifndef _WIN32_IE // 指定要求的最低平台是Internet Explorer 7.0。

#define _WIN32_IE 0x0700 // 将此值更改为相应的值，以适用于IE 的其他版本。

#endif

Part Four:头文件，函数组件声明

#ifndef _SERVICE_H_

#define _SERVICE_H_

#include

#include

#include

#ifndef MAX_PATH

#define MAX_PATH 256

#endif

BOOL Run();//服务主程序入口

/* 安装服务*/

BOOL InstallService();

/* 卸载服务*/

BOOL UnInstallService();

/* 服务主体函数*/

VOID WINAPI ServiceMain(DWORD argc, LPTSTR* argv);;

/* 服务状态处理*/

DWORD WINAPI ServiceCtrlHandlerEx(

DWORD control, // 控制码

DWORD eventType, // 已发生的事件类型

LPVOID lpEnentData, // 事件数据，由dwControl和dwEventType决定其类型

LPVOID lpContext // 用户自定义数据，由RegisterServiceCtrlHandlerEx传入

);

/* 服务状态控制器处理函数*/

BOOL ReportStatusToSCMgr(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwCheckPoint);

/* 停止服务*/

void StopService(LPTSTR lpszMsg);

#endif// __SERVICE_H__