[snippets] updated

git-svn-id: https://pykd.svn.codeplex.com/svn@65717 9b283d60-5439-405e-af05-b73fd8c4d996

samples/drvobj.py

@@ -44,6 +44,8 @@ def getObjectInDir( dirObj, objName ):
               else:
                   dirEntry = 0    
 
+    return None      
+
 
 
                                                                           

samples/proclist.py

@@ -13,21 +13,21 @@ def processInfo():
         print "".join( [chr(i) for i in process.ImageFileName if i != 0] )
 
 
-
+def main():
-if __name__ == "__main__":
-
-
-    while True:
 
     if not isWindbgExt():
         if not loadDump( sys.argv[1] ):
              dprintln( sys.argv[1] + " - load failed" )
-                break
+             return
 
     if not isKernelDebugging():
         dprintln( "not a kernel debugging" )
-            break 
+        return   
                  
     processInfo()
-        break      
+
+	
+if __name__ == "__main__":
+    main()
+
 

snippets/iat.py

@@ -46,7 +46,7 @@ def iat( moduleName, mask = "*" ):
         else:
             iatEntry = ptrQWord( iatAddr + i*pSize )
 
-        if  iatEntry != 0:
+        if  iatEntry != None and iatEntry != 0:
             symbolName = findSymbol( iatEntry ) 
             if fnmatch.fnmatch( symbolName, mask ): 
                 dprintln( symbolName )

snippets/ntobj.py

@@ -73,7 +73,6 @@ def getObjectNameInfoFromInfoMask(p):
 # Select platform-specific function for getting name of object
 getObjectNameInfo = None
 try:
-  getTypeClass("nt", "_OBJECT_HEADER").NameInfoOffset
   getObjectNameInfo = getObjectNameInfoFromHeader
 except AttributeError:
   getObjectNameInfo = getObjectNameInfoFromInfoMask