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

[0.3.x] updated : tests

Browse Source 
git-svn-id: https://pykd.svn.codeplex.com/svn@83966 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\kernelnet_cp 2013-06-06 08:22:53 +00:00 committed by Mikhail I. Izmestev
parent f63d513802
commit 446a006a40
5 changed files with 92 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pykd/memaccess.h
View File

@ -64,6 +64,11 @@ inline python::list loadDoubles( kdlib::MEMOFFSET_64 offset, unsigned long count
return vectorToList( kdlib::loadDoubles( offset, count, phyAddr ) ); return vectorToList( kdlib::loadDoubles( offset, count, phyAddr ) );
} }
inline kdlib::MEMOFFSET_64 ptrPtr( kdlib::MEMOFFSET_64 offset )
{
return kdlib::ptrPtr( offset );
}
inline python::list loadPtrList( kdlib::MEMOFFSET_64 offset ) inline python::list loadPtrList( kdlib::MEMOFFSET_64 offset )
{ {
return vectorToList( kdlib::loadPtrList(offset) ); return vectorToList( kdlib::loadPtrList(offset) );

16
pykd/module.h
View File

@ -4,6 +4,8 @@
#include "kdlib/module.h" #include "kdlib/module.h"
#include "stladaptor.h"
namespace pykd { namespace pykd {
@ -58,6 +60,20 @@ struct ModuleAdapter : public kdlib::Module
std::wstring symbolName = module.findSymbol( offset, displacement ); std::wstring symbolName = module.findSymbol( offset, displacement );
return python::make_tuple( symbolName, displacement ); return python::make_tuple( symbolName, displacement );
} }
static python::list getTypedVarListByTypeName( kdlib::Module& module, kdlib::MEMOFFSET_64 offset, const std::wstring &typeName, const std::wstring &fieldName )
{
kdlib::TypedVarList lst = module.loadTypedVarList( offset, typeName, fieldName );
return vectorToList( lst );
}
static python::list getTypedVarArrayByTypeName( kdlib::Module& module, kdlib::MEMOFFSET_64 offset, const std::wstring &typeName, size_t number )
{
kdlib::TypedVarList lst = module.loadTypedVarArray( offset, typeName, number );
return vectorToList( lst );
}
}; };
} // end namespace pykd } // end namespace pykd

40
pykd/pymod.cpp
View File

@ -70,7 +70,8 @@ BOOST_PYTHON_FUNCTION_OVERLOADS( loadSignQWords_, loadSignQWords, 2, 3 );
BOOST_PYTHON_FUNCTION_OVERLOADS( loadFloats_, loadFloats, 2, 3 ); BOOST_PYTHON_FUNCTION_OVERLOADS( loadFloats_, loadFloats, 2, 3 );
BOOST_PYTHON_FUNCTION_OVERLOADS( loadDoubles_, loadDoubles, 2, 3 ); BOOST_PYTHON_FUNCTION_OVERLOADS( loadDoubles_, loadDoubles, 2, 3 );
BOOST_PYTHON_FUNCTION_OVERLOADS( compareMemory_, kdlib::compareMemory, 3, 4 ); BOOST_PYTHON_FUNCTION_OVERLOADS( compareMemory_, kdlib::compareMemory, 3, 4 );
//
BOOST_PYTHON_FUNCTION_OVERLOADS( getSourceLine_, getSourceLine, 0, 1 ); BOOST_PYTHON_FUNCTION_OVERLOADS( getSourceLine_, getSourceLine, 0, 1 );
BOOST_PYTHON_FUNCTION_OVERLOADS( getSourceFile_, kdlib::getSourceFile, 0, 1 ); BOOST_PYTHON_FUNCTION_OVERLOADS( getSourceFile_, kdlib::getSourceFile, 0, 1 );
// //
@ -194,8 +195,6 @@ BOOST_PYTHON_MODULE( pykd )
"Read an signed 8-byte integer from the target memory" ); "Read an signed 8-byte integer from the target memory" );
python::def( "ptrSignMWord", &kdlib::ptrSignMWord, python::def( "ptrSignMWord", &kdlib::ptrSignMWord,
"Read an signed mashine's word wide integer from the target memory" ); "Read an signed mashine's word wide integer from the target memory" );
python::def( "ptrPtr", &kdlib::ptrPtr,
"Read an pointer value from the target memory" );
python::def( "ptrFloat", &kdlib::ptrSingleFloat, python::def( "ptrFloat", &kdlib::ptrSingleFloat,
"Read a float with single precision from the target memory" ); "Read a float with single precision from the target memory" );
python::def( "ptrDouble", &kdlib::ptrDoubleFloat, python::def( "ptrDouble", &kdlib::ptrDoubleFloat,
@ -229,15 +228,18 @@ BOOST_PYTHON_MODULE( pykd )
// "Return string represention of windows UNICODE_STRING type" ); // "Return string represention of windows UNICODE_STRING type" );
//python::def( "loadAnsiString", &loadAnsiStr, //python::def( "loadAnsiString", &loadAnsiStr,
// "Return string represention of windows ANSI_STRING type" ); // "Return string represention of windows ANSI_STRING type" );
python::def( "loadPtrList", &loadPtrList,
"Return list of pointers, each points to next" );
python::def( "loadPtrs", &loadPtrArray,
"Read the block of the target's memory and return it as a list of pointers" );
python::def( "loadFloats", &loadFloats, loadFloats_( python::args( "offset", "count", "phyAddr" ), python::def( "loadFloats", &loadFloats, loadFloats_( python::args( "offset", "count", "phyAddr" ),
"Read the block of the target's memory and return it as list of floats" ) ); "Read the block of the target's memory and return it as list of floats" ) );
python::def( "loadDoubles", &loadDoubles, loadDoubles_( python::args( "offset", "count", "phyAddr" ), python::def( "loadDoubles", &loadDoubles, loadDoubles_( python::args( "offset", "count", "phyAddr" ),
"Read the block of the target's memory and return it as list of doubles" ) ); "Read the block of the target's memory and return it as list of doubles" ) );
python::def( "ptrPtr", &ptrPtr,
"Read an pointer value from the target memory" );
python::def( "loadPtrList", &loadPtrList,
"Return list of pointers, each points to next" );
python::def( "loadPtrs", &loadPtrArray,
"Read the block of the target's memory and return it as a list of pointers" );
// types and vaiables // types and vaiables
python::def( "getSourceFile", &kdlib::getSourceFile, getSourceFile_( python::args( "offset"), python::def( "getSourceFile", &kdlib::getSourceFile, getSourceFile_( python::args( "offset"),
"Return source file by the specified offset" ) ); "Return source file by the specified offset" ) );
@ -251,14 +253,14 @@ BOOST_PYTHON_MODULE( pykd )
// "Return tuple(symbol_name, displacement) by virtual address" ); // "Return tuple(symbol_name, displacement) by virtual address" );
python::def( "sizeof", &kdlib::getSymbolSize, python::def( "sizeof", &kdlib::getSymbolSize,
"Return a size of the type or variable" ); "Return a size of the type or variable" );
// python::def("typedVarList", &getTypedVarListByTypeName, python::def("typedVarList", &TypedVarAdapter::getTypedVarListByTypeName,
// "Return a list of the typedVar class instances. Each item represents an item of the linked list in the target memory" ); "Return a list of the typedVar class instances. Each item represents an item of the linked list in the target memory" );
// python::def("typedVarList", &getTypedVarListByType, python::def("typedVarList", &TypedVarAdapter::getTypedVarListByType,
// "Return a list of the typedVar class instances. Each item represents an item of the linked list in the target memory" ); "Return a list of the typedVar class instances. Each item represents an item of the linked list in the target memory" );
// python::def("typedVarArray", &getTypedVarArrayByTypeName, python::def("typedVarArray", &TypedVarAdapter::getTypedVarArrayByTypeName,
// "Return a list of the typedVar class instances. Each item represents an item of the counted array in the target memory" ); "Return a list of the typedVar class instances. Each item represents an item of the counted array in the target memory" );
// python::def("typedVarArray", &getTypedVarArrayByType, python::def("typedVarArray", &TypedVarAdapter::getTypedVarArrayByType,
// "Return a list of the typedVar class instances. Each item represents an item of the counted array in the target memory" ); "Return a list of the typedVar class instances. Each item represents an item of the counted array in the target memory" );
python::def("containingRecord", &TypedVarAdapter::containingRecordByName, python::def("containingRecord", &TypedVarAdapter::containingRecordByName,
"Return instance of the typedVar class. It's value are loaded from the target memory." "Return instance of the typedVar class. It's value are loaded from the target memory."
"The start address is calculated by the same method as the standard macro CONTAINING_RECORD does" ); "The start address is calculated by the same method as the standard macro CONTAINING_RECORD does" );
@ -408,10 +410,10 @@ BOOST_PYTHON_MODULE( pykd )
"Return a typedVar class instance" ) "Return a typedVar class instance" )
.def("typedVar",&kdlib::Module::getTypedVarByTypeName, .def("typedVar",&kdlib::Module::getTypedVarByTypeName,
"Return a typedVar class instance" ) "Return a typedVar class instance" )
//.def("typedVarList", &Module::getTypedVarListByTypeName, .def("typedVarList", &ModuleAdapter::getTypedVarListByTypeName,
// "Return a list of the typedVar class instances. Each item represents an item of the linked list in the target memory" ) "Return a list of the typedVar class instances. Each item represents an item of the linked list in the target memory" )
//.def("typedVarArray", &Module::getTypedVarArrayByTypeName, .def("typedVarArray", &ModuleAdapter::getTypedVarArrayByTypeName,
// "Return a list of the typedVar class instances. Each item represents an item of the counted array in the target memory" ) "Return a list of the typedVar class instances. Each item represents an item of the counted array in the target memory" )
.def("containingRecord", &kdlib::Module::containingRecord, .def("containingRecord", &kdlib::Module::containingRecord,
"Return instance of the typedVar class. It's value are loaded from the target memory." "Return instance of the typedVar class. It's value are loaded from the target memory."
"The start address is calculated by the same method as the standard macro CONTAINING_RECORD does" ) "The start address is calculated by the same method as the standard macro CONTAINING_RECORD does" )

28
pykd/typedvar.h
View File

@ -6,6 +6,8 @@ namespace python = boost::python;
#include "kdlib/typedvar.h" #include "kdlib/typedvar.h"
#include "stladaptor.h"
namespace pykd { namespace pykd {
struct TypedVarAdapter { struct TypedVarAdapter {
@ -61,6 +63,32 @@ struct TypedVarAdapter {
return lst; return lst;
} }
static python::list getTypedVarListByTypeName( kdlib::MEMOFFSET_64 offset, const std::wstring &typeName, const std::wstring &fieldName )
{
kdlib::TypedVarList lst = kdlib::loadTypedVarList( offset, typeName, fieldName );
return vectorToList( lst );
}
static python::list getTypedVarListByType( kdlib::MEMOFFSET_64 offset, kdlib::TypeInfoPtr &typeInfo, const std::wstring &fieldName )
{
kdlib::TypedVarList lst = kdlib::loadTypedVarList( offset, typeInfo, fieldName );
return vectorToList( lst );
}
static python::list getTypedVarArrayByTypeName( kdlib::MEMOFFSET_64 offset, const std::wstring &typeName, size_t number )
{
kdlib::TypedVarList lst = kdlib::loadTypedVarArray( offset, typeName, number );
return vectorToList( lst );
}
static python::list getTypedVarArrayByType( kdlib::MEMOFFSET_64 offset, kdlib::TypeInfoPtr &typeInfo, size_t number )
{
kdlib::TypedVarList lst = kdlib::loadTypedVarArray( offset, typeInfo, number );
return vectorToList( lst );
}
}; };
} // end namespace pykd } // end namespace pykd

50
test/scripts/typedvar.py
View File

@ -146,34 +146,34 @@ class TypedVarTest( unittest.TestCase ):
self.assertEqual( -3, tv.m_bit6_8 ) self.assertEqual( -3, tv.m_bit6_8 )
def testTypedVarList(self): def testTypedVarList(self):
tvl = target.module.typedVarList( target.module.g_listHead, "listStruct", "listEntry" ) tvl = target.module.typedVarList( target.module.g_listHead, "listStruct", "next.flink" )
self.assertEqual( 3, len( tvl ) ) self.assertEqual( 5, len( tvl ) )
self.assertEqual( [1,2,3], [ tv.num for tv in tvl ] ) self.assertEqual( range(5), [ tv.num for tv in tvl ] )
tvl = pykd.typedVarList( target.module.g_listHead, target.module.type("listStruct"), "listEntry" ) tvl = pykd.typedVarList( target.module.g_listHead, target.module.type("listStruct"), "next.flink" )
self.assertEqual( 3, len( tvl ) ) self.assertEqual( 5, len( tvl ) )
self.assertEqual( [1,2,3], [ tv.num for tv in tvl ] ) self.assertEqual( range(5), [ tv.num for tv in tvl ] )
tvl = pykd.typedVarList( target.module.g_listHead, target.module.type("listStruct"), "listEntry.Flink" ) tvl = pykd.typedVarList( target.module.g_listHead, target.module.type("listStruct"), "next.flink" )
self.assertEqual( 3, len( tvl ) ) self.assertEqual( 5, len( tvl ) )
self.assertEqual( [1,2,3], [ tv.num for tv in tvl ] ) self.assertEqual( range(5), [ tv.num for tv in tvl ] )
tvl = target.module.typedVarList( target.module.g_listHead1, "listStruct1", "next" ) #tvl = target.module.typedVarList( target.module.g_listHead1, "listStruct1", "next" )
self.assertEqual( 3, len( tvl ) ) #self.assertEqual( 3, len( tvl ) )
self.assertEqual( [100,200,300], [ tv.num for tv in tvl ] ) #self.assertEqual( [100,200,300], [ tv.num for tv in tvl ] )
tvl = pykd.typedVarList( target.module.g_listHead1, target.module.type("listStruct1"), "next" ) #tvl = pykd.typedVarList( target.module.g_listHead1, target.module.type("listStruct1"), "next" )
self.assertEqual( 3, len( tvl ) ) #self.assertEqual( 3, len( tvl ) )
self.assertEqual( [100,200,300], [ tv.num for tv in tvl ] ) #self.assertEqual( [100,200,300], [ tv.num for tv in tvl ] )
tvl = pykd.typedVarList( target.module.g_childListHead, target.module.type("ChildEntryTest"), "m_next" ) #tvl = pykd.typedVarList( target.module.g_childListHead, target.module.type("ChildEntryTest"), "m_next" )
self.assertEqual( 3, len( tvl ) ) #self.assertEqual( 3, len( tvl ) )
self.assertEqual( [1000,2000,3000], [ tv.m_someBaseFiled2 for tv in tvl ] ) #self.assertEqual( [1000,2000,3000], [ tv.m_someBaseFiled2 for tv in tvl ] )
self.assertEqual( [1001,2001,3001], [ tv.m_childFiled1 for tv in tvl ] ) #self.assertEqual( [1001,2001,3001], [ tv.m_childFiled1 for tv in tvl ] )
tvl1 = target.module.typedVarList( target.module.g_listHead, "listStruct", "listEntry" ) #tvl1 = target.module.typedVarList( target.module.g_listHead, "listStruct", "listEntry" )
tvl2 = pykd.typedVarList( target.module.g_listHead, target.moduleName + "!listStruct", "listEntry" ) #tvl2 = pykd.typedVarList( target.module.g_listHead, target.moduleName + "!listStruct", "listEntry" )
self.assertEqual( tvl1, tvl2 ) #self.assertEqual( tvl1, tvl2 )
def testTypedVarArray(self): def testTypedVarArray(self):
tvl = target.module.typedVarArray( target.module.g_testArray, "structTest", 2 ) tvl = target.module.typedVarArray( target.module.g_testArray, "structTest", 2 )
@ -317,9 +317,3 @@ class TypedVarTest( unittest.TestCase ):
for i in range( 0, 100000 ): for i in range( 0, 100000 ):
lst.append(entry) lst.append(entry)
entry = entry.deref().Flink entry = entry.deref().Flink
def testWrongArgs(self):
self.assertRaises( pykd.TypeException, pykd.typedVar, None, 0 )
self.assertRaises( pykd.TypeException, pykd.typedVarList, target.module.g_listHead1, None, "next" )
self.assertRaises( pykd.TypeException, pykd.typedVarArray, target.module.g_testArray, None, 2 )
self.assertRaises( pykd.TypeException, pykd.containingRecord, target.module.offset( "g_structTest" ), None, "m_field2" )