X-Git-Url: https://git.archive.openwrt.org/?p=project%2Fluci.git;a=blobdiff_plain;f=applications%2Fluci-splash%2Froot%2Fetc%2Finit.d%2Fluci_splash;h=d9ef21c217406b01768abf464a67cd3bb4c84762;hp=85ea8ecaafc98fce8380b11d964d762cc164d504;hb=5e9fcc76c02914aa75649492b97aa7d37358c068;hpb=251502e2cd0c47b426cfd4d4f183128ca528b91b

diff --git a/applications/luci-splash/root/etc/init.d/luci_splash b/applications/luci-splash/root/etc/init.d/luci_splash
index 85ea8ecaa..d9ef21c21 100755
--- a/applications/luci-splash/root/etc/init.d/luci_splash
+++ b/applications/luci-splash/root/etc/init.d/luci_splash
@@ -1,5 +1,14 @@
 #!/bin/sh /etc/rc.common
 START=70
+EXTRA_COMMANDS=clear_leases
+SPLASH_INTERFACES=""
+LIMIT_DOWN=0
+LIMIT_DOWN_BURST=0
+LIMIT_UP=0
+
+silent() {
+	"$@" 2>/dev/null
+}
 
 iface_add() {
 	local cfg="$1"
@@ -9,6 +18,9 @@ iface_add() {
 	
 	config_get net "$cfg" network
 	[ -n "$net" ] || return 0
+
+	config_get ifname "$net" ifname
+	[ -n "$ifname" ] || return 0
 	
 	config_get ipaddr "$net" ipaddr
 	[ -n "$ipaddr" ] || return 0
@@ -17,24 +29,137 @@ iface_add() {
 	[ -n "$netmask" ] || return 0
 	
 	eval "$(ipcalc.sh $ipaddr $netmask)"
-	
-	iptables -t nat -A zone_${zone}_prerouting -s "$NETWORK/$PREFIX" -p ! tcp -j luci_splash_portal
-	iptables -t nat -A zone_${zone}_prerouting -s "$NETWORK/$PREFIX" -d ! "$ipaddr" -j luci_splash_portal
-	iptables -t nat -A zone_${zone}_prerouting -s "$NETWORK/$PREFIX" -d "$ipaddr" -p tcp -m multiport ! --dport 22,80,443 -j luci_splash_portal
+
+	iptables -t nat -A prerouting_${zone} -j luci_splash_prerouting
+	iptables -t nat -A luci_splash_prerouting -s "$NETWORK/$PREFIX" -p ! tcp -j luci_splash_portal
+	iptables -t nat -A luci_splash_prerouting -s "$NETWORK/$PREFIX" -d ! "$ipaddr" -j luci_splash_portal
+	iptables -t nat -A luci_splash_prerouting -s "$NETWORK/$PREFIX" -d "$ipaddr" -p tcp -m multiport ! --dport 22,80,443 -j luci_splash_portal
+
+	qos_iface_add "$ifname"
+
+	append SPLASH_INTERFACES "$ifname"
+}
+
+iface_del() {
+	config_get zone "$1" zone                                                                
+	[ -n "$zone" ] || return 0
+
+	config_get ifname "$1" ifname
+	[ -n "$ifname" ] || return 0
+
+	while iptables -t nat -D prerouting_${zone} -j luci_splash_prerouting 2>&-; do :; done
+
+	qos_iface_del "$ifname"
 }
 
 blacklist_add() {
 	local cfg="$1"
 	
 	config_get mac "$cfg" mac
-	[ -n "$mac" ] && iptables -t nat -A luci_splash_portal -m mac --mac-source "$mac" -j DROP
+	[ -n "$mac" ] && {
+		iptables -t filter -I luci_splash_counter -m mac --mac-source "$mac" -j RETURN
+		iptables -t nat    -I luci_splash_leases  -m mac --mac-source "$mac" -j DROP
+	}
 }
 
 whitelist_add() {
 	local cfg="$1"
 	
 	config_get mac "$cfg" mac
-	[ -n "$mac" ] && iptables -t nat -A luci_splash_portal -m mac --mac-source "$mac" -j RETURN
+	[ -n "$mac" ] && {
+		iptables -t filter -I luci_splash_counter -m mac --mac-source "$mac" -j RETURN
+		iptables -t nat    -I luci_splash_leases  -m mac --mac-source "$mac" -j RETURN
+	}
+}
+
+lease_add() {
+	local cfg="$1"
+	
+	config_get mac "$cfg" mac
+	config_get ban "$cfg" kicked
+
+	ban=${ban:+DROP}
+
+	[ -n "$mac" ] && {
+		local oIFS="$IFS"; IFS=":"
+		set -- $mac
+		IFS="$oIFS"; unset oIFS
+	
+		local mac_pre="$1$2"
+		local mac_post="$3$4$5$6"
+		local handle="$6"
+
+		iptables -t filter -I luci_splash_counter -m mac --mac-source "$mac" -j RETURN
+		iptables -t mangle -I luci_splash_mark    -m mac --mac-source "$mac" -j MARK --set-mark 79
+		iptables -t nat    -I luci_splash_leases  -m mac --mac-source "$mac" -j "${ban:-RETURN}"
+
+		for i in $SPLASH_INTERFACES; do
+			tc filter add dev $i parent 77:0 protocol ip prio 2 handle ::$handle u32 \
+				match u16 0x0800 0xFFFF at -2 match u32 0x$mac_post 0xFFFFFFFF at -12 \
+				match u16 0x$mac_pre 0xFFFF at -14 flowid 77:10
+		done
+	}
+}
+
+qos_iface_add() {
+	local iface="$1"
+
+	# 77 -> download root qdisc
+	# 78 -> upload root qdisc
+	# 79 -> fwmark
+
+	silent tc qdisc del dev "$iface" root handle 77:
+
+	if [ "$LIMIT_UP" -gt 0 -a "$LIMIT_DOWN" -gt 0 ]; then
+		tc qdisc add dev "$iface" root handle 77: htb
+
+		# assume maximum rate of 20.000 kilobit for wlan
+		tc class add dev "$iface" parent 77: classid 77:1 htb rate 20000kbit
+
+		# set download limit and burst
+		tc class add dev "$iface" parent 77:1 classid 77:10 htb \
+			rate ${LIMIT_DOWN}kbit ceil ${LIMIT_DOWN_BURST}kbit prio 2
+
+		tc qdisc add dev "$iface" parent 77:10 handle 78: sfq perturb 10
+
+		# adding ingress can result in "File exists" if qos-scripts are active
+		silent tc qdisc add dev "$iface" ingress
+
+		# set client upload speed
+		tc filter add dev "$iface" parent ffff: protocol ip prio 1 \
+			handle 79 fw police rate ${LIMIT_UP}kbit mtu 6k burst 6k drop
+	fi	
+}
+
+qos_iface_del() {
+	local iface="$1"
+
+	silent tc qdisc del dev "$iface" root handle 77:
+	silent tc qdisc del dev "$iface" root handle 78:
+	silent tc filter del dev "$iface" parent ffff: protocol ip prio 1 handle 79 fw
+}
+
+boot() {
+	### Setup splash-relay
+	uci get lucid.splashr || {
+uci batch <<EOF
+	set lucid.splashr=daemon
+	set lucid.splashr.slave=httpd
+	add_list lucid.splashr.address=8082
+	add_list lucid.splashr.publisher=splashredir
+	set lucid.splashr.enabled=1
+
+	set lucid.splashredir=Redirector
+	set lucid.splashredir.name=Splashd
+	set lucid.splashredir.virtual='/'
+	set lucid.splashredir.physical=':80/luci/splash'
+
+	commit lucid
+EOF
+	}
+
+	### We are started by the firewall include
+	exit 0
 }
 
 start() {
@@ -43,17 +168,59 @@ start() {
 	scan_interfaces
 	config_load luci_splash
 	
+	### Find QoS limits
+	config_get LIMIT_UP general limit_up
+	config_get LIMIT_DOWN general limit_down
+	config_get LIMIT_DOWN_BURST general limit_down_burst
+
+	LIMIT_UP="${LIMIT_UP:-0}"
+	LIMIT_DOWN="${LIMIT_DOWN:-0}"
+	LIMIT_DOWN_BURST="${LIMIT_DOWN_BURST:-$(($LIMIT_DOWN * 2))}"
+
+	### Load required modules
+	[ "$LIMIT_IP" -gt 0 -a "$LIMIT_DOWN" -gt 0 ] && {
+		silent insmod cls_fw
+		silent insmod cls_route
+		silent insmod cls_rsvp
+		silent insmod cls_rsvp6
+		silent insmod cls_tcindex
+		silent insmod cls_u32
+		silent insmod sch_cbq
+		silent insmod sch_csz
+		silent insmod sch_dsmark
+		silent insmod sch_gred
+		silent insmod sch_hfsc
+		silent insmod sch_htb
+		silent insmod sch_ingress
+		silent insmod sch_red
+		silent insmod sch_tbf
+		silent insmod sch_teql
+	}
+
 	### Create subchains
-	iptables -t nat -N luci_splash
-	iptables -t nat -N luci_splash_portal
-	iptables -t nat -N luci_splash_leases
-	
+	iptables -t filter -N luci_splash_counter
+	iptables -t nat    -N luci_splash_portal
+	iptables -t nat    -N luci_splash_leases
+	iptables -t nat    -N luci_splash_prerouting
+
+	[ "$LIMIT_UP" -gt 0 -a "$LIMIT_DOWN" -gt 0 ] && \
+		iptables -t mangle -N luci_splash_mark
+
 	### Build the main and portal rule
+	config_foreach iface_add iface
 	config_foreach blacklist_add blacklist
 	config_foreach whitelist_add whitelist
-	config_foreach iface_add iface
+	config_foreach lease_add lease
 	
 	### Build the portal rule
+	iptables -t filter -I INPUT      -j luci_splash_counter
+	iptables -t filter -I FORWARD    -j luci_splash_counter
+
+	[ "$LIMIT_UP" -gt 0 -a "$LIMIT_DOWN" -gt 0 ] && \
+		iptables -t mangle -I PREROUTING -j luci_splash_mark
+	
+	iptables -t nat -A luci_splash_portal -p udp --dport 33434:33523 -j RETURN
+	iptables -t nat -A luci_splash_portal -p icmp -j RETURN
 	iptables -t nat -A luci_splash_portal -p udp --dport 53 -j RETURN
 	iptables -t nat -A luci_splash_portal -j luci_splash_leases
 	
@@ -62,26 +229,42 @@ start() {
 	iptables -t nat -A luci_splash_leases -j DROP
 	
 	### Add crontab entry
-	grep luci-splash /var/spool/cron/crontabs/root >/dev/null 2>&1 || {
-		echo '*/5 * * * * 	/usr/sbin/luci-splash sync' >> /var/spool/cron/crontabs/root
+	test -f /etc/crontabs/root || touch /etc/crontabs/root
+	grep -q luci-splash /etc/crontabs/root || {
+		echo '*/5 * * * * 	/usr/sbin/luci-splash sync' >> /etc/crontabs/root
 	}
-
-	### Start the splash httpd
-	start-stop-daemon -S -b -q -x /usr/bin/luci-splashd
 }
 
-stop() {	
+stop() {
+	### Clear interface rules
+	config_load luci_splash
+	config_foreach iface_del iface
+
+	silent iptables -t filter -D INPUT      -j luci_splash_counter
+	silent iptables -t filter -D FORWARD    -j luci_splash_counter
+	silent iptables -t mangle -D PREROUTING -j luci_splash_mark
+        
 	### Clear subchains
-	iptables -t nat -F luci_splash_leases
-	iptables -t nat -F luci_splash_portal
-	iptables -t nat -F luci_splash	
+	silent iptables -t nat    -F luci_splash_leases
+	silent iptables -t nat    -F luci_splash_portal
+	silent iptables -t nat    -F luci_splash_prerouting
+	silent iptables -t filter -F luci_splash_counter
+	silent iptables -t mangle -F luci_splash_mark
 	
 	### Delete subchains
-	iptables -t nat -X luci_splash_leases
-	iptables -t nat -X luci_splash_portal
-	iptables -t nat -X luci_splash
+	silent iptables -t nat    -X luci_splash_leases
+	silent iptables -t nat    -X luci_splash_portal
+	silent iptables -t nat    -X luci_splash_prerouting
+	silent iptables -t filter -X luci_splash_counter
+	silent iptables -t mangle -X luci_splash_mark
+	
+	sed -ie '/\/usr\/sbin\/luci-splash sync/d' /var/spool/cron/crontabs/root
+}
 
-	### Stop the splash httpd
-	start-stop-daemon -K -q -x /usr/bin/luci-splashd
+	
+clear_leases() {
+	stop
+	while uci -P /var/state del luci_splash.@lease[0] 2>&-;do :; done
+	start
 }