From fb8fb7fed0bb822ccc11ed20229db51a3991a0e5 Mon Sep 17 00:00:00 2001
From: Matt Johnston <matt@ucc.asn.au>
Date: Sun, 21 May 2017 10:54:11 +0800
Subject: add dbmalloc epoch cleanup

--HG--
branch : fuzz
---
 fuzzer-preauth.c | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'fuzzer-preauth.c')

diff --git a/fuzzer-preauth.c b/fuzzer-preauth.c
index 9ca7b84..7f0e136 100644
--- a/fuzzer-preauth.c
+++ b/fuzzer-preauth.c
@@ -23,9 +23,11 @@ int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
 	int fakesock = 1;
 	wrapfd_add(fakesock, fuzz.input, PLAIN);
 
+	m_malloc_set_epoch(1);
 	if (setjmp(fuzz.jmp) == 0) {
 		svr_session(fakesock, fakesock);
 	} else {
+		m_malloc_free_epoch(1);
 		TRACE(("dropbear_exit longjmped"))
 		// dropbear_exit jumped here
 	}
-- 
cgit v1.2.3