Changeset 227 for trunk/python/pyregfi

Timestamp:

04/09/11 12:31:22 (14 years ago)

Author:

tim

Message:

added a SetLogMask? function to pyregfi as an interface to regfi_log_set_mask
made pyregfi-smoketest.py easier to use

Location:

trunk/python/pyregfi

Files:

• __init__.py (modified) (2 diffs)
• structures.py (modified) (1 diff)

trunk/python/pyregfi/__init__.py

@@ -133 +133 @@
 
 
+## An enumeration of log message types
+#
+# @note This is a static class, there is no need to instantiate it. 
+#       Just access its attributes directly as LOG_TYPES.INFO, etc
+class LOG_TYPES(object):
+    ## Informational messages, useful in debugging
+    INFO  =  0x01
+    ## Non-critical problems in structure parsing or intepretation
+    WARN  =  0x04
+    ## Major failures
+    ERROR =  0x10
+
+
 def _buffer2bytearray(char_pointer, length):
     if length == 0 or char_pointer == None:
@@ -180 +193 @@
         return ''
     return msgs.decode('utf-8')
+
+
+## Sets the types of log messages to record
+#
+# @param log_types A sequence of message types that regfi should generate.
+#                  Message types can be found in the LOG_TYPES enumeration.
+#
+# @return True on success, False on failure.  Failures are rare, but could
+#         indicate that global logging is not operating as expected.
+# 
+# Example:
+# @code
+# SetLogMask((LOG_TYPES.ERROR, LOG_TYPES.WARN, LOG_TYPES.INFO))
+# @endcode
+#
+# The message mask is a global (all hives, iterators), thread-specific value.
+# For more information, see @ref regfi_log_set_mask.
+#
+def SetLogMask(log_types):
+    mask = 0
+    for m in log_types:
+        mask |= m
+    return regfi.regfi_log_set_mask(mask)
+
 
 

trunk/python/pyregfi/structures.py

@@ -17 +17 @@
 # XXX: can we always be sure enums are this size?
 REGFI_ENCODING = c_uint32
-REGFI_ENCODING_UTF8 = 1
+REGFI_ENCODING_UTF8 = REGFI_ENCODING(1)
 
 REGFI_DATA_TYPE = c_uint32
-REGFI_REGF_SIZE = 0x1000