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.3.x] fixed: compability with GILState API

Browse Source 
git-svn-id: https://pykd.svn.codeplex.com/svn@86899 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\kernelnet_cp 2013-12-20 09:37:05 +00:00 committed by Mikhail I. Izmestev
parent f3e27d30b8
commit 9e00a5a289
1 changed files with 160 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

148
pykd/windbgext.cpp
View File

@ -94,6 +94,124 @@ void PykdExt::tearDown()
/////////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////////
class InterprterVirt
{
public:
InterprterVirt()
{
PyThreadState *threadState = PyGILState_GetThisThreadState();
modules = threadState->interp->modules;
modules_reloading = threadState->interp->modules_reloading;
sysdict = threadState->interp->sysdict;
builtins = threadState->interp->builtins;
codec_search_path = threadState->interp->codec_search_path;
codec_search_cache = threadState->interp->codec_search_cache;
codec_error_registry = threadState->interp->codec_error_registry;
newThread = NULL;
}
~InterprterVirt()
{
PyThreadState *threadState =PyGILState_GetThisThreadState();
if ( threadState->interp->modules != modules )
{
Py_DECREF(threadState->interp->modules);
threadState->interp->modules = modules;
}
if ( threadState->interp->modules_reloading != modules_reloading )
{
Py_DECREF(threadState->interp->modules_reloading);
threadState->interp->modules_reloading = modules_reloading;
}
if ( threadState->interp->sysdict != sysdict )
{
Py_DECREF(threadState->interp->sysdict);
threadState->interp->sysdict = sysdict;
}
if ( threadState->interp->builtins != builtins )
{
Py_DECREF(threadState->interp->builtins);
threadState->interp->builtins = builtins;
}
if ( threadState->interp->codec_search_path != codec_search_path )
{
Py_DECREF(threadState->interp->codec_search_path);
threadState->interp->codec_search_path = codec_search_path;
}
if ( threadState->interp->codec_search_cache != codec_search_cache )
{
Py_DECREF(threadState->interp->codec_search_cache);
threadState->interp->codec_search_cache = codec_search_cache;
}
if ( threadState->interp->codec_error_registry != codec_error_registry )
{
Py_DECREF(threadState->interp->codec_error_registry);
threadState->interp->codec_search_cache = codec_error_registry;
}
if (newThread)
{
PyThreadState* current = PyThreadState_Get();
PyThreadState_Swap(newThread);
Py_EndInterpreter(newThread);
PyThreadState_Swap(current);
}
}
void fork()
{
PyThreadState *threadState =PyGILState_GetThisThreadState();
newThread = Py_NewInterpreter();
threadState->interp->modules = newThread->interp->modules;
Py_INCREF(threadState->interp->modules);
threadState->interp->modules_reloading = newThread->interp->modules_reloading;
Py_INCREF(threadState->interp->modules_reloading);
threadState->interp->sysdict = newThread->interp->sysdict;
Py_INCREF(threadState->interp->sysdict);
threadState->interp->builtins = newThread->interp->builtins;
Py_INCREF(threadState->interp->builtins);
threadState->interp->codec_search_path = newThread->interp->codec_search_path;
Py_INCREF(threadState->interp->codec_search_path);
threadState->interp->codec_search_cache = newThread->interp->codec_search_cache;
Py_INCREF(threadState->interp->codec_search_cache);
threadState->interp->codec_error_registry = newThread->interp->codec_error_registry;
Py_INCREF(threadState->interp->codec_error_registry);
PyThreadState_Swap(threadState);
}
private:
PyObject *modules;
PyObject *sysdict;
PyObject *builtins;
PyObject *modules_reloading;
PyObject *codec_search_path;
PyObject *codec_search_cache;
PyObject *codec_error_registry;
PyThreadState *newThread;
};
KDLIB_EXT_COMMAND_METHOD_IMPL(PykdExt, py) KDLIB_EXT_COMMAND_METHOD_IMPL(PykdExt, py)
{ {
ArgsList args = getArgs(); ArgsList args = getArgs();
@ -175,11 +293,13 @@ KDLIB_EXT_COMMAND_METHOD_IMPL(PykdExt, py)
PyEval_RestoreThread( m_pyState ); PyEval_RestoreThread( m_pyState );
do {
InterprterVirt interpretrVirt;
if ( !global ) if ( !global )
{ {
globalState = PyThreadState_Swap( NULL ); interpretrVirt.fork();
localState = Py_NewInterpreter();
python::object sys = python::import("sys"); python::object sys = python::import("sys");
@ -223,27 +343,7 @@ KDLIB_EXT_COMMAND_METHOD_IMPL(PykdExt, py)
} }
} }
if ( !global ) } while( false);
{
PyInterpreterState *interpreter = localState->interp;
while( interpreter->tstate_head != NULL )
{
PyThreadState *threadState = (PyThreadState*)(interpreter->tstate_head);
PyThreadState_Clear(threadState);
PyThreadState_Swap( NULL );
PyThreadState_Delete(threadState);
}
PyInterpreterState_Clear(interpreter);
PyInterpreterState_Delete(interpreter);
PyThreadState_Swap( globalState );
}
m_pyState = PyEval_SaveThread(); m_pyState = PyEval_SaveThread();