/* See LICENSE file for copyright and license details. */ #include "common.h" #ifndef TEST void * libsimple_enrealloc(int status, void *ptr, size_t n) { void *ret = realloc(ptr, n); if (!ret) enprintf(status, "realloc:"); return ret; } #else #include "test.h" int main(void) { struct allocinfo *info; void *ptr, *old; assert((ptr = libsimple_enrealloc(1, NULL, 5))); if (have_custom_malloc()) { assert((info = get_allocinfo(ptr))); assert(info->size == 5); assert(!info->zeroed); assert(!((uintptr_t)ptr % (uintptr_t)(info->alignment))); info->refcount += 1; } stpcpy(ptr, "test"); assert((ptr = libsimple_enrealloc(1, old = ptr, 10))); assert(!strcmp(ptr, "test")); if (have_custom_malloc()) { assert((info = get_allocinfo(ptr))); assert(info->size == 10); assert(!info->zeroed); assert(!((uintptr_t)ptr % (uintptr_t)(info->alignment))); assert(ptr != old); free(old); } free(ptr); if (have_custom_malloc()) { alloc_fail_in = 1; assert_exit_ptr(libsimple_enrealloc(2, NULL, 1)); assert(exit_status == 2); assert_stderr("%s: realloc: %s\n", argv0, strerror(ENOMEM)); assert(!alloc_fail_in); } return 0; } #endif