Brian Silverman | 8649792 | 2018-02-10 19:28:39 -0500 | [diff] [blame] | 1 | /* Memory handling for libdw. |
| 2 | Copyright (C) 2003, 2004, 2006 Red Hat, Inc. |
| 3 | This file is part of elfutils. |
| 4 | Written by Ulrich Drepper <drepper@redhat.com>, 2003. |
| 5 | |
| 6 | This file is free software; you can redistribute it and/or modify |
| 7 | it under the terms of either |
| 8 | |
| 9 | * the GNU Lesser General Public License as published by the Free |
| 10 | Software Foundation; either version 3 of the License, or (at |
| 11 | your option) any later version |
| 12 | |
| 13 | or |
| 14 | |
| 15 | * the GNU General Public License as published by the Free |
| 16 | Software Foundation; either version 2 of the License, or (at |
| 17 | your option) any later version |
| 18 | |
| 19 | or both in parallel, as here. |
| 20 | |
| 21 | elfutils is distributed in the hope that it will be useful, but |
| 22 | WITHOUT ANY WARRANTY; without even the implied warranty of |
| 23 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 24 | General Public License for more details. |
| 25 | |
| 26 | You should have received copies of the GNU General Public License and |
| 27 | the GNU Lesser General Public License along with this program. If |
| 28 | not, see <http://www.gnu.org/licenses/>. */ |
| 29 | |
| 30 | #ifdef HAVE_CONFIG_H |
| 31 | # include <config.h> |
| 32 | #endif |
| 33 | |
| 34 | #include <error.h> |
| 35 | #include <errno.h> |
| 36 | #include <stdlib.h> |
| 37 | #include "libdwP.h" |
| 38 | #include "system.h" |
| 39 | |
| 40 | |
| 41 | void * |
| 42 | __libdw_allocate (Dwarf *dbg, size_t minsize, size_t align) |
| 43 | { |
| 44 | size_t size = MAX (dbg->mem_default_size, |
| 45 | (align - 1 + |
| 46 | 2 * minsize + offsetof (struct libdw_memblock, mem))); |
| 47 | struct libdw_memblock *newp = malloc (size); |
| 48 | if (newp == NULL) |
| 49 | dbg->oom_handler (); |
| 50 | |
| 51 | uintptr_t result = ((uintptr_t) newp->mem + align - 1) & ~(align - 1); |
| 52 | |
| 53 | newp->size = size - offsetof (struct libdw_memblock, mem); |
| 54 | newp->remaining = (uintptr_t) newp + size - (result + minsize); |
| 55 | |
| 56 | newp->prev = dbg->mem_tail; |
| 57 | dbg->mem_tail = newp; |
| 58 | |
| 59 | return (void *) result; |
| 60 | } |
| 61 | |
| 62 | |
| 63 | Dwarf_OOM |
| 64 | dwarf_new_oom_handler (Dwarf *dbg, Dwarf_OOM handler) |
| 65 | { |
| 66 | Dwarf_OOM old = dbg->oom_handler; |
| 67 | dbg->oom_handler = handler; |
| 68 | return old; |
| 69 | } |
| 70 | |
| 71 | |
| 72 | void |
| 73 | __attribute ((noreturn)) attribute_hidden |
| 74 | __libdw_oom (void) |
| 75 | { |
| 76 | while (1) |
| 77 | error (EXIT_FAILURE, ENOMEM, "libdw"); |
| 78 | } |