fix: fix get host from request header, #10

src/main.rs

@@ -368,15 +368,15 @@ fn run() -> Result<(), Box<dyn Error>> {
         for (name, plugin) in prox_plugins {
             info!(
                 "Proxy plugin {name}, category:{}, step:{}",
+                plugin.category(),
                 plugin.step(),
-                plugin.category()
             );
         }
         for (name, plugin) in response_plugins {
             info!(
                 "Response plugin {name}, category:{}, step:{}",
+                plugin.category(),
                 plugin.step(),
-                plugin.category()
             );
         }
     }

src/proxy/server.rs

@@ -456,7 +456,7 @@ impl ProxyHttp for Server {
             .locations
             .iter()
             .enumerate()
-            .find(|(_, item)| item.matched(&host, path));
+            .find(|(_, item)| item.matched(host, path));
 
         if location_result.is_none() {
             HttpResponse::unknown_error(Bytes::from(format!(

src/util/mod.rs

@@ -203,11 +203,19 @@ pub fn local_ip_list() -> Vec<String> {
     ip_list
 }
 
-pub fn get_host(header: &RequestHeader) -> Option<String> {
+/// Get request host in this order of precedence:
+/// host name from the request line,
+/// or host name from the "Host" request header field
+pub fn get_host(header: &RequestHeader) -> Option<&str> {
+    if let Some(host) = header.uri.host() {
+        return Some(host);
+    }
     if let Some(host) = header.headers.get("Host") {
-        return Some(host.to_str().unwrap_or_default().to_string());
+        if let Ok(value) = host.to_str().map(|host| host.split(':').next()) {
+            return value;
+        }
     }
-    header.uri.host().map(|host| host.to_string())
+    None
 }
 
 #[cfg(test)]