From f9276a7dd73656f15035c3a47a0981ee48f4069a Mon Sep 17 00:00:00 2001
From: "SND\\kernelnet_cp"
 <SND\kernelnet_cp@9b283d60-5439-405e-af05-b73fd8c4d996>
Date: Thu, 22 Nov 2012 09:11:55 +0000
Subject: [PATCH] [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   | 19 +++++++++++++------
 samples/km/proclist.py |  5 +----
 2 files changed, 14 insertions(+), 10 deletions(-)

diff --git a/samples/km/drvobj.py b/samples/km/drvobj.py
index 6048acf..f0e40d5 100644
--- a/samples/km/drvobj.py
+++ b/samples/km/drvobj.py
@@ -90,14 +90,21 @@ def printDrvMajorTable( drvName ):
         dprintln( "MajorFunction[%d] = %s" % ( i, findSymbol( drvObj.MajorFunction[i] ) ) )
 
 def run():
-   loadSymbols();
-
-   printDrvMajorTable( "afd" )
-
-if __name__ == "__main__":
 
     if not isWindbgExt():
-       loadDump( sys.argv[1] )
+        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__":
     run()
 
+
diff --git a/samples/km/proclist.py b/samples/km/proclist.py
index 70c1d42..9c55cc0 100644
--- a/samples/km/proclist.py
+++ b/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()