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

windows: disable pager when packaged with py2exe...

windows: disable pager when packaged with py2exe With Windows and py3, all output that got directed to the pager was lost. It can be worked around by the user piping to `more`, but that's easy to forget, and can be dangerous if `hg diff` or similar incorrectly shows no changes. The problem appears to be the new WindowsConsoleIO in py3.6[1]. We've worked around it with PyOxidizer by setting the `Py_LegacyWindowsStdioFlag` interpreter option, and worked around it with `hg.bat` and `exewrapper.c` by internally setting `PYTHONLEGACYWINDOWSSTDIO=1`. Unfortunately, py2exe doesn't appear to be able to set the interpreter option, and somehow seems to also ignore the environment variable. The latter isn't a good fix anyway, since setting it in the environment would affect other python programs too. We can't install a global config for this because a config closer to the user (e.g. from before pager was turned on by default) can override it. [1] https://peps.python.org/pep-0528/ Differential Revision: https://phab.mercurial-scm.org/D12556

Augie Fackler - - Load All Authors

File last commit:

r44311:8766728d default


                r49948:30825d15

default

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