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.2.x] add getExtensionSearchPath

Browse Source 
[0.2.x] workitem/12654 fixed

git-svn-id: https://pykd.svn.codeplex.com/svn@87131 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\EreTIk_cp 2014-01-24 10:44:17 +00:00 committed by Mikhail I. Izmestev
parent 39ca17dd9a
commit d4b38e40cb
3 changed files with 71 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pykd/dbgengine.h
View File

@ -203,6 +203,7 @@ void setSymbolPath(const std::string &symPath);
void appendSymbolPath(const std::string &symPath);
// Extensions
std::wstring getExtensionSearchPath();
ULONG64 loadExtension(const std::wstring &extPath );
ULONG64 getExtension(const std::wstring &extPath );
void removeExtension( ULONG64 extHandle );

5
pykd/python/pymod.cpp
View File

@ -81,7 +81,9 @@ BOOST_PYTHON_MODULE( pykd )
// DbgEng services
python::def( "setSymSrvDir", &setSymSrvDir,
"Set directory of SYMSRV.dll library.\nUsually this is a directory of WinDbg");
"Set directory of SYMSRV.dll library.\nUsually this is a directory of WinDbg" );
python::def( "getExtensionSearchPath", &getExtensionSearchPath,
"Return the extension DLL search path" );
python::def( "loadExt", &loadExtension,
"Load a WinDBG extension. Return handle of the loaded extension" );
python::def( "getExt", &getExtension,
@ -91,7 +93,6 @@ BOOST_PYTHON_MODULE( pykd )
python::def( "callExt", &callExtension,
"Call a WinDBG extension's routine. Parameters: handle returned by loadExt; string command line" );
// Manage debug target
python::def( "startProcess", &startProcess, startProcess_( boost::python::args( "commandline", "debugChildren" ),

67
pykd/win/dbgeng.cpp
View File

@ -1289,6 +1289,48 @@ DebugEngine::DbgEngBind* DebugEngine::operator->()
///////////////////////////////////////////////////////////////////////////////
static std::wstring getExtensionSearchPathImpl()
{
ULONG chars = MAX_PATH;
for (; ; )
{
std::vector< wchar_t > rawPath(chars + 1, L'\0');
HRESULT hres =
g_dbgEng->advanced->Request(
DEBUG_REQUEST_GET_EXTENSION_SEARCH_PATH_WIDE,
NULL,
0,
&rawPath[0],
chars * sizeof(wchar_t),
NULL);
if (S_OK == hres)
{
std::wstring result = &rawPath[0];
return result;
}
if (S_FALSE == hres)
{
// The method was successful. However, the output would not fit in the
// output buffer OutBuffer, so truncated output was returned
chars *= 2;
continue;
}
throw DbgException( "IDebugAdvanced::Request", hres );
}
}
///////////////////////////////////////////////////////////////////////////////
std::wstring getExtensionSearchPath()
{
PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
return getExtensionSearchPathImpl();
}
///////////////////////////////////////////////////////////////////////////////
ULONG64 loadExtension(const std::wstring &extPath )
{
PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
@ -1296,10 +1338,35 @@ ULONG64 loadExtension(const std::wstring &extPath )
HRESULT hres;
ULONG64 handle = 0;
std::vector< wchar_t > rawPath(MAX_PATH + 1, L'\0');
DWORD ret =
::SearchPath(
getExtensionSearchPathImpl().c_str(),
extPath.c_str(),
L".dll",
MAX_PATH,
&rawPath[0],
NULL);
if (!ret)
throw DbgException( "extension not found" );
struct _scoped_lib
{
_scoped_lib(const wchar_t *wsz) : m_hmod(::LoadLibrary(wsz)) {}
~_scoped_lib() { if (m_hmod) ::FreeLibrary(m_hmod);}
HMODULE m_hmod;
} scoped_lib(&rawPath[0]);
if (!scoped_lib.m_hmod)
throw DbgException( "extension not found" );
hres = g_dbgEng->control->AddExtensionWide( extPath.c_str(), 0, &handle );
if ( FAILED( hres ) )
throw DbgException( "IDebugControl::AddExtension", hres );
// inderect call of dbgeng!ExtensionInfo::Load
FARPROC dummy = NULL;
g_dbgEng->control->GetExtensionFunctionWide(handle, L"dummy", &dummy);
return handle;
}