initial import of main veilid core

2021-11-22 11:28:30 -05:00
parent c4cd54e020
commit 9e94a6a96f
218 changed files with 34880 additions and 1 deletions
@@ -0,0 +1,99 @@
+use super::*;
+cfg_if! {
+    if #[cfg(target_arch = "wasm32")] {
+
+    } else {
+        use std::net::{TcpListener, UdpSocket};
+    }
+}
+
+pub enum BumpPortType {
+    UDP,
+    TCP,
+}
+
+pub fn tcp_port_available(addr: &SocketAddr) -> bool {
+    cfg_if! {
+        if #[cfg(target_arch = "wasm32")] {
+            false
+        } else {
+            match TcpListener::bind(addr) {
+                Ok(_) => true,
+                Err(_) => false,
+            }
+        }
+    }
+}
+
+pub fn udp_port_available(addr: &SocketAddr) -> bool {
+    cfg_if! {
+        if #[cfg(target_arch = "wasm32")] {
+            false
+        } else {
+            match UdpSocket::bind(addr) {
+                Ok(_) => true,
+                Err(_) => false,
+            }
+        }
+    }
+}
+
+pub fn bump_port(addr: &mut SocketAddr, bpt: BumpPortType) -> Result<bool, String> {
+    cfg_if! {
+        if #[cfg(target_arch = "wasm32")] {
+            Err("unsupported architecture".to_owned())
+        }
+        else
+        {
+            let mut bumped = false;
+            let mut port = addr.port();
+            let mut addr_bump = addr.clone();
+            loop {
+
+                if match bpt {
+                    BumpPortType::TCP => tcp_port_available(&addr_bump),
+                    BumpPortType::UDP => udp_port_available(&addr_bump),
+                } {
+                    *addr = addr_bump;
+                    return Ok(bumped);
+                }
+                if port == u16::MAX {
+                    break;
+                }
+                port += 1;
+                addr_bump.set_port(port);
+                bumped = true;
+            }
+
+            Err("no ports remaining".to_owned())
+        }
+    }
+}
+
+pub fn bump_port_string(addr: &mut String, bpt: BumpPortType) -> Result<bool, String> {
+    cfg_if! {
+        if #[cfg(target_arch = "wasm32")] {
+            return Err("unsupported architecture".to_owned());
+        }
+        else
+        {
+            let savec: Vec<SocketAddr> = addr
+                .to_socket_addrs()
+                .map_err(|x| format!("failed to resolve socket address: {}", x))?
+                .collect();
+
+            if savec.len() == 0 {
+                return Err("No socket addresses resolved".to_owned());
+            }
+            let mut sa = savec.first().unwrap().clone();
+
+            if !bump_port(&mut sa, bpt)? {
+                return Ok(false);
+            }
+
+            *addr = sa.to_string();
+
+            Ok(true)
+        }
+    }
+}