fix: tighten tls fingerprints and approval events
This commit is contained in:
@@ -1,5 +1,6 @@
|
|||||||
import { randomUUID } from "node:crypto";
|
import { randomUUID } from "node:crypto";
|
||||||
import { WebSocket, type ClientOptions, type CertMeta } from "ws";
|
import { WebSocket, type ClientOptions, type CertMeta } from "ws";
|
||||||
|
import { normalizeFingerprint } from "../infra/tls/fingerprint.js";
|
||||||
import { rawDataToString } from "../infra/ws.js";
|
import { rawDataToString } from "../infra/ws.js";
|
||||||
import { logDebug, logError } from "../logger.js";
|
import { logDebug, logError } from "../logger.js";
|
||||||
import type { DeviceIdentity } from "../infra/device-identity.js";
|
import type { DeviceIdentity } from "../infra/device-identity.js";
|
||||||
@@ -99,6 +100,10 @@ export class GatewayClient {
|
|||||||
start() {
|
start() {
|
||||||
if (this.closed) return;
|
if (this.closed) return;
|
||||||
const url = this.opts.url ?? "ws://127.0.0.1:18789";
|
const url = this.opts.url ?? "ws://127.0.0.1:18789";
|
||||||
|
if (this.opts.tlsFingerprint && !url.startsWith("wss://")) {
|
||||||
|
this.opts.onConnectError?.(new Error("gateway tls fingerprint requires wss:// gateway url"));
|
||||||
|
return;
|
||||||
|
}
|
||||||
// Allow node screen snapshots and other large responses.
|
// Allow node screen snapshots and other large responses.
|
||||||
const wsOptions: ClientOptions = {
|
const wsOptions: ClientOptions = {
|
||||||
maxPayload: 25 * 1024 * 1024,
|
maxPayload: 25 * 1024 * 1024,
|
||||||
@@ -399,7 +404,3 @@ export class GatewayClient {
|
|||||||
return p;
|
return p;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function normalizeFingerprint(input: string): string {
|
|
||||||
return input.replace(/[^a-fA-F0-9]/g, "").toLowerCase();
|
|
||||||
}
|
|
||||||
|
|||||||
57
src/gateway/server-broadcast.test.ts
Normal file
57
src/gateway/server-broadcast.test.ts
Normal file
@@ -0,0 +1,57 @@
|
|||||||
|
import { describe, expect, it, vi } from "vitest";
|
||||||
|
|
||||||
|
import { createGatewayBroadcaster } from "./server-broadcast.js";
|
||||||
|
import type { GatewayWsClient } from "./server/ws-types.js";
|
||||||
|
|
||||||
|
type TestSocket = {
|
||||||
|
bufferedAmount: number;
|
||||||
|
send: (payload: string) => void;
|
||||||
|
close: (code: number, reason: string) => void;
|
||||||
|
};
|
||||||
|
|
||||||
|
describe("gateway broadcaster", () => {
|
||||||
|
it("filters approval and pairing events by scope", () => {
|
||||||
|
const approvalsSocket: TestSocket = {
|
||||||
|
bufferedAmount: 0,
|
||||||
|
send: vi.fn(),
|
||||||
|
close: vi.fn(),
|
||||||
|
};
|
||||||
|
const pairingSocket: TestSocket = {
|
||||||
|
bufferedAmount: 0,
|
||||||
|
send: vi.fn(),
|
||||||
|
close: vi.fn(),
|
||||||
|
};
|
||||||
|
const readSocket: TestSocket = {
|
||||||
|
bufferedAmount: 0,
|
||||||
|
send: vi.fn(),
|
||||||
|
close: vi.fn(),
|
||||||
|
};
|
||||||
|
|
||||||
|
const clients = new Set<GatewayWsClient>([
|
||||||
|
{
|
||||||
|
socket: approvalsSocket as unknown as GatewayWsClient["socket"],
|
||||||
|
connect: { role: "operator", scopes: ["operator.approvals"] } as GatewayWsClient["connect"],
|
||||||
|
connId: "c-approvals",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
socket: pairingSocket as unknown as GatewayWsClient["socket"],
|
||||||
|
connect: { role: "operator", scopes: ["operator.pairing"] } as GatewayWsClient["connect"],
|
||||||
|
connId: "c-pairing",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
socket: readSocket as unknown as GatewayWsClient["socket"],
|
||||||
|
connect: { role: "operator", scopes: ["operator.read"] } as GatewayWsClient["connect"],
|
||||||
|
connId: "c-read",
|
||||||
|
},
|
||||||
|
]);
|
||||||
|
|
||||||
|
const { broadcast } = createGatewayBroadcaster({ clients });
|
||||||
|
|
||||||
|
broadcast("exec.approval.requested", { id: "1" });
|
||||||
|
broadcast("device.pair.requested", { requestId: "r1" });
|
||||||
|
|
||||||
|
expect(approvalsSocket.send).toHaveBeenCalledTimes(1);
|
||||||
|
expect(pairingSocket.send).toHaveBeenCalledTimes(1);
|
||||||
|
expect(readSocket.send).toHaveBeenCalledTimes(0);
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -2,6 +2,29 @@ import type { GatewayWsClient } from "./server/ws-types.js";
|
|||||||
import { MAX_BUFFERED_BYTES } from "./server-constants.js";
|
import { MAX_BUFFERED_BYTES } from "./server-constants.js";
|
||||||
import { logWs, summarizeAgentEventForWsLog } from "./ws-log.js";
|
import { logWs, summarizeAgentEventForWsLog } from "./ws-log.js";
|
||||||
|
|
||||||
|
const ADMIN_SCOPE = "operator.admin";
|
||||||
|
const APPROVALS_SCOPE = "operator.approvals";
|
||||||
|
const PAIRING_SCOPE = "operator.pairing";
|
||||||
|
|
||||||
|
const EVENT_SCOPE_GUARDS: Record<string, string[]> = {
|
||||||
|
"exec.approval.requested": [APPROVALS_SCOPE],
|
||||||
|
"exec.approval.resolved": [APPROVALS_SCOPE],
|
||||||
|
"device.pair.requested": [PAIRING_SCOPE],
|
||||||
|
"device.pair.resolved": [PAIRING_SCOPE],
|
||||||
|
"node.pair.requested": [PAIRING_SCOPE],
|
||||||
|
"node.pair.resolved": [PAIRING_SCOPE],
|
||||||
|
};
|
||||||
|
|
||||||
|
function hasEventScope(client: GatewayWsClient, event: string): boolean {
|
||||||
|
const required = EVENT_SCOPE_GUARDS[event];
|
||||||
|
if (!required) return true;
|
||||||
|
const role = client.connect.role ?? "operator";
|
||||||
|
if (role !== "operator") return false;
|
||||||
|
const scopes = Array.isArray(client.connect.scopes) ? client.connect.scopes : [];
|
||||||
|
if (scopes.includes(ADMIN_SCOPE)) return true;
|
||||||
|
return required.some((scope) => scopes.includes(scope));
|
||||||
|
}
|
||||||
|
|
||||||
export function createGatewayBroadcaster(params: { clients: Set<GatewayWsClient> }) {
|
export function createGatewayBroadcaster(params: { clients: Set<GatewayWsClient> }) {
|
||||||
let seq = 0;
|
let seq = 0;
|
||||||
const broadcast = (
|
const broadcast = (
|
||||||
@@ -33,6 +56,7 @@ export function createGatewayBroadcaster(params: { clients: Set<GatewayWsClient>
|
|||||||
}
|
}
|
||||||
logWs("out", "event", logMeta);
|
logWs("out", "event", logMeta);
|
||||||
for (const c of params.clients) {
|
for (const c of params.clients) {
|
||||||
|
if (!hasEventScope(c, event)) continue;
|
||||||
const slow = c.socket.bufferedAmount > MAX_BUFFERED_BYTES;
|
const slow = c.socket.bufferedAmount > MAX_BUFFERED_BYTES;
|
||||||
if (slow && opts?.dropIfSlow) continue;
|
if (slow && opts?.dropIfSlow) continue;
|
||||||
if (slow) {
|
if (slow) {
|
||||||
|
|||||||
11
src/infra/tls/fingerprint.test.ts
Normal file
11
src/infra/tls/fingerprint.test.ts
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
import { describe, expect, it } from "vitest";
|
||||||
|
|
||||||
|
import { normalizeFingerprint } from "./fingerprint.js";
|
||||||
|
|
||||||
|
describe("normalizeFingerprint", () => {
|
||||||
|
it("strips sha256 prefixes and separators", () => {
|
||||||
|
expect(normalizeFingerprint("sha256:AA:BB:cc")).toBe("aabbcc");
|
||||||
|
expect(normalizeFingerprint("SHA-256 11-22-33")).toBe("112233");
|
||||||
|
expect(normalizeFingerprint("aa:bb:cc")).toBe("aabbcc");
|
||||||
|
});
|
||||||
|
});
|
||||||
5
src/infra/tls/fingerprint.ts
Normal file
5
src/infra/tls/fingerprint.ts
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
export function normalizeFingerprint(input: string): string {
|
||||||
|
const trimmed = input.trim();
|
||||||
|
const withoutPrefix = trimmed.replace(/^sha-?256\s*:?\s*/i, "");
|
||||||
|
return withoutPrefix.replace(/[^a-fA-F0-9]/g, "").toLowerCase();
|
||||||
|
}
|
||||||
@@ -7,6 +7,7 @@ import { promisify } from "node:util";
|
|||||||
|
|
||||||
import type { GatewayTlsConfig } from "../../config/types.gateway.js";
|
import type { GatewayTlsConfig } from "../../config/types.gateway.js";
|
||||||
import { CONFIG_DIR, ensureDir, resolveUserPath, shortenHomeInString } from "../../utils.js";
|
import { CONFIG_DIR, ensureDir, resolveUserPath, shortenHomeInString } from "../../utils.js";
|
||||||
|
import { normalizeFingerprint } from "./fingerprint.js";
|
||||||
|
|
||||||
const execFileAsync = promisify(execFile);
|
const execFileAsync = promisify(execFile);
|
||||||
|
|
||||||
@@ -21,10 +22,6 @@ export type GatewayTlsRuntime = {
|
|||||||
error?: string;
|
error?: string;
|
||||||
};
|
};
|
||||||
|
|
||||||
function normalizeFingerprint(input: string): string {
|
|
||||||
return input.replace(/[^a-fA-F0-9]/g, "").toLowerCase();
|
|
||||||
}
|
|
||||||
|
|
||||||
async function fileExists(filePath: string): Promise<boolean> {
|
async function fileExists(filePath: string): Promise<boolean> {
|
||||||
try {
|
try {
|
||||||
await fs.access(filePath);
|
await fs.access(filePath);
|
||||||
|
|||||||
Reference in New Issue
Block a user