add alpn support (#107)

Author: seanmonstar

Cargo.toml

@@ -11,6 +11,7 @@ documentation = "https://docs.rs/hyper-tls"
 edition = "2018"
 
 [features]
+alpn = ["native-tls/alpn"]
 vendored = ["native-tls/vendored"]
 
 [dependencies]

src/lib.rs

@@ -22,6 +22,10 @@
 //!     Ok(())
 //! }
 //! ```
+//!
+//! ## Crate Features
+//!
+//! - `alpn`: Enables `native-tls/alpn`, and if `h2` is negotiated, tells hyper.
 #![doc(html_root_url = "https://docs.rs/hyper-tls/0.5.0")]
 #![cfg_attr(test, deny(warnings))]
 #![deny(missing_docs)]

src/stream.rs

@@ -111,13 +111,28 @@ impl<T: Write + Read + Unpin> Write for MaybeHttpsStream<T> {
     }
 }
 
-impl<T: Connection + Unpin> Connection for MaybeHttpsStream<T> {
+impl<T: Write + Read + Connection + Unpin> Connection for MaybeHttpsStream<T> {
     fn connected(&self) -> Connected {
         match self {
             MaybeHttpsStream::Http(s) => s.connected(),
             MaybeHttpsStream::Https(s) => {
-                s.inner().get_ref().get_ref().get_ref().inner().connected()
+                let c = s.inner().get_ref().get_ref().get_ref().inner().connected();
+                #[cfg(feature = "alpn")]
+                {
+                    if negotiated_h2(s.inner().get_ref()) {
+                        return c.negotiated_h2();
+                    }
+                }
+                c
             }
         }
     }
 }
+
+#[cfg(feature = "alpn")]
+fn negotiated_h2<T: std::io::Read + std::io::Write>(s: &native_tls::TlsStream<T>) -> bool {
+    s.negotiated_alpn()
+        .unwrap_or(None)
+        .map(|list| list == &b"h2"[..])
+        .unwrap_or(false)
+}