about summary refs log tree commit diff
diff options
context:
space:
mode:
authorh1994st <h1994st@gmail.com>2020-03-29 01:07:29 -0400
committerDominik Maier <domenukk@gmail.com>2020-04-01 13:10:07 +0200
commit50fc7327f2556db588c65fe7941198b622ab50af (patch)
tree956aa306628836866a455aed3198d97c86c4124d
parent71edae4a0fd7ca64a6f2c87768d14136ac04b0a1 (diff)
downloadafl++-50fc7327f2556db588c65fe7941198b622ab50af.tar.gz
python mutator: fix nullptr for python mutator data
-rw-r--r--include/afl-fuzz.h3
-rw-r--r--src/afl-fuzz-python.c1
2 files changed, 1 insertions, 3 deletions
diff --git a/include/afl-fuzz.h b/include/afl-fuzz.h
index a265c1a3..fcbc09e5 100644
--- a/include/afl-fuzz.h
+++ b/include/afl-fuzz.h
@@ -566,9 +566,6 @@ typedef struct afl_state {
 
   /* Custom mutators */
   struct custom_mutator *mutator;
-#ifdef USE_PYTHON
-  struct custom_mutator *py_mutator;
-#endif
 
   /* cmplog forkserver ids */
   s32 cmplog_fsrv_ctl_fd, cmplog_fsrv_st_fd;
diff --git a/src/afl-fuzz-python.c b/src/afl-fuzz-python.c
index 01503d2c..12c3a09d 100644
--- a/src/afl-fuzz-python.c
+++ b/src/afl-fuzz-python.c
@@ -303,6 +303,7 @@ void load_custom_mutator_py(afl_state_t *afl, char *module_name) {
 
   py_mutator_t *py_mutator;
   py_mutator = init_py_module(afl, module_name);
+  afl->mutator->data = py_mutator;
   if (!py_mutator) { FATAL("Failed to load python mutator."); }
 
   PyObject **py_functions = py_mutator->py_functions;