make sure to free context in all paths

Cremator · May 22, 2014 · 79a4772 · 79a4772
1 parent f507c1f
commit 79a4772
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cryptonight.c b/cryptonight.c
@@ -279,6 +279,7 @@ int scanhash_cryptonight(int thr_id, uint32_t *pdata, const uint32_t *ptarget,
             cryptonight_hash_ctx_aes_ni(hash, pdata, 76, ctx);
             if (unlikely(hash[7] < ptarget[7])) {
                 *hashes_done = n - first_nonce + 1;
+                free(ctx);
                 return true;
             }
         } while (likely((n <= max_nonce && !work_restart[thr_id].restart)));
@@ -288,6 +289,7 @@ int scanhash_cryptonight(int thr_id, uint32_t *pdata, const uint32_t *ptarget,
             cryptonight_hash_ctx(hash, pdata, 76, ctx);
             if (unlikely(hash[7] < ptarget[7])) {
                 *hashes_done = n - first_nonce + 1;
+                free(ctx);
                 return true;
             }
         } while (likely((n <= max_nonce && !work_restart[thr_id].restart)));