From 2c97de1a26686adf2a6377bc4ef18d1aec6a1f3a Mon Sep 17 00:00:00 2001
From: Neven Sajko <nsajko@gmail.com>
Date: Sun, 11 Aug 2019 00:54:13 +0000
Subject: libmach: plug memory leak

Change-Id: I08cb7227c071c7fc2e30f43e07bcf599fc1d174a
---
 src/libmach/elf.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/libmach')

diff --git a/src/libmach/elf.c b/src/libmach/elf.c
index 2459a291..5b7494d9 100644
--- a/src/libmach/elf.c
+++ b/src/libmach/elf.c
@@ -191,7 +191,7 @@ elfinit(int fd)
 		ElfHdrBytes h32;
 		ElfHdrBytes64 h64;
 	} hdrb;
-	void *p;
+	void *p = nil;
 	ElfSect *s;
 
 	e = mallocz(sizeof(Elf), 1);
@@ -234,6 +234,7 @@ elfinit(int fd)
 		unpackprog(h, &e->prog[i], p);
 	}
 	free(p);
+	p = nil;
 
 	e->nsect = h->shnum;
 	if(e->nsect == 0)
@@ -247,6 +248,7 @@ elfinit(int fd)
 		unpacksect(h, &e->sect[i], p);
 	}
 	free(p);
+	p = nil;
 
 	if(h->shstrndx >= e->nsect){
 		fprint(2, "warning: bad string section index %d >= %d", h->shstrndx, e->nsect);
@@ -287,6 +289,7 @@ nosects:
 	return e;
 
 err:
+	free(p);
 	free(e->sect);
 	free(e->prog);
 	free(e->shstrtab);
-- 
cgit v1.2.3