最近需要将一些命令行程序包装成后台服务, 本来可以用 .NET 完成, 不过又想尝试一下用 C 语言实 现 Windows 服务, 发现用 C 语言做 Windows 服务也是很容易的, 步骤如下:
1 包含必要的头文件, 并定义一些常量, 代码如下:
#include <windows.h> #include <stdio.h> // 服务线程暂停时间; #define SLEEP_TIME 5000 // 日志文件输入路径 #define LOGFILE "C:\\memstatus.txt"
2 写日志文件, 几乎所有的服务都有日志输出, 即使是最简单的服务, 可以说: “无日志不服务!“ 作为一个简单的例子, 仅仅简单的向日志文件输出字符串。
int WriteToLog(char* str) {
FILE* log; log = fopen(LOGFILE, "a+"); if (log == NULL) return -1; fprintf(log, "%s\n", str);
fclose(log); return 0;
}
3 全局服务状态和服务控制处理变量, 这写变量会在多个函数中用到, 所以定义成全局变量。
// 当前服务状态 SERVICE_STATUS ServiceStatus; // 服务控制处理函数 SERVICE_STATUS_HANDLE ServiceStatusHandle;
4 初始化服务, 作为示例, 仅简单向输出日志。
int InitService() { int result;
result = WriteToLog("Monitoring started."); return(result);
}
5 服务控制处理函数, 响应在服务管理器中对服务的操作(停止、重新启动)。
void ServiceControlHandler(DWORD request) { switch (request) { case SERVICE_CONTROL_STOP: case SERVICE_CONTROL_SHUTDOWN: // 服务停止 WriteToLog("Monitoring stopped.");
ServiceStatus.dwWin32ExitCode = 0;
ServiceStatus.dwCurrentState = SERVICE_STOPPED; break; default: break;
} // 向服务管理器汇报服务状态 SetServiceStatus(ServiceStatusHandle, &ServiceStatus); return;
}
6 服务入口函数, 运行在后台服务线程中, 服务的逻辑主要在这个函数中实现。
void ServiceMain(int argc, char** argv) { // 初始化服务类型、 状态、 接受的控制方法以及期待的返回值 ServiceStatus.dwServiceType = SERVICE_WIN32;
ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
ServiceStatus.dwWin32ExitCode = 0;
ServiceStatus.dwServiceSpecificExitCode = 0;
ServiceStatus.dwCheckPoint = 0;
ServiceStatus.dwWaitHint = 0; // 注册服务控制处理函数 ServiceStatusHandle = RegisterServiceCtrlHandler(
L"MemoryStatus", // 服务名称必须与安装服务时的名称一致; (LPHANDLER_FUNCTION)ServiceControlHandler
); if (ServiceStatusHandle == NULL) { // 服务注册失败 return;
} // 初始化服务 int error = InitService(); if (error) { // 初始化服务失败, 设置服务状态为 STOPPED 并返回 ServiceStatus.dwCurrentState = SERVICE_STOPPED;
ServiceStatus.dwWin32ExitCode = -1;
SetServiceStatus(ServiceStatusHandle, &ServiceStatus); return;
} // 现在服务已经成功运行起来了, 向服务管理器汇报状态。 ServiceStatus.dwCurrentState = SERVICE_RUNNING;
SetServiceStatus(ServiceStatusHandle, &ServiceStatus);
MEMORYSTATUS memory; // 服务线程主循环, 每隔 5 秒钟读取系统内存状态, 输出到日志文件 while (ServiceStatus.dwCurrentState == SERVICE_RUNNING) { char buffer[16];
GlobalMemoryStatus(&memory); sprintf(buffer, "%d", memory.dwAvailPhys); int result = WriteToLog(buffer); if (result) {
ServiceStatus.dwCurrentState = SERVICE_STOPPED;
ServiceStatus.dwWin32ExitCode = -1;
SetServiceStatus(ServiceStatusHandle, &ServiceStatus); return;
}
Sleep(SLEEP_TIME);
} return;
}
7 最后, 程序入口函数, 只是简单的注册服务入口函数, 然后就退出了。 因为已经想系统注册了服务 入口函数, 系统会继续保留这个进程以运行服务线程, 服务进程不会退出。
int main(int argc, char** argv) {
SERVICE_TABLE_ENTRY entry;
entry.lpServiceName = L"MemoryStatus";
entry.lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain; // Start the control dispatcher thread for our service StartServiceCtrlDispatcher(&entry); return 0;
}
8 服务安装以及测试, 使用sc create命令可以安装服务:
REM 服务名称必须与代码中的服务名称保持一致 SC CREATE MemoryStatus binPath= %PROJECT_OUTPUT_DIR%\MemoryStatus.exe
注意: 服务名称必须与代码中的服务名称保持一致, binPath= 之后必须保留一个空格!
服务的启动与关闭可以使用NET命令:
NET START MemoryStatus
过几秒钟之后在关闭服务:
NET STOP MemoryStatus
最后, 打开日志文件, 可以看到类似下面的输出:
Monitoring started. -1446236160 -1438883840 -1437360128 Monitoring stopped.