reg: Validate the registry key before commencing the export operation.

Signed-off-by: Hugh McMaster <[email protected]> Signed-off-by: Alexandre Julliard <[email protected]>
Gravemind · Dec 4, 2017 · 1749401 · 1749401
1 parent e0aadc0
commit 1749401
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/programs/reg/export.c b/programs/reg/export.c
@@ -38,7 +38,7 @@ static BOOL is_overwrite_switch(const WCHAR *s)
 
 int reg_export(int argc, WCHAR *argv[])
 {
-    HKEY root;
+    HKEY root, hkey;
     WCHAR *path, *long_key;
 
     if (argc == 3 || argc > 5)
@@ -50,7 +50,16 @@ int reg_export(int argc, WCHAR *argv[])
     if (argc == 5 && !is_overwrite_switch(argv[4]))
         goto error;
 
+    if (RegOpenKeyExW(root, path, 0, KEY_READ, &hkey))
+    {
+        output_message(STRING_INVALID_KEY);
+        return 1;
+    }
+
     FIXME(": operation not yet implemented\n");
+
+    RegCloseKey(hkey);
+
     return 1;
 
 error: