mirror of
https://github.com/binaricat/Netcatty.git
synced 2026-11-06 14:30:13 +00:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
013a46ff22 | ||
|
|
4261859b1e | ||
|
|
c6de217850 |
No files matched your search
@@ -1077,6 +1077,132 @@ for (const rejects of [false, true]) {
|
||||
});
|
||||
}
|
||||
|
||||
for (const rejects of [false, true]) {
|
||||
test(`final failure notification and ${rejects ? "rejection" : "reply"} cannot restart an exhausted cycle`, async (t) => {
|
||||
let statusListener: ((status: PortForwardingRule["status"], error?: string) => void) | undefined;
|
||||
let forceFinalAttempt = true;
|
||||
const reconnectRule = rule({ id: `final-double-failure-${rejects}`, autoStart: true });
|
||||
Object.defineProperty(globalThis, "window", {
|
||||
configurable: true,
|
||||
value: { netcatty: {
|
||||
startPortForward: async () => {
|
||||
if (forceFinalAttempt) getActiveConnection(reconnectRule.id)!.reconnectAttempts = 5;
|
||||
statusListener?.("error", "SSH connection failed");
|
||||
if (rejects) throw new Error("SSH connection failed");
|
||||
return { success: false, error: "SSH connection failed" };
|
||||
},
|
||||
stopPortForwardByRuleId: async () => ({ stopped: 1, failed: 0, errors: [] }),
|
||||
onPortForwardStatus: (_id: string, listener: typeof statusListener) => {
|
||||
statusListener = listener;
|
||||
return () => undefined;
|
||||
},
|
||||
} },
|
||||
});
|
||||
let reconnects = 0;
|
||||
setReconnectCallback(async () => { reconnects += 1; return { success: true }; });
|
||||
t.after(async () => {
|
||||
setReconnectCallback(null);
|
||||
await stopAndCleanupRuleAndWait(reconnectRule.id);
|
||||
});
|
||||
t.mock.timers.enable({ apis: ["setTimeout"] });
|
||||
await startPortForward(reconnectRule, host(), [], [], [], () => undefined, true);
|
||||
t.mock.timers.tick(3000);
|
||||
assert.equal(reconnects, 0);
|
||||
assert.equal(resetReconnectAttempts(reconnectRule.id), true);
|
||||
|
||||
// An explicit restart must still get a new retry cycle if it fails.
|
||||
forceFinalAttempt = false;
|
||||
await startPortForward(reconnectRule, host(), [], [], [], () => undefined, true);
|
||||
assert.equal(getActiveConnection(reconnectRule.id)?.reconnectAttempts, 1);
|
||||
t.mock.timers.tick(3000);
|
||||
assert.equal(reconnects, 1);
|
||||
});
|
||||
}
|
||||
|
||||
for (const adoption of ["sync", "reconcile"] as const) {
|
||||
test(`${adoption} starts a fresh retry cycle when adopting an active backend tunnel`, async (t) => {
|
||||
const reconnectRule = rule({ id: `exhausted-adoption-${adoption}`, autoStart: true });
|
||||
const listeners = new Map<string, (status: PortForwardingRule["status"]) => void>();
|
||||
const backendTunnel = { ruleId: reconnectRule.id, tunnelId: "fresh-backend-tunnel", status: "active" };
|
||||
Object.defineProperty(globalThis, "window", {
|
||||
configurable: true,
|
||||
value: { netcatty: {
|
||||
startPortForward: async () => ({ success: true }),
|
||||
listPortForwards: async () => [backendTunnel],
|
||||
stopPortForwardByRuleId: async () => ({ stopped: 1, failed: 0, errors: [] }),
|
||||
onPortForwardStatus: (id: string, listener: (status: PortForwardingRule["status"]) => void) => {
|
||||
listeners.set(id, listener);
|
||||
return () => undefined;
|
||||
},
|
||||
} },
|
||||
});
|
||||
let reconnects = 0;
|
||||
setReconnectCallback(async () => { reconnects += 1; return { success: true }; }, () => true);
|
||||
t.after(async () => {
|
||||
setReconnectCallback(null);
|
||||
await stopAndCleanupRuleAndWait(reconnectRule.id);
|
||||
});
|
||||
t.mock.timers.enable({ apis: ["setTimeout"] });
|
||||
await startPortForward(reconnectRule, host(), [], [], [], () => undefined, true);
|
||||
const oldConnection = getActiveConnection(reconnectRule.id)!;
|
||||
oldConnection.reconnectAttempts = 5;
|
||||
const oldListener = listeners.get(oldConnection.tunnelId)!;
|
||||
oldListener("error");
|
||||
oldListener("inactive");
|
||||
assert.equal(resetReconnectAttempts(reconnectRule.id), true);
|
||||
|
||||
if (adoption === "sync") await syncWithBackend({ shouldReconnect: () => true });
|
||||
else await reconcileWithBackend();
|
||||
assert.equal(resetReconnectAttempts(reconnectRule.id), false);
|
||||
const replacement = getActiveConnection(reconnectRule.id)!;
|
||||
assert.equal(replacement.tunnelId, backendTunnel.tunnelId);
|
||||
oldListener("error");
|
||||
assert.equal(getActiveConnection(reconnectRule.id), replacement);
|
||||
listeners.get(backendTunnel.tunnelId)!("inactive");
|
||||
assert.equal(getActiveConnection(reconnectRule.id)?.reconnectAttempts, 1);
|
||||
t.mock.timers.tick(3000);
|
||||
assert.equal(reconnects, 1);
|
||||
});
|
||||
}
|
||||
|
||||
for (const adoption of ["sync", "reconcile"] as const) {
|
||||
for (const rejects of [false, true]) {
|
||||
test(`${adoption} preserves exhaustion if backend adoption ${rejects ? "rejects" : "finds a stopped tunnel"}`, async (t) => {
|
||||
const reconnectRule = rule({ id: `failed-adoption-${adoption}-${rejects}`, autoStart: true });
|
||||
let statusListener: ((status: PortForwardingRule["status"]) => void) | undefined;
|
||||
Object.defineProperty(globalThis, "window", {
|
||||
configurable: true,
|
||||
value: { netcatty: {
|
||||
startPortForward: async () => ({ success: true }),
|
||||
listPortForwards: async () => [{ ruleId: reconnectRule.id, tunnelId: "stopped-before-subscribe", status: "active" }],
|
||||
subscribePortForward: async () => {
|
||||
if (rejects) throw new Error("backend unavailable during adoption");
|
||||
return { tunnelId: "stopped-before-subscribe", status: "inactive" };
|
||||
},
|
||||
stopPortForwardByRuleId: async () => ({ stopped: 1, failed: 0, errors: [] }),
|
||||
onPortForwardStatus: (_id: string, listener: typeof statusListener) => {
|
||||
statusListener = listener;
|
||||
return () => undefined;
|
||||
},
|
||||
} },
|
||||
});
|
||||
setReconnectCallback(async () => ({ success: true }), () => true);
|
||||
t.after(async () => {
|
||||
setReconnectCallback(null);
|
||||
await stopAndCleanupRuleAndWait(reconnectRule.id);
|
||||
});
|
||||
await startPortForward(reconnectRule, host(), [], [], [], () => undefined, true);
|
||||
getActiveConnection(reconnectRule.id)!.reconnectAttempts = 5;
|
||||
statusListener!("error");
|
||||
statusListener!("inactive");
|
||||
assert.equal(resetReconnectAttempts(reconnectRule.id), true);
|
||||
if (adoption === "sync") await syncWithBackend({ shouldReconnect: () => true });
|
||||
else await reconcileWithBackend();
|
||||
assert.equal(resetReconnectAttempts(reconnectRule.id), true);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
test("inactive close events preserve an already scheduled reconnect", async (t) => {
|
||||
let statusListener: ((status: PortForwardingRule["status"], error?: string | null) => void) | undefined;
|
||||
Object.defineProperty(globalThis, "window", {
|
||||
|
||||
@@ -234,6 +234,10 @@ const scheduleReconnectIfNeeded = (
|
||||
return true;
|
||||
}
|
||||
|
||||
// One failed dial can report both a status event and a rejected start.
|
||||
// Once exhausted, neither notification may begin another retry cycle.
|
||||
if (exhaustedReconnectRules.has(ruleId)) return false;
|
||||
|
||||
const currentConn = activeConnections.get(ruleId);
|
||||
if (currentConn?.reconnectSuppressed) return false;
|
||||
if (currentConn?.reconnectTimerCallback) {
|
||||
@@ -681,6 +685,10 @@ const subscribeSyncedConnection = async (
|
||||
connection.status = resolveBackendStatus(snapshot.status);
|
||||
connection.error = snapshot.error;
|
||||
}
|
||||
if (connection.status === 'active') {
|
||||
exhaustedReconnectRules.delete(ruleId);
|
||||
connection.reconnectAttempts = 0;
|
||||
}
|
||||
return true;
|
||||
};
|
||||
|
||||
@@ -820,6 +828,10 @@ export const reconcileWithBackend = async (): Promise<{
|
||||
if (existing.status !== backendStatus || existing.error !== tunnel.error) {
|
||||
existing.status = backendStatus;
|
||||
existing.error = tunnel.error;
|
||||
if (backendStatus === 'active') {
|
||||
exhaustedReconnectRules.delete(ruleId);
|
||||
existing.reconnectAttempts = 0;
|
||||
}
|
||||
result.appeared.push(ruleId);
|
||||
}
|
||||
}
|
||||
@@ -893,7 +905,9 @@ export const startPortForward = async (
|
||||
|
||||
// Clear any existing reconnect timer
|
||||
clearReconnectTimer(rule.id);
|
||||
|
||||
// Explicit starts (including network recovery) authorize a fresh cycle.
|
||||
exhaustedReconnectRules.delete(rule.id);
|
||||
|
||||
if (!bridge?.startPortForward) {
|
||||
// Fallback for browser/dev mode - simulate the connection
|
||||
logger.warn('[PortForwardingService] Backend not available, simulating connection...');
|
||||
|
||||
Reference in new issue
Block a user