dirstate.cc
47 lines
| 1.0 KiB
| text/x-c
|
CppLexer
Augie Fackler
|
r41051 | #include <Python.h> | ||
| #include <assert.h> | ||||
| #include <stdlib.h> | ||||
| #include <unistd.h> | ||||
| #include <string> | ||||
| #include "pyutil.h" | ||||
| extern "C" { | ||||
|
Augie Fackler
|
r44311 | static PYCODETYPE *code; | ||
|
Augie Fackler
|
r41051 | |||
| extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) | ||||
| { | ||||
| contrib::initpy(*argv[0]); | ||||
|
Augie Fackler
|
r44311 | code = (PYCODETYPE *)Py_CompileString(R"py( | ||
|
Augie Fackler
|
r41051 | try: | ||
| dmap = {} | ||||
| copymap = {} | ||||
|
Augie Fackler
|
r44311 | p = parsers.parse_dirstate(dmap, copymap, data) | ||
|
Augie Fackler
|
r41051 | except Exception as e: | ||
| pass | ||||
| # uncomment this print if you're editing this Python code | ||||
| # to debug failures. | ||||
| # print e | ||||
| )py", | ||||
|
Augie Fackler
|
r44311 | "fuzzer", Py_file_input); | ||
|
Augie Fackler
|
r41051 | return 0; | ||
| } | ||||
| int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) | ||||
| { | ||||
| PyObject *text = | ||||
| PyBytes_FromStringAndSize((const char *)Data, (Py_ssize_t)Size); | ||||
| PyObject *locals = PyDict_New(); | ||||
| PyDict_SetItemString(locals, "data", text); | ||||
| PyObject *res = PyEval_EvalCode(code, contrib::pyglobals(), locals); | ||||
| if (!res) { | ||||
| PyErr_Print(); | ||||
| } | ||||
| Py_XDECREF(res); | ||||
| Py_DECREF(locals); | ||||
| Py_DECREF(text); | ||||
| return 0; // Non-zero return values are reserved for future use. | ||||
| } | ||||
| } | ||||