- if (image == WIM_ALL_IMAGES)
- ret = extract_all_images(w, output_dir, flags);
- else
- ret = extract_single_image(w, image, output_dir, flags);
- return ret;
+
+ for_lookup_table_entry(w->lookup_table, zero_out_refcnts, NULL);
+
+ if (image == WIM_ALL_IMAGES) {
+ flags |= WIMLIB_EXTRACT_FLAG_MULTI_IMAGE;
+ return extract_all_images(w, output_dir, flags);
+ } else {
+ flags &= ~WIMLIB_EXTRACT_FLAG_MULTI_IMAGE;
+ return extract_single_image(w, image, output_dir, flags);
+ }