From cc2df89ed3b334b9c16e9fbe74cf3b7b2cab2f55 Mon Sep 17 00:00:00 2001 From: "Ycarus (Yannick Chabanois)" Date: Fri, 29 Sep 2023 15:03:16 +0200 Subject: [PATCH] Add XRay and Shadowsocks Rust support for omr-tracker --- omr-tracker/files/bin/omr-tracker-ss | 45 +++++-- omr-tracker/files/bin/omr-tracker-xray | 150 +++++++++++++++++++++++ omr-tracker/files/etc/init.d/omr-tracker | 109 +++++++++++++++- 3 files changed, 289 insertions(+), 15 deletions(-) create mode 100755 omr-tracker/files/bin/omr-tracker-xray diff --git a/omr-tracker/files/bin/omr-tracker-ss b/omr-tracker/files/bin/omr-tracker-ss index b0d7e2ab4..c1f00a345 100755 --- a/omr-tracker/files/bin/omr-tracker-ss +++ b/omr-tracker/files/bin/omr-tracker-ss @@ -65,7 +65,7 @@ proxy=${OMR_TRACKER_PROXY:-127.0.0.1:1111} hosts=${OMR_TRACKER_HOSTS:-1.1.1.1 1.0.0.1} wait_test=${OMR_TRACKER_WAIT_TEST:-0} server=${OMR_TRACKER_SERVER:-sss0} - +type=${OMR_TRACKER_SS_TYPE:-libev} nodns=0 last=0 @@ -103,10 +103,16 @@ while true; do script_alert_up="$(uci -q get omr-tracker.proxy.script_alert_up)" [ -n "$script_alert_up" ] && eval $script_alert_up } - if [ -z "$($IPTABLESSAVE 2>/dev/null | grep :ssr)" ] && [ "$(uci -q get shadowsocks-libev.ss_rules.disabled)" != "1" ]; then - _log "Reload Shadowsocks rules" - /etc/init.d/shadowsocks-libev rules_up 2> /dev/null - _get_ip + if [ -z "$($IPTABLESSAVE 2>/dev/null | grep :ssr)" ]; then + if [ "$type" = "libev" ] && [ "$(uci -q get shadowsocks-libev.ss_rules.disabled)" != "1" ]; then + _log "Reload Shadowsocks rules" + /etc/init.d/shadowsocks-libev rules_up 2> /dev/null + _get_ip + elif [ "$type" = "rust" ] && [ "$(uci -q get shadowsocks-rust.ss_rules.disabled)" != "1" ]; then + _log "Reload Shadowsocks Rust rules" + /etc/init.d/shadowsocks-rust rules_up 2> /dev/null + _get_ip + fi fi [ "$(uci -q get openmptcprouter.omr.detected_public_ipv4)" = "" ] || ([ "$(uci -q get openmptcprouter.settings.disable_ipv6)" != "1" ] && [ "$(uci -q get openmptcprouter.omr.detected_public_ipv6)" = "" ]) && _get_ip last=0 @@ -119,11 +125,19 @@ while true; do OMR_TRACKER_STATUS_MSG="Shadowsocks ${server} is down (can't contact via http ${nocontact})" uci -q set openmptcprouter.omr.ss_${server}="down" uci -q commit openmptcprouter.omr - [ "$(uci show openmptcprouter.omr | grep ss_ | grep up)" = "" ] && /etc/init.d/shadowsocks-libev rules_down 2> /dev/null + if [ "$(uci show openmptcprouter.omr | grep ss_ | grep up)" = "" ]; then + [ "$type" = "libev" ] && /etc/init.d/shadowsocks-libev rules_down 2> /dev/null + [ "$type" = "rust" ] && /etc/init.d/shadowsocks-rust rules_down 2> /dev/null + fi _get_ip server_ping=false - serverip="$(uci -q get shadowsocks-libev.${server}.server)" - disabled="$(uci -q get shadowsocks-libev.${server}.disabled)" + if [ "$type" = "libev" ]; then + serverip="$(uci -q get shadowsocks-libev.${server}.server)" + disabled="$(uci -q get shadowsocks-libev.${server}.disabled)" + elif [ "$type" = "rust" ]; then + serverip="$(uci -q get shadowsocks-rust.${server}.server)" + disabled="$(uci -q get shadowsocks-rust.${server}.disabled)" + fi _ping_server $serverip if [ "$server_ping" = false ]; then _log "Server $server ($serverip) seems down, no answer to ping" @@ -146,10 +160,17 @@ while true; do script_alert_down="$(uci -q get omr-tracker.proxy.script_alert_down)" [ -n "$script_alert_down" ] && eval $script_alert_down - if [ "$disabled" != "1" ] && [ "$(pgrep ss-redir)" = "" ] && [ "$(uci -q get shadowsocks-libev.${server}.key)" != "" ]; then - _log "Can't find shadowsocks, restart it..." - /etc/init.d/shadowsocks-libev restart - sleep 5 + if [ "$disabled" != "1" ]; then + if [ "$type" = "libev" ] && [ "$(pgrep ss-redir)" = "" ] && [ "$(uci -q get shadowsocks-libev.${server}.key)" != "" ]; then + _log "Can't find shadowsocks, restart it..." + /etc/init.d/shadowsocks-libev restart + sleep 5 + fi + if [ "$type" = "rust" ] && [ "$(pgrep ss-redir)" = "" ] && [ "$(uci -q get shadowsocks-libev.${server}.key)" != "" ]; then + _log "Can't find shadowsocks rust, restart it..." + /etc/init.d/shadowsocks-rust restart + sleep 5 + fi fi sleep $wait_test fi diff --git a/omr-tracker/files/bin/omr-tracker-xray b/omr-tracker/files/bin/omr-tracker-xray new file mode 100755 index 000000000..21272b1e6 --- /dev/null +++ b/omr-tracker/files/bin/omr-tracker-xray @@ -0,0 +1,150 @@ +#!/bin/sh +# vim: set noexpandtab tabstop=4 shiftwidth=4 softtabstop=4 : + +name=$0 +basename="$(basename $0)" + +if [ -f /usr/sbin/iptables-legacy ]; then + IPTABLES="/usr/sbin/iptables-legacy" +else + IPTABLES="/usr/sbin/iptables" +fi + + +_log() { + logger -p daemon.info -t "${basename}" "$@" +} + +_ping_server() { + local host=$1 + ret=$(ping \ + -w "$OMR_TRACKER_TIMEOUT" \ + -c 1 \ + -q \ + "${host}" + ) && echo "$ret" | grep -sq " 0% packet loss" && { + server_ping=true + } +} + +_get_ip() { + uci -q set openmptcprouter.omr=router + if [ "$(uci -q get openmptcprouter.settings.external_check)" != "0" ]; then + check_ipv4_website="$(uci -q get openmptcprouter.settings.check_ipv4_website)" + [ -z "$check_ipv4_website" ] && check_ipv4_website="http://ip.openmptcprouter.com" + check_ipv6_website="$(uci -q get openmptcprouter.settings.check_ipv6_website)" + [ -z "$check_ipv6_website" ] && check_ipv6_website="http://ipv6.openmptcprouter.com" + public_ipv4="$(curl -s -4 -m 3 $check_ipv4_website)" + uci -q set openmptcprouter.omr.detected_public_ipv4="${public_ipv4}" + [ -n "${public_ipv4}" ] && { + uci -q set upnpd.config.external_ip="${public_ipv4}" + uci -q commit upnpd + } + if [ "$(uci -q get openmptcprouter.omr.xray)" != "down" ]; then + uci -q set openmptcprouter.omr.detected_ss_ipv4="$(curl -s -4 --socks5 "${proxy}" --max-time 3 $check_ipv4_website)" + else + uci -q del openmptcprouter.omr.detected_ss_ipv4 + fi + if [ "$(uci -q get openmptcprouter.settings.disable_ipv6)" != "1" ]; then + uci -q set openmptcprouter.omr.detected_public_ipv6="$(curl -s -6 -m 3 $check_ipv6_website)" + else + uci -q del openmptcprouter.omr.detected_public_ipv6 + # uci -q set openmptcprouter.omr.detected_ss_ipv6=$(curl -s -6 --socks5 ":::1111" --max-time 3 http://ip.openmptcprouter.com) + fi + fi + uci -q commit openmptcprouter +} + +timeout=${OMR_TRACKER_TIMEOUT:-5} +interval=${OMR_TRACKER_INTERVAL:-10} +retry=${OMR_TRACKER_TRIES:-4} +proxy=${OMR_TRACKER_PROXY:-127.0.0.1:1111} +hosts=${OMR_TRACKER_HOSTS:-1.1.1.1 1.0.0.1} +wait_test=${OMR_TRACKER_WAIT_TEST:-0} + +nodns=0 + +last=0 +nocontact="" +uci -q set openmptcprouter.omr=router +uci -q delete openmptcprouter.omr.xray +_get_ip + +while true; do + host="${hosts%% *}" + [ "$host" = "$hosts" ] || { + hosts="${hosts#* } $host" + } + if [ "$(curl -s -I -w %{http_code} --socks5 ${proxy} --max-time ${timeout} $host -o /dev/null)" != "000" ]; then + nocontact="" + [ "${last}" -ge "${retry}" ] || [ "$(uci -q get openmptcprouter.omr.xray)" = "" ] && { + _log "xray is up (can contact via http ${host})" + OMR_TRACKER_STATUS_MSG="xray is up (can contact via http ${host})" + uci -q set openmptcprouter.omr.xray="up" + uci -q commit openmptcprouter.omr + /etc/init.d/openmptcprouter-vps set_vps_firewall + mail_alert="$(uci -q get omr-tracker.proxy.mail_alert)" + #[ -z "$mail_alert" ] && mail_alert="$(uci -q get omr-tracker.defaults.mail_alert)" + [ "$mail_alert" = "1" ] && [ -n "$(uci -q get mail.default.to)" ] && { + OMR_SYSNAME="$(uci -q get system.@system[0].hostname)" + if [ "$(uci -q get omr-tracker.defaults.mail_up_subject)" != "" ] && [ "$(uci -q get omr-tracker.defaults.mail_up_message)" != "" ]; then + mail_subject="$(uci -q get omr-tracker.defaults.mail_up_subject)" + mail_subject=`echo $mail_subject | sed -e "s/%SYSNAME%/$OMR_SYSNAME/g" -e "s/%INTERFACE%/xray Proxy/g" -e "s/%DEVICE%/Shadowsocks Proxy/g" -e "s/%MESSAGE%/$OMR_TRACKER_STATUS_MSG/g"` + mail_message="$(uci -q get omr-tracker.defaults.mail_up_message)" + mail_message=`echo $mail_message | sed -e "s/%SYSNAME%/$OMR_SYSNAME/g" -e "s/%INTERFACE%/xray Proxy/g" -e "s/%DEVICE%/Shadowsocks Proxy/g" -e "s/%MESSAGE%/$OMR_TRACKER_STATUS_MSG/g"` + echo -e "Subject: ${mail_subject}\n\n${mail_message}" | sendmail $(uci -q get mail.default.to) + else + echo -e "Subject: $OMR_SYSNAME: xray Proxy is UP." | sendmail $(uci -q get mail.default.to) + fi + } + script_alert_up="$(uci -q get omr-tracker.proxy.script_alert_up)" + [ -n "$script_alert_up" ] && eval $script_alert_up + } + if [ -z "$($IPTABLES -w -t nat -L -n 2>/dev/null | grep ^xr)" ]; then + _log "Reload xray rules" + /etc/init.d/xray rules_up 2> /dev/null + _get_ip + fi + [ "$(uci -q get openmptcprouter.omr.detected_public_ipv4)" = "" ] || ([ "$(uci -q get openmptcprouter.settings.disable_ipv6)" != "1" ] && [ "$(uci -q get openmptcprouter.omr.detected_public_ipv6)" = "" ]) && _get_ip + last=0 + else + last=$((last + 1 )) + [ -z "$nocontact" ] && nocontact="$host" || nocontact="$nocontact, $host" + [ "${last}" -ge "${retry}" ] && { + if [ -n "$($IPTABLES -w -t nat -L -n 2>/dev/null | grep ^xr)" ]; then + _log "xray is down (can't contact via http ${nocontact})" + OMR_TRACKER_STATUS_MSG="xray is down (can't contact via http ${nocontact})" + uci -q set openmptcprouter.omr.xray="down" + uci -q commit openmptcprouter.omr + /etc/init.d/openmptcprouter-vps set_vps_firewall + /etc/init.d/xray rules_down 2> /dev/null + _get_ip + server_ping=false + server="$(uci -q get xray.omrout.s_vless_address)" + _ping_server $server + if [ "$server_ping" = false ]; then + _log "Server ($server) seems down, no answer to ping" + OMR_TRACKER_STATUS_MSG="${OMR_TRACKER_STATUS_MSG} - Server ($server) seems down, no answer to ping" + fi + mail_alert="$(uci -q get omr-tracker.proxy.mail_alert)" + #[ -z "$mail_alert" ] && mail_alert="$(uci -q get omr-tracker.defaults.mail_alert)" + [ "$mail_alert" = "1" ] && [ -n "$(uci -q get mail.default.to)" ] && { + OMR_SYSNAME="$(uci -q get system.@system[0].hostname)" + if [ "$(uci -q get omr-tracker.defaults.mail_down_subject)" != "" ] && [ "$(uci -q get omr-tracker.defaults.mail_down_message)" != "" ]; then + mail_subject="$(uci -q get omr-tracker.defaults.mail_down_subject)" + mail_subject=`echo $mail_subject | sed -e "s/%SYSNAME%/$OMR_SYSNAME/g" -e "s/%INTERFACE%/xray Proxy/g" -e "s/%DEVICE%/xray Proxy/g" -e "s/%MESSAGE%/$OMR_TRACKER_STATUS_MSG/g"` + mail_message="$(uci -q get omr-tracker.defaults.mail_down_message)" + mail_message=`echo $mail_message | sed -e "s/%SYSNAME%/$OMR_SYSNAME/g" -e "s/%INTERFACE%/xray Proxy/g" -e "s/%DEVICE%/xray Proxy/g" -e "s/%MESSAGE%/$OMR_TRACKER_STATUS_MSG/g"` + echo -e "Subject: ${mail_subject}\n\n${mail_message}" | sendmail $(uci -q get mail.default.to) + else + echo -e "Subject: $OMR_SYSNAME: xray Proxy is down\n\nConnection failure of xray proxy detected. The reason is \"$OMR_TRACKER_STATUS_MSG\"." | sendmail $(uci -q get mail.default.to) + fi + } + script_alert_down="$(uci -q get omr-tracker.proxy.script_alert_down)" + [ -n "$script_alert_down" ] && eval $script_alert_down + sleep $wait_test + fi + } + fi + sleep "${interval}" +done diff --git a/omr-tracker/files/etc/init.d/omr-tracker b/omr-tracker/files/etc/init.d/omr-tracker index 0f80614ed..9a78b2378 100755 --- a/omr-tracker/files/etc/init.d/omr-tracker +++ b/omr-tracker/files/etc/init.d/omr-tracker @@ -187,6 +187,38 @@ _initialize_shadowsocks_tracker() { # } } +_initialize_shadowsocks_rust_tracker() { + local redir_tcp server tracker_server + server=$1 + + [ -n "$(echo $server | grep sss)" ] || return + [ -z "$server" ] && return + + #redir_tcp=$(uci -q get shadowsocks-libev.ss_rules.redir_tcp) + #config_get tracker_server ss_rules server + config_get ss_rust_disabled $server disabled 0 + [ "$ss_rust_disabled" = "0" ] && ss_rust_enable="1" + [ -z "$(uci -q get shadowsocks-rust.tracker_${server})" ] && [ "$ss_rust_disabled" != "1" ] && { + logger -t "omr-tracker" "Create ShadowSock tracker ss_local..." + uci -q batch <<-EOF >/dev/null + set shadowsocks-rust.tracker_${server}=ss_local + set shadowsocks-rust.tracker_${server}.server=$server + set shadowsocks-rust.tracker_${server}.local_address="127.0.0.1" + set shadowsocks-rust.tracker_${server}.local_port=1111 + set shadowsocks-rust.tracker_${server}.mode=tcp_and_udp + set shadowsocks-rust.tracker_${server}.timeout=600 + set shadowsocks-rust.tracker_${server}.fast_open=0 + set shadowsocks-rust.tracker_${server}.syslog=0 + set shadowsocks-rust.tracker_${server}.reuse_port=1 + set shadowsocks-rust.tracker_${server}.mptcp=1 + set shadowsocks-rust.tracker_${server}.verbose=0 + commit shadowsocks-rust + EOF + logger -t "omr-tracker" "Restart ShadowSocks" + /etc/init.d/shadowsocks-rust restart + } +} + _launch_shadowsocks_tracker() { local hosts timeout tries interval local_port enabled server wait_test @@ -201,7 +233,7 @@ _launch_shadowsocks_tracker() { [ "$enabled" = "0" ] || [ "$disabled" = "1" ] || [ -z "$hosts" ] && return [ -z "$server" ] && return - [ "$(uci -q get shadowsocks-libev.$server.server)" = "192.168.1.3" ] || [ "$(uci -q get shadowsocks-libev.$server.server)" = "" ] && return + [ "$server" = "192.168.1.3" ] || [ "$server" = "" ] && return procd_open_instance # shellcheck disable=SC2086 @@ -211,8 +243,43 @@ _launch_shadowsocks_tracker() { procd_append_param env "OMR_TRACKER_TRIES=$tries" procd_append_param env "OMR_TRACKER_INTERVAL=$interval" procd_append_param env "OMR_TRACKER_PROXY=127.0.0.1:$local_port" - procd_append_param env "OMR_TRACKER_WAIT_TEST=$wait_test" + procd_append_param env "OMR_TRACKER_WAIT_TEST=$type" procd_append_param env "OMR_TRACKER_SERVER=$server" + procd_append_param env "OMR_TRACKER_SS_TYPE=libev" + procd_set_param limits nofile="51200 51200" + procd_set_param respawn 0 10 0 + procd_set_param stderr 1 + procd_close_instance + sleep 1 +} + +_launch_shadowsocks_rust_tracker() { + local hosts timeout tries interval local_port enabled server wait_test + + [ "$(echo $1 | grep tracker)" != "" ] || return + + _validate_section "proxy" "proxy" + + config_get local_port "$1" local_port + local disabled + config_get disabled "$1" disabled 0 + config_get server "$1" server + + [ "$enabled" = "0" ] || [ "$disabled" = "1" ] || [ -z "$hosts" ] && return + [ -z "$server" ] && return + [ "$server" = "192.168.1.3" ] || [ "$server" = "" ] && return + + procd_open_instance + # shellcheck disable=SC2086 + procd_set_param command /bin/omr-tracker-ss "$1" + procd_append_param env "OMR_TRACKER_HOSTS=$hosts" + procd_append_param env "OMR_TRACKER_TIMEOUT=$timeout" + procd_append_param env "OMR_TRACKER_TRIES=$tries" + procd_append_param env "OMR_TRACKER_INTERVAL=$interval" + procd_append_param env "OMR_TRACKER_PROXY=127.0.0.1:$local_port" + procd_append_param env "OMR_TRACKER_WAIT_TEST=$type" + procd_append_param env "OMR_TRACKER_SERVER=$server" + procd_append_param env "OMR_TRACKER_SS_TYPE=rust" procd_set_param limits nofile="51200 51200" procd_set_param respawn 0 10 0 procd_set_param stderr 1 @@ -242,6 +309,28 @@ _launch_v2ray_tracker() { sleep 1 } +_launch_xray_tracker() { + local hosts timeout tries interval local_port enabled server wait_test + + _validate_section "proxy" "proxy" + [ "$enabled" = "0" ] || [ -z "$hosts" ] && return + + procd_open_instance + # shellcheck disable=SC2086 + procd_set_param command /bin/omr-tracker-xray "$1" + procd_append_param env "OMR_TRACKER_HOSTS=$hosts" + procd_append_param env "OMR_TRACKER_TIMEOUT=$timeout" + procd_append_param env "OMR_TRACKER_TRIES=$tries" + procd_append_param env "OMR_TRACKER_INTERVAL=$interval" + procd_append_param env "OMR_TRACKER_PROXY=127.0.0.1:1111" + procd_append_param env "OMR_TRACKER_WAIT_TEST=$wait_test" + procd_set_param limits nofile="51200 51200" + procd_set_param respawn 0 10 0 + procd_set_param stderr 1 + procd_close_instance + sleep 1 +} + _multi_server() { config_get backup $1 backup [ "$backup" = "1" ] && multiserver=true @@ -254,10 +343,14 @@ _gre_tunnel() { start_service() { local ss_enable=0 + local ss_rust_enable=0 logger -t "omr-tracker" "Launching..." config_load shadowsocks-libev config_foreach _initialize_shadowsocks_tracker server + config_load shadowsocks-rust + config_foreach _initialize_shadowsocks_rust_tracker server + config_load network config_foreach _launch_tracker interface @@ -267,11 +360,21 @@ start_service() { #elif [ "$(uci -q get shadowsocks-libev.sss0.disabled)" != "1" ]; then # /etc/init.d/shadowsocks-libev rules_down fi + if [ "$ss_rust_enable" = "1" ]; then + config_load shadowsocks-rust + config_foreach _launch_shadowsocks_rust_tracker ss_local + fi config_load v2ray config_get v2rayenabled main enabled if [ "$v2rayenabled" = "1" ]; then _launch_v2ray_tracker fi + + config_load xray + config_get xrayenabled main enabled + if [ "$xrayenabled" = "1" ]; then + _launch_xray_tracker + fi multiserver=false config_load openmptcprouter @@ -285,7 +388,7 @@ start_service() { } service_triggers() { - procd_add_reload_trigger omr-tracker network shadowsocks-libev v2ray + procd_add_reload_trigger omr-tracker network shadowsocks-libev v2ray xray } reload_service() {