blob: 96602ff15b5d32cffd96ca4b1ff292e49b8b06a0 [file] [log] [blame]
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001
2/* Python interpreter top-level routines, including init/exit */
3
Guido van Rossum82598051997-03-05 00:20:32 +00004#include "Python.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +00005
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00006#include "Python-ast.h"
Kristján Valur Jónsson67387fb2007-04-25 00:17:39 +00007#undef Yield /* undefine macro conflicting with winbase.h */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00008#include "grammar.h"
9#include "node.h"
Fred Drake85f36392000-07-11 17:53:00 +000010#include "token.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000011#include "parsetok.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000012#include "errcode.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000013#include "code.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000014#include "compile.h"
Jeremy Hylton4b38da62001-02-02 18:19:15 +000015#include "symtable.h"
Neal Norwitzadb69fc2005-12-17 20:54:49 +000016#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000017#include "ast.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000018#include "eval.h"
Guido van Rossumfdef2711994-09-14 13:31:04 +000019#include "marshal.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000020
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000021#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000022#include <signal.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +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
Tim Peters62e97f02006-03-28 21:44:32 +000036#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000037#else /* Py_REF_DEBUG */
Tim Peters62e97f02006-03-28 21:44:32 +000038#define PRINT_TOTAL_REFS() fprintf(stderr, \
39 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
Armin Rigoe1709372006-04-12 17:06:05 +000040 _Py_GetRefTotal())
Neal Norwitz4281cef2006-03-04 19:58:13 +000041#endif
42
Anthony Baxterac6bd462006-04-13 02:06:09 +000043#ifdef __cplusplus
44extern "C" {
45#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);
60static void call_sys_exitfunc(void);
61static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000062extern void _PyUnicode_Init(void);
63extern void _PyUnicode_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000064
Mark Hammond8d98d2c2003-04-19 15:41:53 +000065#ifdef WITH_THREAD
66extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
67extern void _PyGILState_Fini(void);
68#endif /* WITH_THREAD */
69
Guido van Rossum82598051997-03-05 00:20:32 +000070int Py_DebugFlag; /* Needed by parser.c */
71int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000072int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Georg Brandl49aafc92007-03-07 00:34:46 +000073int Py_InspectFlag; /* Needed to determine whether to exit at SystemError */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000074int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000075int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000076int Py_FrozenFlag; /* Needed by getpath.c */
Guido van Rossumb16d1972000-05-01 17:55:15 +000077int Py_UnicodeFlag = 0; /* Needed by compile.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000078int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Tim Peters3caca232001-12-06 06:23:26 +000079/* _XXX Py_QnewFlag should go away in 2.3. It's true iff -Qnew is passed,
80 on the command line, and is used in 2.2 by ceval.c to make all "/" divisions
81 true divisions (which they will be in 2.3). */
82int _Py_QnewFlag = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +000083
Mark Hammonda43fd0c2003-02-19 00:33:33 +000084/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000085 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000086*/
Mark Hammondedd07732003-07-15 23:03:55 +000087static PyObject *warnings_module = NULL;
88
89/* Returns a borrowed reference to the 'warnings' module, or NULL.
90 If the module is returned, it is guaranteed to have been obtained
91 without acquiring the import lock
92*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000093PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000094{
95 PyObject *typ, *val, *tb;
96 PyObject *all_modules;
97 /* If we managed to get the module at init time, just use it */
98 if (warnings_module)
99 return warnings_module;
100 /* If it wasn't available at init time, it may be available
101 now in sys.modules (common scenario is frozen apps: import
102 at init time fails, but the frozen init code sets up sys.path
103 correctly, then does an implicit import of warnings for us
104 */
105 /* Save and restore any exceptions */
106 PyErr_Fetch(&typ, &val, &tb);
107
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000108 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000109 if (all_modules) {
110 warnings_module = PyDict_GetItemString(all_modules, "warnings");
111 /* We keep a ref in the global */
112 Py_XINCREF(warnings_module);
113 }
114 PyErr_Restore(typ, val, tb);
115 return warnings_module;
116}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000117
Guido van Rossum25ce5661997-08-02 03:10:38 +0000118static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000119
Thomas Wouters7e474022000-07-16 12:04:32 +0000120/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000121
122int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000123Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000124{
125 return initialized;
126}
127
Guido van Rossum25ce5661997-08-02 03:10:38 +0000128/* Global initializations. Can be undone by Py_Finalize(). Don't
129 call this twice without an intervening Py_Finalize() call. When
130 initializations fail, a fatal error is issued and the function does
131 not return. On return, the first thread and interpreter state have
132 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000133
Guido van Rossum25ce5661997-08-02 03:10:38 +0000134 Locking: you must hold the interpreter lock while calling this.
135 (If the lock has not yet been initialized, that's equivalent to
136 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000137
Guido van Rossum25ce5661997-08-02 03:10:38 +0000138*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000139
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000140static int
141add_flag(int flag, const char *envs)
142{
143 int env = atoi(envs);
144 if (flag < env)
145 flag = env;
146 if (flag < 1)
147 flag = 1;
148 return flag;
149}
150
Guido van Rossuma027efa1997-05-05 20:56:21 +0000151void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000152Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000153{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000154 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000155 PyThreadState *tstate;
156 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000157 char *p;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000158#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
159 char *codeset;
160 char *saved_locale;
161 PyObject *sys_stream, *sys_isatty;
162#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000163 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000164
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000165 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000166 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000167 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000168
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000169 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000170 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000171 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000172 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000173 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000174 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000175
Guido van Rossuma027efa1997-05-05 20:56:21 +0000176 interp = PyInterpreterState_New();
177 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000178 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000179
Guido van Rossuma027efa1997-05-05 20:56:21 +0000180 tstate = PyThreadState_New(interp);
181 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000182 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000183 (void) PyThreadState_Swap(tstate);
184
Guido van Rossum70d893a2001-08-16 08:21:42 +0000185 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000186
Neal Norwitzb2501f42002-12-31 03:42:13 +0000187 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000188 Py_FatalError("Py_Initialize: can't init frames");
189
Neal Norwitzb2501f42002-12-31 03:42:13 +0000190 if (!_PyInt_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000191 Py_FatalError("Py_Initialize: can't init ints");
192
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000193 _PyFloat_Init();
194
Guido van Rossum25ce5661997-08-02 03:10:38 +0000195 interp->modules = PyDict_New();
196 if (interp->modules == NULL)
197 Py_FatalError("Py_Initialize: can't make modules dictionary");
Collin Winter276887b2007-03-12 16:11:39 +0000198 interp->modules_reloading = PyDict_New();
199 if (interp->modules_reloading == NULL)
200 Py_FatalError("Py_Initialize: can't make modules_reloading dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000201
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000202#ifdef Py_USING_UNICODE
Guido van Rossumc94044c2000-03-10 23:03:54 +0000203 /* Init Unicode implementation; relies on the codec registry */
204 _PyUnicode_Init();
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000205#endif
Guido van Rossumc94044c2000-03-10 23:03:54 +0000206
Barry Warsawf242aa02000-05-25 23:09:49 +0000207 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000208 if (bimod == NULL)
209 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000210 interp->builtins = PyModule_GetDict(bimod);
Neal Norwitz0f7dbf72006-08-12 03:17:41 +0000211 if (interp->builtins == NULL)
212 Py_FatalError("Py_Initialize: can't initialize builtins dict");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000213 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000214
215 sysmod = _PySys_Init();
216 if (sysmod == NULL)
217 Py_FatalError("Py_Initialize: can't initialize sys");
218 interp->sysdict = PyModule_GetDict(sysmod);
Neal Norwitz0f7dbf72006-08-12 03:17:41 +0000219 if (interp->sysdict == NULL)
220 Py_FatalError("Py_Initialize: can't initialize sys dict");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000221 Py_INCREF(interp->sysdict);
222 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000223 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000224 PyDict_SetItemString(interp->sysdict, "modules",
225 interp->modules);
226
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000227 _PyImport_Init();
228
Barry Warsawf242aa02000-05-25 23:09:49 +0000229 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000230 _PyExc_Init();
Barry Warsaw5821bc52001-08-13 23:04:56 +0000231 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000232
Barry Warsaw035574d1997-08-29 22:07:17 +0000233 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000234 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000235
Just van Rossum52e14d62002-12-30 22:08:05 +0000236 _PyImportHooks_Init();
237
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000238 if (install_sigs)
239 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000240
241 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000242 if (!Py_NoSiteFlag)
243 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000244
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000245 /* auto-thread-state API, if available */
246#ifdef WITH_THREAD
247 _PyGILState_Init(interp, tstate);
248#endif /* WITH_THREAD */
249
Mark Hammondedd07732003-07-15 23:03:55 +0000250 warnings_module = PyImport_ImportModule("warnings");
251 if (!warnings_module)
252 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000253
254#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
255 /* On Unix, set the file system encoding according to the
256 user's preference, if the CODESET names a well-known
257 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000258 initialized by other means. Also set the encoding of
259 stdin and stdout if these are terminals. */
260
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000261 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000262 setlocale(LC_CTYPE, "");
263 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;
275 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000276 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000277
278 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000279 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000280 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
281 if (!sys_isatty)
282 PyErr_Clear();
Thomas Woutersafea5292007-01-23 13:42:00 +0000283 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
284 PyFile_Check(sys_stream)) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000285 if (!PyFile_SetEncoding(sys_stream, codeset))
286 Py_FatalError("Cannot set codeset of stdin");
287 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000288 Py_XDECREF(sys_isatty);
289
290 sys_stream = PySys_GetObject("stdout");
291 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
292 if (!sys_isatty)
293 PyErr_Clear();
Thomas Woutersafea5292007-01-23 13:42:00 +0000294 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
295 PyFile_Check(sys_stream)) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000296 if (!PyFile_SetEncoding(sys_stream, codeset))
297 Py_FatalError("Cannot set codeset of stdout");
298 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000299 Py_XDECREF(sys_isatty);
300
Martin v. Löwisea62d252006-04-03 10:56:49 +0000301 sys_stream = PySys_GetObject("stderr");
302 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
303 if (!sys_isatty)
304 PyErr_Clear();
Thomas Woutersafea5292007-01-23 13:42:00 +0000305 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
306 PyFile_Check(sys_stream)) {
Martin v. Löwisea62d252006-04-03 10:56:49 +0000307 if (!PyFile_SetEncoding(sys_stream, codeset))
308 Py_FatalError("Cannot set codeset of stderr");
309 }
310 Py_XDECREF(sys_isatty);
311
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000312 if (!Py_FileSystemDefaultEncoding)
313 Py_FileSystemDefaultEncoding = codeset;
314 else
315 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000316 }
317#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000318}
319
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000320void
321Py_Initialize(void)
322{
323 Py_InitializeEx(1);
324}
325
326
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000327#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000328extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000329#endif
330
Guido van Rossum25ce5661997-08-02 03:10:38 +0000331/* Undo the effect of Py_Initialize().
332
333 Beware: if multiple interpreter and/or thread states exist, these
334 are not wiped out; only the current thread and interpreter state
335 are deleted. But since everything else is deleted, those other
336 interpreter and thread states should no longer be used.
337
338 (XXX We should do better, e.g. wipe out all interpreters and
339 threads.)
340
341 Locking: as above.
342
343*/
344
345void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000346Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000347{
348 PyInterpreterState *interp;
349 PyThreadState *tstate;
350
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000351 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000352 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000353
Tim Peters384fd102001-01-21 03:40:37 +0000354 /* The interpreter is still entirely intact at this point, and the
355 * exit funcs may be relying on that. In particular, if some thread
356 * or exit func is still waiting to do an import, the import machinery
357 * expects Py_IsInitialized() to return true. So don't say the
358 * interpreter is uninitialized until after the exit funcs have run.
359 * Note that Threading.py uses an exit func to do a join on all the
360 * threads created thru it, so this also protects pending imports in
361 * the threads created via Threading.
362 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000363 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000364 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000365
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000366 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000367 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000368 interp = tstate->interp;
369
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000370 /* Disable signal handling */
371 PyOS_FiniInterrupts();
372
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000373 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000374 Py_XDECREF(warnings_module);
375 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000376
Guido van Rossume13ddc92003-04-17 17:29:22 +0000377 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000378 * before all modules are destroyed.
379 * XXX If a __del__ or weakref callback is triggered here, and tries to
380 * XXX import a module, bad things can happen, because Python no
381 * XXX longer believes it's initialized.
382 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
383 * XXX is easy to provoke that way. I've also seen, e.g.,
384 * XXX Exception exceptions.ImportError: 'No module named sha'
385 * XXX in <function callback at 0x008F5718> ignored
386 * XXX but I'm unclear on exactly how that one happens. In any case,
387 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000388 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000389 PyGC_Collect();
Martin v. Löwis45294a92006-04-18 06:24:08 +0000390#ifdef COUNT_ALLOCS
391 /* With COUNT_ALLOCS, it helps to run GC multiple times:
392 each collection might release some types from the type
393 list, so they become garbage. */
394 while (PyGC_Collect() > 0)
395 /* nothing */;
396#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000397
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000398 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000399 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000400
Guido van Rossume13ddc92003-04-17 17:29:22 +0000401 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000402 * new-style class definitions, for example.
403 * XXX This is disabled because it caused too many problems. If
404 * XXX a __del__ or weakref callback triggers here, Python code has
405 * XXX a hard time running, because even the sys module has been
406 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
407 * XXX One symptom is a sequence of information-free messages
408 * XXX coming from threads (if a __del__ or callback is invoked,
409 * XXX other threads can execute too, and any exception they encounter
410 * XXX triggers a comedy of errors as subsystem after subsystem
411 * XXX fails to find what it *expects* to find in sys to help report
412 * XXX the exception and consequent unexpected failures). I've also
413 * XXX seen segfaults then, after adding print statements to the
414 * XXX Python code getting called.
415 */
416#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000417 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000418#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000419
Guido van Rossum1707aad1997-12-08 23:43:45 +0000420 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
421 _PyImport_Fini();
422
423 /* Debugging stuff */
424#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000425 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000426#endif
427
Tim Peters62e97f02006-03-28 21:44:32 +0000428 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000429
Tim Peters9cf25ce2003-04-17 15:21:01 +0000430#ifdef Py_TRACE_REFS
431 /* Display all objects still alive -- this can invoke arbitrary
432 * __repr__ overrides, so requires a mostly-intact interpreter.
433 * Alas, a lot of stuff may still be alive now that will be cleaned
434 * up later.
435 */
Tim Peters269b2a62003-04-17 19:52:29 +0000436 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000437 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000438#endif /* Py_TRACE_REFS */
439
Mark Hammond6cb90292003-04-22 11:18:00 +0000440 /* Cleanup auto-thread-state */
441#ifdef WITH_THREAD
442 _PyGILState_Fini();
443#endif /* WITH_THREAD */
444
Guido van Rossumd922fa42003-04-15 14:10:09 +0000445 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000446 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000447
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000448 /* Now we decref the exception classes. After this point nothing
449 can raise an exception. That's okay, because each Fini() method
450 below has been checked to make sure no exceptions are ever
451 raised.
452 */
453
454 _PyExc_Fini();
455
Guido van Rossumd922fa42003-04-15 14:10:09 +0000456 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000457 PyThreadState_Swap(NULL);
458 PyInterpreterState_Delete(interp);
459
Guido van Rossumd922fa42003-04-15 14:10:09 +0000460 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000461 PyMethod_Fini();
462 PyFrame_Fini();
463 PyCFunction_Fini();
464 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000465 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000466 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000467 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000468 PyInt_Fini();
469 PyFloat_Fini();
470
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000471#ifdef Py_USING_UNICODE
472 /* Cleanup Unicode implementation */
473 _PyUnicode_Fini();
474#endif
475
Guido van Rossumcc283f51997-08-05 02:22:03 +0000476 /* XXX Still allocated:
477 - various static ad-hoc pointers to interned strings
478 - int and float free list blocks
479 - whatever various modules and libraries allocate
480 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000481
482 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000483
Tim Peters269b2a62003-04-17 19:52:29 +0000484#ifdef Py_TRACE_REFS
485 /* Display addresses (& refcnts) of all objects still alive.
486 * An address can be used to find the repr of the object, printed
487 * above by _Py_PrintReferences.
488 */
489 if (Py_GETENV("PYTHONDUMPREFS"))
490 _Py_PrintReferenceAddresses(stderr);
491#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000492#ifdef PYMALLOC_DEBUG
493 if (Py_GETENV("PYTHONMALLOCSTATS"))
494 _PyObject_DebugMallocStats();
495#endif
496
Guido van Rossumcc283f51997-08-05 02:22:03 +0000497 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000498}
499
500/* Create and initialize a new interpreter and thread, and return the
501 new thread. This requires that Py_Initialize() has been called
502 first.
503
504 Unsuccessful initialization yields a NULL pointer. Note that *no*
505 exception information is available even in this case -- the
506 exception information is held in the thread, and there is no
507 thread.
508
509 Locking: as above.
510
511*/
512
513PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000514Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000515{
516 PyInterpreterState *interp;
517 PyThreadState *tstate, *save_tstate;
518 PyObject *bimod, *sysmod;
519
520 if (!initialized)
521 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
522
523 interp = PyInterpreterState_New();
524 if (interp == NULL)
525 return NULL;
526
527 tstate = PyThreadState_New(interp);
528 if (tstate == NULL) {
529 PyInterpreterState_Delete(interp);
530 return NULL;
531 }
532
533 save_tstate = PyThreadState_Swap(tstate);
534
535 /* XXX The following is lax in error checking */
536
537 interp->modules = PyDict_New();
Collin Winter276887b2007-03-12 16:11:39 +0000538 interp->modules_reloading = PyDict_New();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000539
540 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
541 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000542 interp->builtins = PyModule_GetDict(bimod);
Neal Norwitz0c6ae5b2006-08-21 20:16:24 +0000543 if (interp->builtins == NULL)
544 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000545 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000546 }
547 sysmod = _PyImport_FindExtension("sys", "sys");
548 if (bimod != NULL && sysmod != NULL) {
549 interp->sysdict = PyModule_GetDict(sysmod);
Neal Norwitz0c6ae5b2006-08-21 20:16:24 +0000550 if (interp->sysdict == NULL)
551 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000552 Py_INCREF(interp->sysdict);
553 PySys_SetPath(Py_GetPath());
554 PyDict_SetItemString(interp->sysdict, "modules",
555 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000556 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000557 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000558 if (!Py_NoSiteFlag)
559 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000560 }
561
562 if (!PyErr_Occurred())
563 return tstate;
564
Neal Norwitz0c6ae5b2006-08-21 20:16:24 +0000565handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000566 /* Oops, it didn't work. Undo it all. */
567
568 PyErr_Print();
569 PyThreadState_Clear(tstate);
570 PyThreadState_Swap(save_tstate);
571 PyThreadState_Delete(tstate);
572 PyInterpreterState_Delete(interp);
573
574 return NULL;
575}
576
577/* Delete an interpreter and its last thread. This requires that the
578 given thread state is current, that the thread has no remaining
579 frames, and that it is its interpreter's only remaining thread.
580 It is a fatal error to violate these constraints.
581
582 (Py_Finalize() doesn't have these constraints -- it zaps
583 everything, regardless.)
584
585 Locking: as above.
586
587*/
588
589void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000590Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000591{
592 PyInterpreterState *interp = tstate->interp;
593
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000594 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000595 Py_FatalError("Py_EndInterpreter: thread is not current");
596 if (tstate->frame != NULL)
597 Py_FatalError("Py_EndInterpreter: thread still has a frame");
598 if (tstate != interp->tstate_head || tstate->next != NULL)
599 Py_FatalError("Py_EndInterpreter: not the last thread");
600
601 PyImport_Cleanup();
602 PyInterpreterState_Clear(interp);
603 PyThreadState_Swap(NULL);
604 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000605}
606
607static char *progname = "python";
608
609void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000610Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000611{
612 if (pn && *pn)
613 progname = pn;
614}
615
616char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000617Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000618{
619 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000620}
621
Guido van Rossuma61691e1998-02-06 22:27:24 +0000622static char *default_home = NULL;
623
624void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000625Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000626{
627 default_home = home;
628}
629
630char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000631Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000632{
633 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000634 if (home == NULL && !Py_IgnoreEnvironmentFlag)
635 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000636 return home;
637}
638
Guido van Rossum6135a871995-01-09 17:53:26 +0000639/* Create __main__ module */
640
641static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000642initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000643{
Guido van Rossum82598051997-03-05 00:20:32 +0000644 PyObject *m, *d;
645 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000646 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000647 Py_FatalError("can't create __main__ module");
648 d = PyModule_GetDict(m);
649 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000650 PyObject *bimod = PyImport_ImportModule("__builtin__");
651 if (bimod == NULL ||
652 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000653 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000654 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000655 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000656}
657
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000658/* Import the site module (not into __main__ though) */
659
660static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000661initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000662{
663 PyObject *m, *f;
664 m = PyImport_ImportModule("site");
665 if (m == NULL) {
666 f = PySys_GetObject("stderr");
667 if (Py_VerboseFlag) {
668 PyFile_WriteString(
669 "'import site' failed; traceback:\n", f);
670 PyErr_Print();
671 }
672 else {
673 PyFile_WriteString(
674 "'import site' failed; use -v for traceback\n", f);
675 PyErr_Clear();
676 }
677 }
678 else {
679 Py_DECREF(m);
680 }
681}
682
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000683/* Parse input from a file and execute it */
684
685int
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000686PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000687 PyCompilerFlags *flags)
688{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000689 if (filename == NULL)
690 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000691 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000692 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000693 if (closeit)
694 fclose(fp);
695 return err;
696 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000697 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000698 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000699}
700
701int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000702PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000703{
Guido van Rossum82598051997-03-05 00:20:32 +0000704 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000705 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000706 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000707
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000708 if (flags == NULL) {
709 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000710 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000711 }
Guido van Rossum82598051997-03-05 00:20:32 +0000712 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000713 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000714 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
715 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000716 }
Guido van Rossum82598051997-03-05 00:20:32 +0000717 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000718 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000719 PySys_SetObject("ps2", v = PyString_FromString("... "));
720 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000721 }
722 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000723 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Tim Peters62e97f02006-03-28 21:44:32 +0000724 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000725 if (ret == E_EOF)
726 return 0;
727 /*
728 if (ret == E_NOMEM)
729 return -1;
730 */
731 }
732}
733
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000734/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000735#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000736 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Neal Norwitzca460d92006-09-06 06:28:06 +0000737 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
738
739#if 0
740/* Keep an example of flags with future keyword support. */
741#define PARSER_FLAGS(flags) \
742 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000743 PyPARSE_DONT_IMPLY_DEDENT : 0) \
Neal Norwitz9589ee22006-03-04 19:01:22 +0000744 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
745 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Neal Norwitzca460d92006-09-06 06:28:06 +0000746#endif
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000747
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000748int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000749PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000750{
Guido van Rossum82598051997-03-05 00:20:32 +0000751 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000752 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000753 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000754 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000755 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000756
Guido van Rossum82598051997-03-05 00:20:32 +0000757 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000758 if (v != NULL) {
759 v = PyObject_Str(v);
760 if (v == NULL)
761 PyErr_Clear();
762 else if (PyString_Check(v))
763 ps1 = PyString_AsString(v);
764 }
Guido van Rossum82598051997-03-05 00:20:32 +0000765 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000766 if (w != NULL) {
767 w = PyObject_Str(w);
768 if (w == NULL)
769 PyErr_Clear();
770 else if (PyString_Check(w))
771 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000772 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000773 arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +0000774 if (arena == NULL) {
775 Py_XDECREF(v);
776 Py_XDECREF(w);
777 return -1;
778 }
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000779 mod = PyParser_ASTFromFile(fp, filename,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000780 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000781 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000782 Py_XDECREF(v);
783 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000784 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000785 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000786 if (errcode == E_EOF) {
787 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000788 return E_EOF;
789 }
Guido van Rossum82598051997-03-05 00:20:32 +0000790 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000791 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000792 }
Guido van Rossum82598051997-03-05 00:20:32 +0000793 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000794 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000795 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000796 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000797 }
Guido van Rossum82598051997-03-05 00:20:32 +0000798 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000799 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000800 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000801 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000802 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000803 return -1;
804 }
Guido van Rossum82598051997-03-05 00:20:32 +0000805 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000806 if (Py_FlushLine())
807 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000808 return 0;
809}
810
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000811/* Check whether a file maybe a pyc file: Look at the extension,
812 the file type, and, if we may close it, at the first few bytes. */
813
814static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000815maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000816{
817 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
818 return 1;
819
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000820 /* Only look into the file if we are allowed to close it, since
821 it then should also be seekable. */
822 if (closeit) {
823 /* Read only two bytes of the magic. If the file was opened in
824 text mode, the bytes 3 and 4 of the magic (\r\n) might not
825 be read as they are on disk. */
826 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
827 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000828 /* Mess: In case of -x, the stream is NOT at its start now,
829 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000830 which makes the current stream position formally undefined,
831 and a x-platform nightmare.
832 Unfortunately, we have no direct way to know whether -x
833 was specified. So we use a terrible hack: if the current
834 stream position is not 0, we assume -x was specified, and
835 give up. Bug 132850 on SourceForge spells out the
836 hopelessness of trying anything else (fseek and ftell
837 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000838 */
Tim Peters3e876562001-02-11 04:35:39 +0000839 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000840 if (ftell(fp) == 0) {
841 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000842 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000843 ispyc = 1;
844 rewind(fp);
845 }
Tim Peters3e876562001-02-11 04:35:39 +0000846 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000847 }
848 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000849}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000850
Guido van Rossum0df002c2000-08-27 19:21:52 +0000851int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000852PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000853 PyCompilerFlags *flags)
854{
Guido van Rossum82598051997-03-05 00:20:32 +0000855 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000856 const char *ext;
Georg Brandlaa2321b2007-03-07 00:40:28 +0000857 int set_file_name = 0, ret;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000858
Guido van Rossum82598051997-03-05 00:20:32 +0000859 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000860 if (m == NULL)
861 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000862 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000863 if (PyDict_GetItemString(d, "__file__") == NULL) {
864 PyObject *f = PyString_FromString(filename);
865 if (f == NULL)
866 return -1;
867 if (PyDict_SetItemString(d, "__file__", f) < 0) {
868 Py_DECREF(f);
869 return -1;
870 }
Georg Brandlaa2321b2007-03-07 00:40:28 +0000871 set_file_name = 1;
Fred Drake8ed02042002-10-17 21:24:58 +0000872 Py_DECREF(f);
873 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000874 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000875 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000876 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000877 if (closeit)
878 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000879 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000880 fprintf(stderr, "python: Can't reopen .pyc file\n");
Georg Brandlaa2321b2007-03-07 00:40:28 +0000881 ret = -1;
882 goto done;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000883 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000884 /* Turn on optimization if a .pyo file is given */
885 if (strcmp(ext, ".pyo") == 0)
886 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000887 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000888 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000889 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000890 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000891 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000892 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000893 PyErr_Print();
Georg Brandlaa2321b2007-03-07 00:40:28 +0000894 ret = -1;
895 goto done;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000896 }
Guido van Rossum82598051997-03-05 00:20:32 +0000897 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000898 if (Py_FlushLine())
899 PyErr_Clear();
Georg Brandlaa2321b2007-03-07 00:40:28 +0000900 ret = 0;
901 done:
902 if (set_file_name && PyDict_DelItemString(d, "__file__"))
903 PyErr_Clear();
904 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000905}
906
907int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000908PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000909{
Guido van Rossum82598051997-03-05 00:20:32 +0000910 PyObject *m, *d, *v;
911 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000912 if (m == NULL)
913 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000914 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000915 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000916 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000917 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000918 return -1;
919 }
Guido van Rossum82598051997-03-05 00:20:32 +0000920 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000921 if (Py_FlushLine())
922 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000923 return 0;
924}
925
Barry Warsaw035574d1997-08-29 22:07:17 +0000926static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000927parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
928 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000929{
930 long hold;
931 PyObject *v;
932
933 /* old style errors */
934 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000935 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000936 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000937
938 /* new style errors. `err' is an instance */
939
940 if (! (v = PyObject_GetAttrString(err, "msg")))
941 goto finally;
942 *message = v;
943
944 if (!(v = PyObject_GetAttrString(err, "filename")))
945 goto finally;
946 if (v == Py_None)
947 *filename = NULL;
948 else if (! (*filename = PyString_AsString(v)))
949 goto finally;
950
951 Py_DECREF(v);
952 if (!(v = PyObject_GetAttrString(err, "lineno")))
953 goto finally;
954 hold = PyInt_AsLong(v);
955 Py_DECREF(v);
956 v = NULL;
957 if (hold < 0 && PyErr_Occurred())
958 goto finally;
959 *lineno = (int)hold;
960
961 if (!(v = PyObject_GetAttrString(err, "offset")))
962 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000963 if (v == Py_None) {
964 *offset = -1;
965 Py_DECREF(v);
966 v = NULL;
967 } else {
968 hold = PyInt_AsLong(v);
969 Py_DECREF(v);
970 v = NULL;
971 if (hold < 0 && PyErr_Occurred())
972 goto finally;
973 *offset = (int)hold;
974 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000975
976 if (!(v = PyObject_GetAttrString(err, "text")))
977 goto finally;
978 if (v == Py_None)
979 *text = NULL;
980 else if (! (*text = PyString_AsString(v)))
981 goto finally;
982 Py_DECREF(v);
983 return 1;
984
985finally:
986 Py_XDECREF(v);
987 return 0;
988}
989
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000990void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000991PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000992{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000993 PyErr_PrintEx(1);
994}
995
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000996static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000997print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000998{
999 char *nl;
1000 if (offset >= 0) {
1001 if (offset > 0 && offset == (int)strlen(text))
1002 offset--;
1003 for (;;) {
1004 nl = strchr(text, '\n');
1005 if (nl == NULL || nl-text >= offset)
1006 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001007 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001008 text = nl+1;
1009 }
1010 while (*text == ' ' || *text == '\t') {
1011 text++;
1012 offset--;
1013 }
1014 }
1015 PyFile_WriteString(" ", f);
1016 PyFile_WriteString(text, f);
1017 if (*text == '\0' || text[strlen(text)-1] != '\n')
1018 PyFile_WriteString("\n", f);
1019 if (offset == -1)
1020 return;
1021 PyFile_WriteString(" ", f);
1022 offset--;
1023 while (offset > 0) {
1024 PyFile_WriteString(" ", f);
1025 offset--;
1026 }
1027 PyFile_WriteString("^\n", f);
1028}
1029
Guido van Rossum66e8e862001-03-23 17:54:43 +00001030static void
1031handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001032{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001033 PyObject *exception, *value, *tb;
1034 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001035
Georg Brandl49aafc92007-03-07 00:34:46 +00001036 if (Py_InspectFlag)
1037 /* Don't exit if -i flag was given. This flag is set to 0
1038 * when entering interactive mode for inspecting. */
1039 return;
1040
Guido van Rossum66e8e862001-03-23 17:54:43 +00001041 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001042 if (Py_FlushLine())
1043 PyErr_Clear();
1044 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);
Neal Norwitza5e4f222006-07-17 00:59:04 +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 if (Py_FlushLine())
1122 PyErr_Clear();
1123 fflush(stdout);
1124 PySys_WriteStderr("Error in sys.excepthook:\n");
1125 PyErr_Display(exception2, v2, tb2);
1126 PySys_WriteStderr("\nOriginal exception was:\n");
1127 PyErr_Display(exception, v, tb);
Neal Norwitza5e4f222006-07-17 00:59:04 +00001128 Py_DECREF(exception2);
1129 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001130 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001131 }
1132 Py_XDECREF(result);
1133 Py_XDECREF(args);
1134 } else {
1135 PySys_WriteStderr("sys.excepthook is missing\n");
1136 PyErr_Display(exception, v, tb);
1137 }
1138 Py_XDECREF(exception);
1139 Py_XDECREF(v);
1140 Py_XDECREF(tb);
1141}
1142
Richard Jones7b9558d2006-05-27 12:29:24 +00001143void
1144PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001145{
1146 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001147 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001148 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001149 if (f == NULL)
1150 fprintf(stderr, "lost sys.stderr\n");
1151 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001152 if (Py_FlushLine())
1153 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001154 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001155 if (tb && tb != Py_None)
1156 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001157 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001158 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001159 {
Guido van Rossum82598051997-03-05 00:20:32 +00001160 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001161 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001162 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001163 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001164 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001165 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001166 else {
1167 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001168 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001169 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001170 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001171 else
Guido van Rossum82598051997-03-05 00:20:32 +00001172 PyFile_WriteString(filename, f);
1173 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001174 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001175 PyFile_WriteString(buf, f);
1176 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001177 if (text != NULL)
1178 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001179 Py_DECREF(value);
1180 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001181 /* Can't be bothered to check all those
1182 PyFile_WriteString() calls */
1183 if (PyErr_Occurred())
1184 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001185 }
1186 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001187 if (err) {
1188 /* Don't do anything else */
1189 }
Brett Cannonbf364092006-03-01 04:25:17 +00001190 else if (PyExceptionClass_Check(exception)) {
Richard Jones7b9558d2006-05-27 12:29:24 +00001191 PyObject* moduleName;
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001192 char* className = PyExceptionClass_Name(exception);
1193 if (className != NULL) {
1194 char *dot = strrchr(className, '.');
1195 if (dot != NULL)
1196 className = dot+1;
1197 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001198
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001199 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001200 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001201 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001202 else {
1203 char* modstr = PyString_AsString(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001204 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001205 {
1206 err = PyFile_WriteString(modstr, f);
1207 err += PyFile_WriteString(".", f);
1208 }
Richard Jones7b9558d2006-05-27 12:29:24 +00001209 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001210 }
1211 if (err == 0) {
1212 if (className == NULL)
1213 err = PyFile_WriteString("<unknown>", f);
1214 else
Brett Cannonbf364092006-03-01 04:25:17 +00001215 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001216 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001217 }
1218 else
1219 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001220 if (err == 0 && (value != Py_None)) {
1221 PyObject *s = PyObject_Str(value);
1222 /* only print colon if the str() of the
1223 object is not the empty string
1224 */
1225 if (s == NULL)
1226 err = -1;
1227 else if (!PyString_Check(s) ||
1228 PyString_GET_SIZE(s) != 0)
1229 err = PyFile_WriteString(": ", f);
1230 if (err == 0)
1231 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1232 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001233 }
Georg Brandl7b9c5552007-03-12 14:30:05 +00001234 /* try to write a newline in any case */
1235 err += PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001236 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001237 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001238 /* If an error happened here, don't show it.
1239 XXX This is wrong, but too many callers rely on this behavior. */
1240 if (err != 0)
1241 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001242}
1243
Guido van Rossum82598051997-03-05 00:20:32 +00001244PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001245PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001246 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001247{
Neal Norwitze92fba02006-03-04 18:52:26 +00001248 PyObject *ret = NULL;
Neal Norwitzd12bd012006-07-21 07:59:47 +00001249 mod_ty mod;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001250 PyArena *arena = PyArena_New();
Neal Norwitzd12bd012006-07-21 07:59:47 +00001251 if (arena == NULL)
1252 return NULL;
1253
1254 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001255 if (mod != NULL)
1256 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001257 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001258 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001259}
1260
1261PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001262PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001263 PyObject *locals, int closeit, PyCompilerFlags *flags)
1264{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001265 PyObject *ret;
Neal Norwitzd12bd012006-07-21 07:59:47 +00001266 mod_ty mod;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001267 PyArena *arena = PyArena_New();
Neal Norwitzd12bd012006-07-21 07:59:47 +00001268 if (arena == NULL)
1269 return NULL;
1270
1271 mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
1272 flags, NULL, arena);
Georg Brandl098cd692007-03-06 12:17:50 +00001273 if (closeit)
1274 fclose(fp);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001275 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001276 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001277 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001278 }
Neal Norwitze92fba02006-03-04 18:52:26 +00001279 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001280 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001281 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001282}
1283
Guido van Rossum82598051997-03-05 00:20:32 +00001284static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001285run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001286 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001287{
Guido van Rossum82598051997-03-05 00:20:32 +00001288 PyCodeObject *co;
1289 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001290 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001291 if (co == NULL)
1292 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001293 v = PyEval_EvalCode(co, globals, locals);
1294 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001295 return v;
1296}
1297
Guido van Rossum82598051997-03-05 00:20:32 +00001298static PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001299run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001300 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001301{
Guido van Rossum82598051997-03-05 00:20:32 +00001302 PyCodeObject *co;
1303 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001304 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001305 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001306
Guido van Rossum82598051997-03-05 00:20:32 +00001307 magic = PyMarshal_ReadLongFromFile(fp);
1308 if (magic != PyImport_GetMagicNumber()) {
1309 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001310 "Bad magic number in .pyc file");
1311 return NULL;
1312 }
Guido van Rossum82598051997-03-05 00:20:32 +00001313 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001314 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001315 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001316 if (v == NULL || !PyCode_Check(v)) {
1317 Py_XDECREF(v);
1318 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001319 "Bad code object in .pyc file");
1320 return NULL;
1321 }
Guido van Rossum82598051997-03-05 00:20:32 +00001322 co = (PyCodeObject *)v;
1323 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001324 if (v && flags)
1325 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001326 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001327 return v;
1328}
1329
Guido van Rossum82598051997-03-05 00:20:32 +00001330PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001331Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001332 PyCompilerFlags *flags)
1333{
Guido van Rossum82598051997-03-05 00:20:32 +00001334 PyCodeObject *co;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001335 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001336 PyArena *arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001337 if (arena == NULL)
1338 return NULL;
1339
1340 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001341 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001342 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001343 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001344 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001345 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001346 PyObject *result = PyAST_mod2obj(mod);
1347 PyArena_Free(arena);
1348 return result;
1349 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001350 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001351 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001352 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001353}
1354
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001355struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001356Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001357{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001358 struct symtable *st;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001359 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001360 PyArena *arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001361 if (arena == NULL)
1362 return NULL;
1363
1364 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001365 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001366 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001367 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001368 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001369 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001370 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001371 return st;
1372}
1373
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001374/* Preferred access to parser is through AST. */
1375mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001376PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001377 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001378{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001379 mod_ty mod;
1380 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001381 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001382 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001383 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001384 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001385 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001386 PyNode_Free(n);
1387 return mod;
1388 }
1389 else {
1390 err_input(&err);
1391 return NULL;
1392 }
1393}
1394
1395mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001396PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001397 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001398 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001399{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001400 mod_ty mod;
1401 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001402 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1403 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;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001423 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1424 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001425 if (n == NULL)
1426 err_input(&err);
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001427
Guido van Rossuma110aa61994-08-29 12:50:44 +00001428 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001429}
1430
Guido van Rossuma110aa61994-08-29 12:50:44 +00001431/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001432
Guido van Rossuma110aa61994-08-29 12:50:44 +00001433node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001434PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001435{
Tim Petersfe2127d2001-07-16 05:37:24 +00001436 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001437 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1438 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001439 if (n == NULL)
1440 err_input(&err);
1441 return n;
1442}
1443
1444node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001445PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001446 int start, int flags)
1447{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001448 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001449 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1450 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001451 if (n == NULL)
1452 err_input(&err);
1453 return n;
1454}
1455
1456node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001457PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001458{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001459 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001460}
1461
Guido van Rossum66ebd912003-04-17 16:02:26 +00001462/* May want to move a more generalized form of this to parsetok.c or
1463 even parser modules. */
1464
1465void
1466PyParser_SetError(perrdetail *err)
1467{
1468 err_input(err);
1469}
1470
Guido van Rossuma110aa61994-08-29 12:50:44 +00001471/* Set the error appropriate to the given input error code (see errcode.h) */
1472
1473static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001474err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001475{
Fred Drake85f36392000-07-11 17:53:00 +00001476 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001477 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001478 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001479 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001480 switch (err->error) {
1481 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001482 errtype = PyExc_IndentationError;
1483 if (err->expected == INDENT)
1484 msg = "expected an indented block";
1485 else if (err->token == INDENT)
1486 msg = "unexpected indent";
1487 else if (err->token == DEDENT)
1488 msg = "unexpected unindent";
1489 else {
1490 errtype = PyExc_SyntaxError;
1491 msg = "invalid syntax";
1492 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001493 break;
1494 case E_TOKEN:
1495 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001496 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001497 case E_EOFS:
1498 msg = "EOF while scanning triple-quoted string";
1499 break;
1500 case E_EOLS:
1501 msg = "EOL while scanning single-quoted string";
1502 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001503 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001504 if (!PyErr_Occurred())
1505 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001506 return;
1507 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001508 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001509 return;
1510 case E_EOF:
1511 msg = "unexpected EOF while parsing";
1512 break;
Fred Drake85f36392000-07-11 17:53:00 +00001513 case E_TABSPACE:
1514 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001515 msg = "inconsistent use of tabs and spaces in indentation";
1516 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001517 case E_OVERFLOW:
1518 msg = "expression too long";
1519 break;
Fred Drake85f36392000-07-11 17:53:00 +00001520 case E_DEDENT:
1521 errtype = PyExc_IndentationError;
1522 msg = "unindent does not match any outer indentation level";
1523 break;
1524 case E_TOODEEP:
1525 errtype = PyExc_IndentationError;
1526 msg = "too many levels of indentation";
1527 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001528 case E_DECODE: {
1529 PyObject *type, *value, *tb;
1530 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001531 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001532 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001533 if (u != NULL) {
1534 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001535 }
1536 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001537 if (msg == NULL)
1538 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001539 Py_XDECREF(type);
1540 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001541 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001542 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001543 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001544 case E_LINECONT:
1545 msg = "unexpected character after line continuation character";
1546 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001547 default:
1548 fprintf(stderr, "error=%d\n", err->error);
1549 msg = "unknown parsing error";
1550 break;
1551 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001552 v = Py_BuildValue("(ziiz)", err->filename,
1553 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001554 if (err->text != NULL) {
Tim Petersc9d78aa2006-03-26 23:27:58 +00001555 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001556 err->text = NULL;
1557 }
1558 w = NULL;
1559 if (v != NULL)
1560 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001561 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001562 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001563 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001564 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001565}
1566
1567/* Print fatal error message and abort */
1568
1569void
Tim Peters7c321a82002-07-09 02:57:01 +00001570Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001571{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001572 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001573#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001574 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001575 OutputDebugString(msg);
1576 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001577#ifdef _DEBUG
1578 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001579#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001580#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001581 abort();
1582}
1583
1584/* Clean up and exit */
1585
Guido van Rossuma110aa61994-08-29 12:50:44 +00001586#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001587#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001588#endif
1589
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001590#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001591static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001592static int nexitfuncs = 0;
1593
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001594int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001595{
1596 if (nexitfuncs >= NEXITFUNCS)
1597 return -1;
1598 exitfuncs[nexitfuncs++] = func;
1599 return 0;
1600}
1601
Guido van Rossumcc283f51997-08-05 02:22:03 +00001602static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001603call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001604{
Guido van Rossum82598051997-03-05 00:20:32 +00001605 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001606
1607 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001608 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001609 Py_INCREF(exitfunc);
1610 PySys_SetObject("exitfunc", (PyObject *)NULL);
1611 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001612 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001613 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1614 PySys_WriteStderr("Error in sys.exitfunc:\n");
1615 }
Guido van Rossum82598051997-03-05 00:20:32 +00001616 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001617 }
Guido van Rossum82598051997-03-05 00:20:32 +00001618 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001619 }
1620
Guido van Rossum0829c751998-02-28 04:31:39 +00001621 if (Py_FlushLine())
1622 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001623}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001624
Guido van Rossumcc283f51997-08-05 02:22:03 +00001625static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001626call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001627{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001628 while (nexitfuncs > 0)
1629 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001630
1631 fflush(stdout);
1632 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001633}
1634
1635void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001636Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001637{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001638 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001639
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001640 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001641}
1642
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001643static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001644initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001645{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001646#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001647 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001648#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001649#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001650 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001651#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001652#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001653 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001654#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001655 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001656}
1657
Guido van Rossum7433b121997-02-14 19:45:36 +00001658
1659/*
1660 * The file descriptor fd is considered ``interactive'' if either
1661 * a) isatty(fd) is TRUE, or
1662 * b) the -i flag was given, and the filename associated with
1663 * the descriptor is NULL or "<stdin>" or "???".
1664 */
1665int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001666Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001667{
1668 if (isatty((int)fileno(fp)))
1669 return 1;
1670 if (!Py_InteractiveFlag)
1671 return 0;
1672 return (filename == NULL) ||
1673 (strcmp(filename, "<stdin>") == 0) ||
1674 (strcmp(filename, "???") == 0);
1675}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001676
1677
Tim Petersd08e3822003-04-17 15:24:21 +00001678#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001679#if defined(WIN32) && defined(_MSC_VER)
1680
1681/* Stack checking for Microsoft C */
1682
1683#include <malloc.h>
1684#include <excpt.h>
1685
Fred Drakee8de31c2000-08-31 05:38:39 +00001686/*
1687 * Return non-zero when we run out of memory on the stack; zero otherwise.
1688 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001689int
Fred Drake399739f2000-08-31 05:52:44 +00001690PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001691{
1692 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001693 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001694 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001695 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001696 return 0;
1697 } __except (EXCEPTION_EXECUTE_HANDLER) {
1698 /* just ignore all errors */
1699 }
1700 return 1;
1701}
1702
1703#endif /* WIN32 && _MSC_VER */
1704
1705/* Alternate implementations can be added here... */
1706
1707#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001708
1709
1710/* Wrappers around sigaction() or signal(). */
1711
1712PyOS_sighandler_t
1713PyOS_getsig(int sig)
1714{
1715#ifdef HAVE_SIGACTION
1716 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001717 if (sigaction(sig, NULL, &context) == -1)
1718 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001719 return context.sa_handler;
1720#else
1721 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001722/* Special signal handling for the secure CRT in Visual Studio 2005 */
1723#if defined(_MSC_VER) && _MSC_VER >= 1400
1724 switch (sig) {
1725 /* Only these signals are valid */
1726 case SIGINT:
1727 case SIGILL:
1728 case SIGFPE:
1729 case SIGSEGV:
1730 case SIGTERM:
1731 case SIGBREAK:
1732 case SIGABRT:
1733 break;
1734 /* Don't call signal() with other values or it will assert */
1735 default:
1736 return SIG_ERR;
1737 }
1738#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001739 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001740 if (handler != SIG_ERR)
1741 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001742 return handler;
1743#endif
1744}
1745
1746PyOS_sighandler_t
1747PyOS_setsig(int sig, PyOS_sighandler_t handler)
1748{
1749#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001750 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001751 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001752 sigemptyset(&context.sa_mask);
1753 context.sa_flags = 0;
1754 if (sigaction(sig, &context, &ocontext) == -1)
1755 return SIG_ERR;
1756 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001757#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001758 PyOS_sighandler_t oldhandler;
1759 oldhandler = signal(sig, handler);
1760#ifdef HAVE_SIGINTERRUPT
1761 siginterrupt(sig, 1);
1762#endif
1763 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001764#endif
1765}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001766
1767/* Deprecated C API functions still provided for binary compatiblity */
1768
1769#undef PyParser_SimpleParseFile
Thomas Heller1b046642006-04-18 18:51:06 +00001770PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001771PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1772{
1773 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1774}
1775
Thomas Heller1b046642006-04-18 18:51:06 +00001776#undef PyParser_SimpleParseString
1777PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001778PyParser_SimpleParseString(const char *str, int start)
1779{
1780 return PyParser_SimpleParseStringFlags(str, start, 0);
1781}
Anthony Baxterac6bd462006-04-13 02:06:09 +00001782
Thomas Heller1b046642006-04-18 18:51:06 +00001783#undef PyRun_AnyFile
1784PyAPI_FUNC(int)
1785PyRun_AnyFile(FILE *fp, const char *name)
1786{
1787 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1788}
1789
1790#undef PyRun_AnyFileEx
1791PyAPI_FUNC(int)
1792PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1793{
1794 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1795}
1796
1797#undef PyRun_AnyFileFlags
1798PyAPI_FUNC(int)
1799PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1800{
1801 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1802}
1803
1804#undef PyRun_File
1805PyAPI_FUNC(PyObject *)
1806PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1807{
1808 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1809}
1810
1811#undef PyRun_FileEx
1812PyAPI_FUNC(PyObject *)
1813PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1814{
1815 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1816}
1817
1818#undef PyRun_FileFlags
1819PyAPI_FUNC(PyObject *)
1820PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1821 PyCompilerFlags *flags)
1822{
1823 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1824}
1825
1826#undef PyRun_SimpleFile
1827PyAPI_FUNC(int)
1828PyRun_SimpleFile(FILE *f, const char *p)
1829{
1830 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1831}
1832
1833#undef PyRun_SimpleFileEx
1834PyAPI_FUNC(int)
1835PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1836{
1837 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1838}
1839
1840
1841#undef PyRun_String
1842PyAPI_FUNC(PyObject *)
1843PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1844{
1845 return PyRun_StringFlags(str, s, g, l, NULL);
1846}
1847
1848#undef PyRun_SimpleString
1849PyAPI_FUNC(int)
1850PyRun_SimpleString(const char *s)
1851{
1852 return PyRun_SimpleStringFlags(s, NULL);
1853}
1854
1855#undef Py_CompileString
1856PyAPI_FUNC(PyObject *)
1857Py_CompileString(const char *str, const char *p, int s)
1858{
1859 return Py_CompileStringFlags(str, p, s, NULL);
1860}
1861
1862#undef PyRun_InteractiveOne
1863PyAPI_FUNC(int)
1864PyRun_InteractiveOne(FILE *f, const char *p)
1865{
1866 return PyRun_InteractiveOneFlags(f, p, NULL);
1867}
1868
1869#undef PyRun_InteractiveLoop
1870PyAPI_FUNC(int)
1871PyRun_InteractiveLoop(FILE *f, const char *p)
1872{
1873 return PyRun_InteractiveLoopFlags(f, p, NULL);
1874}
1875
Anthony Baxterac6bd462006-04-13 02:06:09 +00001876#ifdef __cplusplus
1877}
1878#endif
1879