forked from buzzhub/lsyupdate
132 lines
3.9 KiB
Bash
Executable File
132 lines
3.9 KiB
Bash
Executable File
#!/bin/bash
|
||
|
||
# FRPC 配置目录和文件路径
|
||
LOCK_FILE="/tmp/lsyfrpc_check_config_running.lock"
|
||
FRPC_FILE="/data/lsyfrpc/frpc"
|
||
FRPC_DIR="/data/lsyfrpc"
|
||
CONFIG_FILE="$FRPC_DIR/frpc.ini"
|
||
|
||
# 获取本机 MAC 地址
|
||
function get_mac_address() {
|
||
cat /sys/class/net/eth0/address | tr -d ':'
|
||
}
|
||
|
||
# 从云端获取当前分配的端口
|
||
function get_cloud_port() {
|
||
local mac_address="$1"
|
||
echo "MAC Address: $mac_address"
|
||
|
||
retries=3
|
||
for ((i=1; i<=retries; i++)); do
|
||
response=$(curl -s -X 'POST' \
|
||
'https://frp-box.jxm.cool/request_port/' \
|
||
-H 'accept: application/json' \
|
||
-H 'Authorization: Basic NTI0NzgxNTctMjY1Zi00ZGNjLWE0NDMtODE0YzJhMDMxYjhjOmM2ZDI3Nzc1LTJhMDgtNDkyZS1iMTExLTg5YWQzZDY5ZTliMA==' \
|
||
-H 'Content-Type: application/json' \
|
||
-d "{\"mac_address\": \"$mac_address\"}")
|
||
|
||
cloud_port=$(echo "$response" | grep -o '"port":[0-9]*' | awk -F: '{print $2}')
|
||
|
||
if [ -n "$cloud_port" ]; then
|
||
break
|
||
else
|
||
echo "[ERROR] Failed to get port from the cloud. Retry $i of $retries."
|
||
if [ "$i" -eq "$retries" ]; then
|
||
echo "[ERROR] Max retries reached. Exiting."
|
||
exit 1
|
||
fi
|
||
sleep 10
|
||
fi
|
||
done
|
||
|
||
echo "Cloud Port for 22: $cloud_port"
|
||
echo "Cloud Port for 9100: $((cloud_port + 10000))"
|
||
}
|
||
|
||
# 读取本地配置的端口
|
||
function get_local_port() {
|
||
local section="$1"
|
||
awk -v section="[${section}]" '
|
||
$0 == section { found=1; next }
|
||
found && /^remote_port[ \t]*=[ \t]*/ {
|
||
split($0, parts, "=");
|
||
gsub(/[ \t]/, "", parts[2]);
|
||
print parts[2];
|
||
exit;
|
||
}
|
||
' "$CONFIG_FILE"
|
||
}
|
||
|
||
# 更新 frpc.ini 配置
|
||
function update_config() {
|
||
local mac_address="$1"
|
||
local cloud_port="$2"
|
||
local port_9100=$((cloud_port + 10000))
|
||
|
||
cat > "$CONFIG_FILE" <<EOF
|
||
[common]
|
||
server_addr = box.jxm.cool
|
||
server_port = 8000
|
||
token = 92098d16-1961-4bda-902b-e43e3d41d5a9
|
||
|
||
[${mac_address}-v1]
|
||
type = tcp
|
||
local_ip = 127.0.0.1
|
||
local_port = 22
|
||
remote_port = $cloud_port
|
||
|
||
[${mac_address}-9100]
|
||
type = tcp
|
||
local_ip = 127.0.0.1
|
||
local_port = 9100
|
||
remote_port = $port_9100
|
||
EOF
|
||
echo "[INFO] Configuration updated."
|
||
}
|
||
|
||
# 主函数
|
||
function main() {
|
||
# 确保不会重复运行
|
||
if [ -f "$LOCK_FILE" ]; then
|
||
lock_pid=$(cat "$LOCK_FILE")
|
||
if [ -n "$lock_pid" ] && kill -0 "$lock_pid" 2>/dev/null; then
|
||
echo "[INFO] Lock file exists and script is running (PID: $lock_pid). Exiting."
|
||
exit 0
|
||
else
|
||
echo "[INFO] Stale lock file detected. Removing."
|
||
rm -f "$LOCK_FILE"
|
||
fi
|
||
fi
|
||
|
||
# 创建锁文件
|
||
echo $$ > "$LOCK_FILE"
|
||
trap "rm -f $LOCK_FILE" EXIT SIGHUP SIGINT SIGTERM SIGQUIT
|
||
|
||
# 获取本机 MAC 地址并处理冒号
|
||
mac_address_raw=$(get_mac_address)
|
||
mac_address="${mac_address_raw//:}" # 移除冒号，用于section名称
|
||
mac_address_orig=$(cat /sys/class/net/eth0/address) # 原始MAC用于API请求
|
||
|
||
# 获取云端分配的端口
|
||
get_cloud_port "$mac_address_orig"
|
||
port_9100=$((cloud_port + 10000))
|
||
|
||
# 获取本地端口配置
|
||
local_port_v1=$(get_local_port "${mac_address}-v1")
|
||
local_port_9100=$(get_local_port "${mac_address}-9100")
|
||
|
||
echo "Local Port for [${mac_address}-v1]: $local_port_v1 (Expected: $cloud_port)"
|
||
echo "Local Port for [${mac_address}-9100]: $local_port_9100 (Expected: $port_9100)"
|
||
|
||
# 检查端口是否匹配
|
||
if [ "$local_port_v1" != "$cloud_port" ] || [ "$local_port_9100" != "$port_9100" ]; then
|
||
echo "[INFO] Configuration mismatch detected. Updating frpc.ini..."
|
||
update_config "$mac_address" "$cloud_port"
|
||
systemctl restart lsyfrpc
|
||
echo "[INFO] FRPC restarted with updated configuration."
|
||
else
|
||
echo "[INFO] Configuration is up to date. No changes needed."
|
||
fi
|
||
}
|
||
|
||
main |