[0.2.x] fixed : KM samples

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

samples/km/drvobj.py

@@ -90,14 +90,21 @@ def printDrvMajorTable( drvName ):
         dprintln( "MajorFunction[%d] = %s" % ( i, findSymbol( drvObj.MajorFunction[i] ) ) )
 
 def run():
+
+    if not isWindbgExt():
+        if not loadDump( sys.argv[1] ):
+             dprintln( sys.argv[1] + " - load failed" )
+             return
+
+    if not isKernelDebugging():
+        dprintln( "not a kernel debugging" )
+        return  
+
     loadSymbols();
 
     printDrvMajorTable( "afd" )
 
 if __name__ == "__main__":
-
-    if not isWindbgExt():
-       loadDump( sys.argv[1] )
-
     run()
 
+

samples/km/proclist.py

@@ -12,9 +12,6 @@ def processInfo():
         print "".join( [chr(i) for i in process.ImageFileName if i != 0] )
 
 def run():
-   processInfo()
-
-def main():
 
     if not isWindbgExt():
         if not loadDump( sys.argv[1] ):
@@ -28,6 +25,6 @@ def main():
     processInfo()
 
 if __name__ == "__main__":
-    main()
+    run()