зеркало из https://github.com/mozilla/gecko-dev.git
96 строки
2.6 KiB
JavaScript
96 строки
2.6 KiB
JavaScript
/* 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";
|
|
|
|
var prefs = Services.prefs;
|
|
|
|
function setup() {
|
|
prefs.setBoolPref("network.dns.notifyResolution", true);
|
|
prefs.setCharPref("network.proxy.socks", "127.0.0.1");
|
|
prefs.setIntPref("network.proxy.socks_port", 9000);
|
|
prefs.setIntPref("network.proxy.type", 1);
|
|
prefs.setBoolPref("network.proxy.socks_remote_dns", true);
|
|
}
|
|
|
|
setup();
|
|
registerCleanupFunction(async () => {
|
|
prefs.clearUserPref("network.proxy.socks");
|
|
prefs.clearUserPref("network.proxy.socks_port");
|
|
prefs.clearUserPref("network.proxy.type");
|
|
prefs.clearUserPref("network.proxy.socks_remote_dns");
|
|
prefs.clearUserPref("network.dns.notifyResolution");
|
|
});
|
|
|
|
var url = "ws://dnsleak.example.com";
|
|
|
|
var dnsRequestObserver = {
|
|
register() {
|
|
this.obs = Services.obs;
|
|
this.obs.addObserver(this, "dns-resolution-request");
|
|
},
|
|
|
|
unregister() {
|
|
if (this.obs) {
|
|
this.obs.removeObserver(this, "dns-resolution-request");
|
|
}
|
|
},
|
|
|
|
observe(subject, topic, data) {
|
|
if (topic == "dns-resolution-request") {
|
|
Assert.ok(!data.includes("dnsleak.example.com"), `no dnsleak: ${data}`);
|
|
}
|
|
},
|
|
};
|
|
|
|
function WSListener(closure) {
|
|
this._closure = closure;
|
|
}
|
|
WSListener.prototype = {
|
|
onAcknowledge(aContext, aSize) {},
|
|
onBinaryMessageAvailable(aContext, aMsg) {},
|
|
onMessageAvailable(aContext, aMsg) {},
|
|
onServerClose(aContext, aCode, aReason) {},
|
|
onStart(aContext) {},
|
|
onStop(aContext, aStatusCode) {
|
|
dnsRequestObserver.unregister();
|
|
this._closure();
|
|
},
|
|
};
|
|
|
|
add_task(async function test_dns_websocket_channel() {
|
|
dnsRequestObserver.register();
|
|
|
|
var chan = Cc["@mozilla.org/network/protocol;1?name=ws"].createInstance(
|
|
Ci.nsIWebSocketChannel
|
|
);
|
|
|
|
var uri = Services.io.newURI(url);
|
|
chan.initLoadInfo(
|
|
null, // aLoadingNode
|
|
Services.scriptSecurityManager.createContentPrincipal(uri, {}),
|
|
null, // aTriggeringPrincipal
|
|
Ci.nsILoadInfo.SEC_ALLOW_CROSS_ORIGIN_SEC_CONTEXT_IS_NULL,
|
|
Ci.nsIContentPolicy.TYPE_WEBSOCKET
|
|
);
|
|
|
|
await new Promise(resolve =>
|
|
chan.asyncOpen(uri, url, {}, 0, new WSListener(resolve), null)
|
|
);
|
|
});
|
|
|
|
add_task(async function test_dns_resolve_proxy() {
|
|
dnsRequestObserver.register();
|
|
|
|
let { error } = await new TRRDNSListener("dnsleak.example.com", {
|
|
expectEarlyFail: true,
|
|
});
|
|
Assert.equal(
|
|
error.result,
|
|
Cr.NS_ERROR_UNKNOWN_PROXY_HOST,
|
|
"error is NS_ERROR_UNKNOWN_PROXY_HOST"
|
|
);
|
|
dnsRequestObserver.unregister();
|
|
});
|