Unix域套接字编程

Question

我有一个unix域套接字程序，客户端尝试连接到服务器并发送消息，当服务器接受客户端并读取消息时，它将休眠5秒钟并发送另一条消息。在5秒钟内，如果我使用ctrl + c杀死客户端，然后服务器将退出。如何处理这种情况？我的程序如下：

客户：

#include <unistd.h>
#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/un.h>
#include <string.h>
#include <assert.h>
#include <stddef.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <stdint.h>

#define INFO_SERVER_PATH "/var/info_server_path"


int create_route_client()
{
    int client_fd;
    int addr_len;

    struct sockaddr_un server_addr;

    if ((client_fd = socket(AF_LOCAL, SOCK_STREAM, 0)) < 0) {
        perror("create route info client socket");
        return -1;
    }

    memset(&server_addr, 0, sizeof(struct sockaddr_un));
    server_addr.sun_family = AF_LOCAL;
    strcpy(server_addr.sun_path, INFO_SERVER_PATH);
    addr_len = offsetof(struct sockaddr_un,sun_path) + strlen(server_addr.sun_path);

    if (connect(client_fd, (struct sockaddr*)&server_addr, addr_len) < 0) {
        perror("socket connect");
        return -1;
    }

    return client_fd;
}


int main(int argc, char const *argv[])
{

    char *sendline = "hello server";
    char recvline[512];
    int client_fd;
    int nwrite;
    int nread;

    client_fd = create_route_client();
    assert(client_fd > 0);

    nwrite = write(client_fd, sendline, strlen(sendline));
    if (nwrite < 0) {
        perror("failed to send command to the info server");
        close(client_fd);
        return 1;
    }

    nread = read(client_fd, recvline, sizeof(recvline));
    if (nread < 0) {
        perror("failed to read route state");
        close(client_fd);
        return 1;
    }
    recvline[nread] = '\0';
    printf("%s\n", recvline);
    close(client_fd);

    return 0;
}

服务器：

#include <unistd.h>
 #include <stdio.h>
 #include <sys/socket.h>
 #include <sys/types.h>
 #include <sys/un.h>
 #include <string.h>
 #include <assert.h>
 #include <stddef.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <stdint.h>

 #define INFO_SERVER_PATH "/var/info_server_path"
 int create_command_server()
 {
    struct sockaddr_un server_addr;
    size_t addr_len;
    int server_fd;

    if ((server_fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
        perror("create socket");
        return -1;
    }
    unlink(INFO_SERVER_PATH);

    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sun_family = AF_UNIX;
    strcpy(server_addr.sun_path, INFO_SERVER_PATH);
    addr_len = offsetof(struct sockaddr_un, sun_path) + strlen(INFO_SERVER_PATH);

    if (bind(server_fd, (struct sockaddr*)&server_addr, addr_len) < 0) {
        perror("socket bind");
        return -1;
    }

    if (listen(server_fd, 1) < 0) {
        perror("socket listen");
        return -1;
    }

    return server_fd;
}

int main(int argc, char const *argv[])
{
    int info_server_fd = create_command_server();
    char recvline[512];
    char *sendline = "hello client";
    int nread;
    int nwrite;

    while (1) {
        int info_client_fd = accept(info_server_fd, NULL, NULL);
        nread = read(info_client_fd, recvline, 512);

        if (nread) {
            int i;
            for (i = 0; i < 5; i++) {
                printf("i = %d\n", i);
                sleep(1);
            }
            nwrite = write(info_client_fd , sendline, strlen(sendline));
            printf("nwrite = %d\n", nwrite);
            if (nwrite < 0) 
                perror("failed to send to client");

        }
        close(info_client_fd);
    }


    return 0;
}

Answer 1

在不编译和运行代码的情况下很难说清楚，但是我猜您正在收到SIGPIPE信号，原因是写入了杀死客户端时关闭的连接。 接收SIGPIPE的进程的默认操作是终止该进程。

您可以使用sigprocmask（）阻止SIGPIPE信号，或者使用sigaction（）告诉内核您要阻止它，忽略它或注册一个异步信号处理程序。 然后，当您调用write（）时，它将返回-1并将errno设置为EPIPE。 请参见手册页以获取write（） http://man7.org/linux/man-pages/man2/write.2.html 。

有关信号及其处理方法的更多信息，请参见手册中的信号http://man7.org/linux/man-pages/man7/signal.7.html 。 但是，请注意 ，除非您非常非常小心并且确切知道您在做什么，否则不应使用异步信号处理程序来处理信号。 这是许多错误的根源。 到目前为止，最安全的做法是忽略它们（如果不需要），或者阻塞它们并使用同步信号处理方法，例如sigwait（）或signal fd（特定于Linux）。 就您而言，您不需要它们。 当连接断开时，write（）调用将告诉您。