123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114 |
- /* instrumented_alloc.c -- Memory allocation instrumented to fail when
- requested, for testing purposes.
- Copyright (C) 2018-2022 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 all the header files of alloc here, to make sure they're not
- processed when including alloc.c below, such that the redefinitions of malloc
- and realloc are only effective in alloc.c itself. This does not work for
- config.h, because it's not wrapped in "#ifndef CONFIG_H\n#define CONFIG_H"
- and "#endif" but that does not seem to be harmful. */
- #include "config.h"
- #include <errno.h>
- #include <stdlib.h>
- #include <sys/types.h>
- #include <inttypes.h>
- #include "backtrace.h"
- #include "internal.h"
- extern void *instrumented_malloc (size_t size);
- extern void *instrumented_realloc (void *ptr, size_t size);
- #define malloc instrumented_malloc
- #define realloc instrumented_realloc
- #include "alloc.c"
- #undef malloc
- #undef realloc
- static uint64_t nr_allocs = 0;
- static uint64_t fail_at_alloc = 0;
- extern int at_fail_alloc_p (void);
- extern uint64_t get_nr_allocs (void);
- extern void set_fail_at_alloc (uint64_t);
- void *
- instrumented_malloc (size_t size)
- {
- void *res;
- if (at_fail_alloc_p ())
- return NULL;
- res = malloc (size);
- if (res != NULL)
- nr_allocs++;
- return res;
- }
- void *
- instrumented_realloc (void *ptr, size_t size)
- {
- void *res;
- if (size != 0)
- {
- if (at_fail_alloc_p ())
- return NULL;
- }
- res = realloc (ptr, size);
- if (res != NULL)
- nr_allocs++;
- return res;
- }
- int
- at_fail_alloc_p (void)
- {
- return fail_at_alloc == nr_allocs + 1;
- }
- uint64_t
- get_nr_allocs (void)
- {
- return nr_allocs;
- }
- void
- set_fail_at_alloc (uint64_t nr)
- {
- fail_at_alloc = nr;
- }
|