blob: bb69b1914214fa60b67d8a95736c2f4d6d146cae [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;
157 char *saved_locale;
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
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000165 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000166 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000167 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000168 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000169 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000170 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000171
Guido van Rossuma027efa1997-05-05 20:56:21 +0000172 interp = PyInterpreterState_New();
173 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000174 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000175
Guido van Rossuma027efa1997-05-05 20:56:21 +0000176 tstate = PyThreadState_New(interp);
177 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000178 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000179 (void) PyThreadState_Swap(tstate);
180
Guido van Rossum70d893a2001-08-16 08:21:42 +0000181 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000182
Neal Norwitzb2501f42002-12-31 03:42:13 +0000183 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000184 Py_FatalError("Py_Initialize: can't init frames");
185
Guido van Rossumddefaf32007-01-14 03:31:43 +0000186 if (!_PyLong_Init())
187 Py_FatalError("Py_Initialize: can't init longs");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000188
Neal Norwitz6968b052007-02-27 19:02:19 +0000189 if (!PyBytes_Init())
190 Py_FatalError("Py_Initialize: can't init bytes");
191
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000192 _PyFloat_Init();
193
Guido van Rossum25ce5661997-08-02 03:10:38 +0000194 interp->modules = PyDict_New();
195 if (interp->modules == NULL)
196 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000197 interp->modules_reloading = PyDict_New();
198 if (interp->modules_reloading == NULL)
199 Py_FatalError("Py_Initialize: can't make modules_reloading dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000200
Guido van Rossumc94044c2000-03-10 23:03:54 +0000201 /* Init Unicode implementation; relies on the codec registry */
202 _PyUnicode_Init();
203
Barry Warsawf242aa02000-05-25 23:09:49 +0000204 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000205 if (bimod == NULL)
206 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000207 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000208 if (interp->builtins == NULL)
209 Py_FatalError("Py_Initialize: can't initialize builtins dict");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000210 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000211
212 sysmod = _PySys_Init();
213 if (sysmod == NULL)
214 Py_FatalError("Py_Initialize: can't initialize sys");
215 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000216 if (interp->sysdict == NULL)
217 Py_FatalError("Py_Initialize: can't initialize sys dict");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000218 Py_INCREF(interp->sysdict);
219 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000220 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000221 PyDict_SetItemString(interp->sysdict, "modules",
222 interp->modules);
223
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000224 _PyImport_Init();
225
Barry Warsawf242aa02000-05-25 23:09:49 +0000226 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000227 _PyExc_Init();
Barry Warsawf242aa02000-05-25 23:09:49 +0000228
Barry Warsaw035574d1997-08-29 22:07:17 +0000229 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000230 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000231
Just van Rossum52e14d62002-12-30 22:08:05 +0000232 _PyImportHooks_Init();
233
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000234 if (install_sigs)
235 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000236
237 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000238 if (!Py_NoSiteFlag)
239 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000240
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000241 /* auto-thread-state API, if available */
242#ifdef WITH_THREAD
243 _PyGILState_Init(interp, tstate);
244#endif /* WITH_THREAD */
245
Mark Hammondedd07732003-07-15 23:03:55 +0000246 warnings_module = PyImport_ImportModule("warnings");
247 if (!warnings_module)
248 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000249
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000250#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000251 /* On Unix, set the file system encoding according to the
252 user's preference, if the CODESET names a well-known
253 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000254 initialized by other means. Also set the encoding of
255 stdin and stdout if these are terminals. */
256
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000257 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000258 setlocale(LC_CTYPE, "");
259 codeset = nl_langinfo(CODESET);
260 if (codeset && *codeset) {
261 PyObject *enc = PyCodec_Encoder(codeset);
262 if (enc) {
263 codeset = strdup(codeset);
264 Py_DECREF(enc);
265 } else {
266 codeset = NULL;
267 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000268 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000269 } else
270 codeset = NULL;
271 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000272 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000273
274 if (codeset) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000275 if (!Py_FileSystemDefaultEncoding)
276 Py_FileSystemDefaultEncoding = codeset;
277 else
278 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000279 }
280#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000281}
282
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000283void
284Py_Initialize(void)
285{
286 Py_InitializeEx(1);
287}
288
289
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000290#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000291extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000292#endif
293
Guido van Rossume8432ac2007-07-09 15:04:50 +0000294/* Flush stdout and stderr */
295
296void
Guido van Rossum1bd21222007-07-10 20:14:13 +0000297flush_std_files(void)
Guido van Rossume8432ac2007-07-09 15:04:50 +0000298{
299 PyObject *fout = PySys_GetObject("stdout");
300 PyObject *ferr = PySys_GetObject("stderr");
301 PyObject *tmp;
302
303 if (fout != NULL) {
304 tmp = PyObject_CallMethod(fout, "flush", "");
305 if (tmp == NULL)
306 PyErr_Clear();
307 else
308 Py_DECREF(tmp);
309 }
310
311 if (ferr != NULL) {
312 tmp = PyObject_CallMethod(ferr, "flush", "");
313 if (tmp == NULL)
314 PyErr_Clear();
315 else
316 Py_DECREF(tmp);
317 }
318}
319
Guido van Rossum25ce5661997-08-02 03:10:38 +0000320/* Undo the effect of Py_Initialize().
321
322 Beware: if multiple interpreter and/or thread states exist, these
323 are not wiped out; only the current thread and interpreter state
324 are deleted. But since everything else is deleted, those other
325 interpreter and thread states should no longer be used.
326
327 (XXX We should do better, e.g. wipe out all interpreters and
328 threads.)
329
330 Locking: as above.
331
332*/
333
334void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000335Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000336{
337 PyInterpreterState *interp;
338 PyThreadState *tstate;
339
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000340 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000341 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000342
Tim Peters384fd102001-01-21 03:40:37 +0000343 /* The interpreter is still entirely intact at this point, and the
344 * exit funcs may be relying on that. In particular, if some thread
345 * or exit func is still waiting to do an import, the import machinery
346 * expects Py_IsInitialized() to return true. So don't say the
347 * interpreter is uninitialized until after the exit funcs have run.
348 * Note that Threading.py uses an exit func to do a join on all the
349 * threads created thru it, so this also protects pending imports in
350 * the threads created via Threading.
351 */
Collin Winter670e6922007-03-21 02:57:17 +0000352 call_py_exitfuncs();
Tim Peters384fd102001-01-21 03:40:37 +0000353 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000354
Guido van Rossume8432ac2007-07-09 15:04:50 +0000355 /* Flush stdout+stderr */
356 flush_std_files();
357
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000358 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000359 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000360 interp = tstate->interp;
361
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000362 /* Disable signal handling */
363 PyOS_FiniInterrupts();
364
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000365 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000366 Py_XDECREF(warnings_module);
367 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000368
Guido van Rossume13ddc92003-04-17 17:29:22 +0000369 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000370 * before all modules are destroyed.
371 * XXX If a __del__ or weakref callback is triggered here, and tries to
372 * XXX import a module, bad things can happen, because Python no
373 * XXX longer believes it's initialized.
374 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
375 * XXX is easy to provoke that way. I've also seen, e.g.,
376 * XXX Exception exceptions.ImportError: 'No module named sha'
377 * XXX in <function callback at 0x008F5718> ignored
378 * XXX but I'm unclear on exactly how that one happens. In any case,
379 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000380 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000381 PyGC_Collect();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000382#ifdef COUNT_ALLOCS
383 /* With COUNT_ALLOCS, it helps to run GC multiple times:
384 each collection might release some types from the type
385 list, so they become garbage. */
386 while (PyGC_Collect() > 0)
387 /* nothing */;
388#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000389
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000390 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000391 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000392
Guido van Rossume8432ac2007-07-09 15:04:50 +0000393 /* Flush stdout+stderr (again, in case more was printed) */
394 flush_std_files();
395
Guido van Rossume13ddc92003-04-17 17:29:22 +0000396 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000397 * new-style class definitions, for example.
398 * XXX This is disabled because it caused too many problems. If
399 * XXX a __del__ or weakref callback triggers here, Python code has
400 * XXX a hard time running, because even the sys module has been
401 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
402 * XXX One symptom is a sequence of information-free messages
403 * XXX coming from threads (if a __del__ or callback is invoked,
404 * XXX other threads can execute too, and any exception they encounter
405 * XXX triggers a comedy of errors as subsystem after subsystem
406 * XXX fails to find what it *expects* to find in sys to help report
407 * XXX the exception and consequent unexpected failures). I've also
408 * XXX seen segfaults then, after adding print statements to the
409 * XXX Python code getting called.
410 */
411#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000412 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000413#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000414
Guido van Rossum1707aad1997-12-08 23:43:45 +0000415 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
416 _PyImport_Fini();
417
418 /* Debugging stuff */
419#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000420 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000421#endif
422
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000423 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000424
Tim Peters9cf25ce2003-04-17 15:21:01 +0000425#ifdef Py_TRACE_REFS
426 /* Display all objects still alive -- this can invoke arbitrary
427 * __repr__ overrides, so requires a mostly-intact interpreter.
428 * Alas, a lot of stuff may still be alive now that will be cleaned
429 * up later.
430 */
Tim Peters269b2a62003-04-17 19:52:29 +0000431 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000432 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000433#endif /* Py_TRACE_REFS */
434
Mark Hammond6cb90292003-04-22 11:18:00 +0000435 /* Cleanup auto-thread-state */
436#ifdef WITH_THREAD
437 _PyGILState_Fini();
438#endif /* WITH_THREAD */
439
Guido van Rossumd922fa42003-04-15 14:10:09 +0000440 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000441 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000442
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000443 /* Now we decref the exception classes. After this point nothing
444 can raise an exception. That's okay, because each Fini() method
445 below has been checked to make sure no exceptions are ever
446 raised.
447 */
448
449 _PyExc_Fini();
450
Guido van Rossumd922fa42003-04-15 14:10:09 +0000451 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000452 PyThreadState_Swap(NULL);
453 PyInterpreterState_Delete(interp);
454
Guido van Rossumd922fa42003-04-15 14:10:09 +0000455 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000456 PyMethod_Fini();
457 PyFrame_Fini();
458 PyCFunction_Fini();
459 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000460 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000461 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000462 PyString_Fini();
Neal Norwitz6968b052007-02-27 19:02:19 +0000463 PyBytes_Fini();
Guido van Rossumddefaf32007-01-14 03:31:43 +0000464 PyLong_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000465 PyFloat_Fini();
466
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000467 /* Cleanup Unicode implementation */
468 _PyUnicode_Fini();
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000469
Guido van Rossumcc283f51997-08-05 02:22:03 +0000470 /* XXX Still allocated:
471 - various static ad-hoc pointers to interned strings
472 - int and float free list blocks
473 - whatever various modules and libraries allocate
474 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000475
476 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000477
Tim Peters269b2a62003-04-17 19:52:29 +0000478#ifdef Py_TRACE_REFS
479 /* Display addresses (& refcnts) of all objects still alive.
480 * An address can be used to find the repr of the object, printed
481 * above by _Py_PrintReferences.
482 */
483 if (Py_GETENV("PYTHONDUMPREFS"))
484 _Py_PrintReferenceAddresses(stderr);
485#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000486#ifdef PYMALLOC_DEBUG
487 if (Py_GETENV("PYTHONMALLOCSTATS"))
488 _PyObject_DebugMallocStats();
489#endif
490
Guido van Rossumcc283f51997-08-05 02:22:03 +0000491 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000492}
493
494/* Create and initialize a new interpreter and thread, and return the
495 new thread. This requires that Py_Initialize() has been called
496 first.
497
498 Unsuccessful initialization yields a NULL pointer. Note that *no*
499 exception information is available even in this case -- the
500 exception information is held in the thread, and there is no
501 thread.
502
503 Locking: as above.
504
505*/
506
507PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000508Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000509{
510 PyInterpreterState *interp;
511 PyThreadState *tstate, *save_tstate;
512 PyObject *bimod, *sysmod;
513
514 if (!initialized)
515 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
516
517 interp = PyInterpreterState_New();
518 if (interp == NULL)
519 return NULL;
520
521 tstate = PyThreadState_New(interp);
522 if (tstate == NULL) {
523 PyInterpreterState_Delete(interp);
524 return NULL;
525 }
526
527 save_tstate = PyThreadState_Swap(tstate);
528
529 /* XXX The following is lax in error checking */
530
531 interp->modules = PyDict_New();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000532 interp->modules_reloading = PyDict_New();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000533
534 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
535 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000536 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000537 if (interp->builtins == NULL)
538 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000539 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000540 }
541 sysmod = _PyImport_FindExtension("sys", "sys");
542 if (bimod != NULL && sysmod != NULL) {
543 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000544 if (interp->sysdict == NULL)
545 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000546 Py_INCREF(interp->sysdict);
547 PySys_SetPath(Py_GetPath());
548 PyDict_SetItemString(interp->sysdict, "modules",
549 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000550 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000551 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000552 if (!Py_NoSiteFlag)
553 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000554 }
555
556 if (!PyErr_Occurred())
557 return tstate;
558
Thomas Wouters89f507f2006-12-13 04:49:30 +0000559handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000560 /* Oops, it didn't work. Undo it all. */
561
562 PyErr_Print();
563 PyThreadState_Clear(tstate);
564 PyThreadState_Swap(save_tstate);
565 PyThreadState_Delete(tstate);
566 PyInterpreterState_Delete(interp);
567
568 return NULL;
569}
570
571/* Delete an interpreter and its last thread. This requires that the
572 given thread state is current, that the thread has no remaining
573 frames, and that it is its interpreter's only remaining thread.
574 It is a fatal error to violate these constraints.
575
576 (Py_Finalize() doesn't have these constraints -- it zaps
577 everything, regardless.)
578
579 Locking: as above.
580
581*/
582
583void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000584Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000585{
586 PyInterpreterState *interp = tstate->interp;
587
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000588 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000589 Py_FatalError("Py_EndInterpreter: thread is not current");
590 if (tstate->frame != NULL)
591 Py_FatalError("Py_EndInterpreter: thread still has a frame");
592 if (tstate != interp->tstate_head || tstate->next != NULL)
593 Py_FatalError("Py_EndInterpreter: not the last thread");
594
595 PyImport_Cleanup();
596 PyInterpreterState_Clear(interp);
597 PyThreadState_Swap(NULL);
598 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000599}
600
601static char *progname = "python";
602
603void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000604Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000605{
606 if (pn && *pn)
607 progname = pn;
608}
609
610char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000611Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000612{
613 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000614}
615
Guido van Rossuma61691e1998-02-06 22:27:24 +0000616static char *default_home = NULL;
617
618void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000619Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000620{
621 default_home = home;
622}
623
624char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000625Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000626{
627 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000628 if (home == NULL && !Py_IgnoreEnvironmentFlag)
629 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000630 return home;
631}
632
Guido van Rossum6135a871995-01-09 17:53:26 +0000633/* Create __main__ module */
634
635static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000636initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000637{
Guido van Rossum82598051997-03-05 00:20:32 +0000638 PyObject *m, *d;
639 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000640 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000641 Py_FatalError("can't create __main__ module");
642 d = PyModule_GetDict(m);
643 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000644 PyObject *bimod = PyImport_ImportModule("__builtin__");
645 if (bimod == NULL ||
646 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000647 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000648 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000649 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000650}
651
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000652/* Import the site module (not into __main__ though) */
653
654static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000655initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000656{
657 PyObject *m, *f;
658 m = PyImport_ImportModule("site");
659 if (m == NULL) {
660 f = PySys_GetObject("stderr");
661 if (Py_VerboseFlag) {
662 PyFile_WriteString(
663 "'import site' failed; traceback:\n", f);
664 PyErr_Print();
665 }
666 else {
667 PyFile_WriteString(
668 "'import site' failed; use -v for traceback\n", f);
669 PyErr_Clear();
670 }
671 }
672 else {
673 Py_DECREF(m);
674 }
675}
676
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000677/* Parse input from a file and execute it */
678
679int
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000680PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000681 PyCompilerFlags *flags)
682{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000683 if (filename == NULL)
684 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000685 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000686 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000687 if (closeit)
688 fclose(fp);
689 return err;
690 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000691 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000692 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000693}
694
695int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000696PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000697{
Guido van Rossum82598051997-03-05 00:20:32 +0000698 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000699 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000700 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000701
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000702 if (flags == NULL) {
703 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000704 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000705 }
Guido van Rossum82598051997-03-05 00:20:32 +0000706 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000707 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000708 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
709 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000710 }
Guido van Rossum82598051997-03-05 00:20:32 +0000711 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000712 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000713 PySys_SetObject("ps2", v = PyString_FromString("... "));
714 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000715 }
716 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000717 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000718 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000719 if (ret == E_EOF)
720 return 0;
721 /*
722 if (ret == E_NOMEM)
723 return -1;
724 */
725 }
726}
727
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000728/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000729#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000730 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Guido van Rossum45aecf42006-03-15 04:58:47 +0000731 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000732
Thomas Wouters89f507f2006-12-13 04:49:30 +0000733#if 0
734/* Keep an example of flags with future keyword support. */
735#define PARSER_FLAGS(flags) \
736 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
737 PyPARSE_DONT_IMPLY_DEDENT : 0) \
738 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
739 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
740#endif
741
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000742int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000743PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000744{
Guido van Rossum82598051997-03-05 00:20:32 +0000745 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000746 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000747 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000748 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000749 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000750
Guido van Rossum82598051997-03-05 00:20:32 +0000751 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000752 if (v != NULL) {
753 v = PyObject_Str(v);
754 if (v == NULL)
755 PyErr_Clear();
756 else if (PyString_Check(v))
757 ps1 = PyString_AsString(v);
758 }
Guido van Rossum82598051997-03-05 00:20:32 +0000759 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000760 if (w != NULL) {
761 w = PyObject_Str(w);
762 if (w == NULL)
763 PyErr_Clear();
764 else if (PyString_Check(w))
765 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000766 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000767 arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000768 if (arena == NULL) {
769 Py_XDECREF(v);
770 Py_XDECREF(w);
771 return -1;
772 }
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000773 mod = PyParser_ASTFromFile(fp, filename,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000774 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000775 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000776 Py_XDECREF(v);
777 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000778 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000779 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000780 if (errcode == E_EOF) {
781 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000782 return E_EOF;
783 }
Guido van Rossum82598051997-03-05 00:20:32 +0000784 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000785 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000786 }
Guido van Rossum82598051997-03-05 00:20:32 +0000787 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000788 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000789 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000790 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000791 }
Guido van Rossum82598051997-03-05 00:20:32 +0000792 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000793 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000794 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000795 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000796 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000797 return -1;
798 }
Guido van Rossum82598051997-03-05 00:20:32 +0000799 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000800 return 0;
801}
802
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000803/* Check whether a file maybe a pyc file: Look at the extension,
804 the file type, and, if we may close it, at the first few bytes. */
805
806static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000807maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000808{
809 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
810 return 1;
811
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000812 /* Only look into the file if we are allowed to close it, since
813 it then should also be seekable. */
814 if (closeit) {
815 /* Read only two bytes of the magic. If the file was opened in
816 text mode, the bytes 3 and 4 of the magic (\r\n) might not
817 be read as they are on disk. */
818 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
819 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000820 /* Mess: In case of -x, the stream is NOT at its start now,
821 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000822 which makes the current stream position formally undefined,
823 and a x-platform nightmare.
824 Unfortunately, we have no direct way to know whether -x
825 was specified. So we use a terrible hack: if the current
826 stream position is not 0, we assume -x was specified, and
827 give up. Bug 132850 on SourceForge spells out the
828 hopelessness of trying anything else (fseek and ftell
829 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000830 */
Tim Peters3e876562001-02-11 04:35:39 +0000831 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000832 if (ftell(fp) == 0) {
833 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000834 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000835 ispyc = 1;
836 rewind(fp);
837 }
Tim Peters3e876562001-02-11 04:35:39 +0000838 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000839 }
840 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000841}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000842
Guido van Rossum0df002c2000-08-27 19:21:52 +0000843int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000844PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000845 PyCompilerFlags *flags)
846{
Guido van Rossum82598051997-03-05 00:20:32 +0000847 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000848 const char *ext;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000849 int set_file_name = 0, ret;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000850
Guido van Rossum82598051997-03-05 00:20:32 +0000851 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000852 if (m == NULL)
853 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000854 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000855 if (PyDict_GetItemString(d, "__file__") == NULL) {
856 PyObject *f = PyString_FromString(filename);
857 if (f == NULL)
858 return -1;
859 if (PyDict_SetItemString(d, "__file__", f) < 0) {
860 Py_DECREF(f);
861 return -1;
862 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000863 set_file_name = 1;
Fred Drake8ed02042002-10-17 21:24:58 +0000864 Py_DECREF(f);
865 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000866 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000867 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000868 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000869 if (closeit)
870 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000871 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000872 fprintf(stderr, "python: Can't reopen .pyc file\n");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000873 ret = -1;
874 goto done;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000875 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000876 /* Turn on optimization if a .pyo file is given */
877 if (strcmp(ext, ".pyo") == 0)
878 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000879 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000880 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000881 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000882 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000883 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000884 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000885 PyErr_Print();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000886 ret = -1;
887 goto done;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000888 }
Guido van Rossum82598051997-03-05 00:20:32 +0000889 Py_DECREF(v);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000890 ret = 0;
891 done:
892 if (set_file_name && PyDict_DelItemString(d, "__file__"))
893 PyErr_Clear();
894 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000895}
896
897int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000898PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000899{
Guido van Rossum82598051997-03-05 00:20:32 +0000900 PyObject *m, *d, *v;
901 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000902 if (m == NULL)
903 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000904 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000905 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000906 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000907 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000908 return -1;
909 }
Guido van Rossum82598051997-03-05 00:20:32 +0000910 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000911 return 0;
912}
913
Barry Warsaw035574d1997-08-29 22:07:17 +0000914static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000915parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
916 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000917{
918 long hold;
919 PyObject *v;
920
921 /* old style errors */
922 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000923 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000924 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000925
926 /* new style errors. `err' is an instance */
927
928 if (! (v = PyObject_GetAttrString(err, "msg")))
929 goto finally;
930 *message = v;
931
932 if (!(v = PyObject_GetAttrString(err, "filename")))
933 goto finally;
934 if (v == Py_None)
935 *filename = NULL;
936 else if (! (*filename = PyString_AsString(v)))
937 goto finally;
938
939 Py_DECREF(v);
940 if (!(v = PyObject_GetAttrString(err, "lineno")))
941 goto finally;
942 hold = PyInt_AsLong(v);
943 Py_DECREF(v);
944 v = NULL;
945 if (hold < 0 && PyErr_Occurred())
946 goto finally;
947 *lineno = (int)hold;
948
949 if (!(v = PyObject_GetAttrString(err, "offset")))
950 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000951 if (v == Py_None) {
952 *offset = -1;
953 Py_DECREF(v);
954 v = NULL;
955 } else {
956 hold = PyInt_AsLong(v);
957 Py_DECREF(v);
958 v = NULL;
959 if (hold < 0 && PyErr_Occurred())
960 goto finally;
961 *offset = (int)hold;
962 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000963
964 if (!(v = PyObject_GetAttrString(err, "text")))
965 goto finally;
966 if (v == Py_None)
967 *text = NULL;
968 else if (! (*text = PyString_AsString(v)))
969 goto finally;
970 Py_DECREF(v);
971 return 1;
972
973finally:
974 Py_XDECREF(v);
975 return 0;
976}
977
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000978void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000979PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000980{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000981 PyErr_PrintEx(1);
982}
983
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000984static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000985print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000986{
987 char *nl;
988 if (offset >= 0) {
989 if (offset > 0 && offset == (int)strlen(text))
990 offset--;
991 for (;;) {
992 nl = strchr(text, '\n');
993 if (nl == NULL || nl-text >= offset)
994 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000995 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000996 text = nl+1;
997 }
998 while (*text == ' ' || *text == '\t') {
999 text++;
1000 offset--;
1001 }
1002 }
1003 PyFile_WriteString(" ", f);
1004 PyFile_WriteString(text, f);
1005 if (*text == '\0' || text[strlen(text)-1] != '\n')
1006 PyFile_WriteString("\n", f);
1007 if (offset == -1)
1008 return;
1009 PyFile_WriteString(" ", f);
1010 offset--;
1011 while (offset > 0) {
1012 PyFile_WriteString(" ", f);
1013 offset--;
1014 }
1015 PyFile_WriteString("^\n", f);
1016}
1017
Guido van Rossum66e8e862001-03-23 17:54:43 +00001018static void
1019handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001020{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001021 PyObject *exception, *value, *tb;
1022 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001023
Guido van Rossumd8faa362007-04-27 19:54:29 +00001024 if (Py_InspectFlag)
1025 /* Don't exit if -i flag was given. This flag is set to 0
1026 * when entering interactive mode for inspecting. */
1027 return;
1028
Guido van Rossum66e8e862001-03-23 17:54:43 +00001029 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001030 fflush(stdout);
1031 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001032 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001033 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001034 /* The error code should be in the `code' attribute. */
1035 PyObject *code = PyObject_GetAttrString(value, "code");
1036 if (code) {
1037 Py_DECREF(value);
1038 value = code;
1039 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001040 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001041 }
1042 /* If we failed to dig out the 'code' attribute,
1043 just let the else clause below print the error. */
1044 }
1045 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001046 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001047 else {
1048 PyObject_Print(value, stderr, Py_PRINT_RAW);
1049 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001050 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001051 }
Tim Peterscf615b52003-04-19 18:47:02 +00001052 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001053 /* Restore and clear the exception info, in order to properly decref
1054 * the exception, value, and traceback. If we just exit instead,
1055 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1056 * some finalizers from running.
1057 */
Tim Peterscf615b52003-04-19 18:47:02 +00001058 PyErr_Restore(exception, value, tb);
1059 PyErr_Clear();
1060 Py_Exit(exitcode);
1061 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001062}
1063
1064void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001065PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001066{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001067 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001068
1069 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1070 handle_system_exit();
1071 }
Guido van Rossum82598051997-03-05 00:20:32 +00001072 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001073 if (exception == NULL)
1074 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001075 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001076 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001077 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001078 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001079 if (set_sys_last_vars) {
1080 PySys_SetObject("last_type", exception);
1081 PySys_SetObject("last_value", v);
1082 PySys_SetObject("last_traceback", tb);
1083 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001084 hook = PySys_GetObject("excepthook");
1085 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001086 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001087 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001088 PyObject *result = PyEval_CallObject(hook, args);
1089 if (result == NULL) {
1090 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001091 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1092 handle_system_exit();
1093 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001094 PyErr_Fetch(&exception2, &v2, &tb2);
1095 PyErr_NormalizeException(&exception2, &v2, &tb2);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001096 /* It should not be possible for exception2 or v2
1097 to be NULL. However PyErr_Display() can't
1098 tolerate NULLs, so just be safe. */
1099 if (exception2 == NULL) {
1100 exception2 = Py_None;
1101 Py_INCREF(exception2);
1102 }
1103 if (v2 == NULL) {
1104 v2 = Py_None;
1105 Py_INCREF(v2);
1106 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001107 fflush(stdout);
1108 PySys_WriteStderr("Error in sys.excepthook:\n");
1109 PyErr_Display(exception2, v2, tb2);
1110 PySys_WriteStderr("\nOriginal exception was:\n");
1111 PyErr_Display(exception, v, tb);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001112 Py_DECREF(exception2);
1113 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001114 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001115 }
1116 Py_XDECREF(result);
1117 Py_XDECREF(args);
1118 } else {
1119 PySys_WriteStderr("sys.excepthook is missing\n");
1120 PyErr_Display(exception, v, tb);
1121 }
1122 Py_XDECREF(exception);
1123 Py_XDECREF(v);
1124 Py_XDECREF(tb);
1125}
1126
Thomas Wouters477c8d52006-05-27 19:21:47 +00001127void
1128PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001129{
1130 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001131 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001132 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001133 if (f == NULL)
Martin v. Löwis5b222132007-06-10 09:51:05 +00001134 _PyObject_Dump(value);
1135 if (f == NULL)
Guido van Rossum3165fe61992-09-25 21:59:05 +00001136 fprintf(stderr, "lost sys.stderr\n");
1137 else {
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001138 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001139 if (tb && tb != Py_None)
1140 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001141 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001142 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001143 {
Guido van Rossum82598051997-03-05 00:20:32 +00001144 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001145 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001146 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001147 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001148 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001149 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001150 else {
1151 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001152 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001153 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001154 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001155 else
Guido van Rossum82598051997-03-05 00:20:32 +00001156 PyFile_WriteString(filename, f);
1157 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001158 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001159 PyFile_WriteString(buf, f);
1160 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001161 if (text != NULL)
1162 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001163 Py_DECREF(value);
1164 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001165 /* Can't be bothered to check all those
1166 PyFile_WriteString() calls */
1167 if (PyErr_Occurred())
1168 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001169 }
1170 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001171 if (err) {
1172 /* Don't do anything else */
1173 }
Brett Cannonbf364092006-03-01 04:25:17 +00001174 else if (PyExceptionClass_Check(exception)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001175 PyObject* moduleName;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001176 char* className = PyExceptionClass_Name(exception);
1177 if (className != NULL) {
1178 char *dot = strrchr(className, '.');
1179 if (dot != NULL)
1180 className = dot+1;
1181 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001182
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001183 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001184 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001185 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001186 else {
1187 char* modstr = PyString_AsString(moduleName);
Neal Norwitz2633c692007-02-26 22:22:47 +00001188 if (modstr && strcmp(modstr, "__builtin__"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001189 {
1190 err = PyFile_WriteString(modstr, f);
1191 err += PyFile_WriteString(".", f);
1192 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001193 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001194 }
1195 if (err == 0) {
1196 if (className == NULL)
1197 err = PyFile_WriteString("<unknown>", f);
1198 else
Brett Cannonbf364092006-03-01 04:25:17 +00001199 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001200 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001201 }
1202 else
1203 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001204 if (err == 0 && (value != Py_None)) {
1205 PyObject *s = PyObject_Str(value);
1206 /* only print colon if the str() of the
1207 object is not the empty string
1208 */
1209 if (s == NULL)
1210 err = -1;
1211 else if (!PyString_Check(s) ||
1212 PyString_GET_SIZE(s) != 0)
1213 err = PyFile_WriteString(": ", f);
1214 if (err == 0)
1215 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1216 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001217 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00001218 /* try to write a newline in any case */
1219 err += PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001220 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001221 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001222 /* If an error happened here, don't show it.
1223 XXX This is wrong, but too many callers rely on this behavior. */
1224 if (err != 0)
1225 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001226}
1227
Guido van Rossum82598051997-03-05 00:20:32 +00001228PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001229PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001230 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001231{
Neal Norwitze92fba02006-03-04 18:52:26 +00001232 PyObject *ret = NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001233 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001234 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001235 if (arena == NULL)
1236 return NULL;
1237
1238 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001239 if (mod != NULL)
1240 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001241 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001242 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001243}
1244
1245PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001246PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001247 PyObject *locals, int closeit, PyCompilerFlags *flags)
1248{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001249 PyObject *ret;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001250 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001251 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001252 if (arena == NULL)
1253 return NULL;
1254
1255 mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
1256 flags, NULL, arena);
Guido van Rossumd8faa362007-04-27 19:54:29 +00001257 if (closeit)
1258 fclose(fp);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001259 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001260 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001261 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001262 }
Neal Norwitze92fba02006-03-04 18:52:26 +00001263 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001264 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001265 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001266}
1267
Guido van Rossum82598051997-03-05 00:20:32 +00001268static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001269run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001270 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001271{
Guido van Rossum82598051997-03-05 00:20:32 +00001272 PyCodeObject *co;
1273 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001274 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001275 if (co == NULL)
1276 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001277 v = PyEval_EvalCode(co, globals, locals);
1278 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001279 return v;
1280}
1281
Guido van Rossum82598051997-03-05 00:20:32 +00001282static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001283run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001284 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001285{
Guido van Rossum82598051997-03-05 00:20:32 +00001286 PyCodeObject *co;
1287 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001288 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001289 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001290
Guido van Rossum82598051997-03-05 00:20:32 +00001291 magic = PyMarshal_ReadLongFromFile(fp);
1292 if (magic != PyImport_GetMagicNumber()) {
1293 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001294 "Bad magic number in .pyc file");
1295 return NULL;
1296 }
Guido van Rossum82598051997-03-05 00:20:32 +00001297 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001298 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001299 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001300 if (v == NULL || !PyCode_Check(v)) {
1301 Py_XDECREF(v);
1302 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001303 "Bad code object in .pyc file");
1304 return NULL;
1305 }
Guido van Rossum82598051997-03-05 00:20:32 +00001306 co = (PyCodeObject *)v;
1307 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001308 if (v && flags)
1309 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001310 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001311 return v;
1312}
1313
Guido van Rossum82598051997-03-05 00:20:32 +00001314PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001315Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001316 PyCompilerFlags *flags)
1317{
Guido van Rossum82598051997-03-05 00:20:32 +00001318 PyCodeObject *co;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001319 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001320 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001321 if (arena == NULL)
1322 return NULL;
1323
1324 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001325 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001326 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001327 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001328 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001329 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001330 PyObject *result = PyAST_mod2obj(mod);
1331 PyArena_Free(arena);
1332 return result;
1333 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001334 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001335 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001336 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001337}
1338
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001339struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001340Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001341{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001342 struct symtable *st;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001343 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001344 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001345 if (arena == NULL)
1346 return NULL;
1347
1348 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001349 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001350 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001351 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001352 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001353 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001354 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001355 return st;
1356}
1357
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001358/* Preferred access to parser is through AST. */
1359mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001360PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001361 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001362{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001363 mod_ty mod;
1364 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001365 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001366 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001367 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001368 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001369 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001370 PyNode_Free(n);
1371 return mod;
1372 }
1373 else {
1374 err_input(&err);
1375 return NULL;
1376 }
1377}
1378
1379mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001380PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001381 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001382 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001383{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001384 mod_ty mod;
1385 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001386 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1387 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001388 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001389 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001390 PyNode_Free(n);
1391 return mod;
1392 }
1393 else {
1394 err_input(&err);
1395 if (errcode)
1396 *errcode = err.error;
1397 return NULL;
1398 }
1399}
1400
Guido van Rossuma110aa61994-08-29 12:50:44 +00001401/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001402
Guido van Rossuma110aa61994-08-29 12:50:44 +00001403node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001404PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001405{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001406 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001407 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1408 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001409 if (n == NULL)
1410 err_input(&err);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001411
Guido van Rossuma110aa61994-08-29 12:50:44 +00001412 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001413}
1414
Guido van Rossuma110aa61994-08-29 12:50:44 +00001415/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001416
Guido van Rossuma110aa61994-08-29 12:50:44 +00001417node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001418PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001419{
Tim Petersfe2127d2001-07-16 05:37:24 +00001420 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001421 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1422 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001423 if (n == NULL)
1424 err_input(&err);
1425 return n;
1426}
1427
1428node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001429PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001430 int start, int flags)
1431{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001432 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001433 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1434 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001435 if (n == NULL)
1436 err_input(&err);
1437 return n;
1438}
1439
1440node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001441PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001442{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001443 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001444}
1445
Guido van Rossum66ebd912003-04-17 16:02:26 +00001446/* May want to move a more generalized form of this to parsetok.c or
1447 even parser modules. */
1448
1449void
1450PyParser_SetError(perrdetail *err)
1451{
1452 err_input(err);
1453}
1454
Guido van Rossuma110aa61994-08-29 12:50:44 +00001455/* Set the error appropriate to the given input error code (see errcode.h) */
1456
1457static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001458err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001459{
Fred Drake85f36392000-07-11 17:53:00 +00001460 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001461 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001462 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001463 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001464 switch (err->error) {
1465 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001466 errtype = PyExc_IndentationError;
1467 if (err->expected == INDENT)
1468 msg = "expected an indented block";
1469 else if (err->token == INDENT)
1470 msg = "unexpected indent";
1471 else if (err->token == DEDENT)
1472 msg = "unexpected unindent";
1473 else {
1474 errtype = PyExc_SyntaxError;
1475 msg = "invalid syntax";
1476 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001477 break;
1478 case E_TOKEN:
1479 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001480 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001481 case E_EOFS:
1482 msg = "EOF while scanning triple-quoted string";
1483 break;
1484 case E_EOLS:
1485 msg = "EOL while scanning single-quoted string";
1486 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001487 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001488 if (!PyErr_Occurred())
1489 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001490 return;
1491 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001492 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001493 return;
1494 case E_EOF:
1495 msg = "unexpected EOF while parsing";
1496 break;
Fred Drake85f36392000-07-11 17:53:00 +00001497 case E_TABSPACE:
1498 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001499 msg = "inconsistent use of tabs and spaces in indentation";
1500 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001501 case E_OVERFLOW:
1502 msg = "expression too long";
1503 break;
Fred Drake85f36392000-07-11 17:53:00 +00001504 case E_DEDENT:
1505 errtype = PyExc_IndentationError;
1506 msg = "unindent does not match any outer indentation level";
1507 break;
1508 case E_TOODEEP:
1509 errtype = PyExc_IndentationError;
1510 msg = "too many levels of indentation";
1511 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001512 case E_DECODE: {
1513 PyObject *type, *value, *tb;
1514 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001515 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001516 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001517 if (u != NULL) {
1518 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001519 }
1520 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001521 if (msg == NULL)
1522 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001523 Py_XDECREF(type);
1524 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001525 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001526 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001527 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001528 case E_LINECONT:
1529 msg = "unexpected character after line continuation character";
1530 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001531 default:
1532 fprintf(stderr, "error=%d\n", err->error);
1533 msg = "unknown parsing error";
1534 break;
1535 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001536 v = Py_BuildValue("(ziiz)", err->filename,
1537 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001538 if (err->text != NULL) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001539 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001540 err->text = NULL;
1541 }
1542 w = NULL;
1543 if (v != NULL)
1544 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001545 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001546 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001547 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001548 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001549}
1550
1551/* Print fatal error message and abort */
1552
1553void
Tim Peters7c321a82002-07-09 02:57:01 +00001554Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001555{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001556 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001557#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001558 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001559 OutputDebugString(msg);
1560 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001561#ifdef _DEBUG
1562 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001563#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001564#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001565 abort();
1566}
1567
1568/* Clean up and exit */
1569
Guido van Rossuma110aa61994-08-29 12:50:44 +00001570#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001571#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001572#endif
1573
Collin Winter670e6922007-03-21 02:57:17 +00001574static void (*pyexitfunc)(void) = NULL;
1575/* For the atexit module. */
1576void _Py_PyAtExit(void (*func)(void))
1577{
1578 pyexitfunc = func;
1579}
1580
1581static void
1582call_py_exitfuncs(void)
1583{
1584 if (pyexitfunc == NULL)
1585 return;
1586
1587 (*pyexitfunc)();
1588 PyErr_Clear();
1589}
1590
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001591#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001592static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001593static int nexitfuncs = 0;
1594
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001595int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001596{
1597 if (nexitfuncs >= NEXITFUNCS)
1598 return -1;
1599 exitfuncs[nexitfuncs++] = func;
1600 return 0;
1601}
1602
Guido van Rossumcc283f51997-08-05 02:22:03 +00001603static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001604call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001605{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001606 while (nexitfuncs > 0)
1607 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001608
1609 fflush(stdout);
1610 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001611}
1612
1613void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001614Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001615{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001616 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001617
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001618 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001619}
1620
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001621static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001622initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001623{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001624#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001625 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001626#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001627#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001628 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001629#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001630#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001631 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001632#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001633 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001634}
1635
Guido van Rossum7433b121997-02-14 19:45:36 +00001636
1637/*
1638 * The file descriptor fd is considered ``interactive'' if either
1639 * a) isatty(fd) is TRUE, or
1640 * b) the -i flag was given, and the filename associated with
1641 * the descriptor is NULL or "<stdin>" or "???".
1642 */
1643int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001644Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001645{
1646 if (isatty((int)fileno(fp)))
1647 return 1;
1648 if (!Py_InteractiveFlag)
1649 return 0;
1650 return (filename == NULL) ||
1651 (strcmp(filename, "<stdin>") == 0) ||
1652 (strcmp(filename, "???") == 0);
1653}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001654
1655
Tim Petersd08e3822003-04-17 15:24:21 +00001656#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001657#if defined(WIN32) && defined(_MSC_VER)
1658
1659/* Stack checking for Microsoft C */
1660
1661#include <malloc.h>
1662#include <excpt.h>
1663
Fred Drakee8de31c2000-08-31 05:38:39 +00001664/*
1665 * Return non-zero when we run out of memory on the stack; zero otherwise.
1666 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001667int
Fred Drake399739f2000-08-31 05:52:44 +00001668PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001669{
1670 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001671 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001672 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001673 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001674 return 0;
1675 } __except (EXCEPTION_EXECUTE_HANDLER) {
1676 /* just ignore all errors */
1677 }
1678 return 1;
1679}
1680
1681#endif /* WIN32 && _MSC_VER */
1682
1683/* Alternate implementations can be added here... */
1684
1685#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001686
1687
1688/* Wrappers around sigaction() or signal(). */
1689
1690PyOS_sighandler_t
1691PyOS_getsig(int sig)
1692{
1693#ifdef HAVE_SIGACTION
1694 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001695 if (sigaction(sig, NULL, &context) == -1)
1696 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001697 return context.sa_handler;
1698#else
1699 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001700/* Special signal handling for the secure CRT in Visual Studio 2005 */
1701#if defined(_MSC_VER) && _MSC_VER >= 1400
1702 switch (sig) {
1703 /* Only these signals are valid */
1704 case SIGINT:
1705 case SIGILL:
1706 case SIGFPE:
1707 case SIGSEGV:
1708 case SIGTERM:
1709 case SIGBREAK:
1710 case SIGABRT:
1711 break;
1712 /* Don't call signal() with other values or it will assert */
1713 default:
1714 return SIG_ERR;
1715 }
1716#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001717 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001718 if (handler != SIG_ERR)
1719 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001720 return handler;
1721#endif
1722}
1723
1724PyOS_sighandler_t
1725PyOS_setsig(int sig, PyOS_sighandler_t handler)
1726{
1727#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001728 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001729 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001730 sigemptyset(&context.sa_mask);
1731 context.sa_flags = 0;
1732 if (sigaction(sig, &context, &ocontext) == -1)
1733 return SIG_ERR;
1734 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001735#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001736 PyOS_sighandler_t oldhandler;
1737 oldhandler = signal(sig, handler);
1738#ifdef HAVE_SIGINTERRUPT
1739 siginterrupt(sig, 1);
1740#endif
1741 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001742#endif
1743}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001744
1745/* Deprecated C API functions still provided for binary compatiblity */
1746
1747#undef PyParser_SimpleParseFile
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001748PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001749PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1750{
1751 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1752}
1753
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001754#undef PyParser_SimpleParseString
1755PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001756PyParser_SimpleParseString(const char *str, int start)
1757{
1758 return PyParser_SimpleParseStringFlags(str, start, 0);
1759}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001760
1761#undef PyRun_AnyFile
1762PyAPI_FUNC(int)
1763PyRun_AnyFile(FILE *fp, const char *name)
1764{
1765 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1766}
1767
1768#undef PyRun_AnyFileEx
1769PyAPI_FUNC(int)
1770PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1771{
1772 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1773}
1774
1775#undef PyRun_AnyFileFlags
1776PyAPI_FUNC(int)
1777PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1778{
1779 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1780}
1781
1782#undef PyRun_File
1783PyAPI_FUNC(PyObject *)
1784PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1785{
1786 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1787}
1788
1789#undef PyRun_FileEx
1790PyAPI_FUNC(PyObject *)
1791PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1792{
1793 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1794}
1795
1796#undef PyRun_FileFlags
1797PyAPI_FUNC(PyObject *)
1798PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1799 PyCompilerFlags *flags)
1800{
1801 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1802}
1803
1804#undef PyRun_SimpleFile
1805PyAPI_FUNC(int)
1806PyRun_SimpleFile(FILE *f, const char *p)
1807{
1808 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1809}
1810
1811#undef PyRun_SimpleFileEx
1812PyAPI_FUNC(int)
1813PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1814{
1815 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1816}
1817
1818
1819#undef PyRun_String
1820PyAPI_FUNC(PyObject *)
1821PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1822{
1823 return PyRun_StringFlags(str, s, g, l, NULL);
1824}
1825
1826#undef PyRun_SimpleString
1827PyAPI_FUNC(int)
1828PyRun_SimpleString(const char *s)
1829{
1830 return PyRun_SimpleStringFlags(s, NULL);
1831}
1832
1833#undef Py_CompileString
1834PyAPI_FUNC(PyObject *)
1835Py_CompileString(const char *str, const char *p, int s)
1836{
1837 return Py_CompileStringFlags(str, p, s, NULL);
1838}
1839
1840#undef PyRun_InteractiveOne
1841PyAPI_FUNC(int)
1842PyRun_InteractiveOne(FILE *f, const char *p)
1843{
1844 return PyRun_InteractiveOneFlags(f, p, NULL);
1845}
1846
1847#undef PyRun_InteractiveLoop
1848PyAPI_FUNC(int)
1849PyRun_InteractiveLoop(FILE *f, const char *p)
1850{
1851 return PyRun_InteractiveLoopFlags(f, p, NULL);
1852}
1853
1854#ifdef __cplusplus
1855}
1856#endif