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/shared_impl/private/udp_socket_private_impl.h" |
| 6 |
| 7 #include <string.h> |
| 8 |
| 9 #include <algorithm> |
| 10 |
| 11 #include "base/basictypes.h" |
| 12 #include "base/bind.h" |
| 13 #include "base/logging.h" |
| 14 #include "base/message_loop.h" |
| 15 #include "ppapi/c/pp_completion_callback.h" |
| 16 #include "ppapi/c/pp_errors.h" |
| 17 |
| 18 namespace ppapi { |
| 19 |
| 20 namespace { |
| 21 |
| 22 void AbortCallback(PP_CompletionCallback callback) { |
| 23 PP_RunCompletionCallback(&callback, PP_ERROR_ABORTED); |
| 24 } |
| 25 |
| 26 } // namespace |
| 27 |
| 28 const int32_t UDPSocketPrivateImpl::kMaxReadSize = 1024 * 1024; |
| 29 const int32_t UDPSocketPrivateImpl::kMaxWriteSize = 1024 * 1024; |
| 30 |
| 31 UDPSocketPrivateImpl::UDPSocketPrivateImpl(const HostResource& resource, |
| 32 uint32 socket_id) |
| 33 : Resource(resource) { |
| 34 Init(socket_id); |
| 35 } |
| 36 |
| 37 UDPSocketPrivateImpl::UDPSocketPrivateImpl(PP_Instance instance, |
| 38 uint32 socket_id) |
| 39 : Resource(instance) { |
| 40 Init(socket_id); |
| 41 } |
| 42 |
| 43 UDPSocketPrivateImpl::~UDPSocketPrivateImpl() { |
| 44 } |
| 45 |
| 46 thunk::PPB_UDPSocket_Private_API* |
| 47 UDPSocketPrivateImpl::AsPPB_UDPSocket_Private_API() { |
| 48 return this; |
| 49 } |
| 50 |
| 51 int32_t UDPSocketPrivateImpl::Bind(const PP_NetAddress_Private* addr, |
| 52 PP_CompletionCallback callback) { |
| 53 if (!addr || !callback.func) |
| 54 return PP_ERROR_BADARGUMENT; |
| 55 if (bound_ || closed_) |
| 56 return PP_ERROR_FAILED; |
| 57 if (bind_callback_.func) |
| 58 return PP_ERROR_INPROGRESS; |
| 59 // TODO(dmichael): use some other strategy for determining if an |
| 60 // operation is in progress |
| 61 |
| 62 bind_callback_ = callback; |
| 63 |
| 64 // Send the request, the browser will call us back via BindACK. |
| 65 SendBind(*addr); |
| 66 return PP_OK_COMPLETIONPENDING; |
| 67 } |
| 68 |
| 69 int32_t UDPSocketPrivateImpl::RecvFrom(char* buffer, |
| 70 int32_t num_bytes, |
| 71 PP_CompletionCallback callback) { |
| 72 if (!buffer || num_bytes <= 0 || !callback.func) |
| 73 return PP_ERROR_BADARGUMENT; |
| 74 if (!bound_) |
| 75 return PP_ERROR_FAILED; |
| 76 if (recvfrom_callback_.func) |
| 77 return PP_ERROR_INPROGRESS; |
| 78 |
| 79 read_buffer_ = buffer; |
| 80 bytes_to_read_ = std::min(num_bytes, kMaxReadSize); |
| 81 recvfrom_callback_ = callback; |
| 82 |
| 83 // Send the request, the browser will call us back via RecvFromACK. |
| 84 SendRecvFrom(bytes_to_read_); |
| 85 return PP_OK_COMPLETIONPENDING; |
| 86 } |
| 87 |
| 88 PP_Bool UDPSocketPrivateImpl::GetRecvFromAddress(PP_NetAddress_Private* addr) { |
| 89 if (!addr) |
| 90 return PP_FALSE; |
| 91 |
| 92 *addr = recvfrom_addr_; |
| 93 return PP_TRUE; |
| 94 } |
| 95 |
| 96 int32_t UDPSocketPrivateImpl::SendTo(const char* buffer, |
| 97 int32_t num_bytes, |
| 98 const PP_NetAddress_Private* addr, |
| 99 PP_CompletionCallback callback) { |
| 100 if (!buffer || num_bytes <= 0 || !addr || !callback.func) |
| 101 return PP_ERROR_BADARGUMENT; |
| 102 if (!bound_) |
| 103 return PP_ERROR_FAILED; |
| 104 if (sendto_callback_.func) |
| 105 return PP_ERROR_INPROGRESS; |
| 106 |
| 107 if (num_bytes > kMaxWriteSize) |
| 108 num_bytes = kMaxWriteSize; |
| 109 |
| 110 sendto_callback_ = callback; |
| 111 |
| 112 // Send the request, the browser will call us back via SendToACK. |
| 113 SendSendTo(std::string(buffer, num_bytes), *addr); |
| 114 return PP_OK_COMPLETIONPENDING; |
| 115 } |
| 116 |
| 117 void UDPSocketPrivateImpl::Close() { |
| 118 if(closed_) |
| 119 return; |
| 120 |
| 121 bound_ = false; |
| 122 closed_ = true; |
| 123 |
| 124 SendClose(); |
| 125 |
| 126 socket_id_ = 0; |
| 127 |
| 128 PostAbortAndClearIfNecessary(&bind_callback_); |
| 129 PostAbortAndClearIfNecessary(&recvfrom_callback_); |
| 130 PostAbortAndClearIfNecessary(&sendto_callback_); |
| 131 } |
| 132 |
| 133 void UDPSocketPrivateImpl::OnBindCompleted(bool succeeded) { |
| 134 if (!bind_callback_.func) { |
| 135 NOTREACHED(); |
| 136 return; |
| 137 } |
| 138 |
| 139 if (succeeded) |
| 140 bound_ = true; |
| 141 |
| 142 PP_RunAndClearCompletionCallback(&bind_callback_, |
| 143 succeeded ? PP_OK : PP_ERROR_FAILED); |
| 144 } |
| 145 |
| 146 void UDPSocketPrivateImpl::OnRecvFromCompleted( |
| 147 bool succeeded, |
| 148 const std::string& data, |
| 149 const PP_NetAddress_Private& addr) { |
| 150 if (!recvfrom_callback_.func || !read_buffer_) { |
| 151 NOTREACHED(); |
| 152 return; |
| 153 } |
| 154 |
| 155 if (succeeded) { |
| 156 CHECK_LE(static_cast<int32_t>(data.size()), bytes_to_read_); |
| 157 if (!data.empty()) |
| 158 memcpy(read_buffer_, data.c_str(), data.size()); |
| 159 } |
| 160 read_buffer_ = NULL; |
| 161 bytes_to_read_ = -1; |
| 162 recvfrom_addr_ = addr; |
| 163 |
| 164 PP_RunAndClearCompletionCallback( |
| 165 &recvfrom_callback_, |
| 166 succeeded ? static_cast<int32_t>(data.size()) : |
| 167 static_cast<int32_t>(PP_ERROR_FAILED)); |
| 168 } |
| 169 |
| 170 void UDPSocketPrivateImpl::OnSendToCompleted(bool succeeded, |
| 171 int32_t bytes_written) { |
| 172 if (!sendto_callback_.func) { |
| 173 NOTREACHED(); |
| 174 return; |
| 175 } |
| 176 |
| 177 PP_RunAndClearCompletionCallback( |
| 178 &sendto_callback_, |
| 179 succeeded ? bytes_written : static_cast<int32_t>(PP_ERROR_FAILED)); |
| 180 } |
| 181 |
| 182 void UDPSocketPrivateImpl::Init(uint32 socket_id) { |
| 183 DCHECK(socket_id != 0); |
| 184 socket_id_ = socket_id; |
| 185 bound_ = false; |
| 186 closed_ = false; |
| 187 bind_callback_ = PP_BlockUntilComplete(); |
| 188 recvfrom_callback_ = PP_BlockUntilComplete(); |
| 189 sendto_callback_ = PP_BlockUntilComplete(); |
| 190 read_buffer_ = NULL; |
| 191 bytes_to_read_ = -1; |
| 192 |
| 193 recvfrom_addr_.size = 0; |
| 194 memset(recvfrom_addr_.data, 0, |
| 195 arraysize(recvfrom_addr_.data) * sizeof(*recvfrom_addr_.data)); |
| 196 } |
| 197 |
| 198 void UDPSocketPrivateImpl::PostAbortAndClearIfNecessary( |
| 199 PP_CompletionCallback* callback) { |
| 200 DCHECK(callback); |
| 201 |
| 202 if (callback->func) { |
| 203 MessageLoop::current()->PostTask(FROM_HERE, |
| 204 base::Bind(&AbortCallback, *callback)); |
| 205 *callback = PP_BlockUntilComplete(); |
| 206 } |
| 207 } |
| 208 |
| 209 } // namespace ppapi |
OLD | NEW |