客户端断开连接时套接字服务器崩溃

Question

练习 C 语言和套接字编程。

服务器代码在检查实时客户端并且客户端已断开连接（例如连接断开）时崩溃。我没有阅读 GDB 转储的经验。有人可以指出我在这里缺少什么吗？

这是服务器代码。客户端连接和断开连接。服务器通过发送一条小消息并等待回复来检查断开连接的客户端。收到损坏的管道错误后，代码崩溃。

void * client_hartbeat()
{
    int ret = 0, i;
    char send_msg[50] = "e";
    char recv_msg[50];
    while(1)
    {
        sleep(3);
        printf("\nchecking for disconnected clients\n");
        memset(&recv_msg,0,sizeof(recv_msg));
        for(i=0;i<CLIENTS;i++){
            if(client_pool[i]!=0){
                printf("TEST0, socket: %i thread: %i\n", client_pool[i],pthread_self());
                if(send(client_pool[i],send_msg,sizeof(send_msg),0) < 0){
                    printf("send error: %s\n", strerror(errno));
                }else if(recv(client_pool[i],recv_msg,sizeof(recv_msg),0) < 0){
                    printf("receive error: %s\n", strerror(errno));
                }
                printf("TEST1, socket: %i thread: %d\n", client_pool[i],pthread_self());
            }
        }
    }
    printf("TEST1\n");
    pthread_exit(&th2);
}

这是 GDB 转储：

[Thread debugging using libthread_db enabled]
Using host libthread_db library "/lib/x86_64-linux-gnu/libthread_db.so.1".
waiting connections...
[New Thread 0x7ffff7dba640 (LWP 22567)]
[New Thread 0x7ffff75b9640 (LWP 22568)]

checking for disconnected clients

checking for disconnected clients
TEST0, socket: 4 thread: -144992704
TEST1, socket: 4 thread: -144992704

checking for disconnected clients
TEST0, socket: 4 thread: -144992704
TEST1, socket: 4 thread: -144992704

checking for disconnected clients
TEST0, socket: 4 thread: -144992704

Thread 3 "a.out" received signal SIGPIPE, Broken pipe.
[Switching to Thread 0x7ffff75b9640 (LWP 22568)]
__libc_send (flags=<optimized out>, len=50, buf=0x7ffff75b8e10, fd=4) at ../sysdeps/unix/sysv/linux/send.c:28
28      ../sysdeps/unix/sysv/linux/send.c: No such file or directory.
(gdb)

Answer 1

这些选项中的任何一个都应该可以避免SIGPIPE 信号杀死您的程序：

• 忽略SIGPIPE：

  #include <signal.h>
  
  int main() {
      signal(SIGPIPE, SIG_IGN);
      ...
  

• 当send/recv 尝试通过提供MSG_NOSIGNAL 标志对关闭的套接字进行操作时，不要生成SIGPIPE 信号：

  if (send(client_pool[i], send_msg, strlen(send_msg), MSG_NOSIGNAL) < 0) {
      printf("send error: %s\n", strerror(errno));
  } else if (recv(client_pool[i], recv_msg, sizeof(recv_msg), MSG_NOSIGNAL) < 0) {
      printf("receive error: %s\n", strerror(errno));
  }
  

来自sendman 页面：

MSG_NOSIGNAL（Linux 2.2 起）

如果面向流的套接字上的对等方已关闭连接，则不要生成SIGPIPE 信号。仍然返回 EPIPE 错误。这提供了与使用sigaction(2) 来忽略SIGPIPE 类似的行为，但是，虽然MSG_NOSIGNAL 是一个每次调用的功能，但忽略SIGPIPE 会设置一个影响进程中所有线程的进程属性。 p>

---

您还需要在使用send 后检测套接字的状态，因为它可能会在客户端死亡后的第一个send 上返回成功。

你可以添加一个辅助函数：

int checksock(int sock) { // 0 on success, -1 on failure
    int error = 0;
    socklen_t len = sizeof error;

    int rv = getsockopt(sock, SOL_SOCKET, SO_ERROR, &error, &len);
    if (rv == 0 && error) {
        errno = error;     // set the thread local errno
        rv = -1;
    }
    return rv;
}

然后您的发送部分可以这样做：

if (send(client_pool[i], send_msg, strlen(send_msg), MSG_NOSIGNAL) < 0
    || checksock(client_pool[i]))     // added check
{
   printf("send error: %s\n", strerror(errno));   // Broken pipe
} else if (recv(client_pool[i], recv_msg, sizeof(recv_msg), MSG_NOSIGNAL) < 0) {
   printf("receive error: %s\n", strerror(errno));
}