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

[pykd] fixed: issue 8657( !pycmd does not allow to define functions )

Browse Source 
git-svn-id: https://pykd.svn.codeplex.com/svn@67703 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\kernelnet_cp 2011-07-06 15:45:11 +00:00
parent 5d1bd33f62
commit 8e00dc3612
1 changed files with 101 additions and 93 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

168
pykd/dbgext.cpp
View File

@ -788,112 +788,120 @@ pycmd( PDEBUG_CLIENT4 client, PCSTR args )
try {
if ( !std::string( args ).empty() )
{
try {
// ïåðåíàïðàâëåíèå ñòàíäàðòíûõ ïîòîêîâ ÂÂ
boost::python::object sys = boost::python::import("sys");
boost::python::object retObj = boost::python::eval( args, WindbgGlobalSession::global(), WindbgGlobalSession::global() );
dbgOut dout;
sys.attr("stdout") = boost::python::object( dout );
if ( retObj.ptr() != NULL )
{
PyObject *s = PyObject_Str( retObj.ptr() );
dbgExt->control->Output( DEBUG_OUTPUT_NORMAL, "%s\n", PyString_AS_STRING( s ) );
Py_DECREF( s );
}
}
catch( boost::python::error_already_set const & )
dbgIn din;
sys.attr("stdin") = boost::python::object( din );
boost::python::object syntaxError = boost::python::import("exceptions").attr("SyntaxError");
std::string commandBuffer;
bool commandCompleted = true;
do {
if ( commandCompleted )
{
// îøèáêà â ñêðèïòå
PyObject *errtype = NULL, *errvalue = NULL, *traceback = NULL;
PyErr_Fetch( &errtype, &errvalue, &traceback );
if(errvalue != NULL)
{
PyObject *s = PyObject_Str(errvalue);
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "%s\n", PyString_AS_STRING( s ) );
Py_DECREF(s);
}
Py_XDECREF(errvalue);
Py_XDECREF(errtype);
Py_XDECREF(traceback);
}
}
else
{
char str[100];
ULONG inputSize;
bool stopInput = false;
boost::python::import( "pykd" );
boost::python::object main = boost::python::import("__main__");
boost::python::object global(main.attr("__dict__"));
// ïåðåíàïðàâëåíèå ñòàíäàðòíûõ ïîòîêîâ ÂÂ
boost::python::object sys = boost::python::import("sys");
dbgOut dout;
sys.attr("stdout") = boost::python::object( dout );
dbgIn din;
sys.attr("stdin") = boost::python::object( din );
do {
std::string output;
dbgExt->control->Output( DEBUG_OUTPUT_NORMAL, ">>>" );
}
else
{
dbgExt->control->Output( DEBUG_OUTPUT_NORMAL, "..." );
}
do {
{
char str[100];
ULONG inputSize;
OutputReader outputReader( (IDebugClient*)client );
OutputReader outputReader( (IDebugClient*)client );
HRESULT hres = dbgExt->control->Input( str, sizeof(str), &inputSize );
HRESULT hres = dbgExt->control->Input( str, sizeof(str), &inputSize );
if ( FAILED( hres ) )
throw;
if ( FAILED( hres ) || std::string( str ) == "" )
if ( commandCompleted )
{
if ( std::string( str ) == "" )
break;
commandBuffer = str;
}
else
{
if ( std::string( str ) == "" )
commandCompleted = true;
else
{
stopInput = true;
break;
commandBuffer.append("\n");
commandBuffer.append( str );
}
}
}
} while( FALSE );
if ( commandCompleted )
{
try {
if ( !stopInput )
try {
boost::python::exec( commandBuffer.c_str(), WindbgGlobalSession::global(), WindbgGlobalSession::global() );
boost::python::exec( str, WindbgGlobalSession::global(), WindbgGlobalSession::global() );
commandBuffer.clear();
}
catch( boost::python::error_already_set const & )
}
catch( boost::python::error_already_set const & )
{
PyObject *errtype = NULL, *errvalue = NULL, *traceback = NULL;
PyErr_Fetch( &errtype, &errvalue, &traceback );
if ( errtype && errvalue )
{
// îøèáêà â ñêðèïòå
PyObject *errtype = NULL, *errvalue = NULL, *traceback = NULL;
boost::python::tuple errValueObj( boost::python::handle<>( boost::python::borrowed(errvalue) ) );
PyErr_Fetch( &errtype, &errvalue, &traceback );
PyObject *errvalueStr = PyObject_Str(errvalue);
if(errvalue != NULL)
std::string errValueStr = boost::python::extract<std::string>( errvalueStr );
Py_XDECREF(errvalueStr);
if ( PyErr_GivenExceptionMatches( syntaxError.ptr(), errtype ) )
{
PyObject *s = PyObject_Str(errvalue);
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "%s/n", PyString_AS_STRING( s ) );
Py_DECREF(s);
if ( errValueObj[0] == "unexpected EOF while parsing" )
{
commandCompleted = false;
}
else
{
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "%s\n", errValueStr.c_str() );
}
}
else
{
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "%s\n", errValueStr.c_str() );
}
Py_XDECREF(errvalue);
Py_XDECREF(errtype);
Py_XDECREF(traceback);
}
else
{
Py_XDECREF(errvalue);
Py_XDECREF(errtype);
Py_XDECREF(traceback);
} while( !stopInput );
}
throw;
}
}
}
} while( true );
}
catch(...)
{
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "unexpected error" );