From: Eric Biggers <ebiggers3@gmail.com>
Date: Sun, 15 Mar 2015 01:22:22 +0000 (-0500)
Subject: win32_apply.c: handle target directory more like other directories
X-Git-Tag: v1.8.1~75
X-Git-Url: https://wimlib.net/git/?p=wimlib;a=commitdiff_plain;h=277d7798255f0f5f63b5659cee55eba9d9d1054d

win32_apply.c: handle target directory more like other directories

If we don't send the target directory through the same code path as the
other directories, we can run into several inconsistencies such as empty
named data streams not being created and compression attributes not being
set or cleared.

We actually *can* send the target directory through create_directory() as
long as we don't ask for DELETE access (in case it's the current working
directory) or set its short name (since that doesn't make sense anyway).
---

diff --git a/src/win32_apply.c b/src/win32_apply.c
index 630a77f1..b66717ec 100644
--- a/src/win32_apply.c
+++ b/src/win32_apply.c
@@ -1347,32 +1347,37 @@ create_any_empty_ads(const struct wim_dentry *dentry,
  * Returns 0, WIMLIB_ERR_MKDIR, or WIMLIB_ERR_SET_SHORT_NAME.
  */
 static int
-create_directory(const struct wim_dentry *dentry,
-		 struct win32_apply_ctx *ctx)
+create_directory(const struct wim_dentry *dentry, struct win32_apply_ctx *ctx)
 {
-	HANDLE h;
+	DWORD perms;
 	NTSTATUS status;
+	HANDLE h;
 	int ret;
 
 	/* DELETE is needed for set_short_name(); GENERIC_READ and GENERIC_WRITE
-	 * are needed for adjust_compression_attribute().
-	 *
-	 * FILE_ATTRIBUTE_SYSTEM is needed to ensure that
+	 * are needed for adjust_compression_attribute().  */
+	perms = GENERIC_READ | GENERIC_WRITE;
+	if (!dentry_is_root(dentry))
+		perms |= DELETE;
+
+	/* FILE_ATTRIBUTE_SYSTEM is needed to ensure that
 	 * FILE_ATTRIBUTE_ENCRYPTED doesn't get set before we want it to be.  */
-	status = create_file(&h, GENERIC_READ | GENERIC_WRITE | DELETE, NULL,
-			     FILE_ATTRIBUTE_SYSTEM, FILE_OPEN_IF, FILE_DIRECTORY_FILE,
-			     dentry, ctx);
+	status = create_file(&h, perms, NULL, FILE_ATTRIBUTE_SYSTEM,
+			     FILE_OPEN_IF, FILE_DIRECTORY_FILE, dentry, ctx);
 	if (!NT_SUCCESS(status)) {
 		winnt_error(status, L"Can't create directory \"%ls\"",
 			    current_path(ctx));
 		return WIMLIB_ERR_MKDIR;
 	}
 
-	ret = set_short_name(h, dentry, ctx);
-
-	if (!ret)
-		ret = adjust_compression_attribute(h, dentry, ctx);
+	if (!dentry_is_root(dentry)) {
+		ret = set_short_name(h, dentry, ctx);
+		if (ret)
+			goto out;
+	}
 
+	ret = adjust_compression_attribute(h, dentry, ctx);
+out:
 	(*func_NtClose)(h);
 	return ret;
 }
@@ -1400,19 +1405,14 @@ create_directories(struct list_head *dentry_list,
 		 * FILE_ATTRIBUTE_DIRECTORY|FILE_ATTRIBUTE_REPARSE_POINT, but we
 		 * wait until later to actually set the reparse data.  */
 
-		/* If the root dentry is being extracted, it was already done so
-		 * in prepare_target().  */
-		if (!dentry_is_root(dentry)) {
-			ret = create_directory(dentry, ctx);
-			ret = check_apply_error(dentry, ctx, ret);
-			if (ret)
-				return ret;
+		ret = create_directory(dentry, ctx);
 
+		if (!ret)
 			ret = create_any_empty_ads(dentry, ctx);
-			ret = check_apply_error(dentry, ctx, ret);
-			if (ret)
-				return ret;
-		}
+
+		ret = check_apply_error(dentry, ctx, ret);
+		if (ret)
+			return ret;
 
 		ret = report_file_created(&ctx->common);
 		if (ret)