feat: (tcp) proxy

Cargo.toml

@@ -13,7 +13,6 @@ codegen-units = 1
 [dependencies]
 async-compression = { version = "0.4.27", features = ["tokio", "zstd"] }
 base32 = "0.5.1"
-base64 = "0.22.1"
 bytes = "1.10.1"
 chrono = { version = "0.4.41", default-features = false, features = ["clock", "now"] }
 clap = { version = "4.5.40", features = ["derive", "env"] }
@@ -25,13 +24,12 @@ futures = "0.3.31"
 futures-util = "0.3.31"
 glob = "0.3.2"
 hex = "0.4.3"
-human-units = "0.5.3"
 log = "0.4.27"
 lz4 = "1.28.1"
 nix = { version = "0.30.1", features = ["user"] }
 openssl = "0.10.73"
 page_size = "0.6.0"
-reqwest = { version = "0.13.1", features = ["json", "stream", "native-tls"] }
+reqwest = { version = "0.12.20", features = ["json", "stream", "native-tls"] }
 serde = { version = "1.0.219", features = ["derive"] }
 serde_json = "1.0.140"
 serde_yaml = "0.9.34"

src/apply.rs

@@ -3,8 +3,6 @@ use log::info;
 use std::path::Path;
 use tokio::fs;
 
-use crate::base64_decode;
-
 pub async fn files(files: &[crate::File], root: &str) -> Result<()> {
     for file in files {
         let path = chroot(root, &file.path);
@@ -17,10 +15,6 @@ pub async fn files(files: &[crate::File], root: &str) -> Result<()> {
         use crate::FileKind as K;
         match &file.kind {
             K::Content(content) => fs::write(path, content.as_bytes()).await?,
-            K::Content64(content) => {
-                let content = base64_decode(content)?;
-                fs::write(path, content).await?
-            }
             K::Dir(true) => fs::create_dir(path).await?,
             K::Dir(false) => {} // shouldn't happen, but semantic is to ignore
             K::Symlink(tgt) => fs::symlink(tgt, path).await?,

src/bin/dkl.rs

@@ -1,8 +1,8 @@
 use clap::{CommandFactory, Parser, Subcommand};
 use eyre::{format_err, Result};
-use human_units::Duration;
 use log::{debug, error};
 use std::net::SocketAddr;
+use std::time::Duration;
 use tokio::fs;
 
 #[derive(Parser)]
@@ -71,12 +71,12 @@ enum Command {
         #[arg(long, short = 'l')]
         listen: Vec<SocketAddr>,
         targets: Vec<SocketAddr>,
-        /// target polling interval
-        #[arg(long, default_value = "30s")]
-        poll: Duration,
-        /// connect or check timeout
-        #[arg(long, default_value = "5s")]
-        timeout: Duration,
+        /// target polling interval, in seconds
+        #[arg(long, default_value = "30")]
+        poll: u16,
+        /// connect or check timeout, in seconds
+        #[arg(long, default_value = "5")]
+        timeout: u16,
     },
 }
 
@@ -147,8 +147,8 @@ async fn main() -> Result<()> {
         } => Ok(dkl::proxy::Proxy {
             listen_addrs: listen,
             targets,
-            poll: poll.into(),
-            timeout: timeout.into(),
+            poll_interval: Duration::from_secs(poll.into()),
+            timeout: Duration::from_secs(timeout.into()),
         }
         .run()
         .await

src/bootstrap.rs

@@ -78,17 +78,6 @@ pub struct NetworkInterface {
     pub udev: Option<UdevFilter>,
 }
 
-impl Default for NetworkInterface {
-    fn default() -> Self {
-        Self {
-            var: "iface".into(),
-            n: 1,
-            regexps: Vec::new(),
-            udev: Some(UdevFilter::Eq("INTERFACE".into(), "eth0".into())),
-        }
-    }
-}
-
 #[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
 pub struct SSHServer {
     pub listen: String,
@@ -105,7 +94,7 @@ impl Default for SSHServer {
 
 #[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
 pub struct LvmVG {
-    #[serde(rename = "vg", alias = "name")]
+    #[serde(alias = "vg")]
     pub name: String,
     pub pvs: LvmPV,
 

src/dls.rs

@@ -170,8 +170,6 @@ pub struct Config {
     pub host_templates: Vec<HostConfig>,
     #[serde(default, rename = "SSLConfig")]
     pub ssl_config: String,
-    #[serde(default, deserialize_with = "deserialize_null_as_default")]
-    pub extra_ca_certs: Map<String, String>,
 }
 
 // compensate for go's encoder pitfalls

src/lib.rs

@@ -1,10 +1,10 @@
 pub mod apply;
+pub mod proxy;
 pub mod bootstrap;
 pub mod dls;
 pub mod dynlay;
 pub mod fs;
 pub mod logger;
-pub mod proxy;
 
 #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
 pub struct Config {
@@ -63,10 +63,9 @@ pub struct File {
 }
 
 #[derive(Debug, PartialEq, Eq, serde::Deserialize, serde::Serialize)]
-#[serde(rename_all = "lowercase")]
+#[serde(rename_all = "snake_case")]
 pub enum FileKind {
     Content(String),
-    Content64(String),
     Symlink(String),
     Dir(bool),
 }
@@ -81,8 +80,3 @@ impl Config {
         self.files.iter().find(|f| f.path == path)
     }
 }
-
-pub fn base64_decode(s: &str) -> Result<Vec<u8>, base64::DecodeError> {
-    use base64::{prelude::BASE64_STANDARD_NO_PAD as B64, Engine};
-    B64.decode(s.trim_end_matches('='))
-}

src/proxy.rs

@@ -1,17 +1,16 @@
-use log::{info, log_enabled, warn};
+use log::info;
 use std::convert::Infallible;
-use std::io;
 use std::net::SocketAddr;
-use std::sync::atomic::{AtomicBool, Ordering::Relaxed};
+use std::sync::atomic::Ordering::Relaxed;
+use std::sync::{atomic::AtomicBool, Arc};
 use std::time::Duration;
 use thiserror::Error;
 use tokio::net::{TcpListener, TcpStream};
-use tokio::time;
 
 pub struct Proxy {
     pub listen_addrs: Vec<SocketAddr>,
     pub targets: Vec<SocketAddr>,
-    pub poll: Duration,
+    pub poll_interval: Duration,
     pub timeout: Duration,
 }
 
@@ -37,96 +36,76 @@ impl Proxy {
 
         // all targets are initially ok (better land on a down one than just fail)
         let targets: Vec<_> = (self.targets.into_iter())
-            .map(|addr| TargetStatus {
-                addr,
-                up: AtomicBool::new(true),
-                timeout: self.timeout,
-            })
+            .map(|t| TargetStatus(t, AtomicBool::new(true)))
             .collect();
-
-        // the proxy runs forever -> using 'static is not a leak
-        let targets = targets.leak();
+        let targets = Arc::new(targets);
 
         for listener in listeners {
-            tokio::spawn(proxy_listener(listener, targets));
+            tokio::spawn(proxy_listener(listener, targets.clone()));
         }
 
-        check_targets(targets, self.poll).await
+        check_targets(targets.clone(), self.poll_interval, self.timeout).await
     }
 }
 
-struct TargetStatus {
-    addr: SocketAddr,
-    up: AtomicBool,
-    timeout: Duration,
-}
+struct TargetStatus(SocketAddr, AtomicBool);
 impl TargetStatus {
     fn is_up(&self) -> bool {
-        self.up.load(Relaxed)
-    }
-
-    fn set_up(&self, is_up: bool) {
-        let prev = self.up.swap(is_up, Relaxed);
-        if prev != is_up {
-            if is_up {
-                info!("{} is up", self.addr);
-            } else {
-                warn!("{} is down", self.addr);
-            }
-        }
-    }
-
-    async fn connect(&self) -> io::Result<TcpStream> {
-        let r = match time::timeout(self.timeout, TcpStream::connect(self.addr)).await {
-            Ok(r) => r,
-            Err(e) => Err(io::Error::new(io::ErrorKind::TimedOut, e)),
-        };
-
-        self.set_up(r.is_ok());
-        r
+        self.1.load(Relaxed)
     }
 }
 
-async fn check_targets(targets: &'static [TargetStatus], poll: Duration) -> ! {
+async fn check_targets(targets: Arc<Vec<TargetStatus>>, poll: Duration, timeout: Duration) -> ! {
     use tokio::time;
     let mut poll_ticker = time::interval(poll);
     poll_ticker.set_missed_tick_behavior(time::MissedTickBehavior::Skip);
 
+    let mut first = true;
+
     loop {
         poll_ticker.tick().await;
 
-        let mut tasks = tokio::task::JoinSet::new();
-
-        for target in targets {
-            tasks.spawn(target.connect());
+        for i in 0..targets.len() {
+            let targets = targets.clone();
+            tokio::spawn(async move { check_target(&targets[i], timeout, first).await });
         }
 
-        tasks.join_all().await;
-
-        if log_enabled!(log::Level::Info) {
-            let mut infos = String::new();
-            for ts in targets.iter() {
-                infos.push_str(&format!("{} ", ts.addr));
-                infos.push_str(if ts.is_up() { "up   " } else { "down " });
+        first = false;
     }
-            info!("{infos}");
+}
+
+async fn check_target(ts: &TargetStatus, timeout: Duration, first: bool) {
+    let target = ts.0;
+
+    let is_up = match tokio::time::timeout(timeout, TcpStream::connect(target)).await {
+        Ok(Ok(_)) => true,
+        _ => false,
+    };
+
+    let prev = ts.1.swap(is_up, Relaxed);
+
+    if first || prev != is_up {
+        if is_up {
+            info!("{target} is up");
+        } else {
+            info!("{target} is down");
         }
     }
 }
 
-async fn proxy_listener(listener: TcpListener, targets: &'static [TargetStatus]) {
+async fn proxy_listener(listener: TcpListener, targets: Arc<Vec<TargetStatus>>) {
     let mut rng = fastrand::Rng::new();
 
     loop {
         let mut active = Vec::with_capacity(targets.len());
         let (mut src, _) = listener.accept().await.expect("listener.accept() failed");
 
-        active.extend((targets.iter().enumerate()).filter_map(|(i, ts)| ts.is_up().then_some(i)));
+        active.extend(targets.iter().filter(|ts| ts.is_up()).map(|ts| ts.0));
         rng.shuffle(&mut active);
 
         tokio::spawn(async move {
-            for i in active {
-                if let Ok(mut dst) = targets[i].connect().await {
+            for target in active {
+                if let Ok(mut dst) = TcpStream::connect(target).await {
                     let _ = tokio::io::copy_bidirectional(&mut src, &mut dst).await;
                     break;
                 }