xref: /openbsd-src/regress/sys/netinet6/frag6/frag6_doubleatomic.py (revision 45550d4a98dbed629b022333d58322e23454c128)
1#!/usr/local/bin/python3
2
3print("ping6 fragments with two consecutive atomic fragment header")
4
5# |-IPv6-|-Frag-|-Frag-|-ICMP6-|-payload-|
6
7import os
8from addr import *
9from scapy.all import *
10
11pid=os.getpid()
12eid=pid & 0xffff
13payload=b"ABCDEFGHIJKLMNOP"
14packet=IPv6(src=LOCAL_ADDR6, dst=REMOTE_ADDR6)/ \
15    ICMPv6EchoRequest(id=eid, data=payload)
16frag=[]
17fid=pid & 0xffffffff
18frag.append(IPv6ExtHdrFragment(id=fid)/ \
19    IPv6ExtHdrFragment(nh=58, id=fid)/bytes(packet)[40:64])
20eth=[]
21for f in frag:
22	pkt=IPv6(src=LOCAL_ADDR6, dst=REMOTE_ADDR6)/f
23	eth.append(Ether(src=LOCAL_MAC, dst=REMOTE_MAC)/pkt)
24
25if os.fork() == 0:
26	time.sleep(1)
27	sendp(eth, iface=LOCAL_IF)
28	os._exit(0)
29
30ans=sniff(iface=LOCAL_IF, timeout=3, filter=
31    "ip6 and src "+REMOTE_ADDR6+" and dst "+LOCAL_ADDR6+" and icmp6")
32for a in ans:
33	if a and a.type == ETH_P_IPV6 and \
34	    ipv6nh[a.payload.nh] == 'ICMPv6' and \
35	    icmp6types[a.payload.payload.type] == 'Echo Reply':
36		id=a.payload.payload.id
37		print("id=%#x" % (id))
38		if id != eid:
39			print("WRONG ECHO REPLY ID")
40			exit(2)
41		data=a.payload.payload.data
42		print("payload=%s" % (data))
43		if data == payload:
44			print("double atomic accepted")
45			exit(1)
46		print("PAYLOAD!=%s" % (payload))
47		exit(2)
48print("no echo reply")
49exit(0)
50