Changeset 109 for trunk/lib

Timestamp:

04/29/08 16:38:27 (17 years ago)

Author:

tim

Message:

reworked SK record caching to prevent unnecessary loads.

SK records now cached in iterators on an as-needed basis

File:

• trunk/lib/regfi.c (modified) (7 diffs)

trunk/lib/regfi.c

@@ -681 +681 @@
 
 
+
 /*******************************************************************
-
-static REGF_SK_REC* find_sk_record_by_sec_desc( REGF_FILE *file, SEC_DESC *sd )
-{
-  REGF_SK_REC *p;
-
-  for ( p=file->sec_desc_list; p; p=p->next ) {
-    if ( sec_desc_equal( p->sec_desc, sd ) )
-      return p;
-  }
-
-
-  return NULL;
-}
- *******************************************************************/
-
-/*******************************************************************
- * TODO: Need to add full key and SK record caching using a 
+ * TODO: Need to add full key caching using a 
  *       custom cache structure.
  *******************************************************************/
-REGF_NK_REC* regfi_load_key(REGF_FILE *file, uint32 offset, bool strict)
+REGF_NK_REC* regfi_load_key(REGF_FILE* file, uint32 offset, bool strict)
 {
   REGF_HBIN* hbin;
@@ -780 +765 @@
   }
 
-  /* get the security descriptor.  First look if we have already parsed it */
-  if((nk->sk_off!=REGF_OFFSET_NONE)
-     && !(nk->sec_desc = (REGF_SK_REC*)lru_cache_find(file->sk_recs, 
-                                                      &nk->sk_off, 4)))
-  {
-    sub_hbin = hbin;
-    if(!regfi_offset_in_hbin(hbin, nk->sk_off))
-      sub_hbin = regfi_lookup_hbin(file, nk->sk_off);
-
-    if(sub_hbin == NULL)
-    {
-      free(nk);
-      /* TODO: need convenient way to free nk->values and nk->subkeys deeply 
-       *       in all cases. 
-       */
-      return NULL;
-    }
-
-    off = nk->sk_off + REGF_BLOCKSIZE;
-    max_length = sub_hbin->block_size + sub_hbin->file_off - off;
-    nk->sec_desc = regfi_parse_sk(file, off, max_length, true);
-    if(strict && nk->sec_desc == NULL)
-    {
-      free(nk);
-      /* TODO: need convenient way to free nk->values and nk->subkeys deeply 
-       *       in all cases. 
-       */
-      return NULL;
-    }
-    nk->sec_desc->sk_off = nk->sk_off;
-    
-    lru_cache_update(file->sk_recs, &nk->sk_off, 4, nk->sec_desc);
-  }
-  
   return nk;
 }
@@ -903 +854 @@
   }
   
-  /* TODO: come up with a better secret. */
-  rb->sk_recs = lru_cache_create(127, 0xDEADBEEF, true);
-
   rla = true;
   hbin_off = REGF_BLOCKSIZE;
@@ -942 +890 @@
   range_list_free(file->unalloc_cells);
 
-  lru_cache_destroy(file->sk_recs);
-
   free(file);
 
@@ -1038 +984 @@
   }
 
+  /* TODO: come up with a better secret. */
+  ret_val->sk_recs = lru_cache_create(127, 0xDEADBEEF, true);
+
   ret_val->f = fh;
   ret_val->cur_key = root;
@@ -1062 +1011 @@
   }
   
+  lru_cache_destroy(i->sk_recs);
+
   free(i);
 }
@@ -1201 +1152 @@
   return i->cur_key;
 }
+
+
+/******************************************************************************
+ *****************************************************************************/
+const REGF_SK_REC* regfi_iterator_cur_sk(REGFI_ITERATOR* i)
+{
+  REGF_SK_REC* ret_val;
+  REGF_HBIN* hbin;
+  uint32 max_length, off;
+
+  if(i->cur_key == NULL)
+    return NULL;
+  
+  /* First look if we have already parsed it */
+  if((i->cur_key->sk_off!=REGF_OFFSET_NONE)
+     && !(ret_val =(REGF_SK_REC*)lru_cache_find(i->sk_recs, 
+                                                &i->cur_key->sk_off, 4)))
+  {
+    hbin = regfi_lookup_hbin(i->f, i->cur_key->sk_off);
+
+    if(hbin == NULL)
+      return NULL;
+
+    off = i->cur_key->sk_off + REGF_BLOCKSIZE;
+    max_length = hbin->block_size + hbin->file_off - off;
+    ret_val = regfi_parse_sk(i->f, off, max_length, true);
+    if(ret_val == NULL)
+      return NULL;
+
+    ret_val->sk_off = i->cur_key->sk_off;
+    lru_cache_update(i->sk_recs, &i->cur_key->sk_off, 4, ret_val);
+  }
+
+  return ret_val;
+}
+