Windows 服务中监听系统关机信号的最佳实践和指南

windows

2024-03-21 19:56:32

在 Windows 服务中监听系统关机信号

问题概述：

让 Windows 服务在系统关机时接收到通知信号至关重要，以便进行有序的关机并防止数据丢失。本文将探讨在 Windows 服务中监听系统关机信号的最佳实践和分步指南。

注册服务控制处理程序

首先，我们需要注册一个服务控制处理程序，该处理程序将在系统发送关机信号时被调用。我们使用 RegisterServiceCtrlHandlerExW 函数来完成此任务。

设置服务状态

在服务控制处理程序中，我们需要根据接收到的信号设置服务状态。这包括处理以下不同类型的信号：

停止信号： 当服务被请求停止时。
预关机信号： 在系统准备关闭之前发送。
关机信号： 在系统即将关闭之前发送。

处理关机逻辑

在服务控制处理程序中，我们还需要实现一个 shutdown 函数来执行必要的关机操作。这个函数应返回一个布尔值，表示关机操作是否已完成。

示例代码

以下是一个示例代码片段，展示如何在 Windows 服务中监听系统关机信号：

DWORD HandlerEx(DWORD dwControl, DWORD dwEventType, void* lpEventData, void* lpContext)
{
    SERVICE_STATUS status;
    status.dwCurrentState = SERVICE_RUNNING;
    status.dwWaitHint = 0;

    switch (dwControl)
    {
        case SERVICE_CONTROL_STOP:
            status.dwCurrentState = SERVICE_STOP_PENDING;
            status.dwWaitHint = 20000;
            SetServiceStatus(status_handle, &status);
            return NO_ERROR;
        case SERVICE_CONTROL_PRESHUTDOWN:
            status.dwCurrentState = SERVICE_STOP_PENDING;
            status.dwWaitHint = 20000;
            SetServiceStatus(status_handle, &status);
            return NO_ERROR;
        case SERVICE_CONTROL_SHUTDOWN:
            if (worker->shutdown())
            {
                status.dwCurrentState = SERVICE_STOPPED;
                SetServiceStatus(status_handle, &status);
            }
            return NO_ERROR;
        default:
            return ERROR_CALL_NOT_IMPLEMENTED;
    }
}