mirror of
https://github.com/reactos/reactos.git
synced 2024-12-29 02:25:17 +00:00
[SERVICES] Use RegCopyTreeW from advapi32_vista
This commit is contained in:
Timo Kreuzer
parent
eef22cc036
commit
0db5d8a388
3 changed files with 4 additions and 342 deletions
|
|
@ -23,6 +23,6 @@ add_executable(services ${SOURCE} services.rc)
|
||||||
target_link_libraries(services ${PSEH_LIB})
|
target_link_libraries(services ${PSEH_LIB})
|
||||||
|
|
||||||
set_module_type(services win32gui UNICODE)
|
set_module_type(services win32gui UNICODE)
|
||||||
add_importlibs(services userenv user32 advapi32 rpcrt4 msvcrt kernel32 ntdll)
|
add_importlibs(services userenv user32 advapi32 advapi32_vista rpcrt4 msvcrt kernel32 ntdll)
|
||||||
add_pch(services services.h SOURCE)
|
add_pch(services services.h SOURCE)
|
||||||
add_cd_file(TARGET services DESTINATION reactos/system32 FOR all)
|
add_cd_file(TARGET services DESTINATION reactos/system32 FOR all)
|
||||||
|
|
|
||||||
|
|
@ -14,6 +14,8 @@
|
||||||
#define NDEBUG
|
#define NDEBUG
|
||||||
#include <debug.h>
|
#include <debug.h>
|
||||||
|
|
||||||
|
LSTATUS WINAPI RegCopyTreeW(_In_ HKEY, _In_opt_ LPCWSTR, _In_ HKEY);
|
||||||
|
LSTATUS WINAPI RegDeleteTreeW(_In_ HKEY, _In_opt_ LPCWSTR);
|
||||||
|
|
||||||
/* GLOBALS *******************************************************************/
|
/* GLOBALS *******************************************************************/
|
||||||
|
|
||||||
|
|
@ -22,337 +24,6 @@ static BOOL bBootAccepted = FALSE;
|
||||||
|
|
||||||
/* FUNCTIONS *****************************************************************/
|
/* FUNCTIONS *****************************************************************/
|
||||||
|
|
||||||
#if (_WIN32_WINNT < 0x0600)
|
|
||||||
static
|
|
||||||
DWORD
|
|
||||||
ScmCopyTree(
|
|
||||||
HKEY hSrcKey,
|
|
||||||
HKEY hDstKey)
|
|
||||||
{
|
|
||||||
DWORD dwSubKeys;
|
|
||||||
DWORD dwValues;
|
|
||||||
DWORD dwType;
|
|
||||||
DWORD dwMaxSubKeyNameLength;
|
|
||||||
DWORD dwSubKeyNameLength;
|
|
||||||
DWORD dwMaxValueNameLength;
|
|
||||||
DWORD dwValueNameLength;
|
|
||||||
DWORD dwMaxValueLength;
|
|
||||||
DWORD dwValueLength;
|
|
||||||
DWORD dwDisposition;
|
|
||||||
DWORD i;
|
|
||||||
LPWSTR lpNameBuffer;
|
|
||||||
LPBYTE lpDataBuffer;
|
|
||||||
HKEY hDstSubKey;
|
|
||||||
HKEY hSrcSubKey;
|
|
||||||
DWORD dwError;
|
|
||||||
|
|
||||||
DPRINT("ScmCopyTree()\n");
|
|
||||||
|
|
||||||
dwError = RegQueryInfoKey(hSrcKey,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
&dwSubKeys,
|
|
||||||
&dwMaxSubKeyNameLength,
|
|
||||||
NULL,
|
|
||||||
&dwValues,
|
|
||||||
&dwMaxValueNameLength,
|
|
||||||
&dwMaxValueLength,
|
|
||||||
NULL,
|
|
||||||
NULL);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
{
|
|
||||||
DPRINT1("RegQueryInfoKey() failed (Error %lu)\n", dwError);
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
|
|
||||||
dwMaxSubKeyNameLength++;
|
|
||||||
dwMaxValueNameLength++;
|
|
||||||
|
|
||||||
DPRINT("dwSubKeys %lu\n", dwSubKeys);
|
|
||||||
DPRINT("dwMaxSubKeyNameLength %lu\n", dwMaxSubKeyNameLength);
|
|
||||||
DPRINT("dwValues %lu\n", dwValues);
|
|
||||||
DPRINT("dwMaxValueNameLength %lu\n", dwMaxValueNameLength);
|
|
||||||
DPRINT("dwMaxValueLength %lu\n", dwMaxValueLength);
|
|
||||||
|
|
||||||
/* Copy subkeys */
|
|
||||||
if (dwSubKeys != 0)
|
|
||||||
{
|
|
||||||
lpNameBuffer = HeapAlloc(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
dwMaxSubKeyNameLength * sizeof(WCHAR));
|
|
||||||
if (lpNameBuffer == NULL)
|
|
||||||
{
|
|
||||||
DPRINT1("Buffer allocation failed\n");
|
|
||||||
return ERROR_NOT_ENOUGH_MEMORY;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (i = 0; i < dwSubKeys; i++)
|
|
||||||
{
|
|
||||||
dwSubKeyNameLength = dwMaxSubKeyNameLength;
|
|
||||||
dwError = RegEnumKeyExW(hSrcKey,
|
|
||||||
i,
|
|
||||||
lpNameBuffer,
|
|
||||||
&dwSubKeyNameLength,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
{
|
|
||||||
DPRINT1("Subkey enumeration failed (Error %lu)\n", dwError);
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
|
|
||||||
dwError = RegCreateKeyExW(hDstKey,
|
|
||||||
lpNameBuffer,
|
|
||||||
0,
|
|
||||||
NULL,
|
|
||||||
REG_OPTION_NON_VOLATILE,
|
|
||||||
KEY_WRITE,
|
|
||||||
NULL,
|
|
||||||
&hDstSubKey,
|
|
||||||
&dwDisposition);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
{
|
|
||||||
DPRINT1("Subkey creation failed (Error %lu)\n", dwError);
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
|
|
||||||
dwError = RegOpenKeyExW(hSrcKey,
|
|
||||||
lpNameBuffer,
|
|
||||||
0,
|
|
||||||
KEY_READ,
|
|
||||||
&hSrcSubKey);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
{
|
|
||||||
DPRINT1("Error: %lu\n", dwError);
|
|
||||||
RegCloseKey(hDstSubKey);
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
|
|
||||||
dwError = ScmCopyTree(hSrcSubKey,
|
|
||||||
hDstSubKey);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
{
|
|
||||||
DPRINT1("Error: %lu\n", dwError);
|
|
||||||
RegCloseKey (hSrcSubKey);
|
|
||||||
RegCloseKey (hDstSubKey);
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
|
|
||||||
RegCloseKey(hSrcSubKey);
|
|
||||||
RegCloseKey(hDstSubKey);
|
|
||||||
}
|
|
||||||
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Copy values */
|
|
||||||
if (dwValues != 0)
|
|
||||||
{
|
|
||||||
lpNameBuffer = HeapAlloc(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
dwMaxValueNameLength * sizeof(WCHAR));
|
|
||||||
if (lpNameBuffer == NULL)
|
|
||||||
{
|
|
||||||
DPRINT1("Buffer allocation failed\n");
|
|
||||||
return ERROR_NOT_ENOUGH_MEMORY;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* RegSetValueExW tries to read behind the maximum length, so give it space for that */
|
|
||||||
lpDataBuffer = HeapAlloc(GetProcessHeap(),
|
|
||||||
HEAP_ZERO_MEMORY,
|
|
||||||
dwMaxValueLength + sizeof(WCHAR));
|
|
||||||
if (lpDataBuffer == NULL)
|
|
||||||
{
|
|
||||||
DPRINT1("Buffer allocation failed\n");
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
return ERROR_NOT_ENOUGH_MEMORY;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (i = 0; i < dwValues; i++)
|
|
||||||
{
|
|
||||||
dwValueNameLength = dwMaxValueNameLength;
|
|
||||||
dwValueLength = dwMaxValueLength;
|
|
||||||
dwError = RegEnumValueW(hSrcKey,
|
|
||||||
i,
|
|
||||||
lpNameBuffer,
|
|
||||||
&dwValueNameLength,
|
|
||||||
NULL,
|
|
||||||
&dwType,
|
|
||||||
lpDataBuffer,
|
|
||||||
&dwValueLength);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
{
|
|
||||||
DPRINT1("Error: %lu\n", dwError);
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpDataBuffer);
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
|
|
||||||
dwError = RegSetValueExW(hDstKey,
|
|
||||||
lpNameBuffer,
|
|
||||||
0,
|
|
||||||
dwType,
|
|
||||||
lpDataBuffer,
|
|
||||||
dwValueLength);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
{
|
|
||||||
DPRINT1("Error: %lu\n", dwError);
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpDataBuffer);
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpDataBuffer);
|
|
||||||
|
|
||||||
HeapFree(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
lpNameBuffer);
|
|
||||||
}
|
|
||||||
|
|
||||||
DPRINT("ScmCopyTree() done\n");
|
|
||||||
|
|
||||||
return ERROR_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
DWORD
|
|
||||||
ScmDeleteTree(
|
|
||||||
HKEY hKey,
|
|
||||||
PCWSTR pszSubKey)
|
|
||||||
{
|
|
||||||
DWORD dwMaxSubkeyLength, dwMaxValueLength;
|
|
||||||
DWORD dwMaxLength, dwSize;
|
|
||||||
PWSTR pszName = NULL;
|
|
||||||
HKEY hSubKey = NULL;
|
|
||||||
DWORD dwError;
|
|
||||||
|
|
||||||
if (pszSubKey != NULL)
|
|
||||||
{
|
|
||||||
dwError = RegOpenKeyExW(hKey, pszSubKey, 0, KEY_READ, &hSubKey);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
hSubKey = hKey;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Get highest length for keys, values */
|
|
||||||
dwError = RegQueryInfoKeyW(hSubKey,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
&dwMaxSubkeyLength,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
&dwMaxValueLength,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
goto done;
|
|
||||||
|
|
||||||
dwMaxSubkeyLength++;
|
|
||||||
dwMaxValueLength++;
|
|
||||||
dwMaxLength = max(dwMaxSubkeyLength, dwMaxValueLength);
|
|
||||||
|
|
||||||
/* Allocate a buffer for key and value names */
|
|
||||||
pszName = HeapAlloc(GetProcessHeap(),
|
|
||||||
0,
|
|
||||||
dwMaxLength * sizeof(WCHAR));
|
|
||||||
if (pszName == NULL)
|
|
||||||
{
|
|
||||||
dwError = ERROR_NOT_ENOUGH_MEMORY;
|
|
||||||
goto done;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Recursively delete all the subkeys */
|
|
||||||
while (TRUE)
|
|
||||||
{
|
|
||||||
dwSize = dwMaxLength;
|
|
||||||
if (RegEnumKeyExW(hSubKey,
|
|
||||||
0,
|
|
||||||
pszName,
|
|
||||||
&dwSize,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL))
|
|
||||||
break;
|
|
||||||
|
|
||||||
dwError = ScmDeleteTree(hSubKey, pszName);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
goto done;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (pszSubKey != NULL)
|
|
||||||
{
|
|
||||||
dwError = RegDeleteKeyW(hKey, pszSubKey);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
while (TRUE)
|
|
||||||
{
|
|
||||||
dwSize = dwMaxLength;
|
|
||||||
if (RegEnumValueW(hKey,
|
|
||||||
0,
|
|
||||||
pszName,
|
|
||||||
&dwSize,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL,
|
|
||||||
NULL))
|
|
||||||
break;
|
|
||||||
|
|
||||||
dwError = RegDeleteValueW(hKey, pszName);
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
|
||||||
goto done;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
done:
|
|
||||||
if (pszName != NULL)
|
|
||||||
HeapFree(GetProcessHeap(), 0, pszName);
|
|
||||||
|
|
||||||
if (pszSubKey != NULL)
|
|
||||||
RegCloseKey(hSubKey);
|
|
||||||
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
static
|
static
|
||||||
DWORD
|
DWORD
|
||||||
ScmGetControlSetValues(
|
ScmGetControlSetValues(
|
||||||
|
|
@ -546,14 +217,9 @@ ScmCopyControlSet(
|
||||||
goto done;
|
goto done;
|
||||||
|
|
||||||
/* Copy the source control set to the destination control set */
|
/* Copy the source control set to the destination control set */
|
||||||
#if (_WIN32_WINNT >= 0x0600)
|
|
||||||
dwError = RegCopyTreeW(hSourceControlSetKey,
|
dwError = RegCopyTreeW(hSourceControlSetKey,
|
||||||
NULL,
|
NULL,
|
||||||
hDestinationControlSetKey);
|
hDestinationControlSetKey);
|
||||||
#else
|
|
||||||
dwError = ScmCopyTree(hSourceControlSetKey,
|
|
||||||
hDestinationControlSetKey);
|
|
||||||
#endif
|
|
||||||
if (dwError != ERROR_SUCCESS)
|
if (dwError != ERROR_SUCCESS)
|
||||||
goto done;
|
goto done;
|
||||||
|
|
||||||
|
|
@ -595,13 +261,8 @@ ScmDeleteControlSet(
|
||||||
return dwError;
|
return dwError;
|
||||||
|
|
||||||
/* Delete the control set */
|
/* Delete the control set */
|
||||||
#if (_WIN32_WINNT >= 0x0600)
|
|
||||||
dwError = RegDeleteTreeW(hControlSetKey,
|
dwError = RegDeleteTreeW(hControlSetKey,
|
||||||
NULL);
|
NULL);
|
||||||
#else
|
|
||||||
dwError = ScmDeleteTree(hControlSetKey,
|
|
||||||
NULL);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/* Open the system key */
|
/* Open the system key */
|
||||||
RegCloseKey(hControlSetKey);
|
RegCloseKey(hControlSetKey);
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,5 @@
|
||||||
|
|
||||||
|
@ stdcall RegCopyTreeW(ptr wstr ptr)
|
||||||
@ stdcall RegDeleteTreeA(long str)
|
@ stdcall RegDeleteTreeA(long str)
|
||||||
@ stdcall RegDeleteTreeW(long wstr)
|
@ stdcall RegDeleteTreeW(long wstr)
|
||||||
@ stdcall RegSetKeyValueW(long wstr wstr long ptr long)
|
@ stdcall RegSetKeyValueW(long wstr wstr long ptr long)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue