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

Unified Diff: chrome/installer/util/registry_entry.cc

Issue 623903002: Move shell_util's RegistryEntry class into a separate file. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@shell_util-generic-associations
Patch Set: Created 6 years, 2 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 side-by-side diff with in-line comments
Download patch
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;
+}

Powered by Google App Engine
This is Rietveld 408576698