Skip to content

Commit 29d32a9

Browse files
committed
🤖 Fix SSH exec timeout with ControlMaster socket reuse
## Problem SSH commands with timeout could run indefinitely on remote hosts despite client-side timeouts. When ControlMaster multiplexing was enabled, killing the local SSH client didn't terminate the remote command—it continued running under the shared master connection. ## Solution **Three-layer timeout protection:** 1. **Client-side timeout** - SIGKILL after timeout expires (immediate local cleanup) 2. **Remote-side timeout** - Wrap commands with `timeout -s KILL` utility 3. **SSH connection timeout** - ConnectTimeout + ServerAlive* options Remote timeout is set to `timeout + 1s` to ensure local timeout normally fires first for cleaner error handling. Uses BusyBox-compatible syntax (`-s KILL`) for Alpine Linux compatibility. **ControlMaster always enabled** - Socket reuse now works for all operations, including timed commands. This preserves 50-200ms performance savings per operation from connection pooling. ## Testing - Added timeout integration test for both local and SSH runtimes - Verifies `sleep 10` with 1s timeout completes in ~1s (not 10s) - All 88 integration tests pass - 807 unit tests pass (4 pre-existing gitService failures unrelated) ## Impact Commands with timeout now properly terminate on remote hosts while maintaining the performance benefits of SSH connection multiplexing.
1 parent 71e965c commit 29d32a9

File tree

2 files changed

+61
-4
lines changed

2 files changed

+61
-4
lines changed

src/runtime/SSHRuntime.ts

Lines changed: 38 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -98,10 +98,22 @@ export class SSHRuntime implements Runtime {
9898
parts.push(command);
9999

100100
// Join all parts with && to ensure each step succeeds before continuing
101-
const fullCommand = parts.join(" && ");
101+
let fullCommand = parts.join(" && ");
102+
103+
// Wrap remote command with timeout if specified
104+
// This ensures the command is killed on the remote side even if the local
105+
// SSH client is killed but the ControlMaster connection persists
106+
if (options.timeout !== undefined) {
107+
// Use timeout command with KILL signal
108+
// Set remote timeout slightly longer (+1s) than local timeout to ensure
109+
// the local timeout fires first in normal cases (for cleaner error handling)
110+
// Note: Using BusyBox-compatible syntax (-s KILL) which also works with GNU timeout
111+
const remoteTimeout = Math.ceil(options.timeout) + 1;
112+
fullCommand = `timeout -s KILL ${remoteTimeout} bash -c ${shescape.quote(fullCommand)}`;
113+
}
102114

103115
// Wrap in bash -c with shescape for safe shell execution
104-
const remoteCommand = `bash -c ${shescape.quote(fullCommand)}`;
116+
const remoteCommand = options.timeout ? fullCommand : `bash -c ${shescape.quote(fullCommand)}`;
105117

106118
// Build SSH args
107119
const sshArgs: string[] = ["-T"];
@@ -125,10 +137,32 @@ export class SSHRuntime implements Runtime {
125137
// ControlMaster=auto: Create master connection if none exists, otherwise reuse
126138
// ControlPath: Unix socket path for multiplexing
127139
// ControlPersist=60: Keep master connection alive for 60s after last session
140+
//
141+
// Socket reuse is safe even with timeouts because:
142+
// - Each SSH command gets its own channel within the multiplexed connection
143+
// - SIGKILL on the client immediately closes that channel
144+
// - Remote sshd terminates the command when the channel closes
145+
// - Multiplexing only shares the TCP connection, not command lifetime
128146
sshArgs.push("-o", "ControlMaster=auto");
129147
sshArgs.push("-o", `ControlPath=${this.controlPath}`);
130148
sshArgs.push("-o", "ControlPersist=60");
131149

150+
// Set comprehensive timeout options to ensure SSH respects the timeout
151+
// ConnectTimeout: Maximum time to wait for connection establishment (DNS, TCP handshake, SSH auth)
152+
// ServerAliveInterval: Send keepalive every 5 seconds to detect dead connections
153+
// ServerAliveCountMax: Consider connection dead after 2 missed keepalives (10 seconds total)
154+
// Together these ensure that:
155+
// 1. Connection establishment can't hang indefinitely
156+
// 2. Established connections that die are detected quickly
157+
// 3. The overall timeout is respected from the moment ssh command starts
158+
if (options.timeout !== undefined) {
159+
// Use the configured timeout for connection establishment
160+
sshArgs.push("-o", `ConnectTimeout=${Math.ceil(options.timeout)}`);
161+
// Set aggressive keepalives to detect dead connections
162+
sshArgs.push("-o", "ServerAliveInterval=5");
163+
sshArgs.push("-o", "ServerAliveCountMax=2");
164+
}
165+
132166
sshArgs.push(this.config.host, remoteCommand);
133167

134168
// Debug: log the actual SSH command being executed
@@ -175,14 +209,14 @@ export class SSHRuntime implements Runtime {
175209

176210
// Handle abort signal
177211
if (options.abortSignal) {
178-
options.abortSignal.addEventListener("abort", () => sshProcess.kill());
212+
options.abortSignal.addEventListener("abort", () => sshProcess.kill("SIGKILL"));
179213
}
180214

181215
// Handle timeout
182216
if (options.timeout !== undefined) {
183217
setTimeout(() => {
184218
timedOut = true;
185-
sshProcess.kill();
219+
sshProcess.kill("SIGKILL");
186220
}, options.timeout * 1000);
187221
}
188222

tests/runtime/runtime.test.ts

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,29 @@ describeIntegration("Runtime integration tests", () => {
142142

143143
expect(result.stdout.trim()).toContain(workspace.path);
144144
});
145+
test.concurrent(
146+
"handles timeout correctly",
147+
async () => {
148+
const runtime = createRuntime();
149+
await using workspace = await TestWorkspace.create(runtime, type);
150+
151+
// Command that sleeps longer than timeout
152+
const startTime = performance.now();
153+
const result = await execBuffered(runtime, "sleep 10", {
154+
cwd: workspace.path,
155+
timeout: 1, // 1 second timeout
156+
});
157+
const duration = performance.now() - startTime;
158+
159+
// Exit code should be EXIT_CODE_TIMEOUT (-998)
160+
expect(result.exitCode).toBe(-998);
161+
// Should complete in around 1 second, not 10 seconds
162+
// Allow some margin for overhead (especially on SSH)
163+
expect(duration).toBeLessThan(3000); // 3 seconds max
164+
expect(duration).toBeGreaterThan(500); // At least 0.5 seconds
165+
},
166+
15000
167+
); // 15 second timeout for test (includes workspace creation overhead)
145168
});
146169

147170
describe("readFile() - File reading", () => {

0 commit comments

Comments
 (0)