| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome_frame/test/chrome_frame_test_utils.h" | 5 #include "chrome_frame/test/chrome_frame_test_utils.h" |
| 6 | 6 |
| 7 #include <atlbase.h> | 7 #include <atlbase.h> |
| 8 #include <atlwin.h> | 8 #include <atlwin.h> |
| 9 #include <iepmapi.h> | 9 #include <iepmapi.h> |
| 10 #include <sddl.h> | 10 #include <sddl.h> |
| (...skipping 614 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 625 } | 625 } |
| 626 return NULL; | 626 return NULL; |
| 627 } | 627 } |
| 628 } | 628 } |
| 629 | 629 |
| 630 TempRegKeyOverride::TempRegKeyOverride(HKEY override, const wchar_t* temp_name) | 630 TempRegKeyOverride::TempRegKeyOverride(HKEY override, const wchar_t* temp_name) |
| 631 : override_(override), temp_name_(temp_name) { | 631 : override_(override), temp_name_(temp_name) { |
| 632 DCHECK(temp_name && lstrlenW(temp_name)); | 632 DCHECK(temp_name && lstrlenW(temp_name)); |
| 633 std::wstring key_path(kTempTestKeyPath); | 633 std::wstring key_path(kTempTestKeyPath); |
| 634 key_path += L"\\" + temp_name_; | 634 key_path += L"\\" + temp_name_; |
| 635 EXPECT_TRUE(temp_key_.Create(HKEY_CURRENT_USER, key_path.c_str(), | 635 EXPECT_EQ(ERROR_SUCCESS, temp_key_.Create(HKEY_CURRENT_USER, key_path.c_str(), |
| 636 KEY_ALL_ACCESS)); | 636 KEY_ALL_ACCESS)); |
| 637 EXPECT_EQ(ERROR_SUCCESS, | 637 EXPECT_EQ(ERROR_SUCCESS, |
| 638 ::RegOverridePredefKey(override_, temp_key_.Handle())); | 638 ::RegOverridePredefKey(override_, temp_key_.Handle())); |
| 639 } | 639 } |
| 640 | 640 |
| 641 TempRegKeyOverride::~TempRegKeyOverride() { | 641 TempRegKeyOverride::~TempRegKeyOverride() { |
| 642 ::RegOverridePredefKey(override_, NULL); | 642 ::RegOverridePredefKey(override_, NULL); |
| 643 // The temp key will be deleted via a call to DeleteAllTempKeys(). | 643 // The temp key will be deleted via a call to DeleteAllTempKeys(). |
| 644 } | 644 } |
| 645 | 645 |
| 646 // static | 646 // static |
| 647 void TempRegKeyOverride::DeleteAllTempKeys() { | 647 void TempRegKeyOverride::DeleteAllTempKeys() { |
| 648 base::win::RegKey key; | 648 base::win::RegKey key; |
| 649 if (key.Open(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS)) { | 649 if (key.Open(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS) == ERROR_SUCCESS) { |
| 650 key.DeleteKey(kTempTestKeyPath); | 650 key.DeleteKey(kTempTestKeyPath); |
| 651 } | 651 } |
| 652 } | 652 } |
| 653 | 653 |
| 654 ScopedVirtualizeHklmAndHkcu::ScopedVirtualizeHklmAndHkcu() { | 654 ScopedVirtualizeHklmAndHkcu::ScopedVirtualizeHklmAndHkcu() { |
| 655 TempRegKeyOverride::DeleteAllTempKeys(); | 655 TempRegKeyOverride::DeleteAllTempKeys(); |
| 656 hklm_.reset(new TempRegKeyOverride(HKEY_LOCAL_MACHINE, L"hklm_fake")); | 656 hklm_.reset(new TempRegKeyOverride(HKEY_LOCAL_MACHINE, L"hklm_fake")); |
| 657 hkcu_.reset(new TempRegKeyOverride(HKEY_CURRENT_USER, L"hkcu_fake")); | 657 hkcu_.reset(new TempRegKeyOverride(HKEY_CURRENT_USER, L"hkcu_fake")); |
| 658 } | 658 } |
| 659 | 659 |
| 660 ScopedVirtualizeHklmAndHkcu::~ScopedVirtualizeHklmAndHkcu() { | 660 ScopedVirtualizeHklmAndHkcu::~ScopedVirtualizeHklmAndHkcu() { |
| 661 hkcu_.reset(NULL); | 661 hkcu_.reset(NULL); |
| 662 hklm_.reset(NULL); | 662 hklm_.reset(NULL); |
| 663 TempRegKeyOverride::DeleteAllTempKeys(); | 663 TempRegKeyOverride::DeleteAllTempKeys(); |
| 664 } | 664 } |
| 665 | 665 |
| 666 bool KillProcesses(const std::wstring& executable_name, int exit_code, | 666 bool KillProcesses(const std::wstring& executable_name, int exit_code, |
| 667 bool wait) { | 667 bool wait) { |
| 668 bool result = true; | 668 bool result = true; |
| 669 base::NamedProcessIterator iter(executable_name, NULL); | 669 base::NamedProcessIterator iter(executable_name, NULL); |
| 670 while (const base::ProcessEntry* entry = iter.NextProcessEntry()) { | 670 while (const base::ProcessEntry* entry = iter.NextProcessEntry()) { |
| 671 result &= base::KillProcessById(entry->pid(), exit_code, wait); | 671 result &= base::KillProcessById(entry->pid(), exit_code, wait); |
| 672 } | 672 } |
| 673 return result; | 673 return result; |
| 674 } | 674 } |
| 675 | 675 |
| 676 } // namespace chrome_frame_test | 676 } // namespace chrome_frame_test |
| OLD | NEW |