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