1 files changed, 83 insertions, 0 deletions
diff --git a/tests/sys/netpfil/pf/pft_ping.py b/tests/sys/netpfil/pf/pft_ping.py
new file mode 100644
index 000000000000..a07cfd82b6e8
--- /dev/null
+++ b/tests/sys/netpfil/pf/pft_ping.py
@@ -0,0 +1,83 @@
+#!/usr/local/bin/python2.7
+
+import argparse
+import scapy.all as sp
+import sys
+import threading
+
+PAYLOAD_MAGIC = 0x42c0ffee
+
+class Sniffer(threading.Thread):
+	def __init__(self, recvif):
+		threading.Thread.__init__(self)
+
+		self._recvif = recvif
+
+		self.start()
+
+	def run(self):
+		self.packets = sp.sniff(iface=self._recvif, timeout=3)
+
+def check_ping_request(packet, dst_ip):
+	"""
+	Verify that the packet matches what we'd have sent
+	"""
+	ip = packet.getlayer(sp.IP)
+	if not ip:
+		return False
+	if ip.dst != dst_ip:
+		return False
+
+	icmp = packet.getlayer(sp.ICMP)
+	if not icmp:
+		return False
+	if sp.icmptypes[icmp.type] != 'echo-request':
+		return False
+
+	raw = packet.getlayer(sp.Raw)
+	if not raw:
+		return False
+	if raw.load != str(PAYLOAD_MAGIC):
+		return False
+
+	return True
+
+def ping(send_if, dst_ip):
+	req = sp.Ether() \
+		/ sp.IP(dst=dst_ip) \
+		/ sp.ICMP(type='echo-request') \
+		/ sp.Raw(PAYLOAD_MAGIC)
+	sp.sendp(req, iface=send_if, verbose=False)
+
+def main():
+	parser = argparse.ArgumentParser("pft_ping.py",
+		description="Ping test tool")
+	parser.add_argument('--sendif', nargs=1,
+		required=True,
+		help='The interface through which the packet(s) will be sent')
+	parser.add_argument('--recvif', nargs=1,
+		help='The interface on which to expect the ICMP echo response')
+	parser.add_argument('--to', nargs=1,
+		required=True,
+		help='The destination IP address for the ICMP echo request')
+
+	args = parser.parse_args()
+
+	sniffer = None
+	if not args.recvif is None:
+		sniffer = Sniffer(args.recvif[0])
+
+	ping(args.sendif[0], args.to[0])
+
+	if sniffer:
+		sniffer.join()
+
+		for packet in sniffer.packets:
+			if check_ping_request(packet, args.to[0]):
+				sys.exit(0)
+
+		# We did not get the packet we expected
+		sys.exit(1)
+
+if __name__ == '__main__':
+	main()