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