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

[0.2.x] added: getProcessThreads ( Get all process's threads ( user mode only ) )

Browse Source 
git-svn-id: https://pykd.svn.codeplex.com/svn@81956 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\kernelnet_cp 2012-12-21 14:34:47 +00:00 committed by Mikhail I. Izmestev
parent 18392dca83
commit 4a50b8b563
6 changed files with 264 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pykd/dbgengine.h
View File

@ -141,6 +141,7 @@ ULONG64 getCurrentProcess();
ULONG64 getImplicitThread();
void setCurrentProcess( ULONG64 processAddr );
void setImplicitThread( ULONG64 threadAddr );
void getAllProcessThreads( std::vector<ULONG64> &threadsArray );
// Symbol path
std::string getSymbolPath();

5
pykd/pymod.cpp → pykd/python/pymod.cpp
View File

@ -21,13 +21,14 @@
#include "stkframe.h"
#include "bpoint.h"
#include "eventhandler.h"
#include "pysupport.h"
#include "win/dbgio.h"
#include "win/windbg.h"
using namespace pykd;
////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////
static const std::string pykdVersion = PYKD_VERSION_BUILD_STR
#ifdef _DEBUG
@ -281,6 +282,8 @@ BOOST_PYTHON_MODULE( pykd )
"Set current process by address" );
python::def( "setImplicitThread", &setImplicitThread,
"Set implicit thread for current process" );
python::def( "getProcessThreads", &pysupport::getProcessThreads,
"Get all process's threads ( user mode only )" );
// symbol path
python::def( "getSymbolPath", &getSymbolPath, "Returns current symbol path");

30
pykd/python/pysupport.cpp Normal file
View File

@ -0,0 +1,30 @@
#include "stdafx.h"
#include "pysupport.h"
#include "dbgengine.h"
#include <vector>
///////////////////////////////////////////////////////////////////////////////
namespace pykd {
namespace pysupport {
///////////////////////////////////////////////////////////////////////////////
python::list getProcessThreads()
{
std::vector<ULONG64> threads;
getAllProcessThreads( threads );
python::list threadsLst;
std::vector<ULONG64>::iterator it;
for ( it = threads.begin(); it != threads.end(); ++it )
threadsLst.append( *it );
return threadsLst;
}
///////////////////////////////////////////////////////////////////////////////
} } //pykd::support namespace end

14
pykd/python/pysupport.h Normal file
View File

@ -0,0 +1,14 @@
#pragma once
#include <boost/python/list.hpp>
///////////////////////////////////////////////////////////////////////////////
namespace pykd {
namespace pysupport {
python::list getProcessThreads();
} } //pykd::support namespace end
///////////////////////////////////////////////////////////////////////////////

55
pykd/win/dbgeng.cpp
View File

@ -1365,6 +1365,61 @@ void setImplicitThread( ULONG64 threadAddr )
///////////////////////////////////////////////////////////////////////////////
void getAllProcessThreads( std::vector<ULONG64> &threadsArray )
{
PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
HRESULT hres;
ULONG debugClass, debugQualifier;
hres = g_dbgEng->control->GetDebuggeeType( &debugClass, &debugQualifier );
if ( FAILED( hres ) )
throw DbgException( "IDebugControl::GetDebuggeeType failed" );
if ( debugClass != DEBUG_CLASS_USER_WINDOWS )
throw DbgException( "getAllProcessThreads routine only for usermode" );
ULONG threadsNumber = 0;
hres = g_dbgEng->system->GetNumberThreads( &threadsNumber );
if ( FAILED(hres) )
throw DbgException( "IDebugSystemObjects::GetNumberThreads failed" );
std::vector<ULONG> ids(threadsNumber);
hres = g_dbgEng->system->GetThreadIdsByIndex( 0, threadsNumber, &ids[0], NULL );
if ( FAILED(hres) )
throw DbgException( "IDebugSystemObjects::GetThreadIdsByIndex failed" );
ULONG currentThreadId;
hres = g_dbgEng->system->GetCurrentThreadId( &currentThreadId );
if ( FAILED(hres) )
throw DbgException( "IDebugSystemObjects::GetCurrentThreadId failed" );
threadsArray.resize( threadsNumber );
for ( size_t i = 0; i < threadsNumber; ++i )
{
hres = g_dbgEng->system->SetCurrentThreadId( ids[i] );
if ( FAILED(hres) )
{
g_dbgEng->system->SetCurrentThreadId( currentThreadId );
throw DbgException( "IDebugSystemObjects::SetCurrentThreadId failed" );
}
hres = g_dbgEng->system->GetCurrentThreadTeb( &threadsArray[i] );
if ( FAILED(hres) )
{
g_dbgEng->system->SetCurrentThreadId( currentThreadId );
throw DbgException( "IDebugSystemObjects::GetCurrentThreadTeb failed" );
}
}
g_dbgEng->system->SetCurrentThreadId( currentThreadId );
}
///////////////////////////////////////////////////////////////////////////////
ULONG64 loadExtension(const std::wstring &extPath )
{
PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );

161
test/targetapp/targetapp.vcproj
View File

@ -1,7 +1,7 @@
<?xml version="1.0" encoding="windows-1251"?>
<VisualStudioProject
ProjectType="Visual C++"
Version="9.00"
Version="9,00"
Name="targetapp"
ProjectGUID="{C6254E16-AB8E-41EE-887D-31458E93FC68}"
RootNamespace="targetapp"
@ -315,6 +315,149 @@
Name="VCPostBuildEventTool"
/>
</Configuration>
<Configuration
Name="Debug_2.7|Win32"
OutputDirectory="$(SolutionDir)$(ConfigurationName)"
IntermediateDirectory="$(ConfigurationName)"
ConfigurationType="1"
CharacterSet="1"
>
<Tool
Name="VCPreBuildEventTool"
/>
<Tool
Name="VCCustomBuildTool"
/>
<Tool
Name="VCXMLDataGeneratorTool"
/>
<Tool
Name="VCWebServiceProxyGeneratorTool"
/>
<Tool
Name="VCMIDLTool"
/>
<Tool
Name="VCCLCompilerTool"
Optimization="0"
PreprocessorDefinitions="WIN32;_DEBUG;_CONSOLE"
MinimalRebuild="true"
BasicRuntimeChecks="3"
RuntimeLibrary="3"
UsePrecompiledHeader="2"
WarningLevel="3"
DebugInformationFormat="4"
/>
<Tool
Name="VCManagedResourceCompilerTool"
/>
<Tool
Name="VCResourceCompilerTool"
/>
<Tool
Name="VCPreLinkEventTool"
/>
<Tool
Name="VCLinkerTool"
LinkIncremental="2"
GenerateDebugInformation="true"
SubSystem="1"
TargetMachine="1"
/>
<Tool
Name="VCALinkTool"
/>
<Tool
Name="VCManifestTool"
/>
<Tool
Name="VCXDCMakeTool"
/>
<Tool
Name="VCBscMakeTool"
/>
<Tool
Name="VCFxCopTool"
/>
<Tool
Name="VCAppVerifierTool"
/>
<Tool
Name="VCPostBuildEventTool"
/>
</Configuration>
<Configuration
Name="Debug_2.7|x64"
OutputDirectory="$(SolutionDir)$(PlatformName)\$(ConfigurationName)"
IntermediateDirectory="$(PlatformName)\$(ConfigurationName)"
ConfigurationType="1"
CharacterSet="1"
>
<Tool
Name="VCPreBuildEventTool"
/>
<Tool
Name="VCCustomBuildTool"
/>
<Tool
Name="VCXMLDataGeneratorTool"
/>
<Tool
Name="VCWebServiceProxyGeneratorTool"
/>
<Tool
Name="VCMIDLTool"
TargetEnvironment="3"
/>
<Tool
Name="VCCLCompilerTool"
Optimization="0"
PreprocessorDefinitions="WIN32;_DEBUG;_CONSOLE"
MinimalRebuild="true"
BasicRuntimeChecks="3"
RuntimeLibrary="3"
UsePrecompiledHeader="2"
WarningLevel="3"
DebugInformationFormat="3"
/>
<Tool
Name="VCManagedResourceCompilerTool"
/>
<Tool
Name="VCResourceCompilerTool"
/>
<Tool
Name="VCPreLinkEventTool"
/>
<Tool
Name="VCLinkerTool"
LinkIncremental="2"
GenerateDebugInformation="true"
SubSystem="1"
TargetMachine="17"
/>
<Tool
Name="VCALinkTool"
/>
<Tool
Name="VCManifestTool"
/>
<Tool
Name="VCXDCMakeTool"
/>
<Tool
Name="VCBscMakeTool"
/>
<Tool
Name="VCFxCopTool"
/>
<Tool
Name="VCAppVerifierTool"
/>
<Tool
Name="VCPostBuildEventTool"
/>
</Configuration>
</Configurations>
<References>
</References>
@ -359,6 +502,22 @@
UsePrecompiledHeader="1"
/>
</FileConfiguration>
<FileConfiguration
Name="Debug_2.7|Win32"
>
<Tool
Name="VCCLCompilerTool"
UsePrecompiledHeader="1"
/>
</FileConfiguration>
<FileConfiguration
Name="Debug_2.7|x64"
>
<Tool
Name="VCCLCompilerTool"
UsePrecompiledHeader="1"
/>
</FileConfiguration>
</File>
<File
RelativePath=".\targetapp.cpp"