Changeset 206 for trunk/python/pyregfi/__init__.py

Timestamp:

08/25/10 11:20:32 (14 years ago)

Author:

tim

Message:

simplified part of regfi API to move string encoding to the REGFI_FILE object

additional pyregfi implementation

File:

• trunk/python/pyregfi/__init__.py (modified) (7 diffs)

trunk/python/pyregfi/__init__.py

@@ -29 +29 @@
 regfi.regfi_free_record.restype = None
 
+regfi.regfi_fetch_classname.argtypes = [POINTER(REGFI_FILE), POINTER(REGFI_NK)]
+regfi.regfi_fetch_classname.restype = POINTER(REGFI_CLASSNAME)
+
+regfi.regfi_fetch_sk.argtypes = [POINTER(REGFI_FILE), POINTER(REGFI_NK)]
+regfi.regfi_fetch_sk.restype = POINTER(REGFI_SK)
+
+regfi.regfi_fetch_data.argtypes = [POINTER(REGFI_FILE), POINTER(REGFI_VK)]
+regfi.regfi_fetch_data.restype = POINTER(REGFI_DATA)
+
 regfi.regfi_iterator_new.argtypes = [POINTER(REGFI_FILE), REGFI_ENCODING]
 regfi.regfi_iterator_new.restype = POINTER(REGFI_ITERATOR)
@@ -50 +59 @@
 regfi.regfi_iterator_cur_key.restype = POINTER(REGFI_NK)
 
-regfi.regfi_iterator_cur_sk.argtypes = [POINTER(REGFI_ITERATOR)]
-regfi.regfi_iterator_cur_sk.restype = POINTER(REGFI_SK)
-
 regfi.regfi_iterator_first_subkey.argtypes = [POINTER(REGFI_ITERATOR)]
 regfi.regfi_iterator_first_subkey.restype = c_bool
@@ -76 +82 @@
 regfi.regfi_iterator_find_value.argtypes = [POINTER(REGFI_ITERATOR), c_char_p]
 regfi.regfi_iterator_find_value.restype = c_bool
-
-# XXX: possibly move REGFI_ENCODING to file object and eliminate need for ITERATOR here.
-regfi.regfi_iterator_fetch_classname.argtypes = [POINTER(REGFI_ITERATOR), POINTER(REGFI_NK)]
-regfi.regfi_iterator_fetch_classname.restype = POINTER(REGFI_CLASSNAME)
-
-regfi.regfi_iterator_fetch_data.argtypes = [POINTER(REGFI_ITERATOR), POINTER(REGFI_VK)]
-regfi.regfi_iterator_fetch_data.restype = POINTER(REGFI_DATA)
 
 
@@ -91 +90 @@
 
 def GetLogMessages():
-    return regfi.regfi_log_get_str()
-
+    msgs = regfi.regfi_log_get_str()
+    if msgs == None:
+        return ''
+    return msgs
+
+
+class _StructureWrapper():
+    "Handles memory management and proxies attribute access to base structures"
+    base = None
+
+    def __init__(self, base):
+        # XXX: check for NULL here, throw an exception if so.
+        self.base = base
+
+    def __del__(self):
+        regfi.regfi_free_record(self.base)
+
+    def __getattr__(self, name):
+        return getattr(self.base.contents, name)
+
+    def __eq__(self, other):
+        return (type(self) == type(other)) and (self.offset == other.offset)
+
+    def __ne__(self, other):
+        return (not self.__eq__(other))
+
+
+class Key(_StructureWrapper):
+    pass
+
+class Value(_StructureWrapper):
+    pass
+
+class Data(_StructureWrapper):
+    pass
+
+class Security(_StructureWrapper):
+    pass
 
 class Hive():    
@@ -126 +161 @@
         return HiveIterator(self)
 
+    def subtree(self, path):
+        hi = HiveIterator(self)
+        hi.descend(path)
+        return hi
+
 
 class HiveIterator():
     hive = None
     iter = None
-    root_traversed = False
+    iteration_root = None
 
     def __init__(self, hive):
@@ -147 +187 @@
 
     def __iter__(self):
+        self.iteration_root = None
         return self
 
     def __next__(self):
-        if self.root_traversed:
-            self.root_traversed = True
-            
+        if self.iteration_root == None:
+            self.iteration_root = self.current_key()            
         elif not regfi.regfi_iterator_down(self.iter):
             up_ret = regfi.regfi_iterator_up(self.iter)
-            while up_ret and not regfi.regfi_iterator_next_subkey(self.iter):
+            while (up_ret and
+                   not regfi.regfi_iterator_next_subkey(self.iter)):
+                if self.iteration_root == self.current_key():
+                    self.iteration_root = None
+                    raise StopIteration('')
                 up_ret = regfi.regfi_iterator_up(self.iter)
 
@@ -166 +210 @@
 
         regfi.regfi_iterator_first_subkey(self.iter)
-        print(regfi.regfi_iterator_cur_key(self.iter).contents.keyname)
-        return regfi.regfi_iterator_cur_key(self.iter)
-
+        return self.current_key()
+
+    def down(self):
+        pass
+
+    def up(self):
+        pass
+
+    def descend(self, path):
+        #set up generator
+        cpath = (bytes(p,'ascii') for p in path) 
+
+        # evaluate generator and create char* array
+        apath = (c_char_p*len(path))(*cpath)
+
+        if not regfi.regfi_iterator_walk_path(self.iter,apath):
+            raise Exception('Could not locate path.\n'+GetLogMessages())
+
+    def current_key(self):
+        return Key(regfi.regfi_iterator_cur_key(self.iter))