fix(connect): not negotiate h2 when using native-tls backend

seanmonstar · Mar 11, 2024 · 2674588 · 2674588
1 parent e319263
commit 2674588
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/src/connect.rs b/src/connect.rs
@@ -807,25 +807,41 @@ mod native_tls_conn {
 
     impl Connection for NativeTlsConn<TokioIo<TokioIo<TcpStream>>> {
         fn connected(&self) -> Connected {
-            self.inner
+            let connected = self
+                .inner
                 .inner()
                 .get_ref()
                 .get_ref()
                 .get_ref()
                 .inner()
-                .connected()
+                .connected();
+            #[cfg(feature = "native-tls-alpn")]
+            match self.inner.inner().get_ref().negotiated_alpn().ok() {
+                Some(Some(alpn_protocol)) if alpn_protocol == b"h2" => connected.negotiated_h2(),
+                _ => connected,
+            }
+            #[cfg(not(feature = "native-tls-alpn"))]
+            connected
         }
     }
 
     impl Connection for NativeTlsConn<TokioIo<MaybeHttpsStream<TokioIo<TcpStream>>>> {
         fn connected(&self) -> Connected {
-            self.inner
+            let connected = self
+                .inner
                 .inner()
                 .get_ref()
                 .get_ref()
                 .get_ref()
                 .inner()
-                .connected()
+                .connected();
+            #[cfg(feature = "native-tls-alpn")]
+            match self.inner.inner().get_ref().negotiated_alpn().ok() {
+                Some(Some(alpn_protocol)) if alpn_protocol == b"h2" => connected.negotiated_h2(),
+                _ => connected,
+            }
+            #[cfg(not(feature = "native-tls-alpn"))]
+            connected
         }
     }