ipc: Move methods into AttachmentBroker interface.

ipc/attachment_broker_privileged_mac.cc

 // Copyright 2015 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/attachment_broker_privileged_mac.h"
 
 #include "base/mac/scoped_mach_port.h"
 #include "base/memory/shared_memory.h"
+#include "base/process/port_provider_mac.h"
 #include "base/process/process.h"
 #include "ipc/attachment_broker_messages.h"
 #include "ipc/brokerable_attachment.h"
 #include "ipc/ipc_channel.h"
 #include "ipc/mach_port_attachment_mac.h"
 
 namespace {
 
 // Struct for sending a complex Mach message.
 struct MachSendComplexMessage {
@@ skipping 26 matching lines @@
                   0,                // receive limit
                   MACH_PORT_NULL,   // receive name
                   0,                // timeout
                   MACH_PORT_NULL);  // notification port
 }
 
 }  // namespace
 
 namespace IPC {
 
-AttachmentBrokerPrivilegedMac::AttachmentBrokerPrivilegedMac()
-    : port_provider_(nullptr) {}
-
+AttachmentBrokerPrivilegedMac::AttachmentBrokerPrivilegedMac() {}
 AttachmentBrokerPrivilegedMac::~AttachmentBrokerPrivilegedMac() {}
 
-void AttachmentBrokerPrivilegedMac::SetPortProvider(
-    base::PortProvider* port_provider) {
-  CHECK(!port_provider_);
-  port_provider_ = port_provider;
-}
-
 bool AttachmentBrokerPrivilegedMac::SendAttachmentToProcess(
     BrokerableAttachment* attachment,
     base::ProcessId destination_process) {
   switch (attachment->GetBrokerableType()) {
     case BrokerableAttachment::MACH_PORT: {
       internal::MachPortAttachmentMac* mach_port_attachment =
           static_cast<internal::MachPortAttachmentMac*>(attachment);
       MachPortWireFormat wire_format =
           mach_port_attachment->GetWireFormat(destination_process);
 
@@ skipping 97 matching lines @@
   }
 
   LogError(DESTINATION_FOUND);
   sender->Send(new AttachmentBrokerMsg_MachPortHasBeenDuplicated(wire_format));
 }
 
 mach_port_name_t AttachmentBrokerPrivilegedMac::CreateIntermediateMachPort(
     base::ProcessId pid,
     base::mac::ScopedMachSendRight port_to_insert) {
   DCHECK_NE(pid, base::GetCurrentProcId());
-  mach_port_t task_port = port_provider_->TaskForPid(pid);
+  mach_port_t task_port = get_port_provider()->TaskForPid(pid);
   if (task_port == MACH_PORT_NULL) {
     // TODO(erikchen): UMA metric.
     return MACH_PORT_NULL;
   }
   return CreateIntermediateMachPort(
       task_port, base::mac::ScopedMachSendRight(port_to_insert.release()));
 }
 
 mach_port_name_t AttachmentBrokerPrivilegedMac::CreateIntermediateMachPort(
     mach_port_t task_port,
@@ skipping 53 matching lines @@
     base::ProcessId pid,
     mach_port_name_t named_right) {
   if (pid == base::GetCurrentProcId()) {
     kern_return_t kr = mach_port_mod_refs(mach_task_self(), named_right,
                                           MACH_PORT_RIGHT_SEND, 1);
     if (kr != KERN_SUCCESS)
       return base::mac::ScopedMachSendRight(MACH_PORT_NULL);
     return base::mac::ScopedMachSendRight(named_right);
   }
 
-  mach_port_t task_port = port_provider_->TaskForPid(pid);
+  mach_port_t task_port = get_port_provider()->TaskForPid(pid);
   return ExtractNamedRight(task_port, named_right);
 }
 
 base::mac::ScopedMachSendRight AttachmentBrokerPrivilegedMac::ExtractNamedRight(
     mach_port_t task_port,
     mach_port_name_t named_right) {
   mach_port_t extracted_right = MACH_PORT_NULL;
   mach_msg_type_name_t extracted_right_type;
   kern_return_t kr =
       mach_port_extract_right(task_port, named_right, MACH_MSG_TYPE_COPY_SEND,
@@ skipping 17 matching lines @@
 
 AttachmentBrokerPrivilegedMac::MachPortWireFormat
 AttachmentBrokerPrivilegedMac::CopyWireFormat(
     const MachPortWireFormat& wire_format,
     uint32_t mach_port) {
   return MachPortWireFormat(mach_port, wire_format.destination_process,
                             wire_format.attachment_id);
 }
 
 }  // namespace IPC