ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

reg/tests: Modify delete_tree() to support the use of registry views. 

Signed-off-by: Hugh McMaster <hugh.mcmaster@outlook.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Hugh McMaster 2021-06-25 22:58:10 +10:00 committed by Alexandre Julliard
parent c94ac2e061
commit d790c59349
7 changed files with 44 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
programs/reg/tests/add.c
View File

@ -154,7 +154,7 @@ void delete_key_(const char *file, unsigned line, HKEY root, const char *path, R
}
}
LONG delete_tree(const HKEY key, const char *subkey)
LONG delete_tree(HKEY root, const char *path, REGSAM sam)
{
HKEY hkey;
LONG ret;
@ -162,7 +162,7 @@ LONG delete_tree(const HKEY key, const char *subkey)
DWORD max_subkey_len, subkey_len;
static const char empty[1];
ret = RegOpenKeyExA(key, subkey, 0, KEY_READ, &hkey);
ret = RegOpenKeyExA(root, path, 0, KEY_READ|sam, &hkey);
if (ret) return ret;
ret = RegQueryInfoKeyA(hkey, NULL, NULL, NULL, NULL, &max_subkey_len,
@ -184,11 +184,14 @@ LONG delete_tree(const HKEY key, const char *subkey)
ret = RegEnumKeyExA(hkey, 0, subkey_name, &subkey_len, NULL, NULL, NULL, NULL);
if (ret == ERROR_NO_MORE_ITEMS) break;
if (ret) goto cleanup;
ret = delete_tree(hkey, subkey_name);
ret = delete_tree(hkey, subkey_name, sam);
if (ret) goto cleanup;
}
ret = RegDeleteKeyA(hkey, empty);
if (!sam)
ret = RegDeleteKeyA(hkey, empty);
else
ret = RegDeleteKeyExA(hkey, empty, sam, 0);
cleanup:
HeapFree(GetProcessHeap(), 0, subkey_name);
@ -219,7 +222,7 @@ static void test_command_syntax(void)
{
DWORD r;
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
run_reg_exe("reg add", &r);
@ -322,7 +325,7 @@ static void test_key_formats(void)
verify_key(hkey, "https://winehq.org", 0);
close_key(hkey);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
/* Test validity of trailing backslash after system key */
run_reg_exe("reg add HKCU\\ /v Value1 /t REG_SZ /d foo /f", &r);
@ -409,7 +412,7 @@ static void test_add(void)
verify_reg(hkey, NULL, REG_NONE, "T\0e\0s\0t\0\0", 10, 0);
close_key(hkey);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
}
static void test_reg_none(void)

36
programs/reg/tests/copy.c
View File

@ -126,10 +126,10 @@ static void test_copy_empty_key(void)
{
DWORD r;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
add_key(HKEY_CURRENT_USER, COPY_SRC, 0, NULL);
@ -191,10 +191,10 @@ static void test_copy_simple_data(void)
HKEY hkey;
DWORD r, dword;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
add_key(HKEY_CURRENT_USER, COPY_SRC, 0, &hkey);
@ -261,10 +261,10 @@ static void test_copy_complex_data(void)
HKEY hkey, subkey;
DWORD r, dword;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
add_key(HKEY_CURRENT_USER, COPY_SRC, 0, &hkey);
@ -330,10 +330,10 @@ static void test_copy_key_order(void)
HKEY hkey;
DWORD r;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
add_key(HKEY_CURRENT_USER, COPY_SRC, 0, &hkey);
@ -355,10 +355,10 @@ static void test_copy_value_order(void)
HKEY hkey;
DWORD r;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
add_key(HKEY_CURRENT_USER, COPY_SRC, 0, &hkey);
@ -380,10 +380,10 @@ static void test_copy_hex_data(void)
HKEY hkey;
DWORD r;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
/* Try copying empty hex values */
@ -449,10 +449,10 @@ static void test_copy_embedded_null_values(void)
{
DWORD r;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
test_import_wstr("\xef\xbb\xbfWindows Registry Editor Version 5.00\n\n"
@ -482,10 +482,10 @@ static void test_copy_slashes(void)
HKEY hkey;
DWORD r;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
add_key(HKEY_CURRENT_USER, COPY_SRC, 0, &hkey);
@ -508,10 +508,10 @@ static void test_copy_escaped_null_values(void)
HKEY hkey;
DWORD r;
delete_tree(HKEY_CURRENT_USER, COPY_SRC);
delete_tree(HKEY_CURRENT_USER, COPY_SRC, 0);
verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
add_key(HKEY_CURRENT_USER, COPY_SRC, 0, &hkey);

4
programs/reg/tests/delete.c
View File

@ -23,7 +23,7 @@ static void test_command_syntax(void)
{
DWORD r;
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
run_reg_exe("reg delete", &r);
@ -74,7 +74,7 @@ static void test_delete(void)
DWORD r;
const DWORD deadbeef = 0xdeadbeef;
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
/* Create a test key */

8
programs/reg/tests/export.c
View File

@ -176,7 +176,7 @@ static void test_export(void)
HKEY hkey, subkey;
BYTE hex[4], buffer[8];
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
run_reg_exe("reg export", &r);
@ -287,7 +287,7 @@ static void test_export(void)
run_reg_exe("reg export HKEY_CURRENT_USER\\" KEY_BASE " file.reg /y", &r);
ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
ok(compare_export("file.reg", complex_data_test, 0), "compare_export() failed\n");
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
/* Test the export order of registry keys */
add_key(HKEY_CURRENT_USER, KEY_BASE, 0, &hkey);
@ -413,7 +413,7 @@ static void test_export(void)
run_reg_exe("reg export HKEY_CURRENT_USER\\" KEY_BASE " file.reg /y", &r);
ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
ok(compare_export("file.reg", slashes_test, TODO_REG_COMPARE), "compare_export() failed\n");
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
/* Test escaped null characters */
add_key(HKEY_CURRENT_USER, KEY_BASE, 0, &hkey);
@ -428,7 +428,7 @@ static void test_export(void)
run_reg_exe("reg export HKEY_CURRENT_USER\\" KEY_BASE " file.reg /y", &r);
ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
ok(compare_export("file.reg", escaped_null_test, 0), "compare_export() failed\n");
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
}
START_TEST(export)

12
programs/reg/tests/import.c
View File

@ -101,7 +101,7 @@ static void test_import(void)
LONG err;
BYTE hex[8];
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
run_reg_exe("reg import", &r);
@ -1643,7 +1643,7 @@ static void test_import(void)
close_key(hkey);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
}
static void test_unicode_import(void)
@ -1654,7 +1654,7 @@ static void test_unicode_import(void)
char buffer[24];
BYTE hex[8];
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
test_import_wstr("REGEDIT\n", &r);
@ -3192,7 +3192,7 @@ static void test_unicode_import(void)
close_key(hkey);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
}
static void test_import_with_whitespace(void)
@ -3200,7 +3200,7 @@ static void test_import_with_whitespace(void)
HKEY hkey;
DWORD r, dword;
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
test_import_str(" REGEDIT4\n\n"
@ -3346,7 +3346,7 @@ static void test_unicode_import_with_whitespace(void)
HKEY hkey;
DWORD r, dword;
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
test_import_wstr("\xef\xbb\xbf Windows Registry Editor Version 5.00\n\n"

6
programs/reg/tests/query.c
View File

@ -164,7 +164,7 @@ static void test_query(void)
HKEY hkey, subkey;
BYTE buf[512];
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE, 0);
run_reg_exe("reg query", &r);
@ -270,7 +270,7 @@ static void test_query(void)
ok(r == REG_EXIT_SUCCESS || r == REG_EXIT_FAILURE /* WinXP */,
"got exit code %d, expected 0\n", r);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
/* Subkeys only */
add_key(HKEY_CURRENT_USER, KEY_BASE, 0, &hkey);
@ -288,7 +288,7 @@ static void test_query(void)
ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
compare_query(buf, test8b, FALSE, 0);
delete_tree(HKEY_CURRENT_USER, KEY_BASE);
delete_tree(HKEY_CURRENT_USER, KEY_BASE, 0);
}
START_TEST(query)

2
programs/reg/tests/reg_test.h
View File

@ -64,7 +64,7 @@ void add_key_(const char *file, unsigned line, const HKEY root, const char *path
#define delete_key(r,p,s) delete_key_(__FILE__,__LINE__,r,p,s)
void delete_key_(const char *file, unsigned line, HKEY root, const char *path, REGSAM sam);
LONG delete_tree(const HKEY key, const char *subkey);
LONG delete_tree(HKEY root, const char *path, REGSAM sam);
#define add_value(k,n,t,d,s) add_value_(__FILE__,__LINE__,k,n,t,d,s)
void add_value_(const char *file, unsigned line, HKEY hkey, const char *name,