Devin Jeanpierre | c5bace2 | 2017-09-06 11:15:35 -0700 | [diff] [blame] | 1 | #define PY_SSIZE_T_CLEAN |
| 2 | #include <Python.h> |
| 3 | #include <stdlib.h> |
| 4 | #include <inttypes.h> |
| 5 | |
| 6 | int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size); |
| 7 | |
| 8 | static PyObject* _fuzz_run(PyObject* self, PyObject* args) { |
| 9 | const char* buf; |
| 10 | Py_ssize_t size; |
| 11 | if (!PyArg_ParseTuple(args, "s#", &buf, &size)) { |
| 12 | return NULL; |
| 13 | } |
| 14 | int rv = LLVMFuzzerTestOneInput((const uint8_t*)buf, size); |
| 15 | if (PyErr_Occurred()) { |
| 16 | return NULL; |
| 17 | } |
| 18 | if (rv != 0) { |
| 19 | // Nonzero return codes are reserved for future use. |
| 20 | PyErr_Format( |
| 21 | PyExc_RuntimeError, "Nonzero return code from fuzzer: %d", rv); |
| 22 | return NULL; |
| 23 | } |
| 24 | Py_RETURN_NONE; |
| 25 | } |
| 26 | |
| 27 | static PyMethodDef module_methods[] = { |
| 28 | {"run", (PyCFunction)_fuzz_run, METH_VARARGS, ""}, |
| 29 | {NULL}, |
| 30 | }; |
| 31 | |
| 32 | static struct PyModuleDef _fuzzmodule = { |
| 33 | PyModuleDef_HEAD_INIT, |
| 34 | "_fuzz", |
| 35 | NULL, |
| 36 | 0, |
| 37 | module_methods, |
| 38 | NULL, |
| 39 | NULL, |
| 40 | NULL, |
| 41 | NULL |
| 42 | }; |
| 43 | |
| 44 | PyMODINIT_FUNC |
| 45 | PyInit__xxtestfuzz(void) |
| 46 | { |
| 47 | PyObject *m = NULL; |
| 48 | |
| 49 | if ((m = PyModule_Create(&_fuzzmodule)) == NULL) { |
| 50 | return NULL; |
| 51 | } |
| 52 | return m; |
| 53 | } |