Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(56)

Side by Side Diff: chrome/installer/util/delete_reg_key_work_item.cc

Issue 1976443005: Revert of Add best-effort/allow rollback flags on WorkItem. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@simple_list_tests
Patch Set: Created 4 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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/installer/util/delete_reg_key_work_item.h" 5 #include "chrome/installer/util/delete_reg_key_work_item.h"
6 6
7 #include <shlwapi.h> 7 #include <shlwapi.h>
8 8
9 #include "base/logging.h" 9 #include "base/logging.h"
10 #include "base/win/registry.h" 10 #include "base/win/registry.h"
(...skipping 11 matching lines...) Expand all
22 path_(path), 22 path_(path),
23 wow64_access_(wow64_access) { 23 wow64_access_(wow64_access) {
24 DCHECK(predefined_root); 24 DCHECK(predefined_root);
25 // It's a safe bet that we don't want to delete one of the root trees. 25 // It's a safe bet that we don't want to delete one of the root trees.
26 DCHECK(!path.empty()); 26 DCHECK(!path.empty());
27 DCHECK(wow64_access == 0 || 27 DCHECK(wow64_access == 0 ||
28 wow64_access == KEY_WOW64_32KEY || 28 wow64_access == KEY_WOW64_32KEY ||
29 wow64_access == KEY_WOW64_64KEY); 29 wow64_access == KEY_WOW64_64KEY);
30 } 30 }
31 31
32 bool DeleteRegKeyWorkItem::DoImpl() { 32 bool DeleteRegKeyWorkItem::Do() {
33 DCHECK(!backup_initialized_);
34
35 if (path_.empty()) 33 if (path_.empty())
36 return false; 34 return false;
37 35
38 // Only try to make a backup if rollback is enabled. 36 RegistryKeyBackup backup;
39 if (rollback_enabled()) { 37
40 if (!backup_.Initialize(predefined_root_, path_.c_str(), wow64_access_)) { 38 // Only try to make a backup if we're not configured to ignore failures.
39 if (!ignore_failure_) {
40 if (!backup.Initialize(predefined_root_, path_.c_str(), wow64_access_)) {
41 LOG(ERROR) << "Failed to backup destination for registry key copy."; 41 LOG(ERROR) << "Failed to backup destination for registry key copy.";
42 return false; 42 return false;
43 } 43 }
44 backup_initialized_ = true;
45 } 44 }
46 45
47 // Delete the key. 46 // Delete the key.
48 if (!InstallUtil::DeleteRegistryKey( 47 if (!InstallUtil::DeleteRegistryKey(
49 predefined_root_, path_.c_str(), wow64_access_)) { 48 predefined_root_, path_.c_str(), wow64_access_)) {
50 return false; 49 return ignore_failure_;
51 } 50 }
52 51
52 // We've succeeded, so remember any backup we may have made.
53 backup_.swap(backup);
54
53 return true; 55 return true;
54 } 56 }
55 57
56 void DeleteRegKeyWorkItem::RollbackImpl() { 58 void DeleteRegKeyWorkItem::Rollback() {
57 if (!backup_initialized_) 59 if (ignore_failure_)
58 return; 60 return;
59 61
60 // Delete anything in the key before restoring the backup in case someone else 62 // Delete anything in the key before restoring the backup in case someone else
61 // put new data in the key after Do(). 63 // put new data in the key after Do().
62 InstallUtil::DeleteRegistryKey(predefined_root_, 64 InstallUtil::DeleteRegistryKey(predefined_root_,
63 path_.c_str(), 65 path_.c_str(),
64 wow64_access_); 66 wow64_access_);
65 67
66 // Restore the old contents. The restoration takes on its default security 68 // Restore the old contents. The restoration takes on its default security
67 // attributes; any custom attributes are lost. 69 // attributes; any custom attributes are lost.
68 if (!backup_.WriteTo(predefined_root_, path_.c_str(), wow64_access_)) 70 if (!backup_.WriteTo(predefined_root_, path_.c_str(), wow64_access_))
69 LOG(ERROR) << "Failed to restore key in rollback."; 71 LOG(ERROR) << "Failed to restore key in rollback.";
70 } 72 }
OLDNEW
« no previous file with comments | « chrome/installer/util/delete_reg_key_work_item.h ('k') | chrome/installer/util/delete_reg_value_work_item.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698