| OLD | NEW | 
|    1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |    1 // Copyright (c) 2012 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_value_work_item.h" |    5 #include "chrome/installer/util/delete_reg_value_work_item.h" | 
|    6  |    6  | 
|    7 #include "base/logging.h" |    7 #include "base/logging.h" | 
|    8 #include "base/strings/string_util.h" |    8 #include "base/strings/string_util.h" | 
|    9 #include "base/win/registry.h" |    9 #include "base/win/registry.h" | 
|   10 #include "chrome/installer/util/logging_installer.h" |   10 #include "chrome/installer/util/logging_installer.h" | 
|   11  |   11  | 
|   12 using base::win::RegKey; |   12 using base::win::RegKey; | 
|   13  |   13  | 
|   14 DeleteRegValueWorkItem::DeleteRegValueWorkItem(HKEY predefined_root, |   14 DeleteRegValueWorkItem::DeleteRegValueWorkItem(HKEY predefined_root, | 
|   15                                                const std::wstring& key_path, |   15                                                const std::wstring& key_path, | 
 |   16                                                REGSAM wow64_access, | 
|   16                                                const std::wstring& value_name) |   17                                                const std::wstring& value_name) | 
|   17     : predefined_root_(predefined_root), |   18     : predefined_root_(predefined_root), | 
|   18       key_path_(key_path), |   19       key_path_(key_path), | 
|   19       value_name_(value_name), |   20       value_name_(value_name), | 
 |   21       wow64_access_(wow64_access), | 
|   20       previous_type_(0), |   22       previous_type_(0), | 
|   21       status_(DELETE_VALUE) { |   23       status_(DELETE_VALUE) { | 
 |   24   DCHECK(wow64_access == 0 || | 
 |   25          wow64_access == KEY_WOW64_32KEY || | 
 |   26          wow64_access == KEY_WOW64_64KEY); | 
|   22 } |   27 } | 
|   23  |   28  | 
|   24 DeleteRegValueWorkItem::~DeleteRegValueWorkItem() { |   29 DeleteRegValueWorkItem::~DeleteRegValueWorkItem() { | 
|   25 } |   30 } | 
|   26  |   31  | 
|   27 bool DeleteRegValueWorkItem::Do() { |   32 bool DeleteRegValueWorkItem::Do() { | 
|   28   if (status_ != DELETE_VALUE) { |   33   if (status_ != DELETE_VALUE) { | 
|   29     // we already did something. |   34     // we already did something. | 
|   30     LOG(ERROR) << "multiple calls to Do()"; |   35     LOG(ERROR) << "multiple calls to Do()"; | 
|   31     return false; |   36     return false; | 
|   32   } |   37   } | 
|   33  |   38  | 
|   34   status_ = VALUE_UNCHANGED; |   39   status_ = VALUE_UNCHANGED; | 
|   35  |   40  | 
|   36   RegKey key; |   41   RegKey key; | 
|   37   DWORD type = 0; |   42   DWORD type = 0; | 
|   38   DWORD size = 0; |   43   DWORD size = 0; | 
|   39   LONG result = key.Open(predefined_root_, key_path_.c_str(), |   44   LONG result = key.Open(predefined_root_, | 
|   40                          KEY_READ | KEY_WRITE); |   45                          key_path_.c_str(), | 
 |   46                          KEY_READ | KEY_WRITE | wow64_access_); | 
|   41   if (result == ERROR_SUCCESS) |   47   if (result == ERROR_SUCCESS) | 
|   42     result = key.ReadValue(value_name_.c_str(), NULL, &size, &type); |   48     result = key.ReadValue(value_name_.c_str(), NULL, &size, &type); | 
|   43  |   49  | 
|   44   if (result == ERROR_FILE_NOT_FOUND) { |   50   if (result == ERROR_FILE_NOT_FOUND) { | 
|   45     LOG(INFO) << "(delete value) Key: " << key_path_ << " or Value: " |   51     LOG(INFO) << "(delete value) Key: " << key_path_ << " or Value: " | 
|   46               << value_name_ << " does not exist."; |   52               << value_name_ << " does not exist."; | 
|   47     status_ = VALUE_NOT_FOUND; |   53     status_ = VALUE_NOT_FOUND; | 
|   48     return true; |   54     return true; | 
|   49   } |   55   } | 
|   50  |   56  | 
| (...skipping 25 matching lines...) Expand all  Loading... | 
|   76   if (status_ == DELETE_VALUE || status_ == VALUE_ROLLED_BACK) |   82   if (status_ == DELETE_VALUE || status_ == VALUE_ROLLED_BACK) | 
|   77     return; |   83     return; | 
|   78   if (status_ == VALUE_UNCHANGED || status_ == VALUE_NOT_FOUND) { |   84   if (status_ == VALUE_UNCHANGED || status_ == VALUE_NOT_FOUND) { | 
|   79     status_ = VALUE_ROLLED_BACK; |   85     status_ = VALUE_ROLLED_BACK; | 
|   80     VLOG(1) << "rollback: setting unchanged, nothing to do"; |   86     VLOG(1) << "rollback: setting unchanged, nothing to do"; | 
|   81     return; |   87     return; | 
|   82   } |   88   } | 
|   83  |   89  | 
|   84   // At this point only possible state is VALUE_DELETED. |   90   // At this point only possible state is VALUE_DELETED. | 
|   85   RegKey key; |   91   RegKey key; | 
|   86   LONG result = key.Open(predefined_root_, key_path_.c_str(), |   92   LONG result = key.Open(predefined_root_, | 
|   87                          KEY_READ | KEY_WRITE); |   93                          key_path_.c_str(), | 
 |   94                          KEY_READ | KEY_WRITE | wow64_access_); | 
|   88   if (result == ERROR_SUCCESS) { |   95   if (result == ERROR_SUCCESS) { | 
|   89     // try to restore the previous value |   96     // try to restore the previous value | 
|   90     DWORD previous_size = static_cast<DWORD>(previous_value_.size()); |   97     DWORD previous_size = static_cast<DWORD>(previous_value_.size()); | 
|   91     const char* previous_value = |   98     const char* previous_value = | 
|   92         previous_size ? &previous_value_[0] : NULL; |   99         previous_size ? &previous_value_[0] : NULL; | 
|   93     result = key.WriteValue(value_name_.c_str(), previous_value, |  100     result = key.WriteValue(value_name_.c_str(), previous_value, | 
|   94                             previous_size, previous_type_); |  101                             previous_size, previous_type_); | 
|   95     VLOG_IF(1, result != ERROR_SUCCESS) << "rollback: restoring " |  102     VLOG_IF(1, result != ERROR_SUCCESS) << "rollback: restoring " | 
|   96                                         << value_name_ << " error: " << result; |  103                                         << value_name_ << " error: " << result; | 
|   97   } else { |  104   } else { | 
|   98     VLOG(1) << "can not open " << key_path_ << " error: " << result; |  105     VLOG(1) << "can not open " << key_path_ << " error: " << result; | 
|   99   } |  106   } | 
|  100 } |  107 } | 
| OLD | NEW |