Check the return value of realloc() for error

modified: src/nms.c
bartobri · Sep 20, 2017 · 40da205 · 40da205
1 parent 30bd00f
commit 40da205
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/nms.c b/src/nms.c
@@ -57,6 +57,10 @@ int main(int argc, char *argv[]) {
 		if (inSize > inCapacity) {
 			inCapacity *= INPUT_GROWTH_FACTOR;
 			input = realloc(input, inCapacity + 1);
+			if (input == NULL) {
+				fprintf (stderr, "Memory Allocation Error! Quitting...\n");
+				return 1;
+			}
 		}
 		input[inSize - 1] = c;
 		input[inSize] = '\0';