OLD | NEW |
(Empty) | |
| 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 |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/installer/mini_installer/regkey.h" |
| 6 |
| 7 #include "chrome/installer/mini_installer/mini_installer_constants.h" |
| 8 #include "chrome/installer/mini_installer/mini_string.h" |
| 9 |
| 10 namespace mini_installer { |
| 11 |
| 12 LONG RegKey::Open(HKEY key, const wchar_t* sub_key, REGSAM access) { |
| 13 Close(); |
| 14 return ::RegOpenKeyEx(key, sub_key, NULL, access, &key_); |
| 15 } |
| 16 |
| 17 LONG RegKey::ReadSZValue(const wchar_t* value_name, |
| 18 wchar_t* value, |
| 19 size_t value_size) const { |
| 20 DWORD type = 0; |
| 21 DWORD byte_length = static_cast<DWORD>(value_size * sizeof(wchar_t)); |
| 22 LONG result = ::RegQueryValueEx(key_, value_name, NULL, &type, |
| 23 reinterpret_cast<BYTE*>(value), |
| 24 &byte_length); |
| 25 if (result == ERROR_SUCCESS) { |
| 26 if (type != REG_SZ) { |
| 27 result = ERROR_NOT_SUPPORTED; |
| 28 } else if (byte_length == 0) { |
| 29 *value = L'\0'; |
| 30 } else if (value[byte_length/sizeof(wchar_t) - 1] != L'\0') { |
| 31 if ((byte_length / sizeof(wchar_t)) < value_size) |
| 32 value[byte_length / sizeof(wchar_t)] = L'\0'; |
| 33 else |
| 34 result = ERROR_MORE_DATA; |
| 35 } |
| 36 } |
| 37 return result; |
| 38 } |
| 39 |
| 40 LONG RegKey::ReadDWValue(const wchar_t* value_name, DWORD* value) const { |
| 41 DWORD type = 0; |
| 42 DWORD byte_length = sizeof(*value); |
| 43 LONG result = ::RegQueryValueEx(key_, value_name, NULL, &type, |
| 44 reinterpret_cast<BYTE*>(value), |
| 45 &byte_length); |
| 46 if (result == ERROR_SUCCESS) { |
| 47 if (type != REG_DWORD) { |
| 48 result = ERROR_NOT_SUPPORTED; |
| 49 } else if (byte_length != sizeof(*value)) { |
| 50 result = ERROR_NO_DATA; |
| 51 } |
| 52 } |
| 53 return result; |
| 54 } |
| 55 |
| 56 LONG RegKey::WriteSZValue(const wchar_t* value_name, const wchar_t* value) { |
| 57 return ::RegSetValueEx(key_, value_name, 0, REG_SZ, |
| 58 reinterpret_cast<const BYTE*>(value), |
| 59 (lstrlen(value) + 1) * sizeof(wchar_t)); |
| 60 } |
| 61 |
| 62 LONG RegKey::WriteDWValue(const wchar_t* value_name, DWORD value) { |
| 63 return ::RegSetValueEx(key_, value_name, 0, REG_DWORD, |
| 64 reinterpret_cast<const BYTE*>(&value), |
| 65 sizeof(value)); |
| 66 } |
| 67 |
| 68 void RegKey::Close() { |
| 69 if (key_ != NULL) { |
| 70 ::RegCloseKey(key_); |
| 71 key_ = NULL; |
| 72 } |
| 73 } |
| 74 |
| 75 |
| 76 // static |
| 77 bool RegKey::ReadSZValue(HKEY root_key, const wchar_t *sub_key, |
| 78 const wchar_t *value_name, wchar_t *value, |
| 79 size_t size) { |
| 80 RegKey key; |
| 81 return (key.Open(root_key, sub_key, KEY_QUERY_VALUE) == ERROR_SUCCESS && |
| 82 key.ReadSZValue(value_name, value, size) == ERROR_SUCCESS); |
| 83 } |
| 84 |
| 85 // Opens the Google Update ClientState key for a product. This finds only |
| 86 // registry entries for Chrome; it does not support the Chromium registry |
| 87 // layout. |
| 88 bool OpenClientStateKey(HKEY root_key, const wchar_t* app_guid, |
| 89 REGSAM access, RegKey* key) { |
| 90 StackString<MAX_PATH> client_state_key; |
| 91 return client_state_key.assign(kClientStateKeyBase) && |
| 92 client_state_key.append(app_guid) && |
| 93 (key->Open(root_key, |
| 94 client_state_key.get(), |
| 95 access | KEY_WOW64_32KEY) == ERROR_SUCCESS); |
| 96 } |
| 97 |
| 98 } // namespace mini_installer |
OLD | NEW |