IPC: Make ChannelReader inherit from SupportsAttachmentBrokering.

ipc/mojo/ipc_channel_mojo.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 "ipc/mojo/ipc_channel_mojo.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/lazy_instance.h"
 #include "base/thread_task_runner_handle.h"
@@ skipping 13 matching lines @@
 
 namespace IPC {
 
 namespace {
 
 class MojoChannelFactory : public ChannelFactory {
  public:
   MojoChannelFactory(ChannelMojo::Delegate* delegate,
                      scoped_refptr<base::TaskRunner> io_runner,
                      ChannelHandle channel_handle,
-                     Channel::Mode mode)
+                     Channel::Mode mode,
+                     AttachmentBroker* broker)
       : delegate_(delegate),
         io_runner_(io_runner),
         channel_handle_(channel_handle),
-        mode_(mode) {}
+        mode_(mode),
+        broker_(broker) {}
 
   std::string GetName() const override {
     return channel_handle_.name;
   }
 
   scoped_ptr<Channel> BuildChannel(Listener* listener) override {
     return ChannelMojo::Create(delegate_, io_runner_, channel_handle_, mode_,
-                               listener);
+                               listener, broker_);
   }
 
  private:
   ChannelMojo::Delegate* delegate_;
   scoped_refptr<base::TaskRunner> io_runner_;
   ChannelHandle channel_handle_;
   Channel::Mode mode_;
+  AttachmentBroker* broker_;
 };
 
 //------------------------------------------------------------------------------
 
 class ClientChannelMojo : public ChannelMojo,
                           public ClientChannel,
                           public mojo::ErrorHandler {
  public:
   ClientChannelMojo(ChannelMojo::Delegate* delegate,
                     scoped_refptr<base::TaskRunner> io_runner,
                     const ChannelHandle& handle,
-                    Listener* listener);
+                    Listener* listener,
+                    AttachmentBroker* broker);
   ~ClientChannelMojo() override;
   // MojoBootstrap::Delegate implementation
   void OnPipeAvailable(mojo::embedder::ScopedPlatformHandle handle) override;
   // mojo::ErrorHandler implementation
   void OnConnectionError() override;
+
   // ClientChannel implementation
   void Init(
       mojo::ScopedMessagePipeHandle pipe,
       int32_t peer_pid,
       const mojo::Callback<void(int32_t)>& callback) override;
 
  private:
   void BindPipe(mojo::ScopedMessagePipeHandle handle);
 
   mojo::Binding<ClientChannel> binding_;
   base::WeakPtrFactory<ClientChannelMojo> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ClientChannelMojo);
 };
 
 ClientChannelMojo::ClientChannelMojo(ChannelMojo::Delegate* delegate,
                                      scoped_refptr<base::TaskRunner> io_runner,
                                      const ChannelHandle& handle,
-                                     Listener* listener)
-    : ChannelMojo(delegate, io_runner, handle, Channel::MODE_CLIENT, listener),
+                                     Listener* listener,
+                                     AttachmentBroker* broker)
+    : ChannelMojo(delegate,
+                  io_runner,
+                  handle,
+                  Channel::MODE_CLIENT,
+                  listener,
+                  broker),
       binding_(this),
       weak_factory_(this) {
 }
 
 ClientChannelMojo::~ClientChannelMojo() {
 }
 
 void ClientChannelMojo::OnPipeAvailable(
     mojo::embedder::ScopedPlatformHandle handle) {
   CreateMessagingPipe(handle.Pass(), base::Bind(&ClientChannelMojo::BindPipe,
@@ skipping 16 matching lines @@
   binding_.Bind(handle.Pass());
 }
 
 //------------------------------------------------------------------------------
 
 class ServerChannelMojo : public ChannelMojo, public mojo::ErrorHandler {
  public:
   ServerChannelMojo(ChannelMojo::Delegate* delegate,
                     scoped_refptr<base::TaskRunner> io_runner,
                     const ChannelHandle& handle,
-                    Listener* listener);
+                    Listener* listener,
+                    AttachmentBroker* broker);
   ~ServerChannelMojo() override;
 
   // MojoBootstrap::Delegate implementation
   void OnPipeAvailable(mojo::embedder::ScopedPlatformHandle handle) override;
   // mojo::ErrorHandler implementation
   void OnConnectionError() override;
   // Channel override
   void Close() override;
 
  private:
   void InitClientChannel(mojo::ScopedMessagePipeHandle peer_handle,
                          mojo::ScopedMessagePipeHandle handle);
 
   // ClientChannelClient implementation
   void ClientChannelWasInitialized(int32_t peer_pid);
 
   mojo::InterfacePtr<ClientChannel> client_channel_;
   mojo::ScopedMessagePipeHandle message_pipe_;
   base::WeakPtrFactory<ServerChannelMojo> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ServerChannelMojo);
 };
 
 ServerChannelMojo::ServerChannelMojo(ChannelMojo::Delegate* delegate,
                                      scoped_refptr<base::TaskRunner> io_runner,
                                      const ChannelHandle& handle,
-                                     Listener* listener)
-    : ChannelMojo(delegate, io_runner, handle, Channel::MODE_SERVER, listener),
+                                     Listener* listener,
+                                     AttachmentBroker* broker)
+    : ChannelMojo(delegate,
+                  io_runner,
+                  handle,
+                  Channel::MODE_SERVER,
+                  listener,
+                  broker),
       weak_factory_(this) {
 }
 
 ServerChannelMojo::~ServerChannelMojo() {
   Close();
 }
 
 void ServerChannelMojo::OnPipeAvailable(
     mojo::embedder::ScopedPlatformHandle handle) {
   mojo::ScopedMessagePipeHandle peer;
@@ skipping 75 matching lines @@
   // this at some point. http://crbug.com/500019
   return false;
 }
 
 // static
 scoped_ptr<ChannelMojo> ChannelMojo::Create(
     ChannelMojo::Delegate* delegate,
     scoped_refptr<base::TaskRunner> io_runner,
     const ChannelHandle& channel_handle,
     Mode mode,
-    Listener* listener) {
+    Listener* listener,
+    AttachmentBroker* broker) {
   switch (mode) {
     case Channel::MODE_CLIENT:
-      return make_scoped_ptr(
-          new ClientChannelMojo(delegate, io_runner, channel_handle, listener));
+      return make_scoped_ptr(new ClientChannelMojo(
+          delegate, io_runner, channel_handle, listener, broker));
     case Channel::MODE_SERVER:
-      return make_scoped_ptr(
-          new ServerChannelMojo(delegate, io_runner, channel_handle, listener));
+      return make_scoped_ptr(new ServerChannelMojo(
+          delegate, io_runner, channel_handle, listener, broker));
     default:
       NOTREACHED();
       return nullptr;
   }
 }
 
 // static
 scoped_ptr<ChannelFactory> ChannelMojo::CreateServerFactory(
     ChannelMojo::Delegate* delegate,
     scoped_refptr<base::TaskRunner> io_runner,
-    const ChannelHandle& channel_handle) {
+    const ChannelHandle& channel_handle,
+    AttachmentBroker* broker) {
   return make_scoped_ptr(new MojoChannelFactory(
-      delegate, io_runner, channel_handle, Channel::MODE_SERVER));
+      delegate, io_runner, channel_handle, Channel::MODE_SERVER, broker));
 }
 
 // static
 scoped_ptr<ChannelFactory> ChannelMojo::CreateClientFactory(
     ChannelMojo::Delegate* delegate,
     scoped_refptr<base::TaskRunner> io_runner,
-    const ChannelHandle& channel_handle) {
+    const ChannelHandle& channel_handle,
+    AttachmentBroker* broker) {
   return make_scoped_ptr(new MojoChannelFactory(
-      delegate, io_runner, channel_handle, Channel::MODE_CLIENT));
+      delegate, io_runner, channel_handle, Channel::MODE_CLIENT, broker));
 }
 
 ChannelMojo::ChannelMojo(ChannelMojo::Delegate* delegate,
                          scoped_refptr<base::TaskRunner> io_runner,
                          const ChannelHandle& handle,
                          Mode mode,
-                         Listener* listener)
+                         Listener* listener,
+                         AttachmentBroker* broker)
     : mode_(mode),
       listener_(listener),
       peer_pid_(base::kNullProcessId),
       io_runner_(io_runner),
       channel_info_(nullptr, ChannelInfoDeleter(nullptr)),
       weak_factory_(this) {
   // Create MojoBootstrap after all members are set as it touches
   // ChannelMojo from a different thread.
-  bootstrap_ = MojoBootstrap::Create(handle, mode, this);
+  bootstrap_ = MojoBootstrap::Create(handle, mode, this, broker);
   if (io_runner == base::MessageLoop::current()->task_runner()) {
     InitOnIOThread(delegate);
   } else {
     io_runner->PostTask(FROM_HERE,
                         base::Bind(&ChannelMojo::InitOnIOThread,
                                    base::Unretained(this), delegate));
   }
 }
 
 ChannelMojo::~ChannelMojo() {
@@ skipping 249 matching lines @@
     if (!ok) {
       LOG(ERROR) << "Failed to add new Mojo handle.";
       return MOJO_RESULT_UNKNOWN;
     }
   }
 
   return MOJO_RESULT_OK;
 }
 
 }  // namespace IPC