157 lines
4.2 KiB
Bash
Executable File
157 lines
4.2 KiB
Bash
Executable File
#!/bin/sh
|
||
|
||
CONFIG=passwall
|
||
LOG_FILE=/var/log/$CONFIG.log
|
||
|
||
echolog() {
|
||
local d="$(date "+%Y-%m-%d %H:%M:%S")"
|
||
#echo -e "$d: $1"
|
||
echo -e "$d: $1" >> $LOG_FILE
|
||
}
|
||
|
||
config_t_get() {
|
||
local index=0
|
||
[ -n "$4" ] && index=$4
|
||
local ret=$(uci -q get $CONFIG.@$1[$index].$2 2>/dev/null)
|
||
echo ${ret:=$3}
|
||
}
|
||
|
||
test_url() {
|
||
local url=$1
|
||
local try=1
|
||
[ -n "$2" ] && try=$2
|
||
local timeout=2
|
||
[ -n "$3" ] && timeout=$3
|
||
local extra_params=$4
|
||
status=$(/usr/bin/curl -I -o /dev/null -skL $extra_params --connect-timeout $timeout --retry $try -w %{http_code} "$url")
|
||
case "$status" in
|
||
204|\
|
||
200)
|
||
status=200
|
||
;;
|
||
esac
|
||
echo $status
|
||
}
|
||
|
||
test_proxy() {
|
||
local try=3
|
||
result=0
|
||
status=$(test_url "https://www.google.com/generate_204" $try)
|
||
if [ "$status" = "200" ]; then
|
||
result=0
|
||
else
|
||
status2=$(test_url "https://www.baidu.com" $try)
|
||
if [ "$status2" = "200" ]; then
|
||
result=1
|
||
else
|
||
result=2
|
||
fi
|
||
fi
|
||
echo $result
|
||
}
|
||
|
||
test_auto_switch() {
|
||
local type=$1
|
||
local index=$3
|
||
local b_tcp_nodes=$4
|
||
local now_node
|
||
if [ -f "/var/etc/$CONFIG/id/${type}_${index}" ]; then
|
||
now_node=$(cat /var/etc/$CONFIG/id/${type}_${index})
|
||
else
|
||
return 1
|
||
fi
|
||
|
||
status=$(test_proxy)
|
||
if [ "$status" == 0 ]; then
|
||
#echolog "自动切换检测：${type}_${index}节点正常。"
|
||
return 0
|
||
elif [ "$status" == 2 ]; then
|
||
echolog "自动切换检测：无法连接到网络，请检查网络是否正常！"
|
||
return 2
|
||
elif [ "$status" == 1 ]; then
|
||
echolog "自动切换检测：${type}_${index}节点异常，开始切换节点！"
|
||
|
||
#检测自动切换列表主节点是否能使用
|
||
local main_node=$(echo $b_tcp_nodes | awk -F ' ' '{print $1}')
|
||
if [ "$now_node" != "$main_node" ]; then
|
||
local tmp_port=$(/usr/share/passwall/app.sh get_new_port 61080 tcp)
|
||
/usr/share/passwall/app.sh run_socks "$main_node" "127.0.0.1" "$tmp_port" "/var/etc/passwall/auto_switch_main.json" "10"
|
||
proxy_status=$(test_url "https://www.google.com/generate_204" 3 3 "-x socks5h://127.0.0.1:$tmp_port")
|
||
ps -w | grep -v "grep" | grep "/var/etc/passwall/auto_switch_main.json" | awk '{print $1}' | xargs kill -9 >/dev/null 2>&1
|
||
if [ "$proxy_status" -eq 200 ]; then
|
||
#主节点正常，切换到主节点
|
||
echolog "自动切换检测：${type}_${index}主节点正常，切换到主节点！"
|
||
/usr/share/passwall/app.sh node_switch $type $2 $index $main_node
|
||
return 0
|
||
fi
|
||
fi
|
||
|
||
local new_node
|
||
in_backup_nodes=$(echo $b_tcp_nodes | grep $now_node)
|
||
# 判断当前节点是否存在于备用节点列表里
|
||
if [ -z "$in_backup_nodes" ]; then
|
||
# 如果不存在，设置第一次节点为新的节点
|
||
new_node=$(echo $b_tcp_nodes | awk -F ' ' '{print $1}')
|
||
else
|
||
# 如果存在，设置下一个备用节点为新的节点
|
||
#local count=$(expr $(echo $b_tcp_nodes | grep -o ' ' | wc -l) + 1)
|
||
local next_node=$(echo $b_tcp_nodes | awk -F "$now_node" '{print $2}' | awk -F " " '{print $1}')
|
||
if [ -z "$next_node" ]; then
|
||
new_node=$(echo $b_tcp_nodes | awk -F ' ' '{print $1}')
|
||
else
|
||
new_node=$next_node
|
||
fi
|
||
fi
|
||
/usr/share/passwall/app.sh node_switch $type $2 $index $new_node
|
||
sleep 10s
|
||
# 切换节点后等待10秒后再检测一次，如果还是不通继续切，直到可用为止
|
||
status2=$(test_proxy)
|
||
if [ "$status2" -eq 0 ]; then
|
||
echolog "自动切换检测：${type}_${index}节点切换完毕！"
|
||
return 0
|
||
elif [ "$status2" -eq 1 ]; then
|
||
test_auto_switch $1 $2 $3 "$4"
|
||
elif [ "$status2" -eq 2 ]; then
|
||
return 2
|
||
fi
|
||
fi
|
||
}
|
||
|
||
start() {
|
||
(
|
||
flock -xn 200
|
||
[ "$?" != "0" ] && exit 0
|
||
|
||
ENABLED=$(config_t_get global enabled 0)
|
||
[ "$ENABLED" != 1 ] && return 1
|
||
ENABLED=$(config_t_get auto_switch enable 0)
|
||
[ "$ENABLED" != 1 ] && return 1
|
||
delay=$(config_t_get auto_switch testing_time 1)
|
||
sleep ${delay}m
|
||
while [ "$ENABLED" -eq 1 ]
|
||
do
|
||
# TCP_NODE_NUM=$(config_t_get global_other tcp_node_num 1)
|
||
# 暂时只能检测TCP1
|
||
TCP_NODE_NUM=1
|
||
for i in $(seq 1 $TCP_NODE_NUM); do
|
||
eval TCP_NODE$i=\"$(config_t_get auto_switch tcp_node$i nil)\"
|
||
eval tmp=\$TCP_NODE$i
|
||
[ -n "$tmp" -a "$tmp" != "nil" ] && {
|
||
test_auto_switch TCP tcp $i "$tmp"
|
||
}
|
||
done
|
||
delay=$(config_t_get auto_switch testing_time 1)
|
||
sleep ${delay}m
|
||
done
|
||
) 200>"/var/lock/passwall_auto_switch.lock"
|
||
}
|
||
|
||
case $1 in
|
||
test_url)
|
||
test_url $2
|
||
;;
|
||
*)
|
||
start
|
||
;;
|
||
esac
|