1 // Copyright 2020 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_alias_utility.h"
6
7 #include <string>
8 #include <vector>
9
10 #include "net/dns/public/dns_protocol.h"
11 #include "testing/gmock/include/gmock/gmock.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13
14 namespace net {
15 namespace {
16
TEST(DnsAliasUtilityTest,FixUpDnsAliases)17 TEST(DnsAliasUtilityTest, FixUpDnsAliases) {
18 // TODO(https://crbug.com/1416013) ' ' (0x20: SPACE) should not be escaped.
19 const std::set<std::string> kAliases = {
20 "localhost", "1.2.3.4", "a.com", "", "test",
21 "0", "[::1]", "::1", "-www.e.com", "alias.com",
22 "s .de", "www-1", "2a", "a-", "b..net",
23 "a.com", "b_o.org", "alias.com", "1..3.2", "1.2.3.09",
24 "foo.4", "a,b,c", "f/g", "www?", "[3a2:401f::1]",
25 "0.0.1.2", "a.b.com", "c.org", "123.tld", "d-e.net",
26 "f__g", "h"};
27 const std::set<std::string> kExpectedFixedUpAliases = {
28 "a.com", "test", "-www.e.com", "alias.com", "s%20.de", "www-1",
29 "2a", "a-", "b_o.org", "a,b,c", "a.b.com", "c.org",
30 "123.tld", "d-e.net", "f__g", "h"};
31
32 std::set<std::string> fixed_up_aliases =
33 dns_alias_utility::FixUpDnsAliases(kAliases);
34 EXPECT_EQ(kExpectedFixedUpAliases, fixed_up_aliases);
35
36 std::string long_unqualified_alias =
37 "abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi."
38 "abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi."
39 "abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi."
40 "abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi."
41 "abcdefghi.abcd";
42 std::string long_qualified_alias =
43 "abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi."
44 "abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi."
45 "abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi."
46 "abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi.abcdefghi."
47 "abcdefghi.abc.";
48 std::set<std::string> set_with_long_aliases(
49 {long_unqualified_alias, long_qualified_alias});
50
51 std::set<std::string> fixed_up_set_with_long_aliases =
52 dns_alias_utility::FixUpDnsAliases(set_with_long_aliases);
53 EXPECT_THAT(fixed_up_set_with_long_aliases,
54 testing::ElementsAre(long_qualified_alias));
55
56 std::set<std::string> empty_aliases;
57 std::set<std::string> fixed_up_empty_aliases =
58 dns_alias_utility::FixUpDnsAliases(empty_aliases);
59 EXPECT_TRUE(fixed_up_empty_aliases.empty());
60 }
61
62 } // namespace
63 } // namespace net
64