| Index: chrome_elf/blacklist/test/blacklist_test_main_dll.cc
|
| diff --git a/chrome_elf/blacklist/test/blacklist_test_main_dll.cc b/chrome_elf/blacklist/test/blacklist_test_main_dll.cc
|
| index 3b2c4d7deac0cdb6d6f362c63cead8bb2632a355..5f919f7165dc07a815c5f2bde69775cab228519b 100644
|
| --- a/chrome_elf/blacklist/test/blacklist_test_main_dll.cc
|
| +++ b/chrome_elf/blacklist/test/blacklist_test_main_dll.cc
|
| @@ -14,22 +14,20 @@ namespace {
|
| void GetIpcOverrides() {
|
| DWORD buffer_size = ::GetEnvironmentVariableW(L"hkcu_override", nullptr, 0);
|
| if (buffer_size > 0) {
|
| - wchar_t* content = new wchar_t[buffer_size];
|
| + std::wstring content(buffer_size, L'\0');
|
| buffer_size =
|
| - ::GetEnvironmentVariableW(L"hkcu_override", content, buffer_size);
|
| + ::GetEnvironmentVariableW(L"hkcu_override", &content[0], buffer_size);
|
| if (buffer_size)
|
| - ::wcsncpy(nt::HKCU_override, content, nt::g_kRegMaxPathLen - 1);
|
| - delete[] content;
|
| + nt::SetTestingOverride(nt::HKCU, content);
|
| }
|
|
|
| buffer_size = ::GetEnvironmentVariableW(L"hklm_override", nullptr, 0);
|
| if (buffer_size > 0) {
|
| - wchar_t* content = new wchar_t[buffer_size];
|
| + std::wstring content(buffer_size, L'\0');
|
| buffer_size =
|
| - ::GetEnvironmentVariableW(L"hklm_override", content, buffer_size);
|
| + ::GetEnvironmentVariableW(L"hklm_override", &content[0], buffer_size);
|
| if (buffer_size)
|
| - ::wcsncpy(nt::HKLM_override, content, nt::g_kRegMaxPathLen - 1);
|
| - delete[] content;
|
| + nt::SetTestingOverride(nt::HKLM, content);
|
| }
|
|
|
| return;
|
|
|