123 lines
3.0 KiB
Bash
Executable File
123 lines
3.0 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" >> $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
|
|
status=$(/usr/bin/wget --no-check-certificate --spider --timeout=$timeout --tries $try "$url" 2>/dev/null)
|
|
[ "$?" == 0 ] && status=200
|
|
echo $status
|
|
}
|
|
|
|
test_proxy() {
|
|
local try=5
|
|
result=0
|
|
status=$(test_url "https://www.google.com" $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=$4
|
|
local b_tcp_nodes=$5
|
|
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" == 2 ]; then
|
|
echolog "自动切换检测：无法连接到网络，请检查网络是否正常！"
|
|
return 1
|
|
elif [ "$status" == 1 ]; then
|
|
echolog "自动切换检测：${type}_${index}节点异常，开始切换节点！"
|
|
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 $3 $index $new_node
|
|
echolog "自动切换检测：${type}_${index}节点切换完毕！"
|
|
return 0
|
|
elif [ "$status" == 0 ]; then
|
|
#echolog "自动切换检测：${type}_${index}节点正常。"
|
|
return 0
|
|
fi
|
|
}
|
|
|
|
start() {
|
|
if [ "$(ps -w | grep -v grep | grep $CONFIG/test.sh | wc -l)" -gt 2 ]; then
|
|
exit 1
|
|
fi
|
|
|
|
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 REDIR tcp $i "$tmp"
|
|
}
|
|
done
|
|
delay=$(config_t_get auto_switch testing_time 1)
|
|
sleep ${delay}m
|
|
done
|
|
}
|
|
|
|
case $1 in
|
|
test_url)
|
|
test_url $2
|
|
;;
|
|
*)
|
|
start
|
|
;;
|
|
esac |