diff options
| author | rish9101 <rranjan@cs.iitr.ac.in> | 2020-05-13 19:46:30 +0530 |
|---|---|---|
| committer | rish9101 <rranjan@cs.iitr.ac.in> | 2020-05-13 19:46:30 +0530 |
| commit | 645e331559d01d234be231a7f5076c2ad658b159 (patch) | |
| tree | c039c7192a3aa9f83ac9f7dde9c68dac01f3f607 /src/afl-fuzz-python.c | |
| parent | 45bddcd8082eedeeb09500d764c540bc02ec0d74 (diff) | |
| download | afl++-645e331559d01d234be231a7f5076c2ad658b159.tar.gz | |
Fix previous commit bugs
Diffstat (limited to 'src/afl-fuzz-python.c')
| -rw-r--r-- | src/afl-fuzz-python.c | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/afl-fuzz-python.c b/src/afl-fuzz-python.c index 984561fd..2569cdaf 100644 --- a/src/afl-fuzz-python.c +++ b/src/afl-fuzz-python.c @@ -142,7 +142,7 @@ static py_mutator_t *init_py_module(afl_state_t *afl, u8 *module_name) { py_functions[PY_FUNC_FUZZ] = PyObject_GetAttrString(py_module, "mutate"); if (!py_functions[PY_FUNC_FUZZ]) WARNF("fuzz function not found in python module"); - py_functions[PY_FUNC_post_process] = + py_functions[PY_FUNC_POST_PROCESS] = PyObject_GetAttrString(py_module, "post_process"); py_functions[PY_FUNC_INIT_TRIM] = PyObject_GetAttrString(py_module, "init_trim"); @@ -165,7 +165,7 @@ static py_mutator_t *init_py_module(afl_state_t *afl, u8 *module_name) { if (!py_functions[py_idx] || !PyCallable_Check(py_functions[py_idx])) { - if (py_idx == PY_FUNC_post_process) { + if (py_idx == PY_FUNC_POST_PROCESS) { // Implenting the post_process API is optional for now if (PyErr_Occurred()) { PyErr_Print(); } @@ -330,7 +330,7 @@ struct custom_mutator *load_custom_mutator_py(afl_state_t *afl, is quite different from the custom mutator. */ mutator->afl_custom_fuzz = fuzz_py; - if (py_functions[PY_FUNC_post_process]) { + if (py_functions[PY_FUNC_POST_PROCESS]) { mutator->afl_custom_post_process = post_process_py; @@ -402,7 +402,7 @@ size_t post_process_py(void *py_mutator, u8 *buf, size_t buf_size, u8 **out_buf) PyTuple_SetItem(py_args, 0, py_value); py_value = PyObject_CallObject( - ((py_mutator_t *)py_mutator)->py_functions[PY_FUNC_post_process], py_args); + ((py_mutator_t *)py_mutator)->py_functions[PY_FUNC_POST_PROCESS], py_args); Py_DECREF(py_args); |