From 020919ba7a840547fb30b2d48d8eb26600474ac2 Mon Sep 17 00:00:00 2001
From: Jason Hiser <jdhiser@gmail.com>
Date: Sun, 15 Aug 2021 23:01:59 +0000
Subject: [PATCH] Update to allow persistent mode to work properly via
ZAFL_DRIVER_SETS_UP_MAP
---
src/libzafl.c | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/src/libzafl.c b/src/libzafl.c
index c0a36ec..95c2b01 100644
--- a/src/libzafl.c
+++ b/src/libzafl.c
@@ -279,9 +279,11 @@ static void zafl_setupAflTracemapSharedMemory()
{
if (getenv("ZAFL_DEBUG")) debug = 1;
+
if (tracemap_shared_memory_is_setup)
return;
+
zafl_prev_id = 0;
zafl_trace_map = NULL;
@@ -296,6 +298,12 @@ static void zafl_setupAflTracemapSharedMemory()
// setup the area for storing prev_id and context_id 4K away from the end of the trace map
void* vars_addr = (void*)(shmat_addr + AFL_VARS_OFFSET);
zafl_setupAflVariablesMemory(vars_addr);
+ if (getenv("ZAFL_DRIVER_SETS_UP_MAP"))
+ {
+ tracemap_shared_memory_is_setup=1;
+ zafl_trace_map=shmat_addr;
+ return;
+ }
}
char *shm_env_var = getenv(SHM_ENV_VAR);
--
GitLab