Allow Registry Iterator functions to use a specified WOW64 mode when iterating.

base/win/registry.h

 // Copyright (c) 2012 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.
 
 #ifndef BASE_WIN_REGISTRY_H_
 #define BASE_WIN_REGISTRY_H_
 
 #include <windows.h>
 #include <string>
 #include <vector>
@@ skipping 139 matching lines @@
   HKEY key_;  // The registry key being iterated.
   HANDLE watch_event_;
   REGSAM wow64access_;
 
   DISALLOW_COPY_AND_ASSIGN(RegKey);
 };
 
 // Iterates the entries found in a particular folder on the registry.
 class BASE_EXPORT RegistryValueIterator {
  public:
+  // Construct a Registry Value Iterator with default WOW64 access.
   RegistryValueIterator(HKEY root_key, const wchar_t* folder_key);
 
+  // Construct a Registry Key Iterator with specific WOW64 access, one of
+  // KEY_WOW64_32KEY or KEY_WOW64_64KEY, or 0.
+  // Note: |wow64access| should be the same access used to open |root_key|
+  // previously, or a predefined key (e.g. HKEY_LOCAL_MACHINE).
+  // See http://msdn.microsoft.com/en-us/library/windows/desktop/aa384129.aspx.
+  RegistryValueIterator(HKEY root_key,
+                        const wchar_t* folder_key,
+                        REGSAM wow64access);
+
   ~RegistryValueIterator();
 
   DWORD ValueCount() const;
 
   // True while the iterator is valid.
   bool Valid() const;
 
   // Advances to the next registry entry.
   void operator++();
 
   const wchar_t* Name() const { return name_.c_str(); }
   const wchar_t* Value() const { return vector_as_array(&value_); }
   // ValueSize() is in bytes.
   DWORD ValueSize() const { return value_size_; }
   DWORD Type() const { return type_; }
 
   int Index() const { return index_; }
 
  private:
   // Read in the current values.
   bool Read();
 
+  void Initialize(HKEY root_key, const wchar_t* folder_key, REGSAM wow64access);
+
   // The registry key being iterated.
   HKEY key_;
 
   // Current index of the iteration.
   int index_;
 
   // Current values.
   std::wstring name_;
   std::vector<wchar_t> value_;
   DWORD value_size_;
   DWORD type_;
 
   DISALLOW_COPY_AND_ASSIGN(RegistryValueIterator);
 };
 
 class BASE_EXPORT RegistryKeyIterator {
  public:
+  // Construct a Registry Key Iterator with default WOW64 access.
   RegistryKeyIterator(HKEY root_key, const wchar_t* folder_key);
 
+  // Construct a Registry Value Iterator with specific WOW64 access, one of
+  // KEY_WOW64_32KEY or KEY_WOW64_64KEY, or 0.
+  // Note: |wow64access| should be the same access used to open |root_key|
+  // previously, or a predefined key (e.g. HKEY_LOCAL_MACHINE).
+  // See http://msdn.microsoft.com/en-us/library/windows/desktop/aa384129.aspx.
+  RegistryKeyIterator(HKEY root_key,
+                      const wchar_t* folder_key,
+                      REGSAM wow64access);
+
   ~RegistryKeyIterator();
 
   DWORD SubkeyCount() const;
 
   // True while the iterator is valid.
   bool Valid() const;
 
   // Advances to the next entry in the folder.
   void operator++();
 
   const wchar_t* Name() const { return name_; }
 
   int Index() const { return index_; }
 
  private:
   // Read in the current values.
   bool Read();
 
+  void Initialize(HKEY root_key, const wchar_t* folder_key, REGSAM wow64access);
+
   // The registry key being iterated.
   HKEY key_;
 
   // Current index of the iteration.
   int index_;
 
   wchar_t name_[MAX_PATH];
 
   DISALLOW_COPY_AND_ASSIGN(RegistryKeyIterator);
 };
 
 }  // namespace win
 }  // namespace base
 
 #endif  // BASE_WIN_REGISTRY_H_