xref: /openbsd-src/regress/sys/netinet/frag/frag.py (revision 647ac4014c7881606258ecf2ee637195df4c98ac)
1*647ac401Sbluhm#!/usr/local/bin/python3
23fe81ad4Sbluhm
3*647ac401Sbluhmprint("non-overlapping ping fragments")
43fe81ad4Sbluhm
53fe81ad4Sbluhm# |--------|
63fe81ad4Sbluhm#          |----|
73fe81ad4Sbluhm
83fe81ad4Sbluhmimport os
93fe81ad4Sbluhmfrom addr import *
103fe81ad4Sbluhmfrom scapy.all import *
113fe81ad4Sbluhm
123fe81ad4Sbluhmpid=os.getpid()
133fe81ad4Sbluhmeid=pid & 0xffff
14*647ac401Sbluhmpayload=b"ABCDEFGHIJKLMNOP"
153fe81ad4Sbluhmpacket=IP(src=LOCAL_ADDR, dst=REMOTE_ADDR)/ \
163fe81ad4Sbluhm    ICMP(type='echo-request', id=eid)/payload
173fe81ad4Sbluhmfrag=[]
183fe81ad4Sbluhmfid=pid & 0xffff
197cd47444Sbluhmfrag.append(IP(src=LOCAL_ADDR, dst=REMOTE_ADDR, proto=1, id=fid,
20*647ac401Sbluhm    flags='MF')/bytes(packet)[20:36])
217cd47444Sbluhmfrag.append(IP(src=LOCAL_ADDR, dst=REMOTE_ADDR, proto=1, id=fid,
22*647ac401Sbluhm    frag=2)/bytes(packet)[36:44])
233fe81ad4Sbluhmeth=[]
243fe81ad4Sbluhmfor f in frag:
253fe81ad4Sbluhm	eth.append(Ether(src=LOCAL_MAC, dst=REMOTE_MAC)/f)
263fe81ad4Sbluhm
273fe81ad4Sbluhmif os.fork() == 0:
283fe81ad4Sbluhm	time.sleep(1)
293fe81ad4Sbluhm	sendp(eth, iface=LOCAL_IF)
303fe81ad4Sbluhm	os._exit(0)
313fe81ad4Sbluhm
323fe81ad4Sbluhmans=sniff(iface=LOCAL_IF, timeout=3, filter=
333fe81ad4Sbluhm    "ip and src "+REMOTE_ADDR+" and dst "+LOCAL_ADDR+" and icmp")
343fe81ad4Sbluhmfor a in ans:
353fe81ad4Sbluhm	if a and a.type == ETH_P_IP and \
363fe81ad4Sbluhm	    a.payload.proto == 1 and \
373fe81ad4Sbluhm	    a.payload.frag == 0 and a.payload.flags == 0 and \
383fe81ad4Sbluhm	    icmptypes[a.payload.payload.type] == 'echo-reply':
393fe81ad4Sbluhm		id=a.payload.payload.id
40*647ac401Sbluhm		print("id=%#x" % (id))
413fe81ad4Sbluhm		if id != eid:
42*647ac401Sbluhm			print("WRONG ECHO REPLY ID")
433fe81ad4Sbluhm			exit(2)
443fe81ad4Sbluhm		data=a.payload.payload.payload.load
45*647ac401Sbluhm		print("payload=%s" % (data))
463fe81ad4Sbluhm		if data == payload:
473fe81ad4Sbluhm			exit(0)
48*647ac401Sbluhm		print("PAYLOAD!=%s" % (payload))
493fe81ad4Sbluhm		exit(1)
50*647ac401Sbluhmprint("NO ECHO REPLY")
513fe81ad4Sbluhmexit(2)
52