feat: bound every OAuth-path request with a deadline

Protected-resource-metadata discovery, authorization-server metadata
discovery, dynamic client registration, the token exchange and the
refresh all went out with no AbortSignal at all. Token revocation was
the single exception, and its withDeadline race is the pattern this
generalizes.

Adds core/auth/requestTimeout.ts: withOAuthRequestTimeout wraps a fetch
so every call through it is bounded, composing any caller signal with
AbortSignal.any and racing a timer so the deadline holds even where the
signal is dropped in flight (the browser's remote fetch re-issues the
call to /api/fetch and forwards no signal). The rejection is an
OAuthRequestTimeoutError naming the URL, so a stall says which endpoint
stalled rather than riding the SDK handshake timeout and reporting
"Request timed out".

Applied inside InspectorClient.buildEffectiveAuthFetch rather than to
this.fetchFn, because the transport is handed this.fetchFn directly and
a deadline on a Streamable HTTP / SSE response would sever the stream it
is supposed to hold open. Also applied to the CLI's storedAuthFetch,
which calls SDK discovery directly and so has nothing else bounding it.

Closes #2319

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01YFnM6RqB4nXqNysgUYmjmq
Signed-off-by: cliffhall <cliff@futurescale.com>
This commit is contained in:
cliffhall 2026-09-11 00:45:04 -04:00
commit 8dd7bc2920
2 changed files with 362 additions and 2 deletions

View file

@ -53,6 +53,7 @@ import {
getAuthorizationServerUrlCandidates,
} from "@inspector/core/auth/discovery.js";
import { withRfc8414OidcCompat } from "@inspector/core/auth/oidcDiscoveryCompat.js";
import { withOAuthRequestTimeout } from "@inspector/core/auth/requestTimeout.js";
import { writeStoreFile } from "@inspector/core/storage/store-io.js";
import {
refreshAuthorization,
@ -353,7 +354,14 @@ export async function refreshStoredAuthToken(
// else puts a proxy under it, and a server reachable only through
// `HTTPS_PROXY` would otherwise be probed directly. The same fetch is handed
// to the token request below, so neither leg bypasses the proxy (Copilot).
const storedAuthFetch = withRfc8414OidcCompat(createProxyFetch() ?? fetch);
// #2319: this path runs outside `InspectorClient`, so nothing else bounds it
// — the discovery and the token request below would otherwise hang forever
// against an authorization server that accepts the connection and never
// answers. Innermost, so the compat wrapper's own probe requests inherit the
// deadline too.
const storedAuthFetch = withRfc8414OidcCompat(
withOAuthRequestTimeout(createProxyFetch() ?? fetch),
);
const discover: typeof discoverAuthorizationServerMetadata =
deps.discover ??
((authorizationServerUrl, options) =>

View file

@ -0,0 +1,186 @@
import { describe, it, expect, vi, afterEach } from "vitest";
import {
DEFAULT_OAUTH_REQUEST_TIMEOUT_MS,
OAuthRequestTimeoutError,
withOAuthRequestTimeout,
} from "@inspector/core/auth/requestTimeout.js";
const URL_UNDER_TEST =
"https://as.example.com/.well-known/oauth-authorization-server";
/** A fetch that never settles — the wedged authorization server of #2319. */
const neverSettles: typeof fetch = () => new Promise<Response>(() => {});
afterEach(() => {
vi.useRealTimers();
vi.restoreAllMocks();
});
describe("withOAuthRequestTimeout", () => {
it("passes a prompt response straight through", async () => {
const ok = new Response("{}", { status: 200 });
const inner = vi.fn<typeof fetch>().mockResolvedValue(ok);
const wrapped = withOAuthRequestTimeout(inner, 1000);
await expect(wrapped(URL_UNDER_TEST)).resolves.toBe(ok);
expect(inner).toHaveBeenCalledTimes(1);
});
it("rejects a stalled request with an error naming the endpoint", async () => {
vi.useFakeTimers();
const wrapped = withOAuthRequestTimeout(neverSettles, 1000);
const pending = wrapped(URL_UNDER_TEST);
const assertion = expect(pending).rejects.toThrow(
`OAuth request to ${URL_UNDER_TEST} timed out after 1000ms`,
);
await vi.advanceTimersByTimeAsync(1000);
await assertion;
});
it("reports the timeout as OAuthRequestTimeoutError carrying url and budget", async () => {
vi.useFakeTimers();
const wrapped = withOAuthRequestTimeout(neverSettles, 1000);
const pending = wrapped(new URL(URL_UNDER_TEST));
const assertion = pending.catch((err: unknown) => err);
await vi.advanceTimersByTimeAsync(1000);
const err = await assertion;
expect(err).toBeInstanceOf(OAuthRequestTimeoutError);
const timeout = err as OAuthRequestTimeoutError;
expect(timeout.name).toBe("OAuthRequestTimeoutError");
expect(timeout.url).toBe(URL_UNDER_TEST);
expect(timeout.timeoutMs).toBe(1000);
});
it("takes the URL off a Request object too", async () => {
vi.useFakeTimers();
const wrapped = withOAuthRequestTimeout(neverSettles, 1000);
const pending = wrapped(new Request(URL_UNDER_TEST, { method: "POST" }));
const assertion = pending.catch((err: unknown) => err);
await vi.advanceTimersByTimeAsync(1000);
expect((await assertion) as OAuthRequestTimeoutError).toMatchObject({
url: URL_UNDER_TEST,
});
});
it("aborts the underlying fetch rather than leaving it running detached", async () => {
vi.useFakeTimers();
let seen: AbortSignal | undefined;
const inner: typeof fetch = (_input, init) => {
seen = init?.signal ?? undefined;
return new Promise<Response>((_resolve, reject) => {
init?.signal?.addEventListener("abort", () =>
reject(new Error("aborted by signal")),
);
});
};
const wrapped = withOAuthRequestTimeout(inner, 1000);
const assertion = wrapped(URL_UNDER_TEST).catch((err: unknown) => err);
await vi.advanceTimersByTimeAsync(1000);
expect(seen?.aborted).toBe(true);
// The underlying rejection can win the race against our own; either way the
// caller must be told it was a timeout, and which endpoint stalled.
expect((await assertion) as Error).toBeInstanceOf(OAuthRequestTimeoutError);
});
it("still enforces the deadline when the signal is dropped in flight", async () => {
vi.useFakeTimers();
// The browser's `createRemoteFetch` re-issues the call to `/api/fetch` and
// forwards no signal, so aborting locally cancels nothing. The race is what
// has to bound it.
const signalIgnoring: typeof fetch = () => new Promise<Response>(() => {});
const wrapped = withOAuthRequestTimeout(signalIgnoring, 1000);
const assertion = wrapped(URL_UNDER_TEST).catch((err: unknown) => err);
await vi.advanceTimersByTimeAsync(1000);
expect((await assertion) as Error).toBeInstanceOf(OAuthRequestTimeoutError);
});
it("preserves a caller-supplied signal, which still wins on its own", async () => {
const inner: typeof fetch = (_input, init) =>
new Promise<Response>((_resolve, reject) => {
init?.signal?.addEventListener("abort", () =>
reject(new Error("caller cancelled")),
);
});
const wrapped = withOAuthRequestTimeout(inner, 60_000);
const caller = new AbortController();
const pending = wrapped(URL_UNDER_TEST, { signal: caller.signal });
caller.abort();
await expect(pending).rejects.toThrow("caller cancelled");
});
it("propagates a non-timeout failure unchanged", async () => {
const boom = new TypeError("network error");
const inner = vi.fn<typeof fetch>().mockRejectedValue(boom);
const wrapped = withOAuthRequestTimeout(inner, 1000);
await expect(wrapped(URL_UNDER_TEST)).rejects.toBe(boom);
});
it("rounds a fractional budget, because setTimeout takes an integer", async () => {
vi.useFakeTimers();
// A budget derived from `performance.now()` arrives fractional; a fractional
// delay throws ERR_OUT_OF_RANGE before the request is ever made.
const wrapped = withOAuthRequestTimeout(neverSettles, 1000.4);
const assertion = wrapped(URL_UNDER_TEST).catch((err: unknown) => err);
await vi.advanceTimersByTimeAsync(1000);
expect((await assertion) as OAuthRequestTimeoutError).toMatchObject({
timeoutMs: 1000,
});
});
it("clamps a negative budget to zero rather than throwing", async () => {
vi.useFakeTimers();
const wrapped = withOAuthRequestTimeout(neverSettles, -1);
const assertion = wrapped(URL_UNDER_TEST).catch((err: unknown) => err);
await vi.advanceTimersByTimeAsync(0);
expect((await assertion) as OAuthRequestTimeoutError).toMatchObject({
timeoutMs: 0,
});
});
it("defaults to a generous budget a slow authorization server can meet", async () => {
vi.useFakeTimers();
const wrapped = withOAuthRequestTimeout(neverSettles);
const assertion = wrapped(URL_UNDER_TEST).catch((err: unknown) => err);
await vi.advanceTimersByTimeAsync(DEFAULT_OAUTH_REQUEST_TIMEOUT_MS - 1);
expect(await Promise.race([assertion, Promise.resolve("pending")])).toBe(
"pending",
);
await vi.advanceTimersByTimeAsync(1);
expect((await assertion) as OAuthRequestTimeoutError).toMatchObject({
timeoutMs: DEFAULT_OAUTH_REQUEST_TIMEOUT_MS,
});
});
it("clears its timer once the request settles", async () => {
vi.useFakeTimers();
const clearSpy = vi.spyOn(globalThis, "clearTimeout");
const wrapped = withOAuthRequestTimeout(
vi.fn<typeof fetch>().mockResolvedValue(new Response(null)),
1000,
);
await wrapped(URL_UNDER_TEST);
expect(clearSpy).toHaveBeenCalled();
// Nothing is left to fire, so no unhandled rejection can surface later.
await vi.advanceTimersByTimeAsync(5000);
});
});