+ goto fail;
+
+ res = ntfs_3g_set_timestamps(ni, inode);
+
+ if (ntfs_inode_close(ni) || res)
+ goto fail;
+
+ return 0;
+
+fail:
+ ERROR_WITH_ERRNO("Failed to update timestamps of \"%s\" in NTFS volume",
+ dentry_full_path(inode_first_extraction_dentry(inode)));
+ return WIMLIB_ERR_SET_TIMESTAMPS;
+}
+
+/* Restore the DOS name of the @dentry.
+ * This closes both @ni and @dir_ni.
+ * If either is NULL, then they are opened temporarily. */
+static int
+ntfs_3g_restore_dos_name(ntfs_inode *ni, ntfs_inode *dir_ni,
+ struct wim_dentry *dentry, ntfs_volume *vol)
+{
+ int ret;
+ const char *dos_name;
+ size_t dos_name_nbytes;
+
+ /* Note: ntfs_set_ntfs_dos_name() closes both inodes (even if it fails).
+ * And it takes in a multibyte string, even though it translates it to
+ * UTF-16LE internally... which is annoying because we currently have
+ * the UTF-16LE string but not the multibyte string. */
+
+ ret = utf16le_get_tstr(dentry->d_short_name, dentry->d_short_name_nbytes,
+ &dos_name, &dos_name_nbytes);
+ if (ret)
+ goto out_close;
+
+ if (!dir_ni)
+ dir_ni = ntfs_inode_open(vol, dentry->d_parent->d_inode->i_mft_no);
+ if (!ni)
+ ni = ntfs_inode_open(vol, dentry->d_inode->i_mft_no);
+ if (dir_ni && ni) {
+ ret = ntfs_set_ntfs_dos_name(ni, dir_ni,
+ dos_name, dos_name_nbytes, 0);
+ dir_ni = NULL;
+ ni = NULL;
+ } else {
+ ret = -1;
+ }
+ utf16le_put_tstr(dos_name);
+ if (ret) {
+ ERROR_WITH_ERRNO("Failed to set DOS name of \"%s\" in NTFS "
+ "volume", dentry_full_path(dentry));
+ ret = WIMLIB_ERR_SET_SHORT_NAME;
+ goto out_close;
+ }
+
+ /* Unlike most other NTFS-3g functions, ntfs_set_ntfs_dos_name()
+ * changes the directory's last modification timestamp...
+ * Change it back. */
+ return ntfs_3g_restore_timestamps(vol, dentry->d_parent->d_inode);
+
+out_close:
+ /* ntfs_inode_close() can take a NULL argument, but it's probably best
+ * not to rely on this behavior. */
+ if (ni)
+ ntfs_inode_close(ni);
+ if (dir_ni)
+ ntfs_inode_close(dir_ni);
+ return ret;
+}
+
+/*
+ * Create empty named data streams for the specified file, if there are any.
+ *
+ * Since these won't have blob descriptors, they won't show up in the call to
+ * extract_blob_list(). Hence the need for the special case.
+ */
+static int
+ntfs_3g_create_empty_named_data_streams(ntfs_inode *ni,
+ const struct wim_inode *inode,
+ const struct ntfs_3g_apply_ctx *ctx)
+{
+ for (unsigned i = 0; i < inode->i_num_streams; i++) {
+
+ const struct wim_inode_stream *strm = &inode->i_streams[i];
+
+ if (!stream_is_named_data_stream(strm) ||
+ stream_blob_resolved(strm) != NULL)
+ continue;
+
+ if (ntfs_attr_add(ni, AT_DATA, strm->stream_name,
+ utf16le_len_chars(strm->stream_name),
+ NULL, 0))
+ {
+ ERROR_WITH_ERRNO("Failed to create named data stream "
+ "of \"%s\"", dentry_full_path(
+ inode_first_extraction_dentry(inode)));
+ return WIMLIB_ERR_NTFS_3G;
+ }
+ }
+ return 0;
+}
+
+/* Set attributes, security descriptor, and timestamps on the NTFS inode @ni.
+ */
+static int
+ntfs_3g_set_metadata(ntfs_inode *ni, const struct wim_inode *inode,
+ const struct ntfs_3g_apply_ctx *ctx)
+{
+ int extract_flags;
+ const struct wim_security_data *sd;
+ struct wim_dentry *one_dentry;
+ int ret;
+
+ extract_flags = ctx->common.extract_flags;
+ sd = wim_get_current_security_data(ctx->common.wim);
+ one_dentry = inode_first_extraction_dentry(inode);
+
+ /* Attributes */
+ if (!(extract_flags & WIMLIB_EXTRACT_FLAG_NO_ATTRIBUTES)) {
+ u32 attrib = inode->i_attributes;
+
+ attrib &= ~(FILE_ATTRIBUTE_SPARSE_FILE |
+ FILE_ATTRIBUTE_ENCRYPTED);
+
+ if (ntfs_set_ntfs_attrib(ni, (const char *)&attrib,
+ sizeof(attrib), 0))
+ {
+ ERROR_WITH_ERRNO("Failed to set attributes on \"%s\" "
+ "in NTFS volume",
+ dentry_full_path(one_dentry));
+ return WIMLIB_ERR_SET_ATTRIBUTES;
+ }
+ }
+
+ /* Security descriptor */
+ if (inode_has_security_descriptor(inode)
+ && !(extract_flags & WIMLIB_EXTRACT_FLAG_NO_ACLS))
+ {
+ const void *desc;
+ size_t desc_size;
+
+ desc = sd->descriptors[inode->i_security_id];
+ desc_size = sd->sizes[inode->i_security_id];
+
+ ret = ntfs_3g_set_security_descriptor(ni, desc, desc_size);
+ if (ret) {
+ if (wimlib_print_errors) {
+ ERROR_WITH_ERRNO("Failed to set security descriptor "
+ "on \"%s\" in NTFS volume",
+ dentry_full_path(one_dentry));
+ fprintf(wimlib_error_file,
+ "The security descriptor is: ");
+ print_byte_field(desc, desc_size, wimlib_error_file);
+ fprintf(wimlib_error_file, "\n");
+ }
+ return ret;
+ }
+ }
+
+ /* Timestamps */
+ ret = ntfs_3g_set_timestamps(ni, inode);
+ if (ret) {
+ ERROR_WITH_ERRNO("Failed to set timestamps on \"%s\" "
+ "in NTFS volume",
+ dentry_full_path(one_dentry));
+ return ret;
+ }
+ return 0;
+}
+
+/* Recursively creates all the subdirectories of @dir, which has been created as
+ * the NTFS inode @dir_ni. */
+static int
+ntfs_3g_create_dirs_recursive(ntfs_inode *dir_ni, struct wim_dentry *dir,
+ struct ntfs_3g_apply_ctx *ctx)
+{
+ struct wim_dentry *child;
+
+ for_dentry_child(child, dir) {
+ ntfs_inode *ni;
+ int ret;
+
+ if (!(child->d_inode->i_attributes & FILE_ATTRIBUTE_DIRECTORY))
+ continue;
+ if (!will_extract_dentry(child))
+ continue;
+
+ ni = ntfs_create(dir_ni, 0, child->d_extraction_name,
+ child->d_extraction_name_nchars, S_IFDIR);
+ if (!ni) {
+ ERROR_WITH_ERRNO("Error creating \"%s\" in NTFS volume",
+ dentry_full_path(child));
+ return WIMLIB_ERR_NTFS_3G;
+ }
+
+ child->d_inode->i_mft_no = ni->mft_no;
+
+ ret = report_file_created(&ctx->common);
+ if (!ret)
+ ret = ntfs_3g_set_metadata(ni, child->d_inode, ctx);
+ if (!ret)
+ ret = ntfs_3g_create_empty_named_data_streams(ni, child->d_inode, ctx);
+ if (!ret)
+ ret = ntfs_3g_create_dirs_recursive(ni, child, ctx);
+
+ if (ntfs_inode_close_in_dir(ni, dir_ni) && !ret) {
+ ERROR_WITH_ERRNO("Error closing \"%s\" in NTFS volume",
+ dentry_full_path(child));
+ ret = WIMLIB_ERR_NTFS_3G;
+ }
+ if (ret)
+ return ret;
+ }
+ return 0;
+}
+
+/* For each WIM dentry in the @root tree that represents a directory, create the
+ * corresponding directory in the NTFS volume @ctx->vol. */
+static int
+ntfs_3g_create_directories(struct wim_dentry *root,
+ struct list_head *dentry_list,
+ struct ntfs_3g_apply_ctx *ctx)
+{
+ ntfs_inode *root_ni;
+ int ret;
+ struct wim_dentry *dentry;
+
+ /* Create the directories using POSIX names. */
+
+ root_ni = ntfs_inode_open(ctx->vol, FILE_root);
+ if (!root_ni) {
+ ERROR_WITH_ERRNO("Can't open root of NTFS volume");
+ return WIMLIB_ERR_NTFS_3G;
+ }
+
+ root->d_inode->i_mft_no = FILE_root;
+
+ ret = ntfs_3g_create_dirs_recursive(root_ni, root, ctx);
+
+ if (ntfs_inode_close(root_ni) && !ret) {
+ ERROR_WITH_ERRNO("Error closing root of NTFS volume");
+ ret = WIMLIB_ERR_NTFS_3G;
+ }
+ if (ret)
+ return ret;
+
+ /* Set the DOS name of any directory that has one. */
+ list_for_each_entry(dentry, dentry_list, d_extraction_list_node) {
+ if (!(dentry->d_inode->i_attributes & FILE_ATTRIBUTE_DIRECTORY))
+ continue;
+ if (!dentry_has_short_name(dentry))
+ continue;
+ ret = ntfs_3g_restore_dos_name(NULL, NULL, dentry, ctx->vol);
+ if (ret)
+ return ret;
+ ret = report_file_created(&ctx->common);
+ if (ret)
+ return ret;
+ }
+ return 0;
+}
+
+/* When creating an inode that will have a short (DOS) name, we create it using
+ * the long name associated with the short name. This ensures that the short
+ * name gets associated with the correct long name. */
+static struct wim_dentry *
+ntfs_3g_first_extraction_alias(struct wim_inode *inode)
+{
+ struct wim_dentry *dentry;
+
+ inode_for_each_extraction_alias(dentry, inode)
+ if (dentry_has_short_name(dentry))
+ return dentry;
+ return inode_first_extraction_dentry(inode);
+}
+
+/*
+ * Add a hard link for the NTFS inode @ni at the location corresponding to the
+ * WIM dentry @dentry.
+ *
+ * The parent directory must have already been created on the NTFS volume.
+ *
+ * Returns 0 on success; returns WIMLIB_ERR_NTFS_3G and sets errno on failure.
+ */
+static int
+ntfs_3g_add_link(ntfs_inode *ni, struct wim_dentry *dentry)
+{
+ ntfs_inode *dir_ni;
+ int res;
+
+ /* Open the inode of the parent directory. */
+ dir_ni = ntfs_inode_open(ni->vol, dentry->d_parent->d_inode->i_mft_no);
+ if (!dir_ni)
+ goto fail;
+
+ /* Create the link. */
+ res = ntfs_link(ni, dir_ni, dentry->d_extraction_name,
+ dentry->d_extraction_name_nchars);
+
+ /* Close the parent directory. */
+ if (ntfs_inode_close(dir_ni) || res)
+ goto fail;
+
+ return 0;
+
+fail:
+ ERROR_WITH_ERRNO("Can't create link \"%s\" in NTFS volume",
+ dentry_full_path(dentry));
+ return WIMLIB_ERR_NTFS_3G;
+}
+
+static int
+ntfs_3g_create_nondirectory(struct wim_inode *inode,
+ const struct ntfs_3g_apply_ctx *ctx)
+{
+ struct wim_dentry *first_dentry;
+ ntfs_inode *dir_ni;
+ ntfs_inode *ni;
+ struct wim_dentry *dentry;
+ int ret;
+
+ first_dentry = ntfs_3g_first_extraction_alias(inode);
+
+ /* Create first link. */
+
+ dir_ni = ntfs_inode_open(ctx->vol, first_dentry->d_parent->d_inode->i_mft_no);
+ if (!dir_ni) {
+ ERROR_WITH_ERRNO("Can't open \"%s\" in NTFS volume",
+ dentry_full_path(first_dentry->d_parent));
+ return WIMLIB_ERR_NTFS_3G;
+ }
+
+ ni = ntfs_create(dir_ni, 0, first_dentry->d_extraction_name,
+ first_dentry->d_extraction_name_nchars, S_IFREG);
+
+ if (!ni) {
+ ERROR_WITH_ERRNO("Can't create \"%s\" in NTFS volume",
+ dentry_full_path(first_dentry));
+ ntfs_inode_close(dir_ni);
+ return WIMLIB_ERR_NTFS_3G;
+ }
+
+ inode->i_mft_no = ni->mft_no;
+
+ /* Set short name if present. */
+ if (dentry_has_short_name(first_dentry)) {
+
+ ret = ntfs_3g_restore_dos_name(ni, dir_ni, first_dentry, ctx->vol);
+
+ /* ntfs_3g_restore_dos_name() closed both 'ni' and 'dir_ni'. */
+
+ if (ret)
+ return ret;
+
+ /* Reopen the inode. */
+ ni = ntfs_inode_open(ctx->vol, inode->i_mft_no);
+ if (!ni) {
+ ERROR_WITH_ERRNO("Failed to reopen \"%s\" "
+ "in NTFS volume",
+ dentry_full_path(first_dentry));
+ return WIMLIB_ERR_NTFS_3G;
+ }
+ } else {
+ /* Close the directory in which the first link was created. */
+ if (ntfs_inode_close(dir_ni)) {
+ ERROR_WITH_ERRNO("Failed to close \"%s\" in NTFS volume",
+ dentry_full_path(first_dentry->d_parent));
+ ret = WIMLIB_ERR_NTFS_3G;
+ goto out_close_ni;
+ }
+ }
+
+ /* Create additional links if present. */
+ inode_for_each_extraction_alias(dentry, inode) {
+ if (dentry != first_dentry) {
+ ret = ntfs_3g_add_link(ni, dentry);
+ if (ret)
+ goto out_close_ni;
+ }
+ }
+
+ /* Set metadata. */
+ ret = ntfs_3g_set_metadata(ni, inode, ctx);
+ if (ret)
+ goto out_close_ni;
+
+ ret = ntfs_3g_create_empty_named_data_streams(ni, inode, ctx);
+
+out_close_ni:
+ /* Close the inode. */
+ if (ntfs_inode_close(ni) && !ret) {
+ ERROR_WITH_ERRNO("Error closing \"%s\" in NTFS volume",
+ dentry_full_path(first_dentry));
+ ret = WIMLIB_ERR_NTFS_3G;
+ }