aboutsummaryrefslogtreecommitdiff
path: root/content/content.js
blob: feef5db74d2200dff9ae634d799abc9103ebf669 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
/**
 * This file is part of Haketilo.
 *
 * Function: Content scripts - main script.
 *
 * Copyright (C) 2022 Wojtek Kosior
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * As additional permission under GNU GPL version 3 section 7, you
 * may distribute forms of that code without the copy of the GNU
 * GPL normally required by section 4, provided you include this
 * license notice and, in case of non-source distribution, a URL
 * through which recipients can access the Corresponding Source.
 * If you modify file(s) with this exception, you may extend this
 * exception to your version of the file(s), but you are not
 * obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 *
 * As a special exception to the GPL, any HTML file which merely
 * makes function calls to this code, and for that purpose
 * includes it by reference shall be deemed a separate work for
 * copyright law purposes. If you modify this code, you may extend
 * this exception to your version of the code, but you are not
 * obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 * I, Wojtek Kosior, thereby promise not to sue for violation of this file's
 * license. Although I request that you do not make use of this code in a
 * proprietary program, I am not going to enforce this in court.
 */

#IMPORT content/repo_query_cacher.js

#FROM common/browser.js           IMPORT browser
#FROM common/misc.js              IMPORT is_privileged_url
#FROM common/policy.js            IMPORT decide_policy
#FROM content/policy_enforcing.js IMPORT enforce_blocking

let already_run = false, resolve_page_info,
    page_info_prom = new Promise(cb => resolve_page_info = cb);

function on_page_info_request([type], sender, respond_cb) {
    if (type !== "page_info")
	return;

    page_info_prom.then(respond_cb);

    return true;
}

globalThis.haketilo_content_script_main = async function() {
    if (already_run)
        return;

    already_run = true;

    if (is_privileged_url(document.URL))
	return;

    browser.runtime.onMessage.addListener(on_page_info_request);
    repo_query_cacher.start();

    const policy = decide_policy(globalThis.haketilo_pattern_tree,
				 document.URL,
				 globalThis.haketilo_defualt_allow,
				 globalThis.haketilo_secret);
    const page_info = Object.assign({url: document.URL}, policy);
    ["csp", "nonce"].forEach(prop => delete page_info[prop]);

    if ("payload" in policy) {
	const msg = ["indexeddb_files", policy.payload.identifier];
	var scripts_prom = browser.runtime.sendMessage(msg);
    }

    await enforce_blocking(policy);

    if ("payload" in policy) {
	const script_response = await scripts_prom;

	if ("error" in script_response) {
	    resolve_page_info(Object.assign(page_info, script_response));
	    return;
	} else {
	    for (const script_contents of script_response) {
		const html_ns = "http://www.w3.org/1999/xhtml";
		const script = document.createElementNS(html_ns, "script");

		script.innerText = script_contents;
		script.setAttribute("nonce", policy.nonce);
		document.documentElement.append(script);
		script.remove();
	    }
	}
    }

    resolve_page_info(page_info);
}

function main() {
    if (globalThis.haketilo_pattern_tree !== undefined)
	globalThis.haketilo_content_script_main();
}

#IF !UNIT_TEST
main();
#ENDIF