]> wimlib.net Git - wimlib/blobdiff - src/ntfs-capture.c
mount_image.c: Use fuse_context.umask only when available
[wimlib] / src / ntfs-capture.c
index 9d501aa30b7a98ccee94a757cbf6bb1c1586f982..9896cdcca70292aa10081d0f9457e4d6cfb9d264 100644 (file)
@@ -204,17 +204,23 @@ capture_ntfs_streams(struct wim_inode *inode,
                        ntfs_loc = NULL;
                } else {
                        ntfs_loc = CALLOC(1, sizeof(*ntfs_loc));
-                       if (!ntfs_loc)
+                       if (!ntfs_loc) {
+                               ret = WIMLIB_ERR_NOMEM;
                                goto out_put_actx;
+                       }
                        ntfs_loc->ntfs_vol = vol;
                        ntfs_loc->path = MALLOC(path_len + 1);
-                       if (!ntfs_loc->path)
+                       if (!ntfs_loc->path) {
+                               ret = WIMLIB_ERR_NOMEM;
                                goto out_free_ntfs_loc;
+                       }
                        memcpy(ntfs_loc->path, path, path_len + 1);
                        if (name_length) {
                                ntfs_loc->stream_name = MALLOC(name_length * 2);
-                               if (!ntfs_loc->stream_name)
+                               if (!ntfs_loc->stream_name) {
+                                       ret = WIMLIB_ERR_NOMEM;
                                        goto out_free_ntfs_loc;
+                               }
                                memcpy(ntfs_loc->stream_name,
                                       attr_record_name(actx->attr),
                                       actx->attr->name_length * 2);
@@ -222,8 +228,10 @@ capture_ntfs_streams(struct wim_inode *inode,
                        }
 
                        lte = new_lookup_table_entry();
-                       if (!lte)
+                       if (!lte) {
+                               ret = WIMLIB_ERR_NOMEM;
                                goto out_free_ntfs_loc;
+                       }
                        lte->resource_location = RESOURCE_IN_NTFS_VOLUME;
                        lte->ntfs_loc = ntfs_loc;
                        ntfs_loc = NULL;
@@ -264,8 +272,10 @@ capture_ntfs_streams(struct wim_inode *inode,
                        new_ads_entry = inode_add_ads_utf16le(inode,
                                                              attr_record_name(actx->attr),
                                                              name_length * 2);
-                       if (!new_ads_entry)
+                       if (!new_ads_entry) {
+                               ret = WIMLIB_ERR_NOMEM;
                                goto out_free_lte;
+                       }
                        wimlib_assert(new_ads_entry->stream_name_nbytes == name_length * 2);
                        stream_id = new_ads_entry->stream_id;
                        new_ads_entry->lte = lte;
@@ -431,14 +441,9 @@ struct readdir_ctx {
        ntfs_inode *dir_ni;
        char *path;
        size_t path_len;
-       struct wim_lookup_table *lookup_table;
-       struct wim_inode_table *inode_table;
-       struct sd_set *sd_set;
        struct dos_name_map *dos_name_map;
-       const struct wimlib_capture_config *config;
        ntfs_volume *vol;
-       int add_image_flags;
-       wimlib_progress_func_t progress_func;
+       struct add_image_params *params;
 };
 
 static int
@@ -448,13 +453,8 @@ build_dentry_tree_ntfs_recursive(struct wim_dentry **root_p,
                                 char *path,
                                 size_t path_len,
                                 int name_type,
-                                struct wim_lookup_table *lookup_table,
-                                struct wim_inode_table *inode_table,
-                                struct sd_set *sd_set,
-                                const struct wimlib_capture_config *config,
                                 ntfs_volume *ntfs_vol,
-                                int add_image_flags,
-                                wimlib_progress_func_t progress_func);
+                                struct add_image_params *params);
 
 static int
 wim_ntfs_capture_filldir(void *dirent, const ntfschar *name,
@@ -513,12 +513,7 @@ wim_ntfs_capture_filldir(void *dirent, const ntfschar *name,
        child = NULL;
        ret = build_dentry_tree_ntfs_recursive(&child, ctx->dir_ni,
                                               ni, ctx->path, path_len, name_type,
-                                              ctx->lookup_table,
-                                              ctx->inode_table,
-                                              ctx->sd_set,
-                                              ctx->config, ctx->vol,
-                                              ctx->add_image_flags,
-                                              ctx->progress_func);
+                                              ctx->vol, ctx->params);
        if (child)
                dentry_add_child(ctx->parent, child);
        ntfs_inode_close(ni);
@@ -539,13 +534,8 @@ build_dentry_tree_ntfs_recursive(struct wim_dentry **root_ret,
                                 char *path,
                                 size_t path_len,
                                 int name_type,
-                                struct wim_lookup_table *lookup_table,
-                                struct wim_inode_table *inode_table,
-                                struct sd_set *sd_set,
-                                const struct wimlib_capture_config *config,
                                 ntfs_volume *vol,
-                                int add_image_flags,
-                                wimlib_progress_func_t progress_func)
+                                struct add_image_params *params)
 {
        u32 attributes;
        int ret;
@@ -553,16 +543,16 @@ build_dentry_tree_ntfs_recursive(struct wim_dentry **root_ret,
        struct wim_inode *inode;
        ATTR_TYPES stream_type;
 
-       if (exclude_path(path, path_len, config, false)) {
+       if (exclude_path(path, path_len, params->config, false)) {
                /* Exclude a file or directory tree based on the capture
                 * configuration file */
-               if ((add_image_flags & WIMLIB_ADD_IMAGE_FLAG_EXCLUDE_VERBOSE)
-                   && progress_func)
+               if ((params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_EXCLUDE_VERBOSE)
+                   && params->progress_func)
                {
                        union wimlib_progress_info info;
                        info.scan.cur_path = path;
                        info.scan.excluded = true;
-                       progress_func(WIMLIB_PROGRESS_MSG_SCAN_DENTRY, &info);
+                       params->progress_func(WIMLIB_PROGRESS_MSG_SCAN_DENTRY, &info);
                }
                root = NULL;
                ret = 0;
@@ -582,21 +572,19 @@ build_dentry_tree_ntfs_recursive(struct wim_dentry **root_ret,
                return WIMLIB_ERR_NTFS_3G;
        }
 
-       if ((add_image_flags & WIMLIB_ADD_IMAGE_FLAG_VERBOSE)
-           && progress_func)
+       if ((params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_VERBOSE)
+           && params->progress_func)
        {
                union wimlib_progress_info info;
                info.scan.cur_path = path;
                info.scan.excluded = false;
-               progress_func(WIMLIB_PROGRESS_MSG_SCAN_DENTRY, &info);
+               params->progress_func(WIMLIB_PROGRESS_MSG_SCAN_DENTRY, &info);
        }
 
        /* Create a WIM dentry with an associated inode, which may be shared */
-       ret = inode_table_new_dentry(inode_table,
+       ret = inode_table_new_dentry(params->inode_table,
                                     path_basename_with_len(path, path_len),
-                                    ni->mft_no,
-                                    0,
-                                    &root);
+                                    ni->mft_no, 0, false, &root);
        if (ret)
                return ret;
 
@@ -625,8 +613,8 @@ build_dentry_tree_ntfs_recursive(struct wim_dentry **root_ret,
         * - Directories: capture any named data streams
         * - Reparse points: capture reparse data only
         */
-       ret = capture_ntfs_streams(inode, ni, path, path_len, lookup_table,
-                                  vol, stream_type);
+       ret = capture_ntfs_streams(inode, ni, path, path_len,
+                                  params->lookup_table, vol, stream_type);
        if (ret)
                goto out;
 
@@ -640,14 +628,9 @@ build_dentry_tree_ntfs_recursive(struct wim_dentry **root_ret,
                        .dir_ni          = ni,
                        .path            = path,
                        .path_len        = path_len,
-                       .lookup_table    = lookup_table,
-                       .inode_table     = inode_table,
-                       .sd_set          = sd_set,
                        .dos_name_map    = &dos_name_map,
-                       .config          = config,
                        .vol             = vol,
-                       .add_image_flags = add_image_flags,
-                       .progress_func   = progress_func,
+                       .params          = params,
                };
                ret = ntfs_readdir(ni, &pos, &ctx, wim_ntfs_capture_filldir);
                if (ret) {
@@ -662,7 +645,7 @@ build_dentry_tree_ntfs_recursive(struct wim_dentry **root_ret,
                        goto out;
        }
 
-       if (!(add_image_flags & WIMLIB_ADD_IMAGE_FLAG_NO_ACLS)) {
+       if (!(params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_NO_ACLS)) {
                /* Get security descriptor */
                char _sd[1];
                char *sd = _sd;
@@ -676,7 +659,8 @@ build_dentry_tree_ntfs_recursive(struct wim_dentry **root_ret,
                                                         ni, dir_ni, sd, ret);
                }
                if (ret > 0) {
-                       inode->i_security_id = sd_set_add_sd(sd_set, sd, ret);
+                       inode->i_security_id = sd_set_add_sd(params->sd_set,
+                                                            sd, ret);
                        if (inode->i_security_id == -1) {
                                ERROR("Out of memory");
                                ret = WIMLIB_ERR_NOMEM;
@@ -698,7 +682,7 @@ out:
        if (ret == 0)
                *root_ret = root;
        else
-               free_dentry_tree(root, lookup_table);
+               free_dentry_tree(root, params->lookup_table);
        return ret;
 }
 
@@ -716,13 +700,7 @@ do_ntfs_umount(struct _ntfs_volume *vol)
 int
 build_dentry_tree_ntfs(struct wim_dentry **root_p,
                       const char *device,
-                      struct wim_lookup_table *lookup_table,
-                      struct wim_inode_table *inode_table,
-                      struct sd_set *sd_set,
-                      const struct wimlib_capture_config *config,
-                      int add_image_flags,
-                      wimlib_progress_func_t progress_func,
-                      void *extra_arg)
+                      struct add_image_params *params)
 {
        ntfs_volume *vol;
        ntfs_inode *root_ni;
@@ -770,12 +748,7 @@ build_dentry_tree_ntfs(struct wim_dentry **root_p,
        path[0] = '/';
        path[1] = '\0';
        ret = build_dentry_tree_ntfs_recursive(root_p, NULL, root_ni, path, 1,
-                                              FILE_NAME_POSIX, lookup_table,
-                                              inode_table,
-                                              sd_set,
-                                              config, vol,
-                                              add_image_flags,
-                                              progress_func);
+                                              FILE_NAME_POSIX, vol, params);
 out_cleanup:
        FREE(path);
        ntfs_inode_close(root_ni);
@@ -794,7 +767,7 @@ out:
        } else {
                /* We need to leave the NTFS volume mounted so that we can read
                 * the NTFS files again when we are actually writing the WIM */
-               *(ntfs_volume**)extra_arg = vol;
+               *(ntfs_volume**)params->extra_arg = vol;
        }
        return ret;
 }