Windows下用纯C实现进程自终止后定时重启的技术问询

嘿，我之前在Windows上用纯C做过几乎一样的需求，确实没有Linux下fork那么省心，但有几个靠谱的方案能搞定，给你详细说说：

这个思路是让主程序在收到终止指令时先留下重启标记，然后退出；同时用一个极小的辅助程序负责倒计时，时间到了就检查标记、重启主程序并清理标记。

步骤1：主程序处理终止指令，写入重启标记 #

当你的主程序收到终止信号（比如自定义的控制台命令、窗口关闭消息WM_CLOSE），先创建一个标记文件记录重启原因，再退出：

#include <stdio.h>
#include <stdlib.h>

void handle_termination() {
    // 创建重启标记文件，写入原因
    FILE* flag_file = fopen("restart_triggered.txt", "w");
    if (flag_file) {
        fprintf(flag_file, "Restart caused by manual termination command\n");
        fclose(flag_file);
    }
    // 正常退出程序
    exit(EXIT_SUCCESS);
}

步骤2：写一个轻量的重启辅助程序 #

这个程序只做三件事：等指定时间、检查标记、启动主程序。核心代码如下：

#include <windows.h>
#include <stdio.h>
#include <stdlib.h>

int main(int argc, char* argv[]) {
    if (argc != 3) {
        printf("Usage: restart_helper.exe <timeout_seconds> <main_exe_path>\n");
        return EXIT_FAILURE;
    }

    // 解析参数：超时时间、主程序路径
    int timeout_sec = atoi(argv[1]);
    const char* main_exe = argv[2];

    // 等待指定超时时间
    Sleep(timeout_sec * 1000);

    // 检查重启标记是否存在
    if (GetFileAttributesA("restart_triggered.txt") != INVALID_FILE_ATTRIBUTES) {
        // 启动主程序
        STARTUPINFOA si = {0};
        PROCESS_INFORMATION pi = {0};
        si.cb = sizeof(STARTUPINFOA);
        si.dwFlags = STARTF_USESHOWWINDOW;
        si.wShowWindow = SW_SHOW; // 根据你的需求设置窗口显示状态

        if (CreateProcessA(main_exe, NULL, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi)) {
            // 启动成功，删除标记文件
            DeleteFileA("restart_triggered.txt");
            // 关闭进程句柄，让主程序独立运行
            CloseHandle(pi.hProcess);
            CloseHandle(pi.hThread);
        }
    }

    return EXIT_SUCCESS;
}

步骤3：主程序启动时自动拉起辅助程序 #

主程序一开始就启动这个辅助程序，传递超时时间和自己的路径，确保辅助程序在后台独立运行：

#include <windows.h>
#include <stdio.h>

void start_restart_helper(int timeout_sec) {
    char main_exe_path[MAX_PATH];
    // 获取当前主程序的绝对路径
    GetModuleFileNameA(NULL, main_exe_path, MAX_PATH);

    // 构造辅助程序的命令行参数
    char cmd_line[512];
    sprintf(cmd_line, "restart_helper.exe %d \"%s\"", timeout_sec, main_exe_path);

    STARTUPINFOA si = {0};
    PROCESS_INFORMATION pi = {0};
    si.cb = sizeof(STARTUPINFOA);
    si.dwFlags = STARTF_USESHOWWINDOW;
    si.wShowWindow = SW_HIDE; // 隐藏辅助程序的控制台窗口

    // 启动辅助程序，不继承主程序的句柄
    CreateProcessA(NULL, cmd_line, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);

    // 关闭句柄，让辅助程序脱离主程序的生命周期
    CloseHandle(pi.hProcess);
    CloseHandle(pi.hThread);
}

// 在主程序入口调用
int main() {
    // 先检查是否是重启过来的，记录日志
    check_restart_reason();
    // 启动重启辅助
    start_restart_helper(60); // 超时60秒重启

    // 主程序的业务逻辑...

    return 0;
}

步骤4：主程序启动时检查重启原因 #

主程序启动后，检查标记文件是否存在（如果存在说明是主动重启），把原因记录到日志：

void check_restart_reason() {
    FILE* flag_file = fopen("restart_triggered.txt", "r");
    if (flag_file) {
        char reason[256];
        fgets(reason, sizeof(reason), flag_file);
        fclose(flag_file);

        // 写入日志文件
        FILE* log_file = fopen("app_log.txt", "a");
        if (log_file) {
            fprintf(log_file, "[%s] Restarted with reason: %s", __TIME__, reason);
            fclose(log_file);
        }

        // 保险起见，再次删除标记（辅助程序可能已经删了）
        DeleteFileA("restart_triggered.txt");
    }
}

如果你的程序是后台服务，可以利用Windows服务的机制来实现：

1. 将程序注册为Windows服务，在ServiceMain函数中处理SERVICE_CONTROL_STOP控制请求：收到停止指令时，先在注册表或标记文件中写入重启原因，然后返回NO_ERROR让服务管理器停止服务。
2. 同时，在服务启动时创建一个定时器线程，等待指定超时时间后，调用StartService API重启自己；或者直接在服务属性的“恢复”选项中配置重启，但这种方式没法自定义记录重启原因，还是需要结合标记文件。

• 路径问题：确保主程序和辅助程序在同一目录，或者使用绝对路径传递参数，避免找不到文件。
• 权限问题：如果主程序需要管理员权限，辅助程序也需要以相同权限启动。
• 避免无限重启：可以在标记文件中加入重试次数，超过次数就不再重启，防止死循环。

内容的提问来源于stack exchange，提问作者S.Mori