openwrt/package/wprobe/files/wprobe.init

64 lines
1.3 KiB
Bash
Executable File

#!/bin/sh /etc/rc.common
START=90
EXPORTER=/usr/sbin/wprobe-ipfix
wprobe_ssd() {
local cfg="$1"; shift
local cmd="$1"; shift
start-stop-daemon "$cmd" -p "/var/run/wprobe-$cfg.pid" -b -x "$EXPORTER" -m -- "$@"
}
stop_wprobe() {
local cfg="$1"
[ -f "/var/run/wprobe-$cfg.pid" ] && wprobe_ssd "$cfg" -K
rm -f "/var/run/wprobe-$cfg.pid"
}
config_wprobe() {
config_get ifname "$cfg" ifname
config_get interval "$cfg" interval
[ -n "$interval" ] || interval=100
wprobe-info "$ifname" -c -i "$interval"
}
start_ipfix() {
local cfg="$1"
config_get ifname "$cfg" interface
config_get host "$cfg" host
config_get port "$cfg" port
config_get proto "$cfg" proto
case "$proto" in
sctp) proto="-s";;
tcp) proto="-t";;
udp) proto="-u";;
*) proto="-t";;
esac
[ -z "$ifname" -o -z "$host" ] && {
echo "wprobe-export: missing host or interface name in config $cfg"
return
}
config_wprobe "$cfg"
wprobe_ssd "$cfg" -S "$proto" -i "$ifname" -c "$host" -p "${port:-4739}"
}
start_export() {
config_get export_type "$cfg" type
case "$export_type" in
ipfix) start_ipfix "$cfg";;
esac
}
stop() {
for f in /var/run/wprobe-*.pid; do
CFG="${f%%.pid}"
CFG="${CFG##/var/run/wprobe-}"
stop_wprobe "$CFG"
done
}
start() {
config_load wprobe
config_foreach config_wprobe interface
[ -x "$EXPORTER" ] && config_foreach start_export export
}