blob: e83c4cb4233a49a68fdccada1e69abe703ad32c2 [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();
Thomas Wouterse4713172007-01-23 13:54:30 +0000278 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
279 PyFile_Check(sys_stream)) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000280 if (!PyFile_SetEncoding(sys_stream, codeset))
281 Py_FatalError("Cannot set codeset of stdin");
282 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000283 Py_XDECREF(sys_isatty);
284
285 sys_stream = PySys_GetObject("stdout");
286 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
287 if (!sys_isatty)
288 PyErr_Clear();
Thomas Wouterse4713172007-01-23 13:54:30 +0000289 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
290 PyFile_Check(sys_stream)) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000291 if (!PyFile_SetEncoding(sys_stream, codeset))
292 Py_FatalError("Cannot set codeset of stdout");
293 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000294 Py_XDECREF(sys_isatty);
295
Martin v. Löwisea62d252006-04-03 10:56:49 +0000296 sys_stream = PySys_GetObject("stderr");
297 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
298 if (!sys_isatty)
299 PyErr_Clear();
Thomas Wouterse4713172007-01-23 13:54:30 +0000300 if(sys_isatty && PyObject_IsTrue(sys_isatty) &&
301 PyFile_Check(sys_stream)) {
Martin v. Löwisea62d252006-04-03 10:56:49 +0000302 if (!PyFile_SetEncoding(sys_stream, codeset))
303 Py_FatalError("Cannot set codeset of stderr");
304 }
305 Py_XDECREF(sys_isatty);
306
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000307 if (!Py_FileSystemDefaultEncoding)
308 Py_FileSystemDefaultEncoding = codeset;
309 else
310 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000311 }
312#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000313}
314
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000315void
316Py_Initialize(void)
317{
318 Py_InitializeEx(1);
319}
320
321
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000322#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000323extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000324#endif
325
Guido van Rossum25ce5661997-08-02 03:10:38 +0000326/* Undo the effect of Py_Initialize().
327
328 Beware: if multiple interpreter and/or thread states exist, these
329 are not wiped out; only the current thread and interpreter state
330 are deleted. But since everything else is deleted, those other
331 interpreter and thread states should no longer be used.
332
333 (XXX We should do better, e.g. wipe out all interpreters and
334 threads.)
335
336 Locking: as above.
337
338*/
339
340void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000341Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000342{
343 PyInterpreterState *interp;
344 PyThreadState *tstate;
345
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000346 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000347 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000348
Tim Peters384fd102001-01-21 03:40:37 +0000349 /* The interpreter is still entirely intact at this point, and the
350 * exit funcs may be relying on that. In particular, if some thread
351 * or exit func is still waiting to do an import, the import machinery
352 * expects Py_IsInitialized() to return true. So don't say the
353 * interpreter is uninitialized until after the exit funcs have run.
354 * Note that Threading.py uses an exit func to do a join on all the
355 * threads created thru it, so this also protects pending imports in
356 * the threads created via Threading.
357 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000358 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000359 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000360
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000361 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000362 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000363 interp = tstate->interp;
364
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000365 /* Disable signal handling */
366 PyOS_FiniInterrupts();
367
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000368 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000369 Py_XDECREF(warnings_module);
370 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000371
Guido van Rossume13ddc92003-04-17 17:29:22 +0000372 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000373 * before all modules are destroyed.
374 * XXX If a __del__ or weakref callback is triggered here, and tries to
375 * XXX import a module, bad things can happen, because Python no
376 * XXX longer believes it's initialized.
377 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
378 * XXX is easy to provoke that way. I've also seen, e.g.,
379 * XXX Exception exceptions.ImportError: 'No module named sha'
380 * XXX in <function callback at 0x008F5718> ignored
381 * XXX but I'm unclear on exactly how that one happens. In any case,
382 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000383 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000384 PyGC_Collect();
Martin v. Löwis45294a92006-04-18 06:24:08 +0000385#ifdef COUNT_ALLOCS
386 /* With COUNT_ALLOCS, it helps to run GC multiple times:
387 each collection might release some types from the type
388 list, so they become garbage. */
389 while (PyGC_Collect() > 0)
390 /* nothing */;
391#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000392
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000393 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000394 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000395
Guido van Rossume13ddc92003-04-17 17:29:22 +0000396 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000397 * new-style class definitions, for example.
398 * XXX This is disabled because it caused too many problems. If
399 * XXX a __del__ or weakref callback triggers here, Python code has
400 * XXX a hard time running, because even the sys module has been
401 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
402 * XXX One symptom is a sequence of information-free messages
403 * XXX coming from threads (if a __del__ or callback is invoked,
404 * XXX other threads can execute too, and any exception they encounter
405 * XXX triggers a comedy of errors as subsystem after subsystem
406 * XXX fails to find what it *expects* to find in sys to help report
407 * XXX the exception and consequent unexpected failures). I've also
408 * XXX seen segfaults then, after adding print statements to the
409 * XXX Python code getting called.
410 */
411#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000412 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000413#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000414
Guido van Rossum1707aad1997-12-08 23:43:45 +0000415 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
416 _PyImport_Fini();
417
418 /* Debugging stuff */
419#ifdef COUNT_ALLOCS
Martin v. Löwis45294a92006-04-18 06:24:08 +0000420 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000421#endif
422
Tim Peters62e97f02006-03-28 21:44:32 +0000423 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000424
Tim Peters9cf25ce2003-04-17 15:21:01 +0000425#ifdef Py_TRACE_REFS
426 /* Display all objects still alive -- this can invoke arbitrary
427 * __repr__ overrides, so requires a mostly-intact interpreter.
428 * Alas, a lot of stuff may still be alive now that will be cleaned
429 * up later.
430 */
Tim Peters269b2a62003-04-17 19:52:29 +0000431 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000432 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000433#endif /* Py_TRACE_REFS */
434
Mark Hammond6cb90292003-04-22 11:18:00 +0000435 /* Cleanup auto-thread-state */
436#ifdef WITH_THREAD
437 _PyGILState_Fini();
438#endif /* WITH_THREAD */
439
Guido van Rossumd922fa42003-04-15 14:10:09 +0000440 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000441 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000442
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000443 /* Now we decref the exception classes. After this point nothing
444 can raise an exception. That's okay, because each Fini() method
445 below has been checked to make sure no exceptions are ever
446 raised.
447 */
448
449 _PyExc_Fini();
450
Guido van Rossumd922fa42003-04-15 14:10:09 +0000451 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000452 PyThreadState_Swap(NULL);
453 PyInterpreterState_Delete(interp);
454
Guido van Rossumd922fa42003-04-15 14:10:09 +0000455 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000456 PyMethod_Fini();
457 PyFrame_Fini();
458 PyCFunction_Fini();
459 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000460 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000461 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000462 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000463 PyInt_Fini();
464 PyFloat_Fini();
465
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000466#ifdef Py_USING_UNICODE
467 /* Cleanup Unicode implementation */
468 _PyUnicode_Fini();
469#endif
470
Guido van Rossumcc283f51997-08-05 02:22:03 +0000471 /* XXX Still allocated:
472 - various static ad-hoc pointers to interned strings
473 - int and float free list blocks
474 - whatever various modules and libraries allocate
475 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000476
477 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000478
Tim Peters269b2a62003-04-17 19:52:29 +0000479#ifdef Py_TRACE_REFS
480 /* Display addresses (& refcnts) of all objects still alive.
481 * An address can be used to find the repr of the object, printed
482 * above by _Py_PrintReferences.
483 */
484 if (Py_GETENV("PYTHONDUMPREFS"))
485 _Py_PrintReferenceAddresses(stderr);
486#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000487#ifdef PYMALLOC_DEBUG
488 if (Py_GETENV("PYTHONMALLOCSTATS"))
489 _PyObject_DebugMallocStats();
490#endif
491
Guido van Rossumcc283f51997-08-05 02:22:03 +0000492 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000493}
494
495/* Create and initialize a new interpreter and thread, and return the
496 new thread. This requires that Py_Initialize() has been called
497 first.
498
499 Unsuccessful initialization yields a NULL pointer. Note that *no*
500 exception information is available even in this case -- the
501 exception information is held in the thread, and there is no
502 thread.
503
504 Locking: as above.
505
506*/
507
508PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000509Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000510{
511 PyInterpreterState *interp;
512 PyThreadState *tstate, *save_tstate;
513 PyObject *bimod, *sysmod;
514
515 if (!initialized)
516 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
517
518 interp = PyInterpreterState_New();
519 if (interp == NULL)
520 return NULL;
521
522 tstate = PyThreadState_New(interp);
523 if (tstate == NULL) {
524 PyInterpreterState_Delete(interp);
525 return NULL;
526 }
527
528 save_tstate = PyThreadState_Swap(tstate);
529
530 /* XXX The following is lax in error checking */
531
532 interp->modules = PyDict_New();
533
534 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
535 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000536 interp->builtins = PyModule_GetDict(bimod);
Neal Norwitzbebdc9e2006-08-21 20:20:59 +0000537 if (interp->builtins == NULL)
538 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000539 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000540 }
541 sysmod = _PyImport_FindExtension("sys", "sys");
542 if (bimod != NULL && sysmod != NULL) {
543 interp->sysdict = PyModule_GetDict(sysmod);
Neal Norwitzbebdc9e2006-08-21 20:20:59 +0000544 if (interp->sysdict == NULL)
545 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000546 Py_INCREF(interp->sysdict);
547 PySys_SetPath(Py_GetPath());
548 PyDict_SetItemString(interp->sysdict, "modules",
549 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000550 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000551 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000552 if (!Py_NoSiteFlag)
553 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000554 }
555
556 if (!PyErr_Occurred())
557 return tstate;
558
Neal Norwitzbebdc9e2006-08-21 20:20:59 +0000559handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000560 /* Oops, it didn't work. Undo it all. */
561
562 PyErr_Print();
563 PyThreadState_Clear(tstate);
564 PyThreadState_Swap(save_tstate);
565 PyThreadState_Delete(tstate);
566 PyInterpreterState_Delete(interp);
567
568 return NULL;
569}
570
571/* Delete an interpreter and its last thread. This requires that the
572 given thread state is current, that the thread has no remaining
573 frames, and that it is its interpreter's only remaining thread.
574 It is a fatal error to violate these constraints.
575
576 (Py_Finalize() doesn't have these constraints -- it zaps
577 everything, regardless.)
578
579 Locking: as above.
580
581*/
582
583void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000584Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000585{
586 PyInterpreterState *interp = tstate->interp;
587
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000588 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000589 Py_FatalError("Py_EndInterpreter: thread is not current");
590 if (tstate->frame != NULL)
591 Py_FatalError("Py_EndInterpreter: thread still has a frame");
592 if (tstate != interp->tstate_head || tstate->next != NULL)
593 Py_FatalError("Py_EndInterpreter: not the last thread");
594
595 PyImport_Cleanup();
596 PyInterpreterState_Clear(interp);
597 PyThreadState_Swap(NULL);
598 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000599}
600
601static char *progname = "python";
602
603void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000604Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000605{
606 if (pn && *pn)
607 progname = pn;
608}
609
610char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000611Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000612{
613 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000614}
615
Guido van Rossuma61691e1998-02-06 22:27:24 +0000616static char *default_home = NULL;
617
618void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000619Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000620{
621 default_home = home;
622}
623
624char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000625Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000626{
627 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000628 if (home == NULL && !Py_IgnoreEnvironmentFlag)
629 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000630 return home;
631}
632
Guido van Rossum6135a871995-01-09 17:53:26 +0000633/* Create __main__ module */
634
635static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000636initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000637{
Guido van Rossum82598051997-03-05 00:20:32 +0000638 PyObject *m, *d;
639 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000640 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000641 Py_FatalError("can't create __main__ module");
642 d = PyModule_GetDict(m);
643 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000644 PyObject *bimod = PyImport_ImportModule("__builtin__");
645 if (bimod == NULL ||
646 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000647 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000648 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000649 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000650}
651
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000652/* Import the site module (not into __main__ though) */
653
654static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000655initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000656{
657 PyObject *m, *f;
658 m = PyImport_ImportModule("site");
659 if (m == NULL) {
660 f = PySys_GetObject("stderr");
661 if (Py_VerboseFlag) {
662 PyFile_WriteString(
663 "'import site' failed; traceback:\n", f);
664 PyErr_Print();
665 }
666 else {
667 PyFile_WriteString(
668 "'import site' failed; use -v for traceback\n", f);
669 PyErr_Clear();
670 }
671 }
672 else {
673 Py_DECREF(m);
674 }
675}
676
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000677/* Parse input from a file and execute it */
678
679int
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000680PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000681 PyCompilerFlags *flags)
682{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000683 if (filename == NULL)
684 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000685 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000686 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000687 if (closeit)
688 fclose(fp);
689 return err;
690 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000691 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000692 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000693}
694
695int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000696PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000697{
Guido van Rossum82598051997-03-05 00:20:32 +0000698 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000699 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000700 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000701
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000702 if (flags == NULL) {
703 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000704 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000705 }
Guido van Rossum82598051997-03-05 00:20:32 +0000706 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000707 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000708 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
709 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000710 }
Guido van Rossum82598051997-03-05 00:20:32 +0000711 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000712 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000713 PySys_SetObject("ps2", v = PyString_FromString("... "));
714 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000715 }
716 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000717 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Tim Peters62e97f02006-03-28 21:44:32 +0000718 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000719 if (ret == E_EOF)
720 return 0;
721 /*
722 if (ret == E_NOMEM)
723 return -1;
724 */
725 }
726}
727
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000728/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000729#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000730 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
731 PyPARSE_DONT_IMPLY_DEDENT : 0) \
Neal Norwitz9589ee22006-03-04 19:01:22 +0000732 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
733 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000734
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000735int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000736PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000737{
Guido van Rossum82598051997-03-05 00:20:32 +0000738 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000739 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000740 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000741 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000742 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000743
Guido van Rossum82598051997-03-05 00:20:32 +0000744 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000745 if (v != NULL) {
746 v = PyObject_Str(v);
747 if (v == NULL)
748 PyErr_Clear();
749 else if (PyString_Check(v))
750 ps1 = PyString_AsString(v);
751 }
Guido van Rossum82598051997-03-05 00:20:32 +0000752 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000753 if (w != NULL) {
754 w = PyObject_Str(w);
755 if (w == NULL)
756 PyErr_Clear();
757 else if (PyString_Check(w))
758 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000759 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000760 arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +0000761 if (arena == NULL) {
762 Py_XDECREF(v);
763 Py_XDECREF(w);
764 return -1;
765 }
Tim Peters5e9d6cf2006-05-28 10:41:29 +0000766 mod = PyParser_ASTFromFile(fp, filename,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000767 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000768 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000769 Py_XDECREF(v);
770 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000771 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000772 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000773 if (errcode == E_EOF) {
774 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000775 return E_EOF;
776 }
Guido van Rossum82598051997-03-05 00:20:32 +0000777 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000778 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000779 }
Guido van Rossum82598051997-03-05 00:20:32 +0000780 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000781 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000782 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000783 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000784 }
Guido van Rossum82598051997-03-05 00:20:32 +0000785 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000786 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000787 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000788 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000789 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000790 return -1;
791 }
Guido van Rossum82598051997-03-05 00:20:32 +0000792 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000793 if (Py_FlushLine())
794 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000795 return 0;
796}
797
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000798/* Check whether a file maybe a pyc file: Look at the extension,
799 the file type, and, if we may close it, at the first few bytes. */
800
801static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000802maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000803{
804 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
805 return 1;
806
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000807 /* Only look into the file if we are allowed to close it, since
808 it then should also be seekable. */
809 if (closeit) {
810 /* Read only two bytes of the magic. If the file was opened in
811 text mode, the bytes 3 and 4 of the magic (\r\n) might not
812 be read as they are on disk. */
813 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
814 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000815 /* Mess: In case of -x, the stream is NOT at its start now,
816 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000817 which makes the current stream position formally undefined,
818 and a x-platform nightmare.
819 Unfortunately, we have no direct way to know whether -x
820 was specified. So we use a terrible hack: if the current
821 stream position is not 0, we assume -x was specified, and
822 give up. Bug 132850 on SourceForge spells out the
823 hopelessness of trying anything else (fseek and ftell
824 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000825 */
Tim Peters3e876562001-02-11 04:35:39 +0000826 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000827 if (ftell(fp) == 0) {
828 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000829 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000830 ispyc = 1;
831 rewind(fp);
832 }
Tim Peters3e876562001-02-11 04:35:39 +0000833 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000834 }
835 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000836}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000837
Guido van Rossum0df002c2000-08-27 19:21:52 +0000838int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000839PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000840 PyCompilerFlags *flags)
841{
Guido van Rossum82598051997-03-05 00:20:32 +0000842 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000843 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000844
Guido van Rossum82598051997-03-05 00:20:32 +0000845 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000846 if (m == NULL)
847 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000848 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000849 if (PyDict_GetItemString(d, "__file__") == NULL) {
850 PyObject *f = PyString_FromString(filename);
851 if (f == NULL)
852 return -1;
853 if (PyDict_SetItemString(d, "__file__", f) < 0) {
854 Py_DECREF(f);
855 return -1;
856 }
857 Py_DECREF(f);
858 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000859 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000860 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000861 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000862 if (closeit)
863 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000864 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000865 fprintf(stderr, "python: Can't reopen .pyc file\n");
866 return -1;
867 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000868 /* Turn on optimization if a .pyo file is given */
869 if (strcmp(ext, ".pyo") == 0)
870 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000871 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000872 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000873 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000874 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000875 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000876 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000877 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000878 return -1;
879 }
Guido van Rossum82598051997-03-05 00:20:32 +0000880 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000881 if (Py_FlushLine())
882 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000883 return 0;
884}
885
886int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000887PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000888{
Guido van Rossum82598051997-03-05 00:20:32 +0000889 PyObject *m, *d, *v;
890 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000891 if (m == NULL)
892 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000893 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000894 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000895 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000896 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000897 return -1;
898 }
Guido van Rossum82598051997-03-05 00:20:32 +0000899 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000900 if (Py_FlushLine())
901 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000902 return 0;
903}
904
Barry Warsaw035574d1997-08-29 22:07:17 +0000905static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000906parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
907 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000908{
909 long hold;
910 PyObject *v;
911
912 /* old style errors */
913 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000914 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000915 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000916
917 /* new style errors. `err' is an instance */
918
919 if (! (v = PyObject_GetAttrString(err, "msg")))
920 goto finally;
921 *message = v;
922
923 if (!(v = PyObject_GetAttrString(err, "filename")))
924 goto finally;
925 if (v == Py_None)
926 *filename = NULL;
927 else if (! (*filename = PyString_AsString(v)))
928 goto finally;
929
930 Py_DECREF(v);
931 if (!(v = PyObject_GetAttrString(err, "lineno")))
932 goto finally;
933 hold = PyInt_AsLong(v);
934 Py_DECREF(v);
935 v = NULL;
936 if (hold < 0 && PyErr_Occurred())
937 goto finally;
938 *lineno = (int)hold;
939
940 if (!(v = PyObject_GetAttrString(err, "offset")))
941 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000942 if (v == Py_None) {
943 *offset = -1;
944 Py_DECREF(v);
945 v = NULL;
946 } else {
947 hold = PyInt_AsLong(v);
948 Py_DECREF(v);
949 v = NULL;
950 if (hold < 0 && PyErr_Occurred())
951 goto finally;
952 *offset = (int)hold;
953 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000954
955 if (!(v = PyObject_GetAttrString(err, "text")))
956 goto finally;
957 if (v == Py_None)
958 *text = NULL;
959 else if (! (*text = PyString_AsString(v)))
960 goto finally;
961 Py_DECREF(v);
962 return 1;
963
964finally:
965 Py_XDECREF(v);
966 return 0;
967}
968
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000969void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000970PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000971{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000972 PyErr_PrintEx(1);
973}
974
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000975static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000976print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000977{
978 char *nl;
979 if (offset >= 0) {
980 if (offset > 0 && offset == (int)strlen(text))
981 offset--;
982 for (;;) {
983 nl = strchr(text, '\n');
984 if (nl == NULL || nl-text >= offset)
985 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000986 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000987 text = nl+1;
988 }
989 while (*text == ' ' || *text == '\t') {
990 text++;
991 offset--;
992 }
993 }
994 PyFile_WriteString(" ", f);
995 PyFile_WriteString(text, f);
996 if (*text == '\0' || text[strlen(text)-1] != '\n')
997 PyFile_WriteString("\n", f);
998 if (offset == -1)
999 return;
1000 PyFile_WriteString(" ", f);
1001 offset--;
1002 while (offset > 0) {
1003 PyFile_WriteString(" ", f);
1004 offset--;
1005 }
1006 PyFile_WriteString("^\n", f);
1007}
1008
Guido van Rossum66e8e862001-03-23 17:54:43 +00001009static void
1010handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001011{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001012 PyObject *exception, *value, *tb;
1013 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001014
Guido van Rossum66e8e862001-03-23 17:54:43 +00001015 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001016 if (Py_FlushLine())
1017 PyErr_Clear();
1018 fflush(stdout);
1019 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001020 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001021 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001022 /* The error code should be in the `code' attribute. */
1023 PyObject *code = PyObject_GetAttrString(value, "code");
1024 if (code) {
1025 Py_DECREF(value);
1026 value = code;
1027 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001028 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001029 }
1030 /* If we failed to dig out the 'code' attribute,
1031 just let the else clause below print the error. */
1032 }
1033 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001034 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001035 else {
1036 PyObject_Print(value, stderr, Py_PRINT_RAW);
1037 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001038 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001039 }
Tim Peterscf615b52003-04-19 18:47:02 +00001040 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001041 /* Restore and clear the exception info, in order to properly decref
1042 * the exception, value, and traceback. If we just exit instead,
1043 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1044 * some finalizers from running.
1045 */
Tim Peterscf615b52003-04-19 18:47:02 +00001046 PyErr_Restore(exception, value, tb);
1047 PyErr_Clear();
1048 Py_Exit(exitcode);
1049 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001050}
1051
1052void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001053PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001054{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001055 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001056
1057 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1058 handle_system_exit();
1059 }
Guido van Rossum82598051997-03-05 00:20:32 +00001060 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001061 if (exception == NULL)
1062 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001063 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001064 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001065 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001066 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001067 if (set_sys_last_vars) {
1068 PySys_SetObject("last_type", exception);
1069 PySys_SetObject("last_value", v);
1070 PySys_SetObject("last_traceback", tb);
1071 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001072 hook = PySys_GetObject("excepthook");
1073 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001074 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001075 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001076 PyObject *result = PyEval_CallObject(hook, args);
1077 if (result == NULL) {
1078 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001079 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1080 handle_system_exit();
1081 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001082 PyErr_Fetch(&exception2, &v2, &tb2);
1083 PyErr_NormalizeException(&exception2, &v2, &tb2);
Neal Norwitza5e4f222006-07-17 00:59:04 +00001084 /* It should not be possible for exception2 or v2
1085 to be NULL. However PyErr_Display() can't
1086 tolerate NULLs, so just be safe. */
1087 if (exception2 == NULL) {
1088 exception2 = Py_None;
1089 Py_INCREF(exception2);
1090 }
1091 if (v2 == NULL) {
1092 v2 = Py_None;
1093 Py_INCREF(v2);
1094 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001095 if (Py_FlushLine())
1096 PyErr_Clear();
1097 fflush(stdout);
1098 PySys_WriteStderr("Error in sys.excepthook:\n");
1099 PyErr_Display(exception2, v2, tb2);
1100 PySys_WriteStderr("\nOriginal exception was:\n");
1101 PyErr_Display(exception, v, tb);
Neal Norwitza5e4f222006-07-17 00:59:04 +00001102 Py_DECREF(exception2);
1103 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001104 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001105 }
1106 Py_XDECREF(result);
1107 Py_XDECREF(args);
1108 } else {
1109 PySys_WriteStderr("sys.excepthook is missing\n");
1110 PyErr_Display(exception, v, tb);
1111 }
1112 Py_XDECREF(exception);
1113 Py_XDECREF(v);
1114 Py_XDECREF(tb);
1115}
1116
Richard Jones7b9558d2006-05-27 12:29:24 +00001117void
1118PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001119{
1120 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001121 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001122 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001123 if (f == NULL)
1124 fprintf(stderr, "lost sys.stderr\n");
1125 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001126 if (Py_FlushLine())
1127 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001128 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001129 if (tb && tb != Py_None)
1130 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001131 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001132 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001133 {
Guido van Rossum82598051997-03-05 00:20:32 +00001134 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001135 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001136 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001137 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001138 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001139 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001140 else {
1141 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001142 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001143 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001144 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001145 else
Guido van Rossum82598051997-03-05 00:20:32 +00001146 PyFile_WriteString(filename, f);
1147 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001148 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001149 PyFile_WriteString(buf, f);
1150 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001151 if (text != NULL)
1152 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001153 Py_DECREF(value);
1154 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001155 /* Can't be bothered to check all those
1156 PyFile_WriteString() calls */
1157 if (PyErr_Occurred())
1158 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001159 }
1160 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001161 if (err) {
1162 /* Don't do anything else */
1163 }
Brett Cannonbf364092006-03-01 04:25:17 +00001164 else if (PyExceptionClass_Check(exception)) {
Richard Jones7b9558d2006-05-27 12:29:24 +00001165 PyObject* moduleName;
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001166 char* className = PyExceptionClass_Name(exception);
1167 if (className != NULL) {
1168 char *dot = strrchr(className, '.');
1169 if (dot != NULL)
1170 className = dot+1;
1171 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001172
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001173 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001174 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001175 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001176 else {
1177 char* modstr = PyString_AsString(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001178 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001179 {
1180 err = PyFile_WriteString(modstr, f);
1181 err += PyFile_WriteString(".", f);
1182 }
Richard Jones7b9558d2006-05-27 12:29:24 +00001183 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001184 }
1185 if (err == 0) {
1186 if (className == NULL)
1187 err = PyFile_WriteString("<unknown>", f);
1188 else
Brett Cannonbf364092006-03-01 04:25:17 +00001189 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001190 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001191 }
1192 else
1193 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001194 if (err == 0 && (value != Py_None)) {
1195 PyObject *s = PyObject_Str(value);
1196 /* only print colon if the str() of the
1197 object is not the empty string
1198 */
1199 if (s == NULL)
1200 err = -1;
1201 else if (!PyString_Check(s) ||
1202 PyString_GET_SIZE(s) != 0)
1203 err = PyFile_WriteString(": ", f);
1204 if (err == 0)
1205 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1206 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001207 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001208 if (err == 0)
1209 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001210 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001211 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001212 /* If an error happened here, don't show it.
1213 XXX This is wrong, but too many callers rely on this behavior. */
1214 if (err != 0)
1215 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001216}
1217
Guido van Rossum82598051997-03-05 00:20:32 +00001218PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001219PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001220 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001221{
Neal Norwitze92fba02006-03-04 18:52:26 +00001222 PyObject *ret = NULL;
Neal Norwitzd12bd012006-07-21 07:59:47 +00001223 mod_ty mod;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001224 PyArena *arena = PyArena_New();
Neal Norwitzd12bd012006-07-21 07:59:47 +00001225 if (arena == NULL)
1226 return NULL;
1227
1228 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001229 if (mod != NULL)
1230 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001231 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001232 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001233}
1234
1235PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001236PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001237 PyObject *locals, int closeit, PyCompilerFlags *flags)
1238{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001239 PyObject *ret;
Neal Norwitzd12bd012006-07-21 07:59:47 +00001240 mod_ty mod;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001241 PyArena *arena = PyArena_New();
Neal Norwitzd12bd012006-07-21 07:59:47 +00001242 if (arena == NULL)
1243 return NULL;
1244
1245 mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
1246 flags, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001247 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001248 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001249 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001250 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001251 if (closeit)
1252 fclose(fp);
Neal Norwitze92fba02006-03-04 18:52:26 +00001253 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001254 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001255 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001256}
1257
Guido van Rossum82598051997-03-05 00:20:32 +00001258static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001259run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001260 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001261{
Guido van Rossum82598051997-03-05 00:20:32 +00001262 PyCodeObject *co;
1263 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001264 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001265 if (co == NULL)
1266 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001267 v = PyEval_EvalCode(co, globals, locals);
1268 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001269 return v;
1270}
1271
Guido van Rossum82598051997-03-05 00:20:32 +00001272static PyObject *
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001273run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001274 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001275{
Guido van Rossum82598051997-03-05 00:20:32 +00001276 PyCodeObject *co;
1277 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001278 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001279 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001280
Guido van Rossum82598051997-03-05 00:20:32 +00001281 magic = PyMarshal_ReadLongFromFile(fp);
1282 if (magic != PyImport_GetMagicNumber()) {
1283 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001284 "Bad magic number in .pyc file");
1285 return NULL;
1286 }
Guido van Rossum82598051997-03-05 00:20:32 +00001287 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001288 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001289 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001290 if (v == NULL || !PyCode_Check(v)) {
1291 Py_XDECREF(v);
1292 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001293 "Bad code object in .pyc file");
1294 return NULL;
1295 }
Guido van Rossum82598051997-03-05 00:20:32 +00001296 co = (PyCodeObject *)v;
1297 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001298 if (v && flags)
1299 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001300 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001301 return v;
1302}
1303
Guido van Rossum82598051997-03-05 00:20:32 +00001304PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001305Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001306 PyCompilerFlags *flags)
1307{
Guido van Rossum82598051997-03-05 00:20:32 +00001308 PyCodeObject *co;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001309 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001310 PyArena *arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001311 if (arena == NULL)
1312 return NULL;
1313
1314 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001315 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001316 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001317 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001318 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001319 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001320 PyObject *result = PyAST_mod2obj(mod);
1321 PyArena_Free(arena);
1322 return result;
1323 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001324 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001325 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001326 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001327}
1328
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001329struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001330Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001331{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001332 struct symtable *st;
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001333 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001334 PyArena *arena = PyArena_New();
Neal Norwitzb59d08c2006-07-22 16:20:49 +00001335 if (arena == NULL)
1336 return NULL;
1337
1338 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001339 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001340 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001341 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001342 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001343 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001344 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001345 return st;
1346}
1347
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001348/* Preferred access to parser is through AST. */
1349mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001350PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001351 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001352{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001353 mod_ty mod;
1354 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001355 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001356 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001357 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001358 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001359 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001360 PyNode_Free(n);
1361 return mod;
1362 }
1363 else {
1364 err_input(&err);
1365 return NULL;
1366 }
1367}
1368
1369mod_ty
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001370PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001371 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001372 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001373{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001374 mod_ty mod;
1375 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001376 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1377 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001378 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001379 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001380 PyNode_Free(n);
1381 return mod;
1382 }
1383 else {
1384 err_input(&err);
1385 if (errcode)
1386 *errcode = err.error;
1387 return NULL;
1388 }
1389}
1390
Guido van Rossuma110aa61994-08-29 12:50:44 +00001391/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001392
Guido van Rossuma110aa61994-08-29 12:50:44 +00001393node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001394PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001395{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001396 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001397 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1398 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001399 if (n == NULL)
1400 err_input(&err);
Tim Peters5e9d6cf2006-05-28 10:41:29 +00001401
Guido van Rossuma110aa61994-08-29 12:50:44 +00001402 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001403}
1404
Guido van Rossuma110aa61994-08-29 12:50:44 +00001405/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001406
Guido van Rossuma110aa61994-08-29 12:50:44 +00001407node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001408PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001409{
Tim Petersfe2127d2001-07-16 05:37:24 +00001410 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001411 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1412 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001413 if (n == NULL)
1414 err_input(&err);
1415 return n;
1416}
1417
1418node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001419PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001420 int start, int flags)
1421{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001422 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001423 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1424 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001425 if (n == NULL)
1426 err_input(&err);
1427 return n;
1428}
1429
1430node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001431PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001432{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001433 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001434}
1435
Guido van Rossum66ebd912003-04-17 16:02:26 +00001436/* May want to move a more generalized form of this to parsetok.c or
1437 even parser modules. */
1438
1439void
1440PyParser_SetError(perrdetail *err)
1441{
1442 err_input(err);
1443}
1444
Guido van Rossuma110aa61994-08-29 12:50:44 +00001445/* Set the error appropriate to the given input error code (see errcode.h) */
1446
1447static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001448err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001449{
Fred Drake85f36392000-07-11 17:53:00 +00001450 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001451 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001452 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001453 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001454 switch (err->error) {
1455 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001456 errtype = PyExc_IndentationError;
1457 if (err->expected == INDENT)
1458 msg = "expected an indented block";
1459 else if (err->token == INDENT)
1460 msg = "unexpected indent";
1461 else if (err->token == DEDENT)
1462 msg = "unexpected unindent";
1463 else {
1464 errtype = PyExc_SyntaxError;
1465 msg = "invalid syntax";
1466 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001467 break;
1468 case E_TOKEN:
1469 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001470 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001471 case E_EOFS:
1472 msg = "EOF while scanning triple-quoted string";
1473 break;
1474 case E_EOLS:
1475 msg = "EOL while scanning single-quoted string";
1476 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001477 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001478 if (!PyErr_Occurred())
1479 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001480 return;
1481 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001482 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001483 return;
1484 case E_EOF:
1485 msg = "unexpected EOF while parsing";
1486 break;
Fred Drake85f36392000-07-11 17:53:00 +00001487 case E_TABSPACE:
1488 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001489 msg = "inconsistent use of tabs and spaces in indentation";
1490 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001491 case E_OVERFLOW:
1492 msg = "expression too long";
1493 break;
Fred Drake85f36392000-07-11 17:53:00 +00001494 case E_DEDENT:
1495 errtype = PyExc_IndentationError;
1496 msg = "unindent does not match any outer indentation level";
1497 break;
1498 case E_TOODEEP:
1499 errtype = PyExc_IndentationError;
1500 msg = "too many levels of indentation";
1501 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001502 case E_DECODE: {
1503 PyObject *type, *value, *tb;
1504 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001505 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001506 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001507 if (u != NULL) {
1508 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001509 }
1510 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001511 if (msg == NULL)
1512 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001513 Py_XDECREF(type);
1514 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001515 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001516 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001517 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001518 case E_LINECONT:
1519 msg = "unexpected character after line continuation character";
1520 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001521 default:
1522 fprintf(stderr, "error=%d\n", err->error);
1523 msg = "unknown parsing error";
1524 break;
1525 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001526 v = Py_BuildValue("(ziiz)", err->filename,
1527 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001528 if (err->text != NULL) {
Tim Petersc9d78aa2006-03-26 23:27:58 +00001529 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001530 err->text = NULL;
1531 }
1532 w = NULL;
1533 if (v != NULL)
1534 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001535 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001536 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001537 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001538 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001539}
1540
1541/* Print fatal error message and abort */
1542
1543void
Tim Peters7c321a82002-07-09 02:57:01 +00001544Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001545{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001546 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001547#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001548 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001549 OutputDebugString(msg);
1550 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001551#ifdef _DEBUG
1552 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001553#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001554#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001555 abort();
1556}
1557
1558/* Clean up and exit */
1559
Guido van Rossuma110aa61994-08-29 12:50:44 +00001560#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001561#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001562#endif
1563
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001564#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001565static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001566static int nexitfuncs = 0;
1567
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001568int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001569{
1570 if (nexitfuncs >= NEXITFUNCS)
1571 return -1;
1572 exitfuncs[nexitfuncs++] = func;
1573 return 0;
1574}
1575
Guido van Rossumcc283f51997-08-05 02:22:03 +00001576static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001577call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001578{
Guido van Rossum82598051997-03-05 00:20:32 +00001579 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001580
1581 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001582 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001583 Py_INCREF(exitfunc);
1584 PySys_SetObject("exitfunc", (PyObject *)NULL);
1585 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001586 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001587 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1588 PySys_WriteStderr("Error in sys.exitfunc:\n");
1589 }
Guido van Rossum82598051997-03-05 00:20:32 +00001590 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001591 }
Guido van Rossum82598051997-03-05 00:20:32 +00001592 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001593 }
1594
Guido van Rossum0829c751998-02-28 04:31:39 +00001595 if (Py_FlushLine())
1596 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001597}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001598
Guido van Rossumcc283f51997-08-05 02:22:03 +00001599static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001600call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001601{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001602 while (nexitfuncs > 0)
1603 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001604
1605 fflush(stdout);
1606 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001607}
1608
1609void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001610Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001611{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001612 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001613
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001614 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001615}
1616
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001617static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001618initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001619{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001620#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001621 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001622#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001623#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001624 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001625#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001626#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001627 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001628#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001629 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001630}
1631
Guido van Rossum7433b121997-02-14 19:45:36 +00001632
1633/*
1634 * The file descriptor fd is considered ``interactive'' if either
1635 * a) isatty(fd) is TRUE, or
1636 * b) the -i flag was given, and the filename associated with
1637 * the descriptor is NULL or "<stdin>" or "???".
1638 */
1639int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001640Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001641{
1642 if (isatty((int)fileno(fp)))
1643 return 1;
1644 if (!Py_InteractiveFlag)
1645 return 0;
1646 return (filename == NULL) ||
1647 (strcmp(filename, "<stdin>") == 0) ||
1648 (strcmp(filename, "???") == 0);
1649}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001650
1651
Tim Petersd08e3822003-04-17 15:24:21 +00001652#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001653#if defined(WIN32) && defined(_MSC_VER)
1654
1655/* Stack checking for Microsoft C */
1656
1657#include <malloc.h>
1658#include <excpt.h>
1659
Fred Drakee8de31c2000-08-31 05:38:39 +00001660/*
1661 * Return non-zero when we run out of memory on the stack; zero otherwise.
1662 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001663int
Fred Drake399739f2000-08-31 05:52:44 +00001664PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001665{
1666 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001667 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001668 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001669 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001670 return 0;
1671 } __except (EXCEPTION_EXECUTE_HANDLER) {
1672 /* just ignore all errors */
1673 }
1674 return 1;
1675}
1676
1677#endif /* WIN32 && _MSC_VER */
1678
1679/* Alternate implementations can be added here... */
1680
1681#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001682
1683
1684/* Wrappers around sigaction() or signal(). */
1685
1686PyOS_sighandler_t
1687PyOS_getsig(int sig)
1688{
1689#ifdef HAVE_SIGACTION
1690 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001691 if (sigaction(sig, NULL, &context) == -1)
1692 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001693 return context.sa_handler;
1694#else
1695 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001696/* Special signal handling for the secure CRT in Visual Studio 2005 */
1697#if defined(_MSC_VER) && _MSC_VER >= 1400
1698 switch (sig) {
1699 /* Only these signals are valid */
1700 case SIGINT:
1701 case SIGILL:
1702 case SIGFPE:
1703 case SIGSEGV:
1704 case SIGTERM:
1705 case SIGBREAK:
1706 case SIGABRT:
1707 break;
1708 /* Don't call signal() with other values or it will assert */
1709 default:
1710 return SIG_ERR;
1711 }
1712#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001713 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001714 if (handler != SIG_ERR)
1715 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001716 return handler;
1717#endif
1718}
1719
1720PyOS_sighandler_t
1721PyOS_setsig(int sig, PyOS_sighandler_t handler)
1722{
1723#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001724 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001725 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001726 sigemptyset(&context.sa_mask);
1727 context.sa_flags = 0;
1728 if (sigaction(sig, &context, &ocontext) == -1)
1729 return SIG_ERR;
1730 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001731#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001732 PyOS_sighandler_t oldhandler;
1733 oldhandler = signal(sig, handler);
1734#ifdef HAVE_SIGINTERRUPT
1735 siginterrupt(sig, 1);
1736#endif
1737 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001738#endif
1739}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001740
1741/* Deprecated C API functions still provided for binary compatiblity */
1742
1743#undef PyParser_SimpleParseFile
Thomas Heller1b046642006-04-18 18:51:06 +00001744PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001745PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1746{
1747 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1748}
1749
Thomas Heller1b046642006-04-18 18:51:06 +00001750#undef PyParser_SimpleParseString
1751PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001752PyParser_SimpleParseString(const char *str, int start)
1753{
1754 return PyParser_SimpleParseStringFlags(str, start, 0);
1755}
Anthony Baxterac6bd462006-04-13 02:06:09 +00001756
Thomas Heller1b046642006-04-18 18:51:06 +00001757#undef PyRun_AnyFile
1758PyAPI_FUNC(int)
1759PyRun_AnyFile(FILE *fp, const char *name)
1760{
1761 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1762}
1763
1764#undef PyRun_AnyFileEx
1765PyAPI_FUNC(int)
1766PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1767{
1768 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1769}
1770
1771#undef PyRun_AnyFileFlags
1772PyAPI_FUNC(int)
1773PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1774{
1775 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1776}
1777
1778#undef PyRun_File
1779PyAPI_FUNC(PyObject *)
1780PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1781{
1782 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1783}
1784
1785#undef PyRun_FileEx
1786PyAPI_FUNC(PyObject *)
1787PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1788{
1789 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1790}
1791
1792#undef PyRun_FileFlags
1793PyAPI_FUNC(PyObject *)
1794PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1795 PyCompilerFlags *flags)
1796{
1797 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1798}
1799
1800#undef PyRun_SimpleFile
1801PyAPI_FUNC(int)
1802PyRun_SimpleFile(FILE *f, const char *p)
1803{
1804 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1805}
1806
1807#undef PyRun_SimpleFileEx
1808PyAPI_FUNC(int)
1809PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1810{
1811 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1812}
1813
1814
1815#undef PyRun_String
1816PyAPI_FUNC(PyObject *)
1817PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1818{
1819 return PyRun_StringFlags(str, s, g, l, NULL);
1820}
1821
1822#undef PyRun_SimpleString
1823PyAPI_FUNC(int)
1824PyRun_SimpleString(const char *s)
1825{
1826 return PyRun_SimpleStringFlags(s, NULL);
1827}
1828
1829#undef Py_CompileString
1830PyAPI_FUNC(PyObject *)
1831Py_CompileString(const char *str, const char *p, int s)
1832{
1833 return Py_CompileStringFlags(str, p, s, NULL);
1834}
1835
1836#undef PyRun_InteractiveOne
1837PyAPI_FUNC(int)
1838PyRun_InteractiveOne(FILE *f, const char *p)
1839{
1840 return PyRun_InteractiveOneFlags(f, p, NULL);
1841}
1842
1843#undef PyRun_InteractiveLoop
1844PyAPI_FUNC(int)
1845PyRun_InteractiveLoop(FILE *f, const char *p)
1846{
1847 return PyRun_InteractiveLoopFlags(f, p, NULL);
1848}
1849
Anthony Baxterac6bd462006-04-13 02:06:09 +00001850#ifdef __cplusplus
1851}
1852#endif
1853