boring-rustls-provider/boring-rustls-provider/tests/e2e.rs

use rcgen::CertificateParams;
use std::sync::Arc;
use tokio::{
    io::{AsyncReadExt, AsyncWriteExt},
    net::TcpStream,
};

#[cfg(feature = "tls12")]
use boring_rustls_provider::tls12;
use boring_rustls_provider::tls13;
#[cfg(feature = "tls12")]
use rustls::version::TLS12;
use rustls::{version::TLS13, ClientConfig, ServerConfig, SupportedCipherSuite};
use rustls_pki_types::{CertificateDer, PrivateKeyDer, PrivatePkcs8KeyDer};
use tokio::net::TcpListener;
use tokio_rustls::{TlsAcceptor, TlsConnector};

#[tokio::test]
async fn test_tls13_crypto() {
    let pki = TestPki::new(&rcgen::PKCS_ECDSA_P256_SHA256);

    let root_store = pki.client_root_store();
    let server_config = pki.server_config();

    let ciphers = [
        SupportedCipherSuite::Tls13(&tls13::AES_128_GCM_SHA256),
        SupportedCipherSuite::Tls13(&tls13::AES_256_GCM_SHA384),
        SupportedCipherSuite::Tls13(&tls13::CHACHA20_POLY1305_SHA256),
    ];

    for cipher in ciphers {
        let config = ClientConfig::builder_with_provider(Arc::new(
            boring_rustls_provider::provider_with_ciphers([cipher].to_vec()),
        ))
        .with_protocol_versions(&[&TLS13])
        .unwrap()
        .with_root_certificates(root_store.clone())
        .with_no_client_auth();

        do_exchange(config, server_config.clone()).await;
    }
}

/// Self-to-self TLS 1.3 handshake using only the X25519MLKEM768 PQ hybrid group.
#[cfg(feature = "mlkem")]
#[tokio::test]
async fn test_tls13_pq_x25519_mlkem768() {
    use rustls::NamedGroup;

    let pki = TestPki::new(&rcgen::PKCS_ECDSA_P256_SHA256);

    let root_store = pki.client_root_store();

    // Build server with only X25519MLKEM768
    let server_provider =
        boring_rustls_provider::provider_with_ciphers(vec![SupportedCipherSuite::Tls13(
            &tls13::AES_256_GCM_SHA384,
        )]);
    let server_config = {
        let mut cfg = ServerConfig::builder_with_provider(Arc::new(server_provider))
            .with_protocol_versions(&[&TLS13])
            .unwrap()
            .with_no_client_auth()
            .with_single_cert(
                vec![pki.server_cert_der.clone()],
                pki.server_key_der.clone_key(),
            )
            .unwrap();
        cfg.key_log = Arc::new(rustls::KeyLogFile::new());
        Arc::new(cfg)
    };

    // Build client with only X25519MLKEM768
    let client_provider =
        boring_rustls_provider::provider_with_ciphers(vec![SupportedCipherSuite::Tls13(
            &tls13::AES_256_GCM_SHA384,
        )]);
    let config = ClientConfig::builder_with_provider(Arc::new(client_provider))
        .with_protocol_versions(&[&TLS13])
        .unwrap()
        .with_root_certificates(root_store)
        .with_no_client_auth();

    let negotiated_group = do_exchange(config, server_config).await;
    assert_eq!(negotiated_group, Some(NamedGroup::X25519MLKEM768));
}

/// Connect to Cloudflare's PQ test endpoint and verify X25519MLKEM768
/// was actually negotiated by checking the `/cdn-cgi/trace` response.
/// Marked `#[ignore]` because it depends on an external service.
#[cfg(feature = "mlkem")]
#[ignore]
#[tokio::test]
async fn test_pq_interop_cloudflare() {
    let mut root_store = rustls::RootCertStore::empty();
    root_store.extend(webpki_roots::TLS_SERVER_ROOTS.iter().cloned());

    let provider = boring_rustls_provider::provider();
    let config = ClientConfig::builder_with_provider(Arc::new(provider))
        .with_protocol_versions(&[&TLS13])
        .unwrap()
        .with_root_certificates(root_store)
        .with_no_client_auth();

    let connector = TlsConnector::from(Arc::new(config));
    let stream = TcpStream::connect("pq.cloudflareresearch.com:443")
        .await
        .unwrap();

    let mut stream = connector
        .connect(
            rustls_pki_types::ServerName::try_from("pq.cloudflareresearch.com").unwrap(),
            stream,
        )
        .await
        .expect("TLS handshake with pq.cloudflareresearch.com failed");

    // Hit the trace endpoint which reports negotiated TLS parameters
    stream
        .write_all(
            b"GET /cdn-cgi/trace HTTP/1.1\r\nHost: pq.cloudflareresearch.com\r\nConnection: close\r\n\r\n",
        )
        .await
        .unwrap();

    let mut buf = Vec::new();
    stream.read_to_end(&mut buf).await.unwrap();
    let response = String::from_utf8_lossy(&buf);

    // Verify TLS 1.3 was used
    assert!(
        response.contains("tls=TLSv1.3"),
        "expected TLSv1.3, got: {response}"
    );

    // Verify X25519MLKEM768 was negotiated as the key exchange
    assert!(
        response.contains("kex=X25519MLKEM768"),
        "expected kex=X25519MLKEM768, got: {response}"
    );
}

/// Connect to Cloudflare with TLS 1.2 forced and verify that a classical
/// key exchange is used (PQ groups are TLS 1.3 only).
/// Marked `#[ignore]` because it depends on an external service.
#[cfg(all(feature = "mlkem", feature = "tls12"))]
#[ignore]
#[tokio::test]
async fn test_tls12_interop_cloudflare_no_pq() {
    let mut root_store = rustls::RootCertStore::empty();
    root_store.extend(webpki_roots::TLS_SERVER_ROOTS.iter().cloned());

    let provider = boring_rustls_provider::provider();
    let config = ClientConfig::builder_with_provider(Arc::new(provider))
        .with_protocol_versions(&[&TLS12])
        .unwrap()
        .with_root_certificates(root_store)
        .with_no_client_auth();

    let connector = TlsConnector::from(Arc::new(config));
    let stream = TcpStream::connect("pq.cloudflareresearch.com:443")
        .await
        .unwrap();

    let mut stream = connector
        .connect(
            rustls_pki_types::ServerName::try_from("pq.cloudflareresearch.com").unwrap(),
            stream,
        )
        .await
        .expect("TLS handshake with pq.cloudflareresearch.com (TLS 1.2) failed");

    stream
        .write_all(
            b"GET /cdn-cgi/trace HTTP/1.1\r\nHost: pq.cloudflareresearch.com\r\nConnection: close\r\n\r\n",
        )
        .await
        .unwrap();

    let mut buf = Vec::new();
    stream.read_to_end(&mut buf).await.unwrap();
    let response = String::from_utf8_lossy(&buf);

    // Verify TLS 1.2 was used
    assert!(
        response.contains("tls=TLSv1.2"),
        "expected TLSv1.2, got: {response}"
    );

    // Verify a classical key exchange was used (not PQ)
    assert!(
        !response.contains("kex=X25519MLKEM768"),
        "TLS 1.2 should not use PQ key exchange, got: {response}"
    );
}

#[test]
#[cfg(feature = "fips")]
fn is_fips_enabled() {
    assert!(boring::fips::enabled());
}

#[test]
#[cfg(feature = "fips")]
fn fips_provider_excludes_chacha20_cipher_suites() {
    use rustls::CipherSuite;

    let provider = boring_rustls_provider::provider();
    let disallowed = [
        CipherSuite::TLS13_CHACHA20_POLY1305_SHA256,
        CipherSuite::TLS_ECDHE_ECDSA_WITH_CHACHA20_POLY1305_SHA256,
        CipherSuite::TLS_ECDHE_RSA_WITH_CHACHA20_POLY1305_SHA256,
    ];

    for suite in provider.cipher_suites {
        let selected = suite.suite();
        assert!(
            !disallowed.contains(&selected),
            "FIPS provider exposed disallowed cipher suite: {selected:?}"
        );
    }
}

#[test]
#[cfg(feature = "fips")]
fn fips_provider_restricts_kx_groups() {
    use rustls::NamedGroup;

    let provider = boring_rustls_provider::provider();
    let groups = provider
        .kx_groups
        .iter()
        .map(|group| group.name())
        .collect::<Vec<_>>();

    // fips implies mlkem, so X25519MLKEM768 must be present and preferred
    assert_eq!(
        groups[0],
        NamedGroup::X25519MLKEM768,
        "X25519MLKEM768 should be the first (preferred) FIPS KX group"
    );
    assert!(groups.contains(&NamedGroup::secp256r1));
    assert!(groups.contains(&NamedGroup::secp384r1));
    for group in &groups {
        assert!(
            matches!(
                group,
                NamedGroup::X25519MLKEM768 | NamedGroup::secp256r1 | NamedGroup::secp384r1
            ),
            "FIPS provider exposed disallowed KX group: {group:?}"
        );
    }
}

#[test]
#[cfg(feature = "fips")]
fn fips_provider_excludes_disallowed_signature_schemes() {
    use rustls::SignatureScheme;

    let provider = boring_rustls_provider::provider();
    let schemes = provider
        .signature_verification_algorithms
        .mapping
        .iter()
        .map(|(scheme, _)| *scheme)
        .collect::<Vec<_>>();

    assert!(schemes.contains(&SignatureScheme::RSA_PSS_SHA256));
    assert!(schemes.contains(&SignatureScheme::ECDSA_NISTP256_SHA256));

    for disallowed in [
        SignatureScheme::ECDSA_NISTP521_SHA512,
        SignatureScheme::ED25519,
        SignatureScheme::ED448,
    ] {
        assert!(
            !schemes.contains(&disallowed),
            "FIPS provider exposed disallowed signature scheme: {disallowed:?}"
        );
    }
}

#[test]
#[cfg(not(feature = "fips"))]
fn is_fips_disabled() {
    assert!(!boring::fips::enabled());
}

#[test]
#[cfg(not(feature = "fips"))]
fn non_fips_provider_keeps_non_fips_algorithms() {
    use rustls::{CipherSuite, NamedGroup};

    let provider = boring_rustls_provider::provider();

    assert!(provider
        .cipher_suites
        .iter()
        .any(|suite| { suite.suite() == CipherSuite::TLS13_CHACHA20_POLY1305_SHA256 }));

    assert!(provider
        .kx_groups
        .iter()
        .any(|group| group.name() == NamedGroup::X25519));
}

#[test]
#[cfg(all(not(feature = "fips"), feature = "mlkem"))]
fn non_fips_provider_includes_pq_group() {
    use rustls::NamedGroup;

    let provider = boring_rustls_provider::provider();
    let groups = provider
        .kx_groups
        .iter()
        .map(|group| group.name())
        .collect::<Vec<_>>();

    assert_eq!(
        groups[0],
        NamedGroup::X25519MLKEM768,
        "X25519MLKEM768 should be the first (preferred) KX group"
    );
    assert_eq!(
        groups[1],
        NamedGroup::X25519,
        "X25519 should remain the first classical fallback"
    );
    assert_eq!(
        groups[2],
        NamedGroup::X448,
        "X448 should remain ahead of NIST P-curves in non-FIPS mode"
    );
}

#[cfg(feature = "tls12")]
#[tokio::test]
async fn test_tls12_ec_crypto() {
    let pki = TestPki::new(&rcgen::PKCS_ECDSA_P256_SHA256);

    let root_store = pki.client_root_store();
    let server_config = pki.server_config();

    let ciphers = [
        SupportedCipherSuite::Tls12(&tls12::ECDHE_ECDSA_AES128_GCM_SHA256),
        SupportedCipherSuite::Tls12(&tls12::ECDHE_ECDSA_AES256_GCM_SHA384),
        SupportedCipherSuite::Tls12(&tls12::ECDHE_ECDSA_WITH_CHACHA20_POLY1305_SHA256),
    ];

    for cipher in ciphers {
        let config = ClientConfig::builder_with_provider(Arc::new(
            boring_rustls_provider::provider_with_ciphers([cipher].to_vec()),
        ))
        .with_protocol_versions(&[&TLS12])
        .unwrap()
        .with_root_certificates(root_store.clone())
        .with_no_client_auth();

        do_exchange(config, server_config.clone()).await;
    }
}

#[cfg(feature = "tls12")]
#[tokio::test]
async fn test_tls12_rsa_crypto() {
    let pki = TestPki::new(&rcgen::PKCS_RSA_SHA256);

    let root_store = pki.client_root_store();
    let server_config = pki.server_config();

    let ciphers = [
        SupportedCipherSuite::Tls12(&tls12::ECDHE_RSA_AES128_GCM_SHA256),
        SupportedCipherSuite::Tls12(&tls12::ECDHE_RSA_AES256_GCM_SHA384),
        SupportedCipherSuite::Tls12(&tls12::ECDHE_RSA_WITH_CHACHA20_POLY1305_SHA256),
    ];

    for cipher in ciphers {
        let config = ClientConfig::builder_with_provider(Arc::new(
            boring_rustls_provider::provider_with_ciphers([cipher].to_vec()),
        ))
        .with_protocol_versions(&[&TLS12])
        .unwrap()
        .with_root_certificates(root_store.clone())
        .with_no_client_auth();

        do_exchange(config, server_config.clone()).await;
    }
}

async fn new_listener() -> TcpListener {
    TcpListener::bind("localhost:0").await.unwrap()
}

async fn do_exchange(
    config: ClientConfig,
    server_config: Arc<ServerConfig>,
) -> Option<rustls::NamedGroup> {
    let listener = new_listener().await;
    let addr = listener.local_addr().unwrap();
    tokio::spawn(spawn_echo_server(listener, server_config.clone()));

    let connector = TlsConnector::from(Arc::new(config));
    let stream = TcpStream::connect(&addr).await.unwrap();

    let mut stream = connector
        .connect(
            rustls_pki_types::ServerName::try_from("localhost").unwrap(),
            stream,
        )
        .await
        .unwrap();

    let negotiated_group = stream
        .get_ref()
        .1
        .negotiated_key_exchange_group()
        .map(|group| group.name());

    stream.write_all(b"HELLO").await.unwrap();
    let mut buf = Vec::new();
    let bytes = stream.read_to_end(&mut buf).await.unwrap();
    assert_eq!(&buf[..bytes], b"HELLO");

    negotiated_group
}

async fn spawn_echo_server(listener: TcpListener, config: Arc<ServerConfig>) {
    let acceptor = TlsAcceptor::from(config);

    let (stream, _) = listener.accept().await.unwrap();
    let acceptor = acceptor.clone();
    let mut stream = acceptor.accept(stream).await.unwrap();

    let mut buf = vec![0u8; 5];
    let bytes = stream.read_exact(buf.as_mut_slice()).await.unwrap();
    stream.write_all(&buf[..bytes]).await.unwrap();
    stream.flush().await.unwrap();
    stream.shutdown().await.unwrap();
}

struct TestPki {
    ca_cert_der: CertificateDer<'static>,
    server_cert_der: CertificateDer<'static>,
    server_key_der: PrivateKeyDer<'static>,
}

impl TestPki {
    fn new(alg: &'static rcgen::SignatureAlgorithm) -> Self {
        let mut ca_params = rcgen::CertificateParams::new(Vec::new());
        ca_params
            .distinguished_name
            .push(rcgen::DnType::OrganizationName, "Provider Server Example");
        ca_params
            .distinguished_name
            .push(rcgen::DnType::CommonName, "Example CA");
        ca_params.is_ca = rcgen::IsCa::Ca(rcgen::BasicConstraints::Unconstrained);
        ca_params.key_usages = vec![
            rcgen::KeyUsagePurpose::KeyCertSign,
            rcgen::KeyUsagePurpose::DigitalSignature,
        ];
        keypair_for_alg(&mut ca_params, alg);
        ca_params.alg = alg;
        let ca_cert = rcgen::Certificate::from_params(ca_params).unwrap();

        let ca_cert_der = CertificateDer::from(ca_cert.serialize_der().unwrap());
        // Create a server end entity cert issued by the CA.
        let mut server_ee_params = rcgen::CertificateParams::new(vec!["localhost".to_string()]);
        server_ee_params.is_ca = rcgen::IsCa::NoCa;
        server_ee_params.extended_key_usages = vec![rcgen::ExtendedKeyUsagePurpose::ServerAuth];
        server_ee_params.alg = alg;
        keypair_for_alg(&mut server_ee_params, alg);
        let server_cert = rcgen::Certificate::from_params(server_ee_params).unwrap();
        let server_cert_der =
            CertificateDer::from(server_cert.serialize_der_with_signer(&ca_cert).unwrap());
        let server_key_der =
            PrivatePkcs8KeyDer::from(server_cert.serialize_private_key_der()).into();
        Self {
            ca_cert_der,
            server_cert_der,
            server_key_der,
        }
    }

    fn server_config(self) -> Arc<ServerConfig> {
        #[cfg(feature = "tls12")]
        let versions: &[&'static rustls::SupportedProtocolVersion] = &[&TLS12, &TLS13];
        #[cfg(not(feature = "tls12"))]
        let versions: &[&'static rustls::SupportedProtocolVersion] = &[&TLS13];

        let mut server_config =
            ServerConfig::builder_with_provider(Arc::new(boring_rustls_provider::provider()))
                .with_protocol_versions(versions)
                .unwrap()
                .with_no_client_auth()
                .with_single_cert(vec![self.server_cert_der], self.server_key_der)
                .unwrap();

        server_config.key_log = Arc::new(rustls::KeyLogFile::new());

        Arc::new(server_config)
    }

    fn client_root_store(&self) -> rustls::RootCertStore {
        let mut root_store = rustls::RootCertStore::empty();
        root_store.add(self.ca_cert_der.clone()).unwrap();
        root_store
    }
}

fn gen_rsa_key(bits: u32) -> rcgen::KeyPair {
    let rsa = boring::rsa::Rsa::generate(bits).unwrap();

    let der_pkcs8 = boring::pkey::PKey::from_rsa(rsa)
        .unwrap()
        .private_key_to_der_pkcs8()
        .unwrap();
    rcgen::KeyPair::from_der(&der_pkcs8).unwrap()
}

fn keypair_for_alg(params: &mut CertificateParams, alg: &rcgen::SignatureAlgorithm) {
    if alg == &rcgen::PKCS_RSA_SHA256 {
        params.key_pair = Some(gen_rsa_key(2048));
    } else if alg == &rcgen::PKCS_RSA_SHA384 {
        params.key_pair = Some(gen_rsa_key(3072));
    } else if alg == &rcgen::PKCS_RSA_SHA512 {
        params.key_pair = Some(gen_rsa_key(4096));
    }
}