使用python接收高速率的UDP数据包

Question

我正在使用python来接收来自FPGA的UDP数据包流，试图尽可能少地丢失数据包。数据包速率从大约5kHz到某些MHz，我们希望在特定的时间窗口（代码中的acq_time）内获取数据。我们现在有了这个代码：

BUFSIZE=4096
dataSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
dataSock.settimeout(0.1)
dataSock.bind((self.HOST_IP, self.HOST_PORT))
time0=time.time()
data_list = []
while time.time()-time0<acq_time:
     fast_acquisition(data_list)

def fast_acquisition(data_list_tmp):
    data, addr = dataSock.recvfrom(self.BUFSIZE)
    data_list_tmp.append(data) 
    return len(data)

Answer 1

您可以使用tcpdump（在C中实现）捕获UDP流量，因为它比python:

#!/bin/bash

iface=$1 # interface, 1st arg
port=$2  # port, 2nd arg

tcpdump -i $iface -G acq_time_in_seconds -n udp port $port -w traffic.pcap

然后可以使用scapy来处理该流量

#!/usr/bin/env python

from scapy.all import *

scapy_cap = rdpcap('traffic.pcap')
for packet in scapy_cap:
    # process packets...