x64dbg/x64dbg
x64dbg
/
x64dbg
1
0
Fork 0
Code Releases 2 Activity

DBG: implement DiaLoadCallback for loadDataForExe

This commit is contained in:
Duncan Ogilvie 2017-12-20 14:50:19 +01:00
parent 4fadd01ad4
commit caa5b6273f
No known key found for this signature in database
GPG Key ID: FC89E0AAA0C1AAD8
1 changed files with 75 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
src/dbg/pdbdiafile.cpp
View File

@ -13,6 +13,79 @@
#include "stringutils.h"
#include "console.h"
class DiaLoadCallback : public IDiaLoadCallback2
{
virtual HRESULT STDMETHODCALLTYPE NotifyDebugDir(
/* [in] */ BOOL fExecutable,
/* [in] */ DWORD cbData,
/* [size_is][in] */ BYTE* pbData) override
{
dprintf("[DIA] NotifyDebugDir: %s\n", StringUtils::ToHex(pbData, cbData).c_str());
return S_OK;
}
virtual HRESULT STDMETHODCALLTYPE NotifyOpenDBG(
/* [in] */ LPCOLESTR dbgPath,
/* [in] */ HRESULT resultCode) override
{
dprintf("[DIA] NotifyOpenDBG: %s, %08X\n", StringUtils::Utf16ToUtf8(dbgPath).c_str(), resultCode);
return S_OK;
}
virtual HRESULT STDMETHODCALLTYPE NotifyOpenPDB(
/* [in] */ LPCOLESTR pdbPath,
/* [in] */ HRESULT resultCode) override
{
dprintf("[DIA] NotifyOpenPDB: %s, %08X\n", StringUtils::Utf16ToUtf8(pdbPath).c_str(), resultCode);
return S_OK;
}
virtual HRESULT STDMETHODCALLTYPE RestrictRegistryAccess(void) override { return S_OK; }
virtual HRESULT STDMETHODCALLTYPE RestrictSymbolServerAccess(void) override { return S_OK; }
virtual HRESULT STDMETHODCALLTYPE RestrictDBGAccess(void) override { return S_OK; }
virtual HRESULT STDMETHODCALLTYPE RestrictOriginalPathAccess(void) override { return S_OK; }
virtual HRESULT STDMETHODCALLTYPE RestrictReferencePathAccess(void) override { return S_OK; }
virtual HRESULT STDMETHODCALLTYPE RestrictSystemRootAccess(void) override { return S_OK; }
//TODO: properly implement IUnknown (https://msdn.microsoft.com/en-us/library/office/cc839627.aspx)
HRESULT STDMETHODCALLTYPE QueryInterface(REFIID rid, _Outptr_ void** ppUnk) override
{
if(ppUnk == NULL)
{
return E_INVALIDARG;
}
if(rid == __uuidof(IDiaLoadCallback2))
*ppUnk = (IDiaLoadCallback2*)this;
else if(rid == __uuidof(IDiaLoadCallback))
*ppUnk = (IDiaLoadCallback*)this;
else if(rid == __uuidof(IUnknown))
*ppUnk = (IUnknown*)this;
else
*ppUnk = NULL;
if(*ppUnk != NULL)
{
AddRef();
return S_OK;
}
return E_NOINTERFACE;
}
ULONG STDMETHODCALLTYPE AddRef() override
{
return 1;
}
ULONG STDMETHODCALLTYPE Release() override
{
return 1;
}
};
volatile LONG PDBDiaFile::m_sbInitialized = 0;
template<typename T>
@ -146,7 +219,8 @@ bool PDBDiaFile::open(const wchar_t* file, uint64_t loadAddress, DiaValidationDa
}
else
{
hr = m_dataSource->loadDataForExe(file, fileDir, NULL);
DiaLoadCallback callback;
hr = m_dataSource->loadDataForExe(file, fileDir, &callback);
}
if(testError(hr))