diff options
-rw-r--r-- | kexec/kexec.c | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/kexec/kexec.c b/kexec/kexec.c index 477b426..a1e9317 100644 --- a/kexec/kexec.c +++ b/kexec/kexec.c @@ -197,6 +197,10 @@ unsigned long locate_hole(struct kexec_info *info, /* Compute the free memory ranges */ max_mem_ranges = memory_ranges + info->nr_segments; mem_range = malloc(max_mem_ranges *sizeof(struct memory_range)); + if (!mem_range) { + perror("locate_hole: malloc"); + exit(-1); + } mem_ranges = 0; /* Perform a merge on the 2 sorted lists of memory ranges */ @@ -265,6 +269,7 @@ unsigned long locate_hole(struct kexec_info *info, } } } + free(mem_range); if (hole_base == ULONG_MAX) { fprintf(stderr, "Could not find a free area of memory of " "%lx bytes...\n", hole_size); |