Fix glob()

[wimlib] / programs / imagex-win32.c

diff --git a/programs/imagex-win32.c b/programs/imagex-win32.c
index 3c449a7733b108129503a32b4fbad682e2c2eeae..166485005308ead71bb54b1f1b6508babea06f5c 100644 (file)
--- a/programs/imagex-win32.c
+++ b/programs/imagex-win32.c
@@ -1,6 +1,7 @@
 
 /* Replacements for functions needed specifically by the 'imagex' program in
- * Windows native builds */
+ * Windows native builds; also, Windows-specific code to acquire and release
+ * privileges needed to backup and restore files */
 
 #ifndef __WIN32__
 #  error "This file contains Windows code"
@@ -11,7 +12,7 @@
 #include <errno.h>
 #include <string.h>
 #include <assert.h>
-
+#include <stdio.h>
 
 /* Replacement for glob() in Windows native builds. */
 int glob(const char *pattern, int flags,
@@ -24,6 +25,20 @@ int glob(const char *pattern, int flags,
        int ret;
        size_t nspaces;
 
+       const char *backslash, *forward_slash, *end_slash;
+       size_t prefix_len;
+
+       backslash = strrchr(pattern, '\\');
+       end_slash = strrchr(pattern, '/');
+
+       if (backslash > end_slash)
+               end_slash = backslash;
+
+       if (end_slash)
+               prefix_len = end_slash - pattern + 1;
+       else
+               prefix_len = 0;
+
        /* This function does not support all functionality of the POSIX glob(),
         * so make sure the parameters are consistent with supported
         * functionality. */
@@ -48,7 +63,7 @@ int glob(const char *pattern, int flags,
        pglob->gl_pathv = NULL;
        nspaces = 0;
        do {
-               char *filename;
+               char *path;
                if (pglob->gl_pathc == nspaces) {
                        size_t new_nspaces;
                        char **pathv;
@@ -61,10 +76,16 @@ int glob(const char *pattern, int flags,
                        pglob->gl_pathv = pathv;
                        nspaces = new_nspaces;
                }
-               filename = strdup(dat.cFileName);
-               if (!filename)
+               size_t filename_len = strlen(dat.cFileName);
+               size_t len_needed = prefix_len + filename_len;
+
+               path = malloc(len_needed + 1);
+               if (!path)
                        goto oom;
-               pglob->gl_pathv[pglob->gl_pathc++] = filename;
+
+               memcpy(path, pattern, prefix_len);
+               memcpy(path + prefix_len, dat.cFileName, filename_len + 1);
+               pglob->gl_pathv[pglob->gl_pathc++] = path;
        } while (FindNextFileA(hFind, &dat));
        err = GetLastError();
        CloseHandle(hFind);
@@ -92,5 +113,81 @@ void globfree(glob_t *pglob)
        size_t i;
        for (i = 0; i < pglob->gl_pathc; i++)
                free(pglob->gl_pathv[i]);
-       free(pglob->gl_pathv[i]);
+       free(pglob->gl_pathv);
+}
+
+static bool
+win32_modify_privilege(const char *privilege, bool enable)
+{
+       HANDLE hToken;
+       LUID luid;
+       TOKEN_PRIVILEGES newState;
+       bool ret = false;
+
+       if (!OpenProcessToken(GetCurrentProcess(),
+                             TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
+                             &hToken))
+       {
+               goto out;
+       }
+
+       if (!LookupPrivilegeValue(NULL, privilege, &luid)) {
+               goto out;
+       }
+
+       newState.PrivilegeCount = 1;
+       newState.Privileges[0].Luid = luid;
+       newState.Privileges[0].Attributes = (enable ? SE_PRIVILEGE_ENABLED : 0);
+       ret = AdjustTokenPrivileges(hToken, FALSE, &newState, 0, NULL, NULL);
+       CloseHandle(hToken);
+out:
+       if (!ret) {
+               fprintf(stderr, "WARNING: Failed to %s privilege %s\n",
+                       enable ? "enable" : "disable", privilege);
+               fprintf(stderr,
+                       "WARNING: The program will continue, "
+                       "but if permission issues are\n"
+                       "encountered, you may need to run "
+                       "this program as the administrator\n");
+       }
+       return ret;
+}
+
+static void
+win32_modify_capture_privileges(bool enable)
+{
+       win32_modify_privilege(SE_BACKUP_NAME, enable);
+       win32_modify_privilege(SE_SECURITY_NAME, enable);
+}
+
+static void
+win32_modify_restore_privileges(bool enable)
+{
+       win32_modify_privilege(SE_RESTORE_NAME, enable);
+       win32_modify_privilege(SE_SECURITY_NAME, enable);
+       win32_modify_privilege(SE_TAKE_OWNERSHIP_NAME, enable);
+}
+
+void
+win32_acquire_capture_privileges()
+{
+       win32_modify_capture_privileges(true);
+}
+
+void
+win32_release_capture_privileges()
+{
+       win32_modify_capture_privileges(false);
+}
+
+void
+win32_acquire_restore_privileges()
+{
+       win32_modify_restore_privileges(true);
+}
+
+void
+win32_release_restore_privileges()
+{
+       win32_modify_restore_privileges(false);
 }