1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #include "base/mac/scoped_mach_port.h"
6 
7 #include "base/mac/mach_logging.h"
8 
9 namespace base {
10 namespace mac {
11 namespace internal {
12 
13 // static
Free(mach_port_t port)14 void SendRightTraits::Free(mach_port_t port) {
15   kern_return_t kr = mach_port_deallocate(mach_task_self(), port);
16   MACH_LOG_IF(ERROR, kr != KERN_SUCCESS, kr)
17       << "ScopedMachSendRight mach_port_deallocate";
18 }
19 
20 // static
Free(mach_port_t port)21 void ReceiveRightTraits::Free(mach_port_t port) {
22   kern_return_t kr =
23       mach_port_mod_refs(mach_task_self(), port, MACH_PORT_RIGHT_RECEIVE, -1);
24   MACH_LOG_IF(ERROR, kr != KERN_SUCCESS, kr)
25       << "ScopedMachReceiveRight mach_port_mod_refs";
26 }
27 
28 // static
Free(mach_port_t port)29 void PortSetTraits::Free(mach_port_t port) {
30   kern_return_t kr =
31       mach_port_mod_refs(mach_task_self(), port, MACH_PORT_RIGHT_PORT_SET, -1);
32   MACH_LOG_IF(ERROR, kr != KERN_SUCCESS, kr)
33       << "ScopedMachPortSet mach_port_mod_refs";
34 }
35 
36 }  // namespace internal
37 
CreateMachPort(ScopedMachReceiveRight * receive,ScopedMachSendRight * send,Optional<mach_port_msgcount_t> queue_limit)38 bool CreateMachPort(ScopedMachReceiveRight* receive,
39                     ScopedMachSendRight* send,
40                     Optional<mach_port_msgcount_t> queue_limit) {
41   mach_port_options_t options{};
42   options.flags = (send != nullptr ? MPO_INSERT_SEND_RIGHT : 0);
43 
44   if (queue_limit.has_value()) {
45     options.flags |= MPO_QLIMIT;
46     options.mpl.mpl_qlimit = *queue_limit;
47   }
48 
49   kern_return_t kr =
50       mach_port_construct(mach_task_self(), &options, 0,
51                           ScopedMachReceiveRight::Receiver(*receive).get());
52   if (kr != KERN_SUCCESS) {
53     MACH_LOG(ERROR, kr) << "mach_port_construct";
54     return false;
55   }
56 
57   // Multiple rights are coalesced to the same name in a task, so assign the
58   // send rights to the same name.
59   if (send) {
60     send->reset(receive->get());
61   }
62 
63   return true;
64 }
65 
RetainMachSendRight(mach_port_t port)66 ScopedMachSendRight RetainMachSendRight(mach_port_t port) {
67   kern_return_t kr =
68       mach_port_mod_refs(mach_task_self(), port, MACH_PORT_RIGHT_SEND, 1);
69   if (kr == KERN_SUCCESS)
70     return ScopedMachSendRight(port);
71   MACH_DLOG(ERROR, kr) << "mach_port_mod_refs +1";
72   return {};
73 }
74 
75 }  // namespace mac
76 }  // namespace base
77