From 110656072c127425f7e5e1733e4b006f7cb8beb3 Mon Sep 17 00:00:00 2001 From: Nikolay Sivov Date: Fri, 24 May 2019 16:05:03 +0300 Subject: [PATCH] kernelbase: Add some registry helpers from shlwapi. Signed-off-by: Nikolay Sivov Signed-off-by: Alexandre Julliard --- dlls/kernelbase/Makefile.in | 1 + dlls/kernelbase/kernelbase.spec | 38 +-- dlls/kernelbase/registry.c | 477 ++++++++++++++++++++++++++++++++ 3 files changed, 497 insertions(+), 19 deletions(-) create mode 100644 dlls/kernelbase/registry.c diff --git a/dlls/kernelbase/Makefile.in b/dlls/kernelbase/Makefile.in index 373f79d2936..229b50bc517 100644 --- a/dlls/kernelbase/Makefile.in +++ b/dlls/kernelbase/Makefile.in @@ -4,4 +4,5 @@ IMPORTS = uuid advapi32 C_SRCS = \ main.c \ path.c \ + registry.c \ string.c diff --git a/dlls/kernelbase/kernelbase.spec b/dlls/kernelbase/kernelbase.spec index b721d8f8c6f..24d7b1772b0 100644 --- a/dlls/kernelbase/kernelbase.spec +++ b/dlls/kernelbase/kernelbase.spec @@ -1356,31 +1356,31 @@ @ stdcall SHExpandEnvironmentStringsW(wstr ptr long) kernel32.ExpandEnvironmentStringsW @ stdcall SHLoadIndirectString(wstr ptr long ptr) # @ stub SHLoadIndirectStringInternal -@ stdcall SHRegCloseUSKey(ptr) shlwapi.SHRegCloseUSKey -@ stdcall SHRegCreateUSKeyA(str long long ptr long) shlwapi.SHRegCreateUSKeyA -@ stdcall SHRegCreateUSKeyW(wstr long long ptr long) shlwapi.SHRegCreateUSKeyW -@ stdcall SHRegDeleteEmptyUSKeyA(long str long) shlwapi.SHRegDeleteEmptyUSKeyA -@ stdcall SHRegDeleteEmptyUSKeyW(long wstr long) shlwapi.SHRegDeleteEmptyUSKeyW -@ stdcall SHRegDeleteUSValueA(long str long) shlwapi.SHRegDeleteUSValueA -@ stdcall SHRegDeleteUSValueW(long wstr long) shlwapi.SHRegDeleteUSValueW -@ stdcall SHRegEnumUSKeyA(long long str ptr long) shlwapi.SHRegEnumUSKeyA -@ stdcall SHRegEnumUSKeyW(long long wstr ptr long) shlwapi.SHRegEnumUSKeyW -@ stdcall SHRegEnumUSValueA(long long ptr ptr ptr ptr ptr long) shlwapi.SHRegEnumUSValueA -@ stdcall SHRegEnumUSValueW(long long ptr ptr ptr ptr ptr long) shlwapi.SHRegEnumUSValueW +@ stdcall SHRegCloseUSKey(ptr) +@ stdcall SHRegCreateUSKeyA(str long long ptr long) +@ stdcall SHRegCreateUSKeyW(wstr long long ptr long) +@ stdcall SHRegDeleteEmptyUSKeyA(long str long) +@ stdcall SHRegDeleteEmptyUSKeyW(long wstr long) +@ stdcall SHRegDeleteUSValueA(long str long) +@ stdcall SHRegDeleteUSValueW(long wstr long) +@ stdcall SHRegEnumUSKeyA(long long str ptr long) +@ stdcall SHRegEnumUSKeyW(long long wstr ptr long) +@ stdcall SHRegEnumUSValueA(long long ptr ptr ptr ptr ptr long) +@ stdcall SHRegEnumUSValueW(long long ptr ptr ptr ptr ptr long) @ stdcall SHRegGetBoolUSValueA(str str long long) shlwapi.SHRegGetBoolUSValueA @ stdcall SHRegGetBoolUSValueW(wstr wstr long long) shlwapi.SHRegGetBoolUSValueW @ stdcall SHRegGetUSValueA( str str ptr ptr ptr long ptr long ) shlwapi.SHRegGetUSValueA @ stdcall SHRegGetUSValueW( wstr wstr ptr ptr ptr long ptr long ) shlwapi.SHRegGetUSValueW -@ stdcall SHRegOpenUSKeyA( str long long long long ) shlwapi.SHRegOpenUSKeyA -@ stdcall SHRegOpenUSKeyW( wstr long long long long ) shlwapi.SHRegOpenUSKeyW -@ stdcall SHRegQueryInfoUSKeyA( long ptr ptr ptr ptr long ) shlwapi.SHRegQueryInfoUSKeyA -@ stdcall SHRegQueryInfoUSKeyW( long ptr ptr ptr ptr long ) shlwapi.SHRegQueryInfoUSKeyW +@ stdcall SHRegOpenUSKeyA(str long long long long) +@ stdcall SHRegOpenUSKeyW(wstr long long long long) +@ stdcall SHRegQueryInfoUSKeyA(long ptr ptr ptr ptr long) +@ stdcall SHRegQueryInfoUSKeyW(long ptr ptr ptr ptr long) @ stdcall SHRegQueryUSValueA( long str ptr ptr ptr long ptr long ) shlwapi.SHRegQueryUSValueA @ stdcall SHRegQueryUSValueW( long wstr ptr ptr ptr long ptr long ) shlwapi.SHRegQueryUSValueW -@ stdcall SHRegSetUSValueA( str str long ptr long long) shlwapi.SHRegSetUSValueA -@ stdcall SHRegSetUSValueW( wstr wstr long ptr long long) shlwapi.SHRegSetUSValueW -@ stdcall SHRegWriteUSValueA(long str long ptr long long) shlwapi.SHRegWriteUSValueA -@ stdcall SHRegWriteUSValueW(long wstr long ptr long long) shlwapi.SHRegWriteUSValueW +@ stdcall SHRegSetUSValueA(str str long ptr long long) +@ stdcall SHRegSetUSValueW(wstr wstr long ptr long long) +@ stdcall SHRegWriteUSValueA(long str long ptr long long) +@ stdcall SHRegWriteUSValueW(long wstr long ptr long long) @ stdcall SHTruncateString(str long) # @ stub SaveAlternatePackageRootPath # @ stub SaveStateRootFolderPath diff --git a/dlls/kernelbase/registry.c b/dlls/kernelbase/registry.c new file mode 100644 index 00000000000..bfc7f109429 --- /dev/null +++ b/dlls/kernelbase/registry.c @@ -0,0 +1,477 @@ +/* + * Copyright 2019 Nikolay Sivov for CodeWeavers + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + +#include "shlwapi.h" + +#include "wine/debug.h" +#include "wine/heap.h" + +WINE_DEFAULT_DEBUG_CHANNEL(reg); + +struct USKEY +{ + HKEY HKCUstart; /* Start key in CU hive */ + HKEY HKCUkey; /* Opened key in CU hive */ + HKEY HKLMstart; /* Start key in LM hive */ + HKEY HKLMkey; /* Opened key in LM hive */ + WCHAR path[MAX_PATH]; +}; + +LONG WINAPI SHRegCreateUSKeyA(LPCSTR path, REGSAM samDesired, HUSKEY relative_key, PHUSKEY new_uskey, DWORD flags) +{ + WCHAR *pathW; + LONG ret; + + TRACE("%s, %#x, %p, %p, %#x\n", debugstr_a(path), samDesired, relative_key, new_uskey, flags); + + if (path) + { + INT len = MultiByteToWideChar(CP_ACP, 0, path, -1, NULL, 0); + pathW = heap_alloc(len * sizeof(WCHAR)); + if (!pathW) + return ERROR_NOT_ENOUGH_MEMORY; + MultiByteToWideChar(CP_ACP, 0, path, -1, pathW, len); + } + else + pathW = NULL; + + ret = SHRegCreateUSKeyW(pathW, samDesired, relative_key, new_uskey, flags); + HeapFree(GetProcessHeap(), 0, pathW); + return ret; +} + +static HKEY reg_duplicate_hkey(HKEY hKey) +{ + HKEY newKey = 0; + + RegOpenKeyExW(hKey, 0, 0, MAXIMUM_ALLOWED, &newKey); + return newKey; +} + +static HKEY reg_get_hkey_from_huskey(HUSKEY hUSKey, BOOL is_hkcu) +{ + struct USKEY *mihk = hUSKey; + HKEY test = hUSKey; + + if (test == HKEY_CLASSES_ROOT + || test == HKEY_CURRENT_CONFIG + || test == HKEY_CURRENT_USER + || test == HKEY_DYN_DATA + || test == HKEY_LOCAL_MACHINE + || test == HKEY_PERFORMANCE_DATA + || test == HKEY_USERS) +/* FIXME: need to define for Win2k, ME, XP + * (test == HKEY_PERFORMANCE_TEXT) || + * (test == HKEY_PERFORMANCE_NLSTEXT) || + */ + { + return test; + } + + return is_hkcu ? mihk->HKCUkey : mihk->HKLMkey; +} + +LONG WINAPI SHRegCreateUSKeyW(const WCHAR *path, REGSAM samDesired, HUSKEY relative_key, PHUSKEY new_uskey, DWORD flags) +{ + LONG ret = ERROR_CALL_NOT_IMPLEMENTED; + struct USKEY *ret_key; + + TRACE("%s, %#x, %p, %p, %#x\n", debugstr_w(path), samDesired, relative_key, new_uskey, flags); + + if (!new_uskey) + return ERROR_INVALID_PARAMETER; + + *new_uskey = NULL; + + if (flags & ~SHREGSET_FORCE_HKCU) + { + FIXME("unsupported flags 0x%08x\n", flags); + return ERROR_SUCCESS; + } + + ret_key = heap_alloc_zero(sizeof(*ret_key)); + lstrcpynW(ret_key->path, path, ARRAY_SIZE(ret_key->path)); + + if (relative_key) + { + ret_key->HKCUstart = reg_duplicate_hkey(reg_get_hkey_from_huskey(relative_key, TRUE)); + ret_key->HKLMstart = reg_duplicate_hkey(reg_get_hkey_from_huskey(relative_key, FALSE)); + } + else + { + ret_key->HKCUstart = HKEY_CURRENT_USER; + ret_key->HKLMstart = HKEY_LOCAL_MACHINE; + } + + if (flags & SHREGSET_FORCE_HKCU) + { + ret = RegCreateKeyExW(ret_key->HKCUstart, path, 0, NULL, 0, samDesired, NULL, &ret_key->HKCUkey, NULL); + if (ret == ERROR_SUCCESS) + *new_uskey = ret_key; + else + heap_free(ret_key); + } + + return ret; +} + +LONG WINAPI SHRegCloseUSKey(HUSKEY hUSKey) +{ + struct USKEY *key = hUSKey; + LONG ret = ERROR_SUCCESS; + + if (!key) + return ERROR_INVALID_PARAMETER; + + if (key->HKCUkey) + ret = RegCloseKey(key->HKCUkey); + if (key->HKCUstart && key->HKCUstart != HKEY_CURRENT_USER) + ret = RegCloseKey(key->HKCUstart); + if (key->HKLMkey) + ret = RegCloseKey(key->HKLMkey); + if (key->HKLMstart && key->HKLMstart != HKEY_LOCAL_MACHINE) + ret = RegCloseKey(key->HKLMstart); + + heap_free(key); + return ret; +} + +LONG WINAPI SHRegDeleteEmptyUSKeyA(HUSKEY hUSKey, const char *value, SHREGDEL_FLAGS flags) +{ + FIXME("%p, %s, %#x\n", hUSKey, debugstr_a(value), flags); + return ERROR_SUCCESS; +} + +LONG WINAPI SHRegDeleteEmptyUSKeyW(HUSKEY hUSKey, const WCHAR *value, SHREGDEL_FLAGS flags) +{ + FIXME("%p, %s, %#x\n", hUSKey, debugstr_w(value), flags); + return ERROR_SUCCESS; +} + +LONG WINAPI SHRegDeleteUSValueA(HUSKEY hUSKey, const char *value, SHREGDEL_FLAGS flags) +{ + FIXME("%p, %s, %#x\n", hUSKey, debugstr_a(value), flags); + return ERROR_SUCCESS; +} + +LONG WINAPI SHRegDeleteUSValueW(HUSKEY hUSKey, const WCHAR *value, SHREGDEL_FLAGS flags) +{ + FIXME("%p, %s, %#x\n", hUSKey, debugstr_w(value), flags); + return ERROR_SUCCESS; +} + +LONG WINAPI SHRegEnumUSValueA(HUSKEY hUSKey, DWORD index, char *value_name, DWORD *value_name_len, DWORD *type, + void *data, DWORD *data_len, SHREGENUM_FLAGS flags) +{ + HKEY dokey; + + TRACE("%p, %#x, %p, %p, %p, %p, %p, %#x\n", hUSKey, index, value_name, value_name_len, type, data, data_len, flags); + + if ((flags == SHREGENUM_HKCU || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, TRUE))) + return RegEnumValueA(dokey, index, value_name, value_name_len, NULL, type, data, data_len); + + if ((flags == SHREGENUM_HKLM || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, FALSE))) + return RegEnumValueA(dokey, index, value_name, value_name_len, NULL, type, data, data_len); + + FIXME("no support for SHREGENUM_BOTH\n"); + return ERROR_INVALID_FUNCTION; +} + +LONG WINAPI SHRegEnumUSValueW(HUSKEY hUSKey, DWORD index, WCHAR *value_name, DWORD *value_name_len, DWORD *type, + void *data, DWORD *data_len, SHREGENUM_FLAGS flags) +{ + HKEY dokey; + + TRACE("%p, %#x, %p, %p, %p, %p, %p, %#x\n", hUSKey, index, value_name, value_name_len, type, data, data_len, flags); + + if ((flags == SHREGENUM_HKCU || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, TRUE))) + return RegEnumValueW(dokey, index, value_name, value_name_len, NULL, type, data, data_len); + + if ((flags == SHREGENUM_HKLM || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, FALSE))) + return RegEnumValueW(dokey, index, value_name, value_name_len, NULL, type, data, data_len); + + FIXME("no support for SHREGENUM_BOTH\n"); + return ERROR_INVALID_FUNCTION; +} + +LONG WINAPI SHRegEnumUSKeyA(HUSKEY hUSKey, DWORD index, char *name, DWORD *name_len, SHREGENUM_FLAGS flags) +{ + HKEY dokey; + + TRACE("%p, %d, %p, %p(%d), %d\n", hUSKey, index, name, name_len, *name_len, flags); + + if ((flags == SHREGENUM_HKCU || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, TRUE))) + return RegEnumKeyExA(dokey, index, name, name_len, 0, 0, 0, 0); + + if ((flags == SHREGENUM_HKLM || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, FALSE))) + return RegEnumKeyExA(dokey, index, name, name_len, 0, 0, 0, 0); + + FIXME("no support for SHREGENUM_BOTH\n"); + return ERROR_INVALID_FUNCTION; +} + +LONG WINAPI SHRegEnumUSKeyW(HUSKEY hUSKey, DWORD index, WCHAR *name, DWORD *name_len, SHREGENUM_FLAGS flags) +{ + HKEY dokey; + + TRACE("%p, %d, %p, %p(%d), %d\n", hUSKey, index, name, name_len, *name_len, flags); + + if ((flags == SHREGENUM_HKCU || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, TRUE))) + return RegEnumKeyExW(dokey, index, name, name_len, 0, 0, 0, 0); + + if ((flags == SHREGENUM_HKLM || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, FALSE))) + return RegEnumKeyExW(dokey, index, name, name_len, 0, 0, 0, 0); + + FIXME("no support for SHREGENUM_BOTH\n"); + return ERROR_INVALID_FUNCTION; +} + +LONG WINAPI SHRegOpenUSKeyA(const char *path, REGSAM access_mask, HUSKEY relative_key, HUSKEY *uskey, BOOL ignore_hkcu) +{ + WCHAR pathW[MAX_PATH]; + + if (path) + MultiByteToWideChar(CP_ACP, 0, path, -1, pathW, ARRAY_SIZE(pathW)); + + return SHRegOpenUSKeyW(path ? pathW : NULL, access_mask, relative_key, uskey, ignore_hkcu); +} + +LONG WINAPI SHRegOpenUSKeyW(const WCHAR *path, REGSAM access_mask, HUSKEY relative_key, HUSKEY *uskey, BOOL ignore_hkcu) +{ + LONG ret2, ret1 = ~ERROR_SUCCESS; + struct USKEY *key; + + TRACE("%s, %#x, %p, %p, %d\n", debugstr_w(path), access_mask, relative_key, uskey, ignore_hkcu); + + if (uskey) + *uskey = NULL; + + /* Create internal HUSKEY */ + key = heap_alloc_zero(sizeof(*key)); + lstrcpynW(key->path, path, ARRAY_SIZE(key->path)); + + if (relative_key) + { + key->HKCUstart = reg_duplicate_hkey(reg_get_hkey_from_huskey(relative_key, TRUE)); + key->HKLMstart = reg_duplicate_hkey(reg_get_hkey_from_huskey(relative_key, FALSE)); + + /* FIXME: if either of these keys is NULL, create the start key from + * the relative keys start+path + */ + } + else + { + key->HKCUstart = HKEY_CURRENT_USER; + key->HKLMstart = HKEY_LOCAL_MACHINE; + } + + if (!ignore_hkcu) + { + ret1 = RegOpenKeyExW(key->HKCUstart, key->path, 0, access_mask, &key->HKCUkey); + if (ret1) + key->HKCUkey = 0; + } + + ret2 = RegOpenKeyExW(key->HKLMstart, key->path, 0, access_mask, &key->HKLMkey); + if (ret2) + key->HKLMkey = 0; + + if (ret1 || ret2) + TRACE("one or more opens failed: HKCU=%d HKLM=%d\n", ret1, ret2); + + if (ret1 && ret2) + { + /* Neither open succeeded: fail */ + SHRegCloseUSKey(key); + return ret2; + } + + TRACE("HUSKEY=%p\n", key); + if (uskey) + *uskey = key; + + return ERROR_SUCCESS; +} + +LONG WINAPI SHRegWriteUSValueA(HUSKEY hUSKey, const char *value, DWORD type, void *data, DWORD data_len, DWORD flags) +{ + WCHAR valueW[MAX_PATH]; + + if (value) + MultiByteToWideChar(CP_ACP, 0, value, -1, valueW, ARRAY_SIZE(valueW)); + + return SHRegWriteUSValueW(hUSKey, value ? valueW : NULL, type, data, data_len, flags); +} + +LONG WINAPI SHRegWriteUSValueW(HUSKEY hUSKey, const WCHAR *value, DWORD type, void *data, DWORD data_len, DWORD flags) +{ + struct USKEY *hKey = hUSKey; + LONG ret = ERROR_SUCCESS; + DWORD dummy; + + TRACE("%p, %s, %d, %p, %d, %#x\n", hUSKey, debugstr_w(value), type, data, data_len, flags); + + if (!hUSKey || IsBadWritePtr(hUSKey, sizeof(struct USKEY)) || !(flags & (SHREGSET_FORCE_HKCU|SHREGSET_FORCE_HKLM))) + return ERROR_INVALID_PARAMETER; + + if (flags & (SHREGSET_FORCE_HKCU | SHREGSET_HKCU)) + { + if (!hKey->HKCUkey) + { + /* Create the key */ + ret = RegCreateKeyW(hKey->HKCUstart, hKey->path, &hKey->HKCUkey); + TRACE("Creating HKCU key, ret = %d\n", ret); + if (ret && (flags & SHREGSET_FORCE_HKCU)) + { + hKey->HKCUkey = 0; + return ret; + } + } + + if (!ret) + { + if ((flags & SHREGSET_FORCE_HKCU) || RegQueryValueExW(hKey->HKCUkey, value, NULL, NULL, NULL, &dummy)) + { + /* Doesn't exist or we are forcing: Write value */ + ret = RegSetValueExW(hKey->HKCUkey, value, 0, type, data, data_len); + TRACE("Writing HKCU value, ret = %d\n", ret); + } + } + } + + if (flags & (SHREGSET_FORCE_HKLM | SHREGSET_HKLM)) + { + if (!hKey->HKLMkey) + { + /* Create the key */ + ret = RegCreateKeyW(hKey->HKLMstart, hKey->path, &hKey->HKLMkey); + TRACE("Creating HKLM key, ret = %d\n", ret); + if (ret && (flags & (SHREGSET_FORCE_HKLM))) + { + hKey->HKLMkey = 0; + return ret; + } + } + + if (!ret) + { + if ((flags & SHREGSET_FORCE_HKLM) || RegQueryValueExW(hKey->HKLMkey, value, NULL, NULL, NULL, &dummy)) + { + /* Doesn't exist or we are forcing: Write value */ + ret = RegSetValueExW(hKey->HKLMkey, value, 0, type, data, data_len); + TRACE("Writing HKLM value, ret = %d\n", ret); + } + } + } + + return ret; +} + +LONG WINAPI SHRegSetUSValueA(const char *subkey, const char *value, DWORD type, void *data, DWORD data_len, + DWORD flags) +{ + BOOL ignore_hkcu; + HUSKEY hkey; + LONG ret; + + TRACE("%s, %s, %d, %p, %d, %#x\n", debugstr_a(subkey), debugstr_a(value), type, data, data_len, flags); + + if (!data) + return ERROR_INVALID_FUNCTION; + + ignore_hkcu = !(flags & SHREGSET_HKCU || flags & SHREGSET_FORCE_HKCU); + + ret = SHRegOpenUSKeyA(subkey, KEY_ALL_ACCESS, 0, &hkey, ignore_hkcu); + if (ret == ERROR_SUCCESS) + { + ret = SHRegWriteUSValueA(hkey, value, type, data, data_len, flags); + SHRegCloseUSKey(hkey); + } + + return ret; +} + +LONG WINAPI SHRegSetUSValueW(const WCHAR *subkey, const WCHAR *value, DWORD type, void *data, DWORD data_len, + DWORD flags) +{ + BOOL ignore_hkcu; + HUSKEY hkey; + LONG ret; + + TRACE("%s, %s, %d, %p, %d, %#x\n", debugstr_w(subkey), debugstr_w(value), type, data, data_len, flags); + + if (!data) + return ERROR_INVALID_FUNCTION; + + ignore_hkcu = !(flags & SHREGSET_HKCU || flags & SHREGSET_FORCE_HKCU); + + ret = SHRegOpenUSKeyW(subkey, KEY_ALL_ACCESS, 0, &hkey, ignore_hkcu); + if (ret == ERROR_SUCCESS) + { + ret = SHRegWriteUSValueW(hkey, value, type, data, data_len, flags); + SHRegCloseUSKey(hkey); + } + + return ret; +} + +LONG WINAPI SHRegQueryInfoUSKeyA(HUSKEY hUSKey, DWORD *subkeys, DWORD *max_subkey_len, DWORD *values, DWORD *max_value_name_len, + SHREGENUM_FLAGS flags) +{ + HKEY dokey; + LONG ret; + + TRACE("%p, %p, %p, %p, %p, %#x\n", hUSKey, subkeys, max_subkey_len, values, max_value_name_len, flags); + + if ((flags == SHREGENUM_HKCU || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, TRUE))) + { + ret = RegQueryInfoKeyA(dokey, 0, 0, 0, subkeys, max_subkey_len, 0, values, max_value_name_len, 0, 0, 0); + if (ret == ERROR_SUCCESS || flags == SHREGENUM_HKCU) + return ret; + } + + if ((flags == SHREGENUM_HKLM || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, FALSE))) + { + return RegQueryInfoKeyA(dokey, 0, 0, 0, subkeys, max_subkey_len, 0, values, max_value_name_len, 0, 0, 0); + } + + return ERROR_INVALID_FUNCTION; +} + +LONG WINAPI SHRegQueryInfoUSKeyW(HUSKEY hUSKey, DWORD *subkeys, DWORD *max_subkey_len, DWORD *values, DWORD *max_value_name_len, + SHREGENUM_FLAGS flags) +{ + HKEY dokey; + LONG ret; + + TRACE("%p, %p, %p, %p, %p, %#x\n", hUSKey, subkeys, max_subkey_len, values, max_value_name_len, flags); + + if ((flags == SHREGENUM_HKCU || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, TRUE))) + { + ret = RegQueryInfoKeyW(dokey, 0, 0, 0, subkeys, max_subkey_len, 0, values, max_value_name_len, 0, 0, 0); + if (ret == ERROR_SUCCESS || flags == SHREGENUM_HKCU) + return ret; + } + + if ((flags == SHREGENUM_HKLM || flags == SHREGENUM_DEFAULT) && (dokey = reg_get_hkey_from_huskey(hUSKey, FALSE))) + { + return RegQueryInfoKeyW(dokey, 0, 0, 0, subkeys, max_subkey_len, 0, values, max_value_name_len, 0, 0, 0); + } + + return ERROR_INVALID_FUNCTION; +}