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
|
/* eslint-env mozilla/frame-script */
"use strict";
var Cm = Components.manager;
const { Services } = ChromeUtils.import("resource://gre/modules/Services.jsm");
const { generateUUID } = Cc["@mozilla.org/uuid-generator;1"].getService(
Ci.nsIUUIDGenerator
);
function AboutPage(aboutHost, chromeURL, uriFlags) {
this.chromeURL = chromeURL;
this.aboutHost = aboutHost;
this.classID = Components.ID(generateUUID().number);
this.description = "BrowserTestUtils: " + aboutHost;
this.uriFlags = uriFlags;
}
AboutPage.prototype = {
QueryInterface: ChromeUtils.generateQI(["nsIAboutModule"]),
getURIFlags(aURI) {
// eslint-disable-line no-unused-vars
return this.uriFlags;
},
newChannel(aURI, aLoadInfo) {
let newURI = Services.io.newURI(this.chromeURL);
let channel = Services.io.newChannelFromURIWithLoadInfo(newURI, aLoadInfo);
channel.originalURI = aURI;
if (this.uriFlags & Ci.nsIAboutModule.URI_SAFE_FOR_UNTRUSTED_CONTENT) {
channel.owner = null;
}
return channel;
},
createInstance(outer, iid) {
if (outer !== null) {
throw Components.Exception("", Cr.NS_ERROR_NO_AGGREGATION);
}
return this.QueryInterface(iid);
},
register() {
Cm.QueryInterface(Ci.nsIComponentRegistrar).registerFactory(
this.classID,
this.description,
"@mozilla.org/network/protocol/about;1?what=" + this.aboutHost,
this
);
},
unregister() {
Cm.QueryInterface(Ci.nsIComponentRegistrar).unregisterFactory(
this.classID,
this
);
},
};
const gRegisteredPages = new Map();
addMessageListener("browser-test-utils:about-registration:register", msg => {
let { aboutModule, pageURI, flags } = msg.data;
if (gRegisteredPages.has(aboutModule)) {
gRegisteredPages.get(aboutModule).unregister();
}
let moduleObj = new AboutPage(aboutModule, pageURI, flags);
moduleObj.register();
gRegisteredPages.set(aboutModule, moduleObj);
sendAsyncMessage(
"browser-test-utils:about-registration:registered",
aboutModule
);
});
addMessageListener("browser-test-utils:about-registration:unregister", msg => {
let aboutModule = msg.data;
let moduleObj = gRegisteredPages.get(aboutModule);
if (moduleObj) {
moduleObj.unregister();
gRegisteredPages.delete(aboutModule);
}
sendAsyncMessage(
"browser-test-utils:about-registration:unregistered",
aboutModule
);
});
|
