X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fexport_image.c;h=455baa9152bbd2050b4d98f118667013c86138e8;hp=b62f67c59614a7ae90cad840c32bef1eb419c41d;hb=f5d26d6005ba759762bec8c7cd45a738e577a8a7;hpb=650997e4865a090b6856c7ca34b02f42994e8e29 diff --git a/src/export_image.c b/src/export_image.c index b62f67c5..455baa91 100644 --- a/src/export_image.c +++ b/src/export_image.c @@ -21,10 +21,17 @@ * along with wimlib; if not, see http://www.gnu.org/licenses/. */ -#include "wimlib_internal.h" -#include "dentry.h" -#include "lookup_table.h" -#include "xml.h" +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif + +#include "wimlib.h" +#include "wimlib/dentry.h" +#include "wimlib/error.h" +#include "wimlib/lookup_table.h" +#include "wimlib/metadata.h" +#include "wimlib/swm.h" +#include "wimlib/xml.h" static int inode_allocate_needed_ltes(struct wim_inode *inode, @@ -47,7 +54,8 @@ inode_allocate_needed_ltes(struct wim_inode *inode, dest_lte = clone_lookup_table_entry(src_lte); if (!dest_lte) return WIMLIB_ERR_NOMEM; - list_add_tail(&dest_lte->staging_list, lte_list_head); + list_add_tail(&dest_lte->export_stream_list, + lte_list_head); } } } @@ -62,10 +70,6 @@ inode_move_ltes_to_table(struct wim_inode *inode, { struct wim_lookup_table_entry *src_lte, *dest_lte; unsigned i; - struct wim_dentry *dentry; - - inode_for_each_dentry(dentry, inode) - dentry->refcnt++; for (i = 0; i <= inode->i_num_ads; i++) { src_lte = inode_stream_lte_unresolved(inode, i, src_lookup_table); @@ -80,7 +84,7 @@ inode_move_ltes_to_table(struct wim_inode *inode, list_del(next); dest_lte = container_of(next, struct wim_lookup_table_entry, - staging_list); + export_stream_list); dest_lte->part_number = 1; dest_lte->refcnt = 0; wimlib_assert(hashes_equal(dest_lte->hash, src_lte->hash)); @@ -92,7 +96,7 @@ inode_move_ltes_to_table(struct wim_inode *inode, } /* - * Copies an image, or all the images, from a WIM file, into another WIM file. + * Exports an image, or all the images, from a WIM file, into another WIM file. */ WIMLIBAPI int wimlib_export_image(WIMStruct *src_wim, @@ -106,10 +110,7 @@ wimlib_export_image(WIMStruct *src_wim, wimlib_progress_func_t progress_func) { int ret; - struct wim_security_data *sd; - struct wim_lookup_table *joined_tab, *src_wim_tab_save; struct wim_image_metadata *src_imd; - struct hlist_node *cur_node; struct list_head lte_list_head; struct wim_inode *inode; @@ -155,7 +156,7 @@ wimlib_export_image(WIMStruct *src_wim, additional_swms, num_additional_swms, progress_func); - if (ret != 0) + if (ret) return ret; } return 0; @@ -188,21 +189,21 @@ wimlib_export_image(WIMStruct *src_wim, } ret = verify_swm_set(src_wim, additional_swms, num_additional_swms); - if (ret != 0) + if (ret) return ret; - if (num_additional_swms) { - ret = new_joined_lookup_table(src_wim, additional_swms, - num_additional_swms, - &joined_tab); - if (ret != 0) - return ret; - src_wim_tab_save = src_wim->lookup_table; - src_wim->lookup_table = joined_tab; - } + ret = wim_checksum_unhashed_streams(src_wim); + if (ret) + return ret; + ret = wim_checksum_unhashed_streams(dest_wim); + if (ret) + return ret; + + if (num_additional_swms) + merge_lookup_tables(src_wim, additional_swms, num_additional_swms); ret = select_wim_image(src_wim, src_image); - if (ret != 0) { + if (ret) { ERROR("Could not select image %d from the WIM `%"TS"' " "to export it", src_image, src_wim->filename); goto out; @@ -214,38 +215,34 @@ wimlib_export_image(WIMStruct *src_wim, for_lookup_table_entry(src_wim->lookup_table, lte_zero_out_refcnt, NULL); src_imd = wim_get_current_image_metadata(src_wim); INIT_LIST_HEAD(<e_list_head); - hlist_for_each_entry(inode, cur_node, &src_imd->inode_list, i_hlist) { + image_for_each_inode(inode, src_imd) { ret = inode_allocate_needed_ltes(inode, src_wim->lookup_table, dest_wim->lookup_table, <e_list_head); - if (ret != 0) + if (ret) goto out_free_ltes; } ret = xml_export_image(src_wim->wim_info, src_image, &dest_wim->wim_info, dest_name, dest_description); - if (ret != 0) + if (ret) goto out_free_ltes; - sd = src_imd->security_data; - ret = add_new_dentry_tree(dest_wim, src_imd->root_dentry, sd); - if (ret != 0) + ret = append_image_metadata(dest_wim, src_imd); + if (ret) goto out_xml_delete_image; - dest_wim->image_metadata[ - dest_wim->hdr.image_count - 1].inode_list = src_imd->inode_list; - if (src_imd->inode_list.first) - src_imd->inode_list.first->pprev = NULL; + /* The `struct image_metadata' is now referenced by both the @src_wim + * and the @dest_wim. */ + src_imd->refcnt++; + src_imd->modified = 1; /* All memory allocations have been taken care of, so it's no longer - * possible for this function to fail. Go ahead and increment the - * reference counts of the dentry tree and security data, then update - * the lookup table of the destination WIM and the boot index, if - * needed. */ - sd->refcnt++; - hlist_for_each_entry(inode, cur_node, &src_imd->inode_list, i_hlist) { + * possible for this function to fail. Go ahead and update the lookup + * table of the destination WIM and the boot index, if needed. */ + image_for_each_inode(inode, src_imd) { inode_move_ltes_to_table(inode, src_wim->lookup_table, dest_wim->lookup_table, @@ -253,23 +250,25 @@ wimlib_export_image(WIMStruct *src_wim, } if (export_flags & WIMLIB_EXPORT_FLAG_BOOT) - wimlib_set_boot_idx(dest_wim, dest_wim->hdr.image_count); + dest_wim->hdr.boot_idx = dest_wim->hdr.image_count; + if (src_wim->hdr.flags & WIM_HDR_FLAG_RP_FIX) + { + /* Set the reparse point fixup flag on the destination WIM if + * the flag is set on the source WIM. */ + dest_wim->hdr.flags |= WIM_HDR_FLAG_RP_FIX; + } ret = 0; goto out; - out_xml_delete_image: xml_delete_image(&dest_wim->wim_info, dest_wim->hdr.image_count + 1); out_free_ltes: { struct wim_lookup_table_entry *lte, *tmp; - list_for_each_entry_safe(lte, tmp, <e_list_head, staging_list) + list_for_each_entry_safe(lte, tmp, <e_list_head, export_stream_list) free_lookup_table_entry(lte); } - out: - if (num_additional_swms) { - free_lookup_table(src_wim->lookup_table); - src_wim->lookup_table = src_wim_tab_save; - } + if (num_additional_swms) + unmerge_lookup_table(src_wim); return ret; }