@winspan/claude-forge 8.51.1 → 8.54.3

package/src/web/ssrf-guard.ts

@@ -1,112 +0,0 @@
-/**
- * SSRF 防护 — Web API 对外 fetch 前的 URL 白名单校验
- *
- * 背景：`/api/ai/test` 与 `/api/ai/models` 允许请求方指定上游 `baseUrl`，
- * 若不加约束可被滥用为 SSRF 通道（访问 169.254.169.254 / 127.0.0.1 / 内网 API
- * 等），并通过 res.json 回显响应体。此模块统一做协议、主机、白名单校验。
- */
-
-/** 校验结果 */
-export interface SsrfValidationResult {
-  ok: boolean;
-  error?: string;
-}
-
-/** 默认允许访问的上游主机（已知 AI 服务商的官方域名） */
-export const DEFAULT_ALLOWED_HOSTS: ReadonlySet<string> = new Set([
-  'api.anthropic.com',
-  'api.openai.com',
-  'generativelanguage.googleapis.com',
-]);
-
-/** 统一的跨端点 fetch 超时（毫秒） */
-export const UPSTREAM_TIMEOUT_MS = 10_000;
-
-/** 判断字符串是否为 IPv4 字面量 */
-function isIPv4Literal(host: string): boolean {
-  return /^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$/.test(host);
-}
-
-/** 判断主机是否为 loopback / 保留名 */
-function isLocalhostLike(host: string): boolean {
-  const h = host.toLowerCase();
-  return (
-    h === 'localhost' ||
-    h.endsWith('.localhost') ||
-    h === '0.0.0.0' ||
-    h === '::' ||
-    h === '::1'
-  );
-}
-
-/**
- * 校验用户提供的 baseUrl 是否允许被 daemon 代为访问。
- *
- * 规则：
- * 1. 必须能解析为 URL；
- * 2. 协议必须是 http 或 https；
- * 3. 禁止 IPv4 / IPv6 字面量、localhost 等本机/内网地址；
- * 4. 主机必须在默认白名单或 extraHosts 中。
- *
- * @param baseUrl 待校验的 URL
- * @param extraHosts 额外允许的主机（通常来自 config.yaml 里已配置的 base_url）
- */
-export function validateBaseUrl(
-  baseUrl: unknown,
-  extraHosts: Iterable<string> = [],
-): SsrfValidationResult {
-  if (typeof baseUrl !== 'string' || baseUrl.length === 0) {
-    return { ok: false, error: 'baseUrl is required' };
-  }
-
-  let parsed: URL;
-  try {
-    parsed = new URL(baseUrl);
-  } catch {
-    return { ok: false, error: 'Invalid URL' };
-  }
-
-  if (parsed.protocol !== 'https:' && parsed.protocol !== 'http:') {
-    return { ok: false, error: 'Only http/https protocols allowed' };
-  }
-
-  const host = parsed.hostname.toLowerCase();
-
-  if (isIPv4Literal(host)) {
-    return { ok: false, error: 'Direct IP access blocked (SSRF protection)' };
-  }
-  // URL parser strips brackets from IPv6 hostname; detect by ':' or '::'
-  if (host.includes(':')) {
-    return { ok: false, error: 'Direct IPv6 access blocked (SSRF protection)' };
-  }
-  if (isLocalhostLike(host)) {
-    return { ok: false, error: 'localhost access blocked (SSRF protection)' };
-  }
-
-  const allowed = new Set<string>(DEFAULT_ALLOWED_HOSTS);
-  for (const extra of extraHosts) {
-    if (extra) allowed.add(extra.toLowerCase());
-  }
-
-  if (!allowed.has(host)) {
-    return {
-      ok: false,
-      error: `Host ${host} not in allowlist. Allowed: ${[...allowed].sort().join(', ')}`,
-    };
-  }
-
-  return { ok: true };
-}
-
-/**
- * 从已配置的 base_url 中提取主机名，作为额外允许项。
- * 无效/为空返回 null，调用方可忽略。
- */
-export function extractHostFromBaseUrl(baseUrl: string | undefined | null): string | null {
-  if (!baseUrl) return null;
-  try {
-    return new URL(baseUrl).hostname.toLowerCase();
-  } catch {
-    return null;
-  }
-}