From 1feb3c40514a7c1726f29502ce37966c308bea79 Mon Sep 17 00:00:00 2001
From: amelia squires <ameliasquires@disroot.org>
Date: Mon, 16 Sep 2024 12:16:29 -0500
Subject: fix _copy and mem leaks

---
 src/hash/sha2-256.c | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

(limited to 'src/hash/sha2-256.c')

diff --git a/src/hash/sha2-256.c b/src/hash/sha2-256.c
index 7e37d9d..42f6a81 100644
--- a/src/hash/sha2-256.c
+++ b/src/hash/sha2-256.c
@@ -245,7 +245,13 @@ struct iv sha_iv_gen(int i){
     return (struct iv){.h0 = a.h0, .h1 = a.h1, .h2 = a.h2, .h3 = a.h3, .h4 = a.h4, .h5 = a.h5, .h6 = a.h6, .h7 = a.h7};
 }
 
-common_hash_clone(sha512);
+//common_hash_clone(sha512);
+lua_common_hash_clone_oargs(sha512, sha512, l_sha512_init(L), {
+    uint8_t* old = b->buffer;
+    *b = *a;
+    b->buffer = old;
+    memcpy(b->buffer, a->buffer, bs * sizeof * b->buffer);
+});
 lua_common_hash_init_ni(sha512, sha512, sha512_t_init_l(sha512_iv, L));
 lua_common_hash_update(sha512, sha512);
 
@@ -259,7 +265,13 @@ int l_sha512_final(lua_State* L){
   return 1;
 }
 
-common_hash_clone(sha384);
+//common_hash_clone(sha384);
+lua_common_hash_clone_oargs(sha384, sha384, l_sha384_init(L), {
+    uint8_t* old = b->buffer;
+    *b = *a;
+    b->buffer = old;
+    memcpy(b->buffer, a->buffer, bs * sizeof * b->buffer);
+});
 lua_common_hash_init_ni(sha384, sha384, sha512_t_init_l(sha384_iv, L));
 lua_common_hash_update(sha384, sha384);
 
@@ -278,7 +290,11 @@ int l_sha512_t_clone(lua_State* L){
   lua_pushinteger(L, a->t);
   l_sha512_t_init(L);
   struct sha512_hash* b = (struct sha512_hash*)lua_touserdata(L, -1);
+
+  uint8_t* old = b->buffer;
   *b = *a;
+  b->buffer = old;
+  memcpy(b->buffer, a->buffer, bs * sizeof * b->buffer);
   return 1;
 }
 
-- 
cgit v1.2.3