blob: 7501456ca8713deb121781a2a2e2546fbf64138b [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 Rossum1984f1e1992-08-04 12:41:02 +00007#include "grammar.h"
8#include "node.h"
Fred Drake85f36392000-07-11 17:53:00 +00009#include "token.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000010#include "parsetok.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000011#include "errcode.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000012#include "code.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000013#include "compile.h"
Jeremy Hylton4b38da62001-02-02 18:19:15 +000014#include "symtable.h"
Neal Norwitzadb69fc2005-12-17 20:54:49 +000015#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000016#include "ast.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000017#include "eval.h"
Guido van Rossumfdef2711994-09-14 13:31:04 +000018#include "marshal.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000019
Thomas Wouters0e3f5912006-08-11 14:57:12 +000020#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000021#include <signal.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000022#endif
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000023
Martin v. Löwis73d538b2003-03-05 15:13:47 +000024#ifdef HAVE_LANGINFO_H
25#include <locale.h>
26#include <langinfo.h>
27#endif
28
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000029#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000030#undef BYTE
31#include "windows.h"
32#endif
33
Neal Norwitz4281cef2006-03-04 19:58:13 +000034#ifndef Py_REF_DEBUG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000035#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000036#else /* Py_REF_DEBUG */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000037#define PRINT_TOTAL_REFS() fprintf(stderr, \
38 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
39 _Py_GetRefTotal())
40#endif
41
42#ifdef __cplusplus
43extern "C" {
Neal Norwitz4281cef2006-03-04 19:58:13 +000044#endif
45
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000046extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000047
Guido van Rossum82598051997-03-05 00:20:32 +000048extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000049
Guido van Rossumb73cc041993-11-01 16:28:59 +000050/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000051static void initmain(void);
52static void initsite(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000053static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000054 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000055static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000056 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000057static void err_input(perrdetail *);
58static void initsigs(void);
59static void call_sys_exitfunc(void);
60static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000061extern void _PyUnicode_Init(void);
62extern void _PyUnicode_Fini(void);
Guido van Rossumddefaf32007-01-14 03:31:43 +000063extern int _PyLong_Init(void);
64extern void PyLong_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000065
Mark Hammond8d98d2c2003-04-19 15:41:53 +000066#ifdef WITH_THREAD
67extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
68extern void _PyGILState_Fini(void);
69#endif /* WITH_THREAD */
70
Guido van Rossum82598051997-03-05 00:20:32 +000071int Py_DebugFlag; /* Needed by parser.c */
72int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000073int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
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 */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000079
Mark Hammonda43fd0c2003-02-19 00:33:33 +000080/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000081 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000082*/
Mark Hammondedd07732003-07-15 23:03:55 +000083static PyObject *warnings_module = NULL;
84
85/* Returns a borrowed reference to the 'warnings' module, or NULL.
86 If the module is returned, it is guaranteed to have been obtained
87 without acquiring the import lock
88*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000089PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000090{
91 PyObject *typ, *val, *tb;
92 PyObject *all_modules;
93 /* If we managed to get the module at init time, just use it */
94 if (warnings_module)
95 return warnings_module;
96 /* If it wasn't available at init time, it may be available
97 now in sys.modules (common scenario is frozen apps: import
98 at init time fails, but the frozen init code sets up sys.path
99 correctly, then does an implicit import of warnings for us
100 */
101 /* Save and restore any exceptions */
102 PyErr_Fetch(&typ, &val, &tb);
103
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000104 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000105 if (all_modules) {
106 warnings_module = PyDict_GetItemString(all_modules, "warnings");
107 /* We keep a ref in the global */
108 Py_XINCREF(warnings_module);
109 }
110 PyErr_Restore(typ, val, tb);
111 return warnings_module;
112}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000113
Guido van Rossum25ce5661997-08-02 03:10:38 +0000114static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000115
Thomas Wouters7e474022000-07-16 12:04:32 +0000116/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000117
118int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000119Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000120{
121 return initialized;
122}
123
Guido van Rossum25ce5661997-08-02 03:10:38 +0000124/* Global initializations. Can be undone by Py_Finalize(). Don't
125 call this twice without an intervening Py_Finalize() call. When
126 initializations fail, a fatal error is issued and the function does
127 not return. On return, the first thread and interpreter state have
128 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000129
Guido van Rossum25ce5661997-08-02 03:10:38 +0000130 Locking: you must hold the interpreter lock while calling this.
131 (If the lock has not yet been initialized, that's equivalent to
132 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000133
Guido van Rossum25ce5661997-08-02 03:10:38 +0000134*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000135
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000136static int
137add_flag(int flag, const char *envs)
138{
139 int env = atoi(envs);
140 if (flag < env)
141 flag = env;
142 if (flag < 1)
143 flag = 1;
144 return flag;
145}
146
Guido van Rossuma027efa1997-05-05 20:56:21 +0000147void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000148Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000149{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000150 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000151 PyThreadState *tstate;
152 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000153 char *p;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000154#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
155 char *codeset;
156 char *saved_locale;
157 PyObject *sys_stream, *sys_isatty;
158#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
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000189 _PyFloat_Init();
190
Guido van Rossum25ce5661997-08-02 03:10:38 +0000191 interp->modules = PyDict_New();
192 if (interp->modules == NULL)
193 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000194
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000195#ifdef Py_USING_UNICODE
Guido van Rossumc94044c2000-03-10 23:03:54 +0000196 /* Init Unicode implementation; relies on the codec registry */
197 _PyUnicode_Init();
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000198#endif
Guido van Rossumc94044c2000-03-10 23:03:54 +0000199
Barry Warsawf242aa02000-05-25 23:09:49 +0000200 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000201 if (bimod == NULL)
202 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000203 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000204 if (interp->builtins == NULL)
205 Py_FatalError("Py_Initialize: can't initialize builtins dict");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000206 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000207
208 sysmod = _PySys_Init();
209 if (sysmod == NULL)
210 Py_FatalError("Py_Initialize: can't initialize sys");
211 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000212 if (interp->sysdict == NULL)
213 Py_FatalError("Py_Initialize: can't initialize sys dict");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000214 Py_INCREF(interp->sysdict);
215 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000216 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000217 PyDict_SetItemString(interp->sysdict, "modules",
218 interp->modules);
219
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000220 _PyImport_Init();
221
Barry Warsawf242aa02000-05-25 23:09:49 +0000222 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000223 _PyExc_Init();
Barry Warsawf242aa02000-05-25 23:09:49 +0000224
Barry Warsaw035574d1997-08-29 22:07:17 +0000225 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000226 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000227
Just van Rossum52e14d62002-12-30 22:08:05 +0000228 _PyImportHooks_Init();
229
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000230 if (install_sigs)
231 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000232
233 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000234 if (!Py_NoSiteFlag)
235 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000236
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000237 /* auto-thread-state API, if available */
238#ifdef WITH_THREAD
239 _PyGILState_Init(interp, tstate);
240#endif /* WITH_THREAD */
241
Mark Hammondedd07732003-07-15 23:03:55 +0000242 warnings_module = PyImport_ImportModule("warnings");
243 if (!warnings_module)
244 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000245
246#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
247 /* On Unix, set the file system encoding according to the
248 user's preference, if the CODESET names a well-known
249 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000250 initialized by other means. Also set the encoding of
251 stdin and stdout if these are terminals. */
252
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000253 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000254 setlocale(LC_CTYPE, "");
255 codeset = nl_langinfo(CODESET);
256 if (codeset && *codeset) {
257 PyObject *enc = PyCodec_Encoder(codeset);
258 if (enc) {
259 codeset = strdup(codeset);
260 Py_DECREF(enc);
261 } else {
262 codeset = NULL;
263 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000264 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000265 } else
266 codeset = NULL;
267 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000268 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000269
270 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000271 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000272 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
273 if (!sys_isatty)
274 PyErr_Clear();
Thomas Woutersb2137042007-02-01 18:02:27 +0000275 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
276 PyFile_Check(sys_stream)) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000277 if (!PyFile_SetEncoding(sys_stream, codeset))
278 Py_FatalError("Cannot set codeset of stdin");
279 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000280 Py_XDECREF(sys_isatty);
281
282 sys_stream = PySys_GetObject("stdout");
283 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
284 if (!sys_isatty)
285 PyErr_Clear();
Thomas Woutersb2137042007-02-01 18:02:27 +0000286 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
287 PyFile_Check(sys_stream)) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000288 if (!PyFile_SetEncoding(sys_stream, codeset))
289 Py_FatalError("Cannot set codeset of stdout");
290 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000291 Py_XDECREF(sys_isatty);
292
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000293 sys_stream = PySys_GetObject("stderr");
294 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
295 if (!sys_isatty)
296 PyErr_Clear();
Thomas Woutersb2137042007-02-01 18:02:27 +0000297 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
298 PyFile_Check(sys_stream)) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000299 if (!PyFile_SetEncoding(sys_stream, codeset))
300 Py_FatalError("Cannot set codeset of stderr");
301 }
302 Py_XDECREF(sys_isatty);
303
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000304 if (!Py_FileSystemDefaultEncoding)
305 Py_FileSystemDefaultEncoding = codeset;
306 else
307 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000308 }
309#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000310}
311
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000312void
313Py_Initialize(void)
314{
315 Py_InitializeEx(1);
316}
317
318
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000319#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000320extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000321#endif
322
Guido van Rossum25ce5661997-08-02 03:10:38 +0000323/* Undo the effect of Py_Initialize().
324
325 Beware: if multiple interpreter and/or thread states exist, these
326 are not wiped out; only the current thread and interpreter state
327 are deleted. But since everything else is deleted, those other
328 interpreter and thread states should no longer be used.
329
330 (XXX We should do better, e.g. wipe out all interpreters and
331 threads.)
332
333 Locking: as above.
334
335*/
336
337void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000338Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000339{
340 PyInterpreterState *interp;
341 PyThreadState *tstate;
342
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000343 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000344 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000345
Tim Peters384fd102001-01-21 03:40:37 +0000346 /* The interpreter is still entirely intact at this point, and the
347 * exit funcs may be relying on that. In particular, if some thread
348 * or exit func is still waiting to do an import, the import machinery
349 * expects Py_IsInitialized() to return true. So don't say the
350 * interpreter is uninitialized until after the exit funcs have run.
351 * Note that Threading.py uses an exit func to do a join on all the
352 * threads created thru it, so this also protects pending imports in
353 * the threads created via Threading.
354 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000355 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000356 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000357
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 Rossume13ddc92003-04-17 17:29:22 +0000393 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000394 * new-style class definitions, for example.
395 * XXX This is disabled because it caused too many problems. If
396 * XXX a __del__ or weakref callback triggers here, Python code has
397 * XXX a hard time running, because even the sys module has been
398 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
399 * XXX One symptom is a sequence of information-free messages
400 * XXX coming from threads (if a __del__ or callback is invoked,
401 * XXX other threads can execute too, and any exception they encounter
402 * XXX triggers a comedy of errors as subsystem after subsystem
403 * XXX fails to find what it *expects* to find in sys to help report
404 * XXX the exception and consequent unexpected failures). I've also
405 * XXX seen segfaults then, after adding print statements to the
406 * XXX Python code getting called.
407 */
408#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000409 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000410#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000411
Guido van Rossum1707aad1997-12-08 23:43:45 +0000412 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
413 _PyImport_Fini();
414
415 /* Debugging stuff */
416#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000417 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000418#endif
419
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000420 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000421
Tim Peters9cf25ce2003-04-17 15:21:01 +0000422#ifdef Py_TRACE_REFS
423 /* Display all objects still alive -- this can invoke arbitrary
424 * __repr__ overrides, so requires a mostly-intact interpreter.
425 * Alas, a lot of stuff may still be alive now that will be cleaned
426 * up later.
427 */
Tim Peters269b2a62003-04-17 19:52:29 +0000428 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000429 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000430#endif /* Py_TRACE_REFS */
431
Mark Hammond6cb90292003-04-22 11:18:00 +0000432 /* Cleanup auto-thread-state */
433#ifdef WITH_THREAD
434 _PyGILState_Fini();
435#endif /* WITH_THREAD */
436
Guido van Rossumd922fa42003-04-15 14:10:09 +0000437 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000438 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000439
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000440 /* Now we decref the exception classes. After this point nothing
441 can raise an exception. That's okay, because each Fini() method
442 below has been checked to make sure no exceptions are ever
443 raised.
444 */
445
446 _PyExc_Fini();
447
Guido van Rossumd922fa42003-04-15 14:10:09 +0000448 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000449 PyThreadState_Swap(NULL);
450 PyInterpreterState_Delete(interp);
451
Guido van Rossumd922fa42003-04-15 14:10:09 +0000452 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000453 PyMethod_Fini();
454 PyFrame_Fini();
455 PyCFunction_Fini();
456 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000457 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000458 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000459 PyString_Fini();
Guido van Rossumddefaf32007-01-14 03:31:43 +0000460 PyLong_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000461 PyFloat_Fini();
462
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000463#ifdef Py_USING_UNICODE
464 /* Cleanup Unicode implementation */
465 _PyUnicode_Fini();
466#endif
467
Guido van Rossumcc283f51997-08-05 02:22:03 +0000468 /* XXX Still allocated:
469 - various static ad-hoc pointers to interned strings
470 - int and float free list blocks
471 - whatever various modules and libraries allocate
472 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000473
474 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000475
Tim Peters269b2a62003-04-17 19:52:29 +0000476#ifdef Py_TRACE_REFS
477 /* Display addresses (& refcnts) of all objects still alive.
478 * An address can be used to find the repr of the object, printed
479 * above by _Py_PrintReferences.
480 */
481 if (Py_GETENV("PYTHONDUMPREFS"))
482 _Py_PrintReferenceAddresses(stderr);
483#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000484#ifdef PYMALLOC_DEBUG
485 if (Py_GETENV("PYTHONMALLOCSTATS"))
486 _PyObject_DebugMallocStats();
487#endif
488
Guido van Rossumcc283f51997-08-05 02:22:03 +0000489 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000490}
491
492/* Create and initialize a new interpreter and thread, and return the
493 new thread. This requires that Py_Initialize() has been called
494 first.
495
496 Unsuccessful initialization yields a NULL pointer. Note that *no*
497 exception information is available even in this case -- the
498 exception information is held in the thread, and there is no
499 thread.
500
501 Locking: as above.
502
503*/
504
505PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000506Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000507{
508 PyInterpreterState *interp;
509 PyThreadState *tstate, *save_tstate;
510 PyObject *bimod, *sysmod;
511
512 if (!initialized)
513 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
514
515 interp = PyInterpreterState_New();
516 if (interp == NULL)
517 return NULL;
518
519 tstate = PyThreadState_New(interp);
520 if (tstate == NULL) {
521 PyInterpreterState_Delete(interp);
522 return NULL;
523 }
524
525 save_tstate = PyThreadState_Swap(tstate);
526
527 /* XXX The following is lax in error checking */
528
529 interp->modules = PyDict_New();
530
531 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
532 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000533 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000534 if (interp->builtins == NULL)
535 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000536 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000537 }
538 sysmod = _PyImport_FindExtension("sys", "sys");
539 if (bimod != NULL && sysmod != NULL) {
540 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000541 if (interp->sysdict == NULL)
542 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000543 Py_INCREF(interp->sysdict);
544 PySys_SetPath(Py_GetPath());
545 PyDict_SetItemString(interp->sysdict, "modules",
546 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000547 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000548 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000549 if (!Py_NoSiteFlag)
550 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000551 }
552
553 if (!PyErr_Occurred())
554 return tstate;
555
Thomas Wouters89f507f2006-12-13 04:49:30 +0000556handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000557 /* Oops, it didn't work. Undo it all. */
558
559 PyErr_Print();
560 PyThreadState_Clear(tstate);
561 PyThreadState_Swap(save_tstate);
562 PyThreadState_Delete(tstate);
563 PyInterpreterState_Delete(interp);
564
565 return NULL;
566}
567
568/* Delete an interpreter and its last thread. This requires that the
569 given thread state is current, that the thread has no remaining
570 frames, and that it is its interpreter's only remaining thread.
571 It is a fatal error to violate these constraints.
572
573 (Py_Finalize() doesn't have these constraints -- it zaps
574 everything, regardless.)
575
576 Locking: as above.
577
578*/
579
580void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000581Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000582{
583 PyInterpreterState *interp = tstate->interp;
584
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000585 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000586 Py_FatalError("Py_EndInterpreter: thread is not current");
587 if (tstate->frame != NULL)
588 Py_FatalError("Py_EndInterpreter: thread still has a frame");
589 if (tstate != interp->tstate_head || tstate->next != NULL)
590 Py_FatalError("Py_EndInterpreter: not the last thread");
591
592 PyImport_Cleanup();
593 PyInterpreterState_Clear(interp);
594 PyThreadState_Swap(NULL);
595 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000596}
597
598static char *progname = "python";
599
600void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000601Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000602{
603 if (pn && *pn)
604 progname = pn;
605}
606
607char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000608Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000609{
610 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000611}
612
Guido van Rossuma61691e1998-02-06 22:27:24 +0000613static char *default_home = NULL;
614
615void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000616Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000617{
618 default_home = home;
619}
620
621char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000622Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000623{
624 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000625 if (home == NULL && !Py_IgnoreEnvironmentFlag)
626 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000627 return home;
628}
629
Guido van Rossum6135a871995-01-09 17:53:26 +0000630/* Create __main__ module */
631
632static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000633initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000634{
Guido van Rossum82598051997-03-05 00:20:32 +0000635 PyObject *m, *d;
636 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000637 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000638 Py_FatalError("can't create __main__ module");
639 d = PyModule_GetDict(m);
640 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000641 PyObject *bimod = PyImport_ImportModule("__builtin__");
642 if (bimod == NULL ||
643 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000644 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000645 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000646 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000647}
648
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000649/* Import the site module (not into __main__ though) */
650
651static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000652initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000653{
654 PyObject *m, *f;
655 m = PyImport_ImportModule("site");
656 if (m == NULL) {
657 f = PySys_GetObject("stderr");
658 if (Py_VerboseFlag) {
659 PyFile_WriteString(
660 "'import site' failed; traceback:\n", f);
661 PyErr_Print();
662 }
663 else {
664 PyFile_WriteString(
665 "'import site' failed; use -v for traceback\n", f);
666 PyErr_Clear();
667 }
668 }
669 else {
670 Py_DECREF(m);
671 }
672}
673
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000674/* Parse input from a file and execute it */
675
676int
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000677PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000678 PyCompilerFlags *flags)
679{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000680 if (filename == NULL)
681 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000682 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000683 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000684 if (closeit)
685 fclose(fp);
686 return err;
687 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000688 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000689 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000690}
691
692int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000693PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000694{
Guido van Rossum82598051997-03-05 00:20:32 +0000695 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000696 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000697 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000698
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000699 if (flags == NULL) {
700 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000701 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000702 }
Guido van Rossum82598051997-03-05 00:20:32 +0000703 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000704 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000705 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
706 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000707 }
Guido van Rossum82598051997-03-05 00:20:32 +0000708 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000709 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000710 PySys_SetObject("ps2", v = PyString_FromString("... "));
711 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000712 }
713 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000714 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000715 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000716 if (ret == E_EOF)
717 return 0;
718 /*
719 if (ret == E_NOMEM)
720 return -1;
721 */
722 }
723}
724
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000725/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000726#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000727 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Guido van Rossum45aecf42006-03-15 04:58:47 +0000728 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000729
Thomas Wouters89f507f2006-12-13 04:49:30 +0000730#if 0
731/* Keep an example of flags with future keyword support. */
732#define PARSER_FLAGS(flags) \
733 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
734 PyPARSE_DONT_IMPLY_DEDENT : 0) \
735 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
736 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
737#endif
738
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000739int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000740PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000741{
Guido van Rossum82598051997-03-05 00:20:32 +0000742 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000743 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000744 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000745 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000746 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000747
Guido van Rossum82598051997-03-05 00:20:32 +0000748 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000749 if (v != NULL) {
750 v = PyObject_Str(v);
751 if (v == NULL)
752 PyErr_Clear();
753 else if (PyString_Check(v))
754 ps1 = PyString_AsString(v);
755 }
Guido van Rossum82598051997-03-05 00:20:32 +0000756 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000757 if (w != NULL) {
758 w = PyObject_Str(w);
759 if (w == NULL)
760 PyErr_Clear();
761 else if (PyString_Check(w))
762 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000763 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000764 arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000765 if (arena == NULL) {
766 Py_XDECREF(v);
767 Py_XDECREF(w);
768 return -1;
769 }
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000770 mod = PyParser_ASTFromFile(fp, filename,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000771 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000772 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000773 Py_XDECREF(v);
774 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000775 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000776 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000777 if (errcode == E_EOF) {
778 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000779 return E_EOF;
780 }
Guido van Rossum82598051997-03-05 00:20:32 +0000781 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000782 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000783 }
Guido van Rossum82598051997-03-05 00:20:32 +0000784 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000785 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000786 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000787 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000788 }
Guido van Rossum82598051997-03-05 00:20:32 +0000789 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000790 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000791 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000792 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000793 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000794 return -1;
795 }
Guido van Rossum82598051997-03-05 00:20:32 +0000796 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000797 return 0;
798}
799
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000800/* Check whether a file maybe a pyc file: Look at the extension,
801 the file type, and, if we may close it, at the first few bytes. */
802
803static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000804maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000805{
806 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
807 return 1;
808
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000809 /* Only look into the file if we are allowed to close it, since
810 it then should also be seekable. */
811 if (closeit) {
812 /* Read only two bytes of the magic. If the file was opened in
813 text mode, the bytes 3 and 4 of the magic (\r\n) might not
814 be read as they are on disk. */
815 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
816 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000817 /* Mess: In case of -x, the stream is NOT at its start now,
818 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000819 which makes the current stream position formally undefined,
820 and a x-platform nightmare.
821 Unfortunately, we have no direct way to know whether -x
822 was specified. So we use a terrible hack: if the current
823 stream position is not 0, we assume -x was specified, and
824 give up. Bug 132850 on SourceForge spells out the
825 hopelessness of trying anything else (fseek and ftell
826 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000827 */
Tim Peters3e876562001-02-11 04:35:39 +0000828 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000829 if (ftell(fp) == 0) {
830 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000831 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000832 ispyc = 1;
833 rewind(fp);
834 }
Tim Peters3e876562001-02-11 04:35:39 +0000835 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000836 }
837 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000838}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000839
Guido van Rossum0df002c2000-08-27 19:21:52 +0000840int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000841PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000842 PyCompilerFlags *flags)
843{
Guido van Rossum82598051997-03-05 00:20:32 +0000844 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000845 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000846
Guido van Rossum82598051997-03-05 00:20:32 +0000847 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000848 if (m == NULL)
849 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000850 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000851 if (PyDict_GetItemString(d, "__file__") == NULL) {
852 PyObject *f = PyString_FromString(filename);
853 if (f == NULL)
854 return -1;
855 if (PyDict_SetItemString(d, "__file__", f) < 0) {
856 Py_DECREF(f);
857 return -1;
858 }
859 Py_DECREF(f);
860 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000861 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000862 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000863 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000864 if (closeit)
865 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000866 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000867 fprintf(stderr, "python: Can't reopen .pyc file\n");
868 return -1;
869 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000870 /* Turn on optimization if a .pyo file is given */
871 if (strcmp(ext, ".pyo") == 0)
872 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000873 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000874 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000875 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000876 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000877 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000878 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000879 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000880 return -1;
881 }
Guido van Rossum82598051997-03-05 00:20:32 +0000882 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000883 return 0;
884}
885
886int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000887PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000888{
Guido van Rossum82598051997-03-05 00:20:32 +0000889 PyObject *m, *d, *v;
890 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000891 if (m == NULL)
892 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000893 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000894 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000895 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000896 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000897 return -1;
898 }
Guido van Rossum82598051997-03-05 00:20:32 +0000899 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000900 return 0;
901}
902
Barry Warsaw035574d1997-08-29 22:07:17 +0000903static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000904parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
905 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000906{
907 long hold;
908 PyObject *v;
909
910 /* old style errors */
911 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000912 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000913 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000914
915 /* new style errors. `err' is an instance */
916
917 if (! (v = PyObject_GetAttrString(err, "msg")))
918 goto finally;
919 *message = v;
920
921 if (!(v = PyObject_GetAttrString(err, "filename")))
922 goto finally;
923 if (v == Py_None)
924 *filename = NULL;
925 else if (! (*filename = PyString_AsString(v)))
926 goto finally;
927
928 Py_DECREF(v);
929 if (!(v = PyObject_GetAttrString(err, "lineno")))
930 goto finally;
931 hold = PyInt_AsLong(v);
932 Py_DECREF(v);
933 v = NULL;
934 if (hold < 0 && PyErr_Occurred())
935 goto finally;
936 *lineno = (int)hold;
937
938 if (!(v = PyObject_GetAttrString(err, "offset")))
939 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000940 if (v == Py_None) {
941 *offset = -1;
942 Py_DECREF(v);
943 v = NULL;
944 } else {
945 hold = PyInt_AsLong(v);
946 Py_DECREF(v);
947 v = NULL;
948 if (hold < 0 && PyErr_Occurred())
949 goto finally;
950 *offset = (int)hold;
951 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000952
953 if (!(v = PyObject_GetAttrString(err, "text")))
954 goto finally;
955 if (v == Py_None)
956 *text = NULL;
957 else if (! (*text = PyString_AsString(v)))
958 goto finally;
959 Py_DECREF(v);
960 return 1;
961
962finally:
963 Py_XDECREF(v);
964 return 0;
965}
966
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000967void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000968PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000969{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000970 PyErr_PrintEx(1);
971}
972
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000973static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000974print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000975{
976 char *nl;
977 if (offset >= 0) {
978 if (offset > 0 && offset == (int)strlen(text))
979 offset--;
980 for (;;) {
981 nl = strchr(text, '\n');
982 if (nl == NULL || nl-text >= offset)
983 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000984 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000985 text = nl+1;
986 }
987 while (*text == ' ' || *text == '\t') {
988 text++;
989 offset--;
990 }
991 }
992 PyFile_WriteString(" ", f);
993 PyFile_WriteString(text, f);
994 if (*text == '\0' || text[strlen(text)-1] != '\n')
995 PyFile_WriteString("\n", f);
996 if (offset == -1)
997 return;
998 PyFile_WriteString(" ", f);
999 offset--;
1000 while (offset > 0) {
1001 PyFile_WriteString(" ", f);
1002 offset--;
1003 }
1004 PyFile_WriteString("^\n", f);
1005}
1006
Guido van Rossum66e8e862001-03-23 17:54:43 +00001007static void
1008handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001009{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001010 PyObject *exception, *value, *tb;
1011 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001012
Guido van Rossum66e8e862001-03-23 17:54:43 +00001013 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001014 fflush(stdout);
1015 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001016 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001017 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001018 /* The error code should be in the `code' attribute. */
1019 PyObject *code = PyObject_GetAttrString(value, "code");
1020 if (code) {
1021 Py_DECREF(value);
1022 value = code;
1023 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001024 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001025 }
1026 /* If we failed to dig out the 'code' attribute,
1027 just let the else clause below print the error. */
1028 }
1029 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001030 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001031 else {
1032 PyObject_Print(value, stderr, Py_PRINT_RAW);
1033 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001034 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001035 }
Tim Peterscf615b52003-04-19 18:47:02 +00001036 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001037 /* Restore and clear the exception info, in order to properly decref
1038 * the exception, value, and traceback. If we just exit instead,
1039 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1040 * some finalizers from running.
1041 */
Tim Peterscf615b52003-04-19 18:47:02 +00001042 PyErr_Restore(exception, value, tb);
1043 PyErr_Clear();
1044 Py_Exit(exitcode);
1045 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001046}
1047
1048void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001049PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001050{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001051 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001052
1053 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1054 handle_system_exit();
1055 }
Guido van Rossum82598051997-03-05 00:20:32 +00001056 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001057 if (exception == NULL)
1058 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001059 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001060 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001061 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001062 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001063 if (set_sys_last_vars) {
1064 PySys_SetObject("last_type", exception);
1065 PySys_SetObject("last_value", v);
1066 PySys_SetObject("last_traceback", tb);
1067 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001068 hook = PySys_GetObject("excepthook");
1069 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001070 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001071 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001072 PyObject *result = PyEval_CallObject(hook, args);
1073 if (result == NULL) {
1074 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001075 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1076 handle_system_exit();
1077 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001078 PyErr_Fetch(&exception2, &v2, &tb2);
1079 PyErr_NormalizeException(&exception2, &v2, &tb2);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001080 /* It should not be possible for exception2 or v2
1081 to be NULL. However PyErr_Display() can't
1082 tolerate NULLs, so just be safe. */
1083 if (exception2 == NULL) {
1084 exception2 = Py_None;
1085 Py_INCREF(exception2);
1086 }
1087 if (v2 == NULL) {
1088 v2 = Py_None;
1089 Py_INCREF(v2);
1090 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001091 fflush(stdout);
1092 PySys_WriteStderr("Error in sys.excepthook:\n");
1093 PyErr_Display(exception2, v2, tb2);
1094 PySys_WriteStderr("\nOriginal exception was:\n");
1095 PyErr_Display(exception, v, tb);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001096 Py_DECREF(exception2);
1097 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001098 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001099 }
1100 Py_XDECREF(result);
1101 Py_XDECREF(args);
1102 } else {
1103 PySys_WriteStderr("sys.excepthook is missing\n");
1104 PyErr_Display(exception, v, tb);
1105 }
1106 Py_XDECREF(exception);
1107 Py_XDECREF(v);
1108 Py_XDECREF(tb);
1109}
1110
Thomas Wouters477c8d52006-05-27 19:21:47 +00001111void
1112PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001113{
1114 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001115 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001116 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001117 if (f == NULL)
1118 fprintf(stderr, "lost sys.stderr\n");
1119 else {
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001120 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001121 if (tb && tb != Py_None)
1122 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001123 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001124 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001125 {
Guido van Rossum82598051997-03-05 00:20:32 +00001126 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001127 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001128 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001129 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001130 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001131 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001132 else {
1133 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001134 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001135 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001136 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001137 else
Guido van Rossum82598051997-03-05 00:20:32 +00001138 PyFile_WriteString(filename, f);
1139 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001140 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001141 PyFile_WriteString(buf, f);
1142 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001143 if (text != NULL)
1144 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001145 Py_DECREF(value);
1146 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001147 /* Can't be bothered to check all those
1148 PyFile_WriteString() calls */
1149 if (PyErr_Occurred())
1150 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001151 }
1152 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001153 if (err) {
1154 /* Don't do anything else */
1155 }
Brett Cannonbf364092006-03-01 04:25:17 +00001156 else if (PyExceptionClass_Check(exception)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001157 PyObject* moduleName;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001158 char* className = PyExceptionClass_Name(exception);
1159 if (className != NULL) {
1160 char *dot = strrchr(className, '.');
1161 if (dot != NULL)
1162 className = dot+1;
1163 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001164
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001165 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001166 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001167 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001168 else {
1169 char* modstr = PyString_AsString(moduleName);
Neal Norwitz2633c692007-02-26 22:22:47 +00001170 if (modstr && strcmp(modstr, "__builtin__"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001171 {
1172 err = PyFile_WriteString(modstr, f);
1173 err += PyFile_WriteString(".", f);
1174 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001175 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001176 }
1177 if (err == 0) {
1178 if (className == NULL)
1179 err = PyFile_WriteString("<unknown>", f);
1180 else
Brett Cannonbf364092006-03-01 04:25:17 +00001181 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001182 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001183 }
1184 else
1185 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001186 if (err == 0 && (value != Py_None)) {
1187 PyObject *s = PyObject_Str(value);
1188 /* only print colon if the str() of the
1189 object is not the empty string
1190 */
1191 if (s == NULL)
1192 err = -1;
1193 else if (!PyString_Check(s) ||
1194 PyString_GET_SIZE(s) != 0)
1195 err = PyFile_WriteString(": ", f);
1196 if (err == 0)
1197 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1198 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001199 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001200 if (err == 0)
1201 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001202 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001203 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001204 /* If an error happened here, don't show it.
1205 XXX This is wrong, but too many callers rely on this behavior. */
1206 if (err != 0)
1207 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001208}
1209
Guido van Rossum82598051997-03-05 00:20:32 +00001210PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001211PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001212 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001213{
Neal Norwitze92fba02006-03-04 18:52:26 +00001214 PyObject *ret = NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001215 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001216 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001217 if (arena == NULL)
1218 return NULL;
1219
1220 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001221 if (mod != NULL)
1222 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001223 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001224 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001225}
1226
1227PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001228PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001229 PyObject *locals, int closeit, PyCompilerFlags *flags)
1230{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001231 PyObject *ret;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001232 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001233 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001234 if (arena == NULL)
1235 return NULL;
1236
1237 mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
1238 flags, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001239 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001240 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001241 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001242 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001243 if (closeit)
1244 fclose(fp);
Neal Norwitze92fba02006-03-04 18:52:26 +00001245 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001246 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001247 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001248}
1249
Guido van Rossum82598051997-03-05 00:20:32 +00001250static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001251run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001252 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001253{
Guido van Rossum82598051997-03-05 00:20:32 +00001254 PyCodeObject *co;
1255 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001256 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001257 if (co == NULL)
1258 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001259 v = PyEval_EvalCode(co, globals, locals);
1260 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001261 return v;
1262}
1263
Guido van Rossum82598051997-03-05 00:20:32 +00001264static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001265run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001266 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001267{
Guido van Rossum82598051997-03-05 00:20:32 +00001268 PyCodeObject *co;
1269 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001270 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001271 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001272
Guido van Rossum82598051997-03-05 00:20:32 +00001273 magic = PyMarshal_ReadLongFromFile(fp);
1274 if (magic != PyImport_GetMagicNumber()) {
1275 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001276 "Bad magic number in .pyc file");
1277 return NULL;
1278 }
Guido van Rossum82598051997-03-05 00:20:32 +00001279 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001280 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001281 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001282 if (v == NULL || !PyCode_Check(v)) {
1283 Py_XDECREF(v);
1284 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001285 "Bad code object in .pyc file");
1286 return NULL;
1287 }
Guido van Rossum82598051997-03-05 00:20:32 +00001288 co = (PyCodeObject *)v;
1289 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001290 if (v && flags)
1291 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001292 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001293 return v;
1294}
1295
Guido van Rossum82598051997-03-05 00:20:32 +00001296PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001297Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001298 PyCompilerFlags *flags)
1299{
Guido van Rossum82598051997-03-05 00:20:32 +00001300 PyCodeObject *co;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001301 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001302 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001303 if (arena == NULL)
1304 return NULL;
1305
1306 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001307 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001308 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001309 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001310 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001311 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001312 PyObject *result = PyAST_mod2obj(mod);
1313 PyArena_Free(arena);
1314 return result;
1315 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001316 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001317 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001318 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001319}
1320
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001321struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001322Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001323{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001324 struct symtable *st;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001325 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001326 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001327 if (arena == NULL)
1328 return NULL;
1329
1330 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001331 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001332 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001333 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001334 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001335 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001336 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001337 return st;
1338}
1339
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001340/* Preferred access to parser is through AST. */
1341mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001342PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001343 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001344{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001345 mod_ty mod;
1346 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001347 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001348 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001349 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001350 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001351 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001352 PyNode_Free(n);
1353 return mod;
1354 }
1355 else {
1356 err_input(&err);
1357 return NULL;
1358 }
1359}
1360
1361mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001362PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001363 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001364 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001365{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001366 mod_ty mod;
1367 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001368 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1369 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001370 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001371 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001372 PyNode_Free(n);
1373 return mod;
1374 }
1375 else {
1376 err_input(&err);
1377 if (errcode)
1378 *errcode = err.error;
1379 return NULL;
1380 }
1381}
1382
Guido van Rossuma110aa61994-08-29 12:50:44 +00001383/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001384
Guido van Rossuma110aa61994-08-29 12:50:44 +00001385node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001386PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001387{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001388 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001389 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1390 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001391 if (n == NULL)
1392 err_input(&err);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001393
Guido van Rossuma110aa61994-08-29 12:50:44 +00001394 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001395}
1396
Guido van Rossuma110aa61994-08-29 12:50:44 +00001397/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001398
Guido van Rossuma110aa61994-08-29 12:50:44 +00001399node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001400PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001401{
Tim Petersfe2127d2001-07-16 05:37:24 +00001402 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001403 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1404 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001405 if (n == NULL)
1406 err_input(&err);
1407 return n;
1408}
1409
1410node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001411PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001412 int start, int flags)
1413{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001414 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001415 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1416 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001417 if (n == NULL)
1418 err_input(&err);
1419 return n;
1420}
1421
1422node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001423PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001424{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001425 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001426}
1427
Guido van Rossum66ebd912003-04-17 16:02:26 +00001428/* May want to move a more generalized form of this to parsetok.c or
1429 even parser modules. */
1430
1431void
1432PyParser_SetError(perrdetail *err)
1433{
1434 err_input(err);
1435}
1436
Guido van Rossuma110aa61994-08-29 12:50:44 +00001437/* Set the error appropriate to the given input error code (see errcode.h) */
1438
1439static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001440err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001441{
Fred Drake85f36392000-07-11 17:53:00 +00001442 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001443 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001444 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001445 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001446 switch (err->error) {
1447 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001448 errtype = PyExc_IndentationError;
1449 if (err->expected == INDENT)
1450 msg = "expected an indented block";
1451 else if (err->token == INDENT)
1452 msg = "unexpected indent";
1453 else if (err->token == DEDENT)
1454 msg = "unexpected unindent";
1455 else {
1456 errtype = PyExc_SyntaxError;
1457 msg = "invalid syntax";
1458 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001459 break;
1460 case E_TOKEN:
1461 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001462 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001463 case E_EOFS:
1464 msg = "EOF while scanning triple-quoted string";
1465 break;
1466 case E_EOLS:
1467 msg = "EOL while scanning single-quoted string";
1468 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001469 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001470 if (!PyErr_Occurred())
1471 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001472 return;
1473 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001474 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001475 return;
1476 case E_EOF:
1477 msg = "unexpected EOF while parsing";
1478 break;
Fred Drake85f36392000-07-11 17:53:00 +00001479 case E_TABSPACE:
1480 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001481 msg = "inconsistent use of tabs and spaces in indentation";
1482 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001483 case E_OVERFLOW:
1484 msg = "expression too long";
1485 break;
Fred Drake85f36392000-07-11 17:53:00 +00001486 case E_DEDENT:
1487 errtype = PyExc_IndentationError;
1488 msg = "unindent does not match any outer indentation level";
1489 break;
1490 case E_TOODEEP:
1491 errtype = PyExc_IndentationError;
1492 msg = "too many levels of indentation";
1493 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001494 case E_DECODE: {
1495 PyObject *type, *value, *tb;
1496 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001497 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001498 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001499 if (u != NULL) {
1500 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001501 }
1502 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001503 if (msg == NULL)
1504 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001505 Py_XDECREF(type);
1506 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001507 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001508 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001509 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001510 case E_LINECONT:
1511 msg = "unexpected character after line continuation character";
1512 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001513 default:
1514 fprintf(stderr, "error=%d\n", err->error);
1515 msg = "unknown parsing error";
1516 break;
1517 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001518 v = Py_BuildValue("(ziiz)", err->filename,
1519 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001520 if (err->text != NULL) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001521 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001522 err->text = NULL;
1523 }
1524 w = NULL;
1525 if (v != NULL)
1526 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001527 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001528 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001529 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001530 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001531}
1532
1533/* Print fatal error message and abort */
1534
1535void
Tim Peters7c321a82002-07-09 02:57:01 +00001536Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001537{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001538 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001539#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001540 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001541 OutputDebugString(msg);
1542 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001543#ifdef _DEBUG
1544 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001545#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001546#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001547 abort();
1548}
1549
1550/* Clean up and exit */
1551
Guido van Rossuma110aa61994-08-29 12:50:44 +00001552#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001553#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001554#endif
1555
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001556#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001557static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001558static int nexitfuncs = 0;
1559
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001560int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001561{
1562 if (nexitfuncs >= NEXITFUNCS)
1563 return -1;
1564 exitfuncs[nexitfuncs++] = func;
1565 return 0;
1566}
1567
Guido van Rossumcc283f51997-08-05 02:22:03 +00001568static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001569call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001570{
Guido van Rossum82598051997-03-05 00:20:32 +00001571 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001572
1573 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001574 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001575 Py_INCREF(exitfunc);
1576 PySys_SetObject("exitfunc", (PyObject *)NULL);
1577 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001578 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001579 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1580 PySys_WriteStderr("Error in sys.exitfunc:\n");
1581 }
Guido van Rossum82598051997-03-05 00:20:32 +00001582 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001583 }
Guido van Rossum82598051997-03-05 00:20:32 +00001584 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001585 }
1586
Guido van Rossumcc283f51997-08-05 02:22:03 +00001587}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001588
Guido van Rossumcc283f51997-08-05 02:22:03 +00001589static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001590call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001591{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001592 while (nexitfuncs > 0)
1593 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001594
1595 fflush(stdout);
1596 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001597}
1598
1599void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001600Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001601{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001602 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001603
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001604 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001605}
1606
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001607static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001608initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001609{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001610#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001611 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001612#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001613#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001614 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001615#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001616#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001617 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001618#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001619 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001620}
1621
Guido van Rossum7433b121997-02-14 19:45:36 +00001622
1623/*
1624 * The file descriptor fd is considered ``interactive'' if either
1625 * a) isatty(fd) is TRUE, or
1626 * b) the -i flag was given, and the filename associated with
1627 * the descriptor is NULL or "<stdin>" or "???".
1628 */
1629int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001630Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001631{
1632 if (isatty((int)fileno(fp)))
1633 return 1;
1634 if (!Py_InteractiveFlag)
1635 return 0;
1636 return (filename == NULL) ||
1637 (strcmp(filename, "<stdin>") == 0) ||
1638 (strcmp(filename, "???") == 0);
1639}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001640
1641
Tim Petersd08e3822003-04-17 15:24:21 +00001642#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001643#if defined(WIN32) && defined(_MSC_VER)
1644
1645/* Stack checking for Microsoft C */
1646
1647#include <malloc.h>
1648#include <excpt.h>
1649
Fred Drakee8de31c2000-08-31 05:38:39 +00001650/*
1651 * Return non-zero when we run out of memory on the stack; zero otherwise.
1652 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001653int
Fred Drake399739f2000-08-31 05:52:44 +00001654PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001655{
1656 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001657 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001658 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001659 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001660 return 0;
1661 } __except (EXCEPTION_EXECUTE_HANDLER) {
1662 /* just ignore all errors */
1663 }
1664 return 1;
1665}
1666
1667#endif /* WIN32 && _MSC_VER */
1668
1669/* Alternate implementations can be added here... */
1670
1671#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001672
1673
1674/* Wrappers around sigaction() or signal(). */
1675
1676PyOS_sighandler_t
1677PyOS_getsig(int sig)
1678{
1679#ifdef HAVE_SIGACTION
1680 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001681 if (sigaction(sig, NULL, &context) == -1)
1682 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001683 return context.sa_handler;
1684#else
1685 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001686/* Special signal handling for the secure CRT in Visual Studio 2005 */
1687#if defined(_MSC_VER) && _MSC_VER >= 1400
1688 switch (sig) {
1689 /* Only these signals are valid */
1690 case SIGINT:
1691 case SIGILL:
1692 case SIGFPE:
1693 case SIGSEGV:
1694 case SIGTERM:
1695 case SIGBREAK:
1696 case SIGABRT:
1697 break;
1698 /* Don't call signal() with other values or it will assert */
1699 default:
1700 return SIG_ERR;
1701 }
1702#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001703 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001704 if (handler != SIG_ERR)
1705 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001706 return handler;
1707#endif
1708}
1709
1710PyOS_sighandler_t
1711PyOS_setsig(int sig, PyOS_sighandler_t handler)
1712{
1713#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001714 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001715 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001716 sigemptyset(&context.sa_mask);
1717 context.sa_flags = 0;
1718 if (sigaction(sig, &context, &ocontext) == -1)
1719 return SIG_ERR;
1720 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001721#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001722 PyOS_sighandler_t oldhandler;
1723 oldhandler = signal(sig, handler);
1724#ifdef HAVE_SIGINTERRUPT
1725 siginterrupt(sig, 1);
1726#endif
1727 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001728#endif
1729}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001730
1731/* Deprecated C API functions still provided for binary compatiblity */
1732
1733#undef PyParser_SimpleParseFile
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001734PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001735PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1736{
1737 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1738}
1739
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001740#undef PyParser_SimpleParseString
1741PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001742PyParser_SimpleParseString(const char *str, int start)
1743{
1744 return PyParser_SimpleParseStringFlags(str, start, 0);
1745}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001746
1747#undef PyRun_AnyFile
1748PyAPI_FUNC(int)
1749PyRun_AnyFile(FILE *fp, const char *name)
1750{
1751 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1752}
1753
1754#undef PyRun_AnyFileEx
1755PyAPI_FUNC(int)
1756PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1757{
1758 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1759}
1760
1761#undef PyRun_AnyFileFlags
1762PyAPI_FUNC(int)
1763PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1764{
1765 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1766}
1767
1768#undef PyRun_File
1769PyAPI_FUNC(PyObject *)
1770PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1771{
1772 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1773}
1774
1775#undef PyRun_FileEx
1776PyAPI_FUNC(PyObject *)
1777PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1778{
1779 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1780}
1781
1782#undef PyRun_FileFlags
1783PyAPI_FUNC(PyObject *)
1784PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1785 PyCompilerFlags *flags)
1786{
1787 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1788}
1789
1790#undef PyRun_SimpleFile
1791PyAPI_FUNC(int)
1792PyRun_SimpleFile(FILE *f, const char *p)
1793{
1794 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1795}
1796
1797#undef PyRun_SimpleFileEx
1798PyAPI_FUNC(int)
1799PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1800{
1801 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1802}
1803
1804
1805#undef PyRun_String
1806PyAPI_FUNC(PyObject *)
1807PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1808{
1809 return PyRun_StringFlags(str, s, g, l, NULL);
1810}
1811
1812#undef PyRun_SimpleString
1813PyAPI_FUNC(int)
1814PyRun_SimpleString(const char *s)
1815{
1816 return PyRun_SimpleStringFlags(s, NULL);
1817}
1818
1819#undef Py_CompileString
1820PyAPI_FUNC(PyObject *)
1821Py_CompileString(const char *str, const char *p, int s)
1822{
1823 return Py_CompileStringFlags(str, p, s, NULL);
1824}
1825
1826#undef PyRun_InteractiveOne
1827PyAPI_FUNC(int)
1828PyRun_InteractiveOne(FILE *f, const char *p)
1829{
1830 return PyRun_InteractiveOneFlags(f, p, NULL);
1831}
1832
1833#undef PyRun_InteractiveLoop
1834PyAPI_FUNC(int)
1835PyRun_InteractiveLoop(FILE *f, const char *p)
1836{
1837 return PyRun_InteractiveLoopFlags(f, p, NULL);
1838}
1839
1840#ifdef __cplusplus
1841}
1842#endif