krimedc 0.2.0

A KDC based on libkrimes
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

use crate::config::{CoreAction, TaskName};
use libkrimes::cldap::{CldapConfig, CldapConfigBuilder};
use std::net::SocketAddr;
use std::sync::Arc;
use tokio::net::UdpSocket;
use tokio::sync::broadcast;
use tokio_stream::StreamExt;
use tokio_util::codec::BytesCodec;
use tokio_util::udp::UdpFramed;
use tracing::{error, info};

async fn cldap_udp_acceptor(
    sock: UdpSocket,
    cfg: &'static CldapConfig,
    mut rx: broadcast::Receiver<CoreAction>,
) {
    info!("Started task {}", TaskName::CldapUdp);

    let sock = Arc::new(sock);
    let mut codec = UdpFramed::new(sock.clone(), BytesCodec::new());

    loop {
        tokio::select! {
        Ok(action) = rx.recv() => {
            match action {
                CoreAction::Shutdown => break,
            }
        }
        frame = codec.next() => {
                if let Some(frame) = frame {
                    match frame {
                        Ok((msg, addr)) => {
                            tokio::spawn(libkrimes::cldap::process(cfg, sock.clone(), addr, msg));
                        },
                        Err(e) => {
                            error!("LDAP codec error, no LDAP message: {:?}", e)
                        }
                    }
                }
            }
        }
    }

    info!("Stopped task {}", TaskName::CldapUdp);
}

pub(crate) async fn create_cldap_server(
    cfg: &CldapConfigBuilder,
    rx: broadcast::Receiver<CoreAction>,
) -> Result<tokio::task::JoinHandle<()>, ()> {
    let cfg = cfg.build().expect("TODO");
    let cfg = Box::new(cfg);
    let cfg: &'static CldapConfig = Box::leak(cfg);

    if cfg.address().starts_with(":::") {
        let port = cfg.address().replacen(":::", "", 1);
        error!(
            "Address '{}' looks like an attempt to wildcard bind with IPv6 on port {} - please try using '[::]:{}'",
            cfg.address(),
            port,
            port
        );
    }

    let address = cfg.address().parse::<SocketAddr>().map_err(|e| {
        error!("Could not parse address {} -> {:?}", cfg.address(), e);
    })?;

    let sock = UdpSocket::bind(address).await.map_err(|e| {
        error!("Could not bind to address {} -> {:?}", cfg.address(), e);
    })?;

    let handle = tokio::spawn(cldap_udp_acceptor(sock, cfg, rx));

    Ok(handle)
}