使用Scapy提取TCP数据时遇NoneType异常及AttributeError问题求助

你在提取TCP负载时遇到的异常，核心是两个逻辑错误导致的，咱们一步步拆解：

异常成因分析 #

1. 错误访问了不存在的属性
   你代码里写的packet[TCP].src是完全错误的！TCP协议层根本没有src这个属性——src是IP层的属性（代表源IP地址），TCP层的源端⼝对应的属性是sport。当你强行访问packet[TCP].src时，Scapy找不到这个属性，就抛出了AttributeError: src。

2. 没有过滤非TCP数据包
   你的pcap文件里肯定包含了非TCP协议的数据包（比如ARP、ICMP或者握手包），当你直接用packet[TCP]访问时，非TCP包会返回None，后续再访问属性自然会触发NoneType相关的异常。

3. 未判断TCP负载是否存在
   即使是TCP包，也不是所有包都有负载（比如SYN、ACK这类握手包），直接写入packet[TCP].payload可能会写入空内容或者None，导致潜在问题。

修正后的代码 #

from scapy.all import *

capture = rdpcap('Data.pcapng') 
output = open('output.bin','wb') 

for packet in capture:
    # 先过滤出包含IP和TCP层的数据包
    if packet.haslayer(IP) and packet.haslayer(TCP):
        # 检查IP源地址（这才是你要判断的内容）
        if packet[IP].src == '172.16.139.128':
            # 将TCP负载转为字节，避免None问题
            payload = bytes(packet[TCP].payload)
            # 只写入非空的负载内容
            if payload:
                output.write(payload)

# 务必关闭文件，确保内容写入完成
output.close()

关键修正点说明 #

• 用haslayer(IP)和haslayer(TCP)做前置判断，跳过不符合条件的数据包，避免访问不存在的协议层
• 把packet[TCP].src改为packet[IP].src，正确获取源IP地址
• 将TCP负载转为字节类型，并添加非空判断，只写入有效负载内容
• 新增output.close()，确保文件流正常关闭，避免数据丢失

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