openmptcprouter-feeds/openmptcprouter/files/bin/omr-speedtest

#!/bin/sh
#
# Copyright (C) 2018-2024 Ycarus (Yannick Chabanois) <ycarus@zugaina.org> for OpenMPTCProuter
#
# This is free software, licensed under the GNU General Public License v2.
# See /LICENSE for more information.
#
# This script check speed using Speedtest servers

INTERFACE="$1"

bypass_host_enable() {
	INTERFACE=$1
	HOST=$2
	[ -n "$(tc qdisc show dev $INTERFACE | grep ingress)" ] && /etc/init.d/sqm stop $INTERFACE
	domain=$(echo $HOST | awk -F/ '{print $3}')
	if [ "$IPV6" = true ]; then
		hostip=$(dig +nocmd +noall +answer AAAA $domain | grep -v CNAME | awk '{print $5}' | tr '\n' ' ')
		if [ -n "$(ipset list 2>/dev/null | grep ss_rules6)" ]; then
			for ip in $hostip; do
				ipset add ss_rules6_dst_bypass_all $ip
			done
		fi
		if [ -n "$(nft list set inet fw4 omr_dst_bypass_all_6 2>/dev/null)" ]; then
			for ip in $hostip; do
				nft add element inet fw4 omr_dst_bypass_all_6 { $ip } >/dev/null 2>&1
			done
		fi
	else
		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
	fi
}


bypass_host_disable() {
	INTERFACE=$1
	HOST=$2
	domain=$(echo $HOST | awk -F/ '{print $3}')
	if [ "$IPV6" = true ]; then
		hostip=$(dig +nocmd +noall +answer AAAA $domain | grep -v CNAME | awk '{print $5}' | tr '\n' ' ')
		if [ -n "$(ipset list 2>/dev/null | grep ss_rules6)" ]; then
			for ip in $hostip; do
				ipset del ss_rules6_dst_bypass_all $ip
			done
		fi
		if [ -n "$(nft list set inet fw4 omr_dst_bypass_all_6 2>/dev/null)" ]; then
			for ip in $hostip; do
				nft delete element inet fw4 omr_dst_bypass_all_6 { $ip } >/dev/null 2>&1
			done
		fi
	else
		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 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
	fi
	/etc/init.d/sqm start $INTERFACE
}


echo "Download server list..."
wget -q -O /tmp/speedtest.lst https://www.speedtest.net/api/js/servers
bestuploadurl=""
besthost=""
bestpinghost=""
bestping="999"
echo "Select best server..."
while read line; do
	host=$(echo $line | jsonfilter -e '@.host')
	pinghost=$(echo $host | awk -F: '{print $1}')
	uploadurl=$(echo $line | jsonfilter -e '@.host')
	ping=$(ping -c1 -w1 $pinghost | cut -d "/" -s -f5 | cut -d "." -f1 | tr -d '\n')
	echo -n "."
	if [ -n "$ping" ] && [ "$ping" -lt "$bestping" ]; then
		bestping=$ping
		bestuploadurl=$uploadurl
		besthost=$host
		bestpinghost=$pinghost
	fi
done < <(cat /tmp/speedtest.lst | jsonfilter -e '@[*]')
echo
echo "Done: url: $bestuploadurl - host: $besthost - ping: $bestping"
echo "Download test:"
if [ -z "$INTERFACE" ]; then
	curl -4 $besthost/speedtest/random7000x7000.jpg >/dev/null || echo
else
	bypass_host_enable $INTERFACE $bestpinghost
	curl -4 --interface $INTERFACE $besthost/speedtest/random7000x7000.jpg >/dev/null || echo
	bypass_host_disable $INTERFACE $bestpinghost
fi