mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 05:31:20 +00:00 
			
		
		
		
	 34be807ca4
			
		
	
	
		34be807ca4
		
	
	
	
	
		
			
			Issue #26516: * Add PYTHONMALLOC environment variable to set the Python memory allocators and/or install debug hooks. * PyMem_SetupDebugHooks() can now also be used on Python compiled in release mode. * The PYTHONMALLOCSTATS environment variable can now also be used on Python compiled in release mode. It now has no effect if set to an empty string. * In debug mode, debug hooks are now also installed on Python memory allocators when Python is configured without pymalloc.
		
			
				
	
	
		
			835 lines
		
	
	
	
		
			24 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			835 lines
		
	
	
	
		
			24 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* Python interpreter main program */
 | |
| 
 | |
| #include "Python.h"
 | |
| #include "osdefs.h"
 | |
| 
 | |
| #include <locale.h>
 | |
| 
 | |
| #if defined(MS_WINDOWS) || defined(__CYGWIN__)
 | |
| #include <windows.h>
 | |
| #ifdef HAVE_FCNTL_H
 | |
| #include <fcntl.h>
 | |
| #endif
 | |
| #endif
 | |
| 
 | |
| #ifdef _MSC_VER
 | |
| #include <crtdbg.h>
 | |
| #endif
 | |
| 
 | |
| #if defined(MS_WINDOWS)
 | |
| #define PYTHONHOMEHELP "<prefix>\\lib"
 | |
| #else
 | |
| #define PYTHONHOMEHELP "<prefix>/pythonX.X"
 | |
| #endif
 | |
| 
 | |
| #include "pygetopt.h"
 | |
| 
 | |
| #define COPYRIGHT \
 | |
|     "Type \"help\", \"copyright\", \"credits\" or \"license\" " \
 | |
|     "for more information."
 | |
| 
 | |
| #ifdef __cplusplus
 | |
| extern "C" {
 | |
| #endif
 | |
| 
 | |
| /* For Py_GetArgcArgv(); set by main() */
 | |
| static wchar_t **orig_argv;
 | |
| static int  orig_argc;
 | |
| 
 | |
| /* command line options */
 | |
| #define BASE_OPTS L"bBc:dEhiIJm:OqRsStuvVW:xX:?"
 | |
| 
 | |
| #define PROGRAM_OPTS BASE_OPTS
 | |
| 
 | |
| /* Short usage message (with %s for argv0) */
 | |
| static const char usage_line[] =
 | |
| "usage: %ls [option] ... [-c cmd | -m mod | file | -] [arg] ...\n";
 | |
| 
 | |
| /* Long usage message, split into parts < 512 bytes */
 | |
| static const char usage_1[] = "\
 | |
| Options and arguments (and corresponding environment variables):\n\
 | |
| -b     : issue warnings about str(bytes_instance), str(bytearray_instance)\n\
 | |
|          and comparing bytes/bytearray with str. (-bb: issue errors)\n\
 | |
| -B     : don't write .py[co] files on import; also PYTHONDONTWRITEBYTECODE=x\n\
 | |
| -c cmd : program passed in as string (terminates option list)\n\
 | |
| -d     : debug output from parser; also PYTHONDEBUG=x\n\
 | |
| -E     : ignore PYTHON* environment variables (such as PYTHONPATH)\n\
 | |
| -h     : print this help message and exit (also --help)\n\
 | |
| ";
 | |
| static const char usage_2[] = "\
 | |
| -i     : inspect interactively after running script; forces a prompt even\n\
 | |
|          if stdin does not appear to be a terminal; also PYTHONINSPECT=x\n\
 | |
| -I     : isolate Python from the user's environment (implies -E and -s)\n\
 | |
| -m mod : run library module as a script (terminates option list)\n\
 | |
| -O     : optimize generated bytecode slightly; also PYTHONOPTIMIZE=x\n\
 | |
| -OO    : remove doc-strings in addition to the -O optimizations\n\
 | |
| -q     : don't print version and copyright messages on interactive startup\n\
 | |
| -s     : don't add user site directory to sys.path; also PYTHONNOUSERSITE\n\
 | |
| -S     : don't imply 'import site' on initialization\n\
 | |
| ";
 | |
| static const char usage_3[] = "\
 | |
| -u     : unbuffered binary stdout and stderr, stdin always buffered;\n\
 | |
|          also PYTHONUNBUFFERED=x\n\
 | |
|          see man page for details on internal buffering relating to '-u'\n\
 | |
| -v     : verbose (trace import statements); also PYTHONVERBOSE=x\n\
 | |
|          can be supplied multiple times to increase verbosity\n\
 | |
| -V     : print the Python version number and exit (also --version)\n\
 | |
| -W arg : warning control; arg is action:message:category:module:lineno\n\
 | |
|          also PYTHONWARNINGS=arg\n\
 | |
| -x     : skip first line of source, allowing use of non-Unix forms of #!cmd\n\
 | |
| -X opt : set implementation-specific option\n\
 | |
| ";
 | |
| static const char usage_4[] = "\
 | |
| file   : program read from script file\n\
 | |
| -      : program read from stdin (default; interactive mode if a tty)\n\
 | |
| arg ...: arguments passed to program in sys.argv[1:]\n\n\
 | |
| Other environment variables:\n\
 | |
| PYTHONSTARTUP: file executed on interactive startup (no default)\n\
 | |
| PYTHONPATH   : '%lc'-separated list of directories prefixed to the\n\
 | |
|                default module search path.  The result is sys.path.\n\
 | |
| ";
 | |
| static const char usage_5[] =
 | |
| "PYTHONHOME   : alternate <prefix> directory (or <prefix>%lc<exec_prefix>).\n"
 | |
| "               The default module search path uses %s.\n"
 | |
| "PYTHONCASEOK : ignore case in 'import' statements (Windows).\n"
 | |
| "PYTHONIOENCODING: Encoding[:errors] used for stdin/stdout/stderr.\n"
 | |
| "PYTHONFAULTHANDLER: dump the Python traceback on fatal errors.\n";
 | |
| static const char usage_6[] =
 | |
| "PYTHONHASHSEED: if this variable is set to 'random', a random value is used\n"
 | |
| "   to seed the hashes of str, bytes and datetime objects.  It can also be\n"
 | |
| "   set to an integer in the range [0,4294967295] to get hash values with a\n"
 | |
| "   predictable seed.\n"
 | |
| "PYTHONMALLOC: set the Python memory allocators and/or install debug hooks\n"
 | |
| "   on Python memory allocators. Use PYTHONMALLOC=debug to install debug\n"
 | |
| "   hooks.\n";
 | |
| 
 | |
| static int
 | |
| usage(int exitcode, const wchar_t* program)
 | |
| {
 | |
|     FILE *f = exitcode ? stderr : stdout;
 | |
| 
 | |
|     fprintf(f, usage_line, program);
 | |
|     if (exitcode)
 | |
|         fprintf(f, "Try `python -h' for more information.\n");
 | |
|     else {
 | |
|         fputs(usage_1, f);
 | |
|         fputs(usage_2, f);
 | |
|         fputs(usage_3, f);
 | |
|         fprintf(f, usage_4, (wint_t)DELIM);
 | |
|         fprintf(f, usage_5, (wint_t)DELIM, PYTHONHOMEHELP);
 | |
|         fputs(usage_6, f);
 | |
|     }
 | |
|     return exitcode;
 | |
| }
 | |
| 
 | |
| static void RunStartupFile(PyCompilerFlags *cf)
 | |
| {
 | |
|     char *startup = Py_GETENV("PYTHONSTARTUP");
 | |
|     if (startup != NULL && startup[0] != '\0') {
 | |
|         FILE *fp = _Py_fopen(startup, "r");
 | |
|         if (fp != NULL) {
 | |
|             (void) PyRun_SimpleFileExFlags(fp, startup, 0, cf);
 | |
|             PyErr_Clear();
 | |
|             fclose(fp);
 | |
|         } else {
 | |
|             int save_errno;
 | |
| 
 | |
|             save_errno = errno;
 | |
|             PySys_WriteStderr("Could not open PYTHONSTARTUP\n");
 | |
|             errno = save_errno;
 | |
|             PyErr_SetFromErrnoWithFilename(PyExc_IOError,
 | |
|                             startup);
 | |
|             PyErr_Print();
 | |
|             PyErr_Clear();
 | |
|         }
 | |
|     }
 | |
| }
 | |
| 
 | |
| static void RunInteractiveHook(void)
 | |
| {
 | |
|     PyObject *sys, *hook, *result;
 | |
|     sys = PyImport_ImportModule("sys");
 | |
|     if (sys == NULL)
 | |
|         goto error;
 | |
|     hook = PyObject_GetAttrString(sys, "__interactivehook__");
 | |
|     Py_DECREF(sys);
 | |
|     if (hook == NULL)
 | |
|         PyErr_Clear();
 | |
|     else {
 | |
|         result = PyObject_CallObject(hook, NULL);
 | |
|         Py_DECREF(hook);
 | |
|         if (result == NULL)
 | |
|             goto error;
 | |
|         else
 | |
|             Py_DECREF(result);
 | |
|     }
 | |
|     return;
 | |
| 
 | |
| error:
 | |
|     PySys_WriteStderr("Failed calling sys.__interactivehook__\n");
 | |
|     PyErr_Print();
 | |
|     PyErr_Clear();
 | |
| }
 | |
| 
 | |
| 
 | |
| static int RunModule(wchar_t *modname, int set_argv0)
 | |
| {
 | |
|     PyObject *module, *runpy, *runmodule, *runargs, *result;
 | |
|     runpy = PyImport_ImportModule("runpy");
 | |
|     if (runpy == NULL) {
 | |
|         fprintf(stderr, "Could not import runpy module\n");
 | |
|         PyErr_Print();
 | |
|         return -1;
 | |
|     }
 | |
|     runmodule = PyObject_GetAttrString(runpy, "_run_module_as_main");
 | |
|     if (runmodule == NULL) {
 | |
|         fprintf(stderr, "Could not access runpy._run_module_as_main\n");
 | |
|         PyErr_Print();
 | |
|         Py_DECREF(runpy);
 | |
|         return -1;
 | |
|     }
 | |
|     module = PyUnicode_FromWideChar(modname, wcslen(modname));
 | |
|     if (module == NULL) {
 | |
|         fprintf(stderr, "Could not convert module name to unicode\n");
 | |
|         PyErr_Print();
 | |
|         Py_DECREF(runpy);
 | |
|         Py_DECREF(runmodule);
 | |
|         return -1;
 | |
|     }
 | |
|     runargs = Py_BuildValue("(Oi)", module, set_argv0);
 | |
|     if (runargs == NULL) {
 | |
|         fprintf(stderr,
 | |
|             "Could not create arguments for runpy._run_module_as_main\n");
 | |
|         PyErr_Print();
 | |
|         Py_DECREF(runpy);
 | |
|         Py_DECREF(runmodule);
 | |
|         Py_DECREF(module);
 | |
|         return -1;
 | |
|     }
 | |
|     result = PyObject_Call(runmodule, runargs, NULL);
 | |
|     if (result == NULL) {
 | |
|         PyErr_Print();
 | |
|     }
 | |
|     Py_DECREF(runpy);
 | |
|     Py_DECREF(runmodule);
 | |
|     Py_DECREF(module);
 | |
|     Py_DECREF(runargs);
 | |
|     if (result == NULL) {
 | |
|         return -1;
 | |
|     }
 | |
|     Py_DECREF(result);
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static int
 | |
| RunMainFromImporter(wchar_t *filename)
 | |
| {
 | |
|     PyObject *argv0 = NULL, *importer, *sys_path;
 | |
|     int sts;
 | |
| 
 | |
|     argv0 = PyUnicode_FromWideChar(filename, wcslen(filename));
 | |
|     if (argv0 == NULL)
 | |
|         goto error;
 | |
| 
 | |
|     importer = PyImport_GetImporter(argv0);
 | |
|     if (importer == NULL)
 | |
|         goto error;
 | |
| 
 | |
|     if (importer == Py_None) {
 | |
|         Py_DECREF(argv0);
 | |
|         Py_DECREF(importer);
 | |
|         return -1;
 | |
|     }
 | |
|     Py_DECREF(importer);
 | |
| 
 | |
|     /* argv0 is usable as an import source, so put it in sys.path[0]
 | |
|        and import __main__ */
 | |
|     sys_path = PySys_GetObject("path");
 | |
|     if (sys_path == NULL) {
 | |
|         PyErr_SetString(PyExc_RuntimeError, "unable to get sys.path");
 | |
|         goto error;
 | |
|     }
 | |
|     if (PyList_SetItem(sys_path, 0, argv0)) {
 | |
|         argv0 = NULL;
 | |
|         goto error;
 | |
|     }
 | |
|     Py_INCREF(argv0);
 | |
| 
 | |
|     sts = RunModule(L"__main__", 0);
 | |
|     return sts != 0;
 | |
| 
 | |
| error:
 | |
|     Py_XDECREF(argv0);
 | |
|     PyErr_Print();
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| static int
 | |
| run_command(wchar_t *command, PyCompilerFlags *cf)
 | |
| {
 | |
|     PyObject *unicode, *bytes;
 | |
|     int ret;
 | |
| 
 | |
|     unicode = PyUnicode_FromWideChar(command, -1);
 | |
|     if (unicode == NULL)
 | |
|         goto error;
 | |
|     bytes = PyUnicode_AsUTF8String(unicode);
 | |
|     Py_DECREF(unicode);
 | |
|     if (bytes == NULL)
 | |
|         goto error;
 | |
|     ret = PyRun_SimpleStringFlags(PyBytes_AsString(bytes), cf);
 | |
|     Py_DECREF(bytes);
 | |
|     return ret != 0;
 | |
| 
 | |
| error:
 | |
|     PySys_WriteStderr("Unable to decode the command from the command line:\n");
 | |
|     PyErr_Print();
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| static int
 | |
| run_file(FILE *fp, const wchar_t *filename, PyCompilerFlags *p_cf)
 | |
| {
 | |
|     PyObject *unicode, *bytes = NULL;
 | |
|     char *filename_str;
 | |
|     int run;
 | |
| 
 | |
|     /* call pending calls like signal handlers (SIGINT) */
 | |
|     if (Py_MakePendingCalls() == -1) {
 | |
|         PyErr_Print();
 | |
|         return 1;
 | |
|     }
 | |
| 
 | |
|     if (filename) {
 | |
|         unicode = PyUnicode_FromWideChar(filename, wcslen(filename));
 | |
|         if (unicode != NULL) {
 | |
|             bytes = PyUnicode_EncodeFSDefault(unicode);
 | |
|             Py_DECREF(unicode);
 | |
|         }
 | |
|         if (bytes != NULL)
 | |
|             filename_str = PyBytes_AsString(bytes);
 | |
|         else {
 | |
|             PyErr_Clear();
 | |
|             filename_str = "<encoding error>";
 | |
|         }
 | |
|     }
 | |
|     else
 | |
|         filename_str = "<stdin>";
 | |
| 
 | |
|     run = PyRun_AnyFileExFlags(fp, filename_str, filename != NULL, p_cf);
 | |
|     Py_XDECREF(bytes);
 | |
|     return run != 0;
 | |
| }
 | |
| 
 | |
| 
 | |
| /* Main program */
 | |
| 
 | |
| int
 | |
| Py_Main(int argc, wchar_t **argv)
 | |
| {
 | |
|     int c;
 | |
|     int sts;
 | |
|     wchar_t *command = NULL;
 | |
|     wchar_t *filename = NULL;
 | |
|     wchar_t *module = NULL;
 | |
|     FILE *fp = stdin;
 | |
|     char *p;
 | |
| #ifdef MS_WINDOWS
 | |
|     wchar_t *wp;
 | |
| #endif
 | |
|     int skipfirstline = 0;
 | |
|     int stdin_is_interactive = 0;
 | |
|     int help = 0;
 | |
|     int version = 0;
 | |
|     int saw_unbuffered_flag = 0;
 | |
|     char *opt;
 | |
|     PyCompilerFlags cf;
 | |
|     PyObject *warning_option = NULL;
 | |
|     PyObject *warning_options = NULL;
 | |
| 
 | |
|     cf.cf_flags = 0;
 | |
| 
 | |
|     orig_argc = argc;           /* For Py_GetArgcArgv() */
 | |
|     orig_argv = argv;
 | |
| 
 | |
|     /* Hash randomization needed early for all string operations
 | |
|        (including -W and -X options). */
 | |
|     _PyOS_opterr = 0;  /* prevent printing the error in 1st pass */
 | |
|     while ((c = _PyOS_GetOpt(argc, argv, PROGRAM_OPTS)) != EOF) {
 | |
|         if (c == 'm' || c == 'c') {
 | |
|             /* -c / -m is the last option: following arguments are
 | |
|                not interpreter options. */
 | |
|             break;
 | |
|         }
 | |
|         if (c == 'E') {
 | |
|             Py_IgnoreEnvironmentFlag++;
 | |
|             break;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     opt = Py_GETENV("PYTHONMALLOC");
 | |
|     if (_PyMem_SetupAllocators(opt) < 0) {
 | |
|         fprintf(stderr,
 | |
|                 "Error in PYTHONMALLOC: unknown allocator \"%s\"!\n", opt);
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     Py_HashRandomizationFlag = 1;
 | |
|     _PyRandom_Init();
 | |
| 
 | |
|     PySys_ResetWarnOptions();
 | |
|     _PyOS_ResetGetOpt();
 | |
| 
 | |
|     while ((c = _PyOS_GetOpt(argc, argv, PROGRAM_OPTS)) != EOF) {
 | |
|         if (c == 'c') {
 | |
|             size_t len;
 | |
|             /* -c is the last option; following arguments
 | |
|                that look like options are left for the
 | |
|                command to interpret. */
 | |
| 
 | |
|             len = wcslen(_PyOS_optarg) + 1 + 1;
 | |
|             command = (wchar_t *)PyMem_RawMalloc(sizeof(wchar_t) * len);
 | |
|             if (command == NULL)
 | |
|                 Py_FatalError(
 | |
|                    "not enough memory to copy -c argument");
 | |
|             wcscpy(command, _PyOS_optarg);
 | |
|             command[len - 2] = '\n';
 | |
|             command[len - 1] = 0;
 | |
|             break;
 | |
|         }
 | |
| 
 | |
|         if (c == 'm') {
 | |
|             /* -m is the last option; following arguments
 | |
|                that look like options are left for the
 | |
|                module to interpret. */
 | |
|             module = _PyOS_optarg;
 | |
|             break;
 | |
|         }
 | |
| 
 | |
|         switch (c) {
 | |
|         case 'b':
 | |
|             Py_BytesWarningFlag++;
 | |
|             break;
 | |
| 
 | |
|         case 'd':
 | |
|             Py_DebugFlag++;
 | |
|             break;
 | |
| 
 | |
|         case 'i':
 | |
|             Py_InspectFlag++;
 | |
|             Py_InteractiveFlag++;
 | |
|             break;
 | |
| 
 | |
|         case 'I':
 | |
|             Py_IsolatedFlag++;
 | |
|             Py_NoUserSiteDirectory++;
 | |
|             Py_IgnoreEnvironmentFlag++;
 | |
|             break;
 | |
| 
 | |
|         /* case 'J': reserved for Jython */
 | |
| 
 | |
|         case 'O':
 | |
|             Py_OptimizeFlag++;
 | |
|             break;
 | |
| 
 | |
|         case 'B':
 | |
|             Py_DontWriteBytecodeFlag++;
 | |
|             break;
 | |
| 
 | |
|         case 's':
 | |
|             Py_NoUserSiteDirectory++;
 | |
|             break;
 | |
| 
 | |
|         case 'S':
 | |
|             Py_NoSiteFlag++;
 | |
|             break;
 | |
| 
 | |
|         case 'E':
 | |
|             /* Already handled above */
 | |
|             break;
 | |
| 
 | |
|         case 't':
 | |
|             /* ignored for backwards compatibility */
 | |
|             break;
 | |
| 
 | |
|         case 'u':
 | |
|             Py_UnbufferedStdioFlag = 1;
 | |
|             saw_unbuffered_flag = 1;
 | |
|             break;
 | |
| 
 | |
|         case 'v':
 | |
|             Py_VerboseFlag++;
 | |
|             break;
 | |
| 
 | |
|         case 'x':
 | |
|             skipfirstline = 1;
 | |
|             break;
 | |
| 
 | |
|         case 'h':
 | |
|         case '?':
 | |
|             help++;
 | |
|             break;
 | |
| 
 | |
|         case 'V':
 | |
|             version++;
 | |
|             break;
 | |
| 
 | |
|         case 'W':
 | |
|             if (warning_options == NULL)
 | |
|                 warning_options = PyList_New(0);
 | |
|             if (warning_options == NULL)
 | |
|                 Py_FatalError("failure in handling of -W argument");
 | |
|             warning_option = PyUnicode_FromWideChar(_PyOS_optarg, -1);
 | |
|             if (warning_option == NULL)
 | |
|                 Py_FatalError("failure in handling of -W argument");
 | |
|             PyList_Append(warning_options, warning_option);
 | |
|             Py_DECREF(warning_option);
 | |
|             break;
 | |
| 
 | |
|         case 'X':
 | |
|             PySys_AddXOption(_PyOS_optarg);
 | |
|             break;
 | |
| 
 | |
|         case 'q':
 | |
|             Py_QuietFlag++;
 | |
|             break;
 | |
| 
 | |
|         case 'R':
 | |
|             /* Ignored */
 | |
|             break;
 | |
| 
 | |
|         /* This space reserved for other options */
 | |
| 
 | |
|         default:
 | |
|             return usage(2, argv[0]);
 | |
|             /*NOTREACHED*/
 | |
| 
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     if (help)
 | |
|         return usage(0, argv[0]);
 | |
| 
 | |
|     if (version) {
 | |
|         printf("Python %s\n", PY_VERSION);
 | |
|         return 0;
 | |
|     }
 | |
| 
 | |
|     if (!Py_InspectFlag &&
 | |
|         (p = Py_GETENV("PYTHONINSPECT")) && *p != '\0')
 | |
|         Py_InspectFlag = 1;
 | |
|     if (!saw_unbuffered_flag &&
 | |
|         (p = Py_GETENV("PYTHONUNBUFFERED")) && *p != '\0')
 | |
|         Py_UnbufferedStdioFlag = 1;
 | |
| 
 | |
|     if (!Py_NoUserSiteDirectory &&
 | |
|         (p = Py_GETENV("PYTHONNOUSERSITE")) && *p != '\0')
 | |
|         Py_NoUserSiteDirectory = 1;
 | |
| 
 | |
| #ifdef MS_WINDOWS
 | |
|     if (!Py_IgnoreEnvironmentFlag && (wp = _wgetenv(L"PYTHONWARNINGS")) &&
 | |
|         *wp != L'\0') {
 | |
|         wchar_t *buf, *warning, *context = NULL;
 | |
| 
 | |
|         buf = (wchar_t *)PyMem_RawMalloc((wcslen(wp) + 1) * sizeof(wchar_t));
 | |
|         if (buf == NULL)
 | |
|             Py_FatalError(
 | |
|                "not enough memory to copy PYTHONWARNINGS");
 | |
|         wcscpy(buf, wp);
 | |
|         for (warning = wcstok_s(buf, L",", &context);
 | |
|              warning != NULL;
 | |
|              warning = wcstok_s(NULL, L",", &context)) {
 | |
|             PySys_AddWarnOption(warning);
 | |
|         }
 | |
|         PyMem_RawFree(buf);
 | |
|     }
 | |
| #else
 | |
|     if ((p = Py_GETENV("PYTHONWARNINGS")) && *p != '\0') {
 | |
|         char *buf, *oldloc;
 | |
|         PyObject *unicode;
 | |
| 
 | |
|         /* settle for strtok here as there's no one standard
 | |
|            C89 wcstok */
 | |
|         buf = (char *)PyMem_RawMalloc(strlen(p) + 1);
 | |
|         if (buf == NULL)
 | |
|             Py_FatalError(
 | |
|                "not enough memory to copy PYTHONWARNINGS");
 | |
|         strcpy(buf, p);
 | |
|         oldloc = _PyMem_RawStrdup(setlocale(LC_ALL, NULL));
 | |
|         setlocale(LC_ALL, "");
 | |
|         for (p = strtok(buf, ","); p != NULL; p = strtok(NULL, ",")) {
 | |
| #ifdef __APPLE__
 | |
|             /* Use utf-8 on Mac OS X */
 | |
|             unicode = PyUnicode_FromString(p);
 | |
| #else
 | |
|             unicode = PyUnicode_DecodeLocale(p, "surrogateescape");
 | |
| #endif
 | |
|             if (unicode == NULL) {
 | |
|                 /* ignore errors */
 | |
|                 PyErr_Clear();
 | |
|                 continue;
 | |
|             }
 | |
|             PySys_AddWarnOptionUnicode(unicode);
 | |
|             Py_DECREF(unicode);
 | |
|         }
 | |
|         setlocale(LC_ALL, oldloc);
 | |
|         PyMem_RawFree(oldloc);
 | |
|         PyMem_RawFree(buf);
 | |
|     }
 | |
| #endif
 | |
|     if (warning_options != NULL) {
 | |
|         Py_ssize_t i;
 | |
|         for (i = 0; i < PyList_GET_SIZE(warning_options); i++) {
 | |
|             PySys_AddWarnOptionUnicode(PyList_GET_ITEM(warning_options, i));
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     if (command == NULL && module == NULL && _PyOS_optind < argc &&
 | |
|         wcscmp(argv[_PyOS_optind], L"-") != 0)
 | |
|     {
 | |
|         filename = argv[_PyOS_optind];
 | |
|     }
 | |
| 
 | |
|     stdin_is_interactive = Py_FdIsInteractive(stdin, (char *)0);
 | |
| 
 | |
| #if defined(MS_WINDOWS) || defined(__CYGWIN__)
 | |
|     /* don't translate newlines (\r\n <=> \n) */
 | |
|     _setmode(fileno(stdin), O_BINARY);
 | |
|     _setmode(fileno(stdout), O_BINARY);
 | |
|     _setmode(fileno(stderr), O_BINARY);
 | |
| #endif
 | |
| 
 | |
|     if (Py_UnbufferedStdioFlag) {
 | |
| #ifdef HAVE_SETVBUF
 | |
|         setvbuf(stdin,  (char *)NULL, _IONBF, BUFSIZ);
 | |
|         setvbuf(stdout, (char *)NULL, _IONBF, BUFSIZ);
 | |
|         setvbuf(stderr, (char *)NULL, _IONBF, BUFSIZ);
 | |
| #else /* !HAVE_SETVBUF */
 | |
|         setbuf(stdin,  (char *)NULL);
 | |
|         setbuf(stdout, (char *)NULL);
 | |
|         setbuf(stderr, (char *)NULL);
 | |
| #endif /* !HAVE_SETVBUF */
 | |
|     }
 | |
|     else if (Py_InteractiveFlag) {
 | |
| #ifdef MS_WINDOWS
 | |
|         /* Doesn't have to have line-buffered -- use unbuffered */
 | |
|         /* Any set[v]buf(stdin, ...) screws up Tkinter :-( */
 | |
|         setvbuf(stdout, (char *)NULL, _IONBF, BUFSIZ);
 | |
| #else /* !MS_WINDOWS */
 | |
| #ifdef HAVE_SETVBUF
 | |
|         setvbuf(stdin,  (char *)NULL, _IOLBF, BUFSIZ);
 | |
|         setvbuf(stdout, (char *)NULL, _IOLBF, BUFSIZ);
 | |
| #endif /* HAVE_SETVBUF */
 | |
| #endif /* !MS_WINDOWS */
 | |
|         /* Leave stderr alone - it should be unbuffered anyway. */
 | |
|     }
 | |
| 
 | |
| #ifdef __APPLE__
 | |
|     /* On MacOS X, when the Python interpreter is embedded in an
 | |
|        application bundle, it gets executed by a bootstrapping script
 | |
|        that does os.execve() with an argv[0] that's different from the
 | |
|        actual Python executable. This is needed to keep the Finder happy,
 | |
|        or rather, to work around Apple's overly strict requirements of
 | |
|        the process name. However, we still need a usable sys.executable,
 | |
|        so the actual executable path is passed in an environment variable.
 | |
|        See Lib/plat-mac/bundlebuiler.py for details about the bootstrap
 | |
|        script. */
 | |
|     if ((p = Py_GETENV("PYTHONEXECUTABLE")) && *p != '\0') {
 | |
|         wchar_t* buffer;
 | |
|         size_t len = strlen(p) + 1;
 | |
| 
 | |
|         buffer = PyMem_RawMalloc(len * sizeof(wchar_t));
 | |
|         if (buffer == NULL) {
 | |
|             Py_FatalError(
 | |
|                "not enough memory to copy PYTHONEXECUTABLE");
 | |
|         }
 | |
| 
 | |
|         mbstowcs(buffer, p, len);
 | |
|         Py_SetProgramName(buffer);
 | |
|         /* buffer is now handed off - do not free */
 | |
|     } else {
 | |
| #ifdef WITH_NEXT_FRAMEWORK
 | |
|         char* pyvenv_launcher = getenv("__PYVENV_LAUNCHER__");
 | |
| 
 | |
|         if (pyvenv_launcher && *pyvenv_launcher) {
 | |
|             /* Used by Mac/Tools/pythonw.c to forward
 | |
|              * the argv0 of the stub executable
 | |
|              */
 | |
|             wchar_t* wbuf = Py_DecodeLocale(pyvenv_launcher, NULL);
 | |
| 
 | |
|             if (wbuf == NULL) {
 | |
|                 Py_FatalError("Cannot decode __PYVENV_LAUNCHER__");
 | |
|             }
 | |
|             Py_SetProgramName(wbuf);
 | |
| 
 | |
|             /* Don't free wbuf, the argument to Py_SetProgramName
 | |
|              * must remain valid until Py_FinalizeEx is called.
 | |
|              */
 | |
|         } else {
 | |
|             Py_SetProgramName(argv[0]);
 | |
|         }
 | |
| #else
 | |
|         Py_SetProgramName(argv[0]);
 | |
| #endif
 | |
|     }
 | |
| #else
 | |
|     Py_SetProgramName(argv[0]);
 | |
| #endif
 | |
|     Py_Initialize();
 | |
|     Py_XDECREF(warning_options);
 | |
| 
 | |
|     if (!Py_QuietFlag && (Py_VerboseFlag ||
 | |
|                         (command == NULL && filename == NULL &&
 | |
|                          module == NULL && stdin_is_interactive))) {
 | |
|         fprintf(stderr, "Python %s on %s\n",
 | |
|             Py_GetVersion(), Py_GetPlatform());
 | |
|         if (!Py_NoSiteFlag)
 | |
|             fprintf(stderr, "%s\n", COPYRIGHT);
 | |
|     }
 | |
| 
 | |
|     if (command != NULL) {
 | |
|         /* Backup _PyOS_optind and force sys.argv[0] = '-c' */
 | |
|         _PyOS_optind--;
 | |
|         argv[_PyOS_optind] = L"-c";
 | |
|     }
 | |
| 
 | |
|     if (module != NULL) {
 | |
|         /* Backup _PyOS_optind and force sys.argv[0] = '-m'*/
 | |
|         _PyOS_optind--;
 | |
|         argv[_PyOS_optind] = L"-m";
 | |
|     }
 | |
| 
 | |
|     PySys_SetArgv(argc-_PyOS_optind, argv+_PyOS_optind);
 | |
| 
 | |
|     if ((Py_InspectFlag || (command == NULL && filename == NULL && module == NULL)) &&
 | |
|         isatty(fileno(stdin))) {
 | |
|         PyObject *v;
 | |
|         v = PyImport_ImportModule("readline");
 | |
|         if (v == NULL)
 | |
|             PyErr_Clear();
 | |
|         else
 | |
|             Py_DECREF(v);
 | |
|     }
 | |
| 
 | |
|     if (command) {
 | |
|         sts = run_command(command, &cf);
 | |
|         PyMem_RawFree(command);
 | |
|     } else if (module) {
 | |
|         sts = (RunModule(module, 1) != 0);
 | |
|     }
 | |
|     else {
 | |
| 
 | |
|         if (filename == NULL && stdin_is_interactive) {
 | |
|             Py_InspectFlag = 0; /* do exit on SystemExit */
 | |
|             RunStartupFile(&cf);
 | |
|             RunInteractiveHook();
 | |
|         }
 | |
|         /* XXX */
 | |
| 
 | |
|         sts = -1;               /* keep track of whether we've already run __main__ */
 | |
| 
 | |
|         if (filename != NULL) {
 | |
|             sts = RunMainFromImporter(filename);
 | |
|         }
 | |
| 
 | |
|         if (sts==-1 && filename!=NULL) {
 | |
|             fp = _Py_wfopen(filename, L"r");
 | |
|             if (fp == NULL) {
 | |
|                 char *cfilename_buffer;
 | |
|                 const char *cfilename;
 | |
|                 int err = errno;
 | |
|                 cfilename_buffer = Py_EncodeLocale(filename, NULL);
 | |
|                 if (cfilename_buffer != NULL)
 | |
|                     cfilename = cfilename_buffer;
 | |
|                 else
 | |
|                     cfilename = "<unprintable file name>";
 | |
|                 fprintf(stderr, "%ls: can't open file '%s': [Errno %d] %s\n",
 | |
|                     argv[0], cfilename, err, strerror(err));
 | |
|                 if (cfilename_buffer)
 | |
|                     PyMem_Free(cfilename_buffer);
 | |
|                 return 2;
 | |
|             }
 | |
|             else if (skipfirstline) {
 | |
|                 int ch;
 | |
|                 /* Push back first newline so line numbers
 | |
|                    remain the same */
 | |
|                 while ((ch = getc(fp)) != EOF) {
 | |
|                     if (ch == '\n') {
 | |
|                         (void)ungetc(ch, fp);
 | |
|                         break;
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
|             {
 | |
|                 struct _Py_stat_struct sb;
 | |
|                 if (_Py_fstat_noraise(fileno(fp), &sb) == 0 &&
 | |
|                     S_ISDIR(sb.st_mode)) {
 | |
|                     fprintf(stderr,
 | |
|                             "%ls: '%ls' is a directory, cannot continue\n",
 | |
|                             argv[0], filename);
 | |
|                     fclose(fp);
 | |
|                     return 1;
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         if (sts == -1)
 | |
|             sts = run_file(fp, filename, &cf);
 | |
|     }
 | |
| 
 | |
|     /* Check this environment variable at the end, to give programs the
 | |
|      * opportunity to set it from Python.
 | |
|      */
 | |
|     if (!Py_InspectFlag &&
 | |
|         (p = Py_GETENV("PYTHONINSPECT")) && *p != '\0')
 | |
|     {
 | |
|         Py_InspectFlag = 1;
 | |
|     }
 | |
| 
 | |
|     if (Py_InspectFlag && stdin_is_interactive &&
 | |
|         (filename != NULL || command != NULL || module != NULL)) {
 | |
|         Py_InspectFlag = 0;
 | |
|         RunInteractiveHook();
 | |
|         /* XXX */
 | |
|         sts = PyRun_AnyFileFlags(stdin, "<stdin>", &cf) != 0;
 | |
|     }
 | |
| 
 | |
|     if (Py_FinalizeEx() < 0) {
 | |
|         /* Value unlikely to be confused with a non-error exit status or
 | |
|         other special meaning */
 | |
|         sts = 120;
 | |
|     }
 | |
| 
 | |
| #ifdef __INSURE__
 | |
|     /* Insure++ is a memory analysis tool that aids in discovering
 | |
|      * memory leaks and other memory problems.  On Python exit, the
 | |
|      * interned string dictionaries are flagged as being in use at exit
 | |
|      * (which it is).  Under normal circumstances, this is fine because
 | |
|      * the memory will be automatically reclaimed by the system.  Under
 | |
|      * memory debugging, it's a huge source of useless noise, so we
 | |
|      * trade off slower shutdown for less distraction in the memory
 | |
|      * reports.  -baw
 | |
|      */
 | |
|     _Py_ReleaseInternedUnicodeStrings();
 | |
| #endif /* __INSURE__ */
 | |
| 
 | |
|     return sts;
 | |
| }
 | |
| 
 | |
| /* this is gonna seem *real weird*, but if you put some other code between
 | |
|    Py_Main() and Py_GetArgcArgv() you will need to adjust the test in the
 | |
|    while statement in Misc/gdbinit:ppystack */
 | |
| 
 | |
| /* Make the *original* argc/argv available to other modules.
 | |
|    This is rare, but it is needed by the secureware extension. */
 | |
| 
 | |
| void
 | |
| Py_GetArgcArgv(int *argc, wchar_t ***argv)
 | |
| {
 | |
|     *argc = orig_argc;
 | |
|     *argv = orig_argv;
 | |
| }
 | |
| 
 | |
| #ifdef __cplusplus
 | |
| }
 | |
| #endif
 |