libumem/common/umem_fail.c

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 * Failure routines for libumem (not standalone)
 */

#include <sys/types.h>
#include <signal.h>
#include <stdarg.h>
#include <string.h>

#include "misc.h"

static volatile int umem_exiting = 0;
#define UMEM_EXIT_ABORT 1

static mutex_t umem_exit_lock = DEFAULTMUTEX; /* protects umem_exiting */

static int
firstexit(int type)
{
    if (umem_exiting)
        return (0);

    (void) mutex_lock(&umem_exit_lock);
    if (umem_exiting) {
        (void) mutex_unlock(&umem_exit_lock);
        return (0);
    }
    umem_exiting = type;
    (void) mutex_unlock(&umem_exit_lock);

    return (1);
}

/*
 * We can't use abort(3C), since it closes all of the standard library
 * FILEs, which can call free().
 *
 * In addition, we can't just raise(SIGABRT), since the current handler
 * might do allocation.  We give them once chance, though.
 */
static void __NORETURN
umem_do_abort(void)
{
    if (firstexit(UMEM_EXIT_ABORT))
        (void) raise(SIGABRT);

    for (;;) {
        (void) signal(SIGABRT, SIG_DFL);
        (void) sigrelse(SIGABRT);
        (void) raise(SIGABRT);
    }
}

#define SKIP_FRAMES     1   /* skip the panic frame */
#define ERR_STACK_FRAMES    128

static void
print_stacktrace(void)
{
    uintptr_t cur_stack[ERR_STACK_FRAMES];

    /*
     * if we are in a signal context, checking for it will recurse
     */
    uint_t nframes = getpcstack(cur_stack, ERR_STACK_FRAMES, 0);
    uint_t idx;

    if (nframes > SKIP_FRAMES) {
        umem_printf("stack trace:\n");

        for (idx = SKIP_FRAMES; idx < nframes; idx++) {
            (void) print_sym((void *)cur_stack[idx]);
            umem_printf("\n");
        }
    }
}

void
umem_panic(const char *format, ...)
{
    va_list va;

    va_start(va, format);
    umem_vprintf(format, va);
    va_end(va);

    if (format[strlen(format)-1] != '\n')
        umem_error_enter("\n");

    print_stacktrace();

    umem_do_abort();
}

void
umem_err_recoverable(const char *format, ...)
{
    va_list va;

    va_start(va, format);
    umem_vprintf(format, va);
    va_end(va);

    if (format[strlen(format)-1] != '\n')
        umem_error_enter("\n");

    print_stacktrace();

    if (umem_abort > 0)
        umem_do_abort();
}

int
__umem_assert_failed(const char *assertion, const char *file, int line)
{
    umem_panic("Assertion failed: %s, file %s, line %d\n",
        assertion, file, line);
    /*NOTREACHED*/
    return (0);
}