X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fntfs-apply.c;h=d28dbfac4acdec4379af614cf7345f026f1aad70;hp=6243a914dfd3a2a6e05527e5afb66c85b5d828c9;hb=a3bb2e86f2640f5d593d00250a627d3dcc9747a2;hpb=7be97dea91241da8a7ca5f0ff32f320e692fe53f diff --git a/src/ntfs-apply.c b/src/ntfs-apply.c index 6243a914..d28dbfac 100644 --- a/src/ntfs-apply.c +++ b/src/ntfs-apply.c @@ -1,432 +1,663 @@ /* * ntfs-apply.c * - * Apply a WIM image to a NTFS volume, restoring everything we can, including - * security data and alternate data streams. There should be no loss of - * information. + * Apply a WIM image to a NTFS volume. Restore as much information as possible, + * including security data, file attributes, DOS names, and alternate data + * streams. */ /* - * Copyright (C) 2012 Eric Biggers + * Copyright (C) 2012, 2013 Eric Biggers * * This file is part of wimlib, a library for working with WIM files. * * wimlib is free software; you can redistribute it and/or modify it under the - * terms of the GNU Lesser General Public License as published by the Free - * Software Foundation; either version 2.1 of the License, or (at your option) + * terms of the GNU General Public License as published by the Free + * Software Foundation; either version 3 of the License, or (at your option) * any later version. * * wimlib is distributed in the hope that it will be useful, but WITHOUT ANY * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR - * A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more + * A PARTICULAR PURPOSE. See the GNU General Public License for more * details. * - * You should have received a copy of the GNU Lesser General Public License + * You should have received a copy of the GNU General Public License * along with wimlib; if not, see http://www.gnu.org/licenses/. */ + #include "config.h" -#include "wimlib_internal.h" +#include +#include -#ifdef WITH_NTFS_3G +#include "wimlib_internal.h" +#include "buffer_io.h" #include "dentry.h" #include "lookup_table.h" -#include -#include + #include -#include +#include /* security.h before xattrs.h */ #include -#include -#include -#include -#include +#include +#include +#include -struct ntfs_apply_args { - ntfs_volume *vol; - int extract_flags; - WIMStruct *w; -}; - -extern int _ntfs_set_file_security(ntfs_volume *vol, ntfs_inode *ni, - u32 selection, const char *attr); -extern int _ntfs_set_file_attributes(ntfs_inode *ni, s32 attrib); +static int +extract_wim_chunk_to_ntfs_attr(const void *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. */ -static int extract_resource_to_ntfs_attr(WIMStruct *w, const struct resource_entry *entry, - ntfs_attr *na) +static int +extract_wim_resource_to_ntfs_attr(const struct wim_lookup_table_entry *lte, + ntfs_attr *na) { - return 0; -#if 0 - u8 buf[min(entry->original_size, WIM_CHUNK_SIZE)]; - u64 num_chunks = (entry->original_size + WIM_CHUNK_SIZE - 1) / WIM_CHUNK_SIZE; - u64 n = WIM_CHUNK_SIZE; - int res_ctype = wim_resource_compression_type(w, entry); - u64 offset = 0; - for (u64 i = 0; i < num_chunks; i++) { - DEBUG("Write chunk %u of %u", i + 1, num_chunks); - int ret; - if (i == num_chunks - 1) { - n = entry->original_size % WIM_CHUNK_SIZE; - if (n == 0) { - n = WIM_CHUNK_SIZE; - } - } - - ret = read_resource(w->fp, entry->size, entry->original_size, - entry->offset, res_ctype, n, offset, buf); - if (ret != 0) - return ret; - - if (ntfs_attr_pwrite(na, offset, n, buf) != n) { - ERROR("Failed to write to NTFS data stream"); - return WIMLIB_ERR_WRITE; - } - offset += n; - } - return 0; -#endif + return extract_wim_resource(lte, wim_resource_size(lte), + extract_wim_chunk_to_ntfs_attr, na); } -/* Writes the data streams to a NTFS file +/* Writes the data streams of a WIM inode to the data attributes of a NTFS + * inode. + * + * @ni: The NTFS inode to which the streams are to be extracted. + * + * @dentry: The WIM dentry being extracted. The @d_inode member points to the + * corresponding WIM inode that contains the streams being extracted. + * The WIM dentry itself is only needed to provide a file path for + * better error messages. * - * @ni: The NTFS inode for the file. - * @dentry: The directory entry in the WIM file. - * @w: The WIMStruct for the WIM containing the image we are applying. + * @progress_info: Progress information for the image application. The number + * of extracted bytes will be incremented by the uncompressed + * size of each stream extracted. * * Returns 0 on success, nonzero on failure. */ -static int write_ntfs_data_streams(ntfs_inode *ni, const struct dentry *dentry, - WIMStruct *w) +static int +write_ntfs_data_streams(ntfs_inode *ni, const struct wim_dentry *dentry, + union wimlib_progress_info *progress_info) { - ntfs_attr *na; - struct lookup_table_entry *lte; - int ret; + int ret = 0; + unsigned stream_idx = 0; + ntfschar *stream_name = AT_UNNAMED; + u32 stream_name_nbytes = 0; + const struct wim_inode *inode = dentry->d_inode; + struct wim_lookup_table_entry *lte; + + DEBUG("Writing %u NTFS data stream%s for `%s'", + inode->i_num_ads + 1, + (inode->i_num_ads == 0 ? "" : "s"), + dentry->full_path); + + lte = inode->i_lte; + while (1) { + if (stream_name_nbytes) { + /* Skip special UNIX data entries (see documentation for + * WIMLIB_ADD_IMAGE_FLAG_UNIX_DATA) */ + if (stream_name_nbytes == WIMLIB_UNIX_DATA_TAG_UTF16LE_NBYTES + && !memcmp(stream_name, + WIMLIB_UNIX_DATA_TAG_UTF16LE, + WIMLIB_UNIX_DATA_TAG_UTF16LE_NBYTES)) + goto cont; + + /* Create an empty named stream. */ + ret = ntfs_attr_add(ni, AT_DATA, stream_name, + stream_name_nbytes / 2, NULL, 0); + if (ret != 0) { + ERROR_WITH_ERRNO("Failed to create named data " + "stream for extracted file " + "`%s'", + dentry->full_path); + ret = WIMLIB_ERR_NTFS_3G; + break; + + } + } + /* If there's no lookup table entry, it's an empty stream. + * Otherwise, open the attribute and extract the data. */ + if (lte) { + ntfs_attr *na; + + na = ntfs_attr_open(ni, AT_DATA, stream_name, + stream_name_nbytes / 2); + if (!na) { + ERROR_WITH_ERRNO("Failed to open a data stream of " + "extracted file `%s'", + dentry->full_path); + ret = WIMLIB_ERR_NTFS_3G; + break; + } - DEBUG("Writing NTFS data streams for `%s'", dentry->full_path_utf8); + /* The WIM lookup table entry provides the stream + * length, so the NTFS attribute should be resized to + * this length before starting to extract the data. */ + ret = ntfs_attr_truncate_solid(na, wim_resource_size(lte)); + if (ret != 0) { + ntfs_attr_close(na); + break; + } - wimlib_assert(dentry->num_ads == 0); + /* Actually extract the stream */ + ret = extract_wim_resource_to_ntfs_attr(lte, na); - lte = dentry_stream_lte(dentry, 0, w->lookup_table); - if (lte && lte->resource_entry.original_size != 0) { + /* Close the attribute */ + ntfs_attr_close(na); + if (ret != 0) + break; - na = ntfs_attr_open(ni, AT_DATA, AT_UNNAMED, 0); - if (!na) { - ERROR_WITH_ERRNO("Failed to open unnamed data stream of " - "extracted file `%s'", - dentry->full_path_utf8); - return WIMLIB_ERR_NTFS_3G; + /* Record the number of bytes of uncompressed data that + * have been extracted. */ + progress_info->extract.completed_bytes += wim_resource_size(lte); } - ret = extract_resource_to_ntfs_attr(w, <e->resource_entry, na); - if (ret != 0) - return ret; - ntfs_attr_close(na); + cont: + if (stream_idx == inode->i_num_ads) /* Has the last stream been extracted? */ + break; + + /* Get the name and lookup table entry for the next stream. */ + stream_name = inode->i_ads_entries[stream_idx].stream_name; + stream_name_nbytes = inode->i_ads_entries[stream_idx].stream_name_nbytes; + lte = inode->i_ads_entries[stream_idx].lte; + stream_idx++; } + return ret; +} - return 0; +/* Open the NTFS inode that corresponds to the parent of a WIM dentry. Returns + * the opened inode, or NULL on failure. */ +static ntfs_inode * +dentry_open_parent_ni(const struct wim_dentry *dentry, ntfs_volume *vol) +{ + char *p; + const char *dir_name; + ntfs_inode *dir_ni; + char orig; + + p = dentry->full_path + dentry->full_path_nbytes; + do { + p--; + } while (*p != '/'); + + orig = *p; + *p = '\0'; + dir_name = dentry->full_path; + 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 + * Makes a NTFS hard link. * - * It is named @from_dentry->file_name and is located under the directory - * specified by @dir_ni, and it is made to point to the previously extracted - * file located at @to_dentry->extracted_file. + * The hard link is named @from_dentry->file_name and is located under the + * directory specified by @dir_ni, and it is made to point to the previously + * extracted file located at @inode->i_extracted_file. * - * Return 0 on success, nonzero on failure. + * Or, in other words, this adds a new name @from_dentry->full_path to an + * existing NTFS inode which already has a name @inode->i_extracted_file. + * + * The new name is made in the POSIX namespace (this is the behavior of + * ntfs_link()). + * + * Return 0 on success, nonzero on failure. dir_ni is closed either way. */ -static int wim_apply_hardlink_ntfs(const struct dentry *from_dentry, - const struct dentry *to_dentry, - ntfs_inode *dir_ni) +static int +apply_ntfs_hardlink(const struct wim_dentry *from_dentry, + const struct wim_inode *inode, + ntfs_inode *dir_ni) { int ret; ntfs_inode *to_ni; + ntfs_volume *vol; + + vol = dir_ni->vol; + ret = ntfs_inode_close(dir_ni); + if (ret != 0) { + ERROR_WITH_ERRNO("Error closing directory"); + return WIMLIB_ERR_NTFS_3G; + } DEBUG("Extracting NTFS hard link `%s' => `%s'", - from_dentry->full_path_utf8, to_dentry->extracted_file); + from_dentry->full_path, inode->i_extracted_file); - to_ni = ntfs_pathname_to_inode(dir_ni->vol, NULL, - to_dentry->extracted_file); + to_ni = ntfs_pathname_to_inode(vol, NULL, inode->i_extracted_file); if (!to_ni) { ERROR_WITH_ERRNO("Could not find NTFS inode for `%s'", - to_dentry->extracted_file); + inode->i_extracted_file); return WIMLIB_ERR_NTFS_3G; } + + dir_ni = dentry_open_parent_ni(from_dentry, vol); + if (!dir_ni) { + ntfs_inode_close(to_ni); + return WIMLIB_ERR_NTFS_3G; + } + ret = ntfs_link(to_ni, dir_ni, - (ntfschar*)from_dentry->file_name, - from_dentry->file_name_len / 2); - if (ret != 0) { + from_dentry->file_name, + from_dentry->file_name_nbytes / 2); + ret |= ntfs_inode_close(dir_ni); + ret |= ntfs_inode_close(to_ni); + if (ret) { ERROR_WITH_ERRNO("Could not create hard link `%s' => `%s'", - from_dentry->full_path_utf8, - to_dentry->extracted_file); - ret = WIMLIB_ERR_NTFS_3G; - } - if (ntfs_inode_close(to_ni) != 0) { - ERROR_WITH_ERRNO("Failed to close NTFS inode for `%s'", - to_dentry->extracted_file); + from_dentry->full_path, + inode->i_extracted_file); ret = WIMLIB_ERR_NTFS_3G; } return ret; } -/* +/* Transfers file attributes and possibly a security descriptor from a WIM inode + * to a NTFS inode. + * + * @ni: The NTFS inode to apply the metadata to. + * @dir_ni: The NTFS inode for a directory containing @ni. + * @dentry: The WIM dentry whose inode contains the metadata to apply. + * @w: The WIMStruct for the WIM, through which the table of security + * descriptors can be accessed. + * + * Returns 0 on success, nonzero on failure. + */ +static int +apply_file_attributes_and_security_data(ntfs_inode *ni, + ntfs_inode *dir_ni, + const struct wim_dentry *dentry, + const WIMStruct *w, + int extract_flags) +{ + int ret; + struct SECURITY_CONTEXT ctx; + u32 attributes_le32; + const struct wim_inode *inode; + + inode = dentry->d_inode; + + DEBUG("Setting NTFS file attributes on `%s' to %#"PRIx32, + dentry->full_path, inode->i_attributes); + + attributes_le32 = cpu_to_le32(inode->i_attributes); + memset(&ctx, 0, sizeof(ctx)); + ctx.vol = ni->vol; + ret = ntfs_xattr_system_setxattr(&ctx, XATTR_NTFS_ATTRIB, + ni, dir_ni, + (const char*)&attributes_le32, + sizeof(u32), 0); + if (ret != 0) { + ERROR("Failed to set NTFS file attributes on `%s'", + dentry->full_path); + return WIMLIB_ERR_NTFS_3G; + } + if (inode->i_security_id != -1 && + !(extract_flags & WIMLIB_EXTRACT_FLAG_NOACLS)) + { + const char *desc; + const struct wim_security_data *sd; + + sd = wim_const_security_data(w); + wimlib_assert(inode->i_security_id < sd->num_entries); + desc = (const char *)sd->descriptors[inode->i_security_id]; + DEBUG("Applying security descriptor %d to `%s'", + inode->i_security_id, dentry->full_path); + + ret = ntfs_xattr_system_setxattr(&ctx, XATTR_NTFS_ACL, + ni, dir_ni, desc, + sd->sizes[inode->i_security_id], 0); + + if (ret != 0) { + ERROR_WITH_ERRNO("Failed to set security data on `%s'", + dentry->full_path); + return WIMLIB_ERR_NTFS_3G; + } + } + return 0; +} + +/* + * Transfers the reparse data from a WIM inode (which must represent a reparse + * point) to a NTFS inode. + */ +static int +apply_reparse_data(ntfs_inode *ni, const struct wim_dentry *dentry, + union wimlib_progress_info *progress_info) +{ + struct wim_lookup_table_entry *lte; + int ret = 0; + + lte = inode_unnamed_lte_resolved(dentry->d_inode); + + DEBUG("Applying reparse data to `%s'", dentry->full_path); + + if (!lte) { + ERROR("Could not find reparse data for `%s'", + dentry->full_path); + return WIMLIB_ERR_INVALID_DENTRY; + } + + if (wim_resource_size(lte) >= 0xffff) { + ERROR("Reparse data of `%s' is too long (%"PRIu64" bytes)", + dentry->full_path, wim_resource_size(lte)); + return WIMLIB_ERR_INVALID_DENTRY; + } + + u8 reparse_data_buf[8 + wim_resource_size(lte)]; + u8 *p = reparse_data_buf; + p = put_u32(p, dentry->d_inode->i_reparse_tag); /* ReparseTag */ + DEBUG("ReparseTag = %#x", dentry->d_inode->i_reparse_tag); + p = put_u16(p, wim_resource_size(lte)); /* ReparseDataLength */ + p = put_u16(p, 0); /* Reserved */ + + ret = read_full_wim_resource(lte, p, 0); + if (ret != 0) + return ret; + + ret = ntfs_set_ntfs_reparse_data(ni, (char*)reparse_data_buf, + wim_resource_size(lte) + 8, 0); + if (ret != 0) { + ERROR_WITH_ERRNO("Failed to set NTFS reparse data on `%s'", + dentry->full_path); + return WIMLIB_ERR_NTFS_3G; + } + progress_info->extract.completed_bytes += wim_resource_size(lte); + return 0; +} + +/* * Applies a WIM dentry to a NTFS filesystem. * * @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 wim_dentry *dentry, ntfs_inode *dir_ni, + struct apply_args *args) { - ntfs_inode *ni; - int ret; + int ret = 0; mode_t type; + ntfs_inode *ni = NULL; + struct wim_inode *inode = dentry->d_inode; + dentry->is_extracted = 1; - if (dentry_is_directory(dentry)) { + if (inode->i_attributes & FILE_ATTRIBUTE_DIRECTORY) { type = S_IFDIR; } else { type = S_IFREG; - const struct list_head *head = &dentry->link_group_list; - if (head->next != head) { - /* This dentry is one of a hard link set of at least 2 - * dentries. If one of the other dentries has already - * been extracted, make a hard link to the file - * corresponding to this already-extracted directory. - * Otherwise, extract the file, and set the - * dentry->extracted_file field so that other dentries - * in the hard link group can link to it. */ - struct dentry *other; - list_for_each_entry(other, head, link_group_list) { - if (other->extracted_file) { - return wim_apply_hardlink_ntfs( - dentry, other, dir_ni); + if (inode->i_nlink > 1) { + /* Inode has multiple dentries referencing it. */ + if (inode->i_extracted_file) { + /* Already extracted another dentry in the hard + * link group. Make a hard link instead of + * extracting the file data. */ + ret = apply_ntfs_hardlink(dentry, inode, dir_ni); + /* dir_ni was closed */ + goto out; + } else { + /* None of the dentries of this inode have been + * extracted yet, so go ahead and extract the + * first one. */ + FREE(inode->i_extracted_file); + inode->i_extracted_file = STRDUP(dentry->full_path); + if (!inode->i_extracted_file) { + ret = WIMLIB_ERR_NOMEM; + goto out_close_dir_ni; } } } - FREE(dentry->extracted_file); - dentry->extracted_file = STRDUP(dentry->full_path_utf8); - if (!dentry->extracted_file) { - ERROR("Failed to allocate memory for filename"); - return WIMLIB_ERR_NOMEM; - } } - /* - * Create a directory or file. + /* Create a NTFS directory or file. * - * Note: if it's a reparse point (such as a symbolic link), we still - * pass S_IFREG here, since we manually set the reparse data later. - */ - ni = ntfs_create(dir_ni, 0, (ntfschar*)dentry->file_name, - dentry->file_name_len / 2, type); + * Note: For symbolic links that are not directory junctions, S_IFREG is + * passed here, since the reparse data and file attributes are set + * later. */ + ni = ntfs_create(dir_ni, 0, dentry->file_name, + dentry->file_name_nbytes / 2, type); if (!ni) { - ERROR_WITH_ERRNO("Could not create NTFS object for `%s'", - dentry->full_path_utf8); + ERROR_WITH_ERRNO("Could not create NTFS inode for `%s'", + dentry->full_path); ret = WIMLIB_ERR_NTFS_3G; - goto out; + goto out_close_dir_ni; } /* Write the data streams, unless this is a directory or reparse point * */ - if (!dentry_is_directory(dentry) && - !(dentry->attributes & FILE_ATTRIBUTE_REPARSE_POINT)) { - ret = write_ntfs_data_streams(ni, dentry, w); + if (!(inode->i_attributes & (FILE_ATTRIBUTE_REPARSE_POINT | + FILE_ATTRIBUTE_DIRECTORY))) { + ret = write_ntfs_data_streams(ni, dentry, &args->progress); if (ret != 0) - goto out; + goto out_close_dir_ni; } - DEBUG("Setting NTFS file attributes on `%s' to %#"PRIx32, - dentry->full_path_utf8, dentry->attributes); - - if (!_ntfs_set_file_attributes(ni, dentry->attributes)) { - ERROR("Failed to set NTFS file attributes on `%s'", - dentry->full_path_utf8); - ret = WIMLIB_ERR_NTFS_3G; - goto out; - } + ret = apply_file_attributes_and_security_data(ni, dir_ni, dentry, + args->w, + args->extract_flags); + if (ret != 0) + goto out_close_dir_ni; - if (dentry->security_id != -1) { - const struct wim_security_data *sd = wim_security_data(w); - wimlib_assert(dentry->security_id < sd->num_entries); - DEBUG("Applying security descriptor %d to `%s'", - dentry->security_id, dentry->full_path_utf8); - if (!_ntfs_set_file_security(ni->vol, ni, ~0, - sd->descriptors[dentry->security_id])) - { - ERROR_WITH_ERRNO("Failed to set security data on `%s'", - dentry->full_path_utf8); - ret = WIMLIB_ERR_NTFS_3G; - goto out; - } + if (inode->i_attributes & FILE_ATTR_REPARSE_POINT) { + ret = apply_reparse_data(ni, dentry, &args->progress); + if (ret != 0) + goto out_close_dir_ni; } - if (dentry->attributes & FILE_ATTR_REPARSE_POINT) { - struct lookup_table_entry *lte; - ntfs_inode *ni; - lte = dentry_first_lte(dentry, w->lookup_table); - if (!lte) { - ERROR("Could not find reparse data for `%s'", - dentry->full_path_utf8); - ret = WIMLIB_ERR_INVALID_DENTRY; - goto out; - } + /* Set DOS (short) name if given */ + if (dentry_has_short_name(dentry)) { + char *short_name_mbs; + size_t short_name_mbs_nbytes; + ret = utf16le_to_tstr(dentry->short_name, + dentry->short_name_nbytes, + &short_name_mbs, + &short_name_mbs_nbytes); + if (ret != 0) + goto out_close_dir_ni; - char symlink_buf[wim_resource_size(lte)]; + DEBUG("Setting short (DOS) name of `%s' to %s", + dentry->full_path, short_name_mbs); - ret = read_full_wim_resource(lte, symlink_buf); - if (ret != 0) - goto out; - - ret = ntfs_set_ntfs_reparse_data(ni, symlink_buf, - wim_resource_size(lte), 0); + ret = ntfs_set_ntfs_dos_name(ni, dir_ni, short_name_mbs, + short_name_mbs_nbytes, 0); + FREE(short_name_mbs); if (ret != 0) { - ERROR_WITH_ERRNO("Failed to set NTFS reparse data on " - "`%s'", dentry->full_path_utf8); + ERROR_WITH_ERRNO("Could not set DOS (short) name for `%s'", + dentry->full_path); ret = WIMLIB_ERR_NTFS_3G; - goto out; } - } - - if (ntfs_inode_close_in_dir(ni, dir_ni) != 0) { - ERROR_WITH_ERRNO("Failed to close new inode"); - ret = WIMLIB_ERR_NTFS_3G; + /* inodes have been closed by ntfs_set_ntfs_dos_name(). */ goto out; } +out_close_dir_ni: + if (dir_ni) { + if (ni) { + if (ntfs_inode_close_in_dir(ni, dir_ni)) { + if (ret == 0) + ret = WIMLIB_ERR_NTFS_3G; + ERROR_WITH_ERRNO("Failed to close inode for `%s'", + dentry->full_path); + } + } + if (ntfs_inode_close(dir_ni)) { + if (ret == 0) + ret = WIMLIB_ERR_NTFS_3G; + ERROR_WITH_ERRNO("Failed to close inode of directory " + "containing `%s'", dentry->full_path); + } + } out: return ret; } -static int wim_apply_dentry_ntfs(struct dentry *dentry, void *arg) +static int +apply_root_dentry_ntfs(const struct wim_dentry *dentry, + ntfs_volume *vol, const WIMStruct *w, + int extract_flags) { - struct ntfs_apply_args *args = arg; - ntfs_volume *vol = args->vol; - int extract_flags = args->extract_flags; - WIMStruct *w = args->w; - ntfs_inode *dir_ni; - int ret; - char *p; - char orig; - const char *dir_name; - - wimlib_assert(dentry->full_path_utf8); - - DEBUG("Applying dentry `%s' to NTFS", dentry->full_path_utf8); - - if (dentry_is_root(dentry)) - return 0; - - if (extract_flags & WIMLIB_EXTRACT_FLAG_VERBOSE) - puts(dentry->full_path_utf8); - - p = dentry->full_path_utf8 + dentry->full_path_utf8_len; - do { - p--; - } while (*p != '/'); - - orig = *p; - *p = '\0'; - dir_name = dentry->full_path_utf8; + ntfs_inode *ni; + int ret = 0; - 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); + ni = ntfs_pathname_to_inode(vol, NULL, "/"); + if (!ni) { + ERROR_WITH_ERRNO("Could not find root NTFS inode"); return WIMLIB_ERR_NTFS_3G; } - DEBUG("Found NTFS inode for `%s'", dir_name); - - ret = do_wim_apply_dentry_ntfs(dentry, dir_ni, w); - - if (ntfs_inode_close(dir_ni) != 0) { - if (ret == 0) - ret = WIMLIB_ERR_NTFS_3G; - ERROR_WITH_ERRNO("Failed to close directory inode"); + ret = apply_file_attributes_and_security_data(ni, ni, dentry, w, + extract_flags); + if (ntfs_inode_close(ni) != 0) { + ERROR_WITH_ERRNO("Failed to close NTFS inode for root " + "directory"); + ret = WIMLIB_ERR_NTFS_3G; } return ret; - } -static int do_wim_apply_image_ntfs(WIMStruct *w, const char *device, int extract_flags) +/* Applies a WIM dentry to the NTFS volume */ +int +apply_dentry_ntfs(struct wim_dentry *dentry, void *arg) { - ntfs_volume *vol; + struct apply_args *args = arg; + ntfs_volume *vol = args->vol; + WIMStruct *w = args->w; + struct wim_dentry *orig_dentry; + struct wim_dentry *other; int ret; - - vol = ntfs_mount(device, 0); - if (!vol) { - ERROR_WITH_ERRNO("Failed to mount NTFS volume `%s'", device); - return WIMLIB_ERR_NTFS_3G; + + /* Treat the root dentry specially. */ + if (dentry_is_root(dentry)) + return apply_root_dentry_ntfs(dentry, vol, w, + args->extract_flags); + + /* NTFS filename namespaces need careful consideration. A name for a + * NTFS file may be in either the POSIX, Win32, DOS, or Win32+DOS + * namespaces. A NTFS file (a.k.a. inode) may have multiple names in + * multiple directories (i.e. hard links); however, a NTFS file can have + * at most 1 DOS name total. Furthermore, a Win32 name is always + * associated with a DOS name (either as a Win32+DOS name, or a Win32 + * name and a DOS name separately), which implies that a NTFS file can + * have at most 1 Win32 name. + * + * A WIM dentry just contains a "long name", which wimlib makes sure is + * non-empty, and a "short name", which may be empty. So, wimlib must + * map these to the correct NTFS names. wimlib collects all WIM + * dentries that map to the same NTFS inode and factors out the common + * information into a 'struct wim_inode', so this should make the + * mapping a little more obvious. As a NTFS file can have at most 1 DOS + * name, a WIM inode cannot have more than 1 dentry with a non-empty + * short name, and this is checked in the verify_inode() function in + * verify.c. Furthermore, a WIM dentry, if any, that has a DOS name + * must have a long name that corresponds to a Win32 name or Win32+DOS + * name. + * + * WIM dentries that have a long name but no associated short name are + * assumed to be in the POSIX namespace. + * + * So, given a WIM inode that is to map to a NTFS inode, we must apply + * the Win32 and DOS or Win32+DOS names, if they exist, then any + * additional (POSIX) names. A caveat when actually doing this: as + * confirmed by the libntfs-3g authors, ntfs_set_ntfs_dos_name() is only + * guaranteed to associate a DOS name with the appropriate long name if + * it's called when that long name is the only one in existence for that + * file. So, this implies that the correct ordering of function calls + * to extract a NTFS file are: + * + * if (file has a DOS name) { + * - Call ntfs_create() to create long name associated with + * the DOS name (this initially creates a POSIX name) + * - Call ntfs_set_ntfs_dos_name() to associate a DOS name + * with the long name just created. This either changes + * the POSIX name to Win32+DOS, or changes the POSIX name + * to Win32 and creates a separate DOS name. + * } else { + * - Call ntfs_create() to create the first link to the + * file in the POSIX namespace + * } + * - Call ntfs_link() to create the other names of the file, in the + * POSIX namespace. + */ +again: + orig_dentry = NULL; + if (!dentry->d_inode->i_dos_name_extracted && + !dentry_has_short_name(dentry)) + { + inode_for_each_dentry(other, dentry->d_inode) { + if (dentry_has_short_name(other)) { + orig_dentry = dentry; + dentry = other; + break; + } + } } - struct ntfs_apply_args args = { - .vol = vol, - .extract_flags = extract_flags, - .w = w, - }; - ret = for_dentry_in_tree(wim_root_dentry(w), wim_apply_dentry_ntfs, - &args); - if (ntfs_umount(vol, FALSE) != 0) { - ERROR_WITH_ERRNO("Failed to unmount NTFS volume"); - if (ret == 0) - ret = WIMLIB_ERR_NTFS_3G; + dentry->d_inode->i_dos_name_extracted = 1; + ntfs_inode *dir_ni = dentry_open_parent_ni(dentry, vol); + if (dir_ni) { + ret = do_apply_dentry_ntfs(dentry, dir_ni, arg); + if (ret == 0 && orig_dentry != NULL) { + dentry = orig_dentry; + goto again; + } + } else { + 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) +/* Transfers the 100-nanosecond precision timestamps from a WIM dentry to a NTFS + * inode */ +int +apply_dentry_timestamps_ntfs(struct wim_dentry *dentry, void *arg) { + struct apply_args *args = arg; + ntfs_volume *vol = args->vol; + u8 *p; + u8 buf[24]; + ntfs_inode *ni; int ret; - if (!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; + DEBUG("Setting timestamps on `%s'", dentry->full_path); + + ni = ntfs_pathname_to_inode(vol, NULL, dentry->full_path); + if (!ni) { + ERROR_WITH_ERRNO("Could not find NTFS inode for `%s'", + dentry->full_path); + return WIMLIB_ERR_NTFS_3G; } - ret = wimlib_select_image(w, image); - if (ret != 0) - return ret; -#if 0 - if (getuid() != 0) { - ERROR("We are not root, but NTFS-3g requires root privileges to set arbitrary"); - ERROR("security data on the NTFS filesystem. Please run this program as root"); - ERROR("if you want to extract a WIM image while preserving NTFS-specific"); - ERROR("information."); + p = buf; + p = put_u64(p, dentry->d_inode->i_creation_time); + p = put_u64(p, dentry->d_inode->i_last_write_time); + p = put_u64(p, dentry->d_inode->i_last_access_time); + ret = ntfs_inode_set_times(ni, (const char*)buf, 3 * sizeof(u64), 0); + if (ret != 0) { + ERROR_WITH_ERRNO("Failed to set NTFS timestamps on `%s'", + dentry->full_path); + ret = WIMLIB_ERR_NTFS_3G; + } - return WIMLIB_ERR_NOT_ROOT; + if (ntfs_inode_close(ni) != 0) { + if (ret == 0) + ret = WIMLIB_ERR_NTFS_3G; + ERROR_WITH_ERRNO("Failed to close NTFS inode for `%s'", + dentry->full_path); } -#endif - return do_wim_apply_image_ntfs(w, device, flags); + return ret; } -#else /* WITH_NTFS_3G */ -WIMLIBAPI int wimlib_apply_image_to_ntfs_volume(WIMStruct *w, int image, - const char *device, int flags) +void +libntfs3g_global_init() { - 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; + ntfs_set_char_encoding(setlocale(LC_ALL, "")); } -#endif /* WITH_NTFS_3G */