mslxdff 0.1.176 → 0.1.178

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "mslxdff",
3
- "version": "0.1.176",
3
+ "version": "0.1.178",
4
4
  "description": "测试项目,请勿使用。",
5
5
  "type": "module",
6
6
  "bin": {
@@ -2,7 +2,7 @@
2
2
  import { join } from "node:path";
3
3
  import { logDir } from "../logs.js";
4
4
 
5
- /** 缺省即开:组长升级/重启后无需改任何环境变量即可生效;不想用的节点设 MSLXDFF_WARP_POOL=0。 */
5
+ /** 缺省开 + 429 时换身份换出口:他项目真机实测「同一身份连发 3 次 429,身份一换同一个请求立刻 200」⇒ 换身份=换桶。 */
6
6
  export const DEFAULTS = {
7
7
  enabled: true,
8
8
  mode: "leader",
@@ -14,6 +14,7 @@ export const DEFAULTS = {
14
14
  registerMinGapMs: 6_000,
15
15
  registerMaxGapMs: 15_000,
16
16
  probePath: "/zen/v1/models",
17
+ rotate: true,
17
18
  };
18
19
 
19
20
  const OFF = ["0", "off", "false", "no", "disable", "disabled"];
@@ -50,6 +51,7 @@ export function resolveWarpConfig(env = process.env, { logDirFn = logDir } = {})
50
51
  const bin = envStr(env, "MSLXDFF_WARP_BIN", "");
51
52
  const autodownload = envBool(env, "MSLXDFF_WARP_AUTODOWNLOAD", DEFAULTS.autodownload);
52
53
  const probePath = envStr(env, "MSLXDFF_WARP_PROBE_PATH", DEFAULTS.probePath);
54
+ const rotate = envBool(env, "MSLXDFF_WARP_ROTATE", DEFAULTS.rotate);
53
55
 
54
56
  return {
55
57
  enabled,
@@ -63,6 +65,7 @@ export function resolveWarpConfig(env = process.env, { logDirFn = logDir } = {})
63
65
  bin,
64
66
  autodownload,
65
67
  probePath,
68
+ rotate,
66
69
  registerMinGapMs: DEFAULTS.registerMinGapMs,
67
70
  registerMaxGapMs: DEFAULTS.registerMaxGapMs,
68
71
  /** 节点端口列表:basePort .. basePort+count-1 */
@@ -0,0 +1,141 @@
1
+ // warp-pool · 身份轮换:换一台 WARP 设备身份来换出口 IP。
2
+ //
3
+ // 依据(他项目真机实测,非推测):kyomufs/zen-router 的 internal/zen/errors.go:32-42 记录
4
+ // 同一身份连发 3 次 → 429 FreeUsageLimitError;身份一换,同一个请求立刻 200
5
+ // (104.28.198.244 → 429 ×3,104.28.198.247 → 200)。
6
+ // 故「多名额」的正解不是「N 台设备同时在线」(真机实测同时注册会折叠成同一个出口),
7
+ // 而是**串行换身份**:换一台设备 ⇒ 换一个出口 ⇒ 换一份按 IP 计的名额。
8
+ //
9
+ // 与 zen-router 对齐的三条纪律:
10
+ // 1. 轮换后**必须实测确认出口真的变了**;没变就整个 cycle 重试(它用 4 次,这里同)。
11
+ // 「注册成功」不等于「出口换了」——不校验就会拿同一个桶反复撞 429。
12
+ // 2. 同端口并发轮换去重 + 最小间隔,防 429 风暴把 CF 注册刷到限流(同类项目事故)。
13
+ // 3. 只轮换、不销毁别的节点:轮换失败保持原节点可用,绝不把池打空。
14
+ import { probeTunnel } from "./wireproxy.js";
15
+ import { EXIT_PROBE_URL, parseExitIp } from "./revive.js";
16
+
17
+ /** 经隧道读该端口真实出口地址(CF trace);取不到返回 "",绝不抛。 */
18
+ export async function readExitIp({ port, fetchImpl, timeoutMs = 12_000 } = {}) {
19
+ try {
20
+ const res = await probeTunnel({ port, baseUrl: EXIT_PROBE_URL, path: "", fetchImpl, timeoutMs, wantText: true });
21
+ return typeof res?.text === "string" ? parseExitIp(res.text) : "";
22
+ } catch {
23
+ return "";
24
+ }
25
+ }
26
+
27
+ /**
28
+ * 身份轮换器。
29
+ * @param {{ register: Function, apply: (node:object)=>Promise<void>, readIp?: Function,
30
+ * maxAttempts?: number, settleMs?: number, sleep?: Function, now?: Function,
31
+ * log?: (s:string)=>void }} opts
32
+ * apply 由调用方注入:把新身份落盘并起隧道(含停掉旧隧道)。
33
+ */
34
+ export function createIdentityRotator({
35
+ register,
36
+ apply,
37
+ readIp = readExitIp,
38
+ maxAttempts = 4,
39
+ settleMs = 3000,
40
+ sleep = (ms) => new Promise((r) => setTimeout(r, ms)),
41
+ now = Date.now,
42
+ log = () => {},
43
+ } = {}) {
44
+ /** @type {Map<number, {ip:string, at:number}>} 各端口已知出口 */
45
+ const known = new Map();
46
+ /** @type {Map<number, Promise<object>>} 在途轮换(同端口去重) */
47
+ const inflight = new Map();
48
+
49
+ async function cycle(port, oldIp) {
50
+ for (let i = 1; i <= maxAttempts; i++) {
51
+ let reg;
52
+ try {
53
+ reg = await register({ log });
54
+ } catch (e) {
55
+ log(`[warp-pool] :${port} 轮换 ${i}/${maxAttempts}:注册失败(${String(e?.message || e).slice(0, 120)})`);
56
+ await sleep(settleMs);
57
+ continue;
58
+ }
59
+ try {
60
+ await apply({ port, ...reg });
61
+ } catch (e) {
62
+ log(`[warp-pool] :${port} 轮换 ${i}/${maxAttempts}:起隧道失败(${String(e?.message || e).slice(0, 120)})`);
63
+ }
64
+ await sleep(settleMs);
65
+ const ip = await readIp({ port });
66
+ if (ip && ip !== oldIp) {
67
+ known.set(port, { ip, at: now() });
68
+ log(`[warp-pool] :${port} 身份已轮换:出口 ${oldIp || "?"} → ${ip}(第 ${i}/${maxAttempts} 次)`);
69
+ return { ok: true, ip, attempts: i };
70
+ }
71
+ log(`[warp-pool] :${port} 轮换 ${i}/${maxAttempts}:出口未变(仍取到 ${ip || "空"}),重试`);
72
+ }
73
+ return { ok: false, attempts: maxAttempts };
74
+ }
75
+
76
+ return {
77
+ /** 记下入池时的实测出口(bootstrap 取证后调用)。 */
78
+ note(port, ip) {
79
+ if (ip) known.set(port, { ip, at: now() });
80
+ },
81
+ /** 该端口当前已知出口("" = 未知)。 */
82
+ ipOf(port) {
83
+ return known.get(port)?.ip || "";
84
+ },
85
+ /**
86
+ * 轮换某节点身份换出口。同端口在途时返回同一个 Promise;minGapMs 内重复请求被掐
87
+ * (429 风暴时每个失败请求都触发一次轮换会把 CF 刷黑)。
88
+ * @returns {Promise<{ok:boolean, ip?:string, attempts?:number, skipped?:boolean}>}
89
+ */
90
+ rotate(port, { minGapMs = 60_000, force = false } = {}) {
91
+ const dup = inflight.get(port);
92
+ if (dup) return dup;
93
+ const prev = known.get(port);
94
+ if (!force && prev && now() - prev.at < minGapMs) {
95
+ return Promise.resolve({ ok: false, skipped: true });
96
+ }
97
+ const p = cycle(port, prev?.ip || "")
98
+ .catch((e) => {
99
+ log(`[warp-pool] :${port} 轮换异常:${String(e?.message || e)}`);
100
+ return { ok: false };
101
+ })
102
+ .finally(() => inflight.delete(port));
103
+ inflight.set(port, p);
104
+ return p;
105
+ },
106
+
107
+ /** 入池取证结果(labelExits 的分组)批量记账:每个节点的出口 IP 都要记住,轮换才能比对「是否真变了」。 */
108
+ noteExits(groups) {
109
+ for (const g of groups || []) for (const p of g.nodes || []) if (g.ip) known.set(p, { ip: g.ip, at: now() });
110
+ },
111
+ /** 轮换进行中 = 该节点正在换出口(调用方可据此避开它)。 */
112
+ busy(port) {
113
+ return inflight.has(port);
114
+ },
115
+ stats() {
116
+ return [...known.entries()].map(([port, v]) => ({ port, ip: v.ip, at: v.at }));
117
+ },
118
+ };
119
+ }
120
+
121
+ /**
122
+ * 把「429 → 换身份换出口」接进一个已建好的池。轮换窗口内**先把该节点冷却**(端口已停,
123
+ * 请求过去必失败),换好再放回。依赖全由调用方注入,本模块不反向依赖池。
124
+ * @returns {{note:Function, noteExits:Function, ipOf:Function, rotate:Function, busy:Function, stats:Function}}
125
+ */
126
+ export function wireRotation({ dispatcher, register, tunnels, bringUp, swap = () => {}, log = () => {}, ...rest } = {}) {
127
+ return createIdentityRotator({
128
+ register,
129
+ apply: async (node) => {
130
+ const idx = dispatcher.entries().find((e) => e.port === node.port)?.index;
131
+ if (idx !== undefined) dispatcher.markFailure(idx); // 轮换窗口内先冷却:端口已停,请求过去必失败
132
+ tunnels.get(node.port)?.stop();
133
+ tunnels.delete(node.port);
134
+ swap(node.port, node);
135
+ await bringUp(node);
136
+ if (idx !== undefined) dispatcher.markOk(idx);
137
+ },
138
+ log,
139
+ ...rest,
140
+ });
141
+ }
@@ -1,18 +1,16 @@
1
1
  // warp-pool · 装配入口。
2
2
  //
3
- // 定位(见 MODULES.md §0):WARP 池是**增量**出口,不是替代成员自身出口;
4
- // 开关缺省即开(MSLXDFF_WARP_POOL 缺省 1;设 0 则零副作用);只有组长(MSLXDFF_WARP_MODE=leader)才真建池。
5
- //
6
- // 关键约束:bootstrap(注册设备 + 起隧道)很慢(注册有 6–15s 间隔、可能退避 60–120s),
7
- // 因此**绝不阻塞 daemon 启动**——createWarpPool 同步返回,dispatcher 先是空池(直连兜底),
8
- // bootstrap 完成后 setNodes() 换上真实出口。
3
+ // 定位(MODULES.md §0/§10):WARP 池是**增量**出口,不是替代成员自身出口;缺省**开**,MSLXDFF_WARP_POOL=0 关。
9
4
  import { resolveWarpConfig } from "./config.js";
10
5
  import { createRotatingDispatcher } from "./rotate.js";
11
6
  import { createHealthMonitor } from "./health.js";
12
7
  import { ensureDir, loadNodes, removeNode, saveNode } from "./nodes.js";
13
8
  import { registerWarpDevice } from "./register.js";
14
9
  import { ensureBinary, probeTunnel, resolveAsset, startTunnel, writeTunnelConf } from "./wireproxy.js";
15
- /** 重拨节流:连续失败指数退避。缺省开之后,网络不通的机器绝不能让重注册把 CF 刷到 429。 */
10
+ import { createReviver, labelExits } from "./revive.js";
11
+ import { wireRotation } from "./identity.js";
12
+
13
+ /** 重拨节流:连续失败指数退避。网络不通的机器绝不能让重注册把 CF 刷到 429。 */
16
14
  const REDIAL_MIN_GAP_MS = 5 * 60_000;
17
15
  const REDIAL_MAX_GAP_MS = 30 * 60_000;
18
16
 
@@ -22,9 +20,13 @@ function gapMs(cfg, rnd = Math.random) {
22
20
  return Math.round(cfg.registerMinGapMs + rnd() * span);
23
21
  }
24
22
 
23
+ /** 注册响应 → 节点对象(bootstrap 与重拨共用,避免同一份字段表写两遍)。 */
24
+ function nodeFromReg(port, reg, now) {
25
+ return { port, id: reg.id, token: reg.token, v4: reg.v4, v6: reg.v6, endpoint: reg.endpoint, peerKey: reg.peerKey, privateKey: reg.privateKey, registeredAt: new Date(now()).toISOString() };
26
+ }
25
27
 
26
28
  /**
27
- * 建池。开关缺省即开,故「未启用」= 显式 MSLXDFF_WARP_POOL=0 或 非 leader;那时返回 null,调用方据此不注入。
29
+ * 建池。开关缺省关,故「未启用」= 未设 MSLXDFF_WARP_POOL=1 或 非 leader;那时返回 null,调用方据此不注入。
28
30
  * @returns {null | { dispatcher: object, ready: Promise<void>, close: () => Promise<void>, stats: () => object }}
29
31
  */
30
32
  export function createWarpPool({
@@ -42,9 +44,22 @@ export function createWarpPool({
42
44
  const cfg = resolveWarpConfig(env);
43
45
  if (!cfg.active) return null;
44
46
 
45
- const dispatcher = createRotatingDispatcher({ nodes: [], strategy: cfg.strategy, cooldownMs: cfg.cooldownMs, log });
47
+ /** 身份轮换器:429 时换一台 WARP 设备换出口 IP(串行换身份,见 identity.js 头注)。 */
48
+ let rotator = null;
49
+
50
+ const dispatcher = createRotatingDispatcher({ nodes: [], strategy: cfg.strategy, cooldownMs: cfg.cooldownMs, log, onStatus: (status, port) => { if (status === 429 && cfg.rotate && rotator) void rotator.rotate(port); } });
46
51
  /** @type {Map<number, {stop: () => void}>} */
47
52
  const tunnels = new Map();
53
+
54
+ /** 复活:用磁盘已有 conf 重起,不重注册(进程挂了 ≠ 设备失效,省 CF 配额)。 */
55
+ async function reviveTunnel(port) {
56
+ if (closed) return false;
57
+ const node = loadNodes(cfg.dir, [port])[0];
58
+ if (!node) return false;
59
+ await bringUp(node);
60
+ return tunnels.has(port);
61
+ }
62
+ const reviver = createReviver({ revive: reviveTunnel, log });
48
63
  let binPath = "";
49
64
  let closed = false;
50
65
  let lastRedialAt = 0;
@@ -69,12 +84,23 @@ export function createWarpPool({
69
84
  log,
70
85
  onExit: ({ port, code }) => {
71
86
  tunnels.delete(port);
72
- log(`[warp-pool] 隧道 :${port} 退出(code=${code})`);
87
+ log(`[warp-pool] 隧道 :${port} 退出(code=${code}),尝试复活`);
88
+ void reviver.onExit(port);
73
89
  },
74
90
  });
75
91
  tunnels.set(node.port, t);
76
92
  }
77
93
 
94
+ // 429 = 上游按出口 IP 计的名额耗尽 ⇒ 换一台设备的身份换出口(换身份=换桶)。
95
+ rotator = wireRotation({
96
+ dispatcher,
97
+ register,
98
+ tunnels,
99
+ bringUp,
100
+ swap: (port, node) => { removeNode(cfg.dir, port); saveNode(cfg.dir, node); },
101
+ log,
102
+ });
103
+
78
104
  async function bootstrap() {
79
105
  if (closed) return;
80
106
  ensureDir(cfg.dir);
@@ -102,17 +128,7 @@ export function createWarpPool({
102
128
  try {
103
129
  if (nodes.length > 0) await wait(gapMs(cfg, rnd));
104
130
  const reg = await register({ log });
105
- const node = {
106
- port,
107
- id: reg.id,
108
- token: reg.token,
109
- v4: reg.v4,
110
- v6: reg.v6,
111
- endpoint: reg.endpoint,
112
- peerKey: reg.peerKey,
113
- privateKey: reg.privateKey,
114
- registeredAt: new Date(now()).toISOString(),
115
- };
131
+ const node = nodeFromReg(port, reg, now);
116
132
  saveNode(cfg.dir, node);
117
133
  nodes.push(node);
118
134
  log(`[warp-pool] 注册节点 :${port}(v6=${reg.v6})`);
@@ -130,9 +146,7 @@ export function createWarpPool({
130
146
  }
131
147
  }
132
148
 
133
- // 4. **入池前实测探活**:只判「进程还活着」不够——WARP 握手不通时 wireproxy 进程照样在跑却转不动流量
134
- // (实测经它 curl 挂到超时,不是快速失败)。那种节点一旦进池,dispatcher 会把**所有** opencode
135
- // 请求派给它 ⇒ opencode 整体被黑洞而非直连兜底。探活打上游 models 路径,不烧 token。
149
+ // 入池前实测探活:WARP 握手不通时 wireproxy 进程照样在跑却转不动流量(挂到超时),那种节点进池会黑洞全部请求
136
150
  const live = [];
137
151
  for (const n of nodes) {
138
152
  if (!tunnels.has(n.port)) continue;
@@ -146,6 +160,9 @@ export function createWarpPool({
146
160
  } else {
147
161
  log("[warp-pool] 无可用节点,保持直连");
148
162
  }
163
+
164
+ // 5. 出口取证:把每台设备的真实出口地址打进日志(免费版 WARP 的 IPv4 是共享 NAT,N 台设备很可能折叠成同一个出口)
165
+ if (live.length) rotator.noteExits(await labelExits({ ports: live.map((n) => n.port), log }));
149
166
  }
150
167
 
151
168
  const monitor = createHealthMonitor({
@@ -154,9 +171,7 @@ export function createWarpPool({
154
171
  probePath: cfg.probePath,
155
172
  log,
156
173
  onExhausted: async () => {
157
- // 重拨 = 重新注册 = 换新出口 IP(代价与限流见 MODULES.md)。
158
- // 隧道全活却探活全挂 ⇒ 网络路径问题(实测墙内到 CF 的 UDP 被选择性屏蔽),
159
- // 重注册换不来出路,只会把 CF 注册接口刷到 429 拉黑——故必须过 redialPlan 这道门。
174
+ // 重拨 = 重新注册换出口。只重拨「进程已死」的隧道:全活却全挂 = 网络路径问题,重注册只会把 CF 刷到 429
160
175
  const plan = redialPlan({ ports: cfg.ports, alivePorts: [...tunnels.keys()], lastRedialAt, redialGapMs, now });
161
176
  if (plan.reason === "all-alive") {
162
177
  log("[warp-pool] 隧道均存活但全池不通,判定为网络路径问题,跳过重拨");
@@ -177,17 +192,7 @@ export function createWarpPool({
177
192
  tunnels.delete(port);
178
193
  removeNode(cfg.dir, port);
179
194
  const reg = await register({ log });
180
- saveNode(cfg.dir, {
181
- port,
182
- id: reg.id,
183
- token: reg.token,
184
- v4: reg.v4,
185
- v6: reg.v6,
186
- endpoint: reg.endpoint,
187
- peerKey: reg.peerKey,
188
- privateKey: reg.privateKey,
189
- registeredAt: new Date(now()).toISOString(),
190
- });
195
+ saveNode(cfg.dir, nodeFromReg(port, reg, now));
191
196
  await bringUp({ port, ...reg });
192
197
  } catch (e) {
193
198
  log(`[warp-pool] 重拨 :${port} 失败:${String(e?.message || e)}`);
@@ -195,7 +200,7 @@ export function createWarpPool({
195
200
  await wait(gapMs(cfg, rnd));
196
201
  }
197
202
  if (!closed && [...tunnels.keys()].length) {
198
- // 重拨确实把隧道拉起来了 ⇒ 退避复位,别让一次网络抖动永久拖慢恢复。
203
+ // 重拨确实把隧道拉起来了 ⇒ 退避复位
199
204
  redialGapMs = REDIAL_MIN_GAP_MS;
200
205
  await dispatcher.setNodes([...tunnels.keys()].map((port) => ({ port, uri: `http://127.0.0.1:${port}` })));
201
206
  }
@@ -232,24 +237,20 @@ export function createWarpPool({
232
237
  dir: cfg.dir,
233
238
  tunnels: tunnels.size,
234
239
  nodes: dispatcher.stats(),
240
+ exits: rotator ? rotator.stats() : [],
235
241
  };
236
242
  },
237
243
  };
238
244
  }
239
245
 
240
-
241
- /**
242
- * 只取轮换 dispatcher 的便捷入口(隧道进程由模块内的 process exit 钩子兜底清理)。
243
- * 未启用(开关关 / 非 leader)时返回 undefined —— 调用方直接透传即是"关闭态"。
244
- */
246
+ /** 只取轮换 dispatcher 的便捷入口;未启用时返回 undefined(调用方直接透传即"关闭态")。 */
247
+ /** 隧道进程由模块内的 process exit 钩子兜底清理。 */
245
248
  export function createWarpDispatcher(opts) {
246
249
  return createWarpPool(opts)?.dispatcher;
247
250
  }
248
251
 
249
252
  /**
250
- * 重拨门(纯函数,便于单测):只有「有隧道进程已死」且「距上次重拨超过当前退避间隔」才放行。
251
- * 隧道全活却探活全挂 ⇒ 网络路径问题(实测:墙内到 CF 的 UDP 被选择性屏蔽),
252
- * 此时重拨换不来出路,只会把 CF 注册接口刷到 429 拉黑。
253
+ * 重拨门(纯函数,便于单测):只「有隧道进程已死」且「距上次重拨超过退避间隔」才放行——全活却全挂是网络问题。
253
254
  * @returns {{ dead: number[], reason: "ok" | "all-alive" | "throttled" }}
254
255
  */
255
256
  export function redialPlan({ ports = [], alivePorts = [], lastRedialAt = 0, redialGapMs = REDIAL_MIN_GAP_MS, now = Date.now } = {}) {
@@ -40,6 +40,7 @@ export function loadNode(dir, port) {
40
40
  v6: String(j.v6 || ""),
41
41
  endpoint: String(j.endpoint || ""),
42
42
  peerKey: String(j.peerKey || ""),
43
+ privateKey: String(j.privateKey || ""), // 必须读回:bringUp 靠它渲染 conf,漏了就是 PrivateKey = undefined(2026-10-16 真机事故:复用节点的隧道全起不来)
43
44
  registeredAt: String(j.registeredAt || ""),
44
45
  ...(j.cooldownUntil ? { cooldownUntil: Number(j.cooldownUntil) } : {}),
45
46
  };
@@ -65,7 +66,7 @@ export function loadNodes(dir, ports) {
65
66
  const out = [];
66
67
  for (const p of ports) {
67
68
  const n = loadNode(dir, p);
68
- if (n && n.id && n.token) out.push(n);
69
+ if (n && n.id && n.token && n.privateKey) out.push(n);
69
70
  }
70
71
  return out;
71
72
  }
@@ -0,0 +1,117 @@
1
+ // warp-pool · 复活与取证(真机事故后新增,2026-10-16)。
2
+ //
3
+ // 两件事,都来自组长 VPS 上暴露的缺陷:
4
+ //
5
+ // 1. `revive`: 隧道进程意外退出后**必须自愈**。原先 `onExit` 只把端口从表里删掉、打一行日志就完事,
6
+ // 而唯一的补拉路径 `onExhausted` 要求「全池不通」——3 条活着时探活一直成功,那条死隧道
7
+ // 就**再也不会被拉起**,池静默从 4 降到 3,直到 daemon 重启。(真机::10800 死了就一直死。)
8
+ // 注意:复活走**磁盘上已有的 conf**(不重新注册)——进程挂了不代表设备失效,重注册只会白烧 CF 配额。
9
+ //
10
+ // 2. `labelExits`: 入池后实测各节点**真实出口地址**并标注撞车。WARP 的 IPv4 出口是共享 NAT,
11
+ // 「N 设备」很可能折叠成一个出口;没有这条读数就只能靠人手工 curl 才发现。
12
+ import { probeTunnel } from "./wireproxy.js";
13
+
14
+ /** 出口取证目标:Cloudflare 自己的 trace 端点,回 `ip=` 一行(不烧上游 token)。 */
15
+ export const EXIT_PROBE_URL = "https://cloudflare.com/cdn-cgi/trace";
16
+
17
+ /** 从 cdn-cgi/trace 正文里取出口地址;取不到返回 ""。 */
18
+ export function parseExitIp(text) {
19
+ const m = String(text || "").match(/^\s*ip=(\S+)\s*$/m);
20
+ return m ? m[1] : "";
21
+ }
22
+
23
+ /**
24
+ * 按端口独立的退避复活器。
25
+ * @param {{ revive: (port:number)=>Promise<boolean>, minGapMs?: number, maxGapMs?: number,
26
+ * now?: ()=>number, log?: (s:string)=>void }} opts
27
+ */
28
+ export function createReviver({ revive, minGapMs = 30_000, maxGapMs = 10 * 60_000, now = Date.now, log = () => {} } = {}) {
29
+ /** @type {Map<number, {attempts:number, nextAt:number, busy:boolean}>} */
30
+ const state = new Map();
31
+
32
+ function entry(port) {
33
+ let s = state.get(port);
34
+ if (!s) {
35
+ s = { attempts: 0, nextAt: 0, busy: false };
36
+ state.set(port, s);
37
+ }
38
+ return s;
39
+ }
40
+
41
+ return {
42
+ /** 进程退出时调用。首次立即复活,之后按 30s→10min 指数退避。 */
43
+ async onExit(port) {
44
+ const s = entry(port);
45
+ const wait = Math.max(0, s.nextAt - now());
46
+ if (wait > 0) {
47
+ log(`[warp-pool] :${port} 复活退避中(还剩 ${Math.round(wait / 1000)}s)`);
48
+ return false;
49
+ }
50
+ if (s.busy) return false;
51
+ s.busy = true;
52
+ s.attempts += 1;
53
+ const gap = Math.min(minGapMs * 2 ** (s.attempts - 1), maxGapMs);
54
+ s.nextAt = now() + gap;
55
+ let ok = false;
56
+ try {
57
+ ok = await revive(port);
58
+ } catch (e) {
59
+ log(`[warp-pool] :${port} 复活失败:${String(e?.message || e)}`);
60
+ }
61
+ s.busy = false;
62
+ if (ok) {
63
+ state.delete(port);
64
+ log(`[warp-pool] :${port} 已复活`);
65
+ } else {
66
+ log(`[warp-pool] :${port} 复活未成功(第 ${s.attempts} 次,下次最早 ${Math.round(gap / 1000)}s 后)`);
67
+ }
68
+ return ok;
69
+ },
70
+ /** 探活成功时调用:清掉退避账,恢复正常节奏。 */
71
+ noteHealthy(port) {
72
+ state.delete(port);
73
+ },
74
+ stats() {
75
+ return [...state.entries()].map(([port, s]) => ({
76
+ port,
77
+ attempts: s.attempts,
78
+ nextAt: s.nextAt,
79
+ cooldownMsLeft: Math.max(0, s.nextAt - now()),
80
+ }));
81
+ },
82
+ };
83
+ }
84
+
85
+ /**
86
+ * 入池后实测各节点真实出口地址并打日志(含撞车警告)。
87
+ * 只读探测:打 Cloudflare 的 trace 端点,不碰上游配额。
88
+ * @returns {Promise<{ ip: string, nodes: number[] }[]>} 按出口地址分组的节点清单
89
+ */
90
+ export async function labelExits({ ports = [], fetchImpl, timeoutMs = 12_000, log = () => {} } = {}) {
91
+ const byIp = new Map();
92
+ let failed = 0;
93
+ for (const port of ports) {
94
+ let ip = "";
95
+ try {
96
+ const res = await probeTunnel({ port, baseUrl: EXIT_PROBE_URL, path: "", fetchImpl, timeoutMs, wantText: true });
97
+ ip = typeof res?.text === "string" ? parseExitIp(res.text) : "";
98
+ } catch {
99
+ ip = "";
100
+ }
101
+ if (!ip) {
102
+ failed += 1;
103
+ continue;
104
+ }
105
+ if (!byIp.has(ip)) byIp.set(ip, []);
106
+ byIp.get(ip).push(port);
107
+ }
108
+
109
+ const groups = [...byIp.entries()].map(([ip, nodes]) => ({ ip, nodes }));
110
+ if (groups.length === 1) {
111
+ log(`[warp-pool] ⚠ 出口不分散:${groups[0].nodes.length} 个节点共用同一个出口 ${groups[0].ip}(免费版 WARP 的 IPv4 是共享 NAT)`);
112
+ } else if (groups.length > 1) {
113
+ log(`[warp-pool] 出口分散:${groups.length} 个不同地址 / ${groups.reduce((a, g) => a + g.nodes.length, 0)} 个节点`);
114
+ }
115
+ if (failed) log(`[warp-pool] ${failed} 个节点出口取证失败(不影响其进池)`);
116
+ return groups;
117
+ }
@@ -20,9 +20,21 @@ export function headerValue(headers, name) {
20
20
  return "";
21
21
  }
22
22
 
23
+ /** 让 dispatcher 看见上游状态码:复制一层 handler,只截 onHeaders,其余方法原样转发(全部调用都落在副本上,this 一致)。 */
24
+ export function tapHandler(handler, tap) {
25
+ if (!tap || !handler || typeof handler !== "object") return handler;
26
+ const out = Object.assign(Object.create(Object.getPrototypeOf(handler)), handler);
27
+ const orig = handler.onHeaders;
28
+ out.onHeaders = function (statusCode, headers, resume, statusText) {
29
+ try { tap(statusCode); } catch {}
30
+ return typeof orig === "function" ? orig.call(this, statusCode, headers, resume, statusText) : undefined;
31
+ };
32
+ return out;
33
+ }
34
+
23
35
  /**
24
36
  * @param {{ nodes?: Array<{port:number, uri:string}>, strategy?: "session"|"request",
25
- * cooldownMs?: number, now?: () => number, log?: (s:string)=>void,
37
+ * cooldownMs?: number, now?: () => number, log?: (s:string)=>void, onStatus?: (status:number, port:number)=>void,
26
38
  * agentOptions?: object, selector?: object }} [opts]
27
39
  */
28
40
  export function createRotatingDispatcher({
@@ -31,6 +43,7 @@ export function createRotatingDispatcher({
31
43
  cooldownMs = 600_000,
32
44
  now = Date.now,
33
45
  log = () => {},
46
+ onStatus = null,
34
47
  agentFactory,
35
48
  fallbackAgent,
36
49
  agentOptions = { connections: 8, keepAliveTimeout: 30_000, keepAliveMaxTimeout: 60_000, pipelining: 1 },
@@ -40,7 +53,7 @@ export function createRotatingDispatcher({
40
53
  if (typeof ProxyAgent !== "function") throw new Error("undici ProxyAgent 不可用:无法建立出口池");
41
54
  // 空池兜底:直连(等价 R3 里被顶掉的 keepAlive 池)。
42
55
  const fallback = fallbackAgent !== undefined ? fallbackAgent : typeof Agent === "function" ? new Agent(agentOptions) : null;
43
- let list = [];
56
+ let list = build(nodes);
44
57
 
45
58
  function build(nodesIn) {
46
59
  return (nodesIn || []).map((n, i) => ({
@@ -96,11 +109,11 @@ export function createRotatingDispatcher({
96
109
  throw new Error("warp-pool:节点池为空且无直连兜底");
97
110
  }
98
111
  const avail = availableIndexes();
99
- // 全池冷却时兜底用第一个,避免直接失败(宁可撞限流也别把请求打死)。
100
- if (!avail.length) return list[0].agent.dispatch(opts, handler);
112
+ // 全池冷却 ⇒ 直连兜底优先:硬派给已知不可用的节点 = 把请求挂到超时。
113
+ if (!avail.length) return (fallback || list[0].agent).dispatch(opts, handler);
101
114
  const sessionKey = strategy === "session" ? headerValue(opts?.headers, SESSION_HEADER) : "";
102
- const idx = selector.pick(avail, sessionKey);
103
- return list[idx].agent.dispatch(opts, handler);
115
+ const e = list[selector.pick(avail, sessionKey)];
116
+ return e.agent.dispatch(opts, tapHandler(handler, onStatus ? (sc) => onStatus(sc, e.port) : null));
104
117
  },
105
118
 
106
119
  /** 运行期换上一批出口(bootstrap 完成后调用);旧 Agent 池关闭。 */
@@ -91,6 +91,7 @@ function runCommand(cmd, args, { log = () => {} } = {}) {
91
91
 
92
92
  /** 写出某节点的 wireproxy 配置(含私钥 ⇒ 0600)。 */
93
93
  export function writeTunnelConf(dir, node) {
94
+ if (!node?.privateKey) throw new Error(`节点 :${node?.port} 缺 WireGuard 私钥,拒绝写出一份跑不起来的配置`);
94
95
  const { conf } = nodePaths(dir, node.port);
95
96
  return writeSecret(conf, renderWireproxyConf({ ...node, port: node.port }));
96
97
  }
@@ -135,7 +136,7 @@ export function startTunnel({ binPath: exe, dir, port, log = () => {}, onExit =
135
136
  }
136
137
 
137
138
  /** 探活一个已起的隧道:经该端口访问上游 models 路径(不烧 token)。 */
138
- export async function probeTunnel({ port, baseUrl = "https://opencode.ai", path = "/zen/v1/models", fetchImpl = compatFetch, timeoutMs = 8000, headers = {} } = {}) {
139
+ export async function probeTunnel({ port, baseUrl = "https://opencode.ai", path = "/zen/v1/models", fetchImpl = compatFetch, timeoutMs = 8000, headers = {}, wantText = false } = {}) {
139
140
  const { ProxyAgent } = getUndici();
140
141
  const agent = new ProxyAgent({ uri: `http://127.0.0.1:${port}`, connections: 2, keepAliveTimeout: 10_000 });
141
142
  try {
@@ -144,7 +145,10 @@ export async function probeTunnel({ port, baseUrl = "https://opencode.ai", path
144
145
  signal: AbortSignal.timeout(timeoutMs),
145
146
  dispatcher: agent,
146
147
  });
147
- return { ok: res.ok, status: res.status };
148
+ const base = { ok: res.ok, status: res.status };
149
+ if (!wantText) return base;
150
+ const text = await res.text().catch(() => "");
151
+ return { ...base, text: String(text).slice(0, 2000) };
148
152
  } catch (e) {
149
153
  return { ok: false, status: 0, error: String(e?.message || e) };
150
154
  } finally {