Claude
7d9d478baa
在 arcrun-mcp worker 實作 MCP Authorization 規範（OAuth 2.1 + PKCE S256）， 讓 claude.ai 遠端 connector 安全登入；並修掉「Bearer 明碼 namespace 直接放行」漏洞。 安全模型 - /authorize 同意頁以 owner secret（CF Secrets MCP_OWNER_SECRET）把關，只有 owner 知道 → 只知 URL 的人走不完 OAuth、拿不到 token。 - access_token 是 /mcp 唯一接受的 bearer（預設）；明碼 namespace 舊路徑移除（步驟 5 直接 401）。 實作 endpoint（掛 worker 根路徑） - RFC 9728 /.well-known/oauth-protected-resource（+/mcp 變體）＋ 401 帶 WWW-Authenticate: Bearer resource_metadata=... - RFC 8414 /.well-known/oauth-authorization-server（response_types=code, S256, none） - RFC 7591 /register（public client，無 secret，無狀態不落地） - GET/POST /authorize（PKCE S256 + owner-secret 閘 + redirect_uri 白名單） - POST /token（authorization_code + PKCE 驗證 → access_token 綁定 owner namespace） 儲存鐵律 - authorization code / access token → 短效 KV OAUTH_KV（key 用 SHA-256 hash、帶 TTL、code 一次性） - owner secret / static token → CF Secrets（非 KV、非明碼 var） - DCR client / refresh token → 不落地（無狀態 / 不實作，避免長效機密進 KV） 相容決策 - 本機 CLI/GUI/Claude Code → 用真祕密 MCP_STATIC_TOKEN（CF Secret）取代舊明碼 namespace - 官方 SaaS partner-key 路徑行為不變 - ALLOW_PLAINTEXT_NAMESPACE 逃生門預設關（僅遷移期） 驗證：tsc exit 0；vitest 42/42（oauth 22 + partner-auth 10 改測真實 middleware + 既有 10）； wrangler deploy --dry-run 打包過、OAUTH_KV binding 正確識別。設計文件 mcp/OAUTH.md。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_015d5jDbuqT5Htwv3Q88XXKk
62 lines
2.6 KiB
TypeScript
62 lines
2.6 KiB
TypeScript
// OAuth 加密工具：PKCE S256 驗證、SHA-256 hash、隨機不可猜 token、常數時間比對。
|
||
// 全走 Web Crypto（Workers / Node 18+ 全域 crypto.subtle），無外部依賴。
|
||
|
||
/** base64url 編碼（無 padding，RFC 7636 §A）。 */
|
||
function base64UrlEncode(bytes: Uint8Array): string {
|
||
let bin = "";
|
||
for (const b of bytes) bin += String.fromCharCode(b);
|
||
return btoa(bin).replace(/\+/g, "-").replace(/\//g, "_").replace(/=+$/, "");
|
||
}
|
||
|
||
/** 產生密碼學等級隨機 token（bytes 位元組 → base64url 字串，預設 32 bytes = 256-bit）。 */
|
||
export function randomToken(bytes = 32): string {
|
||
const buf = new Uint8Array(bytes);
|
||
crypto.getRandomValues(buf);
|
||
return base64UrlEncode(buf);
|
||
}
|
||
|
||
/** SHA-256 → base64url（PKCE code_challenge 用；RFC 7636 §4.2 BASE64URL(SHA256(verifier))）。 */
|
||
export async function sha256Base64Url(input: string): Promise<string> {
|
||
const digest = await crypto.subtle.digest("SHA-256", new TextEncoder().encode(input));
|
||
return base64UrlEncode(new Uint8Array(digest));
|
||
}
|
||
|
||
/** SHA-256 → hex（拿來當 KV key：不把可用的 raw token 直接當 key，KV list 也看不到明碼）。 */
|
||
export async function sha256Hex(input: string): Promise<string> {
|
||
const digest = await crypto.subtle.digest("SHA-256", new TextEncoder().encode(input));
|
||
return [...new Uint8Array(digest)].map((b) => b.toString(16).padStart(2, "0")).join("");
|
||
}
|
||
|
||
/**
|
||
* 常數時間字串比對（防 timing attack；比 owner secret / static token 用）。
|
||
* 長度不同直接 false，但仍走完固定迴圈避免長度洩漏。
|
||
*/
|
||
export function constantTimeEqual(a: string, b: string): boolean {
|
||
const ab = new TextEncoder().encode(a);
|
||
const bb = new TextEncoder().encode(b);
|
||
let diff = ab.length ^ bb.length;
|
||
const len = Math.max(ab.length, bb.length);
|
||
for (let i = 0; i < len; i++) {
|
||
diff |= (ab[i] ?? 0) ^ (bb[i] ?? 0);
|
||
}
|
||
return diff === 0;
|
||
}
|
||
|
||
/**
|
||
* 驗證 PKCE：BASE64URL(SHA256(code_verifier)) === code_challenge（僅支援 S256）。
|
||
* method 非 "S256"（含缺省的 "plain"）一律拒絕——OAuth 2.1 + 本 server 只宣告 S256。
|
||
*/
|
||
export async function verifyPkceS256(
|
||
codeVerifier: string,
|
||
codeChallenge: string,
|
||
method: string | undefined,
|
||
): Promise<boolean> {
|
||
if (method !== "S256") return false;
|
||
if (!codeVerifier || !codeChallenge) return false;
|
||
// RFC 7636 §4.1：verifier 43–128 字元、[A-Za-z0-9-._~]。
|
||
if (codeVerifier.length < 43 || codeVerifier.length > 128) return false;
|
||
if (!/^[A-Za-z0-9\-._~]+$/.test(codeVerifier)) return false;
|
||
const computed = await sha256Base64Url(codeVerifier);
|
||
return constantTimeEqual(computed, codeChallenge);
|
||
}
|