X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fextract_image.c;h=fd7453aab9d8d6f34d2e98dcf85ab70dfc80774f;hp=d9912c7bb0b313c9e5c11b59073c568b283eedf1;hb=5684d46f92ac65ace1e8b45fb0e007f9efa0dfee;hpb=4f0d1a0407a25ecae2ad7b73597075bea810baf7

diff --git a/src/extract_image.c b/src/extract_image.c
index d9912c7b..fd7453aa 100644
--- a/src/extract_image.c
+++ b/src/extract_image.c
@@ -2,15 +2,10 @@
  * extract_image.c
  *
  * Support for extracting WIM files.
- *
- * This code does NOT contain any filesystem-specific features.  In particular,
- * security information (i.e. file permissions) and alternate data streams are
- * ignored, except possibly to read an alternate data stream that contains
- * symbolic link data.
  */
 
 /*
- * Copyright (C) 2012 Eric Biggers
+ * Copyright (C) 2012, 2013 Eric Biggers
  *
  * This file is part of wimlib, a library for working with WIM files.
  *
@@ -30,397 +25,118 @@
 
 #include "config.h"
 
-#include <dirent.h>
-#include <errno.h>
-#include <fcntl.h>
-#include <string.h>
-#include <sys/stat.h>
 #include <stdlib.h>
-#include <sys/time.h>
+#include <sys/stat.h>
+#include <errno.h>
+#include <unistd.h>
 
-#ifdef HAVE_UTIME_H
-#include <utime.h>
+#ifdef __WIN32__
+#  include "win32.h"
 #endif
 
-#include <unistd.h>
-
+#include "wimlib_internal.h"
 #include "dentry.h"
 #include "lookup_table.h"
-#include "timestamp.h"
-#include "wimlib_internal.h"
 #include "xml.h"
 
 #ifdef WITH_NTFS_3G
-#include <ntfs-3g/volume.h>
+#  include <ntfs-3g/volume.h>
 #endif
 
-static int extract_regular_file_linked(struct dentry *dentry,
-				       const char *output_path,
-				       struct apply_args *args,
-				       struct lookup_table_entry *lte)
-{
-	/* This mode overrides the normal hard-link extraction and
-	 * instead either symlinks or hardlinks *all* identical files in
-	 * the WIM, even if they are in a different image (in the case
-	 * of a multi-image extraction) */
-
-	if (args->extract_flags & WIMLIB_EXTRACT_FLAG_HARDLINK) {
-		if (link(lte->extracted_file, output_path) != 0) {
-			ERROR_WITH_ERRNO("Failed to hard link "
-					 "`%s' to `%s'",
-					 output_path, lte->extracted_file);
-			return WIMLIB_ERR_LINK;
-		}
-	} else {
-		int num_path_components;
-		int num_output_dir_path_components;
-		size_t extracted_file_len;
-		char *p;
-		const char *p2;
-		size_t i;
-
-		num_path_components =
-			get_num_path_components(dentry->full_path_utf8) - 1;
-		num_output_dir_path_components =
-			get_num_path_components(args->target);
-
-		if (args->extract_flags & WIMLIB_EXTRACT_FLAG_MULTI_IMAGE) {
-			num_path_components++;
-			num_output_dir_path_components--;
-		}
-		extracted_file_len = strlen(lte->extracted_file);
-
-		char buf[extracted_file_len + 3 * num_path_components + 1];
-		p = &buf[0];
-
-		for (i = 0; i < num_path_components; i++) {
-			*p++ = '.';
-			*p++ = '.';
-			*p++ = '/';
-		}
-		p2 = lte->extracted_file;
-		while (*p2 == '/')
-			p2++;
-		while (num_output_dir_path_components--)
-			p2 = path_next_part(p2, NULL);
-		strcpy(p, p2);
-		if (symlink(buf, output_path) != 0) {
-			ERROR_WITH_ERRNO("Failed to symlink `%s' to "
-					 "`%s'",
-					 buf, lte->extracted_file);
-			return WIMLIB_ERR_LINK;
-		}
-	}
-	return 0;
-}
-
-static int extract_regular_file_unlinked(struct dentry *dentry,
-				         struct apply_args *args,
-				         const char *output_path,
-				         struct lookup_table_entry *lte)
-{
-	/* Normal mode of extraction.  Regular files and hard links are
-	 * extracted in the way that they appear in the WIM. */
-
-	int out_fd;
-	int ret;
-	struct inode *inode = dentry->d_inode;
-
-	if (!((args->extract_flags & WIMLIB_EXTRACT_FLAG_MULTI_IMAGE)
-		&& (args->extract_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK |
-				     WIMLIB_EXTRACT_FLAG_HARDLINK))))
-	{
-		/* If the dentry is one of a hard link set of at least 2
-		 * dentries and one of the other dentries has already been
-		 * extracted, make a hard link to the file corresponding to this
-		 * already-extracted directory.  Otherwise, extract the file,
-		 * and set the inode->extracted_file field so that other
-		 * dentries in the hard link group can link to it. */
-		if (inode->link_count > 1) {
-			if (inode->extracted_file) {
-				DEBUG("Extracting hard link `%s' => `%s'",
-				      output_path, inode->extracted_file);
-				if (link(inode->extracted_file, output_path) != 0) {
-					ERROR_WITH_ERRNO("Failed to hard link "
-							 "`%s' to `%s'",
-							 output_path,
-							 inode->extracted_file);
-					return WIMLIB_ERR_LINK;
-				}
-				return 0;
-			}
-			FREE(inode->extracted_file);
-			inode->extracted_file = STRDUP(output_path);
-			if (!inode->extracted_file) {
-				ERROR("Failed to allocate memory for filename");
-				return WIMLIB_ERR_NOMEM;
-			}
-		}
-	}
-
-	/* Extract the contents of the file to @output_path. */
-
-	out_fd = open(output_path, O_WRONLY | O_CREAT | O_TRUNC, 0644);
-	if (out_fd == -1) {
-		ERROR_WITH_ERRNO("Failed to open the file `%s' for writing",
-				 output_path);
-		return WIMLIB_ERR_OPEN;
-	}
-
-	if (!lte) {
-		/* Empty file with no lookup table entry */
-		DEBUG("Empty file `%s'.", output_path);
-		ret = 0;
-		goto out;
-	}
-
-	ret = extract_wim_resource_to_fd(lte, out_fd, wim_resource_size(lte));
-	if (ret != 0) {
-		ERROR("Failed to extract resource to `%s'", output_path);
-		goto out;
-	}
-	args->progress.extract.completed_bytes += wim_resource_size(lte);
-out:
-	if (close(out_fd) != 0) {
-		ERROR_WITH_ERRNO("Failed to close file `%s'", output_path);
-		ret = WIMLIB_ERR_WRITE;
-	}
-	return ret;
-}
-
-/*
- * Extracts a regular file from the WIM archive.
- */
-static int extract_regular_file(struct dentry *dentry,
-				struct apply_args *args,
-				const char *output_path)
+static int
+do_apply_op(struct wim_dentry *dentry, struct apply_args *args,
+	    int (*apply_dentry_func)(const tchar *, size_t,
+				     struct wim_dentry *, struct apply_args *))
 {
-	struct lookup_table_entry *lte;
-	const struct inode *inode = dentry->d_inode;
-
-	lte = inode_unnamed_lte_resolved(inode);
-
-	if (lte && (args->extract_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK |
-					   WIMLIB_EXTRACT_FLAG_HARDLINK)))
-	{
-		if (lte->extracted_file) {
-			return extract_regular_file_linked(dentry, output_path, args, lte);
-		} else {
-			lte->extracted_file = STRDUP(output_path);
-			if (!lte->extracted_file)
-				return WIMLIB_ERR_NOMEM;
-		}
-	}
-	return extract_regular_file_unlinked(dentry, args, output_path, lte);
+	tchar *p;
+	const tchar *full_path;
+	size_t full_path_nchars;
+
+	wimlib_assert(dentry->_full_path != NULL);
+ 	full_path = dentry->_full_path + 1;
+ 	full_path_nchars = dentry->full_path_nbytes / sizeof(tchar) - 1;
+	tchar output_path[args->target_nchars + 1 +
+			 (full_path_nchars - args->wim_source_path_nchars) + 1];
+	p = output_path;
+
+	tmemcpy(p, args->target, args->target_nchars);
+	p += args->target_nchars;
+
+	if (dentry != args->extract_root) {
+		*p++ = T('/');
+		tmemcpy(p, full_path + args->wim_source_path_nchars,
+			full_path_nchars - args->wim_source_path_nchars);
+		p += full_path_nchars - args->wim_source_path_nchars;
+	}
+	*p = T('\0');
+	return (*apply_dentry_func)(output_path, p - output_path,
+				    dentry, args);
 }
 
-static int extract_symlink(struct dentry *dentry,
-			   struct apply_args *args,
-			   const char *output_path)
-{
-	char target[4096];
-	ssize_t ret = inode_readlink(dentry->d_inode, target,
-				     sizeof(target), args->w, 0);
-	struct lookup_table_entry *lte;
-
-	if (ret <= 0) {
-		ERROR("Could not read the symbolic link from dentry `%s'",
-		      dentry->full_path_utf8);
-		return WIMLIB_ERR_INVALID_DENTRY;
-	}
-	ret = symlink(target, output_path);
-	if (ret != 0) {
-		ERROR_WITH_ERRNO("Failed to symlink `%s' to `%s'",
-				 output_path, target);
-		return WIMLIB_ERR_LINK;
-	}
-	lte = inode_unnamed_lte_resolved(dentry->d_inode);
-	args->progress.extract.completed_bytes += wim_resource_size(lte);
-	return 0;
-}
 
-/*
- * Extracts a directory from the WIM archive.
- *
- * @dentry:		The directory entry for the directory.
- * @output_path:   	The path to which the directory is to be extracted to.
- * @return: 		True on success, false on failure.
- */
-static int extract_directory(const char *output_path, bool is_root)
+/* Extracts a file, directory, or symbolic link from the WIM archive. */
+static int
+apply_dentry_normal(struct wim_dentry *dentry, void *arg)
 {
-	int ret;
-	struct stat stbuf;
-	ret = stat(output_path, &stbuf);
-	if (ret == 0) {
-		if (S_ISDIR(stbuf.st_mode)) {
-			/*if (!is_root)*/
-				/*WARNING("`%s' already exists", output_path);*/
-			return 0;
-		} else {
-			ERROR("`%s' is not a directory", output_path);
-			return WIMLIB_ERR_MKDIR;
-		}
-	} else {
-		if (errno != ENOENT) {
-			ERROR_WITH_ERRNO("Failed to stat `%s'", output_path);
-			return WIMLIB_ERR_STAT;
-		}
-	}
-	/* Compute the output path directory to the directory. */
-	if (mkdir(output_path, S_IRWXU | S_IRGRP | S_IXGRP |
-			       S_IROTH | S_IXOTH) != 0) {
-		ERROR_WITH_ERRNO("Cannot create directory `%s'",
-				 output_path);
-		return WIMLIB_ERR_MKDIR;
-	}
-	return 0;
+#ifdef __WIN32__
+	return do_apply_op(dentry, arg, win32_do_apply_dentry);
+#else
+	return do_apply_op(dentry, arg, unix_do_apply_dentry);
+#endif
 }
 
-/*
- * Extracts a file, directory, or symbolic link from the WIM archive.  For use
- * in for_dentry_in_tree().
- */
-static int apply_dentry_normal(struct dentry *dentry, void *arg)
-{
-	struct apply_args *args = arg;
-	struct inode *inode = dentry->d_inode;
-	size_t len;
-
-	len = strlen(args->target);
-	char output_path[len + dentry->full_path_utf8_len + 1];
-	memcpy(output_path, args->target, len);
-	memcpy(output_path + len, dentry->full_path_utf8, dentry->full_path_utf8_len);
-	output_path[len + dentry->full_path_utf8_len] = '\0';
-
-	if (inode_is_symlink(inode))
-		return extract_symlink(dentry, args, output_path);
-	else if (inode_is_directory(inode))
-		return extract_directory(output_path, false);
-	else
-		return extract_regular_file(dentry, args, output_path);
-}
 
-/* Apply timestamp to extracted file */
-static int apply_dentry_timestamps_normal(struct dentry *dentry, void *arg)
+/* Apply timestamps to an extracted file or directory */
+static int
+apply_dentry_timestamps_normal(struct wim_dentry *dentry, void *arg)
 {
-	struct apply_args *args = arg;
-	size_t len = strlen(args->target);
-	char output_path[len + dentry->full_path_utf8_len + 1];
-	const struct inode *inode = dentry->d_inode;
-	int ret;
-
-	memcpy(output_path, args->target, len);
-	memcpy(output_path + len, dentry->full_path_utf8, dentry->full_path_utf8_len);
-	output_path[len + dentry->full_path_utf8_len] = '\0';
-
-	struct timeval tv[2];
-	wim_timestamp_to_timeval(inode->last_access_time, &tv[0]);
-	wim_timestamp_to_timeval(inode->last_write_time, &tv[1]);
-	#ifdef HAVE_LUTIMES
-	ret = lutimes(output_path, tv);
-	#else
-	ret = -1;
-	errno = ENOSYS;
-	#endif
-	if (ret != 0) {
-		#ifdef HAVE_UTIME
-		if (errno == ENOSYS) {
-			struct utimbuf buf;
-			buf.actime = wim_timestamp_to_unix(inode->last_access_time);
-			buf.modtime = wim_timestamp_to_unix(inode->last_write_time);
-			if (utime(output_path, &buf) == 0)
-				return 0;
-		}
-		#endif
-		if (errno != ENOSYS || args->num_lutimes_warnings < 10) {
-			/*WARNING("Failed to set timestamp on file `%s': %s",*/
-				/*output_path, strerror(errno));*/
-			args->num_lutimes_warnings++;
-		}
-	}
-	return 0;
+#ifdef __WIN32__
+	return do_apply_op(dentry, arg, win32_do_apply_dentry_timestamps);
+#else
+	return do_apply_op(dentry, arg, unix_do_apply_dentry_timestamps);
+#endif
 }
 
-static int maybe_apply_dentry(struct dentry *dentry, void *arg)
+/* Extract a dentry if it hasn't already been extracted and either
+ * WIMLIB_EXTRACT_FLAG_NO_STREAMS is not specified, or the dentry is a directory
+ * and/or has no unnamed stream. */
+static int
+maybe_apply_dentry(struct wim_dentry *dentry, void *arg)
 {
 	struct apply_args *args = arg;
 	int ret;
 
-	if (dentry->is_extracted)
+	if (!dentry->needs_extraction)
 		return 0;
 
-	if (args->extract_flags & WIMLIB_EXTRACT_FLAG_NO_STREAMS)
-		if (inode_unnamed_lte_resolved(dentry->d_inode))
-			return 0;
+	if (args->extract_flags & WIMLIB_EXTRACT_FLAG_NO_STREAMS &&
+	    !dentry_is_directory(dentry) &&
+	    inode_unnamed_lte_resolved(dentry->d_inode) != NULL)
+		return 0;
 
 	if ((args->extract_flags & WIMLIB_EXTRACT_FLAG_VERBOSE) &&
 	     args->progress_func) {
-		args->progress.extract.cur_path = dentry->full_path_utf8;
+		args->progress.extract.cur_path = dentry->_full_path;
 		args->progress_func(WIMLIB_PROGRESS_MSG_EXTRACT_DENTRY,
 				    &args->progress);
 	}
 	ret = args->apply_dentry(dentry, args);
 	if (ret == 0)
-		dentry->is_extracted = 1;
+		dentry->needs_extraction = 0;
 	return ret;
 }
 
-static int cmp_streams_by_wim_position(const void *p1, const void *p2)
-{
-	const struct lookup_table_entry *lte1, *lte2;
-	lte1 = *(const struct lookup_table_entry**)p1;
-	lte2 = *(const struct lookup_table_entry**)p2;
-	if (lte1->resource_entry.offset < lte2->resource_entry.offset)
-		return -1;
-	else if (lte1->resource_entry.offset > lte2->resource_entry.offset)
-		return 1;
-	else
-		return 0;
-}
-
-static int sort_stream_list_by_wim_position(struct list_head *stream_list)
-{
-	struct list_head *cur;
-	size_t num_streams;
-	struct lookup_table_entry **array;
-	size_t i;
-	size_t array_size;
-
-	num_streams = 0;
-	list_for_each(cur, stream_list)
-		num_streams++;
-	array_size = num_streams * sizeof(array[0]);
-	array = MALLOC(array_size);
-	if (!array) {
-		ERROR("Failed to allocate %zu bytes to sort stream entries",
-		      array_size);
-		return WIMLIB_ERR_NOMEM;
-	}
-	cur = stream_list->next;
-	for (i = 0; i < num_streams; i++) {
-		array[i] = container_of(cur, struct lookup_table_entry, staging_list);
-		cur = cur->next;
-	}
-
-	qsort(array, num_streams, sizeof(array[0]), cmp_streams_by_wim_position);
-
-	INIT_LIST_HEAD(stream_list);
-	for (i = 0; i < num_streams; i++)
-		list_add_tail(&array[i]->staging_list, stream_list);
-	FREE(array);
-	return 0;
-}
-
-static void calculate_bytes_to_extract(struct list_head *stream_list,
-				       int extract_flags,
-				       union wimlib_progress_info *progress)
+static void
+calculate_bytes_to_extract(struct list_head *stream_list,
+			   int extract_flags,
+			   union wimlib_progress_info *progress)
 {
-	struct lookup_table_entry *lte;
+	struct wim_lookup_table_entry *lte;
 	u64 total_bytes = 0;
 	u64 num_streams = 0;
 
 	/* For each stream to be extracted... */
-	list_for_each_entry(lte, stream_list, staging_list) {
+	list_for_each_entry(lte, stream_list, extraction_list) {
 		if (extract_flags &
 		    (WIMLIB_EXTRACT_FLAG_SYMLINK | WIMLIB_EXTRACT_FLAG_HARDLINK))
 		{
@@ -442,73 +158,118 @@ static void calculate_bytes_to_extract(struct list_head *stream_list,
 	progress->extract.completed_bytes = 0;
 }
 
-static void maybe_add_stream_for_extraction(struct lookup_table_entry *lte,
-					    struct list_head *stream_list)
+static void
+maybe_add_stream_for_extraction(struct wim_lookup_table_entry *lte,
+				struct list_head *stream_list)
 {
 	if (++lte->out_refcnt == 1) {
-		INIT_LIST_HEAD(&lte->inode_list);
-		list_add_tail(&lte->staging_list, stream_list);
+		INIT_LIST_HEAD(&lte->lte_dentry_list);
+		list_add_tail(&lte->extraction_list, stream_list);
 	}
 }
 
-static void inode_find_streams_for_extraction(struct inode *inode,
-					      struct list_head *stream_list,
-					      int extract_flags)
+struct find_streams_ctx {
+	struct list_head stream_list;
+	int extract_flags;
+};
+
+static int
+dentry_find_streams_to_extract(struct wim_dentry *dentry, void *_ctx)
 {
-	struct lookup_table_entry *lte;
-	bool inode_added = false;
+	struct find_streams_ctx *ctx = _ctx;
+	struct wim_inode *inode = dentry->d_inode;
+	struct wim_lookup_table_entry *lte;
+	bool dentry_added = false;
+	struct list_head *stream_list = &ctx->stream_list;
+	int extract_flags = ctx->extract_flags;
 
-	lte = inode_unnamed_lte_resolved(inode);
+	dentry->needs_extraction = 1;
 
+	lte = inode_unnamed_lte_resolved(inode);
 	if (lte) {
-		maybe_add_stream_for_extraction(lte, stream_list);
-		list_add_tail(&inode->lte_inode_list, &lte->inode_list);
-		inode_added = true;
+		if (!inode->i_visited)
+			maybe_add_stream_for_extraction(lte, stream_list);
+		list_add_tail(&dentry->tmp_list, &lte->lte_dentry_list);
+		dentry_added = true;
 	}
-#ifdef WITH_NTFS_3G
-	if (extract_flags & WIMLIB_EXTRACT_FLAG_NTFS) {
-		for (unsigned i = 0; i < inode->num_ads; i++) {
-			if (inode->ads_entries[i].stream_name_len != 0) {
-				lte = inode->ads_entries[i].lte;
+
+	/* Determine whether to include alternate data stream entries or not.
+	 *
+	 * UNIX:  Include them if extracting using NTFS-3g.
+	 *
+	 * Windows: Include them undconditionally, although if the filesystem is
+	 * not NTFS we won't actually be able to extract them. */
+#if defined(WITH_NTFS_3G)
+	if (extract_flags & WIMLIB_EXTRACT_FLAG_NTFS)
+#elif defined(__WIN32__)
+	if (1)
+#else
+	if (0)
+#endif
+	{
+		for (unsigned i = 0; i < inode->i_num_ads; i++) {
+			if (inode->i_ads_entries[i].stream_name_nbytes != 0) {
+				lte = inode->i_ads_entries[i].lte;
 				if (lte) {
-					maybe_add_stream_for_extraction(lte,
-									stream_list);
-					if (!inode_added) {
-						list_add_tail(&inode->lte_inode_list,
-							      &lte->inode_list);
-						inode_added = true;
+					if (!inode->i_visited) {
+						maybe_add_stream_for_extraction(lte,
+										stream_list);
+					}
+					if (!dentry_added) {
+						list_add_tail(&dentry->tmp_list,
+							      &lte->lte_dentry_list);
+						dentry_added = true;
 					}
 				}
 			}
 		}
 	}
-#endif
+	inode->i_visited = 1;
+	return 0;
 }
 
-static void find_streams_for_extraction(struct hlist_head *inode_list,
-					struct list_head *stream_list,
-					struct lookup_table *lookup_table,
-					int extract_flags)
+static int
+dentry_resolve_and_zero_lte_refcnt(struct wim_dentry *dentry, void *_lookup_table)
 {
-	struct inode *inode;
-	struct hlist_node *cur;
-	struct dentry *dentry;
+	struct wim_inode *inode = dentry->d_inode;
+	struct wim_lookup_table *lookup_table = _lookup_table;
+	struct wim_lookup_table_entry *lte;
 
-	for_lookup_table_entry(lookup_table, lte_zero_out_refcnt, NULL);
-	INIT_LIST_HEAD(stream_list);
-	hlist_for_each_entry(inode, cur, inode_list, hlist) {
-		if (!inode->resolved)
-			inode_resolve_ltes(inode, lookup_table);
-		inode_for_each_dentry(dentry, inode)
-			dentry->is_extracted = 0;
-		inode_find_streams_for_extraction(inode, stream_list,
-						  extract_flags);
+	inode_resolve_ltes(inode, lookup_table);
+	for (unsigned i = 0; i <= inode->i_num_ads; i++) {
+		lte = inode_stream_lte_resolved(inode, i);
+		if (lte)
+			lte->out_refcnt = 0;
 	}
+	return 0;
+}
+
+static void
+find_streams_for_extraction(struct wim_dentry *root,
+			    struct list_head *stream_list,
+			    struct wim_lookup_table *lookup_table,
+			    int extract_flags)
+{
+	struct find_streams_ctx ctx;
+
+	INIT_LIST_HEAD(&ctx.stream_list);
+	ctx.extract_flags = extract_flags;
+	for_dentry_in_tree(root, dentry_resolve_and_zero_lte_refcnt, lookup_table);
+	for_dentry_in_tree(root, dentry_find_streams_to_extract, &ctx);
+	list_transfer(&ctx.stream_list, stream_list);
+}
+
+static int
+dentry_reset_needs_extraction(struct wim_dentry *dentry, void *_ignore)
+{
+	dentry->needs_extraction = 0;
+	dentry->d_inode->i_visited = 0;
+	return 0;
 }
 
 struct apply_operations {
-	int (*apply_dentry)(struct dentry *dentry, void *arg);
-	int (*apply_dentry_timestamps)(struct dentry *dentry, void *arg);
+	int (*apply_dentry)(struct wim_dentry *dentry, void *arg);
+	int (*apply_dentry_timestamps)(struct wim_dentry *dentry, void *arg);
 };
 
 static const struct apply_operations normal_apply_operations = {
@@ -523,16 +284,16 @@ static const struct apply_operations ntfs_apply_operations = {
 };
 #endif
 
-static int apply_stream_list(struct list_head *stream_list,
-			     struct apply_args *args,
-			     const struct apply_operations *ops,
-			     wimlib_progress_func_t progress_func)
+static int
+apply_stream_list(struct list_head *stream_list,
+		  struct apply_args *args,
+		  const struct apply_operations *ops,
+		  wimlib_progress_func_t progress_func)
 {
 	uint64_t bytes_per_progress = args->progress.extract.total_bytes / 100;
 	uint64_t next_progress = bytes_per_progress;
-	struct lookup_table_entry *lte;
-	struct inode *inode;
-	struct dentry *dentry;
+	struct wim_lookup_table_entry *lte;
+	struct wim_dentry *dentry;
 	int ret;
 
 	/* This complicated loop is essentially looping through the dentries,
@@ -544,31 +305,29 @@ static int apply_stream_list(struct list_head *stream_list,
 	 * sequential reading of the WIM can be implemented. */
 
 	/* For each distinct stream to be extracted */
-	list_for_each_entry(lte, stream_list, staging_list) {
-		/* For each inode that contains the stream */
-		list_for_each_entry(inode, &lte->inode_list, lte_inode_list) {
-			/* For each dentry that points to the inode */
-			inode_for_each_dentry(dentry, inode) {
-				/* Extract the dentry if it was not already
-				 * extracted */
-				ret = maybe_apply_dentry(dentry, args);
-				if (ret != 0)
-					return ret;
-				if (progress_func &&
-				    args->progress.extract.completed_bytes >= next_progress)
+	list_for_each_entry(lte, stream_list, extraction_list) {
+		/* For each dentry to be extracted that is a name for an inode
+		 * containing the stream */
+		list_for_each_entry(dentry, &lte->lte_dentry_list, tmp_list) {
+			/* Extract the dentry if it was not already
+			 * extracted */
+			ret = maybe_apply_dentry(dentry, args);
+			if (ret)
+				return ret;
+			if (progress_func &&
+			    args->progress.extract.completed_bytes >= next_progress)
+			{
+				progress_func(WIMLIB_PROGRESS_MSG_EXTRACT_STREAMS,
+					      &args->progress);
+				if (args->progress.extract.completed_bytes >=
+				    args->progress.extract.total_bytes)
 				{
-					progress_func(WIMLIB_PROGRESS_MSG_EXTRACT_STREAMS,
-						      &args->progress);
-					if (args->progress.extract.completed_bytes >=
-					    args->progress.extract.total_bytes)
-					{
-						next_progress = ~0ULL;
-					} else {
-						next_progress =
-							min (args->progress.extract.completed_bytes +
-							     bytes_per_progress,
-							     args->progress.extract.total_bytes);
-					}
+					next_progress = ~0ULL;
+				} else {
+					next_progress =
+						min (args->progress.extract.completed_bytes +
+						     bytes_per_progress,
+						     args->progress.extract.total_bytes);
 				}
 			}
 		}
@@ -576,31 +335,129 @@ static int apply_stream_list(struct list_head *stream_list,
 	return 0;
 }
 
-static int extract_single_image(WIMStruct *w, int image,
-				const char *target, int extract_flags,
-				wimlib_progress_func_t progress_func)
+static int
+sort_stream_list_by_wim_position(struct list_head *stream_list)
+{
+	struct list_head *cur;
+	size_t num_streams;
+	struct wim_lookup_table_entry **array;
+	size_t i;
+	size_t array_size;
+
+	num_streams = 0;
+	list_for_each(cur, stream_list)
+		num_streams++;
+	array_size = num_streams * sizeof(array[0]);
+	array = MALLOC(array_size);
+	if (!array) {
+		ERROR("Failed to allocate %zu bytes to sort stream entries",
+		      array_size);
+		return WIMLIB_ERR_NOMEM;
+	}
+	cur = stream_list->next;
+	for (i = 0; i < num_streams; i++) {
+		array[i] = container_of(cur, struct wim_lookup_table_entry, extraction_list);
+		cur = cur->next;
+	}
+
+	qsort(array, num_streams, sizeof(array[0]), cmp_streams_by_wim_position);
+
+	INIT_LIST_HEAD(stream_list);
+	for (i = 0; i < num_streams; i++)
+		list_add_tail(&array[i]->extraction_list, stream_list);
+	FREE(array);
+	return 0;
+}
+
+/*
+ * Extract a dentry to standard output.
+ *
+ * This obviously doesn't make sense in all cases.  We return an error if the
+ * dentry does not correspond to a regular file.  Otherwise we extract the
+ * unnamed data stream only.
+ */
+static int
+extract_dentry_to_stdout(struct wim_dentry *dentry)
+{
+	int ret = 0;
+	if (!dentry_is_regular_file(dentry)) {
+		ERROR("\"%"TS"\" is not a regular file and therefore cannot be "
+		      "extracted to standard output", dentry->_full_path);
+		ret = WIMLIB_ERR_NOT_A_REGULAR_FILE;
+	} else {
+		struct wim_lookup_table_entry *lte;
+
+		lte = inode_unnamed_lte_resolved(dentry->d_inode);
+		if (lte) {
+			ret = extract_wim_resource_to_fd(lte, STDOUT_FILENO,
+							 wim_resource_size(lte));
+		}
+	}
+	return ret;
+}
+
+/*
+ * extract_tree - Extract a file or directory tree from the currently selected
+ *		  WIM image.
+ *
+ * @wim:	WIMStruct for the WIM file, with the desired image selected
+ *		(as wim->current_image).
+ * @wim_source_path:
+ *		"Canonical" (i.e. no leading or trailing slashes, path
+ *		separators forwald slashes) path inside the WIM image to
+ *		extract.  An empty string means the full image.
+ * @target:
+ *		Filesystem path to extract the file or directory tree to.
+ *
+ * @extract_flags:
+ *		WIMLIB_EXTRACT_FLAG_*.  Also, the private flag
+ *		WIMLIB_EXTRACT_FLAG_MULTI_IMAGE will be set if this is being
+ *		called through wimlib_extract_image() with WIMLIB_ALL_IMAGES as
+ *		the image.
+ *
+ * @progress_func:
+ *		If non-NULL, progress function for the extraction.  The messages
+ *		we may in this function are:
+ *
+ *		WIMLIB_PROGRESS_MSG_EXTRACT_TREE_BEGIN or
+ *			WIMLIB_PROGRESS_MSG_EXTRACT_IMAGE_BEGIN;
+ *		WIMLIB_PROGRESS_MSG_EXTRACT_DIR_STRUCTURE_BEGIN;
+ *		WIMLIB_PROGRESS_MSG_EXTRACT_DIR_STRUCTURE_END;
+ *		WIMLIB_PROGRESS_MSG_EXTRACT_DENTRY;
+ *		WIMLIB_PROGRESS_MSG_EXTRACT_STREAMS;
+ *		WIMLIB_PROGRESS_MSG_APPLY_TIMESTAMPS;
+ *		WIMLIB_PROGRESS_MSG_EXTRACT_TREE_END or
+ *			WIMLIB_PROGRESS_MSG_EXTRACT_IMAGE_END.
+ *
+ * Returns 0 on success; nonzero on failure.
+ */
+static int
+extract_tree(WIMStruct *wim, const tchar *wim_source_path, const tchar *target,
+	     int extract_flags, wimlib_progress_func_t progress_func)
 {
 	int ret;
 	struct list_head stream_list;
-	struct hlist_head *inode_list;
-
 	struct apply_args args;
 	const struct apply_operations *ops;
+	struct wim_dentry *root;
+
+	memset(&args, 0, sizeof(args));
 
-	args.w                    = w;
-	args.target               = target;
-	args.extract_flags        = extract_flags;
-	args.num_lutimes_warnings = 0;
-	args.target               = target;
-	args.stream_list          = &stream_list;
-	args.progress_func	  = progress_func;
+	args.w                      = wim;
+	args.target                 = target;
+	args.extract_flags          = extract_flags;
+	args.progress_func          = progress_func;
+	args.target_nchars          = tstrlen(target);
+	args.wim_source_path_nchars = tstrlen(wim_source_path);
 
 	if (progress_func) {
-		args.progress.extract.wimfile_name = w->filename;
-		args.progress.extract.image = image;
+		args.progress.extract.wimfile_name = wim->filename;
+		args.progress.extract.image = wim->current_image;
 		args.progress.extract.extract_flags = (extract_flags &
 						       WIMLIB_EXTRACT_MASK_PUBLIC);
-		args.progress.extract.image_name = wimlib_get_image_name(w, image);
+		args.progress.extract.image_name = wimlib_get_image_name(wim,
+									 wim->current_image);
+		args.progress.extract.extract_root_wim_source_path = wim_source_path;
 		args.progress.extract.target = target;
 	}
 
@@ -608,31 +465,52 @@ static int extract_single_image(WIMStruct *w, int image,
 	if (extract_flags & WIMLIB_EXTRACT_FLAG_NTFS) {
 		args.vol = ntfs_mount(target, 0);
 		if (!args.vol) {
-			ERROR_WITH_ERRNO("Failed to mount NTFS volume `%s'", target);
-			return WIMLIB_ERR_NTFS_3G;
+			ERROR_WITH_ERRNO("Failed to mount NTFS volume `%"TS"'",
+					 target);
+			ret = WIMLIB_ERR_NTFS_3G;
+			goto out;
 		}
 		ops = &ntfs_apply_operations;
 	} else
 #endif
 		ops = &normal_apply_operations;
 
-	ret = select_wim_image(w, image);
-	if (ret != 0)
-		goto out;
+	root = get_dentry(wim, wim_source_path);
+	if (!root) {
+		ERROR("Path \"%"TS"\" does not exist in WIM image %d",
+		      wim_source_path, wim->current_image);
+		ret = WIMLIB_ERR_PATH_DOES_NOT_EXIST;
+		goto out_ntfs_umount;
+	}
+	args.extract_root = root;
 
-	inode_list = &w->image_metadata[image - 1].inode_list;
+	ret = calculate_dentry_tree_full_paths(root);
+	if (ret)
+		goto out_ntfs_umount;
 
-	find_streams_for_extraction(inode_list, &stream_list,
-				    w->lookup_table, extract_flags);
+	/* Build a list of the streams that need to be extracted */
+	find_streams_for_extraction(root,
+				    &stream_list,
+				    wim->lookup_table, extract_flags);
 
+	/* Calculate the number of bytes of data that will be extracted */
 	calculate_bytes_to_extract(&stream_list, extract_flags,
 				   &args.progress);
 
+	if (extract_flags & WIMLIB_EXTRACT_FLAG_TO_STDOUT) {
+		ret = extract_dentry_to_stdout(root);
+		goto out_dentry_reset_needs_extraction;
+	}
+
 	if (progress_func) {
-		progress_func(WIMLIB_PROGRESS_MSG_EXTRACT_IMAGE_BEGIN,
+		progress_func(*wim_source_path ? WIMLIB_PROGRESS_MSG_EXTRACT_TREE_BEGIN :
+			      WIMLIB_PROGRESS_MSG_EXTRACT_IMAGE_BEGIN,
 			      &args.progress);
 	}
 
+	/* If a sequential extraction was specified, sort the streams to be
+	 * extracted by their position in the WIM file, so that the WIM file can
+	 * be read sequentially. */
 	if (extract_flags & WIMLIB_EXTRACT_FLAG_SEQUENTIAL) {
 		ret = sort_stream_list_by_wim_position(&stream_list);
 		if (ret != 0) {
@@ -646,158 +524,456 @@ static int extract_single_image(WIMStruct *w, int image,
 			      &args.progress);
 	}
 
+	/* Make the directory structure and extract empty files */
 	args.extract_flags |= WIMLIB_EXTRACT_FLAG_NO_STREAMS;
 	args.apply_dentry = ops->apply_dentry;
-	ret = for_dentry_in_tree(wim_root_dentry(w), maybe_apply_dentry, &args);
+	ret = for_dentry_in_tree(root, maybe_apply_dentry, &args);
 	args.extract_flags &= ~WIMLIB_EXTRACT_FLAG_NO_STREAMS;
-	if (ret != 0)
-		goto out;
+	if (ret)
+		goto out_dentry_reset_needs_extraction;
 
 	if (progress_func) {
 		progress_func(WIMLIB_PROGRESS_MSG_EXTRACT_DIR_STRUCTURE_END,
 			      &args.progress);
 	}
 
+	if (extract_flags & WIMLIB_EXTRACT_FLAG_RPFIX) {
+		args.target_realpath = realpath(target, NULL);
+		if (!args.target_realpath) {
+			ret = WIMLIB_ERR_NOMEM;
+			goto out_dentry_reset_needs_extraction;
+		}
+		args.target_realpath_len = tstrlen(args.target_realpath);
+	}
+
+	/* Extract non-empty files */
 	ret = apply_stream_list(&stream_list, &args, ops, progress_func);
-	if (ret != 0)
-		goto out;
+	if (ret)
+		goto out_free_target_realpath;
 
 	if (progress_func) {
 		progress_func(WIMLIB_PROGRESS_MSG_APPLY_TIMESTAMPS,
 			      &args.progress);
 	}
 
-	ret = for_dentry_in_tree_depth(wim_root_dentry(w),
+	/* Apply timestamps */
+	ret = for_dentry_in_tree_depth(root,
 				       ops->apply_dentry_timestamps, &args);
-	if (ret != 0)
-		goto out;
+	if (ret)
+		goto out_free_target_realpath;
 
 	if (progress_func) {
-		progress_func(WIMLIB_PROGRESS_MSG_EXTRACT_IMAGE_END,
+		progress_func(*wim_source_path ? WIMLIB_PROGRESS_MSG_EXTRACT_TREE_END :
+			      WIMLIB_PROGRESS_MSG_EXTRACT_IMAGE_END,
 			      &args.progress);
 	}
-out:
+out_free_target_realpath:
+	FREE(args.target_realpath);
+out_dentry_reset_needs_extraction:
+	for_dentry_in_tree(root, dentry_reset_needs_extraction, NULL);
+out_ntfs_umount:
 #ifdef WITH_NTFS_3G
+	/* Unmount the NTFS volume */
 	if (extract_flags & WIMLIB_EXTRACT_FLAG_NTFS) {
 		if (ntfs_umount(args.vol, FALSE) != 0) {
-			ERROR_WITH_ERRNO("Failed to unmount NTFS volume `%s'", args.target);
+			ERROR_WITH_ERRNO("Failed to unmount NTFS volume `%"TS"'",
+					 args.target);
 			if (ret == 0)
 				ret = WIMLIB_ERR_NTFS_3G;
 		}
 	}
 #endif
+out:
 	return ret;
 }
 
+/* Validates a single wimlib_extract_command, mostly checking to make sure the
+ * extract flags make sense. */
+static int
+check_extract_command(struct wimlib_extract_command *cmd, int wim_header_flags)
+{
+	int extract_flags;
+	bool is_entire_image = (cmd->wim_source_path[0] == T('\0'));
+
+	/* Empty destination path? */
+	if (cmd->fs_dest_path[0] == T('\0'))
+		return WIMLIB_ERR_INVALID_PARAM;
+
+	extract_flags = cmd->extract_flags;
+
+	/* Specified both symlink and hardlink modes? */
+	if ((extract_flags &
+	     (WIMLIB_EXTRACT_FLAG_SYMLINK |
+	      WIMLIB_EXTRACT_FLAG_HARDLINK)) == (WIMLIB_EXTRACT_FLAG_SYMLINK |
+						 WIMLIB_EXTRACT_FLAG_HARDLINK))
+		return WIMLIB_ERR_INVALID_PARAM;
+
+#ifdef __WIN32__
+	/* Wanted UNIX data on Windows? */
+	if (extract_flags & WIMLIB_EXTRACT_FLAG_UNIX_DATA) {
+		ERROR("Extracting UNIX data is not supported on Windows");
+		return WIMLIB_ERR_INVALID_PARAM;
+	}
+	/* Wanted linked extraction on Windows?  (XXX This is possible, just not
+	 * implemented yet.) */
+	if (extract_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK |
+			     WIMLIB_EXTRACT_FLAG_HARDLINK))
+	{
+		ERROR("Linked extraction modes are not supported on Windows");
+		return WIMLIB_ERR_INVALID_PARAM;
+	}
+#endif
+
+	if (extract_flags & WIMLIB_EXTRACT_FLAG_NTFS) {
+		/* NTFS-3g extraction mode requested */
+#ifdef WITH_NTFS_3G
+		if ((extract_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK |
+				      WIMLIB_EXTRACT_FLAG_HARDLINK))) {
+			ERROR("Cannot specify symlink or hardlink flags when applying\n"
+			      "        directly to a NTFS volume");
+			return WIMLIB_ERR_INVALID_PARAM;
+		}
+		if (!is_entire_image &&
+		    (extract_flags & WIMLIB_EXTRACT_FLAG_NTFS))
+		{
+			ERROR("When applying directly to a NTFS volume you can "
+			      "only extract a full image, not part of one");
+			return WIMLIB_ERR_INVALID_PARAM;
+		}
+		if (extract_flags & WIMLIB_EXTRACT_FLAG_UNIX_DATA) {
+			ERROR("Cannot restore UNIX-specific data in "
+			      "the NTFS extraction mode");
+			return WIMLIB_ERR_INVALID_PARAM;
+		}
+#else
+		ERROR("wimlib was compiled without support for NTFS-3g, so");
+		ERROR("we cannot apply a WIM image directly to a NTFS volume");
+		return WIMLIB_ERR_UNSUPPORTED;
+#endif
+	}
+
+	if ((extract_flags & (WIMLIB_EXTRACT_FLAG_RPFIX |
+			      WIMLIB_EXTRACT_FLAG_NORPFIX)) ==
+		(WIMLIB_EXTRACT_FLAG_RPFIX | WIMLIB_EXTRACT_FLAG_NORPFIX))
+	{
+		ERROR("Cannot specify RPFIX and NORPFIX flags at the same time!");
+		return WIMLIB_ERR_INVALID_PARAM;
+	}
+
+	if ((extract_flags & (WIMLIB_EXTRACT_FLAG_RPFIX |
+			      WIMLIB_EXTRACT_FLAG_NORPFIX)) == 0)
+	{
+		/* Do reparse point fixups by default if the WIM header says
+		 * they are enabled and we are extracting a full image. */
+		if ((wim_header_flags & WIM_HDR_FLAG_RP_FIX) && is_entire_image)
+			extract_flags |= WIMLIB_EXTRACT_FLAG_RPFIX;
+	}
+
+	if (!is_entire_image && (extract_flags & WIMLIB_EXTRACT_FLAG_RPFIX)) {
+		ERROR("Cannot specify --rpfix when not extracting entire image");
+		return WIMLIB_ERR_INVALID_PARAM;
+	}
 
-/* Extracts all images from the WIM to @output_dir, with the images placed in
- * subdirectories named by their image names. */
-static int extract_all_images(WIMStruct *w, const char *target,
-			      int extract_flags,
-			      wimlib_progress_func_t progress_func)
+	cmd->extract_flags = extract_flags;
+	return 0;
+}
+
+
+/* Internal function to execute extraction commands for a WIM image. */
+static int
+do_wimlib_extract_files(WIMStruct *wim,
+			int image,
+			struct wimlib_extract_command *cmds,
+			size_t num_cmds,
+			wimlib_progress_func_t progress_func)
 {
-	size_t image_name_max_len = max(xml_get_max_image_name_len(w), 20);
-	size_t output_path_len = strlen(target);
-	char buf[output_path_len + 1 + image_name_max_len + 1];
 	int ret;
-	int image;
-	const char *image_name;
+	bool found_link_cmd = false;
+	bool found_nolink_cmd = false;
 
-	ret = extract_directory(target, true);
-	if (ret != 0)
+	/* Select the image from which we are extracting files */
+	ret = select_wim_image(wim, image);
+	if (ret)
 		return ret;
 
-	memcpy(buf, target, output_path_len);
-	buf[output_path_len] = '/';
-	for (image = 1; image <= w->hdr.image_count; image++) {
-		image_name = wimlib_get_image_name(w, image);
-		if (image_name && *image_name) {
-			strcpy(buf + output_path_len + 1, image_name);
+	/* Make sure there are no streams in the WIM that have not been
+	 * checksummed yet. */
+	ret = wim_checksum_unhashed_streams(wim);
+	if (ret)
+		return ret;
+
+	/* Check for problems with the extraction commands */
+	for (size_t i = 0; i < num_cmds; i++) {
+		ret = check_extract_command(&cmds[i], wim->hdr.flags);
+		if (ret)
+			return ret;
+		if (cmds[i].extract_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK |
+					     WIMLIB_EXTRACT_FLAG_HARDLINK)) {
+			found_link_cmd = true;
 		} else {
-			/* Image name is empty. Use image number instead */
-			sprintf(buf + output_path_len + 1, "%d", image);
+			found_nolink_cmd = true;
 		}
-		ret = extract_single_image(w, image, buf, extract_flags,
-					   progress_func);
-		if (ret != 0)
+		if (found_link_cmd && found_nolink_cmd) {
+			ERROR("Symlink or hardlink extraction mode must "
+			      "be set on all extraction commands");
+			return WIMLIB_ERR_INVALID_PARAM;
+		}
+	}
+
+	/* Execute the extraction commands */
+	for (size_t i = 0; i < num_cmds; i++) {
+		ret = extract_tree(wim,
+				   cmds[i].wim_source_path,
+				   cmds[i].fs_dest_path,
+				   cmds[i].extract_flags,
+				   progress_func);
+		if (ret)
 			return ret;
 	}
 	return 0;
 }
 
-/* Extracts a single image or all images from a WIM file. */
-WIMLIBAPI int wimlib_extract_image(WIMStruct *w, int image,
-				   const char *target,
-				   int extract_flags,
-				   WIMStruct **additional_swms,
-				   unsigned num_additional_swms,
-				   wimlib_progress_func_t progress_func)
+/* Extract files or directories from a WIM image. */
+WIMLIBAPI int
+wimlib_extract_files(WIMStruct *wim,
+		     int image,
+		     const struct wimlib_extract_command *cmds,
+		     size_t num_cmds,
+		     int default_extract_flags,
+		     WIMStruct **additional_swms,
+		     unsigned num_additional_swms,
+		     wimlib_progress_func_t progress_func)
 {
-	struct lookup_table *joined_tab, *w_tab_save;
 	int ret;
+	struct wimlib_extract_command *cmds_copy;
+	struct wim_lookup_table *wim_tab_save, *joined_tab;
+	int all_flags = 0;
 
-	if (!target)
-		return WIMLIB_ERR_INVALID_PARAM;
+	default_extract_flags &= WIMLIB_EXTRACT_MASK_PUBLIC;
 
-	extract_flags &= WIMLIB_EXTRACT_MASK_PUBLIC;
+	ret = verify_swm_set(wim, additional_swms, num_additional_swms);
+	if (ret)
+		goto out;
 
-	if ((extract_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK | WIMLIB_EXTRACT_FLAG_HARDLINK))
-			== (WIMLIB_EXTRACT_FLAG_SYMLINK | WIMLIB_EXTRACT_FLAG_HARDLINK))
-		return WIMLIB_ERR_INVALID_PARAM;
+	if (num_cmds == 0)
+		goto out;
 
-	if (extract_flags & WIMLIB_EXTRACT_FLAG_NTFS) {
-#ifdef WITH_NTFS_3G
-		if ((extract_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK | WIMLIB_EXTRACT_FLAG_HARDLINK))) {
-			ERROR("Cannot specify symlink or hardlink flags when applying\n"
-			      "        directly to a NTFS volume");
-			return WIMLIB_ERR_INVALID_PARAM;
+	if (num_additional_swms) {
+		ret = new_joined_lookup_table(wim, additional_swms,
+					      num_additional_swms,
+					      &joined_tab);
+		if (ret)
+			goto out;
+		wim_tab_save = wim->lookup_table;
+		wim->lookup_table = joined_tab;
+	}
+
+	cmds_copy = CALLOC(num_cmds, sizeof(cmds[0]));
+	if (!cmds_copy) {
+		ret = WIMLIB_ERR_NOMEM;
+		goto out_restore_lookup_table;
+	}
+
+	for (size_t i = 0; i < num_cmds; i++) {
+		cmds_copy[i].extract_flags = (default_extract_flags |
+						 cmds[i].extract_flags)
+						& WIMLIB_EXTRACT_MASK_PUBLIC;
+		all_flags |= cmds_copy[i].extract_flags;
+
+		cmds_copy[i].wim_source_path = canonicalize_wim_path(cmds[i].wim_source_path);
+		if (!cmds_copy[i].wim_source_path) {
+			ret = WIMLIB_ERR_NOMEM;
+			goto out_free_cmds_copy;
 		}
-		if (image == WIMLIB_ALL_IMAGES) {
-			ERROR("Can only apply a single image when applying "
-			      "directly to a NTFS volume");
-			return WIMLIB_ERR_INVALID_PARAM;
+
+		cmds_copy[i].fs_dest_path = canonicalize_fs_path(cmds[i].fs_dest_path);
+		if (!cmds_copy[i].fs_dest_path) {
+			ret = WIMLIB_ERR_NOMEM;
+			goto out_free_cmds_copy;
 		}
+
+	}
+	ret = do_wimlib_extract_files(wim, image,
+				      cmds_copy, num_cmds,
+				      progress_func);
+
+	if (all_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK |
+			 WIMLIB_EXTRACT_FLAG_HARDLINK))
+	{
+		for_lookup_table_entry(wim->lookup_table,
+				       lte_free_extracted_file, NULL);
+	}
+out_free_cmds_copy:
+	for (size_t i = 0; i < num_cmds; i++) {
+		FREE(cmds_copy[i].wim_source_path);
+		FREE(cmds_copy[i].fs_dest_path);
+	}
+	FREE(cmds_copy);
+out_restore_lookup_table:
+	if (num_additional_swms) {
+		free_lookup_table(wim->lookup_table);
+		wim->lookup_table = wim_tab_save;
+	}
+out:
+	return ret;
+}
+
+/*
+ * Extracts an image from a WIM file.
+ *
+ * @wim:		WIMStruct for the WIM file.
+ *
+ * @image:		Number of the single image to extract.
+ *
+ * @target:		Directory or NTFS volume to extract the image to.
+ *
+ * @extract_flags:	Bitwise or of WIMLIB_EXTRACT_FLAG_*.
+ *
+ * @progress_func:	If non-NULL, a progress function to be called
+ *			periodically.
+ *
+ * Returns 0 on success; nonzero on failure.
+ */
+static int
+extract_single_image(WIMStruct *wim, int image,
+		     const tchar *target, int extract_flags,
+		     wimlib_progress_func_t progress_func)
+{
+	int ret;
+	tchar *target_copy = canonicalize_fs_path(target);
+	if (!target_copy)
+		return WIMLIB_ERR_NOMEM;
+	struct wimlib_extract_command cmd = {
+		.wim_source_path = T(""),
+		.fs_dest_path = target_copy,
+		.extract_flags = extract_flags,
+	};
+	ret = do_wimlib_extract_files(wim, image, &cmd, 1, progress_func);
+	FREE(target_copy);
+	return ret;
+}
+
+static const tchar * const filename_forbidden_chars =
+T(
+#ifdef __WIN32__
+"<>:\"/\\|?*"
 #else
-		ERROR("wimlib was compiled without support for NTFS-3g, so");
-		ERROR("we cannot apply a WIM image directly to a NTFS volume");
-		return WIMLIB_ERR_UNSUPPORTED;
+"/"
 #endif
+);
+
+/* This function checks if it is okay to use a WIM image's name as a directory
+ * name.  */
+static bool
+image_name_ok_as_dir(const tchar *image_name)
+{
+	return image_name && *image_name &&
+		!tstrpbrk(image_name, filename_forbidden_chars) &&
+		tstrcmp(image_name, T(".")) &&
+		tstrcmp(image_name, T(".."));
+}
+
+/* Extracts all images from the WIM to the directory @target, with the images
+ * placed in subdirectories named by their image names. */
+static int
+extract_all_images(WIMStruct *wim,
+		   const tchar *target,
+		   int extract_flags,
+		   wimlib_progress_func_t progress_func)
+{
+	size_t image_name_max_len = max(xml_get_max_image_name_len(wim), 20);
+	size_t output_path_len = tstrlen(target);
+	tchar buf[output_path_len + 1 + image_name_max_len + 1];
+	int ret;
+	int image;
+	const tchar *image_name;
+	struct stat stbuf;
+
+	if (tstat(target, &stbuf)) {
+		if (errno == ENOENT)
+		{
+			if (tmkdir(target, S_IRWXU | S_IRGRP | S_IXGRP |
+				   	   S_IROTH | S_IXOTH))
+			{
+				ERROR_WITH_ERRNO("Failed to create directory \"%"TS"\"", target);
+				return WIMLIB_ERR_MKDIR;
+			}
+		} else {
+			ERROR_WITH_ERRNO("Failed to stat \"%"TS"\"", target);
+			return WIMLIB_ERR_STAT;
+		}
+	} else if (!S_ISDIR(stbuf.st_mode)) {
+		ERROR("\"%"TS"\" is not a directory", target);
+		return WIMLIB_ERR_NOTDIR;
+	}
+
+	tmemcpy(buf, target, output_path_len);
+	buf[output_path_len] = T('/');
+	for (image = 1; image <= wim->hdr.image_count; image++) {
+		image_name = wimlib_get_image_name(wim, image);
+		if (image_name_ok_as_dir(image_name)) {
+			tstrcpy(buf + output_path_len + 1, image_name);
+		} else {
+			/* Image name is empty or contains forbidden characters.
+			 * Use image number instead. */
+			tsprintf(buf + output_path_len + 1, T("%d"), image);
+		}
+		ret = extract_single_image(wim, image, buf, extract_flags,
+					   progress_func);
+		if (ret)
+			return ret;
 	}
+	return 0;
+}
+
+/* Extracts a single image or all images from a WIM file to a directory or NTFS
+ * volume. */
+WIMLIBAPI int
+wimlib_extract_image(WIMStruct *wim,
+		     int image,
+		     const tchar *target,
+		     int extract_flags,
+		     WIMStruct **additional_swms,
+		     unsigned num_additional_swms,
+		     wimlib_progress_func_t progress_func)
+{
+	struct wim_lookup_table *joined_tab, *wim_tab_save;
+	int ret;
 
-	ret = verify_swm_set(w, additional_swms, num_additional_swms);
-	if (ret != 0)
+	extract_flags &= WIMLIB_EXTRACT_MASK_PUBLIC;
+
+	ret = verify_swm_set(wim, additional_swms, num_additional_swms);
+	if (ret)
 		return ret;
 
 	if (num_additional_swms) {
-		ret = new_joined_lookup_table(w, additional_swms,
+		ret = new_joined_lookup_table(wim, additional_swms,
 					      num_additional_swms, &joined_tab);
-		if (ret != 0)
+		if (ret)
 			return ret;
-		w_tab_save = w->lookup_table;
-		w->lookup_table = joined_tab;
+		wim_tab_save = wim->lookup_table;
+		wim->lookup_table = joined_tab;
 	}
 
 	if (image == WIMLIB_ALL_IMAGES) {
-		extract_flags |= WIMLIB_EXTRACT_FLAG_MULTI_IMAGE;
-		ret = extract_all_images(w, target, extract_flags,
+		ret = extract_all_images(wim, target,
+					 extract_flags | WIMLIB_EXTRACT_FLAG_MULTI_IMAGE,
 					 progress_func);
 	} else {
-		extract_flags &= ~WIMLIB_EXTRACT_FLAG_MULTI_IMAGE;
-		ret = extract_single_image(w, image, target, extract_flags,
+		ret = extract_single_image(wim, image, target, extract_flags,
 					   progress_func);
 	}
 
 	if (extract_flags & (WIMLIB_EXTRACT_FLAG_SYMLINK |
 			     WIMLIB_EXTRACT_FLAG_HARDLINK))
 	{
-		for_lookup_table_entry(w->lookup_table,
+		for_lookup_table_entry(wim->lookup_table,
 				       lte_free_extracted_file,
 				       NULL);
 	}
-
 	if (num_additional_swms) {
-		free_lookup_table(w->lookup_table);
-		w->lookup_table = w_tab_save;
+		free_lookup_table(wim->lookup_table);
+		wim->lookup_table = wim_tab_save;
 	}
 	return ret;
 }