ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-14 02:58:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix previous commit bugs

Browse Source
This commit is contained in:
rish9101
2020-05-13 19:46:30 +05:30
parent 45bddcd808
commit 645e331559
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/afl-fuzz-init.c
View File

@ -314,7 +314,6 @@ void setup_post(afl_state_t *afl) {
/* Do a quick test. It's better to segfault now than later =) */ /* Do a quick test. It's better to segfault now than later =) */
u8 *post_buf = NULL;
mutator->data = mutator->afl_custom_init(afl, rand_below(afl, 0xFFFFFFFF)); mutator->data = mutator->afl_custom_init(afl, rand_below(afl, 0xFFFFFFFF));
if (!mutator->data) { FATAL("Could not initialize post handler."); } if (!mutator->data) { FATAL("Could not initialize post handler."); }

8
src/afl-fuzz-python.c
View File

@ -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"); py_functions[PY_FUNC_FUZZ] = PyObject_GetAttrString(py_module, "mutate");
if (!py_functions[PY_FUNC_FUZZ]) if (!py_functions[PY_FUNC_FUZZ])
WARNF("fuzz function not found in python module"); 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"); PyObject_GetAttrString(py_module, "post_process");
py_functions[PY_FUNC_INIT_TRIM] = py_functions[PY_FUNC_INIT_TRIM] =
PyObject_GetAttrString(py_module, "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_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 // Implenting the post_process API is optional for now
if (PyErr_Occurred()) { PyErr_Print(); } 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. */ is quite different from the custom mutator. */
mutator->afl_custom_fuzz = fuzz_py; 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; 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); PyTuple_SetItem(py_args, 0, py_value);
py_value = PyObject_CallObject( 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); Py_DECREF(py_args);