X-Git-Url: https://wimlib.net/git/?a=blobdiff_plain;f=programs%2Fimagex-win32.c;h=166485005308ead71bb54b1f1b6508babea06f5c;hb=de12c346dc64404821d52d545e2e1b3d44230f2a;hp=3c449a7733b108129503a32b4fbad682e2c2eeae;hpb=cc1b289394f993fa1a75b845553eea44e85e5051;p=wimlib

diff --git a/programs/imagex-win32.c b/programs/imagex-win32.c
index 3c449a77..16648500 100644
--- 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);
 }