diff --git a/src/reverse_proxy.rs b/src/reverse_proxy.rs index 82e7c7a..10f463c 100644 --- a/src/reverse_proxy.rs +++ b/src/reverse_proxy.rs @@ -48,7 +48,7 @@ fn remove_hop_headers(headers: &HeaderMap) -> HeaderMap( // If request does not have host header, add it from original URI authority let host_header_name = "host"; - if let Some(authority) = request.uri().authority() { - if let hyper::header::Entry::Vacant(entry) = - builder.headers_mut().unwrap().entry(host_header_name) - { + if let hyper::header::Entry::Vacant(entry) = + builder.headers_mut().unwrap().entry(host_header_name) + { + if let Some(authority) = request.uri().authority() { entry.insert(authority.as_str().parse()?); } } @@ -111,6 +111,7 @@ fn create_proxied_request( "https".try_into()?, ); + // Proxy upgrade requests properly if let Some(conn) = request.headers().get("connection") { if conn.to_str()?.to_lowercase() == "upgrade" { if let Some(upgrade) = request.headers().get("upgrade") {