summaryrefslogtreecommitdiffstats
path: root/plugins.d/tc-qos-helper.sh
blob: 94eec44a569dac63bb521c661e34a2480032f2ca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/bin/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"
}

# default sleep function
LOOPSLEEPMS_LASTWORK=0
loopsleepms() {
	[ "$1" = "tellwork" ] && shift
	sleep $1
}

# if found and included, this file overwrites loopsleepms()
# with a high resolution timer function for precise looping.
. "${plugins_dir}/loopsleepms.sh.inc"

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"
}

show_tc() {
	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} class show dev ${dev} | wc -l)
		[ $l -ne 0 ] && echo ${dev}
	done
}

# update devices and class names
# once every 2 minutes
names_every=$((qos_get_class_names_every / update_every))

# exit this script every hour
# it will be restarted automatically
exit_after=$((qos_exit_every / update_every))

c=0
gc=0
while [ 1 ]
do
	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

	for d in ${devices}
	do
		show_tc ${d}
	done

	echo "WORKTIME ${LOOPSLEEPMS_LASTWORK}"

	loopsleepms ${update_every}

	[ ${gc} -gt ${exit_after} ] && exit 0
done