2011-09-19 15:05:22 +08:00
|
|
|
#
|
|
|
|
#
|
|
|
|
#
|
|
|
|
|
|
|
|
import unittest
|
|
|
|
import target
|
|
|
|
import pykd
|
|
|
|
|
|
|
|
class ModuleTest( unittest.TestCase ):
|
2012-07-27 18:28:50 +08:00
|
|
|
|
2011-09-19 15:05:22 +08:00
|
|
|
def testCtor( self ):
|
2012-07-27 18:28:50 +08:00
|
|
|
self.assertEqual( target.module.name(), pykd.module(target.module.begin() ).name() )
|
|
|
|
self.assertEqual( target.module.name(), pykd.module(target.module.name() ).name() )
|
2011-09-19 15:05:22 +08:00
|
|
|
|
|
|
|
def testName( self ):
|
2011-09-26 16:46:32 +08:00
|
|
|
self.assertEqual( target.moduleName, target.module.name() )
|
2012-07-27 18:28:50 +08:00
|
|
|
|
2011-09-19 15:05:22 +08:00
|
|
|
def testSize( self ):
|
2011-09-26 16:46:32 +08:00
|
|
|
self.assertNotEqual( 0, target.module.size() )
|
2011-09-19 15:05:22 +08:00
|
|
|
|
|
|
|
def testBegin( self ):
|
2011-09-26 16:46:32 +08:00
|
|
|
self.assertNotEqual( 0, target.module.begin() )
|
2011-12-09 14:57:59 +08:00
|
|
|
self.assertEqual( target.module.begin(), target.module )
|
|
|
|
self.assertEqual( target.module.begin() + 100, target.module + 100 )
|
2011-09-19 15:05:22 +08:00
|
|
|
|
|
|
|
def testEnd( self ):
|
2011-09-26 16:46:32 +08:00
|
|
|
self.assertEqual( target.module.size(), target.module.end() - target.module.begin() )
|
2011-09-21 23:53:02 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# def testPdb( self ):
|
|
|
|
# self.assertNotEqual( "", target.module.pdb() )
|
2011-09-21 23:53:02 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# def testImage( self ):
|
|
|
|
# self.assertEqual( target.module.name() + ".exe", target.module.image() )
|
2011-09-26 16:46:32 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# def testFindModule( self ):
|
2011-09-21 23:53:02 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# try:
|
|
|
|
# pykd.loadModule( target.module.begin() - 0x10 )
|
|
|
|
# self.assertTrue( False )
|
|
|
|
# except pykd.BaseException:
|
|
|
|
# self.assertTrue( True )
|
2011-09-26 16:46:32 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# self.assertNotEqual( None, pykd.loadModule( target.module.begin() ) )
|
|
|
|
# self.assertNotEqual( None, pykd.loadModule( target.module.begin() + 0x10) )
|
2011-09-26 16:46:32 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# try:
|
|
|
|
# pykd.loadModule( target.module.end() )
|
|
|
|
# self.assertTrue( False )
|
|
|
|
# except pykd.BaseException:
|
|
|
|
# self.assertTrue( True )
|
2011-09-21 23:53:02 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# try:
|
|
|
|
# pykd.loadModule( target.module.end() + 0x10 )
|
|
|
|
# self.assertTrue( False )
|
|
|
|
# except pykd.BaseException:
|
|
|
|
# self.assertTrue( True )
|
2011-09-26 16:46:32 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# 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() )
|
2011-10-06 14:26:25 +08:00
|
|
|
|
2012-07-27 18:28:50 +08:00
|
|
|
# def testType( self ):
|
|
|
|
# self.assertEqual( "structTest", target.module.type("structTest").name() );
|
|
|
|
# self.assertEqual( "structTest", target.module.type("g_structTest").name() );
|
2011-12-09 14:57:59 +08:00
|
|
|
|