Add --list-registry function

Change-Id: Ic724e68aee9e5d4581a6d6de74d40aeaff20094a
Reviewed-on: https://gerrit.libreoffice.org/c/core/+/104786
Tested-by: Tor Lillqvist <tml@collabora.com>
Reviewed-by: Tor Lillqvist <tml@collabora.com>

1 files changed, 100 insertions, 40 deletions

diff --git a/solenv/wsl/wsl-lo-helper.cpp b/solenv/wsl/wsl-lo-helper.cpp
index e50ebb7cbe8c..1a90580b6f19 100644
--- a/solenv/wsl/wsl-lo-helper.cpp
+++ b/solenv/wsl/wsl-lo-helper.cpp
@@ -36,6 +36,54 @@ static void print_result(const wchar_t* argv0, const wchar_t* result)
     fflush(stdout);
 }
 
+static REGSAM parse_bitness(const wchar_t* argv0, const wchar_t* command, const wchar_t* bitness)
+{
+    if (wcscmp(bitness, L"32") == 0)
+        return KEY_WOW64_32KEY;
+    else if (wcscmp(bitness, L"64") == 0)
+        return KEY_WOW64_64KEY;
+    else
+    {
+        fprintf(stderr, "%S %S: Use 32 or 64.\n", argv0, command);
+        exit(1);
+    }
+}
+
+static void parse_hive(const wchar_t* argv0, const wchar_t* path, HKEY* hivep,
+                       const wchar_t** hive_namep)
+{
+    if (wcscmp(path, L"HKEY_CLASSES_ROOT") == 0)
+    {
+        *hivep = HKEY_CLASSES_ROOT;
+        *hive_namep = L"HKEY_CLASSES_ROOT";
+    }
+    else if (wcscmp(path, L"HKEY_CURRENT_CONFIG") == 0)
+    {
+        *hivep = HKEY_CURRENT_CONFIG;
+        *hive_namep = L"HKEY_CURRENT_CONFIG";
+    }
+    else if (wcscmp(path, L"HKEY_CURRENT_USER") == 0)
+    {
+        *hivep = HKEY_CURRENT_USER;
+        *hive_namep = L"HKEY_CURRENT_USER";
+    }
+    else if (wcscmp(path, L"HKEY_LOCAL_MACHINE") == 0)
+    {
+        *hivep = HKEY_LOCAL_MACHINE;
+        *hive_namep = L"HKEY_LOCAL_MACHINE";
+    }
+    else if (wcscmp(path, L"HKEY_USERS") == 0)
+    {
+        *hivep = HKEY_USERS;
+        *hive_namep = L"HKEY_USERS";
+    }
+    else
+    {
+        fprintf(stderr, "%S: Invalid Registry hive %S.\n", argv0, path);
+        exit(1);
+    }
+}
+
 int wmain(int argc, wchar_t** argv)
 {
     if (argc <= 1)
@@ -45,6 +93,8 @@ int wmain(int argc, wchar_t** argv)
         fprintf(stderr, "    Print the 8.3 form of a Windows pathnam. Fail it not present.\n");
         fprintf(stderr, "%S --read-registry [32|64] <path>\n", argv[0]);
         fprintf(stderr, "    Read a string value from the Registry and print it.\n");
+        fprintf(stderr, "%S --list-registry [32|64] <path>\n", argv[0]);
+        fprintf(stderr, "    List subkeys of a key in the Registry.\n");
         exit(1);
     }
 
@@ -77,16 +127,7 @@ int wmain(int argc, wchar_t** argv)
             exit(1);
         }
 
-        REGSAM sam;
-        if (wcscmp(argv[2], L"32") == 0)
-            sam = KEY_WOW64_32KEY;
-        else if (wcscmp(argv[2], L"64") == 0)
-            sam = KEY_WOW64_64KEY;
-        else
-        {
-            fprintf(stderr, "%S --read-registry: Use 32 or 64.\n", argv[0]);
-            exit(1);
-        }
+        REGSAM sam = parse_bitness(argv[0], L"--read-registry", argv[2]);
 
         wchar_t* path = wcsdup(argv[3]);
 
@@ -106,36 +147,7 @@ int wmain(int argc, wchar_t** argv)
 
         HKEY hive;
         const wchar_t* hive_name;
-        if (wcscmp(path, L"HKEY_CLASSES_ROOT") == 0)
-        {
-            hive = HKEY_CLASSES_ROOT;
-            hive_name = L"HKEY_CLASSES_ROOT";
-        }
-        else if (wcscmp(path, L"HKEY_CURRENT_CONFIG") == 0)
-        {
-            hive = HKEY_CURRENT_CONFIG;
-            hive_name = L"HKEY_CURRENT_CONFIG";
-        }
-        else if (wcscmp(path, L"HKEY_CURRENT_USER") == 0)
-        {
-            hive = HKEY_CURRENT_USER;
-            hive_name = L"HKEY_CURRENT_USER";
-        }
-        else if (wcscmp(path, L"HKEY_LOCAL_MACHINE") == 0)
-        {
-            hive = HKEY_LOCAL_MACHINE;
-            hive_name = L"HKEY_LOCAL_MACHINE";
-        }
-        else if (wcscmp(path, L"HKEY_USERS") == 0)
-        {
-            hive = HKEY_USERS;
-            hive_name = L"HKEY_USERS";
-        }
-        else
-        {
-            fprintf(stderr, "%S: Invalid Registry hive %S.\n", argv[0], path);
-            exit(1);
-        }
+        parse_hive(argv[0], path, &hive, &hive_name);
 
         DWORD type;
         wchar_t result[1000];
@@ -174,6 +186,54 @@ int wmain(int argc, wchar_t** argv)
         }
         print_result(argv[0], result);
     }
+    else if (wcscmp(argv[1], L"--list-registry") == 0)
+    {
+        if (argc != 4)
+        {
+            fprintf(stderr, "%S --list-registry: Bitness and path arguments expected.\n", argv[0]);
+            exit(1);
+        }
+
+        REGSAM sam = parse_bitness(argv[0], L"--list-registry", argv[2]);
+
+        wchar_t* path = wcsdup(argv[3]);
+
+        for (wchar_t* p = path; *p != L'\0'; p++)
+            if (*p == '/')
+                *p = '\\';
+
+        wchar_t* const first_backslash = wcschr(path, L'\\');
+        if (first_backslash == NULL)
+        {
+            fprintf(stderr, "%S: Invalid path %S to key in the Registry.\n", argv[0], path);
+            exit(1);
+        }
+
+        *first_backslash = L'\0';
+        wchar_t* const key_path = first_backslash + 1;
+
+        HKEY hive;
+        const wchar_t* hive_name;
+        parse_hive(argv[0], path, &hive, &hive_name);
+
+        HKEY key;
+        if (RegOpenKeyExW(hive, key_path, 0, KEY_ENUMERATE_SUB_KEYS | sam, &key) != ERROR_SUCCESS)
+        {
+            fprintf(stderr, "%S: Opening key %S\\%S in %S-bit Registry failed.\n", argv[0],
+                    hive_name, key_path, argv[2]);
+            exit(1);
+        }
+
+        DWORD index = 0;
+        wchar_t name[256];
+        DWORD namelen = sizeof(name) / sizeof(name[0]);
+        while (RegEnumKeyExW(key, index, name, &namelen, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
+        {
+            printf("%S\n", name);
+            index++;
+            namelen = sizeof(name) / sizeof(name[0]);
+        }
+    }
     else
     {
         fprintf(stderr, "%S: Unrecognized sub-command %S.\n", argv[0], argv[1]);