X-Git-Url: https://wimlib.net/git/?a=blobdiff_plain;f=src%2Fresource.c;h=5c0225ab6d55023c5c8ef9c60bdda7b280fdc5ac;hb=8b74fa415edfefeee5869704d39ee8de1e82b422;hp=9d8fe921490a38775975f438f1cb2a5be46c0f81;hpb=50407ddd59aa2d0906225a4526eaefbf6da8450f;p=wimlib

diff --git a/src/resource.c b/src/resource.c
index 9d8fe921..5c0225ab 100644
--- a/src/resource.c
+++ b/src/resource.c
@@ -193,6 +193,7 @@ read_compressed_wim_resource(const struct wim_resource_spec * const rspec,
 		      "expected power-of-2 chunk size (got %"PRIu32")",
 		      chunk_size);
 		ret = WIMLIB_ERR_INVALID_CHUNK_SIZE;
+		errno = EINVAL;
 		goto out_free_memory;
 	}
 
@@ -207,8 +208,11 @@ read_compressed_wim_resource(const struct wim_resource_spec * const rspec,
 	} else {
 		ret = wimlib_create_decompressor(ctype, chunk_size, NULL,
 						 &decompressor);
-		if (ret)
+		if (ret) {
+			if (ret != WIMLIB_ERR_NOMEM)
+				errno = EINVAL;
 			goto out_free_memory;
+		}
 	}
 
 	const u32 chunk_order = bsr32(chunk_size);
@@ -758,7 +762,6 @@ read_wim_stream_prefix(const struct wim_lookup_table_entry *lte, u64 size,
 					 cb, cb_ctx);
 }
 
-#ifndef __WIN32__
 /* This function handles reading stream data that is located in an external
  * file,  such as a file that has been added to the WIM image through execution
  * of a wimlib_add_command.
@@ -779,7 +782,7 @@ read_file_on_disk_prefix(const struct wim_lookup_table_entry *lte, u64 size,
 
 	DEBUG("Reading %"PRIu64" bytes from \"%"TS"\"", size, lte->file_on_disk);
 
-	raw_fd = open(lte->file_on_disk, O_BINARY | O_RDONLY);
+	raw_fd = topen(lte->file_on_disk, O_BINARY | O_RDONLY);
 	if (raw_fd < 0) {
 		ERROR_WITH_ERRNO("Can't open \"%"TS"\"", lte->file_on_disk);
 		return WIMLIB_ERR_OPEN;
@@ -789,7 +792,34 @@ read_file_on_disk_prefix(const struct wim_lookup_table_entry *lte, u64 size,
 	filedes_close(&fd);
 	return ret;
 }
-#endif /* !__WIN32__ */
+
+#ifdef WITH_FUSE
+static int
+read_staging_file_prefix(const struct wim_lookup_table_entry *lte, u64 size,
+			 consume_data_callback_t cb, void *cb_ctx)
+{
+	int raw_fd;
+	struct filedes fd;
+	int ret;
+
+	wimlib_assert(size <= lte->size);
+
+	DEBUG("Reading %"PRIu64" bytes from staging file \"%s\"",
+	      size, lte->staging_file_name);
+
+	raw_fd = openat(lte->staging_dir_fd, lte->staging_file_name,
+			O_RDONLY | O_NOFOLLOW);
+	if (raw_fd < 0) {
+		ERROR_WITH_ERRNO("Can't open staging file \"%s\"",
+				 lte->staging_file_name);
+		return WIMLIB_ERR_OPEN;
+	}
+	filedes_init(&fd, raw_fd);
+	ret = read_raw_file_data(&fd, 0, size, cb, cb_ctx);
+	filedes_close(&fd);
+	return ret;
+}
+#endif
 
 /* This function handles the trivial case of reading stream data that is, in
  * fact, already located in an in-memory buffer.  */
@@ -827,19 +857,16 @@ read_stream_prefix(const struct wim_lookup_table_entry *lte, u64 size,
 {
 	static const read_stream_prefix_handler_t handlers[] = {
 		[RESOURCE_IN_WIM]             = read_wim_stream_prefix,
-	#ifdef __WIN32__
-		[RESOURCE_IN_FILE_ON_DISK]    = read_win32_file_prefix,
-	#else
 		[RESOURCE_IN_FILE_ON_DISK]    = read_file_on_disk_prefix,
-	#endif
 		[RESOURCE_IN_ATTACHED_BUFFER] = read_buffer_prefix,
 	#ifdef WITH_FUSE
-		[RESOURCE_IN_STAGING_FILE]    = read_file_on_disk_prefix,
+		[RESOURCE_IN_STAGING_FILE]    = read_staging_file_prefix,
 	#endif
 	#ifdef WITH_NTFS_3G
 		[RESOURCE_IN_NTFS_VOLUME]     = read_ntfs_file_prefix,
 	#endif
 	#ifdef __WIN32__
+		[RESOURCE_IN_WINNT_FILE_ON_DISK] = read_winnt_file_prefix,
 		[RESOURCE_WIN32_ENCRYPTED]    = read_win32_encrypted_file_prefix,
 	#endif
 	};
@@ -1007,9 +1034,9 @@ streamifier_cb(const void *chunk, size_t size, void *_ctx)
 	/* Consume the chunk.  */
 	ret = (*ctx->cbs.consume_chunk)(chunk, size,
 					ctx->cbs.consume_chunk_ctx);
+	ctx->cur_stream_offset += size;
 	if (ret)
 		return ret;
-	ctx->cur_stream_offset += size;
 
 	if (ctx->cur_stream_offset == ctx->cur_stream->size) {
 		/* Finished reading all the data for a stream.  */