Add OOM-safe must_calloc and must_realloc helper functions

1 files changed, 22 insertions, 0 deletions

diff --git a/src/util.c b/src/util.c
index 9b79ac7..5918968 100644
--- a/src/util.c
+++ b/src/util.c
@@ -11,3 +11,25 @@ must_malloc(size_t size)
 		exit(EXIT_ABNORMAL);
 	}
 }
+
+void *
+must_calloc(size_t nmemb, size_t size)
+{
+	void *ret = calloc(nmemb, size);
+	if (ret == NULL) {
+		fprintf(stderr, "out of memory\n");
+		exit(EXIT_ABNORMAL);
+	}
+	return ret;
+}
+
+void *
+must_realloc(void *ptr, size_t size)
+{
+	void *ret = realloc(ptr, size);
+	if (ret == NULL) {
+		fprintf(stderr, "out of memory\n");
+		exit(EXIT_ABNORMAL);
+	}
+	return ret;
+}