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

node: manually implement Debug...

node: manually implement Debug I got too irritated today with the default Debug implementation of hg::revlog::Node while playing with a new parser. This isn't quite what I wanted, but it wasn't much code and it at least gives you output that's easy to visually compare to a node.hex()ed identifier from the Python side of things. Sadly, this doesn't influence the output in lldb or the VSCode debugger extension that uses lldb under the covers, but it at least means debug prints are a little more useful. Differential Revision: https://phab.mercurial-scm.org/D12608

Augie Fackler - - Load All Authors

File last commit:

r44311:8766728d default


                r50070:34decbaf

default

Download file

             dirstate.cc
        
                    47 lines
            
             | 1.0 KiB
            
                | text/x-c
            
             |
                CppLexer
            
             / contrib / fuzz / dirstate.cc
          
                    History
                
                 |
                  Source
                 | Raw
                 |Copy content
                 |Copy permalink

        Augie Fackler
    
fuzz: new fuzzer for dirstate parser...

              r41051
            
      #include <Python.h>

      #include <assert.h>

      #include <stdlib.h>

      #include <unistd.h>

      #include <string>

      #include "pyutil.h"

      extern "C" {

        Augie Fackler
    
fuzz: add support for fuzzing under either Python 2 or 3...

              r44311
            
      static PYCODETYPE *code;

        Augie Fackler
    
fuzz: new fuzzer for dirstate parser...

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

      {

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

        Augie Fackler
    
fuzz: add support for fuzzing under either Python 2 or 3...

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

        Augie Fackler
    
fuzz: new fuzzer for dirstate parser...

              r41051
            
      try:

          dmap = {}

          copymap = {}

        Augie Fackler
    
fuzz: add support for fuzzing under either Python 2 or 3...

              r44311
            
          p = parsers.parse_dirstate(dmap, copymap, data)

        Augie Fackler
    
fuzz: new fuzzer for dirstate parser...

              r41051
            
      except Exception as e:

          pass

          # uncomment this print if you're editing this Python code

          # to debug failures.

          # print e

      )py",

        Augie Fackler
    
fuzz: add support for fuzzing under either Python 2 or 3...

              r44311
            
      	                                      "fuzzer", Py_file_input);

        Augie Fackler
    
fuzz: new fuzzer for dirstate parser...

              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.

      }

      }

	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

Augie Fackler fuzz: new fuzzer for dirstate parser...	r41051	#include <Python.h>
		#include <assert.h>
		#include <stdlib.h>
		#include <unistd.h>

		#include <string>

		#include "pyutil.h"

		extern "C" {

Augie Fackler fuzz: add support for fuzzing under either Python 2 or 3...	r44311	static PYCODETYPE *code;
Augie Fackler fuzz: new fuzzer for dirstate parser...	r41051
		extern "C" int LLVMFuzzerInitialize(int argc, char **argv)
		{
		contrib::initpy(*argv[0]);
Augie Fackler fuzz: add support for fuzzing under either Python 2 or 3...	r44311	code = (PYCODETYPE *)Py_CompileString(R"py(
Augie Fackler fuzz: new fuzzer for dirstate parser...	r41051	try:
		dmap = {}
		copymap = {}
Augie Fackler fuzz: add support for fuzzing under either Python 2 or 3...	r44311	p = parsers.parse_dirstate(dmap, copymap, data)
Augie Fackler fuzz: new fuzzer for dirstate parser...	r41051	except Exception as e:
		pass
		# uncomment this print if you're editing this Python code
		# to debug failures.
		# print e
		)py",
Augie Fackler fuzz: add support for fuzzing under either Python 2 or 3...	r44311	"fuzzer", Py_file_input);
Augie Fackler fuzz: new fuzzer for dirstate parser...	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.
		}
		}