diff --git a/src/sim/init.cc b/src/sim/init.cc index 50612895b3..66ec408959 100644 --- a/src/sim/init.cc +++ b/src/sim/init.cc @@ -51,6 +51,7 @@ #include #include #include +#include #include "base/cprintf.hh" #include "base/logging.hh" @@ -249,7 +250,7 @@ const char * __attribute__((weak)) m5MainCommands[] = { * main function. */ int -m5Main(int argc, char **argv) +m5Main(int argc, char **_argv) { #if HAVE_PROTOBUF // Verify that the version of the protobuf library that we linked @@ -258,6 +259,23 @@ m5Main(int argc, char **argv) GOOGLE_PROTOBUF_VERIFY_VERSION; #endif + +#if PY_MAJOR_VERSION >= 3 + typedef std::unique_ptr WArgUPtr; + std::vector v_argv; + std::vector vp_argv; + v_argv.reserve(argc); + vp_argv.reserve(argc); + for (int i = 0; i < argc; i++) { + v_argv.emplace_back(Py_DecodeLocale(_argv[i], NULL), &PyMem_RawFree); + vp_argv.emplace_back(v_argv.back().get()); + } + + wchar_t **argv = vp_argv.data(); +#else + char **argv = _argv; +#endif + PySys_SetArgv(argc, argv); // We have to set things up in the special __main__ module diff --git a/src/sim/main.cc b/src/sim/main.cc index 48b1592638..a77c5f578f 100644 --- a/src/sim/main.cc +++ b/src/sim/main.cc @@ -45,7 +45,14 @@ main(int argc, char **argv) // Initialize m5 special signal handling. initSignals(); +#if PY_MAJOR_VERSION >= 3 + std::unique_ptr program( + Py_DecodeLocale(argv[0], NULL), + &PyMem_RawFree); + Py_SetProgramName(program.get()); +#else Py_SetProgramName(argv[0]); +#endif // initialize embedded Python interpreter Py_Initialize();