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

base/win/registry_unittest.cc

 // Copyright (c) 2011 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 "base/win/registry.h"
 
 #include <cstring>
 #include <vector>
 
 #include "base/compiler_specific.h"
@@ skipping 123 matching lines @@
                                       KEY_READ));
   ASSERT_EQ(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, foo_key.c_str(),
                                     KEY_READ | KEY_SET_VALUE));
   ASSERT_TRUE(key.Valid());
 
   // Create a test value that is larger than MAX_PATH.
   std::wstring data(MAX_PATH * 2, L'a');
 
   ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(data.c_str(), data.c_str()));
 
-  RegistryValueIterator iterator(HKEY_CURRENT_USER, foo_key.c_str());
+  RegistryValueIterator iterator(HKEY_CURRENT_USER, foo_key.c_str(), 0);
   ASSERT_TRUE(iterator.Valid());
   EXPECT_STREQ(data.c_str(), iterator.Name());
   EXPECT_STREQ(data.c_str(), iterator.Value());
   // ValueSize() is in bytes, including NUL.
   EXPECT_EQ((MAX_PATH * 2 + 1) * sizeof(wchar_t), iterator.ValueSize());
   ++iterator;
   EXPECT_FALSE(iterator.Valid());
 }
 
 TEST_F(RegistryTest, TruncatedCharTest) {
   RegKey key;
   std::wstring foo_key(kRootKey);
   foo_key += L"\\Foo";
   ASSERT_EQ(ERROR_SUCCESS, key.Create(HKEY_CURRENT_USER, foo_key.c_str(),
                                       KEY_READ));
   ASSERT_EQ(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, foo_key.c_str(),
                                     KEY_READ | KEY_SET_VALUE));
   ASSERT_TRUE(key.Valid());
 
   const wchar_t kName[] = L"name";
   // kData size is not a multiple of sizeof(wchar_t).
   const uint8 kData[] = { 1, 2, 3, 4, 5 };
   EXPECT_EQ(5, arraysize(kData));
   ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(kName, kData,
                                           arraysize(kData), REG_BINARY));
 
-  RegistryValueIterator iterator(HKEY_CURRENT_USER, foo_key.c_str());
+  RegistryValueIterator iterator(HKEY_CURRENT_USER, foo_key.c_str(), 0);
   ASSERT_TRUE(iterator.Valid());
   EXPECT_STREQ(kName, iterator.Name());
   // ValueSize() is in bytes.
   ASSERT_EQ(arraysize(kData), iterator.ValueSize());
   // Value() is NUL terminated.
   int end = (iterator.ValueSize() + sizeof(wchar_t) - 1) / sizeof(wchar_t);
   EXPECT_NE(L'\0', iterator.Value()[end-1]);
   EXPECT_EQ(L'\0', iterator.Value()[end]);
   EXPECT_EQ(0, std::memcmp(kData, iterator.Value(), arraysize(kData)));
   ++iterator;
@@ skipping 164 matching lines @@
             key.Open(HKEY_CURRENT_USER, foo_key.c_str(), KEY_READ));
 
   ASSERT_EQ(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, kRootKey, KEY_WRITE));
   ASSERT_EQ(ERROR_SUCCESS, key.DeleteKey(L"foo"));
 }
 
 }  // namespace
 
 }  // namespace win
 }  // namespace base