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