1 #include <Python.h>
2 #include <assert.h>
3 #include <stdlib.h>
4 #include <unistd.h>
5
6 #include <string>
7
8 #include "pyutil.h"
9
10 extern "C" {
11
12 static PYCODETYPE *code;
13
LLVMFuzzerInitialize(int * argc,char *** argv)14 extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv)
15 {
16 contrib::initpy(*argv[0]);
17 code = (PYCODETYPE *)Py_CompileString(R"py(
18 def maybeint(s, default):
19 try:
20 return int(s)
21 except ValueError:
22 return default
23 try:
24 parts = data.split('\0', 2)
25 if len(parts) == 3:
26 offset, stop, data = parts
27 elif len(parts) == 2:
28 stop, data = parts
29 offset = 0
30 else:
31 offset = stop = 0
32 offset, stop = maybeint(offset, 0), maybeint(stop, len(data))
33 parsers.fm1readmarkers(data, offset, stop)
34 except Exception as e:
35 pass
36 # uncomment this print if you're editing this Python code
37 # to debug failures.
38 # print e
39 )py",
40 "fuzzer", Py_file_input);
41 return 0;
42 }
43
LLVMFuzzerTestOneInput(const uint8_t * Data,size_t Size)44 int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size)
45 {
46 PyObject *text =
47 PyBytes_FromStringAndSize((const char *)Data, (Py_ssize_t)Size);
48 PyObject *locals = PyDict_New();
49 PyDict_SetItemString(locals, "data", text);
50 PyObject *res = PyEval_EvalCode(code, contrib::pyglobals(), locals);
51 if (!res) {
52 PyErr_Print();
53 }
54 Py_XDECREF(res);
55 Py_DECREF(locals);
56 Py_DECREF(text);
57 return 0; // Non-zero return values are reserved for future use.
58 }
59 }
60