[PATCH 05/13] reg: Validate the registry key before commencing the export operation

[Hugh McMaster]

Signed-off-by: Hugh McMaster <hugh.mcmaster(a)outlook.com>
---
 programs/reg/export.c | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/programs/reg/export.c b/programs/reg/export.c
index 720b9ef703..9d0d33274e 100644
--- 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:
-- 
2.15.0