[Chrome ELF] New NT registry API.

chrome_elf/blacklist/test/blacklist_test_main_dll.cc

 // Copyright 2013 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.
 
+#include "chrome_elf/blacklist/blacklist.h"
+
 #include <windows.h>
 
 #include "chrome/install_static/install_util.h"
-#include "chrome_elf/blacklist/blacklist.h"
+#include "chrome_elf/nt_registry/nt_registry.h"
 
-extern "C" void InitBlacklistTestDll() {
+namespace {
+
+void GetIpcOverrides() {
+  DWORD buffer_size = ::GetEnvironmentVariableW(L"hkcu_override", nullptr, 0);
+  if (buffer_size > 0) {
+    wchar_t* content = new wchar_t[buffer_size];
+    buffer_size =
+        ::GetEnvironmentVariableW(L"hkcu_override", content, buffer_size);
+    if (buffer_size)
+      ::wcsncpy(nt::HKCU_override, content, nt::g_kRegMaxPathLen - 1);
+    delete[] content;
+  }
+
+  buffer_size = ::GetEnvironmentVariableW(L"hklm_override", nullptr, 0);
+  if (buffer_size > 0) {
+    wchar_t* content = new wchar_t[buffer_size];
+    buffer_size =
+        ::GetEnvironmentVariableW(L"hklm_override", content, buffer_size);
+    if (buffer_size)
+      ::wcsncpy(nt::HKLM_override, content, nt::g_kRegMaxPathLen - 1);
+    delete[] content;
+  }
+
+  return;
+}
+
+}  // namespace
+
+extern "C" __declspec(dllexport) void InitTestDll() {
+  // Make sure we've got the latest registry overrides.
+  GetIpcOverrides();
 }
 
 BOOL APIENTRY DllMain(HMODULE module, DWORD reason, LPVOID reserved) {
   if (reason == DLL_PROCESS_ATTACH) {
+    GetIpcOverrides();
     install_static::InitializeProcessType();
     blacklist::Initialize(true);  // force always on, no beacon
   }
 
   return TRUE;
 }