Add BootstrapCheckIn and its test

util/mach/bootstrap_test.cc

Index: util/mach/bootstrap_test.cc
diff --git a/util/mach/bootstrap_test.cc b/util/mach/bootstrap_test.cc
new file mode 100644
index 0000000000000000000000000000000000000000..fbf69cad16d38cebf05213989e096e08e108dc5d
--- /dev/null
+++ b/util/mach/bootstrap_test.cc
@@ -0,0 +1,134 @@
+// 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/bootstrap.h"
+
+#include <mach/mach.h>
+#include <servers/bootstrap.h>
+#include <string.h>
+
+#include <algorithm>
+#include <string>
+
+#include "base/basictypes.h"
+#include "base/mac/scoped_mach_port.h"
+#include "base/rand_util.h"
+#include "gtest/gtest.h"
+#include "util/test/mac/mach_errors.h"
+
+namespace {
+
+using namespace crashpad;
+using namespace crashpad::test;
+using namespace testing;
+
+TEST(Bootstrap, BootstrapCheckIn) {
+  std::string service_name = "com.googlecode.crashpad.test.bootstrap.";
+  for (int index = 0; index < 16; ++index) {
+    service_name.append(1, base::RandInt('A', 'Z'));
+  }
+
+  // The service shouldn’t be registered in the bootstrap namespace yet.
+  mach_port_t client_port = MACH_PORT_NULL;
+  kern_return_t kr =
+      bootstrap_look_up(bootstrap_port, service_name.c_str(), &client_port);
+  ASSERT_EQ(BOOTSTRAP_UNKNOWN_SERVICE, kr)
+      << BootstrapErrorMessage(kr, "bootstrap_look_up");
+
+  // Check in, getting a receive right.
+  mach_port_t server_port;
+  kr = BootstrapCheckIn(bootstrap_port, service_name.c_str(), &server_port);
+  ASSERT_EQ(BOOTSTRAP_SUCCESS, kr)
+      << BootstrapErrorMessage(kr, "bootstrap_check_in");
+  ASSERT_NE(static_cast<mach_port_t>(MACH_PORT_NULL), server_port);
+  base::mac::ScopedMachReceiveRight server_port_owner(server_port);
+
+  // A subsequent checkin attempt should fail.
+  mach_port_t fail_port = MACH_PORT_NULL;
+  kr = BootstrapCheckIn(bootstrap_port, service_name.c_str(), &fail_port);
+  EXPECT_EQ(BOOTSTRAP_SERVICE_ACTIVE, kr);
+  EXPECT_EQ(static_cast<mach_port_t>(MACH_PORT_NULL), fail_port);
+
+  // Look up the service, getting a send right.
+  kr = bootstrap_look_up(bootstrap_port, service_name.c_str(), &client_port);
+  ASSERT_EQ(BOOTSTRAP_SUCCESS, kr)
+      << BootstrapErrorMessage(kr, "bootstrap_look_up");
+  base::mac::ScopedMachSendRight client_port_owner(client_port);
+  EXPECT_NE(static_cast<mach_port_t>(MACH_PORT_NULL), client_port);
+
+  // Have the “client” send a message to the “server”.
+  struct SendMessage {
+    mach_msg_header_t header;
+    char data[64];
+  };
+  SendMessage send_message = {};
+  send_message.header.msgh_bits = MACH_MSGH_BITS(MACH_MSG_TYPE_COPY_SEND, 0);
+  send_message.header.msgh_size = sizeof(send_message);
+  send_message.header.msgh_remote_port = client_port;
+  const char kMessageData[] = "Mach messaging is not a big truck";
+  const size_t message_data_size =
+      std::min(sizeof(kMessageData), sizeof(send_message.data));
+  memcpy(send_message.data, kMessageData, message_data_size);
+
+  // The receive operation happens in this same thread after the send, so use a
+  // non-blocking send (MACH_SEND_TIMEOUT with MACH_MSG_TIMEOUT_NONE). This is a
+  // small and simple message and the destination port’s queue should be empty
+  // so a non-blocking send should be successful.
+  kr = mach_msg(&send_message.header,
+                MACH_SEND_MSG | MACH_SEND_TIMEOUT,
+                send_message.header.msgh_size,
+                0,
+                MACH_PORT_NULL,
+                MACH_MSG_TIMEOUT_NONE,
+                MACH_PORT_NULL);
+  ASSERT_EQ(MACH_MSG_SUCCESS, kr) << MachErrorMessage(kr, "mach_msg");
+
+  struct ReceiveMessage : public SendMessage {
+    mach_msg_trailer_t trailer;
+  };
+  ReceiveMessage receive_message;
+  memset(&receive_message, 0xa5, sizeof(receive_message));
+
+  kr = mach_msg(&receive_message.header,
+                MACH_RCV_MSG | MACH_RCV_TIMEOUT,
+                0,
+                sizeof(receive_message),
+                server_port,
+                MACH_MSG_TIMEOUT_NONE,
+                MACH_PORT_NULL);
+  ASSERT_EQ(MACH_MSG_SUCCESS, kr) << MachErrorMessage(kr, "mach_msg");
+
+  EXPECT_EQ(sizeof(send_message), receive_message.header.msgh_size);
+  EXPECT_EQ(0, memcmp(receive_message.data, kMessageData, message_data_size));
+
+  // Make sure that the bootstrap server’s mapping disappears if the service
+  // does.
+  server_port_owner.reset();
+  server_port = MACH_PORT_NULL;
+
+  // With the server port gone, the client port should have become a dead name.
+  mach_port_type_t client_port_type;
+  kr = mach_port_type(mach_task_self(), client_port, &client_port_type);
+  ASSERT_EQ(KERN_SUCCESS, kr) << MachErrorMessage(kr, "mach_port_type");
+  EXPECT_EQ(MACH_PORT_TYPE_DEAD_NAME, client_port_type);
+
+  client_port_owner.reset();
+  client_port = MACH_PORT_NULL;
+
+  kr = bootstrap_look_up(bootstrap_port, service_name.c_str(), &client_port);
+  ASSERT_EQ(BOOTSTRAP_UNKNOWN_SERVICE, kr)
+      << BootstrapErrorMessage(kr, "bootstrap_look_up");
+}
+
+}  // namespace