Files
2026-07-13 13:39:12 +08:00

40 lines
1.3 KiB
TypeScript

import dns from "node:dns/promises";
import { detectIpLiteralFamily, stripIpv6Brackets } from "./proxyFamily.ts";
export type FamilyLookupFn = (
hostname: string
) => Promise<Array<{ address: string; family: number }>>;
const defaultLookup: FamilyLookupFn = (hostname) => dns.lookup(hostname, { all: true });
/**
* Fail-closed guarantee for an IPv6-only (or IPv4-only) proxy given as a hostname:
* refuse early if the hostname has no record in the required family. No-op for IP
* literals (their family is intrinsic).
*/
export async function assertHostnameSupportsFamily(
host: string,
family: 4 | 6,
lookupFn: FamilyLookupFn = defaultLookup
): Promise<void> {
if (detectIpLiteralFamily(host) !== null) return;
let records: Array<{ address: string; family: number }>;
try {
records = await lookupFn(stripIpv6Brackets(host));
} catch (err) {
throw new Error(
`[ProxyFamily] DNS resolution failed for ${host}; refusing to egress (fail-closed): ${
err instanceof Error ? err.message : String(err)
}`
);
}
const hasFamily = records.some((r) => r.family === family);
if (!hasFamily) {
throw new Error(
`[ProxyFamily] Proxy host ${host} has no ${family === 6 ? "IPv6 (AAAA)" : "IPv4 (A)"} record; refusing ${
family === 6 ? "IPv6" : "IPv4"
}-only egress (fail-closed)`
);
}
}