Merge pull request ethereum#27 from zama-ai/louis/add-cgo-destructors

fix: added cgo destructors
sbellem · Jan 23, 2023 · 9223a49 · 9223a49
2 parents 36ac51d + 99e81c5
commit 9223a49
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/core/vm/contracts.go b/core/vm/contracts.go
@@ -120,6 +120,9 @@ void encrypt_integer(BufferView cks_buff_view, uint64_t val, Buffer* ct_buf)
 
 	int ser_ok = shortint_serialize_ciphertext(ct, ct_buf);
 	assert(ser_ok == 0);
+
+	destroy_shortint_ciphertext(ct);
+	destroy_shortint_client_key(cks);
 }
 
 uint64_t decrypt_integer(BufferView cks_buf_view, BufferView ct_buf_view)
@@ -137,6 +140,9 @@ uint64_t decrypt_integer(BufferView cks_buf_view, BufferView ct_buf_view)
 	int ct_decrypt = shortint_bc_client_key_decrypt(cks, ct, &res);
 	assert(ct_decrypt == 0);
 
+	destroy_shortint_ciphertext(ct);
+	destroy_shortint_client_key(cks);
+
 	return res;
 }