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

support custom mutator introspection

Browse Source
This commit is contained in:
van Hauser
2020-11-10 14:08:21 +01:00
parent 1661303248
commit 8e1047f5ef
7 changed files with 159 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -163,6 +163,8 @@ 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_INTROSPECTION] =
PyObject_GetAttrString(py_module, "introspection");
py_functions[PY_FUNC_DEINIT] = PyObject_GetAttrString(py_module, "deinit");
if (!py_functions[PY_FUNC_DEINIT])
FATAL("deinit function not found in python module");
@ -381,6 +383,15 @@ struct custom_mutator *load_custom_mutator_py(afl_state_t *afl,
}
#ifdef INTROSPECTION
if (py_functions[PY_FUNC_INTROSPECTION]) {
mutator->afl_custom_introspection = introspection_py;
}
#endif
OKF("Python mutator '%s' installed successfully.", module_name);
/* Initialize the custom mutator */
@ -679,6 +690,28 @@ u8 havoc_mutation_probability_py(void *py_mutator) {
}
const char *introspection_py(void *py_mutator) {
PyObject *py_args, *py_value;
py_args = PyTuple_New(0);
py_value = PyObject_CallObject(
((py_mutator_t *)py_mutator)->py_functions[PY_FUNC_INTROSPECTION],
py_args);
Py_DECREF(py_args);
if (py_value == NULL) {
return NULL;
} else {
return PyByteArray_AsString(py_value);
}
}
u8 queue_get_py(void *py_mutator, const u8 *filename) {
PyObject *py_args, *py_value;