X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fmodify.c;h=c979162a8ed11aadd2fd95ef012b7ef454a57cba;hp=ff6e106cfba9b9eeb6e2e2bedb2214f10a93c8bf;hb=68b023ab834029db11ce9bf150dadd9c80e59404;hpb=b6fa6d370a09247a608e6a46a3d9e761d10a8a39

diff --git a/src/modify.c b/src/modify.c
index ff6e106c..c979162a 100644
--- a/src/modify.c
+++ b/src/modify.c
@@ -745,6 +745,7 @@ int do_add_image(WIMStruct *w, const char *dir, const char *name,
 	struct image_metadata *imd;
 	struct wim_security_data *sd;
 	struct capture_config config;
+	struct link_group_table *lgt;
 	int ret;
 
 	DEBUG("Adding dentry tree from dir `%s'.", dir);
@@ -803,13 +804,20 @@ int do_add_image(WIMStruct *w, const char *dir, const char *name,
 	if (ret != 0)
 		goto out_free_dentry_tree;
 
+	lgt = w->image_metadata[w->hdr.image_count - 1].lgt;
 	DEBUG("Inserting dentries into hard link group table");
-	ret = for_dentry_in_tree(root_dentry, link_group_table_insert, 
-				 w->image_metadata[w->hdr.image_count - 1].lgt);
+	ret = for_dentry_in_tree(root_dentry, link_group_table_insert, lgt);
+				 
 	if (ret != 0)
 		goto out_destroy_imd;
-	DEBUG("Assigning hard link groups");
-	assign_link_groups(w->image_metadata[w->hdr.image_count - 1].lgt);
+
+	DEBUG("Cleanup up the hard link groups");
+	ret = fix_link_groups(lgt);
+	if (ret != 0)
+		goto out_destroy_imd;
+
+	DEBUG("Assigning hard link group IDs");
+	assign_link_group_ids(w->image_metadata[w->hdr.image_count - 1].lgt);
 
 	if (flags & WIMLIB_ADD_IMAGE_FLAG_BOOT)
 		wimlib_set_boot_idx(w, w->hdr.image_count);