From 7d60a4bb7cd8b646b61a84fa1f0c31916ace3c51 Mon Sep 17 00:00:00 2001
From: "Diego F. Aranha" <dfaranha@gmail.com>
Date: Thu, 8 Aug 2024 12:25:04 +0200
Subject: [PATCH] Fix memory allocation issues.

---
 src/ep/relic_ep_map.c   | 6 ++++--
 src/epx/relic_ep2_map.c | 6 ++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/ep/relic_ep_map.c b/src/ep/relic_ep_map.c
index fb923a61e..c96b78795 100644
--- a/src/ep/relic_ep_map.c
+++ b/src/ep/relic_ep_map.c
@@ -258,6 +258,7 @@ void ep_map_swift(ep_t p, const uint8_t *msg, size_t len) {
 	bn_null(k);
 	fp_null(v);
 	fp_null(w);
+	fp_null(y);
 	fp_null(t1);
 	fp_null(t2);
 	fp_null(x1);
@@ -271,6 +272,7 @@ void ep_map_swift(ep_t p, const uint8_t *msg, size_t len) {
 		bn_new(k);
 		fp_new(v);
 		fp_new(w);
+		fp_new(y);
 		fp_new(t1);
 		fp_new(t2);
 		fp_new(x1);
@@ -478,6 +480,7 @@ void ep_map_swift(ep_t p, const uint8_t *msg, size_t len) {
 		bn_free(k);
 		fp_free(v);
 		fp_free(w);
+		fp_free(y);
 		fp_free(t1);
 		fp_free(t2);
 		fp_free(x1);
@@ -488,8 +491,7 @@ void ep_map_swift(ep_t p, const uint8_t *msg, size_t len) {
 		fp_free(d[2]);
 		RLC_FREE(pseudo_random_bytes);
 		for (size_t i = 0; i < 8; i++) {
-			fp_null(h[i]);
-			fp_new(h[i]);
+			fp_free(h[i]);
 		}
 	}
 }
diff --git a/src/epx/relic_ep2_map.c b/src/epx/relic_ep2_map.c
index a99998f8b..447050d53 100644
--- a/src/epx/relic_ep2_map.c
+++ b/src/epx/relic_ep2_map.c
@@ -516,6 +516,12 @@ void ep2_map_swift(ep2_t p, const uint8_t *msg, size_t len) {
 	}
 	RLC_FINALLY {
 		bn_free(k);
+		fp2_free(a);
+		fp2_free(b);
+		fp2_free(c);
+		fp2_free(d);
+		fp2_free(e);
+		fp2_free(f);
 		fp2_free(t);
 		fp2_free(u);
 		fp2_free(v);