From 4f6ec6cb081bfee7a6e1d6ac211b33a820ec2f71 Mon Sep 17 00:00:00 2001
From: Sergej Schumilo <sergej@schumilo.de>
Date: Fri, 14 Apr 2023 06:21:43 +0200
Subject: [PATCH] add NYX_REUSE_SNAPSHOT env-var option
---
src/afl-forkserver.c | 16 ++++++++++++++++
1 file changed, 16 insertions(+)
diff --git a/src/afl-forkserver.c b/src/afl-forkserver.c
index 33c46b8c..7eb2155e 100644
--- a/src/afl-forkserver.c
+++ b/src/afl-forkserver.c
@@ -509,6 +509,22 @@ void afl_fsrv_start(afl_forkserver_t *fsrv, char **argv,
}
}
+ if (getenv("NYX_REUSE_SNAPSHOT") != NULL){
+
+ if (access(getenv("NYX_REUSE_SNAPSHOT"), F_OK) == -1) {
+ FATAL("NYX_REUSE_SNAPSHOT path does not exist");
+ }
+
+ /* stupid sanity check to avoid passing an empty or invalid snapshot directory */
+ char* snapshot_file_path = alloc_printf("%s/global.state", getenv("NYX_REUSE_SNAPSHOT"));
+ if (access(snapshot_file_path, R_OK) == -1) {
+ FATAL("NYX_REUSE_SNAPSHOT path does not contain a valid Nyx snapshot");
+ }
+ free(snapshot_file_path);
+
+ fsrv->nyx_handlers->nyx_config_set_reuse_snapshot_path(nyx_config, getenv("NYX_REUSE_SNAPSHOT"));
+ }
+
fsrv->nyx_runner = fsrv->nyx_handlers->nyx_new(nyx_config, fsrv->nyx_bind_cpu_id);
ck_free(x);
--
GitLab