uBlock/platform/firefox/frameModule.js

/* global Services, Components, XPCOMUtils */

'use strict';

this.EXPORTED_SYMBOLS = ['contentPolicy', 'docObserver'];

Components.utils['import']('resource://gre/modules/Services.jsm');
Components.utils['import']('resource://gre/modules/XPCOMUtils.jsm');

const Ci = Components.interfaces;
let appName;

try { throw new Error; } catch (ex) {
    appName = ex.fileName.match(/:\/\/([^\/]+)/)[1];
}

let getMessager = function(win) {
    try {
        // e10s
        return win
            .QueryInterface(Ci.nsIInterfaceRequestor)
            .getInterface(Ci.nsIWebNavigation)
            .QueryInterface(Ci.nsIDocShellTreeItem)
            .rootTreeItem
            .QueryInterface(Ci.nsIInterfaceRequestor)
            .getInterface(Ci.nsIContentFrameMessageManager);
    } catch (ex) {
        return win
            .QueryInterface(Ci.nsIInterfaceRequestor)
            .getInterface(Ci.nsIWebNavigation)
            .QueryInterface(Ci.nsIDocShell)
            .QueryInterface(Ci.nsIInterfaceRequestor)
            .getInterface(Ci.nsIContentFrameMessageManager);
    }
};

let contentPolicy = {
    classDescription: 'ContentPolicy implementation',
    classID: Components.ID('{e6d173c8-8dbf-4189-a6fd-189e8acffd27}'),
    contractID: '@' + appName + '/content-policy;1',
    ACCEPT: Ci.nsIContentPolicy.ACCEPT,
    REJECT: Ci.nsIContentPolicy.REJECT_REQUEST,
    requestMessageName: appName + ':onBeforeRequest',
    get componentRegistrar() {
        return Components.manager.QueryInterface(Ci.nsIComponentRegistrar);
    },
    get categoryManager() {
        return Components.classes['@mozilla.org/categorymanager;1']
                .getService(Ci.nsICategoryManager);
    },
    QueryInterface: XPCOMUtils.generateQI([
        Ci.nsIFactory,
        Ci.nsIContentPolicy,
        Ci.nsISupportsWeakReference
    ]),
    createInstance: function(outer, iid) {
        if (outer) {
            throw Components.results.NS_ERROR_NO_AGGREGATION;
        }

        return this.QueryInterface(iid);
    },
    register: function() {
        this.componentRegistrar.registerFactory(
            this.classID,
            this.classDescription,
            this.contractID,
            this
        );
        this.categoryManager.addCategoryEntry(
            'content-policy',
            this.contractID,
            this.contractID,
            false,
            true
        );
    },
    unregister: function() {
        this.componentRegistrar.unregisterFactory(this.classID, this);
        this.categoryManager.deleteCategoryEntry(
            'content-policy',
            this.contractID,
            false
        );
    },
    shouldLoad: function(type, location, origin, context) {
        if (!context || !/^https?$/.test(location.scheme)) {
            return this.ACCEPT;
        }

        let win = type === 6
            ? context.contentWindow
            : (context.ownerDocument || context).defaultView;

        if (!win) {
            return this.ACCEPT;
        }

        let result = getMessager(win).sendSyncMessage(this.requestMessageName, {
            url: location.spec,
            type: type,
            tabId: -1,
            frameId: type === 6 ? -1 : (win === win.top ? 0 : 1),
            parentFrameId: win === win.top ? -1 : 0
        })[0];

        return result === true ? this.REJECT : this.ACCEPT;
    }/*,
    shouldProcess: function() {
        return this.ACCEPT;
    }*/
};

let docObserver = {
    contentBaseURI: 'chrome://' + appName + '/content/js/',
    initContext: function(win, sandbox) {
        let messager = getMessager(win);

        if (sandbox) {
            win = Components.utils.Sandbox([win], {
                sandboxPrototype: win,
                wantComponents: false,
                wantXHRConstructor: false
            });

            win.self = win;
        }

        win.sendAsyncMessage = messager.sendAsyncMessage;
        win.addMessageListener = messager.ublock_addMessageListener;
        win.removeMessageListener = messager.ublock_removeMessageListener;

        return win;
    },
    register: function() {
        Services.obs.addObserver(this, 'document-element-inserted', false);
    },
    unregister: function() {
        Services.obs.removeObserver(this, 'document-element-inserted');
    },
    observe: function(doc) {
        let win = doc.defaultView;

        if (!win) {
            return;
        }

        if (!/^https?:$/.test(win.location.protocol)) {
            if (win.location.protocol === 'chrome:'
                && win.location.host === appName) {
                this.initContext(win);
            }

            return;
        }

        let lss = Services.scriptloader.loadSubScript;
        win = this.initContext(win, true);

        lss(this.contentBaseURI + 'vapi-client.js', win);
        lss(this.contentBaseURI + 'contentscript-start.js', win);

        let docReady = function(e) {
            this.removeEventListener(e.type, docReady, true);
            lss(docObserver.contentBaseURI + 'contentscript-end.js', win);
        };

        doc.addEventListener('DOMContentLoaded', docReady, true);
    }
};

contentPolicy.register();
docObserver.register();