@feiyang666/dsh-mobile-remote 1.9.0 → 2.0.0

package/lib/external.js

@@ -124,9 +124,10 @@ const POOL_COUNT_RE = /^\s*(transport\.)?poolCount\s*=.*(\r?\n|$)/gm
  *      连接复用 tcpMux 自动管理；服务端可能仍在下发该字段，本地一律剥离，
  *      否则 frpc 解析配置失败、启动即退出）。
  * @param {string} cfg - 后端返回的原始 frpc 配置
+ * @param {string} [localHost] - frpc 应当转发到的本地地址（宿主真实监听地址）
  * @returns {string} 修正后的配置
  */
-function finalizeFrpcConfig(cfg) {
+function finalizeFrpcConfig(cfg, localHost) {
   let text = String(cfg || '')
   // 3) 先剥离 client 端已不支持的 poolCount 字段（放最前，保证后续注入干净）
   text = text.replace(POOL_COUNT_RE, '')
@@ -140,7 +141,19 @@ function finalizeFrpcConfig(cfg) {
     }
     text = text.replace(/^(\s*serverAddr\s*=\s*)[^\n]+/m, `$1"${override.trim()}"`)
   }
-  // 2) loginFailExit 必须放顶层（第一个 [[proxies]] 之前），否则 frpc 解析报错
+  // 2.5) localIP 覆写：宿主如果只绑了某块网卡（官方桌面端/第三方壳的 `--host 192.168.x.x`），
+  //      127.0.0.1 上没有任何监听，frpc 转发过去就是 connection refused →
+  //      中转服务器/nginx 对外只能是 502 Bad Gateway。这里把代理的 localIP 改成宿主真实地址。
+  const target = String(localHost || '127.0.0.1').trim() || '127.0.0.1'
+  if (target !== '127.0.0.1') {
+    if (/^\s*localIP\s*=/m.test(text)) {
+      text = text.replace(/^(\s*localIP\s*=\s*)[^\n]*/gm, `$1"${target}"`)
+    } else {
+      // 没有 localIP 行时补在每条 localPort 之前（与 frp 配置顺序无关，读者更直观）
+      text = text.replace(/^(\s*localPort\s*=\s*)/gm, `localIP = "${target}"\n$1`)
+    }
+  }
+  // 3) loginFailExit 必须放顶层（第一个 [[proxies]] 之前），否则 frpc 解析报错
   if (!/loginFailExit\s*=/.test(text)) {
     const injectLine = '# 由 dsh-mobile-remote 注入：连不上 frps 时持续重试不退出\nloginFailExit = false\n'
     const proxiesIdx = text.indexOf('[[proxies]]')
@@ -351,7 +364,7 @@ async function ensureFrpcBinary(base) {
  * 启动外网隧道。做了稳健性处理，避免"反复点/开了又关"。
  * @returns {Promise<{ok:boolean, url?:string, status?:string, error?:string, message?:string, log?:string}>}
  */
-async function startExternal(state, body, patchFile, defaultLocalPort) {
+async function startExternal(state, body, patchFile, defaultLocalPort, localHost) {
   // 已运行则直接返回，避免重复点重复启动；
   // 但状态为 error（如鉴权失败）时允许重试：kill 旧进程并重新绑定。
   if (state.proc && state.proc.exitCode === null && state.status !== 'error') {
@@ -363,8 +376,9 @@ async function startExternal(state, body, patchFile, defaultLocalPort) {
 
   state.status = 'connecting'
   state.error = null
+  state.localHost = String(localHost || '127.0.0.1')
   try {
-    return await startExternalInner(state, body, patchFile, defaultLocalPort)
+    return await startExternalInner(state, body, patchFile, defaultLocalPort, state.localHost)
   } catch (error) {
     // 启动失败必须复位状态，否则前端会一直卡在"连接中"且无法关闭
     state.status = 'error'
@@ -373,7 +387,7 @@ async function startExternal(state, body, patchFile, defaultLocalPort) {
   }
 }
 
-async function startExternalInner(state, body, patchFile, defaultLocalPort) {
+async function startExternalInner(state, body, patchFile, defaultLocalPort, localHost) {
   // 端口优先级：调用方显式指定 > 当前真实监听端口。
   // 不再让过期的持久化 localPort 覆盖当前端口（否则 frpc 会转发到无人监听的端口）。
   const localPort = Number((body && body.localPort) || defaultLocalPort || 0)
@@ -401,7 +415,7 @@ async function startExternalInner(state, body, patchFile, defaultLocalPort) {
 
   // 2) 写 frpc.toml（含 loginFailExit=false 与回环地址覆写）
   const cfgPath = frpcConfigPath()
-  const frpcConfig = finalizeFrpcConfig(data.frpcConfig)
+  const frpcConfig = finalizeFrpcConfig(data.frpcConfig, localHost)
   // 防线：确认代理确实指向当前监听端口。错配时 frpc 会连不上本地服务，
   // 表现为外网域名 502 Bad Gateway（日志：connect to local service ... refused）。
   if (localPort) {
@@ -489,6 +503,9 @@ async function startExternalInner(state, body, patchFile, defaultLocalPort) {
       state.wsClient.connect()
       // 隧道确认建立后启动持续心跳，让后端 last_seen 保持新鲜（管理端在线判定依赖它）
       startKeepAlive(state)
+      // 同时开始探测外网入口的真实可达性：frpc 进程活着并不代表 nginx/中转侧能拿到
+      // 200 —— 本地服务没起来（端口/绑定地址不对）时对外就是 502，必须自己暴露出来。
+      startUrlProbe(state)
     }
 
     const probe = () => {
@@ -575,6 +592,7 @@ async function startExternalInner(state, body, patchFile, defaultLocalPort) {
  */
 function stopExternal(state, opts = {}) {
   stopKeepAlive(state)
+  stopUrlProbe(state)
   // 断开 WS 长连接（中转服务端据此判定离线）
   if (state.wsClient) {
     try { state.wsClient.disconnect() } catch { /* ignore */ }
@@ -601,7 +619,7 @@ function stopExternal(state, opts = {}) {
  * 重新绑定并拉起 frpc，免去每次手动输入验证码与中转服务地址。
  * 失败时把状态置为 error 并保留持久化，供用户手动处理或下次重启重试。
  */
-async function autoRestoreExternal(state, patchFile, defaultLocalPort) {
+async function autoRestoreExternal(state, patchFile, defaultLocalPort, localHost) {
   const saved = loadPersistedState()
   if (!saved) return
   try {
@@ -620,7 +638,7 @@ async function autoRestoreExternal(state, patchFile, defaultLocalPort) {
     if (stale && live && stale !== live) {
       state.lastLog = `检测到端口变化（持久化 ${stale} → 当前 ${live}），外网隧道已改用当前端口`
     }
-    await startExternal(state, body, patchFile, live || stale)
+    await startExternal(state, body, patchFile, live || stale, localHost)
   } catch (e) {
     state.status = 'error'
     state.error = `自动恢复外网访问失败：${String((e && e.message) || e)}`
@@ -709,6 +727,71 @@ function probeTunnelHealth(state) {
   return verdict
 }
 
+/* ------------------------------------------------------------------ */
+/* 外网入口可达性探测                                                    */
+/* ------------------------------------------------------------------ */
+
+/** 探测结果缓存：同一 URL 30 秒内不重复请求（status 每 5s 轮询一次）。 */
+let urlProbeCache = { url: '', at: 0, value: null }
+
+/**
+ * 用 HTTP GET 探测外网入口的真实状态。
+ * frpc 进程活着只代表「已经连上 frps」，不代表中转服务器能把请求送到本机：
+ * 本机端口没人监听 / 绑定地址不对时，nginx 对外一律回 502 —— 这里把它探测出来。
+ * @param {string} url - 外网访问地址（https://xxx）
+ * @returns {Promise<{ok: boolean, status: number, at: number, error?: string}|null>}
+ */
+async function probeExternalUrl(url) {
+  if (!url) return null
+  const now = Date.now()
+  if (urlProbeCache.url === url && now - urlProbeCache.at < 30_000) return urlProbeCache.value
+  let value
+  try {
+    const res = await fetch(url, {
+      method: 'GET',
+      redirect: 'follow',
+      headers: { 'user-agent': 'dsh-mobile-remote/healthcheck' },
+      signal: AbortSignal.timeout(8000),
+    })
+    // 只用状态码判断可达性，正文直接丢弃（避免把整个前端页面读进内存）
+    try { await res.arrayBuffer() } catch { /* ignore */ }
+    value = {
+      ok: res.status >= 200 && res.status < 400,
+      status: res.status,
+      at: Date.now(),
+      error: res.status === 502
+        ? '中转服务器返回 502：frps 已连上，但本机端口没有响应（转发地址/端口不对，或 dsh 未监听该地址）'
+        : null,
+    }
+  } catch (error) {
+    value = { ok: false, status: 0, at: Date.now(), error: String((error && error.message) || error) }
+  }
+  urlProbeCache = { url, at: now, value }
+  return value
+}
+
+/** 启动外网入口可达性轮询（30s 一次，随隧道停止而停止）。 */
+function startUrlProbe(state) {
+  if (state.probeTimer) return
+  const tick = async () => {
+    if (!state.url) return
+    const value = await probeExternalUrl(state.url)
+    if (value) state.probe = value
+  }
+  void tick()
+  state.probeTimer = setInterval(() => { void tick() }, 30_000)
+  if (state.probeTimer.unref) state.probeTimer.unref()
+}
+
+/** 停止外网入口探测并清空缓存结果。 */
+function stopUrlProbe(state) {
+  if (state.probeTimer) {
+    clearInterval(state.probeTimer)
+    state.probeTimer = null
+  }
+  state.probe = null
+}
+
 /** 当前外网状态摘要（供 /status 使用）。 */
 function externalView(state) {
   const running = !!state.proc && state.proc.exitCode === null
@@ -737,6 +820,10 @@ function externalView(state) {
     pid: running && state.proc ? state.proc.pid : null,
     startedAt: state.startedAt || null,
     lastLog: state.lastLog || null,
+    // frpc 实际转发的本地地址（非回环时是宿主的网卡 IP）
+    localHost: state.localHost || null,
+    // 外网入口真实可达性：{ ok, status, at, error }
+    probe: state.probe || null,
     error,
   }
 }
@@ -746,6 +833,8 @@ export {
   externalDir,
   frpcBinaryPath,
   frpcConfigPath,
+  // 供单元测试直接断言配置改写（localIP / loginFailExit / poolCount）
+  finalizeFrpcConfig,
   startExternal,
   stopExternal,
   externalView,