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 "  -k : (\$UDP_CSUM)  enable UDP tx checksum"
15    echo "  -t : (\$THREADS)   threads to start"
16    echo "  -f : (\$F_THREAD)  index of first thread (zero indexed CPU number)"
17    echo "  -c : (\$SKB_CLONE) SKB clones send before alloc new SKB"
18    echo "  -n : (\$COUNT)     num messages to send per thread, 0 means indefinitely"
19    echo "  -b : (\$BURST)     HW level bursting of SKBs"
20    echo "  -v : (\$VERBOSE)   verbose"
21    echo "  -x : (\$DEBUG)     debug"
22    echo "  -6 : (\$IP6)       IPv6"
23    echo "  -w : (\$DELAY)     Tx Delay value (ns)"
24    echo "  -a : (\$APPEND)    Script will not reset generator's state, but will append its config"
25    echo ""
26}
27
28##  --- Parse command line arguments / parameters ---
29## echo "Commandline options:"
30while getopts "s:i:d:m:p:f:t:c:n:b:w:vxh6ak" option; do
31    case $option in
32        i) # interface
33          export DEV=$OPTARG
34	  info "Output device set to: DEV=$DEV"
35          ;;
36        s)
37          export PKT_SIZE=$OPTARG
38	  info "Packet size set to: PKT_SIZE=$PKT_SIZE bytes"
39          ;;
40        d) # destination IP
41          export DEST_IP=$OPTARG
42	  info "Destination IP set to: DEST_IP=$DEST_IP"
43          ;;
44        m) # MAC
45          export DST_MAC=$OPTARG
46	  info "Destination MAC set to: DST_MAC=$DST_MAC"
47          ;;
48        p) # PORT
49          export DST_PORT=$OPTARG
50	  info "Destination PORT set to: DST_PORT=$DST_PORT"
51          ;;
52        f)
53	  export F_THREAD=$OPTARG
54	  info "Index of first thread (zero indexed CPU number): $F_THREAD"
55          ;;
56        t)
57	  export THREADS=$OPTARG
58	  info "Number of threads to start: $THREADS"
59          ;;
60        c)
61	  export CLONE_SKB=$OPTARG
62	  info "CLONE_SKB=$CLONE_SKB"
63          ;;
64        n)
65	  export COUNT=$OPTARG
66	  info "COUNT=$COUNT"
67          ;;
68        b)
69	  export BURST=$OPTARG
70	  info "SKB bursting: BURST=$BURST"
71          ;;
72        w)
73	  export DELAY=$OPTARG
74	  info "DELAY=$DELAY"
75          ;;
76        v)
77          export VERBOSE=yes
78          info "Verbose mode: VERBOSE=$VERBOSE"
79          ;;
80        x)
81          export DEBUG=yes
82          info "Debug mode: DEBUG=$DEBUG"
83          ;;
84	6)
85	  export IP6=6
86	  info "IP6: IP6=$IP6"
87	  ;;
88        a)
89          export APPEND=yes
90          info "Append mode: APPEND=$APPEND"
91          ;;
92        k)
93          export UDP_CSUM=yes
94          info "UDP tx checksum: UDP_CSUM=$UDP_CSUM"
95          ;;
96        h|?|*)
97          usage;
98          err 2 "[ERROR] Unknown parameters!!!"
99    esac
100done
101shift $(( $OPTIND - 1 ))
102
103if [ -z "$PKT_SIZE" ]; then
104    # NIC adds 4 bytes CRC
105    export PKT_SIZE=60
106    info "Default packet size set to: set to: $PKT_SIZE bytes"
107fi
108
109if [ -z "$F_THREAD" ]; then
110    # First thread (F_THREAD) reference the zero indexed CPU number
111    export F_THREAD=0
112fi
113
114if [ -z "$THREADS" ]; then
115    export THREADS=1
116fi
117
118# default DELAY
119[ -z "$DELAY" ] && export DELAY=0 # Zero means max speed
120
121export L_THREAD=$(( THREADS + F_THREAD - 1 ))
122
123if [ -z "$DEV" ]; then
124    usage
125    err 2 "Please specify output device"
126fi
127
128if [ -z "$DST_MAC" ]; then
129    warn "Missing destination MAC address"
130fi
131
132if [ -z "$DEST_IP" ]; then
133    warn "Missing destination IP address"
134fi
135
136if [ ! -d /proc/net/pktgen ]; then
137    info "Loading kernel module: pktgen"
138    modprobe pktgen
139fi
140