reg: Fail if access is denied when adding registry data.

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

reg: Fail if access is denied when adding registry data.
d433d1f1 · Hugh McMaster · Alexandre Julliard · a0dc4d7c · d433d1f1
Commit d433d1f1 authored May 03, 2021 by Hugh McMaster Committed by Alexandre Julliard May 03, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 2 deletions

add.c programs/reg/add.c +10 -2

No files found.
--- a/programs/reg/add.c
+++ b/programs/reg/add.c
@@ -156,6 +156,7 @@ static int run_add(HKEY root, WCHAR *path, WCHAR *value_name, BOOL value_empty,
    HKEY hkey;
    DWORD data_type, data_size;
    BYTE *reg_data = NULL;
+    LONG rc;

    if (RegCreateKeyExW(root, path, 0, NULL, REG_OPTION_NON_VOLATILE,
                        KEY_READ|KEY_WRITE, NULL, &hkey, NULL))
@@ -199,10 +200,17 @@ static int run_add(HKEY root, WCHAR *path, WCHAR *value_name, BOOL value_empty,
        return 1;
    }

-    RegSetValueExW(hkey, value_name, 0, data_type, reg_data, data_size);
-    free(reg_data);
+    rc = RegSetValueExW(hkey, value_name, 0, data_type, reg_data, data_size);

+    free(reg_data);
    RegCloseKey(hkey);
+
+    if (rc)
+    {
+        output_message(STRING_ACCESS_DENIED);
+        return 1;
+    }
+
    output_message(STRING_SUCCESS);

    return 0;