fix: Pin boxed futures before awaiting in connection handler

freenet · Dec 26, 2024 · d5f8d51 · d5f8d51
1 parent f81bd33
commit d5f8d51
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/crates/core/src/transport/connection_handler.rs b/crates/core/src/transport/connection_handler.rs
@@ -291,7 +291,8 @@ impl<S: Socket> UdpPacketsListener<S> {
                             let task = tokio::spawn({
                                 let span = tracing::span!(tracing::Level::DEBUG, "gateway_connection");
                                 async move {
-                                    match gw_ongoing_connection.await {
+                                    let pinned = Box::pin(gw_ongoing_connection);
+                                    match pinned.await {
                                         Ok(result) => Ok(result),
                                         Err(error) => Err((error, remote_addr))
                                     }
@@ -380,7 +381,8 @@ impl<S: Socket> UdpPacketsListener<S> {
                     let task = tokio::spawn({
                         let span = span!(tracing::Level::DEBUG, "traverse_nat");
                         async move {
-                            match ongoing_connection.await {
+                            let pinned = Box::pin(ongoing_connection);
+                            match pinned.await {
                                 Ok(result) => Ok(result),
                                 Err(error) => Err((error, remote_addr))
                             }