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

fixed: #50 ( changed AttributeError to SymbolException )

Browse Source
This commit is contained in:
ussrhero 2019-05-30 00:48:27 +03:00
parent 89d5c3e5f9
commit 11e38853dc
8 changed files with 103 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pykd/pykdver.h
View File

@ -2,7 +2,7 @@
#define PYKD_VERSION_MAJOR 0 #define PYKD_VERSION_MAJOR 0
#define PYKD_VERSION_MINOR 3 #define PYKD_VERSION_MINOR 3
#define PYKD_VERSION_SUBVERSION 4 #define PYKD_VERSION_SUBVERSION 4
#define PYKD_VERSION_BUILDNO 6 #define PYKD_VERSION_BUILDNO 7
#define __VER_STR2__(x) #x #define __VER_STR2__(x) #x
#define __VER_STR1__(x) __VER_STR2__(x) #define __VER_STR1__(x) __VER_STR2__(x)

6
pykd/pymod.cpp
View File

@ -886,8 +886,10 @@ 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("__getattr__", ModuleAdapter::getSymbolVaAttr, .def("__getattr__", ModuleAdapter::getAttrByName,
"Return address of the symbol" ) "Return symbol offset or type as attribute" )
.def("__getitem__", ModuleAdapter::getItemByKey,
"Return symbol offset or type as item" )
.def("__contains__", ModuleAdapter::isContainedSymbol) .def("__contains__", ModuleAdapter::isContainedSymbol)
.def( "__str__", &ModuleAdapter::print ); .def( "__str__", &ModuleAdapter::print );

32
pykd/pymodule.h
View File

@ -82,20 +82,42 @@ struct ModuleAdapter : public kdlib::Module
return module.getSymbolVa(symbolName); return module.getSymbolVa(symbolName);
} }
static kdlib::MEMOFFSET_64 getSymbolVaAttr(kdlib::Module& module, const std::wstring &symbolName) static python::object getAttrByName(kdlib::Module& module, const std::wstring &symbolName)
{ {
AutoRestorePyState pystate; AutoRestorePyState pystate;
try { try {
return module.getSymbolVa(symbolName); return python::object(module.getSymbolVa(symbolName));
}
catch (kdlib::DbgException&)
{ }
return python::object(module.getTypeByName(symbolName));
}
static python::object getItemByKey(kdlib::Module& module, const std::wstring &symbolName)
{
AutoRestorePyState pystate;
try {
return python::object(module.getSymbolVa(symbolName));
} }
catch (kdlib::DbgException&) catch (kdlib::DbgException&)
{ {
std::wstringstream sstr;
sstr << L'\'' << module.getName() << L'\'' << L" module has no symbol " << L'\'' << symbolName << L'\'';
throw AttributeException(std::string(_bstr_t(sstr.str().c_str())).c_str());
} }
try {
return python::object(module.getTypeByName(symbolName));
}
catch (kdlib::DbgException&)
{
}
std::wstringstream sstr;
sstr << L"module hase symbol " << L'\'' << symbolName << L'\'';
throw KeyException(std::string(_bstr_t(sstr.str().c_str())).c_str());
} }
static kdlib::MEMOFFSET_32 getSymbolRva( kdlib::Module& module, const std::wstring &symbolName ) static kdlib::MEMOFFSET_32 getSymbolRva( kdlib::Module& module, const std::wstring &symbolName )
{ {
AutoRestorePyState pystate; AutoRestorePyState pystate;

49
pykd/pytypedvar.cpp
View File

@ -184,28 +184,39 @@ python::list TypedVarAdapter::getElementsDir(kdlib::TypedVar& typedVar)
kdlib::TypedVarPtr TypedVarAdapter::getFieldAttr(kdlib::TypedVar& typedVar, const std::wstring &name) kdlib::TypedVarPtr TypedVarAdapter::getFieldAttr(kdlib::TypedVar& typedVar, const std::wstring &name)
{ {
AutoRestorePyState pystate;
try
{ {
return typedVar.getElement( name );
AutoRestorePyState pystate;
try
{
return typedVar.getElement( name );
}
catch (kdlib::TypeException&)
{}
try
{
return typedVar.getMethod( name );
}
catch (kdlib::TypeException&)
{}
} }
catch (kdlib::TypeException&)
{}
std::wstringstream sstr; return typedVar.getMethod(name);
sstr << L"typed var has no field " << L'\'' << name << L'\'';
throw AttributeException(std::string(_bstr_t(sstr.str().c_str())).c_str()); //{
// AutoRestorePyState pystate;
// try
// {
// return typedVar.getElement( name );
// }
// catch (kdlib::TypeException&)
// {}
// try
// {
// return typedVar.getMethod( name );
// }
// catch (kdlib::TypeException&)
// {}
//}
//std::wstringstream sstr;
//sstr << L"typed var has no field " << L'\'' << name << L'\'';
//throw AttributeException(std::string(_bstr_t(sstr.str().c_str())).c_str());
} }

46
pykd/pytypeinfo.cpp
View File

@ -179,46 +179,36 @@ python::list TypeInfoAdapter::getBaseClasses(kdlib::TypeInfo &typeInfo)
kdlib::TypeInfoPtr TypeInfoAdapter::getElementAttr(kdlib::TypeInfo &typeInfo, const std::wstring &name) kdlib::TypeInfoPtr TypeInfoAdapter::getElementAttr(kdlib::TypeInfo &typeInfo, const std::wstring &name)
{ {
{ AutoRestorePyState pystate;
AutoRestorePyState pystate;
try { try {
return typeInfo.getElement(name); return typeInfo.getElement(name);
}
catch (kdlib::TypeException&)
{}
try {
return typeInfo.getMethod(name);
}
catch (kdlib::TypeException&)
{}
} }
catch (kdlib::TypeException&)
{}
std::wstringstream sstr; return typeInfo.getMethod(name);
sstr << L'\'' << typeInfo.getName() << L'\'' << L" type has no field " << L'\'' << name << L'\'';
throw AttributeException(std::string(_bstr_t(sstr.str().c_str())).c_str());
} }
/////////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////////
kdlib::TypeInfoPtr TypeInfoAdapter::getElementByKey(kdlib::TypeInfo &typeInfo, const std::wstring &name) kdlib::TypeInfoPtr TypeInfoAdapter::getElementByKey(kdlib::TypeInfo &typeInfo, const std::wstring &name)
{ {
{
AutoRestorePyState pystate;
try { AutoRestorePyState pystate;
return typeInfo.getElement(name);
}
catch (kdlib::TypeException&)
{}
try { try {
return typeInfo.getMethod(name); return typeInfo.getElement(name);
}
catch (kdlib::TypeException&)
{}
} }
catch (kdlib::TypeException&)
{}
try {
return typeInfo.getMethod(name);
}
catch (kdlib::TypeException&)
{}
std::wstringstream sstr; std::wstringstream sstr;
sstr << L'\'' << typeInfo.getName() << L'\'' << L" type has no field " << L'\'' << name << L'\''; sstr << L'\'' << typeInfo.getName() << L'\'' << L" type has no field " << L'\'' << name << L'\'';

14
test/scripts/moduletest.py
View File

@ -145,6 +145,20 @@ 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):
self.assertFalse(None == target.module["voidPtr"])
self.assertFalse(None == target.module["structTest"] )
self.assertRaises(KeyError, lambda md : target.module["Not Exist"], target.module)
def testEvalInModuleScope(self):
self.assertEqual( target.module.rva('voidPtr'), eval("voidPtr - %#x" % target.module.begin(), globals(), target.module) )
def testPrint(self): def testPrint(self):
modAsStr = str(target.module) modAsStr = str(target.module)

8
test/scripts/typedvar.py
View File

@ -87,7 +87,7 @@ class TypedVarTest( unittest.TestCase ):
self.assertEqual( True, tv1.m_field2 ) self.assertEqual( True, tv1.m_field2 )
self.assertEqual( 1, tv1.m_field3 ) self.assertEqual( 1, tv1.m_field3 )
self.assertEqual( 1, tv1["m_field3"] ) self.assertEqual( 1, tv1["m_field3"] )
self.assertRaises( AttributeError, lambda t: t.not_exists, tv1) # non-exsisting field self.assertRaises( pykd.SymbolException, lambda t: t.not_exists, tv1) # non-exsisting field
self.assertRaises( KeyError, lambda t: t["not_exists"], tv1) # non-exsisting field self.assertRaises( KeyError, lambda t: t["not_exists"], tv1) # non-exsisting field
def testPtrField(self): def testPtrField(self):
@ -442,7 +442,11 @@ class TypedVarTest( unittest.TestCase ):
self.assertEqual(11, getattr(var, "m_field1")) self.assertEqual(11, getattr(var, "m_field1"))
self.assertFalse(hasattr(var, "noexisit")) self.assertFalse(hasattr(var, "noexisit"))
self.assertRaises(AttributeError, getattr, *(var, "noexists")) self.assertRaises(pykd.SymbolException, lambda x: getattr(x, "noexists"), var)
def testEvalPyScope(self):
var = target.module.typedVar("structTest", [0x55] * 20 )
self.assertEqual( var.m_field1 * 17, eval("m_field1 * 17", globals(), var) )
def testEvalExpr(self): def testEvalExpr(self):
self.assertEqual( 2+2, pykd.evalExpr("2+2") ) self.assertEqual( 2+2, pykd.evalExpr("2+2") )

6
test/scripts/typeinfo.py
View File

@ -40,11 +40,11 @@ class TypeInfoTest( unittest.TestCase ):
""" get field of the complex type """ """ get field of the complex type """
ti1 = target.module.type( "structTest" ) ti1 = target.module.type( "structTest" )
self.assertTrue( "UInt4B", ti1.m_field0.name() ) self.assertTrue( "UInt4B", ti1.m_field0.name() )
self.assertTrue( "UInt4B", ti1["m_field0"].name() ) #self.assertTrue( "UInt4B", ti1["m_field0"].name() )
self.assertTrue( hasattr( ti1, "m_field0" ) ) self.assertTrue( hasattr( ti1, "m_field0" ) )
self.assertFalse( hasattr( ti1, "not_exists" ) ) self.assertFalse( hasattr( ti1, "not_exists" ) )
self.assertRaises( AttributeError, lambda t: t.not_exists, ti1) # non-exsisting field self.assertRaises( pykd.SymbolException, lambda t: t.not_exists, ti1) # non-exsisting field
self.assertRaises( KeyError, lambda t: t["not_exists"], ti1) # non-exsisting field #self.assertRaises( KeyError, lambda t: t["not_exists"], ti1) # non-exsisting field
def testBaseTypes( self ): def testBaseTypes( self ):