Consolidate `struct lookup_table_entry'
authorEric Biggers <ebiggers3@gmail.com>
Sun, 18 Nov 2012 19:01:58 +0000 (13:01 -0600)
committerEric Biggers <ebiggers3@gmail.com>
Sun, 18 Nov 2012 19:01:58 +0000 (13:01 -0600)
src/extract.c
src/lookup_table.c
src/lookup_table.h
src/modify.c

index 6e341cc0f396c414992efe0c5e7935750a26cd71..d2f2a3547c1152381f1328e4d1f0189c8e1d691d 100644 (file)
@@ -459,7 +459,7 @@ WIMLIBAPI int wimlib_extract_image(WIMStruct *w, int image,
                w->lookup_table = joined_tab;
        }
 
-       for_lookup_table_entry(w->lookup_table, lte_free_extracted_file, NULL);
+       for_lookup_table_entry(w->lookup_table, lte_zero_extracted_file, NULL);
 
        if (image == WIM_ALL_IMAGES) {
                flags |= WIMLIB_EXTRACT_FLAG_MULTI_IMAGE;
@@ -472,6 +472,7 @@ WIMLIBAPI int wimlib_extract_image(WIMStruct *w, int image,
                free_lookup_table(w->lookup_table);
                w->lookup_table = w_tab_save;
        }
+       for_lookup_table_entry(w->lookup_table, lte_free_extracted_file, NULL);
        return ret;
 
 }
index 36d2d13f69430518dcdafbf59b2649a75af25152..f0da2b423c59c521612aa7d4f23033bc85b693e4 100644 (file)
@@ -156,7 +156,6 @@ void free_lookup_table_entry(struct lookup_table_entry *lte)
                default:
                        break;
                }
-               FREE(lte->extracted_file);
                FREE(lte);
        }
 }
@@ -423,13 +422,21 @@ int lte_zero_out_refcnt(struct lookup_table_entry *lte, void *ignore)
        return 0;
 }
 
-int lte_free_extracted_file(struct lookup_table_entry *lte, void *ignone)
+int lte_zero_extracted_file(struct lookup_table_entry *lte, void *ignore)
 {
-       FREE(lte->extracted_file);
        lte->extracted_file = NULL;
        return 0;
 }
 
+int lte_free_extracted_file(struct lookup_table_entry *lte, void *ignore)
+{
+       if (lte->extracted_file != NULL) {
+               FREE(lte->extracted_file);
+               lte->extracted_file = NULL;
+       }
+       return 0;
+}
+
 void print_lookup_table_entry(const struct lookup_table_entry *lte)
 {
        if (!lte) {
index dbcf55543254fc18ca48e78d7b18ff11a493b87e..c870a114cd28cb47826e797e1b84d7368e5d6263 100644 (file)
@@ -36,6 +36,45 @@ struct ntfs_location {
 };
 #endif
 
+/* An enumerated type that identifies where the stream corresponding to this
+ * lookup table entry is actually located.
+ *
+ * If we open a WIM and read its lookup table, the location is set to
+ * RESOURCE_IN_WIM since all the streams will initially be located in the WIM.
+ * However, to deal with problems such as image capture and image mount, we
+ * allow the actual location of the stream to be somewhere else, such as an
+ * external file.
+ */
+enum resource_location {
+       /* The lookup table entry does not correspond to a stream (this state
+        * should exist only temporarily) */
+       RESOURCE_NONEXISTENT = 0,
+
+       /* The stream resource is located in a WIM file.  The WIMStruct for the
+        * WIM file will be pointed to by the @wim member. */
+       RESOURCE_IN_WIM,
+
+       /* The stream resource is located in an external file.  The name of the
+        * file will be provided by @file_on_disk member.  In addition, if
+        * @file_on_disk_fp is not NULL, it will be an open FILE * to the file.
+        * */
+       RESOURCE_IN_FILE_ON_DISK,
+
+       /* The stream resource is located in an external file in the staging
+        * directory for a read-write mount.  */
+       RESOURCE_IN_STAGING_FILE,
+
+       /* The stream resource is directly attached in an in-memory buffer
+        * pointed to by @attached_buffer. */
+       RESOURCE_IN_ATTACHED_BUFFER,
+
+       /* The stream resource is located in an NTFS volume.  It is identified
+        * by volume, filename, data stream name, and by whether it is a reparse
+        * point or not. @ntfs_loc points to a structure containing this
+        * information. */
+       RESOURCE_IN_NTFS_VOLUME,
+};
+
 /*
  * An entry in the lookup table in the WIM file.
  *
@@ -50,10 +89,7 @@ struct ntfs_location {
 struct lookup_table_entry {
 
        /* List of lookup table entries in this hash bucket */
-       union {
-               struct hlist_node hash_list;
-               struct list_head list;
-       };
+       struct hlist_node hash_list;
 
        /* Location and size of the stream in the WIM, whether it is compressed
         * or not, and whether it's a metadata resource or not.  This is an
@@ -74,44 +110,8 @@ struct lookup_table_entry {
         */
        u16 part_number;
 
-       /* An enumerated type that identifies where the stream corresponding to
-        * this lookup table entry is actually located.
-        *
-        * Obviously if we open a WIM and read its lookup table, the location is
-        * set to RESOURCE_IN_WIM since all the streams will initially be
-        * located in the WIM.  However, to deal with problems such as image
-        * capture and image mount, we allow the actual location of the stream
-        * to be somewhere else, such as an external file.
-        */
-       enum {
-               /* The lookup table entry does not correspond to a stream (this
-                * state should exist only temporarily) */
-               RESOURCE_NONEXISTENT = 0,
-
-               /* The stream resource is located in a WIM file.  The WIMStruct
-                * for the WIM file will be pointed to by the @wim member. */
-               RESOURCE_IN_WIM,
-
-               /* The stream resource is located in an external file.  The
-                * name of the file will be provided by @file_on_disk member.
-                * In addition, if @file_on_disk_fp is not NULL, it will be an
-                * open FILE * to the file. */
-               RESOURCE_IN_FILE_ON_DISK,
-
-               /* The stream resource is located in an external file in the
-                * staging directory for a read-write mount.  */
-               RESOURCE_IN_STAGING_FILE,
-
-               /* The stream resource is directly attached in an in-memory
-                * buffer pointed to by @attached_buffer. */
-               RESOURCE_IN_ATTACHED_BUFFER,
-
-               /* The stream resource is located in an NTFS volume.  It is
-                * identified by volume, filename, data stream name, and by
-                * whether it is a reparse point or not. @ntfs_loc points to a
-                * structure containing this information. */
-               RESOURCE_IN_NTFS_VOLUME,
-       } resource_location;
+       /* See enum resource_location above */
+       u16 resource_location;
 
        /* (On-disk field)
         * Number of times this lookup table entry is referenced by dentries.
@@ -133,6 +133,10 @@ struct lookup_table_entry {
                size_t hash_short;
        };
 
+       #ifdef WITH_FUSE
+       u16 num_opened_fds;
+       #endif
+
        /* Pointers to somewhere where the stream is actually located.  See the
         * comments for the @resource_location field above. */
        union {
@@ -160,9 +164,6 @@ struct lookup_table_entry {
                 * RESOURCE_IN_STAGING_FILE) */
                struct inode *lte_inode;
        };
-#ifdef WITH_FUSE
-       u16 num_opened_fds;
-#endif
 
        /* When a WIM file is written, out_refcnt starts at 0 and is incremented
         * whenever the file resource pointed to by this lookup table entry
@@ -173,25 +174,26 @@ struct lookup_table_entry {
 
        u32 real_refcnt;
 
-       /* When a WIM file is written, @output_resource_entry is filled
-        * in with the resource entry for the output WIM.  This will not
-        * necessarily be the same as the @resource_entry since:
-        *      - The stream may have a different offset in the new WIM
-        *      - The stream may have a different compressed size in the
-        *      new WIM if the compression type changed
-        */
        union {
+               /* When a WIM file is written, @output_resource_entry is filled
+                * in with the resource entry for the output WIM.  This will not
+                * necessarily be the same as the @resource_entry since: - The
+                * stream may have a different offset in the new WIM - The
+                * stream may have a different compressed size in the new WIM if
+                * the compression type changed
+                */
                struct resource_entry output_resource_entry;
+
                struct list_head msg_list;
        };
 
-       /* This field is used for the special hardlink or symlink image
-        * extraction mode.   In these mode, all identical files are linked
-        * together, and @extracted_file will be set to the filename of the
-        * first extracted file containing this stream.  */
-       char *extracted_file;
-
        union {
+               /* This field is used for the special hardlink or symlink image
+                * extraction mode.   In these mode, all identical files are linked
+                * together, and @extracted_file will be set to the filename of the
+                * first extracted file containing this stream.  */
+               char *extracted_file;
+
                /* List of lookup table entries that correspond to streams that have
                 * been extracted to the staging directory when modifying a read-write
                 * mounted WIM. */
@@ -270,7 +272,8 @@ extern void lte_decrement_num_opened_fds(struct lookup_table_entry *lte);
 
 extern int lte_zero_out_refcnt(struct lookup_table_entry *entry, void *ignore);
 extern int lte_zero_real_refcnt(struct lookup_table_entry *entry, void *ignore);
-extern int lte_free_extracted_file(struct lookup_table_entry *lte, void *ignone);
+extern int lte_zero_extracted_file(struct lookup_table_entry *lte, void *ignore);
+extern int lte_free_extracted_file(struct lookup_table_entry *lte, void *ignore);
 
 extern void print_lookup_table_entry(const struct lookup_table_entry *entry);
 
index 8fb481dc219caa4ea0584e41cdc8947f389d3ce2..4d3b6d41087069ac769d25d278394567da14a450 100644 (file)
@@ -300,7 +300,7 @@ static int allocate_lte_if_needed(struct dentry *dentry, void *arg)
                                dest_lte = clone_lookup_table_entry(src_lte);
                                if (!dest_lte)
                                        return WIMLIB_ERR_NOMEM;
-                               list_add_tail(&dest_lte->list, lte_list_head);
+                               list_add_tail(&dest_lte->staging_list, lte_list_head);
                        }
                }
        }
@@ -348,7 +348,8 @@ static int add_lte_to_dest_wim(struct dentry *dentry, void *arg)
 
                        next = lte_list_head->next;
                        list_del(next);
-                       dest_lte = container_of(next, struct lookup_table_entry, list);
+                       dest_lte = container_of(next, struct lookup_table_entry,
+                                               staging_list);
                        dest_lte->part_number = 1;
                        dest_lte->refcnt = 1;
                        wimlib_assert(hashes_equal(dest_lte->hash, src_lte->hash));
@@ -589,7 +590,7 @@ out_xml_delete_image:
 out_free_ltes:
        {
                struct lookup_table_entry *lte, *tmp;
-               list_for_each_entry_safe(lte, tmp, &wims.lte_list_head, list)
+               list_for_each_entry_safe(lte, tmp, &wims.lte_list_head, staging_list)
                        free_lookup_table_entry(lte);
        }