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
|
/* -*- indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
"use strict";
const { HttpServer } = ChromeUtils.importESModule(
"resource://testing-common/httpd.sys.mjs"
);
let hserv = Cc["@mozilla.org/uriloader/handler-service;1"].getService(
Ci.nsIHandlerService
);
let handlerInfo;
const testScheme = "x-moz-test";
function setup() {
var handler = Cc["@mozilla.org/uriloader/web-handler-app;1"].createInstance(
Ci.nsIWebHandlerApp
);
handler.name = testScheme;
handler.uriTemplate = "http://test.mozilla.org/%s";
var extps = Cc[
"@mozilla.org/uriloader/external-protocol-service;1"
].getService(Ci.nsIExternalProtocolService);
handlerInfo = extps.getProtocolHandlerInfo(testScheme);
handlerInfo.possibleApplicationHandlers.appendElement(handler);
hserv.store(handlerInfo);
Assert.ok(extps.externalProtocolHandlerExists(testScheme));
}
setup();
registerCleanupFunction(() => {
hserv.remove(handlerInfo);
});
function makeChan(url) {
let chan = NetUtil.newChannel({
uri: url,
loadUsingSystemPrincipal: true,
contentPolicyType: Ci.nsIContentPolicy.TYPE_DOCUMENT,
}).QueryInterface(Ci.nsIHttpChannel);
chan.loadFlags = Ci.nsIChannel.LOAD_INITIAL_DOCUMENT_URI;
return chan;
}
function channelOpenPromise(chan, flags) {
return new Promise(resolve => {
function finish(req, buffer) {
resolve([req, buffer]);
}
let internal = chan.QueryInterface(Ci.nsIHttpChannelInternal);
internal.setWaitForHTTPSSVCRecord();
chan.asyncOpen(new ChannelListener(finish, null, flags));
});
}
add_task(async function viewsourceExternalProtocol() {
Assert.throws(
() => makeChan(`view-source:${testScheme}:foo.example.com`),
/NS_ERROR_MALFORMED_URI/
);
});
add_task(async function viewsourceExternalProtocolRedirect() {
let httpserv = new HttpServer();
httpserv.registerPathHandler("/", function handler(metadata, response) {
response.setStatusLine(metadata.httpVersion, 301, "Moved Permanently");
response.setHeader("Location", `${testScheme}:foo@bar.com`, false);
var body = "Moved\n";
response.bodyOutputStream.write(body, body.length);
});
httpserv.start(-1);
let chan = makeChan(
`view-source:http://127.0.0.1:${httpserv.identity.primaryPort}/`
);
let [req] = await channelOpenPromise(chan, CL_EXPECT_FAILURE);
Assert.equal(req.status, Cr.NS_ERROR_MALFORMED_URI);
await httpserv.stop();
});
|
