| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 "ppapi/shared_impl/private/udp_socket_private_impl.h" | 5 #include "ppapi/proxy/udp_socket_private_resource.h" |
| 6 | |
| 7 #include <string.h> | |
| 8 | 6 |
| 9 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <cstring> |
| 10 | 9 |
| 11 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| 12 #include "base/bind.h" | |
| 13 #include "base/logging.h" | 11 #include "base/logging.h" |
| 14 #include "base/message_loop.h" | |
| 15 #include "ppapi/c/pp_bool.h" | 12 #include "ppapi/c/pp_bool.h" |
| 16 #include "ppapi/c/pp_completion_callback.h" | 13 #include "ppapi/c/pp_completion_callback.h" |
| 17 #include "ppapi/c/pp_errors.h" | 14 #include "ppapi/c/pp_errors.h" |
| 15 #include "ppapi/proxy/ppapi_messages.h" |
| 18 | 16 |
| 19 namespace ppapi { | 17 namespace ppapi { |
| 18 namespace proxy { |
| 20 | 19 |
| 21 const int32_t UDPSocketPrivateImpl::kMaxReadSize = 1024 * 1024; | 20 const int32_t UDPSocketPrivateResource::kMaxReadSize = 1024 * 1024; |
| 22 const int32_t UDPSocketPrivateImpl::kMaxWriteSize = 1024 * 1024; | 21 const int32_t UDPSocketPrivateResource::kMaxWriteSize = 1024 * 1024; |
| 23 | 22 |
| 24 UDPSocketPrivateImpl::UDPSocketPrivateImpl(const HostResource& resource, | 23 UDPSocketPrivateResource::UDPSocketPrivateResource(Connection connection, |
| 25 uint32 socket_id) | 24 PP_Instance instance) |
| 26 : Resource(OBJECT_IS_PROXY, resource) { | 25 : PluginResource(connection, instance), |
| 27 Init(socket_id); | 26 bound_(false), |
| 27 closed_(false), |
| 28 read_buffer_(NULL), |
| 29 bytes_to_read_(-1) { |
| 30 recvfrom_addr_.size = 0; |
| 31 memset(recvfrom_addr_.data, 0, |
| 32 arraysize(recvfrom_addr_.data) * sizeof(*recvfrom_addr_.data)); |
| 33 bound_addr_.size = 0; |
| 34 memset(bound_addr_.data, 0, |
| 35 arraysize(bound_addr_.data) * sizeof(*bound_addr_.data)); |
| 36 |
| 37 SendCreate(BROWSER, PpapiHostMsg_UDPSocketPrivate_Create()); |
| 28 } | 38 } |
| 29 | 39 |
| 30 UDPSocketPrivateImpl::UDPSocketPrivateImpl(PP_Instance instance, | 40 UDPSocketPrivateResource::~UDPSocketPrivateResource() { |
| 31 uint32 socket_id) | |
| 32 : Resource(OBJECT_IS_IMPL, instance) { | |
| 33 Init(socket_id); | |
| 34 } | |
| 35 | |
| 36 UDPSocketPrivateImpl::~UDPSocketPrivateImpl() { | |
| 37 } | 41 } |
| 38 | 42 |
| 39 thunk::PPB_UDPSocket_Private_API* | 43 thunk::PPB_UDPSocket_Private_API* |
| 40 UDPSocketPrivateImpl::AsPPB_UDPSocket_Private_API() { | 44 UDPSocketPrivateResource::AsPPB_UDPSocket_Private_API() { |
| 41 return this; | 45 return this; |
| 42 } | 46 } |
| 43 | 47 |
| 44 int32_t UDPSocketPrivateImpl::SetSocketFeature(PP_UDPSocketFeature_Private name, | 48 int32_t UDPSocketPrivateResource::SetSocketFeature( |
| 45 PP_Var value) { | 49 PP_UDPSocketFeature_Private name, |
| 50 PP_Var value) { |
| 46 if (bound_ || closed_) | 51 if (bound_ || closed_) |
| 47 return PP_ERROR_FAILED; | 52 return PP_ERROR_FAILED; |
| 48 | 53 |
| 49 switch (name) { | 54 switch (name) { |
| 50 case PP_UDPSOCKETFEATURE_ADDRESS_REUSE: | 55 case PP_UDPSOCKETFEATURE_ADDRESS_REUSE: |
| 51 case PP_UDPSOCKETFEATURE_BROADCAST: | 56 case PP_UDPSOCKETFEATURE_BROADCAST: |
| 52 if (value.type != PP_VARTYPE_BOOL) | 57 if (value.type != PP_VARTYPE_BOOL) |
| 53 return PP_ERROR_BADARGUMENT; | 58 return PP_ERROR_BADARGUMENT; |
| 54 SendBoolSocketFeature(static_cast<int32_t>(name), | 59 SendBoolSocketFeature(static_cast<int32_t>(name), |
| 55 PP_ToBool(value.value.as_bool)); | 60 PP_ToBool(value.value.as_bool)); |
| 56 break; | 61 break; |
| 57 default: | 62 default: |
| 58 return PP_ERROR_BADARGUMENT; | 63 return PP_ERROR_BADARGUMENT; |
| 59 } | 64 } |
| 60 return PP_OK; | 65 return PP_OK; |
| 61 } | 66 } |
| 62 | 67 |
| 63 int32_t UDPSocketPrivateImpl::Bind(const PP_NetAddress_Private* addr, | 68 int32_t UDPSocketPrivateResource::Bind( |
| 64 scoped_refptr<TrackedCallback> callback) { | 69 const PP_NetAddress_Private* addr, |
| 70 scoped_refptr<TrackedCallback> callback) { |
| 65 if (!addr) | 71 if (!addr) |
| 66 return PP_ERROR_BADARGUMENT; | 72 return PP_ERROR_BADARGUMENT; |
| 67 if (bound_ || closed_) | 73 if (bound_ || closed_) |
| 68 return PP_ERROR_FAILED; | 74 return PP_ERROR_FAILED; |
| 69 if (TrackedCallback::IsPending(bind_callback_)) | 75 if (TrackedCallback::IsPending(bind_callback_)) |
| 70 return PP_ERROR_INPROGRESS; | 76 return PP_ERROR_INPROGRESS; |
| 71 | 77 |
| 72 bind_callback_ = callback; | 78 bind_callback_ = callback; |
| 73 | 79 |
| 74 // Send the request, the browser will call us back via BindACK. | 80 // Send the request, the browser will call us back via BindACK. |
| 75 SendBind(*addr); | 81 SendBind(*addr); |
| 76 return PP_OK_COMPLETIONPENDING; | 82 return PP_OK_COMPLETIONPENDING; |
| 77 } | 83 } |
| 78 | 84 |
| 79 PP_Bool UDPSocketPrivateImpl::GetBoundAddress(PP_NetAddress_Private* addr) { | 85 PP_Bool UDPSocketPrivateResource::GetBoundAddress(PP_NetAddress_Private* addr) { |
| 80 if (!addr || !bound_ || closed_) | 86 if (!addr || !bound_ || closed_) |
| 81 return PP_FALSE; | 87 return PP_FALSE; |
| 82 | 88 |
| 83 *addr = bound_addr_; | 89 *addr = bound_addr_; |
| 84 return PP_TRUE; | 90 return PP_TRUE; |
| 85 } | 91 } |
| 86 | 92 |
| 87 int32_t UDPSocketPrivateImpl::RecvFrom( | 93 int32_t UDPSocketPrivateResource::RecvFrom( |
| 88 char* buffer, | 94 char* buffer, |
| 89 int32_t num_bytes, | 95 int32_t num_bytes, |
| 90 scoped_refptr<TrackedCallback> callback) { | 96 scoped_refptr<TrackedCallback> callback) { |
| 91 if (!buffer || num_bytes <= 0) | 97 if (!buffer || num_bytes <= 0) |
| 92 return PP_ERROR_BADARGUMENT; | 98 return PP_ERROR_BADARGUMENT; |
| 93 if (!bound_) | 99 if (!bound_) |
| 94 return PP_ERROR_FAILED; | 100 return PP_ERROR_FAILED; |
| 95 if (TrackedCallback::IsPending(recvfrom_callback_)) | 101 if (TrackedCallback::IsPending(recvfrom_callback_)) |
| 96 return PP_ERROR_INPROGRESS; | 102 return PP_ERROR_INPROGRESS; |
| 97 | 103 |
| 98 read_buffer_ = buffer; | 104 read_buffer_ = buffer; |
| 99 bytes_to_read_ = std::min(num_bytes, kMaxReadSize); | 105 bytes_to_read_ = std::min(num_bytes, kMaxReadSize); |
| 100 recvfrom_callback_ = callback; | 106 recvfrom_callback_ = callback; |
| 101 | 107 |
| 102 // Send the request, the browser will call us back via RecvFromACK. | 108 // Send the request, the browser will call us back via RecvFromACK. |
| 103 SendRecvFrom(bytes_to_read_); | 109 SendRecvFrom(bytes_to_read_); |
| 104 return PP_OK_COMPLETIONPENDING; | 110 return PP_OK_COMPLETIONPENDING; |
| 105 } | 111 } |
| 106 | 112 |
| 107 PP_Bool UDPSocketPrivateImpl::GetRecvFromAddress(PP_NetAddress_Private* addr) { | 113 PP_Bool UDPSocketPrivateResource::GetRecvFromAddress( |
| 114 PP_NetAddress_Private* addr) { |
| 108 if (!addr) | 115 if (!addr) |
| 109 return PP_FALSE; | 116 return PP_FALSE; |
| 110 | |
| 111 *addr = recvfrom_addr_; | 117 *addr = recvfrom_addr_; |
| 112 return PP_TRUE; | 118 return PP_TRUE; |
| 113 } | 119 } |
| 114 | 120 |
| 115 int32_t UDPSocketPrivateImpl::SendTo(const char* buffer, | 121 void UDPSocketPrivateResource::PostAbortIfNecessary( |
| 116 int32_t num_bytes, | 122 scoped_refptr<TrackedCallback>* callback) { |
| 117 const PP_NetAddress_Private* addr, | 123 if (TrackedCallback::IsPending(*callback)) |
| 118 scoped_refptr<TrackedCallback> callback) { | 124 (*callback)->PostAbort(); |
| 125 } |
| 126 |
| 127 int32_t UDPSocketPrivateResource::SendTo( |
| 128 const char* buffer, |
| 129 int32_t num_bytes, |
| 130 const PP_NetAddress_Private* addr, |
| 131 scoped_refptr<TrackedCallback> callback) { |
| 119 if (!buffer || num_bytes <= 0 || !addr) | 132 if (!buffer || num_bytes <= 0 || !addr) |
| 120 return PP_ERROR_BADARGUMENT; | 133 return PP_ERROR_BADARGUMENT; |
| 121 if (!bound_) | 134 if (!bound_) |
| 122 return PP_ERROR_FAILED; | 135 return PP_ERROR_FAILED; |
| 123 if (TrackedCallback::IsPending(sendto_callback_)) | 136 if (TrackedCallback::IsPending(sendto_callback_)) |
| 124 return PP_ERROR_INPROGRESS; | 137 return PP_ERROR_INPROGRESS; |
| 125 | 138 |
| 126 if (num_bytes > kMaxWriteSize) | 139 if (num_bytes > kMaxWriteSize) |
| 127 num_bytes = kMaxWriteSize; | 140 num_bytes = kMaxWriteSize; |
| 128 | 141 |
| 129 sendto_callback_ = callback; | 142 sendto_callback_ = callback; |
| 130 | 143 |
| 131 // Send the request, the browser will call us back via SendToACK. | 144 // Send the request, the browser will call us back via SendToACK. |
| 132 SendSendTo(std::string(buffer, num_bytes), *addr); | 145 SendSendTo(std::string(buffer, num_bytes), *addr); |
| 133 return PP_OK_COMPLETIONPENDING; | 146 return PP_OK_COMPLETIONPENDING; |
| 134 } | 147 } |
| 135 | 148 |
| 136 void UDPSocketPrivateImpl::Close() { | 149 void UDPSocketPrivateResource::Close() { |
| 137 if(closed_) | 150 if(closed_) |
| 138 return; | 151 return; |
| 139 | 152 |
| 140 bound_ = false; | 153 bound_ = false; |
| 141 closed_ = true; | 154 closed_ = true; |
| 142 | 155 |
| 143 SendClose(); | 156 SendClose(); |
| 144 | 157 |
| 145 socket_id_ = 0; | |
| 146 | |
| 147 PostAbortIfNecessary(&bind_callback_); | 158 PostAbortIfNecessary(&bind_callback_); |
| 148 PostAbortIfNecessary(&recvfrom_callback_); | 159 PostAbortIfNecessary(&recvfrom_callback_); |
| 149 PostAbortIfNecessary(&sendto_callback_); | 160 PostAbortIfNecessary(&sendto_callback_); |
| 150 } | 161 } |
| 151 | 162 |
| 152 void UDPSocketPrivateImpl::OnBindCompleted( | 163 void UDPSocketPrivateResource::SendBoolSocketFeature(int32_t name, bool value) { |
| 164 PpapiHostMsg_UDPSocketPrivate_SetBoolSocketFeature msg(name, value); |
| 165 Post(BROWSER, msg); |
| 166 } |
| 167 |
| 168 void UDPSocketPrivateResource::SendBind(const PP_NetAddress_Private& addr) { |
| 169 PpapiHostMsg_UDPSocketPrivate_Bind msg(addr); |
| 170 Call<PpapiPluginMsg_UDPSocketPrivate_BindReply>( |
| 171 BROWSER, |
| 172 msg, |
| 173 base::Bind(&UDPSocketPrivateResource::OnPluginMsgBindReply, |
| 174 base::Unretained(this))); |
| 175 } |
| 176 |
| 177 void UDPSocketPrivateResource::SendRecvFrom(int32_t num_bytes) { |
| 178 PpapiHostMsg_UDPSocketPrivate_RecvFrom msg(num_bytes); |
| 179 Call<PpapiPluginMsg_UDPSocketPrivate_RecvFromReply>( |
| 180 BROWSER, |
| 181 msg, |
| 182 base::Bind(&UDPSocketPrivateResource::OnPluginMsgRecvFromReply, |
| 183 base::Unretained(this))); |
| 184 } |
| 185 |
| 186 void UDPSocketPrivateResource::SendSendTo(const std::string& buffer, |
| 187 const PP_NetAddress_Private& addr) { |
| 188 PpapiHostMsg_UDPSocketPrivate_SendTo msg(buffer, addr); |
| 189 Call<PpapiPluginMsg_UDPSocketPrivate_SendToReply>( |
| 190 BROWSER, |
| 191 msg, |
| 192 base::Bind(&UDPSocketPrivateResource::OnPluginMsgSendToReply, |
| 193 base::Unretained(this))); |
| 194 } |
| 195 |
| 196 void UDPSocketPrivateResource::SendClose() { |
| 197 PpapiHostMsg_UDPSocketPrivate_Close msg; |
| 198 Post(BROWSER, msg); |
| 199 } |
| 200 |
| 201 void UDPSocketPrivateResource::OnPluginMsgBindReply( |
| 202 const ResourceMessageReplyParams& params, |
| 153 bool succeeded, | 203 bool succeeded, |
| 154 const PP_NetAddress_Private& addr) { | 204 const PP_NetAddress_Private& bound_addr) { |
| 155 if (!TrackedCallback::IsPending(bind_callback_)) { | 205 if (!TrackedCallback::IsPending(bind_callback_)) { |
| 156 NOTREACHED(); | 206 NOTREACHED(); |
| 157 return; | 207 return; |
| 158 } | 208 } |
| 159 | |
| 160 if (succeeded) | 209 if (succeeded) |
| 161 bound_ = true; | 210 bound_ = true; |
| 162 | 211 bound_addr_ = bound_addr; |
| 163 bound_addr_ = addr; | |
| 164 | |
| 165 bind_callback_->Run(succeeded ? PP_OK : PP_ERROR_FAILED); | 212 bind_callback_->Run(succeeded ? PP_OK : PP_ERROR_FAILED); |
| 166 } | 213 } |
| 167 | 214 |
| 168 void UDPSocketPrivateImpl::OnRecvFromCompleted( | 215 void UDPSocketPrivateResource::OnPluginMsgRecvFromReply( |
| 216 const ResourceMessageReplyParams& params, |
| 169 bool succeeded, | 217 bool succeeded, |
| 170 const std::string& data, | 218 const std::string& data, |
| 171 const PP_NetAddress_Private& addr) { | 219 const PP_NetAddress_Private& addr) { |
| 172 if (!TrackedCallback::IsPending(recvfrom_callback_) || !read_buffer_) { | 220 if (!TrackedCallback::IsPending(recvfrom_callback_) || !read_buffer_) { |
| 173 NOTREACHED(); | 221 NOTREACHED(); |
| 174 return; | 222 return; |
| 175 } | 223 } |
| 176 | |
| 177 if (succeeded) { | 224 if (succeeded) { |
| 178 CHECK_LE(static_cast<int32_t>(data.size()), bytes_to_read_); | 225 CHECK_LE(static_cast<int32_t>(data.size()), bytes_to_read_); |
| 179 if (!data.empty()) | 226 if (!data.empty()) |
| 180 memcpy(read_buffer_, data.c_str(), data.size()); | 227 memcpy(read_buffer_, data.c_str(), data.size()); |
| 181 } | 228 } |
| 182 read_buffer_ = NULL; | 229 read_buffer_ = NULL; |
| 183 bytes_to_read_ = -1; | 230 bytes_to_read_ = -1; |
| 184 recvfrom_addr_ = addr; | 231 recvfrom_addr_ = addr; |
| 185 | 232 |
| 186 recvfrom_callback_->Run(succeeded ? static_cast<int32_t>(data.size()) : | 233 recvfrom_callback_->Run(succeeded ? static_cast<int32_t>(data.size()) : |
| 187 static_cast<int32_t>(PP_ERROR_FAILED)); | 234 static_cast<int32_t>(PP_ERROR_FAILED)); |
| 188 } | 235 } |
| 189 | 236 |
| 190 void UDPSocketPrivateImpl::OnSendToCompleted(bool succeeded, | 237 void UDPSocketPrivateResource::OnPluginMsgSendToReply( |
| 191 int32_t bytes_written) { | 238 const ResourceMessageReplyParams& params, |
| 239 bool succeeded, |
| 240 int32_t bytes_written) { |
| 192 if (!TrackedCallback::IsPending(sendto_callback_)) { | 241 if (!TrackedCallback::IsPending(sendto_callback_)) { |
| 193 NOTREACHED(); | 242 NOTREACHED(); |
| 194 return; | 243 return; |
| 195 } | 244 } |
| 196 | |
| 197 sendto_callback_->Run( | 245 sendto_callback_->Run( |
| 198 succeeded ? bytes_written : static_cast<int32_t>(PP_ERROR_FAILED)); | 246 succeeded ? bytes_written : static_cast<int32_t>(PP_ERROR_FAILED)); |
| 199 } | 247 } |
| 200 | 248 |
| 201 void UDPSocketPrivateImpl::Init(uint32 socket_id) { | 249 } // namespace proxy |
| 202 DCHECK(socket_id != 0); | |
| 203 socket_id_ = socket_id; | |
| 204 bound_ = false; | |
| 205 closed_ = false; | |
| 206 read_buffer_ = NULL; | |
| 207 bytes_to_read_ = -1; | |
| 208 | |
| 209 recvfrom_addr_.size = 0; | |
| 210 memset(recvfrom_addr_.data, 0, | |
| 211 arraysize(recvfrom_addr_.data) * sizeof(*recvfrom_addr_.data)); | |
| 212 bound_addr_.size = 0; | |
| 213 memset(bound_addr_.data, 0, | |
| 214 arraysize(bound_addr_.data) * sizeof(*bound_addr_.data)); | |
| 215 } | |
| 216 | |
| 217 void UDPSocketPrivateImpl::PostAbortIfNecessary( | |
| 218 scoped_refptr<TrackedCallback>* callback) { | |
| 219 if (TrackedCallback::IsPending(*callback)) | |
| 220 (*callback)->PostAbort(); | |
| 221 } | |
| 222 | |
| 223 } // namespace ppapi | 250 } // namespace ppapi |
| OLD | NEW |