main.c revision f63dab5a842df7349af73be23fb3acc37aa75703
1667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum/* Python interpreter main program */ 2667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 3667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum#include "Python.h" 4a075ce1618c30c7d0faf4de6e84920027da40e41Guido van Rossum#include "osdefs.h" 5667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 65651eaa720e28fb95b78524524ee75bda2d0ae59Antoine Pitrou#include <locale.h> 75651eaa720e28fb95b78524524ee75bda2d0ae59Antoine Pitrou 89cf424b04bf355df85659da18abc2ac3f9419c37Sjoerd Mullender#if defined(MS_WINDOWS) || defined(__CYGWIN__) 9945362cf971fb2e10f8f2a8e71e8ff10516ebe4cMartin v. Löwis#include <windows.h> 10477c8d5e70240744d24631b18341ad892c8a8e1cThomas Wouters#ifdef HAVE_FCNTL_H 113e7ae7ab173effc320b05a3d5a9fde7eac706771Guido van Rossum#include <fcntl.h> 123e7ae7ab173effc320b05a3d5a9fde7eac706771Guido van Rossum#endif 13477c8d5e70240744d24631b18341ad892c8a8e1cThomas Wouters#endif 143e7ae7ab173effc320b05a3d5a9fde7eac706771Guido van Rossum 15945362cf971fb2e10f8f2a8e71e8ff10516ebe4cMartin v. Löwis#ifdef _MSC_VER 16945362cf971fb2e10f8f2a8e71e8ff10516ebe4cMartin v. Löwis#include <crtdbg.h> 17945362cf971fb2e10f8f2a8e71e8ff10516ebe4cMartin v. Löwis#endif 18945362cf971fb2e10f8f2a8e71e8ff10516ebe4cMartin v. Löwis 19ab70e2ab3264c444ac70a1c98e869b0de0b22c4aJesus Cea#if defined(MS_WINDOWS) 20a075ce1618c30c7d0faf4de6e84920027da40e41Guido van Rossum#define PYTHONHOMEHELP "<prefix>\\lib" 21a075ce1618c30c7d0faf4de6e84920027da40e41Guido van Rossum#else 22da4dbc36c102d5005737365f65627bbca1d2395bMarc-André Lemburg#define PYTHONHOMEHELP "<prefix>/pythonX.X" 23a075ce1618c30c7d0faf4de6e84920027da40e41Guido van Rossum#endif 24a075ce1618c30c7d0faf4de6e84920027da40e41Guido van Rossum 252cffc7d4202fc1197280a05d998075551b459283Thomas Wouters#include "pygetopt.h" 262cffc7d4202fc1197280a05d998075551b459283Thomas Wouters 27a22865ec4400438a8b591c67e26ec69722ac9266Guido van Rossum#define COPYRIGHT \ 2836002d7af1a9971c62e982cbbb1d1ce34610129dGuido van Rossum "Type \"help\", \"copyright\", \"credits\" or \"license\" " \ 2936002d7af1a9971c62e982cbbb1d1ce34610129dGuido van Rossum "for more information." 30a22865ec4400438a8b591c67e26ec69722ac9266Guido van Rossum 3149fd7fa4431da299196d74087df4a04f99f9c46fThomas Wouters#ifdef __cplusplus 3249fd7fa4431da299196d74087df4a04f99f9c46fThomas Woutersextern "C" { 3349fd7fa4431da299196d74087df4a04f99f9c46fThomas Wouters#endif 3449fd7fa4431da299196d74087df4a04f99f9c46fThomas Wouters 35ac56b03f9a931423714ab3a0e8cfae78f81f9b85Guido van Rossum/* For Py_GetArgcArgv(); set by main() */ 36790465fd90e8a72590386465f518db9e67ab843fMartin v. Löwisstatic wchar_t **orig_argv; 37667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossumstatic int orig_argc; 38667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 39bceccf5f43b51d166951ea97cff086c8828b745bGuido van Rossum/* command line options */ 40ad73a9cf97770023665a1bb1c6390a3c99478139Christian Heimes#define BASE_OPTS L"bBc:dEhiIJm:OqRsStuvVW:xX:?" 41bceccf5f43b51d166951ea97cff086c8828b745bGuido van Rossum 42bceccf5f43b51d166951ea97cff086c8828b745bGuido van Rossum#define PROGRAM_OPTS BASE_OPTS 433ed4c15a88c79ae71a01854415a66ca1cd6697c7Guido van Rossum 44667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum/* Short usage message (with %s for argv0) */ 45667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossumstatic char *usage_line = 46790465fd90e8a72590386465f518db9e67ab843fMartin v. Löwis"usage: %ls [option] ... [-c cmd | -m mod | file | -] [arg] ...\n"; 47667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 48667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum/* Long usage message, split into parts < 512 bytes */ 49393661d15fa76ec46e9f7b71e97bf2ed38470620Guido van Rossumstatic char *usage_1 = "\ 50667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van RossumOptions and arguments (and corresponding environment variables):\n\ 512ab344428449792e293c0c4d276fc66d26e919e5Christian Heimes-b : issue warnings about str(bytes_instance), str(bytearray_instance)\n\ 522ab344428449792e293c0c4d276fc66d26e919e5Christian Heimes and comparing bytes/bytearray with str. (-bb: issue errors)\n\ 53790c8232019d0a13c3f0a72b8cffcf3ae69ea7b9Christian Heimes-B : don't write .py[co] files on import; also PYTHONDONTWRITEBYTECODE=x\n\ 54393661d15fa76ec46e9f7b71e97bf2ed38470620Guido van Rossum-c cmd : program passed in as string (terminates option list)\n\ 5589f507fe8c497b3f70fdcecce8bc240f9af2bbe2Thomas Wouters-d : debug output from parser; also PYTHONDEBUG=x\n\ 56790c8232019d0a13c3f0a72b8cffcf3ae69ea7b9Christian Heimes-E : ignore PYTHON* environment variables (such as PYTHONPATH)\n\ 570e3f591aeeef9ed715f8770320f4c4c7332a8794Thomas Wouters-h : print this help message and exit (also --help)\n\ 5861c345fa37573005b708d239c0911218ca22383cGuido van Rossum"; 5961c345fa37573005b708d239c0911218ca22383cGuido van Rossumstatic char *usage_2 = "\ 6098297ee7815939b124156e438b22bd652d67b5dbGuido van Rossum-i : inspect interactively after running script; forces a prompt even\n\ 6198297ee7815939b124156e438b22bd652d67b5dbGuido van Rossum if stdin does not appear to be a terminal; also PYTHONINSPECT=x\n\ 62ad73a9cf97770023665a1bb1c6390a3c99478139Christian Heimes-I : isolate Python from the user's environment (implies -E and -s)\n\ 63db29e0fe8cab58c68d1263487ba5bac7b99c7612Raymond Hettinger-m mod : run library module as a script (terminates option list)\n\ 6489f507fe8c497b3f70fdcecce8bc240f9af2bbe2Thomas Wouters-O : optimize generated bytecode slightly; also PYTHONOPTIMIZE=x\n\ 656b86a42c7f8786f8446d5a48db406282f18ec06aGuido van Rossum-OO : remove doc-strings in addition to the -O optimizations\n\ 669d8711964f114b9f0f5ed8a9a1eccfba2a096030Georg Brandl-q : don't print version and copyright messages on interactive startup\n\ 678dc226fccd2183670ac19a85b89941f5b05167dcChristian Heimes-s : don't add user site directory to sys.path; also PYTHONNOUSERSITE\n\ 687922bd7382c7e22ee9f711f6554718b824ac85a4Guido van Rossum-S : don't imply 'import site' on initialization\n\ 69393661d15fa76ec46e9f7b71e97bf2ed38470620Guido van Rossum"; 70393661d15fa76ec46e9f7b71e97bf2ed38470620Guido van Rossumstatic char *usage_3 = "\ 7161b0c672b5e9ba1cf564947ecb38d175fd70ea7eEzio Melotti-u : unbuffered binary stdout and stderr, stdin always buffered;\n\ 7261b0c672b5e9ba1cf564947ecb38d175fd70ea7eEzio Melotti also PYTHONUNBUFFERED=x\n\ 73db29e0fe8cab58c68d1263487ba5bac7b99c7612Raymond Hettinger see man page for details on internal buffering relating to '-u'\n\ 7489f507fe8c497b3f70fdcecce8bc240f9af2bbe2Thomas Wouters-v : verbose (trace import statements); also PYTHONVERBOSE=x\n\ 7589f507fe8c497b3f70fdcecce8bc240f9af2bbe2Thomas Wouters can be supplied multiple times to increase verbosity\n\ 760e3f591aeeef9ed715f8770320f4c4c7332a8794Thomas Wouters-V : print the Python version number and exit (also --version)\n\ 7789f507fe8c497b3f70fdcecce8bc240f9af2bbe2Thomas Wouters-W arg : warning control; arg is action:message:category:module:lineno\n\ 780805ca3f931ccd4735dcaa231752371deadf5ccfPhilip Jenvey also PYTHONWARNINGS=arg\n\ 79393661d15fa76ec46e9f7b71e97bf2ed38470620Guido van Rossum-x : skip first line of source, allowing use of non-Unix forms of #!cmd\n\ 809583cac6337f9a5f2670fbe5e1f2e85aaad04522Antoine Pitrou-X opt : set implementation-specific option\n\ 817922bd7382c7e22ee9f711f6554718b824ac85a4Guido van Rossum"; 82393661d15fa76ec46e9f7b71e97bf2ed38470620Guido van Rossumstatic char *usage_4 = "\ 8398297ee7815939b124156e438b22bd652d67b5dbGuido van Rossumfile : program read from script file\n\ 8498297ee7815939b124156e438b22bd652d67b5dbGuido van Rossum- : program read from stdin (default; interactive mode if a tty)\n\ 8589f507fe8c497b3f70fdcecce8bc240f9af2bbe2Thomas Woutersarg ...: arguments passed to program in sys.argv[1:]\n\n\ 86667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van RossumOther environment variables:\n\ 87667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van RossumPYTHONSTARTUP: file executed on interactive startup (no default)\n\ 88a075ce1618c30c7d0faf4de6e84920027da40e41Guido van RossumPYTHONPATH : '%c'-separated list of directories prefixed to the\n\ 89667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum default module search path. The result is sys.path.\n\ 90790c8232019d0a13c3f0a72b8cffcf3ae69ea7b9Christian Heimes"; 919802b39c12712e924bed699812756ce5076b77a3Victor Stinnerstatic char *usage_5 = 929802b39c12712e924bed699812756ce5076b77a3Victor Stinner"PYTHONHOME : alternate <prefix> directory (or <prefix>%c<exec_prefix>).\n" 939802b39c12712e924bed699812756ce5076b77a3Victor Stinner" The default module search path uses %s.\n" 949802b39c12712e924bed699812756ce5076b77a3Victor Stinner"PYTHONCASEOK : ignore case in 'import' statements (Windows).\n" 959802b39c12712e924bed699812756ce5076b77a3Victor Stinner"PYTHONIOENCODING: Encoding[:errors] used for stdin/stdout/stderr.\n" 962fb477c0f0284439d40cb3f46eea45ef42446e53Georg Brandl"PYTHONFAULTHANDLER: dump the Python traceback on fatal errors.\n\ 97667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum"; 982daf6ae2495c862adf8bc717bfe9964081ea0b10Georg Brandlstatic char *usage_6 = "\ 9942ae472798d4be46038ea7769ee4cabbcc0e3be0Georg BrandlPYTHONHASHSEED: if this variable is set to 'random', a random value is used\n\ 100c9f54cf512996790266c17f81584c9725ee99d47Benjamin Peterson to seed the hashes of str, bytes and datetime objects. It can also be\n\ 101c9f54cf512996790266c17f81584c9725ee99d47Benjamin Peterson set to an integer in the range [0,4294967295] to get hash values with a\n\ 102c9f54cf512996790266c17f81584c9725ee99d47Benjamin Peterson predictable seed.\n\ 1032daf6ae2495c862adf8bc717bfe9964081ea0b10Georg Brandl"; 104667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 105852ba7eb2a38a4d57cea73bf34b11f8e65f004e1Martin v. Löwisstatic int 106790465fd90e8a72590386465f518db9e67ab843fMartin v. Löwisusage(int exitcode, wchar_t* program) 1073b2aedbdc58c3466469abf4f3fbfc123195ff1f9Barry Warsaw{ 108f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou FILE *f = exitcode ? stderr : stdout; 109f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 110f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(f, usage_line, program); 111f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (exitcode) 112f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(f, "Try `python -h' for more information.\n"); 113f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou else { 114f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fputs(usage_1, f); 115f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fputs(usage_2, f); 116f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fputs(usage_3, f); 117f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(f, usage_4, DELIM); 118f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(f, usage_5, DELIM, PYTHONHOMEHELP); 1192daf6ae2495c862adf8bc717bfe9964081ea0b10Georg Brandl fputs(usage_6, f); 120f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 121f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return exitcode; 1223b2aedbdc58c3466469abf4f3fbfc123195ff1f9Barry Warsaw} 1233b2aedbdc58c3466469abf4f3fbfc123195ff1f9Barry Warsaw 1246caea370ac3e79e8295e456b566a04817eefb0fdMartin v. Löwisstatic void RunStartupFile(PyCompilerFlags *cf) 1256caea370ac3e79e8295e456b566a04817eefb0fdMartin v. Löwis{ 126f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou char *startup = Py_GETENV("PYTHONSTARTUP"); 127f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (startup != NULL && startup[0] != '\0') { 128daf455554bc21b6b5df0a016ab5fa639d36cc595Victor Stinner FILE *fp = _Py_fopen(startup, "r"); 129f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (fp != NULL) { 130f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou (void) PyRun_SimpleFileExFlags(fp, startup, 0, cf); 131f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyErr_Clear(); 132f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fclose(fp); 133f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } else { 134f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int save_errno; 135f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 136f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou save_errno = errno; 137f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PySys_WriteStderr("Could not open PYTHONSTARTUP\n"); 138f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou errno = save_errno; 139f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyErr_SetFromErrnoWithFilename(PyExc_IOError, 140f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou startup); 141f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyErr_Print(); 142f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyErr_Clear(); 143f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 144f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 1456caea370ac3e79e8295e456b566a04817eefb0fdMartin v. Löwis} 1466caea370ac3e79e8295e456b566a04817eefb0fdMartin v. Löwis 1471a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitroustatic void RunInteractiveHook(void) 1481a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou{ 1491a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou PyObject *sys, *hook, *result; 1501a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou sys = PyImport_ImportModule("sys"); 1511a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou if (sys == NULL) 1521a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou goto error; 1531a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou hook = PyObject_GetAttrString(sys, "__interactivehook__"); 1541a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou Py_DECREF(sys); 1551a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou if (hook == NULL) 1561a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou PyErr_Clear(); 1571a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou else { 1581a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou result = PyObject_CallObject(hook, NULL); 1591a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou Py_DECREF(hook); 1601a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou if (result == NULL) 1611a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou goto error; 1621a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou else 1631a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou Py_DECREF(result); 1641a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou } 1651a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou return; 1661a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou 1671a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrouerror: 1681a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou PySys_WriteStderr("Failed calling sys.__interactivehook__\n"); 1691a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou PyErr_Print(); 1701a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou PyErr_Clear(); 1711a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou} 1721a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou 173db29e0fe8cab58c68d1263487ba5bac7b99c7612Raymond Hettinger 1745651eaa720e28fb95b78524524ee75bda2d0ae59Antoine Pitroustatic int RunModule(wchar_t *modname, int set_argv0) 175a977329b6fb0e4c95cabb9043794de69b27a1099Thomas Wouters{ 176f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyObject *module, *runpy, *runmodule, *runargs, *result; 177f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou runpy = PyImport_ImportModule("runpy"); 178f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (runpy == NULL) { 179f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(stderr, "Could not import runpy module\n"); 1807d36e4f074a0141966394284f877bbe2270ccd93Victor Stinner PyErr_Print(); 181f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return -1; 182f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 183f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou runmodule = PyObject_GetAttrString(runpy, "_run_module_as_main"); 184f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (runmodule == NULL) { 185f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(stderr, "Could not access runpy._run_module_as_main\n"); 1867d36e4f074a0141966394284f877bbe2270ccd93Victor Stinner PyErr_Print(); 187f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(runpy); 188f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return -1; 189f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 190f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou module = PyUnicode_FromWideChar(modname, wcslen(modname)); 191f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (module == NULL) { 192f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(stderr, "Could not convert module name to unicode\n"); 1937d36e4f074a0141966394284f877bbe2270ccd93Victor Stinner PyErr_Print(); 194f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(runpy); 195f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(runmodule); 196f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return -1; 197f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 198f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou runargs = Py_BuildValue("(Oi)", module, set_argv0); 199f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (runargs == NULL) { 200f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(stderr, 201f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou "Could not create arguments for runpy._run_module_as_main\n"); 2027d36e4f074a0141966394284f877bbe2270ccd93Victor Stinner PyErr_Print(); 203f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(runpy); 204f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(runmodule); 205f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(module); 206f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return -1; 207f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 208f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou result = PyObject_Call(runmodule, runargs, NULL); 209f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (result == NULL) { 210f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyErr_Print(); 211f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 212f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(runpy); 213f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(runmodule); 214f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(module); 215f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(runargs); 216f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (result == NULL) { 217f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return -1; 218f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 219f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(result); 220f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return 0; 221db29e0fe8cab58c68d1263487ba5bac7b99c7612Raymond Hettinger} 2223b2aedbdc58c3466469abf4f3fbfc123195ff1f9Barry Warsaw 2234726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinnerstatic int 2244726e40e00ad9c2e333c4dfc5005f410a520834cVictor StinnerRunMainFromImporter(wchar_t *filename) 2259cd177526afc086a300b548588880329c32f607dChristian Heimes{ 2264726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner PyObject *argv0 = NULL, *importer, *sys_path; 2274726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner int sts; 228f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 2294726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner argv0 = PyUnicode_FromWideChar(filename, wcslen(filename)); 2304726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner if (argv0 == NULL) 2314726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner goto error; 2324726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner 2334726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner importer = PyImport_GetImporter(argv0); 2344726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner if (importer == NULL) 2354726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner goto error; 2364726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner 237aa93642a35ed570ba91a80c583df2c8a383686d6Brett Cannon if (importer == Py_None) { 2384726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner Py_DECREF(argv0); 2394726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner Py_DECREF(importer); 240f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return -1; 241f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 2424726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner Py_DECREF(importer); 2434726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner 2444726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner /* argv0 is usable as an import source, so put it in sys.path[0] 2454726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner and import __main__ */ 246bd303c165bc914fc943026ece55f4ebd225614c9Victor Stinner sys_path = PySys_GetObject("path"); 2471e53bbacedaed883104454693c29d1ad31f5029bVictor Stinner if (sys_path == NULL) { 2481e53bbacedaed883104454693c29d1ad31f5029bVictor Stinner PyErr_SetString(PyExc_RuntimeError, "unable to get sys.path"); 2494726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner goto error; 2501e53bbacedaed883104454693c29d1ad31f5029bVictor Stinner } 2514726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner if (PyList_SetItem(sys_path, 0, argv0)) { 2524726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner argv0 = NULL; 2534726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner goto error; 2544726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner } 2554726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner Py_INCREF(argv0); 2564726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner 2574726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner sts = RunModule(L"__main__", 0); 2584726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner return sts != 0; 2594726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner 2604726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinnererror: 2614726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner Py_XDECREF(argv0); 2624726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner PyErr_Print(); 2634726e40e00ad9c2e333c4dfc5005f410a520834cVictor Stinner return 1; 2649cd177526afc086a300b548588880329c32f607dChristian Heimes} 2659cd177526afc086a300b548588880329c32f607dChristian Heimes 266a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinnerstatic int 267a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinnerrun_command(wchar_t *command, PyCompilerFlags *cf) 268a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner{ 269a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner PyObject *unicode, *bytes; 270a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner int ret; 271a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner 272a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner unicode = PyUnicode_FromWideChar(command, -1); 273a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner if (unicode == NULL) 274a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner goto error; 275a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner bytes = PyUnicode_AsUTF8String(unicode); 276a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner Py_DECREF(unicode); 277a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner if (bytes == NULL) 278a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner goto error; 279a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner ret = PyRun_SimpleStringFlags(PyBytes_AsString(bytes), cf); 280a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner Py_DECREF(bytes); 281a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner return ret != 0; 282a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner 283a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinnererror: 284398356baaa60b1e46ce45ec7b8ae8ac28ca53091Victor Stinner PySys_WriteStderr("Unable to decode the command from the command line:\n"); 285a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner PyErr_Print(); 286a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner return 1; 287a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner} 288a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner 2890a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinnerstatic int 2900a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinnerrun_file(FILE *fp, const wchar_t *filename, PyCompilerFlags *p_cf) 2910a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner{ 2920a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner PyObject *unicode, *bytes = NULL; 2930a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner char *filename_str; 2940a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner int run; 2950a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner 2960a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner /* call pending calls like signal handlers (SIGINT) */ 2970a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner if (Py_MakePendingCalls() == -1) { 2980a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner PyErr_Print(); 2990a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner return 1; 3000a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner } 3010a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner 3020a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner if (filename) { 3030a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner unicode = PyUnicode_FromWideChar(filename, wcslen(filename)); 3040a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner if (unicode != NULL) { 305e0f32687152615b7266e66cbee216434fbbdd5bcVictor Stinner bytes = PyUnicode_EncodeFSDefault(unicode); 3060a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner Py_DECREF(unicode); 3070a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner } 3080a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner if (bytes != NULL) 3090a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner filename_str = PyBytes_AsString(bytes); 3100a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner else { 3110a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner PyErr_Clear(); 312e0f32687152615b7266e66cbee216434fbbdd5bcVictor Stinner filename_str = "<encoding error>"; 3130a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner } 3140a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner } 3150a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner else 3160a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner filename_str = "<stdin>"; 3170a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner 3180a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner run = PyRun_AnyFileExFlags(fp, filename_str, filename != NULL, p_cf); 3190a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner Py_XDECREF(bytes); 3200a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner return run != 0; 3210a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner} 3220a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner 3239cd177526afc086a300b548588880329c32f607dChristian Heimes 324667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum/* Main program */ 325667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 326fe51c6d66e0fbf6a142036bee2c448bd7fe8fefcMark Hammondint 327790465fd90e8a72590386465f518db9e67ab843fMartin v. LöwisPy_Main(int argc, wchar_t **argv) 328667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum{ 329f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int c; 330f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int sts; 331f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou wchar_t *command = NULL; 332f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou wchar_t *filename = NULL; 333f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou wchar_t *module = NULL; 334f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou FILE *fp = stdin; 335f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou char *p; 336e53de3dc4a9021b5edabd44fd495df7770b6249cPhilip Jenvey#ifdef MS_WINDOWS 337f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou wchar_t *wp; 338e53de3dc4a9021b5edabd44fd495df7770b6249cPhilip Jenvey#endif 339f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int skipfirstline = 0; 340f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int stdin_is_interactive = 0; 341f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int help = 0; 342f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int version = 0; 343f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int saw_unbuffered_flag = 0; 344f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyCompilerFlags cf; 3456999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou PyObject *warning_option = NULL; 3466999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou PyObject *warning_options = NULL; 347f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 348f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou cf.cf_flags = 0; 349f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 350f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou orig_argc = argc; /* For Py_GetArgcArgv() */ 351f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou orig_argv = argv; 352f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 35386838b02f04a01445778584afb7acd0b05081335Antoine Pitrou /* Hash randomization needed early for all string operations 35486838b02f04a01445778584afb7acd0b05081335Antoine Pitrou (including -W and -X options). */ 355a0dd22e5e824195e322cb417ebb34c66475e952dEzio Melotti _PyOS_opterr = 0; /* prevent printing the error in 1st pass */ 35686838b02f04a01445778584afb7acd0b05081335Antoine Pitrou while ((c = _PyOS_GetOpt(argc, argv, PROGRAM_OPTS)) != EOF) { 35786838b02f04a01445778584afb7acd0b05081335Antoine Pitrou if (c == 'm' || c == 'c') { 35886838b02f04a01445778584afb7acd0b05081335Antoine Pitrou /* -c / -m is the last option: following arguments are 35986838b02f04a01445778584afb7acd0b05081335Antoine Pitrou not interpreter options. */ 36086838b02f04a01445778584afb7acd0b05081335Antoine Pitrou break; 36186838b02f04a01445778584afb7acd0b05081335Antoine Pitrou } 362c9f54cf512996790266c17f81584c9725ee99d47Benjamin Peterson if (c == 'E') { 36386838b02f04a01445778584afb7acd0b05081335Antoine Pitrou Py_IgnoreEnvironmentFlag++; 36486838b02f04a01445778584afb7acd0b05081335Antoine Pitrou break; 36586838b02f04a01445778584afb7acd0b05081335Antoine Pitrou } 36686838b02f04a01445778584afb7acd0b05081335Antoine Pitrou } 36786838b02f04a01445778584afb7acd0b05081335Antoine Pitrou 368c9f54cf512996790266c17f81584c9725ee99d47Benjamin Peterson Py_HashRandomizationFlag = 1; 36986838b02f04a01445778584afb7acd0b05081335Antoine Pitrou _PyRandom_Init(); 37086838b02f04a01445778584afb7acd0b05081335Antoine Pitrou 371f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PySys_ResetWarnOptions(); 37286838b02f04a01445778584afb7acd0b05081335Antoine Pitrou _PyOS_ResetGetOpt(); 373f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 374f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou while ((c = _PyOS_GetOpt(argc, argv, PROGRAM_OPTS)) != EOF) { 375f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (c == 'c') { 376f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou size_t len; 377f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* -c is the last option; following arguments 378f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou that look like options are left for the 379f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou command to interpret. */ 380f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 381f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou len = wcslen(_PyOS_optarg) + 1 + 1; 3821a7425f67a0d141483d89ca80ca01e3cb7f6be92Victor Stinner command = (wchar_t *)PyMem_RawMalloc(sizeof(wchar_t) * len); 383f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (command == NULL) 384f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_FatalError( 385f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou "not enough memory to copy -c argument"); 386f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou wcscpy(command, _PyOS_optarg); 387f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou command[len - 2] = '\n'; 388f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou command[len - 1] = 0; 389f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 390f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 391f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 392f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (c == 'm') { 393f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* -m is the last option; following arguments 394f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou that look like options are left for the 395f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou module to interpret. */ 396f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou module = _PyOS_optarg; 397f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 398f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 399f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 400f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou switch (c) { 401f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'b': 402f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_BytesWarningFlag++; 403f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 404f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 405f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'd': 406f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DebugFlag++; 407f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 408f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 409f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'i': 410f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_InspectFlag++; 411f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_InteractiveFlag++; 412f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 413f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 414ad73a9cf97770023665a1bb1c6390a3c99478139Christian Heimes case 'I': 415ad73a9cf97770023665a1bb1c6390a3c99478139Christian Heimes Py_IsolatedFlag++; 416ad73a9cf97770023665a1bb1c6390a3c99478139Christian Heimes Py_NoUserSiteDirectory++; 417ad73a9cf97770023665a1bb1c6390a3c99478139Christian Heimes Py_IgnoreEnvironmentFlag++; 418ad73a9cf97770023665a1bb1c6390a3c99478139Christian Heimes break; 419ad73a9cf97770023665a1bb1c6390a3c99478139Christian Heimes 420f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* case 'J': reserved for Jython */ 421f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 422f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'O': 423f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_OptimizeFlag++; 424f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 425f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 426f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'B': 427f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DontWriteBytecodeFlag++; 428f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 429f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 430f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 's': 431f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_NoUserSiteDirectory++; 432f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 433f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 434f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'S': 435f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_NoSiteFlag++; 436f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 437f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 438f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'E': 43986838b02f04a01445778584afb7acd0b05081335Antoine Pitrou /* Already handled above */ 440f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 441f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 442f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 't': 443f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* ignored for backwards compatibility */ 444f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 445f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 446f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'u': 447f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_UnbufferedStdioFlag = 1; 448f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou saw_unbuffered_flag = 1; 449f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 450f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 451f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'v': 452f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_VerboseFlag++; 453f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 454f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 455f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'x': 456f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou skipfirstline = 1; 457f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 458f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 459f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'h': 460f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case '?': 461f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou help++; 462f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 463f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 464f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'V': 465f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou version++; 466f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 467f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 468f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou case 'W': 4696999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou if (warning_options == NULL) 4706999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou warning_options = PyList_New(0); 4716999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou if (warning_options == NULL) 4726999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou Py_FatalError("failure in handling of -W argument"); 4736999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou warning_option = PyUnicode_FromWideChar(_PyOS_optarg, -1); 4746999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou if (warning_option == NULL) 4756999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou Py_FatalError("failure in handling of -W argument"); 4766999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou PyList_Append(warning_options, warning_option); 4776999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou Py_DECREF(warning_option); 478f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 479f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 4809583cac6337f9a5f2670fbe5e1f2e85aaad04522Antoine Pitrou case 'X': 4819583cac6337f9a5f2670fbe5e1f2e85aaad04522Antoine Pitrou PySys_AddXOption(_PyOS_optarg); 4829583cac6337f9a5f2670fbe5e1f2e85aaad04522Antoine Pitrou break; 4839583cac6337f9a5f2670fbe5e1f2e85aaad04522Antoine Pitrou 4849d8711964f114b9f0f5ed8a9a1eccfba2a096030Georg Brandl case 'q': 4858aa7e999b5ab87cdbefe441649c223647875c110Georg Brandl Py_QuietFlag++; 4869d8711964f114b9f0f5ed8a9a1eccfba2a096030Georg Brandl break; 4879d8711964f114b9f0f5ed8a9a1eccfba2a096030Georg Brandl 4882daf6ae2495c862adf8bc717bfe9964081ea0b10Georg Brandl case 'R': 489c9f54cf512996790266c17f81584c9725ee99d47Benjamin Peterson /* Ignored */ 4902daf6ae2495c862adf8bc717bfe9964081ea0b10Georg Brandl break; 4912daf6ae2495c862adf8bc717bfe9964081ea0b10Georg Brandl 492f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* This space reserved for other options */ 493f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 494f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou default: 495f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return usage(2, argv[0]); 496f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /*NOTREACHED*/ 497f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 498f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 499f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 5007d4bb9f179f1888a472fd5d5a30df4ec3c55fea5Neil Schemenauer 501f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (help) 502f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return usage(0, argv[0]); 503f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 504f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (version) { 505e3ed4edb94cadaef747f6032c52cd903a169883eSerhiy Storchaka printf("Python %s\n", PY_VERSION); 506f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return 0; 507f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 508f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 509f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (!Py_InspectFlag && 510f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou (p = Py_GETENV("PYTHONINSPECT")) && *p != '\0') 511f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_InspectFlag = 1; 512f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (!saw_unbuffered_flag && 513f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou (p = Py_GETENV("PYTHONUNBUFFERED")) && *p != '\0') 514f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_UnbufferedStdioFlag = 1; 515f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 516f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (!Py_NoUserSiteDirectory && 517f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou (p = Py_GETENV("PYTHONNOUSERSITE")) && *p != '\0') 518f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_NoUserSiteDirectory = 1; 5198dc226fccd2183670ac19a85b89941f5b05167dcChristian Heimes 520e53de3dc4a9021b5edabd44fd495df7770b6249cPhilip Jenvey#ifdef MS_WINDOWS 521f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (!Py_IgnoreEnvironmentFlag && (wp = _wgetenv(L"PYTHONWARNINGS")) && 522f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou *wp != L'\0') { 523f63dab5a842df7349af73be23fb3acc37aa75703Steve Dower wchar_t *buf, *warning, *context = NULL; 524f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 5251a7425f67a0d141483d89ca80ca01e3cb7f6be92Victor Stinner buf = (wchar_t *)PyMem_RawMalloc((wcslen(wp) + 1) * sizeof(wchar_t)); 526f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (buf == NULL) 527f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_FatalError( 528f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou "not enough memory to copy PYTHONWARNINGS"); 529f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou wcscpy(buf, wp); 530f63dab5a842df7349af73be23fb3acc37aa75703Steve Dower for (warning = wcstok_s(buf, L",", &context); 531f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou warning != NULL; 532f63dab5a842df7349af73be23fb3acc37aa75703Steve Dower warning = wcstok_s(NULL, L",", &context)) { 533f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PySys_AddWarnOption(warning); 534f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 5351a7425f67a0d141483d89ca80ca01e3cb7f6be92Victor Stinner PyMem_RawFree(buf); 536f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 537e53de3dc4a9021b5edabd44fd495df7770b6249cPhilip Jenvey#else 538f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if ((p = Py_GETENV("PYTHONWARNINGS")) && *p != '\0') { 539f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou char *buf, *oldloc; 540c2d76fd339f5fcc4f0051696e5e57c979289a9c3Victor Stinner PyObject *unicode; 541f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 542f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* settle for strtok here as there's no one standard 543f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou C89 wcstok */ 5441a7425f67a0d141483d89ca80ca01e3cb7f6be92Victor Stinner buf = (char *)PyMem_RawMalloc(strlen(p) + 1); 545f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (buf == NULL) 546f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_FatalError( 547f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou "not enough memory to copy PYTHONWARNINGS"); 548f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou strcpy(buf, p); 54949fc8ece8172162510890f42127d2aa4e13f878bVictor Stinner oldloc = _PyMem_RawStrdup(setlocale(LC_ALL, NULL)); 550f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setlocale(LC_ALL, ""); 551f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou for (p = strtok(buf, ","); p != NULL; p = strtok(NULL, ",")) { 5525c848a84fd59eabd1ab9965d376b78139eaf97f7Victor Stinner#ifdef __APPLE__ 5535c848a84fd59eabd1ab9965d376b78139eaf97f7Victor Stinner /* Use utf-8 on Mac OS X */ 5545c848a84fd59eabd1ab9965d376b78139eaf97f7Victor Stinner unicode = PyUnicode_FromString(p); 5555c848a84fd59eabd1ab9965d376b78139eaf97f7Victor Stinner#else 5561b57967b96daeb505e9d2dbe3cd347625dcb0739Victor Stinner unicode = PyUnicode_DecodeLocale(p, "surrogateescape"); 5575c848a84fd59eabd1ab9965d376b78139eaf97f7Victor Stinner#endif 558af02e1c85a66009cdc645a64de7d7ee1335c8301Victor Stinner if (unicode == NULL) { 559af02e1c85a66009cdc645a64de7d7ee1335c8301Victor Stinner /* ignore errors */ 560af02e1c85a66009cdc645a64de7d7ee1335c8301Victor Stinner PyErr_Clear(); 561c2d76fd339f5fcc4f0051696e5e57c979289a9c3Victor Stinner continue; 562af02e1c85a66009cdc645a64de7d7ee1335c8301Victor Stinner } 563c2d76fd339f5fcc4f0051696e5e57c979289a9c3Victor Stinner PySys_AddWarnOptionUnicode(unicode); 564c2d76fd339f5fcc4f0051696e5e57c979289a9c3Victor Stinner Py_DECREF(unicode); 565f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 566f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setlocale(LC_ALL, oldloc); 56749fc8ece8172162510890f42127d2aa4e13f878bVictor Stinner PyMem_RawFree(oldloc); 5681a7425f67a0d141483d89ca80ca01e3cb7f6be92Victor Stinner PyMem_RawFree(buf); 569f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 570e53de3dc4a9021b5edabd44fd495df7770b6249cPhilip Jenvey#endif 5716999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou if (warning_options != NULL) { 5726999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou Py_ssize_t i; 5736999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou for (i = 0; i < PyList_GET_SIZE(warning_options); i++) { 5746999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou PySys_AddWarnOptionUnicode(PyList_GET_ITEM(warning_options, i)); 5756999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou } 5766999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou } 5770805ca3f931ccd4735dcaa231752371deadf5ccfPhilip Jenvey 578f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (command == NULL && module == NULL && _PyOS_optind < argc && 579f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou wcscmp(argv[_PyOS_optind], L"-") != 0) 580f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou { 581f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou filename = argv[_PyOS_optind]; 582f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 583775af9191149059bf5ebfcda6dd9bf0251ce511aGuido van Rossum 584f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou stdin_is_interactive = Py_FdIsInteractive(stdin, (char *)0); 585775af9191149059bf5ebfcda6dd9bf0251ce511aGuido van Rossum 5869cf424b04bf355df85659da18abc2ac3f9419c37Sjoerd Mullender#if defined(MS_WINDOWS) || defined(__CYGWIN__) 58789e343660623862acbd40cc617b4af262d6c799fVictor Stinner /* don't translate newlines (\r\n <=> \n) */ 58889e343660623862acbd40cc617b4af262d6c799fVictor Stinner _setmode(fileno(stdin), O_BINARY); 58989e343660623862acbd40cc617b4af262d6c799fVictor Stinner _setmode(fileno(stdout), O_BINARY); 59089e343660623862acbd40cc617b4af262d6c799fVictor Stinner _setmode(fileno(stderr), O_BINARY); 591f22d7e2c69f9095edc2703483ec11a3b7ff931f5Guido van Rossum#endif 59289e343660623862acbd40cc617b4af262d6c799fVictor Stinner 59389e343660623862acbd40cc617b4af262d6c799fVictor Stinner if (Py_UnbufferedStdioFlag) { 59422ffac1b1f8288bac7a277a8caa04ca731a54b81Guido van Rossum#ifdef HAVE_SETVBUF 595f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setvbuf(stdin, (char *)NULL, _IONBF, BUFSIZ); 596f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setvbuf(stdout, (char *)NULL, _IONBF, BUFSIZ); 597f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setvbuf(stderr, (char *)NULL, _IONBF, BUFSIZ); 59822ffac1b1f8288bac7a277a8caa04ca731a54b81Guido van Rossum#else /* !HAVE_SETVBUF */ 599f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setbuf(stdin, (char *)NULL); 600f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setbuf(stdout, (char *)NULL); 601f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setbuf(stderr, (char *)NULL); 60222ffac1b1f8288bac7a277a8caa04ca731a54b81Guido van Rossum#endif /* !HAVE_SETVBUF */ 603f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 604f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou else if (Py_InteractiveFlag) { 605b31c7dcb43307b57917150d60a64856d5a845fa7Guido van Rossum#ifdef MS_WINDOWS 606f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* Doesn't have to have line-buffered -- use unbuffered */ 607f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* Any set[v]buf(stdin, ...) screws up Tkinter :-( */ 608f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setvbuf(stdout, (char *)NULL, _IONBF, BUFSIZ); 60922ffac1b1f8288bac7a277a8caa04ca731a54b81Guido van Rossum#else /* !MS_WINDOWS */ 61022ffac1b1f8288bac7a277a8caa04ca731a54b81Guido van Rossum#ifdef HAVE_SETVBUF 611f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setvbuf(stdin, (char *)NULL, _IOLBF, BUFSIZ); 612f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou setvbuf(stdout, (char *)NULL, _IOLBF, BUFSIZ); 61322ffac1b1f8288bac7a277a8caa04ca731a54b81Guido van Rossum#endif /* HAVE_SETVBUF */ 61422ffac1b1f8288bac7a277a8caa04ca731a54b81Guido van Rossum#endif /* !MS_WINDOWS */ 615f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* Leave stderr alone - it should be unbuffered anyway. */ 616f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 617667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 6182ac79ef9e3168691efabad6877e45168a00738acJust van Rossum#ifdef __APPLE__ 619f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* On MacOS X, when the Python interpreter is embedded in an 620f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou application bundle, it gets executed by a bootstrapping script 621f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou that does os.execve() with an argv[0] that's different from the 622f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou actual Python executable. This is needed to keep the Finder happy, 623f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou or rather, to work around Apple's overly strict requirements of 624f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou the process name. However, we still need a usable sys.executable, 625f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou so the actual executable path is passed in an environment variable. 626f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou See Lib/plat-mac/bundlebuiler.py for details about the bootstrap 627f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou script. */ 628f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if ((p = Py_GETENV("PYTHONEXECUTABLE")) && *p != '\0') { 629f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou wchar_t* buffer; 630eb61f8b3a726a8cd1ae3bc3794623e7fe4dd5703Ronald Oussoren size_t len = strlen(p) + 1; 631f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 6321a7425f67a0d141483d89ca80ca01e3cb7f6be92Victor Stinner buffer = PyMem_RawMalloc(len * sizeof(wchar_t)); 633f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (buffer == NULL) { 634f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_FatalError( 635f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou "not enough memory to copy PYTHONEXECUTABLE"); 636f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 637f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 638b94767ff44edf5d461d7cb1c8eb5160f83886358Brett Cannon mbstowcs(buffer, p, len); 639f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_SetProgramName(buffer); 640f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* buffer is now handed off - do not free */ 641f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } else { 64290db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip#ifdef WITH_NEXT_FRAMEWORK 64390db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip char* pyvenv_launcher = getenv("__PYVENV_LAUNCHER__"); 64490db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip 64590db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip if (pyvenv_launcher && *pyvenv_launcher) { 64690db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip /* Used by Mac/Tools/pythonw.c to forward 64790db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip * the argv0 of the stub executable 64890db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip */ 649f6a271ae980d2f3fb450f745b8f87624378156c4Victor Stinner wchar_t* wbuf = Py_DecodeLocale(pyvenv_launcher, NULL); 65090db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip 65190db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip if (wbuf == NULL) { 65290db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip Py_FatalError("Cannot decode __PYVENV_LAUNCHER__"); 65390db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip } 65490db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip Py_SetProgramName(wbuf); 65590db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip 65690db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip /* Don't free wbuf, the argument to Py_SetProgramName 65790db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip * must remain valid until the Py_Finalize is called. 65890db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip */ 65990db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip } else { 66090db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip Py_SetProgramName(argv[0]); 66190db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip } 66290db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip#else 663f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_SetProgramName(argv[0]); 66490db661b4335eaf7a3cb5e9a4524ae9707eeaf76Vinay Sajip#endif 665f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 6662ac79ef9e3168691efabad6877e45168a00738acJust van Rossum#else 667f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_SetProgramName(argv[0]); 668fbd861bbd3532372c6c9160c5d14f17ff2c7de00Jack Jansen#endif 669f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_Initialize(); 6706999441d62cfc6a6273b0e9e3df2cce0ef3f05faAntoine Pitrou Py_XDECREF(warning_options); 671f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 6728aa7e999b5ab87cdbefe441649c223647875c110Georg Brandl if (!Py_QuietFlag && (Py_VerboseFlag || 6739d8711964f114b9f0f5ed8a9a1eccfba2a096030Georg Brandl (command == NULL && filename == NULL && 6749d8711964f114b9f0f5ed8a9a1eccfba2a096030Georg Brandl module == NULL && stdin_is_interactive))) { 675f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(stderr, "Python %s on %s\n", 676f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_GetVersion(), Py_GetPlatform()); 677f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (!Py_NoSiteFlag) 678f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(stderr, "%s\n", COPYRIGHT); 679f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 680f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 681f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (command != NULL) { 682f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* Backup _PyOS_optind and force sys.argv[0] = '-c' */ 683f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou _PyOS_optind--; 684f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou argv[_PyOS_optind] = L"-c"; 685f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 686f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 687f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (module != NULL) { 688d26c18adccf02321592cd58a2dadb0ab68af6906Nick Coghlan /* Backup _PyOS_optind and force sys.argv[0] = '-m'*/ 689f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou _PyOS_optind--; 690d26c18adccf02321592cd58a2dadb0ab68af6906Nick Coghlan argv[_PyOS_optind] = L"-m"; 691f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 692f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 693f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PySys_SetArgv(argc-_PyOS_optind, argv+_PyOS_optind); 694f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 695f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if ((Py_InspectFlag || (command == NULL && filename == NULL && module == NULL)) && 696f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou isatty(fileno(stdin))) { 697f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyObject *v; 698f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou v = PyImport_ImportModule("readline"); 699f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (v == NULL) 700f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou PyErr_Clear(); 701f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou else 702f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_DECREF(v); 703f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 704f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 705f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (command) { 706a62207c5847948b83d698b5fb4b2c1a03434a5a6Victor Stinner sts = run_command(command, &cf); 7071a7425f67a0d141483d89ca80ca01e3cb7f6be92Victor Stinner PyMem_RawFree(command); 708f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } else if (module) { 709f01a337950bdcd6905d12b2335ba4d8773f6b8e5Senthil Kumaran sts = (RunModule(module, 1) != 0); 710f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 711f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou else { 712f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 713f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (filename == NULL && stdin_is_interactive) { 714f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_InspectFlag = 0; /* do exit on SystemExit */ 715f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou RunStartupFile(&cf); 7161a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou RunInteractiveHook(); 717f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 718f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* XXX */ 719f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 720f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou sts = -1; /* keep track of whether we've already run __main__ */ 721f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 722f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (filename != NULL) { 723f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou sts = RunMainFromImporter(filename); 724f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 725f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 726f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (sts==-1 && filename!=NULL) { 7279a6692f6d7784365ae07094694c754d2b153b719Victor Stinner fp = _Py_wfopen(filename, L"r"); 7289a6692f6d7784365ae07094694c754d2b153b719Victor Stinner if (fp == NULL) { 7299a6692f6d7784365ae07094694c754d2b153b719Victor Stinner char *cfilename_buffer; 7309a6692f6d7784365ae07094694c754d2b153b719Victor Stinner const char *cfilename; 731c345ce1a69b3c4a46d87ef56d859bc70abfc74b4Antoine Pitrou int err = errno; 732f6a271ae980d2f3fb450f745b8f87624378156c4Victor Stinner cfilename_buffer = Py_EncodeLocale(filename, NULL); 7339a6692f6d7784365ae07094694c754d2b153b719Victor Stinner if (cfilename_buffer != NULL) 7349a6692f6d7784365ae07094694c754d2b153b719Victor Stinner cfilename = cfilename_buffer; 7359a6692f6d7784365ae07094694c754d2b153b719Victor Stinner else 7369a6692f6d7784365ae07094694c754d2b153b719Victor Stinner cfilename = "<unprintable file name>"; 737f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(stderr, "%ls: can't open file '%s': [Errno %d] %s\n", 738c345ce1a69b3c4a46d87ef56d859bc70abfc74b4Antoine Pitrou argv[0], cfilename, err, strerror(err)); 7399a6692f6d7784365ae07094694c754d2b153b719Victor Stinner if (cfilename_buffer) 7409a6692f6d7784365ae07094694c754d2b153b719Victor Stinner PyMem_Free(cfilename_buffer); 741f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return 2; 742f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 743f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou else if (skipfirstline) { 744f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou int ch; 745f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* Push back first newline so line numbers 746f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou remain the same */ 747f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou while ((ch = getc(fp)) != EOF) { 748f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (ch == '\n') { 749f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou (void)ungetc(ch, fp); 750f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou break; 751f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 752f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 753f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 754f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou { 755f2f373f5931be48efc3f6fa2c2faa1cca79dce75Steve Dower struct _Py_stat_struct sb; 756f2f373f5931be48efc3f6fa2c2faa1cca79dce75Steve Dower if (_Py_fstat(fileno(fp), &sb) == 0 && 757f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou S_ISDIR(sb.st_mode)) { 758f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fprintf(stderr, "%ls: '%ls' is a directory, cannot continue\n", argv[0], filename); 759f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou fclose(fp); 760f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return 1; 761f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 762f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 763f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 764f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 7650a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner if (sts == -1) 7660a3ddad666d3c30f3b232a1825dc6d8e2de04b47Victor Stinner sts = run_file(fp, filename, &cf); 767f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 768f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 769f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* Check this environment variable at the end, to give programs the 770f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou * opportunity to set it from Python. 771f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou */ 772f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (!Py_InspectFlag && 773f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou (p = Py_GETENV("PYTHONINSPECT")) && *p != '\0') 774f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou { 775f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_InspectFlag = 1; 776f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 777f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 778f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou if (Py_InspectFlag && stdin_is_interactive && 779f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou (filename != NULL || command != NULL || module != NULL)) { 780f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_InspectFlag = 0; 7811a6cb30a346ba8812d6abc77fddee636ae06ccffAntoine Pitrou RunInteractiveHook(); 782f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* XXX */ 783f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou sts = PyRun_AnyFileFlags(stdin, "<stdin>", &cf) != 0; 784f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou } 785f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou 786f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou Py_Finalize(); 7873e13b1e48bdd0550929f55bac11571116cd655abBarry Warsaw 7883e13b1e48bdd0550929f55bac11571116cd655abBarry Warsaw#ifdef __INSURE__ 789f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou /* Insure++ is a memory analysis tool that aids in discovering 790f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou * memory leaks and other memory problems. On Python exit, the 791f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou * interned string dictionaries are flagged as being in use at exit 792f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou * (which it is). Under normal circumstances, this is fine because 793f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou * the memory will be automatically reclaimed by the system. Under 794f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou * memory debugging, it's a huge source of useless noise, so we 795f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou * trade off slower shutdown for less distraction in the memory 796f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou * reports. -baw 797f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou */ 798f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou _Py_ReleaseInternedUnicodeStrings(); 7993e13b1e48bdd0550929f55bac11571116cd655abBarry Warsaw#endif /* __INSURE__ */ 8003e13b1e48bdd0550929f55bac11571116cd655abBarry Warsaw 801f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou return sts; 802667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum} 803667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 804786ea6bc23c26a0ec98d5cbc80633615f9fa8cb1Skip Montanaro/* this is gonna seem *real weird*, but if you put some other code between 805786ea6bc23c26a0ec98d5cbc80633615f9fa8cb1Skip Montanaro Py_Main() and Py_GetArgcArgv() you will need to adjust the test in the 806786ea6bc23c26a0ec98d5cbc80633615f9fa8cb1Skip Montanaro while statement in Misc/gdbinit:ppystack */ 807667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 808667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum/* Make the *original* argc/argv available to other modules. 809667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum This is rare, but it is needed by the secureware extension. */ 810667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum 811667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossumvoid 812790465fd90e8a72590386465f518db9e67ab843fMartin v. LöwisPy_GetArgcArgv(int *argc, wchar_t ***argv) 813667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum{ 814f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou *argc = orig_argc; 815f95a1b3c53bdd678b64aa608d4375660033460c3Antoine Pitrou *argv = orig_argv; 816667d704997f26a1a22f4e981bbb3c2f8399cfc41Guido van Rossum} 81749fd7fa4431da299196d74087df4a04f99f9c46fThomas Wouters 81849fd7fa4431da299196d74087df4a04f99f9c46fThomas Wouters#ifdef __cplusplus 81949fd7fa4431da299196d74087df4a04f99f9c46fThomas Wouters} 82049fd7fa4431da299196d74087df4a04f99f9c46fThomas Wouters#endif 821