g3proxy: check out http forward capacilities only when needed

bytedance · Jun 19, 2024 · 4b97ff0 · 4b97ff0
1 parent cc8e786
commit 4b97ff0
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/g3proxy/src/serve/http_proxy/task/pipeline/writer.rs b/g3proxy/src/serve/http_proxy/task/pipeline/writer.rs
@@ -265,21 +265,25 @@ where
             path_selection,
         );
 
-        let forward_capability = self
-            .forward_context
-            .check_in_final_escaper(&task_notes, &req.upstream)
-            .await;
         let remote_protocol = match req.client_protocol {
             HttpProxySubProtocol::TcpConnect => HttpProxySubProtocol::TcpConnect,
             HttpProxySubProtocol::HttpForward => HttpProxySubProtocol::HttpForward,
             HttpProxySubProtocol::HttpsForward => {
+                let forward_capability = self
+                    .forward_context
+                    .check_in_final_escaper(&task_notes, &req.upstream)
+                    .await;
                 if forward_capability.forward_https() {
                     HttpProxySubProtocol::HttpForward
                 } else {
                     HttpProxySubProtocol::HttpsForward
                 }
             }
             HttpProxySubProtocol::FtpOverHttp => {
+                let forward_capability = self
+                    .forward_context
+                    .check_in_final_escaper(&task_notes, &req.upstream)
+                    .await;
                 if forward_capability.forward_ftp(&req.inner.method) {
                     HttpProxySubProtocol::HttpForward
                 } else {