plugins.d/tc-qos-helper.sh

   1 #!/bin/bash
   2
   3 export PATH="${PATH}:/sbin:/usr/sbin:/usr/local/sbin"
   4
   5 PROGRAM_FILE="$0"
   6 PROGRAM_NAME="$(basename $0)"
   7 PROGRAM_NAME="${PROGRAM_NAME/.plugin}"
   8
   9 plugins_dir="${NETDATA_PLUGINS_DIR}"
  10 [ -z "$plugins_dir" ] && plugins_dir="$( dirname $PROGRAM_FILE )"
  11
  12 config_dir=${NETDATA_CONFIG_DIR-/etc/netdata}
  13 tc="$(which tc 2>/dev/null)"
  14 fireqos_run_dir="/var/run/fireqos"
  15 qos_get_class_names_every=120
  16 qos_exit_every=3600
  17
  18 # check if we have a valid number for interval
  19 t=${1}
  20 update_every=$((t))
  21 [ $((update_every)) -lt 1 ] && update_every=${NETDATA_UPDATE_EVERY}
  22 [ $((update_every)) -lt 1 ] && update_every=1
  23
  24 # allow the user to override our defaults
  25 if [ -f "${config_dir}/tc-qos-helper.conf" ]
  26         then
  27         source "${config_dir}/tc-qos-helper.conf"
  28 fi
  29
  30 # default time function
  31 now_ms=
  32 current_time_ms() {
  33         now_ms="$(date +'%s')000"
  34 }
  35
  36 # default sleep function
  37 LOOPSLEEPMS_LASTWORK=0
  38 loopsleepms() {
  39         [ "$1" = "tellwork" ] && shift
  40         sleep $1
  41 }
  42
  43 # if found and included, this file overwrites loopsleepms()
  44 # with a high resolution timer function for precise looping.
  45 . "${plugins_dir}/loopsleepms.sh.inc"
  46
  47 if [ -z "${tc}" -o ! -x "${tc}" ]
  48         then
  49         echo >&2 "${PROGRAM_NAME}: Cannot find command 'tc' in this system."
  50         exit 1
  51 fi
  52
  53 devices=
  54 fix_names=
  55
  56 setclassname() {
  57         echo "SETCLASSNAME $3 $2"
  58 }
  59
  60 show_tc() {
  61         local x="${1}" interface_dev interface_classes interface_classes_monitor
  62
  63         echo "BEGIN ${x}"
  64         ${tc} -s class show dev ${x}
  65
  66         # check FireQOS names for classes
  67         if [ ! -z "${fix_names}" -a -f "${fireqos_run_dir}/ifaces/${x}" ]
  68         then
  69                 name="$(<"${fireqos_run_dir}/ifaces/${x}")"
  70                 echo "SETDEVICENAME ${name}"
  71
  72                 interface_dev=
  73                 interface_classes=
  74                 interface_classes_monitor=
  75                 source "${fireqos_run_dir}/${name}.conf"
  76                 for n in ${interface_classes_monitor}
  77                 do
  78                         setclassname ${n//|/ }
  79                 done
  80                 [ ! -z "${interface_dev}" ] && echo "SETDEVICEGROUP ${interface_dev}"
  81         fi
  82         echo "END ${x}"
  83 }
  84
  85 all_devices() {
  86         cat /proc/net/dev | grep ":" | cut -d ':' -f 1 | while read dev
  87         do
  88                 l=$(${tc} class show dev ${dev} | wc -l)
  89                 [ $l -ne 0 ] && echo ${dev}
  90         done
  91 }
  92
  93 # update devices and class names
  94 # once every 2 minutes
  95 names_every=$((qos_get_class_names_every / update_every))
  96
  97 # exit this script every hour
  98 # it will be restarted automatically
  99 exit_after=$((qos_exit_every / update_every))
 100
 101 c=0
 102 gc=0
 103 while [ 1 ]
 104 do
 105         fix_names=
 106         c=$((c + 1))
 107         gc=$((gc + 1))
 108
 109         if [ ${c} -le 1 -o ${c} -ge ${names_every} ]
 110         then
 111                 c=1
 112                 fix_names="YES"
 113                 devices="$( all_devices )"
 114         fi
 115
 116         for d in ${devices}
 117         do
 118                 show_tc ${d}
 119         done
 120
 121         echo "WORKTIME ${LOOPSLEEPMS_LASTWORK}"
 122
 123         loopsleepms ${update_every}
 124
 125         [ ${gc} -gt ${exit_after} ] && exit 0
 126 done