如何判断已从非阻塞Socket读取完所有数据？

TCP是流式协议，本身不提供消息边界的划分，非阻塞模式下recv返回EAGAIN/EWOULDBLOCK仅表示当前无数据可读，并不代表消息结束。要可靠判断完整消息，必须通过应用层协议定义消息边界，以下是几种常用的实现方案：

1. 长度前缀协议（最常用） #

发送方先发送固定长度的消息头（通常是4字节无符号整数，使用网络字节序），头中包含后续消息体的总长度；接收方先读取完整的消息头，再根据头中指定的长度读取对应字节数的数据，直到满足长度要求。

代码示例（适配非阻塞场景） #

#include <stdint.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <unistd.h>

#define BUF_SIZE 1024

void process_message(char* data, uint32_t len);

void handle_nonblocking_recv(int fd) {
    static uint32_t msg_len = 0;
    static int received_header = 0;
    static char* msg_buffer = NULL;
    static int received_body = 0;

    // 第一步：读取消息头（4字节，网络字节序）
    if (!received_header) {
        int bytes = recv(fd, ((char*)&msg_len) + received_header, sizeof(msg_len) - received_header, 0);
        if (bytes > 0) {
            received_header += bytes;
            if (received_header == sizeof(msg_len)) {
                // 转换为主机字节序
                msg_len = ntohl(msg_len);
                msg_buffer = malloc(msg_len);
                if (!msg_buffer) {
                    // 内存分配失败，清理并关闭连接
                    close(fd);
                    return;
                }
            }
        } else if (bytes < 0) {
            if (errno != EAGAIN && errno != EWOULDBLOCK) {
                // 处理错误，关闭连接
                close(fd);
                return;
            }
            // 无数据可读，等待下次socket就绪
            return;
        } else {
            // 连接已关闭，未完成头读取，异常处理
            close(fd);
            return;
        }
    }

    // 第二步：读取消息体
    if (received_header == sizeof(msg_len) && received_body < msg_len) {
        int bytes = recv(fd, msg_buffer + received_body, msg_len - received_body, 0);
        if (bytes > 0) {
            received_body += bytes;
            // 检查是否读取完完整消息体
            if (received_body == msg_len) {
                process_message(msg_buffer, msg_len);
                // 重置状态，准备接收下一条消息
                free(msg_buffer);
                msg_len = 0;
                received_header = 0;
                received_body = 0;
            }
        } else if (bytes < 0) {
            if (errno != EAGAIN && errno != EWOULDBLOCK) {
                free(msg_buffer);
                close(fd);
                return;
            }
            // 无数据可读，等待下次socket就绪
            return;
        } else {
            // 连接已关闭，未完成消息体读取，异常处理
            free(msg_buffer);
            close(fd);
            return;
        }
    }
}

2. 分隔符协议 #

用特定的字符或字符序列（如\n、\r\n、自定义的END标识）标记消息结束。接收方持续读取数据，缓存已接收内容，直到检测到分隔符，此时判定消息完整。

注意事项 #

• 需处理分隔符跨缓冲区的情况：比如某次recv只读取到分隔符的一部分，需将这部分缓存，和下一次读取的内容拼接后再检测分隔符。
• 分隔符不能出现在消息体中，否则会误判；如果消息体可能包含分隔符，需先对消息体做转义处理。

3. 固定长度消息协议 #

如果所有消息的长度固定，接收方每次读取固定字节数，直到读满该长度，即判定消息完整。

适用场景 #

适合消息格式高度统一的场景（如工控指令、特定硬件通信协议），无需额外的消息头或分隔符，实现简单。

4. 应用层结束标识 #

在消息体中定义明确的结束字段（如HTTP协议的Content-Length头，或Transfer-Encoding: chunked的块结束标记），接收方解析消息内容时，根据字段判断是否到达消息末尾。

核心注意事项 #

1. 非阻塞模式下，必须结合**I/O多路复用（epoll/poll/select）**管理socket状态：每次recv返回EAGAIN/EWOULDBLOCK时，将socket放回多路复用器，等待下次可读事件触发后继续读取。
2. 必须处理连接中途关闭的情况（recv返回0）：此时若未接收完完整消息，属于异常，需及时清理资源并关闭连接。

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