diff options
| author | van Hauser <vh@thc.org> | 2020-08-24 18:06:07 +0200 |
|---|---|---|
| committer | van Hauser <vh@thc.org> | 2020-08-24 18:06:07 +0200 |
| commit | 6a34c5aa3e0a1da06a5268ecabfc92ab36ce96dc (patch) | |
| tree | 0e770e13df70dbeda16cfca2141ddf5e4a2d59a0 /src | |
| parent | c7f0d3066875bca0ec28e9429df40293339dc05c (diff) | |
| download | afl++-6a34c5aa3e0a1da06a5268ecabfc92ab36ce96dc.tar.gz | |
fix python implementation for new function
Diffstat (limited to 'src')
| -rw-r--r-- | src/afl-fuzz-python.c | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/afl-fuzz-python.c b/src/afl-fuzz-python.c index 68540dd7..32c8ac89 100644 --- a/src/afl-fuzz-python.c +++ b/src/afl-fuzz-python.c @@ -144,6 +144,7 @@ static py_mutator_t *init_py_module(afl_state_t *afl, u8 *module_name) { py_functions[PY_FUNC_FUZZ] = PyObject_GetAttrString(py_module, "fuzz"); if (!py_functions[PY_FUNC_FUZZ]) py_functions[PY_FUNC_FUZZ] = PyObject_GetAttrString(py_module, "mutate"); + py_functions[PY_FUNC_FUZZ_COUNT] = PyObject_GetAttrString(py_module, "fuzz_count"); if (!py_functions[PY_FUNC_FUZZ]) WARNF("fuzz function not found in python module"); py_functions[PY_FUNC_POST_PROCESS] = @@ -187,7 +188,7 @@ static py_mutator_t *init_py_module(afl_state_t *afl, u8 *module_name) { // Implenting the havoc and queue API is optional for now if (PyErr_Occurred()) { PyErr_Print(); } - } else { + } else if (py_idx != PY_FUNC_FUZZ_COUNT) { if (PyErr_Occurred()) { PyErr_Print(); } fprintf(stderr, |