IDiaSymbol* PF_OpenAndFindGlobalScopeFromPdbFile(const wchar_t *szFilename)
{
IDiaSession *pSession = NULL;
IDiaSymbol *pGlobalScope = NULL;
IDiaDataSource *pSource;
HRESULT hr = CoCreateInstance(CLSID_DiaSource,
NULL,
CLSCTX_INPROC_SERVER,
__uuidof(IDiaDataSource),
(void **)&pSource);
if (FAILED(hr))
{
fwprintf_s(stdout, L"Could not CoCreate CLSID_DiaSource.\n");
return NULL;
}
if (FAILED(pSource->loadDataFromPdb(szFilename)))
{
fwprintf_s(stdout, L"loadDataFromPdb.\n");
return NULL;
}
if (FAILED(pSource->openSession(&pSession)))
{
fwprintf_s(stdout, L"openSession.\n");
pSession = 0;
return NULL;
}
if (FAILED(pSession->get_globalScope(&pGlobalScope)))
{
fwprintf_s(stdout, L"globalscope.\n");
pSession = 0;
pGlobalScope = 0;
return NULL;
}
pSession = 0;
return pGlobalScope;
}
HRESULT open(
const char *input_file,
const char *user_spath = NULL,
ea_t load_address = BADADDR,
input_exe_reader_t exe_reader = NULL,
input_mem_reader_t mem_reader = NULL)
{
// Already open?
if ( pSession != NULL )
return S_OK;
// When the debugger is active, first try to load debug directory from the memory
ea_t load_address_order[2];
// when remote debugging, don't use files on disk
bool remote_debug = false;
#ifndef BUILDING_EFD
#ifndef PDBTOTIL
if ( get_process_state() != DSTATE_NOTASK )
{
load_address_order[0] = load_address;
load_address_order[1] = BADADDR;
remote_debug = dbg->is_remote();
}
else
#endif
#endif
{
load_address_order[0] = BADADDR;
load_address_order[1] = load_address;
}
HRESULT hr;
do
{
// No interface was created?
hr = create_dia_source();
if ( FAILED(hr) )
break;
qwstring wpath, winput;
get_input_and_sym_path(input_file, user_spath, winput, wpath);
if ( exe_reader == NULL && mem_reader == NULL && !remote_debug )
// Try to load input file as PDB
hr = pSource->loadDataFromPdb(winput.c_str());
else
hr = E_FAIL;
// Failed? Try to load as EXE
if ( FAILED(hr) )
{
CCallback callback(exe_reader, mem_reader);
callback.AddRef();
if ( !remote_debug )
{
// Open the executable
callback.OpenExe(winput.c_str());
}
for ( int i=0; i < qnumber(load_address_order); i++ )
{
callback.SetLoadAddress(load_address_order[i]);
hr = pSource->loadDataForExe(winput.c_str(), wpath.c_str(), (IDiaLoadCallback *)&callback);
if ( SUCCEEDED(hr) )
break;
}
}
// Failed? Then nothing else to try, quit
if ( FAILED(hr) )
break;
// Open a session for querying symbols
hr = pSource->openSession(&pSession);
if ( FAILED(hr) )
break;
// Set load address
if ( load_address != BADADDR )
pSession->put_loadAddress(load_address);
// Retrieve a reference to the global scope
hr = pSession->get_globalScope(&pGlobal);
if ( FAILED(hr) )
break;
hr = S_OK;
} while ( false );
// Make sure we cleanup
if ( FAILED(hr) )
close();
return hr;
}
bool LoadDIASymbols(const wxString &strPDBFile,
wxInt32 modulenum,
std::list<CFunctionDescription> & llFuncs,
stdext::hash_map<BasedAddress,int> & addressmap
)
{
// Create DIA80 Data Source Object
IDiaDataSource *pDataSource;
HRESULT hr = NoRegCoCreate( L"msdia80.dll", _uuidof( DiaSource ), _uuidof( IDiaDataSource ), (void **) &(pDataSource));
if(!SUCCEEDED(hr))
{
return false;
}
/*
HMODULE mod=LoadLibrary(L"msdia80.dll");
if(!mod)
{
return false;
}
TYPEOF_DllGetClassObject *pGetClassObject=(TYPEOF_DllGetClassObject *)GetProcAddress(mod,"DllGetClassObject");
if(!pGetClassObject)
{
FreeLibrary(mod);
return false;
}
IClassFactory *pcf;
HRESULT hr=(*pGetClassObject)(CLSID_DiaSource,IID_IClassFactory,(void **)&pcf);
if(!SUCCEEDED(hr))
{
FreeLibrary(mod);
return false;
}
IDiaDataSource *pDataSource;
hr=pcf->CreateInstance(NULL,_uuidof(IDiaDataSource),(void **)&pDataSource);
if(!SUCCEEDED(hr))
{
pcf->Release();
FreeLibrary(mod);
return false;
}
pcf->Release();
*/
// Load the executable's debug symbols
hr=pDataSource->loadDataFromPdb(strPDBFile);
if(!SUCCEEDED(hr))
{
pDataSource->Release();
return false;
}
// Open a symbol session
IDiaSession *pDIASession;
hr=pDataSource->openSession(&pDIASession);
if(!SUCCEEDED(hr))
{
pDataSource->Release();
return false;
}
// Set the UNBASED address on the session, for resolving BasedAddress addrs
hr=pDIASession->put_loadAddress(0);
if(!SUCCEEDED(hr))
{
pDIASession->Release();
pDataSource->Release();
return false;
}
// Get addresses for this module
std::list<BasedAddress> addresses;
for(stdext::hash_map<BasedAddress,int>::iterator iter=addressmap.begin();iter!=addressmap.end();iter++)
{
if(iter->first.nModule==modulenum)
{
addresses.push_back(iter->first);
}
}
stdext::hash_map<DWORD,wxInt32> functions;
int curidx=(int)llFuncs.size();
for(std::list<BasedAddress>::iterator itera=addresses.begin();itera!=addresses.end();itera++)
{
// Get the symbol for this thing
IDiaSymbol* pFunc;
if(FAILED(pDIASession->findSymbolByVA(itera->nAddr, SymTagFunction, &pFunc)) || pFunc==NULL)
{
continue;
}
// Get the unique symbol index id
DWORD indexId;
if(FAILED(pFunc->get_symIndexId(&indexId)))
{
pFunc->Release();
continue;
}
// See if we have this function already
stdext::hash_map<DWORD,int>::iterator iterf=functions.find(indexId);
if(iterf!=functions.end())
{
addressmap[*itera]=iterf->second;
}
else
{
CFunctionDescription func;
//////
ULONGLONG addr;
if(FAILED(pFunc->get_virtualAddress(&addr)))
{
pFunc->Release();
continue;
}
//////
ULONGLONG length;
if(FAILED(pFunc->get_length(&length)))
{
pFunc->Release();
continue;
}
/////////
BSTR pName;
if(FAILED(pFunc->get_name(&pName)))
{
pFunc->Release();
continue;
}
func.strMangledName=(const LPWSTR)pName;
LocalFree(pName);
char mangled[512];
WideCharToMultiByte(CP_UTF8,0,func.strMangledName,-1,mangled,512,NULL,NULL);
/////////
char outbase[512];
__unDName(outbase,mangled,512,&malloc,&free,0x1000);
wchar_t outbasew[512];
MultiByteToWideChar(CP_UTF8,0,outbase,-1,outbasew,512);
func.strBaseName=(const wchar_t *)(outbasew);
/////////
char outfull[512];
__unDName(outfull,mangled,512,&malloc,&free,0);
wchar_t outfullw[512];
MultiByteToWideChar(CP_UTF8,0,outfull,-1,outfullw,512);
func.strFullName=(const wchar_t *)(outfullw);
//////////
func.nAddress=BasedAddress(addr,modulenum);
func.nByteLength=length;
/////////
IDiaEnumLineNumbers *pdeln;
func.strSourceFile=wxT("<unknown>");
func.nFirstSourceLine=0;
func.nLastSourceLine=0;
bool bGotFile=false;
if(SUCCEEDED(pDIASession->findLinesByVA(addr,length,&pdeln)))
{
IDiaLineNumber *pdln;
int nLineItem=0;
while(SUCCEEDED(pdeln->Item(nLineItem,&pdln)))
{
DWORD dwLineNumber;
IDiaSourceFile *pdsf;
if(SUCCEEDED(pdln->get_lineNumber(&dwLineNumber)) &&
SUCCEEDED(pdln->get_sourceFile(&pdsf)))
{
BSTR filename;
if(SUCCEEDED(pdsf->get_fileName(&filename)))
{
if(bGotFile)
{
if(filename==func.strSourceFile)
{
if(dwLineNumber<func.nFirstSourceLine)
{
func.nFirstSourceLine=dwLineNumber;
}
if(dwLineNumber>func.nLastSourceLine)
{
func.nLastSourceLine=dwLineNumber;
}
}
}
else
{
bGotFile=true;
func.strSourceFile=filename;
func.nFirstSourceLine=dwLineNumber;
func.nLastSourceLine=dwLineNumber;
}
LocalFree(filename);
}
pdsf->Release();
}
pdln->Release();
nLineItem++;
}
pdeln->Release();
}
llFuncs.push_back(func);
functions[indexId]=curidx;
addressmap[*itera]=curidx;
curidx++;
}
}
pDIASession->Release();
pDataSource->Release();
// FreeLibrary(mod);
return true;
}