linux udp 压力测试

import socket
import time

def udp_stress_test(target_ip, target_port, num_packets, packet_size):
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    data = bytearray(packet_size, 0)  # 创建指定大小的数据包
    start_time = time.time()
    for _ in range(num_packets):
        sock.sendto(data, (target_ip, target_port))
    end_time = time.time()
    sock.close()
    elapsed_time = end_time - start_time
    throughput = num_packets / elapsed_time
    print(f"Sent {num_packets} UDP packets of size {packet_size} bytes to {target_ip}:{target_port} in {elapsed_time:.2f} seconds.")
    print(f"Throughput: {throughput:.2f} packets/sec")

# 示例调用
udp_stress_test("192.168.1.100", 5000, 10000, 1024)