Changeset 213 for trunk

Timestamp:

03/25/11 11:48:27 (14 years ago)

Author:

tim

Message:

fixed some pyregfi parameter bugs
fixed some iterator memory management problems
updated smoketest script to use ctypes pyregfi

Location:

trunk

Files:

• lib/regfi.c (modified) (3 diffs)
• python/pyregfi/__init__.py (modified) (14 diffs)
• python/pyregfi/structures.py (modified) (1 diff)
• src/reglookup.c (modified) (3 diffs)

trunk/lib/regfi.c

@@ -1780 +1780 @@
   }
   ret_val->cur_key = root;
-  talloc_reference(ret_val, root);
+  talloc_reparent(NULL, ret_val, root);
 
   ret_val->key_positions = void_stack_new(REGFI_MAX_DEPTH);
@@ -1834 +1834 @@
     return false;
   }
-  talloc_reference(i, subkey);
+  talloc_reparent(NULL, i, subkey);
 
   i->cur_key = subkey;
@@ -1920 +1920 @@
 const REGFI_NK* regfi_iterator_cur_key(REGFI_ITERATOR* i)
 {
-  /* XXX: do we need to add a NULL talloc reference here? */
+  talloc_reference(NULL, i->cur_key);
   return i->cur_key;
 }

trunk/python/pyregfi/__init__.py

@@ -15 +15 @@
 
 
-regfi.regfi_alloc.argtypes = [c_int]
+regfi.regfi_alloc.argtypes = [c_int, REGFI_ENCODING]
 regfi.regfi_alloc.restype = POINTER(REGFI_FILE)
 
-regfi.regfi_alloc_cb.argtypes = [POINTER(REGFI_RAW_FILE)]
+regfi.regfi_alloc_cb.argtypes = [POINTER(REGFI_RAW_FILE), REGFI_ENCODING]
 regfi.regfi_alloc_cb.restype = POINTER(REGFI_FILE)
 
@@ -140 +140 @@
     s = chars_pointer[i]
     while s != None:
-        ret_val.append(s.decode('utf-8'))
+        ret_val.append(s.decode('utf-8', 'replace'))
         i += 1
         s = chars_pointer[i]
@@ -150 +150 @@
 #  access to base structures  
 class _StructureWrapper(object):
-
     hive = None
     base = None
@@ -161 +160 @@
     def __del__(self):
         regfi.regfi_free_record(self.base)
+        hive = None
 
     def __getattr__(self, name):
@@ -205 +205 @@
         self.length = self.fetch_num(key.base)
     
+    def __del__(self):
+        self.key = None
+
     def __len__(self):
         return self.length
@@ -274 +277 @@
                 ret_val = self.name_raw
             else:
-                ret_val = ret_val.decode('utf-8')
+                ret_val = ret_val.decode('utf-8', 'replace')
                 
         elif name == "name_raw":
@@ -307 +310 @@
             elif data_struct.type in (REG_SZ, REG_EXPAND_SZ, REG_LINK):
                 # Unicode strings
-                ret_val = data_struct.interpreted.string.decode('utf-8')
+                ret_val = data_struct.interpreted.string.decode('utf-8', 'replace')
             elif data_struct.type in (REG_DWORD, REG_DWORD_BE):
                 # 32 bit integers
@@ -343 +346 @@
                     ret_val = self.name_raw
                 else:
-                    ret_val = ret_val.decode('utf-8')
+                    ret_val = ret_val.decode('utf-8', 'replace')
 
             elif name == "name_raw":
@@ -370 +373 @@
         try:
             if hasattr(fh, 'fileno'):
-                self.file = regfi.regfi_alloc(fh.fileno())
+                self.file = regfi.regfi_alloc(fh.fileno(), REGFI_ENCODING_UTF8)
                 return
         except:
@@ -379 +382 @@
         self.raw_file.seek = seek_cb_type(self.raw_file.cb_seek)
         self.raw_file.read = read_cb_type(self.raw_file.cb_read)
-        self.file = regfi.regfi_alloc_cb(self.raw_file)
+        self.file = regfi.regfi_alloc_cb(self.raw_file, REGFI_ENCODING_UTF8)
 
     def __getattr__(self, name):
@@ -387 +390 @@
         regfi.regfi_free(self.file)
         if self.raw_file != None:
-            regfi.regfi_free(self.file)
-            
+            self.raw_file = None
 
     def __iter__(self):
@@ -421 +423 @@
 
     def __init__(self, hive):
-        # REGFI_ENCODING_UTF8==1
-        self.iter = regfi.regfi_iterator_new(hive.file, 1)
+        self.iter = regfi.regfi_iterator_new(hive.file, REGFI_ENCODING_UTF8)
         if self.iter == None:
             raise Exception("Could not create iterator.  Current log:\n"
@@ -432 +433 @@
 
     def __del__(self):    
-        regfi.regfi_iterator_free(self.iter)        
+        regfi.regfi_iterator_free(self.iter)
 
     def __iter__(self):
@@ -440 +441 @@
     def __next__(self):
         if self.iteration_root == None:
-            self.iteration_root = self.current_key()            
+            self.iteration_root = self.current_key()
         elif not regfi.regfi_iterator_down(self.iter):
             up_ret = regfi.regfi_iterator_up(self.iter)

trunk/python/pyregfi/structures.py

@@ -10 +10 @@
 # XXX: can we always be sure enums are this size?
 REGFI_ENCODING = c_uint32
+REGFI_ENCODING_UTF8 = 1
+
 REGFI_DATA_TYPE = c_uint32
 

trunk/src/reglookup.c

@@ -406 +406 @@
       if(cur != root)
       {
+        regfi_free_record(cur);
         /* We're done with this sub-tree, going up and hitting other branches. */
         if(!regfi_iterator_up(iter))
@@ -429 +430 @@
        * Let's move down and print this first sub-tree out. 
        */
+      regfi_free_record(cur);
       if(!regfi_iterator_down(iter))
       {
@@ -443 +445 @@
     printMsgs(iter->f);
   } while(!((cur == root) && (sub == NULL)));
+  regfi_free_record(root);
 
   if(print_verbose)