upstream/mercurial-mirror Files · contrib/fuzz/jsonescapeu8fast.cc

tests: use sha256line.py instead of /dev/random in test-censor.t (issue6858)...

tests: use sha256line.py instead of /dev/random in test-censor.t (issue6858) Sometimes the systems that run our test suite don't have enough entropy and they cannot produce target file of the expected size using /dev/random, which results in test failures. Switching to /dev/urandom would give us way more available data at the cost of it being less "random", but we don't really need to use entropy for this task at all, since we only care if the file size after compression is big enough to not be stored inline in the revlog. So let's use something that we already have used to generate this kind of data in other tests.

Augie Fackler - - Load All Authors

File last commit:

r44311:8766728d default


                r52255:e7be2ddf

stable

Download file

             jsonescapeu8fast.cc
        
                    55 lines
            
             | 1.4 KiB
            
                | text/x-c
            
             |
                CppLexer
            
             / contrib / fuzz / jsonescapeu8fast.cc
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      #include <Python.h>

      #include <assert.h>

      #include <stdlib.h>

      #include <unistd.h>

      #include "pyutil.h"

      #include <iostream>

      #include <string>

      #include "FuzzedDataProvider.h"

      extern "C" {

      static PYCODETYPE *code;

      extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv)

      {

      	contrib::initpy(*argv[0]);

      	code = (PYCODETYPE *)Py_CompileString(R"py(

      try:

          parsers.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.

      }

      }

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				#include <Python.h>
				#include <assert.h>
				#include <stdlib.h>
				#include <unistd.h>

				#include "pyutil.h"

				#include <iostream>
				#include <string>
				#include "FuzzedDataProvider.h"

				extern "C" {

				static PYCODETYPE *code;

				extern "C" int LLVMFuzzerInitialize(int argc, char **argv)
				{
				contrib::initpy(*argv[0]);
				code = (PYCODETYPE *)Py_CompileString(R"py(
				try:
				parsers.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.
				}
				}