teraterm/teraterm/ttdebug.cpp

/*
 * (C) 2020- TeraTerm Project
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <windows.h>
#include <stdio.h>
#include <imagehlp.h>

#include "compat_win.h"

#include "ttdebug.h"

/**
 *      �R���\�[���E�B���h�E���\�������f�o�O�p
 *              �f�o�O�p������������printf()�n���\����������������������������
 *
 *      @retval �R���\�[����Window Handle
 */
HWND DebugConsoleOpen(void)
{
        FILE *fp;
        HWND hWnd = pGetConsoleWindow();
        if (hWnd != NULL) {
                return hWnd;
        }
        AllocConsole();
        freopen_s(&fp, "CONOUT$", "w", stdout);
        freopen_s(&fp, "CONOUT$", "w", stderr);

        // �������{�^����������
        hWnd = pGetConsoleWindow();
        HMENU hmenu = GetSystemMenu(hWnd, FALSE);
        RemoveMenu(hmenu, SC_CLOSE, MF_BYCOMMAND);

        return hWnd;
}

//
// ���O�n���h�����t�b�N�i�X�^�b�N�g���[�X���_���v�j
//
// cf. http://svn.collab.net/repos/svn/trunk/subversion/libsvn_subr/win32_crashrpt.c
// (2007.9.30 yutaka)
//
// ���O�R�[�h������������������
#if !defined(_M_X64)
static const char *GetExceptionString(DWORD exception)
{
#define EXCEPTION(x) case EXCEPTION_##x: return (#x);
        static char buf[16];

        switch (exception)
        {
                EXCEPTION(ACCESS_VIOLATION)
                EXCEPTION(DATATYPE_MISALIGNMENT)
                EXCEPTION(BREAKPOINT)
                EXCEPTION(SINGLE_STEP)
                EXCEPTION(ARRAY_BOUNDS_EXCEEDED)
                EXCEPTION(FLT_DENORMAL_OPERAND)
                EXCEPTION(FLT_DIVIDE_BY_ZERO)
                EXCEPTION(FLT_INEXACT_RESULT)
                EXCEPTION(FLT_INVALID_OPERATION)
                EXCEPTION(FLT_OVERFLOW)
                EXCEPTION(FLT_STACK_CHECK)
                EXCEPTION(FLT_UNDERFLOW)
                EXCEPTION(INT_DIVIDE_BY_ZERO)
                EXCEPTION(INT_OVERFLOW)
                EXCEPTION(PRIV_INSTRUCTION)
                EXCEPTION(IN_PAGE_ERROR)
                EXCEPTION(ILLEGAL_INSTRUCTION)
                EXCEPTION(NONCONTINUABLE_EXCEPTION)
                EXCEPTION(STACK_OVERFLOW)
                EXCEPTION(INVALID_DISPOSITION)
                EXCEPTION(GUARD_PAGE)
                EXCEPTION(INVALID_HANDLE)

        default:
                _snprintf_s(buf, sizeof(buf), _TRUNCATE, "0x%x", exception);
                return buf;
                //return "UNKNOWN_ERROR";
        }
#undef EXCEPTION
}

/* ���O�������������������o���������\�������A���O�t�B���^���� */
static LONG CALLBACK ApplicationFaultHandler(EXCEPTION_POINTERS *ExInfo)
{
        HGLOBAL gptr;
        STACKFRAME sf;
        BOOL bResult;
        PIMAGEHLP_SYMBOL pSym;
        DWORD Disp;
        HANDLE hProcess = GetCurrentProcess();
        HANDLE hThread = GetCurrentThread();
        IMAGEHLP_MODULE ih_module;
        IMAGEHLP_LINE   ih_line;
        int frame;
        char msg[3072], buf[256];
        HMODULE h, h2;
        char imagehlp_dll[MAX_PATH];

        // Windows98/Me/NT4�����������������X�L�b�v�����B(2007.10.9 yutaka)
        GetSystemDirectory(imagehlp_dll, sizeof(imagehlp_dll));
        strncat_s(imagehlp_dll, sizeof(imagehlp_dll), "\\imagehlp.dll", _TRUNCATE);
        h2 = LoadLibrary(imagehlp_dll);
        if (((h = GetModuleHandle(imagehlp_dll)) == NULL) ||
                (GetProcAddress(h, "SymGetLineFromAddr") == NULL)) {
                        FreeLibrary(h2);
                        goto error;
        }
        FreeLibrary(h2);

        /* �V���{�������i�[�p�o�b�t�@�������� */
        gptr = GlobalAlloc(GMEM_FIXED, 10000);
        if (gptr == NULL) {
                goto error;
        }
        pSym = (PIMAGEHLP_SYMBOL)GlobalLock(gptr);
        ZeroMemory(pSym, sizeof(IMAGEHLP_SYMBOL));
        pSym->SizeOfStruct = 10000;
        pSym->MaxNameLength = 10000 - sizeof(IMAGEHLP_SYMBOL);

        /* �X�^�b�N�t���[���������� */
        ZeroMemory(&sf, sizeof(sf));
        sf.AddrPC.Offset = ExInfo->ContextRecord->Eip;
        sf.AddrStack.Offset = ExInfo->ContextRecord->Esp;
        sf.AddrFrame.Offset = ExInfo->ContextRecord->Ebp;
        sf.AddrPC.Mode = AddrModeFlat;
        sf.AddrStack.Mode = AddrModeFlat;
        sf.AddrFrame.Mode = AddrModeFlat;

        /* �V���{���n���h���������� */
        SymInitialize(hProcess, NULL, TRUE);

        // ���W�X�^�_���v
        msg[0] = '\0';
        _snprintf_s(buf, sizeof(buf), _TRUNCATE, "eax=%08X ebx=%08X ecx=%08X edx=%08X esi=%08X edi=%08X\r\n"
                   "ebp=%08X esp=%08X eip=%08X efl=%08X\r\n"
                   "cs=%04X ss=%04X ds=%04X es=%04X fs=%04X gs=%04X\r\n",
                   ExInfo->ContextRecord->Eax,
                   ExInfo->ContextRecord->Ebx,
                   ExInfo->ContextRecord->Ecx,
                   ExInfo->ContextRecord->Edx,
                   ExInfo->ContextRecord->Esi,
                   ExInfo->ContextRecord->Edi,
                   ExInfo->ContextRecord->Ebp,
                   ExInfo->ContextRecord->Esp,
                   ExInfo->ContextRecord->Eip,
                   ExInfo->ContextRecord->EFlags,
                   ExInfo->ContextRecord->SegCs,
                   ExInfo->ContextRecord->SegSs,
                   ExInfo->ContextRecord->SegDs,
                   ExInfo->ContextRecord->SegEs,
                   ExInfo->ContextRecord->SegFs,
                   ExInfo->ContextRecord->SegGs
        );
        strncat_s(msg, sizeof(msg), buf, _TRUNCATE);

        if (ExInfo->ExceptionRecord != NULL) {
                _snprintf_s(buf, sizeof(buf), _TRUNCATE, "Exception: %s\r\n", GetExceptionString(ExInfo->ExceptionRecord->ExceptionCode));
                strncat_s(msg, sizeof(msg), buf, _TRUNCATE);
        }

        /* �X�^�b�N�t���[���������\���������� */
        frame = 0;
        for (;;) {
                /* �����X�^�b�N�t���[�������� */
                bResult = StackWalk(
                        IMAGE_FILE_MACHINE_I386,
                        hProcess,
                        hThread,
                        &sf,
                        NULL,
                        NULL,
                        SymFunctionTableAccess,
                        SymGetModuleBase,
                        NULL);

                /* ���s�������A���[�v�������� */
                if (!bResult || sf.AddrFrame.Offset == 0)
                        break;

                frame++;

                /* �v���O�����J�E���^�i���z�A�h���X�j�������������I�t�Z�b�g������ */
                bResult = SymGetSymFromAddr(hProcess, sf.AddrPC.Offset, &Disp, pSym);

                /* �����������\�� */
                _snprintf_s(buf, sizeof(buf), _TRUNCATE, "#%d  0x%08x in ", frame, sf.AddrPC.Offset);
                strncat_s(msg, sizeof(msg), buf, _TRUNCATE);
                if (bResult) {
                        _snprintf_s(buf, sizeof(buf), _TRUNCATE, "%s() + 0x%x ", pSym->Name, Disp);
                        strncat_s(msg, sizeof(msg), buf, _TRUNCATE);
                } else {
                        _snprintf_s(buf, sizeof(buf), _TRUNCATE, " --- ");
                        strncat_s(msg, sizeof(msg), buf, _TRUNCATE);
                }

                // ���s�t�@�C����������
                ZeroMemory( &(ih_module), sizeof(ih_module) );
                ih_module.SizeOfStruct = sizeof(ih_module);
                bResult = SymGetModuleInfo( hProcess, sf.AddrPC.Offset, &(ih_module) );
                strncat_s(msg, sizeof(msg), "at ", _TRUNCATE);
                if (bResult) {
                        _snprintf_s(buf, sizeof(buf), _TRUNCATE, "%s ", ih_module.ImageName );
                        strncat_s(msg, sizeof(msg), buf, _TRUNCATE);
                } else {
                        _snprintf_s(buf, sizeof(buf), _TRUNCATE, "%s ", "<Unknown Module>" );
                        strncat_s(msg, sizeof(msg), buf, _TRUNCATE);
                }

                // �t�@�C�������s����������
                ZeroMemory( &(ih_line), sizeof(ih_line) );
                ih_line.SizeOfStruct = sizeof(ih_line);
                bResult = SymGetLineFromAddr( hProcess, sf.AddrPC.Offset, &Disp, &ih_line );
                if (bResult)
                {
                        _snprintf_s(buf, sizeof(buf), _TRUNCATE, "%s:%lu", ih_line.FileName, ih_line.LineNumber );
                        strncat_s(msg, sizeof(msg), buf, _TRUNCATE);
                }

                strncat_s(msg, sizeof(msg), "\n", _TRUNCATE);
        }

        /* ������ */
        SymCleanup(hProcess);
        GlobalUnlock(pSym);
        GlobalFree(pSym);

        // ���O�������������AAPI�����������o��
        ::MessageBoxA(NULL, msg, "Tera Term: Application fault", MB_OK | MB_ICONEXCLAMATION);

error:
//      return (EXCEPTION_EXECUTE_HANDLER);  /* ���������v���Z�X���I�������� */
        return (EXCEPTION_CONTINUE_SEARCH);  /* ���������m�A�v���P�[�V�����G���[�n�|�b�v�A�b�v���b�Z�[�W�{�b�N�X�������o�� */
}
#endif // !defined(_M_X64 )


/**
 *  ���O�n���h�����t�b�N
 */
void DebugSetException(void)
{
#if !defined(_M_X64)
        SetUnhandledExceptionFilter(ApplicationFaultHandler);
#endif
}