blob: d89f5f95812bcd400fe73dc479d7c098e5026d56 [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);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000054static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000055 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000056static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000057 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000058static void err_input(perrdetail *);
59static void initsigs(void);
Collin Winter670e6922007-03-21 02:57:17 +000060static void call_py_exitfuncs(void);
Tim Petersdbd9ba62000-07-09 03:09:57 +000061static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000062extern void _PyUnicode_Init(void);
63extern void _PyUnicode_Fini(void);
Guido van Rossumddefaf32007-01-14 03:31:43 +000064extern int _PyLong_Init(void);
65extern void PyLong_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000066
Mark Hammond8d98d2c2003-04-19 15:41:53 +000067#ifdef WITH_THREAD
68extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
69extern void _PyGILState_Fini(void);
70#endif /* WITH_THREAD */
71
Guido van Rossum82598051997-03-05 00:20:32 +000072int Py_DebugFlag; /* Needed by parser.c */
73int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000074int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumd8faa362007-04-27 19:54:29 +000075int Py_InspectFlag; /* Needed to determine whether to exit at SystemError */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000076int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000077int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000078int Py_FrozenFlag; /* Needed by getpath.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000079int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000080
Mark Hammonda43fd0c2003-02-19 00:33:33 +000081/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000082 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000083*/
Mark Hammondedd07732003-07-15 23:03:55 +000084static PyObject *warnings_module = NULL;
85
86/* Returns a borrowed reference to the 'warnings' module, or NULL.
87 If the module is returned, it is guaranteed to have been obtained
88 without acquiring the import lock
89*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000090PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000091{
92 PyObject *typ, *val, *tb;
93 PyObject *all_modules;
94 /* If we managed to get the module at init time, just use it */
95 if (warnings_module)
96 return warnings_module;
97 /* If it wasn't available at init time, it may be available
98 now in sys.modules (common scenario is frozen apps: import
99 at init time fails, but the frozen init code sets up sys.path
100 correctly, then does an implicit import of warnings for us
101 */
102 /* Save and restore any exceptions */
103 PyErr_Fetch(&typ, &val, &tb);
104
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000105 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000106 if (all_modules) {
107 warnings_module = PyDict_GetItemString(all_modules, "warnings");
108 /* We keep a ref in the global */
109 Py_XINCREF(warnings_module);
110 }
111 PyErr_Restore(typ, val, tb);
112 return warnings_module;
113}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000114
Guido van Rossum25ce5661997-08-02 03:10:38 +0000115static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000116
Thomas Wouters7e474022000-07-16 12:04:32 +0000117/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000118
119int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000120Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000121{
122 return initialized;
123}
124
Guido van Rossum25ce5661997-08-02 03:10:38 +0000125/* Global initializations. Can be undone by Py_Finalize(). Don't
126 call this twice without an intervening Py_Finalize() call. When
127 initializations fail, a fatal error is issued and the function does
128 not return. On return, the first thread and interpreter state have
129 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000130
Guido van Rossum25ce5661997-08-02 03:10:38 +0000131 Locking: you must hold the interpreter lock while calling this.
132 (If the lock has not yet been initialized, that's equivalent to
133 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000134
Guido van Rossum25ce5661997-08-02 03:10:38 +0000135*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000136
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000137static int
138add_flag(int flag, const char *envs)
139{
140 int env = atoi(envs);
141 if (flag < env)
142 flag = env;
143 if (flag < 1)
144 flag = 1;
145 return flag;
146}
147
Guido van Rossuma027efa1997-05-05 20:56:21 +0000148void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000149Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000150{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000151 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000152 PyThreadState *tstate;
153 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000154 char *p;
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000155#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000156 char *codeset;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000157#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000158 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000159
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000160 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000161 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000162 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000163
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +0000164#ifdef HAVE_SETLOCALE
165 /* Set up the LC_CTYPE locale, so we can obtain
166 the locale's charset without having to switch
167 locales. */
168 setlocale(LC_CTYPE, "");
169#endif
170
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000171 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000172 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000173 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000174 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000175 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000176 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000177
Guido van Rossuma027efa1997-05-05 20:56:21 +0000178 interp = PyInterpreterState_New();
179 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000180 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000181
Guido van Rossuma027efa1997-05-05 20:56:21 +0000182 tstate = PyThreadState_New(interp);
183 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000184 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000185 (void) PyThreadState_Swap(tstate);
186
Guido van Rossum70d893a2001-08-16 08:21:42 +0000187 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000188
Neal Norwitzb2501f42002-12-31 03:42:13 +0000189 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000190 Py_FatalError("Py_Initialize: can't init frames");
191
Guido van Rossumddefaf32007-01-14 03:31:43 +0000192 if (!_PyLong_Init())
193 Py_FatalError("Py_Initialize: can't init longs");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000194
Neal Norwitz6968b052007-02-27 19:02:19 +0000195 if (!PyBytes_Init())
196 Py_FatalError("Py_Initialize: can't init bytes");
197
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000198 _PyFloat_Init();
199
Guido van Rossum25ce5661997-08-02 03:10:38 +0000200 interp->modules = PyDict_New();
201 if (interp->modules == NULL)
202 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000203 interp->modules_reloading = PyDict_New();
204 if (interp->modules_reloading == NULL)
205 Py_FatalError("Py_Initialize: can't make modules_reloading dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000206
Guido van Rossumc94044c2000-03-10 23:03:54 +0000207 /* Init Unicode implementation; relies on the codec registry */
208 _PyUnicode_Init();
209
Barry Warsawf242aa02000-05-25 23:09:49 +0000210 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000211 if (bimod == NULL)
212 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000213 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000214 if (interp->builtins == NULL)
215 Py_FatalError("Py_Initialize: can't initialize builtins dict");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000216 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000217
218 sysmod = _PySys_Init();
219 if (sysmod == NULL)
220 Py_FatalError("Py_Initialize: can't initialize sys");
221 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000222 if (interp->sysdict == NULL)
223 Py_FatalError("Py_Initialize: can't initialize sys dict");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000224 Py_INCREF(interp->sysdict);
225 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000226 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000227 PyDict_SetItemString(interp->sysdict, "modules",
228 interp->modules);
229
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000230 _PyImport_Init();
231
Barry Warsawf242aa02000-05-25 23:09:49 +0000232 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000233 _PyExc_Init();
Barry Warsawf242aa02000-05-25 23:09:49 +0000234
Barry Warsaw035574d1997-08-29 22:07:17 +0000235 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000236 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000237
Just van Rossum52e14d62002-12-30 22:08:05 +0000238 _PyImportHooks_Init();
239
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000240 if (install_sigs)
241 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000242
243 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000244 if (!Py_NoSiteFlag)
245 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000246
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000247 /* auto-thread-state API, if available */
248#ifdef WITH_THREAD
249 _PyGILState_Init(interp, tstate);
250#endif /* WITH_THREAD */
251
Mark Hammondedd07732003-07-15 23:03:55 +0000252 warnings_module = PyImport_ImportModule("warnings");
253 if (!warnings_module)
254 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000255
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000256#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000257 /* On Unix, set the file system encoding according to the
258 user's preference, if the CODESET names a well-known
259 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000260 initialized by other means. Also set the encoding of
261 stdin and stdout if these are terminals. */
262
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000263 codeset = nl_langinfo(CODESET);
264 if (codeset && *codeset) {
265 PyObject *enc = PyCodec_Encoder(codeset);
266 if (enc) {
267 codeset = strdup(codeset);
268 Py_DECREF(enc);
269 } else {
270 codeset = NULL;
271 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000272 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000273 } else
274 codeset = NULL;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000275
276 if (codeset) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000277 if (!Py_FileSystemDefaultEncoding)
278 Py_FileSystemDefaultEncoding = codeset;
279 else
280 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000281 }
282#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000283}
284
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000285void
286Py_Initialize(void)
287{
288 Py_InitializeEx(1);
289}
290
291
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000292#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000293extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000294#endif
295
Guido van Rossume8432ac2007-07-09 15:04:50 +0000296/* Flush stdout and stderr */
297
Neal Norwitz2bad9702007-08-27 06:19:22 +0000298static void
Guido van Rossum1bd21222007-07-10 20:14:13 +0000299flush_std_files(void)
Guido van Rossume8432ac2007-07-09 15:04:50 +0000300{
301 PyObject *fout = PySys_GetObject("stdout");
302 PyObject *ferr = PySys_GetObject("stderr");
303 PyObject *tmp;
304
305 if (fout != NULL) {
306 tmp = PyObject_CallMethod(fout, "flush", "");
307 if (tmp == NULL)
308 PyErr_Clear();
309 else
310 Py_DECREF(tmp);
311 }
312
313 if (ferr != NULL) {
314 tmp = PyObject_CallMethod(ferr, "flush", "");
315 if (tmp == NULL)
316 PyErr_Clear();
317 else
318 Py_DECREF(tmp);
319 }
320}
321
Guido van Rossum25ce5661997-08-02 03:10:38 +0000322/* Undo the effect of Py_Initialize().
323
324 Beware: if multiple interpreter and/or thread states exist, these
325 are not wiped out; only the current thread and interpreter state
326 are deleted. But since everything else is deleted, those other
327 interpreter and thread states should no longer be used.
328
329 (XXX We should do better, e.g. wipe out all interpreters and
330 threads.)
331
332 Locking: as above.
333
334*/
335
336void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000337Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000338{
339 PyInterpreterState *interp;
340 PyThreadState *tstate;
341
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000342 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000343 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000344
Tim Peters384fd102001-01-21 03:40:37 +0000345 /* The interpreter is still entirely intact at this point, and the
346 * exit funcs may be relying on that. In particular, if some thread
347 * or exit func is still waiting to do an import, the import machinery
348 * expects Py_IsInitialized() to return true. So don't say the
349 * interpreter is uninitialized until after the exit funcs have run.
350 * Note that Threading.py uses an exit func to do a join on all the
351 * threads created thru it, so this also protects pending imports in
352 * the threads created via Threading.
353 */
Collin Winter670e6922007-03-21 02:57:17 +0000354 call_py_exitfuncs();
Tim Peters384fd102001-01-21 03:40:37 +0000355 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000356
Guido van Rossume8432ac2007-07-09 15:04:50 +0000357 /* Flush stdout+stderr */
358 flush_std_files();
359
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000360 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000361 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000362 interp = tstate->interp;
363
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000364 /* Disable signal handling */
365 PyOS_FiniInterrupts();
366
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000367 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000368 Py_XDECREF(warnings_module);
369 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000370
Guido van Rossume13ddc92003-04-17 17:29:22 +0000371 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000372 * before all modules are destroyed.
373 * XXX If a __del__ or weakref callback is triggered here, and tries to
374 * XXX import a module, bad things can happen, because Python no
375 * XXX longer believes it's initialized.
376 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
377 * XXX is easy to provoke that way. I've also seen, e.g.,
378 * XXX Exception exceptions.ImportError: 'No module named sha'
379 * XXX in <function callback at 0x008F5718> ignored
380 * XXX but I'm unclear on exactly how that one happens. In any case,
381 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000382 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000383 PyGC_Collect();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000384#ifdef COUNT_ALLOCS
385 /* With COUNT_ALLOCS, it helps to run GC multiple times:
386 each collection might release some types from the type
387 list, so they become garbage. */
388 while (PyGC_Collect() > 0)
389 /* nothing */;
390#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000391
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000392 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000393 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000394
Guido van Rossume8432ac2007-07-09 15:04:50 +0000395 /* Flush stdout+stderr (again, in case more was printed) */
396 flush_std_files();
397
Guido van Rossume13ddc92003-04-17 17:29:22 +0000398 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000399 * new-style class definitions, for example.
400 * XXX This is disabled because it caused too many problems. If
401 * XXX a __del__ or weakref callback triggers here, Python code has
402 * XXX a hard time running, because even the sys module has been
403 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
404 * XXX One symptom is a sequence of information-free messages
405 * XXX coming from threads (if a __del__ or callback is invoked,
406 * XXX other threads can execute too, and any exception they encounter
407 * XXX triggers a comedy of errors as subsystem after subsystem
408 * XXX fails to find what it *expects* to find in sys to help report
409 * XXX the exception and consequent unexpected failures). I've also
410 * XXX seen segfaults then, after adding print statements to the
411 * XXX Python code getting called.
412 */
413#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000414 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000415#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000416
Guido van Rossum1707aad1997-12-08 23:43:45 +0000417 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
418 _PyImport_Fini();
419
420 /* Debugging stuff */
421#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000422 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000423#endif
424
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000425 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000426
Tim Peters9cf25ce2003-04-17 15:21:01 +0000427#ifdef Py_TRACE_REFS
428 /* Display all objects still alive -- this can invoke arbitrary
429 * __repr__ overrides, so requires a mostly-intact interpreter.
430 * Alas, a lot of stuff may still be alive now that will be cleaned
431 * up later.
432 */
Tim Peters269b2a62003-04-17 19:52:29 +0000433 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000434 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000435#endif /* Py_TRACE_REFS */
436
Mark Hammond6cb90292003-04-22 11:18:00 +0000437 /* Cleanup auto-thread-state */
438#ifdef WITH_THREAD
439 _PyGILState_Fini();
440#endif /* WITH_THREAD */
441
Guido van Rossumd922fa42003-04-15 14:10:09 +0000442 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000443 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000444
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000445 /* Now we decref the exception classes. After this point nothing
446 can raise an exception. That's okay, because each Fini() method
447 below has been checked to make sure no exceptions are ever
448 raised.
449 */
450
451 _PyExc_Fini();
452
Guido van Rossumd922fa42003-04-15 14:10:09 +0000453 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000454 PyThreadState_Swap(NULL);
455 PyInterpreterState_Delete(interp);
456
Guido van Rossumd922fa42003-04-15 14:10:09 +0000457 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000458 PyMethod_Fini();
459 PyFrame_Fini();
460 PyCFunction_Fini();
461 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000462 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000463 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000464 PyString_Fini();
Neal Norwitz6968b052007-02-27 19:02:19 +0000465 PyBytes_Fini();
Guido van Rossumddefaf32007-01-14 03:31:43 +0000466 PyLong_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000467 PyFloat_Fini();
468
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000469 /* Cleanup Unicode implementation */
470 _PyUnicode_Fini();
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000471
Guido van Rossumcc283f51997-08-05 02:22:03 +0000472 /* XXX Still allocated:
473 - various static ad-hoc pointers to interned strings
474 - int and float free list blocks
475 - whatever various modules and libraries allocate
476 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000477
478 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000479
Tim Peters269b2a62003-04-17 19:52:29 +0000480#ifdef Py_TRACE_REFS
481 /* Display addresses (& refcnts) of all objects still alive.
482 * An address can be used to find the repr of the object, printed
483 * above by _Py_PrintReferences.
484 */
485 if (Py_GETENV("PYTHONDUMPREFS"))
486 _Py_PrintReferenceAddresses(stderr);
487#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000488#ifdef PYMALLOC_DEBUG
489 if (Py_GETENV("PYTHONMALLOCSTATS"))
490 _PyObject_DebugMallocStats();
491#endif
492
Guido van Rossumcc283f51997-08-05 02:22:03 +0000493 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000494}
495
496/* Create and initialize a new interpreter and thread, and return the
497 new thread. This requires that Py_Initialize() has been called
498 first.
499
500 Unsuccessful initialization yields a NULL pointer. Note that *no*
501 exception information is available even in this case -- the
502 exception information is held in the thread, and there is no
503 thread.
504
505 Locking: as above.
506
507*/
508
509PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000510Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000511{
512 PyInterpreterState *interp;
513 PyThreadState *tstate, *save_tstate;
514 PyObject *bimod, *sysmod;
515
516 if (!initialized)
517 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
518
519 interp = PyInterpreterState_New();
520 if (interp == NULL)
521 return NULL;
522
523 tstate = PyThreadState_New(interp);
524 if (tstate == NULL) {
525 PyInterpreterState_Delete(interp);
526 return NULL;
527 }
528
529 save_tstate = PyThreadState_Swap(tstate);
530
531 /* XXX The following is lax in error checking */
532
533 interp->modules = PyDict_New();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000534 interp->modules_reloading = PyDict_New();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000535
536 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
537 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000538 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000539 if (interp->builtins == NULL)
540 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000541 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000542 }
543 sysmod = _PyImport_FindExtension("sys", "sys");
544 if (bimod != NULL && sysmod != NULL) {
545 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000546 if (interp->sysdict == NULL)
547 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000548 Py_INCREF(interp->sysdict);
549 PySys_SetPath(Py_GetPath());
550 PyDict_SetItemString(interp->sysdict, "modules",
551 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000552 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000553 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000554 if (!Py_NoSiteFlag)
555 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000556 }
557
558 if (!PyErr_Occurred())
559 return tstate;
560
Thomas Wouters89f507f2006-12-13 04:49:30 +0000561handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000562 /* Oops, it didn't work. Undo it all. */
563
564 PyErr_Print();
565 PyThreadState_Clear(tstate);
566 PyThreadState_Swap(save_tstate);
567 PyThreadState_Delete(tstate);
568 PyInterpreterState_Delete(interp);
569
570 return NULL;
571}
572
573/* Delete an interpreter and its last thread. This requires that the
574 given thread state is current, that the thread has no remaining
575 frames, and that it is its interpreter's only remaining thread.
576 It is a fatal error to violate these constraints.
577
578 (Py_Finalize() doesn't have these constraints -- it zaps
579 everything, regardless.)
580
581 Locking: as above.
582
583*/
584
585void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000586Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000587{
588 PyInterpreterState *interp = tstate->interp;
589
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000590 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000591 Py_FatalError("Py_EndInterpreter: thread is not current");
592 if (tstate->frame != NULL)
593 Py_FatalError("Py_EndInterpreter: thread still has a frame");
594 if (tstate != interp->tstate_head || tstate->next != NULL)
595 Py_FatalError("Py_EndInterpreter: not the last thread");
596
597 PyImport_Cleanup();
598 PyInterpreterState_Clear(interp);
599 PyThreadState_Swap(NULL);
600 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000601}
602
603static char *progname = "python";
604
605void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000606Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000607{
608 if (pn && *pn)
609 progname = pn;
610}
611
612char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000613Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000614{
615 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000616}
617
Guido van Rossuma61691e1998-02-06 22:27:24 +0000618static char *default_home = NULL;
619
620void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000621Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000622{
623 default_home = home;
624}
625
626char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000627Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000628{
629 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000630 if (home == NULL && !Py_IgnoreEnvironmentFlag)
631 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000632 return home;
633}
634
Guido van Rossum6135a871995-01-09 17:53:26 +0000635/* Create __main__ module */
636
637static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000638initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000639{
Guido van Rossum82598051997-03-05 00:20:32 +0000640 PyObject *m, *d;
641 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000642 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000643 Py_FatalError("can't create __main__ module");
644 d = PyModule_GetDict(m);
645 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000646 PyObject *bimod = PyImport_ImportModule("__builtin__");
647 if (bimod == NULL ||
648 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000649 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000650 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000651 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000652}
653
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000654/* Import the site module (not into __main__ though) */
655
656static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000657initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000658{
659 PyObject *m, *f;
660 m = PyImport_ImportModule("site");
661 if (m == NULL) {
662 f = PySys_GetObject("stderr");
663 if (Py_VerboseFlag) {
664 PyFile_WriteString(
665 "'import site' failed; traceback:\n", f);
666 PyErr_Print();
667 }
668 else {
669 PyFile_WriteString(
670 "'import site' failed; use -v for traceback\n", f);
671 PyErr_Clear();
672 }
673 }
674 else {
675 Py_DECREF(m);
676 }
677}
678
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000679/* Parse input from a file and execute it */
680
681int
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000682PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000683 PyCompilerFlags *flags)
684{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000685 if (filename == NULL)
686 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000687 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000688 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000689 if (closeit)
690 fclose(fp);
691 return err;
692 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000693 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000694 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000695}
696
697int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000698PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000699{
Guido van Rossum82598051997-03-05 00:20:32 +0000700 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000701 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000702 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000703
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000704 if (flags == NULL) {
705 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000706 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000707 }
Guido van Rossum82598051997-03-05 00:20:32 +0000708 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000709 if (v == NULL) {
Neal Norwitza369c5a2007-08-25 07:41:59 +0000710 PySys_SetObject("ps1", v = PyUnicode_FromString(">>> "));
Guido van Rossum82598051997-03-05 00:20:32 +0000711 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000712 }
Guido van Rossum82598051997-03-05 00:20:32 +0000713 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000714 if (v == NULL) {
Neal Norwitza369c5a2007-08-25 07:41:59 +0000715 PySys_SetObject("ps2", v = PyUnicode_FromString("... "));
Guido van Rossum82598051997-03-05 00:20:32 +0000716 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000717 }
718 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000719 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000720 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000721 if (ret == E_EOF)
722 return 0;
723 /*
724 if (ret == E_NOMEM)
725 return -1;
726 */
727 }
728}
729
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000730/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000731#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000732 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Guido van Rossum45aecf42006-03-15 04:58:47 +0000733 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000734
Thomas Wouters89f507f2006-12-13 04:49:30 +0000735#if 0
736/* Keep an example of flags with future keyword support. */
737#define PARSER_FLAGS(flags) \
738 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
739 PyPARSE_DONT_IMPLY_DEDENT : 0) \
740 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
741 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
742#endif
743
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000744int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000745PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000746{
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000747 PyObject *m, *d, *v, *w, *oenc = NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000748 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000749 PyArena *arena;
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000750 char *ps1 = "", *ps2 = "", *enc = NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000751 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000752
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000753 if (fp == stdin) {
754 /* Fetch encoding from sys.stdin */
755 v = PySys_GetObject("stdin");
756 if (!v)
757 return -1;
758 oenc = PyObject_GetAttrString(v, "encoding");
759 if (!oenc)
760 return -1;
761 enc = PyUnicode_AsString(oenc);
762 }
Guido van Rossum82598051997-03-05 00:20:32 +0000763 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000764 if (v != NULL) {
765 v = PyObject_Str(v);
766 if (v == NULL)
767 PyErr_Clear();
768 else if (PyString_Check(v))
769 ps1 = PyString_AsString(v);
770 }
Guido van Rossum82598051997-03-05 00:20:32 +0000771 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000772 if (w != NULL) {
773 w = PyObject_Str(w);
774 if (w == NULL)
775 PyErr_Clear();
776 else if (PyString_Check(w))
777 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000778 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000779 arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000780 if (arena == NULL) {
781 Py_XDECREF(v);
782 Py_XDECREF(w);
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000783 Py_XDECREF(oenc);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000784 return -1;
785 }
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000786 mod = PyParser_ASTFromFile(fp, filename, enc,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000787 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000788 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000789 Py_XDECREF(v);
790 Py_XDECREF(w);
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000791 Py_XDECREF(oenc);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000792 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000793 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000794 if (errcode == E_EOF) {
795 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000796 return E_EOF;
797 }
Guido van Rossum82598051997-03-05 00:20:32 +0000798 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000799 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000800 }
Guido van Rossum82598051997-03-05 00:20:32 +0000801 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000802 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000803 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000804 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000805 }
Guido van Rossum82598051997-03-05 00:20:32 +0000806 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000807 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000808 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000809 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000810 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000811 return -1;
812 }
Guido van Rossum82598051997-03-05 00:20:32 +0000813 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000814 return 0;
815}
816
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000817/* Check whether a file maybe a pyc file: Look at the extension,
818 the file type, and, if we may close it, at the first few bytes. */
819
820static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000821maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000822{
823 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
824 return 1;
825
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000826 /* Only look into the file if we are allowed to close it, since
827 it then should also be seekable. */
828 if (closeit) {
829 /* Read only two bytes of the magic. If the file was opened in
830 text mode, the bytes 3 and 4 of the magic (\r\n) might not
831 be read as they are on disk. */
832 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
833 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000834 /* Mess: In case of -x, the stream is NOT at its start now,
835 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000836 which makes the current stream position formally undefined,
837 and a x-platform nightmare.
838 Unfortunately, we have no direct way to know whether -x
839 was specified. So we use a terrible hack: if the current
840 stream position is not 0, we assume -x was specified, and
841 give up. Bug 132850 on SourceForge spells out the
842 hopelessness of trying anything else (fseek and ftell
843 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000844 */
Tim Peters3e876562001-02-11 04:35:39 +0000845 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000846 if (ftell(fp) == 0) {
847 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000848 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000849 ispyc = 1;
850 rewind(fp);
851 }
Tim Peters3e876562001-02-11 04:35:39 +0000852 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000853 }
854 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000855}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000856
Guido van Rossum0df002c2000-08-27 19:21:52 +0000857int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000858PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000859 PyCompilerFlags *flags)
860{
Guido van Rossum82598051997-03-05 00:20:32 +0000861 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000862 const char *ext;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000863 int set_file_name = 0, ret;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000864
Guido van Rossum82598051997-03-05 00:20:32 +0000865 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000866 if (m == NULL)
867 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000868 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000869 if (PyDict_GetItemString(d, "__file__") == NULL) {
870 PyObject *f = PyString_FromString(filename);
871 if (f == NULL)
872 return -1;
873 if (PyDict_SetItemString(d, "__file__", f) < 0) {
874 Py_DECREF(f);
875 return -1;
876 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000877 set_file_name = 1;
Fred Drake8ed02042002-10-17 21:24:58 +0000878 Py_DECREF(f);
879 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000880 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000881 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000882 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000883 if (closeit)
884 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000885 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000886 fprintf(stderr, "python: Can't reopen .pyc file\n");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000887 ret = -1;
888 goto done;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000889 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000890 /* Turn on optimization if a .pyo file is given */
891 if (strcmp(ext, ".pyo") == 0)
892 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000893 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000894 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000895 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000896 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000897 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000898 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000899 PyErr_Print();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000900 ret = -1;
901 goto done;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000902 }
Guido van Rossum82598051997-03-05 00:20:32 +0000903 Py_DECREF(v);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000904 ret = 0;
905 done:
906 if (set_file_name && PyDict_DelItemString(d, "__file__"))
907 PyErr_Clear();
908 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000909}
910
911int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000912PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000913{
Guido van Rossum82598051997-03-05 00:20:32 +0000914 PyObject *m, *d, *v;
915 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000916 if (m == NULL)
917 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000918 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000919 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000920 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000921 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000922 return -1;
923 }
Guido van Rossum82598051997-03-05 00:20:32 +0000924 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000925 return 0;
926}
927
Barry Warsaw035574d1997-08-29 22:07:17 +0000928static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000929parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
930 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000931{
932 long hold;
933 PyObject *v;
934
935 /* old style errors */
936 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000937 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000938 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000939
940 /* new style errors. `err' is an instance */
941
942 if (! (v = PyObject_GetAttrString(err, "msg")))
943 goto finally;
944 *message = v;
945
946 if (!(v = PyObject_GetAttrString(err, "filename")))
947 goto finally;
948 if (v == Py_None)
949 *filename = NULL;
950 else if (! (*filename = PyString_AsString(v)))
951 goto finally;
952
953 Py_DECREF(v);
954 if (!(v = PyObject_GetAttrString(err, "lineno")))
955 goto finally;
956 hold = PyInt_AsLong(v);
957 Py_DECREF(v);
958 v = NULL;
959 if (hold < 0 && PyErr_Occurred())
960 goto finally;
961 *lineno = (int)hold;
962
963 if (!(v = PyObject_GetAttrString(err, "offset")))
964 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000965 if (v == Py_None) {
966 *offset = -1;
967 Py_DECREF(v);
968 v = NULL;
969 } else {
970 hold = PyInt_AsLong(v);
971 Py_DECREF(v);
972 v = NULL;
973 if (hold < 0 && PyErr_Occurred())
974 goto finally;
975 *offset = (int)hold;
976 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000977
978 if (!(v = PyObject_GetAttrString(err, "text")))
979 goto finally;
980 if (v == Py_None)
981 *text = NULL;
982 else if (! (*text = PyString_AsString(v)))
983 goto finally;
984 Py_DECREF(v);
985 return 1;
986
987finally:
988 Py_XDECREF(v);
989 return 0;
990}
991
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000992void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000993PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000994{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000995 PyErr_PrintEx(1);
996}
997
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000998static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000999print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001000{
1001 char *nl;
1002 if (offset >= 0) {
1003 if (offset > 0 && offset == (int)strlen(text))
1004 offset--;
1005 for (;;) {
1006 nl = strchr(text, '\n');
1007 if (nl == NULL || nl-text >= offset)
1008 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001009 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001010 text = nl+1;
1011 }
1012 while (*text == ' ' || *text == '\t') {
1013 text++;
1014 offset--;
1015 }
1016 }
1017 PyFile_WriteString(" ", f);
1018 PyFile_WriteString(text, f);
1019 if (*text == '\0' || text[strlen(text)-1] != '\n')
1020 PyFile_WriteString("\n", f);
1021 if (offset == -1)
1022 return;
1023 PyFile_WriteString(" ", f);
1024 offset--;
1025 while (offset > 0) {
1026 PyFile_WriteString(" ", f);
1027 offset--;
1028 }
1029 PyFile_WriteString("^\n", f);
1030}
1031
Guido van Rossum66e8e862001-03-23 17:54:43 +00001032static void
1033handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001034{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001035 PyObject *exception, *value, *tb;
1036 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001037
Guido van Rossumd8faa362007-04-27 19:54:29 +00001038 if (Py_InspectFlag)
1039 /* Don't exit if -i flag was given. This flag is set to 0
1040 * when entering interactive mode for inspecting. */
1041 return;
1042
Guido van Rossum66e8e862001-03-23 17:54:43 +00001043 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001044 fflush(stdout);
1045 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001046 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001047 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001048 /* The error code should be in the `code' attribute. */
1049 PyObject *code = PyObject_GetAttrString(value, "code");
1050 if (code) {
1051 Py_DECREF(value);
1052 value = code;
1053 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001054 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001055 }
1056 /* If we failed to dig out the 'code' attribute,
1057 just let the else clause below print the error. */
1058 }
1059 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001060 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001061 else {
1062 PyObject_Print(value, stderr, Py_PRINT_RAW);
1063 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001064 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001065 }
Tim Peterscf615b52003-04-19 18:47:02 +00001066 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001067 /* Restore and clear the exception info, in order to properly decref
1068 * the exception, value, and traceback. If we just exit instead,
1069 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1070 * some finalizers from running.
1071 */
Tim Peterscf615b52003-04-19 18:47:02 +00001072 PyErr_Restore(exception, value, tb);
1073 PyErr_Clear();
1074 Py_Exit(exitcode);
1075 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001076}
1077
1078void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001079PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001080{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001081 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001082
1083 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1084 handle_system_exit();
1085 }
Guido van Rossum82598051997-03-05 00:20:32 +00001086 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001087 if (exception == NULL)
1088 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001089 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001090 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001091 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001092 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001093 if (set_sys_last_vars) {
1094 PySys_SetObject("last_type", exception);
1095 PySys_SetObject("last_value", v);
1096 PySys_SetObject("last_traceback", tb);
1097 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001098 hook = PySys_GetObject("excepthook");
1099 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001100 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001101 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001102 PyObject *result = PyEval_CallObject(hook, args);
1103 if (result == NULL) {
1104 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001105 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1106 handle_system_exit();
1107 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001108 PyErr_Fetch(&exception2, &v2, &tb2);
1109 PyErr_NormalizeException(&exception2, &v2, &tb2);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001110 /* It should not be possible for exception2 or v2
1111 to be NULL. However PyErr_Display() can't
1112 tolerate NULLs, so just be safe. */
1113 if (exception2 == NULL) {
1114 exception2 = Py_None;
1115 Py_INCREF(exception2);
1116 }
1117 if (v2 == NULL) {
1118 v2 = Py_None;
1119 Py_INCREF(v2);
1120 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001121 fflush(stdout);
1122 PySys_WriteStderr("Error in sys.excepthook:\n");
1123 PyErr_Display(exception2, v2, tb2);
1124 PySys_WriteStderr("\nOriginal exception was:\n");
1125 PyErr_Display(exception, v, tb);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001126 Py_DECREF(exception2);
1127 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001128 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001129 }
1130 Py_XDECREF(result);
1131 Py_XDECREF(args);
1132 } else {
1133 PySys_WriteStderr("sys.excepthook is missing\n");
1134 PyErr_Display(exception, v, tb);
1135 }
1136 Py_XDECREF(exception);
1137 Py_XDECREF(v);
1138 Py_XDECREF(tb);
1139}
1140
Thomas Wouters477c8d52006-05-27 19:21:47 +00001141void
1142PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001143{
1144 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001145 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001146 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001147 if (f == NULL)
Martin v. Löwis5b222132007-06-10 09:51:05 +00001148 _PyObject_Dump(value);
1149 if (f == NULL)
Guido van Rossum3165fe61992-09-25 21:59:05 +00001150 fprintf(stderr, "lost sys.stderr\n");
1151 else {
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001152 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001153 if (tb && tb != Py_None)
1154 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001155 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001156 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001157 {
Guido van Rossum82598051997-03-05 00:20:32 +00001158 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001159 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001160 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001161 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001162 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001163 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001164 else {
1165 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001166 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001167 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001168 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001169 else
Guido van Rossum82598051997-03-05 00:20:32 +00001170 PyFile_WriteString(filename, f);
1171 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001172 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001173 PyFile_WriteString(buf, f);
1174 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001175 if (text != NULL)
1176 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001177 Py_DECREF(value);
1178 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001179 /* Can't be bothered to check all those
1180 PyFile_WriteString() calls */
1181 if (PyErr_Occurred())
1182 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001183 }
1184 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001185 if (err) {
1186 /* Don't do anything else */
1187 }
Brett Cannonbf364092006-03-01 04:25:17 +00001188 else if (PyExceptionClass_Check(exception)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001189 PyObject* moduleName;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001190 char* className = PyExceptionClass_Name(exception);
1191 if (className != NULL) {
1192 char *dot = strrchr(className, '.');
1193 if (dot != NULL)
1194 className = dot+1;
1195 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001196
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001197 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001198 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001199 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001200 else {
1201 char* modstr = PyString_AsString(moduleName);
Neal Norwitz2633c692007-02-26 22:22:47 +00001202 if (modstr && strcmp(modstr, "__builtin__"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001203 {
1204 err = PyFile_WriteString(modstr, f);
1205 err += PyFile_WriteString(".", f);
1206 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001207 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001208 }
1209 if (err == 0) {
1210 if (className == NULL)
1211 err = PyFile_WriteString("<unknown>", f);
1212 else
Brett Cannonbf364092006-03-01 04:25:17 +00001213 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001214 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001215 }
1216 else
1217 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001218 if (err == 0 && (value != Py_None)) {
1219 PyObject *s = PyObject_Str(value);
1220 /* only print colon if the str() of the
1221 object is not the empty string
1222 */
1223 if (s == NULL)
1224 err = -1;
1225 else if (!PyString_Check(s) ||
1226 PyString_GET_SIZE(s) != 0)
1227 err = PyFile_WriteString(": ", f);
1228 if (err == 0)
1229 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1230 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001231 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00001232 /* try to write a newline in any case */
1233 err += PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001234 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001235 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001236 /* If an error happened here, don't show it.
1237 XXX This is wrong, but too many callers rely on this behavior. */
1238 if (err != 0)
1239 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001240}
1241
Guido van Rossum82598051997-03-05 00:20:32 +00001242PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001243PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001244 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001245{
Neal Norwitze92fba02006-03-04 18:52:26 +00001246 PyObject *ret = NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001247 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001248 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001249 if (arena == NULL)
1250 return NULL;
1251
1252 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001253 if (mod != NULL)
1254 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001255 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001256 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001257}
1258
1259PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001260PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001261 PyObject *locals, int closeit, PyCompilerFlags *flags)
1262{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001263 PyObject *ret;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001264 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001265 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001266 if (arena == NULL)
1267 return NULL;
1268
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001269 mod = PyParser_ASTFromFile(fp, filename, NULL, start, 0, 0,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001270 flags, NULL, arena);
Guido van Rossumd8faa362007-04-27 19:54:29 +00001271 if (closeit)
1272 fclose(fp);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001273 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001274 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001275 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001276 }
Neal Norwitze92fba02006-03-04 18:52:26 +00001277 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001278 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001279 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001280}
1281
Guido van Rossum82598051997-03-05 00:20:32 +00001282static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001283run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001284 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001285{
Guido van Rossum82598051997-03-05 00:20:32 +00001286 PyCodeObject *co;
1287 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001288 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001289 if (co == NULL)
1290 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001291 v = PyEval_EvalCode(co, globals, locals);
1292 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001293 return v;
1294}
1295
Guido van Rossum82598051997-03-05 00:20:32 +00001296static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001297run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001298 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001299{
Guido van Rossum82598051997-03-05 00:20:32 +00001300 PyCodeObject *co;
1301 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001302 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001303 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001304
Guido van Rossum82598051997-03-05 00:20:32 +00001305 magic = PyMarshal_ReadLongFromFile(fp);
1306 if (magic != PyImport_GetMagicNumber()) {
1307 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001308 "Bad magic number in .pyc file");
1309 return NULL;
1310 }
Guido van Rossum82598051997-03-05 00:20:32 +00001311 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001312 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001313 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001314 if (v == NULL || !PyCode_Check(v)) {
1315 Py_XDECREF(v);
1316 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001317 "Bad code object in .pyc file");
1318 return NULL;
1319 }
Guido van Rossum82598051997-03-05 00:20:32 +00001320 co = (PyCodeObject *)v;
1321 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001322 if (v && flags)
1323 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001324 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001325 return v;
1326}
1327
Guido van Rossum82598051997-03-05 00:20:32 +00001328PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001329Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001330 PyCompilerFlags *flags)
1331{
Guido van Rossum82598051997-03-05 00:20:32 +00001332 PyCodeObject *co;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001333 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001334 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001335 if (arena == NULL)
1336 return NULL;
1337
1338 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001339 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001340 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001341 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001342 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001343 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001344 PyObject *result = PyAST_mod2obj(mod);
1345 PyArena_Free(arena);
1346 return result;
1347 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001348 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001349 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001350 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001351}
1352
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001353struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001354Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001355{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001356 struct symtable *st;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001357 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001358 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001359 if (arena == NULL)
1360 return NULL;
1361
1362 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001363 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001364 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001365 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001366 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001367 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001368 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001369 return st;
1370}
1371
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001372/* Preferred access to parser is through AST. */
1373mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001374PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001375 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001376{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001377 mod_ty mod;
1378 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001379 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001380 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001381 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001382 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001383 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001384 PyNode_Free(n);
1385 return mod;
1386 }
1387 else {
1388 err_input(&err);
1389 return NULL;
1390 }
1391}
1392
1393mod_ty
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001394PyParser_ASTFromFile(FILE *fp, const char *filename, const char* enc,
1395 int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001396 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001397 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001398{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001399 mod_ty mod;
1400 perrdetail err;
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001401 node *n = PyParser_ParseFileFlags(fp, filename, enc,
1402 &_PyParser_Grammar,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001403 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001404 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001405 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001406 PyNode_Free(n);
1407 return mod;
1408 }
1409 else {
1410 err_input(&err);
1411 if (errcode)
1412 *errcode = err.error;
1413 return NULL;
1414 }
1415}
1416
Guido van Rossuma110aa61994-08-29 12:50:44 +00001417/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001418
Guido van Rossuma110aa61994-08-29 12:50:44 +00001419node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001420PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001421{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001422 perrdetail err;
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001423 node *n = PyParser_ParseFileFlags(fp, filename, NULL,
1424 &_PyParser_Grammar,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001425 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001426 if (n == NULL)
1427 err_input(&err);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001428
Guido van Rossuma110aa61994-08-29 12:50:44 +00001429 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001430}
1431
Guido van Rossuma110aa61994-08-29 12:50:44 +00001432/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001433
Guido van Rossuma110aa61994-08-29 12:50:44 +00001434node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001435PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001436{
Tim Petersfe2127d2001-07-16 05:37:24 +00001437 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001438 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1439 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001440 if (n == NULL)
1441 err_input(&err);
1442 return n;
1443}
1444
1445node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001446PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001447 int start, int flags)
1448{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001449 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001450 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1451 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001452 if (n == NULL)
1453 err_input(&err);
1454 return n;
1455}
1456
1457node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001458PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001459{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001460 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001461}
1462
Guido van Rossum66ebd912003-04-17 16:02:26 +00001463/* May want to move a more generalized form of this to parsetok.c or
1464 even parser modules. */
1465
1466void
1467PyParser_SetError(perrdetail *err)
1468{
1469 err_input(err);
1470}
1471
Guido van Rossuma110aa61994-08-29 12:50:44 +00001472/* Set the error appropriate to the given input error code (see errcode.h) */
1473
1474static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001475err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001476{
Martin v. Löwis5deb2102007-08-31 11:17:42 +00001477 PyObject *v, *w, *errtype, *errtext;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001478 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001479 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001480 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001481 switch (err->error) {
1482 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001483 errtype = PyExc_IndentationError;
1484 if (err->expected == INDENT)
1485 msg = "expected an indented block";
1486 else if (err->token == INDENT)
1487 msg = "unexpected indent";
1488 else if (err->token == DEDENT)
1489 msg = "unexpected unindent";
1490 else {
1491 errtype = PyExc_SyntaxError;
1492 msg = "invalid syntax";
1493 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001494 break;
1495 case E_TOKEN:
1496 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001497 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001498 case E_EOFS:
1499 msg = "EOF while scanning triple-quoted string";
1500 break;
1501 case E_EOLS:
1502 msg = "EOL while scanning single-quoted string";
1503 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001504 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001505 if (!PyErr_Occurred())
1506 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001507 return;
1508 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001509 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001510 return;
1511 case E_EOF:
1512 msg = "unexpected EOF while parsing";
1513 break;
Fred Drake85f36392000-07-11 17:53:00 +00001514 case E_TABSPACE:
1515 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001516 msg = "inconsistent use of tabs and spaces in indentation";
1517 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001518 case E_OVERFLOW:
1519 msg = "expression too long";
1520 break;
Fred Drake85f36392000-07-11 17:53:00 +00001521 case E_DEDENT:
1522 errtype = PyExc_IndentationError;
1523 msg = "unindent does not match any outer indentation level";
1524 break;
1525 case E_TOODEEP:
1526 errtype = PyExc_IndentationError;
1527 msg = "too many levels of indentation";
1528 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001529 case E_DECODE: {
1530 PyObject *type, *value, *tb;
1531 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001532 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001533 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001534 if (u != NULL) {
1535 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001536 }
1537 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001538 if (msg == NULL)
1539 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001540 Py_XDECREF(type);
1541 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001542 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001543 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001544 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001545 case E_LINECONT:
1546 msg = "unexpected character after line continuation character";
1547 break;
Martin v. Löwis47383402007-08-15 07:32:56 +00001548
1549 case E_IDENTIFIER:
1550 msg = "invalid character in identifier";
1551 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001552 default:
1553 fprintf(stderr, "error=%d\n", err->error);
1554 msg = "unknown parsing error";
1555 break;
1556 }
Martin v. Löwis5deb2102007-08-31 11:17:42 +00001557 /* err->text may not be UTF-8 in case of decoding errors.
1558 Explicitly convert to an object. */
1559 if (!err->text) {
1560 errtext = Py_None;
1561 Py_INCREF(Py_None);
1562 } else {
1563 errtext = PyUnicode_DecodeUTF8(err->text, strlen(err->text),
1564 "replace");
1565 }
1566 v = Py_BuildValue("(ziiN)", err->filename,
1567 err->lineno, err->offset, errtext);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001568 if (err->text != NULL) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001569 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001570 err->text = NULL;
1571 }
1572 w = NULL;
1573 if (v != NULL)
1574 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001575 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001576 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001577 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001578 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001579}
1580
1581/* Print fatal error message and abort */
1582
1583void
Tim Peters7c321a82002-07-09 02:57:01 +00001584Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001585{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001586 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001587#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001588 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001589 OutputDebugString(msg);
1590 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001591#ifdef _DEBUG
1592 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001593#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001594#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001595 abort();
1596}
1597
1598/* Clean up and exit */
1599
Guido van Rossuma110aa61994-08-29 12:50:44 +00001600#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001601#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001602#endif
1603
Collin Winter670e6922007-03-21 02:57:17 +00001604static void (*pyexitfunc)(void) = NULL;
1605/* For the atexit module. */
1606void _Py_PyAtExit(void (*func)(void))
1607{
1608 pyexitfunc = func;
1609}
1610
1611static void
1612call_py_exitfuncs(void)
1613{
1614 if (pyexitfunc == NULL)
1615 return;
1616
1617 (*pyexitfunc)();
1618 PyErr_Clear();
1619}
1620
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001621#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001622static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001623static int nexitfuncs = 0;
1624
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001625int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001626{
1627 if (nexitfuncs >= NEXITFUNCS)
1628 return -1;
1629 exitfuncs[nexitfuncs++] = func;
1630 return 0;
1631}
1632
Guido van Rossumcc283f51997-08-05 02:22:03 +00001633static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001634call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001635{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001636 while (nexitfuncs > 0)
1637 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001638
1639 fflush(stdout);
1640 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001641}
1642
1643void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001644Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001645{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001646 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001647
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001648 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001649}
1650
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001651static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001652initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001653{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001654#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001655 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001656#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001657#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001658 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001659#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001660#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001661 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001662#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001663 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001664}
1665
Guido van Rossum7433b121997-02-14 19:45:36 +00001666
1667/*
1668 * The file descriptor fd is considered ``interactive'' if either
1669 * a) isatty(fd) is TRUE, or
1670 * b) the -i flag was given, and the filename associated with
1671 * the descriptor is NULL or "<stdin>" or "???".
1672 */
1673int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001674Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001675{
1676 if (isatty((int)fileno(fp)))
1677 return 1;
1678 if (!Py_InteractiveFlag)
1679 return 0;
1680 return (filename == NULL) ||
1681 (strcmp(filename, "<stdin>") == 0) ||
1682 (strcmp(filename, "???") == 0);
1683}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001684
1685
Tim Petersd08e3822003-04-17 15:24:21 +00001686#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001687#if defined(WIN32) && defined(_MSC_VER)
1688
1689/* Stack checking for Microsoft C */
1690
1691#include <malloc.h>
1692#include <excpt.h>
1693
Fred Drakee8de31c2000-08-31 05:38:39 +00001694/*
1695 * Return non-zero when we run out of memory on the stack; zero otherwise.
1696 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001697int
Fred Drake399739f2000-08-31 05:52:44 +00001698PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001699{
1700 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001701 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001702 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001703 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001704 return 0;
1705 } __except (EXCEPTION_EXECUTE_HANDLER) {
1706 /* just ignore all errors */
1707 }
1708 return 1;
1709}
1710
1711#endif /* WIN32 && _MSC_VER */
1712
1713/* Alternate implementations can be added here... */
1714
1715#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001716
1717
1718/* Wrappers around sigaction() or signal(). */
1719
1720PyOS_sighandler_t
1721PyOS_getsig(int sig)
1722{
1723#ifdef HAVE_SIGACTION
1724 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001725 if (sigaction(sig, NULL, &context) == -1)
1726 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001727 return context.sa_handler;
1728#else
1729 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001730/* Special signal handling for the secure CRT in Visual Studio 2005 */
1731#if defined(_MSC_VER) && _MSC_VER >= 1400
1732 switch (sig) {
1733 /* Only these signals are valid */
1734 case SIGINT:
1735 case SIGILL:
1736 case SIGFPE:
1737 case SIGSEGV:
1738 case SIGTERM:
1739 case SIGBREAK:
1740 case SIGABRT:
1741 break;
1742 /* Don't call signal() with other values or it will assert */
1743 default:
1744 return SIG_ERR;
1745 }
1746#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001747 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001748 if (handler != SIG_ERR)
1749 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001750 return handler;
1751#endif
1752}
1753
1754PyOS_sighandler_t
1755PyOS_setsig(int sig, PyOS_sighandler_t handler)
1756{
1757#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001758 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001759 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001760 sigemptyset(&context.sa_mask);
1761 context.sa_flags = 0;
1762 if (sigaction(sig, &context, &ocontext) == -1)
1763 return SIG_ERR;
1764 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001765#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001766 PyOS_sighandler_t oldhandler;
1767 oldhandler = signal(sig, handler);
1768#ifdef HAVE_SIGINTERRUPT
1769 siginterrupt(sig, 1);
1770#endif
1771 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001772#endif
1773}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001774
1775/* Deprecated C API functions still provided for binary compatiblity */
1776
1777#undef PyParser_SimpleParseFile
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001778PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001779PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1780{
1781 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1782}
1783
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001784#undef PyParser_SimpleParseString
1785PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001786PyParser_SimpleParseString(const char *str, int start)
1787{
1788 return PyParser_SimpleParseStringFlags(str, start, 0);
1789}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001790
1791#undef PyRun_AnyFile
1792PyAPI_FUNC(int)
1793PyRun_AnyFile(FILE *fp, const char *name)
1794{
1795 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1796}
1797
1798#undef PyRun_AnyFileEx
1799PyAPI_FUNC(int)
1800PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1801{
1802 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1803}
1804
1805#undef PyRun_AnyFileFlags
1806PyAPI_FUNC(int)
1807PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1808{
1809 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1810}
1811
1812#undef PyRun_File
1813PyAPI_FUNC(PyObject *)
1814PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1815{
1816 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1817}
1818
1819#undef PyRun_FileEx
1820PyAPI_FUNC(PyObject *)
1821PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1822{
1823 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1824}
1825
1826#undef PyRun_FileFlags
1827PyAPI_FUNC(PyObject *)
1828PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1829 PyCompilerFlags *flags)
1830{
1831 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1832}
1833
1834#undef PyRun_SimpleFile
1835PyAPI_FUNC(int)
1836PyRun_SimpleFile(FILE *f, const char *p)
1837{
1838 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1839}
1840
1841#undef PyRun_SimpleFileEx
1842PyAPI_FUNC(int)
1843PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1844{
1845 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1846}
1847
1848
1849#undef PyRun_String
1850PyAPI_FUNC(PyObject *)
1851PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1852{
1853 return PyRun_StringFlags(str, s, g, l, NULL);
1854}
1855
1856#undef PyRun_SimpleString
1857PyAPI_FUNC(int)
1858PyRun_SimpleString(const char *s)
1859{
1860 return PyRun_SimpleStringFlags(s, NULL);
1861}
1862
1863#undef Py_CompileString
1864PyAPI_FUNC(PyObject *)
1865Py_CompileString(const char *str, const char *p, int s)
1866{
1867 return Py_CompileStringFlags(str, p, s, NULL);
1868}
1869
1870#undef PyRun_InteractiveOne
1871PyAPI_FUNC(int)
1872PyRun_InteractiveOne(FILE *f, const char *p)
1873{
1874 return PyRun_InteractiveOneFlags(f, p, NULL);
1875}
1876
1877#undef PyRun_InteractiveLoop
1878PyAPI_FUNC(int)
1879PyRun_InteractiveLoop(FILE *f, const char *p)
1880{
1881 return PyRun_InteractiveLoopFlags(f, p, NULL);
1882}
1883
1884#ifdef __cplusplus
1885}
1886#endif