mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-29 11:53:23 +08:00
101 lines
3.0 KiB
C++
101 lines
3.0 KiB
C++
#include "stdafx.h"
|
|
|
|
#include "dbgext.h"
|
|
#include "dbgdump.h"
|
|
#include "dbgexcept.h"
|
|
#include "dbgeventcb.h"
|
|
#include "dbgsystem.h"
|
|
#include "dbgcmd.h"
|
|
#include "dbgclient.h"
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
static
|
|
bool dbgStarted = false;
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void
|
|
dbgLoadDump( const std::wstring &fileName )
|
|
{
|
|
HRESULT hres;
|
|
|
|
if ( dbgStarted || isWindbgExt() )
|
|
throw DbgException( "debugger is alread attached" );
|
|
|
|
g_dbgClient.startEventsMgr();
|
|
|
|
hres = dbgExt->client4->OpenDumpFileWide( fileName.c_str(), NULL );
|
|
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugClient4::OpenDumpFileWide failed" );
|
|
|
|
hres = dbgExt->control->WaitForEvent(DEBUG_WAIT_DEFAULT, INFINITE);
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugControl::WaitForEvent failed" );
|
|
|
|
dbgStarted = true;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void
|
|
startProcess( const std::wstring &processName )
|
|
{
|
|
HRESULT hres;
|
|
|
|
if ( dbgStarted || isWindbgExt() )
|
|
throw DbgException( "debugger is alread attached" );
|
|
|
|
g_dbgClient.startEventsMgr();
|
|
|
|
ULONG opt;
|
|
hres = dbgExt->control->GetEngineOptions( &opt );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugControl::GetEngineOptions failed" );
|
|
|
|
opt |= DEBUG_ENGOPT_INITIAL_BREAK;
|
|
hres = dbgExt->control->SetEngineOptions( opt );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugControl::SetEngineOptions failed" );
|
|
|
|
std::vector< std::wstring::value_type> cmdLine( processName.size() + 1 );
|
|
wcscpy_s( &cmdLine[0], cmdLine.size(), processName.c_str() );
|
|
|
|
hres = dbgExt->client4->CreateProcessWide( 0, &cmdLine[0], DEBUG_PROCESS | DETACHED_PROCESS );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugClient4::CreateProcessWide failed" );
|
|
|
|
hres = dbgExt->control->WaitForEvent(DEBUG_WAIT_DEFAULT, INFINITE);
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugControl::WaitForEvent failed" );
|
|
|
|
dbgStarted = true;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void
|
|
attachProcess( ULONG processId )
|
|
{
|
|
HRESULT hres;
|
|
|
|
hres = dbgExt->client->AttachProcess( 0, processId, 0 );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugClient::AttachProcess failed" );
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void
|
|
attachKernel( const std::wstring param )
|
|
{
|
|
HRESULT hres;
|
|
|
|
hres = dbgExt->client5->AttachKernelWide( DEBUG_ATTACH_KERNEL_CONNECTION, param.c_str() );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugClient5::AttachKernelWide failed" );
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|