Commit | Line | Data |
---|---|---|
b64b0d1e JDB |
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 | } |