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