OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "base/compiler_specific.h" | 7 #include "base/compiler_specific.h" |
8 #include "base/scoped_ptr.h" | 8 #include "base/scoped_ptr.h" |
9 #include "net/base/address_list.h" | 9 #include "net/base/address_list.h" |
10 #include "net/base/ip_endpoint.h" | 10 #include "net/base/ip_endpoint.h" |
(...skipping 23 matching lines...) Expand all Loading... |
34 } | 34 } |
35 | 35 |
36 void ParseAddress(std::string ip_str, int port, IPEndPoint* address) { | 36 void ParseAddress(std::string ip_str, int port, IPEndPoint* address) { |
37 IPAddressNumber ip_number; | 37 IPAddressNumber ip_number; |
38 bool rv = ParseIPLiteralToNumber(ip_str, &ip_number); | 38 bool rv = ParseIPLiteralToNumber(ip_str, &ip_number); |
39 if (!rv) | 39 if (!rv) |
40 return; | 40 return; |
41 *address = IPEndPoint(ip_number, port); | 41 *address = IPEndPoint(ip_number, port); |
42 } | 42 } |
43 | 43 |
44 static IPEndPoint GetPeerAddress(ClientSocket* socket) { | 44 static IPEndPoint GetPeerAddress(StreamSocket* socket) { |
45 AddressList address; | 45 AddressList address; |
46 EXPECT_EQ(OK, socket->GetPeerAddress(&address)); | 46 EXPECT_EQ(OK, socket->GetPeerAddress(&address)); |
47 IPEndPoint endpoint; | 47 IPEndPoint endpoint; |
48 EXPECT_TRUE(endpoint.FromSockAddr( | 48 EXPECT_TRUE(endpoint.FromSockAddr( |
49 address.head()->ai_addr, address.head()->ai_addrlen)); | 49 address.head()->ai_addr, address.head()->ai_addrlen)); |
50 return endpoint; | 50 return endpoint; |
51 } | 51 } |
52 | 52 |
53 TCPServerSocket socket_; | 53 TCPServerSocket socket_; |
54 IPEndPoint local_address_; | 54 IPEndPoint local_address_; |
55 }; | 55 }; |
56 | 56 |
57 TEST_F(TCPServerSocketTest, Accept) { | 57 TEST_F(TCPServerSocketTest, Accept) { |
58 TestCompletionCallback connect_callback; | 58 TestCompletionCallback connect_callback; |
59 TCPClientSocket connecting_socket(AddressList(local_address_.address(), | 59 TCPClientSocket connecting_socket(AddressList(local_address_.address(), |
60 local_address_.port(), false), | 60 local_address_.port(), false), |
61 NULL, NetLog::Source()); | 61 NULL, NetLog::Source()); |
62 connecting_socket.Connect(&connect_callback); | 62 connecting_socket.Connect(&connect_callback); |
63 | 63 |
64 TestCompletionCallback accept_callback; | 64 TestCompletionCallback accept_callback; |
65 scoped_ptr<ClientSocket> accepted_socket; | 65 scoped_ptr<StreamSocket> accepted_socket; |
66 int result = socket_.Accept(&accepted_socket, &accept_callback); | 66 int result = socket_.Accept(&accepted_socket, &accept_callback); |
67 if (result == ERR_IO_PENDING) | 67 if (result == ERR_IO_PENDING) |
68 result = accept_callback.WaitForResult(); | 68 result = accept_callback.WaitForResult(); |
69 ASSERT_EQ(OK, result); | 69 ASSERT_EQ(OK, result); |
70 | 70 |
71 ASSERT_TRUE(accepted_socket.get() != NULL); | 71 ASSERT_TRUE(accepted_socket.get() != NULL); |
72 | 72 |
73 // Both sockets should be on the loopback network interface. | 73 // Both sockets should be on the loopback network interface. |
74 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), | 74 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), |
75 local_address_.address()); | 75 local_address_.address()); |
76 | 76 |
77 EXPECT_EQ(OK, connect_callback.WaitForResult()); | 77 EXPECT_EQ(OK, connect_callback.WaitForResult()); |
78 } | 78 } |
79 | 79 |
80 // Test Accept() callback. | 80 // Test Accept() callback. |
81 TEST_F(TCPServerSocketTest, AcceptAsync) { | 81 TEST_F(TCPServerSocketTest, AcceptAsync) { |
82 TestCompletionCallback accept_callback; | 82 TestCompletionCallback accept_callback; |
83 scoped_ptr<ClientSocket> accepted_socket; | 83 scoped_ptr<StreamSocket> accepted_socket; |
84 | 84 |
85 ASSERT_EQ(ERR_IO_PENDING, socket_.Accept(&accepted_socket, &accept_callback)); | 85 ASSERT_EQ(ERR_IO_PENDING, socket_.Accept(&accepted_socket, &accept_callback)); |
86 | 86 |
87 TestCompletionCallback connect_callback; | 87 TestCompletionCallback connect_callback; |
88 TCPClientSocket connecting_socket(AddressList(local_address_.address(), | 88 TCPClientSocket connecting_socket(AddressList(local_address_.address(), |
89 local_address_.port(), false), | 89 local_address_.port(), false), |
90 NULL, NetLog::Source()); | 90 NULL, NetLog::Source()); |
91 connecting_socket.Connect(&connect_callback); | 91 connecting_socket.Connect(&connect_callback); |
92 | 92 |
93 EXPECT_EQ(OK, connect_callback.WaitForResult()); | 93 EXPECT_EQ(OK, connect_callback.WaitForResult()); |
94 EXPECT_EQ(OK, accept_callback.WaitForResult()); | 94 EXPECT_EQ(OK, accept_callback.WaitForResult()); |
95 | 95 |
96 EXPECT_TRUE(accepted_socket != NULL); | 96 EXPECT_TRUE(accepted_socket != NULL); |
97 | 97 |
98 // Both sockets should be on the loopback network interface. | 98 // Both sockets should be on the loopback network interface. |
99 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), | 99 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), |
100 local_address_.address()); | 100 local_address_.address()); |
101 } | 101 } |
102 | 102 |
103 // Accept two connections simultaneously. | 103 // Accept two connections simultaneously. |
104 TEST_F(TCPServerSocketTest, Accept2Connections) { | 104 TEST_F(TCPServerSocketTest, Accept2Connections) { |
105 TestCompletionCallback accept_callback; | 105 TestCompletionCallback accept_callback; |
106 scoped_ptr<ClientSocket> accepted_socket; | 106 scoped_ptr<StreamSocket> accepted_socket; |
107 | 107 |
108 ASSERT_EQ(ERR_IO_PENDING, | 108 ASSERT_EQ(ERR_IO_PENDING, |
109 socket_.Accept(&accepted_socket, &accept_callback)); | 109 socket_.Accept(&accepted_socket, &accept_callback)); |
110 | 110 |
111 TestCompletionCallback connect_callback; | 111 TestCompletionCallback connect_callback; |
112 TCPClientSocket connecting_socket(AddressList(local_address_.address(), | 112 TCPClientSocket connecting_socket(AddressList(local_address_.address(), |
113 local_address_.port(), false), | 113 local_address_.port(), false), |
114 NULL, NetLog::Source()); | 114 NULL, NetLog::Source()); |
115 connecting_socket.Connect(&connect_callback); | 115 connecting_socket.Connect(&connect_callback); |
116 | 116 |
117 TestCompletionCallback connect_callback2; | 117 TestCompletionCallback connect_callback2; |
118 TCPClientSocket connecting_socket2(AddressList(local_address_.address(), | 118 TCPClientSocket connecting_socket2(AddressList(local_address_.address(), |
119 local_address_.port(), false), | 119 local_address_.port(), false), |
120 NULL, NetLog::Source()); | 120 NULL, NetLog::Source()); |
121 connecting_socket2.Connect(&connect_callback2); | 121 connecting_socket2.Connect(&connect_callback2); |
122 | 122 |
123 EXPECT_EQ(OK, accept_callback.WaitForResult()); | 123 EXPECT_EQ(OK, accept_callback.WaitForResult()); |
124 | 124 |
125 TestCompletionCallback accept_callback2; | 125 TestCompletionCallback accept_callback2; |
126 scoped_ptr<ClientSocket> accepted_socket2; | 126 scoped_ptr<StreamSocket> accepted_socket2; |
127 int result = socket_.Accept(&accepted_socket2, &accept_callback2); | 127 int result = socket_.Accept(&accepted_socket2, &accept_callback2); |
128 if (result == ERR_IO_PENDING) | 128 if (result == ERR_IO_PENDING) |
129 result = accept_callback2.WaitForResult(); | 129 result = accept_callback2.WaitForResult(); |
130 ASSERT_EQ(OK, result); | 130 ASSERT_EQ(OK, result); |
131 | 131 |
132 EXPECT_EQ(OK, connect_callback.WaitForResult()); | 132 EXPECT_EQ(OK, connect_callback.WaitForResult()); |
133 | 133 |
134 EXPECT_TRUE(accepted_socket != NULL); | 134 EXPECT_TRUE(accepted_socket != NULL); |
135 EXPECT_TRUE(accepted_socket2 != NULL); | 135 EXPECT_TRUE(accepted_socket2 != NULL); |
136 EXPECT_NE(accepted_socket.get(), accepted_socket2.get()); | 136 EXPECT_NE(accepted_socket.get(), accepted_socket2.get()); |
137 | 137 |
138 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), | 138 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), |
139 local_address_.address()); | 139 local_address_.address()); |
140 EXPECT_EQ(GetPeerAddress(accepted_socket2.get()).address(), | 140 EXPECT_EQ(GetPeerAddress(accepted_socket2.get()).address(), |
141 local_address_.address()); | 141 local_address_.address()); |
142 } | 142 } |
143 | 143 |
144 } // namespace | 144 } // namespace |
145 | 145 |
146 } // namespace net | 146 } // namespace net |
OLD | NEW |