about summary refs log tree commit diff
path: root/src/afl-fuzz-python.c
diff options
context:
space:
mode:
authorh1994st <h1994st@gmail.com>2020-03-02 15:27:29 -0500
committerh1994st <h1994st@gmail.com>2020-03-02 15:27:29 -0500
commit7b59e05600aff4e9deb3110887d6ad49838f02b0 (patch)
tree240b3fecfdf8999593cbfba541e3a059f2409a9a /src/afl-fuzz-python.c
parent95322c11d978ac80d8b841ea8270780fd1b3eb55 (diff)
downloadafl++-7b59e05600aff4e9deb3110887d6ad49838f02b0.tar.gz
Add new APIs for the custom mutator
Diffstat (limited to 'src/afl-fuzz-python.c')
-rw-r--r--src/afl-fuzz-python.c2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/afl-fuzz-python.c b/src/afl-fuzz-python.c
index 42286527..be86ebba 100644
--- a/src/afl-fuzz-python.c
+++ b/src/afl-fuzz-python.c
@@ -55,6 +55,8 @@ int init_py() {
       u8 py_notrim = 0, py_idx;
       py_functions[PY_FUNC_INIT] = PyObject_GetAttrString(py_module, "init");
       py_functions[PY_FUNC_FUZZ] = PyObject_GetAttrString(py_module, "fuzz");
+      py_functions[PY_FUNC_PRE_SAVE] =
+          PyObject_GetAttrString(py_module, "pre_save");
       py_functions[PY_FUNC_INIT_TRIM] =
           PyObject_GetAttrString(py_module, "init_trim");
       py_functions[PY_FUNC_POST_TRIM] =