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

[+, 0.1.x] get symbol by address from module

Browse Source 
git-svn-id: https://pykd.svn.codeplex.com/svn@71164 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\EreTIk_cp 2011-11-08 10:53:07 +00:00 committed by Mikhail I. Izmestev
parent 2d239171a3
commit 82a744d1be
6 changed files with 54 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pykd/dbgext.cpp
View File

@ -334,8 +334,8 @@ BOOST_PYTHON_MODULE( pykd )
"Return rva of the symbol" )
.def("type", &pykd::Module::getTypeByName,
"Return typeInfo class by type name" )
//.def("typedVar", &pykd::Module::getTypedVarByAddr,
// "Return a typedVar class instance" )
.def("typedVar", &pykd::Module::getTypedVarByAddr,
"Return a typedVar class instance" )
.def("typedVar",&pykd::Module::getTypedVarByName,
"Return a typedVar class instance" )
.def("typedVar",&pykd::Module::getTypedVarByType,

47
pykd/module.cpp
View File

@ -147,6 +147,7 @@ Module::reloadSymbols()
if ( FAILED( hres ) )
throw DbgException("IDebugSymbols::Reload failed" );
m_dia.reset();
m_dia = pyDia::GlobalScope::loadPdb( getPdbName() );
}
@ -155,7 +156,7 @@ Module::reloadSymbols()
TypeInfo
Module::getTypeByName( const std::string &typeName )
{
return TypeInfo( m_dia, typeName );
return TypeInfo( getDia(), typeName );
}
///////////////////////////////////////////////////////////////////////////////////
@ -163,7 +164,7 @@ Module::getTypeByName( const std::string &typeName )
TypedVar
Module::getTypedVarByTypeName( const std::string &typeName, ULONG64 addr )
{
return TypedVar( TypeInfo( m_dia, typeName ), addr );
return TypedVar( TypeInfo( getDia(), typeName ), addr );
}
///////////////////////////////////////////////////////////////////////////////////
@ -179,37 +180,29 @@ Module::getTypedVarByType( const TypeInfo &typeInfo, ULONG64 addr )
TypedVar
Module::getTypedVarByName( const std::string &symName )
{
pyDia::SymbolPtr typeSym = m_dia->getChildByName( symName );
pyDia::SymbolPtr typeSym = getDia()->getChildByName( symName );
return TypedVar( TypeInfo( typeSym->getType() ), typeSym->getRva() + m_base );
}
///////////////////////////////////////////////////////////////////////////////////
//TypedVar
//Module::getTypedVarByAddr( ULONG64 addr )
//{
// addr = addr64(addr);
//
// if ( addr < m_base || addr > getEnd() )
// throw DbgException("address is out of the module space" );
//
// ULONG rva = (ULONG)(addr - m_base);
//
// for( ULONG i = 0; i < m_dia->getChildCount(); i++ )
// {
// pyDia::SymbolPtr typeSym = m_dia->getChildByIndex(i);
//
// std::string name = m_dia->getName();
//
// if ( typeSym->getSymTag() == SymTagData && typeSym->getRva() == rva )
// {
// return TypedVar( TypeInfo( typeSym->getType() ), typeSym->getRva() + m_base );
// }
// }
//
// throw DbgException("failed to find type info for this offset" );
//}
TypedVar
Module::getTypedVarByAddr( ULONG64 addr )
{
addr = addr64(addr);
if ( addr < m_base || addr > getEnd() )
throw DbgException( "address is out of the module space" );
LONG displacement;
pyDia::SymbolPtr diaSym =
getDia()->findByRvaImpl((ULONG)(addr - m_base), SymTagData, displacement);
if (displacement)
throw DbgException( "not exactly match by RVA" );
return TypedVar( TypeInfo( diaSym->getType() ), addr );
}
///////////////////////////////////////////////////////////////////////////////////

19
pykd/module.h
View File

@ -47,22 +47,14 @@ public:
ULONG64
getSymbol( const std::string &symbolname ) {
if ( !m_dia )
m_dia = pyDia::GlobalScope::loadPdb( getPdbName() );
pyDia::SymbolPtr sym = m_dia->getChildByName( symbolname );
pyDia::SymbolPtr sym = getDia()->getChildByName( symbolname );
return m_base + sym->getRva();
}
ULONG
getSymbolRva( const std::string &symbolname ) {
if ( !m_dia )
m_dia = pyDia::GlobalScope::loadPdb( getPdbName() );
pyDia::SymbolPtr sym = m_dia->getChildByName( symbolname );
pyDia::SymbolPtr sym = getDia()->getChildByName( symbolname );
return sym->getRva();
}
@ -79,6 +71,13 @@ public:
private:
pyDia::GlobalScopePtr getDia() {
if (!m_dia)
m_dia = pyDia::GlobalScope::loadPdb( getPdbName() );
return m_dia;
}
std::string m_name;
std::string m_imageName;
ULONG64 m_base;

5
test/scripts/typedvar.py
View File

@ -22,6 +22,11 @@ class TypedVarTest( unittest.TestCase ):
#tv2 = target.module.typedVar( "structTest[2]", target.module.g_testArray )
#self.assertEqual( tv1.sizeof()*2, tv2.sizeof() )
def testByAddress( self ):
tv1 = target.module.typedVar( "structTest", target.module.g_structTest )
tv2 = target.module.typedVar( tv1.getAddress() )
self.assertEqual( tv2.getAddress(), tv1.getAddress() )
def testStruct(self):
tv1 = target.module.typedVar( "structTest", target.module.g_structTest )
self.assertEqual( 0, tv1.m_field0 )

3
test/targetapp/targetapp.cpp
View File

@ -24,7 +24,6 @@ struct structWithBits {
ULONG m_bit5 : 1;
ULONG m_bit6_7 : 2;
};
structWithBits g_structWithBits = {0};
union unionTest {
ULONG m_value;
@ -47,6 +46,8 @@ struct structTest {
structTest* m_field4;
};
structWithBits g_structWithBits = {0};
structTest g_structTest = { 0, 500, true, 1, NULL };
structTest g_structTest1 = { 0, 500, true, 1, &g_structTest };