216 lines
9.3 KiB
TypeScript
216 lines
9.3 KiB
TypeScript
import Fastify from "fastify";
|
|
import cookie from "@fastify/cookie";
|
|
import { afterEach, expect, test } from "vitest";
|
|
import { registerAuthRoutes } from "../src/auth/routes.js";
|
|
import type { LoadedAuthConfig, OidcStateRecord } from "../src/auth/types.js";
|
|
import type { AuthSessionStore } from "../src/auth/session-store.js";
|
|
import type { OidcProtocol } from "../src/auth/oidc-client.js";
|
|
|
|
const revision = "a".repeat(64);
|
|
const issuer = "https://issuer.example.test";
|
|
const state = "s".repeat(43);
|
|
const nonce = "n".repeat(43);
|
|
const verifier = "v".repeat(43);
|
|
const createdApps: Array<ReturnType<typeof Fastify>> = [];
|
|
|
|
function config(overrides: Partial<LoadedAuthConfig["value"]> = {}): LoadedAuthConfig {
|
|
return {
|
|
revision,
|
|
sourcePath: "/private/auth.yaml",
|
|
value: {
|
|
version: 1,
|
|
mode: "oidc",
|
|
publicUrl: "https://thothii.example.test",
|
|
session: {
|
|
regularTtlSeconds: 3600, regularIdleSeconds: 300,
|
|
rememberTtlSeconds: 3600, rememberIdleSeconds: 300, oidcTtlSeconds: 3600,
|
|
},
|
|
oidc: {
|
|
issuer, clientId: "thothii", clientSecretRef: "THT_OIDC_CLIENT_SECRET",
|
|
scopes: ["openid", "profile"], groupsClaim: "groups",
|
|
},
|
|
groupCatalog: { driver: "authentik", baseUrl: issuer, apiTokenRef: "THT_AUTHENTIK_API_TOKEN" },
|
|
authorization: { groupRoles: { Users: ["user"], Admins: ["admin"] } },
|
|
...overrides,
|
|
},
|
|
} as LoadedAuthConfig;
|
|
}
|
|
|
|
function stateRecord(extra: Partial<OidcStateRecord> = {}): OidcStateRecord {
|
|
return {
|
|
version: 1, nonce, codeVerifier: verifier, returnTo: "/",
|
|
authConfigRevision: revision, issuer,
|
|
createdAt: "2030-01-01T00:00:00.000Z", expiresAt: "2030-01-01T00:10:00.000Z",
|
|
...extra,
|
|
} as OidcStateRecord;
|
|
}
|
|
|
|
function fixture(options: {
|
|
loaded?: LoadedAuthConfig;
|
|
identity?: Awaited<ReturnType<OidcProtocol["callback"]>>;
|
|
callbackFailure?: boolean;
|
|
stateReturnTo?: string;
|
|
} = {}) {
|
|
let loaded = options.loaded ?? config();
|
|
let protocolAvailable = true;
|
|
let storedState: OidcStateRecord | undefined;
|
|
const stateInputs: Array<Record<string, unknown>> = [];
|
|
const creates: Array<Record<string, unknown>> = [];
|
|
const callbacks: URL[] = [];
|
|
const protocol: OidcProtocol = {
|
|
authorizationUrl: async ({ state: received, nonce: receivedNonce, codeVerifier }) => {
|
|
expect(received).toBe(state);
|
|
expect(receivedNonce).toHaveLength(43);
|
|
expect(codeVerifier).toHaveLength(43);
|
|
return new URL(`https://issuer.example.test/authorize?state=${received}`);
|
|
},
|
|
callback: async ({ currentUrl }) => {
|
|
callbacks.push(currentUrl);
|
|
if (options.callbackFailure) throw new Error("provider failure with access-token-must-not-leak");
|
|
return options.identity ?? {
|
|
issuer, subject: "user-123", displayName: "Ada", groups: ["Users", "Admins", "Unmapped"],
|
|
tokenExpiresAt: new Date(Date.now() + 120_000),
|
|
};
|
|
},
|
|
diagnose: async () => undefined,
|
|
};
|
|
const store = {
|
|
createOidcState: async (input: Record<string, unknown>) => {
|
|
stateInputs.push(input);
|
|
const record = stateRecord({
|
|
nonce: input.nonce as string,
|
|
codeVerifier: input.codeVerifier as string,
|
|
authConfigRevision: input.authConfigRevision as string,
|
|
issuer: input.issuer as string,
|
|
});
|
|
if (options.stateReturnTo) (record as { returnTo: string }).returnTo = options.stateReturnTo;
|
|
storedState = record;
|
|
return { state, record: storedState };
|
|
},
|
|
consumeOidcState: async (received: string) => {
|
|
if (received !== state) return undefined;
|
|
const consumed = storedState;
|
|
storedState = undefined;
|
|
return consumed;
|
|
},
|
|
create: async (input: Record<string, unknown>) => {
|
|
creates.push(input);
|
|
return { token: "opaque-session-token", csrfToken: "c".repeat(43), record: {} };
|
|
},
|
|
} as unknown as AuthSessionStore;
|
|
const app = Fastify();
|
|
app.decorateRequest("authConfigSnapshot", undefined);
|
|
app.decorateRequest("authConfigSnapshotCaptured", false);
|
|
app.decorateRequest("authConfigSnapshotUnavailable", false);
|
|
app.register(cookie);
|
|
registerAuthRoutes(app, {
|
|
authMode: "oidc",
|
|
authentication: { current: () => loaded },
|
|
sessionStore: store,
|
|
resolveOidcProtocol: () => protocolAvailable ? protocol : undefined,
|
|
});
|
|
createdApps.push(app);
|
|
return {
|
|
app, creates, callbacks, stateInputs,
|
|
setConfig(next: LoadedAuthConfig) { loaded = next; },
|
|
setProtocolAvailable(available: boolean) { protocolAvailable = available; },
|
|
stateWasConsumed: () => storedState === undefined,
|
|
};
|
|
}
|
|
|
|
afterEach(async () => {
|
|
await Promise.all(createdApps.splice(0).map((app) => app.close()));
|
|
});
|
|
|
|
test("creates digest-only bound state, maps exact groups, creates a cookie session, and redirects safely", async () => {
|
|
const subject = fixture();
|
|
const start = await subject.app.inject({ method: "GET", url: "/auth/oidc/login" });
|
|
expect(start.statusCode).toBe(302);
|
|
expect(new URL(start.headers.location ?? "").searchParams.get("state")).toBe(state);
|
|
expect(subject.stateInputs[0]).toMatchObject({ returnTo: "/", authConfigRevision: revision, issuer });
|
|
|
|
const callback = await subject.app.inject({
|
|
method: "GET",
|
|
url: `/auth/oidc/callback?code=good&state=${state}`,
|
|
headers: { host: "attacker.example.test" },
|
|
});
|
|
expect(callback.statusCode).toBe(302);
|
|
expect(callback.headers.location).toBe("/");
|
|
expect(callback.headers["set-cookie"]).toContain("HttpOnly");
|
|
expect(callback.headers["set-cookie"]).toContain("SameSite=Lax");
|
|
expect(callback.headers["set-cookie"]).toContain("Secure");
|
|
expect(subject.callbacks[0]?.href).toBe(`https://thothii.example.test/api/auth/oidc/callback?code=good&state=${state}`);
|
|
expect(subject.creates).toHaveLength(1);
|
|
expect(subject.creates[0]).toMatchObject({
|
|
method: "oidc", remembered: false, authConfigRevision: revision,
|
|
principal: { issuer, subject: "user-123", roles: ["user", "admin"] },
|
|
idleTtlMs: 300_000,
|
|
});
|
|
const absoluteTtlMs = subject.creates[0]?.absoluteTtlMs;
|
|
expect(typeof absoluteTtlMs).toBe("number");
|
|
expect(absoluteTtlMs as number).toBeGreaterThan(0);
|
|
expect(absoluteTtlMs as number).toBeLessThanOrEqual(120_000);
|
|
expect(JSON.stringify(subject.creates)).not.toContain("access-token-must-not-leak");
|
|
expect(JSON.stringify(subject.creates)).not.toContain("refresh-token-must-not-leak");
|
|
});
|
|
|
|
test("consumes state on callback failure and refuses replay", async () => {
|
|
const subject = fixture({ callbackFailure: true });
|
|
await subject.app.inject({ method: "GET", url: "/auth/oidc/login" });
|
|
const failed = await subject.app.inject({ method: "GET", url: `/auth/oidc/callback?code=good&state=${state}` });
|
|
expect(failed.statusCode).toBe(401);
|
|
const replay = await subject.app.inject({ method: "GET", url: `/auth/oidc/callback?code=good&state=${state}` });
|
|
expect(replay.statusCode).toBe(401);
|
|
expect(subject.callbacks).toHaveLength(1);
|
|
});
|
|
|
|
test("consumes state when the protocol becomes unavailable before callback", async () => {
|
|
const subject = fixture();
|
|
await subject.app.inject({ method: "GET", url: "/auth/oidc/login" });
|
|
subject.setProtocolAvailable(false);
|
|
const failed = await subject.app.inject({ method: "GET", url: `/auth/oidc/callback?code=good&state=${state}` });
|
|
expect(failed.statusCode).toBe(401);
|
|
expect(subject.stateWasConsumed()).toBe(true);
|
|
});
|
|
|
|
test("rejects a consumed state with a non-root return target", async () => {
|
|
const subject = fixture({ stateReturnTo: "https://attacker.example.test" });
|
|
await subject.app.inject({ method: "GET", url: "/auth/oidc/login" });
|
|
const callback = await subject.app.inject({ method: "GET", url: `/auth/oidc/callback?code=good&state=${state}` });
|
|
expect(callback.statusCode).toBe(401);
|
|
expect(subject.callbacks).toEqual([]);
|
|
expect(subject.creates).toEqual([]);
|
|
});
|
|
|
|
test("rejects an OIDC state when its configuration revision changes before callback", async () => {
|
|
const subject = fixture();
|
|
await subject.app.inject({ method: "GET", url: "/auth/oidc/login" });
|
|
subject.setConfig({ ...config(), revision: "b".repeat(64) });
|
|
const callback = await subject.app.inject({ method: "GET", url: `/auth/oidc/callback?code=good&state=${state}` });
|
|
expect(callback.statusCode).toBe(401);
|
|
expect(subject.callbacks).toEqual([]);
|
|
});
|
|
|
|
test("rejects an OIDC state when its issuer changes before callback", async () => {
|
|
const subject = fixture();
|
|
await subject.app.inject({ method: "GET", url: "/auth/oidc/login" });
|
|
const previous = config();
|
|
subject.setConfig({
|
|
...previous,
|
|
value: { ...previous.value, oidc: { ...previous.value.oidc, issuer: "https://other.example.test" } },
|
|
} as LoadedAuthConfig);
|
|
const callback = await subject.app.inject({ method: "GET", url: `/auth/oidc/callback?code=good&state=${state}` });
|
|
expect(callback.statusCode).toBe(401);
|
|
expect(subject.callbacks).toEqual([]);
|
|
});
|
|
|
|
test("creates an authenticated but forbidden principal for extra unmapped groups", async () => {
|
|
const subject = fixture({ identity: {
|
|
issuer, subject: "user-123", groups: ["Unmapped"], tokenExpiresAt: new Date(Date.now() + 60_000),
|
|
} });
|
|
await subject.app.inject({ method: "GET", url: "/auth/oidc/login" });
|
|
const callback = await subject.app.inject({ method: "GET", url: `/auth/oidc/callback?code=good&state=${state}` });
|
|
expect(callback.statusCode).toBe(302);
|
|
expect(subject.creates[0]).toMatchObject({ principal: { roles: [], permissions: [], isAdmin: false } });
|
|
});
|