6 * Copyright (C) 2012, 2013, 2014 Eric Biggers
8 * This file is free software; you can redistribute it and/or modify it under
9 * the terms of the GNU Lesser General Public License as published by the Free
10 * Software Foundation; either version 3 of the License, or (at your option) any
13 * This file is distributed in the hope that it will be useful, but WITHOUT
14 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
15 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
18 * You should have received a copy of the GNU Lesser General Public License
19 * along with this file; if not, see http://www.gnu.org/licenses/.
27 #include "wimlib/blob_table.h"
28 #include "wimlib/error.h"
29 #include "wimlib/inode.h"
30 #include "wimlib/metadata.h"
31 #include "wimlib/xml.h"
34 blob_set_not_exported(struct blob_descriptor *blob, void *_ignore)
37 blob->was_exported = 0;
42 blob_rollback_export(struct blob_descriptor *blob, void *_blob_table)
44 struct blob_table *blob_table = _blob_table;
46 blob->refcnt -= blob->out_refcnt;
47 if (blob->was_exported) {
48 blob_table_unlink(blob_table, blob);
49 free_blob_descriptor(blob);
55 inode_export_blobs(struct wim_inode *inode, struct blob_table *src_blob_table,
56 struct blob_table *dest_blob_table, bool gift)
60 struct blob_descriptor *src_blob, *dest_blob;
62 for (i = 0; i < inode->i_num_streams; i++) {
64 /* Retrieve SHA-1 message digest of blob to export. */
65 hash = stream_hash(&inode->i_streams[i]);
66 if (is_zero_hash(hash)) /* Empty stream? */
69 /* Search for the blob (via SHA-1 message digest) in the
71 dest_blob = lookup_blob(dest_blob_table, hash);
73 /* Blob not yet present in destination WIM. Search for
74 * it in the source WIM, then export it into the
76 src_blob = stream_blob(&inode->i_streams[i],
79 return blob_not_found_error(inode, hash);
83 blob_table_unlink(src_blob_table, src_blob);
85 dest_blob = clone_blob_descriptor(src_blob);
87 return WIMLIB_ERR_NOMEM;
89 dest_blob->refcnt = 0;
90 dest_blob->out_refcnt = 0;
91 dest_blob->was_exported = 1;
92 blob_table_insert(dest_blob_table, dest_blob);
95 /* Blob is present in destination WIM (either pre-existing,
96 * already exported, or just exported above). Increment its
97 * reference count appropriately. Note: we use 'refcnt' for
98 * the raw reference count, but 'out_refcnt' for references
99 * arising just from the export operation; this is used to roll
100 * back a failed export if needed. */
101 dest_blob->refcnt += inode->i_nlink;
102 dest_blob->out_refcnt += inode->i_nlink;
107 /* API function documented in wimlib.h */
109 wimlib_export_image(WIMStruct *src_wim,
112 const tchar *dest_name,
113 const tchar *dest_description,
119 int orig_dest_image_count;
121 bool all_images = (src_image == WIMLIB_ALL_IMAGES);
123 /* Check for sane parameters. */
124 if (export_flags & ~(WIMLIB_EXPORT_FLAG_BOOT |
125 WIMLIB_EXPORT_FLAG_NO_NAMES |
126 WIMLIB_EXPORT_FLAG_NO_DESCRIPTIONS |
127 WIMLIB_EXPORT_FLAG_GIFT |
128 WIMLIB_EXPORT_FLAG_WIMBOOT))
129 return WIMLIB_ERR_INVALID_PARAM;
131 if (!src_wim || !dest_wim || src_wim == dest_wim)
132 return WIMLIB_ERR_INVALID_PARAM;
134 if (!wim_has_metadata(src_wim) || !wim_has_metadata(dest_wim))
135 return WIMLIB_ERR_METADATA_NOT_FOUND;
138 /* Multi-image export. */
139 if ((!(export_flags & WIMLIB_EXPORT_FLAG_NO_NAMES) &&
141 (!(export_flags & WIMLIB_EXPORT_FLAG_NO_DESCRIPTIONS) &&
144 ERROR("Image name and description must be "
145 "left NULL for multi-image export");
146 return WIMLIB_ERR_INVALID_PARAM;
149 end_src_image = src_wim->hdr.image_count;
151 start_src_image = src_image;
152 end_src_image = src_image;
154 orig_dest_image_count = dest_wim->hdr.image_count;
156 /* Blob checksums must be known before proceeding. */
157 ret = wim_checksum_unhashed_blobs(src_wim);
160 ret = wim_checksum_unhashed_blobs(dest_wim);
164 /* Enable rollbacks */
165 for_blob_in_table(dest_wim->blob_table, blob_set_not_exported, NULL);
167 /* Export each requested image. */
168 for (src_image = start_src_image;
169 src_image <= end_src_image;
172 const tchar *next_dest_name, *next_dest_description;
173 struct wim_image_metadata *src_imd;
174 struct wim_inode *inode;
176 /* Determine destination image name and description. */
178 if (export_flags & WIMLIB_EXPORT_FLAG_NO_NAMES)
179 next_dest_name = T("");
181 next_dest_name = dest_name;
183 next_dest_name = wimlib_get_image_name(src_wim, src_image);
185 if (export_flags & WIMLIB_EXPORT_FLAG_NO_DESCRIPTIONS)
186 next_dest_description = T("");
187 else if (dest_description)
188 next_dest_description = dest_description;
190 next_dest_description = wimlib_get_image_description(src_wim, src_image);
192 /* Check for name conflict. */
193 if (wimlib_image_name_in_use(dest_wim, next_dest_name)) {
194 ERROR("There is already an image named \"%"TS"\" "
195 "in the destination WIM", next_dest_name);
196 ret = WIMLIB_ERR_IMAGE_NAME_COLLISION;
200 /* Load metadata for source image into memory. */
201 ret = select_wim_image(src_wim, src_image);
205 src_imd = wim_get_current_image_metadata(src_wim);
207 /* Iterate through inodes in the source image and export their
208 * blobs into the destination WIM. */
209 image_for_each_inode(inode, src_imd) {
210 ret = inode_export_blobs(inode,
212 dest_wim->blob_table,
213 export_flags & WIMLIB_EXPORT_FLAG_GIFT);
218 /* Export XML information into the destination WIM. */
219 ret = xml_export_image(src_wim->wim_info, src_image,
220 &dest_wim->wim_info, next_dest_name,
221 next_dest_description);
225 /* Reference the source image metadata from the destination WIM.
227 ret = append_image_metadata(dest_wim, src_imd);
232 /* Lock the metadata into memory. XXX: need better solution for
234 src_imd->modified = 1;
238 /* Image export complete. Finish by setting any needed special metadata
239 * on the destination WIM. */
241 if (src_wim->hdr.flags & WIM_HDR_FLAG_RP_FIX)
242 dest_wim->hdr.flags |= WIM_HDR_FLAG_RP_FIX;
244 for (src_image = start_src_image;
245 src_image <= end_src_image;
248 int dst_image = orig_dest_image_count + 1 +
249 (src_image - start_src_image);
251 if (export_flags & WIMLIB_EXPORT_FLAG_WIMBOOT)
252 wim_info_set_wimboot(dest_wim->wim_info, dst_image, true);
254 if ((export_flags & WIMLIB_EXPORT_FLAG_BOOT) &&
255 (!all_images || src_image == src_wim->hdr.boot_idx))
256 dest_wim->hdr.boot_idx = dst_image;
259 if (export_flags & WIMLIB_EXPORT_FLAG_GIFT) {
260 free_blob_table(src_wim->blob_table);
261 src_wim->blob_table = NULL;
266 while ((image = wim_info_get_num_images(dest_wim->wim_info))
267 > orig_dest_image_count)
269 xml_delete_image(&dest_wim->wim_info, image);
271 while (dest_wim->hdr.image_count > orig_dest_image_count)
273 put_image_metadata(dest_wim->image_metadata[
274 --dest_wim->hdr.image_count], NULL);
276 for_blob_in_table(dest_wim->blob_table, blob_rollback_export,
277 dest_wim->blob_table);