Adapting the pty, tun and udp drivers for FreeBSD.

Author: leakingmemory

rust/linkfd/src/pty_dev.rs

@@ -25,6 +25,44 @@ pub(crate) struct PtyDev{
 }
 
 impl PtyDev {
+    #[cfg(not(target_os = "linux"))]
+    pub fn new() -> Option<Self> {
+        let mut ptyname: [u8;10] = [0u8; 10];
+        {
+            let const_ptyname = "/dev/ptyXY".as_bytes();
+            for i in 0..const_ptyname.len() {
+                ptyname[i] = const_ptyname[i];
+            }
+        }
+        let ch = "pqrstuvwxyz".as_bytes();
+        let digit = "0123456789abcdefghijklmnopqrstuv".as_bytes();
+
+        /* This algorithm should work for almost all standard Unices */
+        for l in 0..ch.len() {
+            for m in 0..digit.len() {
+                ptyname[8 as usize] = ch[l];
+                ptyname[9 as usize] = digit[m];
+                /* Open the master */
+                let mr_fd= unsafe { libc::open(ptyname.as_ptr() as *const i8, libc::O_RDWR) };
+                if mr_fd < 0 {
+                    continue;
+                }
+                /* Check the slave */
+                ptyname[5 as usize] = 't' as u8;
+                if unsafe { libc::access(ptyname.as_ptr() as *const i8, libc::R_OK | libc::W_OK) } < 0 {
+                    unsafe { libc::close(mr_fd); }
+                    ptyname[5 as usize] = 'p' as u8;
+                    continue;
+                }
+                return Some(Self {
+                    fd: mr_fd,
+                    ptyname: Box::new(str::from_utf8(&ptyname).unwrap().to_string())
+                });
+            }
+        }
+        None
+    }
+    #[cfg(target_os = "linux")]
     pub fn new() -> Option<Self> {
         /*
          * At first it appears that getpt is available on if not all,

rust/linkfd/src/tun_dev.rs

@@ -101,6 +101,11 @@ impl TunDev {
     pub fn new_from_fd(fd: i32, dev: &str) -> TunDev {
         TunDev { name: Some(Box::new(dev.to_string())), fd: Some(fd) }
     }
+
+    #[cfg(not(target_os = "linux"))]
+    fn linux_prep(&mut self, _name: Option<&str>, _dev_type: TunDevType) -> bool {
+        false
+    }
     #[cfg(target_os = "linux")]
     fn linux_prep(&mut self, name: Option<&str>, dev_type: TunDevType) -> bool {
         let mut ifr_flags: libc::c_short;

rust/linkfd/src/udp_proto.rs

@@ -24,6 +24,28 @@ pub(crate) struct UdpProto {
     pub is_rmt_fd_connected: bool
 }
 
+impl UdpProto {
+    #[cfg(not(target_os = "linux"))] /* freebsd */
+    fn create_sockaddr_in() -> libc::sockaddr_in {
+        libc::sockaddr_in {
+            sin_family: 0,
+            sin_port: 0,
+            sin_addr: libc::in_addr { s_addr: 0 },
+            sin_zero: [0i8; 8],
+            sin_len: 0
+        }
+    }
+    #[cfg(target_os = "linux")] /* linux */
+    fn create_sockaddr_in() -> libc::sockaddr_in {
+        libc::sockaddr_in {
+            sin_family: 0,
+            sin_port: 0,
+            sin_addr: libc::in_addr { s_addr: 0 },
+            sin_zero: [0u8; 8],
+        }
+    }
+}
+
 impl driver::NetworkDriver for UdpProto {
     fn write(&self, buf: &mut Vec<u8>, flags: u16) -> Option<usize> {
         let payloadlen = buf.len();
@@ -56,18 +78,14 @@ impl driver::NetworkDriver for UdpProto {
             return Some(wlen as usize);
         }
     }
+
     fn read(&mut self, buf: &mut Vec<u8>) -> Option<u16> {
         //unsigned short hdr, flen;
         let mut hdrbuf: [u8; 2] = [0u8; 2];
         let mut iv: [libc::iovec; 2] = [libc::iovec { iov_base: null_mut(), iov_len: 0 }; 2];
         //register int rlen;
         //struct sockaddr_in from;
-        let mut from: libc::sockaddr_in = libc::sockaddr_in {
-            sin_family: 0,
-            sin_port: 0,
-            sin_addr: libc::in_addr { s_addr: 0 },
-            sin_zero: [0u8; 8],
-        };
+        let mut from: libc::sockaddr_in = UdpProto::create_sockaddr_in();
         let mut fromlen: libc::socklen_t = size_of::<libc::sockaddr_in>() as libc::socklen_t;
 
         /* Late connect (NAT hack enabled) */