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

[0.3.x] added : typedVar.rawBytes method ( return list of bytes )

Browse Source 
git-svn-id: https://pykd.svn.codeplex.com/svn@91208 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\ussrhero_cp 2017-04-09 08:34:35 +00:00 committed by Mikhail I. Izmestev
parent ee8236c524
commit 01aa2b6c6d
5 changed files with 92 additions and 12 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_MINOR 3
#define PYKD_VERSION_SUBVERSION 2
#define PYKD_VERSION_BUILDNO 2
#define PYKD_VERSION_BUILDNO 3
#define __VER_STR2__(x) #x
#define __VER_STR1__(x) __VER_STR2__(x)

15
pykd/pymod.cpp
View File

@ -21,6 +21,7 @@
#include "pytypeinfo.h"
#include "pycpucontext.h"
#include "pyprocess.h"
#include "pydataaccess.h"
using namespace pykd;
@ -920,6 +921,7 @@ BOOST_PYTHON_MODULE( pykd )
#endif
;
python::class_<kdlib::TypedVar, kdlib::TypedVarPtr, python::bases<kdlib::NumBehavior>, boost::noncopyable >("typedVar",
"Class of non-primitive type object, child class of typeClass. Data from target is copied into object instance", python::no_init )
.def("__init__", python::make_constructor(pykd::getTypedVarByName) )
@ -952,6 +954,8 @@ BOOST_PYTHON_MODULE( pykd )
"Return method of class as an object attribute" )
.def("deref",TypedVarAdapter::deref,
"Return value by pointer" )
.def("rawBytes", TypedVarAdapter::getRawBytes,
"Return list of bytes" )
.def("type", TypedVarAdapter::getType,
"Return typeInfo instance" )
.def("castTo", TypedVarAdapter::castByName,
@ -1311,6 +1315,17 @@ BOOST_PYTHON_MODULE( pykd )
.def("__str__", pykd::printSyntheticSymbol,
"Return object as a string");
//python::class_<pykd::DataAccessor, boost::noncopyable>("dataAccessor",
// "class for abstract representation of data storage",python::no_init)
// .def("readByte", &pykd::DataAccessor::readByte)
// ;
//python::class_<pykd::ListDataAccessor, kdlib::DataAccessorPtr>("objectAccessor",
// "class for data access to python objects", python::no_init)
// .def( python::init<python::object&>() )
// //.def("readByte", &pykd::ListDataAccessor::readByte, "")
// ;
// C++ exception translation to python
pykd::registerExceptions();
}

63
pykd/pytypedvar.cpp
View File

@ -7,6 +7,52 @@ namespace pykd {
///////////////////////////////////////////////////////////////////////////////
kdlib::TypedVarPtr getTypedVarByTypeName(const std::wstring &name, python::object& dataStorage)
{
python::extract<kdlib::MEMOFFSET_64> get_addr(dataStorage);
if ( get_addr.check() )
{
kdlib::MEMOFFSET_64 offset = get_addr();
AutoRestorePyState pystate;
return kdlib::loadTypedVar( name, offset );
}
std::vector<char> byteArray;
for (int i = 0; i < python::len(dataStorage); ++i)
{
byteArray.push_back( python::extract<unsigned char>(dataStorage[i]) );
}
AutoRestorePyState pystate;
return kdlib::loadTypedVar( name, kdlib::getCacheAccessor(byteArray) );
}
///////////////////////////////////////////////////////////////////////////////
kdlib::TypedVarPtr getTypedVarByTypeInfo( const kdlib::TypeInfoPtr &typeInfo, python::object& dataStorage)
{
python::extract<kdlib::MEMOFFSET_64> get_addr(dataStorage);
if ( get_addr.check() )
{
kdlib::MEMOFFSET_64 offset = get_addr();
AutoRestorePyState pystate;
return kdlib::loadTypedVar(typeInfo, offset );
}
std::vector<char> byteArray;
for (int i = 0; i < python::len(dataStorage); ++i)
{
byteArray.push_back( python::extract<unsigned char>(dataStorage[i]) );
}
AutoRestorePyState pystate;
return kdlib::loadTypedVar(typeInfo, kdlib::getCacheAccessor(byteArray) );
}
///////////////////////////////////////////////////////////////////////////////
python::list getTypedVarListByTypeName( kdlib::MEMOFFSET_64 offset, const std::wstring &typeName, const std::wstring &fieldName )
{
kdlib::TypedVarList lst;
@ -154,6 +200,23 @@ kdlib::TypedVarPtr TypedVarAdapter::getFieldAttr(kdlib::TypedVar& typedVar, cons
///////////////////////////////////////////////////////////////////////////////
python::list TypedVarAdapter::getRawBytes(kdlib::TypedVar& typedVar)
{
std::vector<unsigned char> rawBytes;
{
AutoRestorePyState pystate;
kdlib::DataAccessorPtr dataStream = kdlib::getCacheAccessor(typedVar.getSize());
typedVar.writeBytes(dataStream);
dataStream->readBytes(rawBytes, typedVar.getSize());
}
return vectorToList( rawBytes );
}
///////////////////////////////////////////////////////////////////////////////
} // namesapce pykd

16
pykd/pytypedvar.h
View File

@ -14,11 +14,7 @@ namespace python = boost::python;
namespace pykd {
inline kdlib::TypedVarPtr getTypedVarByTypeName( const std::wstring &name, kdlib::MEMOFFSET_64 addr )
{
AutoRestorePyState pystate;
return kdlib::loadTypedVar( name, addr );
}
kdlib::TypedVarPtr getTypedVarByTypeName(const std::wstring &name, python::object& dataStorage);
inline kdlib::TypedVarPtr getTypedVarByName( const std::wstring &name )
{
@ -26,11 +22,7 @@ inline kdlib::TypedVarPtr getTypedVarByName( const std::wstring &name )
return kdlib::loadTypedVar( name );
}
inline kdlib::TypedVarPtr getTypedVarByTypeInfo( const kdlib::TypeInfoPtr &typeInfo, kdlib::MEMOFFSET_64 addr )
{
AutoRestorePyState pystate;
return kdlib::loadTypedVar( typeInfo, addr );
}
kdlib::TypedVarPtr getTypedVarByTypeInfo( const kdlib::TypeInfoPtr &typeInfo, python::object& dataStorage);
inline kdlib::TypedVarPtr getTypedVarWithPrototype( const std::wstring &name, const std::wstring &prototype)
{
@ -38,6 +30,8 @@ inline kdlib::TypedVarPtr getTypedVarWithPrototype( const std::wstring &name, co
return kdlib::loadTypedVar(name, prototype);
}
kdlib::TypedVarPtr getTypedVarByTypeNameWithBuffer(const std::wstring& name, python::object &bytes);
python::list getTypedVarListByTypeName( kdlib::MEMOFFSET_64 offset, const std::wstring &typeName, const std::wstring &fieldName );
python::list getTypedVarListByType( kdlib::MEMOFFSET_64 offset, kdlib::TypeInfoPtr &typeInfo, const std::wstring &fieldName );
python::list getTypedVarArrayByTypeName( kdlib::MEMOFFSET_64 offset, const std::wstring &typeName, size_t number );
@ -185,6 +179,8 @@ struct TypedVarAdapter {
AutoRestorePyState pystate;
return typedVar.castTo(typeInfo);
}
static python::list getRawBytes(kdlib::TypedVar& typedVar);
};
} // end namespace pykd

8
test/scripts/typedvar.py
View File

@ -419,4 +419,10 @@ class TypedVarTest( unittest.TestCase ):
addr = pykd.getOffset("g_structTest")
self.assertTrue( None != target.module.type( "structTest" ).getTypedVar(addr) )
def testByteSequence(self):
self.assertEqual( 0x44332211, pykd.typedVar("UInt4B", [0x11, 0x22, 0x33, 0x44]) )
self.assertEqual( -1, pykd.typedVar( pykd.baseTypes.Int4B, [0xFF, 0xFF, 0xFF, 0xFF] ) )
def testRawBytes(self):
self.assertEqual( [ 0x55, 0x55, 0, 0], target.module.typedVar( "ulongConst" ).rawBytes() )