mirror of
https://github.com/torvalds/linux
synced 2024-07-21 18:51:47 +00:00
a27ac53909
Each sample script sources functions.sh before parameters.sh which makes $APPEND undefined when trapping EXIT no matter in append mode or not. Due to this when sample scripts finished they always do "pgctrl reset" which resets pktgen config. So move trap to each script after sourcing parameters.sh and trap EXIT explicitly. Signed-off-by: J.J. Martzki <mars14850@gmail.com> Signed-off-by: David S. Miller <davem@davemloft.net>
107 lines
3.1 KiB
Bash
Executable file
107 lines
3.1 KiB
Bash
Executable file
#!/bin/bash
|
|
# SPDX-License-Identifier: GPL-2.0
|
|
#
|
|
# Script will generate one flow per thread (-t N)
|
|
# - Same destination IP
|
|
# - Fake source IPs for each flow (fixed based on thread number)
|
|
#
|
|
# Useful for scale testing on receiver, to see whether silo'ing flows
|
|
# works and scales. For optimal scalability (on receiver) each
|
|
# separate-flow should not access shared variables/data. This script
|
|
# helps magnify any of these scaling issues by overloading the receiver.
|
|
#
|
|
basedir=`dirname $0`
|
|
source ${basedir}/functions.sh
|
|
root_check_run_with_sudo "$@"
|
|
|
|
# Parameter parsing via include
|
|
source ${basedir}/parameters.sh
|
|
|
|
# Trap EXIT first
|
|
trap_exit
|
|
|
|
# Set some default params, if they didn't get set
|
|
if [ -z "$DEST_IP" ]; then
|
|
[ -z "$IP6" ] && DEST_IP="198.18.0.42" || DEST_IP="FD00::1"
|
|
fi
|
|
[ -z "$DST_MAC" ] && DST_MAC="90:e2:ba:ff:ff:ff"
|
|
[ -z "$CLONE_SKB" ] && CLONE_SKB="0"
|
|
[ -z "$BURST" ] && BURST=32
|
|
[ -z "$COUNT" ] && COUNT="0" # Zero means indefinitely
|
|
if [ -n "$DEST_IP" ]; then
|
|
validate_addr${IP6} $DEST_IP
|
|
read -r DST_MIN DST_MAX <<< $(parse_addr${IP6} $DEST_IP)
|
|
fi
|
|
if [ -n "$DST_PORT" ]; then
|
|
read -r UDP_DST_MIN UDP_DST_MAX <<< $(parse_ports $DST_PORT)
|
|
validate_ports $UDP_DST_MIN $UDP_DST_MAX
|
|
fi
|
|
|
|
# General cleanup everything since last run
|
|
[ -z "$APPEND" ] && pg_ctrl "reset"
|
|
|
|
# Threads are specified with parameter -t value in $THREADS
|
|
for ((thread = $F_THREAD; thread <= $L_THREAD; thread++)); do
|
|
dev=${DEV}@${thread}
|
|
|
|
# Add remove all other devices and add_device $dev to thread
|
|
[ -z "$APPEND" ] && pg_thread $thread "rem_device_all"
|
|
pg_thread $thread "add_device" $dev
|
|
|
|
# Base config
|
|
pg_set $dev "flag QUEUE_MAP_CPU"
|
|
pg_set $dev "count $COUNT"
|
|
pg_set $dev "clone_skb $CLONE_SKB"
|
|
pg_set $dev "pkt_size $PKT_SIZE"
|
|
pg_set $dev "delay $DELAY"
|
|
pg_set $dev "flag NO_TIMESTAMP"
|
|
|
|
# Single destination
|
|
pg_set $dev "dst_mac $DST_MAC"
|
|
pg_set $dev "dst${IP6}_min $DST_MIN"
|
|
pg_set $dev "dst${IP6}_max $DST_MAX"
|
|
|
|
if [ -n "$DST_PORT" ]; then
|
|
# Single destination port or random port range
|
|
pg_set $dev "flag UDPDST_RND"
|
|
pg_set $dev "udp_dst_min $UDP_DST_MIN"
|
|
pg_set $dev "udp_dst_max $UDP_DST_MAX"
|
|
fi
|
|
|
|
[ ! -z "$UDP_CSUM" ] && pg_set $dev "flag UDPCSUM"
|
|
|
|
# Setup source IP-addresses based on thread number
|
|
pg_set $dev "src_min 198.18.$((thread+1)).1"
|
|
pg_set $dev "src_max 198.18.$((thread+1)).1"
|
|
|
|
# Setup burst, for easy testing -b 0 disable bursting
|
|
# (internally in pktgen default and minimum burst=1)
|
|
if [[ ${BURST} -ne 0 ]]; then
|
|
pg_set $dev "burst $BURST"
|
|
else
|
|
info "$dev: Not using burst"
|
|
fi
|
|
|
|
done
|
|
|
|
# Run if user hits control-c
|
|
function print_result() {
|
|
# Print results
|
|
for ((thread = $F_THREAD; thread <= $L_THREAD; thread++)); do
|
|
dev=${DEV}@${thread}
|
|
echo "Device: $dev"
|
|
cat /proc/net/pktgen/$dev | grep -A2 "Result:"
|
|
done
|
|
}
|
|
# trap keyboard interrupt (Ctrl-C)
|
|
trap true SIGINT
|
|
|
|
if [ -z "$APPEND" ]; then
|
|
echo "Running... ctrl^C to stop" >&2
|
|
pg_ctrl "start"
|
|
|
|
print_result
|
|
else
|
|
echo "Append mode: config done. Do more or use 'pg_ctrl start' to run"
|
|
fi
|