1221 lines
		
	
	
		
			41 KiB
		
	
	
	
		
			C++
		
	
	
	
		
		
			
		
	
	
			1221 lines
		
	
	
		
			41 KiB
		
	
	
	
		
			C++
		
	
	
	
|  | /**********************************************************************
 | |||
|  |  *  | |||
|  |  * StackWalker.cpp | |||
|  |  * | |||
|  |  * | |||
|  |  * History: | |||
|  |  *  2005-07-27   v1    - First public release on http://www.codeproject.com/
 | |||
|  |  *                       http://www.codeproject.com/threads/StackWalker.asp
 | |||
|  |  *  2005-07-28   v2    - Changed the params of the constructor and ShowCallstack | |||
|  |  *                       (to simplify the usage) | |||
|  |  *  2005-08-01   v3    - Changed to use 'CONTEXT_FULL' instead of CONTEXT_ALL  | |||
|  |  *                       (should also be enough) | |||
|  |  *                     - Changed to compile correctly with the PSDK of VC7.0 | |||
|  |  *                       (GetFileVersionInfoSizeA and GetFileVersionInfoA is wrongly defined: | |||
|  |  *                        it uses LPSTR instead of LPCSTR as first paremeter) | |||
|  |  *                     - Added declarations to support VC5/6 without using 'dbghelp.h' | |||
|  |  *                     - Added a 'pUserData' member to the ShowCallstack function and the  | |||
|  |  *                       PReadProcessMemoryRoutine declaration (to pass some user-defined data,  | |||
|  |  *                       which can be used in the readMemoryFunction-callback) | |||
|  |  *  2005-08-02   v4    - OnSymInit now also outputs the OS-Version by default | |||
|  |  *                     - Added example for doing an exception-callstack-walking in main.cpp | |||
|  |  *                       (thanks to owillebo: http://www.codeproject.com/script/profile/whos_who.asp?id=536268)
 | |||
|  |  *  2005-08-05   v5    - Removed most Lint (http://www.gimpel.com/) errors... thanks to Okko Willeboordse!
 | |||
|  |  * | |||
|  |  **********************************************************************/ | |||
|  | #include <windows.h>
 | |||
|  | #include <tchar.h>
 | |||
|  | #include <stdio.h>
 | |||
|  | #pragma comment(lib, "version.lib")  // for "VerQueryValue"
 | |||
|  | 
 | |||
|  | #include "StackWalker.h"
 | |||
|  | 
 | |||
|  | 
 | |||
|  | // If VC7 and later, then use the shipped 'dbghelp.h'-file
 | |||
|  | #if _MSC_VER >= 1300
 | |||
|  | #include <dbghelp.h>
 | |||
|  | #else
 | |||
|  | // inline the important dbghelp.h-declarations...
 | |||
|  | typedef enum { | |||
|  |     SymNone = 0, | |||
|  |     SymCoff, | |||
|  |     SymCv, | |||
|  |     SymPdb, | |||
|  |     SymExport, | |||
|  |     SymDeferred, | |||
|  |     SymSym, | |||
|  |     SymDia, | |||
|  |     SymVirtual, | |||
|  |     NumSymTypes | |||
|  | } SYM_TYPE; | |||
|  | typedef struct _IMAGEHLP_LINE64 { | |||
|  |     DWORD                       SizeOfStruct;           // set to sizeof(IMAGEHLP_LINE64)
 | |||
|  |     PVOID                       Key;                    // internal
 | |||
|  |     DWORD                       LineNumber;             // line number in file
 | |||
|  |     PCHAR                       FileName;               // full filename
 | |||
|  |     DWORD64                     Address;                // first instruction of line
 | |||
|  | } IMAGEHLP_LINE64, *PIMAGEHLP_LINE64; | |||
|  | typedef struct _IMAGEHLP_MODULE64 { | |||
|  |     DWORD                       SizeOfStruct;           // set to sizeof(IMAGEHLP_MODULE64)
 | |||
|  |     DWORD64                     BaseOfImage;            // base load address of module
 | |||
|  |     DWORD                       ImageSize;              // virtual size of the loaded module
 | |||
|  |     DWORD                       TimeDateStamp;          // date/time stamp from pe header
 | |||
|  |     DWORD                       CheckSum;               // checksum from the pe header
 | |||
|  |     DWORD                       NumSyms;                // number of symbols in the symbol table
 | |||
|  |     SYM_TYPE                    SymType;                // type of symbols loaded
 | |||
|  |     CHAR                        ModuleName[32];         // module name
 | |||
|  |     CHAR                        ImageName[256];         // image name
 | |||
|  |     CHAR                        LoadedImageName[256];   // symbol file name
 | |||
|  | } IMAGEHLP_MODULE64, *PIMAGEHLP_MODULE64; | |||
|  | typedef struct _IMAGEHLP_SYMBOL64 { | |||
|  |     DWORD                       SizeOfStruct;           // set to sizeof(IMAGEHLP_SYMBOL64)
 | |||
|  |     DWORD64                     Address;                // virtual address including dll base address
 | |||
|  |     DWORD                       Size;                   // estimated size of symbol, can be zero
 | |||
|  |     DWORD                       Flags;                  // info about the symbols, see the SYMF defines
 | |||
|  |     DWORD                       MaxNameLength;          // maximum size of symbol name in 'Name'
 | |||
|  |     CHAR                        Name[1];                // symbol name (null terminated string)
 | |||
|  | } IMAGEHLP_SYMBOL64, *PIMAGEHLP_SYMBOL64; | |||
|  | typedef enum { | |||
|  |     AddrMode1616, | |||
|  |     AddrMode1632, | |||
|  |     AddrModeReal, | |||
|  |     AddrModeFlat | |||
|  | } ADDRESS_MODE; | |||
|  | typedef struct _tagADDRESS64 { | |||
|  |     DWORD64       Offset; | |||
|  |     WORD          Segment; | |||
|  |     ADDRESS_MODE  Mode; | |||
|  | } ADDRESS64, *LPADDRESS64; | |||
|  | typedef struct _KDHELP64 { | |||
|  |     DWORD64   Thread; | |||
|  |     DWORD   ThCallbackStack; | |||
|  |     DWORD   ThCallbackBStore; | |||
|  |     DWORD   NextCallback; | |||
|  |     DWORD   FramePointer; | |||
|  |     DWORD64   KiCallUserMode; | |||
|  |     DWORD64   KeUserCallbackDispatcher; | |||
|  |     DWORD64   SystemRangeStart; | |||
|  |     DWORD64  Reserved[8]; | |||
|  | } KDHELP64, *PKDHELP64; | |||
|  | typedef struct _tagSTACKFRAME64 { | |||
|  |     ADDRESS64   AddrPC;               // program counter
 | |||
|  |     ADDRESS64   AddrReturn;           // return address
 | |||
|  |     ADDRESS64   AddrFrame;            // frame pointer
 | |||
|  |     ADDRESS64   AddrStack;            // stack pointer
 | |||
|  |     ADDRESS64   AddrBStore;           // backing store pointer
 | |||
|  |     PVOID       FuncTableEntry;       // pointer to pdata/fpo or NULL
 | |||
|  |     DWORD64     Params[4];            // possible arguments to the function
 | |||
|  |     BOOL        Far;                  // WOW far call
 | |||
|  |     BOOL        Virtual;              // is this a virtual frame?
 | |||
|  |     DWORD64     Reserved[3]; | |||
|  |     KDHELP64    KdHelp; | |||
|  | } STACKFRAME64, *LPSTACKFRAME64; | |||
|  | typedef | |||
|  | BOOL | |||
|  | (__stdcall *PREAD_PROCESS_MEMORY_ROUTINE64)( | |||
|  |     HANDLE      hProcess, | |||
|  |     DWORD64     qwBaseAddress, | |||
|  |     PVOID       lpBuffer, | |||
|  |     DWORD       nSize, | |||
|  |     LPDWORD     lpNumberOfBytesRead | |||
|  |     ); | |||
|  | typedef | |||
|  | PVOID | |||
|  | (__stdcall *PFUNCTION_TABLE_ACCESS_ROUTINE64)( | |||
|  |     HANDLE  hProcess, | |||
|  |     DWORD64 AddrBase | |||
|  |     ); | |||
|  | typedef | |||
|  | DWORD64 | |||
|  | (__stdcall *PGET_MODULE_BASE_ROUTINE64)( | |||
|  |     HANDLE  hProcess, | |||
|  |     DWORD64 Address | |||
|  |     ); | |||
|  | typedef | |||
|  | DWORD64 | |||
|  | (__stdcall *PTRANSLATE_ADDRESS_ROUTINE64)( | |||
|  |     HANDLE    hProcess, | |||
|  |     HANDLE    hThread, | |||
|  |     LPADDRESS64 lpaddr | |||
|  |     ); | |||
|  | #define SYMOPT_CASE_INSENSITIVE         0x00000001
 | |||
|  | #define SYMOPT_UNDNAME                  0x00000002
 | |||
|  | #define SYMOPT_DEFERRED_LOADS           0x00000004
 | |||
|  | #define SYMOPT_NO_CPP                   0x00000008
 | |||
|  | #define SYMOPT_LOAD_LINES               0x00000010
 | |||
|  | #define SYMOPT_OMAP_FIND_NEAREST        0x00000020
 | |||
|  | #define SYMOPT_LOAD_ANYTHING            0x00000040
 | |||
|  | #define SYMOPT_IGNORE_CVREC             0x00000080
 | |||
|  | #define SYMOPT_NO_UNQUALIFIED_LOADS     0x00000100
 | |||
|  | #define SYMOPT_FAIL_CRITICAL_ERRORS     0x00000200
 | |||
|  | #define SYMOPT_EXACT_SYMBOLS            0x00000400
 | |||
|  | #define SYMOPT_ALLOW_ABSOLUTE_SYMBOLS   0x00000800
 | |||
|  | #define SYMOPT_IGNORE_NT_SYMPATH        0x00001000
 | |||
|  | #define SYMOPT_INCLUDE_32BIT_MODULES    0x00002000
 | |||
|  | #define SYMOPT_PUBLICS_ONLY             0x00004000
 | |||
|  | #define SYMOPT_NO_PUBLICS               0x00008000
 | |||
|  | #define SYMOPT_AUTO_PUBLICS             0x00010000
 | |||
|  | #define SYMOPT_NO_IMAGE_SEARCH          0x00020000
 | |||
|  | #define SYMOPT_SECURE                   0x00040000
 | |||
|  | #define SYMOPT_DEBUG                    0x80000000
 | |||
|  | #define UNDNAME_COMPLETE                 (0x0000)  // Enable full undecoration
 | |||
|  | #define UNDNAME_NAME_ONLY                (0x1000)  // Crack only the name for primary declaration;
 | |||
|  | #endif  // _MSC_VER < 1300
 | |||
|  | 
 | |||
|  | // Some missing defines (for VC5/6):
 | |||
|  | #ifndef INVALID_FILE_ATTRIBUTES
 | |||
|  | #define INVALID_FILE_ATTRIBUTES ((DWORD)-1)
 | |||
|  | #endif  
 | |||
|  | 
 | |||
|  | 
 | |||
|  | // secure-CRT_functions are only available starting with VC8
 | |||
|  | #if _MSC_VER < 1400
 | |||
|  | #define strcpy_s strcpy
 | |||
|  | #define strcat_s(dst, len, src) strcat(dst, src)
 | |||
|  | #define _snprintf_s _snprintf
 | |||
|  | #define _tcscat_s _tcscat
 | |||
|  | #endif
 | |||
|  | 
 | |||
|  | // Normally it should be enough to use 'CONTEXT_FULL' (better would be 'CONTEXT_ALL')
 | |||
|  | #define USED_CONTEXT_FLAGS CONTEXT_FULL
 | |||
|  | 
 | |||
|  | 
 | |||
|  | class StackWalkerInternal | |||
|  | { | |||
|  | public: | |||
|  |   StackWalkerInternal(StackWalker *parent, HANDLE hProcess) | |||
|  |   { | |||
|  |     m_parent = parent; | |||
|  |     m_hDbhHelp = NULL; | |||
|  |     pSC = NULL; | |||
|  |     m_hProcess = hProcess; | |||
|  |     m_szSymPath = NULL; | |||
|  |     pSFTA = NULL; | |||
|  |     pSGLFA = NULL; | |||
|  |     pSGMB = NULL; | |||
|  |     pSGMI = NULL; | |||
|  |     pSGO = NULL; | |||
|  |     pSGSFA = NULL; | |||
|  |     pSI = NULL; | |||
|  |     pSLM = NULL; | |||
|  |     pSSO = NULL; | |||
|  |     pSW = NULL; | |||
|  |     pUDSN = NULL; | |||
|  |     pSGSP = NULL; | |||
|  |   } | |||
|  |   ~StackWalkerInternal() | |||
|  |   { | |||
|  |     if (pSC != NULL) | |||
|  |       pSC(m_hProcess);  // SymCleanup
 | |||
|  |     if (m_hDbhHelp != NULL) | |||
|  |       FreeLibrary(m_hDbhHelp); | |||
|  |     m_hDbhHelp = NULL; | |||
|  |     m_parent = NULL; | |||
|  |     if(m_szSymPath != NULL) | |||
|  |       free(m_szSymPath); | |||
|  |     m_szSymPath = NULL; | |||
|  |   } | |||
|  |   BOOL Init(LPCSTR szSymPath) | |||
|  |   { | |||
|  |     if (m_parent == NULL) | |||
|  |       return FALSE; | |||
|  |     // Dynamically load the Entry-Points for dbghelp.dll:
 | |||
|  |     // First try to load the newsest one from
 | |||
|  |     TCHAR szTemp[4096]; | |||
|  |     // But before wqe do this, we first check if the ".local" file exists
 | |||
|  |     if (GetModuleFileName(NULL, szTemp, 4096) > 0) | |||
|  |     { | |||
|  |       _tcscat_s(szTemp, _T(".local")); | |||
|  |       if (GetFileAttributes(szTemp) == INVALID_FILE_ATTRIBUTES) | |||
|  |       { | |||
|  |         // ".local" file does not exist, so we can try to load the dbghelp.dll from the "Debugging Tools for Windows"
 | |||
|  |         if (GetEnvironmentVariable(_T("ProgramFiles"), szTemp, 4096) > 0) | |||
|  |         { | |||
|  |           _tcscat_s(szTemp, _T("\\Debugging Tools for Windows\\dbghelp.dll")); | |||
|  |           // now check if the file exists:
 | |||
|  |           if (GetFileAttributes(szTemp) != INVALID_FILE_ATTRIBUTES) | |||
|  |           { | |||
|  |             m_hDbhHelp = LoadLibrary(szTemp); | |||
|  |           } | |||
|  |         } | |||
|  |           // Still not found? Then try to load the 64-Bit version:
 | |||
|  |         if ( (m_hDbhHelp == NULL) && (GetEnvironmentVariable(_T("ProgramFiles"), szTemp, 4096) > 0) ) | |||
|  |         { | |||
|  |           _tcscat_s(szTemp, _T("\\Debugging Tools for Windows 64-Bit\\dbghelp.dll")); | |||
|  |           if (GetFileAttributes(szTemp) != INVALID_FILE_ATTRIBUTES) | |||
|  |           { | |||
|  |             m_hDbhHelp = LoadLibrary(szTemp); | |||
|  |           } | |||
|  |         } | |||
|  |       } | |||
|  |     } | |||
|  |     if (m_hDbhHelp == NULL)  // if not already loaded, try to load a default-one
 | |||
|  |       m_hDbhHelp = LoadLibrary( _T("dbghelp.dll") ); | |||
|  |     if (m_hDbhHelp == NULL) | |||
|  |       return FALSE; | |||
|  |     pSI = (tSI) GetProcAddress(m_hDbhHelp, "SymInitialize" ); | |||
|  |     pSC = (tSC) GetProcAddress(m_hDbhHelp, "SymCleanup" ); | |||
|  | 
 | |||
|  |     pSW = (tSW) GetProcAddress(m_hDbhHelp, "StackWalk64" ); | |||
|  |     pSGO = (tSGO) GetProcAddress(m_hDbhHelp, "SymGetOptions" ); | |||
|  |     pSSO = (tSSO) GetProcAddress(m_hDbhHelp, "SymSetOptions" ); | |||
|  | 
 | |||
|  |     pSFTA = (tSFTA) GetProcAddress(m_hDbhHelp, "SymFunctionTableAccess64" ); | |||
|  |     pSGLFA = (tSGLFA) GetProcAddress(m_hDbhHelp, "SymGetLineFromAddr64" ); | |||
|  |     pSGMB = (tSGMB) GetProcAddress(m_hDbhHelp, "SymGetModuleBase64" ); | |||
|  |     pSGMI = (tSGMI) GetProcAddress(m_hDbhHelp, "SymGetModuleInfo64" ); | |||
|  |     //pSGMI_V3 = (tSGMI_V3) GetProcAddress(m_hDbhHelp, "SymGetModuleInfo64" );
 | |||
|  |     pSGSFA = (tSGSFA) GetProcAddress(m_hDbhHelp, "SymGetSymFromAddr64" ); | |||
|  |     pUDSN = (tUDSN) GetProcAddress(m_hDbhHelp, "UnDecorateSymbolName" ); | |||
|  |     pSLM = (tSLM) GetProcAddress(m_hDbhHelp, "SymLoadModule64" ); | |||
|  |     pSGSP =(tSGSP) GetProcAddress(m_hDbhHelp, "SymGetSearchPath" ); | |||
|  | 
 | |||
|  |     if ( pSC == NULL || pSFTA == NULL || pSGMB == NULL || pSGMI == NULL || | |||
|  |       pSGO == NULL || pSGSFA == NULL || pSI == NULL || pSSO == NULL || | |||
|  |       pSW == NULL || pUDSN == NULL || pSLM == NULL ) | |||
|  |     { | |||
|  |       FreeLibrary(m_hDbhHelp); | |||
|  |       m_hDbhHelp = NULL; | |||
|  |       pSC = NULL; | |||
|  |       return FALSE; | |||
|  |     } | |||
|  | 
 | |||
|  |     // SymInitialize
 | |||
|  |     if (szSymPath != NULL) | |||
|  |       m_szSymPath = _strdup(szSymPath); | |||
|  |     if (this->pSI(m_hProcess, m_szSymPath, FALSE) == FALSE) | |||
|  |       this->m_parent->OnDbgHelpErr("SymInitialize", GetLastError(), 0); | |||
|  |        | |||
|  |     DWORD symOptions = this->pSGO();  // SymGetOptions
 | |||
|  |     symOptions |= SYMOPT_LOAD_LINES; | |||
|  |     symOptions |= SYMOPT_FAIL_CRITICAL_ERRORS; | |||
|  |     //symOptions |= SYMOPT_NO_PROMPTS;
 | |||
|  |     // SymSetOptions
 | |||
|  |     symOptions = this->pSSO(symOptions); | |||
|  | 
 | |||
|  |     /*char buf[StackWalker::STACKWALK_MAX_NAMELEN] = {0};
 | |||
|  |     if (this->pSGSP != NULL) | |||
|  |     { | |||
|  |       if (this->pSGSP(m_hProcess, buf, StackWalker::STACKWALK_MAX_NAMELEN) == FALSE) | |||
|  |         this->m_parent->OnDbgHelpErr("SymGetSearchPath", GetLastError(), 0); | |||
|  |     } | |||
|  |     char szUserName[1024] = {0}; | |||
|  |     DWORD dwSize = 1024; | |||
|  |     GetUserNameA(szUserName, &dwSize); | |||
|  |     this->m_parent->OnSymInit(buf, symOptions, szUserName);*/ | |||
|  | 
 | |||
|  |     return TRUE; | |||
|  |   } | |||
|  | 
 | |||
|  |   StackWalker *m_parent; | |||
|  | 
 | |||
|  |   HMODULE m_hDbhHelp; | |||
|  |   HANDLE m_hProcess; | |||
|  |   LPSTR m_szSymPath; | |||
|  | 
 | |||
|  | /*typedef struct IMAGEHLP_MODULE64_V3 {
 | |||
|  |     DWORD    SizeOfStruct;           // set to sizeof(IMAGEHLP_MODULE64)
 | |||
|  |     DWORD64  BaseOfImage;            // base load address of module
 | |||
|  |     DWORD    ImageSize;              // virtual size of the loaded module
 | |||
|  |     DWORD    TimeDateStamp;          // date/time stamp from pe header
 | |||
|  |     DWORD    CheckSum;               // checksum from the pe header
 | |||
|  |     DWORD    NumSyms;                // number of symbols in the symbol table
 | |||
|  |     SYM_TYPE SymType;                // type of symbols loaded
 | |||
|  |     CHAR     ModuleName[32];         // module name
 | |||
|  |     CHAR     ImageName[256];         // image name
 | |||
|  |     // new elements: 07-Jun-2002
 | |||
|  |     CHAR     LoadedImageName[256];   // symbol file name
 | |||
|  |     CHAR     LoadedPdbName[256];     // pdb file name
 | |||
|  |     DWORD    CVSig;                  // Signature of the CV record in the debug directories
 | |||
|  |     CHAR         CVData[MAX_PATH * 3];   // Contents of the CV record
 | |||
|  |     DWORD    PdbSig;                 // Signature of PDB
 | |||
|  |     GUID     PdbSig70;               // Signature of PDB (VC 7 and up)
 | |||
|  |     DWORD    PdbAge;                 // DBI age of pdb
 | |||
|  |     BOOL     PdbUnmatched;           // loaded an unmatched pdb
 | |||
|  |     BOOL     DbgUnmatched;           // loaded an unmatched dbg
 | |||
|  |     BOOL     LineNumbers;            // we have line number information
 | |||
|  |     BOOL     GlobalSymbols;          // we have internal symbol information
 | |||
|  |     BOOL     TypeInfo;               // we have type information
 | |||
|  |     // new elements: 17-Dec-2003
 | |||
|  |     BOOL     SourceIndexed;          // pdb supports source server
 | |||
|  |     BOOL     Publics;                // contains public symbols
 | |||
|  | }; | |||
|  | */ | |||
|  | struct IMAGEHLP_MODULE64_V2 { | |||
|  |     DWORD    SizeOfStruct;           // set to sizeof(IMAGEHLP_MODULE64)
 | |||
|  |     DWORD64  BaseOfImage;            // base load address of module
 | |||
|  |     DWORD    ImageSize;              // virtual size of the loaded module
 | |||
|  |     DWORD    TimeDateStamp;          // date/time stamp from pe header
 | |||
|  |     DWORD    CheckSum;               // checksum from the pe header
 | |||
|  |     DWORD    NumSyms;                // number of symbols in the symbol table
 | |||
|  |     SYM_TYPE SymType;                // type of symbols loaded
 | |||
|  |     CHAR     ModuleName[32];         // module name
 | |||
|  |     CHAR     ImageName[256];         // image name
 | |||
|  |     CHAR     LoadedImageName[256];   // symbol file name
 | |||
|  | }; | |||
|  | 
 | |||
|  | 
 | |||
|  |   // SymCleanup()
 | |||
|  |   typedef BOOL (__stdcall *tSC)( IN HANDLE hProcess ); | |||
|  |   tSC pSC; | |||
|  | 
 | |||
|  |   // SymFunctionTableAccess64()
 | |||
|  |   typedef PVOID (__stdcall *tSFTA)( HANDLE hProcess, DWORD64 AddrBase ); | |||
|  |   tSFTA pSFTA; | |||
|  | 
 | |||
|  |   // SymGetLineFromAddr64()
 | |||
|  |   typedef BOOL (__stdcall *tSGLFA)( IN HANDLE hProcess, IN DWORD64 dwAddr, | |||
|  |     OUT PDWORD pdwDisplacement, OUT PIMAGEHLP_LINE64 Line ); | |||
|  |   tSGLFA pSGLFA; | |||
|  | 
 | |||
|  |   // SymGetModuleBase64()
 | |||
|  |   typedef DWORD64 (__stdcall *tSGMB)( IN HANDLE hProcess, IN DWORD64 dwAddr ); | |||
|  |   tSGMB pSGMB; | |||
|  | 
 | |||
|  |   // SymGetModuleInfo64()
 | |||
|  |   typedef BOOL (__stdcall *tSGMI)( IN HANDLE hProcess, IN DWORD64 dwAddr, OUT IMAGEHLP_MODULE64_V2 *ModuleInfo ); | |||
|  |   tSGMI pSGMI; | |||
|  | 
 | |||
|  | //  // SymGetModuleInfo64()
 | |||
|  | //  typedef BOOL (__stdcall *tSGMI_V3)( IN HANDLE hProcess, IN DWORD64 dwAddr, OUT IMAGEHLP_MODULE64_V3 *ModuleInfo );
 | |||
|  | //  tSGMI_V3 pSGMI_V3;
 | |||
|  | 
 | |||
|  |   // SymGetOptions()
 | |||
|  |   typedef DWORD (__stdcall *tSGO)( VOID ); | |||
|  |   tSGO pSGO; | |||
|  | 
 | |||
|  |   // SymGetSymFromAddr64()
 | |||
|  |   typedef BOOL (__stdcall *tSGSFA)( IN HANDLE hProcess, IN DWORD64 dwAddr, | |||
|  |     OUT PDWORD64 pdwDisplacement, OUT PIMAGEHLP_SYMBOL64 Symbol ); | |||
|  |   tSGSFA pSGSFA; | |||
|  | 
 | |||
|  |   // SymInitialize()
 | |||
|  |   typedef BOOL (__stdcall *tSI)( IN HANDLE hProcess, IN PSTR UserSearchPath, IN BOOL fInvadeProcess ); | |||
|  |   tSI pSI; | |||
|  | 
 | |||
|  |   // SymLoadModule64()
 | |||
|  |   typedef DWORD64 (__stdcall *tSLM)( IN HANDLE hProcess, IN HANDLE hFile, | |||
|  |     IN PSTR ImageName, IN PSTR ModuleName, IN DWORD64 BaseOfDll, IN DWORD SizeOfDll ); | |||
|  |   tSLM pSLM; | |||
|  | 
 | |||
|  |   // SymSetOptions()
 | |||
|  |   typedef DWORD (__stdcall *tSSO)( IN DWORD SymOptions ); | |||
|  |   tSSO pSSO; | |||
|  | 
 | |||
|  |   // StackWalk64()
 | |||
|  |   typedef BOOL (__stdcall *tSW)(  | |||
|  |     DWORD MachineType,  | |||
|  |     HANDLE hProcess, | |||
|  |     HANDLE hThread,  | |||
|  |     LPSTACKFRAME64 StackFrame,  | |||
|  |     PVOID ContextRecord, | |||
|  |     PREAD_PROCESS_MEMORY_ROUTINE64 ReadMemoryRoutine, | |||
|  |     PFUNCTION_TABLE_ACCESS_ROUTINE64 FunctionTableAccessRoutine, | |||
|  |     PGET_MODULE_BASE_ROUTINE64 GetModuleBaseRoutine, | |||
|  |     PTRANSLATE_ADDRESS_ROUTINE64 TranslateAddress ); | |||
|  |   tSW pSW; | |||
|  | 
 | |||
|  |   // UnDecorateSymbolName()
 | |||
|  |   typedef DWORD (__stdcall WINAPI *tUDSN)( PCSTR DecoratedName, PSTR UnDecoratedName, | |||
|  |     DWORD UndecoratedLength, DWORD Flags ); | |||
|  |   tUDSN pUDSN; | |||
|  | 
 | |||
|  |   typedef BOOL (__stdcall WINAPI *tSGSP)(HANDLE hProcess, PSTR SearchPath, DWORD SearchPathLength); | |||
|  |   tSGSP pSGSP; | |||
|  | 
 | |||
|  | 
 | |||
|  | private: | |||
|  |   // **************************************** ToolHelp32 ************************
 | |||
|  |   #define MAX_MODULE_NAME32 255
 | |||
|  |   #define TH32CS_SNAPMODULE   0x00000008
 | |||
|  |   #pragma pack( push, 8 )
 | |||
|  |   typedef struct tagMODULEENTRY32 | |||
|  |   { | |||
|  |       DWORD   dwSize; | |||
|  |       DWORD   th32ModuleID;       // This module
 | |||
|  |       DWORD   th32ProcessID;      // owning process
 | |||
|  |       DWORD   GlblcntUsage;       // Global usage count on the module
 | |||
|  |       DWORD   ProccntUsage;       // Module usage count in th32ProcessID's context
 | |||
|  |       BYTE  * modBaseAddr;        // Base address of module in th32ProcessID's context
 | |||
|  |       DWORD   modBaseSize;        // Size in bytes of module starting at modBaseAddr
 | |||
|  |       HMODULE hModule;            // The hModule of this module in th32ProcessID's context
 | |||
|  |       char    szModule[MAX_MODULE_NAME32 + 1]; | |||
|  |       char    szExePath[MAX_PATH]; | |||
|  |   } MODULEENTRY32; | |||
|  |   typedef MODULEENTRY32 *  PMODULEENTRY32; | |||
|  |   typedef MODULEENTRY32 *  LPMODULEENTRY32; | |||
|  |   #pragma pack( pop )
 | |||
|  | 
 | |||
|  |   BOOL GetModuleListTH32(HANDLE hProcess, DWORD pid) | |||
|  |   { | |||
|  |     // CreateToolhelp32Snapshot()
 | |||
|  |     typedef HANDLE (__stdcall *tCT32S)(DWORD dwFlags, DWORD th32ProcessID); | |||
|  |     // Module32First()
 | |||
|  |     typedef BOOL (__stdcall *tM32F)(HANDLE hSnapshot, LPMODULEENTRY32 lpme); | |||
|  |     // Module32Next()
 | |||
|  |     typedef BOOL (__stdcall *tM32N)(HANDLE hSnapshot, LPMODULEENTRY32 lpme); | |||
|  | 
 | |||
|  |     // try both dlls...
 | |||
|  |     const TCHAR *dllname[] = { _T("kernel32.dll"), _T("tlhelp32.dll") }; | |||
|  |     HINSTANCE hToolhelp = NULL; | |||
|  |     tCT32S pCT32S = NULL; | |||
|  |     tM32F pM32F = NULL; | |||
|  |     tM32N pM32N = NULL; | |||
|  | 
 | |||
|  |     HANDLE hSnap; | |||
|  |     MODULEENTRY32 me; | |||
|  |     me.dwSize = sizeof(me); | |||
|  |     BOOL keepGoing; | |||
|  |     size_t i; | |||
|  | 
 | |||
|  |     for (i = 0; i<(sizeof(dllname) / sizeof(dllname[0])); i++ ) | |||
|  |     { | |||
|  |       hToolhelp = LoadLibrary( dllname[i] ); | |||
|  |       if (hToolhelp == NULL) | |||
|  |         continue; | |||
|  |       pCT32S = (tCT32S) GetProcAddress(hToolhelp, "CreateToolhelp32Snapshot"); | |||
|  |       pM32F = (tM32F) GetProcAddress(hToolhelp, "Module32First"); | |||
|  |       pM32N = (tM32N) GetProcAddress(hToolhelp, "Module32Next"); | |||
|  |       if ( (pCT32S != NULL) && (pM32F != NULL) && (pM32N != NULL) ) | |||
|  |         break; // found the functions!
 | |||
|  |       FreeLibrary(hToolhelp); | |||
|  |       hToolhelp = NULL; | |||
|  |     } | |||
|  | 
 | |||
|  |     if (hToolhelp == NULL) | |||
|  |       return FALSE; | |||
|  | 
 | |||
|  |     hSnap = pCT32S( TH32CS_SNAPMODULE, pid ); | |||
|  |     if (hSnap == (HANDLE) -1) | |||
|  |       return FALSE; | |||
|  | 
 | |||
|  |     keepGoing = !!pM32F( hSnap, &me ); | |||
|  |     int cnt = 0; | |||
|  |     while (keepGoing) | |||
|  |     { | |||
|  |       this->LoadModule(hProcess, me.szExePath, me.szModule, (DWORD64) me.modBaseAddr, me.modBaseSize); | |||
|  |       cnt++; | |||
|  |       keepGoing = !!pM32N( hSnap, &me ); | |||
|  |     } | |||
|  |     CloseHandle(hSnap); | |||
|  |     FreeLibrary(hToolhelp); | |||
|  |     if (cnt <= 0) | |||
|  |       return FALSE; | |||
|  |     return TRUE; | |||
|  |   }  // GetModuleListTH32
 | |||
|  | 
 | |||
|  |   // **************************************** PSAPI ************************
 | |||
|  |   typedef struct _MODULEINFO { | |||
|  |       LPVOID lpBaseOfDll; | |||
|  |       DWORD SizeOfImage; | |||
|  |       LPVOID EntryPoint; | |||
|  |   } MODULEINFO, *LPMODULEINFO; | |||
|  | 
 | |||
|  |   BOOL GetModuleListPSAPI(HANDLE hProcess) | |||
|  |   { | |||
|  |     // EnumProcessModules()
 | |||
|  |     typedef BOOL (__stdcall *tEPM)(HANDLE hProcess, HMODULE *lphModule, DWORD cb, LPDWORD lpcbNeeded ); | |||
|  |     // GetModuleFileNameEx()
 | |||
|  |     typedef DWORD (__stdcall *tGMFNE)(HANDLE hProcess, HMODULE hModule, LPSTR lpFilename, DWORD nSize ); | |||
|  |     // GetModuleBaseName()
 | |||
|  |     typedef DWORD (__stdcall *tGMBN)(HANDLE hProcess, HMODULE hModule, LPSTR lpFilename, DWORD nSize ); | |||
|  |     // GetModuleInformation()
 | |||
|  |     typedef BOOL (__stdcall *tGMI)(HANDLE hProcess, HMODULE hModule, LPMODULEINFO pmi, DWORD nSize ); | |||
|  | 
 | |||
|  |     HINSTANCE hPsapi; | |||
|  |     tEPM pEPM; | |||
|  |     tGMFNE pGMFNE; | |||
|  |     tGMBN pGMBN; | |||
|  |     tGMI pGMI; | |||
|  | 
 | |||
|  |     DWORD i; | |||
|  |     //ModuleEntry e;
 | |||
|  |     DWORD cbNeeded; | |||
|  |     MODULEINFO mi; | |||
|  |     HMODULE *hMods = 0; | |||
|  |     char *tt = NULL; | |||
|  |     char *tt2 = NULL; | |||
|  |     const SIZE_T TTBUFLEN = 8096; | |||
|  |     int cnt = 0; | |||
|  | 
 | |||
|  |     hPsapi = LoadLibrary( _T("psapi.dll") ); | |||
|  |     if (hPsapi == NULL) | |||
|  |       return FALSE; | |||
|  | 
 | |||
|  |     pEPM = (tEPM) GetProcAddress( hPsapi, "EnumProcessModules" ); | |||
|  |     pGMFNE = (tGMFNE) GetProcAddress( hPsapi, "GetModuleFileNameExA" ); | |||
|  |     pGMBN = (tGMFNE) GetProcAddress( hPsapi, "GetModuleBaseNameA" ); | |||
|  |     pGMI = (tGMI) GetProcAddress( hPsapi, "GetModuleInformation" ); | |||
|  |     if ( (pEPM == NULL) || (pGMFNE == NULL) || (pGMBN == NULL) || (pGMI == NULL) ) | |||
|  |     { | |||
|  |       // we couldn<64>t find all functions
 | |||
|  |       FreeLibrary(hPsapi); | |||
|  |       return FALSE; | |||
|  |     } | |||
|  | 
 | |||
|  |     hMods = (HMODULE*) malloc(sizeof(HMODULE) * (TTBUFLEN / sizeof HMODULE)); | |||
|  |     tt = (char*) malloc(sizeof(char) * TTBUFLEN); | |||
|  |     tt2 = (char*) malloc(sizeof(char) * TTBUFLEN); | |||
|  |     if ( (hMods == NULL) || (tt == NULL) || (tt2 == NULL) ) | |||
|  |       goto cleanup; | |||
|  | 
 | |||
|  |     if ( ! pEPM( hProcess, hMods, TTBUFLEN, &cbNeeded ) ) | |||
|  |     { | |||
|  |       //_ftprintf(fLogFile, _T("%lu: EPM failed, GetLastError = %lu\n"), g_dwShowCount, gle );
 | |||
|  |       goto cleanup; | |||
|  |     } | |||
|  | 
 | |||
|  |     if ( cbNeeded > TTBUFLEN ) | |||
|  |     { | |||
|  |       //_ftprintf(fLogFile, _T("%lu: More than %lu module handles. Huh?\n"), g_dwShowCount, lenof( hMods ) );
 | |||
|  |       goto cleanup; | |||
|  |     } | |||
|  | 
 | |||
|  |     for ( i = 0; i < cbNeeded / sizeof hMods[0]; i++ ) | |||
|  |     { | |||
|  |       // base address, size
 | |||
|  |       pGMI(hProcess, hMods[i], &mi, sizeof mi ); | |||
|  |       // image file name
 | |||
|  |       tt[0] = 0; | |||
|  |       pGMFNE(hProcess, hMods[i], tt, TTBUFLEN ); | |||
|  |       // module name
 | |||
|  |       tt2[0] = 0; | |||
|  |       pGMBN(hProcess, hMods[i], tt2, TTBUFLEN ); | |||
|  | 
 | |||
|  |       DWORD dwRes = this->LoadModule(hProcess, tt, tt2, (DWORD64) mi.lpBaseOfDll, mi.SizeOfImage); | |||
|  |       if (dwRes != ERROR_SUCCESS) | |||
|  |         this->m_parent->OnDbgHelpErr("LoadModule", dwRes, 0); | |||
|  |       cnt++; | |||
|  |     } | |||
|  | 
 | |||
|  |   cleanup: | |||
|  |     if (hPsapi != NULL) FreeLibrary(hPsapi); | |||
|  |     if (tt2 != NULL) free(tt2); | |||
|  |     if (tt != NULL) free(tt); | |||
|  |     if (hMods != NULL) free(hMods); | |||
|  | 
 | |||
|  |     return cnt != 0; | |||
|  |   }  // GetModuleListPSAPI
 | |||
|  | 
 | |||
|  |   DWORD LoadModule(HANDLE hProcess, LPCSTR img, LPCSTR mod, DWORD64 baseAddr, DWORD size) | |||
|  |   { | |||
|  |     CHAR *szImg = _strdup(img); | |||
|  |     CHAR *szMod = _strdup(mod); | |||
|  |     DWORD result = ERROR_SUCCESS; | |||
|  |     if ( (szImg == NULL) || (szMod == NULL) ) | |||
|  |       result = ERROR_NOT_ENOUGH_MEMORY; | |||
|  |     else | |||
|  |     { | |||
|  |       if (pSLM(hProcess, 0, szImg, szMod, baseAddr, size) == 0) | |||
|  |         result = GetLastError(); | |||
|  |     } | |||
|  |     ULONGLONG fileVersion = 0; | |||
|  |     if ( (m_parent != NULL) && (szImg != NULL) ) | |||
|  |     { | |||
|  |       // try to retrive the file-version:
 | |||
|  |       if ( (this->m_parent->m_options & StackWalker::RetrieveFileVersion) != 0) | |||
|  |       { | |||
|  |         VS_FIXEDFILEINFO *fInfo = NULL; | |||
|  |         DWORD dwHandle; | |||
|  |         DWORD dwSize = GetFileVersionInfoSizeA(szImg, &dwHandle); | |||
|  |         if (dwSize > 0) | |||
|  |         { | |||
|  |           LPVOID vData = malloc(dwSize); | |||
|  |           if (vData != NULL) | |||
|  |           { | |||
|  |             if (GetFileVersionInfoA(szImg, dwHandle, dwSize, vData) != 0) | |||
|  |             { | |||
|  |               UINT len; | |||
|  |               TCHAR szSubBlock[] = _T("\\"); | |||
|  |               if (VerQueryValue(vData, szSubBlock, (LPVOID*) &fInfo, &len) == 0) | |||
|  |                 fInfo = NULL; | |||
|  |               else | |||
|  |               { | |||
|  |                 fileVersion = ((ULONGLONG)fInfo->dwFileVersionLS) + ((ULONGLONG)fInfo->dwFileVersionMS << 32); | |||
|  |               } | |||
|  |             } | |||
|  |             free(vData); | |||
|  |           } | |||
|  |         } | |||
|  |       } | |||
|  | 
 | |||
|  |       // Retrive some additional-infos about the module
 | |||
|  |       IMAGEHLP_MODULE64_V2 Module; | |||
|  |       const char *szSymType = "-unknown-"; | |||
|  |       if (this->GetModuleInfo(hProcess, baseAddr, &Module) != FALSE) | |||
|  |       { | |||
|  |         switch(Module.SymType) | |||
|  |         { | |||
|  |           case SymNone: | |||
|  |             szSymType = "-nosymbols-"; | |||
|  |             break; | |||
|  |           case SymCoff: | |||
|  |             szSymType = "COFF"; | |||
|  |             break; | |||
|  |           case SymCv: | |||
|  |             szSymType = "CV"; | |||
|  |             break; | |||
|  |           case SymPdb: | |||
|  |             szSymType = "PDB"; | |||
|  |             break; | |||
|  |           case SymExport: | |||
|  |             szSymType = "-exported-"; | |||
|  |             break; | |||
|  |           case SymDeferred: | |||
|  |             szSymType = "-deferred-"; | |||
|  |             break; | |||
|  |           case SymSym: | |||
|  |             szSymType = "SYM"; | |||
|  |             break; | |||
|  |           case 8: //SymVirtual:
 | |||
|  |             szSymType = "Virtual"; | |||
|  |             break; | |||
|  |           case 9: // SymDia:
 | |||
|  |             szSymType = "DIA"; | |||
|  |             break; | |||
|  |         } | |||
|  |       } | |||
|  |       this->m_parent->OnLoadModule(img, mod, baseAddr, size, result, szSymType, Module.LoadedImageName, fileVersion); | |||
|  |     } | |||
|  |     if (szImg != NULL) free(szImg); | |||
|  |     if (szMod != NULL) free(szMod); | |||
|  |     return result; | |||
|  |   } | |||
|  | public: | |||
|  |   BOOL LoadModules(HANDLE hProcess, DWORD dwProcessId) | |||
|  |   { | |||
|  |     // first try toolhelp32
 | |||
|  |     if (GetModuleListTH32(hProcess, dwProcessId)) | |||
|  |       return true; | |||
|  |     // then try psapi
 | |||
|  |     return GetModuleListPSAPI(hProcess); | |||
|  |   } | |||
|  | 
 | |||
|  | 
 | |||
|  |   BOOL GetModuleInfo(HANDLE hProcess, DWORD64 baseAddr, IMAGEHLP_MODULE64_V2 *pModuleInfo) | |||
|  |   { | |||
|  |     if(this->pSGMI == NULL) | |||
|  |     { | |||
|  |       SetLastError(ERROR_DLL_INIT_FAILED); | |||
|  |       return FALSE; | |||
|  |     } | |||
|  |     // First try to use the larger ModuleInfo-Structure
 | |||
|  | //    memset(pModuleInfo, 0, sizeof(IMAGEHLP_MODULE64_V3));
 | |||
|  | //    pModuleInfo->SizeOfStruct = sizeof(IMAGEHLP_MODULE64_V3);
 | |||
|  | //    if (this->pSGMI_V3 != NULL)
 | |||
|  | //    {
 | |||
|  | //      if (this->pSGMI_V3(hProcess, baseAddr, pModuleInfo) != FALSE)
 | |||
|  | //        return TRUE;
 | |||
|  | //      // check if the parameter was wrong (size is bad...)
 | |||
|  | //      if (GetLastError() != ERROR_INVALID_PARAMETER)
 | |||
|  | //        return FALSE;
 | |||
|  | //    }
 | |||
|  |     // could not retrive the bigger structure, try with the smaller one (as defined in VC7.1)...
 | |||
|  |     pModuleInfo->SizeOfStruct = sizeof(IMAGEHLP_MODULE64_V2); | |||
|  |     void *pData = malloc(4096); // reserve enough memory, so the bug in v6.3.5.1 does not lead to memory-overwrites...
 | |||
|  |     if (pData == NULL) | |||
|  |     { | |||
|  |       SetLastError(ERROR_NOT_ENOUGH_MEMORY); | |||
|  |       return FALSE; | |||
|  |     } | |||
|  |     memcpy(pData, pModuleInfo, sizeof(IMAGEHLP_MODULE64_V2)); | |||
|  |     if (this->pSGMI(hProcess, baseAddr, (IMAGEHLP_MODULE64_V2*) pData) != FALSE) | |||
|  |     { | |||
|  |       // only copy as much memory as is reserved...
 | |||
|  |       memcpy(pModuleInfo, pData, sizeof(IMAGEHLP_MODULE64_V2)); | |||
|  |       pModuleInfo->SizeOfStruct = sizeof(IMAGEHLP_MODULE64_V2); | |||
|  |       free(pData); | |||
|  |       return TRUE; | |||
|  |     } | |||
|  |     free(pData); | |||
|  |     SetLastError(ERROR_DLL_INIT_FAILED); | |||
|  |     return FALSE; | |||
|  |   } | |||
|  | }; | |||
|  | 
 | |||
|  | // #############################################################
 | |||
|  | StackWalker::StackWalker(DWORD dwProcessId, HANDLE hProcess) | |||
|  | { | |||
|  |   this->m_options = OptionsAll; | |||
|  |   this->m_modulesLoaded = FALSE; | |||
|  |   this->m_hProcess = hProcess; | |||
|  |   this->m_sw = new StackWalkerInternal(this, this->m_hProcess); | |||
|  |   this->m_dwProcessId = dwProcessId; | |||
|  |   this->m_szSymPath = NULL; | |||
|  | } | |||
|  | StackWalker::StackWalker(int options, LPCSTR szSymPath, DWORD dwProcessId, HANDLE hProcess) | |||
|  | { | |||
|  |   this->m_options = options; | |||
|  |   this->m_modulesLoaded = FALSE; | |||
|  |   this->m_hProcess = hProcess; | |||
|  |   this->m_sw = new StackWalkerInternal(this, this->m_hProcess); | |||
|  |   this->m_dwProcessId = dwProcessId; | |||
|  |   if (szSymPath != NULL) | |||
|  |   { | |||
|  |     this->m_szSymPath = _strdup(szSymPath); | |||
|  |     this->m_options |= SymBuildPath; | |||
|  |   } | |||
|  |   else | |||
|  |     this->m_szSymPath = NULL; | |||
|  | } | |||
|  | 
 | |||
|  | StackWalker::~StackWalker() | |||
|  | { | |||
|  |   if (m_szSymPath != NULL) | |||
|  |     free(m_szSymPath); | |||
|  |   m_szSymPath = NULL; | |||
|  |   if (this->m_sw != NULL) | |||
|  |     delete this->m_sw; | |||
|  |   this->m_sw = NULL; | |||
|  | } | |||
|  | 
 | |||
|  | BOOL StackWalker::LoadModules() | |||
|  | { | |||
|  |   if (this->m_sw == NULL) | |||
|  |   { | |||
|  |     SetLastError(ERROR_DLL_INIT_FAILED); | |||
|  |     return FALSE; | |||
|  |   } | |||
|  |   if (m_modulesLoaded != FALSE) | |||
|  |     return TRUE; | |||
|  | 
 | |||
|  |   // Build the sym-path:
 | |||
|  |   char *szSymPath = NULL; | |||
|  |   if ( (this->m_options & SymBuildPath) != 0) | |||
|  |   { | |||
|  |     const size_t nSymPathLen = 4096; | |||
|  |     szSymPath = (char*) malloc(nSymPathLen); | |||
|  |     if (szSymPath == NULL) | |||
|  |     { | |||
|  |       SetLastError(ERROR_NOT_ENOUGH_MEMORY); | |||
|  |       return FALSE; | |||
|  |     } | |||
|  |     szSymPath[0] = 0; | |||
|  |     // Now first add the (optional) provided sympath:
 | |||
|  |     if (this->m_szSymPath != NULL) | |||
|  |     { | |||
|  |       strcat_s(szSymPath, nSymPathLen, this->m_szSymPath); | |||
|  |       strcat_s(szSymPath, nSymPathLen, ";"); | |||
|  |     } | |||
|  | 
 | |||
|  |     strcat_s(szSymPath, nSymPathLen, ".;"); | |||
|  | 
 | |||
|  |     const size_t nTempLen = 1024; | |||
|  |     char szTemp[nTempLen]; | |||
|  |     // Now add the current directory:
 | |||
|  |     if (GetCurrentDirectoryA(nTempLen, szTemp) > 0) | |||
|  |     { | |||
|  |       szTemp[nTempLen-1] = 0; | |||
|  |       strcat_s(szSymPath, nSymPathLen, szTemp); | |||
|  |       strcat_s(szSymPath, nSymPathLen, ";"); | |||
|  |     } | |||
|  | 
 | |||
|  |     // Now add the path for the main-module:
 | |||
|  |     if (GetModuleFileNameA(NULL, szTemp, nTempLen) > 0) | |||
|  |     { | |||
|  |       szTemp[nTempLen-1] = 0; | |||
|  |       for (char *p = (szTemp+strlen(szTemp)-1); p >= szTemp; --p) | |||
|  |       { | |||
|  |         // locate the rightmost path separator
 | |||
|  |         if ( (*p == '\\') || (*p == '/') || (*p == ':') ) | |||
|  |         { | |||
|  |           *p = 0; | |||
|  |           break; | |||
|  |         } | |||
|  |       }  // for (search for path separator...)
 | |||
|  |       if (strlen(szTemp) > 0) | |||
|  |       { | |||
|  |         strcat_s(szSymPath, nSymPathLen, szTemp); | |||
|  |         strcat_s(szSymPath, nSymPathLen, ";"); | |||
|  |       } | |||
|  |     } | |||
|  |     if (GetEnvironmentVariableA("_NT_SYMBOL_PATH", szTemp, nTempLen) > 0) | |||
|  |     { | |||
|  |       szTemp[nTempLen-1] = 0; | |||
|  |       strcat_s(szSymPath, nSymPathLen, szTemp); | |||
|  |       strcat_s(szSymPath, nSymPathLen, ";"); | |||
|  |     } | |||
|  |     if (GetEnvironmentVariableA("_NT_ALTERNATE_SYMBOL_PATH", szTemp, nTempLen) > 0) | |||
|  |     { | |||
|  |       szTemp[nTempLen-1] = 0; | |||
|  |       strcat_s(szSymPath, nSymPathLen, szTemp); | |||
|  |       strcat_s(szSymPath, nSymPathLen, ";"); | |||
|  |     } | |||
|  |     if (GetEnvironmentVariableA("SYSTEMROOT", szTemp, nTempLen) > 0) | |||
|  |     { | |||
|  |       szTemp[nTempLen-1] = 0; | |||
|  |       strcat_s(szSymPath, nSymPathLen, szTemp); | |||
|  |       strcat_s(szSymPath, nSymPathLen, ";"); | |||
|  |       // also add the "system32"-directory:
 | |||
|  |       strcat_s(szTemp, nTempLen, "\\system32"); | |||
|  |       strcat_s(szSymPath, nSymPathLen, szTemp); | |||
|  |       strcat_s(szSymPath, nSymPathLen, ";"); | |||
|  |     } | |||
|  | 
 | |||
|  |     if ( (this->m_options & SymBuildPath) != 0) | |||
|  |     { | |||
|  |       if (GetEnvironmentVariableA("SYSTEMDRIVE", szTemp, nTempLen) > 0) | |||
|  |       { | |||
|  |         szTemp[nTempLen-1] = 0; | |||
|  |         strcat_s(szSymPath, nSymPathLen, "SRV*"); | |||
|  |         strcat_s(szSymPath, nSymPathLen, szTemp); | |||
|  |         strcat_s(szSymPath, nSymPathLen, "\\websymbols"); | |||
|  |         strcat_s(szSymPath, nSymPathLen, "*http://msdl.microsoft.com/download/symbols;"); | |||
|  |       } | |||
|  |       else | |||
|  |         strcat_s(szSymPath, nSymPathLen, "SRV*c:\\websymbols*http://msdl.microsoft.com/download/symbols;"); | |||
|  |     } | |||
|  |   } | |||
|  | 
 | |||
|  |   // First Init the whole stuff...
 | |||
|  |   BOOL bRet = this->m_sw->Init(szSymPath); | |||
|  |   if (szSymPath != NULL) free(szSymPath); szSymPath = NULL; | |||
|  |   if (bRet == FALSE) | |||
|  |   { | |||
|  |     this->OnDbgHelpErr("Error while initializing dbghelp.dll", 0, 0); | |||
|  |     SetLastError(ERROR_DLL_INIT_FAILED); | |||
|  |     return FALSE; | |||
|  |   } | |||
|  | 
 | |||
|  |   bRet = this->m_sw->LoadModules(this->m_hProcess, this->m_dwProcessId); | |||
|  |   if (bRet != FALSE) | |||
|  |     m_modulesLoaded = TRUE; | |||
|  |   return bRet; | |||
|  | } | |||
|  | 
 | |||
|  | 
 | |||
|  | // The following is used to pass the "userData"-Pointer to the user-provided readMemoryFunction
 | |||
|  | // This has to be done due to a problem with the "hProcess"-parameter in x64...
 | |||
|  | // Because this class is in no case multi-threading-enabled (because of the limitations 
 | |||
|  | // of dbghelp.dll) it is "safe" to use a static-variable
 | |||
|  | static StackWalker::PReadProcessMemoryRoutine s_readMemoryFunction = NULL; | |||
|  | static LPVOID s_readMemoryFunction_UserData = NULL; | |||
|  | 
 | |||
|  | BOOL StackWalker::ShowCallstack(HANDLE hThread, const CONTEXT *context, PReadProcessMemoryRoutine readMemoryFunction, LPVOID pUserData) | |||
|  | { | |||
|  |   CONTEXT c;; | |||
|  |   CallstackEntry csEntry; | |||
|  |   IMAGEHLP_SYMBOL64 *pSym = NULL; | |||
|  |   StackWalkerInternal::IMAGEHLP_MODULE64_V2 Module; | |||
|  |   IMAGEHLP_LINE64 Line; | |||
|  |   int frameNum; | |||
|  | 
 | |||
|  |   if (m_modulesLoaded == FALSE) | |||
|  |     this->LoadModules();  // ignore the result...
 | |||
|  | 
 | |||
|  |   if (this->m_sw->m_hDbhHelp == NULL) | |||
|  |   { | |||
|  |     SetLastError(ERROR_DLL_INIT_FAILED); | |||
|  |     return FALSE; | |||
|  |   } | |||
|  | 
 | |||
|  |   s_readMemoryFunction = readMemoryFunction; | |||
|  |   s_readMemoryFunction_UserData = pUserData; | |||
|  | 
 | |||
|  |   if (context == NULL) | |||
|  |   { | |||
|  |     // If no context is provided, capture the context
 | |||
|  |     if (hThread == GetCurrentThread()) | |||
|  |     { | |||
|  |       GET_CURRENT_CONTEXT(c, USED_CONTEXT_FLAGS); | |||
|  |     } | |||
|  |     else | |||
|  |     { | |||
|  |       SuspendThread(hThread); | |||
|  |       memset(&c, 0, sizeof(CONTEXT)); | |||
|  |       c.ContextFlags = USED_CONTEXT_FLAGS; | |||
|  |       if (GetThreadContext(hThread, &c) == FALSE) | |||
|  |       { | |||
|  |         ResumeThread(hThread); | |||
|  |         return FALSE; | |||
|  |       } | |||
|  |     } | |||
|  |   } | |||
|  |   else | |||
|  |     c = *context; | |||
|  | 
 | |||
|  |   // init STACKFRAME for first call
 | |||
|  |   STACKFRAME64 s; // in/out stackframe
 | |||
|  |   memset(&s, 0, sizeof(s)); | |||
|  |   DWORD imageType; | |||
|  | #ifdef _M_IX86
 | |||
|  |   // normally, call ImageNtHeader() and use machine info from PE header
 | |||
|  |   imageType = IMAGE_FILE_MACHINE_I386; | |||
|  |   s.AddrPC.Offset = c.Eip; | |||
|  |   s.AddrPC.Mode = AddrModeFlat; | |||
|  |   s.AddrFrame.Offset = c.Ebp; | |||
|  |   s.AddrFrame.Mode = AddrModeFlat; | |||
|  |   s.AddrStack.Offset = c.Esp; | |||
|  |   s.AddrStack.Mode = AddrModeFlat; | |||
|  | #elif _M_X64
 | |||
|  |   imageType = IMAGE_FILE_MACHINE_AMD64; | |||
|  |   s.AddrPC.Offset = c.Rip; | |||
|  |   s.AddrPC.Mode = AddrModeFlat; | |||
|  |   s.AddrFrame.Offset = c.Rsp; | |||
|  |   s.AddrFrame.Mode = AddrModeFlat; | |||
|  |   s.AddrStack.Offset = c.Rsp; | |||
|  |   s.AddrStack.Mode = AddrModeFlat; | |||
|  | #elif _M_IA64
 | |||
|  |   imageType = IMAGE_FILE_MACHINE_IA64; | |||
|  |   s.AddrPC.Offset = c.StIIP; | |||
|  |   s.AddrPC.Mode = AddrModeFlat; | |||
|  |   s.AddrFrame.Offset = c.IntSp; | |||
|  |   s.AddrFrame.Mode = AddrModeFlat; | |||
|  |   s.AddrBStore.Offset = c.RsBSP; | |||
|  |   s.AddrBStore.Mode = AddrModeFlat; | |||
|  |   s.AddrStack.Offset = c.IntSp; | |||
|  |   s.AddrStack.Mode = AddrModeFlat; | |||
|  | #else
 | |||
|  | #error "Platform not supported!"
 | |||
|  | #endif
 | |||
|  | 
 | |||
|  |   pSym = (IMAGEHLP_SYMBOL64 *) malloc(sizeof(IMAGEHLP_SYMBOL64) + STACKWALK_MAX_NAMELEN); | |||
|  |   if (!pSym) goto cleanup;  // not enough memory...
 | |||
|  |   memset(pSym, 0, sizeof(IMAGEHLP_SYMBOL64) + STACKWALK_MAX_NAMELEN); | |||
|  |   pSym->SizeOfStruct = sizeof(IMAGEHLP_SYMBOL64); | |||
|  |   pSym->MaxNameLength = STACKWALK_MAX_NAMELEN; | |||
|  | 
 | |||
|  |   memset(&Line, 0, sizeof(Line)); | |||
|  |   Line.SizeOfStruct = sizeof(Line); | |||
|  | 
 | |||
|  |   memset(&Module, 0, sizeof(Module)); | |||
|  |   Module.SizeOfStruct = sizeof(Module); | |||
|  | 
 | |||
|  |   for (frameNum = 0; ; ++frameNum ) | |||
|  |   { | |||
|  |     // get next stack frame (StackWalk64(), SymFunctionTableAccess64(), SymGetModuleBase64())
 | |||
|  |     // if this returns ERROR_INVALID_ADDRESS (487) or ERROR_NOACCESS (998), you can
 | |||
|  |     // assume that either you are done, or that the stack is so hosed that the next
 | |||
|  |     // deeper frame could not be found.
 | |||
|  |     // CONTEXT need not to be suplied if imageTyp is IMAGE_FILE_MACHINE_I386!
 | |||
|  |     if ( ! this->m_sw->pSW(imageType, this->m_hProcess, hThread, &s, &c, myReadProcMem, this->m_sw->pSFTA, this->m_sw->pSGMB, NULL) ) | |||
|  |     { | |||
|  |       this->OnDbgHelpErr("StackWalk64", GetLastError(), s.AddrPC.Offset); | |||
|  |       break; | |||
|  |     } | |||
|  | 
 | |||
|  |     csEntry.offset = s.AddrPC.Offset; | |||
|  |     csEntry.name[0] = 0; | |||
|  |     csEntry.undName[0] = 0; | |||
|  |     csEntry.undFullName[0] = 0; | |||
|  |     csEntry.offsetFromSmybol = 0; | |||
|  |     csEntry.offsetFromLine = 0; | |||
|  |     csEntry.lineFileName[0] = 0; | |||
|  |     csEntry.lineNumber = 0; | |||
|  |     csEntry.loadedImageName[0] = 0; | |||
|  |     csEntry.moduleName[0] = 0; | |||
|  |     if (s.AddrPC.Offset == s.AddrReturn.Offset) | |||
|  |     { | |||
|  |       this->OnDbgHelpErr("StackWalk64-Endless-Callstack!", 0, s.AddrPC.Offset); | |||
|  |       break; | |||
|  |     } | |||
|  |     if (s.AddrPC.Offset != 0) | |||
|  |     { | |||
|  |       // we seem to have a valid PC
 | |||
|  |       // show procedure info (SymGetSymFromAddr64())
 | |||
|  |       if (this->m_sw->pSGSFA(this->m_hProcess, s.AddrPC.Offset, &(csEntry.offsetFromSmybol), pSym) != FALSE) | |||
|  |       { | |||
|  |         // TODO: Mache dies sicher...!
 | |||
|  |         strcpy_s(csEntry.name, pSym->Name); | |||
|  |         // UnDecorateSymbolName()
 | |||
|  |         this->m_sw->pUDSN( pSym->Name, csEntry.undName, STACKWALK_MAX_NAMELEN, UNDNAME_NAME_ONLY ); | |||
|  |         this->m_sw->pUDSN( pSym->Name, csEntry.undFullName, STACKWALK_MAX_NAMELEN, UNDNAME_COMPLETE ); | |||
|  |       } | |||
|  |       else | |||
|  |       { | |||
|  |         this->OnDbgHelpErr("SymGetSymFromAddr64", GetLastError(), s.AddrPC.Offset); | |||
|  |       } | |||
|  | 
 | |||
|  |       // show line number info, NT5.0-method (SymGetLineFromAddr64())
 | |||
|  |       if (this->m_sw->pSGLFA != NULL ) | |||
|  |       { // yes, we have SymGetLineFromAddr64()
 | |||
|  |         if (this->m_sw->pSGLFA(this->m_hProcess, s.AddrPC.Offset, &(csEntry.offsetFromLine), &Line) != FALSE) | |||
|  |         { | |||
|  |           csEntry.lineNumber = Line.LineNumber; | |||
|  |           // TODO: Mache dies sicher...!
 | |||
|  |           strcpy_s(csEntry.lineFileName, Line.FileName); | |||
|  |         } | |||
|  |         else | |||
|  |         { | |||
|  |           this->OnDbgHelpErr("SymGetLineFromAddr64", GetLastError(), s.AddrPC.Offset); | |||
|  |         } | |||
|  |       } // yes, we have SymGetLineFromAddr64()
 | |||
|  | 
 | |||
|  |       // show module info (SymGetModuleInfo64())
 | |||
|  |       if (this->m_sw->GetModuleInfo(this->m_hProcess, s.AddrPC.Offset, &Module ) != FALSE) | |||
|  |       { // got module info OK
 | |||
|  |         switch ( Module.SymType ) | |||
|  |         { | |||
|  |         case SymNone: | |||
|  |           csEntry.symTypeString = "-nosymbols-"; | |||
|  |           break; | |||
|  |         case SymCoff: | |||
|  |           csEntry.symTypeString = "COFF"; | |||
|  |           break; | |||
|  |         case SymCv: | |||
|  |           csEntry.symTypeString = "CV"; | |||
|  |           break; | |||
|  |         case SymPdb: | |||
|  |           csEntry.symTypeString = "PDB"; | |||
|  |           break; | |||
|  |         case SymExport: | |||
|  |           csEntry.symTypeString = "-exported-"; | |||
|  |           break; | |||
|  |         case SymDeferred: | |||
|  |           csEntry.symTypeString = "-deferred-"; | |||
|  |           break; | |||
|  |         case SymSym: | |||
|  |           csEntry.symTypeString = "SYM"; | |||
|  |           break; | |||
|  | #if API_VERSION_NUMBER >= 9
 | |||
|  |         case SymDia: | |||
|  |           csEntry.symTypeString = "DIA"; | |||
|  |           break; | |||
|  | #endif
 | |||
|  |         case 8: //SymVirtual:
 | |||
|  |           csEntry.symTypeString = "Virtual"; | |||
|  |           break; | |||
|  |         default: | |||
|  |           //_snprintf( ty, sizeof ty, "symtype=%ld", (long) Module.SymType );
 | |||
|  |           csEntry.symTypeString = NULL; | |||
|  |           break; | |||
|  |         } | |||
|  | 
 | |||
|  |         // TODO: Mache dies sicher...!
 | |||
|  |         strcpy_s(csEntry.moduleName, Module.ModuleName); | |||
|  |         csEntry.baseOfImage = Module.BaseOfImage; | |||
|  |         strcpy_s(csEntry.loadedImageName, Module.LoadedImageName); | |||
|  |       } // got module info OK
 | |||
|  |       else | |||
|  |       { | |||
|  |         this->OnDbgHelpErr("SymGetModuleInfo64", GetLastError(), s.AddrPC.Offset); | |||
|  |       } | |||
|  |     } // we seem to have a valid PC
 | |||
|  | 
 | |||
|  |     CallstackEntryType et = nextEntry; | |||
|  |     if (frameNum == 0) | |||
|  |       et = firstEntry; | |||
|  |     this->OnCallstackEntry(et, csEntry); | |||
|  |      | |||
|  |     if (s.AddrReturn.Offset == 0) | |||
|  |     { | |||
|  |       this->OnCallstackEntry(lastEntry, csEntry); | |||
|  |       SetLastError(ERROR_SUCCESS); | |||
|  |       break; | |||
|  |     } | |||
|  |   } // for ( frameNum )
 | |||
|  | 
 | |||
|  |   cleanup: | |||
|  |     if (pSym) free( pSym ); | |||
|  | 
 | |||
|  |   if (context == NULL) | |||
|  |     ResumeThread(hThread); | |||
|  | 
 | |||
|  |   return TRUE; | |||
|  | } | |||
|  | 
 | |||
|  | BOOL __stdcall StackWalker::myReadProcMem( | |||
|  |     HANDLE      hProcess, | |||
|  |     DWORD64     qwBaseAddress, | |||
|  |     PVOID       lpBuffer, | |||
|  |     DWORD       nSize, | |||
|  |     LPDWORD     lpNumberOfBytesRead | |||
|  |     ) | |||
|  | { | |||
|  |   if (s_readMemoryFunction == NULL) | |||
|  |   { | |||
|  |     SIZE_T st; | |||
|  |     BOOL bRet = ReadProcessMemory(hProcess, (LPVOID) qwBaseAddress, lpBuffer, nSize, &st); | |||
|  |     *lpNumberOfBytesRead = (DWORD) st; | |||
|  |     //printf("ReadMemory: hProcess: %p, baseAddr: %p, buffer: %p, size: %d, read: %d, result: %d\n", hProcess, (LPVOID) qwBaseAddress, lpBuffer, nSize, (DWORD) st, (DWORD) bRet);
 | |||
|  |     return bRet; | |||
|  |   } | |||
|  |   else | |||
|  |   { | |||
|  |     return s_readMemoryFunction(hProcess, qwBaseAddress, lpBuffer, nSize, lpNumberOfBytesRead, s_readMemoryFunction_UserData); | |||
|  |   } | |||
|  | } | |||
|  | 
 | |||
|  | void StackWalker::OnLoadModule(LPCSTR img, LPCSTR mod, DWORD64 baseAddr, DWORD size, DWORD result, LPCSTR symType, LPCSTR pdbName, ULONGLONG fileVersion) | |||
|  | { | |||
|  |   /*CHAR buffer[STACKWALK_MAX_NAMELEN];
 | |||
|  |   if (fileVersion == 0) | |||
|  |     _snprintf_s(buffer, STACKWALK_MAX_NAMELEN, "%s:%s (%p), size: %d (result: %d), SymType: '%s', PDB: '%s'\n", img, mod, (LPVOID) baseAddr, size, result, symType, pdbName); | |||
|  |   else | |||
|  |   { | |||
|  |     DWORD v4 = (DWORD) fileVersion & 0xFFFF; | |||
|  |     DWORD v3 = (DWORD) (fileVersion>>16) & 0xFFFF; | |||
|  |     DWORD v2 = (DWORD) (fileVersion>>32) & 0xFFFF; | |||
|  |     DWORD v1 = (DWORD) (fileVersion>>48) & 0xFFFF; | |||
|  |     _snprintf_s(buffer, STACKWALK_MAX_NAMELEN, "%s:%s (%p), size: %d (result: %d), SymType: '%s', PDB: '%s', fileVersion: %d.%d.%d.%d\n", img, mod, (LPVOID) baseAddr, size, result, symType, pdbName, v1, v2, v3, v4); | |||
|  |   } | |||
|  |   OnOutput(buffer);*/ | |||
|  | } | |||
|  | 
 | |||
|  | void StackWalker::OnCallstackEntry(CallstackEntryType eType, CallstackEntry &entry) | |||
|  | { | |||
|  |   CHAR buffer[STACKWALK_MAX_NAMELEN]; | |||
|  |   if ( (eType != lastEntry) && (entry.offset != 0) ) | |||
|  |   { | |||
|  |     if (entry.name[0] == 0) | |||
|  |       strcpy_s(entry.name, "(function-name not available)"); | |||
|  |     if (entry.undName[0] != 0) | |||
|  |       strcpy_s(entry.name, entry.undName); | |||
|  |     if (entry.undFullName[0] != 0) | |||
|  |       strcpy_s(entry.name, entry.undFullName); | |||
|  |     if (entry.lineFileName[0] == 0) | |||
|  |     { | |||
|  |       strcpy_s(entry.lineFileName, "(filename not available)"); | |||
|  |       if (entry.moduleName[0] == 0) | |||
|  |         strcpy_s(entry.moduleName, "(module-name not available)"); | |||
|  |       _snprintf_s(buffer, STACKWALK_MAX_NAMELEN, "%p (%s): %s: %s\n", (LPVOID) entry.offset, entry.moduleName, entry.lineFileName, entry.name); | |||
|  |     } | |||
|  |     else | |||
|  |       _snprintf_s(buffer, STACKWALK_MAX_NAMELEN, "%s (%d): %s\n", entry.lineFileName, entry.lineNumber, entry.name); | |||
|  |     OnOutput(buffer); | |||
|  |   } | |||
|  | } | |||
|  | 
 | |||
|  | void StackWalker::OnDbgHelpErr(LPCSTR szFuncName, DWORD gle, DWORD64 addr) | |||
|  | { | |||
|  |   CHAR buffer[STACKWALK_MAX_NAMELEN]; | |||
|  |   _snprintf_s(buffer, STACKWALK_MAX_NAMELEN, "ERROR: %s, GetLastError: %d (Address: %p)\n", szFuncName, gle, (LPVOID) addr); | |||
|  |   OnOutput(buffer); | |||
|  | } | |||
|  | 
 | |||
|  | void StackWalker::OnSymInit(LPCSTR szSearchPath, DWORD symOptions, LPCSTR szUserName) | |||
|  | { | |||
|  |   /*CHAR buffer[STACKWALK_MAX_NAMELEN];
 | |||
|  |   _snprintf_s(buffer, STACKWALK_MAX_NAMELEN, "SymInit: Symbol-SearchPath: '%s', symOptions: %d, UserName: '%s'\n", szSearchPath, symOptions, szUserName); | |||
|  |   OnOutput(buffer);*/ | |||
|  | //  // Also display the OS-version
 | |||
|  | //#if _MSC_VER <= 1200
 | |||
|  | //  OSVERSIONINFOA ver;
 | |||
|  | //  ZeroMemory(&ver, sizeof(OSVERSIONINFOA));
 | |||
|  | //  ver.dwOSVersionInfoSize = sizeof(ver);
 | |||
|  | //  if (GetVersionExA(&ver) != FALSE)
 | |||
|  | //  {
 | |||
|  | //    _snprintf_s(buffer, STACKWALK_MAX_NAMELEN, "OS-Version: %d.%d.%d (%s)\n", 
 | |||
|  | //      ver.dwMajorVersion, ver.dwMinorVersion, ver.dwBuildNumber,
 | |||
|  | //      ver.szCSDVersion);
 | |||
|  | //    OnOutput(buffer);
 | |||
|  | //  }
 | |||
|  | //#else
 | |||
|  | //  OSVERSIONINFOEXA ver;
 | |||
|  | //  ZeroMemory(&ver, sizeof(OSVERSIONINFOEXA));
 | |||
|  | //  ver.dwOSVersionInfoSize = sizeof(ver);
 | |||
|  | //  if (GetVersionExA( (OSVERSIONINFOA*) &ver) != FALSE)
 | |||
|  | //  {
 | |||
|  | //    _snprintf_s(buffer, STACKWALK_MAX_NAMELEN, "OS-Version: %d.%d.%d (%s) 0x%x-0x%x\n", 
 | |||
|  | //      ver.dwMajorVersion, ver.dwMinorVersion, ver.dwBuildNumber,
 | |||
|  | //      ver.szCSDVersion, ver.wSuiteMask, ver.wProductType);
 | |||
|  | //    OnOutput(buffer);
 | |||
|  | //  }
 | |||
|  | //#endif
 | |||
|  | } | |||
|  | 
 | |||
|  | void StackWalker::OnOutput(LPCSTR buffer) | |||
|  | { | |||
|  |   OutputDebugStringA(buffer); | |||
|  | } |