From c4ed1b8d44c37938aaa9b12674077274554742eb Mon Sep 17 00:00:00 2001
From: Wojtek Kosior <wk@koszkonutek-tmp.pl.eu.org>
Date: Thu, 13 May 2021 12:32:22 +0200
Subject: utilize CSP for blocking

---
 content/main.js         | 32 ++++++++++++++++++++++++++++----
 content/page_actions.js |  6 +++++-
 2 files changed, 33 insertions(+), 5 deletions(-)

(limited to 'content')

diff --git a/content/main.js b/content/main.js
index c7f57bb..507a740 100644
--- a/content/main.js
+++ b/content/main.js
@@ -25,6 +25,8 @@
     let url = url_item(document.URL);
     let unique = gen_unique(url);
 
+    let nonce = unique.substring(1);
+
     var block = true;
     if (first_target !== undefined &&
 	first_target === unique) {
@@ -47,10 +49,20 @@
 	}
 	for (let mutation of mutations) {
 	    for (let node of mutation.addedNodes) {
-		if (node.tagName === "SCRIPT")
+		/*
+		 * Modifying <script> element doesn't always prevent its
+		 * execution in some Mozilla browsers. Additional blocking
+		 * through CSP meta tag injection is required.
+		 */
+		if (node.tagName === "SCRIPT") {
 		    block_script(node);
-		else
-		    sanitize_attributes(node);
+		    continue;
+		}
+
+		sanitize_attributes(node);
+
+		if (node.tagName === "HEAD")
+		    inject_csp(node);
 	    }
 	}
     }
@@ -68,6 +80,18 @@
 	node.setAttribute("type", "application/json");
     }
 
+    function inject_csp(node)
+    {
+	console.log('injecting CSP');
+	let meta = document.createElement("meta");
+	meta.setAttribute("http-equiv", "Content-Security-Policy");
+	meta.setAttribute("content", `\
+script-src 'nonce-${nonce}'; \
+script-src-elem 'nonce-${nonce}';\
+`);
+	node.appendChild(meta);
+    }
+
     function sanitize_attributes(node)
     {
 	if (node.attributes === undefined)
@@ -99,5 +123,5 @@
 	});
     }
 
-    handle_page_actions();
+    handle_page_actions(nonce);
 })();
diff --git a/content/page_actions.js b/content/page_actions.js
index 047bf24..eee0599 100644
--- a/content/page_actions.js
+++ b/content/page_actions.js
@@ -17,6 +17,7 @@
     var port;
     var loaded = false;
     var scripts_awaiting = [];
+    var nonce;
 
     function handle_message(message)
     {
@@ -49,14 +50,17 @@
     {
 	let script = document.createElement("script");
 	script.textContent = script_text;
+	script.setAttribute("nonce", nonce);
 	document.body.appendChild(script);
     }
 
-    function handle_page_actions() {
+    function handle_page_actions(script_nonce) {
 	document.addEventListener("DOMContentLoaded", document_loaded);
 	port = browser.runtime.connect({name : CONNECTION_TYPE.PAGE_ACTIONS});
 	port.onMessage.addListener(handle_message);
 	port.postMessage({url: document.URL});
+
+	nonce = script_nonce;
     }
 
     window.handle_page_actions = handle_page_actions;
-- 
cgit v1.2.3