OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "ppapi/tests/test_tcp_socket_private_shared.h" | |
6 | |
7 #include <vector> | |
8 | |
9 #include "ppapi/cpp/module.h" | |
10 #include "ppapi/tests/test_utils.h" | |
11 #include "ppapi/tests/testing_instance.h" | |
12 | |
13 REGISTER_TEST_CASE(TCPSocketPrivateShared); | |
14 | |
15 TestTCPSocketPrivateShared::TestTCPSocketPrivateShared( | |
16 TestingInstance* instance) | |
17 : TestCase(instance), tcp_socket_private_interface_(NULL) { | |
18 } | |
19 | |
20 bool TestTCPSocketPrivateShared::Init() { | |
21 tcp_socket_private_interface_ = static_cast<const PPB_TCPSocket_Private*>( | |
22 pp::Module::Get()->GetBrowserInterface(PPB_TCPSOCKET_PRIVATE_INTERFACE)); | |
23 if (!tcp_socket_private_interface_) | |
24 instance_->AppendError("TCPSocketPrivate interface not available"); | |
25 | |
26 bool init_host_port = false; | |
27 if (!GetLocalHostPort(instance_->pp_instance(), &host_, &port_)) | |
28 instance_->AppendError("Can't init host and port"); | |
29 else | |
30 init_host_port = true; | |
31 | |
32 return tcp_socket_private_interface_ && | |
33 init_host_port && | |
34 CheckTestingInterface(); | |
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 std::string& host, | |
59 uint16_t port) { | |
60 TestCompletionCallback callback(instance_->pp_instance(), force_async_); | |
61 | |
62 int32_t rv = tcp_socket_private_interface_->Connect( | |
63 socket, host.c_str(), 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::SyncRead(PP_Resource socket, | |
94 char* buffer, | |
95 int32_t num_bytes, | |
96 int32_t* bytes_read) { | |
97 TestCompletionCallback callback(instance_->pp_instance(), force_async_); | |
98 | |
99 int32_t rv = tcp_socket_private_interface_->Read( | |
100 socket, buffer, num_bytes, | |
101 static_cast<pp::CompletionCallback>(callback).pp_completion_callback()); | |
102 | |
103 if (force_async_ && rv != PP_OK_COMPLETIONPENDING) | |
104 return ReportError("PPB_TCPSocket_Private::Read force_async", rv); | |
105 if (rv == PP_OK_COMPLETIONPENDING) | |
106 rv = callback.WaitForResult(); | |
107 | |
108 if (num_bytes != rv) | |
109 return ReportError("PPB_TCPSocket_Private::Read", rv); | |
110 | |
111 *bytes_read = rv; | |
112 PASS(); | |
113 } | |
114 | |
115 std::string TestTCPSocketPrivateShared::SyncWrite(PP_Resource socket, | |
116 const char* buffer, | |
117 int32_t num_bytes, | |
118 int32_t* bytes_wrote) { | |
119 TestCompletionCallback callback(instance_->pp_instance(), force_async_); | |
120 int32_t rv = tcp_socket_private_interface_->Write( | |
121 socket, buffer, num_bytes, | |
122 static_cast<pp::CompletionCallback>(callback).pp_completion_callback()); | |
123 | |
124 if (force_async_ && rv != PP_OK_COMPLETIONPENDING) | |
125 return ReportError("PPB_TCPSocket_Private::Write force_async", rv); | |
126 if (rv == PP_OK_COMPLETIONPENDING) | |
127 rv = callback.WaitForResult(); | |
128 if (num_bytes != rv) | |
129 return ReportError("PPB_TCPSocket_Private::Write", rv); | |
130 | |
131 *bytes_wrote = rv; | |
132 PASS(); | |
133 } | |
134 | |
135 std::string TestTCPSocketPrivateShared::CheckHTTPResponse( | |
136 PP_Resource socket, | |
137 const std::string& request, | |
138 const std::string& response) { | |
139 int32_t rv = 0; | |
140 std::string error_message; | |
141 | |
142 error_message = SyncWrite(socket, request.c_str(), request.size(), &rv); | |
143 if (!error_message.empty()) | |
144 return error_message; | |
145 | |
146 std::vector<char> response_buffer(response.size()); | |
147 error_message = SyncRead(socket, &response_buffer[0], response.size(), &rv); | |
148 if (!error_message.empty()) | |
149 return error_message; | |
150 | |
151 std::string actual_response(&response_buffer[0], rv); | |
152 | |
153 if (response != actual_response) | |
154 return "CheckHTTPResponse failed, expected: " + response + | |
155 ", actual: " + actual_response; | |
156 PASS(); | |
157 } | |
158 | |
159 std::string TestTCPSocketPrivateShared::TestCreate() { | |
160 PP_Resource socket = tcp_socket_private_interface_->Create(0); | |
161 if (0 != socket) | |
162 return "PPB_TCPSocket_Private::Create returns valid socket " \ | |
163 "for invalid instance"; | |
164 | |
165 return CreateSocket(&socket); | |
166 } | |
167 | |
168 std::string TestTCPSocketPrivateShared::TestGetAddress() { | |
169 PP_Resource socket; | |
170 std::string error_message; | |
171 | |
172 error_message = CreateSocket(&socket); | |
173 if (!error_message.empty()) | |
174 return error_message; | |
175 | |
176 error_message = SyncConnect(socket, host_, port_); | |
177 if (!error_message.empty()) | |
178 return error_message; | |
179 | |
180 PP_NetAddress_Private local_address, remote_address; | |
181 | |
182 if (PP_TRUE != tcp_socket_private_interface_->GetLocalAddress( | |
183 socket, &local_address)) | |
184 return "PPB_TCPSocketPrivate::GetLocalAddress failed"; | |
185 if (PP_TRUE != tcp_socket_private_interface_->GetRemoteAddress( | |
186 socket, &remote_address)) | |
187 return "PPB_TCPSocketPrivate::GetRemoteAddress failed"; | |
188 | |
189 tcp_socket_private_interface_->Disconnect(socket); | |
190 | |
191 PASS(); | |
192 } | |
193 | |
194 std::string TestTCPSocketPrivateShared::TestConnect() { | |
195 PP_Resource socket; | |
196 std::string error_message; | |
197 | |
198 error_message = CreateSocket(&socket); | |
199 if (!error_message.empty()) | |
200 return error_message; | |
201 error_message = SyncConnect(socket, host_, port_); | |
202 if (!error_message.empty()) | |
203 return error_message; | |
204 error_message = | |
205 CheckHTTPResponse(socket, "GET / HTTP/1.0\r\n\r\n", "HTTP/1.0 200 OK"); | |
206 if (!error_message.empty()) | |
207 return error_message; | |
208 tcp_socket_private_interface_->Disconnect(socket); | |
209 | |
210 PASS(); | |
211 } | |
212 | |
213 std::string TestTCPSocketPrivateShared::TestReconnect() { | |
214 PP_Resource socket; | |
215 std::string error_message; | |
216 | |
217 error_message = CreateSocket(&socket); | |
218 if (!error_message.empty()) | |
219 return error_message; | |
220 error_message = SyncConnect(socket, host_, port_); | |
221 if (!error_message.empty()) | |
222 return error_message; | |
223 | |
224 PP_NetAddress_Private remote_address; | |
225 if (PP_TRUE != | |
226 tcp_socket_private_interface_->GetRemoteAddress(socket, | |
227 &remote_address)) | |
228 return "PPB_TCPSocketPrivate::GetRemoteAddress failed"; | |
229 tcp_socket_private_interface_->Disconnect(socket); | |
230 | |
231 error_message = CreateSocket(&socket); | |
232 if (!error_message.empty()) | |
233 return error_message; | |
234 error_message = SyncConnectWithNetAddress(socket, remote_address); | |
235 if (!error_message.empty()) | |
236 return error_message; | |
237 error_message = CheckHTTPResponse(socket, | |
238 "GET / HTTP/1.0\r\n\r\n", | |
239 "HTTP/1.0 200 OK"); | |
240 if (!error_message.empty()) | |
241 return error_message; | |
242 tcp_socket_private_interface_->Disconnect(socket); | |
243 | |
244 PASS(); | |
245 } | |
246 | |
247 // TODO(ygorshenin): test SSLHandshake somehow | |
OLD | NEW |