1 files changed, 8 insertions, 0 deletions
diff --git a/src/test.c b/src/test.c
index d309fb7..f3d5f3c 100644
--- a/src/test.c
+++ b/src/test.c
@@ -40,6 +40,14 @@ main(int argc, char **argv)
   }
 
   unsigned char *input = malloc (sizeof(unsigned char) * 16);
+  
+  if (input == NULL) {
+    clean_parser(parser);
+    clean_signed(&error_vec, 4);
+    printf("malloc fails at %s:%d\n", __FILE__, __LINE__);
+
+    return EXIT_FAILURE;
+  }
 
   for (int i = 0; i < 16; i++) *(input+i) = 0;