wimcapture.1: 7-Zip 15.12 supports LZMS decompression

[wimlib] / src / delete_image.c

diff --git a/src/delete_image.c b/src/delete_image.c
index 131ec8d94f83525b2e3833406dea69ca94f54db4..42e1ae4e2ba9e0d6ab667ee136852f897ed36f39 100644 (file)
--- a/src/delete_image.c
+++ b/src/delete_image.c
@@ -3,7 +3,7 @@
  */
 
 /*
- * Copyright (C) 2012, 2013 Eric Biggers
+ * Copyright (C) 2012, 2013, 2014 Eric Biggers
  *
  * This file is free software; you can redistribute it and/or modify it under
  * the terms of the GNU Lesser General Public License as published by the Free
@@ -23,13 +23,53 @@
 #  include "config.h"
 #endif
 
+#include <string.h>
+
 #include "wimlib.h"
-#include "wimlib/error.h"
 #include "wimlib/metadata.h"
-#include "wimlib/util.h"
 #include "wimlib/wim.h"
 #include "wimlib/xml.h"
 
+/* Internal method for single-image deletion.  This doesn't set the
+ * image_deletion_occurred' flag on the WIMStruct.  */
+int
+delete_wim_image(WIMStruct *wim, int image)
+{
+       int ret;
+
+       /* Load the metadata for the image to be deleted.  This is necessary
+        * because blobs referenced by files in the image need to have their
+        * reference counts decremented.  */
+       ret = select_wim_image(wim, image);
+       if (ret)
+               return ret;
+
+       /* Release the reference to the image metadata and decrement reference
+        * counts on the blobs referenced by files in the image.  */
+       put_image_metadata(wim->image_metadata[image - 1], wim->blob_table);
+
+       /* Remove the empty slot from the image metadata array.  */
+       memmove(&wim->image_metadata[image - 1], &wim->image_metadata[image],
+               (wim->hdr.image_count - image) *
+                       sizeof(wim->image_metadata[0]));
+
+       /* Decrement the image count. */
+       --wim->hdr.image_count;
+
+       /* Remove the image from the XML information. */
+       xml_delete_image(wim->xml_info, image);
+
+       /* Fix the boot index. */
+       if (wim->hdr.boot_idx == image)
+               wim->hdr.boot_idx = 0;
+       else if (wim->hdr.boot_idx > image)
+               wim->hdr.boot_idx--;
+
+       /* The image is no longer valid.  */
+       wim->current_image = WIMLIB_NO_IMAGE;
+       return 0;
+}
+
 /* API function documented in wimlib.h  */
 WIMLIBAPI int
 wimlib_delete_image(WIMStruct *wim, int image)
@@ -37,52 +77,21 @@ wimlib_delete_image(WIMStruct *wim, int image)
        int ret;
        int first, last;
 
-       ret = can_delete_from_wim(wim);
-       if (ret)
-               return ret;
-
        if (image == WIMLIB_ALL_IMAGES) {
+               /* Deleting all images  */
                last = wim->hdr.image_count;
                first = 1;
        } else {
+               /* Deleting one image  */
                last = image;
                first = image;
        }
 
        for (image = last; image >= first; image--) {
-               DEBUG("Deleting image %d", image);
-
-               /* Even if the dentry tree is not allocated, we must select it (and
-                * therefore allocate it) so that we can decrement the reference counts
-                * in the lookup table.  */
-               ret = select_wim_image(wim, image);
+               ret = delete_wim_image(wim, image);
                if (ret)
                        return ret;
-
-               /* Unless the image metadata is shared by another WIMStruct, free the
-                * dentry tree, any lookup table entries that have their refcnt
-                * decremented to 0, and the security data. */
-               put_image_metadata(wim->image_metadata[image - 1], wim->lookup_table);
-
-               /* Get rid of the empty slot in the image metadata array. */
-               for (int i = image - 1; i < wim->hdr.image_count - 1; i++)
-                       wim->image_metadata[i] = wim->image_metadata[i + 1];
-
-               /* Decrement the image count. */
-               --wim->hdr.image_count;
-
-               /* Fix the boot index. */
-               if (wim->hdr.boot_idx == image)
-                       wim->hdr.boot_idx = 0;
-               else if (wim->hdr.boot_idx > image)
-                       wim->hdr.boot_idx--;
-
-               wim->current_image = WIMLIB_NO_IMAGE;
-
-               /* Remove the image from the XML information. */
-               xml_delete_image(&wim->wim_info, image);
-
-               wim->deletion_occurred = 1;
+               wim->image_deletion_occurred = 1;
        }
        return 0;
 }