如何在无法控制线程启动代码时，为应用所有线程安装终止处理函数？

这确实是个棘手的场景——没法修改线程启动代码，但又要给所有线程挂上自定义终止处理函数。在Windows平台上，我们可以借助系统钩子机制来实现这个需求，确保不管是你自己创建的线程还是第三方库启动的线程，都会自动设置好std::set_terminate。

核心思路 #

Windows的WH_THREAD钩子可以拦截进程内所有线程的创建事件。我们可以利用这个钩子，在新线程创建完成后，通过APC（异步过程调用）在新线程的上下文中调用std::set_terminate，为其绑定自定义的终止处理函数。

具体实现步骤 #

1. 定义自定义终止处理函数 #

首先写好你想要在线程异常终止时执行的逻辑：

#include <iostream>
#include <cstdlib>

void CustomTerminateHandler() {
    // 这里写你的终止处理逻辑：比如打日志、清理资源、上报监控等
    std::cerr << "[Terminate Handler] Thread terminated unexpectedly!" << std::endl;
    // 按照CRT的约定，终止处理函数最后通常需要调用std::abort()
    std::abort();
}

2. 编写线程钩子处理函数 #

这个函数会在线程创建时被触发，我们通过APC将设置终止函数的逻辑注入到新线程中：

#include <windows.h>

LRESULT CALLBACK ThreadHookProc(int nCode, WPARAM wParam, LPARAM lParam) {
    // 只处理线程创建事件
    if (nCode == THCBT_CREATE) {
        // lParam是新创建线程的句柄
        HANDLE hNewThread = reinterpret_cast<HANDLE>(lParam);
        
        // 使用QueueUserAPC在新线程的上下文中执行set_terminate
        QueueUserAPC(
            reinterpret_cast<PAPCFUNC>([](ULONG_PTR) {
                std::set_terminate(CustomTerminateHandler);
            }),
            hNewThread,
            0
        );
    }
    
    // 一定要调用下一个钩子，保证系统正常运行
    return CallNextHookEx(nullptr, nCode, wParam, lParam);
}

3. 在程序初始化时安装钩子 #

在程序启动的早期（比如main函数开头）安装钩子，确保后续所有线程都能被拦截：

int main() {
    // 安装WH_THREAD钩子，作用于当前进程的所有线程
    HHOOK hThreadHook = SetWindowsHookEx(
        WH_THREAD,
        ThreadHookProc,
        GetModuleHandle(nullptr),  // 当前模块句柄（无需单独DLL）
        0                           // 0表示钩子作用于所有线程
    );
    
    if (!hThreadHook) {
        std::cerr << "Failed to install thread hook! Error: " << GetLastError() << std::endl;
        return 1;
    }

    // 这里是你的业务逻辑，包括第三方库启动线程的代码
    // ...

    // 程序退出前卸载钩子
    UnhookWindowsHookEx(hThreadHook);
    return 0;
}

关键注意事项 #

• CRT兼容性：确保你的程序和所有依赖库使用相同版本的CRT（比如都是MT或MD），因为std::set_terminate依赖CRT的线程本地存储（TLS）实现，版本不匹配会导致未定义行为。
• 钩子作用范围：这个钩子只会作用于当前进程内的线程，如果你需要跨进程拦截，才需要将钩子逻辑放到单独的DLL中。
• APC执行时机：QueueUserAPC会在新线程进入可等待状态时执行，确保在线程执行用户代码前完成终止函数的设置。
• 标准语义说明：虽然C++标准中std::set_terminate是全局的，但Windows CRT的实现将其设计为线程本地的，所以必须为每个线程单独设置——这也是我们需要钩子的核心原因。

内容的提问来源于stack exchange，提问作者tangens