信号（Signal）深入：自定义信号处理

信号（Signal）深入：自定义信号处理

🚀 掌握信号处理的艺术，为你的应用程序增添优雅的响应机制

信号（Signal）是Unix-like操作系统中进程间通信的基本机制之一，它允许进程在特定事件发生时接收通知并采取相应行动。从基本的SIGINT（中断信号）到自定义信号，合理利用信号能显著提升程序的健壮性与灵活性。本文将深入探讨信号的工作原理，并演示如何自定义信号处理。

信号基础 📡

信号是发送给进程的异步通知，通常用于通知进程发生了某种事件。例如，当用户在终端按下Ctrl+C时，会向当前前台进程发送SIGINT信号，默认情况下这会终止该进程。信号可以来自内核、其他进程，或进程自身。

常见的信号包括：

• SIGINT：中断信号，通常由Ctrl+C触发
• SIGTERM：终止信号，请求进程正常退出
• SIGKILL：强制终止信号，不能被捕获或忽略
• SIGSEGV：段错误信号，表示无效内存访问

信号处理机制 ⚙️

每个信号都有一个默认处理方式，通常是终止进程、忽略信号或暂停进程。然而，我们可以通过注册信号处理函数（signal handler）来自定义信号的处理逻辑。这通过signal()或更现代的sigaction()系统调用来实现。

下面是一个简单的例子，演示如何捕获SIGINT信号并自定义处理行为：

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

void handle_sigint(int sig) {
    printf("捕获到信号 %d！自定义处理中...\n", sig);
}

int main() {
    signal(SIGINT, handle_sigint);
    while (1) {
        printf("程序运行中...\n");
        sleep(1);
    }
    return 0;
}

编译并运行上述程序后，尝试按下Ctrl+C，你会看到自定义的消息而不是立即终止进程。要终止程序，你可以使用Ctrl+\（发送SIGQUIT）或另开终端用kill命令结束它。

深入sigaction() 🛠️

虽然signal()简单易用，但sigaction()提供了更强大和灵活的信号处理控制。它允许你精确指定处理函数、阻塞其他信号 during handling、并获取信号的额外信息。

下面是一个使用sigaction()的示例：

#include <stdio.h>
#include <signal.h>
#include <unistd.h>
#include <string.h>

void handler(int sig, siginfo_t *info, void *context) {
    printf("收到信号 %d，发送者PID：%d\n", sig, info->si_pid);
}

int main() {
    struct sigaction sa;
    memset(&sa, 0, sizeof(sa));
    sa.sa_sigaction = handler;
    sa.sa_flags = SA_SIGINFO; // 启用详细信号信息

    if (sigaction(SIGINT, &sa, NULL) == -1) {
        perror("sigaction");
        return 1;
    }

    while (1) {
        pause(); // 等待信号
    }
    return 0;
}

此程序不仅捕获SIGINT，还能获取发送者的进程ID（虽然SIGINT通常来自终端，但kill命令也可发送）。SA_SIGINFO标志启用了更丰富的信号信息传递。

信号集与阻塞 🛑

有时，你可能希望临时阻塞某些信号，防止它们在关键代码段中被处理。sigprocmask()允许我们修改当前阻塞的信号集。下面示例演示如何阻塞SIGINT during一段"关键操作"：

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

void handle_sigint(int sig) {
    printf("信号 %d 处理中\n", sig);
}

int main() {
    signal(SIGINT, handle_sigint);
    sigset_t newset, oldset;
    sigemptyset(&newset);
    sigaddset(&newset, SIGINT);

    // 阻塞SIGINT
    sigprocmask(SIG_BLOCK, &newset, &oldset);

    printf("关键操作开始——SIGINT被阻塞\n");
    sleep(5); // 模拟关键操作
    printf("关键操作结束\n");

    // 解除阻塞，之前pending的信号会被立即传递
    sigprocmask(SIG_SETMASK, &oldset, NULL);

    while (1) {
        pause();
    }
    return 0;
}

如果在5秒睡眠期间按下Ctrl+C，信号会被阻塞，直到关键操作结束才被处理。

自定义信号 🔧

除了标准信号，Linux还提供了实时信号（SIGRTMIN to SIGRTMAX），可用于自定义目的。它们支持排队（多个相同信号可被排队处理），并且优先级高于标准信号。下面示例使用SIGRTMIN作为自定义信号：

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

#define CUSTOM_SIG SIGRTMIN

void custom_handler(int sig) {
    printf("自定义信号 %d 被接收\n", sig);
}

int main() {
    printf("进程PID：%d\n", getpid());
    signal(CUSTOM_SIG, custom_handler);

    // 发送自定义信号给自身
    raise(CUSTOM_SIG);

    // 也可以使用sigqueue发送额外数据
    union sigval value;
    value.sival_int = 42;
    sigqueue(getpid(), CUSTOM_SIG, value);

    sleep(1);
    return 0;
}

sigqueue()允许附加一个整型或指针值到信号，为自定义数据处理提供了可能。

信号处理的最佳实践 ✅

1. 保持处理函数简单：信号处理函数应尽可能简单，避免复杂操作。常见的做法是设置一个标志变量，在主循环中检查并处理。
2. 注意异步信号安全：在信号处理函数中，只能调用异步信号安全的函数（如write()，避免printf()）。
3. 避免全局冲突：使用sig_atomic_t类型标志变量，确保原子读写。
4. 考虑可重入性：如果处理函数中访问全局数据结构，需考虑信号可能中断主程序对该结构的操作，导致不一致状态。

以下示例演示了使用标志变量的安全模式：

#include <stdio.h>
#include <signal.h>
#include <unistd.h>
#include <stdatomic.h>

atomic_int signal_received = 0;

void handle_sigint(int sig) {
    signal_received = 1;
}

int main() {
    signal(SIGINT, handle_sigint);
    while (1) {
        if (signal_received) {
            printf("处理信号...\n");
            signal_received = 0;
        }
        printf("主循环运行中\n");
        sleep(1);
    }
    return 0;
}

信号与多线程 🧵

在多线程程序中，信号处理变得更加复杂。通常，建议将所有信号定向到一个专用线程进行处理，以避免竞态条件。可使用pthread_sigmask()阻塞其他线程的信号接收，然后让一个线程调用sigwait()等待信号。

#include <stdio.h>
#include <signal.h>
#include <pthread.h>
#include <unistd.h>

void* signal_thread(void* arg) {
    sigset_t set;
    int sig;
    sigemptyset(&set);
    sigaddset(&set, SIGINT);

    while (1) {
        sigwait(&set, &sig);
        printf("信号线程处理: %d\n", sig);
    }
    return NULL;
}

int main() {
    pthread_t tid;
    sigset_t set;

    // 阻塞SIGINT在主线程和后续线程
    sigemptyset(&set);
    sigaddset(&set, SIGINT);
    pthread_sigmask(SIG_BLOCK, &set, NULL);

    pthread_create(&tid, NULL, signal_thread, NULL);

    while (1) {
        printf("主线程工作...\n");
        sleep(1);
    }
    return 0;
}

此程序确保只有signal_thread会处理SIGINT，避免了多线程环境下的信号处理冲突。

Mermaid 序列图：信号处理流程 📊

下面通过Mermaid序列图展示信号从生成到处理的完整流程：

此图简要说明了信号如何从发送者经内核到达目标进程，并触发自定义处理函数。

结语 🌟

信号是Unix系统编程中强大且必不可少的工具。通过自定义信号处理，你可以构建更灵活、响应更及时的应用程序。不过，务必谨慎处理信号的异步特性，避免竞态条件和不可重入问题。希望本文助你深入理解信号机制，并在项目中有效应用！

想要进一步学习信号和高级Unix编程，可参考The Linux Programming Interface 等权威资源，或查阅官方signal man page。Happy coding! 🎉