使用Python DPKT从PCAP文件估算TCP拥塞窗口大小问题咨询

• 你对cwnd的统计逻辑存在偏差：cwnd是发送端在未收到ACK的情况下，最多可以连续发送的未确认数据量（通常以段数或字节数计），不是每个RTT内随机统计的发送包数。你之前的方法会把超时重传、快速重传、接收端窗口限制、捎带ACK、延迟ACK这些干扰项都计入统计，自然会得到大量不符合预期的异常值。
• 必须过滤无效统计场景：
  • 排除重传包：重复序列号的发送包属于丢包后的补救行为，不代表当时的真实cwnd大小，不能计入统计，可通过维护已发送最高序列号的方式过滤重传包。
  • 排除接收端窗口（rwnd）限制的情况：如果接收端通告的rwnd小于当前cwnd，发送端会按照rwnd上限发送数据，此时统计到的发送量是rwnd而非cwnd，需要跳过这类窗口的统计。
  • 修正RTT判定逻辑：接收端可能启用延迟ACK机制，每收2个段才回1个ACK，不能每收到1个ACK就算一个RTT结束，要以「发送端发完一批数据，收到对应这批数据最高序列号的ACK」作为一个RTT的结束节点。
  • 限定统计范围：前3个cwnd属于TCP连接建立后的慢启动阶段数据，不能把后续拥塞避免、快恢复阶段的数据纳入统计。
• 前3个cwnd的正确统计时机：第一个cwnd是SYN握手完成后，发送端第一次批量发送的数据量；第二个是收到第一批数据的全部ACK后，发送端第二次批量发送的数据量；第三个是收到第二批全部ACK后，发送端第三次批量发送的数据量，仅这三个值是你需要提取的目标。

• 第一步先按五元组（源IP、目的IP、源端口、目的端口、协议）分离单条TCP流，明确区分发送端和接收端，仅统计发送端发往接收端的数据包、以及接收端发往发送端的ACK包。
• 第二步预处理数据包：过滤SYN、FIN、RST控制包，仅保留带有效数据的TCP段；从SYN包的选项字段提取MSS值，无相关选项时默认按1460字节计算。
• 第三步实现cwnd统计逻辑，参考代码如下：

import dpkt
import socket

def get_first_three_cwnd(pcap_path, send_flow_info):
    # send_flow_info格式：(发送端IP字符串, 接收端IP字符串, 发送端端口, 接收端端口)
    send_ip = socket.inet_aton(send_flow_info[0])
    recv_ip = socket.inet_aton(send_flow_info[1])
    send_port, recv_port = send_flow_info[2], send_flow_info[3]
    mss = 1460
    # 初始化统计变量：current_una是当前未被确认的最低序列号
    current_una = None
    current_sent_bytes = 0
    cwnd_res = []

    with open(pcap_path, 'rb') as f:
        pcap_reader = dpkt.pcap.Reader(f)
        for ts, buf in pcap_reader:
            eth = dpkt.ethernet.Ethernet(buf)
            if not isinstance(eth.data, dpkt.ip.IP):
                continue
            ip = eth.data
            if not isinstance(ip.data, dpkt.tcp.TCP):
                continue
            tcp = ip.data
            # 提取SYN包中的MSS
            if tcp.flags & dpkt.tcp.TH_SYN and ip.src == send_ip and tcp.sport == send_port:
                for opt_type, opt_data in dpkt.tcp.parse_opts(tcp.opts):
                    if opt_type == dpkt.tcp.TCP_OPT_MSS:
                        mss = int.from_bytes(opt_data, byteorder='big')
            # 处理发送端的数据包
            if ip.src == send_ip and tcp.sport == send_port and ip.dst == recv_ip and tcp.dport == recv_port:
                if len(tcp.data) == 0:
                    continue
                if current_una is None:
                    current_una = tcp.seq
                # 跳过重传包
                if tcp.seq >= current_una:
                    current_sent_bytes += len(tcp.data)
            # 处理接收端的ACK包
            elif ip.src == recv_ip and tcp.sport == recv_port and ip.dst == send_ip and tcp.dport == send_port:
                if tcp.flags & dpkt.tcp.TH_ACK and current_una is not None and tcp.ack > current_una:
                    if len(cwnd_res) < 3:
                        # 按MSS段数统计cwnd，也可直接保留字节数
                        cwnd_res.append(round(current_sent_bytes / mss))
                    current_una = tcp.ack
                    current_sent_bytes = 0
                    # 凑齐3个值直接返回
                    if len(cwnd_res) == 3:
                        return cwnd_res
    return cwnd_res

上述代码仅实现基础统计逻辑，你可根据需求补充延迟ACK合并、rwnd限制判断等优化逻辑。

TCP慢启动阶段，cwnd默认每个RTT翻倍，比如初始cwnd为4MSS时，前三个cwnd通常为4、8、16；当cwnd达到慢启动阈值（ssthresh）后进入拥塞避免阶段，每个RTT仅增加1MSS；如果传输过程中出现丢包，cwnd会大幅下降，重新进入慢启动或快恢复阶段。

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