From 42870dff1f80e635d417ee1e83572d213f87f74c Mon Sep 17 00:00:00 2001 From: Mark Storer Date: Wed, 1 Oct 2014 10:05:59 -0400 Subject: [PATCH] Small fix to check the return code when malloc'ing temporary buffers. --- src/erasurecode_preprocessing.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/erasurecode_preprocessing.c b/src/erasurecode_preprocessing.c index c3e5ed6..62134ba 100644 --- a/src/erasurecode_preprocessing.c +++ b/src/erasurecode_preprocessing.c @@ -143,6 +143,10 @@ int prepare_fragments_for_decode( *realloc_bm = *realloc_bm | (1 << i); } else if (!is_addr_aligned((unsigned long)data[i], 16)) { char *tmp_buf = alloc_fragment_buffer(fragment_size - sizeof(fragment_header_t)); + if (NULL == tmp_buf) { + log_error("Could not allocate temp buffer!"); + return -1; + } memcpy(tmp_buf, data[i], fragment_size); data[i] = tmp_buf; *realloc_bm = *realloc_bm | (1 << i); @@ -178,6 +182,10 @@ int prepare_fragments_for_decode( *realloc_bm = *realloc_bm | (1 << (k + i)); } else if (!is_addr_aligned((unsigned long)parity[i], 16)) { char *tmp_buf = alloc_fragment_buffer(fragment_size-sizeof(fragment_header_t)); + if (NULL == tmp_buf) { + log_error("Could not allocate temp buffer!"); + return -1; + } memcpy(tmp_buf, parity[i], fragment_size); parity[i] = tmp_buf; *realloc_bm = *realloc_bm | (1 << (k + i));