Ping local server to avoid connection interruptions.

reu · Apr 28, 2023 · 5cd9d58 · 5cd9d58
1 parent 7009d1a
commit 5cd9d58
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/tinytun-server/src/main.rs b/tinytun-server/src/main.rs
@@ -3,6 +3,7 @@ use std::{
     io::{BufRead, Cursor},
     net::SocketAddr,
     sync::Arc,
+    time::Duration,
 };
 
 use clap::Parser;
@@ -14,6 +15,7 @@ use hyper::{
 };
 use tokio::{
     net::{TcpListener, TcpStream},
+    time::sleep,
     try_join,
 };
 
@@ -99,10 +101,26 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
                                     if let Ok(conn) = hyper::upgrade::on(&mut req).await {
                                         trace!("Tunnel opened");
 
-                                        if let Ok((client, h2)) = h2::client::handshake(conn).await
+                                        if let Ok((client, mut h2)) =
+                                            h2::client::handshake(conn).await
                                         {
                                             tun_entry.add_tunnel(client.into()).await;
 
+                                            let mut ping_pong = h2.ping_pong().unwrap();
+                                            tokio::spawn(
+                                                #[allow(unreachable_code)]
+                                                async move {
+                                                    loop {
+                                                        sleep(Duration::from_secs(10)).await;
+                                                        trace!("Sending ping");
+                                                        ping_pong.ping(h2::Ping::opaque()).await?;
+                                                        trace!("Received pong");
+                                                    }
+                                                    Ok::<_, Box<dyn Error + Send + Sync>>(())
+                                                }
+                                                .in_current_span(),
+                                            );
+
                                             if let Err(err) = h2.await {
                                                 debug!(error = %err, "Error");
                                             }