Chromium Code Reviews| Index: ppapi/proxy/vpn_provider_resource.cc |
| diff --git a/ppapi/proxy/vpn_provider_resource.cc b/ppapi/proxy/vpn_provider_resource.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..1cf173a36c79a2a4e1e6b7d2189213c0c5e28da5 |
| --- /dev/null |
| +++ b/ppapi/proxy/vpn_provider_resource.cc |
| @@ -0,0 +1,254 @@ |
| +// Copyright 2016 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "ppapi/proxy/vpn_provider_resource.h" |
| + |
| +#include "base/bind.h" |
| +#include "base/memory/ptr_util.h" |
| +#include "ppapi/c/pp_errors.h" |
| +#include "ppapi/proxy/dispatch_reply_message.h" |
| +#include "ppapi/proxy/ppapi_messages.h" |
| +#include "ppapi/shared_impl/array_var.h" |
| +#include "ppapi/shared_impl/ppapi_globals.h" |
| +#include "ppapi/shared_impl/tracked_callback.h" |
|
bbudge
2016/05/17 09:50:12
included in header, so redundant
adrian.belgun
2016/05/17 11:05:29
Done.
|
| +#include "ppapi/shared_impl/var_tracker.h" |
| + |
| +namespace ppapi { |
| +namespace proxy { |
| + |
| +VpnProviderResource::VpnProviderResource(Connection connection, |
| + PP_Instance instance) |
| + : PluginResource(connection, instance), |
| + bind_callback_(NULL), |
|
bbudge
2016/05/17 09:50:12
nullptr
adrian.belgun
2016/05/17 11:05:29
Done.
|
| + send_packet_callback_(NULL), |
| + receive_packet_callback_(NULL), |
| + receive_packet_callback_var_(NULL), |
| + bound_(false) { |
| + SendCreate(BROWSER, PpapiHostMsg_VpnProvider_Create()); |
| +} |
| + |
| +VpnProviderResource::~VpnProviderResource() {} |
| + |
| +thunk::PPB_VpnProvider_API* VpnProviderResource::AsPPB_VpnProvider_API() { |
| + return this; |
| +} |
| + |
| +void VpnProviderResource::OnReplyReceived( |
| + const ResourceMessageReplyParams& params, |
| + const IPC::Message& msg) { |
| + PPAPI_BEGIN_MESSAGE_MAP(VpnProviderResource, msg) |
| + PPAPI_DISPATCH_PLUGIN_RESOURCE_CALL(PpapiPluginMsg_VpnProvider_OnUnbind, |
| + OnPluginMsgOnUnbindReceived) |
| + PPAPI_DISPATCH_PLUGIN_RESOURCE_CALL( |
| + PpapiPluginMsg_VpnProvider_OnPacketReceived, |
| + OnPluginMsgOnPacketReceived) |
| + PPAPI_DISPATCH_PLUGIN_RESOURCE_CALL_UNHANDLED( |
| + PluginResource::OnReplyReceived(params, msg)) |
| + PPAPI_END_MESSAGE_MAP() |
| +} |
| + |
| +int32_t VpnProviderResource::Bind( |
| + const PP_Var& configuration_id, |
| + const PP_Var& configuration_name, |
| + const scoped_refptr<TrackedCallback>& callback) { |
| + if (TrackedCallback::IsPending(bind_callback_)) |
| + return PP_ERROR_INPROGRESS; |
| + |
| + StringVar* configuration_id_var = StringVar::FromPPVar(configuration_id); |
| + if (!configuration_id_var) |
| + return PP_ERROR_BADARGUMENT; |
| + StringVar* configuration_name_var = StringVar::FromPPVar(configuration_name); |
| + if (!configuration_name_var) |
| + return PP_ERROR_BADARGUMENT; |
| + |
| + bind_callback_ = callback; |
| + |
| + Call<PpapiPluginMsg_VpnProvider_BindReply>( |
| + BROWSER, PpapiHostMsg_VpnProvider_Bind(configuration_id_var->value(), |
| + configuration_name_var->value()), |
| + base::Bind(&VpnProviderResource::OnPluginMsgBindReply, this)); |
| + |
| + return PP_OK_COMPLETIONPENDING; |
| +} |
| + |
| +int32_t VpnProviderResource::SendPacket( |
| + const PP_Var& packet, |
| + const scoped_refptr<TrackedCallback>& callback) { |
| + if (!bound_) |
| + return PP_ERROR_FAILED; |
| + if (TrackedCallback::IsPending(send_packet_callback_)) |
| + return PP_ERROR_INPROGRESS; |
| + if (!ArrayBufferVar::FromPPVar(packet)) |
| + return PP_ERROR_BADARGUMENT; |
| + |
| + uint32_t id; |
| + if (send_packet_buffer_.get() && send_packet_buffer_->GetAvailable(&id)) { |
| + // Send packet immediately |
| + send_packet_buffer_->SetAvailable(id, false); |
| + return DoSendPacket(packet, id); |
| + } else { |
| + // Packet will be sent later |
| + send_packet_callback_ = callback; |
| + PpapiGlobals::Get()->GetVarTracker()->AddRefVar(packet); |
| + send_packets_.push(packet); |
| + |
| + return PP_OK_COMPLETIONPENDING; |
| + } |
| +} |
| + |
| +int32_t VpnProviderResource::DoSendPacket(const PP_Var& packet, uint32_t id) { |
| + // Convert packet to std::vector<char>, then send it. |
| + scoped_refptr<ArrayBufferVar> packet_arraybuffer = |
| + ArrayBufferVar::FromPPVar(packet); |
| + if (!packet_arraybuffer.get()) |
| + return PP_ERROR_BADARGUMENT; |
| + |
| + char* packet_pointer = static_cast<char*>(packet_arraybuffer->Map()); |
| + uint32_t packet_size = packet_arraybuffer->ByteLength(); |
| + memcpy(send_packet_buffer_->GetBuffer(id), packet_pointer, packet_size); |
| + packet_arraybuffer->Unmap(); |
| + |
| + Call<PpapiPluginMsg_VpnProvider_SendPacketReply>( |
| + BROWSER, PpapiHostMsg_VpnProvider_SendPacket(packet_size, id), |
| + base::Bind(&VpnProviderResource::OnPluginMsgSendPacketReply, this)); |
| + |
| + return PP_OK; |
| +} |
| + |
| +int32_t VpnProviderResource::ReceivePacket( |
| + PP_Var* packet, |
| + const scoped_refptr<TrackedCallback>& callback) { |
| + if (TrackedCallback::IsPending(receive_packet_callback_)) |
| + return PP_ERROR_INPROGRESS; |
| + |
| + // Return previously received packet. |
| + if (!received_packets_.empty()) { |
| + receive_packet_callback_var_ = packet; |
| + WritePacket(); |
| + return PP_OK; |
| + } |
| + |
| + // Or retain packet var and install callback. |
| + receive_packet_callback_var_ = packet; |
| + receive_packet_callback_ = callback; |
| + |
| + return PP_OK_COMPLETIONPENDING; |
| +} |
| + |
| +// Responds to PpapiPluginMsg_VpnProvider_OnUnbind |
| +void VpnProviderResource::OnPluginMsgOnUnbindReceived( |
| + const ResourceMessageReplyParams& params) { |
| + bound_ = false; |
| + |
| + // Cleanup in-flight packets. |
| + while (!received_packets_.empty()) { |
| + received_packets_.pop(); |
| + } |
| + while (!send_packets_.empty()) { |
| + PpapiGlobals::Get()->GetVarTracker()->ReleaseVar(send_packets_.front()); |
| + send_packets_.pop(); |
| + } |
| +} |
| + |
| +// Responds to PpapiPluginMsg_VpnProvider_OnPacketReceived |
| +void VpnProviderResource::OnPluginMsgOnPacketReceived( |
| + const ResourceMessageReplyParams& params, |
| + uint32_t packet_size, |
| + uint32_t id) { |
| + if (!bound_) { |
| + // Ignore packet and mark shared memory as available |
| + Post(BROWSER, PpapiHostMsg_VpnProvider_OnPacketReceivedReply(id)); |
| + return; |
| + } |
| + |
| + // Append received packet to queue. |
| + void* packet_pointer = receive_packet_buffer_->GetBuffer(id); |
| + scoped_refptr<Var> packet_var( |
| + PpapiGlobals::Get()->GetVarTracker()->MakeArrayBufferVar(packet_size, |
| + packet_pointer)); |
| + received_packets_.push(packet_var); |
| + |
| + // Mark shared memory as available for next packet |
| + Post(BROWSER, PpapiHostMsg_VpnProvider_OnPacketReceivedReply(id)); |
| + |
| + if (!TrackedCallback::IsPending(receive_packet_callback_) || |
| + TrackedCallback::IsScheduledToRun(receive_packet_callback_)) { |
| + return; |
| + } |
| + |
| + scoped_refptr<TrackedCallback> callback; |
| + callback.swap(receive_packet_callback_); |
| + WritePacket(); |
| + callback->Run(PP_OK); |
| +} |
| + |
| +// Responds to PpapiPluginMsg_VpnProvider_BindReply |
| +// Forwards to bind_callback_ |
| +void VpnProviderResource::OnPluginMsgBindReply( |
| + const ResourceMessageReplyParams& params, |
| + uint32_t queue_depth, |
| + uint32_t max_packet_size, |
| + int32_t result) { |
| + if (!TrackedCallback::IsPending(bind_callback_)) |
| + return; |
| + |
| + if (params.result() == PP_OK) { |
| + std::vector<base::SharedMemoryHandle> shm_handles; |
| + params.TakeAllSharedMemoryHandles(&shm_handles); |
| + std::unique_ptr<base::SharedMemory> send_shm( |
| + new base::SharedMemory(shm_handles[0], false)); |
| + std::unique_ptr<base::SharedMemory> receive_shm( |
| + new base::SharedMemory(shm_handles[1], false)); |
| + size_t buffer_size = queue_depth * max_packet_size; |
| + if (!send_shm->Map(buffer_size) || !receive_shm->Map(buffer_size)) { |
| + NOTREACHED(); |
| + return; |
| + } |
| + send_packet_buffer_ = base::WrapUnique(new ppapi::VpnProviderSharedBuffer( |
| + queue_depth, max_packet_size, std::move(send_shm))); |
| + receive_packet_buffer_ = |
| + base::WrapUnique(new ppapi::VpnProviderSharedBuffer( |
| + queue_depth, max_packet_size, std::move(receive_shm))); |
| + |
| + bound_ = (result == PP_OK); |
| + } |
| + |
| + scoped_refptr<TrackedCallback> callback; |
| + callback.swap(bind_callback_); |
| + callback->Run(params.result() ? params.result() : result); |
| +} |
| + |
| +// Responds to PpapiPluginMsg_VpnProvider_SendPacketReply |
| +// Forwards to send_packet_callback_ |
| +void VpnProviderResource::OnPluginMsgSendPacketReply( |
| + const ResourceMessageReplyParams& params, |
| + int32_t id) { |
| + if (!send_packets_.empty() && bound_) { |
| + // Process remaining packets |
| + DoSendPacket(send_packets_.front(), id); |
| + PpapiGlobals::Get()->GetVarTracker()->ReleaseVar(send_packets_.front()); |
| + send_packets_.pop(); |
| + } else { |
| + send_packet_buffer_->SetAvailable(id, true); |
| + |
| + // Available slots - Run callback to process new packets. |
| + if (TrackedCallback::IsPending(send_packet_callback_)) { |
| + scoped_refptr<TrackedCallback> callback; |
| + callback.swap(send_packet_callback_); |
| + callback->Run(PP_OK); |
| + } |
| + } |
| +} |
| + |
| +void VpnProviderResource::WritePacket() { |
| + if (!receive_packet_callback_var_) |
| + return; |
| + |
| + *receive_packet_callback_var_ = received_packets_.front()->GetPPVar(); |
| + received_packets_.pop(); |
| + receive_packet_callback_var_ = NULL; |
| +} |
| + |
| +} // namespace proxy |
| +} // namespace ppapi |