X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fwrite.c;h=4c68921cc9164efb926e58325f0e0a99109ec6bc;hp=54257e046f2effae9916681cfe4f60ff3be427a5;hb=58daaca96dc8b23e345342e844c58b1972e5fedf;hpb=1da527afb0ea8fb58838b6ade3a93aea198191da

diff --git a/src/write.c b/src/write.c
index 54257e04..4c68921c 100644
--- a/src/write.c
+++ b/src/write.c
@@ -1798,6 +1798,7 @@ finish_write(WIMStruct *w, int image, int write_flags,
 			struct wim_header checkpoint_hdr;
 			memcpy(&checkpoint_hdr, &hdr, sizeof(struct wim_header));
 			zero_resource_entry(&checkpoint_hdr.integrity);
+			checkpoint_hdr.flags |= WIM_HDR_FLAG_WRITE_IN_PROGRESS;
 			ret = write_header(&checkpoint_hdr, w->out_fd);
 			if (ret)
 				goto out_close_wim;
@@ -2112,11 +2113,11 @@ overwrite_wim_inplace(WIMStruct *w, int write_flags,
 		return ret;
 	}
 
-	/* Write header with write in progress flag set. */
-	w->hdr.flags |= WIM_HDR_FLAG_WRITE_IN_PROGRESS;
-	ret = write_header(&w->hdr, w->out_fd);
-	w->hdr.flags &= ~WIM_HDR_FLAG_WRITE_IN_PROGRESS;
+	/* Set WIM_HDR_FLAG_WRITE_IN_PROGRESS flag in header. */
+	ret = write_header_flags(w->hdr.flags | WIM_HDR_FLAG_WRITE_IN_PROGRESS,
+				 w->out_fd);
 	if (ret) {
+		ERROR_WITH_ERRNO("Error updating WIM header flags");
 		close_wim_writable(w);
 		goto out_unlock_wim;
 	}