| rjw | 1f88458 | 2022-01-06 17:20:42 +0800 | [diff] [blame^] | 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 | set -o errexit |
| 9 | |
| 10 | ## -- General shell logging cmds -- |
| 11 | function err() { |
| 12 | local exitcode=$1 |
| 13 | shift |
| 14 | echo "ERROR: $@" >&2 |
| 15 | exit $exitcode |
| 16 | } |
| 17 | |
| 18 | function warn() { |
| 19 | echo "WARN : $@" >&2 |
| 20 | } |
| 21 | |
| 22 | function info() { |
| 23 | if [[ -n "$VERBOSE" ]]; then |
| 24 | echo "INFO : $@" >&2 |
| 25 | fi |
| 26 | } |
| 27 | |
| 28 | ## -- Pktgen proc config commands -- ## |
| 29 | export PROC_DIR=/proc/net/pktgen |
| 30 | # |
| 31 | # Three different shell functions for configuring the different |
| 32 | # components of pktgen: |
| 33 | # pg_ctrl(), pg_thread() and pg_set(). |
| 34 | # |
| 35 | # These functions correspond to pktgens different components. |
| 36 | # * pg_ctrl() control "pgctrl" (/proc/net/pktgen/pgctrl) |
| 37 | # * pg_thread() control the kernel threads and binding to devices |
| 38 | # * pg_set() control setup of individual devices |
| 39 | function pg_ctrl() { |
| 40 | local proc_file="pgctrl" |
| 41 | proc_cmd ${proc_file} "$@" |
| 42 | } |
| 43 | |
| 44 | function pg_thread() { |
| 45 | local thread=$1 |
| 46 | local proc_file="kpktgend_${thread}" |
| 47 | shift |
| 48 | proc_cmd ${proc_file} "$@" |
| 49 | } |
| 50 | |
| 51 | function pg_set() { |
| 52 | local dev=$1 |
| 53 | local proc_file="$dev" |
| 54 | shift |
| 55 | proc_cmd ${proc_file} "$@" |
| 56 | } |
| 57 | |
| 58 | # More generic replacement for pgset(), that does not depend on global |
| 59 | # variable for proc file. |
| 60 | function proc_cmd() { |
| 61 | local result |
| 62 | local proc_file=$1 |
| 63 | local status=0 |
| 64 | # after shift, the remaining args are contained in $@ |
| 65 | shift |
| 66 | local proc_ctrl=${PROC_DIR}/$proc_file |
| 67 | if [[ ! -e "$proc_ctrl" ]]; then |
| 68 | err 3 "proc file:$proc_ctrl does not exists (dev added to thread?)" |
| 69 | else |
| 70 | if [[ ! -w "$proc_ctrl" ]]; then |
| 71 | err 4 "proc file:$proc_ctrl not writable, not root?!" |
| 72 | fi |
| 73 | fi |
| 74 | |
| 75 | if [[ "$DEBUG" == "yes" ]]; then |
| 76 | echo "cmd: $@ > $proc_ctrl" |
| 77 | fi |
| 78 | # Quoting of "$@" is important for space expansion |
| 79 | echo "$@" > "$proc_ctrl" || status=$? |
| 80 | |
| 81 | if [[ "$proc_file" != "pgctrl" ]]; then |
| 82 | result=$(grep "Result: OK:" $proc_ctrl) || true |
| 83 | if [[ "$result" == "" ]]; then |
| 84 | grep "Result:" $proc_ctrl >&2 |
| 85 | fi |
| 86 | fi |
| 87 | if (( $status != 0 )); then |
| 88 | err 5 "Write error($status) occurred cmd: \"$@ > $proc_ctrl\"" |
| 89 | fi |
| 90 | } |
| 91 | |
| 92 | # Old obsolete "pgset" function, with slightly improved err handling |
| 93 | function pgset() { |
| 94 | local result |
| 95 | |
| 96 | if [[ "$DEBUG" == "yes" ]]; then |
| 97 | echo "cmd: $1 > $PGDEV" |
| 98 | fi |
| 99 | echo $1 > $PGDEV |
| 100 | local status=$? |
| 101 | |
| 102 | result=`cat $PGDEV | fgrep "Result: OK:"` |
| 103 | if [[ "$result" == "" ]]; then |
| 104 | cat $PGDEV | fgrep Result: |
| 105 | fi |
| 106 | if (( $status != 0 )); then |
| 107 | err 5 "Write error($status) occurred cmd: \"$1 > $PGDEV\"" |
| 108 | fi |
| 109 | } |
| 110 | |
| 111 | [[ $EUID -eq 0 ]] && trap 'pg_ctrl "reset"' EXIT |
| 112 | |
| 113 | ## -- General shell tricks -- |
| 114 | |
| 115 | function root_check_run_with_sudo() { |
| 116 | # Trick so, program can be run as normal user, will just use "sudo" |
| 117 | # call as root_check_run_as_sudo "$@" |
| 118 | if [ "$EUID" -ne 0 ]; then |
| 119 | if [ -x $0 ]; then # Directly executable use sudo |
| 120 | info "Not root, running with sudo" |
| 121 | sudo "$0" "$@" |
| 122 | exit $? |
| 123 | fi |
| 124 | err 4 "cannot perform sudo run of $0" |
| 125 | fi |
| 126 | } |