From 31b771119b9c47942cf6a306aef4c97896b8ada8 Mon Sep 17 00:00:00 2001 From: Luke Shumaker Date: Tue, 15 May 2018 23:01:44 -0400 Subject: free --- lib/dedupe-range.c | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/lib/dedupe-range.c b/lib/dedupe-range.c index 32296c0..016f691 100644 --- a/lib/dedupe-range.c +++ b/lib/dedupe-range.c @@ -5,7 +5,7 @@ #include /* for uint64_t, PRIu64 */ #include /* for FIDEDUPRANGE and related */ #include /* for bool, true, false */ -#include /* for exit(3p), EXIT_SUCCESS, EXIT_FAILURE */ +#include /* for exit(3p), EXIT_SUCCESS, EXIT_FAILURE, malloc(3p), calloc(3p), free(3p) */ #include /* for ioctl(2) */ #include /* for sysconf(3p), _SC_PAGESIZE */ @@ -26,6 +26,8 @@ void dedupe_range(struct range src, struct range *dsts) { struct file_dedupe_range_info *range_info = calloc(dst_count, sizeof(struct file_dedupe_range_info)); + if (!range_info) + error(EXIT_FAILURE, errno, "malloc"); for (size_t i = 0; i < dst_count; i++) { int dst_fd = open(dsts[i].filename, dsts[i].flags); if (dst_fd < 0) @@ -37,6 +39,8 @@ void dedupe_range(struct range src, struct range *dsts) { for (size_t files_deduped = 0; files_deduped < dst_count; ) { uint16_t dest_count = MIN(dst_count - files_deduped, max_dst_count); struct file_dedupe_range *range = malloc(sizeof(struct file_dedupe_range) + dest_count * sizeof(struct file_dedupe_range_info)); + if (!range) + error(EXIT_FAILURE, errno, "malloc"); *range = (struct file_dedupe_range){ .src_offset = src.offset, .src_length = src.length, @@ -82,5 +86,7 @@ void dedupe_range(struct range src, struct range *dsts) { range->src_length -= bytes_deduped; } files_deduped += range->dest_count; + free(range); } + free(range_info); } -- cgit v1.2.3