ratelimit: replace script with daemon

Fixes: WIFI-10190
Fixes: WIFI-10194
Signed-off-by: John Crispin <john@phrozen.org>

feeds/ucentral/ratelimit/files/etc/hotplug.d/net/30-ratelimit

@@ -3,11 +3,11 @@
 [ "${INTERFACE:0:4}" == "wlan" ] || exit 0
 
 [ "$ACTION" == remove ] && {
-	ratelimit deliface $INTERFACE
+	[ -f /tmp/run/hostapd-cli-$INTERFACE.pid ] && kill "$(cat /tmp/run/hostapd-cli-$INTERFACE.pid)"
 	exit 0
 }
 
 [ "$ACTION" == add ] && {
-	ratelimit waitiface $INTERFACE &
+	/usr/libexec/ratelimit-wait.sh $INTERFACE &
 	exit 0
 }

feeds/ucentral/ratelimit/files/etc/init.d/ratelimit

@@ -0,0 +1,37 @@
+#!/bin/sh /etc/rc.common
+
+START=80
+
+USE_PROCD=1
+PROG=/usr/bin/ratelimit
+
+add_rate() {
+	local cfg="$1"
+	config_get ssid "$cfg" ssid
+	config_get ingress "$cfg" ingress
+	config_get egress "$cfg" egress
+	ubus call ratelimit defaults_set '{"name": "'$ssid'", "rate_ingress": "'$ingress'mbit", "rate_egress": "'$egress'mbit" }'
+}
+
+reload_service() {
+	logger ratelimit reload
+	config_load ratelimit
+	config_foreach add_rate rate
+}
+
+service_triggers() {
+	procd_add_reload_trigger ratelimit
+}
+
+start_service() {
+	procd_open_instance
+	procd_set_param command "$PROG"
+	procd_set_param respawn
+	procd_close_instance
+}
+
+service_started() {
+	ubus -t 10 wait_for ratelimit
+	[ $? = 0 ] && reload_service
+}
+

feeds/ucentral/ratelimit/files/usr/bin/ratelimit

@@ -1,174 +1,337 @@
-#!/bin/sh
+#!/usr/bin/env ucode
+'use strict';
 
-. /lib/functions.sh
+import { basename, popen } from 'fs';
+import * as ubus from 'ubus';
+import * as uloop from 'uloop';
 
-wrapper() {
+let defaults = {};
-	echo calling $*
+let devices = {};
-	$*
+
+function cmd(command, ignore_error) {
+//	if (ignore_error)
+//		command += "> /dev/null 2>&1";
+	warn(`> ${command}\n`);
+	let rc = system(command);
+	return ignore_error || rc == 0;
 }
 
-TC() {
+function qdisc_add_leaf(iface, id, opts) {
-	wrapper tc $*
+	opts ??= "";
+	return cmd(`tc class replace dev ${iface} parent 1:1 classid 1:${id} htb rate 1mbit ${opts} burst 2k prio 1`) &&
+	       cmd(`tc qdisc replace dev ${iface} parent 1:${id} handle ${id}: fq_codel flows 128 limit 800 quantum 300 noecn`);
 }
 
-IP() {
+function qdisc_del_leaf(iface, id) {
-	wrapper ip $*
+	cmd(`tc class del dev ${iface} parent 1:1 classid 1:${id}`, true);
 }
 
-get_id() {
+function qdisc_add(iface) {
-	addr=$1
+	return cmd(`tc qdisc add dev ${iface} root handle 1: htb default 2`) &&
-	hashval="0x$(echo "$addr" | md5sum | head -c8)"
+	       cmd(`tc class add dev ${iface} parent 1: classid 1:1 htb rate 1000mbit burst 6k`) &&
-	mask=0x4ff
+	       qdisc_add_leaf(iface, 2, "ceil 1000mbit");
-	echo $(($hashval & $mask))
 }
 
-delclient() {
+function qdisc_del(iface) {
-	local ifb=rateifb$1
+	cmd(`tc qdisc del dev ${iface} root`, true);
-	local iface=$1
-	local mac=$2
-	local id=$3
-	
-	logger "ratelimit: delete old client entries $1 $2"
-
-	id=$(get_id ${mac//:})
-
-	TC filter del dev $iface protocol all parent 1: prio 1 u32 match ether dst $mac flowid 1:$id
-	
-	TC filter del dev $ifb protocol all parent 1: prio 1 u32 match ether src $mac flowid 1:$id
 }
 
-ingress=0
+function ifb_dev(iface) {
-egress=0
+	return "ifb-" + iface;
-
-getrate() {
-	config_get ssid $1 ssid
-	[ "$ssid" == "$2" ] || return
-	config_get ingress $1 ingress
-	config_get egress $1 egress
 }
 
-addclient() {
+function ifb_add(iface, ifbdev) {
-	local ifb=rateifb$1
+	return cmd(`ip link add ${ifbdev} type ifb`) &&
-	local iface=$1
+	       cmd(`ip link set ${ifbdev} up`) &&
-	local mac=$2
+	       cmd(`tc qdisc add dev ${iface} clsact`, true) &&
-	local ssid=$(cat /tmp/ratelimit.$iface)
+	       cmd(`tc filter add dev ${iface} ingress protocol all prio 512 matchall action mirred egress redirect dev ${ifbdev}`);
-
-	egress=$3
-	ingress=$4
-
-	logger "ratelimit: adding client"
-
-	[ "$egress" -eq 0 -o $ingress -eq 0 ] && {
-		config_load ratelimit
-		config_foreach getrate rate $ssid
 }
 
-	[ "$egress" -eq 0 -o $ingress -eq 0 ] && {
+function ifb_del(iface, ifbdev) {
-		logger "ratelimit: no valid rates"
+	cmd(`tc filter del dev ${iface} ingress protocol all prio 512`);
-		exit 1
+	cmd(`ip link set ${ifbdev} down`, true);
+	cmd(`ip link del ${ifbdev}`, true);
 }
 
-	local id=$(get_id ${mac//:})
+function macfilter_add(iface, id, type, mac) {
-
+	return cmd(`tc filter add dev ${iface} protocol all parent 1: prio 1 handle 800::${id} u32 match ether ${type} ${mac} flowid 1:${id}`);
-	logger "ratelimit: add new client entries for $1 $2 $egress $ingress"
-
-	TC class add dev $iface parent 1:1 classid 1:$id htb rate 1mbit ceil ${egress}mbit burst 2k prio 1
-	TC qdisc add dev $iface parent 1:$id handle $id: sfq perturb 10
-	TC filter add dev $iface protocol all parent 1: prio 1 u32 match ether dst $mac flowid 1:$id
-	
-	TC class add dev $ifb parent 1:1 classid 1:$id htb rate 1mbit ceil ${ingress}mbit burst 2k prio 1
-	TC filter add dev $ifb protocol all parent 1: prio 1 u32 match ether src $mac flowid 1:$id
 }
 
-deliface() {
+function macfilter_del(iface, id) {
-	local ifb=rateifb$1
+	cmd(`tc filter del dev ${iface} protocol all parent 1: prio 1 handle 800::${id} u32`, true);
-	local iface=$1
-
-	[ -d /sys/class/net/$ifb/ ] || return 0
-
-	logger "ratelimit: deleting old iface settings"
-
-	IP link set $ifb down
-	IP link del $ifb
-
-	TC qdisc del dev $iface root &2> /dev/null
-
-	rm -f /tmp/ratelimit.$iface
-	[ -f /tmp/run/hostapd-cli-$iface.pid ] && kill "$(cat /tmp/run/hostapd-cli-$iface.pid)"
 }
 
-found=0
+function linux_client_del(device, client) {
-find_ssid() {
+	let ifbdev = ifb_dev(device.name);
-	local ssid
+	let id = client.id + 3;
-	config_get ssid $1 ssid
+
-	[ "$ssid" == "$2" ] || return
+	macfilter_del(device.name, id);
-	found=1
+	qdisc_del_leaf(device.name, id);
+	macfilter_del(ifbdev, id);
+	qdisc_del_leaf(ifbdev, id);
 }
 
-addiface() {
+function linux_client_set(device, client) {
-	local ifb=rateifb$1
+	let ifbdev = ifb_dev(device.name);
-	local iface=$1
+	let id = client.id + 3;
-	local ssid
 
+	linux_client_del(device, client);
 
-	[ -f /tmp/ratelimit.$iface -o -d /sys/class/net/$ifb/ ] && {
+	let ret = qdisc_add_leaf(device.name, id, `ceil ${client.data.rate_egress}`) &&
-		return 0
+		  macfilter_add(device.name, id, "dst", client.address) &&
+		  qdisc_add_leaf(ifbdev, id, `ceil ${client.data.rate_ingress}`) &&
+		  macfilter_add(ifbdev, id, "src", client.address);
+
+	if (!ret)
+		linux_client_del(device, client);
+
+	return ret;
 }
 
-	echo -n startup > /tmp/ratelimit.$iface
 
-	sleep 2
+let ops = {
-	ssid=$(ubus call hostapd.$iface get_status | jsonfilter -e '@.ssid')
+	device: {
-	[ -z "$ssid" ] && {
+		add: function(name) {
-		rm /tmp/ratelimit.$iface
+			let ifbdev = ifb_dev(name);
-		logger "ratelimit: failed to lookup ssid"
-		exit 1
-	}
-	config_load ratelimit
-	config_foreach find_ssid rate $ssid
-	[ "$found" -eq 0 ] && {
-		rm /tmp/ratelimit.$iface
-		exit 0
-	}
-	logger "ratelimit: adding new iface settings"
 
-	echo -n $ssid > /tmp/ratelimit.$iface
+			qdisc_del(name);
+			ifb_del(name, ifbdev);
 
-	IP link add name $ifb type ifb
+			let ret = qdisc_add(name) &&
-	IP link set $ifb up
+				  ifb_add(name, ifbdev) &&
+				  qdisc_add(ifbdev);
 
-	sleep 1
+			if (!ret) {
-
+				qdisc_del(name);
-	TC qdisc add dev $iface root handle 1: htb default 30
+				ifb_del(name, ifbdev);
-	TC class add dev $iface parent 1: classid 1:1 htb rate 1000mbit burst 6k
-	TC qdisc add dev $iface ingress
-	TC filter add dev $iface parent ffff: protocol all prio 10 u32 match u32 0 0 flowid 1:1 action mirred egress redirect dev $ifb
-
-	TC qdisc add dev $ifb root handle 1: htb default 10
-	TC class add dev $ifb parent 1: classid 1:1 htb rate 100mbit
-	hostapd_cli -a /usr/libexec/ratelimit.sh -i $iface -P /tmp/run/hostapd-cli-$iface.pid -B
-       
-	for sta in $(ubus call wifi station | jsonfilter -e '@[*][*].mac'); do
-		addclient $iface $sta
-	done
 			}
 
-waitiface() {
+			return ret;
-	local iface=$1
+		},
+		remove: function(name) {
+			let ifbdev = ifb_dev(name);
+			qdisc_del(name);
+			ifb_del(name, ifbdev);
+		}
+	},
+	client: {
+		set: function(device, client) {
+			return linux_client_set(device, client);
+		},
+		remove: function(device, client) {
+			linux_client_del(device, client);
+		}
+	}
+};
 
-	ubus -t 120 wait_for hostapd.$1
+function get_device(devices, name) {
+	let device = devices[name];
 
-	[ $? -eq 0 ] || exit 0
+	if (device)
+		return device;
 
-	addiface $iface
+	if (!ops.device.add(name))
+		return null;
+
+	device = {
+		name: name,
+		clients: {},
+		client_order: [],
+	};
+
+	devices[name] = device;
+
+	return device;
 }
 
-flush() {
+function del_device(name) {
-	for a in `ls /sys/class/net/ | grep rateifb`; do
+	if (!devices[name])
-		deliface ${a:7}
+		return;
-	done
+	ops.device.remove(name);
+	delete devices[name];
 }
 
-cmd=$1
+function get_free_idx(list) {
-shift
+	for (let i = 0; i < length(list); i++)
-$cmd $@
+		if (list[i] == null)
+			return i;
+
+	return length(list);
+}
+
+function del_client(device, address) {
+	let client = device.clients[address];
+
+	if (!client)
+		return false;
+
+	delete device.clients[address];
+	device.client_order[client.id] = null;
+
+	ops.client.remove(device, client);
+	return true;
+}
+
+function get_client(device, address) {
+	let client = device.clients[address];
+
+	if (client)
+		return client;
+
+	let i = get_free_idx(device.client_order);
+	client = {};
+	client.address = address;
+	client.id = i;
+	client.data = {};
+
+	device.clients[address] = client;
+	device.client_order[i] = client;
+
+	return client;
+}
+
+function set_client(device, client, data) {
+	let update = false;
+
+	for (let key in data) {
+		if (client.data[key] != data[key])
+			update = true;
+
+		client.data[key] = data[key];
+	}
+
+	if (update && !ops.client.set(device, client)) {
+		del_client(device, client.address);
+		return false;
+	}
+
+	return true;
+}
+
+function run_service() {
+	let uctx = ubus.connect();
+
+	uctx.publish("ratelimit", {
+		defaults_set: {
+			call: function(req) {
+				let r_i = req.args.rate_ingress ?? req.args.rate;
+				let r_e = req.args.rate_egress ?? req.args.rate;
+				let name = req.args.name;
+
+				if (!name || !r_i || !r_e)
+					return ubus.STATUS_INVALID_ARGUMENT;
+
+				defaults[name] = [ r_e, r_i ];
+
+				return 0;
+			},
+			args: {
+				name:"",
+				rate:"",
+				rate_ingress:"",
+				rate_egress:"",
+			}
+		},
+		client_set: {
+			call: function(req) {
+				let r_i = req.args.rate_ingress ?? req.args.rate;
+				let r_e = req.args.rate_egress ?? req.args.rate;
+
+				if (req.args.defaults && defaults[req.args.defaults]) {
+					let def = defaults[req.args.defaults];
+
+					r_e ??= def[0];
+					r_i ??= def[1];
+				}
+
+				if (!req.args.device || !req.args.address || !r_i || !r_e)
+					return ubus.STATUS_INVALID_ARGUMENT;
+
+				let device = get_device(devices, req.args.device);
+				if (!device)
+					return ubus.STATUS_INVALID_ARGUMENT;
+
+				let client = get_client(device, req.args.address);
+				if (!client)
+					return ubus.STATUS_INVALID_ARGUMENT;
+
+				let data = {
+					rate_ingress: r_i,
+					rate_egress: r_e
+				};
+
+				if (!set_client(device, client, data))
+					return ubus.STATUS_UNKNOWN_ERROR;
+
+				return 0;
+			},
+			args: {
+				device:"",
+				defaults:"",
+				address:"",
+				rate:"",
+				rate_ingress:"",
+				rate_egress:"",
+			}
+		},
+		client_delete: {
+			call: function(req) {
+				if (!req.args.address)
+					return ubus.STATUS_INVALID_ARGUMENT;
+
+				if (req.args.device) {
+					let device = devices[req.args.device];
+					if (!device)
+						return ubus.STATUS_NOT_FOUND;
+
+					if (!del_client(device, req.args.address))
+						return ubus.STATUS_NOT_FOUND;
+				} else {
+					for (let dev in devices) {
+						let device = devices[dev];
+
+						del_client(device, req.args.address);
+					}
+				}
+
+				return 0;
+			},
+			args: {
+				device:"",
+				address:"",
+			}
+		},
+		device_delete: {
+			call: function(req) {
+				let name = req.args.device;
+
+				if (!name)
+					return ubus.STATUS_INVALID_ARGUMENT;
+
+				if (!devices[name])
+					return ubus.STATUS_NOT_FOUND;
+
+				del_device(name);
+
+				return 0;
+			},
+			args: {
+				device:"",
+			}
+		}
+	});
+
+	try {
+		uloop.run();
+	} catch (e) {
+		warn(`Error: ${e}\n${e.stacktrace[0].context}`);
+	}
+
+	for (let dev in devices) {
+		del_device(dev);
+	}
+}
+
+uloop.init();
+run_service();
+uloop.done();

feeds/ucentral/ratelimit/files/usr/libexec/ratelimit-wait.sh

@@ -0,0 +1,4 @@
+#!/bin/sh
+[ -f /tmp/run/hostapd-cli-$1.pid ] && kill "$(cat /tmp/run/hostapd-cli-$1.pid)"
+ubus -t 120 wait_for hostapd.$1
+[ $? = 0 ] && hostapd_cli -a /usr/libexec/ratelimit.sh -i $1 -P /tmp/run/hostapd-cli-$1.pid -B

feeds/ucentral/ratelimit/files/usr/libexec/ratelimit.sh

@@ -2,9 +2,16 @@
 
 case $2 in
 AP-STA-CONNECTED)
-	ratelimit addclient $1 $3 $4 $5
+	[ $4 = 0 -o $5 = 0 ] && {
+		ubus call ratelimit client_set '{"device": "'$1'", "address": "'$3'", "defaults": "'$(ubus call wifi iface | jsonfilter -e "@.$1.ssid")'" }'
+		logger ratelimit addclient $1 $3 $ssid
+		return
+	}
+	ubus call ratelimit client_set '{"device": "'$1'", "address": "'$3'", "rate_ingress": "'$4'mbit", "rate_egress": "'$5'mbit" }'
+	logger ratelimit addclient $1 $3 $4 $5
 	;;
 AP-STA-DISCONNECTED)
-	ratelimit delclient $1 $3
+	ubus call ratelimit client_delete '{ "address": "'$3'" }'
+	logger ratelimit delclient $3
 	;;
 esac