mirror of
https://github.com/freebsd/freebsd-src
synced 2024-07-09 04:36:31 +00:00
6d1471fda8
Teach pft_ether.py to send a range of packet sizes. Use this to move the
size sweep into Python, removing the repeated Python startup overhead
and greatly speeding up the pf.ether.short_pkt test.
This should fix test timeouts seen on ci.freebsd.org.
While here also extend the range of packet sizes tested, because it adds
very little runtime now.
Sponsored by: Rubicon Communications, LLC ("Netgate")
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# SPDX-License-Identifier: BSD-2-Clause
|
|
#
|
|
# Copyright © 2022. Rubicon Communications, LLC (Netgate). All Rights Reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions
|
|
# are met:
|
|
# 1. Redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer.
|
|
# 2. Redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
# SUCH DAMAGE.
|
|
#
|
|
|
|
import argparse
|
|
import logging
|
|
logging.getLogger("scapy").setLevel(logging.CRITICAL)
|
|
import scapy.all as sp
|
|
import socket
|
|
import sys
|
|
|
|
PAYLOAD_MAGIC = bytes.fromhex('42c0ffee')
|
|
|
|
def ping(send_if, dst_ip, length):
|
|
ether = sp.Ether()
|
|
ip = sp.IP(dst=dst_ip)
|
|
icmp = sp.ICMP(type='echo-request')
|
|
raw = sp.raw(PAYLOAD_MAGIC)
|
|
|
|
req = ether / ip / icmp / raw
|
|
req = req.build()[0:length]
|
|
|
|
sp.sendp(req, iface=send_if, verbose=False)
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser("pft_ether.py",
|
|
description="Ethernet test tool")
|
|
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 ICMP echo request')
|
|
parser.add_argument('--len', nargs=1,
|
|
required=True,
|
|
help='The length of the packet')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if '-' in args.len[0]:
|
|
s=args.len[0].split('-')
|
|
for i in range(int(s[0]), int(s[1]) + 1):
|
|
ping(args.sendif[0], args.to[0], i)
|
|
else:
|
|
ping(args.sendif[0], args.to[0], int(args.len[0]))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|