blob: 1a7e01ee4de56e714931c66c5519389027c59c17 [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"
Kristján Valur Jónsson67387fb2007-04-25 00:17:39 +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"
Antoine Pitrou9aece752009-10-27 12:48:52 +000020#include "abstract.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000021
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000022#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000023#include <signal.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000024#endif
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000025
Georg Brandl734373c2009-01-03 21:55:17 +000026#ifdef MS_WINDOWS
27#include "malloc.h" /* for alloca */
28#endif
29
Martin v. Löwis73d538b2003-03-05 15:13:47 +000030#ifdef HAVE_LANGINFO_H
31#include <locale.h>
32#include <langinfo.h>
33#endif
34
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000035#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000036#undef BYTE
37#include "windows.h"
38#endif
39
Neal Norwitz4281cef2006-03-04 19:58:13 +000040#ifndef Py_REF_DEBUG
Tim Peters62e97f02006-03-28 21:44:32 +000041#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000042#else /* Py_REF_DEBUG */
Antoine Pitrouc7c96a92010-05-09 15:15:40 +000043#define PRINT_TOTAL_REFS() fprintf(stderr, \
44 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
45 _Py_GetRefTotal())
Neal Norwitz4281cef2006-03-04 19:58:13 +000046#endif
47
Anthony Baxterac6bd462006-04-13 02:06:09 +000048#ifdef __cplusplus
49extern "C" {
50#endif
51
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000052extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000053
Guido van Rossum82598051997-03-05 00:20:32 +000054extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000055
Guido van Rossumb73cc041993-11-01 16:28:59 +000056/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000057static void initmain(void);
58static void initsite(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000059static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +000060 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000061static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +000062 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000063static void err_input(perrdetail *);
64static void initsigs(void);
Antoine Pitrou9aece752009-10-27 12:48:52 +000065static void wait_for_thread_shutdown(void);
Tim Petersdbd9ba62000-07-09 03:09:57 +000066static void call_sys_exitfunc(void);
67static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000068extern void _PyUnicode_Init(void);
69extern void _PyUnicode_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000070
Mark Hammond8d98d2c2003-04-19 15:41:53 +000071#ifdef WITH_THREAD
72extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
73extern void _PyGILState_Fini(void);
74#endif /* WITH_THREAD */
75
Guido van Rossum82598051997-03-05 00:20:32 +000076int Py_DebugFlag; /* Needed by parser.c */
77int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000078int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Georg Brandl49aafc92007-03-07 00:34:46 +000079int Py_InspectFlag; /* Needed to determine whether to exit at SystemError */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000080int Py_NoSiteFlag; /* Suppress 'import site' */
Christian Heimes1a6387e2008-03-26 12:49:49 +000081int Py_BytesWarningFlag; /* Warn on str(bytes) and str(buffer) */
Georg Brandl2da0fce2008-01-07 17:09:35 +000082int Py_DontWriteBytecodeFlag; /* Suppress writing bytecode files (*.py[co]) */
Barry Warsaw3ce09642000-05-02 19:18:59 +000083int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000084int Py_FrozenFlag; /* Needed by getpath.c */
Guido van Rossumb16d1972000-05-01 17:55:15 +000085int Py_UnicodeFlag = 0; /* Needed by compile.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000086int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Tim Peters3caca232001-12-06 06:23:26 +000087/* _XXX Py_QnewFlag should go away in 2.3. It's true iff -Qnew is passed,
88 on the command line, and is used in 2.2 by ceval.c to make all "/" divisions
89 true divisions (which they will be in 2.3). */
90int _Py_QnewFlag = 0;
Christian Heimesaf748c32008-05-06 22:41:46 +000091int Py_NoUserSiteDirectory = 0; /* for -s and site.py */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000092
Brett Cannone9746892008-04-12 23:44:07 +000093/* PyModule_GetWarningsModule is no longer necessary as of 2.6
94since _warnings is builtin. This API should not be used. */
95PyObject *
96PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000097{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +000098 return PyImport_ImportModule("warnings");
Mark Hammondedd07732003-07-15 23:03:55 +000099}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000100
Guido van Rossum25ce5661997-08-02 03:10:38 +0000101static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000102
Thomas Wouters7e474022000-07-16 12:04:32 +0000103/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000104
105int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000106Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000107{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000108 return initialized;
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000109}
110
Guido van Rossum25ce5661997-08-02 03:10:38 +0000111/* Global initializations. Can be undone by Py_Finalize(). Don't
112 call this twice without an intervening Py_Finalize() call. When
113 initializations fail, a fatal error is issued and the function does
114 not return. On return, the first thread and interpreter state have
115 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000116
Guido van Rossum25ce5661997-08-02 03:10:38 +0000117 Locking: you must hold the interpreter lock while calling this.
118 (If the lock has not yet been initialized, that's equivalent to
119 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000120
Guido van Rossum25ce5661997-08-02 03:10:38 +0000121*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000122
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000123static int
124add_flag(int flag, const char *envs)
125{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000126 int env = atoi(envs);
127 if (flag < env)
128 flag = env;
129 if (flag < 1)
130 flag = 1;
131 return flag;
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000132}
133
Guido van Rossuma027efa1997-05-05 20:56:21 +0000134void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000135Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000136{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000137 PyInterpreterState *interp;
138 PyThreadState *tstate;
139 PyObject *bimod, *sysmod;
140 char *p;
141 char *icodeset = NULL; /* On Windows, input codeset may theoretically
142 differ from output codeset. */
143 char *codeset = NULL;
144 char *errors = NULL;
145 int free_codeset = 0;
146 int overridden = 0;
147 PyObject *sys_stream, *sys_isatty;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000148#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000149 char *saved_locale, *loc_codeset;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000150#endif
Martin v. Löwis99815892008-06-01 07:20:46 +0000151#ifdef MS_WINDOWS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000152 char ibuf[128];
153 char buf[128];
Martin v. Löwis99815892008-06-01 07:20:46 +0000154#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000155 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000156
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000157 if (initialized)
158 return;
159 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000160
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000161 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
162 Py_DebugFlag = add_flag(Py_DebugFlag, p);
163 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
164 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
165 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
166 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
167 if ((p = Py_GETENV("PYTHONDONTWRITEBYTECODE")) && *p != '\0')
168 Py_DontWriteBytecodeFlag = add_flag(Py_DontWriteBytecodeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000169
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000170 interp = PyInterpreterState_New();
171 if (interp == NULL)
172 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000173
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000174 tstate = PyThreadState_New(interp);
175 if (tstate == NULL)
176 Py_FatalError("Py_Initialize: can't make first thread");
177 (void) PyThreadState_Swap(tstate);
Guido van Rossuma027efa1997-05-05 20:56:21 +0000178
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000179 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000180
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000181 if (!_PyFrame_Init())
182 Py_FatalError("Py_Initialize: can't init frames");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000183
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000184 if (!_PyInt_Init())
185 Py_FatalError("Py_Initialize: can't init ints");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000186
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000187 if (!PyByteArray_Init())
188 Py_FatalError("Py_Initialize: can't init bytearray");
Christian Heimes1a6387e2008-03-26 12:49:49 +0000189
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000190 _PyFloat_Init();
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000191
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000192 interp->modules = PyDict_New();
193 if (interp->modules == NULL)
194 Py_FatalError("Py_Initialize: can't make modules dictionary");
195 interp->modules_reloading = PyDict_New();
196 if (interp->modules_reloading == NULL)
197 Py_FatalError("Py_Initialize: can't make modules_reloading dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000198
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000199#ifdef Py_USING_UNICODE
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000200 /* Init Unicode implementation; relies on the codec registry */
201 _PyUnicode_Init();
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000202#endif
Guido van Rossumc94044c2000-03-10 23:03:54 +0000203
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000204 bimod = _PyBuiltin_Init();
205 if (bimod == NULL)
206 Py_FatalError("Py_Initialize: can't initialize __builtin__");
207 interp->builtins = PyModule_GetDict(bimod);
208 if (interp->builtins == NULL)
209 Py_FatalError("Py_Initialize: can't initialize builtins dict");
210 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000211
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000212 sysmod = _PySys_Init();
213 if (sysmod == NULL)
214 Py_FatalError("Py_Initialize: can't initialize sys");
215 interp->sysdict = PyModule_GetDict(sysmod);
216 if (interp->sysdict == NULL)
217 Py_FatalError("Py_Initialize: can't initialize sys dict");
218 Py_INCREF(interp->sysdict);
219 _PyImport_FixupExtension("sys", "sys");
220 PySys_SetPath(Py_GetPath());
221 PyDict_SetItemString(interp->sysdict, "modules",
222 interp->modules);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000223
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000224 _PyImport_Init();
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000225
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000226 /* initialize builtin exceptions */
227 _PyExc_Init();
228 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000229
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000230 /* phase 2 of builtins */
231 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000232
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000233 _PyImportHooks_Init();
Just van Rossum52e14d62002-12-30 22:08:05 +0000234
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000235 if (install_sigs)
236 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000237
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000238 /* Initialize warnings. */
239 _PyWarnings_Init();
240 if (PySys_HasWarnOptions()) {
241 PyObject *warnings_module = PyImport_ImportModule("warnings");
242 if (!warnings_module)
243 PyErr_Clear();
244 Py_XDECREF(warnings_module);
245 }
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000246
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000247 initmain(); /* Module __main__ */
248
249 /* auto-thread-state API, if available */
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000250#ifdef WITH_THREAD
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000251 _PyGILState_Init(interp, tstate);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000252#endif /* WITH_THREAD */
253
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000254 if (!Py_NoSiteFlag)
255 initsite(); /* Module site */
Victor Stinnerea164292010-03-21 14:02:32 +0000256
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000257 if ((p = Py_GETENV("PYTHONIOENCODING")) && *p != '\0') {
258 p = icodeset = codeset = strdup(p);
259 free_codeset = 1;
260 errors = strchr(p, ':');
261 if (errors) {
262 *errors = '\0';
263 errors++;
264 }
265 overridden = 1;
266 }
Martin v. Löwis99815892008-06-01 07:20:46 +0000267
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000268#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000269 /* On Unix, set the file system encoding according to the
270 user's preference, if the CODESET names a well-known
271 Python codec, and Py_FileSystemDefaultEncoding isn't
272 initialized by other means. Also set the encoding of
273 stdin and stdout if these are terminals, unless overridden. */
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000274
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000275 if (!overridden || !Py_FileSystemDefaultEncoding) {
276 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
277 setlocale(LC_CTYPE, "");
278 loc_codeset = nl_langinfo(CODESET);
279 if (loc_codeset && *loc_codeset) {
280 PyObject *enc = PyCodec_Encoder(loc_codeset);
281 if (enc) {
282 loc_codeset = strdup(loc_codeset);
283 Py_DECREF(enc);
284 } else {
285 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
286 PyErr_Clear();
287 loc_codeset = NULL;
288 } else {
289 PyErr_Print();
290 exit(1);
291 }
292 }
293 } else
294 loc_codeset = NULL;
295 setlocale(LC_CTYPE, saved_locale);
296 free(saved_locale);
Martin v. Löwis99815892008-06-01 07:20:46 +0000297
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000298 if (!overridden) {
299 codeset = icodeset = loc_codeset;
300 free_codeset = 1;
301 }
Martin v. Löwis99815892008-06-01 07:20:46 +0000302
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000303 /* Initialize Py_FileSystemDefaultEncoding from
304 locale even if PYTHONIOENCODING is set. */
305 if (!Py_FileSystemDefaultEncoding) {
306 Py_FileSystemDefaultEncoding = loc_codeset;
307 if (!overridden)
308 free_codeset = 0;
309 }
310 }
Martin v. Löwis99815892008-06-01 07:20:46 +0000311#endif
312
313#ifdef MS_WINDOWS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000314 if (!overridden) {
315 icodeset = ibuf;
316 codeset = buf;
317 sprintf(ibuf, "cp%d", GetConsoleCP());
318 sprintf(buf, "cp%d", GetConsoleOutputCP());
319 }
Martin v. Löwis99815892008-06-01 07:20:46 +0000320#endif
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000321
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000322 if (codeset) {
323 sys_stream = PySys_GetObject("stdin");
324 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
325 if (!sys_isatty)
326 PyErr_Clear();
327 if ((overridden ||
328 (sys_isatty && PyObject_IsTrue(sys_isatty))) &&
329 PyFile_Check(sys_stream)) {
330 if (!PyFile_SetEncodingAndErrors(sys_stream, icodeset, errors))
331 Py_FatalError("Cannot set codeset of stdin");
332 }
333 Py_XDECREF(sys_isatty);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000334
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000335 sys_stream = PySys_GetObject("stdout");
336 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
337 if (!sys_isatty)
338 PyErr_Clear();
339 if ((overridden ||
340 (sys_isatty && PyObject_IsTrue(sys_isatty))) &&
341 PyFile_Check(sys_stream)) {
342 if (!PyFile_SetEncodingAndErrors(sys_stream, codeset, errors))
343 Py_FatalError("Cannot set codeset of stdout");
344 }
345 Py_XDECREF(sys_isatty);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000346
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000347 sys_stream = PySys_GetObject("stderr");
348 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
349 if (!sys_isatty)
350 PyErr_Clear();
351 if((overridden ||
352 (sys_isatty && PyObject_IsTrue(sys_isatty))) &&
353 PyFile_Check(sys_stream)) {
354 if (!PyFile_SetEncodingAndErrors(sys_stream, codeset, errors))
355 Py_FatalError("Cannot set codeset of stderr");
356 }
357 Py_XDECREF(sys_isatty);
Martin v. Löwisea62d252006-04-03 10:56:49 +0000358
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000359 if (free_codeset)
360 free(codeset);
361 }
Guido van Rossum25ce5661997-08-02 03:10:38 +0000362}
363
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000364void
365Py_Initialize(void)
366{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000367 Py_InitializeEx(1);
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000368}
369
370
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000371#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000372extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000373#endif
374
Guido van Rossum25ce5661997-08-02 03:10:38 +0000375/* Undo the effect of Py_Initialize().
376
377 Beware: if multiple interpreter and/or thread states exist, these
378 are not wiped out; only the current thread and interpreter state
379 are deleted. But since everything else is deleted, those other
380 interpreter and thread states should no longer be used.
381
382 (XXX We should do better, e.g. wipe out all interpreters and
383 threads.)
384
385 Locking: as above.
386
387*/
388
389void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000390Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000391{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000392 PyInterpreterState *interp;
393 PyThreadState *tstate;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000394
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000395 if (!initialized)
396 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000397
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000398 wait_for_thread_shutdown();
Antoine Pitrou9aece752009-10-27 12:48:52 +0000399
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000400 /* The interpreter is still entirely intact at this point, and the
401 * exit funcs may be relying on that. In particular, if some thread
402 * or exit func is still waiting to do an import, the import machinery
403 * expects Py_IsInitialized() to return true. So don't say the
404 * interpreter is uninitialized until after the exit funcs have run.
405 * Note that Threading.py uses an exit func to do a join on all the
406 * threads created thru it, so this also protects pending imports in
407 * the threads created via Threading.
408 */
409 call_sys_exitfunc();
410 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000411
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000412 /* Get current thread state and interpreter pointer */
413 tstate = PyThreadState_GET();
414 interp = tstate->interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000415
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000416 /* Disable signal handling */
417 PyOS_FiniInterrupts();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000418
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000419 /* Clear type lookup cache */
420 PyType_ClearCache();
Christian Heimes908caac2008-01-27 23:34:59 +0000421
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000422 /* Collect garbage. This may call finalizers; it's nice to call these
423 * before all modules are destroyed.
424 * XXX If a __del__ or weakref callback is triggered here, and tries to
425 * XXX import a module, bad things can happen, because Python no
426 * XXX longer believes it's initialized.
427 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
428 * XXX is easy to provoke that way. I've also seen, e.g.,
429 * XXX Exception exceptions.ImportError: 'No module named sha'
430 * XXX in <function callback at 0x008F5718> ignored
431 * XXX but I'm unclear on exactly how that one happens. In any case,
432 * XXX I haven't seen a real-life report of either of these.
433 */
434 PyGC_Collect();
Martin v. Löwis45294a92006-04-18 06:24:08 +0000435#ifdef COUNT_ALLOCS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000436 /* With COUNT_ALLOCS, it helps to run GC multiple times:
437 each collection might release some types from the type
438 list, so they become garbage. */
439 while (PyGC_Collect() > 0)
440 /* nothing */;
Martin v. Löwis45294a92006-04-18 06:24:08 +0000441#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000442
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000443 /* Destroy all modules */
444 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000445
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000446 /* Collect final garbage. This disposes of cycles created by
447 * new-style class definitions, for example.
448 * XXX This is disabled because it caused too many problems. If
449 * XXX a __del__ or weakref callback triggers here, Python code has
450 * XXX a hard time running, because even the sys module has been
451 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
452 * XXX One symptom is a sequence of information-free messages
453 * XXX coming from threads (if a __del__ or callback is invoked,
454 * XXX other threads can execute too, and any exception they encounter
455 * XXX triggers a comedy of errors as subsystem after subsystem
456 * XXX fails to find what it *expects* to find in sys to help report
457 * XXX the exception and consequent unexpected failures). I've also
458 * XXX seen segfaults then, after adding print statements to the
459 * XXX Python code getting called.
460 */
Tim Peters1d7323e2003-12-01 21:35:27 +0000461#if 0
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000462 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000463#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000464
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000465 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
466 _PyImport_Fini();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000467
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000468 /* Debugging stuff */
Guido van Rossum1707aad1997-12-08 23:43:45 +0000469#ifdef COUNT_ALLOCS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000470 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000471#endif
472
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000473 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000474
Tim Peters9cf25ce2003-04-17 15:21:01 +0000475#ifdef Py_TRACE_REFS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000476 /* Display all objects still alive -- this can invoke arbitrary
477 * __repr__ overrides, so requires a mostly-intact interpreter.
478 * Alas, a lot of stuff may still be alive now that will be cleaned
479 * up later.
480 */
481 if (Py_GETENV("PYTHONDUMPREFS"))
482 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000483#endif /* Py_TRACE_REFS */
484
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000485 /* Clear interpreter state */
486 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000487
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000488 /* Now we decref the exception classes. After this point nothing
489 can raise an exception. That's okay, because each Fini() method
490 below has been checked to make sure no exceptions are ever
491 raised.
492 */
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000493
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000494 _PyExc_Fini();
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000495
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000496 /* Cleanup auto-thread-state */
Amaury Forgeot d'Arc025c3472007-11-29 23:35:25 +0000497#ifdef WITH_THREAD
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000498 _PyGILState_Fini();
Amaury Forgeot d'Arc025c3472007-11-29 23:35:25 +0000499#endif /* WITH_THREAD */
500
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000501 /* Delete current thread */
502 PyThreadState_Swap(NULL);
503 PyInterpreterState_Delete(interp);
Barry Warsawf242aa02000-05-25 23:09:49 +0000504
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000505 /* Sundry finalizers */
506 PyMethod_Fini();
507 PyFrame_Fini();
508 PyCFunction_Fini();
509 PyTuple_Fini();
510 PyList_Fini();
511 PySet_Fini();
512 PyString_Fini();
513 PyByteArray_Fini();
514 PyInt_Fini();
515 PyFloat_Fini();
516 PyDict_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000517
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000518#ifdef Py_USING_UNICODE
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000519 /* Cleanup Unicode implementation */
520 _PyUnicode_Fini();
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000521#endif
522
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000523 /* XXX Still allocated:
524 - various static ad-hoc pointers to interned strings
525 - int and float free list blocks
526 - whatever various modules and libraries allocate
527 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000528
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000529 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000530
Tim Peters269b2a62003-04-17 19:52:29 +0000531#ifdef Py_TRACE_REFS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000532 /* Display addresses (& refcnts) of all objects still alive.
533 * An address can be used to find the repr of the object, printed
534 * above by _Py_PrintReferences.
535 */
536 if (Py_GETENV("PYTHONDUMPREFS"))
537 _Py_PrintReferenceAddresses(stderr);
Tim Peters269b2a62003-04-17 19:52:29 +0000538#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000539#ifdef PYMALLOC_DEBUG
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000540 if (Py_GETENV("PYTHONMALLOCSTATS"))
541 _PyObject_DebugMallocStats();
Tim Peters0e871182002-04-13 08:29:14 +0000542#endif
543
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000544 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000545}
546
547/* Create and initialize a new interpreter and thread, and return the
548 new thread. This requires that Py_Initialize() has been called
549 first.
550
551 Unsuccessful initialization yields a NULL pointer. Note that *no*
552 exception information is available even in this case -- the
553 exception information is held in the thread, and there is no
554 thread.
555
556 Locking: as above.
557
558*/
559
560PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000561Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000562{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000563 PyInterpreterState *interp;
564 PyThreadState *tstate, *save_tstate;
565 PyObject *bimod, *sysmod;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000566
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000567 if (!initialized)
568 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000569
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000570 interp = PyInterpreterState_New();
571 if (interp == NULL)
572 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000573
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000574 tstate = PyThreadState_New(interp);
575 if (tstate == NULL) {
576 PyInterpreterState_Delete(interp);
577 return NULL;
578 }
Guido van Rossum25ce5661997-08-02 03:10:38 +0000579
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000580 save_tstate = PyThreadState_Swap(tstate);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000581
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000582 /* XXX The following is lax in error checking */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000583
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000584 interp->modules = PyDict_New();
585 interp->modules_reloading = PyDict_New();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000586
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000587 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
588 if (bimod != NULL) {
589 interp->builtins = PyModule_GetDict(bimod);
590 if (interp->builtins == NULL)
591 goto handle_error;
592 Py_INCREF(interp->builtins);
593 }
594 sysmod = _PyImport_FindExtension("sys", "sys");
595 if (bimod != NULL && sysmod != NULL) {
596 interp->sysdict = PyModule_GetDict(sysmod);
597 if (interp->sysdict == NULL)
598 goto handle_error;
599 Py_INCREF(interp->sysdict);
600 PySys_SetPath(Py_GetPath());
601 PyDict_SetItemString(interp->sysdict, "modules",
602 interp->modules);
603 _PyImportHooks_Init();
604 initmain();
605 if (!Py_NoSiteFlag)
606 initsite();
607 }
Guido van Rossum25ce5661997-08-02 03:10:38 +0000608
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000609 if (!PyErr_Occurred())
610 return tstate;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000611
Neal Norwitz0c6ae5b2006-08-21 20:16:24 +0000612handle_error:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000613 /* Oops, it didn't work. Undo it all. */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000614
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000615 PyErr_Print();
616 PyThreadState_Clear(tstate);
617 PyThreadState_Swap(save_tstate);
618 PyThreadState_Delete(tstate);
619 PyInterpreterState_Delete(interp);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000620
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000621 return NULL;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000622}
623
624/* Delete an interpreter and its last thread. This requires that the
625 given thread state is current, that the thread has no remaining
626 frames, and that it is its interpreter's only remaining thread.
627 It is a fatal error to violate these constraints.
628
629 (Py_Finalize() doesn't have these constraints -- it zaps
630 everything, regardless.)
631
632 Locking: as above.
633
634*/
635
636void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000637Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000638{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000639 PyInterpreterState *interp = tstate->interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000640
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000641 if (tstate != PyThreadState_GET())
642 Py_FatalError("Py_EndInterpreter: thread is not current");
643 if (tstate->frame != NULL)
644 Py_FatalError("Py_EndInterpreter: thread still has a frame");
645 if (tstate != interp->tstate_head || tstate->next != NULL)
646 Py_FatalError("Py_EndInterpreter: not the last thread");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000647
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000648 PyImport_Cleanup();
649 PyInterpreterState_Clear(interp);
650 PyThreadState_Swap(NULL);
651 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000652}
653
654static char *progname = "python";
655
656void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000657Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000658{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000659 if (pn && *pn)
660 progname = pn;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000661}
662
663char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000664Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000665{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000666 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000667}
668
Guido van Rossuma61691e1998-02-06 22:27:24 +0000669static char *default_home = NULL;
670
671void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000672Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000673{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000674 default_home = home;
Guido van Rossuma61691e1998-02-06 22:27:24 +0000675}
676
677char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000678Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000679{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000680 char *home = default_home;
681 if (home == NULL && !Py_IgnoreEnvironmentFlag)
682 home = Py_GETENV("PYTHONHOME");
683 return home;
Guido van Rossuma61691e1998-02-06 22:27:24 +0000684}
685
Guido van Rossum6135a871995-01-09 17:53:26 +0000686/* Create __main__ module */
687
688static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000689initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000690{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000691 PyObject *m, *d;
692 m = PyImport_AddModule("__main__");
693 if (m == NULL)
694 Py_FatalError("can't create __main__ module");
695 d = PyModule_GetDict(m);
696 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
697 PyObject *bimod = PyImport_ImportModule("__builtin__");
698 if (bimod == NULL ||
699 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
700 Py_FatalError("can't add __builtins__ to __main__");
701 Py_DECREF(bimod);
702 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000703}
704
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000705/* Import the site module (not into __main__ though) */
706
707static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000708initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000709{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000710 PyObject *m, *f;
711 m = PyImport_ImportModule("site");
712 if (m == NULL) {
713 f = PySys_GetObject("stderr");
714 if (Py_VerboseFlag) {
715 PyFile_WriteString(
716 "'import site' failed; traceback:\n", f);
717 PyErr_Print();
718 }
719 else {
720 PyFile_WriteString(
721 "'import site' failed; use -v for traceback\n", f);
722 PyErr_Clear();
723 }
724 }
725 else {
726 Py_DECREF(m);
727 }
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000728}
729
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000730/* Parse input from a file and execute it */
731
732int
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000733PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000734 PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000735{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000736 if (filename == NULL)
737 filename = "???";
738 if (Py_FdIsInteractive(fp, filename)) {
739 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
740 if (closeit)
741 fclose(fp);
742 return err;
743 }
744 else
745 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000746}
747
748int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000749PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000750{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000751 PyObject *v;
752 int ret;
753 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000754
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000755 if (flags == NULL) {
756 flags = &local_flags;
757 local_flags.cf_flags = 0;
758 }
759 v = PySys_GetObject("ps1");
760 if (v == NULL) {
761 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
762 Py_XDECREF(v);
763 }
764 v = PySys_GetObject("ps2");
765 if (v == NULL) {
766 PySys_SetObject("ps2", v = PyString_FromString("... "));
767 Py_XDECREF(v);
768 }
769 for (;;) {
770 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
771 PRINT_TOTAL_REFS();
772 if (ret == E_EOF)
773 return 0;
774 /*
775 if (ret == E_NOMEM)
776 return -1;
777 */
778 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000779}
780
Eric Smith7c478942008-03-18 23:45:49 +0000781#if 0
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000782/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000783#define PARSER_FLAGS(flags) \
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000784 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
785 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
Eric Smith7c478942008-03-18 23:45:49 +0000786#endif
787#if 1
Neal Norwitzca460d92006-09-06 06:28:06 +0000788/* Keep an example of flags with future keyword support. */
789#define PARSER_FLAGS(flags) \
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000790 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
791 PyPARSE_DONT_IMPLY_DEDENT : 0) \
792 | (((flags)->cf_flags & CO_FUTURE_PRINT_FUNCTION) ? \
793 PyPARSE_PRINT_IS_FUNCTION : 0) \
794 | (((flags)->cf_flags & CO_FUTURE_UNICODE_LITERALS) ? \
795 PyPARSE_UNICODE_LITERALS : 0) \
796 ) : 0)
Neal Norwitzca460d92006-09-06 06:28:06 +0000797#endif
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000798
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000799int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000800PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000801{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000802 PyObject *m, *d, *v, *w;
803 mod_ty mod;
804 PyArena *arena;
805 char *ps1 = "", *ps2 = "";
806 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000807
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000808 v = PySys_GetObject("ps1");
809 if (v != NULL) {
810 v = PyObject_Str(v);
811 if (v == NULL)
812 PyErr_Clear();
813 else if (PyString_Check(v))
814 ps1 = PyString_AsString(v);
815 }
816 w = PySys_GetObject("ps2");
817 if (w != NULL) {
818 w = PyObject_Str(w);
819 if (w == NULL)
820 PyErr_Clear();
821 else if (PyString_Check(w))
822 ps2 = PyString_AsString(w);
823 }
824 arena = PyArena_New();
825 if (arena == NULL) {
826 Py_XDECREF(v);
827 Py_XDECREF(w);
828 return -1;
829 }
830 mod = PyParser_ASTFromFile(fp, filename,
831 Py_single_input, ps1, ps2,
832 flags, &errcode, arena);
833 Py_XDECREF(v);
834 Py_XDECREF(w);
835 if (mod == NULL) {
836 PyArena_Free(arena);
837 if (errcode == E_EOF) {
838 PyErr_Clear();
839 return E_EOF;
840 }
841 PyErr_Print();
842 return -1;
843 }
844 m = PyImport_AddModule("__main__");
845 if (m == NULL) {
846 PyArena_Free(arena);
847 return -1;
848 }
849 d = PyModule_GetDict(m);
850 v = run_mod(mod, filename, d, d, flags, arena);
851 PyArena_Free(arena);
852 if (v == NULL) {
853 PyErr_Print();
854 return -1;
855 }
856 Py_DECREF(v);
857 if (Py_FlushLine())
858 PyErr_Clear();
859 return 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000860}
861
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000862/* Check whether a file maybe a pyc file: Look at the extension,
863 the file type, and, if we may close it, at the first few bytes. */
864
865static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000866maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000867{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000868 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
869 return 1;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000870
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000871 /* Only look into the file if we are allowed to close it, since
872 it then should also be seekable. */
873 if (closeit) {
874 /* Read only two bytes of the magic. If the file was opened in
875 text mode, the bytes 3 and 4 of the magic (\r\n) might not
876 be read as they are on disk. */
877 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
878 unsigned char buf[2];
879 /* Mess: In case of -x, the stream is NOT at its start now,
880 and ungetc() was used to push back the first newline,
881 which makes the current stream position formally undefined,
882 and a x-platform nightmare.
883 Unfortunately, we have no direct way to know whether -x
884 was specified. So we use a terrible hack: if the current
885 stream position is not 0, we assume -x was specified, and
886 give up. Bug 132850 on SourceForge spells out the
887 hopelessness of trying anything else (fseek and ftell
888 don't work predictably x-platform for text-mode files).
889 */
890 int ispyc = 0;
891 if (ftell(fp) == 0) {
892 if (fread(buf, 1, 2, fp) == 2 &&
893 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
894 ispyc = 1;
895 rewind(fp);
896 }
897 return ispyc;
898 }
899 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000900}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000901
Guido van Rossum0df002c2000-08-27 19:21:52 +0000902int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000903PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000904 PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000905{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000906 PyObject *m, *d, *v;
907 const char *ext;
908 int set_file_name = 0, ret, len;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000909
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000910 m = PyImport_AddModule("__main__");
911 if (m == NULL)
912 return -1;
913 d = PyModule_GetDict(m);
914 if (PyDict_GetItemString(d, "__file__") == NULL) {
915 PyObject *f = PyString_FromString(filename);
916 if (f == NULL)
917 return -1;
918 if (PyDict_SetItemString(d, "__file__", f) < 0) {
919 Py_DECREF(f);
920 return -1;
921 }
922 set_file_name = 1;
923 Py_DECREF(f);
924 }
925 len = strlen(filename);
926 ext = filename + len - (len > 4 ? 4 : 0);
927 if (maybe_pyc_file(fp, filename, ext, closeit)) {
928 /* Try to run a pyc file. First, re-open in binary */
929 if (closeit)
930 fclose(fp);
931 if ((fp = fopen(filename, "rb")) == NULL) {
932 fprintf(stderr, "python: Can't reopen .pyc file\n");
933 ret = -1;
934 goto done;
935 }
936 /* Turn on optimization if a .pyo file is given */
937 if (strcmp(ext, ".pyo") == 0)
938 Py_OptimizeFlag = 1;
939 v = run_pyc_file(fp, filename, d, d, flags);
940 } else {
941 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
942 closeit, flags);
943 }
944 if (v == NULL) {
945 PyErr_Print();
946 ret = -1;
947 goto done;
948 }
949 Py_DECREF(v);
950 if (Py_FlushLine())
951 PyErr_Clear();
952 ret = 0;
Georg Brandlaa2321b2007-03-07 00:40:28 +0000953 done:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000954 if (set_file_name && PyDict_DelItemString(d, "__file__"))
955 PyErr_Clear();
956 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000957}
958
959int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000960PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000961{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000962 PyObject *m, *d, *v;
963 m = PyImport_AddModule("__main__");
964 if (m == NULL)
965 return -1;
966 d = PyModule_GetDict(m);
967 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
968 if (v == NULL) {
969 PyErr_Print();
970 return -1;
971 }
972 Py_DECREF(v);
973 if (Py_FlushLine())
974 PyErr_Clear();
975 return 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000976}
977
Barry Warsaw035574d1997-08-29 22:07:17 +0000978static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000979parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000980 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000981{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000982 long hold;
983 PyObject *v;
Barry Warsaw035574d1997-08-29 22:07:17 +0000984
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000985 /* old style errors */
986 if (PyTuple_Check(err))
987 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
988 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000989
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000990 /* new style errors. `err' is an instance */
Barry Warsaw035574d1997-08-29 22:07:17 +0000991
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000992 if (! (v = PyObject_GetAttrString(err, "msg")))
993 goto finally;
994 *message = v;
Barry Warsaw035574d1997-08-29 22:07:17 +0000995
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000996 if (!(v = PyObject_GetAttrString(err, "filename")))
997 goto finally;
998 if (v == Py_None)
999 *filename = NULL;
1000 else if (! (*filename = PyString_AsString(v)))
1001 goto finally;
Barry Warsaw035574d1997-08-29 22:07:17 +00001002
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001003 Py_DECREF(v);
1004 if (!(v = PyObject_GetAttrString(err, "lineno")))
1005 goto finally;
1006 hold = PyInt_AsLong(v);
1007 Py_DECREF(v);
1008 v = NULL;
1009 if (hold < 0 && PyErr_Occurred())
1010 goto finally;
1011 *lineno = (int)hold;
Barry Warsaw035574d1997-08-29 22:07:17 +00001012
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001013 if (!(v = PyObject_GetAttrString(err, "offset")))
1014 goto finally;
1015 if (v == Py_None) {
1016 *offset = -1;
1017 Py_DECREF(v);
1018 v = NULL;
1019 } else {
1020 hold = PyInt_AsLong(v);
1021 Py_DECREF(v);
1022 v = NULL;
1023 if (hold < 0 && PyErr_Occurred())
1024 goto finally;
1025 *offset = (int)hold;
1026 }
Barry Warsaw035574d1997-08-29 22:07:17 +00001027
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001028 if (!(v = PyObject_GetAttrString(err, "text")))
1029 goto finally;
1030 if (v == Py_None)
1031 *text = NULL;
1032 else if (! (*text = PyString_AsString(v)))
1033 goto finally;
1034 Py_DECREF(v);
1035 return 1;
Barry Warsaw035574d1997-08-29 22:07:17 +00001036
1037finally:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001038 Py_XDECREF(v);
1039 return 0;
Barry Warsaw035574d1997-08-29 22:07:17 +00001040}
1041
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001042void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001043PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001044{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001045 PyErr_PrintEx(1);
Guido van Rossuma61691e1998-02-06 22:27:24 +00001046}
1047
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001048static void
Martin v. Löwis95292d62002-12-11 14:04:59 +00001049print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001050{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001051 char *nl;
1052 if (offset >= 0) {
1053 if (offset > 0 && offset == (int)strlen(text))
1054 offset--;
1055 for (;;) {
1056 nl = strchr(text, '\n');
1057 if (nl == NULL || nl-text >= offset)
1058 break;
1059 offset -= (int)(nl+1-text);
1060 text = nl+1;
1061 }
1062 while (*text == ' ' || *text == '\t') {
1063 text++;
1064 offset--;
1065 }
1066 }
1067 PyFile_WriteString(" ", f);
1068 PyFile_WriteString(text, f);
1069 if (*text == '\0' || text[strlen(text)-1] != '\n')
1070 PyFile_WriteString("\n", f);
1071 if (offset == -1)
1072 return;
1073 PyFile_WriteString(" ", f);
1074 offset--;
1075 while (offset > 0) {
1076 PyFile_WriteString(" ", f);
1077 offset--;
1078 }
1079 PyFile_WriteString("^\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001080}
1081
Guido van Rossum66e8e862001-03-23 17:54:43 +00001082static void
1083handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001084{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001085 PyObject *exception, *value, *tb;
1086 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001087
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001088 if (Py_InspectFlag)
1089 /* Don't exit if -i flag was given. This flag is set to 0
1090 * when entering interactive mode for inspecting. */
1091 return;
Georg Brandl49aafc92007-03-07 00:34:46 +00001092
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001093 PyErr_Fetch(&exception, &value, &tb);
1094 if (Py_FlushLine())
1095 PyErr_Clear();
1096 fflush(stdout);
1097 if (value == NULL || value == Py_None)
1098 goto done;
1099 if (PyExceptionInstance_Check(value)) {
1100 /* The error code should be in the `code' attribute. */
1101 PyObject *code = PyObject_GetAttrString(value, "code");
1102 if (code) {
1103 Py_DECREF(value);
1104 value = code;
1105 if (value == Py_None)
1106 goto done;
1107 }
1108 /* If we failed to dig out the 'code' attribute,
1109 just let the else clause below print the error. */
1110 }
1111 if (PyInt_Check(value))
1112 exitcode = (int)PyInt_AsLong(value);
1113 else {
Victor Stinnerc3e40e02010-05-25 22:40:38 +00001114 PyObject *sys_stderr = PySys_GetObject("stderr");
1115 if (sys_stderr != NULL && sys_stderr != Py_None) {
1116 PyFile_WriteObject(value, sys_stderr, Py_PRINT_RAW);
1117 } else {
1118 PyObject_Print(value, stderr, Py_PRINT_RAW);
1119 fflush(stderr);
1120 }
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001121 PySys_WriteStderr("\n");
1122 exitcode = 1;
1123 }
Tim Peterscf615b52003-04-19 18:47:02 +00001124 done:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001125 /* Restore and clear the exception info, in order to properly decref
1126 * the exception, value, and traceback. If we just exit instead,
1127 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1128 * some finalizers from running.
1129 */
1130 PyErr_Restore(exception, value, tb);
1131 PyErr_Clear();
1132 Py_Exit(exitcode);
1133 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001134}
1135
1136void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001137PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001138{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001139 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001140
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001141 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1142 handle_system_exit();
1143 }
1144 PyErr_Fetch(&exception, &v, &tb);
1145 if (exception == NULL)
1146 return;
1147 PyErr_NormalizeException(&exception, &v, &tb);
1148 if (exception == NULL)
1149 return;
1150 /* Now we know v != NULL too */
1151 if (set_sys_last_vars) {
1152 PySys_SetObject("last_type", exception);
1153 PySys_SetObject("last_value", v);
1154 PySys_SetObject("last_traceback", tb);
1155 }
1156 hook = PySys_GetObject("excepthook");
1157 if (hook) {
1158 PyObject *args = PyTuple_Pack(3,
1159 exception, v, tb ? tb : Py_None);
1160 PyObject *result = PyEval_CallObject(hook, args);
1161 if (result == NULL) {
1162 PyObject *exception2, *v2, *tb2;
1163 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1164 handle_system_exit();
1165 }
1166 PyErr_Fetch(&exception2, &v2, &tb2);
1167 PyErr_NormalizeException(&exception2, &v2, &tb2);
1168 /* It should not be possible for exception2 or v2
1169 to be NULL. However PyErr_Display() can't
1170 tolerate NULLs, so just be safe. */
1171 if (exception2 == NULL) {
1172 exception2 = Py_None;
1173 Py_INCREF(exception2);
1174 }
1175 if (v2 == NULL) {
1176 v2 = Py_None;
1177 Py_INCREF(v2);
1178 }
1179 if (Py_FlushLine())
1180 PyErr_Clear();
1181 fflush(stdout);
1182 PySys_WriteStderr("Error in sys.excepthook:\n");
1183 PyErr_Display(exception2, v2, tb2);
1184 PySys_WriteStderr("\nOriginal exception was:\n");
1185 PyErr_Display(exception, v, tb);
1186 Py_DECREF(exception2);
1187 Py_DECREF(v2);
1188 Py_XDECREF(tb2);
1189 }
1190 Py_XDECREF(result);
1191 Py_XDECREF(args);
1192 } else {
1193 PySys_WriteStderr("sys.excepthook is missing\n");
1194 PyErr_Display(exception, v, tb);
1195 }
1196 Py_XDECREF(exception);
1197 Py_XDECREF(v);
1198 Py_XDECREF(tb);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001199}
1200
Richard Jones7b9558d2006-05-27 12:29:24 +00001201void
1202PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001203{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001204 int err = 0;
1205 PyObject *f = PySys_GetObject("stderr");
1206 Py_INCREF(value);
1207 if (f == NULL)
1208 fprintf(stderr, "lost sys.stderr\n");
1209 else {
1210 if (Py_FlushLine())
1211 PyErr_Clear();
1212 fflush(stdout);
1213 if (tb && tb != Py_None)
1214 err = PyTraceBack_Print(tb, f);
1215 if (err == 0 &&
1216 PyObject_HasAttrString(value, "print_file_and_line"))
1217 {
1218 PyObject *message;
1219 const char *filename, *text;
1220 int lineno, offset;
1221 if (!parse_syntax_error(value, &message, &filename,
1222 &lineno, &offset, &text))
1223 PyErr_Clear();
1224 else {
1225 char buf[10];
1226 PyFile_WriteString(" File \"", f);
1227 if (filename == NULL)
1228 PyFile_WriteString("<string>", f);
1229 else
1230 PyFile_WriteString(filename, f);
1231 PyFile_WriteString("\", line ", f);
1232 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
1233 PyFile_WriteString(buf, f);
1234 PyFile_WriteString("\n", f);
1235 if (text != NULL)
1236 print_error_text(f, offset, text);
1237 Py_DECREF(value);
1238 value = message;
1239 /* Can't be bothered to check all those
1240 PyFile_WriteString() calls */
1241 if (PyErr_Occurred())
1242 err = -1;
1243 }
1244 }
1245 if (err) {
1246 /* Don't do anything else */
1247 }
1248 else if (PyExceptionClass_Check(exception)) {
1249 PyObject* moduleName;
1250 char* className = PyExceptionClass_Name(exception);
1251 if (className != NULL) {
1252 char *dot = strrchr(className, '.');
1253 if (dot != NULL)
1254 className = dot+1;
1255 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001256
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001257 moduleName = PyObject_GetAttrString(exception, "__module__");
1258 if (moduleName == NULL)
1259 err = PyFile_WriteString("<unknown>", f);
1260 else {
1261 char* modstr = PyString_AsString(moduleName);
1262 if (modstr && strcmp(modstr, "exceptions"))
1263 {
1264 err = PyFile_WriteString(modstr, f);
1265 err += PyFile_WriteString(".", f);
1266 }
1267 Py_DECREF(moduleName);
1268 }
1269 if (err == 0) {
1270 if (className == NULL)
1271 err = PyFile_WriteString("<unknown>", f);
1272 else
1273 err = PyFile_WriteString(className, f);
1274 }
1275 }
1276 else
1277 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
1278 if (err == 0 && (value != Py_None)) {
1279 PyObject *s = PyObject_Str(value);
1280 /* only print colon if the str() of the
1281 object is not the empty string
1282 */
1283 if (s == NULL)
1284 err = -1;
1285 else if (!PyString_Check(s) ||
1286 PyString_GET_SIZE(s) != 0)
1287 err = PyFile_WriteString(": ", f);
1288 if (err == 0)
1289 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1290 Py_XDECREF(s);
1291 }
1292 /* try to write a newline in any case */
1293 err += PyFile_WriteString("\n", f);
1294 }
1295 Py_DECREF(value);
1296 /* If an error happened here, don't show it.
1297 XXX This is wrong, but too many callers rely on this behavior. */
1298 if (err != 0)
1299 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001300}
1301
Guido van Rossum82598051997-03-05 00:20:32 +00001302PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001303PyRun_StringFlags(const char *str, int start, PyObject *globals,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001304 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001305{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001306 PyObject *ret = NULL;
1307 mod_ty mod;
1308 PyArena *arena = PyArena_New();
1309 if (arena == NULL)
1310 return NULL;
1311
1312 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
1313 if (mod != NULL)
1314 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
1315 PyArena_Free(arena);
1316 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001317}
1318
1319PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001320PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001321 PyObject *locals, int closeit, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001322{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001323 PyObject *ret;
1324 mod_ty mod;
1325 PyArena *arena = PyArena_New();
1326 if (arena == NULL)
1327 return NULL;
1328
1329 mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
1330 flags, NULL, arena);
1331 if (closeit)
1332 fclose(fp);
1333 if (mod == NULL) {
1334 PyArena_Free(arena);
1335 return NULL;
1336 }
1337 ret = run_mod(mod, filename, globals, locals, flags, arena);
1338 PyArena_Free(arena);
1339 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001340}
1341
Guido van Rossum82598051997-03-05 00:20:32 +00001342static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001343run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001344 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001345{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001346 PyCodeObject *co;
1347 PyObject *v;
1348 co = PyAST_Compile(mod, filename, flags, arena);
1349 if (co == NULL)
1350 return NULL;
1351 v = PyEval_EvalCode(co, globals, locals);
1352 Py_DECREF(co);
1353 return v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001354}
1355
Guido van Rossum82598051997-03-05 00:20:32 +00001356static PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001357run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001358 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001359{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001360 PyCodeObject *co;
1361 PyObject *v;
1362 long magic;
1363 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001364
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001365 magic = PyMarshal_ReadLongFromFile(fp);
1366 if (magic != PyImport_GetMagicNumber()) {
1367 PyErr_SetString(PyExc_RuntimeError,
1368 "Bad magic number in .pyc file");
1369 return NULL;
1370 }
1371 (void) PyMarshal_ReadLongFromFile(fp);
1372 v = PyMarshal_ReadLastObjectFromFile(fp);
1373 fclose(fp);
1374 if (v == NULL || !PyCode_Check(v)) {
1375 Py_XDECREF(v);
1376 PyErr_SetString(PyExc_RuntimeError,
1377 "Bad code object in .pyc file");
1378 return NULL;
1379 }
1380 co = (PyCodeObject *)v;
1381 v = PyEval_EvalCode(co, globals, locals);
1382 if (v && flags)
1383 flags->cf_flags |= (co->co_flags & PyCF_MASK);
1384 Py_DECREF(co);
1385 return v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001386}
1387
Guido van Rossum82598051997-03-05 00:20:32 +00001388PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001389Py_CompileStringFlags(const char *str, const char *filename, int start,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001390 PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001391{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001392 PyCodeObject *co;
1393 mod_ty mod;
1394 PyArena *arena = PyArena_New();
1395 if (arena == NULL)
1396 return NULL;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001397
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001398 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
1399 if (mod == NULL) {
1400 PyArena_Free(arena);
1401 return NULL;
1402 }
1403 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
1404 PyObject *result = PyAST_mod2obj(mod);
1405 PyArena_Free(arena);
1406 return result;
1407 }
1408 co = PyAST_Compile(mod, filename, flags, arena);
1409 PyArena_Free(arena);
1410 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001411}
1412
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001413struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001414Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001415{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001416 struct symtable *st;
1417 mod_ty mod;
1418 PyCompilerFlags flags;
1419 PyArena *arena = PyArena_New();
1420 if (arena == NULL)
1421 return NULL;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001422
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001423 flags.cf_flags = 0;
Christian Heimes7f23d862008-03-26 22:51:58 +00001424
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001425 mod = PyParser_ASTFromString(str, filename, start, &flags, arena);
1426 if (mod == NULL) {
1427 PyArena_Free(arena);
1428 return NULL;
1429 }
1430 st = PySymtable_Build(mod, filename, 0);
1431 PyArena_Free(arena);
1432 return st;
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001433}
1434
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001435/* Preferred access to parser is through AST. */
1436mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001437PyParser_ASTFromString(const char *s, const char *filename, int start,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001438 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001439{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001440 mod_ty mod;
1441 PyCompilerFlags localflags;
1442 perrdetail err;
1443 int iflags = PARSER_FLAGS(flags);
Christian Heimes3c608332008-03-26 22:01:37 +00001444
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001445 node *n = PyParser_ParseStringFlagsFilenameEx(s, filename,
1446 &_PyParser_Grammar, start, &err,
1447 &iflags);
1448 if (flags == NULL) {
1449 localflags.cf_flags = 0;
1450 flags = &localflags;
1451 }
1452 if (n) {
1453 flags->cf_flags |= iflags & PyCF_MASK;
1454 mod = PyAST_FromNode(n, flags, filename, arena);
1455 PyNode_Free(n);
1456 return mod;
1457 }
1458 else {
1459 err_input(&err);
1460 return NULL;
1461 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001462}
1463
1464mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001465PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001466 char *ps2, PyCompilerFlags *flags, int *errcode,
1467 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001468{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001469 mod_ty mod;
1470 PyCompilerFlags localflags;
1471 perrdetail err;
1472 int iflags = PARSER_FLAGS(flags);
Christian Heimes3c608332008-03-26 22:01:37 +00001473
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001474 node *n = PyParser_ParseFileFlagsEx(fp, filename, &_PyParser_Grammar,
1475 start, ps1, ps2, &err, &iflags);
1476 if (flags == NULL) {
1477 localflags.cf_flags = 0;
1478 flags = &localflags;
1479 }
1480 if (n) {
1481 flags->cf_flags |= iflags & PyCF_MASK;
1482 mod = PyAST_FromNode(n, flags, filename, arena);
1483 PyNode_Free(n);
1484 return mod;
1485 }
1486 else {
1487 err_input(&err);
1488 if (errcode)
1489 *errcode = err.error;
1490 return NULL;
1491 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001492}
1493
Guido van Rossuma110aa61994-08-29 12:50:44 +00001494/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001495
Guido van Rossuma110aa61994-08-29 12:50:44 +00001496node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001497PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001498{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001499 perrdetail err;
1500 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1501 start, NULL, NULL, &err, flags);
1502 if (n == NULL)
1503 err_input(&err);
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001504
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001505 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001506}
1507
Guido van Rossuma110aa61994-08-29 12:50:44 +00001508/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001509
Guido van Rossuma110aa61994-08-29 12:50:44 +00001510node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001511PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001512{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001513 perrdetail err;
1514 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1515 start, &err, flags);
1516 if (n == NULL)
1517 err_input(&err);
1518 return n;
Tim Petersfe2127d2001-07-16 05:37:24 +00001519}
1520
1521node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001522PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001523 int start, int flags)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001524{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001525 perrdetail err;
1526 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1527 &_PyParser_Grammar, start, &err, flags);
1528 if (n == NULL)
1529 err_input(&err);
1530 return n;
Thomas Heller6b17abf2002-07-09 09:23:27 +00001531}
1532
1533node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001534PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001535{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001536 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001537}
1538
Guido van Rossum66ebd912003-04-17 16:02:26 +00001539/* May want to move a more generalized form of this to parsetok.c or
1540 even parser modules. */
1541
1542void
1543PyParser_SetError(perrdetail *err)
1544{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001545 err_input(err);
Guido van Rossum66ebd912003-04-17 16:02:26 +00001546}
1547
Guido van Rossuma110aa61994-08-29 12:50:44 +00001548/* Set the error appropriate to the given input error code (see errcode.h) */
1549
1550static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001551err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001552{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001553 PyObject *v, *w, *errtype;
1554 PyObject* u = NULL;
1555 char *msg = NULL;
1556 errtype = PyExc_SyntaxError;
1557 switch (err->error) {
1558 case E_ERROR:
1559 return;
1560 case E_SYNTAX:
1561 errtype = PyExc_IndentationError;
1562 if (err->expected == INDENT)
1563 msg = "expected an indented block";
1564 else if (err->token == INDENT)
1565 msg = "unexpected indent";
1566 else if (err->token == DEDENT)
1567 msg = "unexpected unindent";
1568 else {
1569 errtype = PyExc_SyntaxError;
1570 msg = "invalid syntax";
1571 }
1572 break;
1573 case E_TOKEN:
1574 msg = "invalid token";
1575 break;
1576 case E_EOFS:
1577 msg = "EOF while scanning triple-quoted string literal";
1578 break;
1579 case E_EOLS:
1580 msg = "EOL while scanning string literal";
1581 break;
1582 case E_INTR:
1583 if (!PyErr_Occurred())
1584 PyErr_SetNone(PyExc_KeyboardInterrupt);
1585 goto cleanup;
1586 case E_NOMEM:
1587 PyErr_NoMemory();
1588 goto cleanup;
1589 case E_EOF:
1590 msg = "unexpected EOF while parsing";
1591 break;
1592 case E_TABSPACE:
1593 errtype = PyExc_TabError;
1594 msg = "inconsistent use of tabs and spaces in indentation";
1595 break;
1596 case E_OVERFLOW:
1597 msg = "expression too long";
1598 break;
1599 case E_DEDENT:
1600 errtype = PyExc_IndentationError;
1601 msg = "unindent does not match any outer indentation level";
1602 break;
1603 case E_TOODEEP:
1604 errtype = PyExc_IndentationError;
1605 msg = "too many levels of indentation";
1606 break;
1607 case E_DECODE: {
1608 PyObject *type, *value, *tb;
1609 PyErr_Fetch(&type, &value, &tb);
1610 if (value != NULL) {
1611 u = PyObject_Str(value);
1612 if (u != NULL) {
1613 msg = PyString_AsString(u);
1614 }
1615 }
1616 if (msg == NULL)
1617 msg = "unknown decode error";
1618 Py_XDECREF(type);
1619 Py_XDECREF(value);
1620 Py_XDECREF(tb);
1621 break;
1622 }
1623 case E_LINECONT:
1624 msg = "unexpected character after line continuation character";
1625 break;
1626 default:
1627 fprintf(stderr, "error=%d\n", err->error);
1628 msg = "unknown parsing error";
1629 break;
1630 }
1631 v = Py_BuildValue("(ziiz)", err->filename,
1632 err->lineno, err->offset, err->text);
1633 w = NULL;
1634 if (v != NULL)
1635 w = Py_BuildValue("(sO)", msg, v);
1636 Py_XDECREF(u);
1637 Py_XDECREF(v);
1638 PyErr_SetObject(errtype, w);
1639 Py_XDECREF(w);
Georg Brandl1ad108d2008-07-19 10:08:55 +00001640cleanup:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001641 if (err->text != NULL) {
1642 PyObject_FREE(err->text);
1643 err->text = NULL;
1644 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001645}
1646
1647/* Print fatal error message and abort */
1648
1649void
Tim Peters7c321a82002-07-09 02:57:01 +00001650Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001651{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001652 fprintf(stderr, "Fatal Python error: %s\n", msg);
1653 fflush(stderr); /* it helps in Windows debug build */
Jesse Noller70b11f02009-03-31 22:25:20 +00001654
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001655#ifdef MS_WINDOWS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001656 {
1657 size_t len = strlen(msg);
1658 WCHAR* buffer;
1659 size_t i;
Georg Brandl734373c2009-01-03 21:55:17 +00001660
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001661 /* Convert the message to wchar_t. This uses a simple one-to-one
1662 conversion, assuming that the this error message actually uses ASCII
1663 only. If this ceases to be true, we will have to convert. */
1664 buffer = alloca( (len+1) * (sizeof *buffer));
1665 for( i=0; i<=len; ++i)
1666 buffer[i] = msg[i];
1667 OutputDebugStringW(L"Fatal Python error: ");
1668 OutputDebugStringW(buffer);
1669 OutputDebugStringW(L"\n");
1670 }
Guido van Rossum0ba35361998-08-13 13:33:16 +00001671#ifdef _DEBUG
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001672 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001673#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001674#endif /* MS_WINDOWS */
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001675 abort();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001676}
1677
1678/* Clean up and exit */
1679
Guido van Rossuma110aa61994-08-29 12:50:44 +00001680#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001681#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001682#endif
1683
Antoine Pitrou9aece752009-10-27 12:48:52 +00001684/* Wait until threading._shutdown completes, provided
1685 the threading module was imported in the first place.
1686 The shutdown routine will wait until all non-daemon
1687 "threading" threads have completed. */
1688static void
1689wait_for_thread_shutdown(void)
1690{
1691#ifdef WITH_THREAD
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001692 PyObject *result;
1693 PyThreadState *tstate = PyThreadState_GET();
1694 PyObject *threading = PyMapping_GetItemString(tstate->interp->modules,
1695 "threading");
1696 if (threading == NULL) {
1697 /* threading not imported */
1698 PyErr_Clear();
1699 return;
1700 }
1701 result = PyObject_CallMethod(threading, "_shutdown", "");
1702 if (result == NULL)
1703 PyErr_WriteUnraisable(threading);
1704 else
1705 Py_DECREF(result);
1706 Py_DECREF(threading);
Antoine Pitrou9aece752009-10-27 12:48:52 +00001707#endif
1708}
1709
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001710#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001711static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001712static int nexitfuncs = 0;
1713
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001714int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001715{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001716 if (nexitfuncs >= NEXITFUNCS)
1717 return -1;
1718 exitfuncs[nexitfuncs++] = func;
1719 return 0;
Guido van Rossum1662dd51994-09-07 14:38:28 +00001720}
1721
Guido van Rossumcc283f51997-08-05 02:22:03 +00001722static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001723call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001724{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001725 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001726
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001727 if (exitfunc) {
1728 PyObject *res;
1729 Py_INCREF(exitfunc);
1730 PySys_SetObject("exitfunc", (PyObject *)NULL);
1731 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
1732 if (res == NULL) {
1733 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1734 PySys_WriteStderr("Error in sys.exitfunc:\n");
1735 }
1736 PyErr_Print();
1737 }
1738 Py_DECREF(exitfunc);
1739 }
Guido van Rossum59bff391992-09-03 20:28:00 +00001740
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001741 if (Py_FlushLine())
1742 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001743}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001744
Guido van Rossumcc283f51997-08-05 02:22:03 +00001745static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001746call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001747{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001748 while (nexitfuncs > 0)
1749 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001750
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001751 fflush(stdout);
1752 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001753}
1754
1755void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001756Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001757{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001758 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001759
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001760 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001761}
1762
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001763static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001764initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001765{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001766#ifdef SIGPIPE
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001767 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001768#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001769#ifdef SIGXFZ
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001770 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001771#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001772#ifdef SIGXFSZ
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001773 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001774#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001775 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001776}
1777
Guido van Rossum7433b121997-02-14 19:45:36 +00001778
1779/*
1780 * The file descriptor fd is considered ``interactive'' if either
1781 * a) isatty(fd) is TRUE, or
1782 * b) the -i flag was given, and the filename associated with
1783 * the descriptor is NULL or "<stdin>" or "???".
1784 */
1785int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001786Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001787{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001788 if (isatty((int)fileno(fp)))
1789 return 1;
1790 if (!Py_InteractiveFlag)
1791 return 0;
1792 return (filename == NULL) ||
1793 (strcmp(filename, "<stdin>") == 0) ||
1794 (strcmp(filename, "???") == 0);
Guido van Rossum7433b121997-02-14 19:45:36 +00001795}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001796
1797
Tim Petersd08e3822003-04-17 15:24:21 +00001798#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001799#if defined(WIN32) && defined(_MSC_VER)
1800
1801/* Stack checking for Microsoft C */
1802
1803#include <malloc.h>
1804#include <excpt.h>
1805
Fred Drakee8de31c2000-08-31 05:38:39 +00001806/*
1807 * Return non-zero when we run out of memory on the stack; zero otherwise.
1808 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001809int
Fred Drake399739f2000-08-31 05:52:44 +00001810PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001811{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001812 __try {
1813 /* alloca throws a stack overflow exception if there's
1814 not enough space left on the stack */
1815 alloca(PYOS_STACK_MARGIN * sizeof(void*));
1816 return 0;
1817 } __except (GetExceptionCode() == STATUS_STACK_OVERFLOW ?
1818 EXCEPTION_EXECUTE_HANDLER :
1819 EXCEPTION_CONTINUE_SEARCH) {
1820 int errcode = _resetstkoflw();
1821 if (errcode == 0)
1822 {
1823 Py_FatalError("Could not reset the stack!");
1824 }
1825 }
1826 return 1;
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001827}
1828
1829#endif /* WIN32 && _MSC_VER */
1830
1831/* Alternate implementations can be added here... */
1832
1833#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001834
1835
1836/* Wrappers around sigaction() or signal(). */
1837
1838PyOS_sighandler_t
1839PyOS_getsig(int sig)
1840{
1841#ifdef HAVE_SIGACTION
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001842 struct sigaction context;
1843 if (sigaction(sig, NULL, &context) == -1)
1844 return SIG_ERR;
1845 return context.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001846#else
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001847 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001848/* Special signal handling for the secure CRT in Visual Studio 2005 */
1849#if defined(_MSC_VER) && _MSC_VER >= 1400
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001850 switch (sig) {
1851 /* Only these signals are valid */
1852 case SIGINT:
1853 case SIGILL:
1854 case SIGFPE:
1855 case SIGSEGV:
1856 case SIGTERM:
1857 case SIGBREAK:
1858 case SIGABRT:
1859 break;
1860 /* Don't call signal() with other values or it will assert */
1861 default:
1862 return SIG_ERR;
1863 }
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001864#endif /* _MSC_VER && _MSC_VER >= 1400 */
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001865 handler = signal(sig, SIG_IGN);
1866 if (handler != SIG_ERR)
1867 signal(sig, handler);
1868 return handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001869#endif
1870}
1871
1872PyOS_sighandler_t
1873PyOS_setsig(int sig, PyOS_sighandler_t handler)
1874{
1875#ifdef HAVE_SIGACTION
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001876 /* Some code in Modules/signalmodule.c depends on sigaction() being
1877 * used here if HAVE_SIGACTION is defined. Fix that if this code
1878 * changes to invalidate that assumption.
1879 */
1880 struct sigaction context, ocontext;
1881 context.sa_handler = handler;
1882 sigemptyset(&context.sa_mask);
1883 context.sa_flags = 0;
1884 if (sigaction(sig, &context, &ocontext) == -1)
1885 return SIG_ERR;
1886 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001887#else
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001888 PyOS_sighandler_t oldhandler;
1889 oldhandler = signal(sig, handler);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001890#ifdef HAVE_SIGINTERRUPT
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001891 siginterrupt(sig, 1);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001892#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001893 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001894#endif
1895}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001896
1897/* Deprecated C API functions still provided for binary compatiblity */
1898
1899#undef PyParser_SimpleParseFile
Thomas Heller1b046642006-04-18 18:51:06 +00001900PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001901PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1902{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001903 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001904}
1905
Thomas Heller1b046642006-04-18 18:51:06 +00001906#undef PyParser_SimpleParseString
1907PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001908PyParser_SimpleParseString(const char *str, int start)
1909{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001910 return PyParser_SimpleParseStringFlags(str, start, 0);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001911}
Anthony Baxterac6bd462006-04-13 02:06:09 +00001912
Thomas Heller1b046642006-04-18 18:51:06 +00001913#undef PyRun_AnyFile
1914PyAPI_FUNC(int)
1915PyRun_AnyFile(FILE *fp, const char *name)
1916{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001917 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001918}
1919
1920#undef PyRun_AnyFileEx
1921PyAPI_FUNC(int)
1922PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1923{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001924 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001925}
1926
1927#undef PyRun_AnyFileFlags
1928PyAPI_FUNC(int)
1929PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1930{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001931 return PyRun_AnyFileExFlags(fp, name, 0, flags);
Thomas Heller1b046642006-04-18 18:51:06 +00001932}
1933
1934#undef PyRun_File
1935PyAPI_FUNC(PyObject *)
1936PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1937{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001938 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001939}
1940
1941#undef PyRun_FileEx
1942PyAPI_FUNC(PyObject *)
1943PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1944{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001945 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001946}
1947
1948#undef PyRun_FileFlags
1949PyAPI_FUNC(PyObject *)
1950PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001951 PyCompilerFlags *flags)
Thomas Heller1b046642006-04-18 18:51:06 +00001952{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001953 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
Thomas Heller1b046642006-04-18 18:51:06 +00001954}
1955
1956#undef PyRun_SimpleFile
1957PyAPI_FUNC(int)
1958PyRun_SimpleFile(FILE *f, const char *p)
1959{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001960 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001961}
1962
1963#undef PyRun_SimpleFileEx
1964PyAPI_FUNC(int)
1965PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1966{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001967 return PyRun_SimpleFileExFlags(f, p, c, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001968}
1969
1970
1971#undef PyRun_String
1972PyAPI_FUNC(PyObject *)
1973PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1974{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001975 return PyRun_StringFlags(str, s, g, l, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001976}
1977
1978#undef PyRun_SimpleString
1979PyAPI_FUNC(int)
1980PyRun_SimpleString(const char *s)
1981{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001982 return PyRun_SimpleStringFlags(s, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001983}
1984
1985#undef Py_CompileString
1986PyAPI_FUNC(PyObject *)
1987Py_CompileString(const char *str, const char *p, int s)
1988{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001989 return Py_CompileStringFlags(str, p, s, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001990}
1991
1992#undef PyRun_InteractiveOne
1993PyAPI_FUNC(int)
1994PyRun_InteractiveOne(FILE *f, const char *p)
1995{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001996 return PyRun_InteractiveOneFlags(f, p, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00001997}
1998
1999#undef PyRun_InteractiveLoop
2000PyAPI_FUNC(int)
2001PyRun_InteractiveLoop(FILE *f, const char *p)
2002{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002003 return PyRun_InteractiveLoopFlags(f, p, NULL);
Thomas Heller1b046642006-04-18 18:51:06 +00002004}
2005
Anthony Baxterac6bd462006-04-13 02:06:09 +00002006#ifdef __cplusplus
2007}
2008#endif
2009