[0.1.x] added : expr routine

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

pykd/dbgclient.h

@@ -62,6 +62,8 @@ public:
 
     void attachKernel( const std::wstring  &param );
 
+    ULONG64 evaluate( const std::wstring  &expression );
+
     Module loadModule( const std::string  &moduleName ) {
         return Module( m_client, moduleName );
     }

pykd/dbgcmd.cpp

@@ -79,6 +79,59 @@ loadExtension( const std::wstring &extPath )
 
 /////////////////////////////////////////////////////////////////////////////////
 
+ULONG64
+DebugClient::evaluate( const std::wstring  &expression )
+{
+    HRESULT             hres;
+    ULONG64             value = 0;
+
+    DEBUG_VALUE  debugValue = {};
+    ULONG        remainderIndex = 0;
+
+    hres = m_control->IsPointer64Bit();
+    if ( FAILED( hres ) )
+        throw  DbgException( "IDebugControl::IsPointer64Bit  failed" );
+    
+    if ( hres == S_OK )
+    {
+        hres = m_control->EvaluateWide( 
+            expression.c_str(), 
+            DEBUG_VALUE_INT64,
+            &debugValue,
+            &remainderIndex );
+            
+        if ( FAILED( hres ) )
+            throw  DbgException( "IDebugControl::Evaluate  failed" );
+            
+        if ( remainderIndex == expression.length() )
+            value = debugValue.I64;
+    }
+    else
+    {
+        hres = m_control->EvaluateWide( 
+            expression.c_str(), 
+            DEBUG_VALUE_INT32,
+            &debugValue,
+            &remainderIndex );
+            
+        if (  FAILED( hres ) )
+            throw  DbgException( "IDebugControl::Evaluate  failed" );
+            
+        if ( remainderIndex == expression.length() )
+            value = debugValue.I32;
+    }      
+
+    return value;
+}
+
+ULONG64
+evaluate( const std::wstring  &expression )
+{
+    return g_dbgClient->evaluate( expression );
+}
+
+/////////////////////////////////////////////////////////////////////////////////
+
 } // end namespace pykd
 
 

pykd/dbgcmd.h

@@ -11,6 +11,9 @@ namespace pykd {
 std::string
 dbgCommand( const std::wstring  &command );
 
+ULONG64
+evaluate( const std::wstring  &expression );
+
 ///////////////////////////////////////////////////////////////////////////////////
 
 class DbgExtension : private DbgObject { 

pykd/dbgext.cpp

@@ -65,6 +65,8 @@ BOOST_PYTHON_MODULE( pykd )
             "Attach debugger to a exsisting process" )
         .def( "attachKernel", &pykd::DebugClient::attachKernel, 
             "Attach debugger to a target's kernel" )
+        .def( "expr", &pykd::DebugClient::evaluate,
+            "Evaluate windbg expression" )
         .def ( "loadExt", &pykd::DebugClient::loadExtension,
             "Load a debuger extension" )
         .def( "loadModule", &pykd::DebugClient::loadModule, 
@@ -88,6 +90,8 @@ BOOST_PYTHON_MODULE( pykd )
         "Attach debugger to a exsisting process" );
     python::def( "attachKernel", &pykd::attachKernel,
         "Attach debugger to a kernel target" );
+    python::def( "expr", &pykd::evaluate,
+        "Evaluate windbg expression" );
     python::def( "loadExt", &pykd::loadExtension,
         "Load a debuger extension" );
     python::def( "loadModule", &pykd::loadModule,

test/scripts/dbgcmd.py

@@ -11,3 +11,7 @@ class DbgcmdTest( unittest.TestCase ):
 #    def testDbgExt( self ):
 #        #ext = pykd.loadExt( "ext" )
 #        #self.assertNotEqual( "", ext.call("help", "") )
+
+    def testExpr( self ):
+        self.assertEqual( 8, pykd.expr( "poi(targetapp!g_ulonglongValue)" ) )
+    

test/scripts/target.py

@@ -2,6 +2,6 @@
 #
 #
 
-
 module = None
 moduleName = None
+dbgext = None