Changeset 107 for trunk/lib

Timestamp:

04/18/08 00:51:21 (16 years ago)

Author:

tim

Message:

removed redundant hbin parsing

added flags parameter to regfi_open

File:

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

trunk/lib/regfi.c

@@ -883 +883 @@
  * first hbin offset.
  *******************************************************************/
-REGF_FILE* regfi_open(const char* filename)
+REGF_FILE* regfi_open(const char* filename, uint32 flags)
 {
   REGF_FILE* rb;
@@ -889 +889 @@
   uint32 hbin_off;
   int fd;
-  int flags = O_RDONLY;
-  bool rla;
+  bool rla, save_unalloc = false;
+
+  if(flags & REGFI_FLAG_SAVE_UNALLOC)
+    save_unalloc = true;
 
   /* open an existing file */
-  if ((fd = open(filename, flags)) == -1) 
+  if ((fd = open(filename, O_RDONLY)) == -1) 
   {
     /* DEBUG(0,("regfi_open: failure to open %s (%s)\n", filename, strerror(errno)));*/
@@ -920 +922 @@
   rla = true;
   hbin_off = REGF_BLOCKSIZE;
-  hbin = regfi_parse_hbin(rb, hbin_off, true, false);
+  hbin = regfi_parse_hbin(rb, hbin_off, true, save_unalloc);
   while(hbin && rla)
   {
     hbin_off = hbin->file_off + hbin->block_size;
     rla = range_list_add(rb->hbins, hbin->file_off, hbin->block_size, hbin);
-    hbin = regfi_parse_hbin(rb, hbin_off, true, false); 
+    hbin = regfi_parse_hbin(rb, hbin_off, true, save_unalloc);
   }
 
@@ -968 +970 @@
   REGF_NK_REC* nk = NULL;
   REGF_HBIN*   hbin;
-  uint32       offset = REGF_BLOCKSIZE;
-  uint32       root_offset;
+  uint32       root_offset, i, num_hbins;
   
   if(!file)
@@ -979 +980 @@
      block (but I'm not assuming that for now) */
 
-  while((hbin = regfi_parse_hbin(file, offset, true, false))) 
-  {
+  num_hbins = range_list_size(file->hbins);
+  for(i=0; i < num_hbins; i++)
+  {
+    hbin = (REGF_HBIN*)range_list_get(file->hbins, i)->data;
     if(regfi_find_root_nk(file, hbin->file_off+HBIN_HEADER_REC_SIZE, 
                           hbin->block_size-HBIN_HEADER_REC_SIZE, &root_offset))
@@ -987 +990 @@
       break;
     }
-
-    offset += hbin->block_size;
   }