From e88509ac224e21738063c25a86e32f57e655f2dd Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Wed, 17 Nov 2021 20:27:45 +0000
Subject: Add __afl_set_persistent_mode

---
 frida_mode/src/js/js_api.c | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

(limited to 'frida_mode/src/js/js_api.c')

diff --git a/frida_mode/src/js/js_api.c b/frida_mode/src/js/js_api.c
index 102423d9..4221fb80 100644
--- a/frida_mode/src/js/js_api.c
+++ b/frida_mode/src/js/js_api.c
@@ -11,6 +11,10 @@
 #include "stats.h"
 #include "util.h"
 
+typedef uint8_t u8;
+
+extern void __afl_set_persistent_mode(u8 mode);
+
 __attribute__((visibility("default"))) void js_api_done() {
 
   js_done = TRUE;
@@ -47,13 +51,7 @@ __attribute__((visibility("default"))) void js_api_set_persistent_address(
 
   persistent_start = GPOINTER_TO_SIZE(address);
 
-  if (getenv("__AFL_PERSISTENT") == NULL) {
-
-    FATAL(
-        "You must set __AFL_PERSISTENT manually if using persistent mode "
-        "configured using JS");
-
-  }
+  __afl_set_persistent_mode(1);
 
 }
 
-- 
cgit 1.4.1