Add xcalloc()

Signed-off-by: Andi Kleen <ak@linux.intel.com>

2 files changed, 17 insertions, 4 deletions

diff --git a/util.c b/util.c
index 17fc0af..811a285 100644
--- a/util.c
+++ b/util.c
@@ -2,13 +2,25 @@
 #include <stdio.h>
 #include "util.h"
 
+void oom(void)
+{
+	fprintf(stderr, "Out of virtual memory\n");
+	exit(1);
+}
+
+void *xcalloc(size_t a, size_t b)
+{
+	void *p = calloc(a, b);
+	if (!p)
+		oom();
+	return p;
+}
+
 void *xalloc(size_t sz)
 {
 	void *p = calloc(sz, 1);
-	if (!p) {
-		fprintf(stderr, "Out of virtual memory\n");
-		exit(1);
-	}
+	if (!p)
+		oom();
 	return p;
 }
 
diff --git a/util.h b/util.h
index 2ed47be..93fd295 100644
--- a/util.h
+++ b/util.h
@@ -3,6 +3,7 @@ void err(const char *msg);
 
 #define NEW(x) ((x) = xalloc(sizeof(*(x))))
 void *xalloc(size_t sz);
+void *xcalloc(size_t a, size_t b);
 
 #ifdef __GNUC__
 #define barrier() asm volatile("" ::: "memory")