diff options
Diffstat (limited to 'encalloc.c')
| -rw-r--r-- | encalloc.c | 37 |
1 files changed, 36 insertions, 1 deletions
@@ -4,7 +4,7 @@ void * -libsimple_encalloc(int status, size_t n, size_t m) /* TODO test */ +libsimple_encalloc(int status, size_t n, size_t m) { void *ret = calloc(n, m); if (!ret) @@ -19,6 +19,41 @@ libsimple_encalloc(int status, size_t n, size_t m) /* TODO test */ int main(void) { + struct allocinfo *info; + void *ptr; + + assert((ptr = libsimple_encalloc(1, 2, 5))); + if (have_custom_malloc()) { + assert((info = get_allocinfo(ptr))); + assert(info->size == 10); + assert(info->zeroed == 10); + } + free(ptr); + + assert((ptr = libsimple_ecalloc(3, 4))); + if (have_custom_malloc()) { + assert((info = get_allocinfo(ptr))); + assert(info->size == 12); + assert(info->zeroed == 12); + } + free(ptr); + + if (have_custom_malloc()) { + alloc_fail_in = 1; + assert_exit_ptr(libsimple_encalloc(5, 1, 1)); + assert(exit_status == 5); + assert_stderr("%s: calloc: %s\n", argv0, strerror(ENOMEM)); + assert(!alloc_fail_in); + + libsimple_default_failure_exit = 103; + alloc_fail_in = 1; + assert_exit_ptr(libsimple_ecalloc(1, 1)); + assert(exit_status == 103); + assert_stderr("%s: calloc: %s\n", argv0, strerror(ENOMEM)); + assert(!alloc_fail_in); + libsimple_default_failure_exit = 1; + } + return 0; } |