Unix环境下如何程序化判断工具例程是否由信号处理程序调用？

你遇到的这个场景其实挺常见的——写一个工具例程，既要给常规多线程代码调用，又得能被信号处理程序调用，输出调试信息还得分情况选IO方式：常规场景用线程友好的printf/fprintf，信号处理场景只能用sprintf+write。那有没有程序化的方法能自动识别当前是哪种调用场景呢？下面给你几个可行的方案：

一、用线程本地存储（TLS）手动标记信号上下文 #

这是移植性最好的方案，思路很简单：

• 先定义一个线程本地的volatile变量，比如static volatile __thread int in_signal_ctx = 0;（__thread是GCC的线程本地存储关键字，Windows平台可替换为__declspec(thread)）。
• 在你的信号处理程序入口，先把这个变量设为1，处理完信号逻辑后再把它清零。
• 你的工具例程里先检查这个变量的值：如果是1，就说明当前在信号处理上下文，用sprintf+write输出；否则就用printf/fprintf。
• 小提醒：一定要加volatile，防止编译器把变量优化掉；如果担心信号嵌套（比如信号处理过程中又触发了另一个信号），可以把变量改成计数器——进入信号处理加1，退出减1，这样嵌套场景也能正确判断。

二、利用平台原生接口检测信号上下文 #

不同操作系统有一些特定接口可以直接判断当前是否处于信号处理中，但移植性较差：

• Linux：可以借助Glibc的内部函数__is_signal_frame（需包含<signal.h>），或者通过getcontext获取当前上下文，检查uc_mcontext里的信号标记。不过这些接口属于未公开的内部实现，版本更新后可能失效。
• BSD/macOS：可以通过thread_self()结合线程属性查询，或者利用sigaction的SA_SIGINFO标志传递的信号上下文信息间接判断。
• 总结：除非你只针对某一个平台开发，否则优先用第一种TLS标记的方法。

三、直接封装兼容两种场景的安全输出函数 #

其实最省心的方法是不用判断上下文，直接实现一个全场景安全的调试输出函数：
POSIX标准规定write、vsnprintf（多数主流平台实现都符合）是信号安全的，所以可以封装一个通用函数：

#include <stdio.h>
#include <unistd.h>
#include <stdarg.h>

void safe_debug_log(const char *fmt, ...) {
    // 用固定大小缓冲区，避免在信号处理中调用malloc
    char buf[2048];
    va_list args;
    va_start(args, fmt);
    // vsnprintf是信号安全的，格式化内容到缓冲区
    int write_len = vsnprintf(buf, sizeof(buf), fmt, args);
    va_end(args);
    // write是信号安全的，直接输出到标准错误
    write(STDERR_FILENO, buf, (write_len > sizeof(buf)-1) ? sizeof(buf)-1 : write_len);
}

这个函数不管是在常规多线程代码里，还是在信号处理程序里调用都安全，不用纠结上下文判断，一步到位。

关键注意事项 #

• 绝对不要在信号处理程序里调用非信号安全的函数，比如printf、malloc、free——这些函数内部可能持有全局锁，信号触发时如果常规线程已经持有锁，会直接导致死锁。
• 如果用TLS标记方案，要确保变量的修改是原子操作（对于int类型的赋值，多数平台默认是原子的，不用额外处理）。
• 封装安全输出函数时，固定缓冲区的大小要根据你的调试信息长度调整，避免内容截断；如果需要输出超长内容，可以分段循环输出。

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