Extract remote host address from authority, then from header

boris-sinyapkin · Aug 15, 2024 · e3e98ee · e3e98ee
1 parent 7bae544
commit e3e98ee
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/server/handlers/http.rs b/src/server/handlers/http.rs
@@ -153,10 +153,9 @@ mod utils {
     };
 
     pub fn get_host_addr(req: &mut Request<body::Incoming>) -> Option<Address> {
-        if req.uri().authority().is_some() {
-            get_host_addr_from_authority(req)
-        } else {
-            get_host_addr_from_header(req)
+        match get_host_addr_from_authority(req) {
+            Some(addr) => Some(addr),
+            None => get_host_addr_from_header(req),
         }
     }