blob: 8b1e61a1c93e48327e5b86b2bb48ec5700a5a2e8 [file] [log] [blame]
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001
2/* Python interpreter top-level routines, including init/exit */
3
Guido van Rossum82598051997-03-05 00:20:32 +00004#include "Python.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +00005
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00006#include "Python-ast.h"
Guido van Rossumd8faa362007-04-27 19:54:29 +00007#undef Yield /* undefine macro conflicting with winbase.h */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00008#include "grammar.h"
9#include "node.h"
Fred Drake85f36392000-07-11 17:53:00 +000010#include "token.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000011#include "parsetok.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000012#include "errcode.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000013#include "code.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000014#include "compile.h"
Jeremy Hylton4b38da62001-02-02 18:19:15 +000015#include "symtable.h"
Neal Norwitzadb69fc2005-12-17 20:54:49 +000016#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000017#include "ast.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000018#include "eval.h"
Guido van Rossumfdef2711994-09-14 13:31:04 +000019#include "marshal.h"
Martin v. Löwis790465f2008-04-05 20:41:37 +000020#include "osdefs.h"
Antoine Pitrou011bd622009-10-20 21:52:47 +000021#include "abstract.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000022
Thomas Wouters0e3f5912006-08-11 14:57:12 +000023#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000024#include <signal.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000025#endif
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000026
Benjamin Peterson80a50ac2009-01-02 21:24:04 +000027#ifdef MS_WINDOWS
Martin v. Löwis5c88d812009-01-02 20:47:48 +000028#include "malloc.h" /* for alloca */
Benjamin Peterson80a50ac2009-01-02 21:24:04 +000029#endif
Martin v. Löwis5c88d812009-01-02 20:47:48 +000030
Martin v. Löwis73d538b2003-03-05 15:13:47 +000031#ifdef HAVE_LANGINFO_H
32#include <locale.h>
33#include <langinfo.h>
34#endif
35
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000036#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000037#undef BYTE
38#include "windows.h"
Martin v. Löwis790465f2008-04-05 20:41:37 +000039#define PATH_MAX MAXPATHLEN
Guido van Rossuma44823b1995-03-14 15:01:17 +000040#endif
41
Neal Norwitz4281cef2006-03-04 19:58:13 +000042#ifndef Py_REF_DEBUG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000043#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000044#else /* Py_REF_DEBUG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000045#define PRINT_TOTAL_REFS() fprintf(stderr, \
46 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
47 _Py_GetRefTotal())
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000048#endif
49
50#ifdef __cplusplus
51extern "C" {
Neal Norwitz4281cef2006-03-04 19:58:13 +000052#endif
53
Martin v. Löwis790465f2008-04-05 20:41:37 +000054extern wchar_t *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000055
Guido van Rossum82598051997-03-05 00:20:32 +000056extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000057
Guido van Rossumb73cc041993-11-01 16:28:59 +000058/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000059static void initmain(void);
Victor Stinnerb744ba12010-05-15 12:27:16 +000060static void initfsencoding(void);
Tim Petersdbd9ba62000-07-09 03:09:57 +000061static void initsite(void);
Guido van Rossumce3a72a2007-10-19 23:16:50 +000062static int initstdio(void);
Amaury Forgeot d'Arc7fedbe52008-04-10 21:03:09 +000063static void flush_io(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000064static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000065 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000066static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000067 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000068static void err_input(perrdetail *);
69static void initsigs(void);
Collin Winter670e6922007-03-21 02:57:17 +000070static void call_py_exitfuncs(void);
Antoine Pitrou011bd622009-10-20 21:52:47 +000071static void wait_for_thread_shutdown(void);
Tim Petersdbd9ba62000-07-09 03:09:57 +000072static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000073extern void _PyUnicode_Init(void);
74extern void _PyUnicode_Fini(void);
Guido van Rossumddefaf32007-01-14 03:31:43 +000075extern int _PyLong_Init(void);
76extern void PyLong_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000077
Mark Hammond8d98d2c2003-04-19 15:41:53 +000078#ifdef WITH_THREAD
79extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
80extern void _PyGILState_Fini(void);
81#endif /* WITH_THREAD */
82
Guido van Rossum82598051997-03-05 00:20:32 +000083int Py_DebugFlag; /* Needed by parser.c */
84int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000085int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumd8faa362007-04-27 19:54:29 +000086int Py_InspectFlag; /* Needed to determine whether to exit at SystemError */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000087int Py_NoSiteFlag; /* Suppress 'import site' */
Guido van Rossum98297ee2007-11-06 21:34:58 +000088int Py_BytesWarningFlag; /* Warn on str(bytes) and str(buffer) */
Christian Heimes790c8232008-01-07 21:14:23 +000089int Py_DontWriteBytecodeFlag; /* Suppress writing bytecode files (*.py[co]) */
Barry Warsaw3ce09642000-05-02 19:18:59 +000090int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000091int Py_FrozenFlag; /* Needed by getpath.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000092int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Christian Heimes8dc226f2008-05-06 23:45:46 +000093int Py_NoUserSiteDirectory = 0; /* for -s and site.py */
Antoine Pitrou05608432009-01-09 18:53:14 +000094int Py_UnbufferedStdioFlag = 0; /* Unbuffered binary std{in,out,err} */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000095
Christian Heimes33fe8092008-04-13 13:53:33 +000096/* PyModule_GetWarningsModule is no longer necessary as of 2.6
97since _warnings is builtin. This API should not be used. */
98PyObject *
99PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +0000100{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000101 return PyImport_ImportModule("warnings");
Mark Hammondedd07732003-07-15 23:03:55 +0000102}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000103
Guido van Rossum25ce5661997-08-02 03:10:38 +0000104static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000105
Thomas Wouters7e474022000-07-16 12:04:32 +0000106/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000107
108int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000109Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000110{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000111 return initialized;
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000112}
113
Guido van Rossum25ce5661997-08-02 03:10:38 +0000114/* Global initializations. Can be undone by Py_Finalize(). Don't
115 call this twice without an intervening Py_Finalize() call. When
116 initializations fail, a fatal error is issued and the function does
117 not return. On return, the first thread and interpreter state have
118 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000119
Guido van Rossum25ce5661997-08-02 03:10:38 +0000120 Locking: you must hold the interpreter lock while calling this.
121 (If the lock has not yet been initialized, that's equivalent to
122 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000123
Guido van Rossum25ce5661997-08-02 03:10:38 +0000124*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000125
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000126static int
127add_flag(int flag, const char *envs)
128{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000129 int env = atoi(envs);
130 if (flag < env)
131 flag = env;
132 if (flag < 1)
133 flag = 1;
134 return flag;
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000135}
136
Christian Heimes5833a2f2008-10-30 21:40:04 +0000137static char*
Victor Stinner94908bb2010-08-18 21:23:25 +0000138get_codec_name(const char *encoding)
Christian Heimes5833a2f2008-10-30 21:40:04 +0000139{
Victor Stinner94908bb2010-08-18 21:23:25 +0000140 char *name_utf8, *name_str;
Victor Stinner386fe712010-05-19 00:34:15 +0000141 PyObject *codec, *name = NULL;
Christian Heimes5833a2f2008-10-30 21:40:04 +0000142
Victor Stinner94908bb2010-08-18 21:23:25 +0000143 codec = _PyCodec_Lookup(encoding);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000144 if (!codec)
145 goto error;
Christian Heimes5833a2f2008-10-30 21:40:04 +0000146
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000147 name = PyObject_GetAttrString(codec, "name");
148 Py_CLEAR(codec);
149 if (!name)
150 goto error;
Christian Heimes5833a2f2008-10-30 21:40:04 +0000151
Victor Stinner94908bb2010-08-18 21:23:25 +0000152 name_utf8 = _PyUnicode_AsString(name);
Victor Stinner386fe712010-05-19 00:34:15 +0000153 if (name == NULL)
154 goto error;
Victor Stinner94908bb2010-08-18 21:23:25 +0000155 name_str = strdup(name_utf8);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000156 Py_DECREF(name);
Victor Stinner94908bb2010-08-18 21:23:25 +0000157 if (name_str == NULL) {
158 PyErr_NoMemory();
159 return NULL;
160 }
161 return name_str;
Christian Heimes5833a2f2008-10-30 21:40:04 +0000162
163error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000164 Py_XDECREF(codec);
Victor Stinner386fe712010-05-19 00:34:15 +0000165 Py_XDECREF(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000166 return NULL;
Christian Heimes5833a2f2008-10-30 21:40:04 +0000167}
Victor Stinner94908bb2010-08-18 21:23:25 +0000168
169#if defined(HAVE_LANGINFO_H) && defined(CODESET)
170static char*
171get_codeset(void)
172{
173 char* codeset = nl_langinfo(CODESET);
174 if (!codeset || codeset[0] == '\0') {
175 PyErr_SetString(PyExc_ValueError, "CODESET is not set or empty");
176 return NULL;
177 }
178 return get_codec_name(codeset);
179}
Christian Heimes5833a2f2008-10-30 21:40:04 +0000180#endif
181
Guido van Rossuma027efa1997-05-05 20:56:21 +0000182void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000183Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000184{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000185 PyInterpreterState *interp;
186 PyThreadState *tstate;
187 PyObject *bimod, *sysmod, *pstderr;
188 char *p;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000189 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000190
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000191 if (initialized)
192 return;
193 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000194
Andrew M. Kuchlingb2ceb3a2010-02-22 23:26:10 +0000195#if defined(HAVE_LANGINFO_H) && defined(HAVE_SETLOCALE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000196 /* Set up the LC_CTYPE locale, so we can obtain
197 the locale's charset without having to switch
198 locales. */
199 setlocale(LC_CTYPE, "");
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +0000200#endif
201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000202 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
203 Py_DebugFlag = add_flag(Py_DebugFlag, p);
204 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
205 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
206 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
207 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
208 if ((p = Py_GETENV("PYTHONDONTWRITEBYTECODE")) && *p != '\0')
209 Py_DontWriteBytecodeFlag = add_flag(Py_DontWriteBytecodeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000210
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000211 interp = PyInterpreterState_New();
212 if (interp == NULL)
213 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000214
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000215 tstate = PyThreadState_New(interp);
216 if (tstate == NULL)
217 Py_FatalError("Py_Initialize: can't make first thread");
218 (void) PyThreadState_Swap(tstate);
Guido van Rossuma027efa1997-05-05 20:56:21 +0000219
Victor Stinner6961bd62010-08-17 22:26:51 +0000220#ifdef WITH_THREAD
Antoine Pitroub0b384b2010-09-20 20:13:48 +0000221 /* We can't call _PyEval_FiniThreads() in Py_Finalize because
222 destroying the GIL might fail when it is being referenced from
223 another running thread (see issue #9901).
224 Instead we destroy the previously created GIL here, which ensures
225 that we can call Py_Initialize / Py_Finalize multiple times. */
226 _PyEval_FiniThreads();
227
228 /* Auto-thread-state API */
Victor Stinner6961bd62010-08-17 22:26:51 +0000229 _PyGILState_Init(interp, tstate);
230#endif /* WITH_THREAD */
231
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000232 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000233
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000234 if (!_PyFrame_Init())
235 Py_FatalError("Py_Initialize: can't init frames");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000236
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000237 if (!_PyLong_Init())
238 Py_FatalError("Py_Initialize: can't init longs");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000239
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000240 if (!PyByteArray_Init())
241 Py_FatalError("Py_Initialize: can't init bytearray");
Neal Norwitz6968b052007-02-27 19:02:19 +0000242
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000243 _PyFloat_Init();
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000244
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000245 interp->modules = PyDict_New();
246 if (interp->modules == NULL)
247 Py_FatalError("Py_Initialize: can't make modules dictionary");
248 interp->modules_reloading = PyDict_New();
249 if (interp->modules_reloading == NULL)
250 Py_FatalError("Py_Initialize: can't make modules_reloading dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000251
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000252 /* Init Unicode implementation; relies on the codec registry */
253 _PyUnicode_Init();
Guido van Rossumc94044c2000-03-10 23:03:54 +0000254
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000255 bimod = _PyBuiltin_Init();
256 if (bimod == NULL)
257 Py_FatalError("Py_Initialize: can't initialize builtins modules");
Victor Stinner49d3f252010-10-17 01:24:53 +0000258 _PyImport_FixupBuiltin(bimod, "builtins");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000259 interp->builtins = PyModule_GetDict(bimod);
260 if (interp->builtins == NULL)
261 Py_FatalError("Py_Initialize: can't initialize builtins dict");
262 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000263
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000264 /* initialize builtin exceptions */
265 _PyExc_Init();
Christian Heimes9a68f8c2007-11-14 00:16:07 +0000266
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000267 sysmod = _PySys_Init();
268 if (sysmod == NULL)
269 Py_FatalError("Py_Initialize: can't initialize sys");
270 interp->sysdict = PyModule_GetDict(sysmod);
271 if (interp->sysdict == NULL)
272 Py_FatalError("Py_Initialize: can't initialize sys dict");
273 Py_INCREF(interp->sysdict);
Victor Stinner49d3f252010-10-17 01:24:53 +0000274 _PyImport_FixupBuiltin(sysmod, "sys");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000275 PySys_SetPath(Py_GetPath());
276 PyDict_SetItemString(interp->sysdict, "modules",
277 interp->modules);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000278
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000279 /* Set up a preliminary stderr printer until we have enough
280 infrastructure for the io module in place. */
281 pstderr = PyFile_NewStdPrinter(fileno(stderr));
282 if (pstderr == NULL)
283 Py_FatalError("Py_Initialize: can't set preliminary stderr");
284 PySys_SetObject("stderr", pstderr);
285 PySys_SetObject("__stderr__", pstderr);
Hirokazu Yamamotodaf83ac2010-10-30 15:08:15 +0000286 Py_DECREF(pstderr);
Guido van Rossum826d8972007-10-30 18:34:07 +0000287
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000288 _PyImport_Init();
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000289
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000290 _PyImportHooks_Init();
Just van Rossum52e14d62002-12-30 22:08:05 +0000291
Victor Stinner7d79b8b2010-05-19 20:40:50 +0000292 /* Initialize _warnings. */
293 _PyWarnings_Init();
294
Alexander Belopolsky6fc4ade2010-08-05 17:34:27 +0000295 _PyTime_Init();
296
Victor Stinnerb744ba12010-05-15 12:27:16 +0000297 initfsencoding();
Martin v. Löwis011e8422009-05-05 04:43:17 +0000298
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000299 if (install_sigs)
300 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000301
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000302 /* Initialize warnings. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000303 if (PySys_HasWarnOptions()) {
304 PyObject *warnings_module = PyImport_ImportModule("warnings");
305 if (!warnings_module)
306 PyErr_Clear();
307 Py_XDECREF(warnings_module);
308 }
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000309
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000310 initmain(); /* Module __main__ */
311 if (initstdio() < 0)
312 Py_FatalError(
313 "Py_Initialize: can't initialize sys standard streams");
314
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000315 if (!Py_NoSiteFlag)
316 initsite(); /* Module site */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000317}
318
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000319void
320Py_Initialize(void)
321{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000322 Py_InitializeEx(1);
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000323}
324
325
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000326#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000327extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000328#endif
329
Guido van Rossume8432ac2007-07-09 15:04:50 +0000330/* Flush stdout and stderr */
331
Neal Norwitz2bad9702007-08-27 06:19:22 +0000332static void
Guido van Rossum1bd21222007-07-10 20:14:13 +0000333flush_std_files(void)
Guido van Rossume8432ac2007-07-09 15:04:50 +0000334{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000335 PyObject *fout = PySys_GetObject("stdout");
336 PyObject *ferr = PySys_GetObject("stderr");
337 PyObject *tmp;
Guido van Rossume8432ac2007-07-09 15:04:50 +0000338
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000339 if (fout != NULL && fout != Py_None) {
340 tmp = PyObject_CallMethod(fout, "flush", "");
341 if (tmp == NULL)
Antoine Pitroubddc9fe2010-08-08 20:46:42 +0000342 PyErr_WriteUnraisable(fout);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000343 else
344 Py_DECREF(tmp);
345 }
Guido van Rossume8432ac2007-07-09 15:04:50 +0000346
Victor Stinner9467b212010-05-14 00:59:09 +0000347 if (ferr != NULL && ferr != Py_None) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000348 tmp = PyObject_CallMethod(ferr, "flush", "");
349 if (tmp == NULL)
350 PyErr_Clear();
351 else
352 Py_DECREF(tmp);
353 }
Guido van Rossume8432ac2007-07-09 15:04:50 +0000354}
355
Guido van Rossum25ce5661997-08-02 03:10:38 +0000356/* Undo the effect of Py_Initialize().
357
358 Beware: if multiple interpreter and/or thread states exist, these
359 are not wiped out; only the current thread and interpreter state
360 are deleted. But since everything else is deleted, those other
361 interpreter and thread states should no longer be used.
362
363 (XXX We should do better, e.g. wipe out all interpreters and
364 threads.)
365
366 Locking: as above.
367
368*/
369
370void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000371Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000372{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000373 PyInterpreterState *interp;
374 PyThreadState *tstate;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000375
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000376 if (!initialized)
377 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000378
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000379 wait_for_thread_shutdown();
Antoine Pitrou011bd622009-10-20 21:52:47 +0000380
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000381 /* The interpreter is still entirely intact at this point, and the
382 * exit funcs may be relying on that. In particular, if some thread
383 * or exit func is still waiting to do an import, the import machinery
384 * expects Py_IsInitialized() to return true. So don't say the
385 * interpreter is uninitialized until after the exit funcs have run.
386 * Note that Threading.py uses an exit func to do a join on all the
387 * threads created thru it, so this also protects pending imports in
388 * the threads created via Threading.
389 */
390 call_py_exitfuncs();
391 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000392
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000393 /* Flush stdout+stderr */
394 flush_std_files();
Guido van Rossume8432ac2007-07-09 15:04:50 +0000395
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000396 /* Get current thread state and interpreter pointer */
397 tstate = PyThreadState_GET();
398 interp = tstate->interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000399
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000400 /* Disable signal handling */
401 PyOS_FiniInterrupts();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000402
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000403 /* Clear type lookup cache */
404 PyType_ClearCache();
Christian Heimes26855632008-01-27 23:50:43 +0000405
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000406 /* Collect garbage. This may call finalizers; it's nice to call these
407 * before all modules are destroyed.
408 * XXX If a __del__ or weakref callback is triggered here, and tries to
409 * XXX import a module, bad things can happen, because Python no
410 * XXX longer believes it's initialized.
411 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
412 * XXX is easy to provoke that way. I've also seen, e.g.,
413 * XXX Exception exceptions.ImportError: 'No module named sha'
414 * XXX in <function callback at 0x008F5718> ignored
415 * XXX but I'm unclear on exactly how that one happens. In any case,
416 * XXX I haven't seen a real-life report of either of these.
417 */
418 PyGC_Collect();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000419#ifdef COUNT_ALLOCS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000420 /* With COUNT_ALLOCS, it helps to run GC multiple times:
421 each collection might release some types from the type
422 list, so they become garbage. */
423 while (PyGC_Collect() > 0)
424 /* nothing */;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000425#endif
Antoine Pitrou696e0352010-08-08 22:18:46 +0000426 /* We run this while most interpreter state is still alive, so that
427 debug information can be printed out */
428 _PyGC_Fini();
Guido van Rossume13ddc92003-04-17 17:29:22 +0000429
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000430 /* Destroy all modules */
431 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000432
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000433 /* Flush stdout+stderr (again, in case more was printed) */
434 flush_std_files();
Guido van Rossume8432ac2007-07-09 15:04:50 +0000435
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000436 /* Collect final garbage. This disposes of cycles created by
437 * new-style class definitions, for example.
438 * XXX This is disabled because it caused too many problems. If
439 * XXX a __del__ or weakref callback triggers here, Python code has
440 * XXX a hard time running, because even the sys module has been
441 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
442 * XXX One symptom is a sequence of information-free messages
443 * XXX coming from threads (if a __del__ or callback is invoked,
444 * XXX other threads can execute too, and any exception they encounter
445 * XXX triggers a comedy of errors as subsystem after subsystem
446 * XXX fails to find what it *expects* to find in sys to help report
447 * XXX the exception and consequent unexpected failures). I've also
448 * XXX seen segfaults then, after adding print statements to the
449 * XXX Python code getting called.
450 */
Tim Peters1d7323e2003-12-01 21:35:27 +0000451#if 0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000452 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000453#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000454
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000455 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
456 _PyImport_Fini();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000457
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000458 /* Debugging stuff */
Guido van Rossum1707aad1997-12-08 23:43:45 +0000459#ifdef COUNT_ALLOCS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000460 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000461#endif
462
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000463 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000464
Tim Peters9cf25ce2003-04-17 15:21:01 +0000465#ifdef Py_TRACE_REFS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000466 /* Display all objects still alive -- this can invoke arbitrary
467 * __repr__ overrides, so requires a mostly-intact interpreter.
468 * Alas, a lot of stuff may still be alive now that will be cleaned
469 * up later.
470 */
471 if (Py_GETENV("PYTHONDUMPREFS"))
472 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000473#endif /* Py_TRACE_REFS */
474
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000475 /* Clear interpreter state */
476 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000477
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000478 /* Now we decref the exception classes. After this point nothing
479 can raise an exception. That's okay, because each Fini() method
480 below has been checked to make sure no exceptions are ever
481 raised.
482 */
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000483
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000484 _PyExc_Fini();
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000485
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000486 /* Cleanup auto-thread-state */
Christian Heimes7d2ff882007-11-30 14:35:04 +0000487#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000488 _PyGILState_Fini();
Christian Heimes7d2ff882007-11-30 14:35:04 +0000489#endif /* WITH_THREAD */
490
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000491 /* Delete current thread */
492 PyThreadState_Swap(NULL);
493 PyInterpreterState_Delete(interp);
Barry Warsawf242aa02000-05-25 23:09:49 +0000494
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000495 /* Sundry finalizers */
496 PyMethod_Fini();
497 PyFrame_Fini();
498 PyCFunction_Fini();
499 PyTuple_Fini();
500 PyList_Fini();
501 PySet_Fini();
502 PyBytes_Fini();
503 PyByteArray_Fini();
504 PyLong_Fini();
505 PyFloat_Fini();
506 PyDict_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000507
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000508 /* Cleanup Unicode implementation */
509 _PyUnicode_Fini();
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000510
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000511 /* reset file system default encoding */
Victor Stinnerb744ba12010-05-15 12:27:16 +0000512 if (!Py_HasFileSystemDefaultEncoding && Py_FileSystemDefaultEncoding) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000513 free((char*)Py_FileSystemDefaultEncoding);
514 Py_FileSystemDefaultEncoding = NULL;
515 }
Christian Heimesc8967002007-11-30 10:18:26 +0000516
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000517 /* XXX Still allocated:
518 - various static ad-hoc pointers to interned strings
519 - int and float free list blocks
520 - whatever various modules and libraries allocate
521 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000522
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000523 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000524
Tim Peters269b2a62003-04-17 19:52:29 +0000525#ifdef Py_TRACE_REFS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000526 /* Display addresses (& refcnts) of all objects still alive.
527 * An address can be used to find the repr of the object, printed
528 * above by _Py_PrintReferences.
529 */
530 if (Py_GETENV("PYTHONDUMPREFS"))
531 _Py_PrintReferenceAddresses(stderr);
Tim Peters269b2a62003-04-17 19:52:29 +0000532#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000533#ifdef PYMALLOC_DEBUG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000534 if (Py_GETENV("PYTHONMALLOCSTATS"))
535 _PyObject_DebugMallocStats();
Tim Peters0e871182002-04-13 08:29:14 +0000536#endif
537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000538 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000539}
540
541/* Create and initialize a new interpreter and thread, and return the
542 new thread. This requires that Py_Initialize() has been called
543 first.
544
545 Unsuccessful initialization yields a NULL pointer. Note that *no*
546 exception information is available even in this case -- the
547 exception information is held in the thread, and there is no
548 thread.
549
550 Locking: as above.
551
552*/
553
554PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000555Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000556{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 PyInterpreterState *interp;
558 PyThreadState *tstate, *save_tstate;
559 PyObject *bimod, *sysmod;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000560
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000561 if (!initialized)
562 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000563
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000564 interp = PyInterpreterState_New();
565 if (interp == NULL)
566 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000567
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 tstate = PyThreadState_New(interp);
569 if (tstate == NULL) {
570 PyInterpreterState_Delete(interp);
571 return NULL;
572 }
Guido van Rossum25ce5661997-08-02 03:10:38 +0000573
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 save_tstate = PyThreadState_Swap(tstate);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000575
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000576 /* XXX The following is lax in error checking */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000577
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000578 interp->modules = PyDict_New();
579 interp->modules_reloading = PyDict_New();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000580
Victor Stinner49d3f252010-10-17 01:24:53 +0000581 bimod = _PyImport_FindBuiltin("builtins");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000582 if (bimod != NULL) {
583 interp->builtins = PyModule_GetDict(bimod);
584 if (interp->builtins == NULL)
585 goto handle_error;
586 Py_INCREF(interp->builtins);
587 }
Christian Heimes6a27efa2008-10-30 21:48:26 +0000588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000589 /* initialize builtin exceptions */
590 _PyExc_Init();
Christian Heimes6a27efa2008-10-30 21:48:26 +0000591
Victor Stinner49d3f252010-10-17 01:24:53 +0000592 sysmod = _PyImport_FindBuiltin("sys");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000593 if (bimod != NULL && sysmod != NULL) {
594 PyObject *pstderr;
595 interp->sysdict = PyModule_GetDict(sysmod);
596 if (interp->sysdict == NULL)
597 goto handle_error;
598 Py_INCREF(interp->sysdict);
599 PySys_SetPath(Py_GetPath());
600 PyDict_SetItemString(interp->sysdict, "modules",
601 interp->modules);
602 /* Set up a preliminary stderr printer until we have enough
603 infrastructure for the io module in place. */
604 pstderr = PyFile_NewStdPrinter(fileno(stderr));
605 if (pstderr == NULL)
606 Py_FatalError("Py_Initialize: can't set preliminary stderr");
607 PySys_SetObject("stderr", pstderr);
608 PySys_SetObject("__stderr__", pstderr);
Hirokazu Yamamotodaf83ac2010-10-30 15:08:15 +0000609 Py_DECREF(pstderr);
Christian Heimes6a27efa2008-10-30 21:48:26 +0000610
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000611 _PyImportHooks_Init();
612 if (initstdio() < 0)
613 Py_FatalError(
614 "Py_Initialize: can't initialize sys standard streams");
615 initmain();
616 if (!Py_NoSiteFlag)
617 initsite();
618 }
Guido van Rossum25ce5661997-08-02 03:10:38 +0000619
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000620 if (!PyErr_Occurred())
621 return tstate;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000622
Thomas Wouters89f507f2006-12-13 04:49:30 +0000623handle_error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000624 /* Oops, it didn't work. Undo it all. */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000625
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000626 PyErr_Print();
627 PyThreadState_Clear(tstate);
628 PyThreadState_Swap(save_tstate);
629 PyThreadState_Delete(tstate);
630 PyInterpreterState_Delete(interp);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000631
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000632 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000633}
634
635/* Delete an interpreter and its last thread. This requires that the
636 given thread state is current, that the thread has no remaining
637 frames, and that it is its interpreter's only remaining thread.
638 It is a fatal error to violate these constraints.
639
640 (Py_Finalize() doesn't have these constraints -- it zaps
641 everything, regardless.)
642
643 Locking: as above.
644
645*/
646
647void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000648Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000649{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000650 PyInterpreterState *interp = tstate->interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000651
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000652 if (tstate != PyThreadState_GET())
653 Py_FatalError("Py_EndInterpreter: thread is not current");
654 if (tstate->frame != NULL)
655 Py_FatalError("Py_EndInterpreter: thread still has a frame");
656 if (tstate != interp->tstate_head || tstate->next != NULL)
657 Py_FatalError("Py_EndInterpreter: not the last thread");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000658
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000659 PyImport_Cleanup();
660 PyInterpreterState_Clear(interp);
661 PyThreadState_Swap(NULL);
662 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000663}
664
Martin v. Löwis790465f2008-04-05 20:41:37 +0000665static wchar_t *progname = L"python";
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000666
667void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000668Py_SetProgramName(wchar_t *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000669{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000670 if (pn && *pn)
671 progname = pn;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000672}
673
Martin v. Löwis790465f2008-04-05 20:41:37 +0000674wchar_t *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000675Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000676{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000677 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000678}
679
Martin v. Löwis790465f2008-04-05 20:41:37 +0000680static wchar_t *default_home = NULL;
681static wchar_t env_home[PATH_MAX+1];
Guido van Rossuma61691e1998-02-06 22:27:24 +0000682
683void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000684Py_SetPythonHome(wchar_t *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000685{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 default_home = home;
Guido van Rossuma61691e1998-02-06 22:27:24 +0000687}
688
Martin v. Löwis790465f2008-04-05 20:41:37 +0000689wchar_t *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000690Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000691{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000692 wchar_t *home = default_home;
693 if (home == NULL && !Py_IgnoreEnvironmentFlag) {
694 char* chome = Py_GETENV("PYTHONHOME");
695 if (chome) {
696 size_t r = mbstowcs(env_home, chome, PATH_MAX+1);
697 if (r != (size_t)-1 && r <= PATH_MAX)
698 home = env_home;
699 }
Martin v. Löwis790465f2008-04-05 20:41:37 +0000700
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000701 }
702 return home;
Guido van Rossuma61691e1998-02-06 22:27:24 +0000703}
704
Guido van Rossum6135a871995-01-09 17:53:26 +0000705/* Create __main__ module */
706
707static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000708initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000709{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 PyObject *m, *d;
711 m = PyImport_AddModule("__main__");
712 if (m == NULL)
713 Py_FatalError("can't create __main__ module");
714 d = PyModule_GetDict(m);
715 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
716 PyObject *bimod = PyImport_ImportModule("builtins");
717 if (bimod == NULL ||
718 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
719 Py_FatalError("can't add __builtins__ to __main__");
720 Py_DECREF(bimod);
721 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000722}
723
Victor Stinnerb744ba12010-05-15 12:27:16 +0000724static void
725initfsencoding(void)
726{
727 PyObject *codec;
728#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Victor Stinner94908bb2010-08-18 21:23:25 +0000729 char *codeset = NULL;
Victor Stinnerb744ba12010-05-15 12:27:16 +0000730
Victor Stinner7f84ab52010-06-11 00:36:33 +0000731 if (Py_FileSystemDefaultEncoding == NULL) {
Victor Stinner8f6b6b02010-10-13 22:02:27 +0000732 /* On Unix, set the file system encoding according to the
733 user's preference, if the CODESET names a well-known
734 Python codec, and Py_FileSystemDefaultEncoding isn't
Victor Stinnere4743092010-10-19 00:05:51 +0000735 initialized by other means. */
Victor Stinner8f6b6b02010-10-13 22:02:27 +0000736 codeset = get_codeset();
Victor Stinnere4743092010-10-19 00:05:51 +0000737 if (codeset == NULL)
738 Py_FatalError("Py_Initialize: Unable to get the locale encoding");
Victor Stinnerb744ba12010-05-15 12:27:16 +0000739
Victor Stinnere4743092010-10-19 00:05:51 +0000740 Py_FileSystemDefaultEncoding = codeset;
741 Py_HasFileSystemDefaultEncoding = 0;
742 return;
Victor Stinner7f84ab52010-06-11 00:36:33 +0000743 }
Victor Stinnerb744ba12010-05-15 12:27:16 +0000744#endif
745
746 /* the encoding is mbcs, utf-8 or ascii */
747 codec = _PyCodec_Lookup(Py_FileSystemDefaultEncoding);
748 if (!codec) {
749 /* Such error can only occurs in critical situations: no more
750 * memory, import a module of the standard library failed,
751 * etc. */
752 Py_FatalError("Py_Initialize: unable to load the file system codec");
753 } else {
754 Py_DECREF(codec);
755 }
756}
757
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000758/* Import the site module (not into __main__ though) */
759
760static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000761initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000762{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000763 PyObject *m;
764 m = PyImport_ImportModule("site");
765 if (m == NULL) {
766 PyErr_Print();
767 Py_Finalize();
768 exit(1);
769 }
770 else {
771 Py_DECREF(m);
772 }
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000773}
774
Antoine Pitrou05608432009-01-09 18:53:14 +0000775static PyObject*
776create_stdio(PyObject* io,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000777 int fd, int write_mode, char* name,
778 char* encoding, char* errors)
Antoine Pitrou05608432009-01-09 18:53:14 +0000779{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000780 PyObject *buf = NULL, *stream = NULL, *text = NULL, *raw = NULL, *res;
781 const char* mode;
782 PyObject *line_buffering;
783 int buffering, isatty;
Antoine Pitrou05608432009-01-09 18:53:14 +0000784
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000785 /* stdin is always opened in buffered mode, first because it shouldn't
786 make a difference in common use cases, second because TextIOWrapper
787 depends on the presence of a read1() method which only exists on
788 buffered streams.
789 */
790 if (Py_UnbufferedStdioFlag && write_mode)
791 buffering = 0;
792 else
793 buffering = -1;
794 if (write_mode)
795 mode = "wb";
796 else
797 mode = "rb";
798 buf = PyObject_CallMethod(io, "open", "isiOOOi",
799 fd, mode, buffering,
800 Py_None, Py_None, Py_None, 0);
801 if (buf == NULL)
802 goto error;
Antoine Pitrou05608432009-01-09 18:53:14 +0000803
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000804 if (buffering) {
805 raw = PyObject_GetAttrString(buf, "raw");
806 if (raw == NULL)
807 goto error;
808 }
809 else {
810 raw = buf;
811 Py_INCREF(raw);
812 }
Antoine Pitrou05608432009-01-09 18:53:14 +0000813
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000814 text = PyUnicode_FromString(name);
815 if (text == NULL || PyObject_SetAttrString(raw, "name", text) < 0)
816 goto error;
817 res = PyObject_CallMethod(raw, "isatty", "");
818 if (res == NULL)
819 goto error;
820 isatty = PyObject_IsTrue(res);
821 Py_DECREF(res);
822 if (isatty == -1)
823 goto error;
824 if (isatty || Py_UnbufferedStdioFlag)
825 line_buffering = Py_True;
826 else
827 line_buffering = Py_False;
Antoine Pitrou91696412009-01-09 22:12:30 +0000828
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000829 Py_CLEAR(raw);
830 Py_CLEAR(text);
Antoine Pitrou91696412009-01-09 22:12:30 +0000831
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000832 stream = PyObject_CallMethod(io, "TextIOWrapper", "OsssO",
833 buf, encoding, errors,
834 "\n", line_buffering);
835 Py_CLEAR(buf);
836 if (stream == NULL)
837 goto error;
Antoine Pitrou05608432009-01-09 18:53:14 +0000838
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000839 if (write_mode)
840 mode = "w";
841 else
842 mode = "r";
843 text = PyUnicode_FromString(mode);
844 if (!text || PyObject_SetAttrString(stream, "mode", text) < 0)
845 goto error;
846 Py_CLEAR(text);
847 return stream;
Antoine Pitrou05608432009-01-09 18:53:14 +0000848
849error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000850 Py_XDECREF(buf);
851 Py_XDECREF(stream);
852 Py_XDECREF(text);
853 Py_XDECREF(raw);
854 return NULL;
Antoine Pitrou05608432009-01-09 18:53:14 +0000855}
856
Georg Brandl1a3284e2007-12-02 09:40:06 +0000857/* Initialize sys.stdin, stdout, stderr and builtins.open */
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000858static int
859initstdio(void)
860{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000861 PyObject *iomod = NULL, *wrapper;
862 PyObject *bimod = NULL;
863 PyObject *m;
864 PyObject *std = NULL;
865 int status = 0, fd;
866 PyObject * encoding_attr;
867 char *encoding = NULL, *errors;
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000869 /* Hack to avoid a nasty recursion issue when Python is invoked
870 in verbose mode: pre-import the Latin-1 and UTF-8 codecs */
871 if ((m = PyImport_ImportModule("encodings.utf_8")) == NULL) {
872 goto error;
873 }
874 Py_DECREF(m);
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000875
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000876 if (!(m = PyImport_ImportModule("encodings.latin_1"))) {
877 goto error;
878 }
879 Py_DECREF(m);
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000880
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000881 if (!(bimod = PyImport_ImportModule("builtins"))) {
882 goto error;
883 }
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000884
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000885 if (!(iomod = PyImport_ImportModule("io"))) {
886 goto error;
887 }
888 if (!(wrapper = PyObject_GetAttrString(iomod, "OpenWrapper"))) {
889 goto error;
890 }
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000891
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000892 /* Set builtins.open */
893 if (PyObject_SetAttrString(bimod, "open", wrapper) == -1) {
894 goto error;
895 }
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000896
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000897 encoding = Py_GETENV("PYTHONIOENCODING");
898 errors = NULL;
899 if (encoding) {
900 encoding = strdup(encoding);
901 errors = strchr(encoding, ':');
902 if (errors) {
903 *errors = '\0';
904 errors++;
905 }
906 }
Martin v. Löwis0f599892008-06-02 11:13:03 +0000907
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000908 /* Set sys.stdin */
909 fd = fileno(stdin);
910 /* Under some conditions stdin, stdout and stderr may not be connected
911 * and fileno() may point to an invalid file descriptor. For example
912 * GUI apps don't have valid standard streams by default.
913 */
914 if (fd < 0) {
Christian Heimes58cb1b82007-11-13 02:19:40 +0000915#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000916 std = Py_None;
917 Py_INCREF(std);
Christian Heimes58cb1b82007-11-13 02:19:40 +0000918#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000919 goto error;
Christian Heimes58cb1b82007-11-13 02:19:40 +0000920#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000921 }
922 else {
923 std = create_stdio(iomod, fd, 0, "<stdin>", encoding, errors);
924 if (std == NULL)
925 goto error;
926 } /* if (fd < 0) */
927 PySys_SetObject("__stdin__", std);
928 PySys_SetObject("stdin", std);
929 Py_DECREF(std);
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000930
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000931 /* Set sys.stdout */
932 fd = fileno(stdout);
933 if (fd < 0) {
Christian Heimes58cb1b82007-11-13 02:19:40 +0000934#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000935 std = Py_None;
936 Py_INCREF(std);
Christian Heimes58cb1b82007-11-13 02:19:40 +0000937#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000938 goto error;
Christian Heimes58cb1b82007-11-13 02:19:40 +0000939#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000940 }
941 else {
942 std = create_stdio(iomod, fd, 1, "<stdout>", encoding, errors);
943 if (std == NULL)
944 goto error;
945 } /* if (fd < 0) */
946 PySys_SetObject("__stdout__", std);
947 PySys_SetObject("stdout", std);
948 Py_DECREF(std);
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000949
Guido van Rossum98297ee2007-11-06 21:34:58 +0000950#if 1 /* Disable this if you have trouble debugging bootstrap stuff */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000951 /* Set sys.stderr, replaces the preliminary stderr */
952 fd = fileno(stderr);
953 if (fd < 0) {
Christian Heimes58cb1b82007-11-13 02:19:40 +0000954#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000955 std = Py_None;
956 Py_INCREF(std);
Christian Heimes58cb1b82007-11-13 02:19:40 +0000957#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000958 goto error;
Christian Heimes58cb1b82007-11-13 02:19:40 +0000959#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000960 }
961 else {
962 std = create_stdio(iomod, fd, 1, "<stderr>", encoding, "backslashreplace");
963 if (std == NULL)
964 goto error;
965 } /* if (fd < 0) */
Trent Nelson39e307e2008-03-19 06:45:48 +0000966
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000967 /* Same as hack above, pre-import stderr's codec to avoid recursion
968 when import.c tries to write to stderr in verbose mode. */
969 encoding_attr = PyObject_GetAttrString(std, "encoding");
970 if (encoding_attr != NULL) {
971 const char * encoding;
972 encoding = _PyUnicode_AsString(encoding_attr);
973 if (encoding != NULL) {
974 _PyCodec_Lookup(encoding);
975 }
Hirokazu Yamamotodaf83ac2010-10-30 15:08:15 +0000976 Py_DECREF(encoding_attr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000977 }
978 PyErr_Clear(); /* Not a fatal error if codec isn't available */
Trent Nelson39e307e2008-03-19 06:45:48 +0000979
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000980 PySys_SetObject("__stderr__", std);
981 PySys_SetObject("stderr", std);
982 Py_DECREF(std);
Guido van Rossum98297ee2007-11-06 21:34:58 +0000983#endif
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000984
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000985 if (0) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000986 error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000987 status = -1;
988 }
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000989
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000990 if (encoding)
991 free(encoding);
992 Py_XDECREF(bimod);
993 Py_XDECREF(iomod);
994 return status;
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000995}
996
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000997/* Parse input from a file and execute it */
998
999int
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001000PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001001 PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001002{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001003 if (filename == NULL)
1004 filename = "???";
1005 if (Py_FdIsInteractive(fp, filename)) {
1006 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
1007 if (closeit)
1008 fclose(fp);
1009 return err;
1010 }
1011 else
1012 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001013}
1014
1015int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001016PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001017{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001018 PyObject *v;
1019 int ret;
1020 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +00001021
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001022 if (flags == NULL) {
1023 flags = &local_flags;
1024 local_flags.cf_flags = 0;
1025 }
1026 v = PySys_GetObject("ps1");
1027 if (v == NULL) {
1028 PySys_SetObject("ps1", v = PyUnicode_FromString(">>> "));
1029 Py_XDECREF(v);
1030 }
1031 v = PySys_GetObject("ps2");
1032 if (v == NULL) {
1033 PySys_SetObject("ps2", v = PyUnicode_FromString("... "));
1034 Py_XDECREF(v);
1035 }
1036 for (;;) {
1037 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
1038 PRINT_TOTAL_REFS();
1039 if (ret == E_EOF)
1040 return 0;
1041 /*
1042 if (ret == E_NOMEM)
1043 return -1;
1044 */
1045 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001046}
1047
Neil Schemenauerc24ea082002-03-22 23:53:36 +00001048/* compute parser flags based on compiler flags */
Benjamin Petersonf5b52242009-03-02 23:31:26 +00001049static int PARSER_FLAGS(PyCompilerFlags *flags)
1050{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001051 int parser_flags = 0;
1052 if (!flags)
1053 return 0;
1054 if (flags->cf_flags & PyCF_DONT_IMPLY_DEDENT)
1055 parser_flags |= PyPARSE_DONT_IMPLY_DEDENT;
1056 if (flags->cf_flags & PyCF_IGNORE_COOKIE)
1057 parser_flags |= PyPARSE_IGNORE_COOKIE;
1058 if (flags->cf_flags & CO_FUTURE_BARRY_AS_BDFL)
1059 parser_flags |= PyPARSE_BARRY_AS_BDFL;
1060 return parser_flags;
Benjamin Petersonf5b52242009-03-02 23:31:26 +00001061}
Neil Schemenauerc24ea082002-03-22 23:53:36 +00001062
Thomas Wouters89f507f2006-12-13 04:49:30 +00001063#if 0
1064/* Keep an example of flags with future keyword support. */
1065#define PARSER_FLAGS(flags) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001066 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
1067 PyPARSE_DONT_IMPLY_DEDENT : 0) \
1068 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
1069 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001070#endif
1071
Jeremy Hylton9f324e92001-03-01 22:59:14 +00001072int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001073PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +00001074{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001075 PyObject *m, *d, *v, *w, *oenc = NULL;
1076 mod_ty mod;
1077 PyArena *arena;
1078 char *ps1 = "", *ps2 = "", *enc = NULL;
1079 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +00001080
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001081 if (fp == stdin) {
1082 /* Fetch encoding from sys.stdin */
1083 v = PySys_GetObject("stdin");
1084 if (v == NULL || v == Py_None)
1085 return -1;
1086 oenc = PyObject_GetAttrString(v, "encoding");
1087 if (!oenc)
1088 return -1;
1089 enc = _PyUnicode_AsString(oenc);
Victor Stinner386fe712010-05-19 00:34:15 +00001090 if (enc == NULL)
1091 return -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001092 }
1093 v = PySys_GetObject("ps1");
1094 if (v != NULL) {
1095 v = PyObject_Str(v);
1096 if (v == NULL)
1097 PyErr_Clear();
Victor Stinner386fe712010-05-19 00:34:15 +00001098 else if (PyUnicode_Check(v)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001099 ps1 = _PyUnicode_AsString(v);
Victor Stinner386fe712010-05-19 00:34:15 +00001100 if (ps1 == NULL) {
1101 PyErr_Clear();
1102 ps1 = "";
1103 }
1104 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001105 }
1106 w = PySys_GetObject("ps2");
1107 if (w != NULL) {
1108 w = PyObject_Str(w);
1109 if (w == NULL)
1110 PyErr_Clear();
Victor Stinner386fe712010-05-19 00:34:15 +00001111 else if (PyUnicode_Check(w)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001112 ps2 = _PyUnicode_AsString(w);
Victor Stinner386fe712010-05-19 00:34:15 +00001113 if (ps2 == NULL) {
1114 PyErr_Clear();
1115 ps2 = "";
1116 }
1117 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001118 }
1119 arena = PyArena_New();
1120 if (arena == NULL) {
1121 Py_XDECREF(v);
1122 Py_XDECREF(w);
1123 Py_XDECREF(oenc);
1124 return -1;
1125 }
1126 mod = PyParser_ASTFromFile(fp, filename, enc,
1127 Py_single_input, ps1, ps2,
1128 flags, &errcode, arena);
1129 Py_XDECREF(v);
1130 Py_XDECREF(w);
1131 Py_XDECREF(oenc);
1132 if (mod == NULL) {
1133 PyArena_Free(arena);
1134 if (errcode == E_EOF) {
1135 PyErr_Clear();
1136 return E_EOF;
1137 }
1138 PyErr_Print();
1139 return -1;
1140 }
1141 m = PyImport_AddModule("__main__");
1142 if (m == NULL) {
1143 PyArena_Free(arena);
1144 return -1;
1145 }
1146 d = PyModule_GetDict(m);
1147 v = run_mod(mod, filename, d, d, flags, arena);
1148 PyArena_Free(arena);
1149 flush_io();
1150 if (v == NULL) {
1151 PyErr_Print();
1152 return -1;
1153 }
1154 Py_DECREF(v);
1155 return 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001156}
1157
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +00001158/* Check whether a file maybe a pyc file: Look at the extension,
1159 the file type, and, if we may close it, at the first few bytes. */
1160
1161static int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001162maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +00001163{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001164 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
1165 return 1;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +00001166
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001167 /* Only look into the file if we are allowed to close it, since
1168 it then should also be seekable. */
1169 if (closeit) {
1170 /* Read only two bytes of the magic. If the file was opened in
1171 text mode, the bytes 3 and 4 of the magic (\r\n) might not
1172 be read as they are on disk. */
1173 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
1174 unsigned char buf[2];
1175 /* Mess: In case of -x, the stream is NOT at its start now,
1176 and ungetc() was used to push back the first newline,
1177 which makes the current stream position formally undefined,
1178 and a x-platform nightmare.
1179 Unfortunately, we have no direct way to know whether -x
1180 was specified. So we use a terrible hack: if the current
1181 stream position is not 0, we assume -x was specified, and
1182 give up. Bug 132850 on SourceForge spells out the
1183 hopelessness of trying anything else (fseek and ftell
1184 don't work predictably x-platform for text-mode files).
1185 */
1186 int ispyc = 0;
1187 if (ftell(fp) == 0) {
1188 if (fread(buf, 1, 2, fp) == 2 &&
1189 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
1190 ispyc = 1;
1191 rewind(fp);
1192 }
1193 return ispyc;
1194 }
1195 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +00001196}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +00001197
Guido van Rossum0df002c2000-08-27 19:21:52 +00001198int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001199PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001200 PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001201{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 PyObject *m, *d, *v;
1203 const char *ext;
1204 int set_file_name = 0, ret, len;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001205
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001206 m = PyImport_AddModule("__main__");
1207 if (m == NULL)
1208 return -1;
1209 d = PyModule_GetDict(m);
1210 if (PyDict_GetItemString(d, "__file__") == NULL) {
1211 PyObject *f;
Victor Stinner4c7c8c32010-10-16 13:14:10 +00001212 f = PyUnicode_DecodeFSDefault(filename);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001213 if (f == NULL)
1214 return -1;
1215 if (PyDict_SetItemString(d, "__file__", f) < 0) {
1216 Py_DECREF(f);
1217 return -1;
1218 }
1219 if (PyDict_SetItemString(d, "__cached__", Py_None) < 0)
1220 return -1;
1221 set_file_name = 1;
1222 Py_DECREF(f);
1223 }
1224 len = strlen(filename);
1225 ext = filename + len - (len > 4 ? 4 : 0);
1226 if (maybe_pyc_file(fp, filename, ext, closeit)) {
1227 /* Try to run a pyc file. First, re-open in binary */
1228 if (closeit)
1229 fclose(fp);
1230 if ((fp = fopen(filename, "rb")) == NULL) {
1231 fprintf(stderr, "python: Can't reopen .pyc file\n");
1232 ret = -1;
1233 goto done;
1234 }
1235 /* Turn on optimization if a .pyo file is given */
1236 if (strcmp(ext, ".pyo") == 0)
1237 Py_OptimizeFlag = 1;
1238 v = run_pyc_file(fp, filename, d, d, flags);
1239 } else {
1240 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
1241 closeit, flags);
1242 }
1243 flush_io();
1244 if (v == NULL) {
1245 PyErr_Print();
1246 ret = -1;
1247 goto done;
1248 }
1249 Py_DECREF(v);
1250 ret = 0;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001251 done:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001252 if (set_file_name && PyDict_DelItemString(d, "__file__"))
1253 PyErr_Clear();
1254 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001255}
1256
1257int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001258PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +00001259{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001260 PyObject *m, *d, *v;
1261 m = PyImport_AddModule("__main__");
1262 if (m == NULL)
1263 return -1;
1264 d = PyModule_GetDict(m);
1265 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
1266 if (v == NULL) {
1267 PyErr_Print();
1268 return -1;
1269 }
1270 Py_DECREF(v);
1271 return 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001272}
1273
Barry Warsaw035574d1997-08-29 22:07:17 +00001274static int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001275parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001276 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +00001277{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001278 long hold;
1279 PyObject *v;
Barry Warsaw035574d1997-08-29 22:07:17 +00001280
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001281 /* old style errors */
1282 if (PyTuple_Check(err))
1283 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
1284 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +00001285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001286 /* new style errors. `err' is an instance */
Barry Warsaw035574d1997-08-29 22:07:17 +00001287
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001288 if (! (v = PyObject_GetAttrString(err, "msg")))
1289 goto finally;
1290 *message = v;
Barry Warsaw035574d1997-08-29 22:07:17 +00001291
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001292 if (!(v = PyObject_GetAttrString(err, "filename")))
1293 goto finally;
1294 if (v == Py_None)
1295 *filename = NULL;
1296 else if (! (*filename = _PyUnicode_AsString(v)))
1297 goto finally;
Barry Warsaw035574d1997-08-29 22:07:17 +00001298
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001299 Py_DECREF(v);
1300 if (!(v = PyObject_GetAttrString(err, "lineno")))
1301 goto finally;
1302 hold = PyLong_AsLong(v);
1303 Py_DECREF(v);
1304 v = NULL;
1305 if (hold < 0 && PyErr_Occurred())
1306 goto finally;
1307 *lineno = (int)hold;
Barry Warsaw035574d1997-08-29 22:07:17 +00001308
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001309 if (!(v = PyObject_GetAttrString(err, "offset")))
1310 goto finally;
1311 if (v == Py_None) {
1312 *offset = -1;
1313 Py_DECREF(v);
1314 v = NULL;
1315 } else {
1316 hold = PyLong_AsLong(v);
1317 Py_DECREF(v);
1318 v = NULL;
1319 if (hold < 0 && PyErr_Occurred())
1320 goto finally;
1321 *offset = (int)hold;
1322 }
Barry Warsaw035574d1997-08-29 22:07:17 +00001323
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001324 if (!(v = PyObject_GetAttrString(err, "text")))
1325 goto finally;
1326 if (v == Py_None)
1327 *text = NULL;
1328 else if (!PyUnicode_Check(v) ||
1329 !(*text = _PyUnicode_AsString(v)))
1330 goto finally;
1331 Py_DECREF(v);
1332 return 1;
Barry Warsaw035574d1997-08-29 22:07:17 +00001333
1334finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001335 Py_XDECREF(v);
1336 return 0;
Barry Warsaw035574d1997-08-29 22:07:17 +00001337}
1338
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001339void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001340PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001341{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001342 PyErr_PrintEx(1);
Guido van Rossuma61691e1998-02-06 22:27:24 +00001343}
1344
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001345static void
Martin v. Löwis95292d62002-12-11 14:04:59 +00001346print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001347{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001348 char *nl;
1349 if (offset >= 0) {
Benjamin Petersona95e9772010-10-29 03:28:14 +00001350 if (offset > 0 && offset == strlen(text) && text[offset - 1] == '\n')
1351 offset--;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001352 for (;;) {
1353 nl = strchr(text, '\n');
1354 if (nl == NULL || nl-text >= offset)
1355 break;
1356 offset -= (int)(nl+1-text);
1357 text = nl+1;
1358 }
1359 while (*text == ' ' || *text == '\t') {
1360 text++;
1361 offset--;
1362 }
1363 }
1364 PyFile_WriteString(" ", f);
1365 PyFile_WriteString(text, f);
1366 if (*text == '\0' || text[strlen(text)-1] != '\n')
1367 PyFile_WriteString("\n", f);
1368 if (offset == -1)
1369 return;
1370 PyFile_WriteString(" ", f);
Benjamin Petersona95e9772010-10-29 03:28:14 +00001371 while (--offset > 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001372 PyFile_WriteString(" ", f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001373 PyFile_WriteString("^\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001374}
1375
Guido van Rossum66e8e862001-03-23 17:54:43 +00001376static void
1377handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001378{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001379 PyObject *exception, *value, *tb;
1380 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001381
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001382 if (Py_InspectFlag)
1383 /* Don't exit if -i flag was given. This flag is set to 0
1384 * when entering interactive mode for inspecting. */
1385 return;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001386
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001387 PyErr_Fetch(&exception, &value, &tb);
1388 fflush(stdout);
1389 if (value == NULL || value == Py_None)
1390 goto done;
1391 if (PyExceptionInstance_Check(value)) {
1392 /* The error code should be in the `code' attribute. */
1393 PyObject *code = PyObject_GetAttrString(value, "code");
1394 if (code) {
1395 Py_DECREF(value);
1396 value = code;
1397 if (value == Py_None)
1398 goto done;
1399 }
1400 /* If we failed to dig out the 'code' attribute,
1401 just let the else clause below print the error. */
1402 }
1403 if (PyLong_Check(value))
1404 exitcode = (int)PyLong_AsLong(value);
1405 else {
Victor Stinnere9fb3192010-05-17 08:58:51 +00001406 PyObject *sys_stderr = PySys_GetObject("stderr");
Victor Stinner7126dbc2010-05-21 23:45:42 +00001407 if (sys_stderr != NULL && sys_stderr != Py_None) {
1408 PyFile_WriteObject(value, sys_stderr, Py_PRINT_RAW);
1409 } else {
1410 PyObject_Print(value, stderr, Py_PRINT_RAW);
1411 fflush(stderr);
1412 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001413 PySys_WriteStderr("\n");
1414 exitcode = 1;
1415 }
Tim Peterscf615b52003-04-19 18:47:02 +00001416 done:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001417 /* Restore and clear the exception info, in order to properly decref
1418 * the exception, value, and traceback. If we just exit instead,
1419 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1420 * some finalizers from running.
1421 */
1422 PyErr_Restore(exception, value, tb);
1423 PyErr_Clear();
1424 Py_Exit(exitcode);
1425 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001426}
1427
1428void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001429PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001430{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001431 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001432
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001433 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1434 handle_system_exit();
1435 }
1436 PyErr_Fetch(&exception, &v, &tb);
1437 if (exception == NULL)
1438 return;
1439 PyErr_NormalizeException(&exception, &v, &tb);
1440 if (tb == NULL) {
1441 tb = Py_None;
1442 Py_INCREF(tb);
1443 }
1444 PyException_SetTraceback(v, tb);
1445 if (exception == NULL)
1446 return;
1447 /* Now we know v != NULL too */
1448 if (set_sys_last_vars) {
1449 PySys_SetObject("last_type", exception);
1450 PySys_SetObject("last_value", v);
1451 PySys_SetObject("last_traceback", tb);
1452 }
1453 hook = PySys_GetObject("excepthook");
1454 if (hook) {
1455 PyObject *args = PyTuple_Pack(3, exception, v, tb);
1456 PyObject *result = PyEval_CallObject(hook, args);
1457 if (result == NULL) {
1458 PyObject *exception2, *v2, *tb2;
1459 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1460 handle_system_exit();
1461 }
1462 PyErr_Fetch(&exception2, &v2, &tb2);
1463 PyErr_NormalizeException(&exception2, &v2, &tb2);
1464 /* It should not be possible for exception2 or v2
1465 to be NULL. However PyErr_Display() can't
1466 tolerate NULLs, so just be safe. */
1467 if (exception2 == NULL) {
1468 exception2 = Py_None;
1469 Py_INCREF(exception2);
1470 }
1471 if (v2 == NULL) {
1472 v2 = Py_None;
1473 Py_INCREF(v2);
1474 }
1475 fflush(stdout);
1476 PySys_WriteStderr("Error in sys.excepthook:\n");
1477 PyErr_Display(exception2, v2, tb2);
1478 PySys_WriteStderr("\nOriginal exception was:\n");
1479 PyErr_Display(exception, v, tb);
1480 Py_DECREF(exception2);
1481 Py_DECREF(v2);
1482 Py_XDECREF(tb2);
1483 }
1484 Py_XDECREF(result);
1485 Py_XDECREF(args);
1486 } else {
1487 PySys_WriteStderr("sys.excepthook is missing\n");
1488 PyErr_Display(exception, v, tb);
1489 }
1490 Py_XDECREF(exception);
1491 Py_XDECREF(v);
1492 Py_XDECREF(tb);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001493}
1494
Benjamin Petersone6528212008-07-15 15:32:09 +00001495static void
1496print_exception(PyObject *f, PyObject *value)
1497{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001498 int err = 0;
1499 PyObject *type, *tb;
Benjamin Petersone6528212008-07-15 15:32:09 +00001500
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001501 if (!PyExceptionInstance_Check(value)) {
1502 PyFile_WriteString("TypeError: print_exception(): Exception expected for value, ", f);
1503 PyFile_WriteString(Py_TYPE(value)->tp_name, f);
1504 PyFile_WriteString(" found\n", f);
1505 return;
1506 }
Benjamin Peterson26582602008-08-23 20:08:07 +00001507
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001508 Py_INCREF(value);
1509 fflush(stdout);
1510 type = (PyObject *) Py_TYPE(value);
1511 tb = PyException_GetTraceback(value);
1512 if (tb && tb != Py_None)
1513 err = PyTraceBack_Print(tb, f);
1514 if (err == 0 &&
1515 PyObject_HasAttrString(value, "print_file_and_line"))
1516 {
1517 PyObject *message;
1518 const char *filename, *text;
1519 int lineno, offset;
1520 if (!parse_syntax_error(value, &message, &filename,
1521 &lineno, &offset, &text))
1522 PyErr_Clear();
1523 else {
1524 char buf[10];
1525 PyFile_WriteString(" File \"", f);
1526 if (filename == NULL)
1527 PyFile_WriteString("<string>", f);
1528 else
1529 PyFile_WriteString(filename, f);
1530 PyFile_WriteString("\", line ", f);
1531 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
1532 PyFile_WriteString(buf, f);
1533 PyFile_WriteString("\n", f);
1534 if (text != NULL)
1535 print_error_text(f, offset, text);
1536 Py_DECREF(value);
1537 value = message;
1538 /* Can't be bothered to check all those
1539 PyFile_WriteString() calls */
1540 if (PyErr_Occurred())
1541 err = -1;
1542 }
1543 }
1544 if (err) {
1545 /* Don't do anything else */
1546 }
1547 else {
1548 PyObject* moduleName;
1549 char* className;
1550 assert(PyExceptionClass_Check(type));
1551 className = PyExceptionClass_Name(type);
1552 if (className != NULL) {
1553 char *dot = strrchr(className, '.');
1554 if (dot != NULL)
1555 className = dot+1;
1556 }
Benjamin Petersone6528212008-07-15 15:32:09 +00001557
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001558 moduleName = PyObject_GetAttrString(type, "__module__");
1559 if (moduleName == NULL || !PyUnicode_Check(moduleName))
1560 {
1561 Py_DECREF(moduleName);
1562 err = PyFile_WriteString("<unknown>", f);
1563 }
1564 else {
1565 char* modstr = _PyUnicode_AsString(moduleName);
1566 if (modstr && strcmp(modstr, "builtins"))
1567 {
1568 err = PyFile_WriteString(modstr, f);
1569 err += PyFile_WriteString(".", f);
1570 }
1571 Py_DECREF(moduleName);
1572 }
1573 if (err == 0) {
1574 if (className == NULL)
1575 err = PyFile_WriteString("<unknown>", f);
1576 else
1577 err = PyFile_WriteString(className, f);
1578 }
1579 }
1580 if (err == 0 && (value != Py_None)) {
1581 PyObject *s = PyObject_Str(value);
1582 /* only print colon if the str() of the
1583 object is not the empty string
1584 */
1585 if (s == NULL)
1586 err = -1;
1587 else if (!PyUnicode_Check(s) ||
1588 PyUnicode_GetSize(s) != 0)
1589 err = PyFile_WriteString(": ", f);
1590 if (err == 0)
1591 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1592 Py_XDECREF(s);
1593 }
1594 /* try to write a newline in any case */
1595 err += PyFile_WriteString("\n", f);
1596 Py_XDECREF(tb);
1597 Py_DECREF(value);
1598 /* If an error happened here, don't show it.
1599 XXX This is wrong, but too many callers rely on this behavior. */
1600 if (err != 0)
1601 PyErr_Clear();
Benjamin Petersone6528212008-07-15 15:32:09 +00001602}
1603
1604static const char *cause_message =
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001605 "\nThe above exception was the direct cause "
1606 "of the following exception:\n\n";
Benjamin Petersone6528212008-07-15 15:32:09 +00001607
1608static const char *context_message =
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001609 "\nDuring handling of the above exception, "
1610 "another exception occurred:\n\n";
Benjamin Petersone6528212008-07-15 15:32:09 +00001611
1612static void
1613print_exception_recursive(PyObject *f, PyObject *value, PyObject *seen)
1614{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001615 int err = 0, res;
1616 PyObject *cause, *context;
Benjamin Petersone6528212008-07-15 15:32:09 +00001617
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001618 if (seen != NULL) {
1619 /* Exception chaining */
1620 if (PySet_Add(seen, value) == -1)
1621 PyErr_Clear();
1622 else if (PyExceptionInstance_Check(value)) {
1623 cause = PyException_GetCause(value);
1624 context = PyException_GetContext(value);
1625 if (cause) {
1626 res = PySet_Contains(seen, cause);
1627 if (res == -1)
1628 PyErr_Clear();
1629 if (res == 0) {
1630 print_exception_recursive(
1631 f, cause, seen);
1632 err |= PyFile_WriteString(
1633 cause_message, f);
1634 }
1635 }
1636 else if (context) {
1637 res = PySet_Contains(seen, context);
1638 if (res == -1)
1639 PyErr_Clear();
1640 if (res == 0) {
1641 print_exception_recursive(
1642 f, context, seen);
1643 err |= PyFile_WriteString(
1644 context_message, f);
1645 }
1646 }
1647 Py_XDECREF(context);
1648 Py_XDECREF(cause);
1649 }
1650 }
1651 print_exception(f, value);
1652 if (err != 0)
1653 PyErr_Clear();
Benjamin Petersone6528212008-07-15 15:32:09 +00001654}
1655
Thomas Wouters477c8d52006-05-27 19:21:47 +00001656void
1657PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001658{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001659 PyObject *seen;
1660 PyObject *f = PySys_GetObject("stderr");
1661 if (f == Py_None) {
1662 /* pass */
1663 }
1664 else if (f == NULL) {
1665 _PyObject_Dump(value);
1666 fprintf(stderr, "lost sys.stderr\n");
1667 }
1668 else {
1669 /* We choose to ignore seen being possibly NULL, and report
1670 at least the main exception (it could be a MemoryError).
1671 */
1672 seen = PySet_New(NULL);
1673 if (seen == NULL)
1674 PyErr_Clear();
1675 print_exception_recursive(f, value, seen);
1676 Py_XDECREF(seen);
1677 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001678}
1679
Guido van Rossum82598051997-03-05 00:20:32 +00001680PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001681PyRun_StringFlags(const char *str, int start, PyObject *globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001682 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001683{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001684 PyObject *ret = NULL;
1685 mod_ty mod;
1686 PyArena *arena = PyArena_New();
1687 if (arena == NULL)
1688 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
1691 if (mod != NULL)
1692 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
1693 PyArena_Free(arena);
1694 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001695}
1696
1697PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001698PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001699 PyObject *locals, int closeit, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001700{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001701 PyObject *ret;
1702 mod_ty mod;
1703 PyArena *arena = PyArena_New();
1704 if (arena == NULL)
1705 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001706
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001707 mod = PyParser_ASTFromFile(fp, filename, NULL, start, 0, 0,
1708 flags, NULL, arena);
1709 if (closeit)
1710 fclose(fp);
1711 if (mod == NULL) {
1712 PyArena_Free(arena);
1713 return NULL;
1714 }
1715 ret = run_mod(mod, filename, globals, locals, flags, arena);
1716 PyArena_Free(arena);
1717 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001718}
1719
Guido van Rossum6c193fa2007-12-05 05:14:58 +00001720static void
1721flush_io(void)
1722{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001723 PyObject *f, *r;
1724 PyObject *type, *value, *traceback;
Amaury Forgeot d'Arc9ed77352008-04-04 23:25:27 +00001725
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001726 /* Save the current exception */
1727 PyErr_Fetch(&type, &value, &traceback);
Amaury Forgeot d'Arc9ed77352008-04-04 23:25:27 +00001728
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001729 f = PySys_GetObject("stderr");
1730 if (f != NULL) {
1731 r = PyObject_CallMethod(f, "flush", "");
1732 if (r)
1733 Py_DECREF(r);
1734 else
1735 PyErr_Clear();
1736 }
1737 f = PySys_GetObject("stdout");
1738 if (f != NULL) {
1739 r = PyObject_CallMethod(f, "flush", "");
1740 if (r)
1741 Py_DECREF(r);
1742 else
1743 PyErr_Clear();
1744 }
Amaury Forgeot d'Arc9ed77352008-04-04 23:25:27 +00001745
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001746 PyErr_Restore(type, value, traceback);
Guido van Rossum6c193fa2007-12-05 05:14:58 +00001747}
1748
Guido van Rossum82598051997-03-05 00:20:32 +00001749static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001750run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001751 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001752{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001753 PyCodeObject *co;
1754 PyObject *v;
1755 co = PyAST_Compile(mod, filename, flags, arena);
1756 if (co == NULL)
1757 return NULL;
1758 v = PyEval_EvalCode(co, globals, locals);
1759 Py_DECREF(co);
1760 return v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001761}
1762
Guido van Rossum82598051997-03-05 00:20:32 +00001763static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001764run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001765 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001766{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001767 PyCodeObject *co;
1768 PyObject *v;
1769 long magic;
1770 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001771
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001772 magic = PyMarshal_ReadLongFromFile(fp);
1773 if (magic != PyImport_GetMagicNumber()) {
1774 PyErr_SetString(PyExc_RuntimeError,
1775 "Bad magic number in .pyc file");
1776 return NULL;
1777 }
1778 (void) PyMarshal_ReadLongFromFile(fp);
1779 v = PyMarshal_ReadLastObjectFromFile(fp);
1780 fclose(fp);
1781 if (v == NULL || !PyCode_Check(v)) {
1782 Py_XDECREF(v);
1783 PyErr_SetString(PyExc_RuntimeError,
1784 "Bad code object in .pyc file");
1785 return NULL;
1786 }
1787 co = (PyCodeObject *)v;
1788 v = PyEval_EvalCode(co, globals, locals);
1789 if (v && flags)
1790 flags->cf_flags |= (co->co_flags & PyCF_MASK);
1791 Py_DECREF(co);
1792 return v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001793}
1794
Guido van Rossum82598051997-03-05 00:20:32 +00001795PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001796Py_CompileStringFlags(const char *str, const char *filename, int start,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001797 PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001798{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001799 PyCodeObject *co;
1800 mod_ty mod;
1801 PyArena *arena = PyArena_New();
1802 if (arena == NULL)
1803 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001804
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001805 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
1806 if (mod == NULL) {
1807 PyArena_Free(arena);
1808 return NULL;
1809 }
1810 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
1811 PyObject *result = PyAST_mod2obj(mod);
1812 PyArena_Free(arena);
1813 return result;
1814 }
1815 co = PyAST_Compile(mod, filename, flags, arena);
1816 PyArena_Free(arena);
1817 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001818}
1819
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001820struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001821Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001822{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001823 struct symtable *st;
1824 mod_ty mod;
1825 PyCompilerFlags flags;
1826 PyArena *arena = PyArena_New();
1827 if (arena == NULL)
1828 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001829
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001830 flags.cf_flags = 0;
1831 mod = PyParser_ASTFromString(str, filename, start, &flags, arena);
1832 if (mod == NULL) {
1833 PyArena_Free(arena);
1834 return NULL;
1835 }
1836 st = PySymtable_Build(mod, filename, 0);
1837 PyArena_Free(arena);
1838 return st;
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001839}
1840
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001841/* Preferred access to parser is through AST. */
1842mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001843PyParser_ASTFromString(const char *s, const char *filename, int start,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001844 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001845{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001846 mod_ty mod;
1847 PyCompilerFlags localflags;
1848 perrdetail err;
1849 int iflags = PARSER_FLAGS(flags);
Christian Heimes4d6ec852008-03-26 22:34:47 +00001850
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001851 node *n = PyParser_ParseStringFlagsFilenameEx(s, filename,
1852 &_PyParser_Grammar, start, &err,
1853 &iflags);
1854 if (flags == NULL) {
1855 localflags.cf_flags = 0;
1856 flags = &localflags;
1857 }
1858 if (n) {
1859 flags->cf_flags |= iflags & PyCF_MASK;
1860 mod = PyAST_FromNode(n, flags, filename, arena);
1861 PyNode_Free(n);
1862 return mod;
1863 }
1864 else {
1865 err_input(&err);
1866 return NULL;
1867 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001868}
1869
1870mod_ty
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001871PyParser_ASTFromFile(FILE *fp, const char *filename, const char* enc,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001872 int start, char *ps1,
1873 char *ps2, PyCompilerFlags *flags, int *errcode,
1874 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001875{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001876 mod_ty mod;
1877 PyCompilerFlags localflags;
1878 perrdetail err;
1879 int iflags = PARSER_FLAGS(flags);
Christian Heimes4d6ec852008-03-26 22:34:47 +00001880
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001881 node *n = PyParser_ParseFileFlagsEx(fp, filename, enc,
1882 &_PyParser_Grammar,
1883 start, ps1, ps2, &err, &iflags);
1884 if (flags == NULL) {
1885 localflags.cf_flags = 0;
1886 flags = &localflags;
1887 }
1888 if (n) {
1889 flags->cf_flags |= iflags & PyCF_MASK;
1890 mod = PyAST_FromNode(n, flags, filename, arena);
1891 PyNode_Free(n);
1892 return mod;
1893 }
1894 else {
1895 err_input(&err);
1896 if (errcode)
1897 *errcode = err.error;
1898 return NULL;
1899 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001900}
1901
Guido van Rossuma110aa61994-08-29 12:50:44 +00001902/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001903
Guido van Rossuma110aa61994-08-29 12:50:44 +00001904node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001905PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001906{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001907 perrdetail err;
1908 node *n = PyParser_ParseFileFlags(fp, filename, NULL,
1909 &_PyParser_Grammar,
1910 start, NULL, NULL, &err, flags);
1911 if (n == NULL)
1912 err_input(&err);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001913
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001914 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001915}
1916
Guido van Rossuma110aa61994-08-29 12:50:44 +00001917/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001918
Guido van Rossuma110aa61994-08-29 12:50:44 +00001919node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001920PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001921{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001922 perrdetail err;
1923 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1924 start, &err, flags);
1925 if (n == NULL)
1926 err_input(&err);
1927 return n;
Tim Petersfe2127d2001-07-16 05:37:24 +00001928}
1929
1930node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001931PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001932 int start, int flags)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001933{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001934 perrdetail err;
1935 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1936 &_PyParser_Grammar, start, &err, flags);
1937 if (n == NULL)
1938 err_input(&err);
1939 return n;
Thomas Heller6b17abf2002-07-09 09:23:27 +00001940}
1941
1942node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001943PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001944{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001945 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001946}
1947
Guido van Rossum66ebd912003-04-17 16:02:26 +00001948/* May want to move a more generalized form of this to parsetok.c or
1949 even parser modules. */
1950
1951void
1952PyParser_SetError(perrdetail *err)
1953{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001954 err_input(err);
Guido van Rossum66ebd912003-04-17 16:02:26 +00001955}
1956
Guido van Rossuma110aa61994-08-29 12:50:44 +00001957/* Set the error appropriate to the given input error code (see errcode.h) */
1958
1959static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001960err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001961{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001962 PyObject *v, *w, *errtype, *errtext;
1963 PyObject *msg_obj = NULL;
Victor Stinner4c7c8c32010-10-16 13:14:10 +00001964 PyObject *filename;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001965 char *msg = NULL;
Victor Stinner4c7c8c32010-10-16 13:14:10 +00001966
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001967 errtype = PyExc_SyntaxError;
1968 switch (err->error) {
1969 case E_ERROR:
1970 return;
1971 case E_SYNTAX:
1972 errtype = PyExc_IndentationError;
1973 if (err->expected == INDENT)
1974 msg = "expected an indented block";
1975 else if (err->token == INDENT)
1976 msg = "unexpected indent";
1977 else if (err->token == DEDENT)
1978 msg = "unexpected unindent";
1979 else {
1980 errtype = PyExc_SyntaxError;
1981 msg = "invalid syntax";
1982 }
1983 break;
1984 case E_TOKEN:
1985 msg = "invalid token";
1986 break;
1987 case E_EOFS:
1988 msg = "EOF while scanning triple-quoted string literal";
1989 break;
1990 case E_EOLS:
1991 msg = "EOL while scanning string literal";
1992 break;
1993 case E_INTR:
1994 if (!PyErr_Occurred())
1995 PyErr_SetNone(PyExc_KeyboardInterrupt);
1996 goto cleanup;
1997 case E_NOMEM:
1998 PyErr_NoMemory();
1999 goto cleanup;
2000 case E_EOF:
2001 msg = "unexpected EOF while parsing";
2002 break;
2003 case E_TABSPACE:
2004 errtype = PyExc_TabError;
2005 msg = "inconsistent use of tabs and spaces in indentation";
2006 break;
2007 case E_OVERFLOW:
2008 msg = "expression too long";
2009 break;
2010 case E_DEDENT:
2011 errtype = PyExc_IndentationError;
2012 msg = "unindent does not match any outer indentation level";
2013 break;
2014 case E_TOODEEP:
2015 errtype = PyExc_IndentationError;
2016 msg = "too many levels of indentation";
2017 break;
2018 case E_DECODE: {
2019 PyObject *type, *value, *tb;
2020 PyErr_Fetch(&type, &value, &tb);
2021 msg = "unknown decode error";
2022 if (value != NULL)
2023 msg_obj = PyObject_Str(value);
2024 Py_XDECREF(type);
2025 Py_XDECREF(value);
2026 Py_XDECREF(tb);
2027 break;
2028 }
2029 case E_LINECONT:
2030 msg = "unexpected character after line continuation character";
2031 break;
Martin v. Löwis47383402007-08-15 07:32:56 +00002032
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002033 case E_IDENTIFIER:
2034 msg = "invalid character in identifier";
2035 break;
2036 default:
2037 fprintf(stderr, "error=%d\n", err->error);
2038 msg = "unknown parsing error";
2039 break;
2040 }
2041 /* err->text may not be UTF-8 in case of decoding errors.
2042 Explicitly convert to an object. */
2043 if (!err->text) {
2044 errtext = Py_None;
2045 Py_INCREF(Py_None);
2046 } else {
2047 errtext = PyUnicode_DecodeUTF8(err->text, strlen(err->text),
2048 "replace");
2049 }
Victor Stinner2f2ed1f2010-10-16 13:42:53 +00002050 if (err->filename != NULL)
2051 filename = PyUnicode_DecodeFSDefault(err->filename);
2052 else {
2053 Py_INCREF(Py_None);
2054 filename = Py_None;
2055 }
Victor Stinner4c7c8c32010-10-16 13:14:10 +00002056 if (filename != NULL)
2057 v = Py_BuildValue("(NiiN)", filename,
2058 err->lineno, err->offset, errtext);
2059 else
2060 v = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002061 if (v != NULL) {
2062 if (msg_obj)
2063 w = Py_BuildValue("(OO)", msg_obj, v);
2064 else
2065 w = Py_BuildValue("(sO)", msg, v);
2066 } else
2067 w = NULL;
2068 Py_XDECREF(v);
2069 PyErr_SetObject(errtype, w);
2070 Py_XDECREF(w);
Georg Brandl3dbca812008-07-23 16:10:53 +00002071cleanup:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002072 Py_XDECREF(msg_obj);
2073 if (err->text != NULL) {
2074 PyObject_FREE(err->text);
2075 err->text = NULL;
2076 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00002077}
2078
2079/* Print fatal error message and abort */
2080
2081void
Tim Peters7c321a82002-07-09 02:57:01 +00002082Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00002083{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002084 fprintf(stderr, "Fatal Python error: %s\n", msg);
2085 fflush(stderr); /* it helps in Windows debug build */
2086 if (PyErr_Occurred()) {
Victor Stinner55a5c782010-06-08 21:00:13 +00002087 PyErr_PrintEx(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002088 }
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002089#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002090 {
2091 size_t len = strlen(msg);
2092 WCHAR* buffer;
2093 size_t i;
Martin v. Löwis5c88d812009-01-02 20:47:48 +00002094
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002095 /* Convert the message to wchar_t. This uses a simple one-to-one
2096 conversion, assuming that the this error message actually uses ASCII
2097 only. If this ceases to be true, we will have to convert. */
2098 buffer = alloca( (len+1) * (sizeof *buffer));
2099 for( i=0; i<=len; ++i)
2100 buffer[i] = msg[i];
2101 OutputDebugStringW(L"Fatal Python error: ");
2102 OutputDebugStringW(buffer);
2103 OutputDebugStringW(L"\n");
2104 }
Guido van Rossum0ba35361998-08-13 13:33:16 +00002105#ifdef _DEBUG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002106 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00002107#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00002108#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002109 abort();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00002110}
2111
2112/* Clean up and exit */
2113
Guido van Rossuma110aa61994-08-29 12:50:44 +00002114#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00002115#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00002116#endif
2117
Collin Winter670e6922007-03-21 02:57:17 +00002118static void (*pyexitfunc)(void) = NULL;
2119/* For the atexit module. */
2120void _Py_PyAtExit(void (*func)(void))
2121{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002122 pyexitfunc = func;
Collin Winter670e6922007-03-21 02:57:17 +00002123}
2124
2125static void
2126call_py_exitfuncs(void)
2127{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002128 if (pyexitfunc == NULL)
2129 return;
Collin Winter670e6922007-03-21 02:57:17 +00002130
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002131 (*pyexitfunc)();
2132 PyErr_Clear();
Collin Winter670e6922007-03-21 02:57:17 +00002133}
2134
Antoine Pitrou011bd622009-10-20 21:52:47 +00002135/* Wait until threading._shutdown completes, provided
2136 the threading module was imported in the first place.
2137 The shutdown routine will wait until all non-daemon
2138 "threading" threads have completed. */
2139static void
2140wait_for_thread_shutdown(void)
2141{
2142#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002143 PyObject *result;
2144 PyThreadState *tstate = PyThreadState_GET();
2145 PyObject *threading = PyMapping_GetItemString(tstate->interp->modules,
2146 "threading");
2147 if (threading == NULL) {
2148 /* threading not imported */
2149 PyErr_Clear();
2150 return;
2151 }
2152 result = PyObject_CallMethod(threading, "_shutdown", "");
2153 if (result == NULL) {
2154 PyErr_WriteUnraisable(threading);
2155 }
2156 else {
2157 Py_DECREF(result);
2158 }
2159 Py_DECREF(threading);
Antoine Pitrou011bd622009-10-20 21:52:47 +00002160#endif
2161}
2162
Guido van Rossum2dcfc961998-10-01 16:01:57 +00002163#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002164static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00002165static int nexitfuncs = 0;
2166
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002167int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00002168{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002169 if (nexitfuncs >= NEXITFUNCS)
2170 return -1;
2171 exitfuncs[nexitfuncs++] = func;
2172 return 0;
Guido van Rossum1662dd51994-09-07 14:38:28 +00002173}
2174
Guido van Rossumcc283f51997-08-05 02:22:03 +00002175static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002176call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00002177{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002178 while (nexitfuncs > 0)
2179 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00002180
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002181 fflush(stdout);
2182 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00002183}
2184
2185void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002186Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00002187{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002188 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00002189
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002190 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00002191}
2192
Guido van Rossumf1dc5661993-07-05 10:31:29 +00002193static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002194initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00002195{
Guido van Rossuma110aa61994-08-29 12:50:44 +00002196#ifdef SIGPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002197 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00002198#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00002199#ifdef SIGXFZ
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002200 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00002201#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00002202#ifdef SIGXFSZ
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002203 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00002204#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002205 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00002206}
2207
Guido van Rossum7433b121997-02-14 19:45:36 +00002208
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002209/* Restore signals that the interpreter has called SIG_IGN on to SIG_DFL.
2210 *
2211 * All of the code in this function must only use async-signal-safe functions,
2212 * listed at `man 7 signal` or
2213 * http://www.opengroup.org/onlinepubs/009695399/functions/xsh_chap02_04.html.
2214 */
2215void
2216_Py_RestoreSignals(void)
2217{
2218#ifdef SIGPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002219 PyOS_setsig(SIGPIPE, SIG_DFL);
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002220#endif
2221#ifdef SIGXFZ
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002222 PyOS_setsig(SIGXFZ, SIG_DFL);
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002223#endif
2224#ifdef SIGXFSZ
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002225 PyOS_setsig(SIGXFSZ, SIG_DFL);
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002226#endif
2227}
2228
2229
Guido van Rossum7433b121997-02-14 19:45:36 +00002230/*
2231 * The file descriptor fd is considered ``interactive'' if either
2232 * a) isatty(fd) is TRUE, or
2233 * b) the -i flag was given, and the filename associated with
2234 * the descriptor is NULL or "<stdin>" or "???".
2235 */
2236int
Martin v. Löwis95292d62002-12-11 14:04:59 +00002237Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00002238{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002239 if (isatty((int)fileno(fp)))
2240 return 1;
2241 if (!Py_InteractiveFlag)
2242 return 0;
2243 return (filename == NULL) ||
2244 (strcmp(filename, "<stdin>") == 0) ||
2245 (strcmp(filename, "???") == 0);
Guido van Rossum7433b121997-02-14 19:45:36 +00002246}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00002247
2248
Tim Petersd08e3822003-04-17 15:24:21 +00002249#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00002250#if defined(WIN32) && defined(_MSC_VER)
2251
2252/* Stack checking for Microsoft C */
2253
2254#include <malloc.h>
2255#include <excpt.h>
2256
Fred Drakee8de31c2000-08-31 05:38:39 +00002257/*
2258 * Return non-zero when we run out of memory on the stack; zero otherwise.
2259 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00002260int
Fred Drake399739f2000-08-31 05:52:44 +00002261PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00002262{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002263 __try {
2264 /* alloca throws a stack overflow exception if there's
2265 not enough space left on the stack */
2266 alloca(PYOS_STACK_MARGIN * sizeof(void*));
2267 return 0;
2268 } __except (GetExceptionCode() == STATUS_STACK_OVERFLOW ?
2269 EXCEPTION_EXECUTE_HANDLER :
2270 EXCEPTION_CONTINUE_SEARCH) {
2271 int errcode = _resetstkoflw();
2272 if (errcode == 0)
2273 {
2274 Py_FatalError("Could not reset the stack!");
2275 }
2276 }
2277 return 1;
Fredrik Lundh2f15b252000-08-27 19:15:31 +00002278}
2279
2280#endif /* WIN32 && _MSC_VER */
2281
2282/* Alternate implementations can be added here... */
2283
2284#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00002285
2286
2287/* Wrappers around sigaction() or signal(). */
2288
2289PyOS_sighandler_t
2290PyOS_getsig(int sig)
2291{
2292#ifdef HAVE_SIGACTION
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002293 struct sigaction context;
2294 if (sigaction(sig, NULL, &context) == -1)
2295 return SIG_ERR;
2296 return context.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00002297#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002298 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00002299/* Special signal handling for the secure CRT in Visual Studio 2005 */
2300#if defined(_MSC_VER) && _MSC_VER >= 1400
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002301 switch (sig) {
2302 /* Only these signals are valid */
2303 case SIGINT:
2304 case SIGILL:
2305 case SIGFPE:
2306 case SIGSEGV:
2307 case SIGTERM:
2308 case SIGBREAK:
2309 case SIGABRT:
2310 break;
2311 /* Don't call signal() with other values or it will assert */
2312 default:
2313 return SIG_ERR;
2314 }
Martin v. Löwisb45b3152005-11-28 17:34:23 +00002315#endif /* _MSC_VER && _MSC_VER >= 1400 */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002316 handler = signal(sig, SIG_IGN);
2317 if (handler != SIG_ERR)
2318 signal(sig, handler);
2319 return handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00002320#endif
2321}
2322
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002323/*
2324 * All of the code in this function must only use async-signal-safe functions,
2325 * listed at `man 7 signal` or
2326 * http://www.opengroup.org/onlinepubs/009695399/functions/xsh_chap02_04.html.
2327 */
Guido van Rossum6f256182000-09-16 16:32:19 +00002328PyOS_sighandler_t
2329PyOS_setsig(int sig, PyOS_sighandler_t handler)
2330{
2331#ifdef HAVE_SIGACTION
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002332 /* Some code in Modules/signalmodule.c depends on sigaction() being
2333 * used here if HAVE_SIGACTION is defined. Fix that if this code
2334 * changes to invalidate that assumption.
2335 */
2336 struct sigaction context, ocontext;
2337 context.sa_handler = handler;
2338 sigemptyset(&context.sa_mask);
2339 context.sa_flags = 0;
2340 if (sigaction(sig, &context, &ocontext) == -1)
2341 return SIG_ERR;
2342 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00002343#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002344 PyOS_sighandler_t oldhandler;
2345 oldhandler = signal(sig, handler);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00002346#ifdef HAVE_SIGINTERRUPT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002347 siginterrupt(sig, 1);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00002348#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002349 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00002350#endif
2351}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002352
2353/* Deprecated C API functions still provided for binary compatiblity */
2354
2355#undef PyParser_SimpleParseFile
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002356PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002357PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
2358{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002359 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002360}
2361
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002362#undef PyParser_SimpleParseString
2363PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002364PyParser_SimpleParseString(const char *str, int start)
2365{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002366 return PyParser_SimpleParseStringFlags(str, start, 0);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002367}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002368
2369#undef PyRun_AnyFile
2370PyAPI_FUNC(int)
2371PyRun_AnyFile(FILE *fp, const char *name)
2372{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002373 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002374}
2375
2376#undef PyRun_AnyFileEx
2377PyAPI_FUNC(int)
2378PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
2379{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002380 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002381}
2382
2383#undef PyRun_AnyFileFlags
2384PyAPI_FUNC(int)
2385PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
2386{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002387 return PyRun_AnyFileExFlags(fp, name, 0, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002388}
2389
2390#undef PyRun_File
2391PyAPI_FUNC(PyObject *)
2392PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
2393{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002394 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002395}
2396
2397#undef PyRun_FileEx
2398PyAPI_FUNC(PyObject *)
2399PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
2400{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002401 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002402}
2403
2404#undef PyRun_FileFlags
2405PyAPI_FUNC(PyObject *)
2406PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002407 PyCompilerFlags *flags)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002408{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002409 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002410}
2411
2412#undef PyRun_SimpleFile
2413PyAPI_FUNC(int)
2414PyRun_SimpleFile(FILE *f, const char *p)
2415{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002416 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002417}
2418
2419#undef PyRun_SimpleFileEx
2420PyAPI_FUNC(int)
2421PyRun_SimpleFileEx(FILE *f, const char *p, int c)
2422{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002423 return PyRun_SimpleFileExFlags(f, p, c, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002424}
2425
2426
2427#undef PyRun_String
2428PyAPI_FUNC(PyObject *)
2429PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
2430{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002431 return PyRun_StringFlags(str, s, g, l, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002432}
2433
2434#undef PyRun_SimpleString
2435PyAPI_FUNC(int)
2436PyRun_SimpleString(const char *s)
2437{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002438 return PyRun_SimpleStringFlags(s, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002439}
2440
2441#undef Py_CompileString
2442PyAPI_FUNC(PyObject *)
2443Py_CompileString(const char *str, const char *p, int s)
2444{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002445 return Py_CompileStringFlags(str, p, s, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002446}
2447
2448#undef PyRun_InteractiveOne
2449PyAPI_FUNC(int)
2450PyRun_InteractiveOne(FILE *f, const char *p)
2451{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002452 return PyRun_InteractiveOneFlags(f, p, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002453}
2454
2455#undef PyRun_InteractiveLoop
2456PyAPI_FUNC(int)
2457PyRun_InteractiveLoop(FILE *f, const char *p)
2458{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002459 return PyRun_InteractiveLoopFlags(f, p, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002460}
2461
2462#ifdef __cplusplus
2463}
2464#endif