| // Copyright 2020 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/updater/activity.h" |
| |
| #include <string> |
| |
| #include "base/logging.h" |
| #include "base/strings/strcat.h" |
| #include "base/strings/string16.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "base/win/registry.h" |
| #include "base/win/windows_types.h" |
| #include "chrome/updater/win/constants.h" |
| |
| namespace updater { |
| namespace { |
| constexpr base::char16 kDidRun[] = L"dr"; |
| |
| base::string16 GetAppClientStateKey(const std::string& id) { |
| return base::ASCIIToUTF16(base::StrCat({CLIENT_STATE_KEY, id})); |
| } |
| |
| } // namespace |
| |
| bool GetActiveBit(const std::string& id, bool is_machine_) { |
| if (is_machine_) { |
| // TODO(crbug.com/1096654): Add support for the machine case. Machine |
| // installs must look for values under HKLM and under every HKU\<sid>. |
| return false; |
| } else { |
| // TODO(crbug/1159498): Standardize registry access. |
| base::win::RegKey key; |
| if (key.Open(HKEY_CURRENT_USER, GetAppClientStateKey(id).c_str(), |
| KEY_READ | KEY_WOW64_32KEY) == ERROR_SUCCESS) { |
| base::string16 value; |
| if (key.ReadValue(kDidRun, &value) == ERROR_SUCCESS && value == L"1") |
| return true; |
| } |
| return false; |
| } |
| } |
| |
| void ClearActiveBit(const std::string& id, bool is_machine_) { |
| if (is_machine_) { |
| // TODO(crbug.com/1096654): Add support for the machine case. Machine |
| // installs must clear values under HKLM and under every HKU\<sid>. |
| } else { |
| // TODO(crbug/1159498): Standardize registry access. |
| base::win::RegKey key; |
| if (key.Open(HKEY_CURRENT_USER, GetAppClientStateKey(id).c_str(), |
| KEY_WRITE | KEY_WOW64_32KEY) == ERROR_SUCCESS) { |
| const LONG result = key.WriteValue(kDidRun, L"0"); |
| if (result) { |
| VLOG(2) << "Failed to clear HKCU activity key for " << id << ": " |
| << result; |
| } |
| } else { |
| VLOG(2) << "Failed to open HKCU activity key with write for " << id; |
| } |
| } |
| } |
| |
| } // namespace updater |