blob: e8f4fa2e5625b7bec1a85d957af1893c63783c5c [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
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000020#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000021#include <signal.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000022#endif
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000023
Martin v. Löwis73d538b2003-03-05 15:13:47 +000024#ifdef HAVE_LANGINFO_H
25#include <locale.h>
26#include <langinfo.h>
27#endif
28
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000029#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000030#undef BYTE
31#include "windows.h"
32#endif
33
Neal Norwitz4281cef2006-03-04 19:58:13 +000034#ifndef Py_REF_DEBUG
Tim Peters62e97f02006-03-28 21:44:32 +000035#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000036#else /* Py_REF_DEBUG */
Tim Peters62e97f02006-03-28 21:44:32 +000037#define PRINT_TOTAL_REFS() fprintf(stderr, \
38 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
Armin Rigoe1709372006-04-12 17:06:05 +000039 _Py_GetRefTotal())
Neal Norwitz4281cef2006-03-04 19:58:13 +000040#endif
41
Anthony Baxterac6bd462006-04-13 02:06:09 +000042#ifdef __cplusplus
43extern "C" {
44#endif
45
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000046extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000047
Guido van Rossum82598051997-03-05 00:20:32 +000048extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000049
Guido van Rossumb73cc041993-11-01 16:28:59 +000050/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000051static void initmain(void);
52static void initsite(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000053static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000054 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000055static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000056 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000057static void err_input(perrdetail *);
58static void initsigs(void);
59static void call_sys_exitfunc(void);
60static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000061extern void _PyUnicode_Init(void);
62extern void _PyUnicode_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000063
Mark Hammond8d98d2c2003-04-19 15:41:53 +000064#ifdef WITH_THREAD
65extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
66extern void _PyGILState_Fini(void);
67#endif /* WITH_THREAD */
68
Guido van Rossum82598051997-03-05 00:20:32 +000069int Py_DebugFlag; /* Needed by parser.c */
70int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000071int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000072int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000073int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000074int Py_FrozenFlag; /* Needed by getpath.c */
Guido van Rossumb16d1972000-05-01 17:55:15 +000075int Py_UnicodeFlag = 0; /* Needed by compile.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000076int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Tim Peters3caca232001-12-06 06:23:26 +000077/* _XXX Py_QnewFlag should go away in 2.3. It's true iff -Qnew is passed,
78 on the command line, and is used in 2.2 by ceval.c to make all "/" divisions
79 true divisions (which they will be in 2.3). */
80int _Py_QnewFlag = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +000081
Mark Hammonda43fd0c2003-02-19 00:33:33 +000082/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000083 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000084*/
Mark Hammondedd07732003-07-15 23:03:55 +000085static PyObject *warnings_module = NULL;
86
87/* Returns a borrowed reference to the 'warnings' module, or NULL.
88 If the module is returned, it is guaranteed to have been obtained
89 without acquiring the import lock
90*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000091PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000092{
93 PyObject *typ, *val, *tb;
94 PyObject *all_modules;
95 /* If we managed to get the module at init time, just use it */
96 if (warnings_module)
97 return warnings_module;
98 /* If it wasn't available at init time, it may be available
99 now in sys.modules (common scenario is frozen apps: import
100 at init time fails, but the frozen init code sets up sys.path
101 correctly, then does an implicit import of warnings for us
102 */
103 /* Save and restore any exceptions */
104 PyErr_Fetch(&typ, &val, &tb);
105
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000106 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000107 if (all_modules) {
108 warnings_module = PyDict_GetItemString(all_modules, "warnings");
109 /* We keep a ref in the global */
110 Py_XINCREF(warnings_module);
111 }
112 PyErr_Restore(typ, val, tb);
113 return warnings_module;
114}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000115
Guido van Rossum25ce5661997-08-02 03:10:38 +0000116static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000117
Thomas Wouters7e474022000-07-16 12:04:32 +0000118/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000119
120int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000121Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000122{
123 return initialized;
124}
125
Guido van Rossum25ce5661997-08-02 03:10:38 +0000126/* Global initializations. Can be undone by Py_Finalize(). Don't
127 call this twice without an intervening Py_Finalize() call. When
128 initializations fail, a fatal error is issued and the function does
129 not return. On return, the first thread and interpreter state have
130 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000131
Guido van Rossum25ce5661997-08-02 03:10:38 +0000132 Locking: you must hold the interpreter lock while calling this.
133 (If the lock has not yet been initialized, that's equivalent to
134 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000135
Guido van Rossum25ce5661997-08-02 03:10:38 +0000136*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000137
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000138static int
139add_flag(int flag, const char *envs)
140{
141 int env = atoi(envs);
142 if (flag < env)
143 flag = env;
144 if (flag < 1)
145 flag = 1;
146 return flag;
147}
148
Guido van Rossuma027efa1997-05-05 20:56:21 +0000149void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000150Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000151{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000152 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000153 PyThreadState *tstate;
154 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000155 char *p;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000156#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
157 char *codeset;
158 char *saved_locale;
159 PyObject *sys_stream, *sys_isatty;
160#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000161 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000162
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000163 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000164 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000165 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000166
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000167 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000168 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000169 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000170 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000171 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000172 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000173
Guido van Rossuma027efa1997-05-05 20:56:21 +0000174 interp = PyInterpreterState_New();
175 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000176 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000177
Guido van Rossuma027efa1997-05-05 20:56:21 +0000178 tstate = PyThreadState_New(interp);
179 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000180 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000181 (void) PyThreadState_Swap(tstate);
182
Guido van Rossum70d893a2001-08-16 08:21:42 +0000183 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000184
Neal Norwitzb2501f42002-12-31 03:42:13 +0000185 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000186 Py_FatalError("Py_Initialize: can't init frames");
187
Neal Norwitzb2501f42002-12-31 03:42:13 +0000188 if (!_PyInt_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000189 Py_FatalError("Py_Initialize: can't init ints");
190
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000191 _PyFloat_Init();
192
Guido van Rossum25ce5661997-08-02 03:10:38 +0000193 interp->modules = PyDict_New();
194 if (interp->modules == NULL)
195 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000196
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000197#ifdef Py_USING_UNICODE
Guido van Rossumc94044c2000-03-10 23:03:54 +0000198 /* Init Unicode implementation; relies on the codec registry */
199 _PyUnicode_Init();
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000200#endif
Guido van Rossumc94044c2000-03-10 23:03:54 +0000201
Barry Warsawf242aa02000-05-25 23:09:49 +0000202 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000203 if (bimod == NULL)
204 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000205 interp->builtins = PyModule_GetDict(bimod);
Neal Norwitz0f7dbf72006-08-12 03:17:41 +0000206 if (interp->builtins == NULL)
207 Py_FatalError("Py_Initialize: can't initialize builtins dict");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000208 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000209
210 sysmod = _PySys_Init();
211 if (sysmod == NULL)
212 Py_FatalError("Py_Initialize: can't initialize sys");
213 interp->sysdict = PyModule_GetDict(sysmod);
Neal Norwitz0f7dbf72006-08-12 03:17:41 +0000214 if (interp->sysdict == NULL)
215 Py_FatalError("Py_Initialize: can't initialize sys dict");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000216 Py_INCREF(interp->sysdict);
217 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000218 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000219 PyDict_SetItemString(interp->sysdict, "modules",
220 interp->modules);
221
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000222 _PyImport_Init();
223
Barry Warsawf242aa02000-05-25 23:09:49 +0000224 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000225 _PyExc_Init();
Barry Warsaw5821bc52001-08-13 23:04:56 +0000226 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000227
Barry Warsaw035574d1997-08-29 22:07:17 +0000228 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000229 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000230
Just van Rossum52e14d62002-12-30 22:08:05 +0000231 _PyImportHooks_Init();
232
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000233 if (install_sigs)
234 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000235
236 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000237 if (!Py_NoSiteFlag)
238 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000239
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000240 /* auto-thread-state API, if available */
241#ifdef WITH_THREAD
242 _PyGILState_Init(interp, tstate);
243#endif /* WITH_THREAD */
244
Mark Hammondedd07732003-07-15 23:03:55 +0000245 warnings_module = PyImport_ImportModule("warnings");
246 if (!warnings_module)
247 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000248
249#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
250 /* On Unix, set the file system encoding according to the
251 user's preference, if the CODESET names a well-known
252 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000253 initialized by other means. Also set the encoding of
254 stdin and stdout if these are terminals. */
255
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000256 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000257 setlocale(LC_CTYPE, "");
258 codeset = nl_langinfo(CODESET);
259 if (codeset && *codeset) {
260 PyObject *enc = PyCodec_Encoder(codeset);
261 if (enc) {
262 codeset = strdup(codeset);
263 Py_DECREF(enc);
264 } else {
265 codeset = NULL;
266 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000267 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000268 } else
269 codeset = NULL;
270 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000271 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000272
273 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000274 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000275 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
276 if (!sys_isatty)
277 PyErr_Clear();
278 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
279 if (!PyFile_SetEncoding(sys_stream, codeset))
280 Py_FatalError("Cannot set codeset of stdin");
281 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000282 Py_XDECREF(sys_isatty);
283
284 sys_stream = PySys_GetObject("stdout");
285 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
286 if (!sys_isatty)
287 PyErr_Clear();
288 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
289 if (!PyFile_SetEncoding(sys_stream, codeset))
290 Py_FatalError("Cannot set codeset of stdout");
291 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000292 Py_XDECREF(sys_isatty);
293
Martin v. Löwisea62d252006-04-03 10:56:49 +0000294 sys_stream = PySys_GetObject("stderr");
295 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
296 if (!sys_isatty)
297 PyErr_Clear();
298 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
299 if (!PyFile_SetEncoding(sys_stream, codeset))
300 Py_FatalError("Cannot set codeset of stderr");
301 }
302 Py_XDECREF(sys_isatty);
303
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000304 if (!Py_FileSystemDefaultEncoding)
305 Py_FileSystemDefaultEncoding = codeset;
306 else
307 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000308 }
309#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000310}
311
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000312void
313Py_Initialize(void)
314{
315 Py_InitializeEx(1);
316}
317
318
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000319#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000320extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000321#endif
322
Guido van Rossum25ce5661997-08-02 03:10:38 +0000323/* Undo the effect of Py_Initialize().
324
325 Beware: if multiple interpreter and/or thread states exist, these
326 are not wiped out; only the current thread and interpreter state
327 are deleted. But since everything else is deleted, those other
328 interpreter and thread states should no longer be used.
329
330 (XXX We should do better, e.g. wipe out all interpreters and
331 threads.)
332
333 Locking: as above.
334
335*/
336
337void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000338Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000339{
340 PyInterpreterState *interp;
341 PyThreadState *tstate;
342
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000343 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000344 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000345
Tim Peters384fd102001-01-21 03:40:37 +0000346 /* The interpreter is still entirely intact at this point, and the
347 * exit funcs may be relying on that. In particular, if some thread
348 * or exit func is still waiting to do an import, the import machinery
349 * expects Py_IsInitialized() to return true. So don't say the
350 * interpreter is uninitialized until after the exit funcs have run.
351 * Note that Threading.py uses an exit func to do a join on all the
352 * threads created thru it, so this also protects pending imports in
353 * the threads created via Threading.
354 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000355 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000356 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000357
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000358 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000359 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000360 interp = tstate->interp;
361
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000362 /* Disable signal handling */
363 PyOS_FiniInterrupts();
364
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000365 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000366 Py_XDECREF(warnings_module);
367 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000368
Guido van Rossume13ddc92003-04-17 17:29:22 +0000369 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000370 * before all modules are destroyed.
371 * XXX If a __del__ or weakref callback is triggered here, and tries to
372 * XXX import a module, bad things can happen, because Python no
373 * XXX longer believes it's initialized.
374 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
375 * XXX is easy to provoke that way. I've also seen, e.g.,
376 * XXX Exception exceptions.ImportError: 'No module named sha'
377 * XXX in <function callback at 0x008F5718> ignored
378 * XXX but I'm unclear on exactly how that one happens. In any case,
379 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000380 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000381 PyGC_Collect();
Martin v. Löwis45294a92006-04-18 06:24:08 +0000382#ifdef COUNT_ALLOCS
383 /* With COUNT_ALLOCS, it helps to run GC multiple times:
384 each collection might release some types from the type
385 list, so they become garbage. */
386 while (PyGC_Collect() > 0)
387 /* nothing */;
388#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000389
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000390 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000391 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000392
Guido van Rossume13ddc92003-04-17 17:29:22 +0000393 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000394 * new-style class definitions, for example.
395 * XXX This is disabled because it caused too many problems. If
396 * XXX a __del__ or weakref callback triggers here, Python code has
397 * XXX a hard time running, because even the sys module has been
398 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
399 * XXX One symptom is a sequence of information-free messages
400 * XXX coming from threads (if a __del__ or callback is invoked,
401 * XXX other threads can execute too, and any exception they encounter
402 * XXX triggers a comedy of errors as subsystem after subsystem
403 * XXX fails to find what it *expects* to find in sys to help report
404 * XXX the exception and consequent unexpected failures). I've also
405 * XXX seen segfaults then, after adding print statements to the
406 * XXX Python code getting called.
407 */
408#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000409 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000410#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000411
Guido van Rossum1707aad1997-12-08 23:43:45 +0000412 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
413 _PyImport_Fini();
414
415 /* Debugging stuff */
416#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000417 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000418#endif
419
Tim Peters62e97f02006-03-28 21:44:32 +0000420 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000421
Tim Peters9cf25ce2003-04-17 15:21:01 +0000422#ifdef Py_TRACE_REFS
423 /* Display all objects still alive -- this can invoke arbitrary
424 * __repr__ overrides, so requires a mostly-intact interpreter.
425 * Alas, a lot of stuff may still be alive now that will be cleaned
426 * up later.
427 */
Tim Peters269b2a62003-04-17 19:52:29 +0000428 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000429 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000430#endif /* Py_TRACE_REFS */
431
Mark Hammond6cb90292003-04-22 11:18:00 +0000432 /* Cleanup auto-thread-state */
433#ifdef WITH_THREAD
434 _PyGILState_Fini();
435#endif /* WITH_THREAD */
436
Guido van Rossumd922fa42003-04-15 14:10:09 +0000437 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000438 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000439
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000440 /* Now we decref the exception classes. After this point nothing
441 can raise an exception. That's okay, because each Fini() method
442 below has been checked to make sure no exceptions are ever
443 raised.
444 */
445
446 _PyExc_Fini();
447
Guido van Rossumd922fa42003-04-15 14:10:09 +0000448 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000449 PyThreadState_Swap(NULL);
450 PyInterpreterState_Delete(interp);
451
Guido van Rossumd922fa42003-04-15 14:10:09 +0000452 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000453 PyMethod_Fini();
454 PyFrame_Fini();
455 PyCFunction_Fini();
456 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000457 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000458 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000459 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000460 PyInt_Fini();
461 PyFloat_Fini();
462
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000463#ifdef Py_USING_UNICODE
464 /* Cleanup Unicode implementation */
465 _PyUnicode_Fini();
466#endif
467
Guido van Rossumcc283f51997-08-05 02:22:03 +0000468 /* XXX Still allocated:
469 - various static ad-hoc pointers to interned strings
470 - int and float free list blocks
471 - whatever various modules and libraries allocate
472 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000473
474 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000475
Tim Peters269b2a62003-04-17 19:52:29 +0000476#ifdef Py_TRACE_REFS
477 /* Display addresses (& refcnts) of all objects still alive.
478 * An address can be used to find the repr of the object, printed
479 * above by _Py_PrintReferences.
480 */
481 if (Py_GETENV("PYTHONDUMPREFS"))
482 _Py_PrintReferenceAddresses(stderr);
483#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000484#ifdef PYMALLOC_DEBUG
485 if (Py_GETENV("PYTHONMALLOCSTATS"))
486 _PyObject_DebugMallocStats();
487#endif
488
Guido van Rossumcc283f51997-08-05 02:22:03 +0000489 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000490}
491
492/* Create and initialize a new interpreter and thread, and return the
493 new thread. This requires that Py_Initialize() has been called
494 first.
495
496 Unsuccessful initialization yields a NULL pointer. Note that *no*
497 exception information is available even in this case -- the
498 exception information is held in the thread, and there is no
499 thread.
500
501 Locking: as above.
502
503*/
504
505PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000506Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000507{
508 PyInterpreterState *interp;
509 PyThreadState *tstate, *save_tstate;
510 PyObject *bimod, *sysmod;
511
512 if (!initialized)
513 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
514
515 interp = PyInterpreterState_New();
516 if (interp == NULL)
517 return NULL;
518
519 tstate = PyThreadState_New(interp);
520 if (tstate == NULL) {
521 PyInterpreterState_Delete(interp);
522 return NULL;
523 }
524
525 save_tstate = PyThreadState_Swap(tstate);
526
527 /* XXX The following is lax in error checking */
528
529 interp->modules = PyDict_New();
530
531 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
532 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000533 interp->builtins = PyModule_GetDict(bimod);
Neal Norwitz0c6ae5b2006-08-21 20:16:24 +0000534 if (interp->builtins == NULL)
535 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000536 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000537 }
538 sysmod = _PyImport_FindExtension("sys", "sys");
539 if (bimod != NULL && sysmod != NULL) {
540 interp->sysdict = PyModule_GetDict(sysmod);
Neal Norwitz0c6ae5b2006-08-21 20:16:24 +0000541 if (interp->sysdict == NULL)
542 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000543 Py_INCREF(interp->sysdict);
544 PySys_SetPath(Py_GetPath());
545 PyDict_SetItemString(interp->sysdict, "modules",
546 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000547 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000548 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000549 if (!Py_NoSiteFlag)
550 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000551 }
552
553 if (!PyErr_Occurred())
554 return tstate;
555
Neal Norwitz0c6ae5b2006-08-21 20:16:24 +0000556handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000557 /* Oops, it didn't work. Undo it all. */
558
559 PyErr_Print();
560 PyThreadState_Clear(tstate);
561 PyThreadState_Swap(save_tstate);
562 PyThreadState_Delete(tstate);
563 PyInterpreterState_Delete(interp);
564
565 return NULL;
566}
567
568/* Delete an interpreter and its last thread. This requires that the
569 given thread state is current, that the thread has no remaining
570 frames, and that it is its interpreter's only remaining thread.
571 It is a fatal error to violate these constraints.
572
573 (Py_Finalize() doesn't have these constraints -- it zaps
574 everything, regardless.)
575
576 Locking: as above.
577
578*/
579
580void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000581Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000582{
583 PyInterpreterState *interp = tstate->interp;
584
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000585 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000586 Py_FatalError("Py_EndInterpreter: thread is not current");
587 if (tstate->frame != NULL)
588 Py_FatalError("Py_EndInterpreter: thread still has a frame");
589 if (tstate != interp->tstate_head || tstate->next != NULL)
590 Py_FatalError("Py_EndInterpreter: not the last thread");
591
592 PyImport_Cleanup();
593 PyInterpreterState_Clear(interp);
594 PyThreadState_Swap(NULL);
595 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000596}
597
598static char *progname = "python";
599
600void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000601Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000602{
603 if (pn && *pn)
604 progname = pn;
605}
606
607char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000608Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000609{
610 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000611}
612
Guido van Rossuma61691e1998-02-06 22:27:24 +0000613static char *default_home = NULL;
614
615void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000616Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000617{
618 default_home = home;
619}
620
621char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000622Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000623{
624 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000625 if (home == NULL && !Py_IgnoreEnvironmentFlag)
626 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000627 return home;
628}
629
Guido van Rossum6135a871995-01-09 17:53:26 +0000630/* Create __main__ module */
631
632static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000633initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000634{
Guido van Rossum82598051997-03-05 00:20:32 +0000635 PyObject *m, *d;
636 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000637 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000638 Py_FatalError("can't create __main__ module");
639 d = PyModule_GetDict(m);
640 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000641 PyObject *bimod = PyImport_ImportModule("__builtin__");
642 if (bimod == NULL ||
643 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000644 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000645 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000646 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000647}
648
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000649/* Import the site module (not into __main__ though) */
650
651static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000652initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000653{
654 PyObject *m, *f;
655 m = PyImport_ImportModule("site");
656 if (m == NULL) {
657 f = PySys_GetObject("stderr");
658 if (Py_VerboseFlag) {
659 PyFile_WriteString(
660 "'import site' failed; traceback:\n", f);
661 PyErr_Print();
662 }
663 else {
664 PyFile_WriteString(
665 "'import site' failed; use -v for traceback\n", f);
666 PyErr_Clear();
667 }
668 }
669 else {
670 Py_DECREF(m);
671 }
672}
673
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000674/* Parse input from a file and execute it */
675
676int
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000677PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000678 PyCompilerFlags *flags)
679{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000680 if (filename == NULL)
681 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000682 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000683 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000684 if (closeit)
685 fclose(fp);
686 return err;
687 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000688 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000689 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000690}
691
692int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000693PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000694{
Guido van Rossum82598051997-03-05 00:20:32 +0000695 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000696 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000697 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000698
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000699 if (flags == NULL) {
700 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000701 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000702 }
Guido van Rossum82598051997-03-05 00:20:32 +0000703 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000704 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000705 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
706 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000707 }
Guido van Rossum82598051997-03-05 00:20:32 +0000708 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000709 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000710 PySys_SetObject("ps2", v = PyString_FromString("... "));
711 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000712 }
713 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000714 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Tim Peters62e97f02006-03-28 21:44:32 +0000715 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000716 if (ret == E_EOF)
717 return 0;
718 /*
719 if (ret == E_NOMEM)
720 return -1;
721 */
722 }
723}
724
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000725/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000726#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000727 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
728 PyPARSE_DONT_IMPLY_DEDENT : 0) \
Neal Norwitz9589ee22006-03-04 19:01:22 +0000729 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
730 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000731
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000732int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000733PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000734{
Guido van Rossum82598051997-03-05 00:20:32 +0000735 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000736 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000737 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000738 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000739 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000740
Guido van Rossum82598051997-03-05 00:20:32 +0000741 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000742 if (v != NULL) {
743 v = PyObject_Str(v);
744 if (v == NULL)
745 PyErr_Clear();
746 else if (PyString_Check(v))
747 ps1 = PyString_AsString(v);
748 }
Guido van Rossum82598051997-03-05 00:20:32 +0000749 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000750 if (w != NULL) {
751 w = PyObject_Str(w);
752 if (w == NULL)
753 PyErr_Clear();
754 else if (PyString_Check(w))
755 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000756 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000757 arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +0000758 if (arena == NULL) {
759 Py_XDECREF(v);
760 Py_XDECREF(w);
761 return -1;
762 }
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000763 mod = PyParser_ASTFromFile(fp, filename,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000764 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000765 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000766 Py_XDECREF(v);
767 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000768 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000769 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000770 if (errcode == E_EOF) {
771 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000772 return E_EOF;
773 }
Guido van Rossum82598051997-03-05 00:20:32 +0000774 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000775 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000776 }
Guido van Rossum82598051997-03-05 00:20:32 +0000777 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000778 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000779 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000780 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000781 }
Guido van Rossum82598051997-03-05 00:20:32 +0000782 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000783 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000784 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000785 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000786 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000787 return -1;
788 }
Guido van Rossum82598051997-03-05 00:20:32 +0000789 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000790 if (Py_FlushLine())
791 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000792 return 0;
793}
794
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000795/* Check whether a file maybe a pyc file: Look at the extension,
796 the file type, and, if we may close it, at the first few bytes. */
797
798static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000799maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000800{
801 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
802 return 1;
803
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000804 /* Only look into the file if we are allowed to close it, since
805 it then should also be seekable. */
806 if (closeit) {
807 /* Read only two bytes of the magic. If the file was opened in
808 text mode, the bytes 3 and 4 of the magic (\r\n) might not
809 be read as they are on disk. */
810 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
811 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000812 /* Mess: In case of -x, the stream is NOT at its start now,
813 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000814 which makes the current stream position formally undefined,
815 and a x-platform nightmare.
816 Unfortunately, we have no direct way to know whether -x
817 was specified. So we use a terrible hack: if the current
818 stream position is not 0, we assume -x was specified, and
819 give up. Bug 132850 on SourceForge spells out the
820 hopelessness of trying anything else (fseek and ftell
821 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000822 */
Tim Peters3e876562001-02-11 04:35:39 +0000823 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000824 if (ftell(fp) == 0) {
825 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000826 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000827 ispyc = 1;
828 rewind(fp);
829 }
Tim Peters3e876562001-02-11 04:35:39 +0000830 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000831 }
832 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000833}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000834
Guido van Rossum0df002c2000-08-27 19:21:52 +0000835int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000836PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000837 PyCompilerFlags *flags)
838{
Guido van Rossum82598051997-03-05 00:20:32 +0000839 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000840 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000841
Guido van Rossum82598051997-03-05 00:20:32 +0000842 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000843 if (m == NULL)
844 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000845 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000846 if (PyDict_GetItemString(d, "__file__") == NULL) {
847 PyObject *f = PyString_FromString(filename);
848 if (f == NULL)
849 return -1;
850 if (PyDict_SetItemString(d, "__file__", f) < 0) {
851 Py_DECREF(f);
852 return -1;
853 }
854 Py_DECREF(f);
855 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000856 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000857 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000858 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000859 if (closeit)
860 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000861 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000862 fprintf(stderr, "python: Can't reopen .pyc file\n");
863 return -1;
864 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000865 /* Turn on optimization if a .pyo file is given */
866 if (strcmp(ext, ".pyo") == 0)
867 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000868 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000869 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000870 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000871 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000872 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000873 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000874 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000875 return -1;
876 }
Guido van Rossum82598051997-03-05 00:20:32 +0000877 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000878 if (Py_FlushLine())
879 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000880 return 0;
881}
882
883int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000884PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000885{
Guido van Rossum82598051997-03-05 00:20:32 +0000886 PyObject *m, *d, *v;
887 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000888 if (m == NULL)
889 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000890 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000891 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000892 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000893 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000894 return -1;
895 }
Guido van Rossum82598051997-03-05 00:20:32 +0000896 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000897 if (Py_FlushLine())
898 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000899 return 0;
900}
901
Barry Warsaw035574d1997-08-29 22:07:17 +0000902static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000903parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
904 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000905{
906 long hold;
907 PyObject *v;
908
909 /* old style errors */
910 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000911 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000912 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000913
914 /* new style errors. `err' is an instance */
915
916 if (! (v = PyObject_GetAttrString(err, "msg")))
917 goto finally;
918 *message = v;
919
920 if (!(v = PyObject_GetAttrString(err, "filename")))
921 goto finally;
922 if (v == Py_None)
923 *filename = NULL;
924 else if (! (*filename = PyString_AsString(v)))
925 goto finally;
926
927 Py_DECREF(v);
928 if (!(v = PyObject_GetAttrString(err, "lineno")))
929 goto finally;
930 hold = PyInt_AsLong(v);
931 Py_DECREF(v);
932 v = NULL;
933 if (hold < 0 && PyErr_Occurred())
934 goto finally;
935 *lineno = (int)hold;
936
937 if (!(v = PyObject_GetAttrString(err, "offset")))
938 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000939 if (v == Py_None) {
940 *offset = -1;
941 Py_DECREF(v);
942 v = NULL;
943 } else {
944 hold = PyInt_AsLong(v);
945 Py_DECREF(v);
946 v = NULL;
947 if (hold < 0 && PyErr_Occurred())
948 goto finally;
949 *offset = (int)hold;
950 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000951
952 if (!(v = PyObject_GetAttrString(err, "text")))
953 goto finally;
954 if (v == Py_None)
955 *text = NULL;
956 else if (! (*text = PyString_AsString(v)))
957 goto finally;
958 Py_DECREF(v);
959 return 1;
960
961finally:
962 Py_XDECREF(v);
963 return 0;
964}
965
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000966void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000967PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000968{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000969 PyErr_PrintEx(1);
970}
971
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000972static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000973print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000974{
975 char *nl;
976 if (offset >= 0) {
977 if (offset > 0 && offset == (int)strlen(text))
978 offset--;
979 for (;;) {
980 nl = strchr(text, '\n');
981 if (nl == NULL || nl-text >= offset)
982 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000983 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000984 text = nl+1;
985 }
986 while (*text == ' ' || *text == '\t') {
987 text++;
988 offset--;
989 }
990 }
991 PyFile_WriteString(" ", f);
992 PyFile_WriteString(text, f);
993 if (*text == '\0' || text[strlen(text)-1] != '\n')
994 PyFile_WriteString("\n", f);
995 if (offset == -1)
996 return;
997 PyFile_WriteString(" ", f);
998 offset--;
999 while (offset > 0) {
1000 PyFile_WriteString(" ", f);
1001 offset--;
1002 }
1003 PyFile_WriteString("^\n", f);
1004}
1005
Guido van Rossum66e8e862001-03-23 17:54:43 +00001006static void
1007handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001008{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001009 PyObject *exception, *value, *tb;
1010 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001011
Guido van Rossum66e8e862001-03-23 17:54:43 +00001012 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001013 if (Py_FlushLine())
1014 PyErr_Clear();
1015 fflush(stdout);
1016 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001017 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001018 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001019 /* The error code should be in the `code' attribute. */
1020 PyObject *code = PyObject_GetAttrString(value, "code");
1021 if (code) {
1022 Py_DECREF(value);
1023 value = code;
1024 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001025 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001026 }
1027 /* If we failed to dig out the 'code' attribute,
1028 just let the else clause below print the error. */
1029 }
1030 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001031 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001032 else {
1033 PyObject_Print(value, stderr, Py_PRINT_RAW);
1034 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001035 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001036 }
Tim Peterscf615b52003-04-19 18:47:02 +00001037 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001038 /* Restore and clear the exception info, in order to properly decref
1039 * the exception, value, and traceback. If we just exit instead,
1040 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1041 * some finalizers from running.
1042 */
Tim Peterscf615b52003-04-19 18:47:02 +00001043 PyErr_Restore(exception, value, tb);
1044 PyErr_Clear();
1045 Py_Exit(exitcode);
1046 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001047}
1048
1049void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001050PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001051{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001052 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001053
1054 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1055 handle_system_exit();
1056 }
Guido van Rossum82598051997-03-05 00:20:32 +00001057 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001058 if (exception == NULL)
1059 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001060 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001061 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001062 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001063 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001064 if (set_sys_last_vars) {
1065 PySys_SetObject("last_type", exception);
1066 PySys_SetObject("last_value", v);
1067 PySys_SetObject("last_traceback", tb);
1068 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001069 hook = PySys_GetObject("excepthook");
1070 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001071 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001072 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001073 PyObject *result = PyEval_CallObject(hook, args);
1074 if (result == NULL) {
1075 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001076 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1077 handle_system_exit();
1078 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001079 PyErr_Fetch(&exception2, &v2, &tb2);
1080 PyErr_NormalizeException(&exception2, &v2, &tb2);
Neal Norwitza5e4f222006-07-17 00:59:04 +00001081 /* It should not be possible for exception2 or v2
1082 to be NULL. However PyErr_Display() can't
1083 tolerate NULLs, so just be safe. */
1084 if (exception2 == NULL) {
1085 exception2 = Py_None;
1086 Py_INCREF(exception2);
1087 }
1088 if (v2 == NULL) {
1089 v2 = Py_None;
1090 Py_INCREF(v2);
1091 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001092 if (Py_FlushLine())
1093 PyErr_Clear();
1094 fflush(stdout);
1095 PySys_WriteStderr("Error in sys.excepthook:\n");
1096 PyErr_Display(exception2, v2, tb2);
1097 PySys_WriteStderr("\nOriginal exception was:\n");
1098 PyErr_Display(exception, v, tb);
Neal Norwitza5e4f222006-07-17 00:59:04 +00001099 Py_DECREF(exception2);
1100 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001101 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001102 }
1103 Py_XDECREF(result);
1104 Py_XDECREF(args);
1105 } else {
1106 PySys_WriteStderr("sys.excepthook is missing\n");
1107 PyErr_Display(exception, v, tb);
1108 }
1109 Py_XDECREF(exception);
1110 Py_XDECREF(v);
1111 Py_XDECREF(tb);
1112}
1113
Richard Jones7b9558d2006-05-27 12:29:24 +00001114void
1115PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001116{
1117 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001118 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001119 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001120 if (f == NULL)
1121 fprintf(stderr, "lost sys.stderr\n");
1122 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001123 if (Py_FlushLine())
1124 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001125 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001126 if (tb && tb != Py_None)
1127 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001128 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001129 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001130 {
Guido van Rossum82598051997-03-05 00:20:32 +00001131 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001132 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001133 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001134 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001135 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001136 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001137 else {
1138 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001139 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001140 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001141 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001142 else
Guido van Rossum82598051997-03-05 00:20:32 +00001143 PyFile_WriteString(filename, f);
1144 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001145 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001146 PyFile_WriteString(buf, f);
1147 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001148 if (text != NULL)
1149 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001150 Py_DECREF(value);
1151 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001152 /* Can't be bothered to check all those
1153 PyFile_WriteString() calls */
1154 if (PyErr_Occurred())
1155 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001156 }
1157 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001158 if (err) {
1159 /* Don't do anything else */
1160 }
Brett Cannonbf364092006-03-01 04:25:17 +00001161 else if (PyExceptionClass_Check(exception)) {
Richard Jones7b9558d2006-05-27 12:29:24 +00001162 PyObject* moduleName;
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001163 char* className = PyExceptionClass_Name(exception);
1164 if (className != NULL) {
1165 char *dot = strrchr(className, '.');
1166 if (dot != NULL)
1167 className = dot+1;
1168 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001169
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001170 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001171 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001172 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001173 else {
1174 char* modstr = PyString_AsString(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001175 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001176 {
1177 err = PyFile_WriteString(modstr, f);
1178 err += PyFile_WriteString(".", f);
1179 }
Richard Jones7b9558d2006-05-27 12:29:24 +00001180 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001181 }
1182 if (err == 0) {
1183 if (className == NULL)
1184 err = PyFile_WriteString("<unknown>", f);
1185 else
Brett Cannonbf364092006-03-01 04:25:17 +00001186 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001187 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001188 }
1189 else
1190 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001191 if (err == 0 && (value != Py_None)) {
1192 PyObject *s = PyObject_Str(value);
1193 /* only print colon if the str() of the
1194 object is not the empty string
1195 */
1196 if (s == NULL)
1197 err = -1;
1198 else if (!PyString_Check(s) ||
1199 PyString_GET_SIZE(s) != 0)
1200 err = PyFile_WriteString(": ", f);
1201 if (err == 0)
1202 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1203 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001204 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001205 if (err == 0)
1206 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001207 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001208 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001209 /* If an error happened here, don't show it.
1210 XXX This is wrong, but too many callers rely on this behavior. */
1211 if (err != 0)
1212 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001213}
1214
Guido van Rossum82598051997-03-05 00:20:32 +00001215PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001216PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001217 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001218{
Neal Norwitze92fba02006-03-04 18:52:26 +00001219 PyObject *ret = NULL;
Neal Norwitzd12bd012006-07-21 07:59:47 +00001220 mod_ty mod;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001221 PyArena *arena = PyArena_New();
Neal Norwitzd12bd012006-07-21 07:59:47 +00001222 if (arena == NULL)
1223 return NULL;
1224
1225 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001226 if (mod != NULL)
1227 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001228 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001229 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001230}
1231
1232PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001233PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001234 PyObject *locals, int closeit, PyCompilerFlags *flags)
1235{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001236 PyObject *ret;
Neal Norwitzd12bd012006-07-21 07:59:47 +00001237 mod_ty mod;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001238 PyArena *arena = PyArena_New();
Neal Norwitzd12bd012006-07-21 07:59:47 +00001239 if (arena == NULL)
1240 return NULL;
1241
1242 mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
1243 flags, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001244 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001245 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001246 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001247 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001248 if (closeit)
1249 fclose(fp);
Neal Norwitze92fba02006-03-04 18:52:26 +00001250 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001251 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001252 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001253}
1254
Guido van Rossum82598051997-03-05 00:20:32 +00001255static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001256run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001257 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001258{
Guido van Rossum82598051997-03-05 00:20:32 +00001259 PyCodeObject *co;
1260 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001261 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001262 if (co == NULL)
1263 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001264 v = PyEval_EvalCode(co, globals, locals);
1265 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001266 return v;
1267}
1268
Guido van Rossum82598051997-03-05 00:20:32 +00001269static PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001270run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001271 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001272{
Guido van Rossum82598051997-03-05 00:20:32 +00001273 PyCodeObject *co;
1274 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001275 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001276 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001277
Guido van Rossum82598051997-03-05 00:20:32 +00001278 magic = PyMarshal_ReadLongFromFile(fp);
1279 if (magic != PyImport_GetMagicNumber()) {
1280 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001281 "Bad magic number in .pyc file");
1282 return NULL;
1283 }
Guido van Rossum82598051997-03-05 00:20:32 +00001284 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001285 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001286 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001287 if (v == NULL || !PyCode_Check(v)) {
1288 Py_XDECREF(v);
1289 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001290 "Bad code object in .pyc file");
1291 return NULL;
1292 }
Guido van Rossum82598051997-03-05 00:20:32 +00001293 co = (PyCodeObject *)v;
1294 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001295 if (v && flags)
1296 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001297 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001298 return v;
1299}
1300
Guido van Rossum82598051997-03-05 00:20:32 +00001301PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001302Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001303 PyCompilerFlags *flags)
1304{
Guido van Rossum82598051997-03-05 00:20:32 +00001305 PyCodeObject *co;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001306 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001307 PyArena *arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001308 if (arena == NULL)
1309 return NULL;
1310
1311 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001312 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001313 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001314 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001315 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001316 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001317 PyObject *result = PyAST_mod2obj(mod);
1318 PyArena_Free(arena);
1319 return result;
1320 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001321 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001322 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001323 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001324}
1325
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001326struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001327Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001328{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001329 struct symtable *st;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001330 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001331 PyArena *arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001332 if (arena == NULL)
1333 return NULL;
1334
1335 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001336 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001337 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001338 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001339 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001340 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001341 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001342 return st;
1343}
1344
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001345/* Preferred access to parser is through AST. */
1346mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001347PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001348 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001349{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001350 mod_ty mod;
1351 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001352 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001353 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001354 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001355 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001356 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001357 PyNode_Free(n);
1358 return mod;
1359 }
1360 else {
1361 err_input(&err);
1362 return NULL;
1363 }
1364}
1365
1366mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001367PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001368 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001369 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001370{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001371 mod_ty mod;
1372 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001373 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1374 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001375 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001376 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001377 PyNode_Free(n);
1378 return mod;
1379 }
1380 else {
1381 err_input(&err);
1382 if (errcode)
1383 *errcode = err.error;
1384 return NULL;
1385 }
1386}
1387
Guido van Rossuma110aa61994-08-29 12:50:44 +00001388/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001389
Guido van Rossuma110aa61994-08-29 12:50:44 +00001390node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001391PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001392{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001393 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001394 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1395 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001396 if (n == NULL)
1397 err_input(&err);
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001398
Guido van Rossuma110aa61994-08-29 12:50:44 +00001399 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001400}
1401
Guido van Rossuma110aa61994-08-29 12:50:44 +00001402/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001403
Guido van Rossuma110aa61994-08-29 12:50:44 +00001404node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001405PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001406{
Tim Petersfe2127d2001-07-16 05:37:24 +00001407 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001408 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1409 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001410 if (n == NULL)
1411 err_input(&err);
1412 return n;
1413}
1414
1415node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001416PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001417 int start, int flags)
1418{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001419 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001420 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1421 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001422 if (n == NULL)
1423 err_input(&err);
1424 return n;
1425}
1426
1427node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001428PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001429{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001430 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001431}
1432
Guido van Rossum66ebd912003-04-17 16:02:26 +00001433/* May want to move a more generalized form of this to parsetok.c or
1434 even parser modules. */
1435
1436void
1437PyParser_SetError(perrdetail *err)
1438{
1439 err_input(err);
1440}
1441
Guido van Rossuma110aa61994-08-29 12:50:44 +00001442/* Set the error appropriate to the given input error code (see errcode.h) */
1443
1444static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001445err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001446{
Fred Drake85f36392000-07-11 17:53:00 +00001447 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001448 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001449 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001450 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001451 switch (err->error) {
1452 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001453 errtype = PyExc_IndentationError;
1454 if (err->expected == INDENT)
1455 msg = "expected an indented block";
1456 else if (err->token == INDENT)
1457 msg = "unexpected indent";
1458 else if (err->token == DEDENT)
1459 msg = "unexpected unindent";
1460 else {
1461 errtype = PyExc_SyntaxError;
1462 msg = "invalid syntax";
1463 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001464 break;
1465 case E_TOKEN:
1466 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001467 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001468 case E_EOFS:
1469 msg = "EOF while scanning triple-quoted string";
1470 break;
1471 case E_EOLS:
1472 msg = "EOL while scanning single-quoted string";
1473 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001474 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001475 if (!PyErr_Occurred())
1476 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001477 return;
1478 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001479 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001480 return;
1481 case E_EOF:
1482 msg = "unexpected EOF while parsing";
1483 break;
Fred Drake85f36392000-07-11 17:53:00 +00001484 case E_TABSPACE:
1485 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001486 msg = "inconsistent use of tabs and spaces in indentation";
1487 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001488 case E_OVERFLOW:
1489 msg = "expression too long";
1490 break;
Fred Drake85f36392000-07-11 17:53:00 +00001491 case E_DEDENT:
1492 errtype = PyExc_IndentationError;
1493 msg = "unindent does not match any outer indentation level";
1494 break;
1495 case E_TOODEEP:
1496 errtype = PyExc_IndentationError;
1497 msg = "too many levels of indentation";
1498 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001499 case E_DECODE: {
1500 PyObject *type, *value, *tb;
1501 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001502 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001503 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001504 if (u != NULL) {
1505 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001506 }
1507 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001508 if (msg == NULL)
1509 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001510 Py_XDECREF(type);
1511 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001512 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001513 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001514 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001515 case E_LINECONT:
1516 msg = "unexpected character after line continuation character";
1517 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001518 default:
1519 fprintf(stderr, "error=%d\n", err->error);
1520 msg = "unknown parsing error";
1521 break;
1522 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001523 v = Py_BuildValue("(ziiz)", err->filename,
1524 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001525 if (err->text != NULL) {
Tim Petersc9d78aa2006-03-26 23:27:58 +00001526 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001527 err->text = NULL;
1528 }
1529 w = NULL;
1530 if (v != NULL)
1531 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001532 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001533 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001534 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001535 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001536}
1537
1538/* Print fatal error message and abort */
1539
1540void
Tim Peters7c321a82002-07-09 02:57:01 +00001541Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001542{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001543 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001544#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001545 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001546 OutputDebugString(msg);
1547 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001548#ifdef _DEBUG
1549 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001550#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001551#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001552 abort();
1553}
1554
1555/* Clean up and exit */
1556
Guido van Rossuma110aa61994-08-29 12:50:44 +00001557#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001558#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001559#endif
1560
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001561#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001562static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001563static int nexitfuncs = 0;
1564
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001565int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001566{
1567 if (nexitfuncs >= NEXITFUNCS)
1568 return -1;
1569 exitfuncs[nexitfuncs++] = func;
1570 return 0;
1571}
1572
Guido van Rossumcc283f51997-08-05 02:22:03 +00001573static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001574call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001575{
Guido van Rossum82598051997-03-05 00:20:32 +00001576 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001577
1578 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001579 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001580 Py_INCREF(exitfunc);
1581 PySys_SetObject("exitfunc", (PyObject *)NULL);
1582 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001583 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001584 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1585 PySys_WriteStderr("Error in sys.exitfunc:\n");
1586 }
Guido van Rossum82598051997-03-05 00:20:32 +00001587 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001588 }
Guido van Rossum82598051997-03-05 00:20:32 +00001589 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001590 }
1591
Guido van Rossum0829c751998-02-28 04:31:39 +00001592 if (Py_FlushLine())
1593 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001594}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001595
Guido van Rossumcc283f51997-08-05 02:22:03 +00001596static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001597call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001598{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001599 while (nexitfuncs > 0)
1600 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001601
1602 fflush(stdout);
1603 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001604}
1605
1606void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001607Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001608{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001609 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001610
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001611 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001612}
1613
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001614static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001615initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001616{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001617#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001618 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001619#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001620#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001621 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001622#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001623#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001624 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001625#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001626 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001627}
1628
Guido van Rossum7433b121997-02-14 19:45:36 +00001629
1630/*
1631 * The file descriptor fd is considered ``interactive'' if either
1632 * a) isatty(fd) is TRUE, or
1633 * b) the -i flag was given, and the filename associated with
1634 * the descriptor is NULL or "<stdin>" or "???".
1635 */
1636int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001637Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001638{
1639 if (isatty((int)fileno(fp)))
1640 return 1;
1641 if (!Py_InteractiveFlag)
1642 return 0;
1643 return (filename == NULL) ||
1644 (strcmp(filename, "<stdin>") == 0) ||
1645 (strcmp(filename, "???") == 0);
1646}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001647
1648
Tim Petersd08e3822003-04-17 15:24:21 +00001649#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001650#if defined(WIN32) && defined(_MSC_VER)
1651
1652/* Stack checking for Microsoft C */
1653
1654#include <malloc.h>
1655#include <excpt.h>
1656
Fred Drakee8de31c2000-08-31 05:38:39 +00001657/*
1658 * Return non-zero when we run out of memory on the stack; zero otherwise.
1659 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001660int
Fred Drake399739f2000-08-31 05:52:44 +00001661PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001662{
1663 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001664 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001665 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001666 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001667 return 0;
1668 } __except (EXCEPTION_EXECUTE_HANDLER) {
1669 /* just ignore all errors */
1670 }
1671 return 1;
1672}
1673
1674#endif /* WIN32 && _MSC_VER */
1675
1676/* Alternate implementations can be added here... */
1677
1678#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001679
1680
1681/* Wrappers around sigaction() or signal(). */
1682
1683PyOS_sighandler_t
1684PyOS_getsig(int sig)
1685{
1686#ifdef HAVE_SIGACTION
1687 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001688 if (sigaction(sig, NULL, &context) == -1)
1689 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001690 return context.sa_handler;
1691#else
1692 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001693/* Special signal handling for the secure CRT in Visual Studio 2005 */
1694#if defined(_MSC_VER) && _MSC_VER >= 1400
1695 switch (sig) {
1696 /* Only these signals are valid */
1697 case SIGINT:
1698 case SIGILL:
1699 case SIGFPE:
1700 case SIGSEGV:
1701 case SIGTERM:
1702 case SIGBREAK:
1703 case SIGABRT:
1704 break;
1705 /* Don't call signal() with other values or it will assert */
1706 default:
1707 return SIG_ERR;
1708 }
1709#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001710 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001711 if (handler != SIG_ERR)
1712 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001713 return handler;
1714#endif
1715}
1716
1717PyOS_sighandler_t
1718PyOS_setsig(int sig, PyOS_sighandler_t handler)
1719{
1720#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001721 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001722 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001723 sigemptyset(&context.sa_mask);
1724 context.sa_flags = 0;
1725 if (sigaction(sig, &context, &ocontext) == -1)
1726 return SIG_ERR;
1727 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001728#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001729 PyOS_sighandler_t oldhandler;
1730 oldhandler = signal(sig, handler);
1731#ifdef HAVE_SIGINTERRUPT
1732 siginterrupt(sig, 1);
1733#endif
1734 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001735#endif
1736}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001737
1738/* Deprecated C API functions still provided for binary compatiblity */
1739
1740#undef PyParser_SimpleParseFile
Thomas Heller1b046642006-04-18 18:51:06 +00001741PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001742PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1743{
1744 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1745}
1746
Thomas Heller1b046642006-04-18 18:51:06 +00001747#undef PyParser_SimpleParseString
1748PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001749PyParser_SimpleParseString(const char *str, int start)
1750{
1751 return PyParser_SimpleParseStringFlags(str, start, 0);
1752}
Anthony Baxterac6bd462006-04-13 02:06:09 +00001753
Thomas Heller1b046642006-04-18 18:51:06 +00001754#undef PyRun_AnyFile
1755PyAPI_FUNC(int)
1756PyRun_AnyFile(FILE *fp, const char *name)
1757{
1758 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1759}
1760
1761#undef PyRun_AnyFileEx
1762PyAPI_FUNC(int)
1763PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1764{
1765 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1766}
1767
1768#undef PyRun_AnyFileFlags
1769PyAPI_FUNC(int)
1770PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1771{
1772 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1773}
1774
1775#undef PyRun_File
1776PyAPI_FUNC(PyObject *)
1777PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1778{
1779 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1780}
1781
1782#undef PyRun_FileEx
1783PyAPI_FUNC(PyObject *)
1784PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1785{
1786 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1787}
1788
1789#undef PyRun_FileFlags
1790PyAPI_FUNC(PyObject *)
1791PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1792 PyCompilerFlags *flags)
1793{
1794 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1795}
1796
1797#undef PyRun_SimpleFile
1798PyAPI_FUNC(int)
1799PyRun_SimpleFile(FILE *f, const char *p)
1800{
1801 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1802}
1803
1804#undef PyRun_SimpleFileEx
1805PyAPI_FUNC(int)
1806PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1807{
1808 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1809}
1810
1811
1812#undef PyRun_String
1813PyAPI_FUNC(PyObject *)
1814PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1815{
1816 return PyRun_StringFlags(str, s, g, l, NULL);
1817}
1818
1819#undef PyRun_SimpleString
1820PyAPI_FUNC(int)
1821PyRun_SimpleString(const char *s)
1822{
1823 return PyRun_SimpleStringFlags(s, NULL);
1824}
1825
1826#undef Py_CompileString
1827PyAPI_FUNC(PyObject *)
1828Py_CompileString(const char *str, const char *p, int s)
1829{
1830 return Py_CompileStringFlags(str, p, s, NULL);
1831}
1832
1833#undef PyRun_InteractiveOne
1834PyAPI_FUNC(int)
1835PyRun_InteractiveOne(FILE *f, const char *p)
1836{
1837 return PyRun_InteractiveOneFlags(f, p, NULL);
1838}
1839
1840#undef PyRun_InteractiveLoop
1841PyAPI_FUNC(int)
1842PyRun_InteractiveLoop(FILE *f, const char *p)
1843{
1844 return PyRun_InteractiveLoopFlags(f, p, NULL);
1845}
1846
Anthony Baxterac6bd462006-04-13 02:06:09 +00001847#ifdef __cplusplus
1848}
1849#endif
1850