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