kesto 0.6.0

//! Proveedor OIDC falso local para las pruebas.
//!
//! **Por qué este arnés y no dobles de las funciones de red.** Un doble de
//! `reqwest` —o de la propia capa de descubrimiento— devolvería lo que el test
//! le pidiera devolver, y con eso el flujo pasaría sin tocar la red. Eso dejaría
//! sin probar precisamente el requisito central de `auth-capability`: que el ID
//! token se **verifica** —firma `RS256` contra el `jwks_uri`, `iss`, `aud`,
//! `exp`/`iat` y `nonce`—, que es uno de los dos defectos de la plantilla que
//! este change corrige. Un doble de la red hace que la verificación de la firma
//! nunca se ejercite: se prueba que el código llama a las cosas, no que las
//! cosas funcionan.
//!
//! Por eso el arnés es `wiremock` sirviendo un proveedor de verdad: un par RSA
//! **real** (con `rsa`), el ID token firmado **`RS256` de verdad** con ese par,
//! y la clave pública publicada como JWK en el `jwks_uri`. La firma que se
//! publica y la que se verifica son la misma matemática que en producción.
//!
//! **Ninguna prueba sale a internet**: `wiremock` levanta un servidor HTTP real
//! atado a `127.0.0.1` con un puerto efímero, y todas las peticiones se quedan
//! en la máquina.
//!
//! Además de los tokens legítimos, el arnés sabe fabricar los de los dos ataques
//! clásicos a la verificación de un ID token: cabeceras a mano —`alg: none` y,
//! para los casos de `kid`, sin `kid` y con uno que el JWKS no publica— y una
//! firma `HS256` cuyo secreto HMAC son los bytes públicos del proveedor, que es
//! la confusión de algoritmo.

use base64::Engine as _;
use jsonwebtoken::jwk::{Jwk, JwkSet};
use jsonwebtoken::{Algorithm, EncodingKey, Header, encode};
use rsa::RsaPrivateKey;
use rsa::pkcs1::{EncodeRsaPrivateKey, EncodeRsaPublicKey};
use serde::{Deserialize, Serialize};
use wiremock::matchers::{method, path};
use wiremock::{Mock, MockServer, ResponseTemplate};

use crate::test_support::{EMAIL, SUB};

/// Ruta del documento de descubrimiento.
const RUTA_DESCUBRIMIENTO: &str = "/.well-known/openid-configuration";
/// Ruta del canje del código.
const RUTA_TOKEN: &str = "/token";
/// Ruta del conjunto de claves públicas.
const RUTA_JWKS: &str = "/jwks";

/// `kid` con el que el proveedor falso etiqueta su clave pública y sus tokens.
///
/// Es coherente entre la cabecera del JWT y el JWK a propósito: la verificación
/// tiene que encontrar la clave por su `kid` y comprobar la firma contra ella,
/// no tropezar antes con una clave ausente.
pub(crate) const KID: &str = "kesto-clave-de-prueba";

/// `kid` que el `jwks_uri` **no** publica.
///
/// Sirve para firmar un token que la verificación no puede resolver por su
/// clave: el JWKS solo publica [`KID`]. El token va firmado con la clave que sí
/// se publica, de modo que el rechazo no puede venir de la firma —cuadra— sino
/// de que la clave que la cabecera nombra no está en el conjunto.
pub(crate) const KID_AJENO: &str = "kid-que-el-jwks-no-publica";

/// Token de acceso del proveedor: distinguible para que buscarlo en el log o en
/// la cookie sea concluyente.
pub(crate) const ACCESS_TOKEN: &str = "access-token-del-proveedor-distinguible";

/// Claims del ID token que el proveedor falso firma.
///
/// Los valores por defecto de [`IdToken::valido`] son distinguibles: `sub` y
/// `email` son los de la sesión de pruebas, y `nonce` viaja tal cual, de modo
/// que los tests localizan la identidad literalmente en la cookie.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub(crate) struct IdToken {
    /// Emisor del token (`iss`).
    pub iss: String,
    /// Audiencia (`aud`): el `client_id` con el que se pidió la sesión.
    pub aud: String,
    /// Usuario (`sub`).
    pub sub: String,
    /// Correo (`email`).
    pub email: String,
    /// `nonce` devuelto: se compara con el emitido en el login.
    pub nonce: String,
    /// Caducidad, en segundos desde la época (`exp`).
    pub exp: u64,
    /// Emisión, en segundos desde la época (`iat`).
    pub iat: u64,
}

impl IdToken {
    /// Un ID token válido para el emisor, el `client_id` y el `nonce` dados.
    ///
    /// Caduca a diez minutos vista y se emite ahora: dentro de la ventana de
    /// vigencia y sin adelantarse al reloj. Los tests que quieren un token que
    /// falle **un** campo parten de aquí y lo tuercen.
    pub(crate) fn valido(issuer: &str, client_id: &str, nonce: &str) -> Self {
        let ahora = jsonwebtoken::get_current_timestamp();

        Self {
            iss: issuer.to_owned(),
            aud: client_id.to_owned(),
            sub: SUB.to_owned(),
            email: EMAIL.to_owned(),
            nonce: nonce.to_owned(),
            exp: ahora + 600,
            iat: ahora,
        }
    }
}

/// Proveedor OIDC falso: un servidor `wiremock` que sirve el descubrimiento, el
/// `token_endpoint` y el `jwks_uri`, con un par RSA real.
pub(crate) struct ProveedorFalso {
    /// El servidor real, vivo mientras lo esté el proveedor.
    server: MockServer,
    /// Emisor: la base contra la que se construyen los endpoints.
    pub(crate) issuer: String,
    /// Clave con la que se firman los ID tokens **válidos**.
    clave: EncodingKey,
    /// Clave que el `jwks_uri` **no** publica: firma los tokens que deben
    /// rechazarse por una firma que no cuadra. Es lo que convierte el escenario
    /// 1.18 en una prueba de la verificación criptográfica y no de una búsqueda.
    clave_no_publicada: EncodingKey,
    /// Clave de firma del token de confusión de algoritmo: un `HMAC` cuyo
    /// secreto son los **bytes públicos** del proveedor —los que cualquiera
    /// tiene del `jwks_uri`—. No es un par RSA: es un atacante firmando en
    /// `HS256` con el material que conoce, convencido de que la verificación
    /// leerá el algoritmo de la cabecera.
    clave_confusa: EncodingKey,
}

impl ProveedorFalso {
    /// Levanta el proveedor y monta el descubrimiento y el JWKS.
    ///
    /// El documento de descubrimiento **no** publica `userinfo_endpoint`, y eso
    /// es deliberado: el flujo no puede depender de `userinfo`, así que el
    /// descubrimiento no lo exige, el proveedor falso no lo ofrece y el
    /// escenario 1.21 lo comprueba.
    pub(crate) async fn servir() -> Self {
        let server = MockServer::start().await;
        let issuer = server.uri();

        // El par de claves es real: la firma que el callback verifica es la
        // misma matemática que en producción. La segunda clave nunca se publica.
        let (clave, clave_publica) = generar_clave();
        let (clave_no_publicada, _) = generar_clave();

        let documento = serde_json::json!({
            "authorization_endpoint": format!("{issuer}/authorize"),
            "token_endpoint": format!("{issuer}{RUTA_TOKEN}"),
            "jwks_uri": format!("{issuer}{RUTA_JWKS}"),
        });

        Mock::given(method("GET"))
            .and(path(RUTA_DESCUBRIMIENTO))
            .respond_with(ResponseTemplate::new(200).set_body_json(documento))
            .mount(&server)
            .await;

        Mock::given(method("GET"))
            .and(path(RUTA_JWKS))
            .respond_with(ResponseTemplate::new(200).set_body_json(jwks(&clave)))
            .mount(&server)
            .await;

        Self {
            server,
            issuer,
            clave,
            clave_no_publicada,
            clave_confusa: EncodingKey::from_secret(&clave_publica),
        }
    }

    /// Firma un ID token `RS256` con la clave que el JWKS **sí** publica.
    pub(crate) fn firmar(&self, claims: &IdToken) -> String {
        firmar_con(&self.clave, claims)
    }

    /// Firma un ID token `RS256` a partir de claims crudos.
    ///
    /// Es lo que permite fabricar tokens que **omiten** un claim obligatorio:
    /// [`IdToken`] declara `exp` e `iat` y no sabe representar su ausencia, y los
    /// candados de A2 necesitan un token sin `exp` y otro sin `iat` para probar
    /// que se rechazan.
    pub(crate) fn firmar_claims(&self, claims: &serde_json::Value) -> String {
        let mut cabecera = Header::new(Algorithm::RS256);
        cabecera.kid = Some(KID.to_owned());

        encode(&cabecera, claims, &self.clave).expect("el ID token de prueba debe firmarse")
    }

    /// Firma un ID token `RS256` con la clave que el JWKS **no** publica.
    ///
    /// La cabecera lleva el mismo `kid` que el JWK publicado: así la
    /// verificación encuentra la clave y el rechazo sale de que la firma no
    /// cuadra, que es lo que el escenario 1.18 quiere cazar. Con un `kid` ajeno,
    /// el token se rechazaría por no encontrar la clave y la comprobación
    /// criptográfica nunca se ejercitaría.
    pub(crate) fn firmar_con_clave_ajena(&self, claims: &IdToken) -> String {
        firmar_con(&self.clave_no_publicada, claims)
    }

    /// Firma un ID token `RS256` **sin `kid`** en la cabecera.
    ///
    /// La firma es legítima —la hace la clave que el JWKS publica—, pero la
    /// cabecera no dice con cuál, así que no hay clave que elegir. Un
    /// «si el conjunto trae una sola clave, úsala» aceptaría este token.
    pub(crate) fn firmar_sin_kid(&self, claims: &serde_json::Value) -> String {
        let cabecera = Header::new(Algorithm::RS256);

        encode(&cabecera, claims, &self.clave).expect("el ID token de prueba debe firmarse")
    }

    /// Firma un ID token `RS256` con la clave publicada pero con el `kid` de
    /// [`KID_AJENO`], que el JWKS no publica.
    ///
    /// La firma cuadra con la única clave del conjunto; lo que falla es la
    /// búsqueda por `kid`. Es el complemento de [`Self::firmar_con_clave_ajena`]:
    /// allí el `kid` resuelve a una clave y la firma no cuadra; aquí la firma
    /// cuadraría y es el `kid` el que no resuelve.
    pub(crate) fn firmar_con_kid_ajeno(&self, claims: &serde_json::Value) -> String {
        let mut cabecera = Header::new(Algorithm::RS256);
        cabecera.kid = Some(KID_AJENO.to_owned());

        encode(&cabecera, claims, &self.clave).expect("el ID token de prueba debe firmarse")
    }

    /// Firma un ID token `HS256` usando como secreto HMAC la **clave pública**
    /// del proveedor: la confusión de algoritmo clásica.
    ///
    /// El `kid` es el legítimo, así que la resolución de la clave no falla: lo
    /// único que separa a este token de la sesión es que la verificación fija
    /// `RS256` en vez de leer el algoritmo de la cabecera.
    pub(crate) fn firmar_confundiendo_algoritmo(&self, claims: &serde_json::Value) -> String {
        let mut cabecera = Header::new(Algorithm::HS256);
        cabecera.kid = Some(KID.to_owned());

        encode(&cabecera, claims, &self.clave_confusa).expect("el ID token de prueba debe firmarse")
    }

    /// Monta el `token_endpoint` para que devuelva el ID token dado.
    ///
    /// Devuelve también un `access_token` distinguible: el flujo no lo usa —con
    /// el ID token verificado sobra— y el escenario 1.38 comprobará que no acaba
    /// ni en el log ni en la cookie.
    pub(crate) async fn montar_token_endpoint(&self, id_token: &str) {
        let cuerpo = serde_json::json!({
            "id_token": id_token,
            "access_token": ACCESS_TOKEN,
            "token_type": "Bearer",
            "expires_in": 600,
        });

        Mock::given(method("POST"))
            .and(path(RUTA_TOKEN))
            .respond_with(ResponseTemplate::new(200).set_body_json(cuerpo))
            .mount(&self.server)
            .await;
    }

    /// Monta el `token_endpoint` para que conteste un error con el cuerpo dado.
    ///
    /// El cuerpo viaja a propósito, y lo elige la prueba: es donde se ponen
    /// credenciales distinguibles para comprobar que un error del proveedor —el
    /// único camino por el que un texto ajeno se interpola en el registro— no
    /// las arrastra al log. Kesto ni lee ese cuerpo: solo mira el código de
    /// estado.
    pub(crate) async fn montar_token_endpoint_con_error(&self, estado: u16, cuerpo: &str) {
        Mock::given(method("POST"))
            .and(path(RUTA_TOKEN))
            .respond_with(ResponseTemplate::new(estado).set_body_string(cuerpo))
            .mount(&self.server)
            .await;
    }

    /// Cuántas peticiones de descubrimiento han llegado al proveedor.
    ///
    /// Es lo que hace creíble la caché: contar las peticiones es la única forma
    /// de probar que un segundo login **no** vuelve a pedir el documento.
    pub(crate) async fn peticiones_de_descubrimiento(&self) -> usize {
        self.peticiones_a(RUTA_DESCUBRIMIENTO).await
    }

    /// Cuántas veces se ha llamado al `token_endpoint`.
    ///
    /// Es lo que prueba el escenario 1.14: un `error` del proveedor rechaza sin
    /// llegar a canjear nada, así que este contador tiene que quedarse en cero.
    pub(crate) async fn peticiones_al_token(&self) -> usize {
        self.peticiones_a(RUTA_TOKEN).await
    }

    /// Cuántas peticiones han ido a un endpoint `userinfo`, si lo hubiera.
    ///
    /// El proveedor falso **no** ofrece `userinfo`: si el flujo lo pidiera, la
    /// petición llegaría igual al servidor —sin mock que la atienda— y este
    /// contador lo delataría. Es lo que hace asertable el escenario 1.21.
    pub(crate) async fn peticiones_a_userinfo(&self) -> usize {
        self.server
            .received_requests()
            .await
            .unwrap_or_default()
            .iter()
            .filter(|peticion| peticion.url.path().contains("userinfo"))
            .count()
    }

    /// Cuenta las peticiones recibidas cuya ruta coincide con `ruta`.
    async fn peticiones_a(&self, ruta: &str) -> usize {
        self.server
            .received_requests()
            .await
            .unwrap_or_default()
            .iter()
            .filter(|peticion| peticion.url.path() == ruta)
            .count()
    }
}

/// Genera un par RSA real y devuelve la clave privada como clave de firma.
///
/// A 2048 bits, que es el mínimo que la verificación `RS256` de `jsonwebtoken`
/// espera. La clave se exporta en PKCS#1 DER porque es lo que acepta
/// `EncodingKey::from_rsa_der` y lo que `Jwk::from_encoding_key` sabe leer.
///
/// Devuelve también la **clave pública** en PKCS#1 DER: son los bytes que
/// cualquiera tiene del `jwks_uri` y con los que el atacante firma su token de
/// confusión de algoritmo.
fn generar_clave() -> (EncodingKey, Vec<u8>) {
    let mut rng = rand::thread_rng();

    let privada = RsaPrivateKey::new(&mut rng, 2048)
        .expect("la generación de un par RSA de prueba no puede fallar");
    let der = privada
        .to_pkcs1_der()
        .expect("la clave privada de prueba debe codificarse en PKCS#1")
        .as_bytes()
        .to_vec();
    let publica = privada
        .to_public_key()
        .to_pkcs1_der()
        .expect("la clave pública de prueba debe codificarse en PKCS#1")
        .as_bytes()
        .to_vec();

    (EncodingKey::from_rsa_der(&der), publica)
}

/// Publica la clave pública de `clave` como JWKS, con su `kid`.
///
/// Se usa `Jwk::from_encoding_key`, que extrae el módulo y el exponente y los
/// codifica en base64url sin relleno —como exige el RFC 7518—, de modo que la
/// misma biblioteca que firma y verifica construye la clave que publica. Nada
/// se codifica a mano.
fn jwks(clave: &EncodingKey) -> serde_json::Value {
    let mut jwk = Jwk::from_encoding_key(clave, Algorithm::RS256)
        .expect("la clave pública de prueba debe poder publicarse como JWK");
    jwk.common.key_id = Some(KID.to_owned());

    serde_json::to_value(JwkSet { keys: vec![jwk] }).expect("el JWKS de prueba debe serializarse")
}

/// Firma `claims` con `clave` y el `kid` del proveedor.
fn firmar_con(clave: &EncodingKey, claims: &IdToken) -> String {
    let mut cabecera = Header::new(Algorithm::RS256);
    cabecera.kid = Some(KID.to_owned());

    encode(&cabecera, claims, clave).expect("el ID token de prueba debe firmarse")
}

/// JWT con la cabecera `{"alg":"none","kid":…}` y **sin firma**.
///
/// Es la confusión de algoritmo más burda: el token declara que no va firmado y
/// los claims son los legítimos. No se construye con `jsonwebtoken` porque su
/// `Algorithm` no tiene la variante `none` —es justo la razón de que este token
/// se rechace al leer la cabecera—, así que las tres partes se codifican a mano
/// en base64url sin relleno, como manda el RFC 7515. La tercera va vacía: eso es
/// exactamente lo que significa `alg: none`.
pub(crate) fn token_con_alg_none(claims: &serde_json::Value) -> String {
    let cabecera = serde_json::json!({ "alg": "none", "kid": KID });

    let codificar = |valor: &serde_json::Value| {
        base64::engine::general_purpose::URL_SAFE_NO_PAD.encode(valor.to_string())
    };

    format!("{}.{}.", codificar(&cabecera), codificar(claims))
}