96 lines
2.6 KiB
C++
96 lines
2.6 KiB
C++
|
#ifndef StackWalk_H
|
||
|
|
#define StackWalk_H 1
|
||
|
|
|
||
|
|
#include <imagehlp.h>
|
||
|
|
#include <wtypes.h>
|
||
|
|
|
||
|
|
|
||
|
|
class Symbol {
|
||
|
|
friend class StackWalker;
|
||
|
|
|
||
|
|
private:
|
||
|
|
Symbol(const char* moduleName, const char* symbolName, DWORD_PTR displacement);
|
||
|
|
void Append(Symbol*);
|
||
|
|
|
||
|
|
|
||
|
|
public:
|
||
|
|
~Symbol();
|
||
|
|
|
||
|
|
const char* moduleName() const { return _moduleName; }
|
||
|
|
const char* symbolName() const { return _symbolName; }
|
||
|
|
DWORD_PTR displacement() const { return _displacement; }
|
||
|
|
void AppendDisplacement(char * sz)
|
||
|
|
{
|
||
|
|
char szDisp[16];
|
||
|
|
wsprintfA(szDisp, " + 0x%X", _displacement);
|
||
|
|
lstrcatA(sz, szDisp);
|
||
|
|
}
|
||
|
|
|
||
|
|
Symbol* next() const { return _next; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
char* _moduleName;
|
||
|
|
char* _symbolName;
|
||
|
|
DWORD_PTR _displacement;
|
||
|
|
|
||
|
|
Symbol* _next;
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
class StackWalker {
|
||
|
|
public:
|
||
|
|
StackWalker(HANDLE hProcess);
|
||
|
|
~StackWalker();
|
||
|
|
|
||
|
|
Symbol* ResolveAddress(DWORD_PTR addr);
|
||
|
|
Symbol* CreateStackTrace(CONTEXT*);
|
||
|
|
BOOL GetCallStack(Symbol * symbol, SIZE_T nChars, WCHAR * sz, SIZE_T nMaxLines=999999999);
|
||
|
|
SIZE_T GetCallStackSize(Symbol* symbol);
|
||
|
|
|
||
|
|
private:
|
||
|
|
static DWORD_PTR __stdcall GetModuleBase(HANDLE hProcess, DWORD_PTR address);
|
||
|
|
static DWORD_PTR LoadModule(HANDLE hProcess, DWORD_PTR address);
|
||
|
|
|
||
|
|
private:
|
||
|
|
typedef BOOL (__stdcall *SymGetModuleInfoFunc)(HANDLE hProcess,
|
||
|
|
DWORD_PTR dwAddr,
|
||
|
|
PIMAGEHLP_MODULE ModuleInfo);
|
||
|
|
typedef BOOL (__stdcall *SymGetSymFromAddrFunc)(HANDLE hProcess,
|
||
|
|
DWORD_PTR dwAddr,
|
||
|
|
PDWORD_PTR pdwDisplacement,
|
||
|
|
PIMAGEHLP_SYMBOL Symbol);
|
||
|
|
typedef DWORD (__stdcall *SymLoadModuleFunc)(HANDLE hProcess,
|
||
|
|
HANDLE hFile,
|
||
|
|
PSTR ImageName,
|
||
|
|
PSTR ModuleName,
|
||
|
|
DWORD_PTR BaseOfDll,
|
||
|
|
DWORD SizeOfDll);
|
||
|
|
typedef BOOL (__stdcall *StackWalkFunc)(DWORD MachineType,
|
||
|
|
HANDLE hProcess,
|
||
|
|
HANDLE hThread,
|
||
|
|
LPSTACKFRAME StackFrame,
|
||
|
|
LPVOID ContextRecord,
|
||
|
|
PREAD_PROCESS_MEMORY_ROUTINE ReadMemoryRoutine,
|
||
|
|
PFUNCTION_TABLE_ACCESS_ROUTINE FunctionTableAccessRoutine,
|
||
|
|
PGET_MODULE_BASE_ROUTINE GetModuleBaseRoutine,
|
||
|
|
PTRANSLATE_ADDRESS_ROUTINE TranslateAddress);
|
||
|
|
typedef BOOL (__stdcall *UndecorateSymbolNameFunc)(LPSTR DecName,
|
||
|
|
LPSTR UnDecName,
|
||
|
|
DWORD UnDecNameLength,
|
||
|
|
DWORD Flags);
|
||
|
|
|
||
|
|
private:
|
||
|
|
HMODULE _imageHlpDLL;
|
||
|
|
HANDLE _hProcess;
|
||
|
|
EXCEPTION_POINTERS m_exceptionpts;
|
||
|
|
|
||
|
|
static SymGetModuleInfoFunc _SymGetModuleInfo;
|
||
|
|
static SymGetSymFromAddrFunc _SymGetSymFromAddr;
|
||
|
|
static SymLoadModuleFunc _SymLoadModule;
|
||
|
|
static StackWalkFunc _StackWalk;
|
||
|
|
static UndecorateSymbolNameFunc _UndecorateSymbolName;
|
||
|
|
static PFUNCTION_TABLE_ACCESS_ROUTINE _SymFunctionTableAccess;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|