Fix leak in encode_gc

helium · May 11, 2020 · 4d1e413 · 4d1e413
1 parent c688c1e
commit 4d1e413
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/c_src/erasure.c b/c_src/erasure.c
@@ -201,6 +201,9 @@ encode_gc(ErlNifEnv * env, int argc, const ERL_NIF_TERM argv[])
     }
     schedule = jerasure_smart_bitmatrix_to_schedule(k, m, w, bitmatrix);
     if (schedule == NULL) {
+        free(shards);
+        free(matrix);
+        free(bitmatrix);
         return enif_make_badarg(env);
     }
     jerasure_schedule_encode(k, m, w, schedule, data_ptrs, coding_ptrs, blocksize, blocksize/w);
@@ -221,6 +224,8 @@ encode_gc(ErlNifEnv * env, int argc, const ERL_NIF_TERM argv[])
     }
     free(shards);
     free(matrix);
+    free(bitmatrix);
+    jerasure_free_schedule(schedule);
     return enif_make_tuple2(env, enif_make_atom(env, "ok"), list);
 }
 
@@ -575,6 +580,9 @@ decode_gc(ErlNifEnv * env, int argc, const ERL_NIF_TERM argv[])
 
 cleanup:
 
+        if (bitmatrix != NULL) {
+            free(bitmatrix);
+        }
         if (matrix != NULL) {
             free(matrix);
         }