4 * Split a WIM file into parts.
8 * Copyright (C) 2012, 2013 Eric Biggers
10 * This file is part of wimlib, a library for working with WIM files.
12 * wimlib is free software; you can redistribute it and/or modify it under the
13 * terms of the GNU General Public License as published by the Free
14 * Software Foundation; either version 3 of the License, or (at your option)
17 * wimlib is distributed in the hope that it will be useful, but WITHOUT ANY
18 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
19 * A PARTICULAR PURPOSE. See the GNU General Public License for more
22 * You should have received a copy of the GNU General Public License
23 * along with wimlib; if not, see http://www.gnu.org/licenses/.
31 #include "wimlib/error.h"
32 #include "wimlib/list.h"
33 #include "wimlib/lookup_table.h"
34 #include "wimlib/metadata.h"
35 #include "wimlib/progress.h"
36 #include "wimlib/resource.h"
37 #include "wimlib/wim.h"
38 #include "wimlib/write.h"
46 struct swm_part_info {
47 struct list_head stream_list;
52 copy_part_info(struct swm_part_info *dst, struct swm_part_info *src)
54 list_transfer(&src->stream_list, &dst->stream_list);
55 dst->size = src->size;
59 struct swm_part_info *parts;
61 unsigned num_alloc_parts;
67 write_split_wim(WIMStruct *orig_wim, const tchar *swm_name,
68 struct swm_info *swm_info, int write_flags)
74 size_t swm_base_name_len;
76 union wimlib_progress_info progress;
79 u8 guid[WIMLIB_GUID_LEN];
81 swm_name_len = tstrlen(swm_name);
82 swm_name_buf = alloca((swm_name_len + 20) * sizeof(tchar));
83 tstrcpy(swm_name_buf, swm_name);
84 dot = tstrchr(swm_name_buf, T('.'));
86 swm_base_name_len = dot - swm_name_buf;
87 swm_suffix = alloca((tstrlen(dot) + 1) * sizeof(tchar));
88 tstrcpy(swm_suffix, dot);
90 swm_base_name_len = swm_name_len;
91 swm_suffix = alloca(1 * sizeof(tchar));
92 swm_suffix[0] = T('\0');
95 progress.split.completed_bytes = 0;
96 progress.split.total_bytes = 0;
97 for (part_number = 1; part_number <= swm_info->num_parts; part_number++)
98 progress.split.total_bytes += swm_info->parts[part_number - 1].size;
99 progress.split.total_parts = swm_info->num_parts;
100 progress.split.part_name = swm_name_buf;
102 randomize_byte_array(guid, WIMLIB_GUID_LEN);
104 for (part_number = 1; part_number <= swm_info->num_parts; part_number++) {
105 int part_write_flags;
106 wimlib_progress_func_t progfunc;
108 if (part_number != 1) {
109 tsprintf(swm_name_buf + swm_base_name_len,
110 T("%u%"TS), part_number, swm_suffix);
113 progress.split.cur_part_number = part_number;
115 ret = call_progress(orig_wim->progfunc,
116 WIMLIB_PROGRESS_MSG_SPLIT_BEGIN_PART,
122 part_write_flags = write_flags;
123 part_write_flags |= WIMLIB_WRITE_FLAG_USE_EXISTING_TOTALBYTES;
124 if (part_number != 1)
125 part_write_flags |= WIMLIB_WRITE_FLAG_NO_METADATA;
127 progfunc = orig_wim->progfunc;
128 orig_wim->progfunc = NULL;
129 ret = write_wim_part(orig_wim,
136 &swm_info->parts[part_number - 1].stream_list,
138 orig_wim->progfunc = progfunc;
142 progress.split.completed_bytes += swm_info->parts[part_number - 1].size;
144 ret = call_progress(orig_wim->progfunc,
145 WIMLIB_PROGRESS_MSG_SPLIT_END_PART,
155 add_stream_to_swm(struct wim_lookup_table_entry *lte, void *_swm_info)
157 struct swm_info *swm_info = _swm_info;
160 if (lte_is_partial(lte)) {
161 ERROR("Splitting of WIM containing packed streams is not supported.\n"
162 " Export it in the default format first.");
163 return WIMLIB_ERR_UNSUPPORTED;
165 if (lte->resource_location == RESOURCE_IN_WIM)
166 stream_size = lte->rspec->size_in_wim;
168 stream_size = lte->size;
170 /* - Start first part if no parts have been started so far;
171 * - Start next part if adding this stream exceeds maximum part size,
172 * UNLESS the stream is metadata or if no streams at all have been
173 * added to the current part.
175 if (swm_info->num_parts == 0 ||
176 ((swm_info->parts[swm_info->num_parts - 1].size +
177 stream_size >= swm_info->max_part_size)
178 && !((lte->flags & WIM_RESHDR_FLAG_METADATA) ||
179 swm_info->parts[swm_info->num_parts - 1].size == 0)))
181 if (swm_info->num_parts == swm_info->num_alloc_parts) {
182 struct swm_part_info *parts;
183 size_t num_alloc_parts = swm_info->num_alloc_parts;
185 num_alloc_parts += 8;
186 parts = MALLOC(num_alloc_parts * sizeof(parts[0]));
188 return WIMLIB_ERR_NOMEM;
190 for (unsigned i = 0; i < swm_info->num_parts; i++)
191 copy_part_info(&parts[i], &swm_info->parts[i]);
193 FREE(swm_info->parts);
194 swm_info->parts = parts;
195 swm_info->num_alloc_parts = num_alloc_parts;
197 swm_info->num_parts++;
198 INIT_LIST_HEAD(&swm_info->parts[swm_info->num_parts - 1].stream_list);
199 swm_info->parts[swm_info->num_parts - 1].size = 0;
201 swm_info->parts[swm_info->num_parts - 1].size += stream_size;
202 if (!(lte->flags & WIM_RESHDR_FLAG_METADATA)) {
203 list_add_tail(<e->write_streams_list,
204 &swm_info->parts[swm_info->num_parts - 1].stream_list);
206 swm_info->total_bytes += stream_size;
210 /* API function documented in wimlib.h */
212 wimlib_split(WIMStruct *wim, const tchar *swm_name,
213 u64 part_size, int write_flags)
215 struct swm_info swm_info;
219 if (swm_name == NULL || swm_name[0] == T('\0') || part_size == 0)
220 return WIMLIB_ERR_INVALID_PARAM;
222 if (write_flags & ~WIMLIB_WRITE_MASK_PUBLIC)
223 return WIMLIB_ERR_INVALID_PARAM;
225 if (!wim_has_metadata(wim))
226 return WIMLIB_ERR_METADATA_NOT_FOUND;
228 memset(&swm_info, 0, sizeof(swm_info));
229 swm_info.max_part_size = part_size;
231 for (i = 0; i < wim->hdr.image_count; i++) {
232 ret = add_stream_to_swm(wim->image_metadata[i]->metadata_lte,
235 goto out_free_swm_info;
238 ret = for_lookup_table_entry_pos_sorted(wim->lookup_table,
242 goto out_free_swm_info;
244 ret = write_split_wim(wim, swm_name, &swm_info, write_flags);
246 FREE(swm_info.parts);