libFenrir/src/tests.rs

use crate::*;

#[::tracing_test::traced_test]
#[::tokio::test]
async fn test_connection_dirsync() {
    use enc::asym::{KeyID, PrivKey, PubKey};
    let rand = enc::Random::new();
    let (priv_exchange_key, pub_exchange_key) =
        match enc::asym::KeyExchangeKind::X25519DiffieHellman.new_keypair(&rand)
        {
            Ok((privkey, pubkey)) => {
                (PrivKey::Exchange(privkey), PubKey::Exchange(pubkey))
            }
            Err(_) => {
                assert!(false, "Can't generate random keypair");
                return;
            }
        };
    let cfg_client = {
        let mut cfg = config::Config::default();
        cfg.threads = Some(::core::num::NonZeroUsize::new(1).unwrap());
        cfg
    };
    let test_domain: Domain = "example.com".into();
    let cfg_server = {
        let mut cfg = cfg_client.clone();
        cfg.server_keys = [config::ServerKey {
            id: KeyID(42),
            priv_key: priv_exchange_key,
            pub_key: pub_exchange_key,
        }]
        .to_vec();
        cfg.servers = [config::AuthServer {
            fqdn: test_domain.clone(),
            keys: [KeyID(42)].to_vec(),
        }]
        .to_vec();
        cfg
    };

    let (server, mut srv_workers) =
        Fenrir::with_workers(&cfg_server).await.unwrap();
    let (client, mut cli_workers) =
        Fenrir::with_workers(&cfg_client).await.unwrap();
    let mut srv_worker = srv_workers.pop().unwrap();
    let mut cli_worker = cli_workers.pop().unwrap();

    ::std::thread::spawn(move || {
        let rt = ::tokio::runtime::Builder::new_current_thread()
            .enable_all()
            .build()
            .unwrap();
        let local_thread = ::tokio::task::LocalSet::new();
        local_thread.spawn_local(async move {
            srv_worker.work_loop().await;
        });

        local_thread.spawn_local(async move {
            ::tokio::time::sleep(::std::time::Duration::from_millis(100)).await;
            cli_worker.work_loop().await;
        });
        rt.block_on(local_thread);
    });

    use crate::dnssec::{record, Record};

    let port: u16 = server.addresses()[0].port();

    let dnssec_record = Record {
        public_keys: [(KeyID(42), pub_exchange_key)].to_vec(),
        addresses: [record::Address {
            ip: ::std::net::IpAddr::V4(::std::net::Ipv4Addr::new(127, 0, 0, 1)),
            port: Some(::core::num::NonZeroU16::new(port).unwrap()),
            priority: record::AddressPriority::P1,
            weight: record::AddressWeight::W1,
            handshake_ids: [handshake::ID::DirectorySynchronized].to_vec(),
            public_key_idx: [record::PubKeyIdx(0)].to_vec(),
        }]
        .to_vec(),
        key_exchanges: [enc::asym::KeyExchangeKind::X25519DiffieHellman]
            .to_vec(),
        hkdfs: [enc::hkdf::Kind::Sha3].to_vec(),
        ciphers: [enc::sym::Kind::XChaCha20Poly1305].to_vec(),
    };

    ::tokio::time::sleep(::std::time::Duration::from_millis(500)).await;
    match client
        .connect_resolved(dnssec_record, &test_domain, auth::SERVICEID_AUTH)
        .await
    {
        Ok((_, _)) => {}
        Err(e) => {
            assert!(false, "Err on client connection: {:?}", e);
        }
    }

    server.graceful_stop().await;
    client.graceful_stop().await;
}