blob: 2e93a3a1991cea5fea6d50dd6d70430184b42877 [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 Rossumd8faa362007-04-27 19:54:29 +00007#undef Yield /* undefine macro conflicting with winbase.h */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00008#include "grammar.h"
9#include "node.h"
Fred Drake85f36392000-07-11 17:53:00 +000010#include "token.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000011#include "parsetok.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000012#include "errcode.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000013#include "code.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000014#include "compile.h"
Jeremy Hylton4b38da62001-02-02 18:19:15 +000015#include "symtable.h"
Neal Norwitzadb69fc2005-12-17 20:54:49 +000016#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000017#include "ast.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000018#include "eval.h"
Guido van Rossumfdef2711994-09-14 13:31:04 +000019#include "marshal.h"
Martin v. Löwis790465f2008-04-05 20:41:37 +000020#include "osdefs.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000021
Thomas Wouters0e3f5912006-08-11 14:57:12 +000022#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000023#include <signal.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000024#endif
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000025
Martin v. Löwis73d538b2003-03-05 15:13:47 +000026#ifdef HAVE_LANGINFO_H
27#include <locale.h>
28#include <langinfo.h>
29#endif
30
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000031#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000032#undef BYTE
33#include "windows.h"
Martin v. Löwis790465f2008-04-05 20:41:37 +000034#define PATH_MAX MAXPATHLEN
Guido van Rossuma44823b1995-03-14 15:01:17 +000035#endif
36
Neal Norwitz4281cef2006-03-04 19:58:13 +000037#ifndef Py_REF_DEBUG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000038#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000039#else /* Py_REF_DEBUG */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000040#define PRINT_TOTAL_REFS() fprintf(stderr, \
41 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
42 _Py_GetRefTotal())
43#endif
44
45#ifdef __cplusplus
46extern "C" {
Neal Norwitz4281cef2006-03-04 19:58:13 +000047#endif
48
Martin v. Löwis790465f2008-04-05 20:41:37 +000049extern wchar_t *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000050
Guido van Rossum82598051997-03-05 00:20:32 +000051extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000052
Guido van Rossumb73cc041993-11-01 16:28:59 +000053/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000054static void initmain(void);
55static void initsite(void);
Guido van Rossumce3a72a2007-10-19 23:16:50 +000056static int initstdio(void);
Amaury Forgeot d'Arc7fedbe52008-04-10 21:03:09 +000057static void flush_io(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000058static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000059 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000060static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000061 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000062static void err_input(perrdetail *);
63static void initsigs(void);
Collin Winter670e6922007-03-21 02:57:17 +000064static void call_py_exitfuncs(void);
Tim Petersdbd9ba62000-07-09 03:09:57 +000065static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000066extern void _PyUnicode_Init(void);
67extern void _PyUnicode_Fini(void);
Guido van Rossumddefaf32007-01-14 03:31:43 +000068extern int _PyLong_Init(void);
69extern void PyLong_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000070
Mark Hammond8d98d2c2003-04-19 15:41:53 +000071#ifdef WITH_THREAD
72extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
73extern void _PyGILState_Fini(void);
74#endif /* WITH_THREAD */
75
Guido van Rossum82598051997-03-05 00:20:32 +000076int Py_DebugFlag; /* Needed by parser.c */
77int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000078int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumd8faa362007-04-27 19:54:29 +000079int Py_InspectFlag; /* Needed to determine whether to exit at SystemError */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000080int Py_NoSiteFlag; /* Suppress 'import site' */
Guido van Rossum98297ee2007-11-06 21:34:58 +000081int Py_BytesWarningFlag; /* Warn on str(bytes) and str(buffer) */
Christian Heimes790c8232008-01-07 21:14:23 +000082int Py_DontWriteBytecodeFlag; /* Suppress writing bytecode files (*.py[co]) */
Barry Warsaw3ce09642000-05-02 19:18:59 +000083int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000084int Py_FrozenFlag; /* Needed by getpath.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000085int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Christian Heimes8dc226f2008-05-06 23:45:46 +000086int Py_NoUserSiteDirectory = 0; /* for -s and site.py */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000087
Christian Heimes33fe8092008-04-13 13:53:33 +000088/* PyModule_GetWarningsModule is no longer necessary as of 2.6
89since _warnings is builtin. This API should not be used. */
90PyObject *
91PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000092{
Christian Heimes33fe8092008-04-13 13:53:33 +000093 return PyImport_ImportModule("warnings");
Mark Hammondedd07732003-07-15 23:03:55 +000094}
Mark Hammonda43fd0c2003-02-19 00:33:33 +000095
Guido van Rossum25ce5661997-08-02 03:10:38 +000096static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +000097
Thomas Wouters7e474022000-07-16 12:04:32 +000098/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +000099
100int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000101Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000102{
103 return initialized;
104}
105
Guido van Rossum25ce5661997-08-02 03:10:38 +0000106/* Global initializations. Can be undone by Py_Finalize(). Don't
107 call this twice without an intervening Py_Finalize() call. When
108 initializations fail, a fatal error is issued and the function does
109 not return. On return, the first thread and interpreter state have
110 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000111
Guido van Rossum25ce5661997-08-02 03:10:38 +0000112 Locking: you must hold the interpreter lock while calling this.
113 (If the lock has not yet been initialized, that's equivalent to
114 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000115
Guido van Rossum25ce5661997-08-02 03:10:38 +0000116*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000117
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000118static int
119add_flag(int flag, const char *envs)
120{
121 int env = atoi(envs);
122 if (flag < env)
123 flag = env;
124 if (flag < 1)
125 flag = 1;
126 return flag;
127}
128
Guido van Rossuma027efa1997-05-05 20:56:21 +0000129void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000130Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000131{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000132 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000133 PyThreadState *tstate;
Guido van Rossum826d8972007-10-30 18:34:07 +0000134 PyObject *bimod, *sysmod, *pstderr;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000135 char *p;
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000136#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000137 char *codeset;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000138#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000139 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000140
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000141 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000142 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000143 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000144
Martin v. Löwisd1cd4d42007-08-11 14:02:14 +0000145#ifdef HAVE_SETLOCALE
146 /* Set up the LC_CTYPE locale, so we can obtain
147 the locale's charset without having to switch
148 locales. */
149 setlocale(LC_CTYPE, "");
150#endif
151
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000152 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000153 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000154 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000155 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000156 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000157 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Christian Heimes790c8232008-01-07 21:14:23 +0000158 if ((p = Py_GETENV("PYTHONDONTWRITEBYTECODE")) && *p != '\0')
159 Py_DontWriteBytecodeFlag = add_flag(Py_DontWriteBytecodeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000160
Guido van Rossuma027efa1997-05-05 20:56:21 +0000161 interp = PyInterpreterState_New();
162 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000163 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000164
Guido van Rossuma027efa1997-05-05 20:56:21 +0000165 tstate = PyThreadState_New(interp);
166 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000167 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000168 (void) PyThreadState_Swap(tstate);
169
Guido van Rossum70d893a2001-08-16 08:21:42 +0000170 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000171
Neal Norwitzb2501f42002-12-31 03:42:13 +0000172 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000173 Py_FatalError("Py_Initialize: can't init frames");
174
Guido van Rossumddefaf32007-01-14 03:31:43 +0000175 if (!_PyLong_Init())
176 Py_FatalError("Py_Initialize: can't init longs");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000177
Christian Heimes9c4756e2008-05-26 13:22:05 +0000178 if (!PyByteArray_Init())
Neal Norwitz6968b052007-02-27 19:02:19 +0000179 Py_FatalError("Py_Initialize: can't init bytes");
180
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000181 _PyFloat_Init();
182
Guido van Rossum25ce5661997-08-02 03:10:38 +0000183 interp->modules = PyDict_New();
184 if (interp->modules == NULL)
185 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000186 interp->modules_reloading = PyDict_New();
187 if (interp->modules_reloading == NULL)
188 Py_FatalError("Py_Initialize: can't make modules_reloading dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000189
Guido van Rossumc94044c2000-03-10 23:03:54 +0000190 /* Init Unicode implementation; relies on the codec registry */
191 _PyUnicode_Init();
192
Barry Warsawf242aa02000-05-25 23:09:49 +0000193 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000194 if (bimod == NULL)
Georg Brandl1a3284e2007-12-02 09:40:06 +0000195 Py_FatalError("Py_Initialize: can't initialize builtins modules");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000196 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000197 if (interp->builtins == NULL)
198 Py_FatalError("Py_Initialize: can't initialize builtins dict");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000199 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000200
Christian Heimes9a68f8c2007-11-14 00:16:07 +0000201 /* initialize builtin exceptions */
202 _PyExc_Init();
203
Guido van Rossum25ce5661997-08-02 03:10:38 +0000204 sysmod = _PySys_Init();
205 if (sysmod == NULL)
206 Py_FatalError("Py_Initialize: can't initialize sys");
207 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000208 if (interp->sysdict == NULL)
209 Py_FatalError("Py_Initialize: can't initialize sys dict");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000210 Py_INCREF(interp->sysdict);
211 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000212 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000213 PyDict_SetItemString(interp->sysdict, "modules",
214 interp->modules);
215
Guido van Rossum826d8972007-10-30 18:34:07 +0000216 /* Set up a preliminary stderr printer until we have enough
217 infrastructure for the io module in place. */
218 pstderr = PyFile_NewStdPrinter(fileno(stderr));
219 if (pstderr == NULL)
220 Py_FatalError("Py_Initialize: can't set preliminary stderr");
221 PySys_SetObject("stderr", pstderr);
Guido van Rossum98297ee2007-11-06 21:34:58 +0000222 PySys_SetObject("__stderr__", pstderr);
Guido van Rossum826d8972007-10-30 18:34:07 +0000223
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000224 _PyImport_Init();
225
Barry Warsaw035574d1997-08-29 22:07:17 +0000226 /* phase 2 of builtins */
Georg Brandl1a3284e2007-12-02 09:40:06 +0000227 _PyImport_FixupExtension("builtins", "builtins");
Barry Warsaw035574d1997-08-29 22:07:17 +0000228
Just van Rossum52e14d62002-12-30 22:08:05 +0000229 _PyImportHooks_Init();
230
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000231 if (install_sigs)
232 initsigs(); /* Signal handling stuff, including initintr() */
Christian Heimes33fe8092008-04-13 13:53:33 +0000233
234 /* Initialize warnings. */
235 _PyWarnings_Init();
236 if (PySys_HasWarnOptions()) {
237 PyObject *warnings_module = PyImport_ImportModule("warnings");
238 if (!warnings_module)
239 PyErr_Clear();
240 Py_XDECREF(warnings_module);
241 }
Guido van Rossum25ce5661997-08-02 03:10:38 +0000242
243 initmain(); /* Module __main__ */
Alexandre Vassalotti77250f42008-05-06 19:48:38 +0000244 if (!Py_NoSiteFlag)
245 initsite(); /* Module site */
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000246 if (initstdio() < 0)
247 Py_FatalError(
248 "Py_Initialize: can't initialize sys standard streams");
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000249
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000250 /* auto-thread-state API, if available */
251#ifdef WITH_THREAD
252 _PyGILState_Init(interp, tstate);
253#endif /* WITH_THREAD */
254
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000255#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000256 /* On Unix, set the file system encoding according to the
257 user's preference, if the CODESET names a well-known
258 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000259 initialized by other means. Also set the encoding of
260 stdin and stdout if these are terminals. */
261
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000262 codeset = nl_langinfo(CODESET);
263 if (codeset && *codeset) {
264 PyObject *enc = PyCodec_Encoder(codeset);
265 if (enc) {
266 codeset = strdup(codeset);
267 Py_DECREF(enc);
268 } else {
269 codeset = NULL;
270 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000271 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000272 } else
273 codeset = NULL;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000274
275 if (codeset) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000276 if (!Py_FileSystemDefaultEncoding)
277 Py_FileSystemDefaultEncoding = codeset;
278 else
279 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000280 }
281#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000282}
283
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000284void
285Py_Initialize(void)
286{
287 Py_InitializeEx(1);
288}
289
290
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000291#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000292extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000293#endif
294
Guido van Rossume8432ac2007-07-09 15:04:50 +0000295/* Flush stdout and stderr */
296
Neal Norwitz2bad9702007-08-27 06:19:22 +0000297static void
Guido van Rossum1bd21222007-07-10 20:14:13 +0000298flush_std_files(void)
Guido van Rossume8432ac2007-07-09 15:04:50 +0000299{
300 PyObject *fout = PySys_GetObject("stdout");
301 PyObject *ferr = PySys_GetObject("stderr");
302 PyObject *tmp;
303
Christian Heimes2be03732007-11-15 02:26:46 +0000304 if (fout != NULL && fout != Py_None) {
Guido van Rossume8432ac2007-07-09 15:04:50 +0000305 tmp = PyObject_CallMethod(fout, "flush", "");
306 if (tmp == NULL)
307 PyErr_Clear();
308 else
309 Py_DECREF(tmp);
310 }
311
Christian Heimes2be03732007-11-15 02:26:46 +0000312 if (ferr != NULL || ferr != Py_None) {
Guido van Rossume8432ac2007-07-09 15:04:50 +0000313 tmp = PyObject_CallMethod(ferr, "flush", "");
314 if (tmp == NULL)
315 PyErr_Clear();
316 else
317 Py_DECREF(tmp);
318 }
319}
320
Guido van Rossum25ce5661997-08-02 03:10:38 +0000321/* Undo the effect of Py_Initialize().
322
323 Beware: if multiple interpreter and/or thread states exist, these
324 are not wiped out; only the current thread and interpreter state
325 are deleted. But since everything else is deleted, those other
326 interpreter and thread states should no longer be used.
327
328 (XXX We should do better, e.g. wipe out all interpreters and
329 threads.)
330
331 Locking: as above.
332
333*/
334
335void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000336Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000337{
338 PyInterpreterState *interp;
339 PyThreadState *tstate;
340
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000341 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000342 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000343
Tim Peters384fd102001-01-21 03:40:37 +0000344 /* The interpreter is still entirely intact at this point, and the
345 * exit funcs may be relying on that. In particular, if some thread
346 * or exit func is still waiting to do an import, the import machinery
347 * expects Py_IsInitialized() to return true. So don't say the
348 * interpreter is uninitialized until after the exit funcs have run.
349 * Note that Threading.py uses an exit func to do a join on all the
350 * threads created thru it, so this also protects pending imports in
351 * the threads created via Threading.
352 */
Collin Winter670e6922007-03-21 02:57:17 +0000353 call_py_exitfuncs();
Tim Peters384fd102001-01-21 03:40:37 +0000354 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000355
Guido van Rossume8432ac2007-07-09 15:04:50 +0000356 /* Flush stdout+stderr */
357 flush_std_files();
358
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000359 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000360 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000361 interp = tstate->interp;
362
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000363 /* Disable signal handling */
364 PyOS_FiniInterrupts();
365
Christian Heimes26855632008-01-27 23:50:43 +0000366 /* Clear type lookup cache */
367 PyType_ClearCache();
368
Guido van Rossume13ddc92003-04-17 17:29:22 +0000369 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000370 * before all modules are destroyed.
371 * XXX If a __del__ or weakref callback is triggered here, and tries to
372 * XXX import a module, bad things can happen, because Python no
373 * XXX longer believes it's initialized.
374 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
375 * XXX is easy to provoke that way. I've also seen, e.g.,
376 * XXX Exception exceptions.ImportError: 'No module named sha'
377 * XXX in <function callback at 0x008F5718> ignored
378 * XXX but I'm unclear on exactly how that one happens. In any case,
379 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000380 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000381 PyGC_Collect();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000382#ifdef COUNT_ALLOCS
383 /* With COUNT_ALLOCS, it helps to run GC multiple times:
384 each collection might release some types from the type
385 list, so they become garbage. */
386 while (PyGC_Collect() > 0)
387 /* nothing */;
388#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000389
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000390 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000391 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000392
Guido van Rossume8432ac2007-07-09 15:04:50 +0000393 /* Flush stdout+stderr (again, in case more was printed) */
394 flush_std_files();
395
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
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000420 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000421#endif
422
Thomas Wouters49fd7fa2006-04-21 10:40:58 +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
Guido van Rossumd922fa42003-04-15 14:10:09 +0000435 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000436 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000437
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000438 /* Now we decref the exception classes. After this point nothing
439 can raise an exception. That's okay, because each Fini() method
440 below has been checked to make sure no exceptions are ever
441 raised.
442 */
443
444 _PyExc_Fini();
445
Christian Heimes7d2ff882007-11-30 14:35:04 +0000446 /* Cleanup auto-thread-state */
447#ifdef WITH_THREAD
448 _PyGILState_Fini();
449#endif /* WITH_THREAD */
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();
Christian Heimes72b710a2008-05-26 13:28:38 +0000462 PyBytes_Fini();
Christian Heimes9c4756e2008-05-26 13:22:05 +0000463 PyByteArray_Fini();
Guido van Rossumddefaf32007-01-14 03:31:43 +0000464 PyLong_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000465 PyFloat_Fini();
Christian Heimes77c02eb2008-02-09 02:18:51 +0000466 PyDict_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000467
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000468 /* Cleanup Unicode implementation */
469 _PyUnicode_Fini();
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000470
Christian Heimesc8967002007-11-30 10:18:26 +0000471 /* reset file system default encoding */
472 if (!Py_HasFileSystemDefaultEncoding) {
473 free((char*)Py_FileSystemDefaultEncoding);
Trent Nelson39e307e2008-03-19 06:45:48 +0000474 Py_FileSystemDefaultEncoding = NULL;
Christian Heimesc8967002007-11-30 10:18:26 +0000475 }
476
Guido van Rossumcc283f51997-08-05 02:22:03 +0000477 /* XXX Still allocated:
478 - various static ad-hoc pointers to interned strings
479 - int and float free list blocks
480 - whatever various modules and libraries allocate
481 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000482
483 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000484
Tim Peters269b2a62003-04-17 19:52:29 +0000485#ifdef Py_TRACE_REFS
486 /* Display addresses (& refcnts) of all objects still alive.
487 * An address can be used to find the repr of the object, printed
488 * above by _Py_PrintReferences.
489 */
490 if (Py_GETENV("PYTHONDUMPREFS"))
491 _Py_PrintReferenceAddresses(stderr);
492#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000493#ifdef PYMALLOC_DEBUG
494 if (Py_GETENV("PYTHONMALLOCSTATS"))
495 _PyObject_DebugMallocStats();
496#endif
497
Guido van Rossumcc283f51997-08-05 02:22:03 +0000498 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000499}
500
501/* Create and initialize a new interpreter and thread, and return the
502 new thread. This requires that Py_Initialize() has been called
503 first.
504
505 Unsuccessful initialization yields a NULL pointer. Note that *no*
506 exception information is available even in this case -- the
507 exception information is held in the thread, and there is no
508 thread.
509
510 Locking: as above.
511
512*/
513
514PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000515Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000516{
517 PyInterpreterState *interp;
518 PyThreadState *tstate, *save_tstate;
519 PyObject *bimod, *sysmod;
520
521 if (!initialized)
522 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
523
524 interp = PyInterpreterState_New();
525 if (interp == NULL)
526 return NULL;
527
528 tstate = PyThreadState_New(interp);
529 if (tstate == NULL) {
530 PyInterpreterState_Delete(interp);
531 return NULL;
532 }
533
534 save_tstate = PyThreadState_Swap(tstate);
535
536 /* XXX The following is lax in error checking */
537
538 interp->modules = PyDict_New();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000539 interp->modules_reloading = PyDict_New();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000540
Georg Brandl1a3284e2007-12-02 09:40:06 +0000541 bimod = _PyImport_FindExtension("builtins", "builtins");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000542 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000543 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000544 if (interp->builtins == NULL)
545 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000546 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000547 }
548 sysmod = _PyImport_FindExtension("sys", "sys");
549 if (bimod != NULL && sysmod != NULL) {
550 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000551 if (interp->sysdict == NULL)
552 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000553 Py_INCREF(interp->sysdict);
554 PySys_SetPath(Py_GetPath());
555 PyDict_SetItemString(interp->sysdict, "modules",
556 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000557 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000558 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000559 if (!Py_NoSiteFlag)
560 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000561 }
562
563 if (!PyErr_Occurred())
564 return tstate;
565
Thomas Wouters89f507f2006-12-13 04:49:30 +0000566handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000567 /* Oops, it didn't work. Undo it all. */
568
569 PyErr_Print();
570 PyThreadState_Clear(tstate);
571 PyThreadState_Swap(save_tstate);
572 PyThreadState_Delete(tstate);
573 PyInterpreterState_Delete(interp);
574
575 return NULL;
576}
577
578/* Delete an interpreter and its last thread. This requires that the
579 given thread state is current, that the thread has no remaining
580 frames, and that it is its interpreter's only remaining thread.
581 It is a fatal error to violate these constraints.
582
583 (Py_Finalize() doesn't have these constraints -- it zaps
584 everything, regardless.)
585
586 Locking: as above.
587
588*/
589
590void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000591Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000592{
593 PyInterpreterState *interp = tstate->interp;
594
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000595 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000596 Py_FatalError("Py_EndInterpreter: thread is not current");
597 if (tstate->frame != NULL)
598 Py_FatalError("Py_EndInterpreter: thread still has a frame");
599 if (tstate != interp->tstate_head || tstate->next != NULL)
600 Py_FatalError("Py_EndInterpreter: not the last thread");
601
602 PyImport_Cleanup();
603 PyInterpreterState_Clear(interp);
604 PyThreadState_Swap(NULL);
605 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000606}
607
Martin v. Löwis790465f2008-04-05 20:41:37 +0000608static wchar_t *progname = L"python";
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000609
610void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000611Py_SetProgramName(wchar_t *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000612{
613 if (pn && *pn)
614 progname = pn;
615}
616
Martin v. Löwis790465f2008-04-05 20:41:37 +0000617wchar_t *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000618Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000619{
620 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000621}
622
Martin v. Löwis790465f2008-04-05 20:41:37 +0000623static wchar_t *default_home = NULL;
624static wchar_t env_home[PATH_MAX+1];
Guido van Rossuma61691e1998-02-06 22:27:24 +0000625
626void
Martin v. Löwis790465f2008-04-05 20:41:37 +0000627Py_SetPythonHome(wchar_t *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000628{
629 default_home = home;
630}
631
Martin v. Löwis790465f2008-04-05 20:41:37 +0000632wchar_t *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000633Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000634{
Martin v. Löwis790465f2008-04-05 20:41:37 +0000635 wchar_t *home = default_home;
636 if (home == NULL && !Py_IgnoreEnvironmentFlag) {
637 char* chome = Py_GETENV("PYTHONHOME");
638 if (chome) {
639 size_t r = mbstowcs(env_home, chome, PATH_MAX+1);
640 if (r != (size_t)-1 && r <= PATH_MAX)
641 home = env_home;
642 }
643
644 }
Guido van Rossuma61691e1998-02-06 22:27:24 +0000645 return home;
646}
647
Guido van Rossum6135a871995-01-09 17:53:26 +0000648/* Create __main__ module */
649
650static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000651initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000652{
Guido van Rossum82598051997-03-05 00:20:32 +0000653 PyObject *m, *d;
654 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000655 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000656 Py_FatalError("can't create __main__ module");
657 d = PyModule_GetDict(m);
658 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Georg Brandl1a3284e2007-12-02 09:40:06 +0000659 PyObject *bimod = PyImport_ImportModule("builtins");
Guido van Rossum858cb731997-11-19 16:15:37 +0000660 if (bimod == NULL ||
661 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000662 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000663 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000664 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000665}
666
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000667/* Import the site module (not into __main__ though) */
668
669static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000670initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000671{
672 PyObject *m, *f;
673 m = PyImport_ImportModule("site");
674 if (m == NULL) {
675 f = PySys_GetObject("stderr");
Christian Heimes2be03732007-11-15 02:26:46 +0000676 if (f == NULL || f == Py_None)
677 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000678 if (Py_VerboseFlag) {
679 PyFile_WriteString(
680 "'import site' failed; traceback:\n", f);
681 PyErr_Print();
682 }
683 else {
684 PyFile_WriteString(
685 "'import site' failed; use -v for traceback\n", f);
686 PyErr_Clear();
687 }
688 }
689 else {
690 Py_DECREF(m);
691 }
692}
693
Georg Brandl1a3284e2007-12-02 09:40:06 +0000694/* Initialize sys.stdin, stdout, stderr and builtins.open */
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000695static int
696initstdio(void)
697{
698 PyObject *iomod = NULL, *wrapper;
699 PyObject *bimod = NULL;
700 PyObject *m;
701 PyObject *std = NULL;
Christian Heimes58cb1b82007-11-13 02:19:40 +0000702 int status = 0, fd;
Trent Nelson39e307e2008-03-19 06:45:48 +0000703 PyObject * encoding_attr;
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000704
705 /* Hack to avoid a nasty recursion issue when Python is invoked
706 in verbose mode: pre-import the Latin-1 and UTF-8 codecs */
707 if ((m = PyImport_ImportModule("encodings.utf_8")) == NULL) {
708 goto error;
709 }
710 Py_DECREF(m);
711
712 if (!(m = PyImport_ImportModule("encodings.latin_1"))) {
713 goto error;
714 }
715 Py_DECREF(m);
716
Georg Brandl1a3284e2007-12-02 09:40:06 +0000717 if (!(bimod = PyImport_ImportModule("builtins"))) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000718 goto error;
719 }
720
721 if (!(iomod = PyImport_ImportModule("io"))) {
722 goto error;
723 }
724 if (!(wrapper = PyObject_GetAttrString(iomod, "OpenWrapper"))) {
725 goto error;
726 }
727
Georg Brandl1a3284e2007-12-02 09:40:06 +0000728 /* Set builtins.open */
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000729 if (PyObject_SetAttrString(bimod, "open", wrapper) == -1) {
730 goto error;
731 }
732
733 /* Set sys.stdin */
Christian Heimes58cb1b82007-11-13 02:19:40 +0000734 fd = fileno(stdin);
735 /* Under some conditions stdin, stdout and stderr may not be connected
736 * and fileno() may point to an invalid file descriptor. For example
737 * GUI apps don't have valid standard streams by default.
738 */
739 if (fd < 0) {
740#ifdef MS_WINDOWS
741 std = Py_None;
742 Py_INCREF(std);
743#else
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000744 goto error;
Christian Heimes58cb1b82007-11-13 02:19:40 +0000745#endif
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000746 }
Christian Heimes58cb1b82007-11-13 02:19:40 +0000747 else {
Guido van Rossume7fc50f2007-12-03 22:54:21 +0000748 if (!(std = PyFile_FromFd(fd, "<stdin>", "r", -1, NULL, NULL,
Christian Heimes58cb1b82007-11-13 02:19:40 +0000749 "\n", 0))) {
750 goto error;
751 }
752 } /* if (fd < 0) */
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000753 PySys_SetObject("__stdin__", std);
754 PySys_SetObject("stdin", std);
755 Py_DECREF(std);
756
757 /* Set sys.stdout */
Christian Heimes58cb1b82007-11-13 02:19:40 +0000758 fd = fileno(stdout);
759 if (fd < 0) {
760#ifdef MS_WINDOWS
761 std = Py_None;
762 Py_INCREF(std);
763#else
764 goto error;
765#endif
766 }
767 else {
Guido van Rossume7fc50f2007-12-03 22:54:21 +0000768 if (!(std = PyFile_FromFd(fd, "<stdout>", "w", -1, NULL, NULL,
Christian Heimes58cb1b82007-11-13 02:19:40 +0000769 "\n", 0))) {
770 goto error;
771 }
772 } /* if (fd < 0) */
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000773 PySys_SetObject("__stdout__", std);
774 PySys_SetObject("stdout", std);
775 Py_DECREF(std);
776
Guido van Rossum98297ee2007-11-06 21:34:58 +0000777#if 1 /* Disable this if you have trouble debugging bootstrap stuff */
Guido van Rossum2dced8b2007-10-30 17:27:30 +0000778 /* Set sys.stderr, replaces the preliminary stderr */
Christian Heimes58cb1b82007-11-13 02:19:40 +0000779 fd = fileno(stderr);
780 if (fd < 0) {
781#ifdef MS_WINDOWS
782 std = Py_None;
783 Py_INCREF(std);
784#else
785 goto error;
786#endif
787 }
788 else {
Guido van Rossume7fc50f2007-12-03 22:54:21 +0000789 if (!(std = PyFile_FromFd(fd, "<stderr>", "w", -1, NULL, NULL,
Christian Heimes58cb1b82007-11-13 02:19:40 +0000790 "\n", 0))) {
791 goto error;
792 }
793 } /* if (fd < 0) */
Trent Nelson39e307e2008-03-19 06:45:48 +0000794
795 /* Same as hack above, pre-import stderr's codec to avoid recursion
796 when import.c tries to write to stderr in verbose mode. */
797 encoding_attr = PyObject_GetAttrString(std, "encoding");
798 if (encoding_attr != NULL) {
799 const char * encoding;
800 encoding = PyUnicode_AsString(encoding_attr);
801 if (encoding != NULL) {
802 _PyCodec_Lookup(encoding);
803 }
804 }
805 PyErr_Clear(); /* Not a fatal error if codec isn't available */
806
Christian Heimesdb233082007-11-13 02:34:21 +0000807 PySys_SetObject("__stderr__", std);
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000808 PySys_SetObject("stderr", std);
809 Py_DECREF(std);
Guido van Rossum98297ee2007-11-06 21:34:58 +0000810#endif
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000811
Christian Heimes58cb1b82007-11-13 02:19:40 +0000812 if (0) {
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000813 error:
Christian Heimesdb233082007-11-13 02:34:21 +0000814 status = -1;
815 }
Guido van Rossumce3a72a2007-10-19 23:16:50 +0000816
817 Py_XDECREF(bimod);
818 Py_XDECREF(iomod);
819 return status;
820}
821
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000822/* Parse input from a file and execute it */
823
824int
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000825PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000826 PyCompilerFlags *flags)
827{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000828 if (filename == NULL)
829 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000830 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000831 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000832 if (closeit)
833 fclose(fp);
834 return err;
835 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000836 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000837 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000838}
839
840int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000841PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000842{
Guido van Rossum82598051997-03-05 00:20:32 +0000843 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000844 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000845 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000846
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000847 if (flags == NULL) {
848 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000849 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000850 }
Guido van Rossum82598051997-03-05 00:20:32 +0000851 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000852 if (v == NULL) {
Neal Norwitza369c5a2007-08-25 07:41:59 +0000853 PySys_SetObject("ps1", v = PyUnicode_FromString(">>> "));
Guido van Rossum82598051997-03-05 00:20:32 +0000854 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000855 }
Guido van Rossum82598051997-03-05 00:20:32 +0000856 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000857 if (v == NULL) {
Neal Norwitza369c5a2007-08-25 07:41:59 +0000858 PySys_SetObject("ps2", v = PyUnicode_FromString("... "));
Guido van Rossum82598051997-03-05 00:20:32 +0000859 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000860 }
861 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000862 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000863 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000864 if (ret == E_EOF)
865 return 0;
866 /*
867 if (ret == E_NOMEM)
868 return -1;
869 */
870 }
871}
872
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000873/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000874#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000875 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Guido van Rossum45aecf42006-03-15 04:58:47 +0000876 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000877
Thomas Wouters89f507f2006-12-13 04:49:30 +0000878#if 0
879/* Keep an example of flags with future keyword support. */
880#define PARSER_FLAGS(flags) \
881 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
882 PyPARSE_DONT_IMPLY_DEDENT : 0) \
883 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
884 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
885#endif
886
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000887int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000888PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000889{
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000890 PyObject *m, *d, *v, *w, *oenc = NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000891 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000892 PyArena *arena;
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000893 char *ps1 = "", *ps2 = "", *enc = NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000894 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000895
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000896 if (fp == stdin) {
897 /* Fetch encoding from sys.stdin */
898 v = PySys_GetObject("stdin");
Christian Heimes2be03732007-11-15 02:26:46 +0000899 if (v == NULL || v == Py_None)
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000900 return -1;
901 oenc = PyObject_GetAttrString(v, "encoding");
902 if (!oenc)
903 return -1;
904 enc = PyUnicode_AsString(oenc);
905 }
Guido van Rossum82598051997-03-05 00:20:32 +0000906 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000907 if (v != NULL) {
Thomas Heller519a0422007-11-15 20:48:54 +0000908 v = PyObject_Str(v);
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000909 if (v == NULL)
910 PyErr_Clear();
Guido van Rossumc5724de2007-10-10 21:38:59 +0000911 else if (PyUnicode_Check(v))
912 ps1 = PyUnicode_AsString(v);
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000913 }
Guido van Rossum82598051997-03-05 00:20:32 +0000914 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000915 if (w != NULL) {
Thomas Heller519a0422007-11-15 20:48:54 +0000916 w = PyObject_Str(w);
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000917 if (w == NULL)
918 PyErr_Clear();
Guido van Rossumc5724de2007-10-10 21:38:59 +0000919 else if (PyUnicode_Check(w))
920 ps2 = PyUnicode_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000921 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000922 arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000923 if (arena == NULL) {
924 Py_XDECREF(v);
925 Py_XDECREF(w);
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000926 Py_XDECREF(oenc);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000927 return -1;
928 }
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000929 mod = PyParser_ASTFromFile(fp, filename, enc,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000930 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000931 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000932 Py_XDECREF(v);
933 Py_XDECREF(w);
Martin v. Löwis85bcc662007-09-04 09:18:06 +0000934 Py_XDECREF(oenc);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000935 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000936 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000937 if (errcode == E_EOF) {
938 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000939 return E_EOF;
940 }
Guido van Rossum82598051997-03-05 00:20:32 +0000941 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000942 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000943 }
Guido van Rossum82598051997-03-05 00:20:32 +0000944 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000945 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000946 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000947 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000948 }
Guido van Rossum82598051997-03-05 00:20:32 +0000949 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000950 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000951 PyArena_Free(arena);
Amaury Forgeot d'Arc7fedbe52008-04-10 21:03:09 +0000952 flush_io();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000953 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000954 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000955 return -1;
956 }
Guido van Rossum82598051997-03-05 00:20:32 +0000957 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000958 return 0;
959}
960
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000961/* Check whether a file maybe a pyc file: Look at the extension,
962 the file type, and, if we may close it, at the first few bytes. */
963
964static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000965maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000966{
967 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
968 return 1;
969
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000970 /* Only look into the file if we are allowed to close it, since
971 it then should also be seekable. */
972 if (closeit) {
973 /* Read only two bytes of the magic. If the file was opened in
974 text mode, the bytes 3 and 4 of the magic (\r\n) might not
975 be read as they are on disk. */
976 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
977 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000978 /* Mess: In case of -x, the stream is NOT at its start now,
979 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000980 which makes the current stream position formally undefined,
981 and a x-platform nightmare.
982 Unfortunately, we have no direct way to know whether -x
983 was specified. So we use a terrible hack: if the current
984 stream position is not 0, we assume -x was specified, and
985 give up. Bug 132850 on SourceForge spells out the
986 hopelessness of trying anything else (fseek and ftell
987 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000988 */
Tim Peters3e876562001-02-11 04:35:39 +0000989 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000990 if (ftell(fp) == 0) {
991 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000992 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000993 ispyc = 1;
994 rewind(fp);
995 }
Tim Peters3e876562001-02-11 04:35:39 +0000996 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000997 }
998 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000999}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +00001000
Guido van Rossum0df002c2000-08-27 19:21:52 +00001001int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001002PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001003 PyCompilerFlags *flags)
1004{
Guido van Rossum82598051997-03-05 00:20:32 +00001005 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001006 const char *ext;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001007 int set_file_name = 0, ret;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001008
Guido van Rossum82598051997-03-05 00:20:32 +00001009 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001010 if (m == NULL)
1011 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +00001012 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +00001013 if (PyDict_GetItemString(d, "__file__") == NULL) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +00001014 PyObject *f;
1015 f = PyUnicode_DecodeFSDefault(filename);
Fred Drake8ed02042002-10-17 21:24:58 +00001016 if (f == NULL)
1017 return -1;
1018 if (PyDict_SetItemString(d, "__file__", f) < 0) {
1019 Py_DECREF(f);
1020 return -1;
1021 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00001022 set_file_name = 1;
Fred Drake8ed02042002-10-17 21:24:58 +00001023 Py_DECREF(f);
1024 }
Guido van Rossumfdef2711994-09-14 13:31:04 +00001025 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +00001026 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +00001027 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +00001028 if (closeit)
1029 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +00001030 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +00001031 fprintf(stderr, "python: Can't reopen .pyc file\n");
Guido van Rossumd8faa362007-04-27 19:54:29 +00001032 ret = -1;
1033 goto done;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001034 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +00001035 /* Turn on optimization if a .pyo file is given */
1036 if (strcmp(ext, ".pyo") == 0)
1037 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001038 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001039 } else {
Tim Petersd08e3822003-04-17 15:24:21 +00001040 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001041 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001042 }
Amaury Forgeot d'Arc7fedbe52008-04-10 21:03:09 +00001043 flush_io();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001044 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +00001045 PyErr_Print();
Guido van Rossumd8faa362007-04-27 19:54:29 +00001046 ret = -1;
1047 goto done;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001048 }
Guido van Rossum82598051997-03-05 00:20:32 +00001049 Py_DECREF(v);
Guido van Rossumd8faa362007-04-27 19:54:29 +00001050 ret = 0;
1051 done:
1052 if (set_file_name && PyDict_DelItemString(d, "__file__"))
1053 PyErr_Clear();
1054 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001055}
1056
1057int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001058PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +00001059{
Guido van Rossum82598051997-03-05 00:20:32 +00001060 PyObject *m, *d, *v;
1061 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001062 if (m == NULL)
1063 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +00001064 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +00001065 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001066 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +00001067 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001068 return -1;
1069 }
Guido van Rossum82598051997-03-05 00:20:32 +00001070 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001071 return 0;
1072}
1073
Barry Warsaw035574d1997-08-29 22:07:17 +00001074static int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001075parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
1076 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +00001077{
1078 long hold;
1079 PyObject *v;
1080
1081 /* old style errors */
1082 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +00001083 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001084 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +00001085
1086 /* new style errors. `err' is an instance */
1087
1088 if (! (v = PyObject_GetAttrString(err, "msg")))
1089 goto finally;
1090 *message = v;
1091
1092 if (!(v = PyObject_GetAttrString(err, "filename")))
1093 goto finally;
1094 if (v == Py_None)
1095 *filename = NULL;
Kurt B. Kaiser43b15092008-01-05 04:32:22 +00001096 else if (! (*filename = PyUnicode_AsString(v)))
Barry Warsaw035574d1997-08-29 22:07:17 +00001097 goto finally;
1098
1099 Py_DECREF(v);
1100 if (!(v = PyObject_GetAttrString(err, "lineno")))
1101 goto finally;
Christian Heimes217cfd12007-12-02 14:31:20 +00001102 hold = PyLong_AsLong(v);
Barry Warsaw035574d1997-08-29 22:07:17 +00001103 Py_DECREF(v);
1104 v = NULL;
1105 if (hold < 0 && PyErr_Occurred())
1106 goto finally;
1107 *lineno = (int)hold;
1108
1109 if (!(v = PyObject_GetAttrString(err, "offset")))
1110 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001111 if (v == Py_None) {
1112 *offset = -1;
1113 Py_DECREF(v);
1114 v = NULL;
1115 } else {
Christian Heimes217cfd12007-12-02 14:31:20 +00001116 hold = PyLong_AsLong(v);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001117 Py_DECREF(v);
1118 v = NULL;
1119 if (hold < 0 && PyErr_Occurred())
1120 goto finally;
1121 *offset = (int)hold;
1122 }
Barry Warsaw035574d1997-08-29 22:07:17 +00001123
1124 if (!(v = PyObject_GetAttrString(err, "text")))
1125 goto finally;
1126 if (v == Py_None)
1127 *text = NULL;
Guido van Rossumc5724de2007-10-10 21:38:59 +00001128 else if (!PyUnicode_Check(v) ||
1129 !(*text = PyUnicode_AsString(v)))
Barry Warsaw035574d1997-08-29 22:07:17 +00001130 goto finally;
1131 Py_DECREF(v);
1132 return 1;
1133
1134finally:
1135 Py_XDECREF(v);
1136 return 0;
1137}
1138
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001139void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001140PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001141{
Guido van Rossuma61691e1998-02-06 22:27:24 +00001142 PyErr_PrintEx(1);
1143}
1144
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001145static void
Martin v. Löwis95292d62002-12-11 14:04:59 +00001146print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001147{
1148 char *nl;
1149 if (offset >= 0) {
1150 if (offset > 0 && offset == (int)strlen(text))
1151 offset--;
1152 for (;;) {
1153 nl = strchr(text, '\n');
1154 if (nl == NULL || nl-text >= offset)
1155 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001156 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001157 text = nl+1;
1158 }
1159 while (*text == ' ' || *text == '\t') {
1160 text++;
1161 offset--;
1162 }
1163 }
1164 PyFile_WriteString(" ", f);
1165 PyFile_WriteString(text, f);
1166 if (*text == '\0' || text[strlen(text)-1] != '\n')
1167 PyFile_WriteString("\n", f);
1168 if (offset == -1)
1169 return;
1170 PyFile_WriteString(" ", f);
1171 offset--;
1172 while (offset > 0) {
1173 PyFile_WriteString(" ", f);
1174 offset--;
1175 }
1176 PyFile_WriteString("^\n", f);
1177}
1178
Guido van Rossum66e8e862001-03-23 17:54:43 +00001179static void
1180handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001181{
Neal Norwitz9589ee22006-03-04 19:01:22 +00001182 PyObject *exception, *value, *tb;
1183 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +00001184
Guido van Rossumd8faa362007-04-27 19:54:29 +00001185 if (Py_InspectFlag)
1186 /* Don't exit if -i flag was given. This flag is set to 0
1187 * when entering interactive mode for inspecting. */
1188 return;
1189
Guido van Rossum66e8e862001-03-23 17:54:43 +00001190 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001191 fflush(stdout);
1192 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001193 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001194 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001195 /* The error code should be in the `code' attribute. */
1196 PyObject *code = PyObject_GetAttrString(value, "code");
1197 if (code) {
1198 Py_DECREF(value);
1199 value = code;
1200 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001201 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001202 }
1203 /* If we failed to dig out the 'code' attribute,
1204 just let the else clause below print the error. */
1205 }
Christian Heimes217cfd12007-12-02 14:31:20 +00001206 if (PyLong_Check(value))
1207 exitcode = (int)PyLong_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001208 else {
1209 PyObject_Print(value, stderr, Py_PRINT_RAW);
1210 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001211 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001212 }
Tim Peterscf615b52003-04-19 18:47:02 +00001213 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001214 /* Restore and clear the exception info, in order to properly decref
1215 * the exception, value, and traceback. If we just exit instead,
1216 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1217 * some finalizers from running.
1218 */
Tim Peterscf615b52003-04-19 18:47:02 +00001219 PyErr_Restore(exception, value, tb);
1220 PyErr_Clear();
1221 Py_Exit(exitcode);
1222 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001223}
1224
1225void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001226PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001227{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001228 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001229
1230 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1231 handle_system_exit();
1232 }
Guido van Rossum82598051997-03-05 00:20:32 +00001233 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001234 if (exception == NULL)
1235 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001236 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001237 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001238 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001239 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001240 if (set_sys_last_vars) {
1241 PySys_SetObject("last_type", exception);
1242 PySys_SetObject("last_value", v);
Guido van Rossumc5724de2007-10-10 21:38:59 +00001243 PySys_SetObject("last_traceback", tb ? tb : Py_None);
Guido van Rossuma61691e1998-02-06 22:27:24 +00001244 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001245 hook = PySys_GetObject("excepthook");
1246 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001247 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001248 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001249 PyObject *result = PyEval_CallObject(hook, args);
1250 if (result == NULL) {
1251 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001252 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1253 handle_system_exit();
1254 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001255 PyErr_Fetch(&exception2, &v2, &tb2);
1256 PyErr_NormalizeException(&exception2, &v2, &tb2);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001257 /* It should not be possible for exception2 or v2
1258 to be NULL. However PyErr_Display() can't
1259 tolerate NULLs, so just be safe. */
1260 if (exception2 == NULL) {
1261 exception2 = Py_None;
1262 Py_INCREF(exception2);
1263 }
1264 if (v2 == NULL) {
1265 v2 = Py_None;
1266 Py_INCREF(v2);
1267 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001268 fflush(stdout);
1269 PySys_WriteStderr("Error in sys.excepthook:\n");
1270 PyErr_Display(exception2, v2, tb2);
1271 PySys_WriteStderr("\nOriginal exception was:\n");
1272 PyErr_Display(exception, v, tb);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001273 Py_DECREF(exception2);
1274 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001275 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001276 }
1277 Py_XDECREF(result);
1278 Py_XDECREF(args);
1279 } else {
1280 PySys_WriteStderr("sys.excepthook is missing\n");
1281 PyErr_Display(exception, v, tb);
1282 }
1283 Py_XDECREF(exception);
1284 Py_XDECREF(v);
1285 Py_XDECREF(tb);
1286}
1287
Thomas Wouters477c8d52006-05-27 19:21:47 +00001288void
1289PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001290{
1291 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001292 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001293 Py_INCREF(value);
Christian Heimes2be03732007-11-15 02:26:46 +00001294 if (f == Py_None) {
1295 /* pass */
1296 }
1297 else if (f == NULL) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00001298 _PyObject_Dump(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001299 fprintf(stderr, "lost sys.stderr\n");
Guido van Rossumce3a72a2007-10-19 23:16:50 +00001300 }
Guido van Rossum3165fe61992-09-25 21:59:05 +00001301 else {
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001302 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001303 if (tb && tb != Py_None)
1304 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001305 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001306 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001307 {
Guido van Rossum82598051997-03-05 00:20:32 +00001308 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001309 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001310 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001311 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001312 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001313 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001314 else {
1315 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001316 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001317 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001318 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001319 else
Guido van Rossum82598051997-03-05 00:20:32 +00001320 PyFile_WriteString(filename, f);
1321 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001322 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001323 PyFile_WriteString(buf, f);
1324 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001325 if (text != NULL)
1326 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001327 Py_DECREF(value);
1328 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001329 /* Can't be bothered to check all those
1330 PyFile_WriteString() calls */
1331 if (PyErr_Occurred())
1332 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001333 }
1334 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001335 if (err) {
1336 /* Don't do anything else */
1337 }
Brett Cannonbf364092006-03-01 04:25:17 +00001338 else if (PyExceptionClass_Check(exception)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001339 PyObject* moduleName;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001340 char* className = PyExceptionClass_Name(exception);
1341 if (className != NULL) {
1342 char *dot = strrchr(className, '.');
1343 if (dot != NULL)
1344 className = dot+1;
1345 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001346
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001347 moduleName = PyObject_GetAttrString(exception, "__module__");
Guido van Rossumc5724de2007-10-10 21:38:59 +00001348 if (moduleName == NULL || !PyUnicode_Check(moduleName))
1349 {
1350 Py_DECREF(moduleName);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001351 err = PyFile_WriteString("<unknown>", f);
Guido van Rossumc5724de2007-10-10 21:38:59 +00001352 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001353 else {
Guido van Rossumc5724de2007-10-10 21:38:59 +00001354 char* modstr = PyUnicode_AsString(moduleName);
Georg Brandl1a3284e2007-12-02 09:40:06 +00001355 if (modstr && strcmp(modstr, "builtins"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001356 {
1357 err = PyFile_WriteString(modstr, f);
1358 err += PyFile_WriteString(".", f);
1359 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001360 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001361 }
1362 if (err == 0) {
1363 if (className == NULL)
1364 err = PyFile_WriteString("<unknown>", f);
1365 else
Brett Cannonbf364092006-03-01 04:25:17 +00001366 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001367 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001368 }
1369 else
1370 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001371 if (err == 0 && (value != Py_None)) {
Thomas Heller519a0422007-11-15 20:48:54 +00001372 PyObject *s = PyObject_Str(value);
Brett Cannon2e63b732006-03-02 18:34:57 +00001373 /* only print colon if the str() of the
1374 object is not the empty string
1375 */
1376 if (s == NULL)
1377 err = -1;
Guido van Rossumc5724de2007-10-10 21:38:59 +00001378 else if (!PyUnicode_Check(s) ||
1379 PyUnicode_GetSize(s) != 0)
Brett Cannon2e63b732006-03-02 18:34:57 +00001380 err = PyFile_WriteString(": ", f);
1381 if (err == 0)
1382 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1383 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001384 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00001385 /* try to write a newline in any case */
1386 err += PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001387 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001388 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001389 /* If an error happened here, don't show it.
1390 XXX This is wrong, but too many callers rely on this behavior. */
1391 if (err != 0)
1392 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001393}
1394
Guido van Rossum82598051997-03-05 00:20:32 +00001395PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001396PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001397 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001398{
Neal Norwitze92fba02006-03-04 18:52:26 +00001399 PyObject *ret = NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001400 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001401 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001402 if (arena == NULL)
1403 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001404
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001405 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001406 if (mod != NULL)
1407 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001408 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001409 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001410}
1411
1412PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001413PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001414 PyObject *locals, int closeit, PyCompilerFlags *flags)
1415{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001416 PyObject *ret;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001417 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001418 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001419 if (arena == NULL)
1420 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001421
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001422 mod = PyParser_ASTFromFile(fp, filename, NULL, start, 0, 0,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001423 flags, NULL, arena);
Guido van Rossumd8faa362007-04-27 19:54:29 +00001424 if (closeit)
1425 fclose(fp);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001426 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001427 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001428 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001429 }
Neal Norwitze92fba02006-03-04 18:52:26 +00001430 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001431 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001432 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001433}
1434
Guido van Rossum6c193fa2007-12-05 05:14:58 +00001435static void
1436flush_io(void)
1437{
1438 PyObject *f, *r;
Amaury Forgeot d'Arc9ed77352008-04-04 23:25:27 +00001439 PyObject *type, *value, *traceback;
1440
1441 /* Save the current exception */
1442 PyErr_Fetch(&type, &value, &traceback);
1443
Guido van Rossum6c193fa2007-12-05 05:14:58 +00001444 f = PySys_GetObject("stderr");
1445 if (f != NULL) {
1446 r = PyObject_CallMethod(f, "flush", "");
1447 if (r)
1448 Py_DECREF(r);
1449 else
1450 PyErr_Clear();
1451 }
1452 f = PySys_GetObject("stdout");
1453 if (f != NULL) {
1454 r = PyObject_CallMethod(f, "flush", "");
1455 if (r)
1456 Py_DECREF(r);
1457 else
1458 PyErr_Clear();
1459 }
Amaury Forgeot d'Arc9ed77352008-04-04 23:25:27 +00001460
1461 PyErr_Restore(type, value, traceback);
Guido van Rossum6c193fa2007-12-05 05:14:58 +00001462}
1463
Guido van Rossum82598051997-03-05 00:20:32 +00001464static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001465run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001466 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001467{
Guido van Rossum82598051997-03-05 00:20:32 +00001468 PyCodeObject *co;
1469 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001470 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001471 if (co == NULL)
1472 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001473 v = PyEval_EvalCode(co, globals, locals);
1474 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001475 return v;
1476}
1477
Guido van Rossum82598051997-03-05 00:20:32 +00001478static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001479run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001480 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001481{
Guido van Rossum82598051997-03-05 00:20:32 +00001482 PyCodeObject *co;
1483 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001484 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001485 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001486
Guido van Rossum82598051997-03-05 00:20:32 +00001487 magic = PyMarshal_ReadLongFromFile(fp);
1488 if (magic != PyImport_GetMagicNumber()) {
1489 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001490 "Bad magic number in .pyc file");
1491 return NULL;
1492 }
Guido van Rossum82598051997-03-05 00:20:32 +00001493 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001494 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001495 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001496 if (v == NULL || !PyCode_Check(v)) {
1497 Py_XDECREF(v);
1498 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001499 "Bad code object in .pyc file");
1500 return NULL;
1501 }
Guido van Rossum82598051997-03-05 00:20:32 +00001502 co = (PyCodeObject *)v;
1503 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001504 if (v && flags)
1505 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001506 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001507 return v;
1508}
1509
Guido van Rossum82598051997-03-05 00:20:32 +00001510PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001511Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001512 PyCompilerFlags *flags)
1513{
Guido van Rossum82598051997-03-05 00:20:32 +00001514 PyCodeObject *co;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001515 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001516 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001517 if (arena == NULL)
1518 return NULL;
1519
1520 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001521 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001522 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001523 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001524 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001525 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001526 PyObject *result = PyAST_mod2obj(mod);
1527 PyArena_Free(arena);
1528 return result;
1529 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001530 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001531 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001532 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001533}
1534
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001535struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001536Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001537{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001538 struct symtable *st;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001539 mod_ty mod;
Christian Heimes4d6ec852008-03-26 22:34:47 +00001540 PyCompilerFlags flags;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001541 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001542 if (arena == NULL)
1543 return NULL;
1544
Christian Heimesb1b3efc2008-03-26 23:24:27 +00001545 flags.cf_flags = 0;
Christian Heimes4d6ec852008-03-26 22:34:47 +00001546 mod = PyParser_ASTFromString(str, filename, start, &flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001547 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001548 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001549 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001550 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001551 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001552 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001553 return st;
1554}
1555
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001556/* Preferred access to parser is through AST. */
1557mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001558PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001559 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001560{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001561 mod_ty mod;
1562 perrdetail err;
Christian Heimes3a932122008-03-26 23:25:24 +00001563 int iflags = PARSER_FLAGS(flags);
Christian Heimes4d6ec852008-03-26 22:34:47 +00001564
1565 node *n = PyParser_ParseStringFlagsFilenameEx(s, filename,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001566 &_PyParser_Grammar, start, &err,
Christian Heimes4d6ec852008-03-26 22:34:47 +00001567 &iflags);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001568 if (n) {
Christian Heimes4d6ec852008-03-26 22:34:47 +00001569 if (flags) {
1570 flags->cf_flags |= iflags & PyCF_MASK;
1571 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001572 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001573 PyNode_Free(n);
1574 return mod;
1575 }
1576 else {
1577 err_input(&err);
1578 return NULL;
1579 }
1580}
1581
1582mod_ty
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001583PyParser_ASTFromFile(FILE *fp, const char *filename, const char* enc,
1584 int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001585 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001586 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001587{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001588 mod_ty mod;
1589 perrdetail err;
Christian Heimes3a932122008-03-26 23:25:24 +00001590 int iflags = PARSER_FLAGS(flags);
Christian Heimes4d6ec852008-03-26 22:34:47 +00001591
Christian Heimes4d6ec852008-03-26 22:34:47 +00001592 node *n = PyParser_ParseFileFlagsEx(fp, filename, enc,
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001593 &_PyParser_Grammar,
Christian Heimes4d6ec852008-03-26 22:34:47 +00001594 start, ps1, ps2, &err, &iflags);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001595 if (n) {
Christian Heimes4d6ec852008-03-26 22:34:47 +00001596 if (flags) {
1597 flags->cf_flags |= iflags & PyCF_MASK;
1598 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001599 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001600 PyNode_Free(n);
1601 return mod;
1602 }
1603 else {
1604 err_input(&err);
1605 if (errcode)
1606 *errcode = err.error;
1607 return NULL;
1608 }
1609}
1610
Guido van Rossuma110aa61994-08-29 12:50:44 +00001611/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001612
Guido van Rossuma110aa61994-08-29 12:50:44 +00001613node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001614PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001615{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001616 perrdetail err;
Martin v. Löwis85bcc662007-09-04 09:18:06 +00001617 node *n = PyParser_ParseFileFlags(fp, filename, NULL,
1618 &_PyParser_Grammar,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001619 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001620 if (n == NULL)
1621 err_input(&err);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001622
Guido van Rossuma110aa61994-08-29 12:50:44 +00001623 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001624}
1625
Guido van Rossuma110aa61994-08-29 12:50:44 +00001626/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001627
Guido van Rossuma110aa61994-08-29 12:50:44 +00001628node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001629PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001630{
Tim Petersfe2127d2001-07-16 05:37:24 +00001631 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001632 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1633 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001634 if (n == NULL)
1635 err_input(&err);
1636 return n;
1637}
1638
1639node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001640PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001641 int start, int flags)
1642{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001643 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001644 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1645 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001646 if (n == NULL)
1647 err_input(&err);
1648 return n;
1649}
1650
1651node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001652PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001653{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001654 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001655}
1656
Guido van Rossum66ebd912003-04-17 16:02:26 +00001657/* May want to move a more generalized form of this to parsetok.c or
1658 even parser modules. */
1659
1660void
1661PyParser_SetError(perrdetail *err)
1662{
1663 err_input(err);
1664}
1665
Guido van Rossuma110aa61994-08-29 12:50:44 +00001666/* Set the error appropriate to the given input error code (see errcode.h) */
1667
1668static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001669err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001670{
Martin v. Löwis5deb2102007-08-31 11:17:42 +00001671 PyObject *v, *w, *errtype, *errtext;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001672 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001673 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001674 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001675 switch (err->error) {
1676 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001677 errtype = PyExc_IndentationError;
1678 if (err->expected == INDENT)
1679 msg = "expected an indented block";
1680 else if (err->token == INDENT)
1681 msg = "unexpected indent";
1682 else if (err->token == DEDENT)
1683 msg = "unexpected unindent";
1684 else {
1685 errtype = PyExc_SyntaxError;
1686 msg = "invalid syntax";
1687 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001688 break;
1689 case E_TOKEN:
1690 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001691 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001692 case E_EOFS:
Alexandre Vassalotti8ae3e052008-05-16 00:41:41 +00001693 msg = "EOF while scanning triple-quoted string literal";
Skip Montanaro118ec702002-08-15 01:20:16 +00001694 break;
1695 case E_EOLS:
Alexandre Vassalotti8ae3e052008-05-16 00:41:41 +00001696 msg = "EOL while scanning string literal";
Skip Montanaro118ec702002-08-15 01:20:16 +00001697 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001698 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001699 if (!PyErr_Occurred())
1700 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001701 return;
1702 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001703 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001704 return;
1705 case E_EOF:
1706 msg = "unexpected EOF while parsing";
1707 break;
Fred Drake85f36392000-07-11 17:53:00 +00001708 case E_TABSPACE:
1709 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001710 msg = "inconsistent use of tabs and spaces in indentation";
1711 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001712 case E_OVERFLOW:
1713 msg = "expression too long";
1714 break;
Fred Drake85f36392000-07-11 17:53:00 +00001715 case E_DEDENT:
1716 errtype = PyExc_IndentationError;
1717 msg = "unindent does not match any outer indentation level";
1718 break;
1719 case E_TOODEEP:
1720 errtype = PyExc_IndentationError;
1721 msg = "too many levels of indentation";
1722 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001723 case E_DECODE: {
1724 PyObject *type, *value, *tb;
1725 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001726 if (value != NULL) {
Thomas Heller519a0422007-11-15 20:48:54 +00001727 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001728 if (u != NULL) {
Guido van Rossumc5724de2007-10-10 21:38:59 +00001729 msg = PyUnicode_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001730 }
1731 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001732 if (msg == NULL)
1733 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001734 Py_XDECREF(type);
1735 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001736 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001737 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001738 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001739 case E_LINECONT:
1740 msg = "unexpected character after line continuation character";
1741 break;
Martin v. Löwis47383402007-08-15 07:32:56 +00001742
1743 case E_IDENTIFIER:
1744 msg = "invalid character in identifier";
1745 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001746 default:
1747 fprintf(stderr, "error=%d\n", err->error);
1748 msg = "unknown parsing error";
1749 break;
1750 }
Martin v. Löwis5deb2102007-08-31 11:17:42 +00001751 /* err->text may not be UTF-8 in case of decoding errors.
1752 Explicitly convert to an object. */
1753 if (!err->text) {
1754 errtext = Py_None;
1755 Py_INCREF(Py_None);
1756 } else {
1757 errtext = PyUnicode_DecodeUTF8(err->text, strlen(err->text),
1758 "replace");
1759 }
1760 v = Py_BuildValue("(ziiN)", err->filename,
1761 err->lineno, err->offset, errtext);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001762 if (err->text != NULL) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001763 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001764 err->text = NULL;
1765 }
1766 w = NULL;
1767 if (v != NULL)
1768 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001769 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001770 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001771 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001772 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001773}
1774
1775/* Print fatal error message and abort */
1776
1777void
Tim Peters7c321a82002-07-09 02:57:01 +00001778Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001779{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001780 fprintf(stderr, "Fatal Python error: %s\n", msg);
Guido van Rossumce3a72a2007-10-19 23:16:50 +00001781 if (PyErr_Occurred()) {
1782 PyErr_Print();
1783 }
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001784#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001785 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001786 OutputDebugString(msg);
1787 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001788#ifdef _DEBUG
1789 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001790#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001791#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001792 abort();
1793}
1794
1795/* Clean up and exit */
1796
Guido van Rossuma110aa61994-08-29 12:50:44 +00001797#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001798#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001799#endif
1800
Collin Winter670e6922007-03-21 02:57:17 +00001801static void (*pyexitfunc)(void) = NULL;
1802/* For the atexit module. */
1803void _Py_PyAtExit(void (*func)(void))
1804{
1805 pyexitfunc = func;
1806}
1807
1808static void
1809call_py_exitfuncs(void)
1810{
Guido van Rossum98297ee2007-11-06 21:34:58 +00001811 if (pyexitfunc == NULL)
Collin Winter670e6922007-03-21 02:57:17 +00001812 return;
1813
1814 (*pyexitfunc)();
1815 PyErr_Clear();
1816}
1817
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001818#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001819static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001820static int nexitfuncs = 0;
1821
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001822int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001823{
1824 if (nexitfuncs >= NEXITFUNCS)
1825 return -1;
1826 exitfuncs[nexitfuncs++] = func;
1827 return 0;
1828}
1829
Guido van Rossumcc283f51997-08-05 02:22:03 +00001830static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001831call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001832{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001833 while (nexitfuncs > 0)
1834 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001835
1836 fflush(stdout);
1837 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001838}
1839
1840void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001841Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001842{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001843 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001844
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001845 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001846}
1847
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001848static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001849initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001850{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001851#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001852 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001853#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001854#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001855 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001856#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001857#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001858 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001859#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001860 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001861}
1862
Guido van Rossum7433b121997-02-14 19:45:36 +00001863
1864/*
1865 * The file descriptor fd is considered ``interactive'' if either
1866 * a) isatty(fd) is TRUE, or
1867 * b) the -i flag was given, and the filename associated with
1868 * the descriptor is NULL or "<stdin>" or "???".
1869 */
1870int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001871Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001872{
1873 if (isatty((int)fileno(fp)))
1874 return 1;
1875 if (!Py_InteractiveFlag)
1876 return 0;
1877 return (filename == NULL) ||
1878 (strcmp(filename, "<stdin>") == 0) ||
1879 (strcmp(filename, "???") == 0);
1880}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001881
1882
Tim Petersd08e3822003-04-17 15:24:21 +00001883#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001884#if defined(WIN32) && defined(_MSC_VER)
1885
1886/* Stack checking for Microsoft C */
1887
1888#include <malloc.h>
1889#include <excpt.h>
1890
Fred Drakee8de31c2000-08-31 05:38:39 +00001891/*
1892 * Return non-zero when we run out of memory on the stack; zero otherwise.
1893 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001894int
Fred Drake399739f2000-08-31 05:52:44 +00001895PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001896{
1897 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001898 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001899 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001900 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001901 return 0;
Christian Heimes7131fd92008-02-19 14:21:46 +00001902 } __except (GetExceptionCode() == STATUS_STACK_OVERFLOW ?
Trent Nelson39e307e2008-03-19 06:45:48 +00001903 EXCEPTION_EXECUTE_HANDLER :
1904 EXCEPTION_CONTINUE_SEARCH) {
Christian Heimes7131fd92008-02-19 14:21:46 +00001905 int errcode = _resetstkoflw();
1906 if (errcode)
1907 {
1908 Py_FatalError("Could not reset the stack!");
1909 }
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001910 }
1911 return 1;
1912}
1913
1914#endif /* WIN32 && _MSC_VER */
1915
1916/* Alternate implementations can be added here... */
1917
1918#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001919
1920
1921/* Wrappers around sigaction() or signal(). */
1922
1923PyOS_sighandler_t
1924PyOS_getsig(int sig)
1925{
1926#ifdef HAVE_SIGACTION
1927 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001928 if (sigaction(sig, NULL, &context) == -1)
1929 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001930 return context.sa_handler;
1931#else
1932 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001933/* Special signal handling for the secure CRT in Visual Studio 2005 */
1934#if defined(_MSC_VER) && _MSC_VER >= 1400
1935 switch (sig) {
1936 /* Only these signals are valid */
1937 case SIGINT:
1938 case SIGILL:
1939 case SIGFPE:
1940 case SIGSEGV:
1941 case SIGTERM:
1942 case SIGBREAK:
1943 case SIGABRT:
1944 break;
1945 /* Don't call signal() with other values or it will assert */
1946 default:
1947 return SIG_ERR;
1948 }
1949#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001950 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001951 if (handler != SIG_ERR)
1952 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001953 return handler;
1954#endif
1955}
1956
1957PyOS_sighandler_t
1958PyOS_setsig(int sig, PyOS_sighandler_t handler)
1959{
1960#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001961 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001962 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001963 sigemptyset(&context.sa_mask);
1964 context.sa_flags = 0;
1965 if (sigaction(sig, &context, &ocontext) == -1)
1966 return SIG_ERR;
1967 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001968#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001969 PyOS_sighandler_t oldhandler;
1970 oldhandler = signal(sig, handler);
1971#ifdef HAVE_SIGINTERRUPT
1972 siginterrupt(sig, 1);
1973#endif
1974 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001975#endif
1976}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001977
1978/* Deprecated C API functions still provided for binary compatiblity */
1979
1980#undef PyParser_SimpleParseFile
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001981PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001982PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1983{
1984 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1985}
1986
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001987#undef PyParser_SimpleParseString
1988PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001989PyParser_SimpleParseString(const char *str, int start)
1990{
1991 return PyParser_SimpleParseStringFlags(str, start, 0);
1992}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001993
1994#undef PyRun_AnyFile
1995PyAPI_FUNC(int)
1996PyRun_AnyFile(FILE *fp, const char *name)
1997{
1998 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1999}
2000
2001#undef PyRun_AnyFileEx
2002PyAPI_FUNC(int)
2003PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
2004{
2005 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
2006}
2007
2008#undef PyRun_AnyFileFlags
2009PyAPI_FUNC(int)
2010PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
2011{
2012 return PyRun_AnyFileExFlags(fp, name, 0, flags);
2013}
2014
2015#undef PyRun_File
2016PyAPI_FUNC(PyObject *)
2017PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
2018{
2019 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
2020}
2021
2022#undef PyRun_FileEx
2023PyAPI_FUNC(PyObject *)
2024PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
2025{
2026 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
2027}
2028
2029#undef PyRun_FileFlags
2030PyAPI_FUNC(PyObject *)
2031PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
2032 PyCompilerFlags *flags)
2033{
2034 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
2035}
2036
2037#undef PyRun_SimpleFile
2038PyAPI_FUNC(int)
2039PyRun_SimpleFile(FILE *f, const char *p)
2040{
2041 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
2042}
2043
2044#undef PyRun_SimpleFileEx
2045PyAPI_FUNC(int)
2046PyRun_SimpleFileEx(FILE *f, const char *p, int c)
2047{
2048 return PyRun_SimpleFileExFlags(f, p, c, NULL);
2049}
2050
2051
2052#undef PyRun_String
2053PyAPI_FUNC(PyObject *)
2054PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
2055{
2056 return PyRun_StringFlags(str, s, g, l, NULL);
2057}
2058
2059#undef PyRun_SimpleString
2060PyAPI_FUNC(int)
2061PyRun_SimpleString(const char *s)
2062{
2063 return PyRun_SimpleStringFlags(s, NULL);
2064}
2065
2066#undef Py_CompileString
2067PyAPI_FUNC(PyObject *)
2068Py_CompileString(const char *str, const char *p, int s)
2069{
2070 return Py_CompileStringFlags(str, p, s, NULL);
2071}
2072
2073#undef PyRun_InteractiveOne
2074PyAPI_FUNC(int)
2075PyRun_InteractiveOne(FILE *f, const char *p)
2076{
2077 return PyRun_InteractiveOneFlags(f, p, NULL);
2078}
2079
2080#undef PyRun_InteractiveLoop
2081PyAPI_FUNC(int)
2082PyRun_InteractiveLoop(FILE *f, const char *p)
2083{
2084 return PyRun_InteractiveLoopFlags(f, p, NULL);
2085}
2086
2087#ifdef __cplusplus
2088}
2089#endif