1#
  2# SPDX-License-Identifier: GPL-2.0
  3# Common parameter parsing for pktgen scripts
  4#
  5
  6function usage() {
  7    echo ""
  8    echo "Usage: $0 [-vx] -i ethX"
  9    echo "  -i : (\$DEV)       output interface/device (required)"
 10    echo "  -s : (\$PKT_SIZE)  packet size"
 11    echo "  -d : (\$DEST_IP)   destination IP. CIDR (e.g. 198.18.0.0/15) is also allowed"
 12    echo "  -m : (\$DST_MAC)   destination MAC-addr"
 13    echo "  -p : (\$DST_PORT)  destination PORT range (e.g. 433-444) is also allowed"
 14    echo "  -t : (\$THREADS)   threads to start"
 15    echo "  -f : (\$F_THREAD)  index of first thread (zero indexed CPU number)"
 16    echo "  -c : (\$SKB_CLONE) SKB clones send before alloc new SKB"
 17    echo "  -n : (\$COUNT)     num messages to send per thread, 0 means indefinitely"
 18    echo "  -b : (\$BURST)     HW level bursting of SKBs"
 19    echo "  -v : (\$VERBOSE)   verbose"
 20    echo "  -x : (\$DEBUG)     debug"
 21    echo "  -6 : (\$IP6)       IPv6"
 22    echo ""
 23}
 24
 25##  --- Parse command line arguments / parameters ---
 26## echo "Commandline options:"
 27while getopts "s:i:d:m:p:f:t:c:n:b:vxh6" option; do
 28    case $option in
 29        i) # interface
 30          export DEV=$OPTARG
 31	  info "Output device set to: DEV=$DEV"
 32          ;;
 33        s)
 34          export PKT_SIZE=$OPTARG
 35	  info "Packet size set to: PKT_SIZE=$PKT_SIZE bytes"
 36          ;;
 37        d) # destination IP
 38          export DEST_IP=$OPTARG
 39	  info "Destination IP set to: DEST_IP=$DEST_IP"
 40          ;;
 41        m) # MAC
 42          export DST_MAC=$OPTARG
 43	  info "Destination MAC set to: DST_MAC=$DST_MAC"
 44          ;;
 45        p) # PORT
 46          export DST_PORT=$OPTARG
 47	  info "Destination PORT set to: DST_PORT=$DST_PORT"
 48          ;;
 49        f)
 50	  export F_THREAD=$OPTARG
 51	  info "Index of first thread (zero indexed CPU number): $F_THREAD"
 52          ;;
 53        t)
 54	  export THREADS=$OPTARG
 55	  info "Number of threads to start: $THREADS"
 56          ;;
 57        c)
 58	  export CLONE_SKB=$OPTARG
 59	  info "CLONE_SKB=$CLONE_SKB"
 60          ;;
 61        n)
 62	  export COUNT=$OPTARG
 63	  info "COUNT=$COUNT"
 64          ;;
 65        b)
 66	  export BURST=$OPTARG
 67	  info "SKB bursting: BURST=$BURST"
 68          ;;
 69        v)
 70          export VERBOSE=yes
 71          info "Verbose mode: VERBOSE=$VERBOSE"
 72          ;;
 73        x)
 74          export DEBUG=yes
 75          info "Debug mode: DEBUG=$DEBUG"
 76          ;;
 77	6)
 78	  export IP6=6
 79	  info "IP6: IP6=$IP6"
 80	  ;;
 81        h|?|*)
 82          usage;
 83          err 2 "[ERROR] Unknown parameters!!!"
 84    esac
 85done
 86shift $(( $OPTIND - 1 ))
 87
 88if [ -z "$PKT_SIZE" ]; then
 89    # NIC adds 4 bytes CRC
 90    export PKT_SIZE=60
 91    info "Default packet size set to: set to: $PKT_SIZE bytes"
 92fi
 93
 94if [ -z "$F_THREAD" ]; then
 95    # First thread (F_THREAD) reference the zero indexed CPU number
 96    export F_THREAD=0
 97fi
 98
 99if [ -z "$THREADS" ]; then
100    export THREADS=1
101fi
102
103export L_THREAD=$(( THREADS + F_THREAD - 1 ))
104
105if [ -z "$DEV" ]; then
106    usage
107    err 2 "Please specify output device"
108fi
109
110if [ -z "$DST_MAC" ]; then
111    warn "Missing destination MAC address"
112fi
113
114if [ -z "$DEST_IP" ]; then
115    warn "Missing destination IP address"
116fi
117
118if [ ! -d /proc/net/pktgen ]; then
119    info "Loading kernel module: pktgen"
120    modprobe pktgen
121fi