linux网络通信屏蔽SIG_PIPE信号

在Linux系统中，SIGPIPE信号通常在向一个已经关闭的管道或套接字写数据时触发。默认情况下，这个信号会导致进程终止。在某些网络编程或IPC（进程间通信）场景中，接收和处理SIGPIPE信号可能并不符合预期。因此，屏蔽或忽略SIGPIPE信号是一个常见的做法。

屏蔽SIGPIPE信号的几种方法

方法一：在代码中忽略SIGPIPE信号

可以使用signal或sigaction函数在代码中忽略SIGPIPE信号。

使用signal函数：

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

int main() {
    // 忽略SIGPIPE信号
    signal(SIGPIPE, SIG_IGN);

    // 示例代码，创建一个管道并关闭读端
    int pipefd[2];
    if (pipe(pipefd) == -1) {
        perror("pipe");
        return 1;
    }

    close(pipefd[0]); // 关闭读端

    // 向管道写数据（会触发SIGPIPE信号）
    if (write(pipefd[1], "data", 4) == -1) {
        perror("write");
    }

    close(pipefd[1]); // 关闭写端
    return 0;
}

使用sigaction函数：

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

int main() {
    struct sigaction sa;
    memset(&sa, 0, sizeof(sa));
    sa.sa_handler = SIG_IGN; // 忽略信号
    sigaction(SIGPIPE, &sa, NULL);

    // 示例代码，创建一个管道并关闭读端
    int pipefd[2];
    if (pipe(pipefd) == -1) {
        perror("pipe");
        return 1;
    }

    close(pipefd[0]); // 关闭读端

    // 向管道写数据（会触发SIGPIPE信号）
    if (write(pipefd[1], "data", 4) == -1) {
        perror("write");
    }

    close(pipefd[1]); // 关闭写端
    return 0;
}

方法二：在套接字上设置SO_NOSIGPIPE选项（仅限于BSD和macOS）

在BSD和macOS系统上，可以在套接字上设置SO_NOSIGPIPE选项来防止产生SIGPIPE信号。然而，这个选项在Linux系统上不可用。

方法三：在send或sendto函数中使用MSG_NOSIGNAL标志

当使用send或sendto函数发送数据时，可以通过传递MSG_NOSIGNAL标志来避免触发SIGPIPE信号。

#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <arpa/inet.h>

int main() {
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        perror("socket");
        return 1;
    }

    struct sockaddr_in server_addr;
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(8080);
    server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");

    // 连接到服务器
    if (connect(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
        perror("connect");
        close(sockfd);
        return 1;
    }

    // 关闭服务器端以模拟关闭连接
    // 发送数据并使用MSG_NOSIGNAL标志
    const char *msg = "Hello, server";
    if (send(sockfd, msg, strlen(msg), MSG_NOSIGNAL) == -1) {
        perror("send");
    }

    close(sockfd);
    return 0;
}

总结

• 忽略SIGPIPE信号： 使用signal或sigaction函数忽略SIGPIPE信号。
• MSG_NOSIGNAL标志： 在使用send或sendto函数发送数据时传递MSG_NOSIGNAL标志。
• SO_NOSIGPIPE选项： 仅在BSD和macOS系统上可用，通过在套接字上设置此选项避免SIGPIPE信号（Linux系统不可用）。

通过这些方法，你可以有效地屏蔽或忽略SIGPIPE信号，避免因为向已关闭的管道或套接字写数据而导致的进程终止。