blob: 56b3ba82993733d2beb7dd09417ab032f74cb12f [file] [log] [blame]
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001/***********************************************************
Guido van Rossumfd71b9e2000-06-30 23:50:40 +00002Copyright (c) 2000, BeOpen.com.
3Copyright (c) 1995-2000, Corporation for National Research Initiatives.
4Copyright (c) 1990-1995, Stichting Mathematisch Centrum.
5All rights reserved.
Guido van Rossum1984f1e1992-08-04 12:41:02 +00006
Guido van Rossumfd71b9e2000-06-30 23:50:40 +00007See the file "Misc/COPYRIGHT" for information on usage and
8redistribution of this file, and for a DISCLAIMER OF ALL WARRANTIES.
Guido van Rossum1984f1e1992-08-04 12:41:02 +00009******************************************************************/
10
11/* Python interpreter top-level routines, including init/exit */
12
Guido van Rossum82598051997-03-05 00:20:32 +000013#include "Python.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000014
15#include "grammar.h"
16#include "node.h"
Fred Drake85f36392000-07-11 17:53:00 +000017#include "token.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000018#include "parsetok.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000019#include "errcode.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000020#include "compile.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000021#include "eval.h"
Guido van Rossumfdef2711994-09-14 13:31:04 +000022#include "marshal.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000023
Guido van Rossum80bb9651996-12-05 23:27:02 +000024#ifdef HAVE_UNISTD_H
25#include <unistd.h>
26#endif
27
Guido van Rossuma110aa61994-08-29 12:50:44 +000028#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000029#include <signal.h>
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000030#endif
31
Guido van Rossum9b38a141996-09-11 23:12:24 +000032#ifdef MS_WIN32
Guido van Rossuma44823b1995-03-14 15:01:17 +000033#undef BYTE
34#include "windows.h"
35#endif
36
Jack Jansencbf630f2000-07-11 21:59:16 +000037#ifdef macintosh
38#include "macglue.h"
39#endif
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000040extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000041
Guido van Rossum82598051997-03-05 00:20:32 +000042extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000043
Guido van Rossumb73cc041993-11-01 16:28:59 +000044/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000045static void initmain(void);
46static void initsite(void);
47static PyObject *run_err_node(node *n, char *filename,
48 PyObject *globals, PyObject *locals);
49static PyObject *run_node(node *n, char *filename,
50 PyObject *globals, PyObject *locals);
51static PyObject *run_pyc_file(FILE *fp, char *filename,
52 PyObject *globals, PyObject *locals);
53static void err_input(perrdetail *);
54static void initsigs(void);
55static void call_sys_exitfunc(void);
56static void call_ll_exitfuncs(void);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000057
Guido van Rossumbffd6832000-01-20 22:32:56 +000058#ifdef Py_TRACE_REFS
59int _Py_AskYesNo(char *prompt);
60#endif
61
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000062extern void _PyUnicode_Init(void);
63extern void _PyUnicode_Fini(void);
64extern void _PyCodecRegistry_Init(void);
65extern void _PyCodecRegistry_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000066
67
Guido van Rossum82598051997-03-05 00:20:32 +000068int Py_DebugFlag; /* Needed by parser.c */
69int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000070int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000071int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000072int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000073int Py_FrozenFlag; /* Needed by getpath.c */
Guido van Rossumb16d1972000-05-01 17:55:15 +000074int Py_UnicodeFlag = 0; /* Needed by compile.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000075
Guido van Rossum25ce5661997-08-02 03:10:38 +000076static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +000077
Thomas Wouters7e474022000-07-16 12:04:32 +000078/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +000079
80int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000081Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +000082{
83 return initialized;
84}
85
Guido van Rossum25ce5661997-08-02 03:10:38 +000086/* Global initializations. Can be undone by Py_Finalize(). Don't
87 call this twice without an intervening Py_Finalize() call. When
88 initializations fail, a fatal error is issued and the function does
89 not return. On return, the first thread and interpreter state have
90 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +000091
Guido van Rossum25ce5661997-08-02 03:10:38 +000092 Locking: you must hold the interpreter lock while calling this.
93 (If the lock has not yet been initialized, that's equivalent to
94 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000095
Guido van Rossum25ce5661997-08-02 03:10:38 +000096*/
Guido van Rossuma027efa1997-05-05 20:56:21 +000097
98void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000099Py_Initialize(void)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000100{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000101 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000102 PyThreadState *tstate;
103 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000104 char *p;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000105
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000106 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000107 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000108 initialized = 1;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000109
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000110 if ((p = getenv("PYTHONDEBUG")) && *p != '\0')
Marc-André Lemburgdc3d6062000-08-25 21:00:46 +0000111 Py_DebugFlag = Py_DebugFlag ? Py_DebugFlag : 1;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000112 if ((p = getenv("PYTHONVERBOSE")) && *p != '\0')
Marc-André Lemburgdc3d6062000-08-25 21:00:46 +0000113 Py_VerboseFlag = Py_VerboseFlag ? Py_VerboseFlag : 1;
Guido van Rossum562f5b11998-10-07 14:50:42 +0000114 if ((p = getenv("PYTHONOPTIMIZE")) && *p != '\0')
Marc-André Lemburgdc3d6062000-08-25 21:00:46 +0000115 Py_OptimizeFlag = Py_OptimizeFlag ? Py_OptimizeFlag : 1;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000116
Guido van Rossuma027efa1997-05-05 20:56:21 +0000117 interp = PyInterpreterState_New();
118 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000119 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000120
Guido van Rossuma027efa1997-05-05 20:56:21 +0000121 tstate = PyThreadState_New(interp);
122 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000123 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000124 (void) PyThreadState_Swap(tstate);
125
Guido van Rossum25ce5661997-08-02 03:10:38 +0000126 interp->modules = PyDict_New();
127 if (interp->modules == NULL)
128 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000129
Guido van Rossumc94044c2000-03-10 23:03:54 +0000130 /* Init codec registry */
131 _PyCodecRegistry_Init();
132
133 /* Init Unicode implementation; relies on the codec registry */
134 _PyUnicode_Init();
135
Jeremy Hylton4a3dd2d2000-04-14 19:13:24 +0000136 _PyCompareState_Key = PyString_InternFromString("cmp_state");
137
Barry Warsawf242aa02000-05-25 23:09:49 +0000138 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000139 if (bimod == NULL)
140 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000141 interp->builtins = PyModule_GetDict(bimod);
142 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000143
144 sysmod = _PySys_Init();
145 if (sysmod == NULL)
146 Py_FatalError("Py_Initialize: can't initialize sys");
147 interp->sysdict = PyModule_GetDict(sysmod);
148 Py_INCREF(interp->sysdict);
149 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000150 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000151 PyDict_SetItemString(interp->sysdict, "modules",
152 interp->modules);
153
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000154 _PyImport_Init();
155
Barry Warsawf242aa02000-05-25 23:09:49 +0000156 /* initialize builtin exceptions */
157 init_exceptions();
158
Barry Warsaw035574d1997-08-29 22:07:17 +0000159 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000160 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000161
Guido van Rossum25ce5661997-08-02 03:10:38 +0000162 initsigs(); /* Signal handling stuff, including initintr() */
163
164 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000165 if (!Py_NoSiteFlag)
166 initsite(); /* Module site */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000167}
168
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000169#ifdef COUNT_ALLOCS
Tim Petersdbd9ba62000-07-09 03:09:57 +0000170extern void dump_counts(void);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000171#endif
172
Guido van Rossum25ce5661997-08-02 03:10:38 +0000173/* Undo the effect of Py_Initialize().
174
175 Beware: if multiple interpreter and/or thread states exist, these
176 are not wiped out; only the current thread and interpreter state
177 are deleted. But since everything else is deleted, those other
178 interpreter and thread states should no longer be used.
179
180 (XXX We should do better, e.g. wipe out all interpreters and
181 threads.)
182
183 Locking: as above.
184
185*/
186
187void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000188Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000189{
190 PyInterpreterState *interp;
191 PyThreadState *tstate;
192
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000193 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000194 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000195 initialized = 0;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000196
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000197 call_sys_exitfunc();
198
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000199 /* Get current thread state and interpreter pointer */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000200 tstate = PyThreadState_Get();
201 interp = tstate->interp;
202
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000203 /* Disable signal handling */
204 PyOS_FiniInterrupts();
205
Guido van Rossumc94044c2000-03-10 23:03:54 +0000206 /* Cleanup Unicode implementation */
207 _PyUnicode_Fini();
208
209 /* Cleanup Codec registry */
210 _PyCodecRegistry_Fini();
211
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000212 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000213 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000214
Guido van Rossum1707aad1997-12-08 23:43:45 +0000215 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
216 _PyImport_Fini();
217
218 /* Debugging stuff */
219#ifdef COUNT_ALLOCS
220 dump_counts();
221#endif
222
223#ifdef Py_REF_DEBUG
224 fprintf(stderr, "[%ld refs]\n", _Py_RefTotal);
225#endif
226
227#ifdef Py_TRACE_REFS
Guido van Rossumeca47842000-04-27 23:44:15 +0000228 if (
229#ifdef MS_WINDOWS /* Only ask on Windows if env var set */
230 getenv("PYTHONDUMPREFS") &&
231#endif /* MS_WINDOWS */
232 _Py_AskYesNo("Print left references?")) {
Guido van Rossum1707aad1997-12-08 23:43:45 +0000233 _Py_PrintReferences(stderr);
234 }
235#endif /* Py_TRACE_REFS */
236
Barry Warsaw035574d1997-08-29 22:07:17 +0000237 /* Now we decref the exception classes. After this point nothing
238 can raise an exception. That's okay, because each Fini() method
239 below has been checked to make sure no exceptions are ever
240 raised.
241 */
Barry Warsawf242aa02000-05-25 23:09:49 +0000242 fini_exceptions();
243
244 /* Delete current thread */
245 PyInterpreterState_Clear(interp);
246 PyThreadState_Swap(NULL);
247 PyInterpreterState_Delete(interp);
248
Guido van Rossumcc283f51997-08-05 02:22:03 +0000249 PyMethod_Fini();
250 PyFrame_Fini();
251 PyCFunction_Fini();
252 PyTuple_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000253 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000254 PyInt_Fini();
255 PyFloat_Fini();
256
257 /* XXX Still allocated:
258 - various static ad-hoc pointers to interned strings
259 - int and float free list blocks
260 - whatever various modules and libraries allocate
261 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000262
263 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000264
265 call_ll_exitfuncs();
266
Guido van Rossumcc283f51997-08-05 02:22:03 +0000267#ifdef Py_TRACE_REFS
Guido van Rossumcc283f51997-08-05 02:22:03 +0000268 _Py_ResetReferences();
269#endif /* Py_TRACE_REFS */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000270}
271
272/* Create and initialize a new interpreter and thread, and return the
273 new thread. This requires that Py_Initialize() has been called
274 first.
275
276 Unsuccessful initialization yields a NULL pointer. Note that *no*
277 exception information is available even in this case -- the
278 exception information is held in the thread, and there is no
279 thread.
280
281 Locking: as above.
282
283*/
284
285PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000286Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000287{
288 PyInterpreterState *interp;
289 PyThreadState *tstate, *save_tstate;
290 PyObject *bimod, *sysmod;
291
292 if (!initialized)
293 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
294
295 interp = PyInterpreterState_New();
296 if (interp == NULL)
297 return NULL;
298
299 tstate = PyThreadState_New(interp);
300 if (tstate == NULL) {
301 PyInterpreterState_Delete(interp);
302 return NULL;
303 }
304
305 save_tstate = PyThreadState_Swap(tstate);
306
307 /* XXX The following is lax in error checking */
308
309 interp->modules = PyDict_New();
310
311 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
312 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000313 interp->builtins = PyModule_GetDict(bimod);
314 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000315 }
316 sysmod = _PyImport_FindExtension("sys", "sys");
317 if (bimod != NULL && sysmod != NULL) {
318 interp->sysdict = PyModule_GetDict(sysmod);
319 Py_INCREF(interp->sysdict);
320 PySys_SetPath(Py_GetPath());
321 PyDict_SetItemString(interp->sysdict, "modules",
322 interp->modules);
323 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000324 if (!Py_NoSiteFlag)
325 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000326 }
327
328 if (!PyErr_Occurred())
329 return tstate;
330
331 /* Oops, it didn't work. Undo it all. */
332
333 PyErr_Print();
334 PyThreadState_Clear(tstate);
335 PyThreadState_Swap(save_tstate);
336 PyThreadState_Delete(tstate);
337 PyInterpreterState_Delete(interp);
338
339 return NULL;
340}
341
342/* Delete an interpreter and its last thread. This requires that the
343 given thread state is current, that the thread has no remaining
344 frames, and that it is its interpreter's only remaining thread.
345 It is a fatal error to violate these constraints.
346
347 (Py_Finalize() doesn't have these constraints -- it zaps
348 everything, regardless.)
349
350 Locking: as above.
351
352*/
353
354void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000355Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000356{
357 PyInterpreterState *interp = tstate->interp;
358
359 if (tstate != PyThreadState_Get())
360 Py_FatalError("Py_EndInterpreter: thread is not current");
361 if (tstate->frame != NULL)
362 Py_FatalError("Py_EndInterpreter: thread still has a frame");
363 if (tstate != interp->tstate_head || tstate->next != NULL)
364 Py_FatalError("Py_EndInterpreter: not the last thread");
365
366 PyImport_Cleanup();
367 PyInterpreterState_Clear(interp);
368 PyThreadState_Swap(NULL);
369 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000370}
371
372static char *progname = "python";
373
374void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000375Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000376{
377 if (pn && *pn)
378 progname = pn;
379}
380
381char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000382Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000383{
384 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000385}
386
Guido van Rossuma61691e1998-02-06 22:27:24 +0000387static char *default_home = NULL;
388
389void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000390Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000391{
392 default_home = home;
393}
394
395char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000396Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000397{
398 char *home = default_home;
399 if (home == NULL)
400 home = getenv("PYTHONHOME");
401 return home;
402}
403
Guido van Rossum6135a871995-01-09 17:53:26 +0000404/* Create __main__ module */
405
406static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000407initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000408{
Guido van Rossum82598051997-03-05 00:20:32 +0000409 PyObject *m, *d;
410 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000411 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000412 Py_FatalError("can't create __main__ module");
413 d = PyModule_GetDict(m);
414 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000415 PyObject *bimod = PyImport_ImportModule("__builtin__");
416 if (bimod == NULL ||
417 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000418 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000419 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000420 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000421}
422
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000423/* Import the site module (not into __main__ though) */
424
425static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000426initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000427{
428 PyObject *m, *f;
429 m = PyImport_ImportModule("site");
430 if (m == NULL) {
431 f = PySys_GetObject("stderr");
432 if (Py_VerboseFlag) {
433 PyFile_WriteString(
434 "'import site' failed; traceback:\n", f);
435 PyErr_Print();
436 }
437 else {
438 PyFile_WriteString(
439 "'import site' failed; use -v for traceback\n", f);
440 PyErr_Clear();
441 }
442 }
443 else {
444 Py_DECREF(m);
445 }
446}
447
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000448/* Parse input from a file and execute it */
449
450int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000451PyRun_AnyFile(FILE *fp, char *filename)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000452{
Guido van Rossum0df002c2000-08-27 19:21:52 +0000453 return PyRun_AnyFileEx(fp, filename, 0);
454}
455
456int
457PyRun_AnyFileEx(FILE *fp, char *filename, int closeit)
458{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000459 if (filename == NULL)
460 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000461 if (Py_FdIsInteractive(fp, filename)) {
462 int err = PyRun_InteractiveLoop(fp, filename);
463 if (closeit)
464 fclose(fp);
465 return err;
466 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000467 else
Guido van Rossum0df002c2000-08-27 19:21:52 +0000468 return PyRun_SimpleFileEx(fp, filename, closeit);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000469}
470
471int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000472PyRun_InteractiveLoop(FILE *fp, char *filename)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000473{
Guido van Rossum82598051997-03-05 00:20:32 +0000474 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000475 int ret;
Guido van Rossum82598051997-03-05 00:20:32 +0000476 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000477 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000478 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
479 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000480 }
Guido van Rossum82598051997-03-05 00:20:32 +0000481 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000482 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000483 PySys_SetObject("ps2", v = PyString_FromString("... "));
484 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000485 }
486 for (;;) {
Guido van Rossum82598051997-03-05 00:20:32 +0000487 ret = PyRun_InteractiveOne(fp, filename);
Guido van Rossumaae0d321996-05-22 16:35:33 +0000488#ifdef Py_REF_DEBUG
Guido van Rossum6f9e4331995-03-29 16:57:48 +0000489 fprintf(stderr, "[%ld refs]\n", _Py_RefTotal);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000490#endif
491 if (ret == E_EOF)
492 return 0;
493 /*
494 if (ret == E_NOMEM)
495 return -1;
496 */
497 }
498}
499
500int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000501PyRun_InteractiveOne(FILE *fp, char *filename)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000502{
Guido van Rossum82598051997-03-05 00:20:32 +0000503 PyObject *m, *d, *v, *w;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000504 node *n;
Guido van Rossuma110aa61994-08-29 12:50:44 +0000505 perrdetail err;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000506 char *ps1 = "", *ps2 = "";
Guido van Rossum82598051997-03-05 00:20:32 +0000507 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000508 if (v != NULL) {
509 v = PyObject_Str(v);
510 if (v == NULL)
511 PyErr_Clear();
512 else if (PyString_Check(v))
513 ps1 = PyString_AsString(v);
514 }
Guido van Rossum82598051997-03-05 00:20:32 +0000515 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000516 if (w != NULL) {
517 w = PyObject_Str(w);
518 if (w == NULL)
519 PyErr_Clear();
520 else if (PyString_Check(w))
521 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000522 }
Guido van Rossum82598051997-03-05 00:20:32 +0000523 n = PyParser_ParseFile(fp, filename, &_PyParser_Grammar,
Guido van Rossumb05a5c71997-05-07 17:46:13 +0000524 Py_single_input, ps1, ps2, &err);
Guido van Rossum82598051997-03-05 00:20:32 +0000525 Py_XDECREF(v);
526 Py_XDECREF(w);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000527 if (n == NULL) {
528 if (err.error == E_EOF) {
529 if (err.text)
Guido van Rossumb18618d2000-05-03 23:44:39 +0000530 PyMem_DEL(err.text);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000531 return E_EOF;
532 }
533 err_input(&err);
Guido van Rossum82598051997-03-05 00:20:32 +0000534 PyErr_Print();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000535 return err.error;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000536 }
Guido van Rossum82598051997-03-05 00:20:32 +0000537 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000538 if (m == NULL)
539 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000540 d = PyModule_GetDict(m);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000541 v = run_node(n, filename, d, d);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000542 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000543 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000544 return -1;
545 }
Guido van Rossum82598051997-03-05 00:20:32 +0000546 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000547 if (Py_FlushLine())
548 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000549 return 0;
550}
551
552int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000553PyRun_SimpleFile(FILE *fp, char *filename)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000554{
Guido van Rossum0df002c2000-08-27 19:21:52 +0000555 return PyRun_SimpleFileEx(fp, filename, 0);
556}
557
558int
559PyRun_SimpleFileEx(FILE *fp, char *filename, int closeit)
560{
Guido van Rossum82598051997-03-05 00:20:32 +0000561 PyObject *m, *d, *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000562 char *ext;
563
Guido van Rossum82598051997-03-05 00:20:32 +0000564 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000565 if (m == NULL)
566 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000567 d = PyModule_GetDict(m);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000568 ext = filename + strlen(filename) - 4;
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000569 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0
Jack Jansenbd06e961995-02-13 11:44:56 +0000570#ifdef macintosh
571 /* On a mac, we also assume a pyc file for types 'PYC ' and 'APPL' */
Jack Jansencbf630f2000-07-11 21:59:16 +0000572 || PyMac_getfiletype(filename) == 'PYC '
573 || PyMac_getfiletype(filename) == 'APPL'
Jack Jansenbd06e961995-02-13 11:44:56 +0000574#endif /* macintosh */
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000575 ) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000576 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000577 if (closeit)
578 fclose(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000579 if( (fp = fopen(filename, "rb")) == NULL ) {
580 fprintf(stderr, "python: Can't reopen .pyc file\n");
581 return -1;
582 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000583 /* Turn on optimization if a .pyo file is given */
584 if (strcmp(ext, ".pyo") == 0)
585 Py_OptimizeFlag = 1;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000586 v = run_pyc_file(fp, filename, d, d);
587 } else {
Guido van Rossum0df002c2000-08-27 19:21:52 +0000588 v = PyRun_FileEx(fp, filename, Py_file_input, d, d, closeit);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000589 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000590 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000591 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000592 return -1;
593 }
Guido van Rossum82598051997-03-05 00:20:32 +0000594 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000595 if (Py_FlushLine())
596 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000597 return 0;
598}
599
600int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000601PyRun_SimpleString(char *command)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000602{
Guido van Rossum82598051997-03-05 00:20:32 +0000603 PyObject *m, *d, *v;
604 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000605 if (m == NULL)
606 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000607 d = PyModule_GetDict(m);
Guido van Rossumb05a5c71997-05-07 17:46:13 +0000608 v = PyRun_String(command, Py_file_input, d, d);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000609 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000610 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000611 return -1;
612 }
Guido van Rossum82598051997-03-05 00:20:32 +0000613 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000614 if (Py_FlushLine())
615 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000616 return 0;
617}
618
Barry Warsaw035574d1997-08-29 22:07:17 +0000619static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000620parse_syntax_error(PyObject *err, PyObject **message, char **filename,
621 int *lineno, int *offset, char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000622{
623 long hold;
624 PyObject *v;
625
626 /* old style errors */
627 if (PyTuple_Check(err))
628 return PyArg_Parse(err, "(O(ziiz))", message, filename,
629 lineno, offset, text);
630
631 /* new style errors. `err' is an instance */
632
633 if (! (v = PyObject_GetAttrString(err, "msg")))
634 goto finally;
635 *message = v;
636
637 if (!(v = PyObject_GetAttrString(err, "filename")))
638 goto finally;
639 if (v == Py_None)
640 *filename = NULL;
641 else if (! (*filename = PyString_AsString(v)))
642 goto finally;
643
644 Py_DECREF(v);
645 if (!(v = PyObject_GetAttrString(err, "lineno")))
646 goto finally;
647 hold = PyInt_AsLong(v);
648 Py_DECREF(v);
649 v = NULL;
650 if (hold < 0 && PyErr_Occurred())
651 goto finally;
652 *lineno = (int)hold;
653
654 if (!(v = PyObject_GetAttrString(err, "offset")))
655 goto finally;
656 hold = PyInt_AsLong(v);
657 Py_DECREF(v);
658 v = NULL;
659 if (hold < 0 && PyErr_Occurred())
660 goto finally;
661 *offset = (int)hold;
662
663 if (!(v = PyObject_GetAttrString(err, "text")))
664 goto finally;
665 if (v == Py_None)
666 *text = NULL;
667 else if (! (*text = PyString_AsString(v)))
668 goto finally;
669 Py_DECREF(v);
670 return 1;
671
672finally:
673 Py_XDECREF(v);
674 return 0;
675}
676
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000677void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000678PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000679{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000680 PyErr_PrintEx(1);
681}
682
683void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000684PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000685{
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000686 int err = 0;
Guido van Rossum82598051997-03-05 00:20:32 +0000687 PyObject *exception, *v, *tb, *f;
688 PyErr_Fetch(&exception, &v, &tb);
Barry Warsaw035574d1997-08-29 22:07:17 +0000689 PyErr_NormalizeException(&exception, &v, &tb);
690
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000691 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +0000692 return;
Barry Warsaw035574d1997-08-29 22:07:17 +0000693
Barry Warsaw36b8f941997-08-26 18:09:48 +0000694 if (PyErr_GivenExceptionMatches(exception, PyExc_SystemExit)) {
Guido van Rossum0829c751998-02-28 04:31:39 +0000695 if (Py_FlushLine())
696 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000697 fflush(stdout);
Guido van Rossum82598051997-03-05 00:20:32 +0000698 if (v == NULL || v == Py_None)
699 Py_Exit(0);
Barry Warsaw035574d1997-08-29 22:07:17 +0000700 if (PyInstance_Check(v)) {
701 /* we expect the error code to be store in the
702 `code' attribute
703 */
704 PyObject *code = PyObject_GetAttrString(v, "code");
705 if (code) {
706 Py_DECREF(v);
707 v = code;
Guido van Rossumaa9606f1997-10-03 13:53:28 +0000708 if (v == Py_None)
709 Py_Exit(0);
Barry Warsaw035574d1997-08-29 22:07:17 +0000710 }
711 /* if we failed to dig out the "code" attribute,
712 then just let the else clause below print the
713 error
714 */
715 }
Guido van Rossum82598051997-03-05 00:20:32 +0000716 if (PyInt_Check(v))
717 Py_Exit((int)PyInt_AsLong(v));
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000718 else {
Guido van Rossum3165fe61992-09-25 21:59:05 +0000719 /* OK to use real stderr here */
Guido van Rossum82598051997-03-05 00:20:32 +0000720 PyObject_Print(v, stderr, Py_PRINT_RAW);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000721 fprintf(stderr, "\n");
Guido van Rossum82598051997-03-05 00:20:32 +0000722 Py_Exit(1);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000723 }
724 }
Guido van Rossuma61691e1998-02-06 22:27:24 +0000725 if (set_sys_last_vars) {
726 PySys_SetObject("last_type", exception);
727 PySys_SetObject("last_value", v);
728 PySys_SetObject("last_traceback", tb);
729 }
Guido van Rossum82598051997-03-05 00:20:32 +0000730 f = PySys_GetObject("stderr");
Guido van Rossum3165fe61992-09-25 21:59:05 +0000731 if (f == NULL)
732 fprintf(stderr, "lost sys.stderr\n");
733 else {
Guido van Rossum0829c751998-02-28 04:31:39 +0000734 if (Py_FlushLine())
735 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000736 fflush(stdout);
Guido van Rossum0829c751998-02-28 04:31:39 +0000737 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +0000738 if (err == 0 &&
739 PyErr_GivenExceptionMatches(exception, PyExc_SyntaxError))
740 {
Guido van Rossum82598051997-03-05 00:20:32 +0000741 PyObject *message;
Guido van Rossuma110aa61994-08-29 12:50:44 +0000742 char *filename, *text;
743 int lineno, offset;
Barry Warsaw035574d1997-08-29 22:07:17 +0000744 if (!parse_syntax_error(v, &message, &filename,
745 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +0000746 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000747 else {
748 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +0000749 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000750 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000751 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000752 else
Guido van Rossum82598051997-03-05 00:20:32 +0000753 PyFile_WriteString(filename, f);
754 PyFile_WriteString("\", line ", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000755 sprintf(buf, "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +0000756 PyFile_WriteString(buf, f);
757 PyFile_WriteString("\n", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000758 if (text != NULL) {
Guido van Rossum798199d1994-09-19 08:08:50 +0000759 char *nl;
760 if (offset > 0 &&
Guido van Rossum478e7181997-05-06 15:24:59 +0000761 offset == (int)strlen(text))
Guido van Rossum798199d1994-09-19 08:08:50 +0000762 offset--;
763 for (;;) {
764 nl = strchr(text, '\n');
765 if (nl == NULL ||
766 nl-text >= offset)
767 break;
768 offset -= (nl+1-text);
769 text = nl+1;
770 }
Guido van Rossuma110aa61994-08-29 12:50:44 +0000771 while (*text == ' ' || *text == '\t') {
772 text++;
773 offset--;
774 }
Guido van Rossum82598051997-03-05 00:20:32 +0000775 PyFile_WriteString(" ", f);
776 PyFile_WriteString(text, f);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000777 if (*text == '\0' ||
778 text[strlen(text)-1] != '\n')
Guido van Rossum82598051997-03-05 00:20:32 +0000779 PyFile_WriteString("\n", f);
780 PyFile_WriteString(" ", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000781 offset--;
782 while (offset > 0) {
Guido van Rossum82598051997-03-05 00:20:32 +0000783 PyFile_WriteString(" ", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000784 offset--;
785 }
Guido van Rossum82598051997-03-05 00:20:32 +0000786 PyFile_WriteString("^\n", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000787 }
Guido van Rossum82598051997-03-05 00:20:32 +0000788 Py_INCREF(message);
789 Py_DECREF(v);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000790 v = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000791 /* Can't be bothered to check all those
792 PyFile_WriteString() calls */
793 if (PyErr_Occurred())
794 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +0000795 }
796 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000797 if (err) {
798 /* Don't do anything else */
799 }
800 else if (PyClass_Check(exception)) {
Barry Warsaw2f5f6a21997-09-16 21:42:03 +0000801 PyClassObject* exc = (PyClassObject*)exception;
802 PyObject* className = exc->cl_name;
803 PyObject* moduleName =
804 PyDict_GetItemString(exc->cl_dict, "__module__");
805
806 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000807 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +0000808 else {
809 char* modstr = PyString_AsString(moduleName);
810 if (modstr && strcmp(modstr, "exceptions"))
811 {
812 err = PyFile_WriteString(modstr, f);
813 err += PyFile_WriteString(".", f);
814 }
815 }
816 if (err == 0) {
817 if (className == NULL)
818 err = PyFile_WriteString("<unknown>", f);
819 else
820 err = PyFile_WriteObject(className, f,
821 Py_PRINT_RAW);
822 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000823 }
824 else
825 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
826 if (err == 0) {
827 if (v != NULL && v != Py_None) {
Barry Warsaw035574d1997-08-29 22:07:17 +0000828 PyObject *s = PyObject_Str(v);
829 /* only print colon if the str() of the
830 object is not the empty string
831 */
Guido van Rossumd6bf45b1997-09-05 19:11:53 +0000832 if (s == NULL)
833 err = -1;
834 else if (!PyString_Check(s) ||
835 PyString_GET_SIZE(s) != 0)
Barry Warsaw035574d1997-08-29 22:07:17 +0000836 err = PyFile_WriteString(": ", f);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000837 if (err == 0)
Guido van Rossumd6bf45b1997-09-05 19:11:53 +0000838 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
839 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +0000840 }
Guido van Rossum262e1241995-02-07 15:30:45 +0000841 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000842 if (err == 0)
843 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000844 }
Guido van Rossum82598051997-03-05 00:20:32 +0000845 Py_XDECREF(exception);
846 Py_XDECREF(v);
847 Py_XDECREF(tb);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000848 /* If an error happened here, don't show it.
849 XXX This is wrong, but too many callers rely on this behavior. */
850 if (err != 0)
851 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000852}
853
Guido van Rossum82598051997-03-05 00:20:32 +0000854PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000855PyRun_String(char *str, int start, PyObject *globals, PyObject *locals)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000856{
Guido van Rossum82598051997-03-05 00:20:32 +0000857 return run_err_node(PyParser_SimpleParseString(str, start),
Guido van Rossuma110aa61994-08-29 12:50:44 +0000858 "<string>", globals, locals);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000859}
860
Guido van Rossum82598051997-03-05 00:20:32 +0000861PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000862PyRun_File(FILE *fp, char *filename, int start, PyObject *globals,
863 PyObject *locals)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000864{
Guido van Rossum0df002c2000-08-27 19:21:52 +0000865 PyRun_FileEx(fp, filename, start, globals, locals, 0);
866}
867
868PyObject *
869PyRun_FileEx(FILE *fp, char *filename, int start, PyObject *globals,
870 PyObject *locals, int closeit)
871{
872 node *n = PyParser_SimpleParseFile(fp, filename, start);
873 if (closeit)
874 fclose(fp);
875 return run_err_node(n, filename, globals, locals);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000876}
877
Guido van Rossum82598051997-03-05 00:20:32 +0000878static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000879run_err_node(node *n, char *filename, PyObject *globals, PyObject *locals)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000880{
Guido van Rossuma110aa61994-08-29 12:50:44 +0000881 if (n == NULL)
882 return NULL;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000883 return run_node(n, filename, globals, locals);
884}
885
Guido van Rossum82598051997-03-05 00:20:32 +0000886static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000887run_node(node *n, char *filename, PyObject *globals, PyObject *locals)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000888{
Guido van Rossum82598051997-03-05 00:20:32 +0000889 PyCodeObject *co;
890 PyObject *v;
891 co = PyNode_Compile(n, filename);
892 PyNode_Free(n);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000893 if (co == NULL)
894 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +0000895 v = PyEval_EvalCode(co, globals, locals);
896 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000897 return v;
898}
899
Guido van Rossum82598051997-03-05 00:20:32 +0000900static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000901run_pyc_file(FILE *fp, char *filename, PyObject *globals, PyObject *locals)
Guido van Rossumfdef2711994-09-14 13:31:04 +0000902{
Guido van Rossum82598051997-03-05 00:20:32 +0000903 PyCodeObject *co;
904 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000905 long magic;
Guido van Rossum82598051997-03-05 00:20:32 +0000906 long PyImport_GetMagicNumber();
Guido van Rossumfdef2711994-09-14 13:31:04 +0000907
Guido van Rossum82598051997-03-05 00:20:32 +0000908 magic = PyMarshal_ReadLongFromFile(fp);
909 if (magic != PyImport_GetMagicNumber()) {
910 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +0000911 "Bad magic number in .pyc file");
912 return NULL;
913 }
Guido van Rossum82598051997-03-05 00:20:32 +0000914 (void) PyMarshal_ReadLongFromFile(fp);
915 v = PyMarshal_ReadObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000916 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +0000917 if (v == NULL || !PyCode_Check(v)) {
918 Py_XDECREF(v);
919 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +0000920 "Bad code object in .pyc file");
921 return NULL;
922 }
Guido van Rossum82598051997-03-05 00:20:32 +0000923 co = (PyCodeObject *)v;
924 v = PyEval_EvalCode(co, globals, locals);
925 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000926 return v;
927}
928
Guido van Rossum82598051997-03-05 00:20:32 +0000929PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000930Py_CompileString(char *str, char *filename, int start)
Guido van Rossum5b722181993-03-30 17:46:03 +0000931{
932 node *n;
Guido van Rossum82598051997-03-05 00:20:32 +0000933 PyCodeObject *co;
934 n = PyParser_SimpleParseString(str, start);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000935 if (n == NULL)
Guido van Rossum5b722181993-03-30 17:46:03 +0000936 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +0000937 co = PyNode_Compile(n, filename);
938 PyNode_Free(n);
939 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +0000940}
941
Guido van Rossuma110aa61994-08-29 12:50:44 +0000942/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000943
Guido van Rossuma110aa61994-08-29 12:50:44 +0000944node *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000945PyParser_SimpleParseFile(FILE *fp, char *filename, int start)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000946{
Guido van Rossuma110aa61994-08-29 12:50:44 +0000947 node *n;
948 perrdetail err;
Guido van Rossum82598051997-03-05 00:20:32 +0000949 n = PyParser_ParseFile(fp, filename, &_PyParser_Grammar, start,
Guido van Rossuma110aa61994-08-29 12:50:44 +0000950 (char *)0, (char *)0, &err);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000951 if (n == NULL)
952 err_input(&err);
953 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000954}
955
Guido van Rossuma110aa61994-08-29 12:50:44 +0000956/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000957
Guido van Rossuma110aa61994-08-29 12:50:44 +0000958node *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000959PyParser_SimpleParseString(char *str, int start)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000960{
Guido van Rossuma110aa61994-08-29 12:50:44 +0000961 node *n;
962 perrdetail err;
Guido van Rossum82598051997-03-05 00:20:32 +0000963 n = PyParser_ParseString(str, &_PyParser_Grammar, start, &err);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000964 if (n == NULL)
965 err_input(&err);
966 return n;
967}
968
969/* Set the error appropriate to the given input error code (see errcode.h) */
970
971static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000972err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +0000973{
Fred Drake85f36392000-07-11 17:53:00 +0000974 PyObject *v, *w, *errtype;
Guido van Rossuma110aa61994-08-29 12:50:44 +0000975 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +0000976 errtype = PyExc_SyntaxError;
Guido van Rossum82598051997-03-05 00:20:32 +0000977 v = Py_BuildValue("(ziiz)", err->filename,
Guido van Rossuma110aa61994-08-29 12:50:44 +0000978 err->lineno, err->offset, err->text);
979 if (err->text != NULL) {
Guido van Rossumb18618d2000-05-03 23:44:39 +0000980 PyMem_DEL(err->text);
Guido van Rossuma110aa61994-08-29 12:50:44 +0000981 err->text = NULL;
982 }
983 switch (err->error) {
984 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +0000985 errtype = PyExc_IndentationError;
986 if (err->expected == INDENT)
987 msg = "expected an indented block";
988 else if (err->token == INDENT)
989 msg = "unexpected indent";
990 else if (err->token == DEDENT)
991 msg = "unexpected unindent";
992 else {
993 errtype = PyExc_SyntaxError;
994 msg = "invalid syntax";
995 }
Guido van Rossuma110aa61994-08-29 12:50:44 +0000996 break;
997 case E_TOKEN:
998 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +0000999 break;
1000 case E_INTR:
Guido van Rossum82598051997-03-05 00:20:32 +00001001 PyErr_SetNone(PyExc_KeyboardInterrupt);
Barry Warsawc80baa31999-01-27 16:39:40 +00001002 Py_XDECREF(v);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001003 return;
1004 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001005 PyErr_NoMemory();
Barry Warsawc80baa31999-01-27 16:39:40 +00001006 Py_XDECREF(v);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001007 return;
1008 case E_EOF:
1009 msg = "unexpected EOF while parsing";
1010 break;
Fred Drake85f36392000-07-11 17:53:00 +00001011 case E_TABSPACE:
1012 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001013 msg = "inconsistent use of tabs and spaces in indentation";
1014 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001015 case E_OVERFLOW:
1016 msg = "expression too long";
1017 break;
Fred Drake85f36392000-07-11 17:53:00 +00001018 case E_DEDENT:
1019 errtype = PyExc_IndentationError;
1020 msg = "unindent does not match any outer indentation level";
1021 break;
1022 case E_TOODEEP:
1023 errtype = PyExc_IndentationError;
1024 msg = "too many levels of indentation";
1025 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001026 default:
1027 fprintf(stderr, "error=%d\n", err->error);
1028 msg = "unknown parsing error";
1029 break;
1030 }
Guido van Rossum82598051997-03-05 00:20:32 +00001031 w = Py_BuildValue("(sO)", msg, v);
Fred Drake85f36392000-07-11 17:53:00 +00001032 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001033 Py_XDECREF(w);
Fred Drake83cb7972000-08-15 15:49:03 +00001034
1035 if (v != NULL) {
1036 PyObject *exc, *tb;
1037
1038 PyErr_Fetch(&errtype, &exc, &tb);
1039 PyErr_NormalizeException(&errtype, &exc, &tb);
1040 if (PyObject_SetAttrString(exc, "filename",
1041 PyTuple_GET_ITEM(v, 0)))
1042 PyErr_Clear();
1043 if (PyObject_SetAttrString(exc, "lineno",
1044 PyTuple_GET_ITEM(v, 1)))
1045 PyErr_Clear();
1046 if (PyObject_SetAttrString(exc, "offset",
1047 PyTuple_GET_ITEM(v, 2)))
1048 PyErr_Clear();
1049 Py_DECREF(v);
1050 PyErr_Restore(errtype, exc, tb);
1051 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001052}
1053
1054/* Print fatal error message and abort */
1055
1056void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001057Py_FatalError(char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001058{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001059 fprintf(stderr, "Fatal Python error: %s\n", msg);
Guido van Rossum8ae87c01995-01-26 00:40:38 +00001060#ifdef macintosh
1061 for (;;);
1062#endif
Guido van Rossum9b38a141996-09-11 23:12:24 +00001063#ifdef MS_WIN32
Guido van Rossum23c94461997-05-22 20:21:30 +00001064 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001065 OutputDebugString(msg);
1066 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001067#ifdef _DEBUG
1068 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001069#endif
Guido van Rossum0ba35361998-08-13 13:33:16 +00001070#endif /* MS_WIN32 */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001071 abort();
1072}
1073
1074/* Clean up and exit */
1075
Guido van Rossuma110aa61994-08-29 12:50:44 +00001076#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001077#include "pythread.h"
Guido van Rossum82598051997-03-05 00:20:32 +00001078int _PyThread_Started = 0; /* Set by threadmodule.c and maybe others */
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001079#endif
1080
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001081#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001082static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001083static int nexitfuncs = 0;
1084
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001085int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001086{
1087 if (nexitfuncs >= NEXITFUNCS)
1088 return -1;
1089 exitfuncs[nexitfuncs++] = func;
1090 return 0;
1091}
1092
Guido van Rossumcc283f51997-08-05 02:22:03 +00001093static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001094call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001095{
Guido van Rossum82598051997-03-05 00:20:32 +00001096 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001097
1098 if (exitfunc) {
Guido van Rossumbf02fb21998-04-03 21:12:12 +00001099 PyObject *res, *f;
Guido van Rossum82598051997-03-05 00:20:32 +00001100 Py_INCREF(exitfunc);
1101 PySys_SetObject("exitfunc", (PyObject *)NULL);
Guido van Rossumbf02fb21998-04-03 21:12:12 +00001102 f = PySys_GetObject("stderr");
Guido van Rossum82598051997-03-05 00:20:32 +00001103 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001104 if (res == NULL) {
Guido van Rossumbf02fb21998-04-03 21:12:12 +00001105 if (f)
1106 PyFile_WriteString("Error in sys.exitfunc:\n", f);
Guido van Rossum82598051997-03-05 00:20:32 +00001107 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001108 }
Guido van Rossum82598051997-03-05 00:20:32 +00001109 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001110 }
1111
Guido van Rossum0829c751998-02-28 04:31:39 +00001112 if (Py_FlushLine())
1113 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001114}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001115
Guido van Rossumcc283f51997-08-05 02:22:03 +00001116static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001117call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001118{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001119 while (nexitfuncs > 0)
1120 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001121
1122 fflush(stdout);
1123 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001124}
1125
1126void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001127Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001128{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001129 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001130
Jack Jansen66a89771995-10-27 13:22:14 +00001131#ifdef macintosh
1132 PyMac_Exit(sts);
1133#else
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001134 exit(sts);
Jack Jansen66a89771995-10-27 13:22:14 +00001135#endif
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001136}
1137
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001138static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001139initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001140{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001141#ifdef HAVE_SIGNAL_H
1142#ifdef SIGPIPE
1143 signal(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001144#endif
Guido van Rossuma110aa61994-08-29 12:50:44 +00001145#endif /* HAVE_SIGNAL_H */
Guido van Rossum82598051997-03-05 00:20:32 +00001146 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001147}
1148
Guido van Rossumaae0d321996-05-22 16:35:33 +00001149#ifdef Py_TRACE_REFS
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001150/* Ask a yes/no question */
1151
Guido van Rossum59bff391992-09-03 20:28:00 +00001152int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001153_Py_AskYesNo(char *prompt)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001154{
1155 char buf[256];
1156
1157 printf("%s [ny] ", prompt);
1158 if (fgets(buf, sizeof buf, stdin) == NULL)
1159 return 0;
1160 return buf[0] == 'y' || buf[0] == 'Y';
1161}
1162#endif
1163
Guido van Rossuma110aa61994-08-29 12:50:44 +00001164#ifdef MPW
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001165
1166/* Check for file descriptor connected to interactive device.
1167 Pretend that stdin is always interactive, other files never. */
1168
1169int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001170isatty(int fd)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001171{
1172 return fd == fileno(stdin);
1173}
1174
1175#endif
Guido van Rossum7433b121997-02-14 19:45:36 +00001176
1177/*
1178 * The file descriptor fd is considered ``interactive'' if either
1179 * a) isatty(fd) is TRUE, or
1180 * b) the -i flag was given, and the filename associated with
1181 * the descriptor is NULL or "<stdin>" or "???".
1182 */
1183int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001184Py_FdIsInteractive(FILE *fp, char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001185{
1186 if (isatty((int)fileno(fp)))
1187 return 1;
1188 if (!Py_InteractiveFlag)
1189 return 0;
1190 return (filename == NULL) ||
1191 (strcmp(filename, "<stdin>") == 0) ||
1192 (strcmp(filename, "???") == 0);
1193}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001194
1195
1196#if defined(USE_STACKCHECK)
1197#if defined(WIN32) && defined(_MSC_VER)
1198
1199/* Stack checking for Microsoft C */
1200
1201#include <malloc.h>
1202#include <excpt.h>
1203
1204int
1205PyOS_CheckStack()
1206{
1207 __try {
1208 /* _alloca throws a stack overflow exception if there's
1209 not enough space left on the stack */
1210 _alloca(PYOS_STACK_MARGIN * sizeof(void*));
1211 return 0;
1212 } __except (EXCEPTION_EXECUTE_HANDLER) {
1213 /* just ignore all errors */
1214 }
1215 return 1;
1216}
1217
1218#endif /* WIN32 && _MSC_VER */
1219
1220/* Alternate implementations can be added here... */
1221
1222#endif /* USE_STACKCHECK */