1#!/bin/bash 2# SPDX-License-Identifier: GPL-2.0 3# 4# Script example for many flows testing 5# 6# Number of simultaneous flows limited by variable $FLOWS 7# and number of packets per flow controlled by variable $FLOWLEN 8# 9basedir=`dirname $0` 10source ${basedir}/functions.sh 11root_check_run_with_sudo "$@" 12 13# Parameter parsing via include 14source ${basedir}/parameters.sh 15# Set some default params, if they didn't get set 16[ -z "$DEST_IP" ] && DEST_IP="198.18.0.42" 17[ -z "$DST_MAC" ] && DST_MAC="90:e2:ba:ff:ff:ff" 18[ -z "$CLONE_SKB" ] && CLONE_SKB="0" 19[ -z "$COUNT" ] && COUNT="0" # Zero means indefinitely 20if [ -n "$DEST_IP" ]; then 21 validate_addr $DEST_IP 22 read -r DST_MIN DST_MAX <<< $(parse_addr $DEST_IP) 23fi 24if [ -n "$DST_PORT" ]; then 25 read -r UDP_DST_MIN UDP_DST_MAX <<< $(parse_ports $DST_PORT) 26 validate_ports $UDP_DST_MIN $UDP_DST_MAX 27fi 28 29# NOTICE: Script specific settings 30# ======= 31# Limiting the number of concurrent flows ($FLOWS) 32# and also set how many packets each flow contains ($FLOWLEN) 33# 34[ -z "$FLOWS" ] && FLOWS="8000" 35[ -z "$FLOWLEN" ] && FLOWLEN="10" 36 37if [[ -n "$BURST" ]]; then 38 err 1 "Bursting not supported for this mode" 39fi 40 41# 198.18.0.0 / 198.19.255.255 42read -r SRC_MIN SRC_MAX <<< $(parse_addr 198.18.0.0/15) 43 44# General cleanup everything since last run 45[ -z "$APPEND" ] && pg_ctrl "reset" 46 47# Threads are specified with parameter -t value in $THREADS 48for ((thread = $F_THREAD; thread <= $L_THREAD; thread++)); do 49 dev=${DEV}@${thread} 50 51 # Add remove all other devices and add_device $dev to thread 52 [ -z "$APPEND" ] && pg_thread $thread "rem_device_all" 53 pg_thread $thread "add_device" $dev 54 55 # Base config 56 pg_set $dev "flag QUEUE_MAP_CPU" 57 pg_set $dev "count $COUNT" 58 pg_set $dev "clone_skb $CLONE_SKB" 59 pg_set $dev "pkt_size $PKT_SIZE" 60 pg_set $dev "delay $DELAY" 61 pg_set $dev "flag NO_TIMESTAMP" 62 63 # Single destination 64 pg_set $dev "dst_mac $DST_MAC" 65 pg_set $dev "dst_min $DST_MIN" 66 pg_set $dev "dst_max $DST_MAX" 67 68 if [ -n "$DST_PORT" ]; then 69 # Single destination port or random port range 70 pg_set $dev "flag UDPDST_RND" 71 pg_set $dev "udp_dst_min $UDP_DST_MIN" 72 pg_set $dev "udp_dst_max $UDP_DST_MAX" 73 fi 74 75 [ ! -z "$UDP_CSUM" ] && pg_set $dev "flag UDPCSUM" 76 77 # Randomize source IP-addresses 78 pg_set $dev "flag IPSRC_RND" 79 pg_set $dev "src_min $SRC_MIN" 80 pg_set $dev "src_max $SRC_MAX" 81 82 # Limit number of flows (max 65535) 83 pg_set $dev "flows $FLOWS" 84 # 85 # How many packets a flow will send, before flow "entry" is 86 # re-generated/setup. 87 pg_set $dev "flowlen $FLOWLEN" 88 # 89 # Flag FLOW_SEQ will cause $FLOWLEN packets from the same flow 90 # being send back-to-back, before next flow is selected 91 # incrementally. This helps lookup caches, and is more realistic. 92 # 93 pg_set $dev "flag FLOW_SEQ" 94 95done 96 97# Run if user hits control-c 98function print_result() { 99 # Print results 100 for ((thread = $F_THREAD; thread <= $L_THREAD; thread++)); do 101 dev=${DEV}@${thread} 102 echo "Device: $dev" 103 cat /proc/net/pktgen/$dev | grep -A2 "Result:" 104 done 105} 106# trap keyboard interrupt (Ctrl-C) 107trap true SIGINT 108 109if [ -z "$APPEND" ]; then 110 echo "Running... ctrl^C to stop" >&2 111 pg_ctrl "start" 112 113 print_result 114else 115 echo "Append mode: config done. Do more or use 'pg_ctrl start' to run" 116fi 117