Add and use scoped-right-returning wrappers for Mach bootstrap routines

util/mach/mach_extensions.cc

 // Copyright 2014 The Crashpad Authors. All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
 #include "util/mach/mach_extensions.h"
 
 #include <AvailabilityMacros.h>
 #include <pthread.h>
 #include <servers/bootstrap.h>
 
 #include "base/mac/mach_logging.h"
 #include "util/mac/mac_util.h"
 
+namespace {
+
+// This forms the internal implementation for BootstrapCheckIn() and
+// BootstrapLookUp(), which follow the same logic aside from the routine called
+// and the right type returned.
+
+struct BootstrapCheckInTraits {
+  using Type = base::mac::ScopedMachReceiveRight;
+  static kern_return_t Call(mach_port_t bootstrap_port,
+                            const char* service_name,
+                            mach_port_t* service_port) {
+    return bootstrap_check_in(bootstrap_port, service_name, service_port);
+  }
+  static const char kName[];
+};
+const char BootstrapCheckInTraits::kName[] = "bootstrap_check_in";

[Robert Sesek, 2015/10/05 19:00:20]

In C++11 this can't be initialized in the struct itself?

[Mark Mentovai, 2015/10/05 19:33:03]

Robert Sesek wrote:
I wish.

mach_extensions.cc:37:21: error: in-class initializer for static data member of
type 'const char [19]' requires 'constexpr' specifier

constexpr shows up on the banned list,
https://chromium-cpp.appspot.com/#blacklist_banned. But even if it didn’t, we’d
need to provide storage:

Undefined symbols for architecture x86_64:
  "(anonymous namespace)::BootstrapLookUpTraits::kName", referenced from:
      crashpad::BootstrapLookUp(std::__1::basic_string<char,
std::__1::char_traits<char>, std::__1::allocator<char> > const&) in
libcrashpad_util.a(crashpad_util.mach_extensions.o)

So it’s not much help. We can write

    static const char kName[];
  };
  const char BootstrapCheckInTraits::kName[] = "bootstrap_check_in";

or the banned

    static constexpr char kName[] = "bootstrap_check_in";
  };
  constexpr char BootstrapCheckInTraits::kName[];

+
+struct BootstrapLookUpTraits {
+  using Type = base::mac::ScopedMachSendRight;
+  static kern_return_t Call(mach_port_t bootstrap_port,
+                            const char* service_name,
+                            mach_port_t* service_port) {
+    return bootstrap_look_up(bootstrap_port, service_name, service_port);
+  }
+  static const char kName[];
+};
+const char BootstrapLookUpTraits::kName[] = "bootstrap_look_up";
+
+template <typename Traits>
+typename Traits::Type BootstrapCheckInOrLookUp(
+    const std::string& service_name) {
+  // bootstrap_check_in() and bootstrap_look_up() silently truncate service
+  // names longer than BOOTSTRAP_MAX_NAME_LEN. This check ensures that the name
+  // will not be truncated.
+  if (service_name.size() >= BOOTSTRAP_MAX_NAME_LEN) {
+    LOG(ERROR) << Traits::kName << " " << service_name << ": name too long";
+    return typename Traits::Type(MACH_PORT_NULL);
+  }
+
+  mach_port_t service_port;
+  kern_return_t kr = Traits::Call(bootstrap_port,
+                                  service_name.c_str(),
+                                  &service_port);
+  if (kr != BOOTSTRAP_SUCCESS) {
+    BOOTSTRAP_LOG(ERROR, kr) << Traits::kName << " " << service_name;
+    service_port = MACH_PORT_NULL;
+  }
+
+  return typename Traits::Type(service_port);
+}
+
+}  // namespace
+
 namespace crashpad {
 
 thread_t MachThreadSelf() {
   // The pthreads library keeps its own copy of the thread port. Using it does
   // not increment its reference count.
   return pthread_mach_thread_np(pthread_self());
 }
 
 mach_port_t NewMachPort(mach_port_right_t right) {
   mach_port_t port = MACH_PORT_NULL;
@@ skipping 56 matching lines @@
     return kExcMaskValid_10_6;
   }
 #endif
 
   // 10.11 added EXC_MASK_CORPSE_NOTIFY. See 10.11 <mach/exception_types.h>.
   const exception_mask_t kExcMaskValid_10_11 =
       kExcMaskValid_10_6 | EXC_MASK_CORPSE_NOTIFY;
   return kExcMaskValid_10_11;
 }
 
+base::mac::ScopedMachReceiveRight BootstrapCheckIn(
+    const std::string& service_name) {
+  return BootstrapCheckInOrLookUp<BootstrapCheckInTraits>(service_name);
+}
+
+base::mac::ScopedMachSendRight BootstrapLookUp(
+    const std::string& service_name) {
+  return BootstrapCheckInOrLookUp<BootstrapLookUpTraits>(service_name);
+}
+
 base::mac::ScopedMachSendRight SystemCrashReporterHandler() {
-  const char kSystemCrashReporterServiceName[] = "com.apple.ReportCrash";
-  exception_handler_t system_crash_reporter_handler;
-  kern_return_t kr = bootstrap_look_up(bootstrap_port,
-                                       kSystemCrashReporterServiceName,
-                                       &system_crash_reporter_handler);
-  if (kr != BOOTSTRAP_SUCCESS) {
-    BOOTSTRAP_LOG(ERROR, kr) << "bootstrap_look_up "
-                             << kSystemCrashReporterServiceName;
-    system_crash_reporter_handler = MACH_PORT_NULL;
-  }
-
-  return base::mac::ScopedMachSendRight(system_crash_reporter_handler);
+  return BootstrapLookUp("com.apple.ReportCrash");
 }
 
 }  // namespace crashpad