mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-19 19:13:22 +08:00
a26edc8208
~ refactoring: DIA findChildren ~ refactoring: failed tests -> tail git-svn-id: https://pykd.svn.codeplex.com/svn@83636 9b283d60-5439-405e-af05-b73fd8c4d996
111 lines
5.4 KiB
Python
111 lines
5.4 KiB
Python
#
|
|
#
|
|
#
|
|
|
|
import unittest
|
|
import target
|
|
import pykd
|
|
import re
|
|
|
|
class ModuleTest( unittest.TestCase ):
|
|
|
|
def testCtor( self ):
|
|
self.assertEqual( target.module.name(), pykd.module(target.module.begin() ).name() )
|
|
self.assertEqual( target.module.name(), pykd.module(target.module.name() ).name() )
|
|
|
|
def testMiscellaneous( self ):
|
|
self.assertFalse( target.module.unloaded() )
|
|
self.assertTrue( target.module.um() )
|
|
|
|
def testName( self ):
|
|
self.assertEqual( target.moduleName, target.module.name() )
|
|
|
|
def testSize( self ):
|
|
self.assertNotEqual( 0, target.module.size() )
|
|
self.assertTrue( pykd.isValid( target.module.begin() + target.module.size() - 1) )
|
|
|
|
def testBegin( self ):
|
|
self.assertNotEqual( 0, target.module.begin() )
|
|
self.assertEqual( target.module.begin(), target.module )
|
|
self.assertEqual( target.module.begin() + 100, target.module + 100 )
|
|
|
|
def testEnd( self ):
|
|
self.assertEqual( target.module.size(), target.module.end() - target.module.begin() )
|
|
self.assertTrue( pykd.isValid( target.module.end() - 1) )
|
|
|
|
def testPdb( self ):
|
|
self.assertNotEqual( "", target.module.symfile() )
|
|
|
|
def testImage( self ):
|
|
self.assertEqual( target.module.name() + ".exe", target.module.image() )
|
|
|
|
def testFindModule( self ):
|
|
self.assertRaises( pykd.BaseException, pykd.module, target.module.begin() - 0x10 )
|
|
|
|
self.assertNotEqual( None, pykd.module( target.module.begin() ) )
|
|
self.assertNotEqual( None, pykd.module( target.module.begin() + 0x10) )
|
|
|
|
self.assertRaises( pykd.BaseException, pykd.module, target.module.end() )
|
|
self.assertRaises( pykd.BaseException, pykd.module, target.module.end() + 0x10 )
|
|
|
|
def testSymbol( self ):
|
|
self.assertEqual( target.module.rva("FuncWithName0"), target.module.offset("FuncWithName0") - target.module.begin() )
|
|
self.assertEqual( target.module.rva("FuncWithName0"), target.module.FuncWithName0 - target.module.begin() )
|
|
self.assertEqual( target.module.rva("FuncWithName0"), pykd.getOffset( target.module.name() + "!FuncWithName0") - target.module.begin() )
|
|
|
|
def testFindSymbol( self ):
|
|
self.assertEqual( "FuncWithName0", target.module.findSymbol( target.module.offset("FuncWithName0") ) )
|
|
self.assertEqual( "_FuncWithName2", target.module.findSymbol( target.module.offset("_FuncWithName2") ) )
|
|
self.assertEqual( "targetapp!FuncWithName0", pykd.findSymbol( target.module.offset("FuncWithName0") ) )
|
|
self.assertEqual( "targetapp!_FuncWithName2", pykd.findSymbol( target.module.offset("_FuncWithName2") ) )
|
|
self.assertEqual( "_FuncWithName2+10", target.module.findSymbol( target.module.offset("_FuncWithName2") + 0x10 ) )
|
|
self.assertEqual( "_FuncWithName2", target.module.findSymbol( target.module.offset("_FuncWithName2") + 0x10, showDisplacement = False ) )
|
|
self.assertEqual( "targetapp!_FuncWithName2+10", pykd.findSymbol( target.module.offset("_FuncWithName2") + 0x10 ) )
|
|
self.assertEqual( "targetapp!_FuncWithName2", pykd.findSymbol( target.module.offset("_FuncWithName2") + 0x10, showDisplacement = False ) )
|
|
|
|
def testFindSymbolAndDisp( self ):
|
|
vaFuncWithName0 = target.module.offset("FuncWithName0")
|
|
self.assertEqual( ("FuncWithName0", 0), target.module.findSymbolAndDisp(vaFuncWithName0) )
|
|
self.assertEqual( ("FuncWithName0", 2), target.module.findSymbolAndDisp(vaFuncWithName0+2) )
|
|
self.assertEqual( ("targetapp!FuncWithName0", 0), pykd.findSymbolAndDisp(vaFuncWithName0) )
|
|
self.assertEqual( ("targetapp!FuncWithName0", 2), pykd.findSymbolAndDisp(vaFuncWithName0+2) )
|
|
|
|
def testType( self ):
|
|
self.assertEqual( "structTest", target.module.type("structTest").name() );
|
|
self.assertEqual( "structTest", target.module.type("g_structTest").name() );
|
|
|
|
def testSourceFile( self ):
|
|
fileName = pykd.getSourceFile(target.module.FuncWithName0 )
|
|
self.assertTrue( re.search('targetapp\\.cpp', fileName ) )
|
|
fileName, lineNo, displacement = pykd.getSourceLine( target.module.FuncWithName0 + 2)
|
|
self.assertEqual( 421, lineNo )
|
|
self.assertTrue( re.search('targetapp\\.cpp', fileName ) )
|
|
self.assertEqual( 2, displacement )
|
|
fileName, lineNo, displacement = pykd.getSourceLine()
|
|
self.assertEqual( 698, lineNo )
|
|
|
|
def testEnumSymbols( self ):
|
|
lst = target.module.enumSymbols()
|
|
self.assertNotEqual( 0, len(lst) )
|
|
lst = target.module.enumSymbols("hello*Str")
|
|
self.assertEqual( 2, len(lst) )
|
|
lst = target.module.enumSymbols( "g_const*Value")
|
|
self.assertEqual( 2, len(lst) )
|
|
lst = target.module.enumSymbols( "*FuncWithName?")
|
|
self.assertEqual( 3, len(lst) )
|
|
lst = target.module.enumSymbols( "*virtFunc*")
|
|
self.assertNotEqual( 0, len(lst) )
|
|
lst = target.module.enumSymbols( "classChild" )
|
|
self.assertEqual( 0, len(lst) )
|
|
|
|
def testGetTypes( self ):
|
|
lst1 = target.module.getUdts()
|
|
self.assertNotEqual( 0, len(lst1) )
|
|
self.assertTrue( "classChild" in lst1 )
|
|
self.assertTrue( "classBase" in lst1 )
|
|
self.assertTrue( "structTest" in lst1 )
|
|
|
|
lst2 = target.module.getEnums()
|
|
self.assertNotEqual( 0, len(lst2) )
|
|
self.assertTrue( "enumType" in lst2 )
|