#include <Tlhelp32.h>#include "dbghelp.h"#include <list>class MiniDumper{public:MiniDumper(bool headless);protected:static MiniDumper *gpDumper;static LONG WINAPI Handler( struct _EXCEPTION_POINTERS *pExceptionInfo );virtual void VSetDumpFileName(void);virtual MINIDUMP_USER_STREAM_INFORMATION *VGetUserStreamArray() { return NULL; }virtual const TCHAR *VGetUserMessage() { return _T(""); }_EXCEPTION_POINTERS *m_pExceptionInfo;time_t m_lTime;TCHAR m_szDumpPath[_MAX_PATH];TCHAR m_szAppPath[_MAX_PATH];TCHAR m_szAppBaseName[_MAX_PATH];LONG WriteMiniDump(_EXCEPTION_POINTERS *pExceptionInfo );BOOL m_bHeadless;};//MiniDump.cpp#include "stdafx.h"#include <time.h>#include <Windows.h>#include <string>#include <tchar.h>#include <assert.h>#include "MiniDump.h"#pragma comment(lib, "version.lib")// based on dbghelp.htypedef BOOL (WINAPI *MINIDUMPWRITEDUMP)(HANDLE hProcess, DWORD dwPid, HANDLE hFile, MINIDUMP_TYPE DumpType,CONST PMINIDUMP_EXCEPTION_INFORMATION ExceptionParam,CONST PMINIDUMP_USER_STREAM_INFORMATION UserStreamParam,CONST PMINIDUMP_CALLBACK_INFORMATION CallbackParam);MiniDumper *MiniDumper::gpDumper = NULL;//// MiniDumper::MiniDumper//MiniDumper::MiniDumper(bool headless){// Detect if there is more than one MiniDumper.assert(!gpDumper);if (!gpDumper){::SetUnhandledExceptionFilter( Handler );gpDumper = this;m_bHeadless = headless;// doesn't throw up a dialog, just writes the file.}}//// MiniDumper::Handler//LONG MiniDumper::Handler(_EXCEPTION_POINTERS *pExceptionInfo){LONG retval = EXCEPTION_CONTINUE_SEARCH;if (!gpDumper){return retval;}return gpDumper->WriteMiniDump(pExceptionInfo);}//// MiniDumper::WriteMiniDump//LONG MiniDumper::WriteMiniDump(_EXCEPTION_POINTERS *pExceptionInfo ){ time(&m_lTime);LONG retval = EXCEPTION_CONTINUE_SEARCH;m_pExceptionInfo = pExceptionInfo;// You have to find the right dbghelp.dll. // Look next to the EXE first since the one in System32 might be old (Win2k)HMODULE hDll = NULL;TCHAR szDbgHelpPath[_MAX_PATH];if (GetModuleFileName(NULL, m_szAppPath, _MAX_PATH)){TCHAR *pSlash = _tcsrchr( m_szAppPath, '\\' );if (pSlash){_tcscpy( m_szAppBaseName, pSlash + 1);*(pSlash+1) = 0;}_tcscpy(szDbgHelpPath, m_szAppPath); _tcscat(szDbgHelpPath, _T("DBGHELP.DLL"));hDll = ::LoadLibrary(szDbgHelpPath);}if (hDll==NULL){// If we haven't found it yet - try one more time.hDll = ::LoadLibrary(_T("DBGHELP.DLL"));}LPCTSTR szResult = NULL;if (hDll){MINIDUMPWRITEDUMP pMiniDumpWriteDump = (MINIDUMPWRITEDUMP)::GetProcAddress(hDll, "MiniDumpWriteDump");if (pMiniDumpWriteDump){TCHAR szScratch[USER_DATA_BUFFER_SIZE];VSetDumpFileName();// ask the user if they want to save a dump file//_tcssprintf(szScratch, _T("There was an unexpected error:\n\n%s\nWould you like to save a diagnostic file?\n\nFilename: %s"), VGetUserMessage(), m_szDumpPath);wsprintf(szScratch, _T("There was an unexpected error:\n\n%s\nWould you like to save a diagnostic file?\n\nFilename: %s"), VGetUserMessage(), m_szDumpPath);if (m_bHeadless || (::MessageBox( NULL, szScratch, NULL, MB_YESNO )==IDYES)){// create the fileHANDLE hFile = ::CreateFile( m_szDumpPath, GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL, NULL );if (hFile!=INVALID_HANDLE_VALUE){_MINIDUMP_EXCEPTION_INFORMATION ExInfo;ExInfo.ThreadId = ::GetCurrentThreadId();ExInfo.ExceptionPointers = pExceptionInfo;ExInfo.ClientPointers = NULL;// write the dumpBOOL bOK = pMiniDumpWriteDump( GetCurrentProcess(), GetCurrentProcessId(), hFile, MiniDumpNormal, &ExInfo, VGetUserStreamArray(), NULL );if (bOK){szResult = NULL;retval = EXCEPTION_EXECUTE_HANDLER;}else{//_tcssprintf( szScratch, _T("Failed to save dump file to '%s' (error %d)"), m_szDumpPath, GetLastError() );wsprintf( szScratch, _T("Failed to save dump file to '%s' (error %d)"), m_szDumpPath, GetLastError() );szResult = szScratch;}::CloseHandle(hFile);}else{//_tcssprintf( szScratch, _T("Failed to create dump file '%s' (error %d)"), m_szDumpPath, GetLastError() );wsprintf( szScratch, _T("Failed to create dump file '%s' (error %d)"), m_szDumpPath, GetLastError() );szResult = szScratch;}}}else{szResult = _T("DBGHELP.DLL too old");}}else{szResult = _T("DBGHELP.DLL not found");}if (szResult && !m_bHeadless)::MessageBox( NULL, szResult, NULL, MB_OK );TerminateProcess(GetCurrentProcess(), 0);// MLM Note: ExitThread will work, and it allows the MiniDumper to kill a crashed thread// without affecting the rest of the application. The question of the day:// Is That A Good Idea??? Answer: ABSOLUTELY NOT!!!!!!!////ExitThread(0);return retval;}//// MiniDumper::VSetDumpFileName//void MiniDumper::VSetDumpFileName(void){//_tcssprintf(m_szDumpPath, _T("%s%s.%ld.dmp"), m_szAppPath, m_szAppBaseName, m_lTime);SYSTEMTIME st = {0};//GetSystemTime(&st);GetLocalTime(&st);wsprintf(m_szDumpPath, _T("%s%s.%d-%02d-%02d %02d-%02d-%02d.dmp"), m_szAppPath, m_szAppBaseName, st.wYear, st.wMonth, st.wDay, st.wHour, st.wMinute, st.wSecond);//wsprintf(m_szDumpPath, _T("%s%s.%ld.dmp"), m_szAppPath, m_szAppBaseName, m_lTime);}
