samples/pktgen/functions.sh

   1 #
   2 # Common functions used by pktgen scripts
   3 #  - Depending on bash 3 (or higher) syntax
   4 #
   5 # Author: Jesper Dangaaard Brouer
   6 # License: GPL
   7
   8 ## -- General shell logging cmds --
   9 function err() {
  10     local exitcode=$1
  11     shift
  12     echo "ERROR: $@" >&2
  13     exit $exitcode
  14 }
  15
  16 function warn() {
  17     echo "WARN : $@" >&2
  18 }
  19
  20 function info() {
  21     if [[ -n "$VERBOSE" ]]; then
  22         echo "INFO : $@" >&2
  23     fi
  24 }
  25
  26 ## -- Pktgen proc config commands -- ##
  27 export PROC_DIR=/proc/net/pktgen
  28 #
  29 # Three different shell functions for configuring the different
  30 # components of pktgen:
  31 #   pg_ctrl(), pg_thread() and pg_set().
  32 #
  33 # These functions correspond to pktgens different components.
  34 # * pg_ctrl()   control "pgctrl" (/proc/net/pktgen/pgctrl)
  35 # * pg_thread() control the kernel threads and binding to devices
  36 # * pg_set()    control setup of individual devices
  37 function pg_ctrl() {
  38     local proc_file="pgctrl"
  39     proc_cmd ${proc_file} "$@"
  40 }
  41
  42 function pg_thread() {
  43     local thread=$1
  44     local proc_file="kpktgend_${thread}"
  45     shift
  46     proc_cmd ${proc_file} "$@"
  47 }
  48
  49 function pg_set() {
  50     local dev=$1
  51     local proc_file="$dev"
  52     shift
  53     proc_cmd ${proc_file} "$@"
  54 }
  55
  56 # More generic replacement for pgset(), that does not depend on global
  57 # variable for proc file.
  58 function proc_cmd() {
  59     local result
  60     local proc_file=$1
  61     # after shift, the remaining args are contained in $@
  62     shift
  63     local proc_ctrl=${PROC_DIR}/$proc_file
  64     if [[ ! -e "$proc_ctrl" ]]; then
  65         err 3 "proc file:$proc_ctrl does not exists (dev added to thread?)"
  66     else
  67         if [[ ! -w "$proc_ctrl" ]]; then
  68             err 4 "proc file:$proc_ctrl not writable, not root?!"
  69         fi
  70     fi
  71
  72     if [[ "$DEBUG" == "yes" ]]; then
  73         echo "cmd: $@ > $proc_ctrl"
  74     fi
  75     # Quoting of "$@" is important for space expansion
  76     echo "$@" > "$proc_ctrl"
  77     local status=$?
  78
  79     result=$(grep "Result: OK:" $proc_ctrl)
  80     # Due to pgctrl, cannot use exit code $? from grep
  81     if [[ "$result" == "" ]]; then
  82         grep "Result:" $proc_ctrl >&2
  83     fi
  84     if (( $status != 0 )); then
  85         err 5 "Write error($status) occurred cmd: \"$@ > $proc_ctrl\""
  86     fi
  87 }
  88
  89 # Old obsolete "pgset" function, with slightly improved err handling
  90 function pgset() {
  91     local result
  92
  93     if [[ "$DEBUG" == "yes" ]]; then
  94         echo "cmd: $1 > $PGDEV"
  95     fi
  96     echo $1 > $PGDEV
  97     local status=$?
  98
  99     result=`cat $PGDEV | fgrep "Result: OK:"`
 100     if [[ "$result" == "" ]]; then
 101          cat $PGDEV | fgrep Result:
 102     fi
 103     if (( $status != 0 )); then
 104         err 5 "Write error($status) occurred cmd: \"$1 > $PGDEV\""
 105     fi
 106 }
 107
 108 ## -- General shell tricks --
 109
 110 function root_check_run_with_sudo() {
 111     # Trick so, program can be run as normal user, will just use "sudo"
 112     #  call as root_check_run_as_sudo "$@"
 113     if [ "$EUID" -ne 0 ]; then
 114         if [ -x $0 ]; then # Directly executable use sudo
 115             info "Not root, running with sudo"
 116             sudo "$0" "$@"
 117             exit $?
 118         fi
 119         err 4 "cannot perform sudo run of $0"
 120     fi
 121 }
 122
 123 # Exact input device's NUMA node info
 124 function get_iface_node()
 125 {
 126     local node=$(</sys/class/net/$1/device/numa_node)
 127     if [[ $node == -1 ]]; then
 128         echo 0
 129     else
 130         echo $node
 131     fi
 132 }
 133
 134 # Given an Dev/iface, get its queues' irq numbers
 135 function get_iface_irqs()
 136 {
 137         local IFACE=$1
 138         local queues="${IFACE}-.*TxRx"
 139
 140         irqs=$(grep "$queues" /proc/interrupts | cut -f1 -d:)
 141         [ -z "$irqs" ] && irqs=$(grep $IFACE /proc/interrupts | cut -f1 -d:)
 142         [ -z "$irqs" ] && irqs=$(for i in `ls -Ux /sys/class/net/$IFACE/device/msi_irqs` ;\
 143             do grep "$i:.*TxRx" /proc/interrupts | grep -v fdir | cut -f 1 -d : ;\
 144             done)
 145         [ -z "$irqs" ] && err 3 "Could not find interrupts for $IFACE"
 146
 147         echo $irqs
 148 }
 149
 150 # Given a NUMA node, return cpu ids belonging to it.
 151 function get_node_cpus()
 152 {
 153         local node=$1
 154         local node_cpu_list
 155         local node_cpu_range_list=`cut -f1- -d, --output-delimiter=" " \
 156                           /sys/devices/system/node/node$node/cpulist`
 157
 158         for cpu_range in $node_cpu_range_list
 159         do
 160             node_cpu_list="$node_cpu_list "`seq -s " " ${cpu_range//-/ }`
 161         done
 162
 163         echo $node_cpu_list
 164 }