From 66c0deefae3295af4b740eea0fe4fc604929dadf Mon Sep 17 00:00:00 2001
From: Martin Szulecki
Date: Wed, 20 Nov 2013 16:57:21 +0100
Subject: img4: Make sure to free created element headers to fix memory leak

---
 src/img4.c | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'src/img4.c')

diff --git a/src/img4.c b/src/img4.c
index ebe2ccd..17a4664 100644
--- a/src/img4.c
+++ b/src/img4.c
@@ -117,6 +117,15 @@ int img4_stitch_component(const char* component_name, const unsigned char* compo
 
 	outbuf = (unsigned char*)malloc(img4header_size + content_size);
 	if (!outbuf) {
+		if (magic_header) {
+			free(magic_header);
+		}
+		if (blob_header) {
+			free(blob_header);
+		}
+		if (img4header) {
+			free(img4header);
+		}
 		return -1;
 	}
 	p = outbuf;
@@ -138,5 +147,15 @@ int img4_stitch_component(const char* component_name, const unsigned char* compo
 	*img4_data = outbuf;
 	*img4_size = (p - outbuf);
 
+	if (magic_header) {
+		free(magic_header);
+	}
+	if (blob_header) {
+		free(blob_header);
+	}
+	if (img4header) {
+		free(img4header);
+	}
+
 	return 0;
 }
-- 
cgit v1.1-32-gdbae