@fitlab-ai/agent-infra 0.9.15 → 0.11.0
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/README.md +2 -6
- package/README.zh-CN.md +2 -6
- package/bin/internal-cli.sh +0 -13
- package/dist/bin/internal-cli.js +11 -49
- package/dist/lib/agent-clients/adapters/codex-lifecycle/sandbox-controller.js +6 -2
- package/dist/lib/agent-clients/adapters/codex-lifecycle/store.js +43 -3
- package/dist/lib/fs/durable-write.js +2 -0
- package/dist/lib/internal/cli-route-inventory.js +3 -4
- package/dist/lib/internal/manual-validation.js +17 -8
- package/dist/lib/internal/platform-comment.js +60 -2
- package/dist/lib/internal/task-artifact.js +1 -1
- package/dist/lib/internal/task-finalization.js +2 -2
- package/dist/lib/internal/task-lifecycle.js +4 -4
- package/dist/lib/internal/task-operation-registry.js +22 -13
- package/dist/lib/internal/task-orchestration.js +2 -2
- package/dist/lib/internal/task-review.js +12 -7
- package/dist/lib/platform/activity-recovery.js +159 -0
- package/dist/lib/platform/issue-comments.js +89 -31
- package/dist/lib/platform/issues.js +1 -1
- package/dist/lib/platform/pr-change-report.js +5 -4
- package/dist/lib/platform/pr-summary.js +64 -32
- package/dist/lib/platform/pull-requests.js +3 -3
- package/dist/lib/platform/task-comment-projection.js +70 -0
- package/dist/lib/platform/task-identities.js +2 -9
- package/dist/lib/platform/task-recovery.js +55 -0
- package/dist/lib/platform/verification-sync.js +45 -7
- package/dist/lib/pr-review/evidence-grading.js +4 -8
- package/dist/lib/sandbox/commands/rm.js +21 -2
- package/dist/lib/sandbox/control/executor.js +4 -2
- package/dist/lib/sandbox/control/server.js +29 -9
- package/dist/lib/sandbox/control/state.js +7 -1
- package/dist/lib/sandbox/control/task-workflow.js +1 -1
- package/dist/lib/sandbox/control/workflow-executor.js +87 -35
- package/dist/lib/sandbox/index.js +1 -2
- package/dist/lib/sandbox/removal.js +116 -1011
- package/dist/lib/sandbox/workspace-identity.js +17 -1
- package/dist/lib/server/process-state.js +17 -5
- package/dist/lib/task/artifact-command.js +11 -30
- package/dist/lib/task/artifact-operations.js +17 -231
- package/dist/lib/task/artifact-recovery.js +476 -0
- package/dist/lib/task/artifact-repair-intent.js +90 -39
- package/dist/lib/task/commit-operation.js +31 -4
- package/dist/lib/task/control-authority.js +39 -8
- package/dist/lib/task/control-recovery.js +14 -2
- package/dist/lib/task/create-service.js +1 -8
- package/dist/lib/task/delegation-receipts.js +64 -3
- package/dist/lib/task/events.js +165 -23
- package/dist/lib/task/human-override.js +5 -8
- package/dist/lib/task/intermediate-cleanup.js +91 -8
- package/dist/lib/task/lifecycle-recovery.js +549 -0
- package/dist/lib/task/lifecycle.js +12 -3
- package/dist/lib/task/local-artifact-finalization.js +212 -123
- package/dist/lib/task/manual-validation-receipt.js +5 -1
- package/dist/lib/task/orchestration.js +52 -2
- package/dist/lib/task/pr-delivery-fact.js +12 -56
- package/dist/lib/task/recovery-warning.js +19 -0
- package/dist/lib/task/review-command.js +3 -2
- package/dist/lib/task/review-finalization.js +294 -74
- package/dist/lib/{host-control → task}/secure-fs.js +8 -19
- package/dist/lib/task/short-id.js +15 -7
- package/dist/lib/task/task-execution-lock.js +1 -1
- package/dist/lib/task/verification-engine.js +3 -14
- package/dist/lib/task/workflow-command.js +2 -1
- package/dist/lib/task/workflow-dispatch.js +161 -0
- package/dist/lib/version.js +1 -11
- package/dist/lifecycle-build-manifest.json +3 -3
- package/lib/agent-clients/adapters/codex-lifecycle/sandbox-controller.ts +7 -2
- package/lib/agent-clients/adapters/codex-lifecycle/store.ts +55 -3
- package/lib/fs/durable-write.ts +1 -0
- package/lib/internal/cli-route-inventory.ts +3 -4
- package/lib/internal/manual-validation.ts +17 -8
- package/lib/internal/platform-comment.ts +53 -2
- package/lib/internal/task-artifact.ts +1 -1
- package/lib/internal/task-finalization.ts +1 -1
- package/lib/internal/task-lifecycle.ts +5 -4
- package/lib/internal/task-operation-registry.ts +22 -13
- package/lib/internal/task-orchestration.ts +1 -1
- package/lib/internal/task-review.ts +11 -7
- package/lib/platform/activity-recovery.ts +179 -0
- package/lib/platform/issue-comments.ts +105 -30
- package/lib/platform/issues.ts +1 -1
- package/lib/platform/pr-change-report.ts +7 -5
- package/lib/platform/pr-summary.ts +64 -32
- package/lib/platform/pull-requests.ts +3 -3
- package/lib/platform/task-comment-projection.ts +79 -0
- package/lib/platform/task-identities.ts +3 -14
- package/lib/platform/task-recovery.ts +54 -0
- package/lib/platform/verification-sync.ts +48 -11
- package/lib/pr-review/evidence-grading.ts +4 -9
- package/lib/sandbox/commands/rm.ts +34 -2
- package/lib/sandbox/control/executor.ts +4 -2
- package/lib/sandbox/control/protocol.ts +6 -1
- package/lib/sandbox/control/server.ts +30 -10
- package/lib/sandbox/control/state.ts +9 -1
- package/lib/sandbox/control/task-workflow.ts +1 -1
- package/lib/sandbox/control/workflow-executor.ts +106 -35
- package/lib/sandbox/index.ts +1 -2
- package/lib/sandbox/removal.ts +126 -1128
- package/lib/sandbox/workspace-identity.ts +19 -3
- package/lib/server/process-state.ts +26 -5
- package/lib/task/artifact-command.ts +12 -29
- package/lib/task/artifact-operations.ts +22 -252
- package/lib/task/artifact-recovery.ts +572 -0
- package/lib/task/artifact-repair-intent.ts +126 -51
- package/lib/task/commit-operation.ts +35 -4
- package/lib/task/control-authority.ts +48 -15
- package/lib/task/control-recovery.ts +13 -1
- package/lib/task/create-service.ts +1 -10
- package/lib/task/delegation-receipts.ts +73 -2
- package/lib/task/events.ts +188 -24
- package/lib/task/human-override.ts +5 -8
- package/lib/task/intermediate-cleanup.ts +66 -12
- package/lib/task/lifecycle-recovery.ts +691 -0
- package/lib/task/lifecycle.ts +13 -4
- package/lib/task/local-artifact-finalization.ts +249 -131
- package/lib/task/manual-validation-receipt.ts +5 -1
- package/lib/task/orchestration.ts +65 -0
- package/lib/task/pr-delivery-fact.ts +15 -62
- package/lib/task/recovery-warning.ts +21 -0
- package/lib/task/review-command.ts +2 -1
- package/lib/task/review-finalization.ts +298 -84
- package/lib/{host-control → task}/secure-fs.ts +10 -32
- package/lib/task/short-id.ts +27 -5
- package/lib/task/task-execution-lock.ts +1 -1
- package/lib/task/verification-engine.ts +3 -17
- package/lib/task/verification-types.ts +0 -2
- package/lib/task/workflow-command.ts +2 -1
- package/lib/task/workflow-dispatch.ts +171 -0
- package/lib/version.ts +1 -15
- package/package.json +1 -1
- package/templates/.agents/rules/create-issue.en.md +1 -1
- package/templates/.agents/rules/create-issue.zh-CN.md +1 -1
- package/templates/.agents/rules/issue-sync.en.md +12 -2
- package/templates/.agents/rules/issue-sync.zh-CN.md +13 -5
- package/templates/.agents/rules/local-artifact-repair.en.md +47 -39
- package/templates/.agents/rules/local-artifact-repair.zh-CN.md +45 -36
- package/templates/.agents/skills/analyze-task/SKILL.en.md +6 -6
- package/templates/.agents/skills/analyze-task/SKILL.zh-CN.md +6 -6
- package/templates/.agents/skills/analyze-task/config/verify.en.json +1 -1
- package/templates/.agents/skills/analyze-task/config/verify.zh-CN.json +1 -1
- package/templates/.agents/skills/block-task/SKILL.en.md +2 -2
- package/templates/.agents/skills/block-task/SKILL.zh-CN.md +2 -2
- package/templates/.agents/skills/block-task/config/verify.json +1 -1
- package/templates/.agents/skills/cancel-task/SKILL.en.md +3 -3
- package/templates/.agents/skills/cancel-task/SKILL.zh-CN.md +3 -3
- package/templates/.agents/skills/cancel-task/config/verify.json +1 -1
- package/templates/.agents/skills/code-task/SKILL.en.md +7 -7
- package/templates/.agents/skills/code-task/SKILL.zh-CN.md +8 -8
- package/templates/.agents/skills/code-task/config/verify.en.json +1 -1
- package/templates/.agents/skills/code-task/config/verify.zh-CN.json +1 -1
- package/templates/.agents/skills/commit/reference/issue-metadata-sync.en.md +1 -1
- package/templates/.agents/skills/commit/reference/issue-metadata-sync.zh-CN.md +1 -1
- package/templates/.agents/skills/complete-manual-validation/SKILL.en.md +1 -1
- package/templates/.agents/skills/complete-manual-validation/SKILL.zh-CN.md +1 -1
- package/templates/.agents/skills/complete-task/SKILL.en.md +2 -2
- package/templates/.agents/skills/complete-task/SKILL.zh-CN.md +2 -2
- package/templates/.agents/skills/complete-task/config/verify.en.json +2 -2
- package/templates/.agents/skills/complete-task/config/verify.zh-CN.json +2 -2
- package/templates/.agents/skills/complete-task/reference/external-delivery.en.md +1 -1
- package/templates/.agents/skills/complete-task/reference/external-delivery.zh-CN.md +1 -1
- package/templates/.agents/skills/create-pr/SKILL.en.md +2 -2
- package/templates/.agents/skills/create-pr/SKILL.zh-CN.md +2 -2
- package/templates/.agents/skills/create-pr/config/verify.json +1 -1
- package/templates/.agents/skills/create-pr/reference/pr-body-template.en.md +2 -2
- package/templates/.agents/skills/create-pr/reference/pr-body-template.zh-CN.md +2 -2
- package/templates/.agents/skills/create-task/SKILL.en.md +5 -5
- package/templates/.agents/skills/create-task/SKILL.zh-CN.md +5 -5
- package/templates/.agents/skills/create-task/config/verify.json +1 -1
- package/templates/.agents/skills/import-issue/SKILL.en.md +8 -8
- package/templates/.agents/skills/import-issue/SKILL.zh-CN.md +8 -8
- package/templates/.agents/skills/import-issue/config/verify.json +2 -2
- package/templates/.agents/skills/plan-task/SKILL.en.md +4 -4
- package/templates/.agents/skills/plan-task/SKILL.zh-CN.md +4 -4
- package/templates/.agents/skills/plan-task/config/verify.en.json +1 -1
- package/templates/.agents/skills/plan-task/config/verify.zh-CN.json +1 -1
- package/templates/.agents/skills/restore-task/SKILL.en.md +11 -6
- package/templates/.agents/skills/restore-task/SKILL.zh-CN.md +11 -6
- package/templates/.agents/skills/restore-task/config/verify.json +1 -1
- package/templates/.agents/skills/review-analysis/SKILL.en.md +3 -3
- package/templates/.agents/skills/review-analysis/SKILL.zh-CN.md +3 -3
- package/templates/.agents/skills/review-analysis/config/verify.en.json +1 -1
- package/templates/.agents/skills/review-analysis/config/verify.zh-CN.json +1 -1
- package/templates/.agents/skills/review-analysis/reference/report-template.en.md +7 -7
- package/templates/.agents/skills/review-analysis/reference/report-template.zh-CN.md +6 -6
- package/templates/.agents/skills/review-code/SKILL.en.md +5 -4
- package/templates/.agents/skills/review-code/SKILL.zh-CN.md +5 -4
- package/templates/.agents/skills/review-code/config/verify.en.json +1 -1
- package/templates/.agents/skills/review-code/config/verify.zh-CN.json +1 -1
- package/templates/.agents/skills/review-code/reference/report-template.en.md +12 -12
- package/templates/.agents/skills/review-code/reference/report-template.zh-CN.md +12 -12
- package/templates/.agents/skills/review-plan/SKILL.en.md +3 -3
- package/templates/.agents/skills/review-plan/SKILL.zh-CN.md +3 -3
- package/templates/.agents/skills/review-plan/config/verify.en.json +1 -1
- package/templates/.agents/skills/review-plan/config/verify.zh-CN.json +1 -1
- package/templates/.agents/skills/review-plan/reference/report-template.en.md +7 -7
- package/templates/.agents/skills/review-plan/reference/report-template.zh-CN.md +7 -7
- package/templates/.agents/skills/review-pr/config/verify.en.json +1 -1
- package/templates/.agents/skills/review-pr/config/verify.zh-CN.json +1 -1
- package/templates/.agents/skills/review-pr/reference/evidence-grading.en.md +3 -3
- package/templates/.agents/skills/review-pr/reference/evidence-grading.zh-CN.md +3 -3
- package/templates/.agents/skills/run-manual-validation/config/verify.en.json +1 -1
- package/templates/.agents/skills/run-manual-validation/config/verify.zh-CN.json +1 -1
- package/templates/.agents/skills/run-task/SKILL.en.md +1 -1
- package/templates/.agents/skills/run-task/SKILL.zh-CN.md +1 -1
- package/templates/.agents/templates/task.en.md +1 -1
- package/templates/.agents/templates/task.zh-CN.md +1 -1
- package/templates/.claude/commands/import-issue.en.md +1 -1
- package/templates/.claude/commands/import-issue.zh-CN.md +1 -1
- package/templates/.claude/commands/restore-task.en.md +1 -1
- package/templates/.claude/commands/restore-task.zh-CN.md +1 -1
- package/dist/lib/host-control/audit.js +0 -19
- package/dist/lib/host-control/client.js +0 -142
- package/dist/lib/host-control/command.js +0 -49
- package/dist/lib/host-control/path.js +0 -160
- package/dist/lib/host-control/server.js +0 -159
- package/dist/lib/host-control/service.js +0 -112
- package/dist/lib/internal/host-control.js +0 -28
- package/lib/host-control/audit.ts +0 -22
- package/lib/host-control/client.ts +0 -172
- package/lib/host-control/command.ts +0 -60
- package/lib/host-control/path.ts +0 -167
- package/lib/host-control/server.ts +0 -180
- package/lib/host-control/service.ts +0 -133
- package/lib/internal/host-control.ts +0 -28
|
@@ -1,172 +0,0 @@
|
|
|
1
|
-
import net from 'node:net';
|
|
2
|
-
import { randomUUID } from 'node:crypto';
|
|
3
|
-
|
|
4
|
-
import {
|
|
5
|
-
inspectHostControlEndpoint,
|
|
6
|
-
resolveHostControlEndpoint
|
|
7
|
-
} from './path.ts';
|
|
8
|
-
|
|
9
|
-
export const HOST_CONTROL_MAX_REQUEST_BYTES = 64 * 1024;
|
|
10
|
-
|
|
11
|
-
export const HOST_CONTROL_COMMANDS = Object.freeze([
|
|
12
|
-
'task-lifecycle', 'task-orchestration', 'task-finalization',
|
|
13
|
-
'task-artifact', 'task-review', 'task-event', 'task-ledger',
|
|
14
|
-
'task-invalidation', 'task-warning'
|
|
15
|
-
] as const);
|
|
16
|
-
export type HostControlCommand = typeof HOST_CONTROL_COMMANDS[number];
|
|
17
|
-
|
|
18
|
-
export type HostControlCommandPayload = Readonly<{
|
|
19
|
-
workingDirectory: string;
|
|
20
|
-
args: readonly string[];
|
|
21
|
-
environment?: Readonly<Record<string, string>>;
|
|
22
|
-
}>;
|
|
23
|
-
|
|
24
|
-
export const HOST_CONTROL_TEST_ENVIRONMENT_KEYS = Object.freeze([
|
|
25
|
-
'TZ', 'AGENT_INFRA_GH_BIN', 'AGENT_INFRA_GH_ARGS_JSON', 'GH_FAKE_PR_PATH', 'GH_FAKE_ARGS_PATH'
|
|
26
|
-
] as const);
|
|
27
|
-
|
|
28
|
-
export type HostControlRequest = Readonly<{
|
|
29
|
-
version: 1;
|
|
30
|
-
id: string;
|
|
31
|
-
taskId: string | null;
|
|
32
|
-
generation: string | null;
|
|
33
|
-
operation: HostControlCommand;
|
|
34
|
-
scope: 'host-command';
|
|
35
|
-
payload: HostControlCommandPayload;
|
|
36
|
-
}>;
|
|
37
|
-
|
|
38
|
-
export type HostControlResponse = Readonly<{
|
|
39
|
-
version: 1;
|
|
40
|
-
id: string;
|
|
41
|
-
status: 'completed' | 'rejected' | 'unknown';
|
|
42
|
-
exitCode: number;
|
|
43
|
-
stdout: string;
|
|
44
|
-
stderr: string;
|
|
45
|
-
error: Readonly<{ code: string; message: string }> | null;
|
|
46
|
-
}>;
|
|
47
|
-
|
|
48
|
-
export class HostControlClientError extends Error {
|
|
49
|
-
readonly code: string;
|
|
50
|
-
readonly retryable: boolean;
|
|
51
|
-
readonly changed: false | null;
|
|
52
|
-
constructor(code: string, message: string, retryable = false, submitted = false) {
|
|
53
|
-
super(`${code}: ${message}`);
|
|
54
|
-
this.name = 'HostControlClientError';
|
|
55
|
-
this.code = code;
|
|
56
|
-
this.retryable = retryable && !submitted;
|
|
57
|
-
this.changed = submitted ? null : false;
|
|
58
|
-
}
|
|
59
|
-
}
|
|
60
|
-
|
|
61
|
-
function reject(message: string): never {
|
|
62
|
-
throw new HostControlClientError('HOST_CONTROL_REQUEST_INVALID', message);
|
|
63
|
-
}
|
|
64
|
-
|
|
65
|
-
export function validateHostControlRequest(value: unknown): HostControlRequest {
|
|
66
|
-
if (!value || typeof value !== 'object' || Array.isArray(value)) reject('request must be an object');
|
|
67
|
-
const request = value as HostControlRequest;
|
|
68
|
-
if (Object.keys(request).sort().join(',') !== 'generation,id,operation,payload,scope,taskId,version'
|
|
69
|
-
|| request.version !== 1 || typeof request.id !== 'string' || !/^[a-f0-9-]{16,64}$/u.test(request.id)
|
|
70
|
-
|| (request.taskId !== null && (typeof request.taskId !== 'string' || !/^TASK-\d{8}-\d{6}$/u.test(request.taskId)))
|
|
71
|
-
|| (request.generation !== null && (typeof request.generation !== 'string' || !request.generation))
|
|
72
|
-
|| request.scope !== 'host-command' || !HOST_CONTROL_COMMANDS.includes(request.operation as HostControlCommand)) reject('request schema is invalid');
|
|
73
|
-
const payload = request.payload;
|
|
74
|
-
if (!payload || typeof payload !== 'object' || Array.isArray(payload)
|
|
75
|
-
|| !['args,environment,workingDirectory', 'args,workingDirectory'].includes(Object.keys(payload).sort().join(','))
|
|
76
|
-
|| !Array.isArray(payload.args) || payload.args.length > 128
|
|
77
|
-
|| !payload.args.every((arg) => typeof arg === 'string' && !/[\r\n]/u.test(arg))
|
|
78
|
-
|| typeof payload.workingDirectory !== 'string' || !payload.workingDirectory.startsWith('/')) reject('host-command payload is invalid');
|
|
79
|
-
if (payload.environment !== undefined) {
|
|
80
|
-
if (!payload.environment || typeof payload.environment !== 'object' || Array.isArray(payload.environment)) reject('host-command environment is invalid');
|
|
81
|
-
for (const [key, value] of Object.entries(payload.environment)) {
|
|
82
|
-
if (!HOST_CONTROL_TEST_ENVIRONMENT_KEYS.includes(key as typeof HOST_CONTROL_TEST_ENVIRONMENT_KEYS[number])
|
|
83
|
-
|| typeof value !== 'string' || /[\r\n]/u.test(value)) reject('host-command environment is invalid');
|
|
84
|
-
}
|
|
85
|
-
}
|
|
86
|
-
return request;
|
|
87
|
-
}
|
|
88
|
-
|
|
89
|
-
export function hostControlRequestForCommand(
|
|
90
|
-
command: HostControlCommand,
|
|
91
|
-
args: readonly string[],
|
|
92
|
-
workingDirectory: string,
|
|
93
|
-
environment: Readonly<Record<string, string>> = {}
|
|
94
|
-
): HostControlRequest {
|
|
95
|
-
if (!HOST_CONTROL_COMMANDS.includes(command)) reject('host-command is unsupported');
|
|
96
|
-
const taskArg = args[0];
|
|
97
|
-
const taskId = taskArg && /^TASK-\d{8}-\d{6}$/u.test(taskArg) ? taskArg : null;
|
|
98
|
-
return validateHostControlRequest({
|
|
99
|
-
version: 1,
|
|
100
|
-
id: randomUUID(),
|
|
101
|
-
taskId,
|
|
102
|
-
generation: null,
|
|
103
|
-
operation: command,
|
|
104
|
-
scope: 'host-command',
|
|
105
|
-
payload: {
|
|
106
|
-
workingDirectory,
|
|
107
|
-
args: [...args],
|
|
108
|
-
...(Object.keys(environment).length > 0 ? { environment } : {})
|
|
109
|
-
}
|
|
110
|
-
});
|
|
111
|
-
}
|
|
112
|
-
|
|
113
|
-
function parseResponse(value: unknown, id: string): HostControlResponse {
|
|
114
|
-
if (!value || typeof value !== 'object' || Array.isArray(value)) throw new HostControlClientError('HOST_CONTROL_RESPONSE_INVALID', 'response must be an object');
|
|
115
|
-
const response = value as Record<string, unknown>;
|
|
116
|
-
if (Object.keys(response).sort().join(',') !== 'error,exitCode,id,status,stderr,stdout,version'
|
|
117
|
-
|| response.version !== 1 || response.id !== id
|
|
118
|
-
|| !['completed', 'rejected', 'unknown'].includes(response.status as string)
|
|
119
|
-
|| !Number.isSafeInteger(response.exitCode)
|
|
120
|
-
|| typeof response.stdout !== 'string' || typeof response.stderr !== 'string'
|
|
121
|
-
|| (response.error !== null && (!response.error || typeof response.error !== 'object'))) {
|
|
122
|
-
throw new HostControlClientError('HOST_CONTROL_RESPONSE_INVALID', 'response schema is invalid');
|
|
123
|
-
}
|
|
124
|
-
return response as HostControlResponse;
|
|
125
|
-
}
|
|
126
|
-
|
|
127
|
-
export async function requestHostControl(params: Readonly<{
|
|
128
|
-
request: HostControlRequest;
|
|
129
|
-
endpoint?: string;
|
|
130
|
-
timeoutMs?: number;
|
|
131
|
-
}>): Promise<HostControlResponse> {
|
|
132
|
-
const request = validateHostControlRequest(params.request);
|
|
133
|
-
const endpoint = params.endpoint ?? resolveHostControlEndpoint();
|
|
134
|
-
const inspection = inspectHostControlEndpoint(endpoint, { uid: process.getuid?.() });
|
|
135
|
-
if (!inspection.ok) throw new HostControlClientError(inspection.code ?? 'HOST_CONTROL_UNAVAILABLE', 'fixed host-control endpoint is unavailable', true);
|
|
136
|
-
const encoded = `${JSON.stringify(request)}\n`;
|
|
137
|
-
if (Buffer.byteLength(encoded, 'utf8') > HOST_CONTROL_MAX_REQUEST_BYTES) throw new HostControlClientError('HOST_CONTROL_REQUEST_TOO_LARGE', 'request exceeds the control limit');
|
|
138
|
-
const timeoutMs = params.timeoutMs ?? 30_000;
|
|
139
|
-
return await new Promise<HostControlResponse>((resolve, rejectPromise) => {
|
|
140
|
-
const socket = net.createConnection(endpoint);
|
|
141
|
-
let output = '';
|
|
142
|
-
let settled = false;
|
|
143
|
-
let submitted = false;
|
|
144
|
-
const failure = (code: string, message: string, retryable = false) =>
|
|
145
|
-
new HostControlClientError(code, message, retryable, submitted);
|
|
146
|
-
const finish = (error?: Error, response?: HostControlResponse): void => {
|
|
147
|
-
if (settled) return;
|
|
148
|
-
settled = true;
|
|
149
|
-
clearTimeout(timer);
|
|
150
|
-
socket.destroy();
|
|
151
|
-
if (error) rejectPromise(error); else resolve(response!);
|
|
152
|
-
};
|
|
153
|
-
const timer = setTimeout(() => finish(failure('HOST_CONTROL_TIMEOUT', 'host-control did not respond; inspect task state before retrying', true)), timeoutMs);
|
|
154
|
-
socket.setEncoding('utf8');
|
|
155
|
-
socket.on('connect', () => { submitted = true; socket.end(encoded); });
|
|
156
|
-
socket.on('data', (chunk: string) => {
|
|
157
|
-
output += chunk;
|
|
158
|
-
if (Buffer.byteLength(output, 'utf8') > HOST_CONTROL_MAX_REQUEST_BYTES) {
|
|
159
|
-
finish(failure('HOST_CONTROL_RESPONSE_TOO_LARGE', 'host-control response exceeds the control limit'));
|
|
160
|
-
return;
|
|
161
|
-
}
|
|
162
|
-
const newline = output.indexOf('\n');
|
|
163
|
-
if (newline < 0) return;
|
|
164
|
-
try { finish(undefined, parseResponse(JSON.parse(output.slice(0, newline)), request.id)); }
|
|
165
|
-
catch (error) { finish(failure('HOST_CONTROL_RESPONSE_INVALID', String(error))); }
|
|
166
|
-
});
|
|
167
|
-
socket.on('error', (error) => finish(failure('HOST_CONTROL_UNAVAILABLE', error.message, true)));
|
|
168
|
-
socket.on('close', () => {
|
|
169
|
-
if (!settled) finish(failure('HOST_CONTROL_RESPONSE_INVALID', 'host-control closed without a response'));
|
|
170
|
-
});
|
|
171
|
-
});
|
|
172
|
-
}
|
|
@@ -1,60 +0,0 @@
|
|
|
1
|
-
import { spawn } from 'node:child_process';
|
|
2
|
-
import fs from 'node:fs';
|
|
3
|
-
import { fileURLToPath } from 'node:url';
|
|
4
|
-
|
|
5
|
-
import type { HostControlRequest } from './client.ts';
|
|
6
|
-
import { readHostControlWorkerToken, resolveHostControlEndpoint } from './path.ts';
|
|
7
|
-
|
|
8
|
-
type HostCommandResult = Readonly<{
|
|
9
|
-
stdout: string;
|
|
10
|
-
stderr: string;
|
|
11
|
-
exitCode: number;
|
|
12
|
-
}>;
|
|
13
|
-
|
|
14
|
-
export async function dispatchHostControlCommand(
|
|
15
|
-
request: Pick<HostControlRequest, 'operation' | 'payload'>
|
|
16
|
-
): Promise<HostCommandResult> {
|
|
17
|
-
const { payload } = request;
|
|
18
|
-
const compiledEntry = fileURLToPath(new URL('../../bin/internal-cli.js', import.meta.url));
|
|
19
|
-
const sourceEntry = fileURLToPath(new URL('../../bin/internal-cli.ts', import.meta.url));
|
|
20
|
-
const entry = fs.existsSync(compiledEntry) ? compiledEntry : sourceEntry;
|
|
21
|
-
const payloadEnvironment = payload.environment ?? {};
|
|
22
|
-
const env = { ...process.env, ...payloadEnvironment };
|
|
23
|
-
for (const key of Object.keys(env)) {
|
|
24
|
-
if (key.startsWith('AGENT_INFRA_CONTROL_')
|
|
25
|
-
|| key === 'AGENT_INFRA_TASK_ID'
|
|
26
|
-
|| key === 'AGENT_INFRA_RUNTIME_DIR'
|
|
27
|
-
|| key === 'AGENT_INFRA_EXECUTOR_MANIFEST'
|
|
28
|
-
|| key === 'AGENT_INFRA_HOST_CONTROL_SOCKET') {
|
|
29
|
-
delete env[key];
|
|
30
|
-
}
|
|
31
|
-
}
|
|
32
|
-
delete env.NODE_OPTIONS;
|
|
33
|
-
delete env.NODE_PATH;
|
|
34
|
-
delete env.AGENT_INFRA_HOST_CONTROL_WORKER;
|
|
35
|
-
delete env.AGENT_INFRA_HOST_CONTROL_WORKER_TOKEN;
|
|
36
|
-
env.AGENT_INFRA_HOST_CONTROL_WORKER = '1';
|
|
37
|
-
env.AGENT_INFRA_HOST_CONTROL_WORKER_TOKEN = readHostControlWorkerToken(
|
|
38
|
-
process.env.AGENT_INFRA_TEST_HOST_CONTROL_ENDPOINT ?? resolveHostControlEndpoint()
|
|
39
|
-
);
|
|
40
|
-
|
|
41
|
-
return await new Promise<HostCommandResult>((resolve, reject) => {
|
|
42
|
-
const runtimeArgs = entry.endsWith('.ts') ? ['--experimental-strip-types', '--no-warnings', entry] : [entry];
|
|
43
|
-
const child = spawn(process.execPath, [...runtimeArgs, request.operation, ...payload.args], {
|
|
44
|
-
cwd: payload.workingDirectory,
|
|
45
|
-
env,
|
|
46
|
-
stdio: ['ignore', 'pipe', 'pipe']
|
|
47
|
-
});
|
|
48
|
-
let stdout = '';
|
|
49
|
-
let stderr = '';
|
|
50
|
-
child.stdout.setEncoding('utf8');
|
|
51
|
-
child.stderr.setEncoding('utf8');
|
|
52
|
-
child.stdout.on('data', (chunk: string) => { stdout += chunk; });
|
|
53
|
-
child.stderr.on('data', (chunk: string) => { stderr += chunk; });
|
|
54
|
-
child.once('error', reject);
|
|
55
|
-
child.once('close', (code, signal) => {
|
|
56
|
-
if (signal) reject(new Error(`HOST_CONTROL_WORKER_TERMINATED: ${signal}`));
|
|
57
|
-
else resolve({ stdout, stderr, exitCode: code ?? 1 });
|
|
58
|
-
});
|
|
59
|
-
});
|
|
60
|
-
}
|
package/lib/host-control/path.ts
DELETED
|
@@ -1,167 +0,0 @@
|
|
|
1
|
-
import fs from 'node:fs';
|
|
2
|
-
import { randomBytes } from 'node:crypto';
|
|
3
|
-
import os from 'node:os';
|
|
4
|
-
import path from 'node:path';
|
|
5
|
-
|
|
6
|
-
export const HOST_CONTROL_DIRECTORY_MODE = 0o700;
|
|
7
|
-
export const HOST_CONTROL_SOCKET_MODE = 0o600;
|
|
8
|
-
export const HOST_CONTROL_WORKER_TOKEN_MODE = 0o600;
|
|
9
|
-
|
|
10
|
-
export type HostControlPlatform = 'linux' | 'darwin' | 'unsupported';
|
|
11
|
-
|
|
12
|
-
export type HostControlEndpointOptions = Readonly<{
|
|
13
|
-
platform?: HostControlPlatform;
|
|
14
|
-
uid?: number;
|
|
15
|
-
username?: string;
|
|
16
|
-
}>;
|
|
17
|
-
|
|
18
|
-
function currentPlatform(): HostControlPlatform {
|
|
19
|
-
if (process.platform === 'linux' || process.platform === 'darwin') return process.platform;
|
|
20
|
-
return 'unsupported';
|
|
21
|
-
}
|
|
22
|
-
|
|
23
|
-
function validUid(uid: number | undefined): number {
|
|
24
|
-
const resolved = uid ?? process.getuid?.();
|
|
25
|
-
if (!Number.isInteger(resolved) || resolved === undefined || resolved < 0) {
|
|
26
|
-
throw new Error('HOST_CONTROL_ENDPOINT_INVALID: uid is unavailable');
|
|
27
|
-
}
|
|
28
|
-
return resolved;
|
|
29
|
-
}
|
|
30
|
-
|
|
31
|
-
function validUsername(username: string | undefined): string {
|
|
32
|
-
const resolved = username ?? os.userInfo({ encoding: 'utf8' }).username;
|
|
33
|
-
if (!/^[A-Za-z0-9._-]+$/u.test(resolved)) {
|
|
34
|
-
throw new Error('HOST_CONTROL_ENDPOINT_INVALID: username is invalid');
|
|
35
|
-
}
|
|
36
|
-
return resolved;
|
|
37
|
-
}
|
|
38
|
-
|
|
39
|
-
export function resolveHostControlEndpoint(options: HostControlEndpointOptions = {}): string {
|
|
40
|
-
const platform = options.platform ?? currentPlatform();
|
|
41
|
-
if (platform === 'linux') return `/run/user/${validUid(options.uid)}/agent-infra/host-control.sock`;
|
|
42
|
-
if (platform === 'darwin') return path.posix.join(
|
|
43
|
-
'/Users',
|
|
44
|
-
validUsername(options.username),
|
|
45
|
-
'Library',
|
|
46
|
-
'Application Support',
|
|
47
|
-
'agent-infra',
|
|
48
|
-
'run',
|
|
49
|
-
'host-control.sock'
|
|
50
|
-
);
|
|
51
|
-
throw new Error('HOST_CONTROL_UNSUPPORTED_PLATFORM');
|
|
52
|
-
}
|
|
53
|
-
|
|
54
|
-
export function hostControlWorkerTokenPath(endpoint: string): string {
|
|
55
|
-
return `${endpoint}.worker-token`;
|
|
56
|
-
}
|
|
57
|
-
|
|
58
|
-
function assertWorkerTokenFile(tokenPath: string, uid = process.getuid?.()): string {
|
|
59
|
-
const stat = fs.lstatSync(tokenPath);
|
|
60
|
-
if (stat.isSymbolicLink() || !stat.isFile()) throw new Error('HOST_CONTROL_WORKER_TOKEN_INVALID');
|
|
61
|
-
if ((stat.mode & 0o777) !== HOST_CONTROL_WORKER_TOKEN_MODE) throw new Error('HOST_CONTROL_WORKER_TOKEN_MODE_INVALID');
|
|
62
|
-
if (uid !== undefined && stat.uid !== uid) throw new Error('HOST_CONTROL_WORKER_TOKEN_OWNER_INVALID');
|
|
63
|
-
const token = fs.readFileSync(tokenPath, 'utf8').trim();
|
|
64
|
-
if (!/^[a-f0-9]{64}$/u.test(token)) throw new Error('HOST_CONTROL_WORKER_TOKEN_INVALID');
|
|
65
|
-
return token;
|
|
66
|
-
}
|
|
67
|
-
|
|
68
|
-
export function ensureHostControlWorkerToken(endpoint: string, uid = process.getuid?.()): string {
|
|
69
|
-
const tokenPath = hostControlWorkerTokenPath(endpoint);
|
|
70
|
-
try {
|
|
71
|
-
return assertWorkerTokenFile(tokenPath, uid);
|
|
72
|
-
} catch (error) {
|
|
73
|
-
if ((error as NodeJS.ErrnoException).code !== 'ENOENT') throw error;
|
|
74
|
-
}
|
|
75
|
-
const token = randomBytes(32).toString('hex');
|
|
76
|
-
try {
|
|
77
|
-
const fd = fs.openSync(tokenPath, 'wx', HOST_CONTROL_WORKER_TOKEN_MODE);
|
|
78
|
-
try {
|
|
79
|
-
fs.writeFileSync(fd, `${token}\n`, 'utf8');
|
|
80
|
-
fs.fchmodSync(fd, HOST_CONTROL_WORKER_TOKEN_MODE);
|
|
81
|
-
if (uid !== undefined && process.getuid?.() === uid) fs.fchownSync(fd, uid, process.getgid?.() ?? -1);
|
|
82
|
-
} finally {
|
|
83
|
-
fs.closeSync(fd);
|
|
84
|
-
}
|
|
85
|
-
} catch (error) {
|
|
86
|
-
if ((error as NodeJS.ErrnoException).code !== 'EEXIST') throw error;
|
|
87
|
-
}
|
|
88
|
-
return assertWorkerTokenFile(tokenPath, uid);
|
|
89
|
-
}
|
|
90
|
-
|
|
91
|
-
export function readHostControlWorkerToken(endpoint: string, uid = process.getuid?.()): string {
|
|
92
|
-
return assertWorkerTokenFile(hostControlWorkerTokenPath(endpoint), uid);
|
|
93
|
-
}
|
|
94
|
-
|
|
95
|
-
export function removeHostControlWorkerToken(endpoint: string): void {
|
|
96
|
-
try {
|
|
97
|
-
fs.unlinkSync(hostControlWorkerTokenPath(endpoint));
|
|
98
|
-
} catch (error) {
|
|
99
|
-
if ((error as NodeJS.ErrnoException).code !== 'ENOENT') throw error;
|
|
100
|
-
}
|
|
101
|
-
}
|
|
102
|
-
|
|
103
|
-
export type HostControlEndpointInspection = Readonly<{
|
|
104
|
-
ok: boolean;
|
|
105
|
-
code: string | null;
|
|
106
|
-
ownerUid?: number;
|
|
107
|
-
mode?: number;
|
|
108
|
-
parentMode?: number;
|
|
109
|
-
}>;
|
|
110
|
-
|
|
111
|
-
export function inspectHostControlEndpoint(
|
|
112
|
-
endpoint: string,
|
|
113
|
-
expected: Readonly<{ uid?: number }> = {}
|
|
114
|
-
): HostControlEndpointInspection {
|
|
115
|
-
let endpointStat: fs.Stats;
|
|
116
|
-
let parentStat: fs.Stats;
|
|
117
|
-
try {
|
|
118
|
-
endpointStat = fs.lstatSync(endpoint);
|
|
119
|
-
parentStat = fs.lstatSync(path.dirname(endpoint));
|
|
120
|
-
} catch {
|
|
121
|
-
return { ok: false, code: 'HOST_CONTROL_ENDPOINT_MISSING' };
|
|
122
|
-
}
|
|
123
|
-
if (endpointStat.isSymbolicLink()) return { ok: false, code: 'HOST_CONTROL_ENDPOINT_SYMLINK' };
|
|
124
|
-
if (!endpointStat.isSocket()) return { ok: false, code: 'HOST_CONTROL_ENDPOINT_NOT_SOCKET' };
|
|
125
|
-
if (!parentStat.isDirectory() || parentStat.isSymbolicLink()) {
|
|
126
|
-
return { ok: false, code: 'HOST_CONTROL_PARENT_INVALID' };
|
|
127
|
-
}
|
|
128
|
-
let ancestor = path.dirname(endpoint);
|
|
129
|
-
while (true) {
|
|
130
|
-
let stat: fs.Stats;
|
|
131
|
-
try { stat = fs.lstatSync(ancestor); } catch { return { ok: false, code: 'HOST_CONTROL_PARENT_INVALID' }; }
|
|
132
|
-
if (!stat.isDirectory() || stat.isSymbolicLink()) return { ok: false, code: 'HOST_CONTROL_PARENT_INVALID' };
|
|
133
|
-
const next = path.dirname(ancestor);
|
|
134
|
-
if (next === ancestor) break;
|
|
135
|
-
ancestor = next;
|
|
136
|
-
}
|
|
137
|
-
const mode = endpointStat.mode & 0o777;
|
|
138
|
-
const parentMode = parentStat.mode & 0o777;
|
|
139
|
-
if (mode !== HOST_CONTROL_SOCKET_MODE) return { ok: false, code: 'HOST_CONTROL_ENDPOINT_MODE_INVALID', ownerUid: endpointStat.uid, mode, parentMode };
|
|
140
|
-
if (parentMode !== HOST_CONTROL_DIRECTORY_MODE) return { ok: false, code: 'HOST_CONTROL_PARENT_MODE_INVALID', ownerUid: endpointStat.uid, mode, parentMode };
|
|
141
|
-
if (expected.uid !== undefined && endpointStat.uid !== expected.uid) {
|
|
142
|
-
return { ok: false, code: 'HOST_CONTROL_ENDPOINT_OWNER_INVALID', ownerUid: endpointStat.uid, mode, parentMode };
|
|
143
|
-
}
|
|
144
|
-
if (expected.uid !== undefined && parentStat.uid !== expected.uid) {
|
|
145
|
-
return { ok: false, code: 'HOST_CONTROL_PARENT_OWNER_INVALID', ownerUid: parentStat.uid, mode, parentMode };
|
|
146
|
-
}
|
|
147
|
-
return { ok: true, code: null, ownerUid: endpointStat.uid, mode, parentMode };
|
|
148
|
-
}
|
|
149
|
-
|
|
150
|
-
export function prepareHostControlDirectory(endpoint: string, uid = process.getuid?.()): string {
|
|
151
|
-
const directory = path.dirname(endpoint);
|
|
152
|
-
const segments = path.resolve(directory).split(path.sep);
|
|
153
|
-
let current = path.isAbsolute(directory) ? path.sep : '';
|
|
154
|
-
for (const segment of segments.filter(Boolean)) {
|
|
155
|
-
current = path.join(current, segment);
|
|
156
|
-
try {
|
|
157
|
-
const stat = fs.lstatSync(current);
|
|
158
|
-
if (!stat.isDirectory() || stat.isSymbolicLink()) throw new Error('HOST_CONTROL_PARENT_INVALID');
|
|
159
|
-
} catch (error) {
|
|
160
|
-
if ((error as NodeJS.ErrnoException).code !== 'ENOENT') throw error;
|
|
161
|
-
}
|
|
162
|
-
}
|
|
163
|
-
fs.mkdirSync(directory, { recursive: true, mode: HOST_CONTROL_DIRECTORY_MODE });
|
|
164
|
-
fs.chmodSync(directory, HOST_CONTROL_DIRECTORY_MODE);
|
|
165
|
-
if (uid !== undefined && process.getuid?.() === uid) fs.chownSync(directory, uid, process.getgid?.() ?? -1);
|
|
166
|
-
return directory;
|
|
167
|
-
}
|
|
@@ -1,180 +0,0 @@
|
|
|
1
|
-
import fs from 'node:fs';
|
|
2
|
-
import net from 'node:net';
|
|
3
|
-
import { createHash } from 'node:crypto';
|
|
4
|
-
|
|
5
|
-
import {
|
|
6
|
-
HOST_CONTROL_DIRECTORY_MODE,
|
|
7
|
-
HOST_CONTROL_SOCKET_MODE,
|
|
8
|
-
ensureHostControlWorkerToken,
|
|
9
|
-
inspectHostControlEndpoint,
|
|
10
|
-
prepareHostControlDirectory,
|
|
11
|
-
removeHostControlWorkerToken,
|
|
12
|
-
resolveHostControlEndpoint
|
|
13
|
-
} from './path.ts';
|
|
14
|
-
import {
|
|
15
|
-
HOST_CONTROL_MAX_REQUEST_BYTES,
|
|
16
|
-
validateHostControlRequest,
|
|
17
|
-
type HostControlRequest,
|
|
18
|
-
type HostControlResponse
|
|
19
|
-
} from './client.ts';
|
|
20
|
-
import { acquireFileLock } from '../fs/file-lock.ts';
|
|
21
|
-
import { appendHostControlAudit } from './audit.ts';
|
|
22
|
-
|
|
23
|
-
export type HostControlDispatch = (request: HostControlRequest) => Promise<{ stdout: string; stderr: string; exitCode: number }>;
|
|
24
|
-
|
|
25
|
-
export type HostControlAudit = Readonly<{
|
|
26
|
-
requestId: string;
|
|
27
|
-
taskId: string | null;
|
|
28
|
-
operation: string;
|
|
29
|
-
scope: string;
|
|
30
|
-
phase: 'accepted' | 'completed' | 'rejected';
|
|
31
|
-
outcome: string;
|
|
32
|
-
authorityKind: 'host-broker';
|
|
33
|
-
transportKind: 'host-fixed-transport';
|
|
34
|
-
identityDigest: string;
|
|
35
|
-
}>;
|
|
36
|
-
|
|
37
|
-
export type HostControlServerOptions = Readonly<{
|
|
38
|
-
endpoint?: string;
|
|
39
|
-
dispatch: HostControlDispatch;
|
|
40
|
-
audit?: (entry: HostControlAudit) => void;
|
|
41
|
-
}>;
|
|
42
|
-
|
|
43
|
-
function auditFor(request: HostControlRequest, phase: HostControlAudit['phase'], outcome: string): HostControlAudit {
|
|
44
|
-
return {
|
|
45
|
-
requestId: request.id,
|
|
46
|
-
taskId: request.taskId,
|
|
47
|
-
operation: request.operation,
|
|
48
|
-
scope: request.scope,
|
|
49
|
-
phase,
|
|
50
|
-
outcome,
|
|
51
|
-
authorityKind: 'host-broker',
|
|
52
|
-
transportKind: 'host-fixed-transport',
|
|
53
|
-
identityDigest: createHash('sha256').update(`${request.taskId ?? '-'}\0${request.generation ?? '-'}`, 'utf8').digest('hex')
|
|
54
|
-
};
|
|
55
|
-
}
|
|
56
|
-
|
|
57
|
-
function failedResponse(id: string, status: 'rejected' | 'unknown', code: string, message: string): HostControlResponse {
|
|
58
|
-
const error = { code, message };
|
|
59
|
-
const stdout = `${JSON.stringify({ status: 'failed', changed: status === 'rejected' ? false : null, error })}\n`;
|
|
60
|
-
return { version: 1, id, status, exitCode: 1, stdout, stderr: '', error };
|
|
61
|
-
}
|
|
62
|
-
|
|
63
|
-
async function handleConnection(socket: net.Socket, options: HostControlServerOptions): Promise<void> {
|
|
64
|
-
await new Promise<void>((resolve) => {
|
|
65
|
-
let input = '';
|
|
66
|
-
let completed = false;
|
|
67
|
-
let processing = false;
|
|
68
|
-
socket.setEncoding('utf8');
|
|
69
|
-
// A disconnected caller does not cancel an already dispatched domain operation.
|
|
70
|
-
socket.on('error', () => socket.destroy());
|
|
71
|
-
socket.on('close', () => { if (!processing) resolve(); });
|
|
72
|
-
socket.on('end', () => { if (!processing) socket.end(); });
|
|
73
|
-
const finish = (value: HostControlResponse): void => {
|
|
74
|
-
if (completed || socket.destroyed) return;
|
|
75
|
-
completed = true;
|
|
76
|
-
socket.end(`${JSON.stringify(value)}\n`);
|
|
77
|
-
};
|
|
78
|
-
socket.on('data', async (chunk: string) => {
|
|
79
|
-
if (completed || processing) return;
|
|
80
|
-
input += chunk;
|
|
81
|
-
if (Buffer.byteLength(input, 'utf8') > HOST_CONTROL_MAX_REQUEST_BYTES) {
|
|
82
|
-
finish(failedResponse('invalid', 'rejected', 'HOST_CONTROL_REQUEST_TOO_LARGE', 'request exceeds the control limit'));
|
|
83
|
-
return;
|
|
84
|
-
}
|
|
85
|
-
const newline = input.indexOf('\n');
|
|
86
|
-
if (newline < 0) return;
|
|
87
|
-
const raw = input.slice(0, newline);
|
|
88
|
-
let request: HostControlRequest;
|
|
89
|
-
try { request = validateHostControlRequest(JSON.parse(raw)); }
|
|
90
|
-
catch (error) {
|
|
91
|
-
finish(failedResponse('invalid', 'rejected', 'HOST_CONTROL_REQUEST_INVALID', error instanceof Error ? error.message : String(error)));
|
|
92
|
-
return;
|
|
93
|
-
}
|
|
94
|
-
processing = true;
|
|
95
|
-
let started = false;
|
|
96
|
-
try {
|
|
97
|
-
options.audit?.(auditFor(request, 'accepted', 'in-progress'));
|
|
98
|
-
started = true;
|
|
99
|
-
const result = await options.dispatch(request);
|
|
100
|
-
options.audit?.(auditFor(request, 'completed', result.exitCode === 0 ? 'success' : 'failure'));
|
|
101
|
-
finish({ version: 1, id: request.id, status: 'completed', ...result, error: null });
|
|
102
|
-
} catch (error) {
|
|
103
|
-
const message = error instanceof Error ? error.message : String(error);
|
|
104
|
-
const code = /^([A-Z][A-Z0-9_]+)/u.exec(message)?.[1] ?? 'HOST_CONTROL_DISPATCH_FAILED';
|
|
105
|
-
const status = started ? 'unknown' : 'rejected';
|
|
106
|
-
try { options.audit?.(auditFor(request, started ? 'completed' : 'rejected', status)); }
|
|
107
|
-
catch { /* Return the uncertainty even when the audit sink is unavailable. */ }
|
|
108
|
-
finish(failedResponse(request.id, status, code, message));
|
|
109
|
-
} finally {
|
|
110
|
-
resolve();
|
|
111
|
-
}
|
|
112
|
-
});
|
|
113
|
-
});
|
|
114
|
-
}
|
|
115
|
-
|
|
116
|
-
export type RunningHostControlServer = Readonly<{
|
|
117
|
-
endpoint: string;
|
|
118
|
-
server: net.Server;
|
|
119
|
-
close: () => Promise<void>;
|
|
120
|
-
}>;
|
|
121
|
-
|
|
122
|
-
export async function startHostControlServer(options: HostControlServerOptions): Promise<RunningHostControlServer> {
|
|
123
|
-
const endpoint = options.endpoint ?? resolveHostControlEndpoint();
|
|
124
|
-
const audit = options.audit ?? ((entry: HostControlAudit) => appendHostControlAudit(endpoint, entry));
|
|
125
|
-
prepareHostControlDirectory(endpoint);
|
|
126
|
-
const lock = acquireFileLock(`${endpoint}.lock`);
|
|
127
|
-
let server: net.Server | undefined;
|
|
128
|
-
try {
|
|
129
|
-
ensureHostControlWorkerToken(endpoint);
|
|
130
|
-
try {
|
|
131
|
-
const existing = fs.lstatSync(endpoint);
|
|
132
|
-
if (existing.isSymbolicLink() || !existing.isSocket()) throw new Error('HOST_CONTROL_ENDPOINT_INVALID');
|
|
133
|
-
fs.unlinkSync(endpoint);
|
|
134
|
-
} catch (error) {
|
|
135
|
-
if ((error as NodeJS.ErrnoException).code !== 'ENOENT') throw error;
|
|
136
|
-
}
|
|
137
|
-
const connections = new Set<Promise<void>>();
|
|
138
|
-
server = net.createServer({ allowHalfOpen: true }, (socket) => {
|
|
139
|
-
const connection = handleConnection(socket, { ...options, audit });
|
|
140
|
-
connections.add(connection);
|
|
141
|
-
void connection.finally(() => connections.delete(connection));
|
|
142
|
-
});
|
|
143
|
-
const listeningServer = server;
|
|
144
|
-
await new Promise<void>((resolve, reject) => {
|
|
145
|
-
listeningServer.once('error', reject);
|
|
146
|
-
listeningServer.listen(endpoint, () => { listeningServer.off('error', reject); resolve(); });
|
|
147
|
-
});
|
|
148
|
-
fs.chmodSync(endpoint, HOST_CONTROL_SOCKET_MODE);
|
|
149
|
-
const inspection = inspectHostControlEndpoint(endpoint, { uid: process.getuid?.() });
|
|
150
|
-
if (!inspection.ok) {
|
|
151
|
-
await new Promise<void>((resolve) => listeningServer.close(() => resolve()));
|
|
152
|
-
removeHostControlWorkerToken(endpoint);
|
|
153
|
-
throw new Error(inspection.code ?? 'HOST_CONTROL_ENDPOINT_INVALID');
|
|
154
|
-
}
|
|
155
|
-
let closing: Promise<void> | undefined;
|
|
156
|
-
return {
|
|
157
|
-
endpoint,
|
|
158
|
-
server,
|
|
159
|
-
close: () => closing ??= (async () => {
|
|
160
|
-
try {
|
|
161
|
-
await new Promise<void>((resolve) => listeningServer.close(() => resolve()));
|
|
162
|
-
await Promise.all(connections);
|
|
163
|
-
await fs.promises.rm(endpoint, { force: true });
|
|
164
|
-
removeHostControlWorkerToken(endpoint);
|
|
165
|
-
} finally { lock.release(); }
|
|
166
|
-
})()
|
|
167
|
-
};
|
|
168
|
-
} catch (error) {
|
|
169
|
-
if (server?.listening) await new Promise<void>((resolve) => server!.close(() => resolve()));
|
|
170
|
-
lock.release();
|
|
171
|
-
throw error;
|
|
172
|
-
}
|
|
173
|
-
}
|
|
174
|
-
|
|
175
|
-
export async function serveHostControl(options: HostControlServerOptions, signal?: AbortSignal): Promise<void> {
|
|
176
|
-
const running = await startHostControlServer(options);
|
|
177
|
-
if (!signal) return;
|
|
178
|
-
if (signal.aborted) { await running.close(); return; }
|
|
179
|
-
await new Promise<void>((resolve) => signal.addEventListener('abort', () => { void running.close().then(resolve); }, { once: true }));
|
|
180
|
-
}
|