zjgcjy/pykd
1
0
Fork 0
mirror of https://github.com/ivellioscolin/pykd.git synced 2025-04-21 04:13:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

added : module.hasSymbol method (check if a module has the specified symbol)

Browse Source
This commit is contained in:
ussrhero 2019-06-01 10:33:17 +03:00
parent 5d5e634ace
commit 3ab64bdc85
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pykd/pymod.cpp
View File

@ -886,6 +886,8 @@ void pykd_init()
"Return tuple of the module's file version") "Return tuple of the module's file version")
.def("getFixedFileInfo", ModuleAdapter::getFixedFileInfo, .def("getFixedFileInfo", ModuleAdapter::getFixedFileInfo,
"Return FixedFileInfo" ) "Return FixedFileInfo" )
.def("hasSymbol", ModuleAdapter::isContainedSymbol,
"Check if module has the specified symbol")
.def("__getattr__", ModuleAdapter::getAttrByName, .def("__getattr__", ModuleAdapter::getAttrByName,
"Return symbol offset or type as attribute" ) "Return symbol offset or type as attribute" )
.def("__getitem__", ModuleAdapter::getItemByKey, .def("__getitem__", ModuleAdapter::getItemByKey,

11
test/scripts/moduletest.py
View File

@ -145,12 +145,6 @@ class ModuleTest( unittest.TestCase ):
self.assertFalse("NotExist" in target.module) self.assertFalse("NotExist" in target.module)
self.assertRaises(Exception, lambda md : 2 in md, target.module) self.assertRaises(Exception, lambda md : 2 in md, target.module)
def testHasAttr(self):
self.assertTrue(hasattr(target.module, "voidPtr"))
self.assertTrue(hasattr(target.module, "structTest"))
self.assertFalse(hasattr(target.module, "NotExist"))
self.assertRaises(Exception, lambda md : hasattr(2, target.module), target.module)
def testGetByKey(self): def testGetByKey(self):
self.assertFalse(None == target.module["voidPtr"]) self.assertFalse(None == target.module["voidPtr"])
self.assertFalse(None == target.module["structTest"] ) self.assertFalse(None == target.module["structTest"] )
@ -159,6 +153,11 @@ class ModuleTest( unittest.TestCase ):
def testEvalInModuleScope(self): def testEvalInModuleScope(self):
self.assertEqual( target.module.rva('voidPtr'), eval("voidPtr - %#x" % target.module.begin(), globals(), target.module) ) self.assertEqual( target.module.rva('voidPtr'), eval("voidPtr - %#x" % target.module.begin(), globals(), target.module) )
def testHasSymbol(self):
self.assertTrue(target.module.hasSymbol("voidPtr"))
self.assertTrue(target.module.hasSymbol("structTest"))
self.assertFalse(target.module.hasSymbol("notExists"))
def testPrint(self): def testPrint(self):
modAsStr = str(target.module) modAsStr = str(target.module)