about summary refs log tree commit diff
path: root/src
diff options
context:
space:
mode:
authorh1994st <h1994st@gmail.com>2020-03-28 00:52:29 -0400
committerDominik Maier <domenukk@gmail.com>2020-04-01 13:10:07 +0200
commit0dd8ed9171cdbee3360f4b7f6a9fd91e7478a508 (patch)
tree1526ed8333ad947577f23127572931da088b1498 /src
parentd568559f01b1a7609f8a0c4f7afea513375725e4 (diff)
downloadafl++-0dd8ed9171cdbee3360f4b7f6a9fd91e7478a508.tar.gz
Fix invalid memory access bug in `afl_custom_pre_save` of example.c
Diffstat (limited to 'src')
-rw-r--r--src/afl-fuzz-python.c3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/afl-fuzz-python.c b/src/afl-fuzz-python.c
index 91e5b084..01503d2c 100644
--- a/src/afl-fuzz-python.c
+++ b/src/afl-fuzz-python.c
@@ -133,8 +133,8 @@ static py_mutator_t *init_py_module(afl_state_t *afl, u8 *module_name) {
   if (py_module != NULL) {
 
     u8 py_notrim = 0, py_idx;
+    /* init, required */
     py_functions[PY_FUNC_INIT] = PyObject_GetAttrString(py_module, "init");
-    py_functions[PY_FUNC_DEINIT] = PyObject_GetAttrString(py_module, "deinit");
     py_functions[PY_FUNC_FUZZ] = PyObject_GetAttrString(py_module, "fuzz");
     py_functions[PY_FUNC_PRE_SAVE] =
         PyObject_GetAttrString(py_module, "pre_save");
@@ -151,6 +151,7 @@ static py_mutator_t *init_py_module(afl_state_t *afl, u8 *module_name) {
         PyObject_GetAttrString(py_module, "queue_get");
     py_functions[PY_FUNC_QUEUE_NEW_ENTRY] =
         PyObject_GetAttrString(py_module, "queue_new_entry");
+    py_functions[PY_FUNC_DEINIT] = PyObject_GetAttrString(py_module, "deinit");
 
     for (py_idx = 0; py_idx < PY_FUNC_COUNT; ++py_idx) {