n_signals.c

Go to the documentation of this file.

/*
 * Nilorea Library
 * Copyright (C) 2005-2026 Castagnier Mickael
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#include <signal.h>
#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include <stdbool.h>
#include <stdint.h>
#include <errno.h>
#include <string.h>
 
#include "nilorea/n_log.h"
#include "nilorea/n_signals.h"
#include "nilorea/n_str.h"
 
#define LOGFPRT(...) fprintf(stderr, "Error: " __VA_ARGS__)
 
#define LOGNLOG(...) n_log(LOG_ERR, __VA_ARGS__)
 
#define LOGSIG LOGNLOG
 
static const char* __n_stack_traced_progam_name;
 
#ifdef __windows__
#include "nilorea/n_windows.h"
#include <imagehlp.h>
#else
#include <err.h>
#include <execinfo.h>
#endif
 
// void almost_c99_signal_handler(int sig)
// {
//   switch(sig)
//   {
//     case SIGABRT:
//       LOGSIG("Caught SIGABRT: usually caused by an abort() or assert()" );
//       break;
//     case SIGFPE:
//       LOGSIG("Caught SIGFPE: arithmetic exception, such as divide by zero" );
//       break;
//     case SIGILL:
//       LOGSIG("Caught SIGILL: illegal instruction" );
//       break;
//     case SIGINT:
//       LOGSIG("Caught SIGINT: interactive attention signal, probably a ctrl+c" );
//       break;
//     case SIGSEGV:
//       LOGSIG("Caught SIGSEGV: segfault" );
//       break;
//     case SIGTERM:
//     default:
//       LOGSIG("Caught SIGTERM: a termination request was sent to the program" );
//       break;
//   }
//   _Exit(1);
// }
 
// void set_signal_handler()
// {
//   signal(SIGABRT, almost_c99_signal_handler);
//   signal(SIGFPE,  almost_c99_signal_handler);
//   signal(SIGILL,  almost_c99_signal_handler);
//   signal(SIGINT,  almost_c99_signal_handler);
//   signal(SIGSEGV, almost_c99_signal_handler);
//   signal(SIGTERM, almost_c99_signal_handler);
// }
 
int addr2line(char const* const program_name, void const* const addr) {
    char addr2line_cmd[4093] = "";
 
    /* have addr2line map the address to the relent line in the code */
#ifdef __APPLE__
    /* apple does things differently... */
    sprintf(addr2line_cmd, "atos -o %.256s ./%p", program_name, addr);
#else
#ifdef __windows__
    snprintf(addr2line_cmd, sizeof(addr2line_cmd), "addr2line -f -p -e ./%s %p", program_name, addr);
    LOGSIG("%s", addr2line_cmd);
    return 0;
#else
    sprintf(addr2line_cmd, "addr2line -f -p -e %.256s %p", program_name, addr);
#endif
#endif
    N_STR* output = NULL;
    int ret = -1;
    n_popen(addr2line_cmd, 1024, (void**)&output, &ret);
    if (!output || !output->data) {
        free_nstr(&output);
        return -1;
    }
    LOGSIG("%s", output->data);
    free_nstr(&output);
    return ret;
} /* addr2line(...) */
 
#ifdef __windows__
void windows_print_stacktrace(CONTEXT* context) {
    SymInitialize(GetCurrentProcess(), 0, true);
 
    STACKFRAME frame = {0};
 
    /* setup initial stack frame */
#if (_WIN32_WINNT >= _WIN32_WINNT_WIN8)
    /* Wrong values for W10 ? */
    frame.AddrPC.Offset = context->Eip;
    frame.AddrPC.Mode = AddrModeFlat;
    frame.AddrStack.Offset = context->Esp;
    frame.AddrStack.Mode = AddrModeFlat;
    frame.AddrFrame.Offset = context->Ebp;
    frame.AddrFrame.Mode = AddrModeFlat;
#else
    frame.AddrPC.Offset = context->Rip;
    frame.AddrPC.Mode = AddrModeFlat;
    frame.AddrStack.Offset = context->Rsp;
    frame.AddrStack.Mode = AddrModeFlat;
    frame.AddrFrame.Offset = context->Rbp;
    frame.AddrFrame.Mode = AddrModeFlat;
#endif
 
    while (StackWalk(IMAGE_FILE_MACHINE_I386,
                     GetCurrentProcess(),
                     GetCurrentThread(),
                     &frame,
                     context,
                     0,
                     SymFunctionTableAccess,
                     SymGetModuleBase,
                     0)) {
        addr2line(__n_stack_traced_progam_name, (void*)frame.AddrPC.Offset);
    }
 
    SymCleanup(GetCurrentProcess());
} /* windows_print_stacktrace(...) */
 
LONG WINAPI windows_exception_handler(EXCEPTION_POINTERS* ExceptionInfo) {
    switch (ExceptionInfo->ExceptionRecord->ExceptionCode) {
        case EXCEPTION_ACCESS_VIOLATION:
            LOGSIG("Error: EXCEPTION_ACCESS_VIOLATION");
            break;
        case EXCEPTION_ARRAY_BOUNDS_EXCEEDED:
            LOGSIG("Error: EXCEPTION_ARRAY_BOUNDS_EXCEEDED");
            break;
        case EXCEPTION_BREAKPOINT:
            LOGSIG("Error: EXCEPTION_BREAKPOINT");
            break;
        case EXCEPTION_DATATYPE_MISALIGNMENT:
            LOGSIG("Error: EXCEPTION_DATATYPE_MISALIGNMENT");
            break;
        case EXCEPTION_FLT_DENORMAL_OPERAND:
            LOGSIG("Error: EXCEPTION_FLT_DENORMAL_OPERAND");
            break;
        case EXCEPTION_FLT_DIVIDE_BY_ZERO:
            LOGSIG("Error: EXCEPTION_FLT_DIVIDE_BY_ZERO");
            break;
        case EXCEPTION_FLT_INEXACT_RESULT:
            LOGSIG("Error: EXCEPTION_FLT_INEXACT_RESULT");
            break;
        case EXCEPTION_FLT_INVALID_OPERATION:
            LOGSIG("Error: EXCEPTION_FLT_INVALID_OPERATION");
            break;
        case EXCEPTION_FLT_OVERFLOW:
            LOGSIG("Error: EXCEPTION_FLT_OVERFLOW");
            break;
        case EXCEPTION_FLT_STACK_CHECK:
            LOGSIG("Error: EXCEPTION_FLT_STACK_CHECK");
            break;
        case EXCEPTION_FLT_UNDERFLOW:
            LOGSIG("Error: EXCEPTION_FLT_UNDERFLOW");
            break;
        case EXCEPTION_ILLEGAL_INSTRUCTION:
            LOGSIG("Error: EXCEPTION_ILLEGAL_INSTRUCTION");
            break;
        case EXCEPTION_IN_PAGE_ERROR:
            LOGSIG("Error: EXCEPTION_IN_PAGE_ERROR");
            break;
        case EXCEPTION_INT_DIVIDE_BY_ZERO:
            LOGSIG("Error: EXCEPTION_INT_DIVIDE_BY_ZERO");
            break;
        case EXCEPTION_INT_OVERFLOW:
            LOGSIG("Error: EXCEPTION_INT_OVERFLOW");
            break;
        case EXCEPTION_INVALID_DISPOSITION:
            LOGSIG("Error: EXCEPTION_INVALID_DISPOSITION");
            break;
        case EXCEPTION_NONCONTINUABLE_EXCEPTION:
            LOGSIG("Error: EXCEPTION_NONCONTINUABLE_EXCEPTION");
            break;
        case EXCEPTION_PRIV_INSTRUCTION:
            LOGSIG("Error: EXCEPTION_PRIV_INSTRUCTION");
            break;
        case EXCEPTION_SINGLE_STEP:
            LOGSIG("Error: EXCEPTION_SINGLE_STEP");
            break;
        case EXCEPTION_STACK_OVERFLOW:
            LOGSIG("Error: EXCEPTION_STACK_OVERFLOW");
            break;
        default:
            LOGSIG("Error: Unrecognized Exception");
            break;
    }
    /* If this is a stack overflow then we can't walk the stack, so just show
       where the error happened */
    if (EXCEPTION_STACK_OVERFLOW != ExceptionInfo->ExceptionRecord->ExceptionCode) {
        windows_print_stacktrace(ExceptionInfo->ContextRecord);
    } else {
        addr2line(__n_stack_traced_progam_name, (void*)ExceptionInfo->ContextRecord->Rip);
    }
    fflush(stdout);
    fflush(stderr);
    return EXCEPTION_EXECUTE_HANDLER;
} /* windows_exception_handler( ... ) */
 
void set_signal_handler(const char* progname) {
    __n_stack_traced_progam_name = progname;
    SetUnhandledExceptionFilter(windows_exception_handler);
}
#else
static void* stack_traces[MAX_STACK_FRAMES];
 
void posix_print_stack_trace() {
    int i, trace_size = 0;
    char** messages = (char**)NULL;
 
    memset(stack_traces, 0, MAX_STACK_FRAMES * sizeof(void*));
 
    trace_size = backtrace(stack_traces, MAX_STACK_FRAMES);
    messages = backtrace_symbols(stack_traces, trace_size);
    if (!messages) return;
 
    /* skip the first couple stack frames (as they are this function and
       our handler) and also skip the last frame as it's (always?) junk. */
    // for (i = 3; i < (trace_size - 1); ++i)
    for (i = 0; i < trace_size; ++i)  // we'll use this for now so you can see what's going on
    {
        if (addr2line(__n_stack_traced_progam_name, stack_traces[i]) != 0) {
            LOGSIG("  error determining line # for: %s", messages[i]);
        }
    }
    if (messages) {
        free(messages);
    }
}
 
// cppcheck-suppress constParameterCallback -- callback signature must match system typedef
void posix_signal_handler(int sig, siginfo_t* siginfo, void* context) {
    (void)context;
    switch (sig) {
        case SIGSEGV:
            LOGSIG("Caught SIGSEGV: Segmentation Fault");
            break;
        case SIGINT:
            LOGSIG("Caught SIGINT: Interactive attention signal, (usually ctrl+c)");
            break;
        case SIGFPE:
            switch (siginfo->si_code) {
                case FPE_INTDIV:
                    LOGSIG("Caught SIGFPE: (integer divide by zero)");
                    break;
                case FPE_INTOVF:
                    LOGSIG("Caught SIGFPE: (integer overflow)");
                    break;
                case FPE_FLTDIV:
                    LOGSIG("Caught SIGFPE: (floating-point divide by zero)");
                    break;
                case FPE_FLTOVF:
                    LOGSIG("Caught SIGFPE: (floating-point overflow)");
                    break;
                case FPE_FLTUND:
                    LOGSIG("Caught SIGFPE: (floating-point underflow)");
                    break;
                case FPE_FLTRES:
                    LOGSIG("Caught SIGFPE: (floating-point inexact result)");
                    break;
                case FPE_FLTINV:
                    LOGSIG("Caught SIGFPE: (floating-point invalid operation)");
                    break;
                case FPE_FLTSUB:
                    LOGSIG("Caught SIGFPE: (subscript out of range)");
                    break;
                default:
                    LOGSIG("Caught SIGFPE: Arithmetic Exception");
                    break;
            }
            break;
        case SIGILL:
            switch (siginfo->si_code) {
                case ILL_ILLOPC:
                    LOGSIG("Caught SIGILL: (illegal opcode)");
                    break;
                case ILL_ILLOPN:
                    LOGSIG("Caught SIGILL: (illegal operand)");
                    break;
                case ILL_ILLADR:
                    LOGSIG("Caught SIGILL: (illegal addressing mode)");
                    break;
                case ILL_ILLTRP:
                    LOGSIG("Caught SIGILL: (illegal trap)");
                    break;
                case ILL_PRVOPC:
                    LOGSIG("Caught SIGILL: (privileged opcode)");
                    break;
                case ILL_PRVREG:
                    LOGSIG("Caught SIGILL: (privileged register)");
                    break;
                case ILL_COPROC:
                    LOGSIG("Caught SIGILL: (coprocessor error)");
                    break;
                case ILL_BADSTK:
                    LOGSIG("Caught SIGILL: (internal stack error)");
                    break;
                default:
                    LOGSIG("Caught SIGILL: Illegal Instruction");
                    break;
            }
            break;
        case SIGTERM:
            LOGSIG("Caught SIGTERM: a termination request was sent to the program");
            break;
        case SIGABRT:
            LOGSIG("Caught SIGABRT: usually caused by an abort() or assert()");
            break;
        default:
            break;
    }
    posix_print_stack_trace();
    _Exit(1);
}
 
void set_signal_handler(const char* progname) {
    __n_stack_traced_progam_name = progname;
 
#ifdef RLIMIT_STACK
    /* Before starting the endless recursion, try to be friendly to the user's
       machine.  On some Linux 2.2.x systems, there is no stack limit for user
       processes at all.  We don't want to kill such systems.  */
    struct rlimit rl;
    rl.rlim_cur = rl.rlim_max = 0x100000; /* 1 MB */
    setrlimit(RLIMIT_STACK, &rl);
#endif
 
#ifdef __linux__
    /* setup alternate stack */
    {
        static uint8_t alternate_stack[SIGALTSTACK_SIZE];
        stack_t ss;
 
        /* malloc is usually used here, I'm not 100% sure my static allocation
           is valid but it seems to work just fine. */
        ss.ss_sp = alternate_stack;
        ss.ss_size = sizeof(alternate_stack);
        ss.ss_flags = 0;
 
        if (sigaltstack(&ss, NULL) != 0) {
            err(1, "sigaltstack");
        }
    }
#endif
    /* register our signal handlers */
    {
        struct sigaction sig_action;
        sigemptyset(&sig_action.sa_mask);
        sig_action.sa_sigaction = posix_signal_handler;
 
#ifdef __APPLE__
        /* for some reason we backtrace() doesn't work on osx
           when we use an alternate stack */
        sig_action.sa_flags = SA_SIGINFO;
#else
        sig_action.sa_flags = SA_SIGINFO | SA_ONSTACK;
#endif
 
        if (sigaction(SIGSEGV, &sig_action, NULL) != 0) {
            err(1, "sigaction");
        }
        if (sigaction(SIGFPE, &sig_action, NULL) != 0) {
            err(1, "sigaction");
        }
        if (sigaction(SIGINT, &sig_action, NULL) != 0) {
            err(1, "sigaction");
        }
        if (sigaction(SIGILL, &sig_action, NULL) != 0) {
            err(1, "sigaction");
        }
        /* No TERM catch
           if (sigaction(SIGTERM, &sig_action, NULL) != 0) { err(1, "sigaction"); } */
        if (sigaction(SIGABRT, &sig_action, NULL) != 0) {
            err(1, "sigaction");
        }
    }
}
#endif