jerasure.c: add check for result of malloc()

Add check for bitmatrix and return NULL if malloc failed.

Signed-off-by: Danny Al-Gaaf <danny.al-gaaf@bisect.de>

1 files changed, 1 insertions, 0 deletions

diff --git a/src/jerasure.c b/src/jerasure.c
index de54fcf..a2bd3e7 100644
--- a/src/jerasure.c
+++ b/src/jerasure.c
@@ -276,6 +276,7 @@ int *jerasure_matrix_to_bitmatrix(int k, int m, int w, int *matrix)
   if (matrix == NULL) { return NULL; }
 
   bitmatrix = talloc(int, k*m*w*w);
+  if (!bitmatrix) return NULL;
 
   rowelts = k * w;
   rowindex = 0;