From 3303d7d70d4b9749c39ca87085d17495beab6030 Mon Sep 17 00:00:00 2001
From: Wojtek Kosior <koszko@koszko.org>
Date: Thu, 26 Aug 2021 15:53:23 +0200
Subject: filter HTTP request headers to remove Hachette cookies in case they
 slip through

---
 background/main.js | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

(limited to 'background/main.js')

diff --git a/background/main.js b/background/main.js
index 2c8a87b..5d6e680 100644
--- a/background/main.js
+++ b/background/main.js
@@ -17,6 +17,7 @@
  * IMPORT gen_nonce
  * IMPORT inject_csp_headers
  * IMPORT apply_stream_filter
+ * IMPORT filter_cookie_headers
  * IMPORT is_chrome
  * IMPORTS_END
  */
@@ -81,18 +82,37 @@ function on_headers_received(details)
     return {responseHeaders: headers};
 }
 
+function on_before_send_headers(details)
+{
+    let headers = details.requestHeaders;
+    headers = filter_cookie_headers(headers);
+    return {requestHeaders: headers};
+}
+
+const all_types = [
+    "main_frame", "sub_frame", "stylesheet", "script", "image", "font",
+    "object", "xmlhttprequest", "ping", "csp_report", "media", "websocket",
+    "other", "main_frame", "sub_frame"
+];
+
 async function start_webRequest_operations()
 {
     storage = await get_storage();
 
-    const extra_opts = ["blocking", "responseHeaders"];
+    const extra_opts = ["blocking"];
     if (is_chrome)
 	extra_opts.push("extraHeaders");
 
     browser.webRequest.onHeadersReceived.addListener(
 	on_headers_received,
 	{urls: ["<all_urls>"], types: ["main_frame", "sub_frame"]},
-	extra_opts
+	extra_opts.concat("responseHeaders")
+    );
+
+    browser.webRequest.onBeforeSendHeaders.addListener(
+	on_before_send_headers,
+	{urls: ["<all_urls>"], types: all_types},
+	extra_opts.concat("requestHeaders")
     );
 }
 
-- 
cgit v1.2.3