Implement WebResourceRequested Event. (#34961)

* Implement WebResourceRequested Event on the Embedder Layer Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com> * fix and add test Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com> * resolve comments Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com> * remove sample code in webview Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com> * remove typo Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com> * ./mach format Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com> * fix test fail caused by interception message Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com> * update impl for is_for_main_frame Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com> --------- Signed-off-by: zhuhaichao518 <zhuhaichao518@gmail.com>
2025-08-03 04:30:10 +01:00 · 2025-01-16 15:44:16 +08:00 · 2025-01-16 15:44:16 +08:00 · a1326a7cf6
commit a1326a7cf6
parent 7256590599
13 changed files with 364 additions and 9 deletions
--- a/components/net/request_intercepter.rs
+++ b/components/net/request_intercepter.rs
@ -0,0 +1,104 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+use content_security_policy::Destination;
+use embedder_traits::{
+    EmbedderMsg, EmbedderProxy, HttpBodyData, WebResourceRequest, WebResourceResponseMsg,
+};
+use ipc_channel::ipc;
+use net_traits::http_status::HttpStatus;
+use net_traits::request::Request;
+use net_traits::response::{Response, ResponseBody};
+use net_traits::NetworkError;
+
+use crate::fetch::methods::FetchContext;
+
+#[derive(Clone)]
+pub struct RequestIntercepter {
+    embedder_proxy: EmbedderProxy,
+}
+
+impl RequestIntercepter {
+    pub fn new(embedder_proxy: EmbedderProxy) -> RequestIntercepter {
+        RequestIntercepter { embedder_proxy }
+    }
+
+    pub fn intercept_request(
+        &self,
+        request: &mut Request,
+        response: &mut Option<Response>,
+        context: &FetchContext,
+    ) {
+        let (tx, rx) = ipc::channel().unwrap();
+        let is_for_main_frame = matches!(request.destination, Destination::Document);
+        let req = WebResourceRequest::new(
+            request.method.clone(),
+            request.headers.clone(),
+            request.url(),
+            is_for_main_frame,
+            request.redirect_count > 0,
+        );
+
+        self.embedder_proxy.send((
+            request.target_browsing_context_id,
+            EmbedderMsg::WebResourceRequested(req, tx),
+        ));
+        let mut response_received = false;
+
+        // TODO: use done_chan and run in CoreResourceThreadPool.
+        while let Ok(msg) = rx.recv() {
+            match msg {
+                WebResourceResponseMsg::Start(webresource_response) => {
+                    response_received = true;
+                    let timing = context.timing.lock().unwrap().clone();
+                    let mut res = Response::new(webresource_response.url.clone(), timing);
+                    res.headers = webresource_response.headers;
+                    res.status = HttpStatus::new(
+                        webresource_response.status_code,
+                        webresource_response.status_message,
+                    );
+                    *res.body.lock().unwrap() = ResponseBody::Receiving(Vec::new());
+                    *response = Some(res);
+                },
+                WebResourceResponseMsg::Body(data) => {
+                    if !response_received {
+                        panic!("Receive body before initializing a Response!");
+                    }
+                    if let Some(ref mut res) = *response {
+                        match data {
+                            HttpBodyData::Chunk(chunk) => {
+                                if let ResponseBody::Receiving(ref mut body) =
+                                    *res.body.lock().unwrap()
+                                {
+                                    body.extend_from_slice(&chunk);
+                                } else {
+                                    panic!("Receive Playload Message when Response body is not in Receiving state!");
+                                }
+                            },
+                            HttpBodyData::Done => {
+                                let mut res_body = res.body.lock().unwrap();
+                                if let ResponseBody::Receiving(ref mut body) = *res_body {
+                                    let completed_body = std::mem::take(body);
+                                    *res_body = ResponseBody::Done(completed_body);
+                                } else {
+                                    panic!("Receive Done Message when Response body is not in Receiving state!");
+                                }
+                                break;
+                            },
+                            HttpBodyData::Cancelled => {
+                                *response =
+                                    Some(Response::network_error(NetworkError::LoadCancelled));
+                                break;
+                            },
+                        }
+                    }
+                },
+                WebResourceResponseMsg::None => {
+                    // Will not intercept the response. Continue.
+                    break;
+                },
+            }
+        }
+    }
+}