Expose the inner QUIC connection

BiagioFesta · Sep 15, 2024 · 31d83d5 · 31d83d5
1 parent 194b583
commit 31d83d5
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/wtransport/Cargo.toml b/wtransport/Cargo.toml
@@ -69,6 +69,7 @@ rustdoc-args = ["--cfg", "docsrs"]
 allowed_external_types = [
     "bytes::bytes::Bytes",
     "quinn",
+    "quinn::connection::Connection",
     "quinn::recv_stream::ReadError",
     "quinn::recv_stream::RecvStream",
     "quinn::send_stream::SendStream",

diff --git a/wtransport/src/connection.rs b/wtransport/src/connection.rs
@@ -404,4 +404,20 @@ impl Connection {
             server_name: hd.server_name,
         }
     }
+
+    /// Returns a reference to the inner QUIC connection.
+    #[cfg(feature = "quinn")]
+    #[cfg_attr(docsrs, doc(cfg(feature = "quinn")))]
+    #[inline(always)]
+    pub fn quic_connection(&self) -> &quinn::Connection {
+        &self.quic_connection
+    }
+
+    /// Returns a mutable reference to the inner QUIC connection.
+    #[cfg(feature = "quinn")]
+    #[cfg_attr(docsrs, doc(cfg(feature = "quinn")))]
+    #[inline(always)]
+    pub fn quic_connection_mut(&mut self) -> &mut quinn::Connection {
+        &mut self.quic_connection
+    }
 }