#include <stdio.h>
#include "memutil.h"

void *safe_malloc(unsigned size)
{
#undef malloc
  void *p = malloc(size);
  if (size > 0 && p == NULL) {
    fprintf(stderr, "\nsafe_malloc: out of memory\n");
    exit(1);
  }
  return p;
}

void *safe_realloc(void *p, unsigned size)
{
#undef realloc
  void *q = realloc(p, size);
  if (size > 0 && q == NULL) {
    fprintf(stderr, "\nsafe_realloc: out of memory\n");
    exit(1);
  }
  return q;
}

char *safe_strdup(char *str)
{
#undef strdup
  char *new_str = safe_malloc(strlen(str) + 1);
  strcpy(new_str, str);
  return new_str;
}

void safe_free(void *p)
{
#undef free
  free(p);
}

void *ref(MEMORY_MANAGED_OBJECT *p)
{
  if (p->n_refs < 0) {
    fprintf(stderr, "ref caught negative ref count\n");
    exit(1);
  }
  ++p->n_refs;
  return (void*)p;
}

void deref(MEMORY_MANAGED_OBJECT *p)
{
  if (p->n_refs <= 0) {
    fprintf(stderr, "deref caught non-positive ref count\n");
    exit(1);
  }
  if (--p->n_refs == 0)
    safe_free(p);
}