[Chrome ELF] New NT registry API.

chrome_elf/breakpad.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // This module contains the necessary code to register the Breakpad exception
 // handler. This implementation is based on Chrome's crash reporting code.
 
 #include "chrome_elf/breakpad.h"
 
-#include <sddl.h>
-
 #include "base/macros.h"
 #include "breakpad/src/client/windows/handler/exception_handler.h"
 #include "chrome/common/chrome_version.h"
+#include "chrome_elf/chrome_elf_reg.h"
 #include "chrome_elf/chrome_elf_util.h"
 
 google_breakpad::ExceptionHandler* g_elf_breakpad = NULL;
 
 namespace {
 
 const wchar_t kBreakpadProductName[] = L"Chrome";
 const wchar_t kBreakpadVersionEntry[] = L"ver";
 const wchar_t kBreakpadProdEntry[] = L"prod";
 const wchar_t kBreakpadPlatformEntry[] = L"plat";
@@ skipping 32 matching lines @@
       kBreakpadProcessEntry, process.c_str());
   static google_breakpad::CustomInfoEntry channel_entry(
       kBreakpadChannelEntry, channel.c_str());
   static google_breakpad::CustomInfoEntry entries[] = {
       ver_entry, prod_entry, plat_entry, proc_entry, channel_entry};
   static google_breakpad::CustomClientInfo custom_info = {
       entries, arraysize(entries) };
   return &custom_info;
 }
 
-base::string16 GetUserSidString() {
-  // Get the current token.
-  HANDLE token = NULL;
-  base::string16 user_sid;
-  if (!::OpenProcessToken(::GetCurrentProcess(), TOKEN_QUERY, &token))
-    return user_sid;
-
-  DWORD size = sizeof(TOKEN_USER) + SECURITY_MAX_SID_SIZE;
-  BYTE user_bytes[sizeof(TOKEN_USER) + SECURITY_MAX_SID_SIZE] = {};
-  TOKEN_USER* user = reinterpret_cast<TOKEN_USER*>(user_bytes);
-
-  wchar_t* sid_string = NULL;
-  if (::GetTokenInformation(token, TokenUser, user, size, &size) &&
-      user->User.Sid &&
-      ::ConvertSidToStringSid(user->User.Sid, &sid_string)) {
-    user_sid = sid_string;
-    ::LocalFree(sid_string);
-  }
-
-  CloseHandle(token);
-  return user_sid;
-}
-
 bool IsHeadless() {
   DWORD ret = ::GetEnvironmentVariable(L"CHROME_HEADLESS", NULL, 0);
   if (ret != 0)
     return true;
 
   wchar_t* command_line = ::GetCommandLine();
 
   // Note: Since this is a pure substring search rather than a check for a
   // switch, there is a small chance that this code will match things that the
   // Chrome code (which executes a similar check) does not. However, as long as
@@ skipping 51 matching lines @@
   if (!use_policy && IsHeadless()) {
     pipe_name = kChromePipeName;
   } else if (use_policy ?
                  enabled_by_policy :
                  (is_official_chrome_build && AreUsageStatsEnabled(exe_path))) {
     // Build the pipe name. It can be one of:
     // 32-bit system: \\.\pipe\GoogleCrashServices\S-1-5-18
     // 32-bit user: \\.\pipe\GoogleCrashServices\<user SID>
     // 64-bit system: \\.\pipe\GoogleCrashServices\S-1-5-18-x64
     // 64-bit user: \\.\pipe\GoogleCrashServices\<user SID>-x64
-    base::string16 user_sid = IsSystemInstall(exe_path) ? kSystemPrincipalSid :
-                                                          GetUserSidString();
+    base::string16 user_sid = IsSystemInstall(exe_path)
+                                  ? kSystemPrincipalSid
+                                  : nt::GetCurrentUserSidString();
     if (user_sid.empty())
       return;
 
     pipe_name = kGoogleUpdatePipeName;
     pipe_name += user_sid;
 
 #if defined(_WIN64)
     pipe_name += L"-x64";
 #endif
   } else {
@@ skipping 10 matching lines @@
       google_breakpad::ExceptionHandler::HANDLER_ALL,
       dump_type,
       pipe_name.c_str(),
       GetCustomInfo());
 
   if (g_elf_breakpad->IsOutOfProcess()) {
     // Tells breakpad to handle breakpoint and single step exceptions.
     g_elf_breakpad->set_handle_debug_exceptions(true);
   }
 }