Hugh McMaster : reg/tests: Add more data tests for the 'copy' command.

Alexandre Julliard julliard at winehq.org
Fri May 14 15:47:04 CDT 2021


Module: wine
Branch: master
Commit: baf0254bbb617c3b21c5b068e645615418a5be19
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=baf0254bbb617c3b21c5b068e645615418a5be19

Author: Hugh McMaster <hugh.mcmaster at outlook.com>
Date:   Fri May 14 22:58:02 2021 +1000

reg/tests: Add more data tests for the 'copy' command.

Signed-off-by: Hugh McMaster <hugh.mcmaster at outlook.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 programs/reg/tests/copy.c     | 93 +++++++++++++++++++++++++++++++++++++++++++
 programs/reg/tests/export.c   | 58 +++++++++++++--------------
 programs/reg/tests/reg_test.h |  3 ++
 3 files changed, 125 insertions(+), 29 deletions(-)

diff --git a/programs/reg/tests/copy.c b/programs/reg/tests/copy.c
index 579d91a35ab..3ed75acad4d 100644
--- a/programs/reg/tests/copy.c
+++ b/programs/reg/tests/copy.c
@@ -395,6 +395,96 @@ static void test_copy_hex_data(void)
     todo_wine ok(compare_export("file.reg", hex_types_test, 0), "compare_export() failed\n");
 }
 
+static void test_copy_embedded_null_values(void)
+{
+    DWORD r;
+
+    delete_tree(HKEY_CURRENT_USER, COPY_SRC);
+    verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC);
+
+    delete_tree(HKEY_CURRENT_USER, KEY_BASE);
+    verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE);
+
+    test_import_wstr("\xef\xbb\xbfWindows Registry Editor Version 5.00\n\n"
+                     "[HKEY_CURRENT_USER\\" COPY_SRC "]\n"
+                     "\"Wine4a\"=dword:00000005\n"
+                     "\"Wine4b\"=hex(1):00,00,00,00,00,00,00,00\n"
+                     "\"Wine4c\"=\"Value\"\n"
+                     "\"Wine4d\"=hex(1):00,00,61,00,62,00,63,00\n"
+                     "\"Wine4e\"=dword:00000100\n"
+                     "\"Wine4f\"=hex(1):00,00,56,00,61,00,6c,00,75,00,65,00,00,00\n"
+                     "\"Wine4g\"=\"Value2\"\n"
+                     "\"Wine4h\"=hex(1):61,00,62,00,63,00,00,00, \\\n"
+                     "  64,00,65,00,66,00,00,00\n\n", &r);
+    ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
+
+    run_reg_exe("reg copy HKCU\\" COPY_SRC " HKCU\\" KEY_BASE " /f", &r);
+    todo_wine ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
+    todo_wine verify_key(HKEY_CURRENT_USER, KEY_BASE);
+
+    run_reg_exe("reg export HKEY_CURRENT_USER\\" KEY_BASE " file.reg /y", &r);
+    todo_wine ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
+    todo_wine ok(compare_export("file.reg", embedded_null_test, 0), "compare_export() failed\n");
+}
+
+static void test_copy_slashes(void)
+{
+    HKEY hkey;
+    DWORD r;
+
+    delete_tree(HKEY_CURRENT_USER, COPY_SRC);
+    verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC);
+
+    delete_tree(HKEY_CURRENT_USER, KEY_BASE);
+    verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE);
+
+    add_key(HKEY_CURRENT_USER, COPY_SRC, &hkey);
+    add_key(hkey, "https://winehq.org", NULL);
+    add_value(hkey, "count/up", REG_SZ, "one/two/three", 14);
+    add_value(hkey, "\\foo\\bar", REG_SZ, "", 1);
+    close_key(hkey);
+
+    run_reg_exe("reg copy HKCU\\" COPY_SRC " HKCU\\" KEY_BASE " /s /f", &r);
+    todo_wine ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
+    todo_wine verify_key(HKEY_CURRENT_USER, KEY_BASE);
+
+    run_reg_exe("reg export HKCU\\" KEY_BASE " file.reg /y", &r);
+    todo_wine ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
+    todo_wine ok(compare_export("file.reg", slashes_test, 0), "compare_export() failed\n");
+}
+
+static void test_copy_escaped_null_values(void)
+{
+    HKEY hkey;
+    DWORD r;
+
+    delete_tree(HKEY_CURRENT_USER, COPY_SRC);
+    verify_key_nonexist(HKEY_CURRENT_USER, COPY_SRC);
+
+    delete_tree(HKEY_CURRENT_USER, KEY_BASE);
+    verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE);
+
+    add_key(HKEY_CURRENT_USER, COPY_SRC, &hkey);
+    add_value(hkey, "Wine5a", REG_SZ, "\\0", 3);
+    add_value(hkey, "Wine5b", REG_SZ, "\\0\\0", 5);
+    add_value(hkey, "Wine5c", REG_SZ, "Value1\\0", 9);
+    add_value(hkey, "Wine5d", REG_SZ, "Value2\\0\\0\\0\\0", 15);
+    add_value(hkey, "Wine5e", REG_SZ, "Value3\\0Value4", 15);
+    add_value(hkey, "Wine5f", REG_SZ, "\\0Value5", 9);
+    close_key(hkey);
+
+    run_reg_exe("reg copy HKCU\\" COPY_SRC " HKCU\\" KEY_BASE " /f", &r);
+    todo_wine ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
+    todo_wine verify_key(HKEY_CURRENT_USER, KEY_BASE);
+
+    run_reg_exe("reg export HKCU\\" KEY_BASE " file.reg /y", &r);
+    todo_wine ok(r == REG_EXIT_SUCCESS, "got exit code %d, expected 0\n", r);
+    todo_wine ok(compare_export("file.reg", escaped_null_test, 0), "compare_export() failed\n");
+
+    delete_key(HKEY_CURRENT_USER, COPY_SRC);
+    todo_wine delete_key(HKEY_CURRENT_USER, KEY_BASE);
+}
+
 START_TEST(copy)
 {
     DWORD r;
@@ -409,4 +499,7 @@ START_TEST(copy)
     test_copy_simple_data();
     test_copy_complex_data();
     test_copy_hex_data();
+    test_copy_embedded_null_values();
+    test_copy_slashes();
+    test_copy_escaped_null_values();
 }
diff --git a/programs/reg/tests/export.c b/programs/reg/tests/export.c
index 8ea9bda3f1c..be1d9ea4f86 100644
--- a/programs/reg/tests/export.c
+++ b/programs/reg/tests/export.c
@@ -125,6 +125,35 @@ const char *hex_types_test =
     "\"Wine3b\"=hex:12,34,56,78\r\n"
     "\"Wine3c\"=dword:10203040\r\n\r\n";
 
+const char *embedded_null_test =
+    "\xef\xbb\xbfWindows Registry Editor Version 5.00\r\n\r\n"
+    "[HKEY_CURRENT_USER\\" KEY_BASE "]\r\n"
+    "\"Wine4a\"=dword:00000005\r\n"
+    "\"Wine4b\"=\"\"\r\n"
+    "\"Wine4c\"=\"Value\"\r\n"
+    "\"Wine4d\"=\"\"\r\n"
+    "\"Wine4e\"=dword:00000100\r\n"
+    "\"Wine4f\"=\"\"\r\n"
+    "\"Wine4g\"=\"Value2\"\r\n"
+    "\"Wine4h\"=\"abc\"\r\n\r\n";
+
+const char *slashes_test =
+    "\xef\xbb\xbfWindows Registry Editor Version 5.00\r\n\r\n"
+    "[HKEY_CURRENT_USER\\" KEY_BASE "]\r\n"
+    "\"count/up\"=\"one/two/three\"\r\n"
+    "\"\\\\foo\\\\bar\"=\"\"\r\n\r\n"
+    "[HKEY_CURRENT_USER\\" KEY_BASE "\\https://winehq.org]\r\n\r\n";
+
+const char *escaped_null_test =
+    "\xef\xbb\xbfWindows Registry Editor Version 5.00\r\n\r\n"
+    "[HKEY_CURRENT_USER\\" KEY_BASE "]\r\n"
+    "\"Wine5a\"=\"\\\\0\"\r\n"
+    "\"Wine5b\"=\"\\\\0\\\\0\"\r\n"
+    "\"Wine5c\"=\"Value1\\\\0\"\r\n"
+    "\"Wine5d\"=\"Value2\\\\0\\\\0\\\\0\\\\0\"\r\n"
+    "\"Wine5e\"=\"Value3\\\\0Value4\"\r\n"
+    "\"Wine5f\"=\"\\\\0Value5\"\r\n\r\n";
+
 
 /* Unit tests */
 
@@ -147,35 +176,6 @@ static void test_export(void)
         "\"Value 2\"=\"I was added first!\"\r\n"
         "\"Value 1\"=\"I was added second!\"\r\n\r\n";
 
-    const char *embedded_null_test =
-        "\xef\xbb\xbfWindows Registry Editor Version 5.00\r\n\r\n"
-        "[HKEY_CURRENT_USER\\" KEY_BASE "]\r\n"
-        "\"Wine4a\"=dword:00000005\r\n"
-        "\"Wine4b\"=\"\"\r\n"
-        "\"Wine4c\"=\"Value\"\r\n"
-        "\"Wine4d\"=\"\"\r\n"
-        "\"Wine4e\"=dword:00000100\r\n"
-        "\"Wine4f\"=\"\"\r\n"
-        "\"Wine4g\"=\"Value2\"\r\n"
-        "\"Wine4h\"=\"abc\"\r\n\r\n";
-
-    const char *slashes_test =
-        "\xef\xbb\xbfWindows Registry Editor Version 5.00\r\n\r\n"
-        "[HKEY_CURRENT_USER\\" KEY_BASE "]\r\n"
-        "\"count/up\"=\"one/two/three\"\r\n"
-        "\"\\\\foo\\\\bar\"=\"\"\r\n\r\n"
-        "[HKEY_CURRENT_USER\\" KEY_BASE "\\https://winehq.org]\r\n\r\n";
-
-    const char *escaped_null_test =
-        "\xef\xbb\xbfWindows Registry Editor Version 5.00\r\n\r\n"
-        "[HKEY_CURRENT_USER\\" KEY_BASE "]\r\n"
-        "\"Wine5a\"=\"\\\\0\"\r\n"
-        "\"Wine5b\"=\"\\\\0\\\\0\"\r\n"
-        "\"Wine5c\"=\"Value1\\\\0\"\r\n"
-        "\"Wine5d\"=\"Value2\\\\0\\\\0\\\\0\\\\0\"\r\n"
-        "\"Wine5e\"=\"Value3\\\\0Value4\"\r\n"
-        "\"Wine5f\"=\"\\\\0Value5\"\r\n\r\n";
-
     delete_tree(HKEY_CURRENT_USER, KEY_BASE);
     verify_key_nonexist(HKEY_CURRENT_USER, KEY_BASE);
 
diff --git a/programs/reg/tests/reg_test.h b/programs/reg/tests/reg_test.h
index 40880fa6c0a..14636b4edc9 100644
--- a/programs/reg/tests/reg_test.h
+++ b/programs/reg/tests/reg_test.h
@@ -84,6 +84,9 @@ extern const char *complex_data_test;
 extern const char *empty_hex_test;
 extern const char *empty_hex_test2;
 extern const char *hex_types_test;
+extern const char *slashes_test;
+extern const char *embedded_null_test;
+extern const char *escaped_null_test;
 
 /* import.c */
 #define test_import_str(c,r) import_reg(__FILE__,__LINE__,c,FALSE,r)




More information about the wine-cvs mailing list