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 |