diff options
Diffstat (limited to 'src/modules')
-rw-r--r-- | src/modules/hosts.rs | 4 |
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 |