test: exit on malloc failure

This commit is contained in:
Max Bruckner 2016-12-06 08:41:51 +07:00
parent eeb6d237b6
commit f10e2f884a

15
test.c
View File

@ -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) {