Enable overriding interface binders for any services running in current process.

services/service_manager/public/cpp/service_context.cc

 // Copyright 2014 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 "services/service_manager/public/cpp/service_context.h"
 
 #include <utility>
 
 #include "base/bind.h"
+#include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "mojo/public/cpp/bindings/interface_request.h"
+#include "services/service_manager/public/cpp/binder_registry.h"
 #include "services/service_manager/public/cpp/connector.h"
 #include "services/service_manager/public/cpp/service.h"
 
 namespace service_manager {
 
+namespace {
+
+using IdentityToBinderRegistryMap = std::map<Identity, BinderRegistry>;
+
+base::LazyInstance<std::unique_ptr<IdentityToBinderRegistryMap>>::Leaky
+    g_overridden_binder_registries = LAZY_INSTANCE_INITIALIZER;
+
+// Returns the overridden binder registry which can intercept interface bind
+// requests to the |service|, returns nullptr if no such one.
+BinderRegistry* GetGlobalBinderRegistryForService(const Identity& service) {
+  const auto& registries = g_overridden_binder_registries.Get();
+  if (registries) {
+    // Find the registry matching the exact service identity.
+    auto it = registries->find(service);
+    if (it != registries->end())
+      return &it->second;
+    // Find the general registry matching the service name.
+    it = registries->find(Identity(service.name()));
+    if (it != registries->end())
+      return &it->second;
+  }
+
+  return nullptr;
+}
+
+}  // namespace
+
 ////////////////////////////////////////////////////////////////////////////////
 // ServiceContext, public:
 
+// static
+void ServiceContext::TestApi::SetGlobalBinder(
+    const Identity& service,
+    const std::string& interface_name,
+    const base::Callback<void(mojo::ScopedMessagePipeHandle)>& binder,
+    const scoped_refptr<base::SingleThreadTaskRunner>& task_runner) {
+  if (!g_overridden_binder_registries.Get()) {
+    g_overridden_binder_registries.Get() =
+        base::MakeUnique<IdentityToBinderRegistryMap>();
+  }
+
+  (*g_overridden_binder_registries.Get())[service].AddInterface(
+      interface_name, binder, task_runner);
+}
+
+// static
+void ServiceContext::TestApi::ClearGlobalBinders(const Identity& service) {
+  if (!g_overridden_binder_registries.Get()) {
+    return;
+  }
+
+  g_overridden_binder_registries.Get()->erase(service);
+}
+
 ServiceContext::ServiceContext(
     std::unique_ptr<service_manager::Service> service,
     mojom::ServiceRequest request,
     std::unique_ptr<Connector> connector,
     mojom::ConnectorRequest connector_request)
     : pending_connector_request_(std::move(connector_request)),
       service_(std::move(service)),
       binding_(this, std::move(request)),
       connector_(std::move(connector)),
       weak_factory_(this) {
@@ skipping 46 matching lines @@
 }
 
 void ServiceContext::OnBindInterface(
     const BindSourceInfo& source_info,
     const std::string& interface_name,
     mojo::ScopedMessagePipeHandle interface_pipe,
     const OnBindInterfaceCallback& callback) {
   // Acknowledge the request regardless of whether it's accepted.
   callback.Run();
 
+  BinderRegistry* global_registry =
+      GetGlobalBinderRegistryForService(identity_);
+  if (global_registry && global_registry->CanBindInterface(interface_name)) {
+    // Just use the binder overridden globally.
+    global_registry->BindInterface(source_info, interface_name,
+                                   std::move(interface_pipe));
+    return;
+  }
   service_->OnBindInterface(source_info, interface_name,
                             std::move(interface_pipe));
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 // ServiceContext, private:
 
 void ServiceContext::OnConnectionError() {
   if (service_->OnServiceManagerConnectionLost()) {
     // CAUTION: May delete |this|.
     QuitNow();
   }
 }
 
 }  // namespace service_manager