upstream/mercurial-mirror Commit - r45488:ee5f27d7

1

#include "pyutil.h"

1

#include "pyutil.h"

2

3

#include <iostream>

3

#include <iostream>

4

#include <string>

4

#include <string>

5

6

namespace contrib

6

namespace contrib

7

{

7

{

8

9

#if PY_MAJOR_VERSION >= 3

9

#if PY_MAJOR_VERSION >= 3

10

#define HG_FUZZER_PY3 1

10

#define HG_FUZZER_PY3 1

11

PyMODINIT_FUNC PyInit_parsers(void);

11

PyMODINIT_FUNC PyInit_parsers(void);

12

#else

12

#else

13

PyMODINIT_FUNC initparsers(void);

13

PyMODINIT_FUNC initparsers(void);

14

#endif

14

#endif

15

16

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

16

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

17

18

static PyObject *mainmod;

18

static PyObject *mainmod;

19

static PyObject *globals;

19

static PyObject *globals;

20

21

void initpy(const char *cselfpath)

21

void initpy(const char *cselfpath)

22

{

22

{

23

#ifdef HG_FUZZER_PY3

23

#ifdef HG_FUZZER_PY3

24

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

24

const std::string subdir = "/sanpy/lib/python3.8";

25

#else

25

#else

26

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

26

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

27

#endif

27

#endif

28

29

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

29

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

30

pymalloc and with ASAN, so we dump one in

30

pymalloc and with ASAN, so we dump one in

31

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

31

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

32

std::string selfpath(cselfpath);

32

std::string selfpath(cselfpath);

33

std::string pypath;

33

std::string pypath;

34

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

34

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

35

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

35

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

36

char wd[8192];

36

char wd[8192];

37

if (!getcwd(wd, 8192)) {

37

if (!getcwd(wd, 8192)) {

38

std::cerr << "Failed to call getcwd: errno " << errno

38

std::cerr << "Failed to call getcwd: errno " << errno

39

<< std::endl;

39

<< std::endl;

40

exit(1);

40

exit(1);

41

}

41

}

42

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

42

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

43

} else {

43

} else {

44

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

44

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

45

}

45

}

46

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

46

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

47

setenv("PYTHONPATH", cpypath, 1);

47

setenv("PYTHONPATH", cpypath, 1);

48

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

48

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

49

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

49

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

50

setenv("PYTHONUSERBASE", cpypath, 1);

50

setenv("PYTHONUSERBASE", cpypath, 1);

51

#ifdef HG_FUZZER_PY3

51

#ifdef HG_FUZZER_PY3

52

std::wstring wcpypath(pypath.begin(), pypath.end());

52

std::wstring wcpypath(pypath.begin(), pypath.end());

53

Py_SetPythonHome(wcpypath.c_str());

53

Py_SetPythonHome(wcpypath.c_str());

54

#else

54

#else

55

Py_SetPythonHome(cpypath);

55

Py_SetPythonHome(cpypath);

56

#endif

56

#endif

57

Py_InitializeEx(0);

57

Py_InitializeEx(0);

58

mainmod = PyImport_AddModule("__main__");

58

mainmod = PyImport_AddModule("__main__");

59

globals = PyModule_GetDict(mainmod);

59

globals = PyModule_GetDict(mainmod);

60

61

#ifdef HG_FUZZER_PY3

61

#ifdef HG_FUZZER_PY3

62

PyObject *mod = PyInit_parsers();

62

PyObject *mod = PyInit_parsers();

63

#else

63

#else

64

initparsers();

64

initparsers();

65

PyObject *mod = PyImport_ImportModule("parsers");

65

PyObject *mod = PyImport_ImportModule("parsers");

66

#endif

66

#endif

67

68

PyDict_SetItemString(globals, "parsers", mod);

68

PyDict_SetItemString(globals, "parsers", mod);

69

}

69

}

70

71

PyObject *pyglobals()

71

PyObject *pyglobals()

72

{

72

{

73

return globals;

73

return globals;

74

}

74

}

75

76

} // namespace contrib

76

} // 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 <iostream>
             #include <string>
             namespace contrib
             {
             #if PY_MAJOR_VERSION >= 3
             #define HG_FUZZER_PY3 1
             PyMODINIT_FUNC PyInit_parsers(void);
             #else
             PyMODINIT_FUNC initparsers(void);
             #endif
             static char cpypath[8192] = "\0";
             static PyObject *mainmod;
             static PyObject *globals;
             void initpy(const char *cselfpath)
             {
             #ifdef HG_FUZZER_PY3
-            	const std::string subdir = "/sanpy/lib/python3.7";
+            	const std::string subdir = "/sanpy/lib/python3.8";
             #else
             	const std::string subdir = "/sanpy/lib/python2.7";
             #endif
             	/* 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];
             		if (!getcwd(wd, 8192)) {
             			std::cerr << "Failed to call getcwd: errno " << errno
             			          << std::endl;
             			exit(1);
             		}
             		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);
             #ifdef HG_FUZZER_PY3
             	std::wstring wcpypath(pypath.begin(), pypath.end());
             	Py_SetPythonHome(wcpypath.c_str());
             #else
             	Py_SetPythonHome(cpypath);
             #endif
             	Py_InitializeEx(0);
             	mainmod = PyImport_AddModule("__main__");
             	globals = PyModule_GetDict(mainmod);
             #ifdef HG_FUZZER_PY3
             	PyObject *mod = PyInit_parsers();
             #else
             	initparsers();
             	PyObject *mod = PyImport_ImportModule("parsers");
             #endif
             	PyDict_SetItemString(globals, "parsers", mod);
             }
             PyObject *pyglobals()
             {
             	return globals;
             }
             } // namespace contrib