#!/bin/sh
# (c) Yannick Chabanois (ycarus@zugaina.org) for OpenMPTCProuter
#
#

FORCEVPS=false
FASTTEST=false
if [ "$1" = "forcevps" ]; then
	FORCEVPS=true
fi
if [ "$1" = "fasttest" ]; then
	FASTTEST=true
fi
for i in $@; do :; done
INTERFACE="$i"
[ "$INTERFACE" = "forcevps" ] || [ "$INTERFACE" = "fasttest" ] && INTERFACE=""

[ -n "$INTERFACE" ] && [ ! -d "/sys/class/net/$INTERFACE" ] && {
	echo "You must use a real interface. You wan find them using 'ip a' for example (not $INTERFACE)"
	exit 0
}

[ "$FASTTEST" = true ] || echo "Select best test server..."
HOSTLST="http://scaleway.testdebit.info/10G.iso http://bordeaux.testdebit.info/10G.iso http://aix-marseille.testdebit.info/10G.iso http://lyon.testdebit.info/10G.iso http://lille.testdebit.info/10G.iso http://paris.testdebit.info/10G.iso http://appliwave.testdebit.info/10G/10G.iso http://speedtest.frankfurt.linode.com/garbage.php?ckSize=10000 http://speedtest.tokyo2.linode.com/garbage.php?ckSize=10000 http://speedtest.singapore.linode.com/garbage.php?ckSize=10000 http://speedtest.newark.linode.com/garbage.php?ckSize=10000 http://speedtest.atlanta.linode.com/garbage.php?ckSize=10000 http://speedtest.dallas.linode.com/garbage.php?ckSize=10000 http://speedtest.fremont.linode.com/garbage.php?ckSize=10000 http://ipv4.bouygues.testdebit.info/10G.iso http://par.download.datapacket.com/10000mb.bin http://nyc.download.datapacket.com/10000mb.bin http://ams.download.datapacket.com/10000mb.bin http://fra.download.datapacket.com/10000mb.bin http://lon.download.datapacket.com/10000mb.bin http://mad.download.datapacket.com/10000mb.bin http://prg.download.datapacket.com/10000mb.bin http://sto.download.datapacket.com/10000mb.bin http://vie.download.datapacket.com/10000mb.bin http://war.download.datapacket.com/10000mb.bin http://atl.download.datapacket.com/10000mb.bin http://chi.download.datapacket.com/10000mb.bin http://lax.download.datapacket.com/10000mb.bin http://mia.download.datapacket.com/10000mb.bin http://nyc.download.datapacket.com/10000mb.bin http://speedtest.milkywan.fr/files/10G.iso"
bestping="9999"
for pinghost in $HOSTLST; do
	domain=$(echo $pinghost | awk -F/ '{print $3}')
	if [ -z "$INTERFACE" ] || [ "$FORCEVPS" = true ]; then
		ping=$(ping -4 -c1 -w2 $domain | cut -d "/" -s -f5 | cut -d "." -f1 | tr -d '\n')
	else
		ping=$(ping -4 -c1 -w2 -I $INTERFACE -B $domain | cut -d "/" -s -f5 | cut -d "." -f1 | tr -d '\n')
	fi
	[ "$FASTTEST" = true ] || echo "host: $domain - ping: $ping"
	if [ -n "$ping" ] && [ "$ping" -lt "$bestping" ]; then
		bestping=$ping
		HOST=$pinghost
	fi
done

[ -z "$HOST" ] && HOST="http://speedtest.milkywan.fr/files/10G.iso"

[ "$FASTTEST" = true ] || echo "Best server is $HOST, running test:"
trap : HUP INT TERM
if [ -z "$INTERFACE" ]; then
	if [ "$FASTTEST" = true ]; then 
		avg_speed=$(curl -4 --max-time 10 -o /dev/null -qfsS -w '%{speed_download}' $HOST 2>/dev/null)
		echo "$avg_speed"
	else
		curl -4 -o /dev/null $HOST || echo
	fi
else
	[ -n "$(tc qdisc show dev $INTERFACE | grep ingress)" ] && /etc/init.d/sqm stop $INTERFACE
	domain=$(echo $HOST | awk -F/ '{print $3}')
	hostip=$(dig +nocmd +noall +answer A $domain | grep -v CNAME | awk '{print $5}' | tr '\n' ' ')
	if [ -n "$(ipset list 2>/dev/null | grep ss_rules)" ]; then
		for ip in $hostip; do
			ipset add ss_rules_dst_bypass_all $ip
		done
	fi
	if [ -n "$(nft list set inet fw4 omr_dst_bypass_all_4 2>/dev/null)" ]; then
		for ip in $hostip; do
			nft add element inet fw4 omr_dst_bypass_all_4 { $ip } >/dev/null 2>&1
		done
	fi
	if [ "$FASTTEST" = true ]; then 
		avg_speed=$(curl -4 --max-time 10 -o /dev/null -qfsS -w '%{speed_download}' --interface $INTERFACE $HOST 2>/dev/null)
		echo "$avg_speed"
	else
		curl -4 -o /dev/null --interface $INTERFACE $HOST || echo
	fi
	if [ -n "$(ipset list 2>/dev/null | grep ss_rules)" ]; then
		for ip in $hostip; do
			ipset del ss_rules_dst_bypass_all $ip
		done
	fi
	if [ -n "$(nft list set inet fw4 omr_dst_bypass_all_4 2>/dev/null)" ]; then
		for ip in $hostip; do
			nft delete element inet fw4 omr_dst_bypass_all_4 { $ip } >/dev/null 2>&1
		done
	fi
	/etc/init.d/sqm start $INTERFACE
fi