fix hosts module for HTTP/2

1 files changed, 3 insertions, 1 deletions

diff --git a/src/modules/hosts.rs b/src/modules/hosts.rs
index 7ad7481..03a8b57 100644
--- a/src/modules/hosts.rs
+++ b/src/modules/hosts.rs
@@ -26,13 +26,15 @@ impl Node for Hosts {
         request: NodeRequest,
     ) -> Pin<Box<dyn Future<Output = Result<NodeResponse, ServiceError>> + Send + Sync + 'a>> {
         Box::pin(async move {
+            // use Host header but if missing like in HTTP/2 uses uri authority
             let host = request
                 .headers()
                 .get(HOST)
                 .and_then(|e| e.to_str().ok())
+                .map(remove_port)
+                .or(request.uri().authority().map(|a| a.host()))
                 .ok_or(ServiceError::NoHost)?;
 
-            let host = remove_port(host);
             let node = self.0.get(host).ok_or(ServiceError::UnknownHost)?;
 
             node.handle(context, request).await