blob: 8283fc5c98a319f46050c283c461a833984409cd [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
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000020#include <signal.h>
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000021
Martin v. Löwis73d538b2003-03-05 15:13:47 +000022#ifdef HAVE_LANGINFO_H
23#include <locale.h>
24#include <langinfo.h>
25#endif
26
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000027#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000028#undef BYTE
29#include "windows.h"
30#endif
31
Neal Norwitz4281cef2006-03-04 19:58:13 +000032#ifndef Py_REF_DEBUG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000033#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000034#else /* Py_REF_DEBUG */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000035#define PRINT_TOTAL_REFS() fprintf(stderr, \
36 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
37 _Py_GetRefTotal())
38#endif
39
40#ifdef __cplusplus
41extern "C" {
Neal Norwitz4281cef2006-03-04 19:58:13 +000042#endif
43
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000044extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000045
Guido van Rossum82598051997-03-05 00:20:32 +000046extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000047
Guido van Rossumb73cc041993-11-01 16:28:59 +000048/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000049static void initmain(void);
50static void initsite(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000051static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000052 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000053static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000054 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000055static void err_input(perrdetail *);
56static void initsigs(void);
57static void call_sys_exitfunc(void);
58static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000059extern void _PyUnicode_Init(void);
60extern void _PyUnicode_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000061
Mark Hammond8d98d2c2003-04-19 15:41:53 +000062#ifdef WITH_THREAD
63extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
64extern void _PyGILState_Fini(void);
65#endif /* WITH_THREAD */
66
Guido van Rossum82598051997-03-05 00:20:32 +000067int Py_DebugFlag; /* Needed by parser.c */
68int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000069int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000070int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000071int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000072int Py_FrozenFlag; /* Needed by getpath.c */
Guido van Rossumb16d1972000-05-01 17:55:15 +000073int Py_UnicodeFlag = 0; /* Needed by compile.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000074int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000075
Mark Hammonda43fd0c2003-02-19 00:33:33 +000076/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000077 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000078*/
Mark Hammondedd07732003-07-15 23:03:55 +000079static PyObject *warnings_module = NULL;
80
81/* Returns a borrowed reference to the 'warnings' module, or NULL.
82 If the module is returned, it is guaranteed to have been obtained
83 without acquiring the import lock
84*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000085PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000086{
87 PyObject *typ, *val, *tb;
88 PyObject *all_modules;
89 /* If we managed to get the module at init time, just use it */
90 if (warnings_module)
91 return warnings_module;
92 /* If it wasn't available at init time, it may be available
93 now in sys.modules (common scenario is frozen apps: import
94 at init time fails, but the frozen init code sets up sys.path
95 correctly, then does an implicit import of warnings for us
96 */
97 /* Save and restore any exceptions */
98 PyErr_Fetch(&typ, &val, &tb);
99
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000100 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000101 if (all_modules) {
102 warnings_module = PyDict_GetItemString(all_modules, "warnings");
103 /* We keep a ref in the global */
104 Py_XINCREF(warnings_module);
105 }
106 PyErr_Restore(typ, val, tb);
107 return warnings_module;
108}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000109
Guido van Rossum25ce5661997-08-02 03:10:38 +0000110static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000111
Thomas Wouters7e474022000-07-16 12:04:32 +0000112/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000113
114int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000115Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000116{
117 return initialized;
118}
119
Guido van Rossum25ce5661997-08-02 03:10:38 +0000120/* Global initializations. Can be undone by Py_Finalize(). Don't
121 call this twice without an intervening Py_Finalize() call. When
122 initializations fail, a fatal error is issued and the function does
123 not return. On return, the first thread and interpreter state have
124 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000125
Guido van Rossum25ce5661997-08-02 03:10:38 +0000126 Locking: you must hold the interpreter lock while calling this.
127 (If the lock has not yet been initialized, that's equivalent to
128 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000129
Guido van Rossum25ce5661997-08-02 03:10:38 +0000130*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000131
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000132static int
133add_flag(int flag, const char *envs)
134{
135 int env = atoi(envs);
136 if (flag < env)
137 flag = env;
138 if (flag < 1)
139 flag = 1;
140 return flag;
141}
142
Guido van Rossuma027efa1997-05-05 20:56:21 +0000143void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000144Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000145{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000146 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000147 PyThreadState *tstate;
148 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000149 char *p;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000150#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
151 char *codeset;
152 char *saved_locale;
153 PyObject *sys_stream, *sys_isatty;
154#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000155 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000156
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000157 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000158 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000159 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000160
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000161 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000162 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000163 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000164 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000165 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000166 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000167
Guido van Rossuma027efa1997-05-05 20:56:21 +0000168 interp = PyInterpreterState_New();
169 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000170 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000171
Guido van Rossuma027efa1997-05-05 20:56:21 +0000172 tstate = PyThreadState_New(interp);
173 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000174 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000175 (void) PyThreadState_Swap(tstate);
176
Guido van Rossum70d893a2001-08-16 08:21:42 +0000177 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000178
Neal Norwitzb2501f42002-12-31 03:42:13 +0000179 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000180 Py_FatalError("Py_Initialize: can't init frames");
181
Neal Norwitzb2501f42002-12-31 03:42:13 +0000182 if (!_PyInt_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000183 Py_FatalError("Py_Initialize: can't init ints");
184
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000185 _PyFloat_Init();
186
Guido van Rossum25ce5661997-08-02 03:10:38 +0000187 interp->modules = PyDict_New();
188 if (interp->modules == NULL)
189 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000190
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000191#ifdef Py_USING_UNICODE
Guido van Rossumc94044c2000-03-10 23:03:54 +0000192 /* Init Unicode implementation; relies on the codec registry */
193 _PyUnicode_Init();
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000194#endif
Guido van Rossumc94044c2000-03-10 23:03:54 +0000195
Barry Warsawf242aa02000-05-25 23:09:49 +0000196 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000197 if (bimod == NULL)
198 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000199 interp->builtins = PyModule_GetDict(bimod);
200 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000201
202 sysmod = _PySys_Init();
203 if (sysmod == NULL)
204 Py_FatalError("Py_Initialize: can't initialize sys");
205 interp->sysdict = PyModule_GetDict(sysmod);
206 Py_INCREF(interp->sysdict);
207 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000208 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000209 PyDict_SetItemString(interp->sysdict, "modules",
210 interp->modules);
211
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000212 _PyImport_Init();
213
Barry Warsawf242aa02000-05-25 23:09:49 +0000214 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000215 _PyExc_Init();
Barry Warsaw5821bc52001-08-13 23:04:56 +0000216 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000217
Barry Warsaw035574d1997-08-29 22:07:17 +0000218 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000219 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000220
Just van Rossum52e14d62002-12-30 22:08:05 +0000221 _PyImportHooks_Init();
222
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000223 if (install_sigs)
224 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000225
226 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000227 if (!Py_NoSiteFlag)
228 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000229
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000230 /* auto-thread-state API, if available */
231#ifdef WITH_THREAD
232 _PyGILState_Init(interp, tstate);
233#endif /* WITH_THREAD */
234
Mark Hammondedd07732003-07-15 23:03:55 +0000235 warnings_module = PyImport_ImportModule("warnings");
236 if (!warnings_module)
237 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000238
239#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
240 /* On Unix, set the file system encoding according to the
241 user's preference, if the CODESET names a well-known
242 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000243 initialized by other means. Also set the encoding of
244 stdin and stdout if these are terminals. */
245
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000246 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000247 setlocale(LC_CTYPE, "");
248 codeset = nl_langinfo(CODESET);
249 if (codeset && *codeset) {
250 PyObject *enc = PyCodec_Encoder(codeset);
251 if (enc) {
252 codeset = strdup(codeset);
253 Py_DECREF(enc);
254 } else {
255 codeset = NULL;
256 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000257 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000258 } else
259 codeset = NULL;
260 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000261 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000262
263 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000264 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000265 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
266 if (!sys_isatty)
267 PyErr_Clear();
268 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
269 if (!PyFile_SetEncoding(sys_stream, codeset))
270 Py_FatalError("Cannot set codeset of stdin");
271 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000272 Py_XDECREF(sys_isatty);
273
274 sys_stream = PySys_GetObject("stdout");
275 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
276 if (!sys_isatty)
277 PyErr_Clear();
278 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
279 if (!PyFile_SetEncoding(sys_stream, codeset))
280 Py_FatalError("Cannot set codeset of stdout");
281 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000282 Py_XDECREF(sys_isatty);
283
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000284 sys_stream = PySys_GetObject("stderr");
285 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
286 if (!sys_isatty)
287 PyErr_Clear();
288 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
289 if (!PyFile_SetEncoding(sys_stream, codeset))
290 Py_FatalError("Cannot set codeset of stderr");
291 }
292 Py_XDECREF(sys_isatty);
293
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000294 if (!Py_FileSystemDefaultEncoding)
295 Py_FileSystemDefaultEncoding = codeset;
296 else
297 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000298 }
299#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000300}
301
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000302void
303Py_Initialize(void)
304{
305 Py_InitializeEx(1);
306}
307
308
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000309#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000310extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000311#endif
312
Guido van Rossum25ce5661997-08-02 03:10:38 +0000313/* Undo the effect of Py_Initialize().
314
315 Beware: if multiple interpreter and/or thread states exist, these
316 are not wiped out; only the current thread and interpreter state
317 are deleted. But since everything else is deleted, those other
318 interpreter and thread states should no longer be used.
319
320 (XXX We should do better, e.g. wipe out all interpreters and
321 threads.)
322
323 Locking: as above.
324
325*/
326
327void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000328Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000329{
330 PyInterpreterState *interp;
331 PyThreadState *tstate;
332
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000333 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000334 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000335
Tim Peters384fd102001-01-21 03:40:37 +0000336 /* The interpreter is still entirely intact at this point, and the
337 * exit funcs may be relying on that. In particular, if some thread
338 * or exit func is still waiting to do an import, the import machinery
339 * expects Py_IsInitialized() to return true. So don't say the
340 * interpreter is uninitialized until after the exit funcs have run.
341 * Note that Threading.py uses an exit func to do a join on all the
342 * threads created thru it, so this also protects pending imports in
343 * the threads created via Threading.
344 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000345 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000346 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000347
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000348 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000349 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000350 interp = tstate->interp;
351
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000352 /* Disable signal handling */
353 PyOS_FiniInterrupts();
354
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000355 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000356 Py_XDECREF(warnings_module);
357 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000358
Guido van Rossume13ddc92003-04-17 17:29:22 +0000359 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000360 * before all modules are destroyed.
361 * XXX If a __del__ or weakref callback is triggered here, and tries to
362 * XXX import a module, bad things can happen, because Python no
363 * XXX longer believes it's initialized.
364 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
365 * XXX is easy to provoke that way. I've also seen, e.g.,
366 * XXX Exception exceptions.ImportError: 'No module named sha'
367 * XXX in <function callback at 0x008F5718> ignored
368 * XXX but I'm unclear on exactly how that one happens. In any case,
369 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000370 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000371 PyGC_Collect();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000372#ifdef COUNT_ALLOCS
373 /* With COUNT_ALLOCS, it helps to run GC multiple times:
374 each collection might release some types from the type
375 list, so they become garbage. */
376 while (PyGC_Collect() > 0)
377 /* nothing */;
378#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000379
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000380 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000381 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000382
Guido van Rossume13ddc92003-04-17 17:29:22 +0000383 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000384 * new-style class definitions, for example.
385 * XXX This is disabled because it caused too many problems. If
386 * XXX a __del__ or weakref callback triggers here, Python code has
387 * XXX a hard time running, because even the sys module has been
388 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
389 * XXX One symptom is a sequence of information-free messages
390 * XXX coming from threads (if a __del__ or callback is invoked,
391 * XXX other threads can execute too, and any exception they encounter
392 * XXX triggers a comedy of errors as subsystem after subsystem
393 * XXX fails to find what it *expects* to find in sys to help report
394 * XXX the exception and consequent unexpected failures). I've also
395 * XXX seen segfaults then, after adding print statements to the
396 * XXX Python code getting called.
397 */
398#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000399 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000400#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000401
Guido van Rossum1707aad1997-12-08 23:43:45 +0000402 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
403 _PyImport_Fini();
404
405 /* Debugging stuff */
406#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000407 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000408#endif
409
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000410 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000411
Tim Peters9cf25ce2003-04-17 15:21:01 +0000412#ifdef Py_TRACE_REFS
413 /* Display all objects still alive -- this can invoke arbitrary
414 * __repr__ overrides, so requires a mostly-intact interpreter.
415 * Alas, a lot of stuff may still be alive now that will be cleaned
416 * up later.
417 */
Tim Peters269b2a62003-04-17 19:52:29 +0000418 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000419 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000420#endif /* Py_TRACE_REFS */
421
Mark Hammond6cb90292003-04-22 11:18:00 +0000422 /* Cleanup auto-thread-state */
423#ifdef WITH_THREAD
424 _PyGILState_Fini();
425#endif /* WITH_THREAD */
426
Guido van Rossumd922fa42003-04-15 14:10:09 +0000427 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000428 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000429
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000430 /* Now we decref the exception classes. After this point nothing
431 can raise an exception. That's okay, because each Fini() method
432 below has been checked to make sure no exceptions are ever
433 raised.
434 */
435
436 _PyExc_Fini();
437
Guido van Rossumd922fa42003-04-15 14:10:09 +0000438 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000439 PyThreadState_Swap(NULL);
440 PyInterpreterState_Delete(interp);
441
Guido van Rossumd922fa42003-04-15 14:10:09 +0000442 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000443 PyMethod_Fini();
444 PyFrame_Fini();
445 PyCFunction_Fini();
446 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000447 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000448 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000449 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000450 PyInt_Fini();
451 PyFloat_Fini();
452
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000453#ifdef Py_USING_UNICODE
454 /* Cleanup Unicode implementation */
455 _PyUnicode_Fini();
456#endif
457
Guido van Rossumcc283f51997-08-05 02:22:03 +0000458 /* XXX Still allocated:
459 - various static ad-hoc pointers to interned strings
460 - int and float free list blocks
461 - whatever various modules and libraries allocate
462 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000463
464 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000465
Tim Peters269b2a62003-04-17 19:52:29 +0000466#ifdef Py_TRACE_REFS
467 /* Display addresses (& refcnts) of all objects still alive.
468 * An address can be used to find the repr of the object, printed
469 * above by _Py_PrintReferences.
470 */
471 if (Py_GETENV("PYTHONDUMPREFS"))
472 _Py_PrintReferenceAddresses(stderr);
473#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000474#ifdef PYMALLOC_DEBUG
475 if (Py_GETENV("PYTHONMALLOCSTATS"))
476 _PyObject_DebugMallocStats();
477#endif
478
Guido van Rossumcc283f51997-08-05 02:22:03 +0000479 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000480}
481
482/* Create and initialize a new interpreter and thread, and return the
483 new thread. This requires that Py_Initialize() has been called
484 first.
485
486 Unsuccessful initialization yields a NULL pointer. Note that *no*
487 exception information is available even in this case -- the
488 exception information is held in the thread, and there is no
489 thread.
490
491 Locking: as above.
492
493*/
494
495PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000496Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000497{
498 PyInterpreterState *interp;
499 PyThreadState *tstate, *save_tstate;
500 PyObject *bimod, *sysmod;
501
502 if (!initialized)
503 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
504
505 interp = PyInterpreterState_New();
506 if (interp == NULL)
507 return NULL;
508
509 tstate = PyThreadState_New(interp);
510 if (tstate == NULL) {
511 PyInterpreterState_Delete(interp);
512 return NULL;
513 }
514
515 save_tstate = PyThreadState_Swap(tstate);
516
517 /* XXX The following is lax in error checking */
518
519 interp->modules = PyDict_New();
520
521 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
522 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000523 interp->builtins = PyModule_GetDict(bimod);
524 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000525 }
526 sysmod = _PyImport_FindExtension("sys", "sys");
527 if (bimod != NULL && sysmod != NULL) {
528 interp->sysdict = PyModule_GetDict(sysmod);
529 Py_INCREF(interp->sysdict);
530 PySys_SetPath(Py_GetPath());
531 PyDict_SetItemString(interp->sysdict, "modules",
532 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000533 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000534 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000535 if (!Py_NoSiteFlag)
536 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000537 }
538
539 if (!PyErr_Occurred())
540 return tstate;
541
542 /* Oops, it didn't work. Undo it all. */
543
544 PyErr_Print();
545 PyThreadState_Clear(tstate);
546 PyThreadState_Swap(save_tstate);
547 PyThreadState_Delete(tstate);
548 PyInterpreterState_Delete(interp);
549
550 return NULL;
551}
552
553/* Delete an interpreter and its last thread. This requires that the
554 given thread state is current, that the thread has no remaining
555 frames, and that it is its interpreter's only remaining thread.
556 It is a fatal error to violate these constraints.
557
558 (Py_Finalize() doesn't have these constraints -- it zaps
559 everything, regardless.)
560
561 Locking: as above.
562
563*/
564
565void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000566Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000567{
568 PyInterpreterState *interp = tstate->interp;
569
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000570 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000571 Py_FatalError("Py_EndInterpreter: thread is not current");
572 if (tstate->frame != NULL)
573 Py_FatalError("Py_EndInterpreter: thread still has a frame");
574 if (tstate != interp->tstate_head || tstate->next != NULL)
575 Py_FatalError("Py_EndInterpreter: not the last thread");
576
577 PyImport_Cleanup();
578 PyInterpreterState_Clear(interp);
579 PyThreadState_Swap(NULL);
580 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000581}
582
583static char *progname = "python";
584
585void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000586Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000587{
588 if (pn && *pn)
589 progname = pn;
590}
591
592char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000593Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000594{
595 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000596}
597
Guido van Rossuma61691e1998-02-06 22:27:24 +0000598static char *default_home = NULL;
599
600void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000601Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000602{
603 default_home = home;
604}
605
606char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000607Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000608{
609 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000610 if (home == NULL && !Py_IgnoreEnvironmentFlag)
611 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000612 return home;
613}
614
Guido van Rossum6135a871995-01-09 17:53:26 +0000615/* Create __main__ module */
616
617static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000618initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000619{
Guido van Rossum82598051997-03-05 00:20:32 +0000620 PyObject *m, *d;
621 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000622 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000623 Py_FatalError("can't create __main__ module");
624 d = PyModule_GetDict(m);
625 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000626 PyObject *bimod = PyImport_ImportModule("__builtin__");
627 if (bimod == NULL ||
628 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000629 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000630 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000631 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000632}
633
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000634/* Import the site module (not into __main__ though) */
635
636static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000637initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000638{
639 PyObject *m, *f;
640 m = PyImport_ImportModule("site");
641 if (m == NULL) {
642 f = PySys_GetObject("stderr");
643 if (Py_VerboseFlag) {
644 PyFile_WriteString(
645 "'import site' failed; traceback:\n", f);
646 PyErr_Print();
647 }
648 else {
649 PyFile_WriteString(
650 "'import site' failed; use -v for traceback\n", f);
651 PyErr_Clear();
652 }
653 }
654 else {
655 Py_DECREF(m);
656 }
657}
658
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000659/* Parse input from a file and execute it */
660
661int
Martin v. Löwis056a69c2006-03-01 16:55:42 +0000662PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000663 PyCompilerFlags *flags)
664{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000665 if (filename == NULL)
666 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000667 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000668 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000669 if (closeit)
670 fclose(fp);
671 return err;
672 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000673 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000674 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000675}
676
677int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000678PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000679{
Guido van Rossum82598051997-03-05 00:20:32 +0000680 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000681 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000682 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000683
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000684 if (flags == NULL) {
685 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000686 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000687 }
Guido van Rossum82598051997-03-05 00:20:32 +0000688 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000689 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000690 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
691 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000692 }
Guido van Rossum82598051997-03-05 00:20:32 +0000693 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000694 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000695 PySys_SetObject("ps2", v = PyString_FromString("... "));
696 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000697 }
698 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000699 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000700 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000701 if (ret == E_EOF)
702 return 0;
703 /*
704 if (ret == E_NOMEM)
705 return -1;
706 */
707 }
708}
709
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000710/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000711#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000712 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Guido van Rossum45aecf42006-03-15 04:58:47 +0000713 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000714
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000715int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000716PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000717{
Guido van Rossum82598051997-03-05 00:20:32 +0000718 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000719 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000720 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000721 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000722 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000723
Guido van Rossum82598051997-03-05 00:20:32 +0000724 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000725 if (v != NULL) {
726 v = PyObject_Str(v);
727 if (v == NULL)
728 PyErr_Clear();
729 else if (PyString_Check(v))
730 ps1 = PyString_AsString(v);
731 }
Guido van Rossum82598051997-03-05 00:20:32 +0000732 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000733 if (w != NULL) {
734 w = PyObject_Str(w);
735 if (w == NULL)
736 PyErr_Clear();
737 else if (PyString_Check(w))
738 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000739 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000740 arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000741 mod = PyParser_ASTFromFile(fp, filename,
742 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000743 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000744 Py_XDECREF(v);
745 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000746 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000747 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000748 if (errcode == E_EOF) {
749 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000750 return E_EOF;
751 }
Guido van Rossum82598051997-03-05 00:20:32 +0000752 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000753 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000754 }
Guido van Rossum82598051997-03-05 00:20:32 +0000755 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000756 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000757 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000758 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000759 }
Guido van Rossum82598051997-03-05 00:20:32 +0000760 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000761 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000762 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000763 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000764 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000765 return -1;
766 }
Guido van Rossum82598051997-03-05 00:20:32 +0000767 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000768 if (Py_FlushLine())
769 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000770 return 0;
771}
772
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000773/* Check whether a file maybe a pyc file: Look at the extension,
774 the file type, and, if we may close it, at the first few bytes. */
775
776static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000777maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000778{
779 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
780 return 1;
781
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000782 /* Only look into the file if we are allowed to close it, since
783 it then should also be seekable. */
784 if (closeit) {
785 /* Read only two bytes of the magic. If the file was opened in
786 text mode, the bytes 3 and 4 of the magic (\r\n) might not
787 be read as they are on disk. */
788 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
789 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000790 /* Mess: In case of -x, the stream is NOT at its start now,
791 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000792 which makes the current stream position formally undefined,
793 and a x-platform nightmare.
794 Unfortunately, we have no direct way to know whether -x
795 was specified. So we use a terrible hack: if the current
796 stream position is not 0, we assume -x was specified, and
797 give up. Bug 132850 on SourceForge spells out the
798 hopelessness of trying anything else (fseek and ftell
799 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000800 */
Tim Peters3e876562001-02-11 04:35:39 +0000801 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000802 if (ftell(fp) == 0) {
803 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000804 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000805 ispyc = 1;
806 rewind(fp);
807 }
Tim Peters3e876562001-02-11 04:35:39 +0000808 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000809 }
810 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000811}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000812
Guido van Rossum0df002c2000-08-27 19:21:52 +0000813int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000814PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000815 PyCompilerFlags *flags)
816{
Guido van Rossum82598051997-03-05 00:20:32 +0000817 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000818 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000819
Guido van Rossum82598051997-03-05 00:20:32 +0000820 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000821 if (m == NULL)
822 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000823 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000824 if (PyDict_GetItemString(d, "__file__") == NULL) {
825 PyObject *f = PyString_FromString(filename);
826 if (f == NULL)
827 return -1;
828 if (PyDict_SetItemString(d, "__file__", f) < 0) {
829 Py_DECREF(f);
830 return -1;
831 }
832 Py_DECREF(f);
833 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000834 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000835 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000836 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000837 if (closeit)
838 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000839 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000840 fprintf(stderr, "python: Can't reopen .pyc file\n");
841 return -1;
842 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000843 /* Turn on optimization if a .pyo file is given */
844 if (strcmp(ext, ".pyo") == 0)
845 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000846 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000847 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000848 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000849 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000850 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000851 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000852 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000853 return -1;
854 }
Guido van Rossum82598051997-03-05 00:20:32 +0000855 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000856 if (Py_FlushLine())
857 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000858 return 0;
859}
860
861int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000862PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000863{
Guido van Rossum82598051997-03-05 00:20:32 +0000864 PyObject *m, *d, *v;
865 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000866 if (m == NULL)
867 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000868 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000869 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000870 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000871 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000872 return -1;
873 }
Guido van Rossum82598051997-03-05 00:20:32 +0000874 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000875 if (Py_FlushLine())
876 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000877 return 0;
878}
879
Barry Warsaw035574d1997-08-29 22:07:17 +0000880static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000881parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
882 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000883{
884 long hold;
885 PyObject *v;
886
887 /* old style errors */
888 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000889 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000890 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000891
892 /* new style errors. `err' is an instance */
893
894 if (! (v = PyObject_GetAttrString(err, "msg")))
895 goto finally;
896 *message = v;
897
898 if (!(v = PyObject_GetAttrString(err, "filename")))
899 goto finally;
900 if (v == Py_None)
901 *filename = NULL;
902 else if (! (*filename = PyString_AsString(v)))
903 goto finally;
904
905 Py_DECREF(v);
906 if (!(v = PyObject_GetAttrString(err, "lineno")))
907 goto finally;
908 hold = PyInt_AsLong(v);
909 Py_DECREF(v);
910 v = NULL;
911 if (hold < 0 && PyErr_Occurred())
912 goto finally;
913 *lineno = (int)hold;
914
915 if (!(v = PyObject_GetAttrString(err, "offset")))
916 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000917 if (v == Py_None) {
918 *offset = -1;
919 Py_DECREF(v);
920 v = NULL;
921 } else {
922 hold = PyInt_AsLong(v);
923 Py_DECREF(v);
924 v = NULL;
925 if (hold < 0 && PyErr_Occurred())
926 goto finally;
927 *offset = (int)hold;
928 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000929
930 if (!(v = PyObject_GetAttrString(err, "text")))
931 goto finally;
932 if (v == Py_None)
933 *text = NULL;
934 else if (! (*text = PyString_AsString(v)))
935 goto finally;
936 Py_DECREF(v);
937 return 1;
938
939finally:
940 Py_XDECREF(v);
941 return 0;
942}
943
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000944void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000945PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000946{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000947 PyErr_PrintEx(1);
948}
949
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000950static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000951print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000952{
953 char *nl;
954 if (offset >= 0) {
955 if (offset > 0 && offset == (int)strlen(text))
956 offset--;
957 for (;;) {
958 nl = strchr(text, '\n');
959 if (nl == NULL || nl-text >= offset)
960 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000961 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000962 text = nl+1;
963 }
964 while (*text == ' ' || *text == '\t') {
965 text++;
966 offset--;
967 }
968 }
969 PyFile_WriteString(" ", f);
970 PyFile_WriteString(text, f);
971 if (*text == '\0' || text[strlen(text)-1] != '\n')
972 PyFile_WriteString("\n", f);
973 if (offset == -1)
974 return;
975 PyFile_WriteString(" ", f);
976 offset--;
977 while (offset > 0) {
978 PyFile_WriteString(" ", f);
979 offset--;
980 }
981 PyFile_WriteString("^\n", f);
982}
983
Guido van Rossum66e8e862001-03-23 17:54:43 +0000984static void
985handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000986{
Neal Norwitz9589ee22006-03-04 19:01:22 +0000987 PyObject *exception, *value, *tb;
988 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +0000989
Guido van Rossum66e8e862001-03-23 17:54:43 +0000990 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000991 if (Py_FlushLine())
992 PyErr_Clear();
993 fflush(stdout);
994 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +0000995 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +0000996 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000997 /* The error code should be in the `code' attribute. */
998 PyObject *code = PyObject_GetAttrString(value, "code");
999 if (code) {
1000 Py_DECREF(value);
1001 value = code;
1002 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001003 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001004 }
1005 /* If we failed to dig out the 'code' attribute,
1006 just let the else clause below print the error. */
1007 }
1008 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001009 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001010 else {
1011 PyObject_Print(value, stderr, Py_PRINT_RAW);
1012 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001013 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001014 }
Tim Peterscf615b52003-04-19 18:47:02 +00001015 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001016 /* Restore and clear the exception info, in order to properly decref
1017 * the exception, value, and traceback. If we just exit instead,
1018 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1019 * some finalizers from running.
1020 */
Tim Peterscf615b52003-04-19 18:47:02 +00001021 PyErr_Restore(exception, value, tb);
1022 PyErr_Clear();
1023 Py_Exit(exitcode);
1024 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001025}
1026
1027void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001028PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001029{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001030 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001031
1032 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1033 handle_system_exit();
1034 }
Guido van Rossum82598051997-03-05 00:20:32 +00001035 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001036 if (exception == NULL)
1037 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001038 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001039 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001040 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001041 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001042 if (set_sys_last_vars) {
1043 PySys_SetObject("last_type", exception);
1044 PySys_SetObject("last_value", v);
1045 PySys_SetObject("last_traceback", tb);
1046 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001047 hook = PySys_GetObject("excepthook");
1048 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001049 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001050 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001051 PyObject *result = PyEval_CallObject(hook, args);
1052 if (result == NULL) {
1053 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001054 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1055 handle_system_exit();
1056 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001057 PyErr_Fetch(&exception2, &v2, &tb2);
1058 PyErr_NormalizeException(&exception2, &v2, &tb2);
1059 if (Py_FlushLine())
1060 PyErr_Clear();
1061 fflush(stdout);
1062 PySys_WriteStderr("Error in sys.excepthook:\n");
1063 PyErr_Display(exception2, v2, tb2);
1064 PySys_WriteStderr("\nOriginal exception was:\n");
1065 PyErr_Display(exception, v, tb);
Jeremy Hylton07028582001-12-07 15:35:35 +00001066 Py_XDECREF(exception2);
1067 Py_XDECREF(v2);
1068 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001069 }
1070 Py_XDECREF(result);
1071 Py_XDECREF(args);
1072 } else {
1073 PySys_WriteStderr("sys.excepthook is missing\n");
1074 PyErr_Display(exception, v, tb);
1075 }
1076 Py_XDECREF(exception);
1077 Py_XDECREF(v);
1078 Py_XDECREF(tb);
1079}
1080
Thomas Wouters477c8d52006-05-27 19:21:47 +00001081void
1082PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001083{
1084 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001085 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001086 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001087 if (f == NULL)
1088 fprintf(stderr, "lost sys.stderr\n");
1089 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001090 if (Py_FlushLine())
1091 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001092 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001093 if (tb && tb != Py_None)
1094 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001095 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001096 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001097 {
Guido van Rossum82598051997-03-05 00:20:32 +00001098 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001099 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001100 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001101 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001102 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001103 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001104 else {
1105 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001106 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001107 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001108 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001109 else
Guido van Rossum82598051997-03-05 00:20:32 +00001110 PyFile_WriteString(filename, f);
1111 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001112 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001113 PyFile_WriteString(buf, f);
1114 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001115 if (text != NULL)
1116 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001117 Py_DECREF(value);
1118 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001119 /* Can't be bothered to check all those
1120 PyFile_WriteString() calls */
1121 if (PyErr_Occurred())
1122 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001123 }
1124 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001125 if (err) {
1126 /* Don't do anything else */
1127 }
Brett Cannonbf364092006-03-01 04:25:17 +00001128 else if (PyExceptionClass_Check(exception)) {
1129 char* className = PyExceptionClass_Name(exception);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001130 char *dot = strrchr(className, '.');
1131 PyObject* moduleName;
1132 if (dot != NULL)
1133 className = dot+1;
1134 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001135
1136 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001137 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001138 else {
1139 char* modstr = PyString_AsString(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001140 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001141 {
1142 err = PyFile_WriteString(modstr, f);
1143 err += PyFile_WriteString(".", f);
1144 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001145 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001146 }
1147 if (err == 0) {
1148 if (className == NULL)
1149 err = PyFile_WriteString("<unknown>", f);
1150 else
Brett Cannonbf364092006-03-01 04:25:17 +00001151 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001152 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001153 }
1154 else
1155 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001156 if (err == 0 && (value != Py_None)) {
1157 PyObject *s = PyObject_Str(value);
1158 /* only print colon if the str() of the
1159 object is not the empty string
1160 */
1161 if (s == NULL)
1162 err = -1;
1163 else if (!PyString_Check(s) ||
1164 PyString_GET_SIZE(s) != 0)
1165 err = PyFile_WriteString(": ", f);
1166 if (err == 0)
1167 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1168 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001169 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001170 if (err == 0)
1171 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001172 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001173 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001174 /* If an error happened here, don't show it.
1175 XXX This is wrong, but too many callers rely on this behavior. */
1176 if (err != 0)
1177 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001178}
1179
Guido van Rossum82598051997-03-05 00:20:32 +00001180PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001181PyRun_StringFlags(const char *str, int start, PyObject *globals,
1182 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001183{
Neal Norwitze92fba02006-03-04 18:52:26 +00001184 PyObject *ret = NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001185 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001186 mod_ty mod = PyParser_ASTFromString(str, "<string>", start, flags,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001187 arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001188 if (mod != NULL)
1189 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001190 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001191 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001192}
1193
1194PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001195PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001196 PyObject *locals, int closeit, PyCompilerFlags *flags)
1197{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001198 PyObject *ret;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001199 PyArena *arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001200 mod_ty mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001201 flags, NULL, arena);
1202 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001203 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001204 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001205 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001206 if (closeit)
1207 fclose(fp);
Neal Norwitze92fba02006-03-04 18:52:26 +00001208 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001209 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001210 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001211}
1212
Guido van Rossum82598051997-03-05 00:20:32 +00001213static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001214run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001215 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001216{
Guido van Rossum82598051997-03-05 00:20:32 +00001217 PyCodeObject *co;
1218 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001219 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001220 if (co == NULL)
1221 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001222 v = PyEval_EvalCode(co, globals, locals);
1223 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001224 return v;
1225}
1226
Guido van Rossum82598051997-03-05 00:20:32 +00001227static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001228run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
1229 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001230{
Guido van Rossum82598051997-03-05 00:20:32 +00001231 PyCodeObject *co;
1232 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001233 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001234 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001235
Guido van Rossum82598051997-03-05 00:20:32 +00001236 magic = PyMarshal_ReadLongFromFile(fp);
1237 if (magic != PyImport_GetMagicNumber()) {
1238 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001239 "Bad magic number in .pyc file");
1240 return NULL;
1241 }
Guido van Rossum82598051997-03-05 00:20:32 +00001242 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001243 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001244 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001245 if (v == NULL || !PyCode_Check(v)) {
1246 Py_XDECREF(v);
1247 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001248 "Bad code object in .pyc file");
1249 return NULL;
1250 }
Guido van Rossum82598051997-03-05 00:20:32 +00001251 co = (PyCodeObject *)v;
1252 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001253 if (v && flags)
1254 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001255 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001256 return v;
1257}
1258
Guido van Rossum82598051997-03-05 00:20:32 +00001259PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001260Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001261 PyCompilerFlags *flags)
1262{
Guido van Rossum82598051997-03-05 00:20:32 +00001263 PyCodeObject *co;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001264 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001265 mod_ty mod = PyParser_ASTFromString(str, filename, start, flags, arena);
1266 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001267 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001268 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001269 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001270 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001271 PyObject *result = PyAST_mod2obj(mod);
1272 PyArena_Free(arena);
1273 return result;
1274 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001275 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001276 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001277 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001278}
1279
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001280struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001281Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001282{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001283 struct symtable *st;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001284 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001285 mod_ty mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
1286 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001287 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001288 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001289 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001290 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001291 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001292 return st;
1293}
1294
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001295/* Preferred access to parser is through AST. */
1296mod_ty
1297PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001298 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001299{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001300 mod_ty mod;
1301 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001302 node *n = PyParser_ParseStringFlagsFilename(s, filename,
1303 &_PyParser_Grammar, start, &err,
1304 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001305 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001306 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001307 PyNode_Free(n);
1308 return mod;
1309 }
1310 else {
1311 err_input(&err);
1312 return NULL;
1313 }
1314}
1315
1316mod_ty
1317PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001318 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001319 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001320{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001321 mod_ty mod;
1322 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001323 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1324 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001325 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001326 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001327 PyNode_Free(n);
1328 return mod;
1329 }
1330 else {
1331 err_input(&err);
1332 if (errcode)
1333 *errcode = err.error;
1334 return NULL;
1335 }
1336}
1337
Guido van Rossuma110aa61994-08-29 12:50:44 +00001338/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001339
Guido van Rossuma110aa61994-08-29 12:50:44 +00001340node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001341PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001342{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001343 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001344 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1345 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001346 if (n == NULL)
1347 err_input(&err);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001348
Guido van Rossuma110aa61994-08-29 12:50:44 +00001349 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001350}
1351
Guido van Rossuma110aa61994-08-29 12:50:44 +00001352/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001353
Guido van Rossuma110aa61994-08-29 12:50:44 +00001354node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001355PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001356{
Tim Petersfe2127d2001-07-16 05:37:24 +00001357 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001358 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1359 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001360 if (n == NULL)
1361 err_input(&err);
1362 return n;
1363}
1364
1365node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001366PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001367 int start, int flags)
1368{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001369 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001370 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1371 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001372 if (n == NULL)
1373 err_input(&err);
1374 return n;
1375}
1376
1377node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001378PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001379{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001380 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001381}
1382
Guido van Rossum66ebd912003-04-17 16:02:26 +00001383/* May want to move a more generalized form of this to parsetok.c or
1384 even parser modules. */
1385
1386void
1387PyParser_SetError(perrdetail *err)
1388{
1389 err_input(err);
1390}
1391
Guido van Rossuma110aa61994-08-29 12:50:44 +00001392/* Set the error appropriate to the given input error code (see errcode.h) */
1393
1394static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001395err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001396{
Fred Drake85f36392000-07-11 17:53:00 +00001397 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001398 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001399 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001400 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001401 switch (err->error) {
1402 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001403 errtype = PyExc_IndentationError;
1404 if (err->expected == INDENT)
1405 msg = "expected an indented block";
1406 else if (err->token == INDENT)
1407 msg = "unexpected indent";
1408 else if (err->token == DEDENT)
1409 msg = "unexpected unindent";
1410 else {
1411 errtype = PyExc_SyntaxError;
1412 msg = "invalid syntax";
1413 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001414 break;
1415 case E_TOKEN:
1416 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001417 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001418 case E_EOFS:
1419 msg = "EOF while scanning triple-quoted string";
1420 break;
1421 case E_EOLS:
1422 msg = "EOL while scanning single-quoted string";
1423 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001424 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001425 if (!PyErr_Occurred())
1426 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001427 return;
1428 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001429 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001430 return;
1431 case E_EOF:
1432 msg = "unexpected EOF while parsing";
1433 break;
Fred Drake85f36392000-07-11 17:53:00 +00001434 case E_TABSPACE:
1435 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001436 msg = "inconsistent use of tabs and spaces in indentation";
1437 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001438 case E_OVERFLOW:
1439 msg = "expression too long";
1440 break;
Fred Drake85f36392000-07-11 17:53:00 +00001441 case E_DEDENT:
1442 errtype = PyExc_IndentationError;
1443 msg = "unindent does not match any outer indentation level";
1444 break;
1445 case E_TOODEEP:
1446 errtype = PyExc_IndentationError;
1447 msg = "too many levels of indentation";
1448 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001449 case E_DECODE: {
1450 PyObject *type, *value, *tb;
1451 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001452 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001453 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001454 if (u != NULL) {
1455 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001456 }
1457 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001458 if (msg == NULL)
1459 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001460 Py_XDECREF(type);
1461 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001462 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001463 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001464 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001465 case E_LINECONT:
1466 msg = "unexpected character after line continuation character";
1467 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001468 default:
1469 fprintf(stderr, "error=%d\n", err->error);
1470 msg = "unknown parsing error";
1471 break;
1472 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001473 v = Py_BuildValue("(ziiz)", err->filename,
1474 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001475 if (err->text != NULL) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001476 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001477 err->text = NULL;
1478 }
1479 w = NULL;
1480 if (v != NULL)
1481 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001482 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001483 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001484 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001485 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001486}
1487
1488/* Print fatal error message and abort */
1489
1490void
Tim Peters7c321a82002-07-09 02:57:01 +00001491Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001492{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001493 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001494#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001495 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001496 OutputDebugString(msg);
1497 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001498#ifdef _DEBUG
1499 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001500#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001501#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001502 abort();
1503}
1504
1505/* Clean up and exit */
1506
Guido van Rossuma110aa61994-08-29 12:50:44 +00001507#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001508#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001509#endif
1510
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001511#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001512static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001513static int nexitfuncs = 0;
1514
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001515int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001516{
1517 if (nexitfuncs >= NEXITFUNCS)
1518 return -1;
1519 exitfuncs[nexitfuncs++] = func;
1520 return 0;
1521}
1522
Guido van Rossumcc283f51997-08-05 02:22:03 +00001523static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001524call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001525{
Guido van Rossum82598051997-03-05 00:20:32 +00001526 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001527
1528 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001529 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001530 Py_INCREF(exitfunc);
1531 PySys_SetObject("exitfunc", (PyObject *)NULL);
1532 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001533 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001534 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1535 PySys_WriteStderr("Error in sys.exitfunc:\n");
1536 }
Guido van Rossum82598051997-03-05 00:20:32 +00001537 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001538 }
Guido van Rossum82598051997-03-05 00:20:32 +00001539 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001540 }
1541
Guido van Rossum0829c751998-02-28 04:31:39 +00001542 if (Py_FlushLine())
1543 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001544}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001545
Guido van Rossumcc283f51997-08-05 02:22:03 +00001546static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001547call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001548{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001549 while (nexitfuncs > 0)
1550 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001551
1552 fflush(stdout);
1553 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001554}
1555
1556void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001557Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001558{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001559 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001560
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001561 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001562}
1563
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001564static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001565initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001566{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001567#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001568 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001569#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001570#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001571 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001572#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001573#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001574 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001575#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001576 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001577}
1578
Guido van Rossum7433b121997-02-14 19:45:36 +00001579
1580/*
1581 * The file descriptor fd is considered ``interactive'' if either
1582 * a) isatty(fd) is TRUE, or
1583 * b) the -i flag was given, and the filename associated with
1584 * the descriptor is NULL or "<stdin>" or "???".
1585 */
1586int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001587Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001588{
1589 if (isatty((int)fileno(fp)))
1590 return 1;
1591 if (!Py_InteractiveFlag)
1592 return 0;
1593 return (filename == NULL) ||
1594 (strcmp(filename, "<stdin>") == 0) ||
1595 (strcmp(filename, "???") == 0);
1596}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001597
1598
Tim Petersd08e3822003-04-17 15:24:21 +00001599#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001600#if defined(WIN32) && defined(_MSC_VER)
1601
1602/* Stack checking for Microsoft C */
1603
1604#include <malloc.h>
1605#include <excpt.h>
1606
Fred Drakee8de31c2000-08-31 05:38:39 +00001607/*
1608 * Return non-zero when we run out of memory on the stack; zero otherwise.
1609 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001610int
Fred Drake399739f2000-08-31 05:52:44 +00001611PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001612{
1613 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001614 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001615 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001616 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001617 return 0;
1618 } __except (EXCEPTION_EXECUTE_HANDLER) {
1619 /* just ignore all errors */
1620 }
1621 return 1;
1622}
1623
1624#endif /* WIN32 && _MSC_VER */
1625
1626/* Alternate implementations can be added here... */
1627
1628#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001629
1630
1631/* Wrappers around sigaction() or signal(). */
1632
1633PyOS_sighandler_t
1634PyOS_getsig(int sig)
1635{
1636#ifdef HAVE_SIGACTION
1637 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001638 if (sigaction(sig, NULL, &context) == -1)
1639 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001640 return context.sa_handler;
1641#else
1642 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001643/* Special signal handling for the secure CRT in Visual Studio 2005 */
1644#if defined(_MSC_VER) && _MSC_VER >= 1400
1645 switch (sig) {
1646 /* Only these signals are valid */
1647 case SIGINT:
1648 case SIGILL:
1649 case SIGFPE:
1650 case SIGSEGV:
1651 case SIGTERM:
1652 case SIGBREAK:
1653 case SIGABRT:
1654 break;
1655 /* Don't call signal() with other values or it will assert */
1656 default:
1657 return SIG_ERR;
1658 }
1659#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001660 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001661 if (handler != SIG_ERR)
1662 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001663 return handler;
1664#endif
1665}
1666
1667PyOS_sighandler_t
1668PyOS_setsig(int sig, PyOS_sighandler_t handler)
1669{
1670#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001671 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001672 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001673 sigemptyset(&context.sa_mask);
1674 context.sa_flags = 0;
1675 if (sigaction(sig, &context, &ocontext) == -1)
1676 return SIG_ERR;
1677 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001678#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001679 PyOS_sighandler_t oldhandler;
1680 oldhandler = signal(sig, handler);
1681#ifdef HAVE_SIGINTERRUPT
1682 siginterrupt(sig, 1);
1683#endif
1684 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001685#endif
1686}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001687
1688/* Deprecated C API functions still provided for binary compatiblity */
1689
1690#undef PyParser_SimpleParseFile
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001691PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001692PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1693{
1694 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1695}
1696
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001697#undef PyParser_SimpleParseString
1698PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001699PyParser_SimpleParseString(const char *str, int start)
1700{
1701 return PyParser_SimpleParseStringFlags(str, start, 0);
1702}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001703
1704#undef PyRun_AnyFile
1705PyAPI_FUNC(int)
1706PyRun_AnyFile(FILE *fp, const char *name)
1707{
1708 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1709}
1710
1711#undef PyRun_AnyFileEx
1712PyAPI_FUNC(int)
1713PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1714{
1715 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1716}
1717
1718#undef PyRun_AnyFileFlags
1719PyAPI_FUNC(int)
1720PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1721{
1722 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1723}
1724
1725#undef PyRun_File
1726PyAPI_FUNC(PyObject *)
1727PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1728{
1729 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1730}
1731
1732#undef PyRun_FileEx
1733PyAPI_FUNC(PyObject *)
1734PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1735{
1736 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1737}
1738
1739#undef PyRun_FileFlags
1740PyAPI_FUNC(PyObject *)
1741PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1742 PyCompilerFlags *flags)
1743{
1744 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1745}
1746
1747#undef PyRun_SimpleFile
1748PyAPI_FUNC(int)
1749PyRun_SimpleFile(FILE *f, const char *p)
1750{
1751 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1752}
1753
1754#undef PyRun_SimpleFileEx
1755PyAPI_FUNC(int)
1756PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1757{
1758 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1759}
1760
1761
1762#undef PyRun_String
1763PyAPI_FUNC(PyObject *)
1764PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1765{
1766 return PyRun_StringFlags(str, s, g, l, NULL);
1767}
1768
1769#undef PyRun_SimpleString
1770PyAPI_FUNC(int)
1771PyRun_SimpleString(const char *s)
1772{
1773 return PyRun_SimpleStringFlags(s, NULL);
1774}
1775
1776#undef Py_CompileString
1777PyAPI_FUNC(PyObject *)
1778Py_CompileString(const char *str, const char *p, int s)
1779{
1780 return Py_CompileStringFlags(str, p, s, NULL);
1781}
1782
1783#undef PyRun_InteractiveOne
1784PyAPI_FUNC(int)
1785PyRun_InteractiveOne(FILE *f, const char *p)
1786{
1787 return PyRun_InteractiveOneFlags(f, p, NULL);
1788}
1789
1790#undef PyRun_InteractiveLoop
1791PyAPI_FUNC(int)
1792PyRun_InteractiveLoop(FILE *f, const char *p)
1793{
1794 return PyRun_InteractiveLoopFlags(f, p, NULL);
1795}
1796
1797#ifdef __cplusplus
1798}
1799#endif
1800