mirror of
https://github.com/kiddin9/op-packages.git
synced 2026-10-26 20:22:51 +08:00
225 lines
7.1 KiB
Bash
Executable File
225 lines
7.1 KiB
Bash
Executable File
#!/bin/sh
|
|
|
|
. /usr/share/passwall2/utils.sh
|
|
APP_FILE=${APP_PATH}/app.sh
|
|
|
|
flag=0
|
|
|
|
check_process() {
|
|
while busybox pgrep -af "${CONFIG}/" | grep -E 'app\.sh.*(start|stop)|nftables\.sh|iptables\.sh|subscribe\.lua' >/dev/null; do
|
|
sleep 6s
|
|
done
|
|
}
|
|
|
|
test_url() {
|
|
local url="$1"
|
|
local try="${2:-1}"
|
|
local timeout="${3:-2}"
|
|
local extra_params="$4"
|
|
local repeat="$5"
|
|
|
|
if [ -z "$curl_retry_all_errors" ]; then
|
|
if /usr/bin/curl --help all | grep -q "\-\-retry-all-errors"; then
|
|
curl_retry_all_errors=1
|
|
fi
|
|
fi
|
|
[ "$curl_retry_all_errors" = "1" ] && extra_params="--retry-all-errors ${extra_params}"
|
|
|
|
local max_time=$((timeout * (try + 1) + try + 3))
|
|
curl_test() {
|
|
/usr/bin/curl -skIL -o /dev/null ${extra_params} --max-time ${max_time} --connect-timeout ${timeout} --retry ${try} --retry-delay 1 -w "%{http_code}" "$url"
|
|
}
|
|
|
|
local status=$(curl_test)
|
|
[ "$status" = "204" ] && status=200
|
|
if [ "$status" = "200" ] && [ "$repeat" = "1" ]; then
|
|
sleep 3s
|
|
status=$(curl_test)
|
|
[ "$status" = "204" ] && status=200
|
|
fi
|
|
echo $status
|
|
}
|
|
|
|
test_proxy() {
|
|
local result=0
|
|
local status=$(test_url "${probe_url}" ${retry_num} ${connect_timeout} "-x socks5h://127.0.0.1:${socks_port}")
|
|
if [ "$status" = "200" ]; then
|
|
result=0
|
|
else
|
|
local status2=$(test_url "https://www.baidu.com" ${retry_num} ${connect_timeout})
|
|
if [ "$status2" = "200" ]; then
|
|
result=1
|
|
else
|
|
result=2
|
|
ping -c 3 -W 1 223.5.5.5 > /dev/null 2>&1
|
|
[ $? -eq 0 ] && {
|
|
result=1
|
|
}
|
|
fi
|
|
fi
|
|
echo $result
|
|
}
|
|
|
|
test_node() {
|
|
local node_id=$1
|
|
local _type=$(echo $(config_n_get ${node_id} type) | tr 'A-Z' 'a-z')
|
|
[ -n "${_type}" ] && {
|
|
check_process
|
|
local _tmp_port=$(get_new_port 48800 tcp,udp)
|
|
NO_REC_PROCESS=1 $APP_FILE run_socks flag="test_node_${node_id}" node=${node_id} bind=127.0.0.1 socks_port=${_tmp_port} config_file=test_node_${node_id}.json
|
|
sleep 2s
|
|
local curlx="socks5h://127.0.0.1:${_tmp_port}"
|
|
local _proxy_status=$(test_url "${probe_url}" ${retry_num} ${connect_timeout} "-x $curlx" 1)
|
|
# Kill the SS plugin process
|
|
local pid_file="${TMP_PATH}/test_node_${node_id}_plugin.pid"
|
|
[ -s "$pid_file" ] && kill -9 "$(head -n 1 "$pid_file")" >/dev/null 2>&1
|
|
busybox pgrep -af "test_node_${node_id}" | awk '! /socks_auto_switch\.sh/{print $1}' | xargs kill -9 >/dev/null 2>&1
|
|
rm -rf ${TMP_PATH}/test_node_${node_id}*.*
|
|
if [ "${_proxy_status}" -eq 200 ]; then
|
|
return 0
|
|
fi
|
|
}
|
|
return 1
|
|
}
|
|
|
|
try_switch_backup() {
|
|
local b_nodes="$1"
|
|
local now_node="$2"
|
|
local total tried
|
|
local new_node msg
|
|
local first_node found node
|
|
|
|
# Only one backup node, alternate with the primary node.
|
|
if [ "$backup_node_num" -eq 1 ]; then
|
|
b_nodes="$b_nodes $main_node"
|
|
fi
|
|
|
|
total=$(printf "%s\n" "$b_nodes" | wc -w)
|
|
tried=0
|
|
|
|
while [ "$tried" -lt "$total" ]; do
|
|
new_node=""
|
|
first_node=""
|
|
found=""
|
|
for node in $b_nodes; do
|
|
[ -z "$first_node" ] && first_node="$node" # Record first node.
|
|
[ "$found" = "1" ] && { new_node="$node"; break; } # Find the current node and then get the next one.
|
|
[ "$node" = "$now_node" ] && found=1 # flag the found current node.
|
|
done
|
|
# If the current node is not found, or if it is the last node, select the first node.
|
|
[ -z "$new_node" ] && new_node="$first_node"
|
|
|
|
local node_role node_type node_remarks
|
|
if [ "$new_node" = "$main_node" ]; then
|
|
node_role="$(i18n "main node")"
|
|
else
|
|
node_role="$(i18n "next backup node")"
|
|
fi
|
|
node_type=$(config_n_get $new_node type)
|
|
node_remarks=$(config_n_get $new_node remarks)
|
|
log_i18n 0 "Socks switch detection: Port [%s] try %s [%s:%s]." "${socks_port}" "${node_role}" "${node_type}" "${node_remarks}"
|
|
|
|
if test_node ${new_node}; then
|
|
check_process
|
|
log_i18n 0 "Socks switch detection: Port [%s] [%s:[%s]] normal, switch to this node!" "${socks_port}" "${node_type}" "${node_remarks}"
|
|
NO_REC_PROCESS=1 $APP_FILE socks_node_switch flag=${id} new_node=${new_node}
|
|
[ $? -eq 0 ] && {
|
|
log_i18n 0 "Socks switch detection: Port [%s] node switch complete!" "${socks_port}"
|
|
}
|
|
return 0
|
|
fi
|
|
log_i18n 0 "Socks switch detection: Port [%s] [%s:[%s]] abnormal." "${socks_port}" "${node_type}" "${node_remarks}"
|
|
now_node="$new_node"
|
|
tried=$((tried + 1))
|
|
done
|
|
|
|
log_i18n 0 "Socks switch detection: Port [%s] all nodes are unavailable!" "${socks_port}"
|
|
return 1
|
|
}
|
|
|
|
test_auto_switch() {
|
|
flag=$((flag + 1))
|
|
local b_nodes=$1
|
|
local now_node=$2
|
|
[ -z "$now_node" ] && {
|
|
if [ -n "$(get_cache_var "${id}")" ]; then
|
|
now_node=$(get_cache_var "${id}")
|
|
else
|
|
#log_i18n 0 "Socks switch detection: Port [%s] Unknown error." "${socks_port}"
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
[ $flag -le 1 ] && main_node=$now_node
|
|
|
|
local status=$(test_proxy)
|
|
if [ "$status" = "2" ]; then
|
|
log_i18n 0 "Socks switch detection: Port [%s] Unable to connect to the network. Please check if the network is working properly!" "${socks_port}"
|
|
return 2
|
|
fi
|
|
|
|
# Check if the main node is usable
|
|
if [ "$restore_switch" = "1" ] && [ -n "$main_node" ] && [ "$now_node" != "$main_node" ]; then
|
|
test_node ${main_node}
|
|
[ $? -eq 0 ] && {
|
|
check_process
|
|
# The main node is working properly; switch to the main node.
|
|
log_i18n 0 "Socks switch detection: Port [%s] Primary node [%s: [%s]] is normal. Switch to the primary node!" "${socks_port}" "$(config_n_get $main_node type)" "$(config_n_get $main_node remarks)"
|
|
NO_REC_PROCESS=1 $APP_FILE socks_node_switch flag=${id} new_node=${main_node}
|
|
[ $? -eq 0 ] && {
|
|
log_i18n 0 "Socks switch detection: Port [%s] node switch complete!" "${socks_port}"
|
|
}
|
|
return 0
|
|
}
|
|
fi
|
|
|
|
[ "$status" = "0" ] && {
|
|
#log_i18n 0 "Socks switch detection: Port [%s] [%s:[%s]] normal." "${socks_port}" "$(config_n_get $now_node type)" "$(config_n_get $now_node remarks)"
|
|
return 0
|
|
}
|
|
|
|
log_i18n 0 "Socks switch detection: Port [%s] [%s:[%s]] abnormal." "${socks_port}" "$(config_n_get $now_node type)" "$(config_n_get $now_node remarks)"
|
|
try_switch_backup "$b_nodes" "$now_node"
|
|
return $?
|
|
}
|
|
|
|
start() {
|
|
id=$1
|
|
LOCK_FILE=${LOCK_PATH}/${CONFIG}_socks_auto_switch_${id}.lock
|
|
main_node=$(config_n_get $id node)
|
|
socks_port=$(config_n_get $id port 0)
|
|
delay=$(config_n_get $id autoswitch_testing_time 30)
|
|
connect_timeout=$(config_n_get $id autoswitch_connect_timeout 3)
|
|
retry_num=$(config_n_get $id autoswitch_retry_num 1)
|
|
restore_switch=$(config_n_get $id autoswitch_restore_switch 0)
|
|
probe_url=$(config_n_get $id autoswitch_probe_url "https://www.google.com/generate_204")
|
|
backup_node=$(lua_api "get_socks_backup_nodes(\"${id}\")")
|
|
if [ -n "$backup_node" ]; then
|
|
backup_node_num=$(printf "%s\n" "$backup_node" | wc -w)
|
|
log_i18n 0 "Socks switch detection: Port [%s] Number of backup nodes: %s" "${socks_port}" "${backup_node_num}"
|
|
if [ "$backup_node_num" -eq 1 ]; then
|
|
[ "$main_node" = "$backup_node" ] && return
|
|
elif [ "$backup_node_num" -gt 1 ]; then
|
|
[ "$restore_switch" != "1" ] && {
|
|
[ -z "$(echo $backup_node | grep -F "$main_node")" ] && backup_node="${backup_node} ${main_node}"
|
|
}
|
|
fi
|
|
else
|
|
log_i18n 0 "Socks switch detection: Port [%s] Number of backup nodes: %s" "${socks_port}" "0"
|
|
return
|
|
fi
|
|
while [ -n "$backup_node" ]; do
|
|
[ -f "$LOCK_FILE" ] && {
|
|
sleep 6s
|
|
continue
|
|
}
|
|
check_process
|
|
touch $LOCK_FILE
|
|
test_auto_switch "$backup_node"
|
|
rm -f $LOCK_FILE
|
|
sleep ${delay}
|
|
done
|
|
}
|
|
|
|
start $@
|