fix(auth): SecretString redacted Serialize + PKCE verifier wired
Two findings from KeiSeiKit2.0 pr-review (~/Projects/KeiSeiKit2.0/skills/pr-review)
applied to commit range b346250..HEAD.
1. BLOCKER — SecretString silently leaked plaintext via Serialize.
File: _primitives/_rust/kei-runtime-core/src/secrets.rs
Was: derive(Serialize) + serde(transparent) -> serde_json::to_string(&secret)
emitted the raw plaintext in any parent struct with #[derive(Serialize)].
Debug was redacted but Serialize was not. Defeated the type's purpose.
Now: manual Serialize impl always emits literal "<redacted>". Deserialize
derive kept (callers need to read secrets from config/env).
Test serialize_emits_redacted_literal asserts JSON output is "\"<redacted>\"".
2. WARNING — PKCE code_verifier dropped before token exchange.
build_auth_url generated code_challenge = SHA256(verifier) but verify() never
threaded the verifier to the token endpoint. Token exchange submitted no
code_verifier, defeating the PKCE protection.
Files:
- _primitives/_rust/kei-runtime-core/src/traits/auth.rs:
AuthChallenge::OAuthCode now carries code_verifier: Option<String>.
Caller stores verifier alongside state in their session-store, exactly as
they already store state for CSRF check.
- _primitives/_rust/kei-auth-google/src/provider.rs:
verify() destructures code_verifier and passes to client.exchange_code(...).
- _primitives/_rust/kei-auth-apple/src/provider.rs:
same change.
Tests added (wiremock body assertions):
- google_smoke / apple_smoke: assert exchange request body contains
code_verifier=<value> when challenge carried Some(verifier).
- existing tests updated to construct OAuthCode { ..., code_verifier: None }.
Test split (Constructor Pattern 200 LOC):
- apple_smoke.rs grew over 200 LOC after PKCE test addition. Split into
apple_smoke.rs (provider tests) + apple_client_smoke.rs (client tests).
- same for google_smoke.rs / google_client_smoke.rs.
Test results: 31 passed; 0 failed across kei-auth, kei-auth-apple, kei-auth-google,
kei-runtime-core unit + integration tests. cargo check --workspace clean.
Breaking change: any caller that constructs AuthChallenge::OAuthCode outside this
workspace must add code_verifier field (None for legacy no-PKCE; Some for PKCE).
Compile-time surfaced gap, not runtime regression.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
parent
ddd13e6422
commit
a2b4dd6d66
9 changed files with 271 additions and 135 deletions
|
|
@ -106,13 +106,13 @@ impl AuthProvider for AppleAuthProvider {
|
||||||
async fn verify(&self, c: &AuthChallenge) -> CoreResult<AuthSession> {
|
async fn verify(&self, c: &AuthChallenge) -> CoreResult<AuthSession> {
|
||||||
let (code, state, expected_state, code_verifier) = match c {
|
let (code, state, expected_state, code_verifier) = match c {
|
||||||
AuthChallenge::OAuthCode {
|
AuthChallenge::OAuthCode {
|
||||||
provider, code, state, expected_state,
|
provider, code, state, expected_state, code_verifier,
|
||||||
} if provider == "apple" => {
|
} if provider == "apple" => (
|
||||||
// code_verifier is not threaded through AuthChallenge;
|
code.as_str(),
|
||||||
// callers pass it via the exchange directly if desired.
|
state.as_str(),
|
||||||
// Here we use None as the challenge only carries state.
|
expected_state.as_str(),
|
||||||
(code.as_str(), state.as_str(), expected_state.as_str(), None::<&str>)
|
code_verifier.as_deref(),
|
||||||
}
|
),
|
||||||
AuthChallenge::OAuthCode { provider, .. } => {
|
AuthChallenge::OAuthCode { provider, .. } => {
|
||||||
return Err(kei_runtime_core::Error::Auth(format!(
|
return Err(kei_runtime_core::Error::Auth(format!(
|
||||||
"wrong provider: expected apple, got {provider}"
|
"wrong provider: expected apple, got {provider}"
|
||||||
|
|
|
||||||
57
_primitives/_rust/kei-auth-apple/tests/apple_client_smoke.rs
Normal file
57
_primitives/_rust/kei-auth-apple/tests/apple_client_smoke.rs
Normal file
|
|
@ -0,0 +1,57 @@
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
|
// Copyright 2026 <author org>
|
||||||
|
//
|
||||||
|
//! Wiremock smoke tests for `AppleAuthClient` HTTP layer. No live HTTP.
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
|
mod helpers;
|
||||||
|
use helpers::{sign_id_token, token_response_body};
|
||||||
|
|
||||||
|
use kei_auth_apple::{AppleAuthClient, Error};
|
||||||
|
use wiremock::matchers::{method, path};
|
||||||
|
use wiremock::{Mock, MockServer, ResponseTemplate};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn token_endpoint_200_returns_token_response() {
|
||||||
|
let server = MockServer::start().await;
|
||||||
|
let id_token = sign_id_token(
|
||||||
|
r#"{"sub":"001234.abc","email":"x@y.example","iss":"https://appleid.apple.com","aud":"com.example.web"}"#,
|
||||||
|
);
|
||||||
|
Mock::given(method("POST"))
|
||||||
|
.and(path("/auth/token"))
|
||||||
|
.respond_with(
|
||||||
|
ResponseTemplate::new(200).set_body_json(token_response_body(&id_token)),
|
||||||
|
)
|
||||||
|
.mount(&server)
|
||||||
|
.await;
|
||||||
|
let token_url = format!("{}/auth/token", server.uri());
|
||||||
|
let c = AppleAuthClient::with_url(
|
||||||
|
token_url, "com.example.web", "JWT-CS", "https://app.example/cb",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let resp = c.exchange_code("auth-code-123", None).await.unwrap();
|
||||||
|
assert_eq!(resp.access_token, "at-1234");
|
||||||
|
assert_eq!(resp.expires_in, 3600);
|
||||||
|
assert_eq!(resp.id_token, id_token);
|
||||||
|
assert_eq!(resp.refresh_token.as_deref(), Some("rt-5678"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn token_endpoint_400_maps_to_api_error() {
|
||||||
|
let server = MockServer::start().await;
|
||||||
|
Mock::given(method("POST"))
|
||||||
|
.and(path("/auth/token"))
|
||||||
|
.respond_with(
|
||||||
|
ResponseTemplate::new(400)
|
||||||
|
.set_body_string("{\"error\":\"invalid_grant\"}"),
|
||||||
|
)
|
||||||
|
.mount(&server)
|
||||||
|
.await;
|
||||||
|
let token_url = format!("{}/auth/token", server.uri());
|
||||||
|
let c = AppleAuthClient::with_url(
|
||||||
|
token_url, "com.example.web", "JWT-CS", "https://app.example/cb",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let err = c.exchange_code("bad-code", None).await.unwrap_err();
|
||||||
|
assert!(matches!(err, Error::Api(_)), "expected Api(_), got {err:?}");
|
||||||
|
}
|
||||||
|
|
@ -1,67 +1,17 @@
|
||||||
// SPDX-License-Identifier: Apache-2.0
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
// Copyright 2026 <author org>
|
// Copyright 2026 <author org>
|
||||||
//
|
//
|
||||||
//! `wiremock`-driven smoke tests for [`AppleAuthClient`] +
|
//! Wiremock smoke tests for `AppleAuthProvider`. No live calls to appleid.apple.com.
|
||||||
//! [`AppleAuthProvider`]. No live calls to appleid.apple.com.
|
|
||||||
|
|
||||||
mod helpers;
|
mod helpers;
|
||||||
use helpers::{sign_id_token, token_response_body, TEST_JWKS_JSON};
|
use helpers::{sign_id_token, token_response_body, TEST_JWKS_JSON};
|
||||||
|
|
||||||
use kei_auth_apple::{AppleAuthClient, AppleAuthProvider, Error};
|
use kei_auth_apple::{AppleAuthClient, AppleAuthProvider};
|
||||||
use kei_runtime_core::HasDna;
|
use kei_runtime_core::HasDna;
|
||||||
use kei_runtime_core::traits::auth::{AuthChallenge, AuthProvider};
|
use kei_runtime_core::traits::auth::{AuthChallenge, AuthProvider};
|
||||||
use wiremock::matchers::{method, path};
|
use wiremock::matchers::{body_string_contains, method, path};
|
||||||
use wiremock::{Mock, MockServer, ResponseTemplate};
|
use wiremock::{Mock, MockServer, ResponseTemplate};
|
||||||
|
|
||||||
// ── Client-level tests ────────────────────────────────────────────────────────
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn token_endpoint_200_returns_token_response() {
|
|
||||||
let server = MockServer::start().await;
|
|
||||||
let id_token = sign_id_token(
|
|
||||||
r#"{"sub":"001234.abc","email":"x@y.example","iss":"https://appleid.apple.com","aud":"com.example.web"}"#,
|
|
||||||
);
|
|
||||||
Mock::given(method("POST"))
|
|
||||||
.and(path("/auth/token"))
|
|
||||||
.respond_with(
|
|
||||||
ResponseTemplate::new(200).set_body_json(token_response_body(&id_token)),
|
|
||||||
)
|
|
||||||
.mount(&server)
|
|
||||||
.await;
|
|
||||||
let token_url = format!("{}/auth/token", server.uri());
|
|
||||||
let c = AppleAuthClient::with_url(
|
|
||||||
token_url, "com.example.web", "JWT-CS", "https://app.example/cb",
|
|
||||||
)
|
|
||||||
.unwrap();
|
|
||||||
let resp = c.exchange_code("auth-code-123", None).await.unwrap();
|
|
||||||
assert_eq!(resp.access_token, "at-1234");
|
|
||||||
assert_eq!(resp.expires_in, 3600);
|
|
||||||
assert_eq!(resp.id_token, id_token);
|
|
||||||
assert_eq!(resp.refresh_token.as_deref(), Some("rt-5678"));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn token_endpoint_400_maps_to_api_error() {
|
|
||||||
let server = MockServer::start().await;
|
|
||||||
Mock::given(method("POST"))
|
|
||||||
.and(path("/auth/token"))
|
|
||||||
.respond_with(
|
|
||||||
ResponseTemplate::new(400)
|
|
||||||
.set_body_string("{\"error\":\"invalid_grant\"}"),
|
|
||||||
)
|
|
||||||
.mount(&server)
|
|
||||||
.await;
|
|
||||||
let token_url = format!("{}/auth/token", server.uri());
|
|
||||||
let c = AppleAuthClient::with_url(
|
|
||||||
token_url, "com.example.web", "JWT-CS", "https://app.example/cb",
|
|
||||||
)
|
|
||||||
.unwrap();
|
|
||||||
let err = c.exchange_code("bad-code", None).await.unwrap_err();
|
|
||||||
assert!(matches!(err, Error::Api(_)), "expected Api(_), got {err:?}");
|
|
||||||
}
|
|
||||||
|
|
||||||
// ── Provider-level tests ──────────────────────────────────────────────────────
|
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn provider_verify_end_to_end_returns_session_with_sub_user_id() {
|
async fn provider_verify_end_to_end_returns_session_with_sub_user_id() {
|
||||||
let server = MockServer::start().await;
|
let server = MockServer::start().await;
|
||||||
|
|
@ -86,6 +36,7 @@ async fn provider_verify_end_to_end_returns_session_with_sub_user_id() {
|
||||||
code: "auth-code-123".into(),
|
code: "auth-code-123".into(),
|
||||||
state: "csrf-token".into(),
|
state: "csrf-token".into(),
|
||||||
expected_state: "csrf-token".into(),
|
expected_state: "csrf-token".into(),
|
||||||
|
code_verifier: None,
|
||||||
};
|
};
|
||||||
let session = provider.verify(&challenge).await.unwrap();
|
let session = provider.verify(&challenge).await.unwrap();
|
||||||
assert_eq!(session.user_id, "001999.zzz");
|
assert_eq!(session.user_id, "001999.zzz");
|
||||||
|
|
@ -109,6 +60,7 @@ async fn provider_verify_csrf_mismatch_rejected() {
|
||||||
code: "code".into(),
|
code: "code".into(),
|
||||||
state: "DIFFERENT".into(),
|
state: "DIFFERENT".into(),
|
||||||
expected_state: "EXPECTED".into(),
|
expected_state: "EXPECTED".into(),
|
||||||
|
code_verifier: None,
|
||||||
};
|
};
|
||||||
let err = provider.verify(&challenge).await.unwrap_err();
|
let err = provider.verify(&challenge).await.unwrap_err();
|
||||||
assert!(
|
assert!(
|
||||||
|
|
@ -140,6 +92,7 @@ async fn jwt_decode_rejects_malformed_id_token() {
|
||||||
code: "auth-code-123".into(),
|
code: "auth-code-123".into(),
|
||||||
state: "csrf".into(),
|
state: "csrf".into(),
|
||||||
expected_state: "csrf".into(),
|
expected_state: "csrf".into(),
|
||||||
|
code_verifier: None,
|
||||||
};
|
};
|
||||||
let err = provider.verify(&challenge).await.unwrap_err();
|
let err = provider.verify(&challenge).await.unwrap_err();
|
||||||
let msg = format!("{err}");
|
let msg = format!("{err}");
|
||||||
|
|
@ -163,7 +116,40 @@ async fn provider_rejects_non_apple_oauth_code() {
|
||||||
code: "x".into(),
|
code: "x".into(),
|
||||||
state: "y".into(),
|
state: "y".into(),
|
||||||
expected_state: "y".into(),
|
expected_state: "y".into(),
|
||||||
|
code_verifier: None,
|
||||||
};
|
};
|
||||||
let err = provider.verify(&challenge).await.unwrap_err();
|
let err = provider.verify(&challenge).await.unwrap_err();
|
||||||
assert!(format!("{err}").contains("wrong provider"));
|
assert!(format!("{err}").contains("wrong provider"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn verify_sends_code_verifier_when_challenge_carries_some() {
|
||||||
|
let server = MockServer::start().await;
|
||||||
|
let id_token = sign_id_token(
|
||||||
|
r#"{"sub":"pkce-sub","email":"pkce@apple.example","iss":"https://appleid.apple.com","aud":"com.example.web"}"#,
|
||||||
|
);
|
||||||
|
Mock::given(method("POST"))
|
||||||
|
.and(path("/auth/token"))
|
||||||
|
.and(body_string_contains("code_verifier=apple-pkce-verifier"))
|
||||||
|
.respond_with(
|
||||||
|
ResponseTemplate::new(200).set_body_json(token_response_body(&id_token)),
|
||||||
|
)
|
||||||
|
.expect(1)
|
||||||
|
.mount(&server)
|
||||||
|
.await;
|
||||||
|
let token_url = format!("{}/auth/token", server.uri());
|
||||||
|
let client = AppleAuthClient::with_url(
|
||||||
|
token_url, "com.example.web", "JWT-CS", "https://app.example/cb",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
let provider = AppleAuthProvider::new(client, TEST_JWKS_JSON, None).unwrap();
|
||||||
|
let challenge = AuthChallenge::OAuthCode {
|
||||||
|
provider: "apple".into(),
|
||||||
|
code: "auth-code-pkce".into(),
|
||||||
|
state: "st".into(),
|
||||||
|
expected_state: "st".into(),
|
||||||
|
code_verifier: Some("apple-pkce-verifier".into()),
|
||||||
|
};
|
||||||
|
let session = provider.verify(&challenge).await.unwrap();
|
||||||
|
assert_eq!(session.user_id, "pkce-sub");
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -30,6 +30,7 @@
|
||||||
//! code: "<code from redirect>".into(),
|
//! code: "<code from redirect>".into(),
|
||||||
//! state: "<csrf state from callback>".into(),
|
//! state: "<csrf state from callback>".into(),
|
||||||
//! expected_state: "<csrf state you generated>".into(),
|
//! expected_state: "<csrf state you generated>".into(),
|
||||||
|
//! code_verifier: Some("<pkce verifier stored in session>".into()),
|
||||||
//! };
|
//! };
|
||||||
//! let session = provider.verify(&challenge).await?;
|
//! let session = provider.verify(&challenge).await?;
|
||||||
//! # let _ = session;
|
//! # let _ = session;
|
||||||
|
|
|
||||||
|
|
@ -87,10 +87,15 @@ impl AuthProvider for GoogleAuthProvider {
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn verify(&self, c: &AuthChallenge) -> kei_runtime_core::Result<AuthSession> {
|
async fn verify(&self, c: &AuthChallenge) -> kei_runtime_core::Result<AuthSession> {
|
||||||
let (code, state, expected_state) = match c {
|
let (code, state, expected_state, code_verifier) = match c {
|
||||||
AuthChallenge::OAuthCode {
|
AuthChallenge::OAuthCode {
|
||||||
provider, code, state, expected_state,
|
provider, code, state, expected_state, code_verifier,
|
||||||
} if provider == "google" => (code.as_str(), state.as_str(), expected_state.as_str()),
|
} if provider == "google" => (
|
||||||
|
code.as_str(),
|
||||||
|
state.as_str(),
|
||||||
|
expected_state.as_str(),
|
||||||
|
code_verifier.as_deref(),
|
||||||
|
),
|
||||||
AuthChallenge::OAuthCode { provider, .. } => {
|
AuthChallenge::OAuthCode { provider, .. } => {
|
||||||
return Err(kei_runtime_core::Error::Auth(format!(
|
return Err(kei_runtime_core::Error::Auth(format!(
|
||||||
"wrong provider for google: {provider}"
|
"wrong provider for google: {provider}"
|
||||||
|
|
@ -99,7 +104,7 @@ impl AuthProvider for GoogleAuthProvider {
|
||||||
_ => return Err(kei_runtime_core::Error::from(Error::MissingState)),
|
_ => return Err(kei_runtime_core::Error::from(Error::MissingState)),
|
||||||
};
|
};
|
||||||
check_state(state, expected_state)?;
|
check_state(state, expected_state)?;
|
||||||
let token = self.client.exchange_code(code, None).await
|
let token = self.client.exchange_code(code, code_verifier).await
|
||||||
.map_err(kei_runtime_core::Error::from)?;
|
.map_err(kei_runtime_core::Error::from)?;
|
||||||
let info = self.client.userinfo(&token.access_token).await
|
let info = self.client.userinfo(&token.access_token).await
|
||||||
.map_err(kei_runtime_core::Error::from)?;
|
.map_err(kei_runtime_core::Error::from)?;
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,86 @@
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
|
// Copyright 2026 <author org>
|
||||||
|
//!
|
||||||
|
//! Wiremock smoke tests for `GoogleAuthClient` HTTP layer. No live HTTP.
|
||||||
|
|
||||||
|
use kei_auth_google::GoogleAuthClient;
|
||||||
|
use serde_json::json;
|
||||||
|
use wiremock::matchers::{body_string_contains, header, method, path};
|
||||||
|
use wiremock::{Mock, MockServer, ResponseTemplate};
|
||||||
|
|
||||||
|
fn client_for(server: &MockServer) -> GoogleAuthClient {
|
||||||
|
GoogleAuthClient::with_urls(
|
||||||
|
format!("{}/token", server.uri()),
|
||||||
|
format!("{}/userinfo", server.uri()),
|
||||||
|
"client-id-xyz",
|
||||||
|
"client-secret-xyz",
|
||||||
|
"https://example.com/cb",
|
||||||
|
)
|
||||||
|
.unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn token_endpoint_200_returns_access_token() {
|
||||||
|
let server = MockServer::start().await;
|
||||||
|
Mock::given(method("POST"))
|
||||||
|
.and(path("/token"))
|
||||||
|
.and(body_string_contains("grant_type=authorization_code"))
|
||||||
|
.and(body_string_contains("code=abc123"))
|
||||||
|
.and(body_string_contains("client_id=client-id-xyz"))
|
||||||
|
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
|
||||||
|
"access_token": "ya29.a0AfH-test",
|
||||||
|
"expires_in": 3600,
|
||||||
|
"id_token": "eyJ.fake.jwt"
|
||||||
|
})))
|
||||||
|
.expect(1)
|
||||||
|
.mount(&server)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let client = client_for(&server);
|
||||||
|
let token = client.exchange_code("abc123", None).await.unwrap();
|
||||||
|
assert_eq!(token.access_token, "ya29.a0AfH-test");
|
||||||
|
assert_eq!(token.expires_in, 3600);
|
||||||
|
assert_eq!(token.id_token.as_deref(), Some("eyJ.fake.jwt"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn userinfo_200_returns_email_and_sub() {
|
||||||
|
let server = MockServer::start().await;
|
||||||
|
Mock::given(method("GET"))
|
||||||
|
.and(path("/userinfo"))
|
||||||
|
.and(header("authorization", "Bearer ya29.a0AfH-test"))
|
||||||
|
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
|
||||||
|
"sub": "1234567890",
|
||||||
|
"email": "alice@example.com",
|
||||||
|
"name": "Alice"
|
||||||
|
})))
|
||||||
|
.expect(1)
|
||||||
|
.mount(&server)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let client = client_for(&server);
|
||||||
|
let info = client.userinfo("ya29.a0AfH-test").await.unwrap();
|
||||||
|
assert_eq!(info.sub, "1234567890");
|
||||||
|
assert_eq!(info.email, "alice@example.com");
|
||||||
|
assert_eq!(info.name, "Alice");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn exchange_code_400_returns_api_error() {
|
||||||
|
let server = MockServer::start().await;
|
||||||
|
Mock::given(method("POST"))
|
||||||
|
.and(path("/token"))
|
||||||
|
.respond_with(ResponseTemplate::new(400).set_body_json(json!({
|
||||||
|
"error": "invalid_grant",
|
||||||
|
"error_description": "Bad code"
|
||||||
|
})))
|
||||||
|
.expect(1)
|
||||||
|
.mount(&server)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let client = client_for(&server);
|
||||||
|
let err = client.exchange_code("bad-code", None).await.unwrap_err();
|
||||||
|
let msg = format!("{err}");
|
||||||
|
assert!(msg.contains("api"), "expected api variant, got {msg}");
|
||||||
|
assert!(msg.contains("400"), "expected status 400 in message, got {msg}");
|
||||||
|
}
|
||||||
|
|
@ -1,8 +1,7 @@
|
||||||
// SPDX-License-Identifier: Apache-2.0
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
// Copyright 2026 <author org>
|
// Copyright 2026 <author org>
|
||||||
//!
|
//!
|
||||||
//! Wiremock smoke tests for `kei-auth-google`. No live HTTP — every
|
//! Wiremock smoke tests for `GoogleAuthProvider`. No live HTTP.
|
||||||
//! assertion is local to the test process.
|
|
||||||
|
|
||||||
use kei_auth_google::{GoogleAuthClient, GoogleAuthProvider};
|
use kei_auth_google::{GoogleAuthClient, GoogleAuthProvider};
|
||||||
use kei_runtime_core::traits::auth::{AuthChallenge, AuthProvider};
|
use kei_runtime_core::traits::auth::{AuthChallenge, AuthProvider};
|
||||||
|
|
@ -21,52 +20,6 @@ fn client_for(server: &MockServer) -> GoogleAuthClient {
|
||||||
.unwrap()
|
.unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn token_endpoint_200_returns_access_token() {
|
|
||||||
let server = MockServer::start().await;
|
|
||||||
Mock::given(method("POST"))
|
|
||||||
.and(path("/token"))
|
|
||||||
.and(body_string_contains("grant_type=authorization_code"))
|
|
||||||
.and(body_string_contains("code=abc123"))
|
|
||||||
.and(body_string_contains("client_id=client-id-xyz"))
|
|
||||||
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
|
|
||||||
"access_token": "ya29.a0AfH-test",
|
|
||||||
"expires_in": 3600,
|
|
||||||
"id_token": "eyJ.fake.jwt"
|
|
||||||
})))
|
|
||||||
.expect(1)
|
|
||||||
.mount(&server)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
let client = client_for(&server);
|
|
||||||
let token = client.exchange_code("abc123", None).await.unwrap();
|
|
||||||
assert_eq!(token.access_token, "ya29.a0AfH-test");
|
|
||||||
assert_eq!(token.expires_in, 3600);
|
|
||||||
assert_eq!(token.id_token.as_deref(), Some("eyJ.fake.jwt"));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn userinfo_200_returns_email_and_sub() {
|
|
||||||
let server = MockServer::start().await;
|
|
||||||
Mock::given(method("GET"))
|
|
||||||
.and(path("/userinfo"))
|
|
||||||
.and(header("authorization", "Bearer ya29.a0AfH-test"))
|
|
||||||
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
|
|
||||||
"sub": "1234567890",
|
|
||||||
"email": "alice@example.com",
|
|
||||||
"name": "Alice"
|
|
||||||
})))
|
|
||||||
.expect(1)
|
|
||||||
.mount(&server)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
let client = client_for(&server);
|
|
||||||
let info = client.userinfo("ya29.a0AfH-test").await.unwrap();
|
|
||||||
assert_eq!(info.sub, "1234567890");
|
|
||||||
assert_eq!(info.email, "alice@example.com");
|
|
||||||
assert_eq!(info.name, "Alice");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn verify_end_to_end_builds_auth_session() {
|
async fn verify_end_to_end_builds_auth_session() {
|
||||||
let server = MockServer::start().await;
|
let server = MockServer::start().await;
|
||||||
|
|
@ -99,6 +52,7 @@ async fn verify_end_to_end_builds_auth_session() {
|
||||||
code: "code-xyz".into(),
|
code: "code-xyz".into(),
|
||||||
state: "csrf-state-xyz".into(),
|
state: "csrf-state-xyz".into(),
|
||||||
expected_state: "csrf-state-xyz".into(),
|
expected_state: "csrf-state-xyz".into(),
|
||||||
|
code_verifier: None,
|
||||||
};
|
};
|
||||||
let session = provider.verify(&challenge).await.unwrap();
|
let session = provider.verify(&challenge).await.unwrap();
|
||||||
assert_eq!(session.user_id, "bob@example.com");
|
assert_eq!(session.user_id, "bob@example.com");
|
||||||
|
|
@ -109,26 +63,6 @@ async fn verify_end_to_end_builds_auth_session() {
|
||||||
assert!(session.expires_unix_ms > 0);
|
assert!(session.expires_unix_ms > 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn exchange_code_400_returns_api_error() {
|
|
||||||
let server = MockServer::start().await;
|
|
||||||
Mock::given(method("POST"))
|
|
||||||
.and(path("/token"))
|
|
||||||
.respond_with(ResponseTemplate::new(400).set_body_json(json!({
|
|
||||||
"error": "invalid_grant",
|
|
||||||
"error_description": "Bad code"
|
|
||||||
})))
|
|
||||||
.expect(1)
|
|
||||||
.mount(&server)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
let client = client_for(&server);
|
|
||||||
let err = client.exchange_code("bad-code", None).await.unwrap_err();
|
|
||||||
let msg = format!("{err}");
|
|
||||||
assert!(msg.contains("api"), "expected api variant, got {msg}");
|
|
||||||
assert!(msg.contains("400"), "expected status 400 in message, got {msg}");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn issue_challenge_rejects_non_oauth() {
|
async fn issue_challenge_rejects_non_oauth() {
|
||||||
let client = GoogleAuthClient::with_urls(
|
let client = GoogleAuthClient::with_urls(
|
||||||
|
|
@ -150,6 +84,7 @@ async fn verify_rejects_wrong_provider() {
|
||||||
code: "x".into(),
|
code: "x".into(),
|
||||||
state: "y".into(),
|
state: "y".into(),
|
||||||
expected_state: "y".into(),
|
expected_state: "y".into(),
|
||||||
|
code_verifier: None,
|
||||||
};
|
};
|
||||||
assert!(provider.verify(&challenge).await.is_err());
|
assert!(provider.verify(&challenge).await.is_err());
|
||||||
}
|
}
|
||||||
|
|
@ -168,6 +103,7 @@ async fn verify_rejects_csrf_state_mismatch() {
|
||||||
code: "code".into(),
|
code: "code".into(),
|
||||||
state: "got-this-state".into(),
|
state: "got-this-state".into(),
|
||||||
expected_state: "expected-state".into(),
|
expected_state: "expected-state".into(),
|
||||||
|
code_verifier: None,
|
||||||
};
|
};
|
||||||
let err = provider.verify(&challenge).await.unwrap_err();
|
let err = provider.verify(&challenge).await.unwrap_err();
|
||||||
assert!(
|
assert!(
|
||||||
|
|
@ -175,3 +111,41 @@ async fn verify_rejects_csrf_state_mismatch() {
|
||||||
"expected CSRF error, got: {err}"
|
"expected CSRF error, got: {err}"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn verify_sends_code_verifier_when_challenge_carries_some() {
|
||||||
|
let server = MockServer::start().await;
|
||||||
|
Mock::given(method("POST"))
|
||||||
|
.and(path("/token"))
|
||||||
|
.and(body_string_contains("code_verifier=my-pkce-verifier"))
|
||||||
|
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
|
||||||
|
"access_token": "tok-pkce",
|
||||||
|
"expires_in": 900,
|
||||||
|
"id_token": null
|
||||||
|
})))
|
||||||
|
.expect(1)
|
||||||
|
.mount(&server)
|
||||||
|
.await;
|
||||||
|
Mock::given(method("GET"))
|
||||||
|
.and(path("/userinfo"))
|
||||||
|
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
|
||||||
|
"sub": "pkce-sub",
|
||||||
|
"email": "pkce@example.com",
|
||||||
|
"name": "PKCE"
|
||||||
|
})))
|
||||||
|
.expect(1)
|
||||||
|
.mount(&server)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let client = client_for(&server);
|
||||||
|
let provider = GoogleAuthProvider::new(client, None).unwrap();
|
||||||
|
let challenge = AuthChallenge::OAuthCode {
|
||||||
|
provider: "google".into(),
|
||||||
|
code: "code-pkce".into(),
|
||||||
|
state: "st".into(),
|
||||||
|
expected_state: "st".into(),
|
||||||
|
code_verifier: Some("my-pkce-verifier".into()),
|
||||||
|
};
|
||||||
|
let session = provider.verify(&challenge).await.unwrap();
|
||||||
|
assert_eq!(session.user_id, "pkce@example.com");
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -10,12 +10,15 @@
|
||||||
//! The value is exposed only via [`SecretString::expose`], forcing callers
|
//! The value is exposed only via [`SecretString::expose`], forcing callers
|
||||||
//! to be explicit about accessing the secret.
|
//! to be explicit about accessing the secret.
|
||||||
|
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize, Serializer};
|
||||||
use std::fmt;
|
use std::fmt;
|
||||||
|
|
||||||
/// A string whose `Debug` impl is redacted and whose `Drop` zeroes memory.
|
/// A string whose `Debug` impl is redacted and whose `Drop` zeroes memory.
|
||||||
#[derive(Clone, Serialize, Deserialize)]
|
///
|
||||||
#[serde(transparent)]
|
/// `Serialize` emits the literal `"<redacted>"` so that parent structs
|
||||||
|
/// that derive `Serialize` never accidentally leak the secret value.
|
||||||
|
/// Use [`SecretString::expose`] to access the real value explicitly.
|
||||||
|
#[derive(Clone, Deserialize)]
|
||||||
pub struct SecretString(String);
|
pub struct SecretString(String);
|
||||||
|
|
||||||
impl SecretString {
|
impl SecretString {
|
||||||
|
|
@ -30,6 +33,16 @@ impl SecretString {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Serializes as the literal `"<redacted>"` — never the secret value.
|
||||||
|
///
|
||||||
|
/// This prevents accidental secret leaks when a parent struct that holds a
|
||||||
|
/// `SecretString` field also derives `Serialize`.
|
||||||
|
impl Serialize for SecretString {
|
||||||
|
fn serialize<S: Serializer>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error> {
|
||||||
|
serializer.serialize_str("<redacted>")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Always prints `<redacted>` — never the secret value.
|
/// Always prints `<redacted>` — never the secret value.
|
||||||
impl fmt::Debug for SecretString {
|
impl fmt::Debug for SecretString {
|
||||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
|
|
@ -78,4 +91,11 @@ mod tests {
|
||||||
let b = a.clone();
|
let b = a.clone();
|
||||||
assert_eq!(a.expose(), b.expose());
|
assert_eq!(a.expose(), b.expose());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn serialize_emits_redacted_literal() {
|
||||||
|
let s = SecretString::new("secret123");
|
||||||
|
let json = serde_json::to_string(&s).expect("serialize must not fail");
|
||||||
|
assert_eq!(json, "\"<redacted>\"");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -25,11 +25,18 @@ pub enum AuthChallenge {
|
||||||
/// `expected_state` — the nonce generated when the auth URL was built;
|
/// `expected_state` — the nonce generated when the auth URL was built;
|
||||||
/// must equal `state` (verified via constant-time comparison in each
|
/// must equal `state` (verified via constant-time comparison in each
|
||||||
/// provider's `verify()` impl).
|
/// provider's `verify()` impl).
|
||||||
|
///
|
||||||
|
/// `code_verifier` — the plain PKCE verifier (RFC 7636) originally passed
|
||||||
|
/// to `build_auth_url`. Store this alongside `state` in the session-store
|
||||||
|
/// when building the auth URL; pass it back here at callback time so
|
||||||
|
/// `verify()` can thread it through to the token exchange endpoint.
|
||||||
|
/// `None` retains legacy "no PKCE" behavior.
|
||||||
OAuthCode {
|
OAuthCode {
|
||||||
provider: String,
|
provider: String,
|
||||||
code: String,
|
code: String,
|
||||||
state: String,
|
state: String,
|
||||||
expected_state: String,
|
expected_state: String,
|
||||||
|
code_verifier: Option<String>,
|
||||||
},
|
},
|
||||||
SshKeySig { key_id: String, signature: String },
|
SshKeySig { key_id: String, signature: String },
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue