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 |