Implement more of CrashpadClient on Windows

client/crashpad_client_win.cc

Index: client/crashpad_client_win.cc
diff --git a/client/crashpad_client_win.cc b/client/crashpad_client_win.cc
index 734bedfd84301af5cf81162fe1723d21ba43240e..3e10900530cc0bb8269d52ad5bd4d02e85b9191e 100644
--- a/client/crashpad_client_win.cc
+++ b/client/crashpad_client_win.cc
@@ -16,14 +16,21 @@
 
 #include <windows.h>
 
+#include "base/atomicops.h"
 #include "base/logging.h"
+#include "base/strings/string16.h"
+#include "base/strings/utf_string_conversions.h"
+#include "client/crashpad_info.h"
+#include "client/registration_protocol_win.h"
+#include "util/file/file_io.h"
+#include "util/win/scoped_handle.h"
 
 namespace {
 // Time to wait for the handler to create a dump. This is tricky to figure out.
 const DWORD kMillisecondsUntilTerminate = 5000;
 
 // This is the exit code that the process will return to the system once the
-// crash has been handled by Crashpad. We don't want to clash with the 
+// crash has been handled by Crashpad. We don't want to clash with the
 // application-defined exit codes but we don't know them so we use one that is
 // unlikely to be used.
 const UINT kCrashExitCode = 0xffff7001;
@@ -32,8 +39,30 @@ const UINT kCrashExitCode = 0xffff7001;
 HANDLE g_signal_exception = nullptr;
 HANDLE g_wait_termination = nullptr;
 
+// Tracks whether a thread has already entered UnhandledExceptionHandler.
+base::subtle::AtomicWord g_have_crashed;
+
 LONG WINAPI UnhandledExceptionHandler(EXCEPTION_POINTERS* exception_pointers) {
-  // TODO (cpu): Here write |exception_pointers| to g_crashpad_info.
+  // This is a per-process handler. While this handler is being invoked, other
+  // threads are still executing as usual, so multiple threads could enter at
+  // the same time. Because we're in a crashing state, we shouldn't be doing
+  // anything that might cause allocations, call into kernel mode, etc. So, we
+  // don't want to take a critical section here to avoid simultaneous access to
+  // the global exception pointers in CrashpadInfo. Because the crash handler
+  // will record all threads, it's fine to simply have the second and subsequent
+  // entrants block here. They will soon be suspended by the crash handler, and
+  // then the entire process will be terminated below. This means that we won't
+  // save the exception pointers from the second and further crashes, but
+  // contention here is very unlikely, and we'll still have a stack that's
+  // blocked at this location.
+  if (base::subtle::Barrier_AtomicIncrement(&g_have_crashed, 1) > 1) {
+    SleepEx(false, INFINITE);
+  }
+
+  // Otherwise, we're the first thread, so record the exception pointer and
+  // signal the crash handler.
+  crashpad::CrashpadInfo::GetCrashpadInfo()->set_exception_pointers(
+      exception_pointers);
   DWORD rv = SignalObjectAndWait(g_signal_exception,
                                  g_wait_termination,
                                  kMillisecondsUntilTerminate,
@@ -51,6 +80,45 @@ LONG WINAPI UnhandledExceptionHandler(EXCEPTION_POINTERS* exception_pointers) {
   return 0L;
 }
 
+// Returns a pipe handle connected to the RegistrationServer.
+crashpad::ScopedFileHANDLE Connect(const base::string16& pipe_name) {
+  crashpad::ScopedFileHANDLE pipe;
+  const int kMaxTries = 5;
+  for (int tries = 0; tries < kMaxTries; ++tries) {
+    pipe.reset(CreateFile(pipe_name.c_str(),
+                          GENERIC_READ | GENERIC_WRITE,
+                          0,
+                          nullptr,
+                          OPEN_EXISTING,
+                          SECURITY_SQOS_PRESENT | SECURITY_ANONYMOUS,
+                          nullptr));
+    if (pipe.is_valid())
+      break;
+
+    // If busy, wait 60s before retrying.
+    if (GetLastError() != ERROR_PIPE_BUSY) {
+      PLOG(ERROR) << "CreateFile pipe connection";
+      return crashpad::ScopedFileHANDLE();
+    } else if (!WaitNamedPipe(pipe_name.c_str(), 60000)) {
+      PLOG(ERROR) << "WaitNamedPipe";
+    }
+  }
+
+  if (!pipe.is_valid())
+    return crashpad::ScopedFileHANDLE();
+
+  DWORD mode = PIPE_READMODE_MESSAGE;
+  if (!SetNamedPipeHandleState(pipe.get(),
+                               &mode,
+                               nullptr,  // Don't set maximum bytes.
+                               nullptr)) {  // Don't set maximum time.
+    PLOG(ERROR) << "SetNamedPipeHandleState";
+    return crashpad::ScopedFileHANDLE();
+  }
+
+  return pipe.Pass();
+}
+
 }  // namespace
 
 namespace crashpad {
@@ -67,14 +135,30 @@ bool CrashpadClient::StartHandler(
     const std::string& url,
     const std::map<std::string, std::string>& annotations,
     const std::vector<std::string>& arguments) {
-  // TODO(cpu): Provide a reference implementation.
   return false;
 }
 
 bool CrashpadClient::SetHandler(const std::string& ipc_port) {
-  // TODO (cpu): Contact the handler and obtain g_signal_exception and
-  // g_wait_termination.
-  return false;
+  RegistrationRequest request = {0};
+  request.client_process_id = GetCurrentProcessId();
+  request.crashpad_info_address =
+      reinterpret_cast<WinVMAddress>(CrashpadInfo::GetCrashpadInfo());
+
+  RegistrationResponse response = {0};
+
+  ScopedFileHANDLE pipe = Connect(base::UTF8ToUTF16(ipc_port));
+  if (!pipe.is_valid())
+    return false;
+  bool result = LoggingWriteFile(pipe.get(), &request, sizeof(request)) &&
+                LoggingReadFile(pipe.get(), &response, sizeof(response));
+  if (!result)
+    return result;
+
+  // The server returns these already duplicated to be valid in this process.
+  g_signal_exception = reinterpret_cast<HANDLE>(response.request_report_event);
+  g_wait_termination =
+      reinterpret_cast<HANDLE>(response.report_complete_event);
+  return true;
 }
 
 bool CrashpadClient::UseHandler() {