### 解析 pcap 文件并生成传输负载的 Python 完整代码 以下是基于 Scapy 和 Pandas 的完整 Python 程序，用于解析 pcap 文件、提取数据包信息以及计算传输负载，并最终将结果保存到 CSV 文件中。 #### 依赖项说明 为了运行此脚本，请确保安装以下库： - `scapy`：用于读取和解析 pcap 文件。 - `pandas`：用于存储和导出数据帧至文件。 可以通过以下命令安装这些库： ```bash pip install scapy pandas ``` #### 完整代码示例 ```python from scapy.all import rdpcap, IP, TCP, UDP import pandas as pd def parse_pcap(file_path): """ 解析 pcap 文件并返回每条数据包的相关统计信息。 参数: file_path (str): 输入的 pcap 文件路径 返回: DataFrame: 包含数据包统计信息的数据帧 """ packets = rdpcap(file_path) data = [] for packet in packets: if IP in packet and (TCP in packet or UDP in packet): ip_layer = packet[IP] transport_layer = packet[TCP] if TCP in packet else packet[UDP] src_ip = ip_layer.src dst_ip = ip_layer.dst protocol = 'TCP' if TCP in packet else 'UDP' payload_size = len(packet.payload) entry = { 'Source_IP': src_ip, 'Destination_IP': dst_ip, 'Protocol': protocol, 'Payload_Size_Bytes': payload_size } data.append(entry) df = pd.DataFrame(data) return df def calculate_traffic_load(df): """ 计算每个源 IP 地址的总传输负载。 参数: df (DataFrame): 数据包统计信息的数据帧 返回: DataFrame: 每个源 IP 总传输负载的结果表 """ traffic_load_df = df.groupby('Source_IP')['Payload_Size_Bytes'].sum().reset_index() traffic_load_df.rename(columns={'Payload_Size_Bytes': 'Total_Load_Bytes'}, inplace=True) return traffic_load_df if __name__ == "__main__": input_file = "input.pcap" # 替换为实际的 pcap 文件名 output_csv = "traffic_load.csv" try: # 步骤1：解析 pcap 文件 packet_data = parse_pcap(input_file)[^3] # 步骤2：计算传输负载 traffic_load_result = calculate_traffic_load(packet_data)[^3] # 步骤3：保存结果到 CSV 文件 traffic_load_result.to_csv(output_csv, index=False) print(f"传输负载已成功写入 {output_csv}") except FileNotFoundError: print("错误：指定的 pcap 文件未找到。") except Exception as e: print(f"发生未知错误：{e}") ``` --- ### 功能描述 1. **解析 pcap 文件**: 使用 Scapy 库加载 pcap 文件，并逐个遍历其中的数据包。 2. **筛选有效数据包**: 提取包含 IP 层和传输层（TCP 或 UDP）的有效数据包。 3. **记录关键字段**: 对于每个符合条件的数据包，记录其源 IP、目标 IP、协议类型及负载大小。 4. **汇总传输负载**: 利用 Pandas 组合功能按源 IP 聚合计数，得出各源 IP 的总传输负载。 5. **输出结果**: 将汇总后的结果保存为 CSV 文件以便进一步分析。 --- ### 注意事项 - 如果输入的 pcap 文件较大，则可能需要优化内存使用效率[^4]。 - 需要确认输入文件是否存在且格式正确；否则会抛出异常提示用户检查文件路径或内容有效性。 ---