Linux下如何等待TCP Socket消息发送并被对等端确认？

针对你提到的嵌入式设备固件升级后需确认消息送达再重启的场景，Linux下有几种可靠的方式实现需求：

1. 利用shutdown(SHUT_WR) + read等待连接关闭 #

这是TCP层面最稳妥的方法，原理是通过关闭写方向触发TCP的FIN握手，直到对等端回应FIN，确认所有数据已被接收：

• 步骤：
  1. 发送完所有目标消息后，调用shutdown(sockfd, SHUT_WR)，此时Socket不再接受本地写操作，TCP会将发送缓冲区剩余数据全部发出，随后向对等端发送FIN包。
  2. 循环调用read读取Socket，直到read返回0（表示收到对等端的FIN包）或出错。此时可确认之前发送的所有数据已被对等端接收并ACK。
• 代码示例：

// 发送升级完成消息
const char* msg = "{\"status\": \"upgrade_success\"}";
send(sockfd, msg, strlen(msg), 0);

// 关闭写方向，触发FIN流程
shutdown(sockfd, SHUT_WR);

// 等待对等端关闭连接
char buf[1024];
ssize_t ret;
while ((ret = read(sockfd, buf, sizeof(buf))) > 0) {
    // 无需处理接收的数据，仅等待连接关闭
}

// 此时所有数据已确认送达，可执行重启
system("reboot");

• 注意：若对等端（如Web UI浏览器）未及时关闭连接，需添加超时逻辑，避免无限等待。

2. 通过TCP_INFO选项监控发送状态 #

无需关闭Socket，直接通过getsockopt获取TCP内部状态，判断未ACK字节数和发送缓冲区状态：

• 步骤：
  1. 发送消息后，循环调用getsockopt获取TCP_INFO结构体。
  2. 检查结构体中的tcpi_unacked（未被对等端ACK的字节数）和tcpi_sndbuf（发送缓冲区剩余字节数），当两者都为0时，说明所有数据已发送并被确认。
• 代码示例：

#include <linux/tcp.h>

struct tcp_info info;
socklen_t len = sizeof(info);
int timeout = 5000; // 超时5秒
int elapsed = 0;

while (elapsed < timeout) {
    getsockopt(sockfd, IPPROTO_TCP, TCP_INFO, &info, &len);
    if (info.tcpi_unacked == 0 && info.tcpi_sndbuf == info.tcpi_snd_wnd) {
        break; // 数据已全部确认
    }
    usleep(10000); // 10ms后重试
    elapsed += 10;
}

if (elapsed >= timeout) {
    // 处理超时情况
} else {
    // 安全重启
}

• 注意：不同Linux内核版本的TCP_INFO结构体字段可能略有差异，需根据目标系统调整判断条件。

3. 应用层确认机制（最可靠） #

如果需要确保对等端不仅收到数据，还完成了业务处理，建议在应用层增加确认逻辑：

• 步骤：
  1. 设备发送升级完成的消息后，等待Web UI返回一个明确的"接收确认"报文（如{"ack": "upgrade_received"}）。
  2. 收到确认报文后，再执行设备重启操作。
• 优势：不受TCP层中间设备（如代理、NAT）的影响，能确保业务层面的消息送达。

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