Imported Upstream version 1.3.0+dfsgupstream/1.3.0+dfsg

1 files changed, 80 insertions, 61 deletions

diff --git a/plugins.d/tc-qos-helper.sh b/plugins.d/tc-qos-helper.sh
index 7b4739815..bff5217d2 100755
--- a/plugins.d/tc-qos-helper.sh
+++ b/plugins.d/tc-qos-helper.sh
@@ -1,107 +1,126 @@
-#!/bin/bash
+#!/usr/bin/env bash
 
 export PATH="${PATH}:/sbin:/usr/sbin:/usr/local/sbin"
 
+PROGRAM_FILE="$0"
+PROGRAM_NAME="$(basename $0)"
+PROGRAM_NAME="${PROGRAM_NAME/.plugin}"
+
+plugins_dir="${NETDATA_PLUGINS_DIR}"
+[ -z "$plugins_dir" ] && plugins_dir="$( dirname $PROGRAM_FILE )"
+
+config_dir=${NETDATA_CONFIG_DIR-/etc/netdata}
+tc="$(which tc 2>/dev/null)"
+fireqos_run_dir="/var/run/fireqos"
+qos_get_class_names_every=120
+qos_exit_every=3600
+
+# check if we have a valid number for interval
+t=${1}
+update_every=$((t))
+[ $((update_every)) -lt 1 ] && update_every=${NETDATA_UPDATE_EVERY}
+[ $((update_every)) -lt 1 ] && update_every=1
+
+# allow the user to override our defaults
+if [ -f "${config_dir}/tc-qos-helper.conf" ]
+    then
+    source "${config_dir}/tc-qos-helper.conf"
+fi
+
 # default time function
 now_ms=
 current_time_ms() {
-	now_ms="$(date +'%s')000"
+    now_ms="$(date +'%s')000"
 }
 
 # default sleep function
 LOOPSLEEPMS_LASTWORK=0
 loopsleepms() {
-	[ "$1" = "tellwork" ] && shift
-	sleep $1
+    [ "$1" = "tellwork" ] && shift
+    sleep $1
 }
 
 # if found and included, this file overwrites loopsleepms()
 # with a high resolution timer function for precise looping.
-. "$NETDATA_PLUGINS_DIR/loopsleepms.sh.inc"
+. "${plugins_dir}/loopsleepms.sh.inc"
 
-# check if we have a valid number for interval
-t=$1
-sleep_time=$((t))
-[ $((sleep_time)) -lt 1 ] && $NETDATA_UPDATE_EVERY
-[ $((sleep_time)) -lt 1 ] && sleep_time=1
-
-tc_cmd="$(which tc)"
-if [ -z "$tc_cmd" ]
-	then
-	echo >&2 "tc: Cannot find a 'tc' command in this system."
-	exit 1
+if [ -z "${tc}" -o ! -x "${tc}" ]
+    then
+    echo >&2 "${PROGRAM_NAME}: Cannot find command 'tc' in this system."
+    exit 1
 fi
 
 devices=
 fix_names=
 
 setclassname() {
-	echo "SETCLASSNAME $3 $2"
+    echo "SETCLASSNAME $3 $2"
 }
 
 show_tc() {
-	local x="$1"
-
-	echo "BEGIN $x"
-	$tc_cmd -s class show dev $x
-
-	# check FireQOS names for classes
-	if [ ! -z "$fix_names" -a -f /var/run/fireqos/ifaces/$x ]
-	then
-		name="$(cat /var/run/fireqos/ifaces/$x)"
-		echo "SETDEVICENAME $name"
-
-		interface_classes=
-		interface_classes_monitor=
-		. /var/run/fireqos/$name.conf
-		for n in $interface_classes_monitor
-		do
-			setclassname $(echo $n | tr '|' ' ')
-		done
-		echo "SETDEVICEGROUP $interface_dev"
-	fi
-	echo "END $x"
+    local x="${1}" interface_dev interface_classes interface_classes_monitor
+
+    echo "BEGIN ${x}"
+    ${tc} -s class show dev ${x}
+
+    # check FireQOS names for classes
+    if [ ! -z "${fix_names}" -a -f "${fireqos_run_dir}/ifaces/${x}" ]
+    then
+        name="$(<"${fireqos_run_dir}/ifaces/${x}")"
+        echo "SETDEVICENAME ${name}"
+
+        interface_dev=
+        interface_classes=
+        interface_classes_monitor=
+        source "${fireqos_run_dir}/${name}.conf"
+        for n in ${interface_classes_monitor}
+        do
+            setclassname ${n//|/ }
+        done
+        [ ! -z "${interface_dev}" ] && echo "SETDEVICEGROUP ${interface_dev}"
+    fi
+    echo "END ${x}"
 }
 
 all_devices() {
-	cat /proc/net/dev | grep ":" | cut -d ':' -f 1 | while read dev
-	do
-		l=$($tc_cmd class show dev $dev | wc -l)
-		[ $l -ne 0 ] && echo $dev
-	done
+    cat /proc/net/dev | grep ":" | cut -d ':' -f 1 | while read dev
+    do
+        l=$(${tc} class show dev ${dev} | wc -l)
+        [ $l -ne 0 ] && echo ${dev}
+    done
 }
 
 # update devices and class names
 # once every 2 minutes
-names_every=$((120 / sleep_time))
+names_every=$((qos_get_class_names_every / update_every))
 
 # exit this script every hour
 # it will be restarted automatically
-exit_after=$((3600 / sleep_time))
+exit_after=$((qos_exit_every / update_every))
 
 c=0
 gc=0
 while [ 1 ]
 do
-	fix_names=
-	c=$((c + 1))
-	gc=$((gc + 1))
+    fix_names=
+    c=$((c + 1))
+    gc=$((gc + 1))
 
-	if [ $c -le 1 -o $c -ge $names_every ]
-	then
-		c=1
-		fix_names="YES"
-		devices="$( all_devices )"
-	fi
+    if [ ${c} -le 1 -o ${c} -ge ${names_every} ]
+    then
+        c=1
+        fix_names="YES"
+        devices="$( all_devices )"
+    fi
 
-	for d in $devices
-	do
-		show_tc $d
-	done
+    for d in ${devices}
+    do
+        show_tc ${d}
+    done
 
-	echo "WORKTIME $LOOPSLEEPMS_LASTWORK"
+    echo "WORKTIME ${LOOPSLEEPMS_LASTWORK}"
 
-	loopsleepms $sleep_time
+    loopsleepms ${update_every}
 
-	[ $gc -gt $exit_after ] && exit 0
+    [ ${gc} -gt ${exit_after} ] && exit 0
 done