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/nt_registry/nt_registry.h"

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;
}