diff -r b37dd26935ee -r 741fb1a95da2 contrib/fuzz/jsonescapeu8fast.cc --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/contrib/fuzz/jsonescapeu8fast.cc Wed Oct 09 20:49:58 2019 -0700 @@ -0,0 +1,57 @@ +#include +#include +#include +#include + +#include "pyutil.h" + +#include +#include +#include + +extern "C" { + +static PyCodeObject *code; + +extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) +{ + contrib::initpy(*argv[0]); + code = (PyCodeObject *)Py_CompileString(R"py( +from parsers import jsonescapeu8fast + +try: + jsonescapeu8fast(data, paranoid) +except Exception as e: + pass + # uncomment this print if you're editing this Python code + # to debug failures. + # print(e) +)py", + "fuzzer", Py_file_input); + if (!code) { + std::cerr << "failed to compile Python code!" << std::endl; + } + return 0; +} + +int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) +{ + FuzzedDataProvider provider(Data, Size); + bool paranoid = provider.ConsumeBool(); + std::string remainder = provider.ConsumeRemainingBytesAsString(); + + PyObject *mtext = PyBytes_FromStringAndSize( + (const char *)remainder.c_str(), remainder.size()); + PyObject *locals = PyDict_New(); + PyDict_SetItemString(locals, "data", mtext); + PyDict_SetItemString(locals, "paranoid", paranoid ? Py_True : Py_False); + PyObject *res = PyEval_EvalCode(code, contrib::pyglobals(), locals); + if (!res) { + PyErr_Print(); + } + Py_XDECREF(res); + Py_DECREF(locals); + Py_DECREF(mtext); + return 0; // Non-zero return values are reserved for future use. +} +}