PPAPI: Force async callback invocation option.

webkit/plugins/ppapi/ppb_transport_impl.cc

 // Copyright (c) 2011 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 "webkit/plugins/ppapi/ppb_transport_impl.h"
 
 #include "base/message_loop.h"
 #include "base/string_util.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/socket/socket.h"
 #include "ppapi/c/dev/ppb_transport_dev.h"
 #include "ppapi/c/pp_completion_callback.h"
 #include "ppapi/c/pp_errors.h"
+#include "ppapi/thunk/common.h"
 #include "webkit/plugins/ppapi/common.h"
 #include "webkit/plugins/ppapi/plugin_module.h"
 #include "webkit/plugins/ppapi/ppapi_plugin_instance.h"
 #include "webkit/plugins/ppapi/var.h"
 
+using ppapi::thunk::MayForceCallback;
 using webkit_glue::P2PTransport;
 
 namespace webkit {
 namespace ppapi {
 
 namespace {
 
 const char kUdpProtocolName[] = "udp";
 const char kTcpProtocolName[] = "tcp";
 
@@ skipping 16 matching lines @@
 
 PP_Bool IsWritable(PP_Resource resource) {
   scoped_refptr<PPB_Transport_Impl> t(
       Resource::GetAs<PPB_Transport_Impl>(resource));
   return BoolToPPBool((t.get()) ? t->IsWritable() : false);
 }
 
 int32_t Connect(PP_Resource resource, PP_CompletionCallback callback) {
   scoped_refptr<PPB_Transport_Impl> t(
       Resource::GetAs<PPB_Transport_Impl>(resource));
-  return (t.get()) ? t->Connect(callback) : PP_ERROR_BADRESOURCE;
+  if (!t.get())
+    return MayForceCallback(callback, PP_ERROR_BADRESOURCE);
+  return MayForceCallback(callback, t->Connect(callback));
 }
 
 int32_t GetNextAddress(PP_Resource resource, PP_Var* address,
                        PP_CompletionCallback callback) {
   scoped_refptr<PPB_Transport_Impl> t(
       Resource::GetAs<PPB_Transport_Impl>(resource));
-  return (t.get())? t->GetNextAddress(address, callback) : PP_ERROR_BADRESOURCE;
+  if (!t.get())
+    return MayForceCallback(callback, PP_ERROR_BADRESOURCE);
+  return MayForceCallback(callback, t->GetNextAddress(address, callback));
 }
 
 int32_t ReceiveRemoteAddress(PP_Resource resource, PP_Var address) {
   scoped_refptr<PPB_Transport_Impl> t(
       Resource::GetAs<PPB_Transport_Impl>(resource));
   return (t.get())? t->ReceiveRemoteAddress(address) : PP_ERROR_BADRESOURCE;
 }
 
 int32_t Recv(PP_Resource resource, void* data, uint32_t len,
              PP_CompletionCallback callback) {
   scoped_refptr<PPB_Transport_Impl> t(
       Resource::GetAs<PPB_Transport_Impl>(resource));
-  return (t.get())? t->Recv(data, len, callback) : PP_ERROR_BADRESOURCE;
+  if (!t.get())
+    return MayForceCallback(callback, PP_ERROR_BADRESOURCE);
+  return MayForceCallback(callback, t->Recv(data, len, callback));
 }
 
 int32_t Send(PP_Resource resource, const void* data, uint32_t len,
              PP_CompletionCallback callback) {
   scoped_refptr<PPB_Transport_Impl> t(
       Resource::GetAs<PPB_Transport_Impl>(resource));
-  return (t.get())? t->Send(data, len, callback) : PP_ERROR_BADRESOURCE;
+  if (!t.get())
+    return MayForceCallback(callback, PP_ERROR_BADRESOURCE);
+  return MayForceCallback(callback, t->Send(data, len, callback));
 }
 
 // Disconnects from the remote peer.
 int32_t Close(PP_Resource resource) {
   scoped_refptr<PPB_Transport_Impl> t(
       Resource::GetAs<PPB_Transport_Impl>(resource));
   return (t.get())? t->Close() : PP_ERROR_BADRESOURCE;
 }
 
 const PPB_Transport_Dev ppb_transport = {
@@ skipping 227 matching lines @@
 void PPB_Transport_Impl::OnWritten(int result) {
   DCHECK(send_callback_.get() && !send_callback_->completed());
 
   scoped_refptr<TrackedCompletionCallback> callback;
   callback.swap(send_callback_);
   callback->Run(MapNetError(result));
 }
 
 }  // namespace ppapi
 }  // namespace webkit