blob: ec9ed02ab9bd20abf9884ab6fe0fce3bfe8c075f [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"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000020
Thomas Wouters0e3f5912006-08-11 14:57:12 +000021#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000022#include <signal.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000023#endif
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000024
Martin v. Löwis73d538b2003-03-05 15:13:47 +000025#ifdef HAVE_LANGINFO_H
26#include <locale.h>
27#include <langinfo.h>
28#endif
29
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000030#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000031#undef BYTE
32#include "windows.h"
33#endif
34
Neal Norwitz4281cef2006-03-04 19:58:13 +000035#ifndef Py_REF_DEBUG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000036#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000037#else /* Py_REF_DEBUG */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000038#define PRINT_TOTAL_REFS() fprintf(stderr, \
39 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
40 _Py_GetRefTotal())
41#endif
42
43#ifdef __cplusplus
44extern "C" {
Neal Norwitz4281cef2006-03-04 19:58:13 +000045#endif
46
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000047extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000048
Guido van Rossum82598051997-03-05 00:20:32 +000049extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000050
Guido van Rossumb73cc041993-11-01 16:28:59 +000051/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000052static void initmain(void);
53static void initsite(void);
Guido van Rossumce3a72a2007-10-19 23:16:50 +000054static int initstdio(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000055static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000056 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000057static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000058 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000059static void err_input(perrdetail *);
60static void initsigs(void);
Collin Winter670e6922007-03-21 02:57:17 +000061static void call_py_exitfuncs(void);
Tim Petersdbd9ba62000-07-09 03:09:57 +000062static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000063extern void _PyUnicode_Init(void);
64extern void _PyUnicode_Fini(void);
Guido van Rossumddefaf32007-01-14 03:31:43 +000065extern int _PyLong_Init(void);
66extern void PyLong_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000067
Mark Hammond8d98d2c2003-04-19 15:41:53 +000068#ifdef WITH_THREAD
69extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
70extern void _PyGILState_Fini(void);
71#endif /* WITH_THREAD */
72
Guido van Rossum82598051997-03-05 00:20:32 +000073int Py_DebugFlag; /* Needed by parser.c */
74int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000075int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumd8faa362007-04-27 19:54:29 +000076int Py_InspectFlag; /* Needed to determine whether to exit at SystemError */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000077int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000078int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000079int Py_FrozenFlag; /* Needed by getpath.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000080int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000081
Mark Hammonda43fd0c2003-02-19 00:33:33 +000082/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000083 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000084*/
Mark Hammondedd07732003-07-15 23:03:55 +000085static PyObject *warnings_module = NULL;
86
87/* Returns a borrowed reference to the 'warnings' module, or NULL.
88 If the module is returned, it is guaranteed to have been obtained
89 without acquiring the import lock
90*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000091PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000092{
93 PyObject *typ, *val, *tb;
94 PyObject *all_modules;
95 /* If we managed to get the module at init time, just use it */
96 if (warnings_module)
97 return warnings_module;
98 /* If it wasn't available at init time, it may be available
99 now in sys.modules (common scenario is frozen apps: import
100 at init time fails, but the frozen init code sets up sys.path
101 correctly, then does an implicit import of warnings for us
102 */
103 /* Save and restore any exceptions */
104 PyErr_Fetch(&typ, &val, &tb);
105
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000106 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000107 if (all_modules) {
108 warnings_module = PyDict_GetItemString(all_modules, "warnings");
109 /* We keep a ref in the global */
110 Py_XINCREF(warnings_module);
111 }
112 PyErr_Restore(typ, val, tb);
113 return warnings_module;
114}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000115
Guido van Rossum25ce5661997-08-02 03:10:38 +0000116static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000117
Thomas Wouters7e474022000-07-16 12:04:32 +0000118/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000119
120int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000121Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000122{
123 return initialized;
124}
125
Guido van Rossum25ce5661997-08-02 03:10:38 +0000126/* Global initializations. Can be undone by Py_Finalize(). Don't
127 call this twice without an intervening Py_Finalize() call. When
128 initializations fail, a fatal error is issued and the function does
129 not return. On return, the first thread and interpreter state have
130 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000131
Guido van Rossum25ce5661997-08-02 03:10:38 +0000132 Locking: you must hold the interpreter lock while calling this.
133 (If the lock has not yet been initialized, that's equivalent to
134 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000135
Guido van Rossum25ce5661997-08-02 03:10:38 +0000136*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000137
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000138static int
139add_flag(int flag, const char *envs)
140{
141 int env = atoi(envs);
142 if (flag < env)
143 flag = env;
144 if (flag < 1)
145 flag = 1;
146 return flag;
147}
148
Guido van Rossuma027efa1997-05-05 20:56:21 +0000149void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000150Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000151{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000152 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000153 PyThreadState *tstate;
Guido van Rossum826d8972007-10-30 18:34:07 +0000154 PyObject *bimod, *sysmod, *pstderr;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000155 char *p;
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000156#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000157 char *codeset;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000158#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000159 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000160
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000161 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000162 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000163 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000164
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +0000165#ifdef HAVE_SETLOCALE
166 /* Set up the LC_CTYPE locale, so we can obtain
167 the locale's charset without having to switch
168 locales. */
169 setlocale(LC_CTYPE, "");
170#endif
171
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000172 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000173 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000174 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000175 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000176 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000177 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000178
Guido van Rossuma027efa1997-05-05 20:56:21 +0000179 interp = PyInterpreterState_New();
180 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000181 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000182
Guido van Rossuma027efa1997-05-05 20:56:21 +0000183 tstate = PyThreadState_New(interp);
184 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000185 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000186 (void) PyThreadState_Swap(tstate);
187
Guido van Rossum70d893a2001-08-16 08:21:42 +0000188 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000189
Neal Norwitzb2501f42002-12-31 03:42:13 +0000190 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000191 Py_FatalError("Py_Initialize: can't init frames");
192
Guido van Rossumddefaf32007-01-14 03:31:43 +0000193 if (!_PyLong_Init())
194 Py_FatalError("Py_Initialize: can't init longs");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000195
Neal Norwitz6968b052007-02-27 19:02:19 +0000196 if (!PyBytes_Init())
197 Py_FatalError("Py_Initialize: can't init bytes");
198
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000199 _PyFloat_Init();
200
Guido van Rossum25ce5661997-08-02 03:10:38 +0000201 interp->modules = PyDict_New();
202 if (interp->modules == NULL)
203 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000204 interp->modules_reloading = PyDict_New();
205 if (interp->modules_reloading == NULL)
206 Py_FatalError("Py_Initialize: can't make modules_reloading dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000207
Guido van Rossumc94044c2000-03-10 23:03:54 +0000208 /* Init Unicode implementation; relies on the codec registry */
209 _PyUnicode_Init();
210
Barry Warsawf242aa02000-05-25 23:09:49 +0000211 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000212 if (bimod == NULL)
213 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000214 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000215 if (interp->builtins == NULL)
216 Py_FatalError("Py_Initialize: can't initialize builtins dict");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000217 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000218
219 sysmod = _PySys_Init();
220 if (sysmod == NULL)
221 Py_FatalError("Py_Initialize: can't initialize sys");
222 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000223 if (interp->sysdict == NULL)
224 Py_FatalError("Py_Initialize: can't initialize sys dict");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000225 Py_INCREF(interp->sysdict);
226 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000227 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000228 PyDict_SetItemString(interp->sysdict, "modules",
229 interp->modules);
230
Guido van Rossum826d8972007-10-30 18:34:07 +0000231 /* Set up a preliminary stderr printer until we have enough
232 infrastructure for the io module in place. */
233 pstderr = PyFile_NewStdPrinter(fileno(stderr));
234 if (pstderr == NULL)
235 Py_FatalError("Py_Initialize: can't set preliminary stderr");
236 PySys_SetObject("stderr", pstderr);
237
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000238 _PyImport_Init();
239
Barry Warsawf242aa02000-05-25 23:09:49 +0000240 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000241 _PyExc_Init();
Barry Warsawf242aa02000-05-25 23:09:49 +0000242
Barry Warsaw035574d1997-08-29 22:07:17 +0000243 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000244 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000245
Just van Rossum52e14d62002-12-30 22:08:05 +0000246 _PyImportHooks_Init();
247
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000248 if (install_sigs)
249 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000250
251 initmain(); /* Module __main__ */
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000252 if (initstdio() < 0)
253 Py_FatalError(
254 "Py_Initialize: can't initialize sys standard streams");
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000255 if (!Py_NoSiteFlag)
256 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000257
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000258 /* auto-thread-state API, if available */
259#ifdef WITH_THREAD
260 _PyGILState_Init(interp, tstate);
261#endif /* WITH_THREAD */
262
Mark Hammondedd07732003-07-15 23:03:55 +0000263 warnings_module = PyImport_ImportModule("warnings");
264 if (!warnings_module)
265 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000266
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000267#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000268 /* On Unix, set the file system encoding according to the
269 user's preference, if the CODESET names a well-known
270 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000271 initialized by other means. Also set the encoding of
272 stdin and stdout if these are terminals. */
273
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000274 codeset = nl_langinfo(CODESET);
275 if (codeset && *codeset) {
276 PyObject *enc = PyCodec_Encoder(codeset);
277 if (enc) {
278 codeset = strdup(codeset);
279 Py_DECREF(enc);
280 } else {
281 codeset = NULL;
282 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000283 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000284 } else
285 codeset = NULL;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000286
287 if (codeset) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000288 if (!Py_FileSystemDefaultEncoding)
289 Py_FileSystemDefaultEncoding = codeset;
290 else
291 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000292 }
293#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000294}
295
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000296void
297Py_Initialize(void)
298{
299 Py_InitializeEx(1);
300}
301
302
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000303#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000304extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000305#endif
306
Guido van Rossume8432ac2007-07-09 15:04:50 +0000307/* Flush stdout and stderr */
308
Neal Norwitz2bad9702007-08-27 06:19:22 +0000309static void
Guido van Rossum1bd21222007-07-10 20:14:13 +0000310flush_std_files(void)
Guido van Rossume8432ac2007-07-09 15:04:50 +0000311{
312 PyObject *fout = PySys_GetObject("stdout");
313 PyObject *ferr = PySys_GetObject("stderr");
314 PyObject *tmp;
315
316 if (fout != NULL) {
317 tmp = PyObject_CallMethod(fout, "flush", "");
318 if (tmp == NULL)
319 PyErr_Clear();
320 else
321 Py_DECREF(tmp);
322 }
323
324 if (ferr != NULL) {
325 tmp = PyObject_CallMethod(ferr, "flush", "");
326 if (tmp == NULL)
327 PyErr_Clear();
328 else
329 Py_DECREF(tmp);
330 }
331}
332
Guido van Rossum25ce5661997-08-02 03:10:38 +0000333/* Undo the effect of Py_Initialize().
334
335 Beware: if multiple interpreter and/or thread states exist, these
336 are not wiped out; only the current thread and interpreter state
337 are deleted. But since everything else is deleted, those other
338 interpreter and thread states should no longer be used.
339
340 (XXX We should do better, e.g. wipe out all interpreters and
341 threads.)
342
343 Locking: as above.
344
345*/
346
347void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000348Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000349{
350 PyInterpreterState *interp;
351 PyThreadState *tstate;
352
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000353 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000354 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000355
Tim Peters384fd102001-01-21 03:40:37 +0000356 /* The interpreter is still entirely intact at this point, and the
357 * exit funcs may be relying on that. In particular, if some thread
358 * or exit func is still waiting to do an import, the import machinery
359 * expects Py_IsInitialized() to return true. So don't say the
360 * interpreter is uninitialized until after the exit funcs have run.
361 * Note that Threading.py uses an exit func to do a join on all the
362 * threads created thru it, so this also protects pending imports in
363 * the threads created via Threading.
364 */
Collin Winter670e6922007-03-21 02:57:17 +0000365 call_py_exitfuncs();
Tim Peters384fd102001-01-21 03:40:37 +0000366 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000367
Guido van Rossume8432ac2007-07-09 15:04:50 +0000368 /* Flush stdout+stderr */
369 flush_std_files();
370
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000371 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000372 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000373 interp = tstate->interp;
374
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000375 /* Disable signal handling */
376 PyOS_FiniInterrupts();
377
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000378 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000379 Py_XDECREF(warnings_module);
380 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000381
Guido van Rossume13ddc92003-04-17 17:29:22 +0000382 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000383 * before all modules are destroyed.
384 * XXX If a __del__ or weakref callback is triggered here, and tries to
385 * XXX import a module, bad things can happen, because Python no
386 * XXX longer believes it's initialized.
387 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
388 * XXX is easy to provoke that way. I've also seen, e.g.,
389 * XXX Exception exceptions.ImportError: 'No module named sha'
390 * XXX in <function callback at 0x008F5718> ignored
391 * XXX but I'm unclear on exactly how that one happens. In any case,
392 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000393 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000394 PyGC_Collect();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000395#ifdef COUNT_ALLOCS
396 /* With COUNT_ALLOCS, it helps to run GC multiple times:
397 each collection might release some types from the type
398 list, so they become garbage. */
399 while (PyGC_Collect() > 0)
400 /* nothing */;
401#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000402
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000403 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000404 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000405
Guido van Rossume8432ac2007-07-09 15:04:50 +0000406 /* Flush stdout+stderr (again, in case more was printed) */
407 flush_std_files();
408
Guido van Rossume13ddc92003-04-17 17:29:22 +0000409 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000410 * new-style class definitions, for example.
411 * XXX This is disabled because it caused too many problems. If
412 * XXX a __del__ or weakref callback triggers here, Python code has
413 * XXX a hard time running, because even the sys module has been
414 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
415 * XXX One symptom is a sequence of information-free messages
416 * XXX coming from threads (if a __del__ or callback is invoked,
417 * XXX other threads can execute too, and any exception they encounter
418 * XXX triggers a comedy of errors as subsystem after subsystem
419 * XXX fails to find what it *expects* to find in sys to help report
420 * XXX the exception and consequent unexpected failures). I've also
421 * XXX seen segfaults then, after adding print statements to the
422 * XXX Python code getting called.
423 */
424#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000425 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000426#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000427
Guido van Rossum1707aad1997-12-08 23:43:45 +0000428 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
429 _PyImport_Fini();
430
431 /* Debugging stuff */
432#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000433 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000434#endif
435
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000436 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000437
Tim Peters9cf25ce2003-04-17 15:21:01 +0000438#ifdef Py_TRACE_REFS
439 /* Display all objects still alive -- this can invoke arbitrary
440 * __repr__ overrides, so requires a mostly-intact interpreter.
441 * Alas, a lot of stuff may still be alive now that will be cleaned
442 * up later.
443 */
Tim Peters269b2a62003-04-17 19:52:29 +0000444 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000445 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000446#endif /* Py_TRACE_REFS */
447
Mark Hammond6cb90292003-04-22 11:18:00 +0000448 /* Cleanup auto-thread-state */
449#ifdef WITH_THREAD
450 _PyGILState_Fini();
451#endif /* WITH_THREAD */
452
Guido van Rossumd922fa42003-04-15 14:10:09 +0000453 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000454 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000455
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000456 /* Now we decref the exception classes. After this point nothing
457 can raise an exception. That's okay, because each Fini() method
458 below has been checked to make sure no exceptions are ever
459 raised.
460 */
461
462 _PyExc_Fini();
463
Guido van Rossumd922fa42003-04-15 14:10:09 +0000464 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000465 PyThreadState_Swap(NULL);
466 PyInterpreterState_Delete(interp);
467
Guido van Rossumd922fa42003-04-15 14:10:09 +0000468 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000469 PyMethod_Fini();
470 PyFrame_Fini();
471 PyCFunction_Fini();
472 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000473 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000474 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000475 PyString_Fini();
Neal Norwitz6968b052007-02-27 19:02:19 +0000476 PyBytes_Fini();
Guido van Rossumddefaf32007-01-14 03:31:43 +0000477 PyLong_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000478 PyFloat_Fini();
479
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000480 /* Cleanup Unicode implementation */
481 _PyUnicode_Fini();
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000482
Guido van Rossumcc283f51997-08-05 02:22:03 +0000483 /* XXX Still allocated:
484 - various static ad-hoc pointers to interned strings
485 - int and float free list blocks
486 - whatever various modules and libraries allocate
487 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000488
489 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000490
Tim Peters269b2a62003-04-17 19:52:29 +0000491#ifdef Py_TRACE_REFS
492 /* Display addresses (& refcnts) of all objects still alive.
493 * An address can be used to find the repr of the object, printed
494 * above by _Py_PrintReferences.
495 */
496 if (Py_GETENV("PYTHONDUMPREFS"))
497 _Py_PrintReferenceAddresses(stderr);
498#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000499#ifdef PYMALLOC_DEBUG
500 if (Py_GETENV("PYTHONMALLOCSTATS"))
501 _PyObject_DebugMallocStats();
502#endif
503
Guido van Rossumcc283f51997-08-05 02:22:03 +0000504 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000505}
506
507/* Create and initialize a new interpreter and thread, and return the
508 new thread. This requires that Py_Initialize() has been called
509 first.
510
511 Unsuccessful initialization yields a NULL pointer. Note that *no*
512 exception information is available even in this case -- the
513 exception information is held in the thread, and there is no
514 thread.
515
516 Locking: as above.
517
518*/
519
520PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000521Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000522{
523 PyInterpreterState *interp;
524 PyThreadState *tstate, *save_tstate;
525 PyObject *bimod, *sysmod;
526
527 if (!initialized)
528 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
529
530 interp = PyInterpreterState_New();
531 if (interp == NULL)
532 return NULL;
533
534 tstate = PyThreadState_New(interp);
535 if (tstate == NULL) {
536 PyInterpreterState_Delete(interp);
537 return NULL;
538 }
539
540 save_tstate = PyThreadState_Swap(tstate);
541
542 /* XXX The following is lax in error checking */
543
544 interp->modules = PyDict_New();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000545 interp->modules_reloading = PyDict_New();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000546
547 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
548 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000549 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000550 if (interp->builtins == NULL)
551 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000552 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000553 }
554 sysmod = _PyImport_FindExtension("sys", "sys");
555 if (bimod != NULL && sysmod != NULL) {
556 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000557 if (interp->sysdict == NULL)
558 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000559 Py_INCREF(interp->sysdict);
560 PySys_SetPath(Py_GetPath());
561 PyDict_SetItemString(interp->sysdict, "modules",
562 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000563 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000564 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000565 if (!Py_NoSiteFlag)
566 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000567 }
568
569 if (!PyErr_Occurred())
570 return tstate;
571
Thomas Wouters89f507f2006-12-13 04:49:30 +0000572handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000573 /* Oops, it didn't work. Undo it all. */
574
575 PyErr_Print();
576 PyThreadState_Clear(tstate);
577 PyThreadState_Swap(save_tstate);
578 PyThreadState_Delete(tstate);
579 PyInterpreterState_Delete(interp);
580
581 return NULL;
582}
583
584/* Delete an interpreter and its last thread. This requires that the
585 given thread state is current, that the thread has no remaining
586 frames, and that it is its interpreter's only remaining thread.
587 It is a fatal error to violate these constraints.
588
589 (Py_Finalize() doesn't have these constraints -- it zaps
590 everything, regardless.)
591
592 Locking: as above.
593
594*/
595
596void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000597Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000598{
599 PyInterpreterState *interp = tstate->interp;
600
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000601 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000602 Py_FatalError("Py_EndInterpreter: thread is not current");
603 if (tstate->frame != NULL)
604 Py_FatalError("Py_EndInterpreter: thread still has a frame");
605 if (tstate != interp->tstate_head || tstate->next != NULL)
606 Py_FatalError("Py_EndInterpreter: not the last thread");
607
608 PyImport_Cleanup();
609 PyInterpreterState_Clear(interp);
610 PyThreadState_Swap(NULL);
611 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000612}
613
614static char *progname = "python";
615
616void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000617Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000618{
619 if (pn && *pn)
620 progname = pn;
621}
622
623char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000624Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000625{
626 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000627}
628
Guido van Rossuma61691e1998-02-06 22:27:24 +0000629static char *default_home = NULL;
630
631void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000632Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000633{
634 default_home = home;
635}
636
637char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000638Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000639{
640 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000641 if (home == NULL && !Py_IgnoreEnvironmentFlag)
642 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000643 return home;
644}
645
Guido van Rossum6135a871995-01-09 17:53:26 +0000646/* Create __main__ module */
647
648static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000649initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000650{
Guido van Rossum82598051997-03-05 00:20:32 +0000651 PyObject *m, *d;
652 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000653 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000654 Py_FatalError("can't create __main__ module");
655 d = PyModule_GetDict(m);
656 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000657 PyObject *bimod = PyImport_ImportModule("__builtin__");
658 if (bimod == NULL ||
659 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000660 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000661 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000662 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000663}
664
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000665/* Import the site module (not into __main__ though) */
666
667static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000668initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000669{
670 PyObject *m, *f;
671 m = PyImport_ImportModule("site");
672 if (m == NULL) {
673 f = PySys_GetObject("stderr");
674 if (Py_VerboseFlag) {
675 PyFile_WriteString(
676 "'import site' failed; traceback:\n", f);
677 PyErr_Print();
678 }
679 else {
680 PyFile_WriteString(
681 "'import site' failed; use -v for traceback\n", f);
682 PyErr_Clear();
683 }
684 }
685 else {
686 Py_DECREF(m);
687 }
688}
689
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000690/* Initialize sys.stdin, stdout, stderr and __builtin__.open */
691static int
692initstdio(void)
693{
694 PyObject *iomod = NULL, *wrapper;
695 PyObject *bimod = NULL;
696 PyObject *m;
697 PyObject *std = NULL;
698 int status = 0;
699
700 /* Hack to avoid a nasty recursion issue when Python is invoked
701 in verbose mode: pre-import the Latin-1 and UTF-8 codecs */
702 if ((m = PyImport_ImportModule("encodings.utf_8")) == NULL) {
703 goto error;
704 }
705 Py_DECREF(m);
706
707 if (!(m = PyImport_ImportModule("encodings.latin_1"))) {
708 goto error;
709 }
710 Py_DECREF(m);
711
712 if (!(bimod = PyImport_ImportModule("__builtin__"))) {
713 goto error;
714 }
715
716 if (!(iomod = PyImport_ImportModule("io"))) {
717 goto error;
718 }
719 if (!(wrapper = PyObject_GetAttrString(iomod, "OpenWrapper"))) {
720 goto error;
721 }
722
723 /* Set __builtin__.open */
724 if (PyObject_SetAttrString(bimod, "open", wrapper) == -1) {
725 goto error;
726 }
727
728 /* Set sys.stdin */
Guido van Rossum40d20bc2007-10-22 00:09:51 +0000729 if (!(std = PyFile_FromFd(fileno(stdin), "<stdin>", "r", -1,
Guido van Rossum2dced8b2007-10-30 17:27:30 +0000730 NULL, "\n", 0))) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000731 goto error;
732 }
733 PySys_SetObject("__stdin__", std);
734 PySys_SetObject("stdin", std);
735 Py_DECREF(std);
736
737 /* Set sys.stdout */
Guido van Rossum40d20bc2007-10-22 00:09:51 +0000738 if (!(std = PyFile_FromFd(fileno(stdout), "<stdout>", "w", -1,
Guido van Rossum2dced8b2007-10-30 17:27:30 +0000739 NULL, "\n", 0))) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000740 goto error;
741 }
742 PySys_SetObject("__stdout__", std);
743 PySys_SetObject("stdout", std);
744 Py_DECREF(std);
745
Guido van Rossum2dced8b2007-10-30 17:27:30 +0000746 /* Set sys.stderr, replaces the preliminary stderr */
Guido van Rossum40d20bc2007-10-22 00:09:51 +0000747 if (!(std = PyFile_FromFd(fileno(stderr), "<stderr>", "w", -1,
Guido van Rossum2dced8b2007-10-30 17:27:30 +0000748 NULL, "\n", 0))) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000749 goto error;
750 }
751 PySys_SetObject("__stderr__", std);
752 PySys_SetObject("stderr", std);
753 Py_DECREF(std);
754
755 if (0) {
756 error:
757 status = -1;
758 }
759
760 Py_XDECREF(bimod);
761 Py_XDECREF(iomod);
762 return status;
763}
764
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000765/* Parse input from a file and execute it */
766
767int
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000768PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000769 PyCompilerFlags *flags)
770{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000771 if (filename == NULL)
772 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000773 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000774 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000775 if (closeit)
776 fclose(fp);
777 return err;
778 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000779 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000780 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000781}
782
783int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000784PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000785{
Guido van Rossum82598051997-03-05 00:20:32 +0000786 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000787 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000788 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000789
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000790 if (flags == NULL) {
791 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000792 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000793 }
Guido van Rossum82598051997-03-05 00:20:32 +0000794 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000795 if (v == NULL) {
Neal Norwitza369c5a2007-08-25 07:41:59 +0000796 PySys_SetObject("ps1", v = PyUnicode_FromString(">>> "));
Guido van Rossum82598051997-03-05 00:20:32 +0000797 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000798 }
Guido van Rossum82598051997-03-05 00:20:32 +0000799 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000800 if (v == NULL) {
Neal Norwitza369c5a2007-08-25 07:41:59 +0000801 PySys_SetObject("ps2", v = PyUnicode_FromString("... "));
Guido van Rossum82598051997-03-05 00:20:32 +0000802 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000803 }
804 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000805 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000806 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000807 if (ret == E_EOF)
808 return 0;
809 /*
810 if (ret == E_NOMEM)
811 return -1;
812 */
813 }
814}
815
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000816/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000817#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000818 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Guido van Rossum45aecf42006-03-15 04:58:47 +0000819 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000820
Thomas Wouters89f507f2006-12-13 04:49:30 +0000821#if 0
822/* Keep an example of flags with future keyword support. */
823#define PARSER_FLAGS(flags) \
824 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
825 PyPARSE_DONT_IMPLY_DEDENT : 0) \
826 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
827 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
828#endif
829
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000830int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000831PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000832{
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000833 PyObject *m, *d, *v, *w, *oenc = NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000834 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000835 PyArena *arena;
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000836 char *ps1 = "", *ps2 = "", *enc = NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000837 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000838
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000839 if (fp == stdin) {
840 /* Fetch encoding from sys.stdin */
841 v = PySys_GetObject("stdin");
842 if (!v)
843 return -1;
844 oenc = PyObject_GetAttrString(v, "encoding");
845 if (!oenc)
846 return -1;
847 enc = PyUnicode_AsString(oenc);
848 }
Guido van Rossum82598051997-03-05 00:20:32 +0000849 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000850 if (v != NULL) {
Guido van Rossumc5724de2007-10-10 21:38:59 +0000851 v = PyObject_Unicode(v);
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000852 if (v == NULL)
853 PyErr_Clear();
Guido van Rossumc5724de2007-10-10 21:38:59 +0000854 else if (PyUnicode_Check(v))
855 ps1 = PyUnicode_AsString(v);
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000856 }
Guido van Rossum82598051997-03-05 00:20:32 +0000857 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000858 if (w != NULL) {
Guido van Rossumc5724de2007-10-10 21:38:59 +0000859 w = PyObject_Unicode(w);
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000860 if (w == NULL)
861 PyErr_Clear();
Guido van Rossumc5724de2007-10-10 21:38:59 +0000862 else if (PyUnicode_Check(w))
863 ps2 = PyUnicode_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000864 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000865 arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000866 if (arena == NULL) {
867 Py_XDECREF(v);
868 Py_XDECREF(w);
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000869 Py_XDECREF(oenc);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000870 return -1;
871 }
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000872 mod = PyParser_ASTFromFile(fp, filename, enc,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000873 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000874 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000875 Py_XDECREF(v);
876 Py_XDECREF(w);
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000877 Py_XDECREF(oenc);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000878 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000879 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000880 if (errcode == E_EOF) {
881 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000882 return E_EOF;
883 }
Guido van Rossum82598051997-03-05 00:20:32 +0000884 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000885 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000886 }
Guido van Rossum82598051997-03-05 00:20:32 +0000887 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000888 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000889 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000890 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000891 }
Guido van Rossum82598051997-03-05 00:20:32 +0000892 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000893 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000894 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000895 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000896 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000897 return -1;
898 }
Guido van Rossum82598051997-03-05 00:20:32 +0000899 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000900 return 0;
901}
902
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000903/* Check whether a file maybe a pyc file: Look at the extension,
904 the file type, and, if we may close it, at the first few bytes. */
905
906static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000907maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000908{
909 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
910 return 1;
911
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000912 /* Only look into the file if we are allowed to close it, since
913 it then should also be seekable. */
914 if (closeit) {
915 /* Read only two bytes of the magic. If the file was opened in
916 text mode, the bytes 3 and 4 of the magic (\r\n) might not
917 be read as they are on disk. */
918 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
919 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000920 /* Mess: In case of -x, the stream is NOT at its start now,
921 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000922 which makes the current stream position formally undefined,
923 and a x-platform nightmare.
924 Unfortunately, we have no direct way to know whether -x
925 was specified. So we use a terrible hack: if the current
926 stream position is not 0, we assume -x was specified, and
927 give up. Bug 132850 on SourceForge spells out the
928 hopelessness of trying anything else (fseek and ftell
929 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000930 */
Tim Peters3e876562001-02-11 04:35:39 +0000931 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000932 if (ftell(fp) == 0) {
933 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000934 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000935 ispyc = 1;
936 rewind(fp);
937 }
Tim Peters3e876562001-02-11 04:35:39 +0000938 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000939 }
940 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000941}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000942
Guido van Rossum0df002c2000-08-27 19:21:52 +0000943int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000944PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000945 PyCompilerFlags *flags)
946{
Guido van Rossum82598051997-03-05 00:20:32 +0000947 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000948 const char *ext;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000949 int set_file_name = 0, ret;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000950
Guido van Rossum82598051997-03-05 00:20:32 +0000951 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000952 if (m == NULL)
953 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000954 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000955 if (PyDict_GetItemString(d, "__file__") == NULL) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +0000956 PyObject *f;
957 f = PyUnicode_DecodeFSDefault(filename);
Fred Drake8ed02042002-10-17 21:24:58 +0000958 if (f == NULL)
959 return -1;
960 if (PyDict_SetItemString(d, "__file__", f) < 0) {
961 Py_DECREF(f);
962 return -1;
963 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000964 set_file_name = 1;
Fred Drake8ed02042002-10-17 21:24:58 +0000965 Py_DECREF(f);
966 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000967 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000968 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000969 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000970 if (closeit)
971 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000972 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000973 fprintf(stderr, "python: Can't reopen .pyc file\n");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000974 ret = -1;
975 goto done;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000976 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000977 /* Turn on optimization if a .pyo file is given */
978 if (strcmp(ext, ".pyo") == 0)
979 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000980 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000981 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000982 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000983 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000984 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000985 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000986 PyErr_Print();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000987 ret = -1;
988 goto done;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000989 }
Guido van Rossum82598051997-03-05 00:20:32 +0000990 Py_DECREF(v);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000991 ret = 0;
992 done:
993 if (set_file_name && PyDict_DelItemString(d, "__file__"))
994 PyErr_Clear();
995 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000996}
997
998int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000999PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +00001000{
Guido van Rossum82598051997-03-05 00:20:32 +00001001 PyObject *m, *d, *v;
1002 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001003 if (m == NULL)
1004 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +00001005 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +00001006 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001007 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +00001008 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001009 return -1;
1010 }
Guido van Rossum82598051997-03-05 00:20:32 +00001011 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001012 return 0;
1013}
1014
Barry Warsaw035574d1997-08-29 22:07:17 +00001015static int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001016parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
1017 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +00001018{
1019 long hold;
1020 PyObject *v;
1021
1022 /* old style errors */
1023 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +00001024 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001025 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +00001026
1027 /* new style errors. `err' is an instance */
1028
1029 if (! (v = PyObject_GetAttrString(err, "msg")))
1030 goto finally;
1031 *message = v;
1032
1033 if (!(v = PyObject_GetAttrString(err, "filename")))
1034 goto finally;
1035 if (v == Py_None)
1036 *filename = NULL;
1037 else if (! (*filename = PyString_AsString(v)))
1038 goto finally;
1039
1040 Py_DECREF(v);
1041 if (!(v = PyObject_GetAttrString(err, "lineno")))
1042 goto finally;
1043 hold = PyInt_AsLong(v);
1044 Py_DECREF(v);
1045 v = NULL;
1046 if (hold < 0 && PyErr_Occurred())
1047 goto finally;
1048 *lineno = (int)hold;
1049
1050 if (!(v = PyObject_GetAttrString(err, "offset")))
1051 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001052 if (v == Py_None) {
1053 *offset = -1;
1054 Py_DECREF(v);
1055 v = NULL;
1056 } else {
1057 hold = PyInt_AsLong(v);
1058 Py_DECREF(v);
1059 v = NULL;
1060 if (hold < 0 && PyErr_Occurred())
1061 goto finally;
1062 *offset = (int)hold;
1063 }
Barry Warsaw035574d1997-08-29 22:07:17 +00001064
1065 if (!(v = PyObject_GetAttrString(err, "text")))
1066 goto finally;
1067 if (v == Py_None)
1068 *text = NULL;
Guido van Rossumc5724de2007-10-10 21:38:59 +00001069 else if (!PyUnicode_Check(v) ||
1070 !(*text = PyUnicode_AsString(v)))
Barry Warsaw035574d1997-08-29 22:07:17 +00001071 goto finally;
1072 Py_DECREF(v);
1073 return 1;
1074
1075finally:
1076 Py_XDECREF(v);
1077 return 0;
1078}
1079
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001080void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001081PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001082{
Guido van Rossuma61691e1998-02-06 22:27:24 +00001083 PyErr_PrintEx(1);
1084}
1085
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001086static void
Martin v. Löwis95292d62002-12-11 14:04:59 +00001087print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001088{
1089 char *nl;
1090 if (offset >= 0) {
1091 if (offset > 0 && offset == (int)strlen(text))
1092 offset--;
1093 for (;;) {
1094 nl = strchr(text, '\n');
1095 if (nl == NULL || nl-text >= offset)
1096 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001097 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001098 text = nl+1;
1099 }
1100 while (*text == ' ' || *text == '\t') {
1101 text++;
1102 offset--;
1103 }
1104 }
1105 PyFile_WriteString(" ", f);
1106 PyFile_WriteString(text, f);
1107 if (*text == '\0' || text[strlen(text)-1] != '\n')
1108 PyFile_WriteString("\n", f);
1109 if (offset == -1)
1110 return;
1111 PyFile_WriteString(" ", f);
1112 offset--;
1113 while (offset > 0) {
1114 PyFile_WriteString(" ", f);
1115 offset--;
1116 }
1117 PyFile_WriteString("^\n", f);
1118}
1119
Guido van Rossum66e8e862001-03-23 17:54:43 +00001120static void
1121handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001122{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001123 PyObject *exception, *value, *tb;
1124 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001125
Guido van Rossumd8faa362007-04-27 19:54:29 +00001126 if (Py_InspectFlag)
1127 /* Don't exit if -i flag was given. This flag is set to 0
1128 * when entering interactive mode for inspecting. */
1129 return;
1130
Guido van Rossum66e8e862001-03-23 17:54:43 +00001131 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001132 fflush(stdout);
1133 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001134 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001135 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001136 /* The error code should be in the `code' attribute. */
1137 PyObject *code = PyObject_GetAttrString(value, "code");
1138 if (code) {
1139 Py_DECREF(value);
1140 value = code;
1141 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001142 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001143 }
1144 /* If we failed to dig out the 'code' attribute,
1145 just let the else clause below print the error. */
1146 }
1147 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001148 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001149 else {
1150 PyObject_Print(value, stderr, Py_PRINT_RAW);
1151 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001152 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001153 }
Tim Peterscf615b52003-04-19 18:47:02 +00001154 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001155 /* Restore and clear the exception info, in order to properly decref
1156 * the exception, value, and traceback. If we just exit instead,
1157 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1158 * some finalizers from running.
1159 */
Tim Peterscf615b52003-04-19 18:47:02 +00001160 PyErr_Restore(exception, value, tb);
1161 PyErr_Clear();
1162 Py_Exit(exitcode);
1163 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001164}
1165
1166void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001167PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001168{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001169 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001170
1171 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1172 handle_system_exit();
1173 }
Guido van Rossum82598051997-03-05 00:20:32 +00001174 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001175 if (exception == NULL)
1176 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001177 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001178 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001179 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001180 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001181 if (set_sys_last_vars) {
1182 PySys_SetObject("last_type", exception);
1183 PySys_SetObject("last_value", v);
Guido van Rossumc5724de2007-10-10 21:38:59 +00001184 PySys_SetObject("last_traceback", tb ? tb : Py_None);
Guido van Rossuma61691e1998-02-06 22:27:24 +00001185 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001186 hook = PySys_GetObject("excepthook");
1187 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001188 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001189 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001190 PyObject *result = PyEval_CallObject(hook, args);
1191 if (result == NULL) {
1192 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001193 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1194 handle_system_exit();
1195 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001196 PyErr_Fetch(&exception2, &v2, &tb2);
1197 PyErr_NormalizeException(&exception2, &v2, &tb2);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001198 /* It should not be possible for exception2 or v2
1199 to be NULL. However PyErr_Display() can't
1200 tolerate NULLs, so just be safe. */
1201 if (exception2 == NULL) {
1202 exception2 = Py_None;
1203 Py_INCREF(exception2);
1204 }
1205 if (v2 == NULL) {
1206 v2 = Py_None;
1207 Py_INCREF(v2);
1208 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001209 fflush(stdout);
1210 PySys_WriteStderr("Error in sys.excepthook:\n");
1211 PyErr_Display(exception2, v2, tb2);
1212 PySys_WriteStderr("\nOriginal exception was:\n");
1213 PyErr_Display(exception, v, tb);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001214 Py_DECREF(exception2);
1215 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001216 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001217 }
1218 Py_XDECREF(result);
1219 Py_XDECREF(args);
1220 } else {
1221 PySys_WriteStderr("sys.excepthook is missing\n");
1222 PyErr_Display(exception, v, tb);
1223 }
1224 Py_XDECREF(exception);
1225 Py_XDECREF(v);
1226 Py_XDECREF(tb);
1227}
1228
Thomas Wouters477c8d52006-05-27 19:21:47 +00001229void
1230PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001231{
1232 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001233 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001234 Py_INCREF(value);
Guido van Rossumce3a72a2007-10-19 23:16:50 +00001235 if (f == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00001236 _PyObject_Dump(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001237 fprintf(stderr, "lost sys.stderr\n");
Guido van Rossumce3a72a2007-10-19 23:16:50 +00001238 }
Guido van Rossum3165fe61992-09-25 21:59:05 +00001239 else {
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001240 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001241 if (tb && tb != Py_None)
1242 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001243 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001244 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001245 {
Guido van Rossum82598051997-03-05 00:20:32 +00001246 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001247 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001248 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001249 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001250 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001251 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001252 else {
1253 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001254 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001255 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001256 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001257 else
Guido van Rossum82598051997-03-05 00:20:32 +00001258 PyFile_WriteString(filename, f);
1259 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001260 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001261 PyFile_WriteString(buf, f);
1262 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001263 if (text != NULL)
1264 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001265 Py_DECREF(value);
1266 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001267 /* Can't be bothered to check all those
1268 PyFile_WriteString() calls */
1269 if (PyErr_Occurred())
1270 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001271 }
1272 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001273 if (err) {
1274 /* Don't do anything else */
1275 }
Brett Cannonbf364092006-03-01 04:25:17 +00001276 else if (PyExceptionClass_Check(exception)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001277 PyObject* moduleName;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001278 char* className = PyExceptionClass_Name(exception);
1279 if (className != NULL) {
1280 char *dot = strrchr(className, '.');
1281 if (dot != NULL)
1282 className = dot+1;
1283 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001284
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001285 moduleName = PyObject_GetAttrString(exception, "__module__");
Guido van Rossumc5724de2007-10-10 21:38:59 +00001286 if (moduleName == NULL || !PyUnicode_Check(moduleName))
1287 {
1288 Py_DECREF(moduleName);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001289 err = PyFile_WriteString("<unknown>", f);
Guido van Rossumc5724de2007-10-10 21:38:59 +00001290 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001291 else {
Guido van Rossumc5724de2007-10-10 21:38:59 +00001292 char* modstr = PyUnicode_AsString(moduleName);
Neal Norwitz2633c692007-02-26 22:22:47 +00001293 if (modstr && strcmp(modstr, "__builtin__"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001294 {
1295 err = PyFile_WriteString(modstr, f);
1296 err += PyFile_WriteString(".", f);
1297 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001298 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001299 }
1300 if (err == 0) {
1301 if (className == NULL)
1302 err = PyFile_WriteString("<unknown>", f);
1303 else
Brett Cannonbf364092006-03-01 04:25:17 +00001304 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001305 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001306 }
1307 else
1308 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001309 if (err == 0 && (value != Py_None)) {
Guido van Rossumc5724de2007-10-10 21:38:59 +00001310 PyObject *s = PyObject_Unicode(value);
Brett Cannon2e63b732006-03-02 18:34:57 +00001311 /* only print colon if the str() of the
1312 object is not the empty string
1313 */
1314 if (s == NULL)
1315 err = -1;
Guido van Rossumc5724de2007-10-10 21:38:59 +00001316 else if (!PyUnicode_Check(s) ||
1317 PyUnicode_GetSize(s) != 0)
Brett Cannon2e63b732006-03-02 18:34:57 +00001318 err = PyFile_WriteString(": ", f);
1319 if (err == 0)
1320 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1321 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001322 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00001323 /* try to write a newline in any case */
1324 err += PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001325 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001326 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001327 /* If an error happened here, don't show it.
1328 XXX This is wrong, but too many callers rely on this behavior. */
1329 if (err != 0)
1330 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001331}
1332
Guido van Rossum82598051997-03-05 00:20:32 +00001333PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001334PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001335 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001336{
Neal Norwitze92fba02006-03-04 18:52:26 +00001337 PyObject *ret = NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001338 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001339 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001340 if (arena == NULL)
1341 return NULL;
1342
1343 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001344 if (mod != NULL)
1345 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001346 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001347 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001348}
1349
1350PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001351PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001352 PyObject *locals, int closeit, PyCompilerFlags *flags)
1353{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001354 PyObject *ret;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001355 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001356 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001357 if (arena == NULL)
1358 return NULL;
1359
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001360 mod = PyParser_ASTFromFile(fp, filename, NULL, start, 0, 0,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001361 flags, NULL, arena);
Guido van Rossumd8faa362007-04-27 19:54:29 +00001362 if (closeit)
1363 fclose(fp);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001364 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001365 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001366 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001367 }
Neal Norwitze92fba02006-03-04 18:52:26 +00001368 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001369 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001370 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001371}
1372
Guido van Rossum82598051997-03-05 00:20:32 +00001373static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001374run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001375 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001376{
Guido van Rossum82598051997-03-05 00:20:32 +00001377 PyCodeObject *co;
1378 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001379 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001380 if (co == NULL)
1381 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001382 v = PyEval_EvalCode(co, globals, locals);
1383 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001384 return v;
1385}
1386
Guido van Rossum82598051997-03-05 00:20:32 +00001387static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001388run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001389 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001390{
Guido van Rossum82598051997-03-05 00:20:32 +00001391 PyCodeObject *co;
1392 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001393 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001394 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001395
Guido van Rossum82598051997-03-05 00:20:32 +00001396 magic = PyMarshal_ReadLongFromFile(fp);
1397 if (magic != PyImport_GetMagicNumber()) {
1398 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001399 "Bad magic number in .pyc file");
1400 return NULL;
1401 }
Guido van Rossum82598051997-03-05 00:20:32 +00001402 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001403 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001404 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001405 if (v == NULL || !PyCode_Check(v)) {
1406 Py_XDECREF(v);
1407 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001408 "Bad code object in .pyc file");
1409 return NULL;
1410 }
Guido van Rossum82598051997-03-05 00:20:32 +00001411 co = (PyCodeObject *)v;
1412 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001413 if (v && flags)
1414 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001415 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001416 return v;
1417}
1418
Guido van Rossum82598051997-03-05 00:20:32 +00001419PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001420Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001421 PyCompilerFlags *flags)
1422{
Guido van Rossum82598051997-03-05 00:20:32 +00001423 PyCodeObject *co;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001424 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001425 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001426 if (arena == NULL)
1427 return NULL;
1428
1429 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001430 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001431 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001432 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001433 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001434 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001435 PyObject *result = PyAST_mod2obj(mod);
1436 PyArena_Free(arena);
1437 return result;
1438 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001439 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001440 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001441 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001442}
1443
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001444struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001445Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001446{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001447 struct symtable *st;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001448 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001449 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001450 if (arena == NULL)
1451 return NULL;
1452
1453 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001454 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001455 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001456 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001457 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001458 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001459 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001460 return st;
1461}
1462
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001463/* Preferred access to parser is through AST. */
1464mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001465PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001466 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001467{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001468 mod_ty mod;
1469 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001470 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001471 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001472 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001473 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001474 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001475 PyNode_Free(n);
1476 return mod;
1477 }
1478 else {
1479 err_input(&err);
1480 return NULL;
1481 }
1482}
1483
1484mod_ty
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001485PyParser_ASTFromFile(FILE *fp, const char *filename, const char* enc,
1486 int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001487 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001488 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001489{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001490 mod_ty mod;
1491 perrdetail err;
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001492 node *n = PyParser_ParseFileFlags(fp, filename, enc,
1493 &_PyParser_Grammar,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001494 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001495 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001496 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001497 PyNode_Free(n);
1498 return mod;
1499 }
1500 else {
1501 err_input(&err);
1502 if (errcode)
1503 *errcode = err.error;
1504 return NULL;
1505 }
1506}
1507
Guido van Rossuma110aa61994-08-29 12:50:44 +00001508/* Simplified interface to parsefile -- 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_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001512{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001513 perrdetail err;
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001514 node *n = PyParser_ParseFileFlags(fp, filename, NULL,
1515 &_PyParser_Grammar,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001516 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001517 if (n == NULL)
1518 err_input(&err);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001519
Guido van Rossuma110aa61994-08-29 12:50:44 +00001520 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001521}
1522
Guido van Rossuma110aa61994-08-29 12:50:44 +00001523/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001524
Guido van Rossuma110aa61994-08-29 12:50:44 +00001525node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001526PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001527{
Tim Petersfe2127d2001-07-16 05:37:24 +00001528 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001529 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1530 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001531 if (n == NULL)
1532 err_input(&err);
1533 return n;
1534}
1535
1536node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001537PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001538 int start, int flags)
1539{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001540 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001541 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1542 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001543 if (n == NULL)
1544 err_input(&err);
1545 return n;
1546}
1547
1548node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001549PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001550{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001551 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001552}
1553
Guido van Rossum66ebd912003-04-17 16:02:26 +00001554/* May want to move a more generalized form of this to parsetok.c or
1555 even parser modules. */
1556
1557void
1558PyParser_SetError(perrdetail *err)
1559{
1560 err_input(err);
1561}
1562
Guido van Rossuma110aa61994-08-29 12:50:44 +00001563/* Set the error appropriate to the given input error code (see errcode.h) */
1564
1565static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001566err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001567{
Martin v. Löwis5deb2102007-08-31 11:17:42 +00001568 PyObject *v, *w, *errtype, *errtext;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001569 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001570 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001571 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001572 switch (err->error) {
1573 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001574 errtype = PyExc_IndentationError;
1575 if (err->expected == INDENT)
1576 msg = "expected an indented block";
1577 else if (err->token == INDENT)
1578 msg = "unexpected indent";
1579 else if (err->token == DEDENT)
1580 msg = "unexpected unindent";
1581 else {
1582 errtype = PyExc_SyntaxError;
1583 msg = "invalid syntax";
1584 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001585 break;
1586 case E_TOKEN:
1587 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001588 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001589 case E_EOFS:
1590 msg = "EOF while scanning triple-quoted string";
1591 break;
1592 case E_EOLS:
1593 msg = "EOL while scanning single-quoted string";
1594 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001595 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001596 if (!PyErr_Occurred())
1597 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001598 return;
1599 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001600 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001601 return;
1602 case E_EOF:
1603 msg = "unexpected EOF while parsing";
1604 break;
Fred Drake85f36392000-07-11 17:53:00 +00001605 case E_TABSPACE:
1606 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001607 msg = "inconsistent use of tabs and spaces in indentation";
1608 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001609 case E_OVERFLOW:
1610 msg = "expression too long";
1611 break;
Fred Drake85f36392000-07-11 17:53:00 +00001612 case E_DEDENT:
1613 errtype = PyExc_IndentationError;
1614 msg = "unindent does not match any outer indentation level";
1615 break;
1616 case E_TOODEEP:
1617 errtype = PyExc_IndentationError;
1618 msg = "too many levels of indentation";
1619 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001620 case E_DECODE: {
1621 PyObject *type, *value, *tb;
1622 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001623 if (value != NULL) {
Guido van Rossumc5724de2007-10-10 21:38:59 +00001624 u = PyObject_Unicode(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001625 if (u != NULL) {
Guido van Rossumc5724de2007-10-10 21:38:59 +00001626 msg = PyUnicode_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001627 }
1628 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001629 if (msg == NULL)
1630 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001631 Py_XDECREF(type);
1632 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001633 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001634 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001635 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001636 case E_LINECONT:
1637 msg = "unexpected character after line continuation character";
1638 break;
Martin v. Löwis47383402007-08-15 07:32:56 +00001639
1640 case E_IDENTIFIER:
1641 msg = "invalid character in identifier";
1642 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001643 default:
1644 fprintf(stderr, "error=%d\n", err->error);
1645 msg = "unknown parsing error";
1646 break;
1647 }
Martin v. Löwis5deb2102007-08-31 11:17:42 +00001648 /* err->text may not be UTF-8 in case of decoding errors.
1649 Explicitly convert to an object. */
1650 if (!err->text) {
1651 errtext = Py_None;
1652 Py_INCREF(Py_None);
1653 } else {
1654 errtext = PyUnicode_DecodeUTF8(err->text, strlen(err->text),
1655 "replace");
1656 }
1657 v = Py_BuildValue("(ziiN)", err->filename,
1658 err->lineno, err->offset, errtext);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001659 if (err->text != NULL) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001660 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001661 err->text = NULL;
1662 }
1663 w = NULL;
1664 if (v != NULL)
1665 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001666 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001667 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001668 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001669 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001670}
1671
1672/* Print fatal error message and abort */
1673
1674void
Tim Peters7c321a82002-07-09 02:57:01 +00001675Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001676{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001677 fprintf(stderr, "Fatal Python error: %s\n", msg);
Guido van Rossumce3a72a2007-10-19 23:16:50 +00001678 if (PyErr_Occurred()) {
1679 PyErr_Print();
1680 }
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001681#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001682 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001683 OutputDebugString(msg);
1684 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001685#ifdef _DEBUG
1686 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001687#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001688#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001689 abort();
1690}
1691
1692/* Clean up and exit */
1693
Guido van Rossuma110aa61994-08-29 12:50:44 +00001694#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001695#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001696#endif
1697
Collin Winter670e6922007-03-21 02:57:17 +00001698static void (*pyexitfunc)(void) = NULL;
1699/* For the atexit module. */
1700void _Py_PyAtExit(void (*func)(void))
1701{
1702 pyexitfunc = func;
1703}
1704
1705static void
1706call_py_exitfuncs(void)
1707{
1708 if (pyexitfunc == NULL)
1709 return;
1710
1711 (*pyexitfunc)();
1712 PyErr_Clear();
1713}
1714
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001715#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001716static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001717static int nexitfuncs = 0;
1718
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001719int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001720{
1721 if (nexitfuncs >= NEXITFUNCS)
1722 return -1;
1723 exitfuncs[nexitfuncs++] = func;
1724 return 0;
1725}
1726
Guido van Rossumcc283f51997-08-05 02:22:03 +00001727static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001728call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001729{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001730 while (nexitfuncs > 0)
1731 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001732
1733 fflush(stdout);
1734 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001735}
1736
1737void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001738Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001739{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001740 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001741
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001742 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001743}
1744
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001745static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001746initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001747{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001748#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001749 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001750#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001751#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001752 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001753#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001754#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001755 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001756#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001757 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001758}
1759
Guido van Rossum7433b121997-02-14 19:45:36 +00001760
1761/*
1762 * The file descriptor fd is considered ``interactive'' if either
1763 * a) isatty(fd) is TRUE, or
1764 * b) the -i flag was given, and the filename associated with
1765 * the descriptor is NULL or "<stdin>" or "???".
1766 */
1767int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001768Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001769{
1770 if (isatty((int)fileno(fp)))
1771 return 1;
1772 if (!Py_InteractiveFlag)
1773 return 0;
1774 return (filename == NULL) ||
1775 (strcmp(filename, "<stdin>") == 0) ||
1776 (strcmp(filename, "???") == 0);
1777}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001778
1779
Tim Petersd08e3822003-04-17 15:24:21 +00001780#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001781#if defined(WIN32) && defined(_MSC_VER)
1782
1783/* Stack checking for Microsoft C */
1784
1785#include <malloc.h>
1786#include <excpt.h>
1787
Fred Drakee8de31c2000-08-31 05:38:39 +00001788/*
1789 * Return non-zero when we run out of memory on the stack; zero otherwise.
1790 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001791int
Fred Drake399739f2000-08-31 05:52:44 +00001792PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001793{
1794 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001795 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001796 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001797 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001798 return 0;
1799 } __except (EXCEPTION_EXECUTE_HANDLER) {
1800 /* just ignore all errors */
1801 }
1802 return 1;
1803}
1804
1805#endif /* WIN32 && _MSC_VER */
1806
1807/* Alternate implementations can be added here... */
1808
1809#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001810
1811
1812/* Wrappers around sigaction() or signal(). */
1813
1814PyOS_sighandler_t
1815PyOS_getsig(int sig)
1816{
1817#ifdef HAVE_SIGACTION
1818 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001819 if (sigaction(sig, NULL, &context) == -1)
1820 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001821 return context.sa_handler;
1822#else
1823 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001824/* Special signal handling for the secure CRT in Visual Studio 2005 */
1825#if defined(_MSC_VER) && _MSC_VER >= 1400
1826 switch (sig) {
1827 /* Only these signals are valid */
1828 case SIGINT:
1829 case SIGILL:
1830 case SIGFPE:
1831 case SIGSEGV:
1832 case SIGTERM:
1833 case SIGBREAK:
1834 case SIGABRT:
1835 break;
1836 /* Don't call signal() with other values or it will assert */
1837 default:
1838 return SIG_ERR;
1839 }
1840#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001841 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001842 if (handler != SIG_ERR)
1843 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001844 return handler;
1845#endif
1846}
1847
1848PyOS_sighandler_t
1849PyOS_setsig(int sig, PyOS_sighandler_t handler)
1850{
1851#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001852 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001853 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001854 sigemptyset(&context.sa_mask);
1855 context.sa_flags = 0;
1856 if (sigaction(sig, &context, &ocontext) == -1)
1857 return SIG_ERR;
1858 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001859#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001860 PyOS_sighandler_t oldhandler;
1861 oldhandler = signal(sig, handler);
1862#ifdef HAVE_SIGINTERRUPT
1863 siginterrupt(sig, 1);
1864#endif
1865 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001866#endif
1867}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001868
1869/* Deprecated C API functions still provided for binary compatiblity */
1870
1871#undef PyParser_SimpleParseFile
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001872PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001873PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1874{
1875 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1876}
1877
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001878#undef PyParser_SimpleParseString
1879PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001880PyParser_SimpleParseString(const char *str, int start)
1881{
1882 return PyParser_SimpleParseStringFlags(str, start, 0);
1883}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001884
1885#undef PyRun_AnyFile
1886PyAPI_FUNC(int)
1887PyRun_AnyFile(FILE *fp, const char *name)
1888{
1889 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1890}
1891
1892#undef PyRun_AnyFileEx
1893PyAPI_FUNC(int)
1894PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1895{
1896 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1897}
1898
1899#undef PyRun_AnyFileFlags
1900PyAPI_FUNC(int)
1901PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1902{
1903 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1904}
1905
1906#undef PyRun_File
1907PyAPI_FUNC(PyObject *)
1908PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1909{
1910 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1911}
1912
1913#undef PyRun_FileEx
1914PyAPI_FUNC(PyObject *)
1915PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1916{
1917 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1918}
1919
1920#undef PyRun_FileFlags
1921PyAPI_FUNC(PyObject *)
1922PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1923 PyCompilerFlags *flags)
1924{
1925 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1926}
1927
1928#undef PyRun_SimpleFile
1929PyAPI_FUNC(int)
1930PyRun_SimpleFile(FILE *f, const char *p)
1931{
1932 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1933}
1934
1935#undef PyRun_SimpleFileEx
1936PyAPI_FUNC(int)
1937PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1938{
1939 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1940}
1941
1942
1943#undef PyRun_String
1944PyAPI_FUNC(PyObject *)
1945PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1946{
1947 return PyRun_StringFlags(str, s, g, l, NULL);
1948}
1949
1950#undef PyRun_SimpleString
1951PyAPI_FUNC(int)
1952PyRun_SimpleString(const char *s)
1953{
1954 return PyRun_SimpleStringFlags(s, NULL);
1955}
1956
1957#undef Py_CompileString
1958PyAPI_FUNC(PyObject *)
1959Py_CompileString(const char *str, const char *p, int s)
1960{
1961 return Py_CompileStringFlags(str, p, s, NULL);
1962}
1963
1964#undef PyRun_InteractiveOne
1965PyAPI_FUNC(int)
1966PyRun_InteractiveOne(FILE *f, const char *p)
1967{
1968 return PyRun_InteractiveOneFlags(f, p, NULL);
1969}
1970
1971#undef PyRun_InteractiveLoop
1972PyAPI_FUNC(int)
1973PyRun_InteractiveLoop(FILE *f, const char *p)
1974{
1975 return PyRun_InteractiveLoopFlags(f, p, NULL);
1976}
1977
1978#ifdef __cplusplus
1979}
1980#endif