117 lines
3.4 KiB
JavaScript
117 lines
3.4 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";
|
|
|
|
const gOverride = Cc["@mozilla.org/network/native-dns-override;1"].getService(
|
|
Ci.nsINativeDNSResolverOverride
|
|
);
|
|
|
|
trr_test_setup();
|
|
registerCleanupFunction(async () => {
|
|
trr_clear_prefs();
|
|
});
|
|
|
|
let trrServer = null;
|
|
add_task(async function start_trr_server() {
|
|
trrServer = new TRRServer();
|
|
registerCleanupFunction(async () => {
|
|
await trrServer.stop();
|
|
});
|
|
await trrServer.start();
|
|
dump(`port = ${trrServer.port()}\n`);
|
|
Services.prefs.setBoolPref("network.trr.skip-AAAA-when-not-supported", false);
|
|
Services.prefs.setCharPref(
|
|
"network.trr.uri",
|
|
`https://foo.example.com:${trrServer.port()}/dns-query`
|
|
);
|
|
Services.prefs.setIntPref("network.trr.mode", Ci.nsIDNSService.MODE_TRRFIRST);
|
|
});
|
|
|
|
add_task(async function unspec_first() {
|
|
gOverride.clearOverrides();
|
|
Services.dns.clearCache(true);
|
|
|
|
gOverride.addIPOverride("example.org", "1.1.1.1");
|
|
gOverride.addIPOverride("example.org", "::1");
|
|
|
|
await trrServer.registerDoHAnswers("example.org", "A", {
|
|
answers: [
|
|
{
|
|
name: "example.org",
|
|
ttl: 55,
|
|
type: "A",
|
|
flush: false,
|
|
data: "1.2.3.4",
|
|
},
|
|
],
|
|
});
|
|
// This first request gets cached. IPv6 response gets served from the cache
|
|
await new TRRDNSListener("example.org", { expectedAnswer: "1.2.3.4" });
|
|
await new TRRDNSListener("example.org", {
|
|
flags: Ci.nsIDNSService.RESOLVE_DISABLE_IPV6,
|
|
expectedAnswer: "1.2.3.4",
|
|
});
|
|
let { inStatus } = await new TRRDNSListener("example.org", {
|
|
flags: Ci.nsIDNSService.RESOLVE_DISABLE_IPV4,
|
|
expectedSuccess: false,
|
|
});
|
|
equal(inStatus, Cr.NS_ERROR_UNKNOWN_HOST);
|
|
});
|
|
|
|
add_task(async function A_then_AAAA_fails() {
|
|
gOverride.clearOverrides();
|
|
Services.dns.clearCache(true);
|
|
|
|
gOverride.addIPOverride("example.org", "1.1.1.1");
|
|
gOverride.addIPOverride("example.org", "::1");
|
|
|
|
await trrServer.registerDoHAnswers("example.org", "A", {
|
|
answers: [
|
|
{
|
|
name: "example.org",
|
|
ttl: 55,
|
|
type: "A",
|
|
flush: false,
|
|
data: "1.2.3.4",
|
|
},
|
|
],
|
|
});
|
|
// We do individual IPv4/IPv6 requests - we expect IPv6 not to fallback to Do53 because we have an IPv4 record
|
|
await new TRRDNSListener("example.org", {
|
|
flags: Ci.nsIDNSService.RESOLVE_DISABLE_IPV6,
|
|
expectedAnswer: "1.2.3.4",
|
|
});
|
|
let { inStatus } = await new TRRDNSListener("example.org", {
|
|
flags: Ci.nsIDNSService.RESOLVE_DISABLE_IPV4,
|
|
expectedSuccess: false,
|
|
});
|
|
equal(inStatus, Cr.NS_ERROR_UNKNOWN_HOST);
|
|
});
|
|
|
|
add_task(async function just_AAAA_fails() {
|
|
gOverride.clearOverrides();
|
|
Services.dns.clearCache(true);
|
|
|
|
gOverride.addIPOverride("example.org", "1.1.1.1");
|
|
gOverride.addIPOverride("example.org", "::1");
|
|
|
|
await trrServer.registerDoHAnswers("example.org", "A", {
|
|
answers: [
|
|
{
|
|
name: "example.org",
|
|
ttl: 55,
|
|
type: "A",
|
|
flush: false,
|
|
data: "1.2.3.4",
|
|
},
|
|
],
|
|
});
|
|
// We only do an IPv6 req - we expect IPv6 not to fallback to Do53 because we have an IPv4 record
|
|
let { inStatus } = await new TRRDNSListener("example.org", {
|
|
flags: Ci.nsIDNSService.RESOLVE_DISABLE_IPV4,
|
|
expectedSuccess: false,
|
|
});
|
|
equal(inStatus, Cr.NS_ERROR_UNKNOWN_HOST);
|
|
});
|