136 lines
3.4 KiB
C
136 lines
3.4 KiB
C
/* allocfail.c -- Test for libbacktrace library
|
|
Copyright (C) 2018-2021 Free Software Foundation, Inc.
|
|
|
|
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 AUTHOR ``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 AUTHOR 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 <assert.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include "filenames.h"
|
|
|
|
#include "backtrace.h"
|
|
#include "backtrace-supported.h"
|
|
|
|
#include "testlib.h"
|
|
|
|
extern uint64_t get_nr_allocs (void);
|
|
extern void set_fail_at_alloc (uint64_t);
|
|
extern int at_fail_alloc_p (void);
|
|
|
|
static int test1 (void) __attribute__ ((noinline, unused));
|
|
static int f2 (int) __attribute__ ((noinline));
|
|
static int f3 (int, int) __attribute__ ((noinline));
|
|
|
|
static unsigned callback_errors = 0;
|
|
|
|
static void
|
|
error_callback_full (void *vdata ATTRIBUTE_UNUSED,
|
|
const char *msg ATTRIBUTE_UNUSED,
|
|
int errnum ATTRIBUTE_UNUSED)
|
|
{
|
|
if (at_fail_alloc_p ())
|
|
{
|
|
set_fail_at_alloc (0);
|
|
return;
|
|
}
|
|
|
|
callback_errors++;
|
|
}
|
|
|
|
static int
|
|
callback_full (void *vdata ATTRIBUTE_UNUSED, uintptr_t pc ATTRIBUTE_UNUSED,
|
|
const char *filename ATTRIBUTE_UNUSED,
|
|
int lineno ATTRIBUTE_UNUSED,
|
|
const char *function ATTRIBUTE_UNUSED)
|
|
{
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
test1 (void)
|
|
{
|
|
return f2 (__LINE__) + 1;
|
|
}
|
|
|
|
static int
|
|
f2 (int f1line)
|
|
{
|
|
return f3 (f1line, __LINE__) + 2;
|
|
}
|
|
|
|
static int
|
|
f3 (int f1line ATTRIBUTE_UNUSED, int f2line ATTRIBUTE_UNUSED)
|
|
{
|
|
int i;
|
|
|
|
i = backtrace_full (state, 0, callback_full, error_callback_full, NULL);
|
|
|
|
if (i != 0)
|
|
{
|
|
fprintf (stderr, "test1: unexpected return value %d\n", i);
|
|
++failures;
|
|
}
|
|
|
|
if (callback_errors)
|
|
++failures;
|
|
|
|
return failures;
|
|
}
|
|
|
|
/* Run all the tests. */
|
|
|
|
int
|
|
main (int argc, char **argv)
|
|
{
|
|
uint64_t fail_at = 0;
|
|
|
|
if (argc == 2)
|
|
{
|
|
fail_at = atoi (argv[1]);
|
|
set_fail_at_alloc (fail_at);
|
|
}
|
|
|
|
state = backtrace_create_state (argv[0], BACKTRACE_SUPPORTS_THREADS,
|
|
error_callback_full, NULL);
|
|
if (state == NULL)
|
|
exit (failures ? EXIT_FAILURE : EXIT_SUCCESS);
|
|
|
|
#if BACKTRACE_SUPPORTED
|
|
test1 ();
|
|
#endif
|
|
|
|
if (argc == 1)
|
|
fprintf (stderr, "%llu\n", (long long unsigned) get_nr_allocs ());
|
|
|
|
exit (failures ? EXIT_FAILURE : EXIT_SUCCESS);
|
|
}
|