远程关闭Socket后阻塞recv调用无法终止的技术问询

问题背景 #

在Socket编程场景中，当远程端关闭Socket时，本地处于阻塞状态的recv调用无法自动终止；即使调用close或shutdown关闭本地Socket，线程仍会卡在recv上无法正常退出。已尝试::shutdown(m_socketId, SHUT_RDWR)、切换非阻塞IO、设置SO_RCVTIMEO等方案，但均不符合需求，希望在尽量保留现有架构的前提下解决问题。

可行解决方案 #

1. 向阻塞线程发送信号打断recv #

利用POSIX线程的pthread_kill向处于recv阻塞的线程发送自定义信号（如SIGUSR1），触发recv返回错误，从而让线程退出循环。

• 实现步骤：
  • 在recv线程初始化时注册空信号处理函数：

    #include <signal.h>
    #include <pthread.h>
    
    void sig_handler(int sig) {}
    
    // 线程启动时执行
    signal(SIGUSR1, sig_handler);
    

  • 在disConnect函数中，向目标recv线程发送信号：

    pthread_kill(recv_thread_id, SIGUSR1);
    

  • 修改recv循环逻辑，处理信号中断错误：

    ssize_t ret;
    while (true) {
        ret = recv(m_socketId, buf, sizeof(buf), 0);
        if (ret == -1) {
            if (errno == EINTR) {
                // 被信号打断，退出循环
                break;
            }
            // 处理其他错误
            break;
        } else if (ret == 0) {
            // 远程正常关闭连接
            break;
        }
        // 处理业务数据
    }
    

• 注意：Windows平台可使用QueueUserAPC替代pthread_kill实现类似逻辑，确保信号仅作用于目标线程。

2. 设置SO_LINGER选项强制重置连接 #

通过配置SO_LINGER选项，让close调用时立即向对方发送RST包，触发对方recv返回ECONNRESET错误，从而退出阻塞。

• 实现代码：

  #include <sys/socket.h>
  
  void disConnect() {
      struct linger l_opt;
      l_opt.l_onoff = 1;
      l_opt.l_linger = 0; // 立即重置连接，不等待未发送数据
      setsockopt(m_socketId, SOL_SOCKET, SO_LINGER, &l_opt, sizeof(l_opt));
      close(m_socketId);
  }
  

• 注意：该方式会直接断开连接，未发送完成的数据可能丢失，适合无需保证数据完整性的场景；本地recv线程也会因Socket关闭，在后续调用中返回错误。

3. 发送自定义终止标记触发recv退出 #

在调用disConnect前，向本地Socket发送一个约定的终止数据包（如空字节），让recv读取到后主动退出循环。

• 实现步骤：
  • 在disConnect中添加终止标记发送逻辑：

    // 发送空字节作为终止信号
    send(m_socketId, "", 1, 0);
    

  • 修改recv逻辑，识别终止标记：

    ssize_t ret;
    while (true) {
        ret = recv(m_socketId, buf, sizeof(buf), 0);
        if (ret == -1) {
            // 处理错误
            break;
        } else if (ret == 0 || (ret == 1 && buf[0] == '\0')) {
            // 远程关闭或收到终止标记，退出循环
            break;
        }
        // 处理正常业务数据
    }
    

• 注意：需确保终止标记不会与正常业务数据冲突，该方案改动极小，仅需在recv逻辑中增加判断。

内容的提问来源于stack exchange，提问作者bourne