]> wimlib.net Git - wimlib/blobdiff - src/mount.c
Cleanup wimlib_mount() preparations
[wimlib] / src / mount.c
index 1a6f1dcecff8e48ad8c513c64d23fd5a414fdc0d..6096319688e1cd8f03ca5359091864287aecb15b 100644 (file)
@@ -53,8 +53,8 @@
 
 /* File descriptor to a file open on the WIM filesystem. */
 struct wimlib_fd {
-       struct inode *inode;
-       struct lookup_table_entry *lte;
+       struct inode *f_inode;
+       struct lookup_table_entry *f_lte;
        int staging_fd;
        u16 idx;
        u32 stream_id;
@@ -64,7 +64,7 @@ struct wimlib_fd {
 static WIMStruct *w;
 
 /* Working directory when `imagex mount' is run. */
-static const char *working_directory;
+static char *working_directory;
 
 /* Name of the staging directory for a read-write mount.  Whenever a new file is
  * created, it is done so in the staging directory.  Furthermore, whenever a
@@ -77,9 +77,6 @@ static size_t staging_dir_name_len;
 /* Flags passed to wimlib_mount(). */
 static int mount_flags;
 
-/* Name of the directory on which the WIM file is mounted. */
-static const char *mount_dir;
-
 /* Next inode number to be assigned. */
 static u64 next_ino;
 
@@ -100,6 +97,16 @@ static inline int flags_writable(int open_flags)
        return open_flags & (O_RDWR | O_WRONLY);
 }
 
+/*
+ * Allocate a file descriptor for a stream.
+ *
+ * @inode:     inode containing the stream we're opening
+ * @stream_id: ID of the stream we're opening
+ * @lte:       Lookup table entry for the stream (may be NULL)
+ * @fd_ret:    Return the allocated file descriptor if successful.
+ *
+ * Return 0 iff successful or error code if unsuccessful.
+ */
 static int alloc_wimlib_fd(struct inode *inode,
                           u32 stream_id,
                           struct lookup_table_entry *lte,
@@ -118,10 +125,12 @@ static int alloc_wimlib_fd(struct inode *inode,
 
                if (inode->num_allocated_fds == max_fds)
                        return -EMFILE;
-               num_new_fds = min(fds_per_alloc, max_fds - inode->num_allocated_fds);
-               
-               fds = REALLOC(inode->fds, (inode->num_allocated_fds + num_new_fds) *
-                              sizeof(inode->fds[0]));
+               num_new_fds = min(fds_per_alloc,
+                                 max_fds - inode->num_allocated_fds);
+
+               fds = REALLOC(inode->fds,
+                             (inode->num_allocated_fds + num_new_fds) *
+                               sizeof(inode->fds[0]));
                if (!fds)
                        return -ENOMEM;
                memset(&fds[inode->num_allocated_fds], 0,
@@ -134,8 +143,8 @@ static int alloc_wimlib_fd(struct inode *inode,
                        struct wimlib_fd *fd = CALLOC(1, sizeof(*fd));
                        if (!fd)
                                return -ENOMEM;
-                       fd->inode      = inode;
-                       fd->lte        = lte;
+                       fd->f_inode    = inode;
+                       fd->f_lte      = lte;
                        fd->staging_fd = -1;
                        fd->idx        = i;
                        fd->stream_id  = stream_id;
@@ -154,7 +163,7 @@ static void inode_put_fd(struct inode *inode, struct wimlib_fd *fd)
 {
        wimlib_assert(fd);
        wimlib_assert(inode);
-       wimlib_assert(fd->inode == inode);
+       wimlib_assert(fd->f_inode == inode);
        wimlib_assert(inode->num_opened_fds);
        wimlib_assert(fd->idx < inode->num_opened_fds);
        wimlib_assert(inode->fds[fd->idx] == fd);
@@ -168,16 +177,16 @@ static void inode_put_fd(struct inode *inode, struct wimlib_fd *fd)
 static int lte_put_fd(struct lookup_table_entry *lte, struct wimlib_fd *fd)
 {
        wimlib_assert(fd);
-       wimlib_assert(fd->lte == lte);
+       wimlib_assert(fd->f_lte == lte);
 
        if (!lte) /* Empty stream with no lookup table entry */
                return 0;
 
-       wimlib_assert(lte->num_opened_fds);
+       /* Close staging file descriptor if needed. */
 
-       if (lte->resource_location == RESOURCE_IN_STAGING_FILE) {
-               wimlib_assert(lte->staging_file_name);
-               wimlib_assert(fd->staging_fd != -1);
+       if (lte->resource_location == RESOURCE_IN_STAGING_FILE
+            && fd->staging_fd != -1)
+       {
                if (close(fd->staging_fd) != 0) {
                        ERROR_WITH_ERRNO("Failed to close staging file");
                        return -errno;
@@ -187,34 +196,37 @@ static int lte_put_fd(struct lookup_table_entry *lte, struct wimlib_fd *fd)
        return 0;
 }
 
+/* Close a file descriptor. */
 static int close_wimlib_fd(struct wimlib_fd *fd)
 {
        int ret;
+       wimlib_assert(fd);
        DEBUG("Closing fd (inode = %lu, opened = %u, allocated = %u)",
-             fd->inode->ino, fd->inode->num_opened_fds,
-             fd->inode->num_allocated_fds);
-       ret = lte_put_fd(fd->lte, fd);
+             fd->f_inode->ino, fd->f_inode->num_opened_fds,
+             fd->f_inode->num_allocated_fds);
+       ret = lte_put_fd(fd->f_lte, fd);
        if (ret != 0)
                return ret;
 
-       inode_put_fd(fd->inode, fd);
+       inode_put_fd(fd->f_inode, fd);
        return 0;
 }
 
-/* Remove a dentry; i.e. remove a reference to the inode.
+/* Remove a dentry; i.e. remove a reference to the corresponding inode.
  *
- * If there are no remaining references to the inode either through detnries or
- * open file descriptors, the inode is freed.
+ * If there are no remaining references to the inode either through dentries or
+ * open file descriptors, the inode is freed.  Otherwise, the inode is not
+ * removed, but the dentry is unlinked and freed.
  *
- * All lookup table entries referenced by the inode have their reference count
- * decremented.  If a lookup table entry has no open file descriptors and no
- * references remaining, it is freed, and the staging file is unlinked.
- *
- * Otherwise, the inode is not removed, but the dentry is unlinked and freed. */
+ * Either way, all lookup table entries referenced by the inode have their
+ * reference count decremented.  If a lookup table entry has no open file
+ * descriptors and no references remaining, it is freed, and the staging file is
+ * unlinked.
+ */
 static void remove_dentry(struct dentry *dentry,
                          struct lookup_table *lookup_table)
 {
-       struct inode *inode = dentry->inode;
+       struct inode *inode = dentry->d_inode;
        struct lookup_table_entry *lte;
        unsigned i;
 
@@ -227,76 +239,6 @@ static void remove_dentry(struct dentry *dentry,
        put_dentry(dentry);
 }
 
-/* Remove an alternate data stream from the inode  */
-static void inode_remove_ads(struct inode *inode, u16 idx,
-                            struct lookup_table *lookup_table)
-{
-       struct ads_entry *ads_entry;
-       struct lookup_table_entry *lte;
-
-       ads_entry = inode->ads_entries[idx];
-
-       wimlib_assert(ads_entry);
-
-       lte = ads_entry->lte;
-
-       if (lte)
-               lte_decrement_refcnt(lte, lookup_table);
-
-       free_ads_entry(ads_entry);
-
-       wimlib_assert(inode->num_ads);
-       memcpy(&inode->ads_entries[idx],
-              &inode->ads_entries[idx + 1],
-              (inode->num_ads - idx - 1) * sizeof(inode->ads_entries[0]));
-       inode->num_ads--;
-}
-
-/* Transfers file attributes from a struct inode to a `stat' buffer. 
- *
- * The lookup table entry tells us which stream in the inode we are statting.
- * For a named data stream, everything returned is the same as the unnamed data
- * stream except possibly the size and block count. */
-int inode_to_stbuf(const struct inode *inode, struct lookup_table_entry *lte,
-                  struct stat *stbuf)
-{
-       if (inode_is_symlink(inode))
-               stbuf->st_mode = S_IFLNK | 0777;
-       else if (inode_is_directory(inode))
-               stbuf->st_mode = S_IFDIR | 0755;
-       else
-               stbuf->st_mode = S_IFREG | 0644;
-
-       stbuf->st_ino   = (ino_t)inode->ino;
-       stbuf->st_nlink = inode->link_count;
-       stbuf->st_uid   = getuid();
-       stbuf->st_gid   = getgid();
-
-       if (lte) {
-               if (lte->resource_location == RESOURCE_IN_STAGING_FILE) {
-                       wimlib_assert(mount_flags & WIMLIB_MOUNT_FLAG_READWRITE);
-                       wimlib_assert(lte->staging_file_name);
-                       struct stat native_stat;
-                       if (stat(lte->staging_file_name, &native_stat) != 0) {
-                               DEBUG("Failed to stat `%s': %m",
-                                     lte->staging_file_name);
-                               return -errno;
-                       }
-                       stbuf->st_size = native_stat.st_size;
-               } else {
-                       stbuf->st_size = wim_resource_size(lte);
-               }
-       } else {
-               stbuf->st_size = 0;
-       }
-
-       stbuf->st_atime   = wim_timestamp_to_unix(inode->last_access_time);
-       stbuf->st_mtime   = wim_timestamp_to_unix(inode->last_write_time);
-       stbuf->st_ctime   = wim_timestamp_to_unix(inode->creation_time);
-       stbuf->st_blocks  = (stbuf->st_size + 511) / 512;
-       return 0;
-}
-
 /* Creates a new staging file and returns its file descriptor opened for
  * writing.
  *
@@ -341,7 +283,7 @@ static int create_staging_file(char **name_ret, int open_flags)
 
        DEBUG("Creating staging file `%s'", name);
 
-       fd = open(name, open_flags | O_CREAT | O_TRUNC, 0600); 
+       fd = open(name, open_flags | O_CREAT | O_TRUNC, 0600);
        if (fd == -1) {
                errno_save = errno;
                FREE(name);
@@ -352,7 +294,7 @@ static int create_staging_file(char **name_ret, int open_flags)
        return fd;
 }
 
-/* 
+/*
  * Extract a WIM resource to the staging directory.
  *
  * @inode:  Inode that contains the stream we are extracting
@@ -376,6 +318,9 @@ static int extract_resource_to_staging_dir(struct inode *inode,
        int fd;
        struct lookup_table_entry *old_lte, *new_lte;
 
+       DEBUG("Extracting resource to staging dir: inode %"PRIu64", "
+             "stream id %"PRIu32, inode->ino, stream_id);
+
        old_lte = *lte;
        fd = create_staging_file(&staging_file_name, O_WRONLY);
        if (fd == -1)
@@ -429,7 +374,7 @@ static int extract_resource_to_staging_dir(struct inode *inode,
                 * opening it read-write.  Identify those file descriptors and
                 * change their lookup table entry pointers to point to the new
                 * lookup table entry, and open staging file descriptors for
-                * them. 
+                * them.
                 *
                 * At the same time, we need to count the number of these opened
                 * file descriptors to the new lookup table entry.  If there's
@@ -439,8 +384,9 @@ static int extract_resource_to_staging_dir(struct inode *inode,
                        struct wimlib_fd *fd = inode->fds[i];
                        if (fd) {
                                if (fd->stream_id == stream_id) {
-                                       wimlib_assert(fd->lte == old_lte);
-                                       fd->lte = new_lte;
+                                       wimlib_assert(fd->f_lte == old_lte);
+                                       wimlib_assert(fd->staging_fd == -1);
+                                       fd->f_lte = new_lte;
                                        new_lte->num_opened_fds++;
                                        fd->staging_fd = open(staging_file_name, O_RDONLY);
                                        if (fd->staging_fd == -1) {
@@ -461,17 +407,17 @@ static int extract_resource_to_staging_dir(struct inode *inode,
 
        new_lte->resource_entry.original_size = size;
        new_lte->refcnt                       = inode->link_count;
-       new_lte->staging_file_name            = staging_file_name;
        new_lte->resource_location            = RESOURCE_IN_STAGING_FILE;
-       new_lte->inode                        = inode;
+       new_lte->staging_file_name            = staging_file_name;
+       new_lte->lte_inode                    = inode;
        random_hash(new_lte->hash);
 
        if (stream_id == 0)
                inode->lte = new_lte;
        else
                for (u16 i = 0; i < inode->num_ads; i++)
-                       if (inode->ads_entries[i]->stream_id == stream_id)
-                               inode->ads_entries[i]->lte = new_lte;
+                       if (inode->ads_entries[i].stream_id == stream_id)
+                               inode->ads_entries[i].lte = new_lte;
 
        lookup_table_insert(w->lookup_table, new_lte);
        list_add(&new_lte->staging_list, &staging_list);
@@ -480,8 +426,8 @@ static int extract_resource_to_staging_dir(struct inode *inode,
 out_revert_fd_changes:
        for (u16 i = 0, j = 0; j < new_lte->num_opened_fds; i++) {
                struct wimlib_fd *fd = inode->fds[i];
-               if (fd && fd->stream_id == stream_id && fd->lte == new_lte) {
-                       fd->lte = old_lte;
+               if (fd && fd->stream_id == stream_id && fd->f_lte == new_lte) {
+                       fd->f_lte = old_lte;
                        if (fd->staging_fd != -1) {
                                close(fd->staging_fd);
                                fd->staging_fd = -1;
@@ -497,7 +443,7 @@ out_delete_staging_file:
        return ret;
 }
 
-/* 
+/*
  * Creates a randomly named staging directory and returns its name into the
  * static variable staging_dir_name.
  *
@@ -546,13 +492,13 @@ static int remove_file_or_directory(const char *fpath, const struct stat *sb,
 }
 
 
-/* 
- * Deletes the staging directory and all the files contained in it. 
+/*
+ * Deletes the staging directory and all the files contained in it.
  */
 static int delete_staging_dir()
 {
        int ret;
-       
+
        ret = nftw(staging_dir_name, remove_file_or_directory,10, FTW_DEPTH);
        staging_dir_name = NULL;
        return ret;
@@ -570,7 +516,7 @@ static mqd_t unmount_to_daemon_mq;
 static mqd_t daemon_to_unmount_mq;
 
 /* Simple function that returns the concatenation of 4 strings. */
-static char *strcat_dup(const char *s1, const char *s2, const char *s3, 
+static char *strcat_dup(const char *s1, const char *s2, const char *s3,
                        const char *s4)
 {
        size_t len = strlen(s1) + strlen(s2) + strlen(s3) + strlen(s4) + 1;
@@ -605,17 +551,7 @@ static void s_slashes_underscores_g(char *s)
        }
 }
 
-/* 
- * Opens two POSIX message queue: one for sending messages from the unmount
- * process to the daemon process, and one to go the other way.  The names of the
- * message queues, which must be system-wide unique, are be based on the mount
- * point.  (There of course is still a possibility of a collision if one were to
- * unmount two identically named directories simultaneously...)
- *
- * @daemon specifies whether the calling process is the filesystem daemon or the
- * unmount process.
- */
-static int open_message_queues(bool daemon)
+static int set_message_queue_names(const char *mount_dir)
 {
        static const char *slash = "/";
        static const char *prefix = "wimlib-";
@@ -623,8 +559,6 @@ static int open_message_queues(bool daemon)
        static const char *d2u_suffix = "daemon-to-unmount-mq";
 
        const char *mount_dir_basename = path_basename(mount_dir);
-       int flags;
-       int ret;
 
        unmount_to_daemon_mq_name = strcat_dup(slash, mount_dir_basename,
                                                prefix, u2d_suffix);
@@ -636,27 +570,56 @@ static int open_message_queues(bool daemon)
                                                prefix, d2u_suffix);
        if (!daemon_to_unmount_mq_name) {
                ERROR("Out of memory");
-               ret = WIMLIB_ERR_NOMEM;
-               goto err1;
+               FREE(unmount_to_daemon_mq_name);
+               unmount_to_daemon_mq_name = NULL;
+               return WIMLIB_ERR_NOMEM;
        }
 
        remove_trailing_slashes(unmount_to_daemon_mq_name);
        remove_trailing_slashes(daemon_to_unmount_mq_name);
        s_slashes_underscores_g(unmount_to_daemon_mq_name + 1);
        s_slashes_underscores_g(daemon_to_unmount_mq_name + 1);
+       return 0;
+}
+
+static void free_message_queue_names()
+{
+       FREE(unmount_to_daemon_mq_name);
+       FREE(daemon_to_unmount_mq_name);
+       unmount_to_daemon_mq_name = NULL;
+       daemon_to_unmount_mq_name = NULL;
+}
+
+/*
+ * Opens two POSIX message queue: one for sending messages from the unmount
+ * process to the daemon process, and one to go the other way.  The names of the
+ * message queues, which must be system-wide unique, are be based on the mount
+ * point.  (There of course is still a possibility of a collision if one were to
+ * unmount two identically named directories simultaneously...)
+ *
+ * @daemon specifies whether the calling process is the filesystem daemon or the
+ * unmount process.
+ */
+static int open_message_queues(bool daemon)
+{
+       int flags;
+       int ret;
+
+       wimlib_assert(unmount_to_daemon_mq_name != NULL &&
+                     daemon_to_unmount_mq_name != NULL);
 
        if (daemon)
                flags = O_RDONLY | O_CREAT;
        else
                flags = O_WRONLY | O_CREAT;
 
-       unmount_to_daemon_mq = mq_open(unmount_to_daemon_mq_name, flags, 
+       unmount_to_daemon_mq = mq_open(unmount_to_daemon_mq_name, flags,
                                       0700, NULL);
 
        if (unmount_to_daemon_mq == (mqd_t)-1) {
                ERROR_WITH_ERRNO("mq_open()");
                ret = WIMLIB_ERR_MQUEUE;
-               goto err2;
+               goto out;
        }
 
        if (daemon)
@@ -664,22 +627,21 @@ static int open_message_queues(bool daemon)
        else
                flags = O_RDONLY | O_CREAT;
 
-       daemon_to_unmount_mq = mq_open(daemon_to_unmount_mq_name, flags, 
+       daemon_to_unmount_mq = mq_open(daemon_to_unmount_mq_name, flags,
                                       0700, NULL);
 
        if (daemon_to_unmount_mq == (mqd_t)-1) {
                ERROR_WITH_ERRNO("mq_open()");
                ret = WIMLIB_ERR_MQUEUE;
-               goto err3;
+               goto out_close_unmount_to_daemon_mq;
        }
-       return 0;
-err3:
+       ret = 0;
+       goto out;
+out_close_unmount_to_daemon_mq:
        mq_close(unmount_to_daemon_mq);
        mq_unlink(unmount_to_daemon_mq_name);
-err2:
-       FREE(daemon_to_unmount_mq_name);
-err1:
-       FREE(unmount_to_daemon_mq_name);
+       unmount_to_daemon_mq = (mqd_t)-1;
+out:
        return ret;
 }
 
@@ -712,13 +674,20 @@ static int mq_get_msgsize(mqd_t mq)
        return msgsize;
 }
 
+static void unlink_message_queues()
+{
+       mq_unlink(unmount_to_daemon_mq_name);
+       mq_unlink(daemon_to_unmount_mq_name);
+}
+
 /* Closes the message queues, which are allocated in static variables */
 static void close_message_queues()
 {
        mq_close(unmount_to_daemon_mq);
+       unmount_to_daemon_mq = (mqd_t)(-1);
        mq_close(daemon_to_unmount_mq);
-       mq_unlink(unmount_to_daemon_mq_name);
-       mq_unlink(daemon_to_unmount_mq_name);
+       daemon_to_unmount_mq = (mqd_t)(-1);
+       unlink_message_queues();
 }
 
 static int wimfs_access(const char *path, int mask)
@@ -727,6 +696,22 @@ static int wimfs_access(const char *path, int mask)
        return 0;
 }
 
+static void inode_update_lte_ptr(struct inode *inode,
+                                struct lookup_table_entry *old_lte,
+                                struct lookup_table_entry *new_lte)
+{
+       if (inode->lte == old_lte) {
+               inode->lte = new_lte;
+       } else {
+               for (unsigned i = 0; i < inode->num_ads; i++) {
+                       if (inode->ads_entries[i].lte == old_lte) {
+                               inode->ads_entries[i].lte = new_lte;
+                               break;
+                       }
+               }
+       }
+}
+
 static int update_lte_of_staging_file(struct lookup_table_entry *lte,
                                      struct lookup_table *table)
 {
@@ -750,30 +735,37 @@ static int update_lte_of_staging_file(struct lookup_table_entry *lte,
                /* Merge duplicate lookup table entries */
                duplicate_lte->refcnt += lte->refcnt;
                list_del(&lte->staging_list);
+               inode_update_lte_ptr(lte->lte_inode, lte, duplicate_lte);
                free_lookup_table_entry(lte);
        } else {
                if (stat(lte->staging_file_name, &stbuf) != 0) {
                        ERROR_WITH_ERRNO("Failed to stat `%s'", lte->staging_file_name);
                        return WIMLIB_ERR_STAT;
                }
-               wimlib_assert(&lte->file_on_disk == &lte->staging_file_name);
-               lte->resource_location = RESOURCE_IN_FILE_ON_DISK;
-               copy_hash(lte->hash, hash);
-               lte->resource_entry.original_size = stbuf.st_size;
-               lte->resource_entry.size = stbuf.st_size;
-               lte->inode = NULL;
-               lookup_table_insert(table, lte);
+               if (stbuf.st_size == 0) {
+                       /* Zero-length stream.  No lookup table entry needed. */
+                       inode_update_lte_ptr(lte->lte_inode, lte, NULL);
+                       free_lookup_table_entry(lte);
+               } else {
+                       wimlib_assert(&lte->file_on_disk == &lte->staging_file_name);
+                       lte->resource_entry.original_size = stbuf.st_size;
+                       lte->resource_entry.size = stbuf.st_size;
+                       lte->resource_location = RESOURCE_IN_FILE_ON_DISK;
+                       lte->lte_inode = NULL;
+                       copy_hash(lte->hash, hash);
+                       lookup_table_insert(table, lte);
+               }
        }
-
        return 0;
 }
 
 static int inode_close_fds(struct inode *inode)
 {
-       for (u16 i = 0, j = 0; j < inode->num_opened_fds; i++) {
+       u16 num_opened_fds = inode->num_opened_fds;
+       for (u16 i = 0, j = 0; j < num_opened_fds; i++) {
                struct wimlib_fd *fd = inode->fds[i];
                if (fd) {
-                       wimlib_assert(fd->inode == inode);
+                       wimlib_assert(fd->f_inode == inode);
                        int ret = close_wimlib_fd(fd);
                        if (ret != 0)
                                return ret;
@@ -783,11 +775,6 @@ static int inode_close_fds(struct inode *inode)
        return 0;
 }
 
-/*static int dentry_close_fds(struct dentry *dentry, void *ignore)*/
-/*{*/
-       /*return inode_close_fds(dentry->inode);*/
-/*}*/
-
 /* Overwrites the WIM file, with changes saved. */
 static int rebuild_wim(WIMStruct *w, bool check_integrity)
 {
@@ -797,13 +784,10 @@ static int rebuild_wim(WIMStruct *w, bool check_integrity)
 
        DEBUG("Closing all staging file descriptors.");
        list_for_each_entry(lte, &staging_list, staging_list) {
-               ret = inode_close_fds(lte->inode);
+               ret = inode_close_fds(lte->lte_inode);
                if (ret != 0)
                        return ret;
        }
-       /*ret = for_dentry_in_tree(wim_root_dentry(w), dentry_close_fds, NULL);*/
-       /*if (ret != 0)*/
-               /*return ret;*/
 
        DEBUG("Calculating SHA1 checksums for all new staging files.");
        list_for_each_entry_safe(lte, tmp, &staging_list, staging_list) {
@@ -842,7 +826,7 @@ static void wimfs_destroy(void *p)
 
        ret = open_message_queues(true);
        if (ret != 0)
-               exit(1);
+               return;
 
        msgsize = mq_get_msgsize(unmount_to_daemon_mq);
        char msg[msgsize];
@@ -856,19 +840,20 @@ static void wimfs_destroy(void *p)
        DEBUG("Waiting for message telling us whether to commit or not, and "
              "whether to include integrity checks.");
 
-       bytes_received = mq_timedreceive(unmount_to_daemon_mq, msg, 
+       bytes_received = mq_timedreceive(unmount_to_daemon_mq, msg,
                                         msgsize, NULL, &timeout);
-       commit = msg[0];
-       check_integrity = msg[1];
        if (bytes_received == -1) {
-               if (errno == ETIMEDOUT) {
+               if (errno == ETIMEDOUT)
                        ERROR("Timed out.");
-               } else {
+               else
                        ERROR_WITH_ERRNO("mq_timedreceive()");
-               }
                ERROR("Not committing.");
+               commit = 0;
+               check_integrity = 0;
        } else {
                DEBUG("Received message: [%d %d]", msg[0], msg[1]);
+               commit = msg[0];
+               check_integrity = msg[1];
        }
 
        status = 0;
@@ -878,7 +863,7 @@ static void wimfs_destroy(void *p)
                        if (status != 0) {
                                ERROR_WITH_ERRNO("chdir()");
                                status = WIMLIB_ERR_NOTDIR;
-                               goto done;
+                               goto out;
                        }
                        status = rebuild_wim(w, (check_integrity != 0));
                }
@@ -892,8 +877,8 @@ static void wimfs_destroy(void *p)
        } else {
                DEBUG("Read-only mount");
        }
-done:
-       DEBUG("Sending status %u", status);
+out:
+       DEBUG("Sending status %hhd", status);
        ret = mq_send(daemon_to_unmount_mq, &status, 1, 1);
        if (ret == -1)
                ERROR_WITH_ERRNO("Failed to send status to unmount process");
@@ -915,7 +900,7 @@ static int wimfs_fgetattr(const char *path, struct stat *stbuf,
                          struct fuse_file_info *fi)
 {
        struct wimlib_fd *fd = (struct wimlib_fd*)(uintptr_t)fi->fh;
-       return inode_to_stbuf(fd->inode, fd->lte, stbuf);
+       return inode_to_stbuf(fd->f_inode, fd->f_lte, stbuf);
 }
 
 static int wimfs_ftruncate(const char *path, off_t size,
@@ -925,8 +910,8 @@ static int wimfs_ftruncate(const char *path, off_t size,
        int ret = ftruncate(fd->staging_fd, size);
        if (ret != 0)
                return ret;
-       if (fd->lte && size < fd->lte->resource_entry.original_size)
-               fd->lte->resource_entry.original_size = size;
+       if (fd->f_lte && size < fd->f_lte->resource_entry.original_size)
+               fd->f_lte->resource_entry.original_size = size;
        return 0;
 }
 
@@ -944,7 +929,7 @@ static int wimfs_getattr(const char *path, struct stat *stbuf)
                              &dentry, &lte, NULL);
        if (ret != 0)
                return ret;
-       return inode_to_stbuf(dentry->inode, lte, stbuf);
+       return inode_to_stbuf(dentry->d_inode, lte, stbuf);
 }
 
 #ifdef ENABLE_XATTR
@@ -953,7 +938,7 @@ static int wimfs_getxattr(const char *path, const char *name, char *value,
                          size_t size)
 {
        int ret;
-       struct dentry *dentry;
+       struct inode *inode;
        struct ads_entry *ads_entry;
        size_t res_size;
        struct lookup_table_entry *lte;
@@ -965,10 +950,11 @@ static int wimfs_getxattr(const char *path, const char *name, char *value,
                return -ENOATTR;
        name += 5;
 
-       dentry = get_dentry(w, path);
-       if (!dentry)
+       inode = wim_pathname_to_inode(w, path);
+       if (!inode)
                return -ENOENT;
-       ads_entry = inode_get_ads_entry(dentry->inode, name, NULL);
+
+       ads_entry = inode_get_ads_entry(inode, name, NULL);
        if (!ads_entry)
                return -ENOATTR;
 
@@ -977,11 +963,14 @@ static int wimfs_getxattr(const char *path, const char *name, char *value,
 
        if (size == 0)
                return res_size;
+
        if (res_size > size)
                return -ERANGE;
+
        ret = read_full_wim_resource(lte, (u8*)value);
        if (ret != 0)
                return -EIO;
+
        return res_size;
 }
 #endif
@@ -989,18 +978,15 @@ static int wimfs_getxattr(const char *path, const char *name, char *value,
 /* Create a hard link */
 static int wimfs_link(const char *to, const char *from)
 {
-       struct dentry *to_dentry, *from_dentry, *from_dentry_parent;
+       struct dentry *from_dentry, *from_dentry_parent;
        const char *link_name;
        struct inode *inode;
-       unsigned i;
        struct lookup_table_entry *lte;
 
-       to_dentry = get_dentry(w, to);
-       if (!to_dentry)
+       inode = wim_pathname_to_inode(w, to);
+       if (!inode)
                return -ENOENT;
 
-       inode = to_dentry->inode;
-
        if (!inode_is_regular_file(inode))
                return -EPERM;
 
@@ -1019,10 +1005,10 @@ static int wimfs_link(const char *to, const char *from)
 
 
        inode_add_dentry(from_dentry, inode);
-       from_dentry->inode = inode;
+       from_dentry->d_inode = inode;
        inode->link_count++;
 
-       for (i = 0; i <= inode->num_ads; i++) {
+       for (unsigned i = 0; i <= inode->num_ads; i++) {
                lte = inode_stream_lte_resolved(inode, i);
                if (lte)
                        lte->refcnt++;
@@ -1035,34 +1021,33 @@ static int wimfs_link(const char *to, const char *from)
 #ifdef ENABLE_XATTR
 static int wimfs_listxattr(const char *path, char *list, size_t size)
 {
-       struct dentry *dentry;
        int ret;
-       char *p = list;
        size_t needed_size;
        unsigned i;
        struct inode *inode;
+
        if (!(mount_flags & WIMLIB_MOUNT_FLAG_STREAM_INTERFACE_XATTR))
                return -ENOTSUP;
 
        /* List alternate data streams, or get the list size */
 
-       ret = lookup_resource(w, path, get_lookup_flags(), &dentry, NULL, NULL);
-       if (ret != 0)
-               return ret;
-       inode = dentry->inode;
+       inode = wim_pathname_to_inode(w, path);
+       if (!inode)
+               return -ENOENT;
 
        if (size == 0) {
                needed_size = 0;
                for (i = 0; i < inode->num_ads; i++)
-                       needed_size += inode->ads_entries[i]->stream_name_utf8_len + 6;
+                       needed_size += inode->ads_entries[i].stream_name_utf8_len + 6;
                return needed_size;
        } else {
+               char *p = list;
                for (i = 0; i < inode->num_ads; i++) {
-                       needed_size = inode->ads_entries[i]->stream_name_utf8_len + 6;
+                       needed_size = inode->ads_entries[i].stream_name_utf8_len + 6;
                        if (needed_size > size)
                                return -ERANGE;
                        p += sprintf(p, "user.%s",
-                                    inode->ads_entries[i]->stream_name_utf8) + 1;
+                                    inode->ads_entries[i].stream_name_utf8) + 1;
                        size -= needed_size;
                }
                return p - list;
@@ -1070,8 +1055,8 @@ static int wimfs_listxattr(const char *path, char *list, size_t size)
 }
 #endif
 
-/* 
- * Create a directory in the WIM.  
+/*
+ * Create a directory in the WIM.
  * @mode is currently ignored.
  */
 static int wimfs_mkdir(const char *path, mode_t mode)
@@ -1079,7 +1064,7 @@ static int wimfs_mkdir(const char *path, mode_t mode)
        struct dentry *parent;
        struct dentry *newdir;
        const char *basename;
-       
+
        parent = get_parent_dentry(w, path);
        if (!parent)
                return -ENOENT;
@@ -1092,14 +1077,13 @@ static int wimfs_mkdir(const char *path, mode_t mode)
                return -EEXIST;
 
        newdir = new_dentry_with_inode(basename);
-       newdir->inode->attributes |= FILE_ATTRIBUTE_DIRECTORY;
-       newdir->inode->resolved = true;
-       newdir->inode->ino = next_ino++;
+       newdir->d_inode->attributes |= FILE_ATTRIBUTE_DIRECTORY;
+       newdir->d_inode->resolved = true;
+       newdir->d_inode->ino = next_ino++;
        link_dentry(newdir, parent);
        return 0;
 }
 
-
 /* Creates a regular file. */
 static int wimfs_mknod(const char *path, mode_t mode, dev_t rdev)
 {
@@ -1108,20 +1092,22 @@ static int wimfs_mknod(const char *path, mode_t mode, dev_t rdev)
             && (stream_name = path_stream_name(path))) {
                /* Make an alternate data stream */
                struct ads_entry *new_entry;
-               struct dentry *dentry;
+               struct inode *inode;
 
                char *p = (char*)stream_name - 1;
                wimlib_assert(*p == ':');
                *p = '\0';
 
-               dentry = get_dentry(w, path);
-               if (!dentry || !dentry_is_regular_file(dentry))
+               inode = wim_pathname_to_inode(w, path);
+               if (!inode)
+                       return -ENOENT;
+               if (!inode_is_regular_file(inode))
                        return -ENOENT;
-               if (inode_get_ads_entry(dentry->inode, stream_name, NULL))
+               if (inode_get_ads_entry(inode, stream_name, NULL))
                        return -EEXIST;
-               new_entry = inode_add_ads(dentry->inode, stream_name);
+               new_entry = inode_add_ads(inode, stream_name);
                if (!new_entry)
-                       return -ENOENT;
+                       return -ENOMEM;
        } else {
                struct dentry *dentry, *parent;
                const char *basename;
@@ -1143,8 +1129,8 @@ static int wimfs_mknod(const char *path, mode_t mode, dev_t rdev)
                dentry = new_dentry_with_inode(basename);
                if (!dentry)
                        return -ENOMEM;
-               dentry->inode->resolved = true;
-               dentry->inode->ino = next_ino++;
+               dentry->d_inode->resolved = true;
+               dentry->d_inode->ino = next_ino++;
                link_dentry(dentry, parent);
        }
        return 0;
@@ -1167,12 +1153,12 @@ static int wimfs_open(const char *path, struct fuse_file_info *fi)
        if (ret != 0)
                return ret;
 
-       inode = dentry->inode;
+       inode = dentry->d_inode;
 
        if (stream_idx == 0)
                stream_id = 0;
        else
-               stream_id = inode->ads_entries[stream_idx - 1]->stream_id;
+               stream_id = inode->ads_entries[stream_idx - 1].stream_id;
 
        /* The file resource may be in the staging directory (read-write mounts
         * only) or in the WIM.  If it's in the staging directory, we need to
@@ -1180,6 +1166,7 @@ static int wimfs_open(const char *path, struct fuse_file_info *fi)
         * we can read the file resource directly from the WIM file if we are
         * opening it read-only, but we need to extract the resource to the
         * staging directory if we are opening it writable. */
+
        if (flags_writable(fi->flags) &&
             (!lte || lte->resource_location != RESOURCE_IN_STAGING_FILE)) {
                u64 size = (lte) ? wim_resource_size(lte) : 0;
@@ -1210,7 +1197,7 @@ static int wimfs_opendir(const char *path, struct fuse_file_info *fi)
        struct inode *inode;
        int ret;
        struct wimlib_fd *fd = NULL;
-       
+
        inode = wim_pathname_to_inode(w, path);
        if (!inode)
                return -ENOENT;
@@ -1223,9 +1210,9 @@ static int wimfs_opendir(const char *path, struct fuse_file_info *fi)
 
 
 /*
- * Read data from a file in the WIM or in the staging directory. 
+ * Read data from a file in the WIM or in the staging directory.
  */
-static int wimfs_read(const char *path, char *buf, size_t size, 
+static int wimfs_read(const char *path, char *buf, size_t size,
                      off_t offset, struct fuse_file_info *fi)
 {
        struct wimlib_fd *fd = (struct wimlib_fd*)(uintptr_t)fi->fh;
@@ -1233,13 +1220,13 @@ static int wimfs_read(const char *path, char *buf, size_t size,
        if (!fd)
                return -EBADF;
 
-       if (!fd->lte) /* Empty stream with no lookup table entry */
+       if (!fd->f_lte) /* Empty stream with no lookup table entry */
                return 0;
 
-       if (fd->lte->resource_location == RESOURCE_IN_STAGING_FILE) {
+       if (fd->f_lte->resource_location == RESOURCE_IN_STAGING_FILE) {
                /* Read from staging file */
 
-               wimlib_assert(fd->lte->staging_file_name);
+               wimlib_assert(fd->f_lte->staging_file_name);
                wimlib_assert(fd->staging_fd != -1);
 
                ssize_t ret;
@@ -1254,14 +1241,14 @@ static int wimfs_read(const char *path, char *buf, size_t size,
        } else {
                /* Read from WIM */
 
-               u64 res_size = wim_resource_size(fd->lte);
-               
+               u64 res_size = wim_resource_size(fd->f_lte);
+
                if (offset > res_size)
                        return -EOVERFLOW;
 
                size = min(size, res_size - offset);
 
-               if (read_wim_resource(fd->lte, (u8*)buf,
+               if (read_wim_resource(fd->f_lte, (u8*)buf,
                                      size, offset, false) != 0)
                        return -EIO;
                return size;
@@ -1270,7 +1257,7 @@ static int wimfs_read(const char *path, char *buf, size_t size,
 
 /* Fills in the entries of the directory specified by @path using the
  * FUSE-provided function @filler.  */
-static int wimfs_readdir(const char *path, void *buf, fuse_fill_dir_t filler, 
+static int wimfs_readdir(const char *path, void *buf, fuse_fill_dir_t filler,
                         off_t offset, struct fuse_file_info *fi)
 {
        struct wimlib_fd *fd = (struct wimlib_fd*)(uintptr_t)fi->fh;
@@ -1280,7 +1267,7 @@ static int wimfs_readdir(const char *path, void *buf, fuse_fill_dir_t filler,
        if (!fd)
                return -EBADF;
 
-       inode = fd->inode;
+       inode = fd->f_inode;
 
        filler(buf, ".", NULL, 0);
        filler(buf, "..", NULL, 0);
@@ -1301,14 +1288,14 @@ static int wimfs_readdir(const char *path, void *buf, fuse_fill_dir_t filler,
 
 static int wimfs_readlink(const char *path, char *buf, size_t buf_len)
 {
-       struct dentry *dentry = get_dentry(w, path);
+       struct inode *inode = wim_pathname_to_inode(w, path);
        int ret;
-       if (!dentry)
+       if (!inode)
                return -ENOENT;
-       if (!dentry_is_symlink(dentry))
+       if (!inode_is_symlink(inode))
                return -EINVAL;
 
-       ret = inode_readlink(dentry->inode, buf, buf_len, w);
+       ret = inode_readlink(inode, buf, buf_len, w);
        if (ret > 0)
                ret = 0;
        return ret;
@@ -1318,14 +1305,13 @@ static int wimfs_readlink(const char *path, char *buf, size_t buf_len)
 static int wimfs_release(const char *path, struct fuse_file_info *fi)
 {
        struct wimlib_fd *fd = (struct wimlib_fd*)(uintptr_t)fi->fh;
-       wimlib_assert(fd);
        return close_wimlib_fd(fd);
 }
 
+/* Close a directory */
 static int wimfs_releasedir(const char *path, struct fuse_file_info *fi)
 {
        struct wimlib_fd *fd = (struct wimlib_fd*)(uintptr_t)fi->fh;
-       wimlib_assert(fd);
        return close_wimlib_fd(fd);
 }
 
@@ -1333,7 +1319,7 @@ static int wimfs_releasedir(const char *path, struct fuse_file_info *fi)
 /* Remove an alternate data stream through the XATTR interface */
 static int wimfs_removexattr(const char *path, const char *name)
 {
-       struct dentry *dentry;
+       struct inode *inode;
        struct ads_entry *ads_entry;
        u16 ads_idx;
        if (!(mount_flags & WIMLIB_MOUNT_FLAG_STREAM_INTERFACE_XATTR))
@@ -1343,14 +1329,14 @@ static int wimfs_removexattr(const char *path, const char *name)
                return -ENOATTR;
        name += 5;
 
-       dentry = get_dentry(w, path);
-       if (!dentry)
+       inode = wim_pathname_to_inode(w, path);
+       if (!inode)
                return -ENOENT;
 
-       ads_entry = inode_get_ads_entry(dentry->inode, name, &ads_idx);
+       ads_entry = inode_get_ads_entry(inode, name, &ads_idx);
        if (!ads_entry)
                return -ENOATTR;
-       inode_remove_ads(dentry->inode, ads_idx, w->lookup_table);
+       inode_remove_ads(inode, ads_idx, w->lookup_table);
        return 0;
 }
 #endif
@@ -1367,7 +1353,7 @@ static int wimfs_rename(const char *from, const char *to)
 
        /* This rename() implementation currently only supports actual files
         * (not alternate data streams) */
-       
+
        src = get_dentry(w, from);
        if (!src)
                return -ENOENT;
@@ -1396,7 +1382,7 @@ static int wimfs_rename(const char *from, const char *to)
                         * directory */
                        if (!dentry_is_directory(dst))
                                return -ENOTDIR;
-                       if (dst->inode->children != NULL)
+                       if (dst->d_inode->children != NULL)
                                return -ENOTEMPTY;
                }
                parent_of_dst = dst->parent;
@@ -1427,7 +1413,7 @@ static int wimfs_rename(const char *from, const char *to)
 static int wimfs_rmdir(const char *path)
 {
        struct dentry *dentry;
-       
+
        dentry = get_dentry(w, path);
        if (!dentry)
                return -ENOENT;
@@ -1512,7 +1498,7 @@ static int wimfs_symlink(const char *to, const char *from)
        struct dentry *dentry_parent, *dentry;
        const char *link_name;
        struct inode *inode;
-       
+
        dentry_parent = get_parent_dentry(w, from);
        if (!dentry_parent)
                return -ENOENT;
@@ -1526,7 +1512,7 @@ static int wimfs_symlink(const char *to, const char *from)
        dentry = new_dentry_with_inode(link_name);
        if (!dentry)
                return -ENOMEM;
-       inode = dentry->inode;
+       inode = dentry->d_inode;
 
        inode->attributes  = FILE_ATTRIBUTE_REPARSE_POINT;
        inode->reparse_tag = WIM_IO_REPARSE_TAG_SYMLINK;
@@ -1553,7 +1539,7 @@ static int wimfs_truncate(const char *path, off_t size)
        u16 stream_idx;
        u32 stream_id;
        struct inode *inode;
-       
+
        ret = lookup_resource(w, path, get_lookup_flags(), &dentry,
                              &lte, &stream_idx);
 
@@ -1563,12 +1549,12 @@ static int wimfs_truncate(const char *path, off_t size)
        if (!lte) /* Already a zero-length file */
                return 0;
 
-       inode = dentry->inode;
+       inode = dentry->d_inode;
 
        if (stream_idx == 0)
                stream_id = 0;
        else
-               stream_id = inode->ads_entries[stream_idx - 1]->stream_id;
+               stream_id = inode->ads_entries[stream_idx - 1].stream_id;
 
        if (lte->resource_location == RESOURCE_IN_STAGING_FILE) {
                wimlib_assert(lte->staging_file_name);
@@ -1595,7 +1581,7 @@ static int wimfs_unlink(const char *path)
        int ret;
        u16 stream_idx;
        unsigned i;
-       
+
        ret = lookup_resource(w, path, get_lookup_flags(), &dentry,
                              &lte, &stream_idx);
 
@@ -1605,13 +1591,13 @@ static int wimfs_unlink(const char *path)
        if (stream_idx == 0)
                remove_dentry(dentry, w->lookup_table);
        else
-               inode_remove_ads(dentry->inode, stream_idx - 1, w->lookup_table);
+               inode_remove_ads(dentry->d_inode, stream_idx - 1, w->lookup_table);
        return 0;
 }
 
 #ifdef HAVE_UTIMENSAT
-/* 
- * Change the timestamp on a file dentry. 
+/*
+ * Change the timestamp on a file dentry.
  *
  * Note that alternate data streams do not have their own timestamps.
  */
@@ -1622,7 +1608,7 @@ static int wimfs_utimens(const char *path, const struct timespec tv[2])
        dentry = get_dentry(w, path);
        if (!dentry)
                return -ENOENT;
-       inode = dentry->inode;
+       inode = dentry->d_inode;
 
        if (tv[0].tv_nsec != UTIME_OMIT) {
                if (tv[0].tv_nsec == UTIME_NOW)
@@ -1646,7 +1632,7 @@ static int wimfs_utime(const char *path, struct utimbuf *times)
        dentry = get_dentry(w, path);
        if (!dentry)
                return -ENOENT;
-       inode = dentry->inode;
+       inode = dentry->d_inode;
 
        inode->last_write_time = unix_timestamp_to_wim(times->modtime);
        inode->last_access_time = unix_timestamp_to_wim(times->actime);
@@ -1654,10 +1640,10 @@ static int wimfs_utime(const char *path, struct utimbuf *times)
 }
 #endif
 
-/* Writes to a file in the WIM filesystem. 
+/* Writes to a file in the WIM filesystem.
  * It may be an alternate data stream, but here we don't even notice because we
  * just get a lookup table entry. */
-static int wimfs_write(const char *path, const char *buf, size_t size, 
+static int wimfs_write(const char *path, const char *buf, size_t size,
                       off_t offset, struct fuse_file_info *fi)
 {
        struct wimlib_fd *fd = (struct wimlib_fd*)(uintptr_t)fi->fh;
@@ -1667,10 +1653,10 @@ static int wimfs_write(const char *path, const char *buf, size_t size,
        if (!fd)
                return -EBADF;
 
-       wimlib_assert(fd->lte);
-       wimlib_assert(fd->lte->staging_file_name);
+       wimlib_assert(fd->f_lte);
+       wimlib_assert(fd->f_lte->staging_file_name);
        wimlib_assert(fd->staging_fd != -1);
-       wimlib_assert(fd->inode);
+       wimlib_assert(fd->f_inode);
 
        /* Seek to the requested position */
        if (lseek(fd->staging_fd, offset, SEEK_SET) == -1)
@@ -1682,8 +1668,8 @@ static int wimfs_write(const char *path, const char *buf, size_t size,
                return -errno;
 
        now = get_wim_timestamp();
-       fd->inode->last_write_time = now;
-       fd->inode->last_access_time = now;
+       fd->f_inode->last_write_time = now;
+       fd->f_inode->last_access_time = now;
        return ret;
 }
 
@@ -1733,14 +1719,14 @@ static struct fuse_operations wimfs_operations = {
 
 
 /* Mounts a WIM file. */
-WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir, 
+WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
                           int flags, WIMStruct **additional_swms,
                           unsigned num_additional_swms)
 {
        int argc = 0;
        char *argv[16];
        int ret;
-       char *p;
+       char *dir_copy;
        struct lookup_table *joined_tab, *wim_tab_save;
        struct image_metadata *imd;
 
@@ -1764,7 +1750,7 @@ WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
                wim->lookup_table = joined_tab;
        }
 
-       ret = wimlib_select_image(wim, image);
+       ret = select_wim_image(wim, image);
 
        if (ret != 0)
                goto out;
@@ -1773,17 +1759,12 @@ WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
 
        DEBUG("Selected image %d", image);
 
-       next_ino = assign_inode_numbers(&imd->inode_list);
-
-       DEBUG("(next_ino = %"PRIu64")", next_ino);
-
-       /* Resolve all the lookup table entries of the dentry tree */
-       DEBUG("Resolving lookup table entries");
-       for_dentry_in_tree(imd->root_dentry, dentry_resolve_ltes,
-                          wim->lookup_table);
-
-       if (flags & WIMLIB_MOUNT_FLAG_READWRITE)
-               imd->modified = true;
+       if (imd->root_dentry->refcnt != 1) {
+               ERROR("Cannot mount image that was just exported with "
+                     "wimlib_export()");
+               ret = WIMLIB_ERR_INVALID_PARAM;
+               goto out;
+       }
 
        if (!(flags & (WIMLIB_MOUNT_FLAG_STREAM_INTERFACE_NONE |
                       WIMLIB_MOUNT_FLAG_STREAM_INTERFACE_XATTR |
@@ -1792,7 +1773,6 @@ WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
 
        DEBUG("Getting current directory");
 
-       mount_dir = dir;
        working_directory = getcwd(NULL, 0);
        if (!working_directory) {
                ERROR_WITH_ERRNO("Could not determine current directory");
@@ -1800,36 +1780,30 @@ WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
                goto out;
        }
 
-       DEBUG("Closing POSIX message queues");
-       /* XXX hack to get rid of the message queues if they already exist for
-        * some reason (maybe left over from a previous mount that wasn't
-        * unmounted correctly) */
-       ret = open_message_queues(true);
+       DEBUG("Unlinking message queues in case they already exist");
+       ret = set_message_queue_names(dir);
        if (ret != 0)
-               goto out;
-       close_message_queues();
+               goto out_free_working_directory;
+       unlink_message_queues();
 
        DEBUG("Preparing arguments to fuse_main()");
 
-
-       p = STRDUP(dir);
-       if (!p) {
-               ret = WIMLIB_ERR_NOMEM;
-               goto out;
-       }
+       dir_copy = STRDUP(dir);
+       if (!dir_copy)
+               goto out_free_message_queue_names;
 
        argv[argc++] = "imagex";
-       argv[argc++] = p;
+       argv[argc++] = dir_copy;
        argv[argc++] = "-s"; /* disable multi-threaded operation */
 
        if (flags & WIMLIB_MOUNT_FLAG_DEBUG)
                argv[argc++] = "-d";
 
-       /* 
+       /*
         * We provide the use_ino option because we are going to assign inode
-        * numbers oursides.  We've already numbered the inodes with unique
-        * numbers in the assign_inode_numbers() function, and the static
-        * variable next_ino is set to the next available inode number.
+        * numbers oursides.  The inodes will be given unique numbers in the
+        * assign_inode_numbers() function, and the static variable @next_ino is
+        * set to the next available inode number.
         */
        char optstring[256] = "use_ino,subtype=wimfs,attr_timeout=0";
        argv[argc++] = "-o";
@@ -1838,9 +1812,8 @@ WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
                /* Read-write mount.  Make the staging directory */
                make_staging_dir();
                if (!staging_dir_name) {
-                       FREE(p);
                        ret = WIMLIB_ERR_MKDIR;
-                       goto out;
+                       goto out_free_dir_copy;
                }
        } else {
                /* Read-only mount */
@@ -1861,6 +1834,20 @@ WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
        }
 #endif
 
+
+       /* Mark dentry tree as modified if read-write mount. */
+       if (flags & WIMLIB_MOUNT_FLAG_READWRITE)
+               imd->modified = true;
+
+       next_ino = assign_inode_numbers(&imd->inode_list);
+
+       DEBUG("(next_ino = %"PRIu64")", next_ino);
+
+       /* Resolve all the lookup table entries of the dentry tree */
+       DEBUG("Resolving lookup table entries");
+       for_dentry_in_tree(imd->root_dentry, dentry_resolve_ltes,
+                          wim->lookup_table);
+
        /* Set static variables. */
        w = wim;
        mount_flags = flags;
@@ -1868,6 +1855,13 @@ WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
        ret = fuse_main(argc, argv, &wimfs_operations, NULL);
        if (ret)
                ret = WIMLIB_ERR_FUSE;
+out_free_dir_copy:
+       FREE(dir_copy);
+out_free_message_queue_names:
+       free_message_queue_names();
+out_free_working_directory:
+       FREE(working_directory);
+       working_directory = NULL;
 out:
        if (num_additional_swms) {
                free_lookup_table(wim->lookup_table);
@@ -1877,7 +1871,7 @@ out:
 }
 
 
-/* 
+/*
  * Unmounts the WIM file that was previously mounted on @dir by using
  * wimlib_mount().
  */
@@ -1892,10 +1886,12 @@ WIMLIBAPI int wimlib_unmount(const char *dir, int flags)
        int msgsize;
        int errno_save;
 
-       mount_dir = dir;
-
        /* Open message queues between the unmount process and the
         * filesystem daemon. */
+       ret = set_message_queue_names(dir);
+       if (ret != 0)
+               return ret;
+
        ret = open_message_queues(false);
        if (ret != 0)
                return ret;
@@ -1905,7 +1901,7 @@ WIMLIBAPI int wimlib_unmount(const char *dir, int flags)
        msg[0] = (flags & WIMLIB_UNMOUNT_FLAG_COMMIT) ? 1 : 0;
        msg[1] = (flags & WIMLIB_UNMOUNT_FLAG_CHECK_INTEGRITY) ? 1 : 0;
 
-       DEBUG("Sending message: %s, %s", 
+       DEBUG("Sending message: %s, %s",
                        (msg[0] == 0) ? "don't commit" : "commit",
                        (msg[1] == 0) ? "don't check"  : "check");
        ret = mq_send(unmount_to_daemon_mq, msg, 2, 1);
@@ -1925,7 +1921,7 @@ WIMLIBAPI int wimlib_unmount(const char *dir, int flags)
         * send a message from this process to the filesystem daemon telling
         * whether --commit was specified or not.  However, after that, the
         * unmount process must wait for the filesystem daemon to finish writing
-        * the WIM file. 
+        * the WIM file.
         */
 
 
@@ -1987,7 +1983,7 @@ WIMLIBAPI int wimlib_unmount(const char *dir, int flags)
         * filesystem daemon has crashed or failed for some reason.
         *
         * XXX come up with some method to determine if the filesystem
-        * daemon has really crashed or not. 
+        * daemon has really crashed or not.
         *
         * XXX Idea: have mount daemon write its PID into the WIM file header?
         * */
@@ -2040,7 +2036,7 @@ WIMLIBAPI int wimlib_unmount(const char *dir, int flags)
        return mount_unsupported_error();
 }
 
-WIMLIBAPI int wimlib_mount(WIMStruct *wim_p, int image, const char *dir, 
+WIMLIBAPI int wimlib_mount(WIMStruct *wim_p, int image, const char *dir,
                           int flags, WIMStruct **additional_swms,
                           unsigned num_additional_swms)
 {