Add ability to bind a different InterfaceRegistry/InterfaceProvider to an incoming Connection.

services/shell/public/cpp/lib/connection_impl.h

 // 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.
 
 #ifndef SERVICES_SHELL_PUBLIC_CPP_LIB_CONNECTION_IMPL_H_
 #define SERVICES_SHELL_PUBLIC_CPP_LIB_CONNECTION_IMPL_H_
 
 #include <stdint.h>
 
 #include <set>
@@ skipping 15 matching lines @@
 // other.
 class ConnectionImpl : public Connection {
  public:
   ConnectionImpl();
   // |allowed_interfaces| are the set of interfaces that the shell has allowed
   // an application to expose to another application. If this set contains only
   // the string value "*" all interfaces may be exposed.
   ConnectionImpl(const std::string& connection_name,
                  const Identity& remote,
                  uint32_t remote_id,
-                 shell::mojom::InterfaceProviderPtr remote_interfaces,
-                 shell::mojom::InterfaceProviderRequest local_interfaces,
                  const CapabilityRequest& capability_request,
                  State initial_state);
   ~ConnectionImpl() override;
 
+  // Sets the local registry & remote provider, transferring ownership to the
+  // ConnectionImpl.
+  void SetExposedInterfaces(
+      std::unique_ptr<InterfaceRegistry> exposed_interfaces);
+  void SetRemoteInterfaces(
+      std::unique_ptr<InterfaceProvider> remote_interfaces);
+
+  // Sets the local registry & remote provider, without transferring ownership.
+  void set_exposed_interfaces(InterfaceRegistry* exposed_interfaces) {
+    exposed_interfaces_ = exposed_interfaces;
+  }
+  void set_remote_interfaces(InterfaceProvider* remote_interfaces) {
+    remote_interfaces_ = remote_interfaces;
+  }
+
   shell::mojom::Connector::ConnectCallback GetConnectCallback();
 
  private:
   // Connection:
   bool HasCapabilityClass(const std::string& class_name) const override;
   const std::string& GetConnectionName() override;
   const Identity& GetRemoteIdentity() const override;
   void SetConnectionLostClosure(const base::Closure& handler) override;
   shell::mojom::ConnectResult GetResult() const override;
   bool IsPending() const override;
@@ skipping 10 matching lines @@
                              uint32_t target_application_id);
 
   const std::string connection_name_;
   Identity remote_;
   uint32_t remote_id_ = shell::mojom::kInvalidInstanceID;
 
   State state_;
   shell::mojom::ConnectResult result_ = shell::mojom::ConnectResult::SUCCEEDED;
   std::vector<base::Closure> connection_completed_callbacks_;
 
-  InterfaceRegistry interfaces_;
-  InterfaceProvider remote_interfaces_;
+  InterfaceRegistry* exposed_interfaces_ = nullptr;
+  InterfaceProvider* remote_interfaces_ = nullptr;
+
+  std::unique_ptr<InterfaceRegistry> exposed_interfaces_owner_;
+  std::unique_ptr<InterfaceProvider> remote_interfaces_owner_;
 
   const CapabilityRequest capability_request_;
   const bool allow_all_interfaces_;
 
   base::WeakPtrFactory<ConnectionImpl> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ConnectionImpl);
 };
 
 }  // namespace internal
 }  // namespace shell
 
 #endif  // SERVICES_SHELL_PUBLIC_CPP_LIB_CONNECTION_IMPL_H_