83 lines
2.5 KiB
JavaScript
83 lines
2.5 KiB
JavaScript
/* -*- 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]);
|
|
}
|
|
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();
|
|
});
|