Abort on allocation errors

1 files changed, 6 insertions, 0 deletions

diff --git a/choices.c b/choices.c
index 933b8a0..d1eaa3c 100644
--- a/choices.c
+++ b/choices.c
@@ -1,5 +1,6 @@
 #define _GNU_SOURCE
 #include <stdlib.h>
+#include <stdio.h>
 
 #include "choices.h"
 #include "match.h"
@@ -23,6 +24,11 @@ static void choices_resize(choices_t *c, int new_capacity){
 	c->scores  = realloc(c->scores,  new_capacity * sizeof(double));
 	c->sorted  = realloc(c->sorted,  new_capacity * sizeof(size_t));
 
+	if(!c->strings || !c->scores || !c->sorted){
+		fprintf(stderr, "Error: Can't allocate memory\n");
+		abort();
+	}
+
 	for(int i = c->capacity; i < new_capacity; i++){
 		c->strings[i] = NULL;
 	}