]> wimlib.net Git - wimlib/blobdiff - src/lookup_table.c
inode updates (IN PROGRESS)
[wimlib] / src / lookup_table.c
index 5408baf654a80756411ff51e18c9404b69e894bb..61cc5a83ab663132dc87b3d581c9b64fefa2862a 100644 (file)
 #include "io.h"
 #include <errno.h>
 
+#ifdef WITH_FUSE
+#include <unistd.h>
+#endif
+
 struct lookup_table *new_lookup_table(size_t capacity)
 {
        struct lookup_table *table;
@@ -66,7 +70,6 @@ struct lookup_table_entry *new_lookup_table_entry()
 
        lte->part_number  = 1;
        lte->refcnt       = 1;
-       INIT_LIST_HEAD(&lte->lte_group_list);
        return lte;
 }
 
@@ -99,6 +102,7 @@ void free_lookup_table_entry(struct lookup_table_entry *lte)
                default:
                        break;
                }
+               FREE(lte->extracted_file);
                FREE(lte);
        }
 }
@@ -141,7 +145,14 @@ void lookup_table_insert(struct lookup_table *table,
        table->num_entries++;
 }
 
-
+static void finalize_lte(struct lookup_table_entry *lte)
+{
+       #ifdef WITH_FUSE
+       if (lte->resource_location == RESOURCE_IN_STAGING_FILE)
+               unlink(lte->staging_file_name);
+       #endif
+       free_lookup_table_entry(lte);
+}
 
 /* Decrements the reference count for the lookup table entry @lte.  If its
  * reference count reaches 0, it is unlinked from the lookup table.  If,
@@ -158,7 +169,7 @@ lte_decrement_refcnt(struct lookup_table_entry *lte, struct lookup_table *table)
                        if (lte->num_opened_fds == 0)
                #endif
                        {
-                               free_lookup_table_entry(lte);
+                               finalize_lte(lte);
                                lte = NULL;
                        }
                }
@@ -166,6 +177,23 @@ lte_decrement_refcnt(struct lookup_table_entry *lte, struct lookup_table *table)
        return lte;
 }
 
+#ifdef WITH_FUSE
+struct lookup_table_entry *
+lte_decrement_num_opened_fds(struct lookup_table_entry *lte,
+                            struct lookup_table *table)
+{
+       if (lte) {
+               wimlib_assert(lte->num_opened_fds);
+               if (--lte->num_opened_fds == 0 && lte->refcnt == 0) {
+                       lookup_table_unlink(table, lte);
+                       finalize_lte(lte);
+                       lte = NULL;
+               }
+       }
+       return lte;
+}
+#endif
+
 /* 
  * Calls a function on all the entries in the lookup table.  Stop early and
  * return nonzero if any call to the function returns nonzero.
@@ -330,10 +358,22 @@ int write_lookup_table_entry(struct lookup_table_entry *lte, void *__out)
 }
 
 
+int lte_zero_real_refcnt(struct lookup_table_entry *lte, void *ignore)
+{
+       lte->real_refcnt = 0;
+       return 0;
+}
+
+int lte_zero_out_refcnt(struct lookup_table_entry *lte, void *ignore)
+{
+       lte->out_refcnt = 0;
+       return 0;
+}
 
-int zero_out_refcnts(struct lookup_table_entry *entry, void *ignore)
+int lte_free_extracted_file(struct lookup_table_entry *lte, void *ignone)
 {
-       entry->out_refcnt = 0;
+       FREE(lte->extracted_file);
+       lte->extracted_file = NULL;
        return 0;
 }
 
@@ -422,18 +462,19 @@ __lookup_resource(const struct lookup_table *table, const u8 hash[])
  * Finds the dentry, lookup table entry, and stream index for a WIM file stream,
  * given a path name.
  *
- * This is only for pre-resolved dentries.
+ * This is only for pre-resolved inodes.
  */
 int lookup_resource(WIMStruct *w, const char *path,
                    int lookup_flags,
                    struct dentry **dentry_ret,
                    struct lookup_table_entry **lte_ret,
-                   unsigned *stream_idx_ret)
+                   u16 *stream_idx_ret)
 {
        struct dentry *dentry;
        struct lookup_table_entry *lte;
-       unsigned stream_idx;
+       u16 stream_idx;
        const char *stream_name = NULL;
+       struct inode *inode;
        char *p = NULL;
 
        if (lookup_flags & LOOKUP_FLAG_ADS_OK) {
@@ -450,26 +491,29 @@ int lookup_resource(WIMStruct *w, const char *path,
        if (!dentry)
                return -ENOENT;
 
-       wimlib_assert(dentry->resolved);
+       inode = dentry->inode;
+
+       wimlib_assert(inode->resolved);
 
-       lte = dentry->lte;
        if (!(lookup_flags & LOOKUP_FLAG_DIRECTORY_OK)
-             && dentry_is_directory(dentry))
+             && inode_is_directory(inode))
                return -EISDIR;
-       stream_idx = 0;
+
        if (stream_name) {
-               size_t stream_name_len = strlen(stream_name);
-               for (u16 i = 0; i < dentry->num_ads; i++) {
-                       if (ads_entry_has_name(&dentry->ads_entries[i],
-                                              stream_name,
-                                              stream_name_len))
-                       {
-                               stream_idx = i + 1;
-                               lte = dentry->ads_entries[i].lte;
-                               goto out;
-                       }
+               struct ads_entry *ads_entry;
+               u16 ads_idx;
+               ads_entry = inode_get_ads_entry(inode, stream_name,
+                                               &ads_idx);
+               if (ads_entry) {
+                       stream_idx = ads_idx + 1;
+                       lte = ads_entry->lte;
+                       goto out;
+               } else {
+                       return -ENOENT;
                }
-               return -ENOENT;
+       } else {
+               lte = inode->lte;
+               stream_idx = 0;
        }
 out:
        if (dentry_ret)
@@ -481,6 +525,24 @@ out:
        return 0;
 }
 
+static void inode_resolve_ltes(struct inode *inode, struct lookup_table *table)
+{
+       struct lookup_table_entry *lte;
+
+       /* Resolve the default file stream */
+       lte = __lookup_resource(table, inode->hash);
+       inode->lte = lte;
+       inode->resolved = true;
+
+       /* Resolve the alternate data streams */
+       for (u16 i = 0; i < inode->num_ads; i++) {
+               struct ads_entry *cur_entry = inode->ads_entries[i];
+
+               lte = __lookup_resource(table, cur_entry->hash);
+               cur_entry->lte = lte;
+       }
+}
+
 /* Resolve a dentry's lookup table entries 
  *
  * This replaces the SHA1 hash fields (which are used to lookup an entry in the
@@ -490,65 +552,36 @@ out:
  * This function always succeeds; unresolved lookup table entries are given a
  * NULL pointer.
  */
-int dentry_resolve_ltes(struct dentry *dentry, void *__table)
+int dentry_resolve_ltes(struct dentry *dentry, void *table)
 {
-       struct lookup_table *table = __table;
-       struct lookup_table_entry *lte;
-
-       if (dentry->resolved)
-               return 0;
-
-       /* Resolve the default file stream */
-       lte = __lookup_resource(table, dentry->hash);
-       if (lte)
-               list_add(&dentry->lte_group_list.list, &lte->lte_group_list);
-       else
-               INIT_LIST_HEAD(&dentry->lte_group_list.list);
-       dentry->lte = lte;
-       dentry->lte_group_list.type = STREAM_TYPE_NORMAL;
-       dentry->resolved = true;
-
-       /* Resolve the alternate data streams */
-       if (dentry->ads_entries_status != ADS_ENTRIES_USER) {
-               for (u16 i = 0; i < dentry->num_ads; i++) {
-                       struct ads_entry *cur_entry = &dentry->ads_entries[i];
-
-                       lte = __lookup_resource(table, cur_entry->hash);
-                       if (lte)
-                               list_add(&cur_entry->lte_group_list.list,
-                                        &lte->lte_group_list);
-                       else
-                               INIT_LIST_HEAD(&cur_entry->lte_group_list.list);
-                       cur_entry->lte = lte;
-                       cur_entry->lte_group_list.type = STREAM_TYPE_ADS;
-               }
-       }
+       if (!dentry->inode->resolved)
+               inode_resolve_ltes(dentry->inode, table);
        return 0;
 }
 
-/* Return the lookup table entry for the unnamed data stream of a dentry, or
+/* Return the lookup table entry for the unnamed data stream of an inode, or
  * NULL if there is none.
  *
  * You'd think this would be easier than it actually is, since the unnamed data
- * stream should be the one referenced from the dentry itself.  Alas, if there
+ * stream should be the one referenced from the inode itself.  Alas, if there
  * are named data streams, Microsoft's "imagex.exe" program will put the unnamed
- * data stream in one of the alternate data streams instead of inside the
- * dentry.  So we need to check the alternate data streams too.
+ * data stream in one of the alternate data streams instead of inside the WIM
+ * dentry itself.  So we need to check the alternate data streams too.
  *
  * Also, note that a dentry may appear to have than one unnamed stream, but if
  * the SHA1 message digest is all 0's then the corresponding stream does not
- * really "count" (this is the case for the dentry's own file stream when the
+ * really "count" (this is the case for the inode's own file stream when the
  * file stream that should be there is actually in one of the alternate stream
  * entries.).  This is despite the fact that we may need to extract such a
  * missing entry as an empty file or empty named data stream.
  */
 struct lookup_table_entry *
-dentry_unnamed_lte(const struct dentry *dentry,
+inode_unnamed_lte(const struct inode *inode,
                   const struct lookup_table *table)
 {
-       if (dentry->resolved)
-               return dentry_unnamed_lte_resolved(dentry);
+       if (inode->resolved)
+               return inode_unnamed_lte_resolved(inode);
        else
-               return dentry_unnamed_lte_unresolved(dentry, table);
+               return inode_unnamed_lte_unresolved(inode, table);
 }