Project

General

Profile

Download (3.89 KB) Statistics
| Branch: | Tag: | Revision:

haketilo / content / content.js @ fbfddb02

1
/**
2
 * This file is part of Haketilo.
3
 *
4
 * Function: Content scripts - main script.
5
 *
6
 * Copyright (C) 2022 Wojtek Kosior
7
 *
8
 * This program is free software: you can redistribute it and/or modify
9
 * it under the terms of the GNU General Public License as published by
10
 * the Free Software Foundation, either version 3 of the License, or
11
 * (at your option) any later version.
12
 *
13
 * This program is distributed in the hope that it will be useful,
14
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16
 * GNU General Public License for more details.
17
 *
18
 * As additional permission under GNU GPL version 3 section 7, you
19
 * may distribute forms of that code without the copy of the GNU
20
 * GPL normally required by section 4, provided you include this
21
 * license notice and, in case of non-source distribution, a URL
22
 * through which recipients can access the Corresponding Source.
23
 * If you modify file(s) with this exception, you may extend this
24
 * exception to your version of the file(s), but you are not
25
 * obligated to do so. If you do not wish to do so, delete this
26
 * exception statement from your version.
27
 *
28
 * As a special exception to the GPL, any HTML file which merely
29
 * makes function calls to this code, and for that purpose
30
 * includes it by reference shall be deemed a separate work for
31
 * copyright law purposes. If you modify this code, you may extend
32
 * this exception to your version of the code, but you are not
33
 * obligated to do so. If you do not wish to do so, delete this
34
 * exception statement from your version.
35
 *
36
 * You should have received a copy of the GNU General Public License
37
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
38
 *
39
 * I, Wojtek Kosior, thereby promise not to sue for violation of this file's
40
 * license. Although I request that you do not make use of this code in a
41
 * proprietary program, I am not going to enforce this in court.
42
 */
43

    
44
#IMPORT content/repo_query_cacher.js
45

    
46
#FROM common/browser.js           IMPORT browser
47
#FROM common/misc.js              IMPORT is_privileged_url
48
#FROM common/policy.js            IMPORT decide_policy
49
#FROM content/policy_enforcing.js IMPORT enforce_blocking
50

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

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

    
58
    page_info_prom.then(respond_cb);
59

    
60
    return true;
61
}
62

    
63
globalThis.haketilo_content_script_main = async function() {
64
    if (already_run)
65
        return;
66

    
67
    already_run = true;
68

    
69
    if (is_privileged_url(document.URL))
70
	return;
71

    
72
    browser.runtime.onMessage.addListener(on_page_info_request);
73
    repo_query_cacher.start();
74

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

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

    
87
    await enforce_blocking(policy);
88

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

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

    
100
		script.innerText = script_contents;
101
		script.setAttribute("nonce", policy.nonce);
102
		document.documentElement.append(script);
103
		script.remove();
104
	    }
105
	}
106
    }
107

    
108
    resolve_page_info(page_info);
109
}
110

    
111
function main() {
112
    if (globalThis.haketilo_pattern_tree !== undefined)
113
	globalThis.haketilo_content_script_main();
114
}
115

    
116
#IF !UNIT_TEST
117
main();
118
#ENDIF
(2-2/7)