polyresolver 0.1.1

DNS resolver for client-side split horizon resolution over multiple domain arenas
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use std::{
    net::{IpAddr, SocketAddr},
    path::PathBuf,
    str::FromStr,
    time::Duration,
};

use rustls::{Certificate, PrivateKey};
use tokio::net::{TcpListener, UdpSocket};
use tracing::{error, info};
use trust_dns_server::ServerFuture;

use crate::catalog::PolyCatalog;

// listener routine for TCP, UDP, and TLS.
pub async fn listen<'a>(
    listen_ip: Option<IpAddr>,
    config_dir: PathBuf,
    tcp_timeout: Duration,
    certs: Option<Vec<Certificate>>,
    key: Option<PrivateKey>,
) -> Result<(), anyhow::Error> {
    let listen_ip = listen_ip.unwrap_or(IpAddr::from_str("127.0.0.1")?);

    let sa = SocketAddr::new(listen_ip, 53);
    let tcp = TcpListener::bind(sa).await?;
    let udp = UdpSocket::bind(sa).await?;

    let (closer_s, closer_r) = std::sync::mpsc::channel();
    let catalog = PolyCatalog::new();

    let mut sf = ServerFuture::new(catalog.clone());
    tokio::spawn(catalog.sync_catalog(config_dir, closer_r));

    if certs.is_some() && key.is_some() {
        info!("Configuring DoT Listener");

        let tls = TcpListener::bind(SocketAddr::new(listen_ip, 853)).await?;
        match sf.register_tls_listener(tls, tcp_timeout, (certs.unwrap(), key.unwrap())) {
            Ok(_) => {}
            Err(e) => error!("Cannot start DoT listener: {}", e),
        }
    }

    sf.register_socket(udp);
    sf.register_listener(tcp, tcp_timeout);

    let res = match sf.block_until_done().await {
        Ok(_) => Ok(()),
        Err(e) => Err(anyhow::anyhow!("{}", e)),
    };

    closer_s.send(()).expect("Could not close sync_catalog");
    res
}