Chromium Code Reviews| Index: chrome/installer/util/registry_entry.cc |
| diff --git a/chrome/installer/util/registry_entry.cc b/chrome/installer/util/registry_entry.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..4df95df04fa1a93b97d19a98f3e0165463338511 |
| --- /dev/null |
| +++ b/chrome/installer/util/registry_entry.cc |
| @@ -0,0 +1,97 @@ |
| +// Copyright 2014 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/installer/util/registry_entry.h" |
| + |
| +#include "base/strings/string_util.h" |
| +#include "base/win/registry.h" |
| +#include "chrome/installer/util/work_item.h" |
| +#include "chrome/installer/util/work_item_list.h" |
| + |
| +RegistryEntry::RegistryEntry(const base::string16& key_path, |
| + const base::string16& value) |
| + : key_path_(key_path), |
| + name_(), |
| + is_string_(true), |
|
grt (UTC plus 2)
2014/10/03 14:31:07
type_(REG_SZ),
value_(reinterpret_cast<const
Matt Giuca
2015/01/20 02:41:17
Acknowledged.
|
| + value_(value), |
| + int_value_(0) { |
| +} |
| + |
| +RegistryEntry::RegistryEntry(const base::string16& key_path, |
| + const base::string16& name, |
| + const base::string16& value) |
| + : key_path_(key_path), |
| + name_(name), |
| + is_string_(true), |
|
grt (UTC plus 2)
2014/10/03 14:31:07
as above
Matt Giuca
2015/01/20 02:41:17
Acknowledged.
|
| + value_(value), |
| + int_value_(0) { |
| +} |
| + |
| +RegistryEntry::RegistryEntry(const base::string16& key_path, |
| + const base::string16& name, |
| + DWORD value) |
| + : key_path_(key_path), |
| + name_(name), |
| + is_string_(false), |
|
grt (UTC plus 2)
2014/10/03 14:31:07
type_(REG_DWORD),
value_(reinterpret_cast<co
Matt Giuca
2015/01/20 02:41:17
Acknowledged.
|
| + value_(), |
| + int_value_(value) { |
| +} |
| + |
| +void RegistryEntry::AddToWorkItemList(HKEY root, WorkItemList* items) const { |
| + items->AddCreateRegKeyWorkItem(root, key_path_, WorkItem::kWow64Default); |
|
Will Harris
2014/10/03 05:13:45
Will this class ever be used to access 64-bit or 3
Matt Giuca
2014/10/03 10:09:39
No idea --- note that I'm just moving this code he
grt (UTC plus 2)
2015/01/20 20:05:31
This CL takes a private class and makes it public.
Matt Giuca
2015/01/20 23:16:35
OK, if you really feel strongly about this, then I
grt (UTC plus 2)
2015/01/20 23:30:55
I think it's a waste of both of our time to do it
|
| + if (is_string_) { |
|
grt (UTC plus 2)
2014/10/03 14:31:07
items->AddSetRegValueWorkItem(root_, key_path_, Wo
Matt Giuca
2015/01/20 02:41:17
Again, should not be changed here.
|
| + items->AddSetRegValueWorkItem( |
| + root, key_path_, WorkItem::kWow64Default, name_, value_, true); |
| + } else { |
| + items->AddSetRegValueWorkItem( |
| + root, key_path_, WorkItem::kWow64Default, name_, int_value_, true); |
| + } |
| +} |
| + |
| +bool RegistryEntry::ExistsInRegistry(uint32 look_for_in) const { |
| + DCHECK(look_for_in); |
| + |
| + RegistryStatus status = DOES_NOT_EXIST; |
| + if (look_for_in & LOOK_IN_HKCU) |
| + status = StatusInRegistryUnderRoot(HKEY_CURRENT_USER); |
| + if (status == DOES_NOT_EXIST && (look_for_in & LOOK_IN_HKLM)) |
| + status = StatusInRegistryUnderRoot(HKEY_LOCAL_MACHINE); |
| + return status == SAME_VALUE; |
| +} |
| + |
| +bool RegistryEntry::KeyExistsInRegistry(uint32 look_for_in) const { |
| + DCHECK(look_for_in); |
| + |
| + RegistryStatus status = DOES_NOT_EXIST; |
| + if (look_for_in & LOOK_IN_HKCU) |
| + status = StatusInRegistryUnderRoot(HKEY_CURRENT_USER); |
| + if (status == DOES_NOT_EXIST && (look_for_in & LOOK_IN_HKLM)) |
| + status = StatusInRegistryUnderRoot(HKEY_LOCAL_MACHINE); |
| + return status != DOES_NOT_EXIST; |
| +} |
| + |
| +RegistryEntry::RegistryStatus RegistryEntry::StatusInRegistryUnderRoot( |
| + HKEY root) const { |
| + base::win::RegKey key(root, key_path_.c_str(), KEY_QUERY_VALUE); |
| + bool found = false; |
| + bool correct_value = false; |
| + if (is_string_) { |
|
grt (UTC plus 2)
2014/10/03 14:31:07
since this has string-specific handling, you could
Matt Giuca
2015/01/20 02:41:17
Again, not here.
|
| + base::string16 read_value; |
| + found = key.ReadValue(name_.c_str(), &read_value) == ERROR_SUCCESS; |
| + if (found) { |
| + correct_value = read_value.size() == value_.size() && |
| + std::equal(value_.begin(), |
| + value_.end(), |
| + read_value.begin(), |
| + base::CaseInsensitiveCompare<wchar_t>()); |
| + } |
| + } else { |
| + DWORD read_value; |
| + found = key.ReadValueDW(name_.c_str(), &read_value) == ERROR_SUCCESS; |
| + if (found) |
| + correct_value = read_value == int_value_; |
| + } |
| + return found ? (correct_value ? SAME_VALUE : DIFFERENT_VALUE) |
| + : DOES_NOT_EXIST; |
| +} |