blob: b98d6fb991bba5508420bf0970aeb22e9d8e2542 [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
Tim Peters62e97f02006-03-28 21:44:32 +000033#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000034#else /* Py_REF_DEBUG */
Tim Peters62e97f02006-03-28 21:44:32 +000035#define PRINT_TOTAL_REFS() fprintf(stderr, \
36 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
Armin Rigoe1709372006-04-12 17:06:05 +000037 _Py_GetRefTotal())
Neal Norwitz4281cef2006-03-04 19:58:13 +000038#endif
39
Anthony Baxterac6bd462006-04-13 02:06:09 +000040#ifdef __cplusplus
41extern "C" {
42#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 */
Tim Peters3caca232001-12-06 06:23:26 +000075/* _XXX Py_QnewFlag should go away in 2.3. It's true iff -Qnew is passed,
76 on the command line, and is used in 2.2 by ceval.c to make all "/" divisions
77 true divisions (which they will be in 2.3). */
78int _Py_QnewFlag = 0;
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
Neal Norwitzb2501f42002-12-31 03:42:13 +0000186 if (!_PyInt_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000187 Py_FatalError("Py_Initialize: can't init ints");
188
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);
204 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000205
206 sysmod = _PySys_Init();
207 if (sysmod == NULL)
208 Py_FatalError("Py_Initialize: can't initialize sys");
209 interp->sysdict = PyModule_GetDict(sysmod);
210 Py_INCREF(interp->sysdict);
211 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000212 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000213 PyDict_SetItemString(interp->sysdict, "modules",
214 interp->modules);
215
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000216 _PyImport_Init();
217
Barry Warsawf242aa02000-05-25 23:09:49 +0000218 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000219 _PyExc_Init();
Barry Warsaw5821bc52001-08-13 23:04:56 +0000220 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000221
Barry Warsaw035574d1997-08-29 22:07:17 +0000222 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000223 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000224
Just van Rossum52e14d62002-12-30 22:08:05 +0000225 _PyImportHooks_Init();
226
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000227 if (install_sigs)
228 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000229
230 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000231 if (!Py_NoSiteFlag)
232 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000233
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000234 /* auto-thread-state API, if available */
235#ifdef WITH_THREAD
236 _PyGILState_Init(interp, tstate);
237#endif /* WITH_THREAD */
238
Mark Hammondedd07732003-07-15 23:03:55 +0000239 warnings_module = PyImport_ImportModule("warnings");
240 if (!warnings_module)
241 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000242
243#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
244 /* On Unix, set the file system encoding according to the
245 user's preference, if the CODESET names a well-known
246 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000247 initialized by other means. Also set the encoding of
248 stdin and stdout if these are terminals. */
249
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000250 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000251 setlocale(LC_CTYPE, "");
252 codeset = nl_langinfo(CODESET);
253 if (codeset && *codeset) {
254 PyObject *enc = PyCodec_Encoder(codeset);
255 if (enc) {
256 codeset = strdup(codeset);
257 Py_DECREF(enc);
258 } else {
259 codeset = NULL;
260 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000261 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000262 } else
263 codeset = NULL;
264 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000265 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000266
267 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000268 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000269 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
270 if (!sys_isatty)
271 PyErr_Clear();
272 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
273 if (!PyFile_SetEncoding(sys_stream, codeset))
274 Py_FatalError("Cannot set codeset of stdin");
275 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000276 Py_XDECREF(sys_isatty);
277
278 sys_stream = PySys_GetObject("stdout");
279 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
280 if (!sys_isatty)
281 PyErr_Clear();
282 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
283 if (!PyFile_SetEncoding(sys_stream, codeset))
284 Py_FatalError("Cannot set codeset of stdout");
285 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000286 Py_XDECREF(sys_isatty);
287
Martin v. Löwisea62d252006-04-03 10:56:49 +0000288 sys_stream = PySys_GetObject("stderr");
289 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
290 if (!sys_isatty)
291 PyErr_Clear();
292 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
293 if (!PyFile_SetEncoding(sys_stream, codeset))
294 Py_FatalError("Cannot set codeset of stderr");
295 }
296 Py_XDECREF(sys_isatty);
297
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000298 if (!Py_FileSystemDefaultEncoding)
299 Py_FileSystemDefaultEncoding = codeset;
300 else
301 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000302 }
303#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000304}
305
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000306void
307Py_Initialize(void)
308{
309 Py_InitializeEx(1);
310}
311
312
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000313#ifdef COUNT_ALLOCS
Tim Petersdbd9ba62000-07-09 03:09:57 +0000314extern void dump_counts(void);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000315#endif
316
Guido van Rossum25ce5661997-08-02 03:10:38 +0000317/* Undo the effect of Py_Initialize().
318
319 Beware: if multiple interpreter and/or thread states exist, these
320 are not wiped out; only the current thread and interpreter state
321 are deleted. But since everything else is deleted, those other
322 interpreter and thread states should no longer be used.
323
324 (XXX We should do better, e.g. wipe out all interpreters and
325 threads.)
326
327 Locking: as above.
328
329*/
330
331void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000332Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000333{
334 PyInterpreterState *interp;
335 PyThreadState *tstate;
336
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000337 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000338 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000339
Tim Peters384fd102001-01-21 03:40:37 +0000340 /* The interpreter is still entirely intact at this point, and the
341 * exit funcs may be relying on that. In particular, if some thread
342 * or exit func is still waiting to do an import, the import machinery
343 * expects Py_IsInitialized() to return true. So don't say the
344 * interpreter is uninitialized until after the exit funcs have run.
345 * Note that Threading.py uses an exit func to do a join on all the
346 * threads created thru it, so this also protects pending imports in
347 * the threads created via Threading.
348 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000349 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000350 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000351
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000352 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000353 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000354 interp = tstate->interp;
355
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000356 /* Disable signal handling */
357 PyOS_FiniInterrupts();
358
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000359 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000360 Py_XDECREF(warnings_module);
361 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000362
Guido van Rossume13ddc92003-04-17 17:29:22 +0000363 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000364 * before all modules are destroyed.
365 * XXX If a __del__ or weakref callback is triggered here, and tries to
366 * XXX import a module, bad things can happen, because Python no
367 * XXX longer believes it's initialized.
368 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
369 * XXX is easy to provoke that way. I've also seen, e.g.,
370 * XXX Exception exceptions.ImportError: 'No module named sha'
371 * XXX in <function callback at 0x008F5718> ignored
372 * XXX but I'm unclear on exactly how that one happens. In any case,
373 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000374 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000375 PyGC_Collect();
376
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000377 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000378 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000379
Guido van Rossume13ddc92003-04-17 17:29:22 +0000380 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000381 * new-style class definitions, for example.
382 * XXX This is disabled because it caused too many problems. If
383 * XXX a __del__ or weakref callback triggers here, Python code has
384 * XXX a hard time running, because even the sys module has been
385 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
386 * XXX One symptom is a sequence of information-free messages
387 * XXX coming from threads (if a __del__ or callback is invoked,
388 * XXX other threads can execute too, and any exception they encounter
389 * XXX triggers a comedy of errors as subsystem after subsystem
390 * XXX fails to find what it *expects* to find in sys to help report
391 * XXX the exception and consequent unexpected failures). I've also
392 * XXX seen segfaults then, after adding print statements to the
393 * XXX Python code getting called.
394 */
395#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000396 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000397#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000398
Guido van Rossum1707aad1997-12-08 23:43:45 +0000399 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
400 _PyImport_Fini();
401
402 /* Debugging stuff */
403#ifdef COUNT_ALLOCS
404 dump_counts();
405#endif
406
Tim Peters62e97f02006-03-28 21:44:32 +0000407 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000408
Tim Peters9cf25ce2003-04-17 15:21:01 +0000409#ifdef Py_TRACE_REFS
410 /* Display all objects still alive -- this can invoke arbitrary
411 * __repr__ overrides, so requires a mostly-intact interpreter.
412 * Alas, a lot of stuff may still be alive now that will be cleaned
413 * up later.
414 */
Tim Peters269b2a62003-04-17 19:52:29 +0000415 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000416 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000417#endif /* Py_TRACE_REFS */
418
Mark Hammond6cb90292003-04-22 11:18:00 +0000419 /* Cleanup auto-thread-state */
420#ifdef WITH_THREAD
421 _PyGILState_Fini();
422#endif /* WITH_THREAD */
423
Guido van Rossumd922fa42003-04-15 14:10:09 +0000424 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000425 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000426
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000427 /* Now we decref the exception classes. After this point nothing
428 can raise an exception. That's okay, because each Fini() method
429 below has been checked to make sure no exceptions are ever
430 raised.
431 */
432
433 _PyExc_Fini();
434
Guido van Rossumd922fa42003-04-15 14:10:09 +0000435 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000436 PyThreadState_Swap(NULL);
437 PyInterpreterState_Delete(interp);
438
Guido van Rossumd922fa42003-04-15 14:10:09 +0000439 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000440 PyMethod_Fini();
441 PyFrame_Fini();
442 PyCFunction_Fini();
443 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000444 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000445 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000446 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000447 PyInt_Fini();
448 PyFloat_Fini();
449
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000450#ifdef Py_USING_UNICODE
451 /* Cleanup Unicode implementation */
452 _PyUnicode_Fini();
453#endif
454
Guido van Rossumcc283f51997-08-05 02:22:03 +0000455 /* XXX Still allocated:
456 - various static ad-hoc pointers to interned strings
457 - int and float free list blocks
458 - whatever various modules and libraries allocate
459 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000460
461 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000462
Tim Peters269b2a62003-04-17 19:52:29 +0000463#ifdef Py_TRACE_REFS
464 /* Display addresses (& refcnts) of all objects still alive.
465 * An address can be used to find the repr of the object, printed
466 * above by _Py_PrintReferences.
467 */
468 if (Py_GETENV("PYTHONDUMPREFS"))
469 _Py_PrintReferenceAddresses(stderr);
470#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000471#ifdef PYMALLOC_DEBUG
472 if (Py_GETENV("PYTHONMALLOCSTATS"))
473 _PyObject_DebugMallocStats();
474#endif
475
Guido van Rossumcc283f51997-08-05 02:22:03 +0000476 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000477}
478
479/* Create and initialize a new interpreter and thread, and return the
480 new thread. This requires that Py_Initialize() has been called
481 first.
482
483 Unsuccessful initialization yields a NULL pointer. Note that *no*
484 exception information is available even in this case -- the
485 exception information is held in the thread, and there is no
486 thread.
487
488 Locking: as above.
489
490*/
491
492PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000493Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000494{
495 PyInterpreterState *interp;
496 PyThreadState *tstate, *save_tstate;
497 PyObject *bimod, *sysmod;
498
499 if (!initialized)
500 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
501
502 interp = PyInterpreterState_New();
503 if (interp == NULL)
504 return NULL;
505
506 tstate = PyThreadState_New(interp);
507 if (tstate == NULL) {
508 PyInterpreterState_Delete(interp);
509 return NULL;
510 }
511
512 save_tstate = PyThreadState_Swap(tstate);
513
514 /* XXX The following is lax in error checking */
515
516 interp->modules = PyDict_New();
517
518 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
519 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000520 interp->builtins = PyModule_GetDict(bimod);
521 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000522 }
523 sysmod = _PyImport_FindExtension("sys", "sys");
524 if (bimod != NULL && sysmod != NULL) {
525 interp->sysdict = PyModule_GetDict(sysmod);
526 Py_INCREF(interp->sysdict);
527 PySys_SetPath(Py_GetPath());
528 PyDict_SetItemString(interp->sysdict, "modules",
529 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000530 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000531 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000532 if (!Py_NoSiteFlag)
533 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000534 }
535
536 if (!PyErr_Occurred())
537 return tstate;
538
539 /* Oops, it didn't work. Undo it all. */
540
541 PyErr_Print();
542 PyThreadState_Clear(tstate);
543 PyThreadState_Swap(save_tstate);
544 PyThreadState_Delete(tstate);
545 PyInterpreterState_Delete(interp);
546
547 return NULL;
548}
549
550/* Delete an interpreter and its last thread. This requires that the
551 given thread state is current, that the thread has no remaining
552 frames, and that it is its interpreter's only remaining thread.
553 It is a fatal error to violate these constraints.
554
555 (Py_Finalize() doesn't have these constraints -- it zaps
556 everything, regardless.)
557
558 Locking: as above.
559
560*/
561
562void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000563Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000564{
565 PyInterpreterState *interp = tstate->interp;
566
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000567 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000568 Py_FatalError("Py_EndInterpreter: thread is not current");
569 if (tstate->frame != NULL)
570 Py_FatalError("Py_EndInterpreter: thread still has a frame");
571 if (tstate != interp->tstate_head || tstate->next != NULL)
572 Py_FatalError("Py_EndInterpreter: not the last thread");
573
574 PyImport_Cleanup();
575 PyInterpreterState_Clear(interp);
576 PyThreadState_Swap(NULL);
577 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000578}
579
580static char *progname = "python";
581
582void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000583Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000584{
585 if (pn && *pn)
586 progname = pn;
587}
588
589char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000590Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000591{
592 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000593}
594
Guido van Rossuma61691e1998-02-06 22:27:24 +0000595static char *default_home = NULL;
596
597void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000598Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000599{
600 default_home = home;
601}
602
603char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000604Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000605{
606 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000607 if (home == NULL && !Py_IgnoreEnvironmentFlag)
608 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000609 return home;
610}
611
Guido van Rossum6135a871995-01-09 17:53:26 +0000612/* Create __main__ module */
613
614static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000615initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000616{
Guido van Rossum82598051997-03-05 00:20:32 +0000617 PyObject *m, *d;
618 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000619 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000620 Py_FatalError("can't create __main__ module");
621 d = PyModule_GetDict(m);
622 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000623 PyObject *bimod = PyImport_ImportModule("__builtin__");
624 if (bimod == NULL ||
625 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000626 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000627 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000628 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000629}
630
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000631/* Import the site module (not into __main__ though) */
632
633static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000634initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000635{
636 PyObject *m, *f;
637 m = PyImport_ImportModule("site");
638 if (m == NULL) {
639 f = PySys_GetObject("stderr");
640 if (Py_VerboseFlag) {
641 PyFile_WriteString(
642 "'import site' failed; traceback:\n", f);
643 PyErr_Print();
644 }
645 else {
646 PyFile_WriteString(
647 "'import site' failed; use -v for traceback\n", f);
648 PyErr_Clear();
649 }
650 }
651 else {
652 Py_DECREF(m);
653 }
654}
655
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000656/* Parse input from a file and execute it */
657
658int
Martin v. Löwis056a69c2006-03-01 16:55:42 +0000659PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000660 PyCompilerFlags *flags)
661{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000662 if (filename == NULL)
663 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000664 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000665 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000666 if (closeit)
667 fclose(fp);
668 return err;
669 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000670 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000671 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000672}
673
674int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000675PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000676{
Guido van Rossum82598051997-03-05 00:20:32 +0000677 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000678 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000679 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000680
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000681 if (flags == NULL) {
682 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000683 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000684 }
Guido van Rossum82598051997-03-05 00:20:32 +0000685 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000686 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000687 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
688 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000689 }
Guido van Rossum82598051997-03-05 00:20:32 +0000690 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000691 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000692 PySys_SetObject("ps2", v = PyString_FromString("... "));
693 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000694 }
695 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000696 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Tim Peters62e97f02006-03-28 21:44:32 +0000697 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000698 if (ret == E_EOF)
699 return 0;
700 /*
701 if (ret == E_NOMEM)
702 return -1;
703 */
704 }
705}
706
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000707/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000708#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000709 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
710 PyPARSE_DONT_IMPLY_DEDENT : 0) \
Neal Norwitz9589ee22006-03-04 19:01:22 +0000711 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
712 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000713
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000714int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000715PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000716{
Guido van Rossum82598051997-03-05 00:20:32 +0000717 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000718 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000719 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000720 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000721 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000722
Guido van Rossum82598051997-03-05 00:20:32 +0000723 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000724 if (v != NULL) {
725 v = PyObject_Str(v);
726 if (v == NULL)
727 PyErr_Clear();
728 else if (PyString_Check(v))
729 ps1 = PyString_AsString(v);
730 }
Guido van Rossum82598051997-03-05 00:20:32 +0000731 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000732 if (w != NULL) {
733 w = PyObject_Str(w);
734 if (w == NULL)
735 PyErr_Clear();
736 else if (PyString_Check(w))
737 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000738 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000739 arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000740 mod = PyParser_ASTFromFile(fp, filename,
741 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000742 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000743 Py_XDECREF(v);
744 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000745 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000746 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000747 if (errcode == E_EOF) {
748 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000749 return E_EOF;
750 }
Guido van Rossum82598051997-03-05 00:20:32 +0000751 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000752 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000753 }
Guido van Rossum82598051997-03-05 00:20:32 +0000754 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000755 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000756 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000757 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000758 }
Guido van Rossum82598051997-03-05 00:20:32 +0000759 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000760 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000761 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000762 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000763 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000764 return -1;
765 }
Guido van Rossum82598051997-03-05 00:20:32 +0000766 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000767 if (Py_FlushLine())
768 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000769 return 0;
770}
771
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000772/* Check whether a file maybe a pyc file: Look at the extension,
773 the file type, and, if we may close it, at the first few bytes. */
774
775static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000776maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000777{
778 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
779 return 1;
780
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000781 /* Only look into the file if we are allowed to close it, since
782 it then should also be seekable. */
783 if (closeit) {
784 /* Read only two bytes of the magic. If the file was opened in
785 text mode, the bytes 3 and 4 of the magic (\r\n) might not
786 be read as they are on disk. */
787 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
788 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000789 /* Mess: In case of -x, the stream is NOT at its start now,
790 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000791 which makes the current stream position formally undefined,
792 and a x-platform nightmare.
793 Unfortunately, we have no direct way to know whether -x
794 was specified. So we use a terrible hack: if the current
795 stream position is not 0, we assume -x was specified, and
796 give up. Bug 132850 on SourceForge spells out the
797 hopelessness of trying anything else (fseek and ftell
798 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000799 */
Tim Peters3e876562001-02-11 04:35:39 +0000800 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000801 if (ftell(fp) == 0) {
802 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000803 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000804 ispyc = 1;
805 rewind(fp);
806 }
Tim Peters3e876562001-02-11 04:35:39 +0000807 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000808 }
809 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000810}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000811
Guido van Rossum0df002c2000-08-27 19:21:52 +0000812int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000813PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000814 PyCompilerFlags *flags)
815{
Guido van Rossum82598051997-03-05 00:20:32 +0000816 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000817 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000818
Guido van Rossum82598051997-03-05 00:20:32 +0000819 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000820 if (m == NULL)
821 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000822 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000823 if (PyDict_GetItemString(d, "__file__") == NULL) {
824 PyObject *f = PyString_FromString(filename);
825 if (f == NULL)
826 return -1;
827 if (PyDict_SetItemString(d, "__file__", f) < 0) {
828 Py_DECREF(f);
829 return -1;
830 }
831 Py_DECREF(f);
832 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000833 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000834 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000835 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000836 if (closeit)
837 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000838 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000839 fprintf(stderr, "python: Can't reopen .pyc file\n");
840 return -1;
841 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000842 /* Turn on optimization if a .pyo file is given */
843 if (strcmp(ext, ".pyo") == 0)
844 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000845 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000846 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000847 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000848 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000849 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000850 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000851 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000852 return -1;
853 }
Guido van Rossum82598051997-03-05 00:20:32 +0000854 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000855 if (Py_FlushLine())
856 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000857 return 0;
858}
859
860int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000861PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000862{
Guido van Rossum82598051997-03-05 00:20:32 +0000863 PyObject *m, *d, *v;
864 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000865 if (m == NULL)
866 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000867 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000868 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000869 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000870 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000871 return -1;
872 }
Guido van Rossum82598051997-03-05 00:20:32 +0000873 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000874 if (Py_FlushLine())
875 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000876 return 0;
877}
878
Barry Warsaw035574d1997-08-29 22:07:17 +0000879static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000880parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
881 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000882{
883 long hold;
884 PyObject *v;
885
886 /* old style errors */
887 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000888 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000889 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000890
891 /* new style errors. `err' is an instance */
892
893 if (! (v = PyObject_GetAttrString(err, "msg")))
894 goto finally;
895 *message = v;
896
897 if (!(v = PyObject_GetAttrString(err, "filename")))
898 goto finally;
899 if (v == Py_None)
900 *filename = NULL;
901 else if (! (*filename = PyString_AsString(v)))
902 goto finally;
903
904 Py_DECREF(v);
905 if (!(v = PyObject_GetAttrString(err, "lineno")))
906 goto finally;
907 hold = PyInt_AsLong(v);
908 Py_DECREF(v);
909 v = NULL;
910 if (hold < 0 && PyErr_Occurred())
911 goto finally;
912 *lineno = (int)hold;
913
914 if (!(v = PyObject_GetAttrString(err, "offset")))
915 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000916 if (v == Py_None) {
917 *offset = -1;
918 Py_DECREF(v);
919 v = NULL;
920 } else {
921 hold = PyInt_AsLong(v);
922 Py_DECREF(v);
923 v = NULL;
924 if (hold < 0 && PyErr_Occurred())
925 goto finally;
926 *offset = (int)hold;
927 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000928
929 if (!(v = PyObject_GetAttrString(err, "text")))
930 goto finally;
931 if (v == Py_None)
932 *text = NULL;
933 else if (! (*text = PyString_AsString(v)))
934 goto finally;
935 Py_DECREF(v);
936 return 1;
937
938finally:
939 Py_XDECREF(v);
940 return 0;
941}
942
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000943void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000944PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000945{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000946 PyErr_PrintEx(1);
947}
948
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000949static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000950print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000951{
952 char *nl;
953 if (offset >= 0) {
954 if (offset > 0 && offset == (int)strlen(text))
955 offset--;
956 for (;;) {
957 nl = strchr(text, '\n');
958 if (nl == NULL || nl-text >= offset)
959 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000960 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000961 text = nl+1;
962 }
963 while (*text == ' ' || *text == '\t') {
964 text++;
965 offset--;
966 }
967 }
968 PyFile_WriteString(" ", f);
969 PyFile_WriteString(text, f);
970 if (*text == '\0' || text[strlen(text)-1] != '\n')
971 PyFile_WriteString("\n", f);
972 if (offset == -1)
973 return;
974 PyFile_WriteString(" ", f);
975 offset--;
976 while (offset > 0) {
977 PyFile_WriteString(" ", f);
978 offset--;
979 }
980 PyFile_WriteString("^\n", f);
981}
982
Guido van Rossum66e8e862001-03-23 17:54:43 +0000983static void
984handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000985{
Neal Norwitz9589ee22006-03-04 19:01:22 +0000986 PyObject *exception, *value, *tb;
987 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +0000988
Guido van Rossum66e8e862001-03-23 17:54:43 +0000989 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000990 if (Py_FlushLine())
991 PyErr_Clear();
992 fflush(stdout);
993 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +0000994 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +0000995 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000996 /* The error code should be in the `code' attribute. */
997 PyObject *code = PyObject_GetAttrString(value, "code");
998 if (code) {
999 Py_DECREF(value);
1000 value = code;
1001 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001002 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001003 }
1004 /* If we failed to dig out the 'code' attribute,
1005 just let the else clause below print the error. */
1006 }
1007 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001008 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001009 else {
1010 PyObject_Print(value, stderr, Py_PRINT_RAW);
1011 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001012 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001013 }
Tim Peterscf615b52003-04-19 18:47:02 +00001014 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001015 /* Restore and clear the exception info, in order to properly decref
1016 * the exception, value, and traceback. If we just exit instead,
1017 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1018 * some finalizers from running.
1019 */
Tim Peterscf615b52003-04-19 18:47:02 +00001020 PyErr_Restore(exception, value, tb);
1021 PyErr_Clear();
1022 Py_Exit(exitcode);
1023 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001024}
1025
1026void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001027PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001028{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001029 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001030
1031 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1032 handle_system_exit();
1033 }
Guido van Rossum82598051997-03-05 00:20:32 +00001034 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001035 if (exception == NULL)
1036 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001037 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001038 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001039 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001040 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001041 if (set_sys_last_vars) {
1042 PySys_SetObject("last_type", exception);
1043 PySys_SetObject("last_value", v);
1044 PySys_SetObject("last_traceback", tb);
1045 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001046 hook = PySys_GetObject("excepthook");
1047 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001048 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001049 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001050 PyObject *result = PyEval_CallObject(hook, args);
1051 if (result == NULL) {
1052 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001053 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1054 handle_system_exit();
1055 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001056 PyErr_Fetch(&exception2, &v2, &tb2);
1057 PyErr_NormalizeException(&exception2, &v2, &tb2);
1058 if (Py_FlushLine())
1059 PyErr_Clear();
1060 fflush(stdout);
1061 PySys_WriteStderr("Error in sys.excepthook:\n");
1062 PyErr_Display(exception2, v2, tb2);
1063 PySys_WriteStderr("\nOriginal exception was:\n");
1064 PyErr_Display(exception, v, tb);
Jeremy Hylton07028582001-12-07 15:35:35 +00001065 Py_XDECREF(exception2);
1066 Py_XDECREF(v2);
1067 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001068 }
1069 Py_XDECREF(result);
1070 Py_XDECREF(args);
1071 } else {
1072 PySys_WriteStderr("sys.excepthook is missing\n");
1073 PyErr_Display(exception, v, tb);
1074 }
1075 Py_XDECREF(exception);
1076 Py_XDECREF(v);
1077 Py_XDECREF(tb);
1078}
1079
1080void PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
1081{
1082 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001083 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001084 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001085 if (f == NULL)
1086 fprintf(stderr, "lost sys.stderr\n");
1087 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001088 if (Py_FlushLine())
1089 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001090 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001091 if (tb && tb != Py_None)
1092 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001093 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001094 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001095 {
Guido van Rossum82598051997-03-05 00:20:32 +00001096 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001097 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001098 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001099 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001100 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001101 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001102 else {
1103 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001104 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001105 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001106 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001107 else
Guido van Rossum82598051997-03-05 00:20:32 +00001108 PyFile_WriteString(filename, f);
1109 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001110 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001111 PyFile_WriteString(buf, f);
1112 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001113 if (text != NULL)
1114 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001115 Py_DECREF(value);
1116 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001117 /* Can't be bothered to check all those
1118 PyFile_WriteString() calls */
1119 if (PyErr_Occurred())
1120 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001121 }
1122 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001123 if (err) {
1124 /* Don't do anything else */
1125 }
Brett Cannonbf364092006-03-01 04:25:17 +00001126 else if (PyExceptionClass_Check(exception)) {
1127 char* className = PyExceptionClass_Name(exception);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001128 PyObject* moduleName =
Brett Cannonbf364092006-03-01 04:25:17 +00001129 PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001130
1131 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001132 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001133 else {
1134 char* modstr = PyString_AsString(moduleName);
Brett Cannon2e63b732006-03-02 18:34:57 +00001135 Py_DECREF(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001136 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001137 {
1138 err = PyFile_WriteString(modstr, f);
1139 err += PyFile_WriteString(".", f);
1140 }
1141 }
1142 if (err == 0) {
1143 if (className == NULL)
1144 err = PyFile_WriteString("<unknown>", f);
1145 else
Brett Cannonbf364092006-03-01 04:25:17 +00001146 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001147 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001148 }
1149 else
1150 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001151 if (err == 0 && (value != Py_None)) {
1152 PyObject *s = PyObject_Str(value);
1153 /* only print colon if the str() of the
1154 object is not the empty string
1155 */
1156 if (s == NULL)
1157 err = -1;
1158 else if (!PyString_Check(s) ||
1159 PyString_GET_SIZE(s) != 0)
1160 err = PyFile_WriteString(": ", f);
1161 if (err == 0)
1162 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1163 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001164 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001165 if (err == 0)
1166 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001167 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001168 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001169 /* If an error happened here, don't show it.
1170 XXX This is wrong, but too many callers rely on this behavior. */
1171 if (err != 0)
1172 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001173}
1174
Guido van Rossum82598051997-03-05 00:20:32 +00001175PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001176PyRun_StringFlags(const char *str, int start, PyObject *globals,
1177 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001178{
Neal Norwitze92fba02006-03-04 18:52:26 +00001179 PyObject *ret = NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001180 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001181 mod_ty mod = PyParser_ASTFromString(str, "<string>", start, flags,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001182 arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001183 if (mod != NULL)
1184 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001185 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001186 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001187}
1188
1189PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001190PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001191 PyObject *locals, int closeit, PyCompilerFlags *flags)
1192{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001193 PyObject *ret;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001194 PyArena *arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001195 mod_ty mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001196 flags, NULL, arena);
1197 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001198 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001199 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001200 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001201 if (closeit)
1202 fclose(fp);
Neal Norwitze92fba02006-03-04 18:52:26 +00001203 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001204 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001205 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001206}
1207
Guido van Rossum82598051997-03-05 00:20:32 +00001208static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001209run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001210 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001211{
Guido van Rossum82598051997-03-05 00:20:32 +00001212 PyCodeObject *co;
1213 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001214 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001215 if (co == NULL)
1216 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001217 v = PyEval_EvalCode(co, globals, locals);
1218 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001219 return v;
1220}
1221
Guido van Rossum82598051997-03-05 00:20:32 +00001222static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001223run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
1224 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001225{
Guido van Rossum82598051997-03-05 00:20:32 +00001226 PyCodeObject *co;
1227 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001228 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001229 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001230
Guido van Rossum82598051997-03-05 00:20:32 +00001231 magic = PyMarshal_ReadLongFromFile(fp);
1232 if (magic != PyImport_GetMagicNumber()) {
1233 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001234 "Bad magic number in .pyc file");
1235 return NULL;
1236 }
Guido van Rossum82598051997-03-05 00:20:32 +00001237 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001238 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001239 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001240 if (v == NULL || !PyCode_Check(v)) {
1241 Py_XDECREF(v);
1242 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001243 "Bad code object in .pyc file");
1244 return NULL;
1245 }
Guido van Rossum82598051997-03-05 00:20:32 +00001246 co = (PyCodeObject *)v;
1247 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001248 if (v && flags)
1249 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001250 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001251 return v;
1252}
1253
Guido van Rossum82598051997-03-05 00:20:32 +00001254PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001255Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001256 PyCompilerFlags *flags)
1257{
Guido van Rossum82598051997-03-05 00:20:32 +00001258 PyCodeObject *co;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001259 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001260 mod_ty mod = PyParser_ASTFromString(str, filename, start, flags, arena);
1261 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001262 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001263 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001264 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001265 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001266 PyObject *result = PyAST_mod2obj(mod);
1267 PyArena_Free(arena);
1268 return result;
1269 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001270 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001271 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001272 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001273}
1274
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001275struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001276Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001277{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001278 struct symtable *st;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001279 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001280 mod_ty mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
1281 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001282 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001283 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001284 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001285 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001286 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001287 return st;
1288}
1289
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001290/* Preferred access to parser is through AST. */
1291mod_ty
1292PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001293 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001294{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001295 mod_ty mod;
1296 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001297 node *n = PyParser_ParseStringFlagsFilename(s, filename,
1298 &_PyParser_Grammar, start, &err,
1299 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001300 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001301 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001302 PyNode_Free(n);
1303 return mod;
1304 }
1305 else {
1306 err_input(&err);
1307 return NULL;
1308 }
1309}
1310
1311mod_ty
1312PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001313 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001314 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001315{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001316 mod_ty mod;
1317 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001318 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1319 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001320 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001321 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001322 PyNode_Free(n);
1323 return mod;
1324 }
1325 else {
1326 err_input(&err);
1327 if (errcode)
1328 *errcode = err.error;
1329 return NULL;
1330 }
1331}
1332
Guido van Rossuma110aa61994-08-29 12:50:44 +00001333/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001334
Guido van Rossuma110aa61994-08-29 12:50:44 +00001335node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001336PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001337{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001338 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001339 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1340 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001341 if (n == NULL)
1342 err_input(&err);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001343
Guido van Rossuma110aa61994-08-29 12:50:44 +00001344 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001345}
1346
Guido van Rossuma110aa61994-08-29 12:50:44 +00001347/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001348
Guido van Rossuma110aa61994-08-29 12:50:44 +00001349node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001350PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001351{
Tim Petersfe2127d2001-07-16 05:37:24 +00001352 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001353 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1354 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001355 if (n == NULL)
1356 err_input(&err);
1357 return n;
1358}
1359
1360node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001361PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001362 int start, int flags)
1363{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001364 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001365 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1366 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001367 if (n == NULL)
1368 err_input(&err);
1369 return n;
1370}
1371
1372node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001373PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001374{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001375 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001376}
1377
Guido van Rossum66ebd912003-04-17 16:02:26 +00001378/* May want to move a more generalized form of this to parsetok.c or
1379 even parser modules. */
1380
1381void
1382PyParser_SetError(perrdetail *err)
1383{
1384 err_input(err);
1385}
1386
Guido van Rossuma110aa61994-08-29 12:50:44 +00001387/* Set the error appropriate to the given input error code (see errcode.h) */
1388
1389static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001390err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001391{
Fred Drake85f36392000-07-11 17:53:00 +00001392 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001393 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001394 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001395 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001396 switch (err->error) {
1397 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001398 errtype = PyExc_IndentationError;
1399 if (err->expected == INDENT)
1400 msg = "expected an indented block";
1401 else if (err->token == INDENT)
1402 msg = "unexpected indent";
1403 else if (err->token == DEDENT)
1404 msg = "unexpected unindent";
1405 else {
1406 errtype = PyExc_SyntaxError;
1407 msg = "invalid syntax";
1408 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001409 break;
1410 case E_TOKEN:
1411 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001412 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001413 case E_EOFS:
1414 msg = "EOF while scanning triple-quoted string";
1415 break;
1416 case E_EOLS:
1417 msg = "EOL while scanning single-quoted string";
1418 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001419 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001420 if (!PyErr_Occurred())
1421 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001422 return;
1423 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001424 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001425 return;
1426 case E_EOF:
1427 msg = "unexpected EOF while parsing";
1428 break;
Fred Drake85f36392000-07-11 17:53:00 +00001429 case E_TABSPACE:
1430 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001431 msg = "inconsistent use of tabs and spaces in indentation";
1432 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001433 case E_OVERFLOW:
1434 msg = "expression too long";
1435 break;
Fred Drake85f36392000-07-11 17:53:00 +00001436 case E_DEDENT:
1437 errtype = PyExc_IndentationError;
1438 msg = "unindent does not match any outer indentation level";
1439 break;
1440 case E_TOODEEP:
1441 errtype = PyExc_IndentationError;
1442 msg = "too many levels of indentation";
1443 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001444 case E_DECODE: {
1445 PyObject *type, *value, *tb;
1446 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001447 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001448 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001449 if (u != NULL) {
1450 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001451 }
1452 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001453 if (msg == NULL)
1454 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001455 Py_XDECREF(type);
1456 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001457 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001458 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001459 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001460 case E_LINECONT:
1461 msg = "unexpected character after line continuation character";
1462 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001463 default:
1464 fprintf(stderr, "error=%d\n", err->error);
1465 msg = "unknown parsing error";
1466 break;
1467 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001468 v = Py_BuildValue("(ziiz)", err->filename,
1469 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001470 if (err->text != NULL) {
Tim Petersc9d78aa2006-03-26 23:27:58 +00001471 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001472 err->text = NULL;
1473 }
1474 w = NULL;
1475 if (v != NULL)
1476 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001477 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001478 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001479 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001480 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001481}
1482
1483/* Print fatal error message and abort */
1484
1485void
Tim Peters7c321a82002-07-09 02:57:01 +00001486Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001487{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001488 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001489#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001490 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001491 OutputDebugString(msg);
1492 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001493#ifdef _DEBUG
1494 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001495#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001496#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001497 abort();
1498}
1499
1500/* Clean up and exit */
1501
Guido van Rossuma110aa61994-08-29 12:50:44 +00001502#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001503#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001504#endif
1505
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001506#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001507static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001508static int nexitfuncs = 0;
1509
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001510int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001511{
1512 if (nexitfuncs >= NEXITFUNCS)
1513 return -1;
1514 exitfuncs[nexitfuncs++] = func;
1515 return 0;
1516}
1517
Guido van Rossumcc283f51997-08-05 02:22:03 +00001518static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001519call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001520{
Guido van Rossum82598051997-03-05 00:20:32 +00001521 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001522
1523 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001524 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001525 Py_INCREF(exitfunc);
1526 PySys_SetObject("exitfunc", (PyObject *)NULL);
1527 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001528 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001529 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1530 PySys_WriteStderr("Error in sys.exitfunc:\n");
1531 }
Guido van Rossum82598051997-03-05 00:20:32 +00001532 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001533 }
Guido van Rossum82598051997-03-05 00:20:32 +00001534 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001535 }
1536
Guido van Rossum0829c751998-02-28 04:31:39 +00001537 if (Py_FlushLine())
1538 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001539}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001540
Guido van Rossumcc283f51997-08-05 02:22:03 +00001541static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001542call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001543{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001544 while (nexitfuncs > 0)
1545 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001546
1547 fflush(stdout);
1548 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001549}
1550
1551void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001552Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001553{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001554 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001555
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001556 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001557}
1558
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001559static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001560initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001561{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001562#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001563 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001564#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001565#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001566 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001567#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001568#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001569 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001570#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001571 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001572}
1573
Guido van Rossum7433b121997-02-14 19:45:36 +00001574
1575/*
1576 * The file descriptor fd is considered ``interactive'' if either
1577 * a) isatty(fd) is TRUE, or
1578 * b) the -i flag was given, and the filename associated with
1579 * the descriptor is NULL or "<stdin>" or "???".
1580 */
1581int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001582Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001583{
1584 if (isatty((int)fileno(fp)))
1585 return 1;
1586 if (!Py_InteractiveFlag)
1587 return 0;
1588 return (filename == NULL) ||
1589 (strcmp(filename, "<stdin>") == 0) ||
1590 (strcmp(filename, "???") == 0);
1591}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001592
1593
Tim Petersd08e3822003-04-17 15:24:21 +00001594#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001595#if defined(WIN32) && defined(_MSC_VER)
1596
1597/* Stack checking for Microsoft C */
1598
1599#include <malloc.h>
1600#include <excpt.h>
1601
Fred Drakee8de31c2000-08-31 05:38:39 +00001602/*
1603 * Return non-zero when we run out of memory on the stack; zero otherwise.
1604 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001605int
Fred Drake399739f2000-08-31 05:52:44 +00001606PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001607{
1608 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001609 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001610 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001611 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001612 return 0;
1613 } __except (EXCEPTION_EXECUTE_HANDLER) {
1614 /* just ignore all errors */
1615 }
1616 return 1;
1617}
1618
1619#endif /* WIN32 && _MSC_VER */
1620
1621/* Alternate implementations can be added here... */
1622
1623#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001624
1625
1626/* Wrappers around sigaction() or signal(). */
1627
1628PyOS_sighandler_t
1629PyOS_getsig(int sig)
1630{
1631#ifdef HAVE_SIGACTION
1632 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001633 if (sigaction(sig, NULL, &context) == -1)
1634 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001635 return context.sa_handler;
1636#else
1637 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001638/* Special signal handling for the secure CRT in Visual Studio 2005 */
1639#if defined(_MSC_VER) && _MSC_VER >= 1400
1640 switch (sig) {
1641 /* Only these signals are valid */
1642 case SIGINT:
1643 case SIGILL:
1644 case SIGFPE:
1645 case SIGSEGV:
1646 case SIGTERM:
1647 case SIGBREAK:
1648 case SIGABRT:
1649 break;
1650 /* Don't call signal() with other values or it will assert */
1651 default:
1652 return SIG_ERR;
1653 }
1654#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001655 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001656 if (handler != SIG_ERR)
1657 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001658 return handler;
1659#endif
1660}
1661
1662PyOS_sighandler_t
1663PyOS_setsig(int sig, PyOS_sighandler_t handler)
1664{
1665#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001666 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001667 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001668 sigemptyset(&context.sa_mask);
1669 context.sa_flags = 0;
1670 if (sigaction(sig, &context, &ocontext) == -1)
1671 return SIG_ERR;
1672 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001673#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001674 PyOS_sighandler_t oldhandler;
1675 oldhandler = signal(sig, handler);
1676#ifdef HAVE_SIGINTERRUPT
1677 siginterrupt(sig, 1);
1678#endif
1679 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001680#endif
1681}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001682
1683/* Deprecated C API functions still provided for binary compatiblity */
1684
1685#undef PyParser_SimpleParseFile
1686#undef PyParser_SimpleParseString
1687
1688node *
1689PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1690{
1691 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1692}
1693
1694node *
1695PyParser_SimpleParseString(const char *str, int start)
1696{
1697 return PyParser_SimpleParseStringFlags(str, start, 0);
1698}
Anthony Baxterac6bd462006-04-13 02:06:09 +00001699
1700#ifdef __cplusplus
1701}
1702#endif
1703