Linux进程信号处理：如何自定义信号处理函数

在Linux中，信号是一种用于进程间通信的机制。当一个进程收到一个信号时，操作系统会中断其当前的工作，转而执行一个特定的信号处理函数。自定义信号处理函数可以帮助你处理特定的信号事件，例如捕获程序中断（SIGINT）或其他自定义信号。

以下是如何自定义信号处理函数的步骤：

1. 包含必要的头文件

首先，你需要包含处理信号所需的头文件：

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

2. 定义信号处理函数

定义一个函数来处理你感兴趣的信号。这个函数的原型应该是：

void signal_handler(int signum);

其中，signum 是接收到的信号编号。

例如，定义一个处理 SIGINT 信号的函数：

void signal_handler(int signum) {
    printf("Received signal %d\n", signum);
    // 在这里添加你的处理代码
    exit(signum); // 退出程序
}

3. 注册信号处理函数

使用 signal 或 sigaction 函数来注册你的信号处理函数。

使用 signal 函数

int main() {
    // 注册 SIGINT 信号的处理函数
    signal(SIGINT, signal_handler);

    printf("Program is running. Press Ctrl+C to interrupt.\n");

    while (1) {
        // 主程序循环
        sleep(1);
    }

    return 0;
}

使用 sigaction 函数

sigaction 提供了更灵活和安全的信号处理方式。以下是使用 sigaction 的示例：

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

void signal_handler(int signum) {
    printf("Received signal %d\n", signum);
    // 在这里添加你的处理代码
    exit(signum); // 退出程序
}

int main() {
    struct sigaction sa;

    // 清空 sa 结构体
    memset(&sa, 0, sizeof(sa));

    // 设置信号处理函数
    sa.sa_handler = signal_handler;

    // 阻塞信号（可选）
    sigemptyset(&sa.sa_mask);
    sa.sa_flags = 0;

    // 注册 SIGINT 信号的处理函数
    if (sigaction(SIGINT, &sa, NULL) == -1) {
        perror("sigaction");
        exit(EXIT_FAILURE);
    }

    printf("Program is running. Press Ctrl+C to interrupt.\n");

    while (1) {
        // 主程序循环
        sleep(1);
    }

    return 0;
}

4. 编译和运行程序

使用 gcc 编译你的程序：

gcc -o signal_example signal_example.c

然后运行程序：

./signal_example

现在，当你按下 Ctrl+C 时，程序会捕获 SIGINT 信号并执行你定义的信号处理函数。

注意事项

• signal 函数在不同的系统上可能有不同的行为，而 sigaction 提供了更一致和可靠的行为。
• 在信号处理函数中，尽量避免调用不可重入的函数，例如 printf 和 malloc。
• 信号处理函数应该尽可能简单，复杂的逻辑应该在主程序中处理。

通过以上步骤，你可以自定义信号处理函数来处理特定的信号事件。