#include "ns.h"	// Must come before everything else!

#include "compat.h"
#include "mdump.h"

#include <tchar.h>

BEGIN_DUKE_NS

LPCSTR MiniDumper::m_szAppName;

MiniDumper g_dumper("eduke32");

MiniDumper::MiniDumper(LPCSTR szAppName)
{
    // if this assert fires then you have two instances of MiniDumper
    // which is not allowed
    assert(m_szAppName == NULL);

    m_szAppName = szAppName ? _strdup(szAppName) : "Application";

    ::SetUnhandledExceptionFilter(TopLevelFilter);
}

LONG MiniDumper::TopLevelFilter(struct _EXCEPTION_POINTERS *pExceptionInfo)
{
    HMODULE hDll = NULL;
    char    szDbgHelpPath[_MAX_PATH];

    if (GetModuleFileNameW(NULL, szDbgHelpPath, _MAX_PATH))
    {
        auto pSlash = _tcsrchr(szDbgHelpPath, '\\');

        if (pSlash)
        {
            _tcscpy(pSlash + 1, "DBGHELP.DLL");
            hDll = ::LoadLibrary(szDbgHelpPath);
        }
    }

    if (hDll == NULL)
    {
        // load any version we can
        hDll = ::LoadLibrary("DBGHELP.DLL");
    }

    LONG    retval   = EXCEPTION_CONTINUE_SEARCH;
    LPCTSTR szResult = "DBGHELP.DLL not found";
    char    szScratch[_MAX_PATH];

    if (hDll)
    {
        auto pDump = MINIDUMPWRITEDUMP(::GetProcAddress(hDll, "MiniDumpWriteDump"));

        if (pDump)
        {
            char szDumpPath[_MAX_PATH];

            sprintf(szDumpPath, "%s_%u.dmp", m_szAppName, timeGetTime());

            HANDLE hFile = ::CreateFile(szDumpPath, GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, NULL);

            if (hFile != INVALID_HANDLE_VALUE)
            {
                _MINIDUMP_EXCEPTION_INFORMATION ExInfo = { ::GetCurrentThreadId(), pExceptionInfo, NULL };

                // take a dump
                BOOL bOK = pDump(GetCurrentProcess(), GetCurrentProcessId(), hFile, MiniDumpNormal, &ExInfo, NULL, NULL);

                if (bOK)
                {
                    sprintf(szScratch, "Saved dump file to \"%s\"", szDumpPath);
                    szResult = szScratch;
                    retval   = EXCEPTION_EXECUTE_HANDLER;
                }
                else
                {
                    sprintf(szScratch, "Failed to save dump file to \"%s\" (error %d)", szDumpPath, GetLastError());
                    szResult = szScratch;
                }

                ::CloseHandle(hFile);
            }
            else
            {
                sprintf(szScratch, "Failed to create dump file \"%s\" (error %d)", szDumpPath, GetLastError());
                szResult = szScratch;
            }
        }
        else
        {
            szResult = "DBGHELP.DLL too old";
        }

        ::FreeLibrary(hDll);
    }

    ::MessageBox(NULL, szResult, m_szAppName, MB_OK);

    return retval;
}

END_DUKE_NS