X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fntfs-apply.c;h=214ac2d617688a8049852f4dcf8bbd5d71930969;hp=32d981082da0a9e0edb4754904fe09f62561f296;hb=89ef4adba6571c53200b0c72852784506670b596;hpb=b61e9afca45dacd9baefbcfa17dbea2af75259b8 diff --git a/src/ntfs-apply.c b/src/ntfs-apply.c index 32d98108..214ac2d6 100644 --- a/src/ntfs-apply.c +++ b/src/ntfs-apply.c @@ -27,18 +27,14 @@ #include "config.h" -#ifdef WITH_NTFS_3G + #include #include -#endif #include "wimlib_internal.h" - - -#ifdef WITH_NTFS_3G #include "dentry.h" #include "lookup_table.h" -#include "io.h" +#include "buffer_io.h" #include #include #include @@ -48,11 +44,17 @@ #include #include -struct ntfs_apply_args { - ntfs_volume *vol; - int extract_flags; - WIMStruct *w; -}; +static int extract_wim_chunk_to_ntfs_attr(const u8 *buf, size_t len, + u64 offset, void *arg) +{ + ntfs_attr *na = arg; + if (ntfs_attr_pwrite(na, offset, len, buf) == len) { + return 0; + } else { + ERROR_WITH_ERRNO("Error extracting WIM resource to NTFS attribute"); + return WIMLIB_ERR_WRITE; + } +} /* * Extracts a WIM resource to a NTFS attribute. @@ -61,49 +63,19 @@ static int extract_wim_resource_to_ntfs_attr(const struct lookup_table_entry *lte, ntfs_attr *na) { - u64 bytes_remaining = wim_resource_size(lte); - u8 buf[min(WIM_CHUNK_SIZE, bytes_remaining)]; - u64 offset = 0; - int ret = 0; - u8 hash[SHA1_HASH_SIZE]; - - SHA_CTX ctx; - sha1_init(&ctx); - - while (bytes_remaining) { - u64 to_read = min(bytes_remaining, WIM_CHUNK_SIZE); - ret = read_wim_resource(lte, buf, to_read, offset, 0); - if (ret != 0) - break; - sha1_update(&ctx, buf, to_read); - if (ntfs_attr_pwrite(na, offset, to_read, buf) != to_read) { - ERROR_WITH_ERRNO("Error extracting WIM resource"); - return WIMLIB_ERR_WRITE; - } - bytes_remaining -= to_read; - offset += to_read; - } - sha1_final(hash, &ctx); - if (!hashes_equal(hash, lte->hash)) { - ERROR("Invalid checksum on a WIM resource " - "(detected when extracting to NTFS stream)"); - ERROR("The following WIM resource is invalid:"); - print_lookup_table_entry(lte); - return WIMLIB_ERR_INVALID_RESOURCE_HASH; - } - return 0; + return extract_wim_resource(lte, wim_resource_size(lte), + extract_wim_chunk_to_ntfs_attr, na); } /* Writes the data streams to a NTFS file * * @ni: The NTFS inode for the file. * @inode: The WIM dentry that has an inode containing the streams. - * @w: The WIMStruct for the WIM containing the image we are applying. * * Returns 0 on success, nonzero on failure. */ static int write_ntfs_data_streams(ntfs_inode *ni, const struct dentry *dentry, - WIMStruct *w) + union wimlib_progress_info *progress_info) { int ret = 0; unsigned stream_idx = 0; @@ -118,9 +90,8 @@ static int write_ntfs_data_streams(ntfs_inode *ni, const struct dentry *dentry, while (1) { struct lookup_table_entry *lte; - ntfs_attr *na; - lte = inode_stream_lte(inode, stream_idx, w->lookup_table); + lte = inode_stream_lte_resolved(inode, stream_idx); if (stream_name_len) { /* Create an empty named stream. */ @@ -140,6 +111,8 @@ static int write_ntfs_data_streams(ntfs_inode *ni, const struct dentry *dentry, * Otherwise, we must open the attribute and extract the data. * */ if (lte) { + ntfs_attr *na; + na = ntfs_attr_open(ni, AT_DATA, stream_name, stream_name_len); if (!na) { ERROR_WITH_ERRNO("Failed to open a data stream of " @@ -148,10 +121,17 @@ static int write_ntfs_data_streams(ntfs_inode *ni, const struct dentry *dentry, ret = WIMLIB_ERR_NTFS_3G; break; } + ret = ntfs_attr_truncate_solid(na, wim_resource_size(lte)); + if (ret != 0) { + ntfs_attr_close(na); + break; + } + ret = extract_wim_resource_to_ntfs_attr(lte, na); + ntfs_attr_close(na); if (ret != 0) break; - ntfs_attr_close(na); + progress_info->extract.completed_bytes += wim_resource_size(lte); } if (stream_idx == inode->num_ads) break; @@ -162,6 +142,32 @@ static int write_ntfs_data_streams(ntfs_inode *ni, const struct dentry *dentry, return ret; } +/* Open the NTFS inode that corresponds to the parent of a WIM dentry. */ +static ntfs_inode *dentry_open_parent_ni(const struct dentry *dentry, + ntfs_volume *vol) +{ + char *p; + const char *dir_name; + ntfs_inode *dir_ni; + char orig; + + p = dentry->full_path_utf8 + dentry->full_path_utf8_len; + do { + p--; + } while (*p != '/'); + + orig = *p; + *p = '\0'; + dir_name = dentry->full_path_utf8; + dir_ni = ntfs_pathname_to_inode(vol, NULL, dir_name); + if (!dir_ni) { + ERROR_WITH_ERRNO("Could not find NTFS inode for `%s'", + dir_name); + } + *p = orig; + return dir_ni; +} + /* * Makes a NTFS hard link * @@ -171,10 +177,10 @@ static int write_ntfs_data_streams(ntfs_inode *ni, const struct dentry *dentry, * * Return 0 on success, nonzero on failure. */ -static int wim_apply_hardlink_ntfs(const struct dentry *from_dentry, - const struct inode *inode, - ntfs_inode *dir_ni, - ntfs_inode **to_ni_ret) +static int apply_hardlink_ntfs(const struct dentry *from_dentry, + const struct inode *inode, + ntfs_inode *dir_ni, + ntfs_inode **to_ni_ret) { int ret; char *p; @@ -184,9 +190,6 @@ static int wim_apply_hardlink_ntfs(const struct dentry *from_dentry, ntfs_inode *to_ni; ntfs_volume *vol; - wimlib_assert(dentry_is_regular_file(from_dentry) - && inode_is_regular_file(inode)); - if (ntfs_inode_close(dir_ni) != 0) { ERROR_WITH_ERRNO("Error closing directory"); return WIMLIB_ERR_NTFS_3G; @@ -203,23 +206,12 @@ static int wim_apply_hardlink_ntfs(const struct dentry *from_dentry, inode->extracted_file); return WIMLIB_ERR_NTFS_3G; } - p = from_dentry->full_path_utf8 + from_dentry->full_path_utf8_len; - do { - p--; - } while (*p != '/'); - orig = *p; - *p = '\0'; - dir_name = from_dentry->full_path_utf8; + *to_ni_ret = to_ni; - dir_ni = ntfs_pathname_to_inode(vol, NULL, dir_name); - if (!dir_ni) { - ERROR_WITH_ERRNO("Could not find NTFS inode for `%s'", - from_dentry->full_path_utf8); - *p = orig; + dir_ni = dentry_open_parent_ni(from_dentry, vol); + if (!dir_ni) return WIMLIB_ERR_NTFS_3G; - } - *p = orig; ret = ntfs_link(to_ni, dir_ni, (ntfschar*)from_dentry->file_name, @@ -230,7 +222,6 @@ static int wim_apply_hardlink_ntfs(const struct dentry *from_dentry, inode->extracted_file); ret = WIMLIB_ERR_NTFS_3G; } - *to_ni_ret = to_ni; return ret; } @@ -281,14 +272,12 @@ apply_file_attributes_and_security_data(ntfs_inode *ni, } static int apply_reparse_data(ntfs_inode *ni, const struct dentry *dentry, - const WIMStruct *w) + union wimlib_progress_info *progress_info) { struct lookup_table_entry *lte; int ret = 0; - wimlib_assert(dentry->d_inode->attributes & FILE_ATTRIBUTE_REPARSE_POINT); - - lte = inode_unnamed_lte(dentry->d_inode, w->lookup_table); + lte = inode_unnamed_lte_resolved(dentry->d_inode); DEBUG("Applying reparse data to `%s'", dentry->full_path_utf8); @@ -321,11 +310,12 @@ static int apply_reparse_data(ntfs_inode *ni, const struct dentry *dentry, dentry->full_path_utf8); return WIMLIB_ERR_NTFS_3G; } + progress_info->extract.completed_bytes += wim_resource_size(lte); return 0; } -static int do_wim_apply_dentry_ntfs(struct dentry *dentry, ntfs_inode *dir_ni, - WIMStruct *w); +static int do_apply_dentry_ntfs(struct dentry *dentry, ntfs_inode *dir_ni, + struct apply_args *args); /* * If @dentry is part of a hard link group, search for hard-linked dentries in @@ -337,7 +327,7 @@ static int do_wim_apply_dentry_ntfs(struct dentry *dentry, ntfs_inode *dir_ni, */ static int preapply_dentry_with_dos_name(struct dentry *dentry, ntfs_inode **dir_ni_p, - WIMStruct *w) + struct apply_args *args) { struct dentry *other; struct dentry *dentry_with_dos_name; @@ -357,7 +347,7 @@ static int preapply_dentry_with_dos_name(struct dentry *dentry, } } /* If there's a dentry with a DOS name, extract it first */ - if (dentry_with_dos_name && !dentry_is_extracted(dentry)) { + if (dentry_with_dos_name && !dentry_with_dos_name->is_extracted) { char *p; const char *dir_name; char orig; @@ -366,26 +356,14 @@ static int preapply_dentry_with_dos_name(struct dentry *dentry, DEBUG("pre-applying DOS name `%s'", dentry_with_dos_name->full_path_utf8); - ret = do_wim_apply_dentry_ntfs(dentry_with_dos_name, - *dir_ni_p, w); + ret = do_apply_dentry_ntfs(dentry_with_dos_name, + *dir_ni_p, args); if (ret != 0) return ret; - p = dentry->full_path_utf8 + dentry->full_path_utf8_len; - do { - p--; - } while (*p != '/'); - - orig = *p; - *p = '\0'; - dir_name = dentry->full_path_utf8; - - *dir_ni_p = ntfs_pathname_to_inode(vol, NULL, dir_name); - *p = orig; - if (!*dir_ni_p) { - ERROR_WITH_ERRNO("Could not find NTFS inode for `%s'", - dir_name); + + *dir_ni_p = dentry_open_parent_ni(dentry, vol); + if (!*dir_ni_p) return WIMLIB_ERR_NTFS_3G; - } } return 0; } @@ -395,12 +373,11 @@ static int preapply_dentry_with_dos_name(struct dentry *dentry, * * @dentry: The WIM dentry to apply * @dir_ni: The NTFS inode for the parent directory - * @w: The WIMStruct for the WIM containing the image we are applying. * * @return: 0 on success; nonzero on failure. */ -static int do_wim_apply_dentry_ntfs(struct dentry *dentry, ntfs_inode *dir_ni, - WIMStruct *w) +static int do_apply_dentry_ntfs(struct dentry *dentry, ntfs_inode *dir_ni, + struct apply_args *args) { int ret = 0; mode_t type; @@ -408,16 +385,18 @@ static int do_wim_apply_dentry_ntfs(struct dentry *dentry, ntfs_inode *dir_ni, bool is_hardlink = false; ntfs_volume *vol = dir_ni->vol; struct inode *inode = dentry->d_inode; - dentry->is_extracted = true; + dentry->is_extracted = 1; if (inode->attributes & FILE_ATTRIBUTE_DIRECTORY) { type = S_IFDIR; } else { - /* Apply hard-linked directory in same directory with DOS name - * (if there is one) before this dentry */ + /* If this dentry is hard-linked to any other dentries in the + * same directory, make sure to apply the one (if any) with a + * DOS name first. Otherwise, NTFS-3g might not assign the file + * names correctly. */ if (dentry->short_name_len == 0) { ret = preapply_dentry_with_dos_name(dentry, - &dir_ni, w); + &dir_ni, args); if (ret != 0) return ret; } @@ -429,8 +408,8 @@ static int do_wim_apply_dentry_ntfs(struct dentry *dentry, ntfs_inode *dir_ni, * group. We can make a hard link instead of extracting * the file data. */ if (inode->extracted_file) { - ret = wim_apply_hardlink_ntfs(dentry, inode, - dir_ni, &ni); + ret = apply_hardlink_ntfs(dentry, inode, + dir_ni, &ni); is_hardlink = true; if (ret) goto out_close_dir_ni; @@ -467,18 +446,19 @@ static int do_wim_apply_dentry_ntfs(struct dentry *dentry, ntfs_inode *dir_ni, * */ if (!(inode->attributes & (FILE_ATTRIBUTE_REPARSE_POINT | FILE_ATTRIBUTE_DIRECTORY))) { - ret = write_ntfs_data_streams(ni, dentry, w); + ret = write_ntfs_data_streams(ni, dentry, &args->progress); if (ret != 0) goto out_close_dir_ni; } - ret = apply_file_attributes_and_security_data(ni, dir_ni, dentry, w); + ret = apply_file_attributes_and_security_data(ni, dir_ni, + dentry, args->w); if (ret != 0) goto out_close_dir_ni; if (inode->attributes & FILE_ATTR_REPARSE_POINT) { - ret = apply_reparse_data(ni, dentry, w); + ret = apply_reparse_data(ni, dentry, &args->progress); if (ret != 0) goto out_close_dir_ni; } @@ -489,14 +469,12 @@ out_set_dos_name: char *short_name_utf8; size_t short_name_utf8_len; - short_name_utf8 = utf16_to_utf8(dentry->short_name, - dentry->short_name_len, - &short_name_utf8_len); - if (!short_name_utf8) { - ERROR("Out of memory"); - ret = WIMLIB_ERR_NOMEM; + ret = utf16_to_utf8(dentry->short_name, + dentry->short_name_len, + &short_name_utf8, + &short_name_utf8_len); + if (ret != 0) goto out_close_dir_ni; - } if (is_hardlink) { char *p; @@ -510,34 +488,29 @@ out_set_dos_name: if (ret == 0) ret = WIMLIB_ERR_NTFS_3G; ERROR_WITH_ERRNO("Failed to close directory inode"); + goto out_close_ni; } if (ntfs_inode_close(ni) != 0) { if (ret == 0) ret = WIMLIB_ERR_NTFS_3G; ERROR_WITH_ERRNO("Failed to close hard link target inode"); + goto out; } - p = dentry->full_path_utf8 + dentry->full_path_utf8_len; - do { - p--; - } while (*p != '/'); - - orig = *p; - *p = '\0'; - dir_name = dentry->full_path_utf8; - dir_ni = ntfs_pathname_to_inode(vol, NULL, dir_name); - *p = orig; + dir_ni = dentry_open_parent_ni(dentry, vol); if (!dir_ni) { - ERROR_WITH_ERRNO("Could not find NTFS inode for `%s'", - dir_name); - return WIMLIB_ERR_NTFS_3G; + ret = WIMLIB_ERR_NTFS_3G; + goto out; } + ni = ntfs_pathname_to_inode(vol, dir_ni, dentry->file_name_utf8); if (!ni) { ERROR_WITH_ERRNO("Could not find NTFS inode for `%s'", - dir_name); - return WIMLIB_ERR_NTFS_3G; + dentry->full_path_utf8); + ntfs_inode_close(dir_ni); + ret = WIMLIB_ERR_NTFS_3G; + goto out_close_dir_ni; } } @@ -557,22 +530,33 @@ out_set_dos_name: } out_close_dir_ni: + if (ni && dir_ni) { + if (ntfs_inode_close_in_dir(ni, dir_ni) != 0) { + ni = NULL; + if (ret == 0) + ret = WIMLIB_ERR_NTFS_3G; + ERROR_WITH_ERRNO("Failed to close inode for `%s'", + dentry->full_path_utf8); + } + } if (ntfs_inode_close(dir_ni) != 0) { if (ret == 0) ret = WIMLIB_ERR_NTFS_3G; ERROR_WITH_ERRNO("Failed to close directory inode"); } +out_close_ni: if (ni && ntfs_inode_close(ni) != 0) { if (ret == 0) ret = WIMLIB_ERR_NTFS_3G; - ERROR_WITH_ERRNO("Failed to close inode"); + ERROR_WITH_ERRNO("Failed to close inode for `%s'", + dentry->full_path_utf8); } +out: return ret; } -static int wim_apply_root_dentry_ntfs(const struct dentry *dentry, - ntfs_volume *vol, - const WIMStruct *w) +static int apply_root_dentry_ntfs(const struct dentry *dentry, + ntfs_volume *vol, const WIMStruct *w) { ntfs_inode *ni; int ret = 0; @@ -593,59 +577,32 @@ static int wim_apply_root_dentry_ntfs(const struct dentry *dentry, } /* Applies a WIM dentry to the NTFS volume */ -static int wim_apply_dentry_ntfs(struct dentry *dentry, void *arg) +int apply_dentry_ntfs(struct dentry *dentry, void *arg) { - struct ntfs_apply_args *args = arg; - ntfs_volume *vol = args->vol; - int extract_flags = args->extract_flags; - WIMStruct *w = args->w; + struct apply_args *args = arg; + ntfs_volume *vol = args->vol; + WIMStruct *w = args->w; ntfs_inode *dir_ni; - char *p; - char orig; - const char *dir_name; - - if (dentry_is_extracted(dentry)) - return 0; - - wimlib_assert(dentry->full_path_utf8); - - DEBUG("Applying dentry `%s' to NTFS", dentry->full_path_utf8); - - if (extract_flags & WIMLIB_EXTRACT_FLAG_VERBOSE) - puts(dentry->full_path_utf8); if (dentry_is_root(dentry)) - return wim_apply_root_dentry_ntfs(dentry, vol, w); - - p = dentry->full_path_utf8 + dentry->full_path_utf8_len; - do { - p--; - } while (*p != '/'); - - orig = *p; - *p = '\0'; - dir_name = dentry->full_path_utf8; + return apply_root_dentry_ntfs(dentry, vol, w); - dir_ni = ntfs_pathname_to_inode(vol, NULL, dir_name); - *p = orig; - if (!dir_ni) { - ERROR_WITH_ERRNO("Could not find NTFS inode for `%s'", - dir_name); + dir_ni = dentry_open_parent_ni(dentry, vol); + if (dir_ni) + return do_apply_dentry_ntfs(dentry, dir_ni, arg); + else return WIMLIB_ERR_NTFS_3G; - } - return do_wim_apply_dentry_ntfs(dentry, dir_ni, w); } -static int wim_apply_dentry_timestamps(struct dentry *dentry, void *arg) +int apply_dentry_timestamps_ntfs(struct dentry *dentry, void *arg) { - struct ntfs_apply_args *args = arg; - ntfs_volume *vol = args->vol; + struct apply_args *args = arg; + ntfs_volume *vol = args->vol; u8 *p; u8 buf[24]; ntfs_inode *ni; int ret = 0; - DEBUG("Setting timestamps on `%s'", dentry->full_path_utf8); ni = ntfs_pathname_to_inode(vol, NULL, dentry->full_path_utf8); @@ -674,126 +631,3 @@ static int wim_apply_dentry_timestamps(struct dentry *dentry, void *arg) } return ret; } - -static int dentry_set_unextracted(struct dentry *dentry, void *ignore) -{ - dentry->is_extracted = false; - return 0; -} - -static int do_wim_apply_image_ntfs(WIMStruct *w, const char *device, int extract_flags) -{ - ntfs_volume *vol; - int ret; - struct dentry *root; - struct ntfs_apply_args args; - - DEBUG("Mounting NTFS volume `%s'", device); - vol = ntfs_mount(device, 0); - if (!vol) { - ERROR_WITH_ERRNO("Failed to mount NTFS volume `%s'", device); - return WIMLIB_ERR_NTFS_3G; - } - args.vol = vol; - args.extract_flags = extract_flags; - args.w = w; - root = wim_root_dentry(w); - - for_dentry_in_tree(root, dentry_set_unextracted, NULL); - ret = for_dentry_in_tree(root, wim_apply_dentry_ntfs, &args); - if (ret != 0) - goto out; - - if (extract_flags & WIMLIB_EXTRACT_FLAG_VERBOSE) - printf("Setting timestamps of extracted files on NTFS " - "volume `%s'\n", device); - ret = for_dentry_in_tree_depth(root, wim_apply_dentry_timestamps, - &args); - - if (ret == 0 && (extract_flags & WIMLIB_EXTRACT_FLAG_VERBOSE)) - printf("Finished applying image %d of %s to NTFS " - "volume `%s'\n", - w->current_image, - w->filename ? w->filename : "WIM", - device); -out: - DEBUG("Unmounting NTFS volume `%s'", device); - if (ntfs_umount(vol, FALSE) != 0) { - ERROR_WITH_ERRNO("Failed to unmount NTFS volume `%s'", device); - if (ret == 0) - ret = WIMLIB_ERR_NTFS_3G; - } - return ret; -} - - -/* - * API entry point for applying a WIM image to a NTFS volume. - * - * Please note that this is a NTFS *volume* and not a directory. The intention - * is that the volume contain an empty filesystem, and the WIM image contain a - * full filesystem to be applied to the volume. - */ -WIMLIBAPI int wimlib_apply_image_to_ntfs_volume(WIMStruct *w, int image, - const char *device, int flags, - WIMStruct **additional_swms, - unsigned num_additional_swms) -{ - struct lookup_table *joined_tab, *w_tab_save; - int ret; - - DEBUG("w->filename = %s, image = %d, device = %s, flags = 0x%x, " - "num_additional_swms = %u", - w->filename, image, device, flags, num_additional_swms); - - if (!w || !device) - return WIMLIB_ERR_INVALID_PARAM; - if (image == WIM_ALL_IMAGES) { - ERROR("Can only apply a single image when applying " - "directly to a NTFS volume"); - return WIMLIB_ERR_INVALID_PARAM; - } - if (flags & (WIMLIB_EXTRACT_FLAG_SYMLINK | WIMLIB_EXTRACT_FLAG_HARDLINK)) { - ERROR("Cannot specify symlink or hardlink flags when applying "); - ERROR("directly to a NTFS volume"); - return WIMLIB_ERR_INVALID_PARAM; - } - - ret = verify_swm_set(w, additional_swms, num_additional_swms); - if (ret != 0) - return ret; - - if (num_additional_swms) { - ret = new_joined_lookup_table(w, additional_swms, - num_additional_swms, &joined_tab); - if (ret != 0) - return ret; - w_tab_save = w->lookup_table; - w->lookup_table = joined_tab; - } - - ret = select_wim_image(w, image); - if (ret != 0) - goto out; - - ret = do_wim_apply_image_ntfs(w, device, flags); - -out: - if (num_additional_swms) { - free_lookup_table(w->lookup_table); - w->lookup_table = w_tab_save; - } - return ret; -} - -#else /* WITH_NTFS_3G */ -WIMLIBAPI int wimlib_apply_image_to_ntfs_volume(WIMStruct *w, int image, - const char *device, int flags, - WIMStruct **additional_swms, - unsigned num_additional_swms) -{ - 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 /* WITH_NTFS_3G */