blob: b6dcdd0a30a8bb57dc109fdb43f030f9bf770032 [file] [log] [blame]
Guido van Rossum667d7041995-08-04 04:20:48 +00001/* Python interpreter main program */
2
3#include "Python.h"
Guido van Rossuma075ce11997-12-05 21:56:45 +00004#include "osdefs.h"
Guido van Rossum667d7041995-08-04 04:20:48 +00005
Antoine Pitrou5651eaa2008-09-06 20:46:58 +00006#include <locale.h>
7
Sjoerd Mullender9cf424b2002-08-09 13:35:18 +00008#if defined(MS_WINDOWS) || defined(__CYGWIN__)
Martin v. Löwis945362c2007-08-30 14:57:25 +00009#include <windows.h>
Thomas Wouters477c8d52006-05-27 19:21:47 +000010#ifdef HAVE_FCNTL_H
Guido van Rossum3e7ae7a1997-01-17 22:05:38 +000011#include <fcntl.h>
12#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +000013#endif
Guido van Rossum3e7ae7a1997-01-17 22:05:38 +000014
Martin v. Löwis945362c2007-08-30 14:57:25 +000015#ifdef _MSC_VER
16#include <crtdbg.h>
17#endif
18
Jesus Ceaab70e2a2012-10-05 01:48:08 +020019#if defined(MS_WINDOWS)
Guido van Rossuma075ce11997-12-05 21:56:45 +000020#define PYTHONHOMEHELP "<prefix>\\lib"
21#else
Marc-André Lemburgda4dbc32001-06-12 16:13:51 +000022#define PYTHONHOMEHELP "<prefix>/pythonX.X"
Guido van Rossuma075ce11997-12-05 21:56:45 +000023#endif
24
Thomas Wouters2cffc7d2000-11-03 08:18:37 +000025#include "pygetopt.h"
26
Guido van Rossuma22865e2000-09-05 04:41:18 +000027#define COPYRIGHT \
Guido van Rossum36002d72001-07-18 16:59:46 +000028 "Type \"help\", \"copyright\", \"credits\" or \"license\" " \
29 "for more information."
Guido van Rossuma22865e2000-09-05 04:41:18 +000030
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000031#ifdef __cplusplus
32extern "C" {
33#endif
34
Guido van Rossumac56b031996-07-21 02:33:38 +000035/* For Py_GetArgcArgv(); set by main() */
Martin v. Löwis790465f2008-04-05 20:41:37 +000036static wchar_t **orig_argv;
Guido van Rossum667d7041995-08-04 04:20:48 +000037static int orig_argc;
38
Guido van Rossumbceccf52001-04-10 22:07:43 +000039/* command line options */
Christian Heimesad73a9c2013-08-10 16:36:18 +020040#define BASE_OPTS L"bBc:dEhiIJm:OqRsStuvVW:xX:?"
Guido van Rossumbceccf52001-04-10 22:07:43 +000041
Guido van Rossumbceccf52001-04-10 22:07:43 +000042#define PROGRAM_OPTS BASE_OPTS
Guido van Rossum3ed4c152001-03-02 06:18:03 +000043
Guido van Rossum667d7041995-08-04 04:20:48 +000044/* Short usage message (with %s for argv0) */
Serhiy Storchaka2d06e842015-12-25 19:53:18 +020045static const char usage_line[] =
Martin v. Löwis790465f2008-04-05 20:41:37 +000046"usage: %ls [option] ... [-c cmd | -m mod | file | -] [arg] ...\n";
Guido van Rossum667d7041995-08-04 04:20:48 +000047
48/* Long usage message, split into parts < 512 bytes */
Serhiy Storchaka2d06e842015-12-25 19:53:18 +020049static const char usage_1[] = "\
Guido van Rossum667d7041995-08-04 04:20:48 +000050Options and arguments (and corresponding environment variables):\n\
Christian Heimes2ab34442008-09-03 20:31:07 +000051-b : issue warnings about str(bytes_instance), str(bytearray_instance)\n\
52 and comparing bytes/bytearray with str. (-bb: issue errors)\n\
Christian Heimes790c8232008-01-07 21:14:23 +000053-B : don't write .py[co] files on import; also PYTHONDONTWRITEBYTECODE=x\n\
Guido van Rossum393661d2001-08-31 17:40:15 +000054-c cmd : program passed in as string (terminates option list)\n\
Thomas Wouters89f507f2006-12-13 04:49:30 +000055-d : debug output from parser; also PYTHONDEBUG=x\n\
Christian Heimes790c8232008-01-07 21:14:23 +000056-E : ignore PYTHON* environment variables (such as PYTHONPATH)\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +000057-h : print this help message and exit (also --help)\n\
Guido van Rossum61c345f2001-09-04 03:26:15 +000058";
Serhiy Storchaka2d06e842015-12-25 19:53:18 +020059static const char usage_2[] = "\
Guido van Rossum98297ee2007-11-06 21:34:58 +000060-i : inspect interactively after running script; forces a prompt even\n\
61 if stdin does not appear to be a terminal; also PYTHONINSPECT=x\n\
Christian Heimesad73a9c2013-08-10 16:36:18 +020062-I : isolate Python from the user's environment (implies -E and -s)\n\
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +000063-m mod : run library module as a script (terminates option list)\n\
Thomas Wouters89f507f2006-12-13 04:49:30 +000064-O : optimize generated bytecode slightly; also PYTHONOPTIMIZE=x\n\
Guido van Rossum6b86a421999-01-28 15:07:47 +000065-OO : remove doc-strings in addition to the -O optimizations\n\
Georg Brandl9d871192010-12-04 10:47:18 +000066-q : don't print version and copyright messages on interactive startup\n\
Christian Heimes8dc226f2008-05-06 23:45:46 +000067-s : don't add user site directory to sys.path; also PYTHONNOUSERSITE\n\
Guido van Rossum7922bd71997-08-29 22:34:47 +000068-S : don't imply 'import site' on initialization\n\
Guido van Rossum393661d2001-08-31 17:40:15 +000069";
Serhiy Storchaka2d06e842015-12-25 19:53:18 +020070static const char usage_3[] = "\
Ezio Melotti61b0c672013-07-25 05:04:02 +020071-u : unbuffered binary stdout and stderr, stdin always buffered;\n\
72 also PYTHONUNBUFFERED=x\n\
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +000073 see man page for details on internal buffering relating to '-u'\n\
Thomas Wouters89f507f2006-12-13 04:49:30 +000074-v : verbose (trace import statements); also PYTHONVERBOSE=x\n\
75 can be supplied multiple times to increase verbosity\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +000076-V : print the Python version number and exit (also --version)\n\
Thomas Wouters89f507f2006-12-13 04:49:30 +000077-W arg : warning control; arg is action:message:category:module:lineno\n\
Philip Jenvey0805ca32010-04-07 04:04:10 +000078 also PYTHONWARNINGS=arg\n\
Guido van Rossum393661d2001-08-31 17:40:15 +000079-x : skip first line of source, allowing use of non-Unix forms of #!cmd\n\
Antoine Pitrou9583cac2010-10-21 13:42:28 +000080-X opt : set implementation-specific option\n\
Guido van Rossum7922bd71997-08-29 22:34:47 +000081";
Serhiy Storchaka2d06e842015-12-25 19:53:18 +020082static const char usage_4[] = "\
Guido van Rossum98297ee2007-11-06 21:34:58 +000083file : program read from script file\n\
84- : program read from stdin (default; interactive mode if a tty)\n\
Thomas Wouters89f507f2006-12-13 04:49:30 +000085arg ...: arguments passed to program in sys.argv[1:]\n\n\
Guido van Rossum667d7041995-08-04 04:20:48 +000086Other environment variables:\n\
87PYTHONSTARTUP: file executed on interactive startup (no default)\n\
Serhiy Storchaka1ba01612015-12-30 09:28:19 +020088PYTHONPATH : '%lc'-separated list of directories prefixed to the\n\
Guido van Rossum667d7041995-08-04 04:20:48 +000089 default module search path. The result is sys.path.\n\
Christian Heimes790c8232008-01-07 21:14:23 +000090";
Serhiy Storchaka2d06e842015-12-25 19:53:18 +020091static const char usage_5[] =
Serhiy Storchaka1ba01612015-12-30 09:28:19 +020092"PYTHONHOME : alternate <prefix> directory (or <prefix>%lc<exec_prefix>).\n"
Victor Stinner9802b392010-08-19 11:36:43 +000093" The default module search path uses %s.\n"
94"PYTHONCASEOK : ignore case in 'import' statements (Windows).\n"
95"PYTHONIOENCODING: Encoding[:errors] used for stdin/stdout/stderr.\n"
Victor Stinner34be8072016-03-14 12:04:26 +010096"PYTHONFAULTHANDLER: dump the Python traceback on fatal errors.\n";
97static const char usage_6[] =
98"PYTHONHASHSEED: if this variable is set to 'random', a random value is used\n"
99" to seed the hashes of str, bytes and datetime objects. It can also be\n"
100" set to an integer in the range [0,4294967295] to get hash values with a\n"
101" predictable seed.\n"
102"PYTHONMALLOC: set the Python memory allocators and/or install debug hooks\n"
103" on Python memory allocators. Use PYTHONMALLOC=debug to install debug\n"
104" hooks.\n";
Guido van Rossum667d7041995-08-04 04:20:48 +0000105
Martin v. Löwis852ba7e2003-03-30 17:09:58 +0000106static int
Serhiy Storchakaef1585e2015-12-25 20:01:53 +0200107usage(int exitcode, const wchar_t* program)
Barry Warsaw3b2aedb2000-09-15 18:40:42 +0000108{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000109 FILE *f = exitcode ? stderr : stdout;
Guido van Rossum393661d2001-08-31 17:40:15 +0000110
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000111 fprintf(f, usage_line, program);
112 if (exitcode)
113 fprintf(f, "Try `python -h' for more information.\n");
114 else {
115 fputs(usage_1, f);
116 fputs(usage_2, f);
117 fputs(usage_3, f);
Serhiy Storchaka1ba01612015-12-30 09:28:19 +0200118 fprintf(f, usage_4, (wint_t)DELIM);
119 fprintf(f, usage_5, (wint_t)DELIM, PYTHONHOMEHELP);
Georg Brandl2daf6ae2012-02-20 19:54:16 +0100120 fputs(usage_6, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000121 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000122 return exitcode;
Barry Warsaw3b2aedb2000-09-15 18:40:42 +0000123}
124
Martin v. Löwis6caea372003-11-18 19:46:25 +0000125static void RunStartupFile(PyCompilerFlags *cf)
126{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000127 char *startup = Py_GETENV("PYTHONSTARTUP");
128 if (startup != NULL && startup[0] != '\0') {
Victor Stinnerdaf45552013-08-28 00:53:59 +0200129 FILE *fp = _Py_fopen(startup, "r");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000130 if (fp != NULL) {
131 (void) PyRun_SimpleFileExFlags(fp, startup, 0, cf);
132 PyErr_Clear();
133 fclose(fp);
134 } else {
135 int save_errno;
136
137 save_errno = errno;
138 PySys_WriteStderr("Could not open PYTHONSTARTUP\n");
139 errno = save_errno;
140 PyErr_SetFromErrnoWithFilename(PyExc_IOError,
141 startup);
142 PyErr_Print();
143 PyErr_Clear();
144 }
145 }
Martin v. Löwis6caea372003-11-18 19:46:25 +0000146}
147
Antoine Pitrou1a6cb302013-05-04 20:08:35 +0200148static void RunInteractiveHook(void)
149{
150 PyObject *sys, *hook, *result;
151 sys = PyImport_ImportModule("sys");
152 if (sys == NULL)
153 goto error;
154 hook = PyObject_GetAttrString(sys, "__interactivehook__");
155 Py_DECREF(sys);
156 if (hook == NULL)
157 PyErr_Clear();
158 else {
159 result = PyObject_CallObject(hook, NULL);
160 Py_DECREF(hook);
161 if (result == NULL)
162 goto error;
163 else
164 Py_DECREF(result);
165 }
166 return;
167
168error:
169 PySys_WriteStderr("Failed calling sys.__interactivehook__\n");
170 PyErr_Print();
171 PyErr_Clear();
172}
173
Thomas Woutersa9773292006-04-21 09:43:23 +0000174
Antoine Pitrou5651eaa2008-09-06 20:46:58 +0000175static int RunModule(wchar_t *modname, int set_argv0)
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +0000176{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000177 PyObject *module, *runpy, *runmodule, *runargs, *result;
178 runpy = PyImport_ImportModule("runpy");
179 if (runpy == NULL) {
180 fprintf(stderr, "Could not import runpy module\n");
Victor Stinner7d36e4f2013-04-10 00:27:23 +0200181 PyErr_Print();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000182 return -1;
183 }
184 runmodule = PyObject_GetAttrString(runpy, "_run_module_as_main");
185 if (runmodule == NULL) {
186 fprintf(stderr, "Could not access runpy._run_module_as_main\n");
Victor Stinner7d36e4f2013-04-10 00:27:23 +0200187 PyErr_Print();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000188 Py_DECREF(runpy);
189 return -1;
190 }
191 module = PyUnicode_FromWideChar(modname, wcslen(modname));
192 if (module == NULL) {
193 fprintf(stderr, "Could not convert module name to unicode\n");
Victor Stinner7d36e4f2013-04-10 00:27:23 +0200194 PyErr_Print();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000195 Py_DECREF(runpy);
196 Py_DECREF(runmodule);
197 return -1;
198 }
199 runargs = Py_BuildValue("(Oi)", module, set_argv0);
200 if (runargs == NULL) {
201 fprintf(stderr,
202 "Could not create arguments for runpy._run_module_as_main\n");
Victor Stinner7d36e4f2013-04-10 00:27:23 +0200203 PyErr_Print();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000204 Py_DECREF(runpy);
205 Py_DECREF(runmodule);
206 Py_DECREF(module);
207 return -1;
208 }
209 result = PyObject_Call(runmodule, runargs, NULL);
210 if (result == NULL) {
211 PyErr_Print();
212 }
213 Py_DECREF(runpy);
214 Py_DECREF(runmodule);
215 Py_DECREF(module);
216 Py_DECREF(runargs);
217 if (result == NULL) {
218 return -1;
219 }
220 Py_DECREF(result);
221 return 0;
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +0000222}
Barry Warsaw3b2aedb2000-09-15 18:40:42 +0000223
Victor Stinner4726e402010-10-06 23:24:57 +0000224static int
225RunMainFromImporter(wchar_t *filename)
Christian Heimes9cd17752007-11-18 19:35:23 +0000226{
Victor Stinner4726e402010-10-06 23:24:57 +0000227 PyObject *argv0 = NULL, *importer, *sys_path;
228 int sts;
Christian Heimes9cd17752007-11-18 19:35:23 +0000229
Victor Stinner4726e402010-10-06 23:24:57 +0000230 argv0 = PyUnicode_FromWideChar(filename, wcslen(filename));
231 if (argv0 == NULL)
232 goto error;
233
234 importer = PyImport_GetImporter(argv0);
235 if (importer == NULL)
236 goto error;
237
Brett Cannonaa936422012-04-27 15:30:58 -0400238 if (importer == Py_None) {
Victor Stinner4726e402010-10-06 23:24:57 +0000239 Py_DECREF(argv0);
240 Py_DECREF(importer);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000241 return -1;
242 }
Victor Stinner4726e402010-10-06 23:24:57 +0000243 Py_DECREF(importer);
244
245 /* argv0 is usable as an import source, so put it in sys.path[0]
246 and import __main__ */
Victor Stinnerbd303c12013-11-07 23:07:29 +0100247 sys_path = PySys_GetObject("path");
Victor Stinner1e53bba2013-07-16 22:26:05 +0200248 if (sys_path == NULL) {
249 PyErr_SetString(PyExc_RuntimeError, "unable to get sys.path");
Victor Stinner4726e402010-10-06 23:24:57 +0000250 goto error;
Victor Stinner1e53bba2013-07-16 22:26:05 +0200251 }
Victor Stinner4726e402010-10-06 23:24:57 +0000252 if (PyList_SetItem(sys_path, 0, argv0)) {
253 argv0 = NULL;
254 goto error;
255 }
256 Py_INCREF(argv0);
257
258 sts = RunModule(L"__main__", 0);
259 return sts != 0;
260
261error:
262 Py_XDECREF(argv0);
263 PyErr_Print();
264 return 1;
Christian Heimes9cd17752007-11-18 19:35:23 +0000265}
266
Victor Stinnera62207c2010-08-07 10:57:17 +0000267static int
268run_command(wchar_t *command, PyCompilerFlags *cf)
269{
270 PyObject *unicode, *bytes;
271 int ret;
272
273 unicode = PyUnicode_FromWideChar(command, -1);
274 if (unicode == NULL)
275 goto error;
276 bytes = PyUnicode_AsUTF8String(unicode);
277 Py_DECREF(unicode);
278 if (bytes == NULL)
279 goto error;
280 ret = PyRun_SimpleStringFlags(PyBytes_AsString(bytes), cf);
281 Py_DECREF(bytes);
282 return ret != 0;
283
284error:
Victor Stinner398356b2010-08-18 22:23:22 +0000285 PySys_WriteStderr("Unable to decode the command from the command line:\n");
Victor Stinnera62207c2010-08-07 10:57:17 +0000286 PyErr_Print();
287 return 1;
288}
289
Victor Stinner0a3ddad2010-08-07 16:34:25 +0000290static int
291run_file(FILE *fp, const wchar_t *filename, PyCompilerFlags *p_cf)
292{
293 PyObject *unicode, *bytes = NULL;
294 char *filename_str;
295 int run;
296
297 /* call pending calls like signal handlers (SIGINT) */
298 if (Py_MakePendingCalls() == -1) {
299 PyErr_Print();
300 return 1;
301 }
302
303 if (filename) {
304 unicode = PyUnicode_FromWideChar(filename, wcslen(filename));
305 if (unicode != NULL) {
Victor Stinnere0f32682010-10-17 19:34:51 +0000306 bytes = PyUnicode_EncodeFSDefault(unicode);
Victor Stinner0a3ddad2010-08-07 16:34:25 +0000307 Py_DECREF(unicode);
308 }
309 if (bytes != NULL)
310 filename_str = PyBytes_AsString(bytes);
311 else {
312 PyErr_Clear();
Victor Stinnere0f32682010-10-17 19:34:51 +0000313 filename_str = "<encoding error>";
Victor Stinner0a3ddad2010-08-07 16:34:25 +0000314 }
315 }
316 else
317 filename_str = "<stdin>";
318
319 run = PyRun_AnyFileExFlags(fp, filename_str, filename != NULL, p_cf);
320 Py_XDECREF(bytes);
321 return run != 0;
322}
323
Christian Heimes9cd17752007-11-18 19:35:23 +0000324
Guido van Rossum667d7041995-08-04 04:20:48 +0000325/* Main program */
326
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000327int
Martin v. Löwis790465f2008-04-05 20:41:37 +0000328Py_Main(int argc, wchar_t **argv)
Guido van Rossum667d7041995-08-04 04:20:48 +0000329{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000330 int c;
331 int sts;
332 wchar_t *command = NULL;
333 wchar_t *filename = NULL;
334 wchar_t *module = NULL;
335 FILE *fp = stdin;
336 char *p;
Philip Jenveye53de3d2010-04-14 03:01:39 +0000337#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000338 wchar_t *wp;
Philip Jenveye53de3d2010-04-14 03:01:39 +0000339#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000340 int skipfirstline = 0;
341 int stdin_is_interactive = 0;
342 int help = 0;
343 int version = 0;
344 int saw_unbuffered_flag = 0;
Victor Stinner34be8072016-03-14 12:04:26 +0100345 char *opt;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000346 PyCompilerFlags cf;
Antoine Pitrou69994412014-04-29 00:56:08 +0200347 PyObject *warning_option = NULL;
348 PyObject *warning_options = NULL;
Guido van Rossum667d7041995-08-04 04:20:48 +0000349
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000350 cf.cf_flags = 0;
Guido van Rossum393661d2001-08-31 17:40:15 +0000351
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000352 orig_argc = argc; /* For Py_GetArgcArgv() */
353 orig_argv = argv;
Guido van Rossum667d7041995-08-04 04:20:48 +0000354
Antoine Pitrou86838b02012-02-21 19:03:47 +0100355 /* Hash randomization needed early for all string operations
356 (including -W and -X options). */
Ezio Melottia0dd22e2012-11-23 18:48:32 +0200357 _PyOS_opterr = 0; /* prevent printing the error in 1st pass */
Antoine Pitrou86838b02012-02-21 19:03:47 +0100358 while ((c = _PyOS_GetOpt(argc, argv, PROGRAM_OPTS)) != EOF) {
359 if (c == 'm' || c == 'c') {
360 /* -c / -m is the last option: following arguments are
361 not interpreter options. */
362 break;
363 }
Benjamin Petersonc9f54cf2012-02-21 16:08:05 -0500364 if (c == 'E') {
Antoine Pitrou86838b02012-02-21 19:03:47 +0100365 Py_IgnoreEnvironmentFlag++;
366 break;
Antoine Pitrou86838b02012-02-21 19:03:47 +0100367 }
368 }
Antoine Pitrou86838b02012-02-21 19:03:47 +0100369
Victor Stinner34be8072016-03-14 12:04:26 +0100370 opt = Py_GETENV("PYTHONMALLOC");
371 if (_PyMem_SetupAllocators(opt) < 0) {
372 fprintf(stderr,
373 "Error in PYTHONMALLOC: unknown allocator \"%s\"!\n", opt);
374 exit(1);
375 }
376
Benjamin Petersonc9f54cf2012-02-21 16:08:05 -0500377 Py_HashRandomizationFlag = 1;
Antoine Pitrou86838b02012-02-21 19:03:47 +0100378 _PyRandom_Init();
379
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000380 PySys_ResetWarnOptions();
Antoine Pitrou86838b02012-02-21 19:03:47 +0100381 _PyOS_ResetGetOpt();
Guido van Rossum47f5fdc2000-12-15 22:00:54 +0000382
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000383 while ((c = _PyOS_GetOpt(argc, argv, PROGRAM_OPTS)) != EOF) {
384 if (c == 'c') {
385 size_t len;
386 /* -c is the last option; following arguments
387 that look like options are left for the
388 command to interpret. */
Amaury Forgeot d'Arc9a5499b2008-11-11 23:04:59 +0000389
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000390 len = wcslen(_PyOS_optarg) + 1 + 1;
Victor Stinner1a7425f2013-07-07 16:25:15 +0200391 command = (wchar_t *)PyMem_RawMalloc(sizeof(wchar_t) * len);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000392 if (command == NULL)
393 Py_FatalError(
394 "not enough memory to copy -c argument");
395 wcscpy(command, _PyOS_optarg);
396 command[len - 2] = '\n';
397 command[len - 1] = 0;
398 break;
399 }
Guido van Rossum393661d2001-08-31 17:40:15 +0000400
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000401 if (c == 'm') {
402 /* -m is the last option; following arguments
403 that look like options are left for the
404 module to interpret. */
405 module = _PyOS_optarg;
406 break;
407 }
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +0000408
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000409 switch (c) {
410 case 'b':
411 Py_BytesWarningFlag++;
412 break;
Guido van Rossum667d7041995-08-04 04:20:48 +0000413
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000414 case 'd':
415 Py_DebugFlag++;
416 break;
Guido van Rossum667d7041995-08-04 04:20:48 +0000417
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000418 case 'i':
419 Py_InspectFlag++;
420 Py_InteractiveFlag++;
421 break;
Guido van Rossum667d7041995-08-04 04:20:48 +0000422
Christian Heimesad73a9c2013-08-10 16:36:18 +0200423 case 'I':
424 Py_IsolatedFlag++;
425 Py_NoUserSiteDirectory++;
426 Py_IgnoreEnvironmentFlag++;
427 break;
428
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000429 /* case 'J': reserved for Jython */
Christian Heimes33fe8092008-04-13 13:53:33 +0000430
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000431 case 'O':
432 Py_OptimizeFlag++;
433 break;
Guido van Rossum7614da61997-03-03 19:14:45 +0000434
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000435 case 'B':
436 Py_DontWriteBytecodeFlag++;
437 break;
Christian Heimes790c8232008-01-07 21:14:23 +0000438
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000439 case 's':
440 Py_NoUserSiteDirectory++;
441 break;
Christian Heimes8dc226f2008-05-06 23:45:46 +0000442
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000443 case 'S':
444 Py_NoSiteFlag++;
445 break;
Guido van Rossum7922bd71997-08-29 22:34:47 +0000446
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000447 case 'E':
Antoine Pitrou86838b02012-02-21 19:03:47 +0100448 /* Already handled above */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000449 break;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000450
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000451 case 't':
452 /* ignored for backwards compatibility */
453 break;
Guido van Rossumbba92ca1998-04-10 19:39:15 +0000454
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000455 case 'u':
456 Py_UnbufferedStdioFlag = 1;
457 saw_unbuffered_flag = 1;
458 break;
Guido van Rossum667d7041995-08-04 04:20:48 +0000459
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000460 case 'v':
461 Py_VerboseFlag++;
462 break;
Guido van Rossum667d7041995-08-04 04:20:48 +0000463
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000464 case 'x':
465 skipfirstline = 1;
466 break;
Guido van Rossuma075ce11997-12-05 21:56:45 +0000467
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000468 case 'h':
469 case '?':
470 help++;
471 break;
Guido van Rossum45aecf42006-03-15 04:58:47 +0000472
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000473 case 'V':
474 version++;
475 break;
Guido van Rossumc15a9a12000-05-01 17:54:33 +0000476
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000477 case 'W':
Antoine Pitrou69994412014-04-29 00:56:08 +0200478 if (warning_options == NULL)
479 warning_options = PyList_New(0);
480 if (warning_options == NULL)
481 Py_FatalError("failure in handling of -W argument");
482 warning_option = PyUnicode_FromWideChar(_PyOS_optarg, -1);
483 if (warning_option == NULL)
484 Py_FatalError("failure in handling of -W argument");
485 PyList_Append(warning_options, warning_option);
486 Py_DECREF(warning_option);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000487 break;
Guido van Rossum47f5fdc2000-12-15 22:00:54 +0000488
Antoine Pitrou9583cac2010-10-21 13:42:28 +0000489 case 'X':
490 PySys_AddXOption(_PyOS_optarg);
491 break;
492
Georg Brandl9d871192010-12-04 10:47:18 +0000493 case 'q':
Georg Brandl8aa7e992010-12-28 18:30:18 +0000494 Py_QuietFlag++;
Georg Brandl9d871192010-12-04 10:47:18 +0000495 break;
496
Georg Brandl2daf6ae2012-02-20 19:54:16 +0100497 case 'R':
Benjamin Petersonc9f54cf2012-02-21 16:08:05 -0500498 /* Ignored */
Georg Brandl2daf6ae2012-02-20 19:54:16 +0100499 break;
500
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000501 /* This space reserved for other options */
Guido van Rossum667d7041995-08-04 04:20:48 +0000502
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000503 default:
504 return usage(2, argv[0]);
505 /*NOTREACHED*/
Guido van Rossum667d7041995-08-04 04:20:48 +0000506
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000507 }
508 }
Guido van Rossum667d7041995-08-04 04:20:48 +0000509
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000510 if (help)
511 return usage(0, argv[0]);
Barry Warsaw3b2aedb2000-09-15 18:40:42 +0000512
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000513 if (version) {
Serhiy Storchakae3ed4ed2013-07-11 20:01:17 +0300514 printf("Python %s\n", PY_VERSION);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000515 return 0;
516 }
Barry Warsaw3b2aedb2000-09-15 18:40:42 +0000517
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000518 if (!Py_InspectFlag &&
519 (p = Py_GETENV("PYTHONINSPECT")) && *p != '\0')
520 Py_InspectFlag = 1;
521 if (!saw_unbuffered_flag &&
522 (p = Py_GETENV("PYTHONUNBUFFERED")) && *p != '\0')
523 Py_UnbufferedStdioFlag = 1;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000524
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000525 if (!Py_NoUserSiteDirectory &&
526 (p = Py_GETENV("PYTHONNOUSERSITE")) && *p != '\0')
527 Py_NoUserSiteDirectory = 1;
Christian Heimes8dc226f2008-05-06 23:45:46 +0000528
Philip Jenveye53de3d2010-04-14 03:01:39 +0000529#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000530 if (!Py_IgnoreEnvironmentFlag && (wp = _wgetenv(L"PYTHONWARNINGS")) &&
531 *wp != L'\0') {
Steve Dowerf63dab52015-02-25 20:48:01 -0800532 wchar_t *buf, *warning, *context = NULL;
Philip Jenvey0805ca32010-04-07 04:04:10 +0000533
Victor Stinner1a7425f2013-07-07 16:25:15 +0200534 buf = (wchar_t *)PyMem_RawMalloc((wcslen(wp) + 1) * sizeof(wchar_t));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000535 if (buf == NULL)
536 Py_FatalError(
537 "not enough memory to copy PYTHONWARNINGS");
538 wcscpy(buf, wp);
Steve Dowerf63dab52015-02-25 20:48:01 -0800539 for (warning = wcstok_s(buf, L",", &context);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000540 warning != NULL;
Steve Dowerf63dab52015-02-25 20:48:01 -0800541 warning = wcstok_s(NULL, L",", &context)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000542 PySys_AddWarnOption(warning);
543 }
Victor Stinner1a7425f2013-07-07 16:25:15 +0200544 PyMem_RawFree(buf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000545 }
Philip Jenveye53de3d2010-04-14 03:01:39 +0000546#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000547 if ((p = Py_GETENV("PYTHONWARNINGS")) && *p != '\0') {
548 char *buf, *oldloc;
Victor Stinnerc2d76fd2010-09-10 23:13:52 +0000549 PyObject *unicode;
Philip Jenveye53de3d2010-04-14 03:01:39 +0000550
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000551 /* settle for strtok here as there's no one standard
552 C89 wcstok */
Victor Stinner1a7425f2013-07-07 16:25:15 +0200553 buf = (char *)PyMem_RawMalloc(strlen(p) + 1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000554 if (buf == NULL)
555 Py_FatalError(
556 "not enough memory to copy PYTHONWARNINGS");
557 strcpy(buf, p);
Victor Stinner49fc8ec2013-07-07 23:30:24 +0200558 oldloc = _PyMem_RawStrdup(setlocale(LC_ALL, NULL));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000559 setlocale(LC_ALL, "");
560 for (p = strtok(buf, ","); p != NULL; p = strtok(NULL, ",")) {
Victor Stinner5c848a82010-09-12 08:00:41 +0000561#ifdef __APPLE__
562 /* Use utf-8 on Mac OS X */
563 unicode = PyUnicode_FromString(p);
564#else
Victor Stinner1b579672011-12-17 05:47:23 +0100565 unicode = PyUnicode_DecodeLocale(p, "surrogateescape");
Victor Stinner5c848a82010-09-12 08:00:41 +0000566#endif
Victor Stinneraf02e1c2011-12-16 23:56:01 +0100567 if (unicode == NULL) {
568 /* ignore errors */
569 PyErr_Clear();
Victor Stinnerc2d76fd2010-09-10 23:13:52 +0000570 continue;
Victor Stinneraf02e1c2011-12-16 23:56:01 +0100571 }
Victor Stinnerc2d76fd2010-09-10 23:13:52 +0000572 PySys_AddWarnOptionUnicode(unicode);
573 Py_DECREF(unicode);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 }
575 setlocale(LC_ALL, oldloc);
Victor Stinner49fc8ec2013-07-07 23:30:24 +0200576 PyMem_RawFree(oldloc);
Victor Stinner1a7425f2013-07-07 16:25:15 +0200577 PyMem_RawFree(buf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000578 }
Philip Jenveye53de3d2010-04-14 03:01:39 +0000579#endif
Antoine Pitrou69994412014-04-29 00:56:08 +0200580 if (warning_options != NULL) {
581 Py_ssize_t i;
582 for (i = 0; i < PyList_GET_SIZE(warning_options); i++) {
583 PySys_AddWarnOptionUnicode(PyList_GET_ITEM(warning_options, i));
584 }
585 }
Philip Jenvey0805ca32010-04-07 04:04:10 +0000586
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000587 if (command == NULL && module == NULL && _PyOS_optind < argc &&
588 wcscmp(argv[_PyOS_optind], L"-") != 0)
589 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000590 filename = argv[_PyOS_optind];
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000591 }
Guido van Rossum775af911997-02-14 19:50:32 +0000592
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000593 stdin_is_interactive = Py_FdIsInteractive(stdin, (char *)0);
Guido van Rossum775af911997-02-14 19:50:32 +0000594
Sjoerd Mullender9cf424b2002-08-09 13:35:18 +0000595#if defined(MS_WINDOWS) || defined(__CYGWIN__)
Victor Stinner89e34362011-01-07 18:47:22 +0000596 /* don't translate newlines (\r\n <=> \n) */
597 _setmode(fileno(stdin), O_BINARY);
598 _setmode(fileno(stdout), O_BINARY);
599 _setmode(fileno(stderr), O_BINARY);
Guido van Rossumf22d7e21997-01-11 19:28:55 +0000600#endif
Victor Stinner89e34362011-01-07 18:47:22 +0000601
602 if (Py_UnbufferedStdioFlag) {
Guido van Rossum22ffac11998-03-06 15:30:39 +0000603#ifdef HAVE_SETVBUF
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000604 setvbuf(stdin, (char *)NULL, _IONBF, BUFSIZ);
605 setvbuf(stdout, (char *)NULL, _IONBF, BUFSIZ);
606 setvbuf(stderr, (char *)NULL, _IONBF, BUFSIZ);
Guido van Rossum22ffac11998-03-06 15:30:39 +0000607#else /* !HAVE_SETVBUF */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000608 setbuf(stdin, (char *)NULL);
609 setbuf(stdout, (char *)NULL);
610 setbuf(stderr, (char *)NULL);
Guido van Rossum22ffac11998-03-06 15:30:39 +0000611#endif /* !HAVE_SETVBUF */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000612 }
613 else if (Py_InteractiveFlag) {
Guido van Rossumb31c7dc1997-04-11 22:19:12 +0000614#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000615 /* Doesn't have to have line-buffered -- use unbuffered */
616 /* Any set[v]buf(stdin, ...) screws up Tkinter :-( */
617 setvbuf(stdout, (char *)NULL, _IONBF, BUFSIZ);
Guido van Rossum22ffac11998-03-06 15:30:39 +0000618#else /* !MS_WINDOWS */
619#ifdef HAVE_SETVBUF
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000620 setvbuf(stdin, (char *)NULL, _IOLBF, BUFSIZ);
621 setvbuf(stdout, (char *)NULL, _IOLBF, BUFSIZ);
Guido van Rossum22ffac11998-03-06 15:30:39 +0000622#endif /* HAVE_SETVBUF */
623#endif /* !MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000624 /* Leave stderr alone - it should be unbuffered anyway. */
625 }
Guido van Rossum667d7041995-08-04 04:20:48 +0000626
Just van Rossum2ac79ef2003-03-05 15:46:54 +0000627#ifdef __APPLE__
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000628 /* On MacOS X, when the Python interpreter is embedded in an
629 application bundle, it gets executed by a bootstrapping script
630 that does os.execve() with an argv[0] that's different from the
631 actual Python executable. This is needed to keep the Finder happy,
632 or rather, to work around Apple's overly strict requirements of
633 the process name. However, we still need a usable sys.executable,
634 so the actual executable path is passed in an environment variable.
635 See Lib/plat-mac/bundlebuiler.py for details about the bootstrap
636 script. */
637 if ((p = Py_GETENV("PYTHONEXECUTABLE")) && *p != '\0') {
638 wchar_t* buffer;
Ronald Oussoreneb61f8b2012-08-22 14:24:14 +0200639 size_t len = strlen(p) + 1;
Ronald Oussoren3e264e12009-02-12 15:55:38 +0000640
Victor Stinner1a7425f2013-07-07 16:25:15 +0200641 buffer = PyMem_RawMalloc(len * sizeof(wchar_t));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000642 if (buffer == NULL) {
643 Py_FatalError(
644 "not enough memory to copy PYTHONEXECUTABLE");
645 }
Ronald Oussoren3e264e12009-02-12 15:55:38 +0000646
Brett Cannonb94767f2011-02-22 20:15:44 +0000647 mbstowcs(buffer, p, len);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000648 Py_SetProgramName(buffer);
649 /* buffer is now handed off - do not free */
650 } else {
Vinay Sajip90db6612012-07-17 17:33:46 +0100651#ifdef WITH_NEXT_FRAMEWORK
652 char* pyvenv_launcher = getenv("__PYVENV_LAUNCHER__");
653
654 if (pyvenv_launcher && *pyvenv_launcher) {
655 /* Used by Mac/Tools/pythonw.c to forward
656 * the argv0 of the stub executable
657 */
Victor Stinnerf6a271a2014-08-01 12:28:48 +0200658 wchar_t* wbuf = Py_DecodeLocale(pyvenv_launcher, NULL);
Vinay Sajip90db6612012-07-17 17:33:46 +0100659
660 if (wbuf == NULL) {
661 Py_FatalError("Cannot decode __PYVENV_LAUNCHER__");
662 }
663 Py_SetProgramName(wbuf);
664
665 /* Don't free wbuf, the argument to Py_SetProgramName
Martin Panterb4ce1fc2015-11-30 03:18:29 +0000666 * must remain valid until Py_FinalizeEx is called.
Vinay Sajip90db6612012-07-17 17:33:46 +0100667 */
668 } else {
669 Py_SetProgramName(argv[0]);
670 }
671#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000672 Py_SetProgramName(argv[0]);
Vinay Sajip90db6612012-07-17 17:33:46 +0100673#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000674 }
Just van Rossum2ac79ef2003-03-05 15:46:54 +0000675#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000676 Py_SetProgramName(argv[0]);
Jack Jansenfbd861b2003-03-05 16:00:15 +0000677#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000678 Py_Initialize();
Antoine Pitrou69994412014-04-29 00:56:08 +0200679 Py_XDECREF(warning_options);
Guido van Rossumed52aac1997-07-19 19:20:32 +0000680
Georg Brandl8aa7e992010-12-28 18:30:18 +0000681 if (!Py_QuietFlag && (Py_VerboseFlag ||
Georg Brandl9d871192010-12-04 10:47:18 +0000682 (command == NULL && filename == NULL &&
683 module == NULL && stdin_is_interactive))) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000684 fprintf(stderr, "Python %s on %s\n",
685 Py_GetVersion(), Py_GetPlatform());
686 if (!Py_NoSiteFlag)
687 fprintf(stderr, "%s\n", COPYRIGHT);
688 }
Guido van Rossum393661d2001-08-31 17:40:15 +0000689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000690 if (command != NULL) {
691 /* Backup _PyOS_optind and force sys.argv[0] = '-c' */
692 _PyOS_optind--;
693 argv[_PyOS_optind] = L"-c";
694 }
Guido van Rossum667d7041995-08-04 04:20:48 +0000695
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000696 if (module != NULL) {
Nick Coghland26c18a2010-08-17 13:06:11 +0000697 /* Backup _PyOS_optind and force sys.argv[0] = '-m'*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000698 _PyOS_optind--;
Nick Coghland26c18a2010-08-17 13:06:11 +0000699 argv[_PyOS_optind] = L"-m";
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000700 }
Raymond Hettingerdb29e0f2004-10-07 06:46:25 +0000701
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000702 PySys_SetArgv(argc-_PyOS_optind, argv+_PyOS_optind);
Guido van Rossum667d7041995-08-04 04:20:48 +0000703
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000704 if ((Py_InspectFlag || (command == NULL && filename == NULL && module == NULL)) &&
705 isatty(fileno(stdin))) {
706 PyObject *v;
707 v = PyImport_ImportModule("readline");
708 if (v == NULL)
709 PyErr_Clear();
710 else
711 Py_DECREF(v);
712 }
Guido van Rossum3d26cc91997-09-16 16:11:28 +0000713
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000714 if (command) {
Victor Stinnera62207c2010-08-07 10:57:17 +0000715 sts = run_command(command, &cf);
Victor Stinner1a7425f2013-07-07 16:25:15 +0200716 PyMem_RawFree(command);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000717 } else if (module) {
Senthil Kumaranf01a3372012-07-04 19:28:16 -0700718 sts = (RunModule(module, 1) != 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000719 }
720 else {
Christian Heimes9cd17752007-11-18 19:35:23 +0000721
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000722 if (filename == NULL && stdin_is_interactive) {
723 Py_InspectFlag = 0; /* do exit on SystemExit */
724 RunStartupFile(&cf);
Antoine Pitrou1a6cb302013-05-04 20:08:35 +0200725 RunInteractiveHook();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000726 }
727 /* XXX */
Christian Heimes9cd17752007-11-18 19:35:23 +0000728
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000729 sts = -1; /* keep track of whether we've already run __main__ */
Christian Heimes9cd17752007-11-18 19:35:23 +0000730
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000731 if (filename != NULL) {
732 sts = RunMainFromImporter(filename);
733 }
Christian Heimes9cd17752007-11-18 19:35:23 +0000734
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000735 if (sts==-1 && filename!=NULL) {
Victor Stinner9a6692f2010-10-14 10:51:24 +0000736 fp = _Py_wfopen(filename, L"r");
737 if (fp == NULL) {
738 char *cfilename_buffer;
739 const char *cfilename;
Antoine Pitrouc345ce12011-12-16 12:28:32 +0100740 int err = errno;
Victor Stinnerf6a271a2014-08-01 12:28:48 +0200741 cfilename_buffer = Py_EncodeLocale(filename, NULL);
Victor Stinner9a6692f2010-10-14 10:51:24 +0000742 if (cfilename_buffer != NULL)
743 cfilename = cfilename_buffer;
744 else
745 cfilename = "<unprintable file name>";
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000746 fprintf(stderr, "%ls: can't open file '%s': [Errno %d] %s\n",
Antoine Pitrouc345ce12011-12-16 12:28:32 +0100747 argv[0], cfilename, err, strerror(err));
Victor Stinner9a6692f2010-10-14 10:51:24 +0000748 if (cfilename_buffer)
749 PyMem_Free(cfilename_buffer);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000750 return 2;
751 }
752 else if (skipfirstline) {
753 int ch;
754 /* Push back first newline so line numbers
755 remain the same */
756 while ((ch = getc(fp)) != EOF) {
757 if (ch == '\n') {
758 (void)ungetc(ch, fp);
759 break;
760 }
761 }
762 }
763 {
Steve Dowerf2f373f2015-02-21 08:44:05 -0800764 struct _Py_stat_struct sb;
Victor Stinnere134a7f2015-03-30 10:09:31 +0200765 if (_Py_fstat_noraise(fileno(fp), &sb) == 0 &&
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000766 S_ISDIR(sb.st_mode)) {
Victor Stinnere134a7f2015-03-30 10:09:31 +0200767 fprintf(stderr,
768 "%ls: '%ls' is a directory, cannot continue\n",
769 argv[0], filename);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000770 fclose(fp);
771 return 1;
772 }
773 }
774 }
Christian Heimes9cd17752007-11-18 19:35:23 +0000775
Victor Stinner0a3ddad2010-08-07 16:34:25 +0000776 if (sts == -1)
777 sts = run_file(fp, filename, &cf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000778 }
Barry Warsawd86dcd32003-06-29 17:07:06 +0000779
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000780 /* Check this environment variable at the end, to give programs the
781 * opportunity to set it from Python.
782 */
783 if (!Py_InspectFlag &&
784 (p = Py_GETENV("PYTHONINSPECT")) && *p != '\0')
785 {
786 Py_InspectFlag = 1;
787 }
Guido van Rossum667d7041995-08-04 04:20:48 +0000788
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000789 if (Py_InspectFlag && stdin_is_interactive &&
790 (filename != NULL || command != NULL || module != NULL)) {
791 Py_InspectFlag = 0;
Antoine Pitrou1a6cb302013-05-04 20:08:35 +0200792 RunInteractiveHook();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000793 /* XXX */
794 sts = PyRun_AnyFileFlags(stdin, "<stdin>", &cf) != 0;
795 }
796
Martin Panterb4ce1fc2015-11-30 03:18:29 +0000797 if (Py_FinalizeEx() < 0) {
798 /* Value unlikely to be confused with a non-error exit status or
799 other special meaning */
800 sts = 120;
801 }
Barry Warsaw3e13b1e2001-02-23 16:46:39 +0000802
803#ifdef __INSURE__
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000804 /* Insure++ is a memory analysis tool that aids in discovering
805 * memory leaks and other memory problems. On Python exit, the
806 * interned string dictionaries are flagged as being in use at exit
807 * (which it is). Under normal circumstances, this is fine because
808 * the memory will be automatically reclaimed by the system. Under
809 * memory debugging, it's a huge source of useless noise, so we
810 * trade off slower shutdown for less distraction in the memory
811 * reports. -baw
812 */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000813 _Py_ReleaseInternedUnicodeStrings();
Barry Warsaw3e13b1e2001-02-23 16:46:39 +0000814#endif /* __INSURE__ */
815
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000816 return sts;
Guido van Rossum667d7041995-08-04 04:20:48 +0000817}
818
Skip Montanaro786ea6b2004-03-01 15:44:05 +0000819/* this is gonna seem *real weird*, but if you put some other code between
820 Py_Main() and Py_GetArgcArgv() you will need to adjust the test in the
821 while statement in Misc/gdbinit:ppystack */
Guido van Rossum667d7041995-08-04 04:20:48 +0000822
Guido van Rossum667d7041995-08-04 04:20:48 +0000823/* Make the *original* argc/argv available to other modules.
824 This is rare, but it is needed by the secureware extension. */
825
826void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000827Py_GetArgcArgv(int *argc, wchar_t ***argv)
Guido van Rossum667d7041995-08-04 04:20:48 +0000828{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000829 *argc = orig_argc;
830 *argv = orig_argv;
Guido van Rossum667d7041995-08-04 04:20:48 +0000831}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000832
833#ifdef __cplusplus
834}
835#endif