git: 4c3c6c9271c0 - stable/13 - netpfil tests: improve pft_ping.py
- Go to: [ bottom of page ] [ top of archives ] [ this month ]
Date: Sun, 29 Jan 2023 07:50:51 UTC
The branch stable/13 has been updated by kp: URL: https://cgit.FreeBSD.org/src/commit/?id=4c3c6c9271c0d28da5e0d6832c06aa74c530f456 commit 4c3c6c9271c0d28da5e0d6832c06aa74c530f456 Author: Kajetan Staszkiewicz <vegeta@tuxpowered.net> AuthorDate: 2023-01-21 02:44:52 +0000 Commit: Kristof Provost <kp@FreeBSD.org> CommitDate: 2023-01-29 05:37:46 +0000 netpfil tests: improve pft_ping.py Multiple improvements to pft_ping.py: * Automatically use IPv6 when IPv6 addresses are used, --ip6 is not needed. * Building of ping requests and parsing of ping replies is done layer by layer. This way most arguments are available both for IPv6 and IPv4, for ICMP and TCP. * Use argument groups for improved readability. * Change ToS and TTL argument name to TC and HL to reflect the modern IPv6 nomenclature. The argument still set related IPv4 header fields properly. * Instead of sniffing for the very specific case of duplicated packets, allow for sniffing on multiple interfaces. * Report which sniffer has failed by setting bits of error code. * Raise meaningful exceptions when irrecoverable errors happen. * Make IPv4 fragmentation flags configurable. * Make IPv6 HL / IPv4 TTL configurable. * Make TCP MSS configurable. * Make TCP sequence number configurable. * Make ICMP payload size configurable. * Add debug output. * Move command line argument parsing out of network functions. * Make the code somehow PEP-8 compliant. MFC after: 1 week Sponsored by: InnoGames GmbH Differential Revision: https://reviews.freebsd.org/D38122 (cherry picked from commit f57218e469a7b1ab40521ea75ebfd45b493851ca) --- tests/sys/netpfil/common/pft_ping.py | 780 ++++++++++++++++++++++------------- tests/sys/netpfil/common/tos.sh | 10 +- tests/sys/netpfil/pf/checksum.sh | 6 +- tests/sys/netpfil/pf/dup.sh | 3 +- tests/sys/netpfil/pf/forward.sh | 5 - tests/sys/netpfil/pf/killstate.sh | 5 - tests/sys/netpfil/pf/set_tos.sh | 16 +- 7 files changed, 505 insertions(+), 320 deletions(-) diff --git a/tests/sys/netpfil/common/pft_ping.py b/tests/sys/netpfil/common/pft_ping.py index 20d4164c6e3e..380b92a662c7 100644 --- a/tests/sys/netpfil/common/pft_ping.py +++ b/tests/sys/netpfil/common/pft_ping.py @@ -3,6 +3,7 @@ # SPDX-License-Identifier: BSD-2-Clause # # Copyright (c) 2017 Kristof Provost <kp@FreeBSD.org> +# Copyright (c) 2023 Kajetan Staszkiewicz <vegeta@tuxpowered.net> # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions @@ -29,306 +30,501 @@ import argparse import logging logging.getLogger("scapy").setLevel(logging.CRITICAL) +import math import scapy.all as sp -import socket import sys + +from copy import copy from sniffer import Sniffer -PAYLOAD_MAGIC = bytes.fromhex('42c0ffee') +logging.basicConfig(format='%(message)s') +LOGGER = logging.getLogger(__name__) -dup_found = 0 - -def check_dup(args, packet): - """ - Verify that this is an ICMP packet, and that we only see one - """ - global dup_found - - icmp = packet.getlayer(sp.ICMP) - if not icmp: - return False - - raw = packet.getlayer(sp.Raw) - if not raw: - return False - if raw.load != PAYLOAD_MAGIC: - return False - - dup_found = dup_found + 1 - return False - -def check_ping_request(args, packet): - if args.ip6: - return check_ping6_request(args, packet) - else: - return check_ping4_request(args, packet) - -def check_ping4_request(args, packet): - """ - Verify that the packet matches what we'd have sent - """ - dst_ip = args.to[0] - - 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 != PAYLOAD_MAGIC: - return False - - # Wait to check expectations until we've established this is the packet we - # sent. - if args.expect_tos: - if ip.tos != int(args.expect_tos[0]): - print("Unexpected ToS value %d, expected %d" \ - % (ip.tos, int(args.expect_tos[0]))) - return False - - return True - -def check_ping6_request(args, packet): - """ - Verify that the packet matches what we'd have sent - """ - dst_ip = args.to[0] - - ip = packet.getlayer(sp.IPv6) - if not ip: - return False - if ip.dst != dst_ip: - return False - - icmp = packet.getlayer(sp.ICMPv6EchoRequest) - if not icmp: - return False - if icmp.data != PAYLOAD_MAGIC: - return False - - return True - -def check_ping_reply(args, packet): - if args.ip6: - return check_ping6_reply(args, packet) - else: - return check_ping4_reply(args, packet) - -def check_ping4_reply(args, packet): - """ - Check that this is a reply to the ping request we sent - """ - dst_ip = args.to[0] - - ip = packet.getlayer(sp.IP) - if not ip: - return False - if ip.src != dst_ip: - return False - - icmp = packet.getlayer(sp.ICMP) - if not icmp: - return False - if sp.icmptypes[icmp.type] != 'echo-reply': - return False - - raw = packet.getlayer(sp.Raw) - if not raw: - return False - if raw.load != PAYLOAD_MAGIC: - return False - - return True - -def check_ping6_reply(args, packet): - """ - Check that this is a reply to the ping request we sent - """ - dst_ip = args.to[0] - - ip = packet.getlayer(sp.IPv6) - if not ip: - return False - if ip.src != dst_ip: - return False - - icmp = packet.getlayer(sp.ICMPv6EchoReply) - if not icmp: - print("No echo reply!") - return False - - if icmp.data != PAYLOAD_MAGIC: - print("data mismatch") - return False - - return True - -def ping(send_if, dst_ip, args): - ether = sp.Ether() - ip = sp.IP(dst=dst_ip) - icmp = sp.ICMP(type='echo-request') - raw = sp.raw(PAYLOAD_MAGIC) - - if args.send_tos: - ip.tos = int(args.send_tos[0]) - - if args.fromaddr: - ip.src = args.fromaddr[0] - - req = ether / ip / icmp / raw - sp.sendp(req, iface=send_if, verbose=False) - -def ping6(send_if, dst_ip, args): - ether = sp.Ether() - ip6 = sp.IPv6(dst=dst_ip) - icmp = sp.ICMPv6EchoRequest(data=sp.raw(PAYLOAD_MAGIC)) - - if args.fromaddr: - ip.src = args.fromaddr[0] - - req = ether / ip6 / icmp - sp.sendp(req, iface=send_if, verbose=False) - -def check_tcpsyn(args, packet): - dst_ip = args.to[0] - - ip = packet.getlayer(sp.IP) - if not ip: - return False - if ip.dst != dst_ip: - return False - - tcp = packet.getlayer(sp.TCP) - if not tcp: - return False - - # Verify IP checksum - chksum = ip.chksum - ip.chksum = None - new_chksum = sp.IP(sp.raw(ip)).chksum - if chksum != new_chksum: - print("Expected IP checksum %x but found %x\n" % (new_cshkum, chksum)) - return False - - # Verify TCP checksum - chksum = tcp.chksum - packet_raw = sp.raw(packet) - tcp.chksum = None - newpacket = sp.Ether(sp.raw(packet[sp.Ether])) - new_chksum = newpacket[sp.TCP].chksum - if chksum != new_chksum: - print("Expected TCP checksum %x but found %x\n" % (new_chksum, chksum)) - return False - - return True - -def tcpsyn(send_if, dst_ip, args): - opts=[('Timestamp', (1, 1)), ('MSS', 1280)] - - if args.tcpopt_unaligned: - opts = [('NOP', 0 )] + opts - - ether = sp.Ether() - ip = sp.IP(dst=dst_ip) - tcp = sp.TCP(dport=666, flags='S', options=opts) +PAYLOAD_MAGIC = bytes.fromhex('42c0ffee') - req = ether / ip / tcp - sp.sendp(req, iface=send_if, verbose=False) +def build_payload(l): + pl = len(PAYLOAD_MAGIC) + ret = PAYLOAD_MAGIC * math.floor(l/pl) + ret += PAYLOAD_MAGIC[0:(l % pl)] + return ret + + +def prepare_ipv6(dst_address, send_params): + src_address = send_params.get('src_address') + hlim = send_params.get('hlim') + tc = send_params.get('tc') + ip6 = sp.IPv6(dst=dst_address) + if src_address: + ip6.src = src_address + if hlim: + ip6.hlim = hlim + if tc: + ip6.tc = tc + return ip6 + + +def prepare_ipv4(dst_address, send_params): + src_address = send_params.get('src_address') + flags = send_params.get('flags') + tos = send_params.get('tc') + ttl = send_params.get('hlim') + ip = sp.IP(dst=dst_address) + if src_address: + ip.src = src_address + if flags: + ip.flags = flags + if tos: + ip.tos = tos + if ttl: + ip.ttl = ttl + return ip + + +def send_icmp_ping(dst_address, sendif, send_params): + send_length = send_params['length'] + ether = sp.Ether() + if ':' in dst_address: + ip6 = prepare_ipv6(dst_address, send_params) + icmp = sp.ICMPv6EchoRequest(data=sp.raw(build_payload(send_length))) + req = ether / ip6 / icmp + else: + ip = prepare_ipv4(dst_address, send_params) + icmp = sp.ICMP(type='echo-request') + raw = sp.raw(build_payload(send_length)) + req = ether / ip / icmp / raw + sp.sendp(req, sendif, verbose=False) + + +def send_tcp_syn(dst_address, sendif, send_params): + tcpopt_unaligned = send_params.get('tcpopt_unaligned') + seq = send_params.get('seq') + mss = send_params.get('mss') + ether = sp.Ether() + opts=[('Timestamp', (1, 1)), ('MSS', mss if mss else 1280)] + if tcpopt_unaligned: + opts = [('NOP', 0 )] + opts + if ':' in dst_address: + ip = prepare_ipv6(dst_address, send_params) + else: + ip = prepare_ipv4(dst_address, send_params) + tcp = sp.TCP(dport=666, flags='S', options=opts, seq=seq) + req = ether / ip / tcp + sp.sendp(req, iface=sendif, verbose=False) + + +def send_ping(dst_address, sendif, ping_type, send_params): + if ping_type == 'icmp': + send_icmp_ping(dst_address, sendif, send_params) + elif ping_type == 'tcpsyn': + send_tcp_syn(dst_address, sendif, send_params) + else: + raise Exception('Unspported ping type') + + +def check_ipv4(expect_params, packet): + src_address = expect_params.get('src_address') + dst_address = expect_params.get('dst_address') + flags = expect_params.get('flags') + tos = expect_params.get('tc') + ttl = expect_params.get('hlim') + ip = packet.getlayer(sp.IP) + if not ip: + LOGGER.debug('Packet is not IPv4!') + return False + if src_address and ip.src != src_address: + LOGGER.debug('Source IPv4 address does not match!') + return False + if dst_address and ip.dst != dst_address: + LOGGER.debug('Destination IPv4 address does not match!') + return False + chksum = ip.chksum + ip.chksum = None + new_chksum = sp.IP(sp.raw(ip)).chksum + if chksum != new_chksum: + LOGGER.debug(f'Expected IP checksum {new_chksum} but found {chksum}') + return False + if flags and ip.flags != flags: + LOGGER.debug(f'Wrong IP flags value {ip.flags}, expected {flags}') + return False + if tos and ip.tos != tos: + LOGGER.debug(f'Wrong ToS value {ip.tos}, expected {tos}') + return False + if ttl and ip.ttl != ttl: + LOGGER.debug(f'Wrong TTL value {ip.ttl}, expected {ttl}') + return False + return True + + +def check_ipv6(expect_params, packet): + src_address = expect_params.get('src_address') + dst_address = expect_params.get('dst_address') + flags = expect_params.get('flags') + hlim = expect_params.get('hlim') + tc = expect_params.get('tc') + ip6 = packet.getlayer(sp.IPv6) + if not ip6: + LOGGER.debug('Packet is not IPv6!') + return False + if src_address and ip6.src != src_address: + LOGGER.debug('Source IPv6 address does not match!') + return False + if dst_address and ip6.dst != dst_address: + LOGGER.debug('Destination IPv6 address does not match!') + return False + # IPv6 has no IP-level checksum. + if flags: + raise Exception("There's no fragmentation flags in IPv6") + if hlim and ip6.hlim != hlim: + LOGGER.debug(f'Wrong Hop Limit value {ip6.hlim}, expected {hlim}') + return False + if tc and ip6.tc != tc: + LOGGER.debug(f'Wrong TC value {ip6.tc}, expected {tc}') + return False + return True + +def check_ping_4(expect_params, packet): + expect_length = expect_params['length'] + if not check_ipv4(expect_params, packet): + return False + icmp = packet.getlayer(sp.ICMP) + if not icmp: + LOGGER.debug('Packet is not IPv4 ICMP!') + return False + raw = packet.getlayer(sp.Raw) + if not raw: + LOGGER.debug('Packet contains no payload!') + return False + if raw.load != build_payload(expect_length): + LOGGER.debug('Payload magic does not match!') + return False + return True + +def check_ping_request_4(expect_params, packet): + if not check_ping_4(expect_params, packet): + return False + icmp = packet.getlayer(sp.ICMP) + if sp.icmptypes[icmp.type] != 'echo-request': + LOGGER.debug('Packet is not IPv4 ICMP Echo Request!') + return False + return True + + +def check_ping_reply_4(expect_params, packet): + if not check_ping_4(expect_params, packet): + return False + icmp = packet.getlayer(sp.ICMP) + if sp.icmptypes[icmp.type] != 'echo-reply': + LOGGER.debug('Packet is not IPv4 ICMP Echo Reply!') + return False + return True + + +def check_ping_request_6(expect_params, packet): + expect_length = expect_params['length'] + if not check_ipv6(expect_params, packet): + return False + icmp = packet.getlayer(sp.ICMPv6EchoRequest) + if not icmp: + LOGGER.debug('Packet is not IPv6 ICMP Echo Request!') + return False + if icmp.data != build_payload(expect_length): + LOGGER.debug('Payload magic does not match!') + return False + return True + + +def check_ping_reply_6(expect_params, packet): + expect_length = expect_params['length'] + if not check_ipv6(expect_params, packet): + return False + icmp = packet.getlayer(sp.ICMPv6EchoReply) + if not icmp: + LOGGER.debug('Packet is not IPv6 ICMP Echo Reply!') + return False + if icmp.data != build_payload(expect_length): + LOGGER.debug('Payload magic does not match!') + return False + return True + + +def check_ping_request(expect_params, packet): + src_address = expect_params.get('src_address') + dst_address = expect_params.get('dst_address') + if not (src_address or dst_address): + raise Exception('Source or destination address must be given to match the ping request!') + if ( + (src_address and ':' in src_address) or + (dst_address and ':' in dst_address) + ): + return check_ping_request_6(expect_params, packet) + else: + return check_ping_request_4(expect_params, packet) + +def check_ping_reply(expect_params, packet): + src_address = expect_params.get('src_address') + dst_address = expect_params.get('dst_address') + if not (src_address or dst_address): + raise Exception('Source or destination address must be given to match the ping reply!') + if ( + (src_address and ':' in src_address) or + (dst_address and ':' in dst_address) + ): + return check_ping_reply_6(expect_params, packet) + else: + return check_ping_reply_4(expect_params, packet) + +def check_tcp(expect_params, packet): + tcp_flags = expect_params.get('tcp_flags') + mss = expect_params.get('mss') + seq = expect_params.get('seq') + tcp = packet.getlayer(sp.TCP) + if not tcp: + LOGGER.debug('Packet is not TCP!') + return False + chksum = tcp.chksum + tcp.chksum = None + newpacket = sp.Ether(sp.raw(packet[sp.Ether])) + new_chksum = newpacket[sp.TCP].chksum + if chksum != new_chksum: + LOGGER.debug(f'Wrong TCP checksum {chksum}, expected {new_chksum}!') + return False + if tcp_flags and tcp.flags != tcp_flags: + LOGGER.debug(f'Wrong TCP flags {tcp.flags}, expected {tcp_flags}!') + return False + if seq: + if tcp_flags == 'S': + tcp_seq = tcp.seq + elif tcp_flags == 'SA': + tcp_seq = tcp.ack - 1 + if seq != tcp_seq: + LOGGER.debug(f'Wrong TCP Sequence Number {tcp_seq}, expected {seq}') + return False + if mss: + for option in tcp.options: + if option[0] == 'MSS': + if option[1] != mss: + LOGGER.debug(f'Wrong TCP MSS {option[1]}, expected {mss}') + return False + return True + + +def check_tcp_syn_request_4(expect_params, packet): + if not check_ipv4(expect_params, packet): + return False + if not check_tcp(expect_params | {'tcp_flags': 'S'}, packet): + return False + return True + + +def check_tcp_syn_reply_4(expect_params, packet): + if not check_ipv4(expect_params, packet): + return False + if not check_tcp(expect_params | {'tcp_flags': 'SA'}, packet): + return False + return True + + +def check_tcp_syn_request_6(expect_params, packet): + if not check_ipv6(expect_params, packet): + return False + if not check_tcp(expect_params | {'tcp_flags': 'S'}, packet): + return False + return True + + +def check_tcp_syn_reply_6(expect_params, packet): + if not check_ipv6(expect_params, packet): + return False + if not check_tcp(expect_params | {'tcp_flags': 'SA'}, packet): + return False + return True + + +def check_tcp_syn_request(expect_params, packet): + src_address = expect_params.get('src_address') + dst_address = expect_params.get('dst_address') + if not (src_address or dst_address): + raise Exception('Source or destination address must be given to match the tcp syn request!') + if ( + (src_address and ':' in src_address) or + (dst_address and ':' in dst_address) + ): + return check_tcp_syn_request_6(expect_params, packet) + else: + return check_tcp_syn_request_4(expect_params, packet) + + +def check_tcp_syn_reply(expect_params, packet): + src_address = expect_params.get('src_address') + dst_address = expect_params.get('dst_address') + if not (src_address or dst_address): + raise Exception('Source or destination address must be given to match the tcp syn reply!') + if ( + (src_address and ':' in src_address) or + (dst_address and ':' in dst_address) + ): + return check_tcp_syn_reply_6(expect_params, packet) + else: + return check_tcp_syn_reply_4(expect_params, packet) + + +def setup_sniffer(recvif, ping_type, sniff_type, expect_params): + if ping_type == 'icmp' and sniff_type == 'request': + checkfn = check_ping_request + elif ping_type == 'icmp' and sniff_type == 'reply': + checkfn = check_ping_reply + elif ping_type == 'tcpsyn' and sniff_type == 'request': + checkfn = check_tcp_syn_request + elif ping_type == 'tcpsyn' and sniff_type == 'reply': + checkfn = check_tcp_syn_reply + else: + raise Exception('Unspported ping or sniff type') + + return Sniffer(expect_params, checkfn, recvif) + + +def parse_args(): + parser = argparse.ArgumentParser("pft_ping.py", + description="Ping test tool") + + # Parameters of sent ping request + parser.add_argument('--sendif', nargs=1, + required=True, + help='The interface through which the packet(s) will be sent') + parser.add_argument('--to', nargs=1, + required=True, + help='The destination IP address for the ping request') + parser.add_argument('--ping-type', + choices=('icmp', 'tcpsyn'), + help='Type of ping: ICMP (default) or TCP SYN', + default='icmp') + parser.add_argument('--fromaddr', nargs=1, + help='The source IP address for the ping request') + + # Where to look for packets to analyze. + # The '+' format is ugly as it mixes positional with optional syntax. + # But we have no positional parameters so I guess it's fine to use it. + parser.add_argument('--recvif', nargs='+', + help='The interfaces on which to expect the ping request') + parser.add_argument('--replyif', nargs='+', + help='The interfaces which to expect the ping response') + + # Packet settings + parser_send = parser.add_argument_group('Values set in transmitted packets') + parser_send.add_argument('--send-flags', nargs=1, type=str, + help='IPv4 fragmentation flags') + parser_send.add_argument('--send-hlim', nargs=1, type=int, + help='IPv6 Hop Limit or IPv4 Time To Live') + parser_send.add_argument('--send-mss', nargs=1, type=int, + help='TCP Maximum Segment Size') + parser_send.add_argument('--send-seq', nargs=1, type=int, + help='TCP sequence number') + parser_send.add_argument('--send-length', nargs=1, type=int, + default=[len(PAYLOAD_MAGIC)], help='ICMP Echo Request payload size') + parser_send.add_argument('--send-tc', nargs=1, type=int, + help='IPv6 Traffic Class or IPv4 DiffServ / ToS') + parser_send.add_argument('--send-tcpopt-unaligned', action='store_true', + help='Include unaligned TCP options') + + # Expectations + parser_expect = parser.add_argument_group('Values expected in sniffed packets') + parser_expect.add_argument('--expect-flags', nargs=1, type=str, + help='IPv4 fragmentation flags') + parser_expect.add_argument('--expect-hlim', nargs=1, type=int, + help='IPv6 Hop Limit or IPv4 Time To Live') + parser_expect.add_argument('--expect-mss', nargs=1, type=int, + help='TCP Maximum Segment Size') + parser_send.add_argument('--expect-seq', nargs=1, type=int, + help='TCP sequence number') + parser_expect.add_argument('--expect-tc', nargs=1, type=int, + help='IPv6 Traffic Class or IPv4 DiffServ / ToS') + + parser.add_argument('-v', '--verbose', action='store_true', + help=('Enable verbose logging. Apart of potentially useful information ' + 'you might see warnings from parsing packets like NDP or other ' + 'packets not related to the test being run. Use only when ' + 'developing because real tests expect empty stderr and stdout.')) + + return parser.parse_args() 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 request') - parser.add_argument('--replyif', nargs=1, - help='The interface on which to expect the ICMP echo response') - parser.add_argument('--checkdup', nargs=1, - help='The interface on which to expect the duplicated ICMP packets') - parser.add_argument('--ip6', action='store_true', - help='Use IPv6') - parser.add_argument('--to', nargs=1, - required=True, - help='The destination IP address for the ICMP echo request') - parser.add_argument('--fromaddr', nargs=1, - help='The source IP address for the ICMP echo request') - - # TCP options - parser.add_argument('--tcpsyn', action='store_true', - help='Send a TCP SYN packet') - parser.add_argument('--tcpopt_unaligned', action='store_true', - help='Include unaligned TCP options') - - # Packet settings - parser.add_argument('--send-tos', nargs=1, - help='Set the ToS value for the transmitted packet') - - # Expectations - parser.add_argument('--expect-tos', nargs=1, - help='The expected ToS value in the received packet') - - args = parser.parse_args() - - # We may not have a default route. Tell scapy where to start looking for routes - sp.conf.iface6 = args.sendif[0] - - sniffer = None - if not args.recvif is None: - checkfn=check_ping_request - if args.tcpsyn: - checkfn=check_tcpsyn - - sniffer = Sniffer(args, checkfn, args.recvif[0]) - - replysniffer = None - if not args.replyif is None: - checkfn=check_ping_reply - replysniffer = Sniffer(args, checkfn, args.replyif[0]) - - dupsniffer = None - if args.checkdup is not None: - dupsniffer = Sniffer(args, check_dup, args.checkdup[0]) - - if args.tcpsyn: - tcpsyn(args.sendif[0], args.to[0], args) - else: - if args.ip6: - ping6(args.sendif[0], args.to[0], args) - else: - ping(args.sendif[0], args.to[0], args) - - if dupsniffer: - dupsniffer.join() - if dup_found != 1: - sys.exit(1) - - if sniffer: - sniffer.join() - - if sniffer.correctPackets: - sys.exit(0) - else: - sys.exit(1) - - if replysniffer: - replysniffer.join() - - if replysniffer.correctPackets: - sys.exit(0) - else: - sys.exit(1) + args = parse_args() + + if args.verbose: + LOGGER.setLevel(logging.DEBUG) + + # Dig out real values of program arguments + send_if = args.sendif[0] + reply_ifs = args.replyif + recv_ifs = args.recvif + dst_address = args.to[0] + + # Standardize parameters which have nargs=1. + send_params = {} + expect_params = {} + for param_name in ('flags', 'hlim', 'length', 'mss', 'seq', 'tc'): + param_arg = vars(args).get(f'send_{param_name}') + send_params[param_name] = param_arg[0] if param_arg else None + param_arg = vars(args).get(f'expect_{param_name}') + expect_params[param_name] = param_arg[0] if param_arg else None + + expect_params['length'] = send_params['length'] + send_params['tcpopt_unaligned'] = args.send_tcpopt_unaligned + send_params['src_address'] = args.fromaddr[0] if args.fromaddr else None + + # We may not have a default route. Tell scapy where to start looking for routes + sp.conf.iface6 = send_if + + # Configuration sanity checking. + if not (reply_ifs or recv_ifs): + raise Exception('With no reply or recv interface specified no traffic ' + 'can be sniffed and verified!' + ) + + sniffers = [] + + if recv_ifs: + sniffer_params = copy(expect_params) + sniffer_params['src_address'] = None + sniffer_params['dst_address'] = dst_address + for iface in recv_ifs: + LOGGER.debug(f'Installing receive sniffer on {iface}') + sniffers.append( + setup_sniffer(iface, args.ping_type, 'request', sniffer_params, + )) + + if reply_ifs: + sniffer_params = copy(expect_params) + sniffer_params['src_address'] = dst_address + sniffer_params['dst_address'] = None + for iface in reply_ifs: + LOGGER.debug(f'Installing reply sniffer on {iface}') + sniffers.append( + setup_sniffer(iface, args.ping_type, 'reply', sniffer_params, + )) + + LOGGER.debug(f'Installed {len(sniffers)} sniffers') + + send_ping(dst_address, send_if, args.ping_type, send_params) + + err = 0 + sniffer_num = 0 + for sniffer in sniffers: + sniffer.join() + if sniffer.correctPackets == 1: + LOGGER.debug(f'Expected ping has been sniffed on {sniffer._recvif}.') + else: + # Set a bit in err for each failed sniffer. + err |= 1<<sniffer_num + if sniffer.correctPackets > 1: + LOGGER.debug(f'Duplicated ping has been sniffed on {sniffer._recvif}!') + else: + LOGGER.debug(f'Expected ping has not been sniffed on {sniffer._recvif}!') + sniffer_num += 1 + + return err + if __name__ == '__main__': - main() + sys.exit(main()) diff --git a/tests/sys/netpfil/common/tos.sh b/tests/sys/netpfil/common/tos.sh index 452c6a002bbf..bd5e657ecf72 100644 --- a/tests/sys/netpfil/common/tos.sh +++ b/tests/sys/netpfil/common/tos.sh @@ -68,7 +68,7 @@ tos_body() --sendif ${epair_send}a \ --to 198.51.100.3 \ --recvif ${epair_recv}a \ - --expect-tos 36 + --expect-tc 36 # Check if the firewall is able to set the ToS bits # and persists the EN bits (if already set) @@ -82,8 +82,8 @@ tos_body() --sendif ${epair_send}a \ --to 198.51.100.3 \ --recvif ${epair_recv}a \ - --send-tos 3 \ - --expect-tos 39 + --send-tc 3 \ + --expect-tc 39 # Check if the firewall is able to filter the # packets based on the ToS value @@ -97,13 +97,13 @@ tos_body() --sendif ${epair_send}a \ --to 198.51.100.3 \ --recvif ${epair_recv}a \ - --send-tos 36 + --send-tc 36 atf_check -s exit:0 $(atf_get_srcdir)/pft_ping.py \ --sendif ${epair_send}a \ --to 198.51.100.3 \ --recvif ${epair_recv}a \ - --send-tos 32 + --send-tc 32 } tos_cleanup() diff --git a/tests/sys/netpfil/pf/checksum.sh b/tests/sys/netpfil/pf/checksum.sh index 836bc1233963..9060e763d18d 100644 --- a/tests/sys/netpfil/pf/checksum.sh +++ b/tests/sys/netpfil/pf/checksum.sh @@ -64,15 +64,15 @@ unaligned_body() --sendif ${epair_in}a \ --to 198.51.100.2 \ --recvif ${epair_out}b \ - --tcpsyn + --ping-type tcpsyn # And unaligned atf_check -s exit:0 ${common_dir}/pft_ping.py \ --sendif ${epair_in}a \ --to 198.51.100.2 \ --recvif ${epair_out}b \ - --tcpsyn \ - --tcpopt_unaligned + --ping-type tcpsyn \ + --send-tcpopt-unaligned } unaligned_cleanup() diff --git a/tests/sys/netpfil/pf/dup.sh b/tests/sys/netpfil/pf/dup.sh index 3b3bef976fc2..83ef57760eef 100644 --- a/tests/sys/netpfil/pf/dup.sh +++ b/tests/sys/netpfil/pf/dup.sh @@ -67,8 +67,7 @@ dup_to_body() atf_check -s exit:0 ${common_dir}/pft_ping.py \ --sendif ${epair_send}a \ --to 198.51.100.3 \ - --recv ${epair_recv}a \ - --checkdup ${epair_dupto}a + --recv ${epair_recv}a ${epair_dupto}a } dup_to_cleanup() diff --git a/tests/sys/netpfil/pf/forward.sh b/tests/sys/netpfil/pf/forward.sh index 7d5bde00d137..0f00a366878b 100644 --- a/tests/sys/netpfil/pf/forward.sh +++ b/tests/sys/netpfil/pf/forward.sh @@ -118,7 +118,6 @@ v6_body() # Sanity check, can we forward ICMP echo requests without pf? atf_check -s exit:0 ${common_dir}/pft_ping.py \ - --ip6 \ --sendif ${epair_send}a \ --to 2001:db8:43::3 \ --recvif ${epair_recv}a @@ -129,7 +128,6 @@ v6_body() pft_set_rules alcatraz \ "block in inet6 proto icmp6 icmp6-type echoreq" atf_check -s exit:1 ${common_dir}/pft_ping.py \ - --ip6 \ --sendif ${epair_send}a \ --to 2001:db8:43::3 \ --recvif ${epair_recv}a @@ -138,7 +136,6 @@ v6_body() pft_set_rules alcatraz \ "block out inet6 proto icmp6 icmp6-type echoreq" atf_check -s exit:1 -e ignore ${common_dir}/pft_ping.py \ - --ip6 \ --sendif ${epair_send}a \ --to 2001:db8:43::3 \ --recvif ${epair_recv}a @@ -148,7 +145,6 @@ v6_body() "block out" \ "pass out inet6 proto icmp6" atf_check -s exit:0 ${common_dir}/pft_ping.py \ - --ip6 \ --sendif ${epair_send}a \ --to 2001:db8:43::3 \ --recvif ${epair_recv}a @@ -158,7 +154,6 @@ v6_body() "block out inet6 proto icmp6 icmp6-type echoreq" \ "pass in proto icmp" atf_check -s exit:1 ${common_dir}/pft_ping.py \ - --ip6 \ --sendif ${epair_send}a \ --to 2001:db8:43::3 \ --recvif ${epair_recv}a diff --git a/tests/sys/netpfil/pf/killstate.sh b/tests/sys/netpfil/pf/killstate.sh index 91fa9b42c8be..c2942aab41f2 100644 --- a/tests/sys/netpfil/pf/killstate.sh +++ b/tests/sys/netpfil/pf/killstate.sh @@ -127,7 +127,6 @@ v6_body() # Note: use pft_ping so we always use the same ID, so pf considers all # echo requests part of the same flow. atf_check -s exit:0 -o ignore ${common_dir}/pft_ping.py \ - --ip6 \ --sendif ${epair}a \ *** 95 LINES SKIPPED ***