fix_true_inode(): Fix bug

[wimlib] / src / hardlink.c

diff --git a/src/hardlink.c b/src/hardlink.c
index b88f4555f237a605a9cfc480668b86d25c6a3e33..93030d8f7e10e6b89874a089c28da575c63e850d 100644 (file)
--- a/src/hardlink.c
+++ b/src/hardlink.c
@@ -51,8 +51,34 @@
  *                            -----------------
  */
 
+/* Hash table to find inodes, identified by their inode ID.
+ * */
+struct inode_table {
+       /* Fields for the hash table */
+       struct hlist_head *array;
+       u64 num_entries;
+       u64 capacity;
+
+       /*
+        * Linked list of "extra" inodes.  These may be:
+        *
+        * - inodes with link count 1, which are all allowed to have 0 for their
+        *   inode number, meaning we cannot insert them into the hash table
+        *   before calling assign_inode_numbers().
+         *
+        * - Groups we create ourselves by splitting a nominal inode due to
+        *   inconsistencies in the dentries.  These inodes will share a inode
+        *   ID with some other inode until assign_inode_numbers() is called.
+        */
+       struct hlist_head extra_inodes;
+};
+
+static inline void destroy_inode_table(struct inode_table *table)
+{
+       FREE(table->array);
+}
 
-int init_inode_table(struct inode_table *table, size_t capacity)
+static int init_inode_table(struct inode_table *table, size_t capacity)
 {
        table->array = CALLOC(capacity, sizeof(table->array[0]));
        if (!table->array) {
@@ -65,8 +91,7 @@ int init_inode_table(struct inode_table *table, size_t capacity)
        return 0;
 }
 
-
-static size_t inode_link_count(const struct inode *inode)
+static inline size_t inode_link_count(const struct inode *inode)
 {
        const struct list_head *cur;
        size_t size = 0;
@@ -90,7 +115,7 @@ static size_t inode_link_count(const struct inode *inode)
  * we keep a linked list of the single dentries, and assign them inode
  * numbers later.
  */
-int inode_table_insert(struct dentry *dentry, void *__table)
+static int inode_table_insert(struct dentry *dentry, void *__table)
 {
        struct inode_table *table = __table;
        struct inode *d_inode = dentry->d_inode;
@@ -162,10 +187,12 @@ static void print_inode_dentries(const struct inode *inode)
 
 static void inconsistent_inode(const struct inode *inode)
 {
-       ERROR("An inconsistent hard link group that we cannot correct has been "
-             "detected");
+       ERROR("An inconsistent hard link group that cannot be corrected has "
+             "been detected");
        ERROR("The dentries are located at the following paths:");
+#ifdef ENABLE_ERROR_MESSAGES
        print_inode_dentries(inode);
+#endif
 }
 
 static bool ref_inodes_consistent(const struct inode * restrict ref_inode_1,
@@ -251,6 +278,8 @@ static int fix_true_inode(struct inode *inode, struct hlist_head *inode_list)
                                return WIMLIB_ERR_INVALID_DENTRY;
                        }
                        /* Free the unneeded `struct inode'. */
+                       dentry->d_inode->hlist.next = NULL;
+                       dentry->d_inode->hlist.pprev = NULL;
                        free_inode(dentry->d_inode);
                        dentry->d_inode = ref_inode;
                        ref_inode->link_count++;
@@ -277,10 +306,10 @@ static int fix_true_inode(struct inode *inode, struct hlist_head *inode_list)
  * inodes'.  There will be just one `struct inode' for each hard link group
  * remaining.
  */
-static int
-fix_nominal_inode(struct inode *inode, struct hlist_head *inode_list)
+static int fix_nominal_inode(struct inode *inode,
+                            struct hlist_head *inode_list)
 {
-       struct dentry *dentry, *ref_dentry;
+       struct dentry *dentry;
        struct hlist_node *cur, *tmp;
        int ret;
        size_t num_true_inodes;
@@ -315,7 +344,7 @@ fix_nominal_inode(struct inode *inode, struct hlist_head *inode_list)
        if (list_empty(&dentries_with_data_streams)) {
        #ifdef ENABLE_DEBUG
                if (inode->link_count > 1) {
-                       DEBUG("Found link group of size %zu without "
+                       DEBUG("Found link group of size %u without "
                              "any data streams:", inode->link_count);
                        print_inode_dentries(inode);
                        DEBUG("We are going to interpret it as true "
@@ -411,7 +440,7 @@ next_dentry_2:
  * split inodes as well as the inodes that were good before, is returned in the
  * list @inode_list.
  */
-int fix_inodes(struct inode_table *table, struct hlist_head *inode_list)
+static int fix_inodes(struct inode_table *table, struct hlist_head *inode_list)
 {
        struct inode *inode;
        struct hlist_node *cur, *tmp;
@@ -428,3 +457,21 @@ int fix_inodes(struct inode_table *table, struct hlist_head *inode_list)
                hlist_add_head(cur, inode_list);
        return 0;
 }
+
+int dentry_tree_fix_inodes(struct dentry *root, struct hlist_head *inode_list)
+{
+       struct inode_table inode_tab;
+       int ret;
+
+       DEBUG("Inserting dentries into inode table");
+       ret = init_inode_table(&inode_tab, 9001);
+       if (ret != 0)
+               return ret;
+
+       for_dentry_in_tree(root, inode_table_insert, &inode_tab);
+
+       DEBUG("Cleaning up the hard link groups");
+       ret = fix_inodes(&inode_tab, inode_list);
+       destroy_inode_table(&inode_tab);
+       return ret;
+}