[PATCH] reg: Fail if REG_DWORD is passed with no data or if the data cannot be converted to a DWORD
Hugh McMaster
hugh.mcmaster at outlook.com
Fri Feb 12 06:39:11 CST 2016
Signed-off-by: Hugh McMaster <hugh.mcmaster at outlook.com>
---
programs/reg/reg.c | 3 +++
programs/reg/tests/reg.c | 6 +++---
2 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/programs/reg/reg.c b/programs/reg/reg.c
index 63b518f..ac9b4aa 100644
--- a/programs/reg/reg.c
+++ b/programs/reg/reg.c
@@ -369,6 +369,9 @@ static int reg_add(WCHAR *key_name, WCHAR *value_name, BOOL value_empty,
if (data)
reg_data = get_regdata(data,reg_type,separator,®_count);
+ if (!reg_data && reg_type == REG_DWORD)
+ return 1;
+
RegSetValueExW(subkey,value_name,0,reg_type,reg_data,reg_count);
HeapFree(GetProcessHeap(),0,reg_data);
}
diff --git a/programs/reg/tests/reg.c b/programs/reg/tests/reg.c
index cdf8359..17bff04 100644
--- a/programs/reg/tests/reg.c
+++ b/programs/reg/tests/reg.c
@@ -257,12 +257,12 @@ static void test_add(void)
run_reg_exe("reg add HKCU\\" KEY_BASE " /v dword0 /t REG_DWORD /f /d", &r);
ok(r == REG_EXIT_FAILURE, "got exit code %d, expected 1\n", r);
run_reg_exe("reg add HKCU\\" KEY_BASE " /v dword1 /t REG_DWORD /f", &r);
- todo_wine ok(r == REG_EXIT_FAILURE || broken(r == REG_EXIT_SUCCESS /* WinXP */),
+ ok(r == REG_EXIT_FAILURE || broken(r == REG_EXIT_SUCCESS /* WinXP */),
"got exit code %d, expected 1\n", r);
run_reg_exe("reg add HKCU\\" KEY_BASE " /v dword2 /t REG_DWORD /d zzz /f", &r);
- todo_wine ok(r == REG_EXIT_FAILURE, "got exit code %d, expected 1\n", r);
+ ok(r == REG_EXIT_FAILURE, "got exit code %d, expected 1\n", r);
run_reg_exe("reg add HKCU\\" KEY_BASE " /v dword3 /t REG_DWORD /d deadbeef /f", &r);
- todo_wine ok(r == REG_EXIT_FAILURE, "got exit code %d, expected 1\n", r);
+ ok(r == REG_EXIT_FAILURE, "got exit code %d, expected 1\n", r);
run_reg_exe("reg add HKCU\\" KEY_BASE " /v dword4 /t REG_DWORD /d 123xyz /f", &r);
todo_wine ok(r == REG_EXIT_FAILURE, "got exit code %d, expected 1\n", r);
--
1.9.1
More information about the wine-patches
mailing list