mirror of
git://git.musl-libc.org/musl
synced 2025-01-07 07:00:24 +00:00
allow interposition/replacement of allocator (malloc)
replacement is subject to conditions on the replacement functions.
they may only call functions which are async-signal-safe, as specified
either by POSIX or as an implementation-defined extension. if any
allocator functions are replaced, at least malloc, realloc, and free
must be provided. if calloc is not provided, it will behave as
malloc+memset. any of the memalign-family functions not provided will
fail with ENOMEM.
in order to implement the above properties, calloc and __memalign
check that they are using their own malloc or free, respectively.
choice to check malloc or free is based on considerations of
supporting __simple_malloc. in order to make this work, calloc is
split into separate versions for __simple_malloc and full malloc;
commit ba819787ee
already did most of
the split anyway, and completing it saves an extra call frame.
previously, use of -Bsymbolic-functions made dynamic interposition
impossible. now, we are using an explicit dynamic-list, so add
allocator functions to the list. most are not referenced anyway, but
all are added for completeness.
This commit is contained in:
parent
c1014a812c
commit
c9f415d7ea
@ -6,6 +6,15 @@ stdin;
|
|||||||
stdout;
|
stdout;
|
||||||
stderr;
|
stderr;
|
||||||
|
|
||||||
|
malloc;
|
||||||
|
calloc;
|
||||||
|
realloc;
|
||||||
|
free;
|
||||||
|
memalign;
|
||||||
|
posix_memalign;
|
||||||
|
aligned_alloc;
|
||||||
|
malloc_usable_size;
|
||||||
|
|
||||||
timezone;
|
timezone;
|
||||||
daylight;
|
daylight;
|
||||||
tzname;
|
tzname;
|
||||||
|
@ -1,13 +0,0 @@
|
|||||||
#include <stdlib.h>
|
|
||||||
#include <errno.h>
|
|
||||||
|
|
||||||
void *__malloc0(size_t);
|
|
||||||
|
|
||||||
void *calloc(size_t m, size_t n)
|
|
||||||
{
|
|
||||||
if (n && m > (size_t)-1/n) {
|
|
||||||
errno = ENOMEM;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
return __malloc0(n * m);
|
|
||||||
}
|
|
@ -47,4 +47,14 @@ static void *__simple_malloc(size_t n)
|
|||||||
}
|
}
|
||||||
|
|
||||||
weak_alias(__simple_malloc, malloc);
|
weak_alias(__simple_malloc, malloc);
|
||||||
weak_alias(__simple_malloc, __malloc0);
|
|
||||||
|
static void *__simple_calloc(size_t m, size_t n)
|
||||||
|
{
|
||||||
|
if (n && m > (size_t)-1/n || malloc != __simple_malloc) {
|
||||||
|
errno = ENOMEM;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return __simple_malloc(n * m);
|
||||||
|
}
|
||||||
|
|
||||||
|
weak_alias(__simple_calloc, calloc);
|
||||||
|
@ -368,6 +368,8 @@ void *malloc(size_t n)
|
|||||||
return CHUNK_TO_MEM(c);
|
return CHUNK_TO_MEM(c);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
weak_alias(malloc, __internal_malloc);
|
||||||
|
|
||||||
static size_t mal0_clear(char *p, size_t pagesz, size_t n)
|
static size_t mal0_clear(char *p, size_t pagesz, size_t n)
|
||||||
{
|
{
|
||||||
#ifdef __GNUC__
|
#ifdef __GNUC__
|
||||||
@ -386,13 +388,21 @@ static size_t mal0_clear(char *p, size_t pagesz, size_t n)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void *__malloc0(size_t n)
|
void *calloc(size_t m, size_t n)
|
||||||
{
|
{
|
||||||
|
if (n && m > (size_t)-1/n) {
|
||||||
|
errno = ENOMEM;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
n *= m;
|
||||||
void *p = malloc(n);
|
void *p = malloc(n);
|
||||||
if (!p || IS_MMAPPED(MEM_TO_CHUNK(p)))
|
if (!p) return p;
|
||||||
return p;
|
if (malloc == __internal_malloc) {
|
||||||
if (n >= PAGE_SIZE)
|
if (IS_MMAPPED(MEM_TO_CHUNK(p)))
|
||||||
n = mal0_clear(p, PAGE_SIZE, n);
|
return p;
|
||||||
|
if (n >= PAGE_SIZE)
|
||||||
|
n = mal0_clear(p, PAGE_SIZE, n);
|
||||||
|
}
|
||||||
return memset(p, 0, n);
|
return memset(p, 0, n);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -558,6 +568,8 @@ void free(void *p)
|
|||||||
bin_chunk(self);
|
bin_chunk(self);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
weak_alias(free, __internal_free);
|
||||||
|
|
||||||
void __malloc_donate(char *start, char *end)
|
void __malloc_donate(char *start, char *end)
|
||||||
{
|
{
|
||||||
size_t align_start_up = (SIZE_ALIGN-1) & (-(uintptr_t)start - OVERHEAD);
|
size_t align_start_up = (SIZE_ALIGN-1) & (-(uintptr_t)start - OVERHEAD);
|
||||||
|
@ -3,9 +3,7 @@
|
|||||||
#include <errno.h>
|
#include <errno.h>
|
||||||
#include "libc.h"
|
#include "libc.h"
|
||||||
|
|
||||||
/* This function should work with most dlmalloc-like chunk bookkeeping
|
void __internal_free(void *);
|
||||||
* systems, but it's only guaranteed to work with the native implementation
|
|
||||||
* used in this library. */
|
|
||||||
|
|
||||||
void *__memalign(size_t align, size_t len)
|
void *__memalign(size_t align, size_t len)
|
||||||
{
|
{
|
||||||
@ -17,7 +15,7 @@ void *__memalign(size_t align, size_t len)
|
|||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (len > SIZE_MAX - align) {
|
if (len > SIZE_MAX - align || free != __internal_free) {
|
||||||
errno = ENOMEM;
|
errno = ENOMEM;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user