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

py3: manually import pycompat.open into files that need it...

py3: manually import pycompat.open into files that need it We want to eliminate the source transformer. Currently it inserts a `from mercurial.pycompat import ...` at the top of files to alias some builtins. This commit replaces the implicit import of `open` with an explicit import on files that need it and changes the source transformer to no longer import `open`. As part of this, we needed to store an explicit local for `open` in the Python 2 code path in `pycompat` so the import works. (Builtins that are automatically in scope cannot be imported.) Differential Revision: https://phab.mercurial-scm.org/D7005

Augie Fackler - - Load All Authors

File last commit:

r41049:ef103c96 default


                r43355:eef9a2d6

default

Download file

             pyutil.cc
        
                    49 lines
            
             | 1.2 KiB
            
                | text/x-c
            
             |
                CppLexer
            
             / contrib / fuzz / pyutil.cc
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      #include "pyutil.h"

      #include <string>

      namespace contrib

      {

      static char cpypath[8192] = "\0";

      static PyObject *mainmod;

      static PyObject *globals;

      /* TODO: use Python 3 for this fuzzing? */

      PyMODINIT_FUNC initparsers(void);

      void initpy(const char *cselfpath)

      {

      	const std::string subdir = "/sanpy/lib/python2.7";

      	/* HACK ALERT: we need a full Python installation built without

      	   pymalloc and with ASAN, so we dump one in

      	   $OUT/sanpy/lib/python2.7. This helps us wire that up. */

      	std::string selfpath(cselfpath);

      	std::string pypath;

      	auto pos = selfpath.rfind("/");

      	if (pos == std::string::npos) {

      		char wd[8192];

      		getcwd(wd, 8192);

      		pypath = std::string(wd) + subdir;

      	} else {

      		pypath = selfpath.substr(0, pos) + subdir;

      	}

      	strncpy(cpypath, pypath.c_str(), pypath.size());

      	setenv("PYTHONPATH", cpypath, 1);

      	setenv("PYTHONNOUSERSITE", "1", 1);

      	/* prevent Python from looking up users in the fuzz environment */

      	setenv("PYTHONUSERBASE", cpypath, 1);

      	Py_SetPythonHome(cpypath);

      	Py_InitializeEx(0);

      	mainmod = PyImport_AddModule("__main__");

      	globals = PyModule_GetDict(mainmod);

      	initparsers();

      }

      PyObject *pyglobals()

      {

      	return globals;

      }

      } // namespace contrib

	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 "pyutil.h"

				#include <string>

				namespace contrib
				{

				static char cpypath[8192] = "\0";

				static PyObject *mainmod;
				static PyObject *globals;

				/* TODO: use Python 3 for this fuzzing? */
				PyMODINIT_FUNC initparsers(void);

				void initpy(const char *cselfpath)
				{
				const std::string subdir = "/sanpy/lib/python2.7";
				/* HACK ALERT: we need a full Python installation built without
				pymalloc and with ASAN, so we dump one in
				$OUT/sanpy/lib/python2.7. This helps us wire that up. */
				std::string selfpath(cselfpath);
				std::string pypath;
				auto pos = selfpath.rfind("/");
				if (pos == std::string::npos) {
				char wd[8192];
				getcwd(wd, 8192);
				pypath = std::string(wd) + subdir;
				} else {
				pypath = selfpath.substr(0, pos) + subdir;
				}
				strncpy(cpypath, pypath.c_str(), pypath.size());
				setenv("PYTHONPATH", cpypath, 1);
				setenv("PYTHONNOUSERSITE", "1", 1);
				/* prevent Python from looking up users in the fuzz environment */
				setenv("PYTHONUSERBASE", cpypath, 1);
				Py_SetPythonHome(cpypath);
				Py_InitializeEx(0);
				mainmod = PyImport_AddModule("__main__");
				globals = PyModule_GetDict(mainmod);
				initparsers();
				}

				PyObject *pyglobals()
				{
				return globals;
				}

				} // namespace contrib