+ 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.
+ *
+ * Since these won't have 'struct wim_lookup_table_entry's, they won't show up
+ * in the call to extract_stream_list(). Hence the need for the special case.
+ */
+static int
+ntfs_3g_create_any_empty_ads(ntfs_inode *ni, const struct wim_inode *inode,
+ const struct ntfs_3g_apply_ctx *ctx)
+{
+ for (u16 i = 0; i < inode->i_num_ads; i++) {
+ const struct wim_ads_entry *entry;
+
+ entry = &inode->i_ads_entries[i];
+
+ /* Not named? */
+ if (!entry->stream_name_nbytes)
+ continue;
+
+ /* Not empty? */
+ if (entry->lte)
+ continue;
+
+ if (ntfs_attr_add(ni, AT_DATA, entry->stream_name,
+ entry->stream_name_nbytes /
+ sizeof(utf16lechar),
+ 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->i_security_id >= 0)
+ && !(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_any_empty_ads(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 list_head *next = inode->i_extraction_aliases.next;
+ struct wim_dentry *dentry;
+
+ do {
+ dentry = list_entry(next, struct wim_dentry,
+ d_extraction_alias_node);
+ if (dentry_has_short_name(dentry))
+ break;
+ next = next->next;
+ } while (next != &inode->i_extraction_aliases);
+ return dentry;
+}
+
+/*
+ * 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 list_head *next;
+ 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. */
+ next = inode->i_extraction_aliases.next;
+ do {
+ dentry = list_entry(next, struct wim_dentry,
+ d_extraction_alias_node);
+ if (dentry != first_dentry) {
+ ret = ntfs_3g_add_link(ni, dentry);
+ if (ret)
+ goto out_close_ni;
+ }
+ next = next->next;
+ } while (next != &inode->i_extraction_aliases);
+
+ /* Set metadata. */
+ ret = ntfs_3g_set_metadata(ni, inode, ctx);
+ if (ret)
+ goto out_close_ni;
+
+ ret = ntfs_3g_create_any_empty_ads(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;
+ }