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

Side by Side Diff: rlz/win/lib/rlz_lib_win.cc

Issue 63153009: Use RegistryOverrideManager properly in RLZ tests. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: comment fix Created 7 years, 1 month 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 unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « rlz/win/lib/rlz_lib.h ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 // 4 //
5 // A library to manage RLZ information for access-points shared 5 // A library to manage RLZ information for access-points shared
6 // across different client applications. 6 // across different client applications.
7 7
8 #include "rlz/win/lib/rlz_lib.h" 8 #include "rlz/lib/rlz_lib.h"
9 9
10 #include <windows.h> 10 #include <windows.h>
11 #include <aclapi.h> 11 #include <aclapi.h>
12 #include <winerror.h> 12 #include <winerror.h>
13 13
14 #include "base/basictypes.h" 14 #include "base/basictypes.h"
15 #include "base/win/registry.h" 15 #include "base/win/registry.h"
16 #include "base/win/windows_version.h"
17 #include "rlz/lib/assert.h" 16 #include "rlz/lib/assert.h"
18 #include "rlz/lib/rlz_value_store.h" 17 #include "rlz/lib/rlz_value_store.h"
19 #include "rlz/win/lib/machine_deal.h" 18 #include "rlz/win/lib/machine_deal.h"
20 #include "rlz/win/lib/rlz_value_store_registry.h" 19 #include "rlz/win/lib/rlz_value_store_registry.h"
21 20
22 namespace {
23
24 // Path to recursively copy into the replacemment hives. These are needed
25 // to make sure certain win32 APIs continue to run correctly once the real
26 // hives are replaced.
27 const wchar_t* kHKLMAccessProviders =
28 L"System\\CurrentControlSet\\Control\\Lsa\\AccessProviders";
29
30 // Helper functions
31
32 void CopyRegistryTree(const base::win::RegKey& src, base::win::RegKey* dest) {
33 // First copy values.
34 for (base::win::RegistryValueIterator i(src.Handle(), L"");
35 i.Valid(); ++i) {
36 dest->WriteValue(i.Name(), reinterpret_cast<const void*>(i.Value()),
37 i.ValueSize(), i.Type());
38 }
39
40 // Next copy subkeys recursively.
41 for (base::win::RegistryKeyIterator i(src.Handle(), L"");
42 i.Valid(); ++i) {
43 base::win::RegKey subkey(dest->Handle(), i.Name(), KEY_ALL_ACCESS);
44 CopyRegistryTree(base::win::RegKey(src.Handle(), i.Name(), KEY_READ),
45 &subkey);
46 }
47 }
48
49 } // namespace anonymous
50
51
52 namespace rlz_lib { 21 namespace rlz_lib {
53 22
54 // OEM Deal confirmation storage functions. 23 // OEM Deal confirmation storage functions.
55 24
56 template<class T> 25 template<class T>
57 class typed_buffer_ptr { 26 class typed_buffer_ptr {
58 scoped_ptr<char[]> buffer_; 27 scoped_ptr<char[]> buffer_;
59 28
60 public: 29 public:
61 typed_buffer_ptr() { 30 typed_buffer_ptr() {
(...skipping 168 matching lines...) Expand 10 before | Expand all | Expand 10 after
230 bool GetMachineDealCode(char* dcc, size_t dcc_size) { 199 bool GetMachineDealCode(char* dcc, size_t dcc_size) {
231 return MachineDealCode::Get(dcc, dcc_size); 200 return MachineDealCode::Get(dcc, dcc_size);
232 } 201 }
233 202
234 // Combined functions. 203 // Combined functions.
235 204
236 bool SetMachineDealCodeFromPingResponse(const char* response) { 205 bool SetMachineDealCodeFromPingResponse(const char* response) {
237 return MachineDealCode::SetFromPingResponse(response); 206 return MachineDealCode::SetFromPingResponse(response);
238 } 207 }
239 208
240 void InitializeTempHivesForTesting(const base::win::RegKey& temp_hklm_key,
241 const base::win::RegKey& temp_hkcu_key) {
242 // For the moment, the HKCU hive requires no initialization.
243
244 if (base::win::GetVersion() >= base::win::VERSION_WIN7) {
245 // Copy the following HKLM subtrees to the temporary location so that the
246 // win32 APIs used by the tests continue to work:
247 //
248 // HKLM\System\CurrentControlSet\Control\Lsa\AccessProviders
249 //
250 // This seems to be required since Win7.
251 base::win::RegKey dest(temp_hklm_key.Handle(), kHKLMAccessProviders,
252 KEY_ALL_ACCESS);
253 CopyRegistryTree(base::win::RegKey(HKEY_LOCAL_MACHINE,
254 kHKLMAccessProviders,
255 KEY_READ),
256 &dest);
257 }
258 }
259
260 } // namespace rlz_lib 209 } // namespace rlz_lib
OLDNEW
« no previous file with comments | « rlz/win/lib/rlz_lib.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698