-
Notifications
You must be signed in to change notification settings - Fork 2
/
Symbols.cpp
75 lines (69 loc) · 1.67 KB
/
Symbols.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#include "stdafx.h"
#include "Symbols.h"
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;
}
ULONG64 PF_GetVirtualAddressOffsetForSymbolName(IDiaSymbol *pGlobalScope, const wchar_t *nameToFind)
{
ULONG length = wcslen(nameToFind);
CComPtr<IDiaEnumSymbols> pEnum;
pGlobalScope->findChildren(SymTagEnum::SymTagPublicSymbol, NULL, 0, &pEnum);
CComPtr< IDiaSymbol > pSymbol;
DWORD tag;
DWORD celt;
while (pEnum != NULL && SUCCEEDED(pEnum->Next(1, &pSymbol, &celt)) && celt == 1)
{
pSymbol->get_symTag(&tag);
if (tag == SymTagPublicSymbol)
{
BSTR name;
ULONG64 va;
pSymbol->get_undecoratedNameEx(0x1000, &name);
if (name != NULL)
{
if (wcscmp(name, nameToFind) == 0)
{
if (pSymbol->get_virtualAddress(&va) == S_OK)
return va;
else
return 0;
}
}
}
pSymbol = NULL;
}
return 0;
}