OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "ppapi/tests/test_tcp_socket_private_shared.h" | |
6 | |
7 #include <string.h> | |
8 #include <new> | |
9 #include <string> | |
10 #include <vector> | |
11 | |
12 #include "ppapi/cpp/module.h" | |
13 #include "ppapi/tests/test_utils.h" | |
14 #include "ppapi/tests/testing_instance.h" | |
15 | |
16 REGISTER_TEST_CASE(TCPSocketPrivateShared); | |
17 | |
18 // TODO(ygorshenin): get rid of using external server in tests, | |
19 // http://crbug.com/105863 | |
20 const char* const TestTCPSocketPrivateShared::kHost = "www.google.com"; | |
21 | |
22 TestTCPSocketPrivateShared::TestTCPSocketPrivateShared( | |
23 TestingInstance* instance) | |
24 : TestCase(instance), tcp_socket_private_interface_(NULL) { | |
25 } | |
26 | |
27 bool TestTCPSocketPrivateShared::Init() { | |
28 tcp_socket_private_interface_ = | |
29 reinterpret_cast<PPB_TCPSocket_Private const*>( | |
30 pp::Module::Get()->GetBrowserInterface( | |
31 PPB_TCPSOCKET_PRIVATE_INTERFACE)); | |
32 if (!tcp_socket_private_interface_) | |
33 instance_->AppendError("TCPSocketPrivate interface not available"); | |
34 return tcp_socket_private_interface_ && InitTestingInterface(); | |
35 } | |
36 | |
37 void TestTCPSocketPrivateShared::RunTests(const std::string& filter) { | |
38 RUN_TEST(Create, filter); | |
39 RUN_TEST_FORCEASYNC_AND_NOT(GetAddress, filter); | |
40 RUN_TEST_FORCEASYNC_AND_NOT(Connect, filter); | |
41 RUN_TEST_FORCEASYNC_AND_NOT(Reconnect, filter); | |
42 } | |
43 | |
44 void TestTCPSocketPrivateShared::QuitMessageLoop() { | |
45 testing_interface_->QuitMessageLoop(instance_->pp_instance()); | |
46 } | |
47 | |
48 std::string TestTCPSocketPrivateShared::CreateSocket(PP_Resource* socket) { | |
49 *socket = tcp_socket_private_interface_->Create(instance_->pp_instance()); | |
50 if (0 == *socket) | |
51 return "PPB_TCPSocket_Private::Create failed"; | |
52 if (!tcp_socket_private_interface_->IsTCPSocket(*socket)) | |
53 return "PPB_TCPSocket_Private::IsTCPSocket failed"; | |
54 PASS(); | |
55 } | |
56 | |
57 std::string TestTCPSocketPrivateShared::SyncConnect(PP_Resource socket, | |
58 const char* host, | |
59 int port) { | |
60 TestCompletionCallback callback(instance_->pp_instance(), force_async_); | |
61 | |
62 int32_t rv = tcp_socket_private_interface_->Connect( | |
63 socket, host, port, | |
64 static_cast<pp::CompletionCallback>(callback).pp_completion_callback()); | |
65 | |
66 if (force_async_ && rv != PP_OK_COMPLETIONPENDING) | |
67 return ReportError("PPB_TCPSocket_Private::Connect force_async", rv); | |
68 if (rv == PP_OK_COMPLETIONPENDING) | |
69 rv = callback.WaitForResult(); | |
70 if (rv != PP_OK) | |
71 return ReportError("PPB_TCPSocket_Private::Connect", rv); | |
72 PASS(); | |
73 } | |
74 | |
75 std::string TestTCPSocketPrivateShared::SyncConnectWithNetAddress( | |
76 PP_Resource socket, const PP_NetAddress_Private& addr) { | |
77 TestCompletionCallback callback(instance_->pp_instance(), force_async_); | |
78 | |
79 int32_t rv = tcp_socket_private_interface_->ConnectWithNetAddress( | |
80 socket, &addr, | |
81 static_cast<pp::CompletionCallback>(callback).pp_completion_callback()); | |
82 | |
83 if (force_async_ && rv != PP_OK_COMPLETIONPENDING) | |
84 return ReportError( | |
85 "PPB_TCPSocket_Private::ConnectWithNetAddress force_async", rv); | |
86 if (rv == PP_OK_COMPLETIONPENDING) | |
87 rv = callback.WaitForResult(); | |
88 if (rv != PP_OK) | |
89 return ReportError("PPB_TCPSocket_Private::ConnectWithNetAddress", rv); | |
90 PASS(); | |
91 } | |
92 | |
93 std::string TestTCPSocketPrivateShared::SyncSSLHandshake(PP_Resource socket, | |
94 const char* host, | |
95 int port) { | |
96 TestCompletionCallback callback(instance_->pp_instance(), force_async_); | |
97 | |
98 int32_t rv = tcp_socket_private_interface_->SSLHandshake( | |
99 socket, host, port, | |
100 static_cast<pp::CompletionCallback>(callback).pp_completion_callback()); | |
101 | |
102 if (force_async_ && rv != PP_OK_COMPLETIONPENDING) | |
103 return ReportError("PPB_TCPSocket_Private::SSLHandshake force_async", rv); | |
104 if (rv == PP_OK_COMPLETIONPENDING) | |
105 rv = callback.WaitForResult(); | |
106 if (rv != PP_OK) | |
107 return ReportError("PPB_TCPSocket_Private::SSLHandshake", rv); | |
108 PASS(); | |
109 } | |
110 | |
111 std::string TestTCPSocketPrivateShared::SyncRead(PP_Resource socket, | |
112 char* buffer, | |
113 int32_t num_bytes, | |
114 int32_t* bytes_read) { | |
115 TestCompletionCallback callback(instance_->pp_instance(), force_async_); | |
116 | |
117 int32_t rv = tcp_socket_private_interface_->Read( | |
118 socket, buffer, num_bytes, | |
119 static_cast<pp::CompletionCallback>(callback).pp_completion_callback()); | |
120 | |
121 if (force_async_ && rv != PP_OK_COMPLETIONPENDING) | |
122 return ReportError("PPB_TCPSocket_Private::Read force_async", rv); | |
123 if (rv == PP_OK_COMPLETIONPENDING) | |
124 rv = callback.WaitForResult(); | |
125 | |
126 if (num_bytes != rv) | |
127 return ReportError("PPB_TCPSocket_Private::Read", rv); | |
128 | |
129 *bytes_read = rv; | |
130 PASS(); | |
131 } | |
132 | |
133 std::string TestTCPSocketPrivateShared::SyncWrite(PP_Resource socket, | |
134 const char* buffer, | |
135 int32_t num_bytes, | |
136 int32_t* bytes_wrote) { | |
137 TestCompletionCallback callback(instance_->pp_instance(), force_async_); | |
138 int32_t rv = tcp_socket_private_interface_->Write( | |
139 socket, buffer, num_bytes, | |
140 static_cast<pp::CompletionCallback>(callback).pp_completion_callback()); | |
141 | |
142 if (force_async_ && rv != PP_OK_COMPLETIONPENDING) | |
143 return ReportError("PPB_TCPSocket_Private::Write force_async", rv); | |
144 if (rv == PP_OK_COMPLETIONPENDING) | |
145 rv = callback.WaitForResult(); | |
146 if (num_bytes != rv) | |
147 return ReportError("PPB_TCPSocket_Private::Write", rv); | |
148 | |
149 *bytes_wrote = rv; | |
150 PASS(); | |
151 } | |
152 | |
153 std::string TestTCPSocketPrivateShared::CheckHTTPResponse( | |
154 PP_Resource socket, | |
155 const std::string& request, | |
156 const std::string& response) { | |
157 int32_t rv; | |
158 std::string error_message; | |
159 | |
160 error_message = SyncWrite(socket, request.c_str(), request.size(), &rv); | |
161 if (!error_message.empty()) | |
162 return error_message; | |
163 | |
164 std::vector<char> response_buffer(response.size()); | |
165 error_message = SyncRead(socket, &response_buffer[0], response.size(), &rv); | |
166 if (!error_message.empty()) | |
167 return error_message; | |
168 | |
169 std::string actual_response(&response_buffer[0], rv); | |
170 | |
171 if (response != actual_response) | |
172 return "CheckHTTPResponse failed, expected: " + response + | |
173 ", actual: " + actual_response; | |
174 PASS(); | |
175 } | |
176 | |
177 std::string TestTCPSocketPrivateShared::TestCreate() { | |
178 PP_Resource socket = tcp_socket_private_interface_->Create(0); | |
179 if (0 != socket) | |
180 return "PPB_TCPSocket_Private::Create returns valid socket " \ | |
181 "for invalid instance"; | |
182 | |
183 return CreateSocket(&socket); | |
184 } | |
185 | |
186 std::string TestTCPSocketPrivateShared::TestGetAddress() { | |
187 PP_Resource socket; | |
188 std::string error_message; | |
189 | |
190 error_message = CreateSocket(&socket); | |
191 if (!error_message.empty()) | |
192 return error_message; | |
193 | |
194 error_message = SyncConnect(socket, kHost, kPort); | |
195 if (!error_message.empty()) | |
196 return error_message; | |
197 | |
198 PP_NetAddress_Private local_address, remote_address; | |
199 | |
200 if (PP_TRUE != tcp_socket_private_interface_->GetLocalAddress( | |
201 socket, &local_address)) | |
202 return "PPB_TCPSocketPrivate::GetLocalAddress failed"; | |
203 if (PP_TRUE != tcp_socket_private_interface_->GetRemoteAddress( | |
204 socket, &remote_address)) | |
205 return "PPB_TCPSocketPrivate::GetRemoteAddress failed"; | |
206 | |
207 tcp_socket_private_interface_->Disconnect(socket); | |
208 | |
209 PASS(); | |
210 } | |
211 | |
212 std::string TestTCPSocketPrivateShared::TestConnect() { | |
213 PP_Resource socket; | |
214 std::string error_message; | |
215 | |
216 error_message = CreateSocket(&socket); | |
217 if (!error_message.empty()) | |
218 return error_message; | |
219 error_message = SyncConnect(socket, kHost, kPort); | |
220 if (!error_message.empty()) | |
221 return error_message; | |
222 error_message = SyncSSLHandshake(socket, kHost, kPort); | |
223 if (!error_message.empty()) | |
224 return error_message; | |
225 error_message = | |
226 CheckHTTPResponse(socket, "GET /robots.txt\r\n", "HTTP/1.0 200 OK"); | |
227 if (!error_message.empty()) | |
228 return error_message; | |
229 tcp_socket_private_interface_->Disconnect(socket); | |
230 | |
231 PASS(); | |
232 } | |
233 | |
234 std::string TestTCPSocketPrivateShared::TestReconnect() { | |
235 PP_Resource socket; | |
236 std::string error_message; | |
237 | |
238 error_message = CreateSocket(&socket); | |
239 if (!error_message.empty()) | |
240 return error_message; | |
241 error_message = SyncConnect(socket, kHost, kPort); | |
242 if (!error_message.empty()) | |
243 return error_message; | |
244 error_message = SyncSSLHandshake(socket, kHost, kPort); | |
245 if (!error_message.empty()) | |
246 return error_message; | |
247 | |
248 PP_NetAddress_Private remote_address; | |
249 if (PP_TRUE != | |
250 tcp_socket_private_interface_->GetRemoteAddress(socket, | |
251 &remote_address)) | |
252 return "PPB_TCPSocketPrivate::GetRemoteAddress failed"; | |
253 tcp_socket_private_interface_->Disconnect(socket); | |
254 | |
255 error_message = CreateSocket(&socket); | |
256 if (!error_message.empty()) | |
257 return error_message; | |
258 error_message = SyncConnectWithNetAddress(socket, remote_address); | |
259 if (!error_message.empty()) | |
260 return error_message; | |
261 error_message = SyncSSLHandshake(socket, kHost, kPort); | |
262 if (!error_message.empty()) | |
263 return error_message; | |
264 error_message = | |
265 CheckHTTPResponse(socket, "GET /robots.txt\r\n", "HTTP/1.0 200 OK"); | |
266 if (!error_message.empty()) | |
267 return error_message; | |
268 tcp_socket_private_interface_->Disconnect(socket); | |
269 | |
270 PASS(); | |
271 } | |
OLD | NEW |