blob: 88fd67c1e57ac19b1687b5229700e85a84342e64 [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);
206 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000207
208 sysmod = _PySys_Init();
209 if (sysmod == NULL)
210 Py_FatalError("Py_Initialize: can't initialize sys");
211 interp->sysdict = PyModule_GetDict(sysmod);
212 Py_INCREF(interp->sysdict);
213 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000214 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000215 PyDict_SetItemString(interp->sysdict, "modules",
216 interp->modules);
217
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000218 _PyImport_Init();
219
Barry Warsawf242aa02000-05-25 23:09:49 +0000220 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000221 _PyExc_Init();
Barry Warsaw5821bc52001-08-13 23:04:56 +0000222 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000223
Barry Warsaw035574d1997-08-29 22:07:17 +0000224 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000225 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000226
Just van Rossum52e14d62002-12-30 22:08:05 +0000227 _PyImportHooks_Init();
228
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000229 if (install_sigs)
230 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000231
232 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000233 if (!Py_NoSiteFlag)
234 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000235
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000236 /* auto-thread-state API, if available */
237#ifdef WITH_THREAD
238 _PyGILState_Init(interp, tstate);
239#endif /* WITH_THREAD */
240
Mark Hammondedd07732003-07-15 23:03:55 +0000241 warnings_module = PyImport_ImportModule("warnings");
242 if (!warnings_module)
243 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000244
245#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
246 /* On Unix, set the file system encoding according to the
247 user's preference, if the CODESET names a well-known
248 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000249 initialized by other means. Also set the encoding of
250 stdin and stdout if these are terminals. */
251
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000252 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000253 setlocale(LC_CTYPE, "");
254 codeset = nl_langinfo(CODESET);
255 if (codeset && *codeset) {
256 PyObject *enc = PyCodec_Encoder(codeset);
257 if (enc) {
258 codeset = strdup(codeset);
259 Py_DECREF(enc);
260 } else {
261 codeset = NULL;
262 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000263 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000264 } else
265 codeset = NULL;
266 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000267 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000268
269 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000270 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000271 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
272 if (!sys_isatty)
273 PyErr_Clear();
274 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
275 if (!PyFile_SetEncoding(sys_stream, codeset))
276 Py_FatalError("Cannot set codeset of stdin");
277 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000278 Py_XDECREF(sys_isatty);
279
280 sys_stream = PySys_GetObject("stdout");
281 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
282 if (!sys_isatty)
283 PyErr_Clear();
284 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
285 if (!PyFile_SetEncoding(sys_stream, codeset))
286 Py_FatalError("Cannot set codeset of stdout");
287 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000288 Py_XDECREF(sys_isatty);
289
Martin v. Löwisea62d252006-04-03 10:56:49 +0000290 sys_stream = PySys_GetObject("stderr");
291 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
292 if (!sys_isatty)
293 PyErr_Clear();
294 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
295 if (!PyFile_SetEncoding(sys_stream, codeset))
296 Py_FatalError("Cannot set codeset of stderr");
297 }
298 Py_XDECREF(sys_isatty);
299
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000300 if (!Py_FileSystemDefaultEncoding)
301 Py_FileSystemDefaultEncoding = codeset;
302 else
303 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000304 }
305#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000306}
307
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000308void
309Py_Initialize(void)
310{
311 Py_InitializeEx(1);
312}
313
314
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000315#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000316extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000317#endif
318
Guido van Rossum25ce5661997-08-02 03:10:38 +0000319/* Undo the effect of Py_Initialize().
320
321 Beware: if multiple interpreter and/or thread states exist, these
322 are not wiped out; only the current thread and interpreter state
323 are deleted. But since everything else is deleted, those other
324 interpreter and thread states should no longer be used.
325
326 (XXX We should do better, e.g. wipe out all interpreters and
327 threads.)
328
329 Locking: as above.
330
331*/
332
333void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000334Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000335{
336 PyInterpreterState *interp;
337 PyThreadState *tstate;
338
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000339 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000340 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000341
Tim Peters384fd102001-01-21 03:40:37 +0000342 /* The interpreter is still entirely intact at this point, and the
343 * exit funcs may be relying on that. In particular, if some thread
344 * or exit func is still waiting to do an import, the import machinery
345 * expects Py_IsInitialized() to return true. So don't say the
346 * interpreter is uninitialized until after the exit funcs have run.
347 * Note that Threading.py uses an exit func to do a join on all the
348 * threads created thru it, so this also protects pending imports in
349 * the threads created via Threading.
350 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000351 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000352 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000353
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000354 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000355 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000356 interp = tstate->interp;
357
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000358 /* Disable signal handling */
359 PyOS_FiniInterrupts();
360
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000361 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000362 Py_XDECREF(warnings_module);
363 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000364
Guido van Rossume13ddc92003-04-17 17:29:22 +0000365 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000366 * before all modules are destroyed.
367 * XXX If a __del__ or weakref callback is triggered here, and tries to
368 * XXX import a module, bad things can happen, because Python no
369 * XXX longer believes it's initialized.
370 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
371 * XXX is easy to provoke that way. I've also seen, e.g.,
372 * XXX Exception exceptions.ImportError: 'No module named sha'
373 * XXX in <function callback at 0x008F5718> ignored
374 * XXX but I'm unclear on exactly how that one happens. In any case,
375 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000376 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000377 PyGC_Collect();
Martin v. Löwis45294a92006-04-18 06:24:08 +0000378#ifdef COUNT_ALLOCS
379 /* With COUNT_ALLOCS, it helps to run GC multiple times:
380 each collection might release some types from the type
381 list, so they become garbage. */
382 while (PyGC_Collect() > 0)
383 /* nothing */;
384#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000385
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000386 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000387 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000388
Guido van Rossume13ddc92003-04-17 17:29:22 +0000389 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000390 * new-style class definitions, for example.
391 * XXX This is disabled because it caused too many problems. If
392 * XXX a __del__ or weakref callback triggers here, Python code has
393 * XXX a hard time running, because even the sys module has been
394 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
395 * XXX One symptom is a sequence of information-free messages
396 * XXX coming from threads (if a __del__ or callback is invoked,
397 * XXX other threads can execute too, and any exception they encounter
398 * XXX triggers a comedy of errors as subsystem after subsystem
399 * XXX fails to find what it *expects* to find in sys to help report
400 * XXX the exception and consequent unexpected failures). I've also
401 * XXX seen segfaults then, after adding print statements to the
402 * XXX Python code getting called.
403 */
404#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000405 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000406#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000407
Guido van Rossum1707aad1997-12-08 23:43:45 +0000408 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
409 _PyImport_Fini();
410
411 /* Debugging stuff */
412#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000413 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000414#endif
415
Tim Peters62e97f02006-03-28 21:44:32 +0000416 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000417
Tim Peters9cf25ce2003-04-17 15:21:01 +0000418#ifdef Py_TRACE_REFS
419 /* Display all objects still alive -- this can invoke arbitrary
420 * __repr__ overrides, so requires a mostly-intact interpreter.
421 * Alas, a lot of stuff may still be alive now that will be cleaned
422 * up later.
423 */
Tim Peters269b2a62003-04-17 19:52:29 +0000424 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000425 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000426#endif /* Py_TRACE_REFS */
427
Mark Hammond6cb90292003-04-22 11:18:00 +0000428 /* Cleanup auto-thread-state */
429#ifdef WITH_THREAD
430 _PyGILState_Fini();
431#endif /* WITH_THREAD */
432
Guido van Rossumd922fa42003-04-15 14:10:09 +0000433 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000434 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000435
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000436 /* Now we decref the exception classes. After this point nothing
437 can raise an exception. That's okay, because each Fini() method
438 below has been checked to make sure no exceptions are ever
439 raised.
440 */
441
442 _PyExc_Fini();
443
Guido van Rossumd922fa42003-04-15 14:10:09 +0000444 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000445 PyThreadState_Swap(NULL);
446 PyInterpreterState_Delete(interp);
447
Guido van Rossumd922fa42003-04-15 14:10:09 +0000448 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000449 PyMethod_Fini();
450 PyFrame_Fini();
451 PyCFunction_Fini();
452 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000453 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000454 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000455 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000456 PyInt_Fini();
457 PyFloat_Fini();
458
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000459#ifdef Py_USING_UNICODE
460 /* Cleanup Unicode implementation */
461 _PyUnicode_Fini();
462#endif
463
Guido van Rossumcc283f51997-08-05 02:22:03 +0000464 /* XXX Still allocated:
465 - various static ad-hoc pointers to interned strings
466 - int and float free list blocks
467 - whatever various modules and libraries allocate
468 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000469
470 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000471
Tim Peters269b2a62003-04-17 19:52:29 +0000472#ifdef Py_TRACE_REFS
473 /* Display addresses (& refcnts) of all objects still alive.
474 * An address can be used to find the repr of the object, printed
475 * above by _Py_PrintReferences.
476 */
477 if (Py_GETENV("PYTHONDUMPREFS"))
478 _Py_PrintReferenceAddresses(stderr);
479#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000480#ifdef PYMALLOC_DEBUG
481 if (Py_GETENV("PYTHONMALLOCSTATS"))
482 _PyObject_DebugMallocStats();
483#endif
484
Guido van Rossumcc283f51997-08-05 02:22:03 +0000485 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000486}
487
488/* Create and initialize a new interpreter and thread, and return the
489 new thread. This requires that Py_Initialize() has been called
490 first.
491
492 Unsuccessful initialization yields a NULL pointer. Note that *no*
493 exception information is available even in this case -- the
494 exception information is held in the thread, and there is no
495 thread.
496
497 Locking: as above.
498
499*/
500
501PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000502Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000503{
504 PyInterpreterState *interp;
505 PyThreadState *tstate, *save_tstate;
506 PyObject *bimod, *sysmod;
507
508 if (!initialized)
509 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
510
511 interp = PyInterpreterState_New();
512 if (interp == NULL)
513 return NULL;
514
515 tstate = PyThreadState_New(interp);
516 if (tstate == NULL) {
517 PyInterpreterState_Delete(interp);
518 return NULL;
519 }
520
521 save_tstate = PyThreadState_Swap(tstate);
522
523 /* XXX The following is lax in error checking */
524
525 interp->modules = PyDict_New();
526
527 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
528 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000529 interp->builtins = PyModule_GetDict(bimod);
530 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000531 }
532 sysmod = _PyImport_FindExtension("sys", "sys");
533 if (bimod != NULL && sysmod != NULL) {
534 interp->sysdict = PyModule_GetDict(sysmod);
535 Py_INCREF(interp->sysdict);
536 PySys_SetPath(Py_GetPath());
537 PyDict_SetItemString(interp->sysdict, "modules",
538 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000539 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000540 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000541 if (!Py_NoSiteFlag)
542 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000543 }
544
545 if (!PyErr_Occurred())
546 return tstate;
547
548 /* Oops, it didn't work. Undo it all. */
549
550 PyErr_Print();
551 PyThreadState_Clear(tstate);
552 PyThreadState_Swap(save_tstate);
553 PyThreadState_Delete(tstate);
554 PyInterpreterState_Delete(interp);
555
556 return NULL;
557}
558
559/* Delete an interpreter and its last thread. This requires that the
560 given thread state is current, that the thread has no remaining
561 frames, and that it is its interpreter's only remaining thread.
562 It is a fatal error to violate these constraints.
563
564 (Py_Finalize() doesn't have these constraints -- it zaps
565 everything, regardless.)
566
567 Locking: as above.
568
569*/
570
571void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000572Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000573{
574 PyInterpreterState *interp = tstate->interp;
575
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000576 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000577 Py_FatalError("Py_EndInterpreter: thread is not current");
578 if (tstate->frame != NULL)
579 Py_FatalError("Py_EndInterpreter: thread still has a frame");
580 if (tstate != interp->tstate_head || tstate->next != NULL)
581 Py_FatalError("Py_EndInterpreter: not the last thread");
582
583 PyImport_Cleanup();
584 PyInterpreterState_Clear(interp);
585 PyThreadState_Swap(NULL);
586 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000587}
588
589static char *progname = "python";
590
591void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000592Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000593{
594 if (pn && *pn)
595 progname = pn;
596}
597
598char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000599Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000600{
601 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000602}
603
Guido van Rossuma61691e1998-02-06 22:27:24 +0000604static char *default_home = NULL;
605
606void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000607Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000608{
609 default_home = home;
610}
611
612char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000613Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000614{
615 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000616 if (home == NULL && !Py_IgnoreEnvironmentFlag)
617 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000618 return home;
619}
620
Guido van Rossum6135a871995-01-09 17:53:26 +0000621/* Create __main__ module */
622
623static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000624initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000625{
Guido van Rossum82598051997-03-05 00:20:32 +0000626 PyObject *m, *d;
627 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000628 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000629 Py_FatalError("can't create __main__ module");
630 d = PyModule_GetDict(m);
631 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000632 PyObject *bimod = PyImport_ImportModule("__builtin__");
633 if (bimod == NULL ||
634 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000635 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000636 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000637 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000638}
639
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000640/* Import the site module (not into __main__ though) */
641
642static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000643initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000644{
645 PyObject *m, *f;
646 m = PyImport_ImportModule("site");
647 if (m == NULL) {
648 f = PySys_GetObject("stderr");
649 if (Py_VerboseFlag) {
650 PyFile_WriteString(
651 "'import site' failed; traceback:\n", f);
652 PyErr_Print();
653 }
654 else {
655 PyFile_WriteString(
656 "'import site' failed; use -v for traceback\n", f);
657 PyErr_Clear();
658 }
659 }
660 else {
661 Py_DECREF(m);
662 }
663}
664
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000665/* Parse input from a file and execute it */
666
667int
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000668PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000669 PyCompilerFlags *flags)
670{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000671 if (filename == NULL)
672 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000673 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000674 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000675 if (closeit)
676 fclose(fp);
677 return err;
678 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000679 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000680 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000681}
682
683int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000684PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000685{
Guido van Rossum82598051997-03-05 00:20:32 +0000686 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000687 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000688 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000689
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000690 if (flags == NULL) {
691 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000692 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000693 }
Guido van Rossum82598051997-03-05 00:20:32 +0000694 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000695 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000696 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
697 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000698 }
Guido van Rossum82598051997-03-05 00:20:32 +0000699 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000700 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000701 PySys_SetObject("ps2", v = PyString_FromString("... "));
702 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000703 }
704 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000705 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Tim Peters62e97f02006-03-28 21:44:32 +0000706 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000707 if (ret == E_EOF)
708 return 0;
709 /*
710 if (ret == E_NOMEM)
711 return -1;
712 */
713 }
714}
715
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000716/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000717#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000718 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
719 PyPARSE_DONT_IMPLY_DEDENT : 0) \
Neal Norwitz9589ee22006-03-04 19:01:22 +0000720 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
721 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000722
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000723int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000724PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000725{
Guido van Rossum82598051997-03-05 00:20:32 +0000726 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000727 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000728 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000729 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000730 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000731
Guido van Rossum82598051997-03-05 00:20:32 +0000732 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000733 if (v != NULL) {
734 v = PyObject_Str(v);
735 if (v == NULL)
736 PyErr_Clear();
737 else if (PyString_Check(v))
738 ps1 = PyString_AsString(v);
739 }
Guido van Rossum82598051997-03-05 00:20:32 +0000740 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000741 if (w != NULL) {
742 w = PyObject_Str(w);
743 if (w == NULL)
744 PyErr_Clear();
745 else if (PyString_Check(w))
746 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000747 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000748 arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +0000749 if (arena == NULL) {
750 Py_XDECREF(v);
751 Py_XDECREF(w);
752 return -1;
753 }
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000754 mod = PyParser_ASTFromFile(fp, filename,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000755 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000756 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000757 Py_XDECREF(v);
758 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000759 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000760 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000761 if (errcode == E_EOF) {
762 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000763 return E_EOF;
764 }
Guido van Rossum82598051997-03-05 00:20:32 +0000765 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000766 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000767 }
Guido van Rossum82598051997-03-05 00:20:32 +0000768 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000769 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000770 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000771 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000772 }
Guido van Rossum82598051997-03-05 00:20:32 +0000773 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000774 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000775 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000776 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000777 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000778 return -1;
779 }
Guido van Rossum82598051997-03-05 00:20:32 +0000780 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000781 if (Py_FlushLine())
782 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000783 return 0;
784}
785
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000786/* Check whether a file maybe a pyc file: Look at the extension,
787 the file type, and, if we may close it, at the first few bytes. */
788
789static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000790maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000791{
792 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
793 return 1;
794
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000795 /* Only look into the file if we are allowed to close it, since
796 it then should also be seekable. */
797 if (closeit) {
798 /* Read only two bytes of the magic. If the file was opened in
799 text mode, the bytes 3 and 4 of the magic (\r\n) might not
800 be read as they are on disk. */
801 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
802 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000803 /* Mess: In case of -x, the stream is NOT at its start now,
804 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000805 which makes the current stream position formally undefined,
806 and a x-platform nightmare.
807 Unfortunately, we have no direct way to know whether -x
808 was specified. So we use a terrible hack: if the current
809 stream position is not 0, we assume -x was specified, and
810 give up. Bug 132850 on SourceForge spells out the
811 hopelessness of trying anything else (fseek and ftell
812 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000813 */
Tim Peters3e876562001-02-11 04:35:39 +0000814 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000815 if (ftell(fp) == 0) {
816 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000817 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000818 ispyc = 1;
819 rewind(fp);
820 }
Tim Peters3e876562001-02-11 04:35:39 +0000821 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000822 }
823 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000824}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000825
Guido van Rossum0df002c2000-08-27 19:21:52 +0000826int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000827PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000828 PyCompilerFlags *flags)
829{
Guido van Rossum82598051997-03-05 00:20:32 +0000830 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000831 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000832
Guido van Rossum82598051997-03-05 00:20:32 +0000833 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000834 if (m == NULL)
835 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000836 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000837 if (PyDict_GetItemString(d, "__file__") == NULL) {
838 PyObject *f = PyString_FromString(filename);
839 if (f == NULL)
840 return -1;
841 if (PyDict_SetItemString(d, "__file__", f) < 0) {
842 Py_DECREF(f);
843 return -1;
844 }
845 Py_DECREF(f);
846 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000847 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000848 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000849 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000850 if (closeit)
851 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000852 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000853 fprintf(stderr, "python: Can't reopen .pyc file\n");
854 return -1;
855 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000856 /* Turn on optimization if a .pyo file is given */
857 if (strcmp(ext, ".pyo") == 0)
858 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000859 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000860 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000861 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000862 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000863 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000864 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000865 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000866 return -1;
867 }
Guido van Rossum82598051997-03-05 00:20:32 +0000868 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000869 if (Py_FlushLine())
870 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000871 return 0;
872}
873
874int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000875PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000876{
Guido van Rossum82598051997-03-05 00:20:32 +0000877 PyObject *m, *d, *v;
878 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000879 if (m == NULL)
880 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000881 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000882 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000883 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000884 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000885 return -1;
886 }
Guido van Rossum82598051997-03-05 00:20:32 +0000887 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000888 if (Py_FlushLine())
889 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000890 return 0;
891}
892
Barry Warsaw035574d1997-08-29 22:07:17 +0000893static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000894parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
895 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000896{
897 long hold;
898 PyObject *v;
899
900 /* old style errors */
901 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000902 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000903 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000904
905 /* new style errors. `err' is an instance */
906
907 if (! (v = PyObject_GetAttrString(err, "msg")))
908 goto finally;
909 *message = v;
910
911 if (!(v = PyObject_GetAttrString(err, "filename")))
912 goto finally;
913 if (v == Py_None)
914 *filename = NULL;
915 else if (! (*filename = PyString_AsString(v)))
916 goto finally;
917
918 Py_DECREF(v);
919 if (!(v = PyObject_GetAttrString(err, "lineno")))
920 goto finally;
921 hold = PyInt_AsLong(v);
922 Py_DECREF(v);
923 v = NULL;
924 if (hold < 0 && PyErr_Occurred())
925 goto finally;
926 *lineno = (int)hold;
927
928 if (!(v = PyObject_GetAttrString(err, "offset")))
929 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000930 if (v == Py_None) {
931 *offset = -1;
932 Py_DECREF(v);
933 v = NULL;
934 } else {
935 hold = PyInt_AsLong(v);
936 Py_DECREF(v);
937 v = NULL;
938 if (hold < 0 && PyErr_Occurred())
939 goto finally;
940 *offset = (int)hold;
941 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000942
943 if (!(v = PyObject_GetAttrString(err, "text")))
944 goto finally;
945 if (v == Py_None)
946 *text = NULL;
947 else if (! (*text = PyString_AsString(v)))
948 goto finally;
949 Py_DECREF(v);
950 return 1;
951
952finally:
953 Py_XDECREF(v);
954 return 0;
955}
956
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000957void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000958PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000959{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000960 PyErr_PrintEx(1);
961}
962
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000963static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000964print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000965{
966 char *nl;
967 if (offset >= 0) {
968 if (offset > 0 && offset == (int)strlen(text))
969 offset--;
970 for (;;) {
971 nl = strchr(text, '\n');
972 if (nl == NULL || nl-text >= offset)
973 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000974 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000975 text = nl+1;
976 }
977 while (*text == ' ' || *text == '\t') {
978 text++;
979 offset--;
980 }
981 }
982 PyFile_WriteString(" ", f);
983 PyFile_WriteString(text, f);
984 if (*text == '\0' || text[strlen(text)-1] != '\n')
985 PyFile_WriteString("\n", f);
986 if (offset == -1)
987 return;
988 PyFile_WriteString(" ", f);
989 offset--;
990 while (offset > 0) {
991 PyFile_WriteString(" ", f);
992 offset--;
993 }
994 PyFile_WriteString("^\n", f);
995}
996
Guido van Rossum66e8e862001-03-23 17:54:43 +0000997static void
998handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000999{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001000 PyObject *exception, *value, *tb;
1001 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001002
Guido van Rossum66e8e862001-03-23 17:54:43 +00001003 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001004 if (Py_FlushLine())
1005 PyErr_Clear();
1006 fflush(stdout);
1007 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001008 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001009 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001010 /* The error code should be in the `code' attribute. */
1011 PyObject *code = PyObject_GetAttrString(value, "code");
1012 if (code) {
1013 Py_DECREF(value);
1014 value = code;
1015 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001016 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001017 }
1018 /* If we failed to dig out the 'code' attribute,
1019 just let the else clause below print the error. */
1020 }
1021 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001022 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001023 else {
1024 PyObject_Print(value, stderr, Py_PRINT_RAW);
1025 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001026 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001027 }
Tim Peterscf615b52003-04-19 18:47:02 +00001028 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001029 /* Restore and clear the exception info, in order to properly decref
1030 * the exception, value, and traceback. If we just exit instead,
1031 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1032 * some finalizers from running.
1033 */
Tim Peterscf615b52003-04-19 18:47:02 +00001034 PyErr_Restore(exception, value, tb);
1035 PyErr_Clear();
1036 Py_Exit(exitcode);
1037 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001038}
1039
1040void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001041PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001042{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001043 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001044
1045 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1046 handle_system_exit();
1047 }
Guido van Rossum82598051997-03-05 00:20:32 +00001048 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001049 if (exception == NULL)
1050 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001051 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001052 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001053 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001054 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001055 if (set_sys_last_vars) {
1056 PySys_SetObject("last_type", exception);
1057 PySys_SetObject("last_value", v);
1058 PySys_SetObject("last_traceback", tb);
1059 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001060 hook = PySys_GetObject("excepthook");
1061 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001062 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001063 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001064 PyObject *result = PyEval_CallObject(hook, args);
1065 if (result == NULL) {
1066 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001067 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1068 handle_system_exit();
1069 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001070 PyErr_Fetch(&exception2, &v2, &tb2);
1071 PyErr_NormalizeException(&exception2, &v2, &tb2);
Neal Norwitza5e4f222006-07-17 00:59:04 +00001072 /* It should not be possible for exception2 or v2
1073 to be NULL. However PyErr_Display() can't
1074 tolerate NULLs, so just be safe. */
1075 if (exception2 == NULL) {
1076 exception2 = Py_None;
1077 Py_INCREF(exception2);
1078 }
1079 if (v2 == NULL) {
1080 v2 = Py_None;
1081 Py_INCREF(v2);
1082 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001083 if (Py_FlushLine())
1084 PyErr_Clear();
1085 fflush(stdout);
1086 PySys_WriteStderr("Error in sys.excepthook:\n");
1087 PyErr_Display(exception2, v2, tb2);
1088 PySys_WriteStderr("\nOriginal exception was:\n");
1089 PyErr_Display(exception, v, tb);
Neal Norwitza5e4f222006-07-17 00:59:04 +00001090 Py_DECREF(exception2);
1091 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001092 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001093 }
1094 Py_XDECREF(result);
1095 Py_XDECREF(args);
1096 } else {
1097 PySys_WriteStderr("sys.excepthook is missing\n");
1098 PyErr_Display(exception, v, tb);
1099 }
1100 Py_XDECREF(exception);
1101 Py_XDECREF(v);
1102 Py_XDECREF(tb);
1103}
1104
Richard Jones7b9558d2006-05-27 12:29:24 +00001105void
1106PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001107{
1108 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001109 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001110 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001111 if (f == NULL)
1112 fprintf(stderr, "lost sys.stderr\n");
1113 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001114 if (Py_FlushLine())
1115 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001116 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001117 if (tb && tb != Py_None)
1118 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001119 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001120 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001121 {
Guido van Rossum82598051997-03-05 00:20:32 +00001122 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001123 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001124 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001125 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001126 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001127 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001128 else {
1129 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001130 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001131 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001132 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001133 else
Guido van Rossum82598051997-03-05 00:20:32 +00001134 PyFile_WriteString(filename, f);
1135 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001136 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001137 PyFile_WriteString(buf, f);
1138 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001139 if (text != NULL)
1140 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001141 Py_DECREF(value);
1142 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001143 /* Can't be bothered to check all those
1144 PyFile_WriteString() calls */
1145 if (PyErr_Occurred())
1146 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001147 }
1148 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001149 if (err) {
1150 /* Don't do anything else */
1151 }
Brett Cannonbf364092006-03-01 04:25:17 +00001152 else if (PyExceptionClass_Check(exception)) {
Richard Jones7b9558d2006-05-27 12:29:24 +00001153 PyObject* moduleName;
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001154 char* className = PyExceptionClass_Name(exception);
1155 if (className != NULL) {
1156 char *dot = strrchr(className, '.');
1157 if (dot != NULL)
1158 className = dot+1;
1159 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001160
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001161 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001162 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001163 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001164 else {
1165 char* modstr = PyString_AsString(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001166 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001167 {
1168 err = PyFile_WriteString(modstr, f);
1169 err += PyFile_WriteString(".", f);
1170 }
Richard Jones7b9558d2006-05-27 12:29:24 +00001171 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001172 }
1173 if (err == 0) {
1174 if (className == NULL)
1175 err = PyFile_WriteString("<unknown>", f);
1176 else
Brett Cannonbf364092006-03-01 04:25:17 +00001177 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001178 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001179 }
1180 else
1181 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001182 if (err == 0 && (value != Py_None)) {
1183 PyObject *s = PyObject_Str(value);
1184 /* only print colon if the str() of the
1185 object is not the empty string
1186 */
1187 if (s == NULL)
1188 err = -1;
1189 else if (!PyString_Check(s) ||
1190 PyString_GET_SIZE(s) != 0)
1191 err = PyFile_WriteString(": ", f);
1192 if (err == 0)
1193 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1194 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001195 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001196 if (err == 0)
1197 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001198 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001199 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001200 /* If an error happened here, don't show it.
1201 XXX This is wrong, but too many callers rely on this behavior. */
1202 if (err != 0)
1203 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001204}
1205
Guido van Rossum82598051997-03-05 00:20:32 +00001206PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001207PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001208 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001209{
Neal Norwitze92fba02006-03-04 18:52:26 +00001210 PyObject *ret = NULL;
Neal Norwitzd12bd012006-07-21 07:59:47 +00001211 mod_ty mod;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001212 PyArena *arena = PyArena_New();
Neal Norwitzd12bd012006-07-21 07:59:47 +00001213 if (arena == NULL)
1214 return NULL;
1215
1216 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001217 if (mod != NULL)
1218 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001219 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001220 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001221}
1222
1223PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001224PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001225 PyObject *locals, int closeit, PyCompilerFlags *flags)
1226{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001227 PyObject *ret;
Neal Norwitzd12bd012006-07-21 07:59:47 +00001228 mod_ty mod;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001229 PyArena *arena = PyArena_New();
Neal Norwitzd12bd012006-07-21 07:59:47 +00001230 if (arena == NULL)
1231 return NULL;
1232
1233 mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
1234 flags, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001235 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001236 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001237 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001238 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001239 if (closeit)
1240 fclose(fp);
Neal Norwitze92fba02006-03-04 18:52:26 +00001241 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001242 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001243 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001244}
1245
Guido van Rossum82598051997-03-05 00:20:32 +00001246static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001247run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001248 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001249{
Guido van Rossum82598051997-03-05 00:20:32 +00001250 PyCodeObject *co;
1251 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001252 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001253 if (co == NULL)
1254 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001255 v = PyEval_EvalCode(co, globals, locals);
1256 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001257 return v;
1258}
1259
Guido van Rossum82598051997-03-05 00:20:32 +00001260static PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001261run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001262 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001263{
Guido van Rossum82598051997-03-05 00:20:32 +00001264 PyCodeObject *co;
1265 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001266 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001267 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001268
Guido van Rossum82598051997-03-05 00:20:32 +00001269 magic = PyMarshal_ReadLongFromFile(fp);
1270 if (magic != PyImport_GetMagicNumber()) {
1271 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001272 "Bad magic number in .pyc file");
1273 return NULL;
1274 }
Guido van Rossum82598051997-03-05 00:20:32 +00001275 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001276 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001277 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001278 if (v == NULL || !PyCode_Check(v)) {
1279 Py_XDECREF(v);
1280 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001281 "Bad code object in .pyc file");
1282 return NULL;
1283 }
Guido van Rossum82598051997-03-05 00:20:32 +00001284 co = (PyCodeObject *)v;
1285 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001286 if (v && flags)
1287 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001288 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001289 return v;
1290}
1291
Guido van Rossum82598051997-03-05 00:20:32 +00001292PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001293Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001294 PyCompilerFlags *flags)
1295{
Guido van Rossum82598051997-03-05 00:20:32 +00001296 PyCodeObject *co;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001297 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001298 PyArena *arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001299 if (arena == NULL)
1300 return NULL;
1301
1302 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001303 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001304 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001305 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001306 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001307 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001308 PyObject *result = PyAST_mod2obj(mod);
1309 PyArena_Free(arena);
1310 return result;
1311 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001312 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001313 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001314 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001315}
1316
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001317struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001318Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001319{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001320 struct symtable *st;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001321 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001322 PyArena *arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001323 if (arena == NULL)
1324 return NULL;
1325
1326 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001327 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001328 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001329 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001330 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001331 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001332 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001333 return st;
1334}
1335
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001336/* Preferred access to parser is through AST. */
1337mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001338PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001339 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001340{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001341 mod_ty mod;
1342 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001343 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001344 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001345 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001346 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001347 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001348 PyNode_Free(n);
1349 return mod;
1350 }
1351 else {
1352 err_input(&err);
1353 return NULL;
1354 }
1355}
1356
1357mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001358PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001359 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001360 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001361{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001362 mod_ty mod;
1363 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001364 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1365 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001366 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001367 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001368 PyNode_Free(n);
1369 return mod;
1370 }
1371 else {
1372 err_input(&err);
1373 if (errcode)
1374 *errcode = err.error;
1375 return NULL;
1376 }
1377}
1378
Guido van Rossuma110aa61994-08-29 12:50:44 +00001379/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001380
Guido van Rossuma110aa61994-08-29 12:50:44 +00001381node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001382PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001383{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001384 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001385 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1386 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001387 if (n == NULL)
1388 err_input(&err);
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001389
Guido van Rossuma110aa61994-08-29 12:50:44 +00001390 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001391}
1392
Guido van Rossuma110aa61994-08-29 12:50:44 +00001393/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001394
Guido van Rossuma110aa61994-08-29 12:50:44 +00001395node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001396PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001397{
Tim Petersfe2127d2001-07-16 05:37:24 +00001398 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001399 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1400 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001401 if (n == NULL)
1402 err_input(&err);
1403 return n;
1404}
1405
1406node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001407PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001408 int start, int flags)
1409{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001410 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001411 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1412 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001413 if (n == NULL)
1414 err_input(&err);
1415 return n;
1416}
1417
1418node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001419PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001420{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001421 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001422}
1423
Guido van Rossum66ebd912003-04-17 16:02:26 +00001424/* May want to move a more generalized form of this to parsetok.c or
1425 even parser modules. */
1426
1427void
1428PyParser_SetError(perrdetail *err)
1429{
1430 err_input(err);
1431}
1432
Guido van Rossuma110aa61994-08-29 12:50:44 +00001433/* Set the error appropriate to the given input error code (see errcode.h) */
1434
1435static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001436err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001437{
Fred Drake85f36392000-07-11 17:53:00 +00001438 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001439 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001440 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001441 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001442 switch (err->error) {
1443 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001444 errtype = PyExc_IndentationError;
1445 if (err->expected == INDENT)
1446 msg = "expected an indented block";
1447 else if (err->token == INDENT)
1448 msg = "unexpected indent";
1449 else if (err->token == DEDENT)
1450 msg = "unexpected unindent";
1451 else {
1452 errtype = PyExc_SyntaxError;
1453 msg = "invalid syntax";
1454 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001455 break;
1456 case E_TOKEN:
1457 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001458 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001459 case E_EOFS:
1460 msg = "EOF while scanning triple-quoted string";
1461 break;
1462 case E_EOLS:
1463 msg = "EOL while scanning single-quoted string";
1464 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001465 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001466 if (!PyErr_Occurred())
1467 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001468 return;
1469 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001470 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001471 return;
1472 case E_EOF:
1473 msg = "unexpected EOF while parsing";
1474 break;
Fred Drake85f36392000-07-11 17:53:00 +00001475 case E_TABSPACE:
1476 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001477 msg = "inconsistent use of tabs and spaces in indentation";
1478 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001479 case E_OVERFLOW:
1480 msg = "expression too long";
1481 break;
Fred Drake85f36392000-07-11 17:53:00 +00001482 case E_DEDENT:
1483 errtype = PyExc_IndentationError;
1484 msg = "unindent does not match any outer indentation level";
1485 break;
1486 case E_TOODEEP:
1487 errtype = PyExc_IndentationError;
1488 msg = "too many levels of indentation";
1489 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001490 case E_DECODE: {
1491 PyObject *type, *value, *tb;
1492 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001493 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001494 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001495 if (u != NULL) {
1496 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001497 }
1498 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001499 if (msg == NULL)
1500 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001501 Py_XDECREF(type);
1502 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001503 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001504 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001505 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001506 case E_LINECONT:
1507 msg = "unexpected character after line continuation character";
1508 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001509 default:
1510 fprintf(stderr, "error=%d\n", err->error);
1511 msg = "unknown parsing error";
1512 break;
1513 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001514 v = Py_BuildValue("(ziiz)", err->filename,
1515 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001516 if (err->text != NULL) {
Tim Petersc9d78aa2006-03-26 23:27:58 +00001517 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001518 err->text = NULL;
1519 }
1520 w = NULL;
1521 if (v != NULL)
1522 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001523 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001524 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001525 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001526 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001527}
1528
1529/* Print fatal error message and abort */
1530
1531void
Tim Peters7c321a82002-07-09 02:57:01 +00001532Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001533{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001534 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001535#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001536 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001537 OutputDebugString(msg);
1538 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001539#ifdef _DEBUG
1540 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001541#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001542#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001543 abort();
1544}
1545
1546/* Clean up and exit */
1547
Guido van Rossuma110aa61994-08-29 12:50:44 +00001548#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001549#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001550#endif
1551
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001552#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001553static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001554static int nexitfuncs = 0;
1555
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001556int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001557{
1558 if (nexitfuncs >= NEXITFUNCS)
1559 return -1;
1560 exitfuncs[nexitfuncs++] = func;
1561 return 0;
1562}
1563
Guido van Rossumcc283f51997-08-05 02:22:03 +00001564static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001565call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001566{
Guido van Rossum82598051997-03-05 00:20:32 +00001567 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001568
1569 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001570 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001571 Py_INCREF(exitfunc);
1572 PySys_SetObject("exitfunc", (PyObject *)NULL);
1573 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001574 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001575 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1576 PySys_WriteStderr("Error in sys.exitfunc:\n");
1577 }
Guido van Rossum82598051997-03-05 00:20:32 +00001578 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001579 }
Guido van Rossum82598051997-03-05 00:20:32 +00001580 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001581 }
1582
Guido van Rossum0829c751998-02-28 04:31:39 +00001583 if (Py_FlushLine())
1584 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001585}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001586
Guido van Rossumcc283f51997-08-05 02:22:03 +00001587static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001588call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001589{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001590 while (nexitfuncs > 0)
1591 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001592
1593 fflush(stdout);
1594 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001595}
1596
1597void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001598Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001599{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001600 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001601
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001602 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001603}
1604
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001605static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001606initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001607{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001608#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001609 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001610#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001611#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001612 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001613#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001614#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001615 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001616#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001617 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001618}
1619
Guido van Rossum7433b121997-02-14 19:45:36 +00001620
1621/*
1622 * The file descriptor fd is considered ``interactive'' if either
1623 * a) isatty(fd) is TRUE, or
1624 * b) the -i flag was given, and the filename associated with
1625 * the descriptor is NULL or "<stdin>" or "???".
1626 */
1627int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001628Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001629{
1630 if (isatty((int)fileno(fp)))
1631 return 1;
1632 if (!Py_InteractiveFlag)
1633 return 0;
1634 return (filename == NULL) ||
1635 (strcmp(filename, "<stdin>") == 0) ||
1636 (strcmp(filename, "???") == 0);
1637}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001638
1639
Tim Petersd08e3822003-04-17 15:24:21 +00001640#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001641#if defined(WIN32) && defined(_MSC_VER)
1642
1643/* Stack checking for Microsoft C */
1644
1645#include <malloc.h>
1646#include <excpt.h>
1647
Fred Drakee8de31c2000-08-31 05:38:39 +00001648/*
1649 * Return non-zero when we run out of memory on the stack; zero otherwise.
1650 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001651int
Fred Drake399739f2000-08-31 05:52:44 +00001652PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001653{
1654 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001655 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001656 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001657 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001658 return 0;
1659 } __except (EXCEPTION_EXECUTE_HANDLER) {
1660 /* just ignore all errors */
1661 }
1662 return 1;
1663}
1664
1665#endif /* WIN32 && _MSC_VER */
1666
1667/* Alternate implementations can be added here... */
1668
1669#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001670
1671
1672/* Wrappers around sigaction() or signal(). */
1673
1674PyOS_sighandler_t
1675PyOS_getsig(int sig)
1676{
1677#ifdef HAVE_SIGACTION
1678 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001679 if (sigaction(sig, NULL, &context) == -1)
1680 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001681 return context.sa_handler;
1682#else
1683 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001684/* Special signal handling for the secure CRT in Visual Studio 2005 */
1685#if defined(_MSC_VER) && _MSC_VER >= 1400
1686 switch (sig) {
1687 /* Only these signals are valid */
1688 case SIGINT:
1689 case SIGILL:
1690 case SIGFPE:
1691 case SIGSEGV:
1692 case SIGTERM:
1693 case SIGBREAK:
1694 case SIGABRT:
1695 break;
1696 /* Don't call signal() with other values or it will assert */
1697 default:
1698 return SIG_ERR;
1699 }
1700#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001701 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001702 if (handler != SIG_ERR)
1703 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001704 return handler;
1705#endif
1706}
1707
1708PyOS_sighandler_t
1709PyOS_setsig(int sig, PyOS_sighandler_t handler)
1710{
1711#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001712 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001713 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001714 sigemptyset(&context.sa_mask);
1715 context.sa_flags = 0;
1716 if (sigaction(sig, &context, &ocontext) == -1)
1717 return SIG_ERR;
1718 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001719#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001720 PyOS_sighandler_t oldhandler;
1721 oldhandler = signal(sig, handler);
1722#ifdef HAVE_SIGINTERRUPT
1723 siginterrupt(sig, 1);
1724#endif
1725 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001726#endif
1727}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001728
1729/* Deprecated C API functions still provided for binary compatiblity */
1730
1731#undef PyParser_SimpleParseFile
Thomas Heller1b046642006-04-18 18:51:06 +00001732PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001733PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1734{
1735 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1736}
1737
Thomas Heller1b046642006-04-18 18:51:06 +00001738#undef PyParser_SimpleParseString
1739PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001740PyParser_SimpleParseString(const char *str, int start)
1741{
1742 return PyParser_SimpleParseStringFlags(str, start, 0);
1743}
Anthony Baxterac6bd462006-04-13 02:06:09 +00001744
Thomas Heller1b046642006-04-18 18:51:06 +00001745#undef PyRun_AnyFile
1746PyAPI_FUNC(int)
1747PyRun_AnyFile(FILE *fp, const char *name)
1748{
1749 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1750}
1751
1752#undef PyRun_AnyFileEx
1753PyAPI_FUNC(int)
1754PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1755{
1756 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1757}
1758
1759#undef PyRun_AnyFileFlags
1760PyAPI_FUNC(int)
1761PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1762{
1763 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1764}
1765
1766#undef PyRun_File
1767PyAPI_FUNC(PyObject *)
1768PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1769{
1770 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1771}
1772
1773#undef PyRun_FileEx
1774PyAPI_FUNC(PyObject *)
1775PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1776{
1777 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1778}
1779
1780#undef PyRun_FileFlags
1781PyAPI_FUNC(PyObject *)
1782PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1783 PyCompilerFlags *flags)
1784{
1785 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1786}
1787
1788#undef PyRun_SimpleFile
1789PyAPI_FUNC(int)
1790PyRun_SimpleFile(FILE *f, const char *p)
1791{
1792 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1793}
1794
1795#undef PyRun_SimpleFileEx
1796PyAPI_FUNC(int)
1797PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1798{
1799 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1800}
1801
1802
1803#undef PyRun_String
1804PyAPI_FUNC(PyObject *)
1805PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1806{
1807 return PyRun_StringFlags(str, s, g, l, NULL);
1808}
1809
1810#undef PyRun_SimpleString
1811PyAPI_FUNC(int)
1812PyRun_SimpleString(const char *s)
1813{
1814 return PyRun_SimpleStringFlags(s, NULL);
1815}
1816
1817#undef Py_CompileString
1818PyAPI_FUNC(PyObject *)
1819Py_CompileString(const char *str, const char *p, int s)
1820{
1821 return Py_CompileStringFlags(str, p, s, NULL);
1822}
1823
1824#undef PyRun_InteractiveOne
1825PyAPI_FUNC(int)
1826PyRun_InteractiveOne(FILE *f, const char *p)
1827{
1828 return PyRun_InteractiveOneFlags(f, p, NULL);
1829}
1830
1831#undef PyRun_InteractiveLoop
1832PyAPI_FUNC(int)
1833PyRun_InteractiveLoop(FILE *f, const char *p)
1834{
1835 return PyRun_InteractiveLoopFlags(f, p, NULL);
1836}
1837
Anthony Baxterac6bd462006-04-13 02:06:09 +00001838#ifdef __cplusplus
1839}
1840#endif
1841