Chromium Code Reviews| 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/ready_mode/internal/registry_ready_mode_state.h" | 5 #include "chrome_frame/ready_mode/internal/registry_ready_mode_state.h" |
| 6 | 6 |
| 7 #include "base/time.h" | 7 #include "base/time.h" |
| 8 #include "base/task.h" | 8 #include "base/task.h" |
| 9 #include "base/win/registry.h" | 9 #include "base/win/registry.h" |
| 10 #include "chrome_frame/ready_mode/internal/installation_state.h" | 10 #include "chrome_frame/ready_mode/internal/installation_state.h" |
| (...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 65 return READY_MODE_ACTIVE; | 65 return READY_MODE_ACTIVE; |
| 66 | 66 |
| 67 return READY_MODE_TEMPORARILY_DECLINED; | 67 return READY_MODE_TEMPORARILY_DECLINED; |
| 68 } | 68 } |
| 69 | 69 |
| 70 bool RegistryReadyModeState::GetValue(int64* value, bool* exists) { | 70 bool RegistryReadyModeState::GetValue(int64* value, bool* exists) { |
| 71 *exists = false; | 71 *exists = false; |
| 72 *value = 0; | 72 *value = 0; |
| 73 | 73 |
| 74 base::win::RegKey config_key; | 74 base::win::RegKey config_key; |
| 75 if (!config_key.Open(HKEY_CURRENT_USER, key_name_.c_str(), KEY_QUERY_VALUE)) { | 75 LONG result = config_key.Open(HKEY_CURRENT_USER, key_name_.c_str(), |
| 76 DLOG(ERROR) << "Failed to open registry key " << key_name_; | 76 KEY_QUERY_VALUE); |
| 77 if (result != ERROR_SUCCESS) { | |
| 78 DLOG(ERROR) << "Failed to open registry key: " << key_name_ << | |
| 79 " Error: " << result; | |
| 77 return false; | 80 return false; |
| 78 } | 81 } |
| 79 | 82 |
| 80 if (!config_key.ValueExists(kReadyModeStateValue)) | 83 if (!config_key.ValueExists(kReadyModeStateValue)) |
| 81 return true; | 84 return true; |
| 82 | 85 |
| 83 int64 temp; | 86 result = config_key.ReadValueInt64(kReadyModeStateValue, value); |
| 84 DWORD value_size = sizeof(temp); | 87 if (result != ERROR_SUCCESS) { |
| 85 DWORD type = 0; | 88 DLOG(ERROR) << "Failed to open registry key: " << key_name_ << |
| 86 if (!config_key.ReadValue(kReadyModeStateValue, &temp, &value_size, &type)) { | 89 " Error: " << result; |
| 87 DLOG(ERROR) << "Failed to open registry key " << key_name_; | |
| 88 return false; | 90 return false; |
| 89 } | 91 } |
| 90 | 92 |
| 91 if (value_size != sizeof(temp) || type != REG_QWORD) { | |
| 92 DLOG(ERROR) << "Unexpected state found under registry key " << key_name_ | |
| 93 << " and value " << kReadyModeStateValue; | |
| 94 config_key.DeleteValue(kReadyModeStateValue); | |
| 95 return true; | |
| 96 } | |
| 97 | |
| 98 *value = temp; | |
| 99 *exists = true; | 93 *exists = true; |
| 100 return true; | 94 return true; |
| 101 } | 95 } |
| 102 | 96 |
| 103 bool RegistryReadyModeState::StoreValue(int64 value) { | 97 bool RegistryReadyModeState::StoreValue(const int64& value) { |
| 104 base::win::RegKey config_key; | 98 base::win::RegKey config_key; |
| 105 if (config_key.Open(HKEY_CURRENT_USER, key_name_.c_str(), KEY_SET_VALUE) && | 99 LONG result = config_key.Open(HKEY_CURRENT_USER, key_name_.c_str(), |
| 106 config_key.WriteValue(kReadyModeStateValue, &value, sizeof(value), | 100 KEY_SET_VALUE); |
| 107 REG_QWORD)) { | 101 if (result == ERROR_SUCCESS) { |
| 108 return true; | 102 result = config_key.WriteValue(kReadyModeStateValue, &value, sizeof(value), |
| 103 REG_QWORD); | |
| 109 } | 104 } |
| 110 | 105 |
| 111 DLOG(ERROR) << "Failed to open or write to registry key " << key_name_ | 106 if (result != ERROR_SUCCESS) { |
| 112 << " and value " << kReadyModeStateValue; | 107 DLOG(ERROR) << "Failed to open or write to registry key " << key_name_ |
| 108 << " and value " << kReadyModeStateValue << " Error: " | |
| 109 << result; | |
| 110 return false; | |
| 111 } | |
|
robertshield
2011/01/12 15:01:34
More nitty whining:
I found the original code easi
| |
| 113 | 112 |
| 114 return false; | 113 return true; |
| 115 } | 114 } |
| 116 | 115 |
| 117 void RegistryReadyModeState::TemporarilyDeclineChromeFrame() { | 116 void RegistryReadyModeState::TemporarilyDeclineChromeFrame() { |
| 118 int64 value = GetNow().ToInternalValue(); | 117 int64 value = GetNow().ToInternalValue(); |
| 119 | 118 |
| 120 if (StoreValue(value)) | 119 if (StoreValue(value)) |
| 121 observer_->OnStateChange(); | 120 observer_->OnStateChange(); |
| 122 } | 121 } |
| 123 | 122 |
| 124 void RegistryReadyModeState::PermanentlyDeclineChromeFrame() { | 123 void RegistryReadyModeState::PermanentlyDeclineChromeFrame() { |
| 125 bool success = false; | 124 bool success = false; |
| 126 | 125 |
| 127 // Either change, by itself, will deactivate Ready Mode, though we prefer to | 126 // Either change, by itself, will deactivate Ready Mode, though we prefer to |
| 128 // also unregister, in order to free up resources. | 127 // also unregister, in order to free up resources. |
| 129 | 128 |
| 130 if (StoreValue(0)) | 129 if (StoreValue(0)) |
| 131 success = true; | 130 success = true; |
| 132 | 131 |
| 133 if (installation_state_->UnregisterProduct()) | 132 if (installation_state_->UnregisterProduct()) |
| 134 success = true; | 133 success = true; |
| 135 | 134 |
| 136 if (success) | 135 if (success) |
| 137 observer_->OnStateChange(); | 136 observer_->OnStateChange(); |
| 138 } | 137 } |
| 139 | 138 |
| 140 void RegistryReadyModeState::AcceptChromeFrame() { | 139 void RegistryReadyModeState::AcceptChromeFrame() { |
| 141 if (installation_state_->InstallProduct()) | 140 if (installation_state_->InstallProduct()) |
| 142 observer_->OnStateChange(); | 141 observer_->OnStateChange(); |
| 143 } | 142 } |
| OLD | NEW |