Fix reconstruct to return an error when memory allocation failed

As like current code, if we do go jump to just "out" section when
memory allocation failed, liberaurecode/PyECLib will report it as
success. However, in fact, the returned binary is not the one
reconstructed. This patch fixes liberasurecode_reconstruct to return
error code (ENOMEM) if either malloc or memset for working space is
failed.

Change-Id: I7bfc481c7a9bbc64288760df2dc6053c57657b41

1 files changed, 3 insertions, 0 deletions

diff --git a/src/erasurecode.c b/src/erasurecode.c
index 31fc884..540cc26 100644
--- a/src/erasurecode.c
+++ b/src/erasurecode.c
@@ -812,18 +812,21 @@ int liberasurecode_reconstruct_fragment(int desc,
     data = alloc_zeroed_buffer(sizeof(char*) * k);
     if (NULL == data) {
         log_error("Could not allocate data buffer!");
+        ret = -ENOMEM;
         goto out;
     }
 
     parity = alloc_zeroed_buffer(sizeof(char*) * m);
     if (NULL == parity) {
         log_error("Could not allocate parity buffer!");
+        ret = -ENOMEM;
         goto out;
     }
 
     missing_idxs = alloc_and_set_buffer(sizeof(int*) * (k + m), -1);
     if (NULL == missing_idxs) {
         log_error("Could not allocate missing_idxs buffer!");
+        ret = -ENOMEM;
         goto out;
     }