blob: db21dafaae96bb4b95ba28f62fe3be039cbe60bd [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"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000020
Thomas Wouters0e3f5912006-08-11 14:57:12 +000021#ifdef HAVE_SIGNAL_H
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000022#include <signal.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +000023#endif
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000024
Martin v. Löwis73d538b2003-03-05 15:13:47 +000025#ifdef HAVE_LANGINFO_H
26#include <locale.h>
27#include <langinfo.h>
28#endif
29
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000030#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000031#undef BYTE
32#include "windows.h"
33#endif
34
Neal Norwitz4281cef2006-03-04 19:58:13 +000035#ifndef Py_REF_DEBUG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000036#define PRINT_TOTAL_REFS()
Neal Norwitz4281cef2006-03-04 19:58:13 +000037#else /* Py_REF_DEBUG */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000038#define PRINT_TOTAL_REFS() fprintf(stderr, \
39 "[%" PY_FORMAT_SIZE_T "d refs]\n", \
40 _Py_GetRefTotal())
41#endif
42
43#ifdef __cplusplus
44extern "C" {
Neal Norwitz4281cef2006-03-04 19:58:13 +000045#endif
46
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000047extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000048
Guido van Rossum82598051997-03-05 00:20:32 +000049extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000050
Guido van Rossumb73cc041993-11-01 16:28:59 +000051/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000052static void initmain(void);
53static void initsite(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000054static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000055 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000056static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000057 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000058static void err_input(perrdetail *);
59static void initsigs(void);
Collin Winter670e6922007-03-21 02:57:17 +000060static void call_py_exitfuncs(void);
Tim Petersdbd9ba62000-07-09 03:09:57 +000061static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000062extern void _PyUnicode_Init(void);
63extern void _PyUnicode_Fini(void);
Guido van Rossumddefaf32007-01-14 03:31:43 +000064extern int _PyLong_Init(void);
65extern void PyLong_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000066
Mark Hammond8d98d2c2003-04-19 15:41:53 +000067#ifdef WITH_THREAD
68extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
69extern void _PyGILState_Fini(void);
70#endif /* WITH_THREAD */
71
Guido van Rossum82598051997-03-05 00:20:32 +000072int Py_DebugFlag; /* Needed by parser.c */
73int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000074int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumd8faa362007-04-27 19:54:29 +000075int Py_InspectFlag; /* Needed to determine whether to exit at SystemError */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000076int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000077int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000078int Py_FrozenFlag; /* Needed by getpath.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000079int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000080
Mark Hammonda43fd0c2003-02-19 00:33:33 +000081/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000082 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000083*/
Mark Hammondedd07732003-07-15 23:03:55 +000084static PyObject *warnings_module = NULL;
85
86/* Returns a borrowed reference to the 'warnings' module, or NULL.
87 If the module is returned, it is guaranteed to have been obtained
88 without acquiring the import lock
89*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000090PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000091{
92 PyObject *typ, *val, *tb;
93 PyObject *all_modules;
94 /* If we managed to get the module at init time, just use it */
95 if (warnings_module)
96 return warnings_module;
97 /* If it wasn't available at init time, it may be available
98 now in sys.modules (common scenario is frozen apps: import
99 at init time fails, but the frozen init code sets up sys.path
100 correctly, then does an implicit import of warnings for us
101 */
102 /* Save and restore any exceptions */
103 PyErr_Fetch(&typ, &val, &tb);
104
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000105 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000106 if (all_modules) {
107 warnings_module = PyDict_GetItemString(all_modules, "warnings");
108 /* We keep a ref in the global */
109 Py_XINCREF(warnings_module);
110 }
111 PyErr_Restore(typ, val, tb);
112 return warnings_module;
113}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000114
Guido van Rossum25ce5661997-08-02 03:10:38 +0000115static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000116
Thomas Wouters7e474022000-07-16 12:04:32 +0000117/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000118
119int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000120Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000121{
122 return initialized;
123}
124
Guido van Rossum25ce5661997-08-02 03:10:38 +0000125/* Global initializations. Can be undone by Py_Finalize(). Don't
126 call this twice without an intervening Py_Finalize() call. When
127 initializations fail, a fatal error is issued and the function does
128 not return. On return, the first thread and interpreter state have
129 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000130
Guido van Rossum25ce5661997-08-02 03:10:38 +0000131 Locking: you must hold the interpreter lock while calling this.
132 (If the lock has not yet been initialized, that's equivalent to
133 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000134
Guido van Rossum25ce5661997-08-02 03:10:38 +0000135*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000136
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000137static int
138add_flag(int flag, const char *envs)
139{
140 int env = atoi(envs);
141 if (flag < env)
142 flag = env;
143 if (flag < 1)
144 flag = 1;
145 return flag;
146}
147
Guido van Rossuma027efa1997-05-05 20:56:21 +0000148void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000149Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000150{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000151 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000152 PyThreadState *tstate;
153 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000154 char *p;
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000155#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000156 char *codeset;
157 char *saved_locale;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000158#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000159 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000160
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000161 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000162 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000163 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000164
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000165 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000166 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000167 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000168 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000169 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000170 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000171
Guido van Rossuma027efa1997-05-05 20:56:21 +0000172 interp = PyInterpreterState_New();
173 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000174 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000175
Guido van Rossuma027efa1997-05-05 20:56:21 +0000176 tstate = PyThreadState_New(interp);
177 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000178 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000179 (void) PyThreadState_Swap(tstate);
180
Guido van Rossum70d893a2001-08-16 08:21:42 +0000181 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000182
Neal Norwitzb2501f42002-12-31 03:42:13 +0000183 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000184 Py_FatalError("Py_Initialize: can't init frames");
185
Guido van Rossumddefaf32007-01-14 03:31:43 +0000186 if (!_PyLong_Init())
187 Py_FatalError("Py_Initialize: can't init longs");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000188
Neal Norwitz6968b052007-02-27 19:02:19 +0000189 if (!PyBytes_Init())
190 Py_FatalError("Py_Initialize: can't init bytes");
191
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000192 _PyFloat_Init();
193
Guido van Rossum25ce5661997-08-02 03:10:38 +0000194 interp->modules = PyDict_New();
195 if (interp->modules == NULL)
196 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000197 interp->modules_reloading = PyDict_New();
198 if (interp->modules_reloading == NULL)
199 Py_FatalError("Py_Initialize: can't make modules_reloading dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000200
Guido van Rossumc94044c2000-03-10 23:03:54 +0000201 /* Init Unicode implementation; relies on the codec registry */
202 _PyUnicode_Init();
203
Barry Warsawf242aa02000-05-25 23:09:49 +0000204 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000205 if (bimod == NULL)
206 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000207 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000208 if (interp->builtins == NULL)
209 Py_FatalError("Py_Initialize: can't initialize builtins dict");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000210 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000211
212 sysmod = _PySys_Init();
213 if (sysmod == NULL)
214 Py_FatalError("Py_Initialize: can't initialize sys");
215 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000216 if (interp->sysdict == NULL)
217 Py_FatalError("Py_Initialize: can't initialize sys dict");
Guido van Rossum25ce5661997-08-02 03:10:38 +0000218 Py_INCREF(interp->sysdict);
219 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000220 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000221 PyDict_SetItemString(interp->sysdict, "modules",
222 interp->modules);
223
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000224 _PyImport_Init();
225
Barry Warsawf242aa02000-05-25 23:09:49 +0000226 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000227 _PyExc_Init();
Barry Warsawf242aa02000-05-25 23:09:49 +0000228
Barry Warsaw035574d1997-08-29 22:07:17 +0000229 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000230 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000231
Just van Rossum52e14d62002-12-30 22:08:05 +0000232 _PyImportHooks_Init();
233
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000234 if (install_sigs)
235 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000236
237 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000238 if (!Py_NoSiteFlag)
239 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000240
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000241 /* auto-thread-state API, if available */
242#ifdef WITH_THREAD
243 _PyGILState_Init(interp, tstate);
244#endif /* WITH_THREAD */
245
Mark Hammondedd07732003-07-15 23:03:55 +0000246 warnings_module = PyImport_ImportModule("warnings");
247 if (!warnings_module)
248 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000249
Guido van Rossum8d30cc02007-05-03 17:49:24 +0000250#if defined(HAVE_LANGINFO_H) && defined(CODESET)
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000251 /* On Unix, set the file system encoding according to the
252 user's preference, if the CODESET names a well-known
253 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000254 initialized by other means. Also set the encoding of
255 stdin and stdout if these are terminals. */
256
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000257 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000258 setlocale(LC_CTYPE, "");
259 codeset = nl_langinfo(CODESET);
260 if (codeset && *codeset) {
261 PyObject *enc = PyCodec_Encoder(codeset);
262 if (enc) {
263 codeset = strdup(codeset);
264 Py_DECREF(enc);
265 } else {
266 codeset = NULL;
267 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000268 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000269 } else
270 codeset = NULL;
271 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000272 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000273
274 if (codeset) {
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000275 if (!Py_FileSystemDefaultEncoding)
276 Py_FileSystemDefaultEncoding = codeset;
277 else
278 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000279 }
280#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000281}
282
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000283void
284Py_Initialize(void)
285{
286 Py_InitializeEx(1);
287}
288
289
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000290#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000291extern void dump_counts(FILE*);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000292#endif
293
Guido van Rossum25ce5661997-08-02 03:10:38 +0000294/* Undo the effect of Py_Initialize().
295
296 Beware: if multiple interpreter and/or thread states exist, these
297 are not wiped out; only the current thread and interpreter state
298 are deleted. But since everything else is deleted, those other
299 interpreter and thread states should no longer be used.
300
301 (XXX We should do better, e.g. wipe out all interpreters and
302 threads.)
303
304 Locking: as above.
305
306*/
307
308void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000309Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000310{
311 PyInterpreterState *interp;
312 PyThreadState *tstate;
313
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000314 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000315 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000316
Tim Peters384fd102001-01-21 03:40:37 +0000317 /* The interpreter is still entirely intact at this point, and the
318 * exit funcs may be relying on that. In particular, if some thread
319 * or exit func is still waiting to do an import, the import machinery
320 * expects Py_IsInitialized() to return true. So don't say the
321 * interpreter is uninitialized until after the exit funcs have run.
322 * Note that Threading.py uses an exit func to do a join on all the
323 * threads created thru it, so this also protects pending imports in
324 * the threads created via Threading.
325 */
Collin Winter670e6922007-03-21 02:57:17 +0000326 call_py_exitfuncs();
Tim Peters384fd102001-01-21 03:40:37 +0000327 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000328
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000329 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000330 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000331 interp = tstate->interp;
332
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000333 /* Disable signal handling */
334 PyOS_FiniInterrupts();
335
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000336 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000337 Py_XDECREF(warnings_module);
338 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000339
Guido van Rossume13ddc92003-04-17 17:29:22 +0000340 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000341 * before all modules are destroyed.
342 * XXX If a __del__ or weakref callback is triggered here, and tries to
343 * XXX import a module, bad things can happen, because Python no
344 * XXX longer believes it's initialized.
345 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
346 * XXX is easy to provoke that way. I've also seen, e.g.,
347 * XXX Exception exceptions.ImportError: 'No module named sha'
348 * XXX in <function callback at 0x008F5718> ignored
349 * XXX but I'm unclear on exactly how that one happens. In any case,
350 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000351 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000352 PyGC_Collect();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000353#ifdef COUNT_ALLOCS
354 /* With COUNT_ALLOCS, it helps to run GC multiple times:
355 each collection might release some types from the type
356 list, so they become garbage. */
357 while (PyGC_Collect() > 0)
358 /* nothing */;
359#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000360
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000361 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000362 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000363
Guido van Rossume13ddc92003-04-17 17:29:22 +0000364 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000365 * new-style class definitions, for example.
366 * XXX This is disabled because it caused too many problems. If
367 * XXX a __del__ or weakref callback triggers here, Python code has
368 * XXX a hard time running, because even the sys module has been
369 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
370 * XXX One symptom is a sequence of information-free messages
371 * XXX coming from threads (if a __del__ or callback is invoked,
372 * XXX other threads can execute too, and any exception they encounter
373 * XXX triggers a comedy of errors as subsystem after subsystem
374 * XXX fails to find what it *expects* to find in sys to help report
375 * XXX the exception and consequent unexpected failures). I've also
376 * XXX seen segfaults then, after adding print statements to the
377 * XXX Python code getting called.
378 */
379#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000380 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000381#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000382
Guido van Rossum1707aad1997-12-08 23:43:45 +0000383 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
384 _PyImport_Fini();
385
386 /* Debugging stuff */
387#ifdef COUNT_ALLOCS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000388 dump_counts(stdout);
Guido van Rossum1707aad1997-12-08 23:43:45 +0000389#endif
390
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000391 PRINT_TOTAL_REFS();
Guido van Rossum1707aad1997-12-08 23:43:45 +0000392
Tim Peters9cf25ce2003-04-17 15:21:01 +0000393#ifdef Py_TRACE_REFS
394 /* Display all objects still alive -- this can invoke arbitrary
395 * __repr__ overrides, so requires a mostly-intact interpreter.
396 * Alas, a lot of stuff may still be alive now that will be cleaned
397 * up later.
398 */
Tim Peters269b2a62003-04-17 19:52:29 +0000399 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000400 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000401#endif /* Py_TRACE_REFS */
402
Mark Hammond6cb90292003-04-22 11:18:00 +0000403 /* Cleanup auto-thread-state */
404#ifdef WITH_THREAD
405 _PyGILState_Fini();
406#endif /* WITH_THREAD */
407
Guido van Rossumd922fa42003-04-15 14:10:09 +0000408 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000409 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000410
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000411 /* Now we decref the exception classes. After this point nothing
412 can raise an exception. That's okay, because each Fini() method
413 below has been checked to make sure no exceptions are ever
414 raised.
415 */
416
417 _PyExc_Fini();
418
Guido van Rossumd922fa42003-04-15 14:10:09 +0000419 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000420 PyThreadState_Swap(NULL);
421 PyInterpreterState_Delete(interp);
422
Guido van Rossumd922fa42003-04-15 14:10:09 +0000423 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000424 PyMethod_Fini();
425 PyFrame_Fini();
426 PyCFunction_Fini();
427 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000428 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000429 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000430 PyString_Fini();
Neal Norwitz6968b052007-02-27 19:02:19 +0000431 PyBytes_Fini();
Guido van Rossumddefaf32007-01-14 03:31:43 +0000432 PyLong_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000433 PyFloat_Fini();
434
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000435 /* Cleanup Unicode implementation */
436 _PyUnicode_Fini();
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000437
Guido van Rossumcc283f51997-08-05 02:22:03 +0000438 /* XXX Still allocated:
439 - various static ad-hoc pointers to interned strings
440 - int and float free list blocks
441 - whatever various modules and libraries allocate
442 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000443
444 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000445
Tim Peters269b2a62003-04-17 19:52:29 +0000446#ifdef Py_TRACE_REFS
447 /* Display addresses (& refcnts) of all objects still alive.
448 * An address can be used to find the repr of the object, printed
449 * above by _Py_PrintReferences.
450 */
451 if (Py_GETENV("PYTHONDUMPREFS"))
452 _Py_PrintReferenceAddresses(stderr);
453#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000454#ifdef PYMALLOC_DEBUG
455 if (Py_GETENV("PYTHONMALLOCSTATS"))
456 _PyObject_DebugMallocStats();
457#endif
458
Guido van Rossumcc283f51997-08-05 02:22:03 +0000459 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000460}
461
462/* Create and initialize a new interpreter and thread, and return the
463 new thread. This requires that Py_Initialize() has been called
464 first.
465
466 Unsuccessful initialization yields a NULL pointer. Note that *no*
467 exception information is available even in this case -- the
468 exception information is held in the thread, and there is no
469 thread.
470
471 Locking: as above.
472
473*/
474
475PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000476Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000477{
478 PyInterpreterState *interp;
479 PyThreadState *tstate, *save_tstate;
480 PyObject *bimod, *sysmod;
481
482 if (!initialized)
483 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
484
485 interp = PyInterpreterState_New();
486 if (interp == NULL)
487 return NULL;
488
489 tstate = PyThreadState_New(interp);
490 if (tstate == NULL) {
491 PyInterpreterState_Delete(interp);
492 return NULL;
493 }
494
495 save_tstate = PyThreadState_Swap(tstate);
496
497 /* XXX The following is lax in error checking */
498
499 interp->modules = PyDict_New();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000500 interp->modules_reloading = PyDict_New();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000501
502 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
503 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000504 interp->builtins = PyModule_GetDict(bimod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000505 if (interp->builtins == NULL)
506 goto handle_error;
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000507 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000508 }
509 sysmod = _PyImport_FindExtension("sys", "sys");
510 if (bimod != NULL && sysmod != NULL) {
511 interp->sysdict = PyModule_GetDict(sysmod);
Thomas Wouters89f507f2006-12-13 04:49:30 +0000512 if (interp->sysdict == NULL)
513 goto handle_error;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000514 Py_INCREF(interp->sysdict);
515 PySys_SetPath(Py_GetPath());
516 PyDict_SetItemString(interp->sysdict, "modules",
517 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000518 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000519 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000520 if (!Py_NoSiteFlag)
521 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000522 }
523
524 if (!PyErr_Occurred())
525 return tstate;
526
Thomas Wouters89f507f2006-12-13 04:49:30 +0000527handle_error:
Guido van Rossum25ce5661997-08-02 03:10:38 +0000528 /* Oops, it didn't work. Undo it all. */
529
530 PyErr_Print();
531 PyThreadState_Clear(tstate);
532 PyThreadState_Swap(save_tstate);
533 PyThreadState_Delete(tstate);
534 PyInterpreterState_Delete(interp);
535
536 return NULL;
537}
538
539/* Delete an interpreter and its last thread. This requires that the
540 given thread state is current, that the thread has no remaining
541 frames, and that it is its interpreter's only remaining thread.
542 It is a fatal error to violate these constraints.
543
544 (Py_Finalize() doesn't have these constraints -- it zaps
545 everything, regardless.)
546
547 Locking: as above.
548
549*/
550
551void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000552Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000553{
554 PyInterpreterState *interp = tstate->interp;
555
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000556 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000557 Py_FatalError("Py_EndInterpreter: thread is not current");
558 if (tstate->frame != NULL)
559 Py_FatalError("Py_EndInterpreter: thread still has a frame");
560 if (tstate != interp->tstate_head || tstate->next != NULL)
561 Py_FatalError("Py_EndInterpreter: not the last thread");
562
563 PyImport_Cleanup();
564 PyInterpreterState_Clear(interp);
565 PyThreadState_Swap(NULL);
566 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000567}
568
569static char *progname = "python";
570
571void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000572Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000573{
574 if (pn && *pn)
575 progname = pn;
576}
577
578char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000579Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000580{
581 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000582}
583
Guido van Rossuma61691e1998-02-06 22:27:24 +0000584static char *default_home = NULL;
585
586void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000587Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000588{
589 default_home = home;
590}
591
592char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000593Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000594{
595 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000596 if (home == NULL && !Py_IgnoreEnvironmentFlag)
597 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000598 return home;
599}
600
Guido van Rossum6135a871995-01-09 17:53:26 +0000601/* Create __main__ module */
602
603static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000604initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000605{
Guido van Rossum82598051997-03-05 00:20:32 +0000606 PyObject *m, *d;
607 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000608 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000609 Py_FatalError("can't create __main__ module");
610 d = PyModule_GetDict(m);
611 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000612 PyObject *bimod = PyImport_ImportModule("__builtin__");
613 if (bimod == NULL ||
614 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000615 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000616 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000617 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000618}
619
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000620/* Import the site module (not into __main__ though) */
621
622static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000623initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000624{
625 PyObject *m, *f;
626 m = PyImport_ImportModule("site");
627 if (m == NULL) {
628 f = PySys_GetObject("stderr");
629 if (Py_VerboseFlag) {
630 PyFile_WriteString(
631 "'import site' failed; traceback:\n", f);
632 PyErr_Print();
633 }
634 else {
635 PyFile_WriteString(
636 "'import site' failed; use -v for traceback\n", f);
637 PyErr_Clear();
638 }
639 }
640 else {
641 Py_DECREF(m);
642 }
643}
644
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000645/* Parse input from a file and execute it */
646
647int
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000648PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000649 PyCompilerFlags *flags)
650{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000651 if (filename == NULL)
652 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000653 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000654 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000655 if (closeit)
656 fclose(fp);
657 return err;
658 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000659 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000660 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000661}
662
663int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000664PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000665{
Guido van Rossum82598051997-03-05 00:20:32 +0000666 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000667 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000668 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000669
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000670 if (flags == NULL) {
671 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000672 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000673 }
Guido van Rossum82598051997-03-05 00:20:32 +0000674 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000675 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000676 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
677 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000678 }
Guido van Rossum82598051997-03-05 00:20:32 +0000679 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000680 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000681 PySys_SetObject("ps2", v = PyString_FromString("... "));
682 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000683 }
684 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000685 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000686 PRINT_TOTAL_REFS();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000687 if (ret == E_EOF)
688 return 0;
689 /*
690 if (ret == E_NOMEM)
691 return -1;
692 */
693 }
694}
695
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000696/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000697#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000698 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
Guido van Rossum45aecf42006-03-15 04:58:47 +0000699 PyPARSE_DONT_IMPLY_DEDENT : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000700
Thomas Wouters89f507f2006-12-13 04:49:30 +0000701#if 0
702/* Keep an example of flags with future keyword support. */
703#define PARSER_FLAGS(flags) \
704 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
705 PyPARSE_DONT_IMPLY_DEDENT : 0) \
706 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
707 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
708#endif
709
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000710int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000711PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000712{
Guido van Rossum82598051997-03-05 00:20:32 +0000713 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000714 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000715 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000716 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000717 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000718
Guido van Rossum82598051997-03-05 00:20:32 +0000719 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000720 if (v != NULL) {
721 v = PyObject_Str(v);
722 if (v == NULL)
723 PyErr_Clear();
724 else if (PyString_Check(v))
725 ps1 = PyString_AsString(v);
726 }
Guido van Rossum82598051997-03-05 00:20:32 +0000727 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000728 if (w != NULL) {
729 w = PyObject_Str(w);
730 if (w == NULL)
731 PyErr_Clear();
732 else if (PyString_Check(w))
733 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000734 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000735 arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000736 if (arena == NULL) {
737 Py_XDECREF(v);
738 Py_XDECREF(w);
739 return -1;
740 }
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000741 mod = PyParser_ASTFromFile(fp, filename,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000742 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000743 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000744 Py_XDECREF(v);
745 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000746 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000747 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000748 if (errcode == E_EOF) {
749 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000750 return E_EOF;
751 }
Guido van Rossum82598051997-03-05 00:20:32 +0000752 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000753 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000754 }
Guido van Rossum82598051997-03-05 00:20:32 +0000755 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000756 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000757 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000758 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000759 }
Guido van Rossum82598051997-03-05 00:20:32 +0000760 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000761 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000762 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000763 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000764 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000765 return -1;
766 }
Guido van Rossum82598051997-03-05 00:20:32 +0000767 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000768 return 0;
769}
770
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000771/* Check whether a file maybe a pyc file: Look at the extension,
772 the file type, and, if we may close it, at the first few bytes. */
773
774static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000775maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000776{
777 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
778 return 1;
779
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000780 /* Only look into the file if we are allowed to close it, since
781 it then should also be seekable. */
782 if (closeit) {
783 /* Read only two bytes of the magic. If the file was opened in
784 text mode, the bytes 3 and 4 of the magic (\r\n) might not
785 be read as they are on disk. */
786 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
787 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000788 /* Mess: In case of -x, the stream is NOT at its start now,
789 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000790 which makes the current stream position formally undefined,
791 and a x-platform nightmare.
792 Unfortunately, we have no direct way to know whether -x
793 was specified. So we use a terrible hack: if the current
794 stream position is not 0, we assume -x was specified, and
795 give up. Bug 132850 on SourceForge spells out the
796 hopelessness of trying anything else (fseek and ftell
797 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000798 */
Tim Peters3e876562001-02-11 04:35:39 +0000799 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000800 if (ftell(fp) == 0) {
801 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000802 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000803 ispyc = 1;
804 rewind(fp);
805 }
Tim Peters3e876562001-02-11 04:35:39 +0000806 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000807 }
808 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000809}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000810
Guido van Rossum0df002c2000-08-27 19:21:52 +0000811int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000812PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000813 PyCompilerFlags *flags)
814{
Guido van Rossum82598051997-03-05 00:20:32 +0000815 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000816 const char *ext;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000817 int set_file_name = 0, ret;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000818
Guido van Rossum82598051997-03-05 00:20:32 +0000819 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000820 if (m == NULL)
821 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000822 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000823 if (PyDict_GetItemString(d, "__file__") == NULL) {
824 PyObject *f = PyString_FromString(filename);
825 if (f == NULL)
826 return -1;
827 if (PyDict_SetItemString(d, "__file__", f) < 0) {
828 Py_DECREF(f);
829 return -1;
830 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000831 set_file_name = 1;
Fred Drake8ed02042002-10-17 21:24:58 +0000832 Py_DECREF(f);
833 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000834 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000835 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000836 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000837 if (closeit)
838 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000839 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000840 fprintf(stderr, "python: Can't reopen .pyc file\n");
Guido van Rossumd8faa362007-04-27 19:54:29 +0000841 ret = -1;
842 goto done;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000843 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000844 /* Turn on optimization if a .pyo file is given */
845 if (strcmp(ext, ".pyo") == 0)
846 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000847 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000848 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000849 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000850 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000851 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000852 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000853 PyErr_Print();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000854 ret = -1;
855 goto done;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000856 }
Guido van Rossum82598051997-03-05 00:20:32 +0000857 Py_DECREF(v);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000858 ret = 0;
859 done:
860 if (set_file_name && PyDict_DelItemString(d, "__file__"))
861 PyErr_Clear();
862 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000863}
864
865int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000866PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000867{
Guido van Rossum82598051997-03-05 00:20:32 +0000868 PyObject *m, *d, *v;
869 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000870 if (m == NULL)
871 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000872 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000873 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000874 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000875 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000876 return -1;
877 }
Guido van Rossum82598051997-03-05 00:20:32 +0000878 Py_DECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000879 return 0;
880}
881
Barry Warsaw035574d1997-08-29 22:07:17 +0000882static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000883parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
884 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000885{
886 long hold;
887 PyObject *v;
888
889 /* old style errors */
890 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000891 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000892 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000893
894 /* new style errors. `err' is an instance */
895
896 if (! (v = PyObject_GetAttrString(err, "msg")))
897 goto finally;
898 *message = v;
899
900 if (!(v = PyObject_GetAttrString(err, "filename")))
901 goto finally;
902 if (v == Py_None)
903 *filename = NULL;
904 else if (! (*filename = PyString_AsString(v)))
905 goto finally;
906
907 Py_DECREF(v);
908 if (!(v = PyObject_GetAttrString(err, "lineno")))
909 goto finally;
910 hold = PyInt_AsLong(v);
911 Py_DECREF(v);
912 v = NULL;
913 if (hold < 0 && PyErr_Occurred())
914 goto finally;
915 *lineno = (int)hold;
916
917 if (!(v = PyObject_GetAttrString(err, "offset")))
918 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000919 if (v == Py_None) {
920 *offset = -1;
921 Py_DECREF(v);
922 v = NULL;
923 } else {
924 hold = PyInt_AsLong(v);
925 Py_DECREF(v);
926 v = NULL;
927 if (hold < 0 && PyErr_Occurred())
928 goto finally;
929 *offset = (int)hold;
930 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000931
932 if (!(v = PyObject_GetAttrString(err, "text")))
933 goto finally;
934 if (v == Py_None)
935 *text = NULL;
936 else if (! (*text = PyString_AsString(v)))
937 goto finally;
938 Py_DECREF(v);
939 return 1;
940
941finally:
942 Py_XDECREF(v);
943 return 0;
944}
945
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000946void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000947PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000948{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000949 PyErr_PrintEx(1);
950}
951
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000952static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000953print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000954{
955 char *nl;
956 if (offset >= 0) {
957 if (offset > 0 && offset == (int)strlen(text))
958 offset--;
959 for (;;) {
960 nl = strchr(text, '\n');
961 if (nl == NULL || nl-text >= offset)
962 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000963 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000964 text = nl+1;
965 }
966 while (*text == ' ' || *text == '\t') {
967 text++;
968 offset--;
969 }
970 }
971 PyFile_WriteString(" ", f);
972 PyFile_WriteString(text, f);
973 if (*text == '\0' || text[strlen(text)-1] != '\n')
974 PyFile_WriteString("\n", f);
975 if (offset == -1)
976 return;
977 PyFile_WriteString(" ", f);
978 offset--;
979 while (offset > 0) {
980 PyFile_WriteString(" ", f);
981 offset--;
982 }
983 PyFile_WriteString("^\n", f);
984}
985
Guido van Rossum66e8e862001-03-23 17:54:43 +0000986static void
987handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000988{
Neal Norwitz9589ee22006-03-04 19:01:22 +0000989 PyObject *exception, *value, *tb;
990 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +0000991
Guido van Rossumd8faa362007-04-27 19:54:29 +0000992 if (Py_InspectFlag)
993 /* Don't exit if -i flag was given. This flag is set to 0
994 * when entering interactive mode for inspecting. */
995 return;
996
Guido van Rossum66e8e862001-03-23 17:54:43 +0000997 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000998 fflush(stdout);
999 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001000 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +00001001 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001002 /* The error code should be in the `code' attribute. */
1003 PyObject *code = PyObject_GetAttrString(value, "code");
1004 if (code) {
1005 Py_DECREF(value);
1006 value = code;
1007 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +00001008 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001009 }
1010 /* If we failed to dig out the 'code' attribute,
1011 just let the else clause below print the error. */
1012 }
1013 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +00001014 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001015 else {
1016 PyObject_Print(value, stderr, Py_PRINT_RAW);
1017 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001018 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001019 }
Tim Peterscf615b52003-04-19 18:47:02 +00001020 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001021 /* Restore and clear the exception info, in order to properly decref
1022 * the exception, value, and traceback. If we just exit instead,
1023 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1024 * some finalizers from running.
1025 */
Tim Peterscf615b52003-04-19 18:47:02 +00001026 PyErr_Restore(exception, value, tb);
1027 PyErr_Clear();
1028 Py_Exit(exitcode);
1029 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001030}
1031
1032void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001033PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001034{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001035 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001036
1037 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1038 handle_system_exit();
1039 }
Guido van Rossum82598051997-03-05 00:20:32 +00001040 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001041 if (exception == NULL)
1042 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001043 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001044 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001045 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001046 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001047 if (set_sys_last_vars) {
1048 PySys_SetObject("last_type", exception);
1049 PySys_SetObject("last_value", v);
1050 PySys_SetObject("last_traceback", tb);
1051 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001052 hook = PySys_GetObject("excepthook");
1053 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001054 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001055 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001056 PyObject *result = PyEval_CallObject(hook, args);
1057 if (result == NULL) {
1058 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001059 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1060 handle_system_exit();
1061 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001062 PyErr_Fetch(&exception2, &v2, &tb2);
1063 PyErr_NormalizeException(&exception2, &v2, &tb2);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001064 /* It should not be possible for exception2 or v2
1065 to be NULL. However PyErr_Display() can't
1066 tolerate NULLs, so just be safe. */
1067 if (exception2 == NULL) {
1068 exception2 = Py_None;
1069 Py_INCREF(exception2);
1070 }
1071 if (v2 == NULL) {
1072 v2 = Py_None;
1073 Py_INCREF(v2);
1074 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001075 fflush(stdout);
1076 PySys_WriteStderr("Error in sys.excepthook:\n");
1077 PyErr_Display(exception2, v2, tb2);
1078 PySys_WriteStderr("\nOriginal exception was:\n");
1079 PyErr_Display(exception, v, tb);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001080 Py_DECREF(exception2);
1081 Py_DECREF(v2);
Jeremy Hylton07028582001-12-07 15:35:35 +00001082 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001083 }
1084 Py_XDECREF(result);
1085 Py_XDECREF(args);
1086 } else {
1087 PySys_WriteStderr("sys.excepthook is missing\n");
1088 PyErr_Display(exception, v, tb);
1089 }
1090 Py_XDECREF(exception);
1091 Py_XDECREF(v);
1092 Py_XDECREF(tb);
1093}
1094
Thomas Wouters477c8d52006-05-27 19:21:47 +00001095void
1096PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001097{
1098 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001099 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001100 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001101 if (f == NULL)
Martin v. Löwis5b222132007-06-10 09:51:05 +00001102 _PyObject_Dump(value);
1103 if (f == NULL)
Guido van Rossum3165fe61992-09-25 21:59:05 +00001104 fprintf(stderr, "lost sys.stderr\n");
1105 else {
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001106 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001107 if (tb && tb != Py_None)
1108 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001109 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001110 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001111 {
Guido van Rossum82598051997-03-05 00:20:32 +00001112 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001113 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001114 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001115 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001116 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001117 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001118 else {
1119 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001120 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001121 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001122 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001123 else
Guido van Rossum82598051997-03-05 00:20:32 +00001124 PyFile_WriteString(filename, f);
1125 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001126 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001127 PyFile_WriteString(buf, f);
1128 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001129 if (text != NULL)
1130 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001131 Py_DECREF(value);
1132 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001133 /* Can't be bothered to check all those
1134 PyFile_WriteString() calls */
1135 if (PyErr_Occurred())
1136 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001137 }
1138 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001139 if (err) {
1140 /* Don't do anything else */
1141 }
Brett Cannonbf364092006-03-01 04:25:17 +00001142 else if (PyExceptionClass_Check(exception)) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00001143 PyObject* moduleName;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001144 char* className = PyExceptionClass_Name(exception);
1145 if (className != NULL) {
1146 char *dot = strrchr(className, '.');
1147 if (dot != NULL)
1148 className = dot+1;
1149 }
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001150
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001151 moduleName = PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001152 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001153 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001154 else {
1155 char* modstr = PyString_AsString(moduleName);
Neal Norwitz2633c692007-02-26 22:22:47 +00001156 if (modstr && strcmp(modstr, "__builtin__"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001157 {
1158 err = PyFile_WriteString(modstr, f);
1159 err += PyFile_WriteString(".", f);
1160 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001161 Py_DECREF(moduleName);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001162 }
1163 if (err == 0) {
1164 if (className == NULL)
1165 err = PyFile_WriteString("<unknown>", f);
1166 else
Brett Cannonbf364092006-03-01 04:25:17 +00001167 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001168 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001169 }
1170 else
1171 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001172 if (err == 0 && (value != Py_None)) {
1173 PyObject *s = PyObject_Str(value);
1174 /* only print colon if the str() of the
1175 object is not the empty string
1176 */
1177 if (s == NULL)
1178 err = -1;
1179 else if (!PyString_Check(s) ||
1180 PyString_GET_SIZE(s) != 0)
1181 err = PyFile_WriteString(": ", f);
1182 if (err == 0)
1183 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1184 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001185 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00001186 /* try to write a newline in any case */
1187 err += PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001188 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001189 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001190 /* If an error happened here, don't show it.
1191 XXX This is wrong, but too many callers rely on this behavior. */
1192 if (err != 0)
1193 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001194}
1195
Guido van Rossum82598051997-03-05 00:20:32 +00001196PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001197PyRun_StringFlags(const char *str, int start, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001198 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001199{
Neal Norwitze92fba02006-03-04 18:52:26 +00001200 PyObject *ret = NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001201 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001202 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001203 if (arena == NULL)
1204 return NULL;
1205
1206 mod = PyParser_ASTFromString(str, "<string>", start, flags, arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001207 if (mod != NULL)
1208 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001209 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001210 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001211}
1212
1213PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001214PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001215 PyObject *locals, int closeit, PyCompilerFlags *flags)
1216{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001217 PyObject *ret;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001218 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001219 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001220 if (arena == NULL)
1221 return NULL;
1222
1223 mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
1224 flags, NULL, arena);
Guido van Rossumd8faa362007-04-27 19:54:29 +00001225 if (closeit)
1226 fclose(fp);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001227 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001228 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001229 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001230 }
Neal Norwitze92fba02006-03-04 18:52:26 +00001231 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001232 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001233 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001234}
1235
Guido van Rossum82598051997-03-05 00:20:32 +00001236static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001237run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001238 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001239{
Guido van Rossum82598051997-03-05 00:20:32 +00001240 PyCodeObject *co;
1241 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001242 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001243 if (co == NULL)
1244 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001245 v = PyEval_EvalCode(co, globals, locals);
1246 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001247 return v;
1248}
1249
Guido van Rossum82598051997-03-05 00:20:32 +00001250static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001251run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001252 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001253{
Guido van Rossum82598051997-03-05 00:20:32 +00001254 PyCodeObject *co;
1255 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001256 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001257 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001258
Guido van Rossum82598051997-03-05 00:20:32 +00001259 magic = PyMarshal_ReadLongFromFile(fp);
1260 if (magic != PyImport_GetMagicNumber()) {
1261 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001262 "Bad magic number in .pyc file");
1263 return NULL;
1264 }
Guido van Rossum82598051997-03-05 00:20:32 +00001265 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001266 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001267 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001268 if (v == NULL || !PyCode_Check(v)) {
1269 Py_XDECREF(v);
1270 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001271 "Bad code object in .pyc file");
1272 return NULL;
1273 }
Guido van Rossum82598051997-03-05 00:20:32 +00001274 co = (PyCodeObject *)v;
1275 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001276 if (v && flags)
1277 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001278 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001279 return v;
1280}
1281
Guido van Rossum82598051997-03-05 00:20:32 +00001282PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001283Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001284 PyCompilerFlags *flags)
1285{
Guido van Rossum82598051997-03-05 00:20:32 +00001286 PyCodeObject *co;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001287 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001288 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001289 if (arena == NULL)
1290 return NULL;
1291
1292 mod = PyParser_ASTFromString(str, filename, start, flags, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001293 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001294 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001295 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001296 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001297 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001298 PyObject *result = PyAST_mod2obj(mod);
1299 PyArena_Free(arena);
1300 return result;
1301 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001302 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001303 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001304 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001305}
1306
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001307struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001308Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001309{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001310 struct symtable *st;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001311 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001312 PyArena *arena = PyArena_New();
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001313 if (arena == NULL)
1314 return NULL;
1315
1316 mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001317 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001318 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001319 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001320 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001321 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001322 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001323 return st;
1324}
1325
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001326/* Preferred access to parser is through AST. */
1327mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001328PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001329 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001330{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001331 mod_ty mod;
1332 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001333 node *n = PyParser_ParseStringFlagsFilename(s, filename,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001334 &_PyParser_Grammar, start, &err,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001335 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001336 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001337 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001338 PyNode_Free(n);
1339 return mod;
1340 }
1341 else {
1342 err_input(&err);
1343 return NULL;
1344 }
1345}
1346
1347mod_ty
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001348PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001349 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001350 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001351{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001352 mod_ty mod;
1353 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001354 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1355 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001356 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001357 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001358 PyNode_Free(n);
1359 return mod;
1360 }
1361 else {
1362 err_input(&err);
1363 if (errcode)
1364 *errcode = err.error;
1365 return NULL;
1366 }
1367}
1368
Guido van Rossuma110aa61994-08-29 12:50:44 +00001369/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001370
Guido van Rossuma110aa61994-08-29 12:50:44 +00001371node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001372PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001373{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001374 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001375 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1376 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001377 if (n == NULL)
1378 err_input(&err);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001379
Guido van Rossuma110aa61994-08-29 12:50:44 +00001380 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001381}
1382
Guido van Rossuma110aa61994-08-29 12:50:44 +00001383/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001384
Guido van Rossuma110aa61994-08-29 12:50:44 +00001385node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001386PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001387{
Tim Petersfe2127d2001-07-16 05:37:24 +00001388 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001389 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1390 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001391 if (n == NULL)
1392 err_input(&err);
1393 return n;
1394}
1395
1396node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001397PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001398 int start, int flags)
1399{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001400 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001401 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1402 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001403 if (n == NULL)
1404 err_input(&err);
1405 return n;
1406}
1407
1408node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001409PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001410{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001411 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001412}
1413
Guido van Rossum66ebd912003-04-17 16:02:26 +00001414/* May want to move a more generalized form of this to parsetok.c or
1415 even parser modules. */
1416
1417void
1418PyParser_SetError(perrdetail *err)
1419{
1420 err_input(err);
1421}
1422
Guido van Rossuma110aa61994-08-29 12:50:44 +00001423/* Set the error appropriate to the given input error code (see errcode.h) */
1424
1425static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001426err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001427{
Fred Drake85f36392000-07-11 17:53:00 +00001428 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001429 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001430 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001431 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001432 switch (err->error) {
1433 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001434 errtype = PyExc_IndentationError;
1435 if (err->expected == INDENT)
1436 msg = "expected an indented block";
1437 else if (err->token == INDENT)
1438 msg = "unexpected indent";
1439 else if (err->token == DEDENT)
1440 msg = "unexpected unindent";
1441 else {
1442 errtype = PyExc_SyntaxError;
1443 msg = "invalid syntax";
1444 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001445 break;
1446 case E_TOKEN:
1447 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001448 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001449 case E_EOFS:
1450 msg = "EOF while scanning triple-quoted string";
1451 break;
1452 case E_EOLS:
1453 msg = "EOL while scanning single-quoted string";
1454 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001455 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001456 if (!PyErr_Occurred())
1457 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001458 return;
1459 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001460 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001461 return;
1462 case E_EOF:
1463 msg = "unexpected EOF while parsing";
1464 break;
Fred Drake85f36392000-07-11 17:53:00 +00001465 case E_TABSPACE:
1466 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001467 msg = "inconsistent use of tabs and spaces in indentation";
1468 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001469 case E_OVERFLOW:
1470 msg = "expression too long";
1471 break;
Fred Drake85f36392000-07-11 17:53:00 +00001472 case E_DEDENT:
1473 errtype = PyExc_IndentationError;
1474 msg = "unindent does not match any outer indentation level";
1475 break;
1476 case E_TOODEEP:
1477 errtype = PyExc_IndentationError;
1478 msg = "too many levels of indentation";
1479 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001480 case E_DECODE: {
1481 PyObject *type, *value, *tb;
1482 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001483 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001484 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001485 if (u != NULL) {
1486 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001487 }
1488 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001489 if (msg == NULL)
1490 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001491 Py_XDECREF(type);
1492 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001493 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001494 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001495 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001496 case E_LINECONT:
1497 msg = "unexpected character after line continuation character";
1498 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001499 default:
1500 fprintf(stderr, "error=%d\n", err->error);
1501 msg = "unknown parsing error";
1502 break;
1503 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001504 v = Py_BuildValue("(ziiz)", err->filename,
1505 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001506 if (err->text != NULL) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001507 PyObject_FREE(err->text);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001508 err->text = NULL;
1509 }
1510 w = NULL;
1511 if (v != NULL)
1512 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001513 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001514 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001515 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001516 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001517}
1518
1519/* Print fatal error message and abort */
1520
1521void
Tim Peters7c321a82002-07-09 02:57:01 +00001522Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001523{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001524 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001525#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001526 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001527 OutputDebugString(msg);
1528 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001529#ifdef _DEBUG
1530 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001531#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001532#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001533 abort();
1534}
1535
1536/* Clean up and exit */
1537
Guido van Rossuma110aa61994-08-29 12:50:44 +00001538#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001539#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001540#endif
1541
Collin Winter670e6922007-03-21 02:57:17 +00001542static void (*pyexitfunc)(void) = NULL;
1543/* For the atexit module. */
1544void _Py_PyAtExit(void (*func)(void))
1545{
1546 pyexitfunc = func;
1547}
1548
1549static void
1550call_py_exitfuncs(void)
1551{
1552 if (pyexitfunc == NULL)
1553 return;
1554
1555 (*pyexitfunc)();
1556 PyErr_Clear();
1557}
1558
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001559#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001560static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001561static int nexitfuncs = 0;
1562
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001563int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001564{
1565 if (nexitfuncs >= NEXITFUNCS)
1566 return -1;
1567 exitfuncs[nexitfuncs++] = func;
1568 return 0;
1569}
1570
Guido van Rossumcc283f51997-08-05 02:22:03 +00001571static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001572call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001573{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001574 while (nexitfuncs > 0)
1575 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001576
1577 fflush(stdout);
1578 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001579}
1580
1581void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001582Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001583{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001584 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001585
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001586 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001587}
1588
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001589static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001590initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001591{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001592#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001593 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001594#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001595#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001596 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001597#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001598#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001599 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001600#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001601 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001602}
1603
Guido van Rossum7433b121997-02-14 19:45:36 +00001604
1605/*
1606 * The file descriptor fd is considered ``interactive'' if either
1607 * a) isatty(fd) is TRUE, or
1608 * b) the -i flag was given, and the filename associated with
1609 * the descriptor is NULL or "<stdin>" or "???".
1610 */
1611int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001612Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001613{
1614 if (isatty((int)fileno(fp)))
1615 return 1;
1616 if (!Py_InteractiveFlag)
1617 return 0;
1618 return (filename == NULL) ||
1619 (strcmp(filename, "<stdin>") == 0) ||
1620 (strcmp(filename, "???") == 0);
1621}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001622
1623
Tim Petersd08e3822003-04-17 15:24:21 +00001624#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001625#if defined(WIN32) && defined(_MSC_VER)
1626
1627/* Stack checking for Microsoft C */
1628
1629#include <malloc.h>
1630#include <excpt.h>
1631
Fred Drakee8de31c2000-08-31 05:38:39 +00001632/*
1633 * Return non-zero when we run out of memory on the stack; zero otherwise.
1634 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001635int
Fred Drake399739f2000-08-31 05:52:44 +00001636PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001637{
1638 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001639 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001640 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001641 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001642 return 0;
1643 } __except (EXCEPTION_EXECUTE_HANDLER) {
1644 /* just ignore all errors */
1645 }
1646 return 1;
1647}
1648
1649#endif /* WIN32 && _MSC_VER */
1650
1651/* Alternate implementations can be added here... */
1652
1653#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001654
1655
1656/* Wrappers around sigaction() or signal(). */
1657
1658PyOS_sighandler_t
1659PyOS_getsig(int sig)
1660{
1661#ifdef HAVE_SIGACTION
1662 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001663 if (sigaction(sig, NULL, &context) == -1)
1664 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001665 return context.sa_handler;
1666#else
1667 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001668/* Special signal handling for the secure CRT in Visual Studio 2005 */
1669#if defined(_MSC_VER) && _MSC_VER >= 1400
1670 switch (sig) {
1671 /* Only these signals are valid */
1672 case SIGINT:
1673 case SIGILL:
1674 case SIGFPE:
1675 case SIGSEGV:
1676 case SIGTERM:
1677 case SIGBREAK:
1678 case SIGABRT:
1679 break;
1680 /* Don't call signal() with other values or it will assert */
1681 default:
1682 return SIG_ERR;
1683 }
1684#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001685 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001686 if (handler != SIG_ERR)
1687 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001688 return handler;
1689#endif
1690}
1691
1692PyOS_sighandler_t
1693PyOS_setsig(int sig, PyOS_sighandler_t handler)
1694{
1695#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001696 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001697 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001698 sigemptyset(&context.sa_mask);
1699 context.sa_flags = 0;
1700 if (sigaction(sig, &context, &ocontext) == -1)
1701 return SIG_ERR;
1702 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001703#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001704 PyOS_sighandler_t oldhandler;
1705 oldhandler = signal(sig, handler);
1706#ifdef HAVE_SIGINTERRUPT
1707 siginterrupt(sig, 1);
1708#endif
1709 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001710#endif
1711}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001712
1713/* Deprecated C API functions still provided for binary compatiblity */
1714
1715#undef PyParser_SimpleParseFile
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001716PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001717PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1718{
1719 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1720}
1721
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001722#undef PyParser_SimpleParseString
1723PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001724PyParser_SimpleParseString(const char *str, int start)
1725{
1726 return PyParser_SimpleParseStringFlags(str, start, 0);
1727}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001728
1729#undef PyRun_AnyFile
1730PyAPI_FUNC(int)
1731PyRun_AnyFile(FILE *fp, const char *name)
1732{
1733 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
1734}
1735
1736#undef PyRun_AnyFileEx
1737PyAPI_FUNC(int)
1738PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1739{
1740 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
1741}
1742
1743#undef PyRun_AnyFileFlags
1744PyAPI_FUNC(int)
1745PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1746{
1747 return PyRun_AnyFileExFlags(fp, name, 0, flags);
1748}
1749
1750#undef PyRun_File
1751PyAPI_FUNC(PyObject *)
1752PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1753{
1754 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
1755}
1756
1757#undef PyRun_FileEx
1758PyAPI_FUNC(PyObject *)
1759PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1760{
1761 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
1762}
1763
1764#undef PyRun_FileFlags
1765PyAPI_FUNC(PyObject *)
1766PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
1767 PyCompilerFlags *flags)
1768{
1769 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
1770}
1771
1772#undef PyRun_SimpleFile
1773PyAPI_FUNC(int)
1774PyRun_SimpleFile(FILE *f, const char *p)
1775{
1776 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
1777}
1778
1779#undef PyRun_SimpleFileEx
1780PyAPI_FUNC(int)
1781PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1782{
1783 return PyRun_SimpleFileExFlags(f, p, c, NULL);
1784}
1785
1786
1787#undef PyRun_String
1788PyAPI_FUNC(PyObject *)
1789PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1790{
1791 return PyRun_StringFlags(str, s, g, l, NULL);
1792}
1793
1794#undef PyRun_SimpleString
1795PyAPI_FUNC(int)
1796PyRun_SimpleString(const char *s)
1797{
1798 return PyRun_SimpleStringFlags(s, NULL);
1799}
1800
1801#undef Py_CompileString
1802PyAPI_FUNC(PyObject *)
1803Py_CompileString(const char *str, const char *p, int s)
1804{
1805 return Py_CompileStringFlags(str, p, s, NULL);
1806}
1807
1808#undef PyRun_InteractiveOne
1809PyAPI_FUNC(int)
1810PyRun_InteractiveOne(FILE *f, const char *p)
1811{
1812 return PyRun_InteractiveOneFlags(f, p, NULL);
1813}
1814
1815#undef PyRun_InteractiveLoop
1816PyAPI_FUNC(int)
1817PyRun_InteractiveLoop(FILE *f, const char *p)
1818{
1819 return PyRun_InteractiveLoopFlags(f, p, NULL);
1820}
1821
1822#ifdef __cplusplus
1823}
1824#endif