基于BSD Socket的服务器误判Socket关闭问题排查

你的服务器错误关闭连接的核心原因是混淆了TCP流套接字的行为逻辑，并且错误处理了select的读就绪事件，以下是具体分析和修正方案：

核心错误点 #

1. 对TCP套接字select读就绪的理解错误
   对于TCP流套接字，select触发读事件的场景只有三种：

   • 套接字接收缓冲区有数据可读
   • 对方发送FIN包（正常关闭连接）
   • 套接字发生错误（如收到RST包）
     不存在“连接存活但当前无数据”的触发场景——如果连接正常且没有数据，select根本不会触发读事件。

2. 冗余且错误的连接关闭检测逻辑
   你用ioctl(sd, FIONREAD, &count)判断是否有数据，再用recv(MSG_PEEK)检测连接状态的逻辑完全多余：

   • 当FIONREAD返回0且select触发了读事件，这已经明确是对方发送了FIN包，此时直接调用recv会返回0，不需要用MSG_PEEK。
   • MSG_PEEK仅用于查看数据而不消费，对检测连接关闭毫无帮助——TCP下recv返回0的唯一含义就是对方关闭了连接，和是否使用MSG_PEEK无关。

3. 未处理部分到达的数据
   当已接收数据量小于expected_message_size时，你的代码什么都不做，导致数据一直留在接收缓冲区，select会持续触发读事件，不仅浪费CPU，还可能引发后续逻辑异常。

修正后的代码示例 #

#include <errno.h>
#include <unistd.h>
#include <stdint.h>

// 假设你有会话缓冲区管理的辅助函数
void append_to_session_buffer(int sd, uint8_t* data, size_t len);
size_t session_buffer_length(int sd);
void process_full_message(int sd);
void remove_processed_data(int sd, size_t len);

#define expected_message_size 128 // 根据你的实际报文大小调整
#define BUFFER_SIZE 1024

if (FD_ISSET(sd, &readfds)) {
    uint8_t buffer[BUFFER_SIZE];
    ssize_t count = recv(sd, buffer, sizeof(buffer), 0);
    
    if (count == 0) {
        // 对方正常关闭连接，关闭套接字
        close(sd);
    } else if (count == -1) {
        // 处理错误：非阻塞模式下的暂时无数据无需关闭，其他错误直接关闭
        if (errno != EAGAIN && errno != EWOULDBLOCK) {
            close(sd);
        }
    } else {
        // 将读取到的数据追加到会话缓冲区
        append_to_session_buffer(sd, buffer, count);
        
        // 循环检查是否有足够数据处理完整报文
        while (session_buffer_length(sd) >= expected_message_size) {
            process_full_message(sd);
            remove_processed_data(sd, expected_message_size);
        }
    }
}

关键说明 #

• 直接用recv读取数据：无需提前用ioctl查询数据量，recv会直接返回实际读取到的字节数，效率更高且逻辑清晰。
• 累积部分数据：收到部分报文时，必须将数据存入会话缓冲区，直到累积到足够长度再处理完整报文，避免数据滞留在内核缓冲区导致select反复触发。
• 区分TCP与UDP行为：你引用的man recv中关于零长度数据报的内容仅适用于UDP等数据报套接字，TCP是流协议，不存在零长度报文，recv返回0就代表连接正常关闭。

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