Remove all uses of the global Dispatcher Get function.

ppapi/proxy/plugin_dispatcher.cc

 // Copyright (c) 2010 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/plugin_dispatcher.h"
 
 #include <map>
 
 #include "base/compiler_specific.h"
 #include "base/logging.h"
 #include "ipc/ipc_message.h"
 #include "ipc/ipc_sync_channel.h"
 #include "ppapi/c/pp_errors.h"
 #include "ppapi/proxy/interface_proxy.h"
 #include "ppapi/proxy/plugin_var_serialization_rules.h"
 #include "ppapi/proxy/ppapi_messages.h"
 #include "ppapi/proxy/ppp_class_proxy.h"
 
 namespace pp {
 namespace proxy {
 
 namespace {
 
-PluginDispatcher* g_dispatcher = NULL;
+typedef std::map<PP_Instance, PluginDispatcher*> InstanceToDispatcherMap;
+InstanceToDispatcherMap* g_instance_to_dispatcher = NULL;
 
 const void* GetInterfaceFromDispatcher(const char* interface) {
-  // TODO(brettw) need some kind of lock for multi-thread access.
-  return pp::proxy::PluginDispatcher::Get()->GetProxiedInterface(interface);
+  // All interfaces the plugin requests of the browser are "PPB".
+  const InterfaceProxy::Info* info = Dispatcher::GetPPBInterfaceInfo(interface);
+  if (!info)
+    return NULL;
+  return info->interface;
 }
 
 }  // namespace
 
 PluginDispatcher::PluginDispatcher(base::ProcessHandle remote_process_handle,
                                    GetInterfaceFunc get_interface,
                                    InitModuleFunc init_module,
                                    ShutdownModuleFunc shutdown_module)
     : Dispatcher(remote_process_handle, get_interface),
       init_module_(init_module),
       shutdown_module_(shutdown_module) {
   SetSerializationRules(new PluginVarSerializationRules);
 
   // As a plugin, we always support the PPP_Class interface. There's no
   // GetInterface call or name for it, so we insert it into our table now.
-  InjectProxy(INTERFACE_ID_PPP_CLASS, "$Internal_PPP_Class",
-              new PPP_Class_Proxy(this));
+  target_proxies_[INTERFACE_ID_PPP_CLASS].reset(new PPP_Class_Proxy(this));
 }
 
 PluginDispatcher::~PluginDispatcher() {
   if (shutdown_module_)
     shutdown_module_();
 }
 
 // static
-PluginDispatcher* PluginDispatcher::Get() {
-  return g_dispatcher;
-}
-
-// static
-void PluginDispatcher::SetGlobal(PluginDispatcher* dispatcher) {
-  DCHECK(!dispatcher || !g_dispatcher);
-  g_dispatcher = dispatcher;
-}
-
-// static
 PluginDispatcher* PluginDispatcher::GetForInstance(PP_Instance instance) {
-  // TODO(brettw) implement "real" per-instance dispatcher map.
-  DCHECK(instance != 0);
-  return Get();
+  if (!g_instance_to_dispatcher)
+    return NULL;
+  InstanceToDispatcherMap::iterator found = g_instance_to_dispatcher->find(
+      instance);
+  if (found == g_instance_to_dispatcher->end())
+    return NULL;
+  return found->second;
+}
+
+// static
+void PluginDispatcher::SetForInstance(PP_Instance instance,
+                                      PluginDispatcher* dispatcher) {
+  if (!g_instance_to_dispatcher)
+    g_instance_to_dispatcher = new InstanceToDispatcherMap;
+  (*g_instance_to_dispatcher)[instance] = dispatcher;
+}
+
+// static
+void PluginDispatcher::RemoveForInstance(PP_Instance instance) {
+  if (!g_instance_to_dispatcher)
+    return;
+  InstanceToDispatcherMap::iterator found = g_instance_to_dispatcher->find(
+      instance);
+  if (found != g_instance_to_dispatcher->end())
+    g_instance_to_dispatcher->erase(found);
 }
 
 bool PluginDispatcher::IsPlugin() const {
   return true;
 }
 
 bool PluginDispatcher::OnMessageReceived(const IPC::Message& msg) {
+  // Handle common control messages.
+  if (Dispatcher::OnMessageReceived(msg))
+    return true;
+
   if (msg.routing_id() == MSG_ROUTING_CONTROL) {
     // Handle some plugin-specific control messages.
     bool handled = true;
     IPC_BEGIN_MESSAGE_MAP(PluginDispatcher, msg)
+      IPC_MESSAGE_HANDLER(PpapiMsg_SupportsInterface, OnMsgSupportsInterface)
       IPC_MESSAGE_HANDLER(PpapiMsg_InitializeModule, OnMsgInitializeModule)
       IPC_MESSAGE_HANDLER(PpapiMsg_Shutdown, OnMsgShutdown)
-
-      // Forward all other control messages to the superclass.
-      IPC_MESSAGE_UNHANDLED(handled = Dispatcher::OnMessageReceived(msg))
     IPC_END_MESSAGE_MAP()
     return handled;
   }
 
-  // All non-control messages get handled by the superclass.
-  return Dispatcher::OnMessageReceived(msg);
+  if (msg.routing_id() <= 0 && msg.routing_id() >= INTERFACE_ID_COUNT) {
+    // Host is sending us garbage. Since it's supposed to be trusted, this
+    // isn't supposed to happen.

[piman, 2011/02/07 21:29:22]

Should we just crash here ? To prevent a compromised renderer from escalating
privilege ?

[brettw, 2011/02/08 00:28:04]

Sure.

+    NOTREACHED();
+    return true;
+  }
+
+  // There are two cases:
+  //
+  // * The first case is that the host is calling a PPP interface. It will
+  //   always do a check for the interface before sending messages, and this
+  //   will create the necessary interface proxy at that time. So when we
+  //   actually receive a message, we know such a proxy will exist.
+  //
+  // * The second case is that the host is sending a response to the plugin
+  //   side of a PPB interface (some, like the URL loader, have complex
+  //   response messages). Since the host is trusted and not supposed to be
+  //   doing silly things, we can just create a PPB proxy project on demand the
+  //   first time it's needed.
+
+  InterfaceProxy* proxy = target_proxies_[msg.routing_id()].get();
+  if (!proxy) {
+    // Handle the first time the host calls a PPB reply interface by
+    // autocreating it.
+    const InterfaceProxy::Info* info = GetPPBInterfaceInfo(
+        static_cast<InterfaceID>(msg.routing_id()));
+    if (!info) {
+      NOTREACHED();
+      return true;
+    }
+    proxy = info->create(this, NULL);
+    target_proxies_[info->id].reset(proxy);
+  }
+
+  return proxy->OnMessageReceived(msg);
 }
 
 void PluginDispatcher::DidCreateInstance(PP_Instance instance) {
   instance_map_[instance] = InstanceData();
 }
 
 void PluginDispatcher::DidDestroyInstance(PP_Instance instance) {
   InstanceDataMap::iterator it = instance_map_.find(instance);
   if (it != instance_map_.end())
     instance_map_.erase(it);
 }
 
 InstanceData* PluginDispatcher::GetInstanceData(PP_Instance instance) {
   InstanceDataMap::iterator it = instance_map_.find(instance);
   return (it == instance_map_.end()) ? NULL : &it->second;
 }
 
 void PluginDispatcher::OnMsgInitializeModule(PP_Module pp_module,
                                              bool* result) {
   set_pp_module(pp_module);
   *result = init_module_(pp_module, &GetInterfaceFromDispatcher) == PP_OK;
 }
 
 void PluginDispatcher::OnMsgShutdown() {
   if (shutdown_module_)
     shutdown_module_();
   MessageLoop::current()->Quit();
 }
 
+void PluginDispatcher::OnMsgSupportsInterface(
+    const std::string& interface_name,
+    bool* result) {
+  *result = false;
+
+  // Setup a proxy for receiving the messages from this interface.
+  const InterfaceProxy::Info* info = GetPPPInterfaceInfo(interface_name);
+  if (!info)
+    return;  // Interface not supported by proxy.
+
+  // Check for a cached result.
+  if (target_proxies_[info->id].get()) {
+    *result = true;
+    return;
+  }
+
+  // Query the plugin & cache the result.
+  const void* interface_functions = GetLocalInterface(interface_name.c_str());
+  if (!interface_functions)
+    return;
+  target_proxies_[info->id].reset(info->create(this, interface_functions));
+  *result = true;
+}
+
 }  // namespace proxy
 }  // namespace pp