From f10e2f884a55c285b9c99c1a640e14333180470f Mon Sep 17 00:00:00 2001
From: Max Bruckner <max@maxbruckner.de>
Date: Tue, 6 Dec 2016 08:41:51 +0700
Subject: [PATCH] test: exit on malloc failure

---
 test.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/test.c b/test.c
index 2b204ea..cfbb217 100644
--- a/test.c
+++ b/test.c
@@ -61,6 +61,11 @@ static void dofile(char *filename)
     fseek(f, 0, SEEK_SET);
 
     data = (char*)malloc(len + 1);
+    if (data == NULL)
+    {
+        printf("Failed to allocate memory.\n");
+        exit(1);
+    }
 
     fread(data, 1, len, f);
     data[len] = '\0';
@@ -102,10 +107,20 @@ static int print_preallocated(cJSON *root)
     /* the extra 64 bytes are in case a floating point value is printed */
     len = strlen(out) + 64;
     buf = malloc(len);
+    if (buf == NULL)
+    {
+        printf("Failed to allocate memory.\n");
+        exit(1);
+    }
 
     /* create buffer to fail */
     len_fail = strlen(out);
     buf_fail = malloc(len_fail);
+    if (buf_fail == NULL)
+    {
+        printf("Failed to allocate memory.\n");
+        exit(1);
+    }
 
     /* Print to buffer */
     if (cJSON_PrintPreallocated(root, buf, len, 1) != 0) {