diff options
| author | van Hauser <vh@thc.org> | 2019-12-03 10:11:39 +0100 |
|---|---|---|
| committer | van Hauser <vh@thc.org> | 2019-12-03 10:11:39 +0100 |
| commit | 674fbc39f8bcac77a543693edace00c843fc411b (patch) | |
| tree | f039d2b98af5d3f8b04e543da17cac9b95a33f26 /src | |
| parent | 2b0cfe1ab53f2a43db098a66e8b585420352677d (diff) | |
| download | afl++-674fbc39f8bcac77a543693edace00c843fc411b.tar.gz | |
python3.7 support for afl-fuzz py mutator
Diffstat (limited to 'src')
| -rw-r--r-- | src/afl-fuzz-python.c | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/src/afl-fuzz-python.c b/src/afl-fuzz-python.c index 70a547d2..c1ee143b 100644 --- a/src/afl-fuzz-python.c +++ b/src/afl-fuzz-python.c @@ -35,7 +35,11 @@ int init_py() { if (module_name) { +#if PY_MAJOR_VERSION >= 3 + PyObject* py_name = PyUnicode_FromString(module_name); +#else PyObject* py_name = PyString_FromString(module_name); +#endif py_module = PyImport_Import(py_name); Py_DECREF(py_name); @@ -91,7 +95,12 @@ int init_py() { /* Provide the init function a seed for the Python RNG */ py_args = PyTuple_New(1); +#if PY_MAJOR_VERSION >= 3 + py_value = PyLong_FromLong(UR(0xFFFFFFFF)); +#else py_value = PyInt_FromLong(UR(0xFFFFFFFF)); +#endif + if (!py_value) { Py_DECREF(py_args); @@ -216,7 +225,11 @@ u32 init_trim_py(char* buf, size_t buflen) { if (py_value != NULL) { +#if PY_MAJOR_VERSION >= 3 + u32 retcnt = (u32) PyLong_AsLong(py_value); +#else u32 retcnt = PyInt_AsLong(py_value); +#endif Py_DECREF(py_value); return retcnt; @@ -250,7 +263,11 @@ u32 post_trim_py(char success) { if (py_value != NULL) { +#if PY_MAJOR_VERSION >= 3 + u32 retcnt = (u32) PyLong_AsLong(py_value); +#else u32 retcnt = PyInt_AsLong(py_value); +#endif Py_DECREF(py_value); return retcnt; |