Bump version v7.9.1

src/tunnel/io.rs

@@ -6,6 +6,7 @@ use std::time::Duration;
 use tokio::io::{AsyncRead, AsyncReadExt, AsyncWrite, AsyncWriteExt, ReadHalf, WriteHalf};
 use tokio::select;
 use tokio::sync::oneshot;
+use tokio::time::Instant;
 use tracing::log::debug;
 use tracing::{error, info, trace, warn};
 
@@ -24,7 +25,10 @@ pub(super) async fn propagate_read(
 
     // We do our own pin_mut! to avoid shadowing timeout and be able to reset it, on next loop iteration
     // We reuse the future to avoid creating a timer in the tight loop
-    let timeout = tokio::time::interval_at(tokio::time::Instant::now() + ping_frequency, ping_frequency);
+    let start_at = Instant::now()
+        .checked_add(ping_frequency)
+        .unwrap_or(Instant::now() + Duration::from_secs(3600 * 24));
+    let timeout = tokio::time::interval_at(start_at, ping_frequency);
     pin_mut!(timeout);
 
     pin_mut!(local_rx);

src/udp.rs

@@ -8,7 +8,6 @@ use std::future::Future;
 use std::io;
 use std::io::{Error, ErrorKind};
 use std::net::{Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6};
-use std::os::fd::AsRawFd;
 
 use log::warn;
 use std::pin::{pin, Pin};
@@ -36,24 +35,20 @@ struct UdpServer {
 }
 
 impl UdpServer {
-    pub fn new(listener: Arc<UdpSocket>, timeout: Option<Duration>) -> Self {
-        unsafe {
-            // Increase socket buffer length to 64MB
-            let buf_len: libc::c_int = 64 * 1024 * 1024;
-            let ret = libc::setsockopt(
-                listener.as_raw_fd(),
-                libc::SOL_SOCKET,
-                libc::SO_RCVBUF,
-                &buf_len as *const _ as *const libc::c_void,
-                std::mem::size_of_val(&buf_len) as libc::socklen_t,
-            );
-            if ret != 0 {
-                warn!("Cannot set UDP server recv buffer: {}", io::Error::last_os_error());
-            }
-        };
+    pub fn new(listener: UdpSocket, timeout: Option<Duration>) -> Self {
+        let socket = socket2::Socket::from(listener.into_std().unwrap());
+
+        // Increase receive buffer
+        if let Err(err) = socket.set_recv_buffer_size(64 * 1024 * 1024) {
+            warn!("Cannot set UDP server recv buffer: {}", err);
+        }
+
+        if let Err(err) = socket.set_send_buffer_size(64 * 1024 * 1024) {
+            warn!("Cannot set UDP server recv buffer: {}", err);
+        }
 
         Self {
-            listener,
+            listener: Arc::new(UdpSocket::from_std(socket.into()).unwrap()),
             peers: HashMap::with_hasher(ahash::RandomState::new()),
             keys_to_delete: Default::default(),
             cnx_timeout: timeout,
@@ -210,7 +205,7 @@ pub async fn run_server(
         .await
         .with_context(|| format!("Cannot create UDP server {:?}", bind))?;
 
-    let udp_server = UdpServer::new(Arc::new(listener), timeout);
+    let udp_server = UdpServer::new(listener, timeout);
     let stream = stream::unfold((udp_server, None), |(mut server, peer_with_data)| async move {
         // New returned peer hasn't read its data yet, await for it.
         if let Some(await_peer) = peer_with_data {