1 // Copyright 2009 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "net/dns/dns_util.h"
6
7 #include <string_view>
8
9 #include "base/test/scoped_feature_list.h"
10 #include "net/dns/dns_test_util.h"
11 #include "net/dns/public/dns_over_https_config.h"
12 #include "net/dns/public/dns_protocol.h"
13 #include "net/dns/public/doh_provider_entry.h"
14 #include "testing/gmock/include/gmock/gmock.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16
17 namespace net {
18
19 namespace {
20 // Returns the DoH provider entry in `DohProviderEntry::GetList()` that matches
21 // `provider`. Crashes if there is no matching entry.
GetDohProviderEntry(std::string_view provider)22 const DohProviderEntry& GetDohProviderEntry(std::string_view provider) {
23 auto provider_list = DohProviderEntry::GetList();
24 auto it =
25 base::ranges::find(provider_list, provider, &DohProviderEntry::provider);
26 CHECK(it != provider_list.end());
27 return **it;
28 }
29 } // namespace
30
31 class DNSUtilTest : public testing::Test {};
32
TEST_F(DNSUtilTest,GetURLFromTemplateWithoutParameters)33 TEST_F(DNSUtilTest, GetURLFromTemplateWithoutParameters) {
34 EXPECT_EQ("https://dnsserver.example.net/dns-query",
35 GetURLFromTemplateWithoutParameters(
36 "https://dnsserver.example.net/dns-query{?dns}"));
37 }
38
TEST_F(DNSUtilTest,GetDohUpgradeServersFromDotHostname)39 TEST_F(DNSUtilTest, GetDohUpgradeServersFromDotHostname) {
40 std::vector<DnsOverHttpsServerConfig> doh_servers =
41 GetDohUpgradeServersFromDotHostname("");
42 EXPECT_EQ(0u, doh_servers.size());
43
44 doh_servers = GetDohUpgradeServersFromDotHostname("unrecognized");
45 EXPECT_EQ(0u, doh_servers.size());
46
47 doh_servers = GetDohUpgradeServersFromDotHostname(
48 "family-filter-dns.cleanbrowsing.org");
49 EXPECT_EQ(1u, doh_servers.size());
50 EXPECT_EQ("https://doh.cleanbrowsing.org/doh/family-filter{?dns}",
51 doh_servers[0].server_template());
52
53 base::test::ScopedFeatureList scoped_feature_list;
54 scoped_feature_list.InitWithFeatures(
55 /*enabled_features=*/{}, /*disabled_features=*/{
56 GetDohProviderEntry("CleanBrowsingFamily").feature});
57 doh_servers = GetDohUpgradeServersFromDotHostname(
58 "family-filter-dns.cleanbrowsing.org");
59 EXPECT_EQ(0u, doh_servers.size());
60 }
61
TEST_F(DNSUtilTest,GetDohUpgradeServersFromNameservers)62 TEST_F(DNSUtilTest, GetDohUpgradeServersFromNameservers) {
63 std::vector<IPEndPoint> nameservers;
64 // Cloudflare upgradeable IPs
65 IPAddress dns_ip0(1, 0, 0, 1);
66 IPAddress dns_ip1;
67 EXPECT_TRUE(dns_ip1.AssignFromIPLiteral("2606:4700:4700::1111"));
68 // SafeBrowsing family filter upgradeable IP
69 IPAddress dns_ip2;
70 EXPECT_TRUE(dns_ip2.AssignFromIPLiteral("2a0d:2a00:2::"));
71 // SafeBrowsing security filter upgradeable IP
72 IPAddress dns_ip3(185, 228, 169, 9);
73 // None-upgradeable IP
74 IPAddress dns_ip4(1, 2, 3, 4);
75
76 nameservers.emplace_back(dns_ip0, dns_protocol::kDefaultPort);
77 nameservers.emplace_back(dns_ip1, dns_protocol::kDefaultPort);
78 nameservers.emplace_back(dns_ip2, 54);
79 nameservers.emplace_back(dns_ip3, dns_protocol::kDefaultPort);
80 nameservers.emplace_back(dns_ip4, dns_protocol::kDefaultPort);
81
82 std::vector<DnsOverHttpsServerConfig> doh_servers =
83 GetDohUpgradeServersFromNameservers(std::vector<IPEndPoint>());
84 EXPECT_EQ(0u, doh_servers.size());
85
86 doh_servers = GetDohUpgradeServersFromNameservers(nameservers);
87 auto expected_config = *DnsOverHttpsConfig::FromTemplatesForTesting(
88 {"https://chrome.cloudflare-dns.com/dns-query",
89 "https://doh.cleanbrowsing.org/doh/family-filter{?dns}",
90 "https://doh.cleanbrowsing.org/doh/security-filter{?dns}"});
91 EXPECT_EQ(expected_config.servers(), doh_servers);
92
93 base::test::ScopedFeatureList scoped_feature_list;
94 scoped_feature_list.InitWithFeatures(
95 /*enabled_features=*/{},
96 /*disabled_features=*/{GetDohProviderEntry("CleanBrowsingSecure").feature,
97 GetDohProviderEntry("Cloudflare").feature});
98
99 doh_servers = GetDohUpgradeServersFromNameservers(nameservers);
100 EXPECT_THAT(doh_servers,
101 testing::ElementsAre(*DnsOverHttpsServerConfig::FromString(
102 "https://doh.cleanbrowsing.org/doh/family-filter{?dns}")));
103 }
104
TEST_F(DNSUtilTest,GetDohProviderIdForHistogramFromServerConfig)105 TEST_F(DNSUtilTest, GetDohProviderIdForHistogramFromServerConfig) {
106 EXPECT_EQ("Cloudflare",
107 GetDohProviderIdForHistogramFromServerConfig(
108 *DnsOverHttpsServerConfig::FromString(
109 "https://chrome.cloudflare-dns.com/dns-query")));
110 EXPECT_EQ("Other", GetDohProviderIdForHistogramFromServerConfig(
111 *DnsOverHttpsServerConfig::FromString(
112 "https://unexpected.dohserver.com/dns-query")));
113 }
114
TEST_F(DNSUtilTest,GetDohProviderIdForHistogramFromNameserver)115 TEST_F(DNSUtilTest, GetDohProviderIdForHistogramFromNameserver) {
116 EXPECT_EQ("CleanBrowsingSecure",
117 GetDohProviderIdForHistogramFromNameserver(IPEndPoint(
118 IPAddress(185, 228, 169, 9), dns_protocol::kDefaultPort)));
119 EXPECT_EQ("Other", GetDohProviderIdForHistogramFromNameserver(IPEndPoint(
120 IPAddress(1, 2, 3, 4), dns_protocol::kDefaultPort)));
121 }
122
123 } // namespace net
124