Haketilo

# SPDX-License-Identifier: CC0-1.0

"""

Haketilo unit tests - main content script

"""

# This file is part of Haketilo

#

# Copyright (C) 2022 Wojtek Kosior <koszko@koszko.org>

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the CC0 1.0 Universal License as published by

# the Creative Commons Corporation.

#

# 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

# CC0 1.0 Universal License for more details.

import pytest

import json

from selenium.webdriver.support.ui import WebDriverWait

from ..script_loader import load_script

# From:

# https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/contentScripts/register

# it is unclear whether the dynamically-registered content script is guaranteed

# to be always executed after statically-registered ones. We want to test both

# cases, so we'll make the mocked dynamic content script execute before

# content.js on http:// pages and after it on https:// pages.

dynamic_script = \

    ''';

    this.haketilo_secret        = "abracadabra";

    this.haketilo_pattern_tree  = {};

    this.haketilo_defualt_allow = false;

    if (this.haketilo_content_script_main)

        this.haketilo_content_script_main();

    '''

content_script = \

    '''

    /* Mock dynamic content script - case 'before'. */

    if (/dynamic_before/.test(document.URL)) {

        %s;

    }

    /* Place amalgamated content.js here. */

    %s;

    /* Rest of mocks */

    function mock_decide_policy() {

        nonce = "12345";

        return {

            allow: false,

            mapping: "what-is-programmers-favorite-drinking-place",

            payload: {identifier: "foo-bar"},

            nonce,

            csp: "prefetch-src 'none'; script-src-attr 'none'; script-src 'nonce-12345'; script-src-elem 'nonce-12345';"

        };

    }

    async function mock_payload_error([type, res_id]) {

        if (type === "indexeddb_files")

            return {error: {haketilo_error_type: "missing", id: res_id}};

    }

    async function mock_payload_ok([type, res_id]) {

        if (type === "indexeddb_files")

            return [1, 2].map(n => `window.haketilo_injected_${n} = ${n}${n};`);

    }

    if (/payload_error/.test(document.URL)) {

        browser.runtime.sendMessage = mock_payload_error;

        decide_policy = mock_decide_policy;

    } else if (/payload_ok/.test(document.URL)) {

        browser.runtime.sendMessage = mock_payload_ok;

        decide_policy = mock_decide_policy;

    }

    /* Otherwise, script blocking policy without payload to inject is used. */

    const data_to_verify = {};

    function data_set(prop, val) {

        data_to_verify[prop] = val;

        window.wrappedJSObject.data_to_verify = JSON.stringify(data_to_verify);

    }

    repo_query_cacher.start = () => data_set("cacher_started", true);

    enforce_blocking = policy => data_set("enforcing", policy);

    browser.runtime.onMessage.addListener = async function (listener_cb) {

        await new Promise(cb => setTimeout(cb, 10));

        /* Mock a good request. */

        const set_good = val => data_set("good_request_result", val);

        data_set("good_request_returned",

                 !!listener_cb(["page_info"], {}, val => set_good(val)));

        /* Mock a bad request. */

        const set_bad = val => data_set("bad_request_result", val);

        data_set("bad_request_returned",

                 !!listener_cb(["???"], {}, val => set_bad(val)));

    }

    /* main() call - normally present in content.js, inside '#IF !UNIT_TEST'. */

    main();

    /* Mock dynamic content script - case 'after'. */

    if (/#dynamic_after/.test(document.URL)) {

        %s;

    }

    data_set("script_run_without_errors", true);

    ''' % (dynamic_script, load_script('content/content.js'), dynamic_script)

@pytest.mark.ext_data({'content_script': content_script})

@pytest.mark.usefixtures('webextension')

@pytest.mark.parametrize('target1', ['dynamic_before'])#, 'dynamic_after'])

@pytest.mark.parametrize('target2', [

    'scripts_blocked',

    'payload_error',

    'payload_ok'

])

def test_content_unprivileged_page(driver, execute_in_page, target1, target2):

    """

    Test functioning of content.js on an page using unprivileged schema (e.g.

    'https://' and not 'about:').

    """

    driver.get(f'https://gotmyowndoma.in/index.html#{target1}-{target2}')

    def get_data(driver):

        data = driver.execute_script('return window.data_to_verify;')

        return data if 'good_request_result' in data else False

    data = json.loads(WebDriverWait(driver, 10).until(get_data))

    assert 'gotmyowndoma.in' in data['good_request_result']['url']

    assert 'bad_request_result' not in data

    assert data['good_request_returned'] == True

    assert data['bad_request_returned'] == False

    assert data['cacher_started'] == True

    for obj in (data['good_request_result'], data['enforcing']):

        assert obj['allow'] == False

    assert 'error' not in data['enforcing']

    if target2.startswith('payload'):

        for obj in (data['good_request_result'], data['enforcing']):

            assert obj['payload']['identifier'] == 'foo-bar'

            assert 'mapping' in obj

    else:

        assert 'payload' not in data['enforcing']

        assert 'mapping' not in data['enforcing']

    assert data['script_run_without_errors'] == True

    def vars_made_by_payload(driver):

        vars_values = driver.execute_script(

            'return [1, 2].map(n => window[`haketilo_injected_${n}`]);'

        )

        if vars_values != [None, None]:

            return vars_values

    if target2 == 'payload_error':

        assert data['good_request_result']['error'] == {

            'haketilo_error_type': 'missing',

            'id': 'foo-bar'

        }

    elif target2 == 'payload_ok':

        vars_values = WebDriverWait(driver, 10).until(vars_made_by_payload)

        assert vars_values == [11, 22]

@pytest.mark.ext_data({'content_script': content_script})

@pytest.mark.usefixtures('webextension')

@pytest.mark.parametrize('target', ['dynamic_before', 'dynamic_after'])

def test_content_privileged_page(driver, execute_in_page, target):

    """

    Test functioning of content.js on an page considered privileged (e.g. a

    directory listing at 'file:///').

    """

    driver.get(f'file:///#{target}')

    data = json.loads(driver.execute_script('return window.data_to_verify;'))

    assert data == {'script_run_without_errors': True}