| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 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 "ipc/test_util_mac.h" | |
| 6 | |
| 7 #include <mach/mach_vm.h> | |
| 8 #include <servers/bootstrap.h> | |
| 9 #include <stddef.h> | |
| 10 | |
| 11 #include "base/mac/mach_logging.h" | |
| 12 #include "base/mac/scoped_mach_port.h" | |
| 13 #include "base/rand_util.h" | |
| 14 #include "base/strings/stringprintf.h" | |
| 15 | |
| 16 namespace { | |
| 17 | |
| 18 // Structs used to pass a Mach port over mach_msg(). | |
| 19 struct MachSendPortMessage { | |
| 20 mach_msg_header_t header; | |
| 21 mach_msg_body_t body; | |
| 22 mach_msg_port_descriptor_t data; | |
| 23 }; | |
| 24 | |
| 25 struct MachReceivePortMessage : public MachSendPortMessage { | |
| 26 mach_msg_trailer_t trailer; | |
| 27 }; | |
| 28 | |
| 29 } // namespace | |
| 30 | |
| 31 namespace IPC { | |
| 32 | |
| 33 std::string CreateRandomServiceName() { | |
| 34 return base::StringPrintf("TestUtilMac.%llu", base::RandUint64()); | |
| 35 } | |
| 36 | |
| 37 base::mac::ScopedMachReceiveRight BecomeMachServer(const char* service_name) { | |
| 38 mach_port_t port; | |
| 39 kern_return_t kr = bootstrap_check_in(bootstrap_port, service_name, &port); | |
| 40 MACH_CHECK(kr == KERN_SUCCESS, kr) << "BecomeMachServer "; | |
| 41 return base::mac::ScopedMachReceiveRight(port); | |
| 42 } | |
| 43 | |
| 44 base::mac::ScopedMachSendRight LookupServer(const char* service_name) { | |
| 45 mach_port_t server_port; | |
| 46 kern_return_t kr = | |
| 47 bootstrap_look_up(bootstrap_port, service_name, &server_port); | |
| 48 MACH_CHECK(kr == KERN_SUCCESS, kr) << "LookupServer"; | |
| 49 return base::mac::ScopedMachSendRight(server_port); | |
| 50 } | |
| 51 | |
| 52 base::mac::ScopedMachReceiveRight MakeReceivingPort() { | |
| 53 mach_port_t client_port; | |
| 54 kern_return_t kr = mach_port_allocate(mach_task_self(), | |
| 55 MACH_PORT_RIGHT_RECEIVE, &client_port); | |
| 56 MACH_CHECK(kr == KERN_SUCCESS, kr) << "MakeReceivingPort"; | |
| 57 return base::mac::ScopedMachReceiveRight(client_port); | |
| 58 } | |
| 59 | |
| 60 base::mac::ScopedMachSendRight ReceiveMachPort(mach_port_t port_to_listen_on) { | |
| 61 MachReceivePortMessage recv_msg; | |
| 62 mach_msg_header_t* recv_hdr = &recv_msg.header; | |
| 63 recv_hdr->msgh_local_port = port_to_listen_on; | |
| 64 recv_hdr->msgh_size = sizeof(recv_msg); | |
| 65 kern_return_t kr = | |
| 66 mach_msg(recv_hdr, MACH_RCV_MSG, 0, recv_hdr->msgh_size, | |
| 67 port_to_listen_on, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL); | |
| 68 MACH_CHECK(kr == KERN_SUCCESS, kr) << "ReceiveMachPort"; | |
| 69 mach_port_t other_task_port = recv_msg.data.name; | |
| 70 return base::mac::ScopedMachSendRight(other_task_port); | |
| 71 } | |
| 72 | |
| 73 // Passes a copy of the send right of |port_to_send| to |receiving_port|. | |
| 74 void SendMachPort(mach_port_t receiving_port, | |
| 75 mach_port_t port_to_send, | |
| 76 int disposition) { | |
| 77 MachSendPortMessage send_msg; | |
| 78 send_msg.header.msgh_bits = | |
| 79 MACH_MSGH_BITS(MACH_MSG_TYPE_COPY_SEND, 0) | MACH_MSGH_BITS_COMPLEX; | |
| 80 send_msg.header.msgh_size = sizeof(send_msg); | |
| 81 send_msg.header.msgh_remote_port = receiving_port; | |
| 82 send_msg.header.msgh_local_port = MACH_PORT_NULL; | |
| 83 send_msg.header.msgh_reserved = 0; | |
| 84 send_msg.header.msgh_id = 0; | |
| 85 send_msg.body.msgh_descriptor_count = 1; | |
| 86 send_msg.data.name = port_to_send; | |
| 87 send_msg.data.disposition = disposition; | |
| 88 send_msg.data.type = MACH_MSG_PORT_DESCRIPTOR; | |
| 89 int kr = mach_msg(&send_msg.header, MACH_SEND_MSG, send_msg.header.msgh_size, | |
| 90 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL); | |
| 91 MACH_CHECK(kr == KERN_SUCCESS, kr) << "SendMachPort"; | |
| 92 } | |
| 93 | |
| 94 mach_msg_type_number_t GetActiveNameCount() { | |
| 95 mach_port_name_array_t name_array; | |
| 96 mach_msg_type_number_t names_count; | |
| 97 mach_port_type_array_t type_array; | |
| 98 mach_msg_type_number_t types_count; | |
| 99 kern_return_t kr = mach_port_names(mach_task_self(), &name_array, | |
| 100 &names_count, &type_array, &types_count); | |
| 101 MACH_CHECK(kr == KERN_SUCCESS, kr) << "GetActiveNameCount"; | |
| 102 return names_count; | |
| 103 } | |
| 104 | |
| 105 mach_port_urefs_t GetMachRefCount(mach_port_name_t name, | |
| 106 mach_port_right_t right) { | |
| 107 mach_port_urefs_t ref_count; | |
| 108 kern_return_t kr = mach_port_get_refs(mach_task_self(), name, | |
| 109 MACH_PORT_RIGHT_SEND, &ref_count); | |
| 110 MACH_CHECK(kr == KERN_SUCCESS, kr) << "GetRefCount"; | |
| 111 return ref_count; | |
| 112 } | |
| 113 | |
| 114 void IncrementMachRefCount(mach_port_name_t name, mach_port_right_t right) { | |
| 115 kern_return_t kr = mach_port_mod_refs(mach_task_self(), name, right, 1); | |
| 116 MACH_CHECK(kr == KERN_SUCCESS, kr) << "GetRefCount"; | |
| 117 } | |
| 118 | |
| 119 bool GetMachProtections(void* address, size_t size, int* current, int* max) { | |
| 120 vm_region_info_t region_info; | |
| 121 mach_vm_address_t mem_address = reinterpret_cast<mach_vm_address_t>(address); | |
| 122 mach_vm_size_t mem_size = size; | |
| 123 vm_region_basic_info_64 basic_info; | |
| 124 | |
| 125 region_info = reinterpret_cast<vm_region_recurse_info_t>(&basic_info); | |
| 126 vm_region_flavor_t flavor = VM_REGION_BASIC_INFO_64; | |
| 127 memory_object_name_t memory_object; | |
| 128 mach_msg_type_number_t count = VM_REGION_BASIC_INFO_COUNT_64; | |
| 129 | |
| 130 kern_return_t kr = | |
| 131 mach_vm_region(mach_task_self(), &mem_address, &mem_size, flavor, | |
| 132 region_info, &count, &memory_object); | |
| 133 if (kr != KERN_SUCCESS) { | |
| 134 MACH_LOG(ERROR, kr) << "Failed to get region info."; | |
| 135 return false; | |
| 136 } | |
| 137 | |
| 138 *current = basic_info.protection; | |
| 139 *max = basic_info.max_protection; | |
| 140 return true; | |
| 141 } | |
| 142 | |
| 143 } // namespace IPC | |
| OLD | NEW |