]> wimlib.net Git - wimlib/blobdiff - src/mount.c
Store dentry children in red-black trees
[wimlib] / src / mount.c
index c130f618c2777deb6f89f4d6d5db26676b7bfe00..ed428fca58907d1b943fb51b7590cbd934c92ace 100644 (file)
@@ -35,6 +35,7 @@
 #include "xml.h"
 #include "io.h"
 #include "timestamp.h"
+#include <limits.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <sys/wait.h>
@@ -541,72 +542,65 @@ static int delete_staging_dir(struct wimfs_context *ctx)
 }
 
 
-/* Simple function that returns the concatenation of 4 strings. */
-static char *strcat_dup(const char *s1, const char *s2, const char *s3,
-                       const char *s4)
+/* Simple function that returns the concatenation of 2 strings. */
+static char *strcat_dup(const char *s1, const char *s2, size_t max_len)
 {
-       size_t len = strlen(s1) + strlen(s2) + strlen(s3) + strlen(s4) + 1;
-       char *p = MALLOC(len);
+       size_t len = strlen(s1) + strlen(s2);
+       if (len > max_len)
+               len = max_len;
+       char *p = MALLOC(len + 1);
        if (!p)
                return NULL;
-       p = strcpy(p, s1);
-       p = strcat(p, s2);
-       p = strcat(p, s3);
-       return strcat(p, s4);
+       snprintf(p, len + 1, "%s%s", s1, s2);
+       return p;
 }
 
-/* Removes trailing forward slashes in a string. */
-static void remove_trailing_slashes(char *s)
+static int set_message_queue_names(struct wimfs_context *ctx,
+                                  const char *mount_dir)
 {
-       long len = strlen(s);
-       for (long i = len - 1; i >= 1; i--) {
-               if (s[i] == '/')
-                       s[i] = '\0';
-               else
-                       break;
-       }
-}
+       static const char *u2d_prefix = "/wimlib-unmount-to-daemon-mq";
+       static const char *d2u_prefix = "/wimlib-daemon-to-unmount-mq";
+       char *dir_path;
+       char *p;
+       int ret;
 
-/* Changes forward slashes to underscores in a string. */
-static void s_slashes_underscores_g(char *s)
-{
-       while (*s) {
-               if (*s == '/')
-                       *s = '_';
-               s++;
+       dir_path = realpath(mount_dir, NULL);
+       if (!dir_path) {
+               ERROR_WITH_ERRNO("Failed to resolve path \"%s\"", mount_dir);
+               return WIMLIB_ERR_NOTDIR;
        }
-}
 
-static int set_message_queue_names(struct wimfs_context *ctx,
-                                  const char *mount_dir)
-{
-       static const char *slash = "/";
-       static const char *prefix = "wimlib-";
-       static const char *u2d_suffix = "unmount-to-daemon-mq";
-       static const char *d2u_suffix = "daemon-to-unmount-mq";
+       DEBUG("Using absolute dir_path = `%s'", dir_path);
 
-       const char *mount_dir_basename = path_basename(mount_dir);
 
-       ctx->unmount_to_daemon_mq_name = strcat_dup(slash, mount_dir_basename,
-                                                   prefix, u2d_suffix);
+       p = dir_path;
+       while (*p) {
+               if (*p == '/')
+                       *p = 0xff;
+               p++;
+       }
+
+       ctx->unmount_to_daemon_mq_name = strcat_dup(u2d_prefix, dir_path,
+                                                   NAME_MAX);
        if (!ctx->unmount_to_daemon_mq_name) {
-               ERROR("Out of memory");
-               return WIMLIB_ERR_NOMEM;
+               ret = WIMLIB_ERR_NOMEM;
+               goto out_free_dir_path;
        }
-       ctx->daemon_to_unmount_mq_name = strcat_dup(slash, mount_dir_basename,
-                                                   prefix, d2u_suffix);
+       ctx->daemon_to_unmount_mq_name = strcat_dup(d2u_prefix, dir_path,
+                                                   NAME_MAX);
        if (!ctx->daemon_to_unmount_mq_name) {
-               ERROR("Out of memory");
-               FREE(ctx->unmount_to_daemon_mq_name);
-               ctx->unmount_to_daemon_mq_name = NULL;
-               return WIMLIB_ERR_NOMEM;
+               ret = WIMLIB_ERR_NOMEM;
+               goto out_free_unmount_to_daemon_mq_name;
        }
 
-       remove_trailing_slashes(ctx->unmount_to_daemon_mq_name);
-       remove_trailing_slashes(ctx->daemon_to_unmount_mq_name);
-       s_slashes_underscores_g(ctx->unmount_to_daemon_mq_name + 1);
-       s_slashes_underscores_g(ctx->daemon_to_unmount_mq_name + 1);
-       return 0;
+       ret = 0;
+       goto out_free_dir_path;
+out_free_unmount_to_daemon_mq_name:
+       FREE(ctx->unmount_to_daemon_mq_name);
+       ctx->unmount_to_daemon_mq_name = NULL;
+out_free_dir_path:
+       FREE(dir_path);
+       return ret;
 }
 
 static void free_message_queue_names(struct wimfs_context *ctx)
@@ -640,6 +634,7 @@ static int open_message_queues(struct wimfs_context *ctx, bool daemon)
        else
                flags = O_WRONLY | O_CREAT;
 
+       DEBUG("Opening message queue \"%s\"", ctx->unmount_to_daemon_mq_name);
        ctx->unmount_to_daemon_mq = mq_open(ctx->unmount_to_daemon_mq_name,
                                            flags, 0700, NULL);
 
@@ -654,6 +649,7 @@ static int open_message_queues(struct wimfs_context *ctx, bool daemon)
        else
                flags = O_RDONLY | O_CREAT;
 
+       DEBUG("Opening message queue \"%s\"", ctx->daemon_to_unmount_mq_name);
        ctx->daemon_to_unmount_mq = mq_open(ctx->daemon_to_unmount_mq_name,
                                            flags, 0700, NULL);
 
@@ -960,7 +956,7 @@ static int wimfs_ftruncate(const char *path, off_t size,
        struct wimlib_fd *fd = (struct wimlib_fd*)(uintptr_t)fi->fh;
        int ret = ftruncate(fd->staging_fd, size);
        if (ret != 0)
-               return ret;
+               return -errno;
        if (fd->f_lte && size < fd->f_lte->resource_entry.original_size)
                fd->f_lte->resource_entry.original_size = size;
        return 0;
@@ -1253,9 +1249,9 @@ static int wimfs_opendir(const char *path, struct fuse_file_info *fi)
        struct inode *inode;
        int ret;
        struct wimlib_fd *fd = NULL;
-       struct wimfs_context *ctx = wimfs_get_context();
+       WIMStruct *w = wimfs_get_WIMStruct();
 
-       inode = wim_pathname_to_inode(ctx->wim, path);
+       inode = wim_pathname_to_inode(w, path);
        if (!inode)
                return -ENOENT;
        if (!inode_is_directory(inode))
@@ -1312,6 +1308,18 @@ static int wimfs_read(const char *path, char *buf, size_t size,
        }
 }
 
+struct fill_params {
+       void *buf;
+       fuse_fill_dir_t filler;
+};
+
+static int dentry_fuse_fill(struct dentry *dentry, void *arg)
+{
+       struct fill_params *fill_params = arg;
+       return fill_params->filler(fill_params->buf, dentry->file_name_utf8,
+                                  NULL, 0);
+}
+
 /* 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,
@@ -1319,27 +1327,22 @@ static int wimfs_readdir(const char *path, void *buf, fuse_fill_dir_t filler,
 {
        struct wimlib_fd *fd = (struct wimlib_fd*)(uintptr_t)fi->fh;
        struct inode *inode;
-       struct dentry *child;
 
        if (!fd)
                return -EBADF;
 
        inode = fd->f_inode;
 
+       struct fill_params fill_params = {
+               .buf = buf,
+               .filler = filler,
+       };
+
        filler(buf, ".", NULL, 0);
        filler(buf, "..", NULL, 0);
 
-       child = inode->children;
-
-       if (!child)
-               return 0;
-
-       do {
-               if (filler(buf, child->file_name_utf8, NULL, 0))
-                       return 0;
-               child = child->next;
-       } while (child != inode->children);
-       return 0;
+       return for_dentry_in_rbtree(inode->children.rb_node,
+                                   dentry_fuse_fill, &fill_params);
 }
 
 
@@ -1443,7 +1446,7 @@ static int wimfs_rename(const char *from, const char *to)
                         * directory */
                        if (!dentry_is_directory(dst))
                                return -ENOTDIR;
-                       if (dst->d_inode->children != NULL)
+                       if (inode_has_children(dst->d_inode))
                                return -ENOTEMPTY;
                }
                parent_of_dst = dst->parent;
@@ -1696,6 +1699,8 @@ static int wimfs_utime(const char *path, struct utimbuf *times)
 {
        struct dentry *dentry;
        struct inode *inode;
+       WIMStruct *w = wimfs_get_WIMStruct();
+
        dentry = get_dentry(w, path);
        if (!dentry)
                return -ENOENT;
@@ -1919,7 +1924,11 @@ WIMLIBAPI int wimlib_mount(WIMStruct *wim, int image, const char *dir,
        ctx.wim = wim;
        ctx.mount_flags = flags;
 
+       DEBUG("Calling fuse_main()");
+
        ret = fuse_main(argc, argv, &wimfs_operations, &ctx);
+
+       DEBUG("Returned from fuse_main() (ret = %d)", ret);
        if (ret)
                ret = WIMLIB_ERR_FUSE;
 out_free_dir_copy: