python测试--scapy构造报文

记录一些测试经验。

---

前言

软件测试工作中经常需要用到收发报文验证一些逻辑，最常见的就是http包，一般的web应用接口测试的时候常用。越是上级协议，工具越多，越方便，越是底层约麻烦。到了需要用到IP协议这一级别的反而经常缺少合适的工具。
工作中遇到需要修改IP.src构造UDP报文，并大量发送（性能测试），特此记录，主要工具是python的scapy库。

---

一、scapy是什么？

scapy是python中用于构造报文的一个库，非常实用。Scapy is a Python program that enables the user to send, sniff and dissect and forge network packets. This capability allows construction of tools that can probe, scan or attack networks.
链接: 文档.

二、步骤

1.快速构造发包

以下示例实现了构造包并发送，但速度不快：

# 引入包
from scapy.all import *
# 最基本的udp构造
test_bytes = b'test'
# 构造一个从ip 192.168.1.1发往192.168.1.100:10000的udp包，其内容为test
udp_packet = IP(src='192.168.1.1'， dst='192.168.1.100')/UDP(dport=10000)/Raw(load=test_bytes)

# Send packets at layer 3 不用关心路由网口等
send(udp_packet)
# Send packets at layer 2
# sendp()

# 使用很方便，就是windows马上就可能遇上：scapy.error.Scapy_Exception: Interface is invalid (no pcap match found)
# windows上使用需要安装npcap https://nmap.org/npcap/#download， 可以先安装wiresharck这个抓包工具，我记得有带npcap的 

2.复用socket大量发包

复用socket：

# 直接使用send，可以说每次都在创建socket，再发送数据，很显然到了性能测试的时候，是无法接受的。
# 所以需要手动创建一个socket，复用发送数据
# 理论上完全可以撇开scapy，自己重复造个轮子，然后使用python的socket创建rawsocket发送报文

# linux
# 可以直接用socket库创建一个支持发送原始报文的socket，参数有兴趣可以了解下
# raw_socket = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_RAW)
#