cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

parameters.sh (3767B)


      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