refactor: 迁移函数到新的文件

2026-07-20 15:25:58 +08:00
parent af8b8ac64f
commit 0e6a4c1300
4 changed files with 32 additions and 33 deletions
@@ -0,0 +1,97 @@
+#!/bin/bash
+
+# 查找随机空闲端口
+# 用途：
+#   在指定端口范围内，随机选择一个当前未被监听（TCP/UDP）的端口
+# 参数：
+#   $1: 起始端口（默认 10001）
+#   $2: 结束端口（默认 65535）
+#   $3: 最大随机尝试次数（默认 100）
+find_free_port() {
+  local start="${1:-10001}"
+  local max="${2:-65535}"
+  local attempts="${3:-100}"
+
+  # ------------------------------------------------------------
+  # 收集当前正在监听的端口（TCP + UDP）
+  # 优先使用 ss，其次 netstat
+  # ------------------------------------------------------------
+  local used_ports raw
+  if command -v ss >/dev/null 2>&1; then
+    # ss 输出示例：
+    # LISTEN 0 128 0.0.0.0:22
+    raw=$(ss -lntu 2>/dev/null || true)
+    used_ports=$(printf "%s\n" "$raw" \
+      | awk '{print $5}' \
+      | sed -E 's/.*[:]//g')
+  elif command -v netstat >/dev/null 2>&1; then
+    # netstat 输出示例：
+    # tcp 0 0 0.0.0.0:22
+    raw=$(netstat -lntu 2>/dev/null || true)
+    used_ports=$(printf "%s\n" "$raw" \
+      | awk '{print $4}' \
+      | sed -E 's/.*[:]//g')
+  else
+    echo "Error: ss or netstat is required to check listening ports." >&2
+    return 2
+  fi
+
+  # ------------------------------------------------------------
+  # 将已占用端口存入关联数组，便于 O(1) 判断
+  # 需要 bash 4+
+  # ------------------------------------------------------------
+  declare -A used_map
+  local p
+  for p in $used_ports; do
+    # 过滤非数字字段（如 *、:::）
+    if [[ $p =~ ^[0-9]+$ ]]; then
+      used_map["$p"]=1
+    fi
+  done
+
+  # ------------------------------------------------------------
+  # 随机尝试若干次
+  # 每次随机生成一个端口，只要未被监听就立即返回
+  # ------------------------------------------------------------
+  local port i
+  for ((i=0; i<attempts; i++)); do
+    # 在 [start, max] 范围内生成随机端口
+    port=$(( RANDOM % (max - start + 1) + start ))
+
+    # 如果端口不在已监听表中，则认为可用
+    if [[ -z "${used_map[$port]}" ]]; then
+      echo "$port"
+      return 0
+    fi
+  done
+
+  # ------------------------------------------------------------
+  # 多次随机尝试后仍未找到可用端口
+  # 通常意味着端口范围过小或已被大量占用
+  # ------------------------------------------------------------
+  echo "Error: no free port found after $attempts random attempts." >&2
+  return 1
+}
+
+
+update_port(){
+  local script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" # 脚本文件夹路径
+  local config_file="$script_dir/../conf/config.json"
+
+  local port=$1
+  # 如果参数为空，则设置为443
+  if [ -z "$port" ]; then
+    port=443
+  fi
+  modify_json_file "$config_file" ".inbounds[0].port" "$port"
+  echo "设置端口成功"
+}
+
+ask_update_port(){
+    local script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+    local config_file="$script_dir/../conf/config.json"
+
+    local current_port=$(jq -r '.inbounds[0].port' "$config_file")
+    local new_port=$(ask_input "请输入新的端口号" "$current_port")
+    update_port "$new_port"
+}