shithub: puzzles

ref: b967a3ea86d6285457abc2a89e455c8f62d2c22a
dir: /malloc.c/

View raw version
/*
 * malloc.c: safe wrappers around malloc, realloc, free, strdup
 */

#ifndef NO_STDINT_H
#include <stdint.h>
#endif
#include <stdlib.h>
#include <string.h>
#include "puzzles.h"

/*
 * smalloc should guarantee to return a useful pointer - we
 * can do nothing except die when it's out of memory anyway.
 */
void *smalloc(size_t size) {
    void *p;
#ifdef PTRDIFF_MAX
    if (size > PTRDIFF_MAX)
	fatal("allocation too large");
#endif
    p = malloc(size);
    if (!p)
	fatal("out of memory");
    return p;
}

/*
 * sfree should guaranteeably deal gracefully with freeing NULL
 */
void sfree(void *p) {
    if (p) {
	free(p);
    }
}

/*
 * srealloc should guaranteeably be able to realloc NULL
 */
void *srealloc(void *p, size_t size) {
    void *q;
#ifdef PTRDIFF_MAX
    if (size > PTRDIFF_MAX)
	fatal("allocation too large");
#endif
    if (p) {
	q = realloc(p, size);
    } else {
	q = malloc(size);
    }
    if (!q)
	fatal("out of memory");
    return q;
}

/*
 * dupstr is like strdup, but with the never-return-NULL property
 * of smalloc (and also reliably defined in all environments :-)
 */
char *dupstr(const char *s) {
    char *r = smalloc(1+strlen(s));
    strcpy(r,s);
    return r;
}