| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/base/address_list.h" | 5 #include "net/base/address_list.h" |
| 6 | 6 |
| 7 #if defined(OS_POSIX) | |
| 8 #include <arpa/inet.h> // For inet_pton | |
| 9 #endif // defined(OS_POSIX) | |
| 10 | |
| 11 #include "base/string_util.h" | 7 #include "base/string_util.h" |
| 12 #include "net/base/host_resolver_proc.h" | 8 #include "net/base/host_resolver_proc.h" |
| 13 #include "net/base/net_util.h" | 9 #include "net/base/net_util.h" |
| 14 #include "net/base/sys_addrinfo.h" | 10 #include "net/base/sys_addrinfo.h" |
| 15 #if defined(OS_WIN) | 11 #if defined(OS_WIN) |
| 16 #include "net/base/winsock_init.h" | 12 #include "net/base/winsock_init.h" |
| 17 #endif | 13 #endif |
| 18 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 19 | 15 |
| 20 namespace { | 16 namespace { |
| 21 | 17 |
| 22 // Use getaddrinfo() to allocate an addrinfo structure. | 18 // Use getaddrinfo() to allocate an addrinfo structure. |
| 23 void CreateAddressList(const std::string& hostname, | 19 void CreateAddressList(const std::string& hostname, |
| 24 int port, | 20 net::AddressList* addrlist, int port) { |
| 25 net::AddressList* addrlist) { | |
| 26 #if defined(OS_WIN) | 21 #if defined(OS_WIN) |
| 27 net::EnsureWinsockInit(); | 22 net::EnsureWinsockInit(); |
| 28 #endif | 23 #endif |
| 29 int rv = SystemHostResolverProc(hostname, | 24 int rv = SystemHostResolverProc(hostname, |
| 30 net::ADDRESS_FAMILY_UNSPECIFIED, | 25 net::ADDRESS_FAMILY_UNSPECIFIED, |
| 31 0, | |
| 32 addrlist); | 26 addrlist); |
| 33 EXPECT_EQ(0, rv); | 27 EXPECT_EQ(0, rv); |
| 34 addrlist->SetPort(port); | 28 addrlist->SetPort(port); |
| 35 } | 29 } |
| 36 | 30 |
| 37 void CreateLongAddressList(net::AddressList* addrlist, int port) { | 31 void CreateLongAddressList(net::AddressList* addrlist, int port) { |
| 38 CreateAddressList("192.168.1.1", port, addrlist); | 32 CreateAddressList("192.168.1.1", addrlist, port); |
| 39 net::AddressList second_list; | 33 net::AddressList second_list; |
| 40 CreateAddressList("192.168.1.2", port, &second_list); | 34 CreateAddressList("192.168.1.2", &second_list, port); |
| 41 addrlist->Append(second_list.head()); | 35 addrlist->Append(second_list.head()); |
| 42 } | 36 } |
| 43 | 37 |
| 44 TEST(AddressListTest, GetPort) { | 38 TEST(AddressListTest, GetPort) { |
| 45 net::AddressList addrlist; | 39 net::AddressList addrlist; |
| 46 CreateAddressList("192.168.1.1", 81, &addrlist); | 40 CreateAddressList("192.168.1.1", &addrlist, 81); |
| 47 EXPECT_EQ(81, addrlist.GetPort()); | 41 EXPECT_EQ(81, addrlist.GetPort()); |
| 48 | 42 |
| 49 addrlist.SetPort(83); | 43 addrlist.SetPort(83); |
| 50 EXPECT_EQ(83, addrlist.GetPort()); | 44 EXPECT_EQ(83, addrlist.GetPort()); |
| 51 } | 45 } |
| 52 | 46 |
| 53 TEST(AddressListTest, Assignment) { | 47 TEST(AddressListTest, Assignment) { |
| 54 net::AddressList addrlist1; | 48 net::AddressList addrlist1; |
| 55 CreateAddressList("192.168.1.1", 85, &addrlist1); | 49 CreateAddressList("192.168.1.1", &addrlist1, 85); |
| 56 EXPECT_EQ(85, addrlist1.GetPort()); | 50 EXPECT_EQ(85, addrlist1.GetPort()); |
| 57 | 51 |
| 58 // Should reference the same data as addrlist1 -- so when we change addrlist1 | 52 // Should reference the same data as addrlist1 -- so when we change addrlist1 |
| 59 // both are changed. | 53 // both are changed. |
| 60 net::AddressList addrlist2 = addrlist1; | 54 net::AddressList addrlist2 = addrlist1; |
| 61 EXPECT_EQ(85, addrlist2.GetPort()); | 55 EXPECT_EQ(85, addrlist2.GetPort()); |
| 62 | 56 |
| 63 addrlist1.SetPort(80); | 57 addrlist1.SetPort(80); |
| 64 EXPECT_EQ(80, addrlist1.GetPort()); | 58 EXPECT_EQ(80, addrlist1.GetPort()); |
| 65 EXPECT_EQ(80, addrlist2.GetPort()); | 59 EXPECT_EQ(80, addrlist2.GetPort()); |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 104 // Changes to addrlist1 are not reflected in addrlist2. | 98 // Changes to addrlist1 are not reflected in addrlist2. |
| 105 addrlist1.SetPort(70); | 99 addrlist1.SetPort(70); |
| 106 addrlist2.SetPort(90); | 100 addrlist2.SetPort(90); |
| 107 | 101 |
| 108 EXPECT_EQ(70, addrlist1.GetPort()); | 102 EXPECT_EQ(70, addrlist1.GetPort()); |
| 109 EXPECT_EQ(90, addrlist2.GetPort()); | 103 EXPECT_EQ(90, addrlist2.GetPort()); |
| 110 } | 104 } |
| 111 | 105 |
| 112 TEST(AddressListTest, Append) { | 106 TEST(AddressListTest, Append) { |
| 113 net::AddressList addrlist1; | 107 net::AddressList addrlist1; |
| 114 CreateAddressList("192.168.1.1", 11, &addrlist1); | 108 CreateAddressList("192.168.1.1", &addrlist1, 11); |
| 115 EXPECT_EQ(11, addrlist1.GetPort()); | 109 EXPECT_EQ(11, addrlist1.GetPort()); |
| 116 net::AddressList addrlist2; | 110 net::AddressList addrlist2; |
| 117 CreateAddressList("192.168.1.2", 12, &addrlist2); | 111 CreateAddressList("192.168.1.2", &addrlist2, 12); |
| 118 EXPECT_EQ(12, addrlist2.GetPort()); | 112 EXPECT_EQ(12, addrlist2.GetPort()); |
| 119 | 113 |
| 120 ASSERT_TRUE(addrlist1.head()->ai_next == NULL); | 114 ASSERT_TRUE(addrlist1.head()->ai_next == NULL); |
| 121 addrlist1.Append(addrlist2.head()); | 115 addrlist1.Append(addrlist2.head()); |
| 122 ASSERT_TRUE(addrlist1.head()->ai_next != NULL); | 116 ASSERT_TRUE(addrlist1.head()->ai_next != NULL); |
| 123 | 117 |
| 124 net::AddressList addrlist3; | 118 net::AddressList addrlist3; |
| 125 addrlist3.Copy(addrlist1.head()->ai_next, false); | 119 addrlist3.Copy(addrlist1.head()->ai_next, false); |
| 126 EXPECT_EQ(12, addrlist3.GetPort()); | 120 EXPECT_EQ(12, addrlist3.GetPort()); |
| 127 } | 121 } |
| 128 | 122 |
| 129 static const char* kCanonicalHost = "canonical.bar.com"; | |
| 130 | |
| 131 TEST(AddressListTest, Canonical) { | |
| 132 // Create an addrinfo with a canonical name. | |
| 133 sockaddr_in address; | |
| 134 EXPECT_EQ(1, inet_pton(AF_INET, "10.0.0.1", &address)); | |
| 135 struct addrinfo ai; | |
| 136 memset(&ai, 0x0, sizeof(ai)); | |
| 137 ai.ai_family = AF_INET; | |
| 138 ai.ai_socktype = SOCK_STREAM; | |
| 139 ai.ai_addrlen = sizeof(address); | |
| 140 ai.ai_addr = reinterpret_cast<sockaddr*>(&address); | |
| 141 ai.ai_canonname = const_cast<char *>(kCanonicalHost); | |
| 142 | |
| 143 // Copy the addrinfo struct into an AddressList object and | |
| 144 // make sure it seems correct. | |
| 145 net::AddressList addrlist1; | |
| 146 addrlist1.Copy(&ai, true); | |
| 147 const struct addrinfo* addrinfo1 = addrlist1.head(); | |
| 148 EXPECT_TRUE(addrinfo1 != NULL); | |
| 149 EXPECT_TRUE(addrinfo1->ai_next == NULL); | |
| 150 std::string canon_name1; | |
| 151 EXPECT_TRUE(addrlist1.GetCanonicalName(&canon_name1)); | |
| 152 EXPECT_EQ("canonical.bar.com", canon_name1); | |
| 153 | |
| 154 // Copy the AddressList to another one. | |
| 155 net::AddressList addrlist2; | |
| 156 addrlist2.Copy(addrinfo1, true); | |
| 157 const struct addrinfo* addrinfo2 = addrlist2.head(); | |
| 158 EXPECT_TRUE(addrinfo2 != NULL); | |
| 159 EXPECT_TRUE(addrinfo2->ai_next == NULL); | |
| 160 EXPECT_TRUE(addrinfo2->ai_canonname != NULL); | |
| 161 EXPECT_NE(addrinfo1, addrinfo2); | |
| 162 EXPECT_NE(addrinfo1->ai_canonname, addrinfo2->ai_canonname); | |
| 163 std::string canon_name2; | |
| 164 EXPECT_TRUE(addrlist2.GetCanonicalName(&canon_name2)); | |
| 165 EXPECT_EQ("canonical.bar.com", canon_name2); | |
| 166 | |
| 167 // Make sure that GetCanonicalName correctly returns false | |
| 168 // when ai_canonname is NULL. | |
| 169 ai.ai_canonname = NULL; | |
| 170 net::AddressList addrlist_no_canon; | |
| 171 addrlist_no_canon.Copy(&ai, true); | |
| 172 std::string canon_name3 = "blah"; | |
| 173 EXPECT_FALSE(addrlist_no_canon.GetCanonicalName(&canon_name3)); | |
| 174 EXPECT_EQ("blah", canon_name3); | |
| 175 } | |
| 176 | |
| 177 } // namespace | 123 } // namespace |
| OLD | NEW |