about summary refs log tree commit diff
path: root/src/afl-sharedmem.c
diff options
context:
space:
mode:
authorvan Hauser <vh@thc.org>2021-02-01 12:01:23 +0100
committervan Hauser <vh@thc.org>2021-02-01 12:01:23 +0100
commit981ffb27a8a166b51a06d57fce044ed1eaf1aa62 (patch)
treef4ee467c9b3df7fbfc84533429717f64f7b6e4ab /src/afl-sharedmem.c
parent522eacce71a91f0495834215b08ac38750a4f5d8 (diff)
downloadafl++-981ffb27a8a166b51a06d57fce044ed1eaf1aa62.tar.gz
making AFL_MAP_SIZE obsolete
Diffstat (limited to 'src/afl-sharedmem.c')
-rw-r--r--src/afl-sharedmem.c14
1 files changed, 12 insertions, 2 deletions
diff --git a/src/afl-sharedmem.c b/src/afl-sharedmem.c
index fe641d0d..3241a130 100644
--- a/src/afl-sharedmem.c
+++ b/src/afl-sharedmem.c
@@ -66,9 +66,17 @@ static list_t shm_list = {.element_prealloc_count = 0};
 
 void afl_shm_deinit(sharedmem_t *shm) {
 
-  if (shm == NULL) return;
-
+  if (shm == NULL) { return; }
   list_remove(&shm_list, shm);
+  if (shm->shmemfuzz_mode) {
+
+    unsetenv(SHM_FUZZ_ENV_VAR);
+
+  } else {
+
+    unsetenv(SHM_ENV_VAR);
+
+  }
 
 #ifdef USEMMAP
   if (shm->map != NULL) {
@@ -94,6 +102,8 @@ void afl_shm_deinit(sharedmem_t *shm) {
 
   if (shm->cmplog_mode) {
 
+    unsetenv(CMPLOG_SHM_ENV_VAR);
+
     if (shm->cmp_map != NULL) {
 
       munmap(shm->cmp_map, shm->map_size);