OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/socket/tcp_server_socket.h" | 5 #include "net/socket/tcp_server_socket.h" |
6 | 6 |
| 7 #include <memory> |
7 #include <string> | 8 #include <string> |
8 #include <vector> | 9 #include <vector> |
9 | 10 |
10 #include "base/compiler_specific.h" | 11 #include "base/compiler_specific.h" |
11 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
12 #include "base/memory/scoped_ptr.h" | |
13 #include "net/base/address_list.h" | 13 #include "net/base/address_list.h" |
14 #include "net/base/io_buffer.h" | 14 #include "net/base/io_buffer.h" |
15 #include "net/base/ip_address.h" | 15 #include "net/base/ip_address.h" |
16 #include "net/base/ip_endpoint.h" | 16 #include "net/base/ip_endpoint.h" |
17 #include "net/base/net_errors.h" | 17 #include "net/base/net_errors.h" |
18 #include "net/base/test_completion_callback.h" | 18 #include "net/base/test_completion_callback.h" |
19 #include "net/socket/tcp_client_socket.h" | 19 #include "net/socket/tcp_client_socket.h" |
20 #include "testing/gtest/include/gtest/gtest.h" | 20 #include "testing/gtest/include/gtest/gtest.h" |
21 #include "testing/platform_test.h" | 21 #include "testing/platform_test.h" |
22 | 22 |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
65 | 65 |
66 TEST_F(TCPServerSocketTest, Accept) { | 66 TEST_F(TCPServerSocketTest, Accept) { |
67 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); | 67 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); |
68 | 68 |
69 TestCompletionCallback connect_callback; | 69 TestCompletionCallback connect_callback; |
70 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, | 70 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, |
71 NetLog::Source()); | 71 NetLog::Source()); |
72 connecting_socket.Connect(connect_callback.callback()); | 72 connecting_socket.Connect(connect_callback.callback()); |
73 | 73 |
74 TestCompletionCallback accept_callback; | 74 TestCompletionCallback accept_callback; |
75 scoped_ptr<StreamSocket> accepted_socket; | 75 std::unique_ptr<StreamSocket> accepted_socket; |
76 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); | 76 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); |
77 if (result == ERR_IO_PENDING) | 77 if (result == ERR_IO_PENDING) |
78 result = accept_callback.WaitForResult(); | 78 result = accept_callback.WaitForResult(); |
79 ASSERT_EQ(OK, result); | 79 ASSERT_EQ(OK, result); |
80 | 80 |
81 ASSERT_TRUE(accepted_socket.get() != NULL); | 81 ASSERT_TRUE(accepted_socket.get() != NULL); |
82 | 82 |
83 // Both sockets should be on the loopback network interface. | 83 // Both sockets should be on the loopback network interface. |
84 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), | 84 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), |
85 local_address_.address()); | 85 local_address_.address()); |
86 | 86 |
87 EXPECT_EQ(OK, connect_callback.WaitForResult()); | 87 EXPECT_EQ(OK, connect_callback.WaitForResult()); |
88 } | 88 } |
89 | 89 |
90 // Test Accept() callback. | 90 // Test Accept() callback. |
91 TEST_F(TCPServerSocketTest, AcceptAsync) { | 91 TEST_F(TCPServerSocketTest, AcceptAsync) { |
92 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); | 92 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); |
93 | 93 |
94 TestCompletionCallback accept_callback; | 94 TestCompletionCallback accept_callback; |
95 scoped_ptr<StreamSocket> accepted_socket; | 95 std::unique_ptr<StreamSocket> accepted_socket; |
96 | 96 |
97 ASSERT_EQ(ERR_IO_PENDING, | 97 ASSERT_EQ(ERR_IO_PENDING, |
98 socket_.Accept(&accepted_socket, accept_callback.callback())); | 98 socket_.Accept(&accepted_socket, accept_callback.callback())); |
99 | 99 |
100 TestCompletionCallback connect_callback; | 100 TestCompletionCallback connect_callback; |
101 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, | 101 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, |
102 NetLog::Source()); | 102 NetLog::Source()); |
103 connecting_socket.Connect(connect_callback.callback()); | 103 connecting_socket.Connect(connect_callback.callback()); |
104 | 104 |
105 EXPECT_EQ(OK, connect_callback.WaitForResult()); | 105 EXPECT_EQ(OK, connect_callback.WaitForResult()); |
106 EXPECT_EQ(OK, accept_callback.WaitForResult()); | 106 EXPECT_EQ(OK, accept_callback.WaitForResult()); |
107 | 107 |
108 EXPECT_TRUE(accepted_socket != NULL); | 108 EXPECT_TRUE(accepted_socket != NULL); |
109 | 109 |
110 // Both sockets should be on the loopback network interface. | 110 // Both sockets should be on the loopback network interface. |
111 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), | 111 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), |
112 local_address_.address()); | 112 local_address_.address()); |
113 } | 113 } |
114 | 114 |
115 // Accept two connections simultaneously. | 115 // Accept two connections simultaneously. |
116 TEST_F(TCPServerSocketTest, Accept2Connections) { | 116 TEST_F(TCPServerSocketTest, Accept2Connections) { |
117 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); | 117 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); |
118 | 118 |
119 TestCompletionCallback accept_callback; | 119 TestCompletionCallback accept_callback; |
120 scoped_ptr<StreamSocket> accepted_socket; | 120 std::unique_ptr<StreamSocket> accepted_socket; |
121 | 121 |
122 ASSERT_EQ(ERR_IO_PENDING, | 122 ASSERT_EQ(ERR_IO_PENDING, |
123 socket_.Accept(&accepted_socket, accept_callback.callback())); | 123 socket_.Accept(&accepted_socket, accept_callback.callback())); |
124 | 124 |
125 TestCompletionCallback connect_callback; | 125 TestCompletionCallback connect_callback; |
126 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, | 126 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, |
127 NetLog::Source()); | 127 NetLog::Source()); |
128 connecting_socket.Connect(connect_callback.callback()); | 128 connecting_socket.Connect(connect_callback.callback()); |
129 | 129 |
130 TestCompletionCallback connect_callback2; | 130 TestCompletionCallback connect_callback2; |
131 TCPClientSocket connecting_socket2(local_address_list(), NULL, NULL, | 131 TCPClientSocket connecting_socket2(local_address_list(), NULL, NULL, |
132 NetLog::Source()); | 132 NetLog::Source()); |
133 connecting_socket2.Connect(connect_callback2.callback()); | 133 connecting_socket2.Connect(connect_callback2.callback()); |
134 | 134 |
135 EXPECT_EQ(OK, accept_callback.WaitForResult()); | 135 EXPECT_EQ(OK, accept_callback.WaitForResult()); |
136 | 136 |
137 TestCompletionCallback accept_callback2; | 137 TestCompletionCallback accept_callback2; |
138 scoped_ptr<StreamSocket> accepted_socket2; | 138 std::unique_ptr<StreamSocket> accepted_socket2; |
139 int result = socket_.Accept(&accepted_socket2, accept_callback2.callback()); | 139 int result = socket_.Accept(&accepted_socket2, accept_callback2.callback()); |
140 if (result == ERR_IO_PENDING) | 140 if (result == ERR_IO_PENDING) |
141 result = accept_callback2.WaitForResult(); | 141 result = accept_callback2.WaitForResult(); |
142 ASSERT_EQ(OK, result); | 142 ASSERT_EQ(OK, result); |
143 | 143 |
144 EXPECT_EQ(OK, connect_callback.WaitForResult()); | 144 EXPECT_EQ(OK, connect_callback.WaitForResult()); |
145 | 145 |
146 EXPECT_TRUE(accepted_socket != NULL); | 146 EXPECT_TRUE(accepted_socket != NULL); |
147 EXPECT_TRUE(accepted_socket2 != NULL); | 147 EXPECT_TRUE(accepted_socket2 != NULL); |
148 EXPECT_NE(accepted_socket.get(), accepted_socket2.get()); | 148 EXPECT_NE(accepted_socket.get(), accepted_socket2.get()); |
149 | 149 |
150 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), | 150 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), |
151 local_address_.address()); | 151 local_address_.address()); |
152 EXPECT_EQ(GetPeerAddress(accepted_socket2.get()).address(), | 152 EXPECT_EQ(GetPeerAddress(accepted_socket2.get()).address(), |
153 local_address_.address()); | 153 local_address_.address()); |
154 } | 154 } |
155 | 155 |
156 TEST_F(TCPServerSocketTest, AcceptIPv6) { | 156 TEST_F(TCPServerSocketTest, AcceptIPv6) { |
157 bool initialized = false; | 157 bool initialized = false; |
158 ASSERT_NO_FATAL_FAILURE(SetUpIPv6(&initialized)); | 158 ASSERT_NO_FATAL_FAILURE(SetUpIPv6(&initialized)); |
159 if (!initialized) | 159 if (!initialized) |
160 return; | 160 return; |
161 | 161 |
162 TestCompletionCallback connect_callback; | 162 TestCompletionCallback connect_callback; |
163 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, | 163 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, |
164 NetLog::Source()); | 164 NetLog::Source()); |
165 connecting_socket.Connect(connect_callback.callback()); | 165 connecting_socket.Connect(connect_callback.callback()); |
166 | 166 |
167 TestCompletionCallback accept_callback; | 167 TestCompletionCallback accept_callback; |
168 scoped_ptr<StreamSocket> accepted_socket; | 168 std::unique_ptr<StreamSocket> accepted_socket; |
169 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); | 169 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); |
170 if (result == ERR_IO_PENDING) | 170 if (result == ERR_IO_PENDING) |
171 result = accept_callback.WaitForResult(); | 171 result = accept_callback.WaitForResult(); |
172 ASSERT_EQ(OK, result); | 172 ASSERT_EQ(OK, result); |
173 | 173 |
174 ASSERT_TRUE(accepted_socket.get() != NULL); | 174 ASSERT_TRUE(accepted_socket.get() != NULL); |
175 | 175 |
176 // Both sockets should be on the loopback network interface. | 176 // Both sockets should be on the loopback network interface. |
177 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), | 177 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), |
178 local_address_.address()); | 178 local_address_.address()); |
179 | 179 |
180 EXPECT_EQ(OK, connect_callback.WaitForResult()); | 180 EXPECT_EQ(OK, connect_callback.WaitForResult()); |
181 } | 181 } |
182 | 182 |
183 TEST_F(TCPServerSocketTest, AcceptIO) { | 183 TEST_F(TCPServerSocketTest, AcceptIO) { |
184 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); | 184 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); |
185 | 185 |
186 TestCompletionCallback connect_callback; | 186 TestCompletionCallback connect_callback; |
187 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, | 187 TCPClientSocket connecting_socket(local_address_list(), NULL, NULL, |
188 NetLog::Source()); | 188 NetLog::Source()); |
189 connecting_socket.Connect(connect_callback.callback()); | 189 connecting_socket.Connect(connect_callback.callback()); |
190 | 190 |
191 TestCompletionCallback accept_callback; | 191 TestCompletionCallback accept_callback; |
192 scoped_ptr<StreamSocket> accepted_socket; | 192 std::unique_ptr<StreamSocket> accepted_socket; |
193 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); | 193 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); |
194 ASSERT_EQ(OK, accept_callback.GetResult(result)); | 194 ASSERT_EQ(OK, accept_callback.GetResult(result)); |
195 | 195 |
196 ASSERT_TRUE(accepted_socket.get() != NULL); | 196 ASSERT_TRUE(accepted_socket.get() != NULL); |
197 | 197 |
198 // Both sockets should be on the loopback network interface. | 198 // Both sockets should be on the loopback network interface. |
199 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), | 199 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), |
200 local_address_.address()); | 200 local_address_.address()); |
201 | 201 |
202 EXPECT_EQ(OK, connect_callback.WaitForResult()); | 202 EXPECT_EQ(OK, connect_callback.WaitForResult()); |
(...skipping 30 matching lines...) Expand all Loading... |
233 bytes_read += read_result; | 233 bytes_read += read_result; |
234 } | 234 } |
235 | 235 |
236 std::string received_message(buffer.begin(), buffer.end()); | 236 std::string received_message(buffer.begin(), buffer.end()); |
237 ASSERT_EQ(message, received_message); | 237 ASSERT_EQ(message, received_message); |
238 } | 238 } |
239 | 239 |
240 } // namespace | 240 } // namespace |
241 | 241 |
242 } // namespace net | 242 } // namespace net |
OLD | NEW |