X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/4d78228c5d21c6320ecca02f587e167d0c6f024d..2417770134dfb5e942934143c41b4545400f0018:/sbr/utils.c diff --git a/sbr/utils.c b/sbr/utils.c index d3228909..c9769fb0 100644 --- a/sbr/utils.c +++ b/sbr/utils.c @@ -39,44 +39,61 @@ void *mh_xmalloc(size_t size) return p; } -/* - * Safely call realloc - */ -void * -mh_xrealloc(void *ptr, size_t size) +/* Call realloc(3), exiting on NULL return. */ +void *mh_xrealloc(void *ptr, size_t size) { - void *memory; + void *new; - /* Some non-POSIX realloc()s don't cope with realloc(NULL,sz) */ + /* Copy POSIX behaviour, coping with non-POSIX systems. */ + if (size == 0) { + if (ptr) { + free(ptr); + } + return mh_xmalloc(1); /* Get a unique pointer. */ + } if (!ptr) return mh_xmalloc(size); - if (size == 0) - adios(NULL, "Tried to realloc 0 bytes"); + new = realloc(ptr, size); + if (!new) + adios(NULL, "realloc failed, size wanted: %zu", size); - memory = realloc(ptr, size); - if (!memory) - adios(NULL, "Realloc failed"); + return new; +} - return memory; +/* Call calloc(3), exiting on NULL return. */ +void *mh_xcalloc(size_t nelem, size_t elsize) +{ + void *p; + + if (!nelem || !elsize) + return mh_xmalloc(1); /* Get a unique pointer. */ + + p = calloc(nelem, elsize); + if (!p) + adios(NULL, "calloc failed, size wanted: %zu * %zu", nelem, elsize); + + return p; } -/* - * Safely call calloc - */ -void * -mh_xcalloc(size_t nmemb, size_t size) +/* Duplicate a NUL-terminated string, exit on failure. */ +char *mh_xstrdup(const char *src) { - void *memory; + size_t n; + char *dest; - if (nmemb == 0 || size == 0) - adios(NULL, "Tried to calloc 0 bytes"); + n = strlen(src) + 1; /* Ignore possibility of overflow. */ + dest = mh_xmalloc(n); + memcpy(dest, src, n); - if ((memory = calloc(nmemb, size))) { - return memory; - } else { - adios(NULL, "calloc failed"); - } + return dest; +} + +/* Call free(3), if ptr isn't NULL. */ +void mh_xfree(void *ptr) +{ + if (ptr) + free(ptr); /* Some very old platforms can't cope with NULL. */ } /* @@ -86,7 +103,7 @@ mh_xcalloc(size_t nmemb, size_t size) char * pwd(void) { - register char *cp; + char *cp; static char curwd[PATH_MAX]; if (!getcwd (curwd, PATH_MAX)) {