blob: c4e387605c842fa5691fb0a125c5446dfa825a2a [file] [log] [blame]
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001
2/* Python interpreter top-level routines, including init/exit */
3
Guido van Rossum82598051997-03-05 00:20:32 +00004#include "Python.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +00005
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00006#include "Python-ast.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +00007#include "grammar.h"
8#include "node.h"
Fred Drake85f36392000-07-11 17:53:00 +00009#include "token.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000010#include "parsetok.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000011#include "errcode.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000012#include "code.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000013#include "compile.h"
Jeremy Hylton4b38da62001-02-02 18:19:15 +000014#include "symtable.h"
Neal Norwitzadb69fc2005-12-17 20:54:49 +000015#include "pyarena.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000016#include "ast.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +000017#include "eval.h"
Guido van Rossumfdef2711994-09-14 13:31:04 +000018#include "marshal.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000019
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000020#include <signal.h>
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000021
Martin v. Löwis73d538b2003-03-05 15:13:47 +000022#ifdef HAVE_LANGINFO_H
23#include <locale.h>
24#include <langinfo.h>
25#endif
26
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000027#ifdef MS_WINDOWS
Guido van Rossuma44823b1995-03-14 15:01:17 +000028#undef BYTE
29#include "windows.h"
30#endif
31
Neal Norwitz4281cef2006-03-04 19:58:13 +000032#ifndef Py_REF_DEBUG
33# define PRINT_TOTAL_REFS()
34#else /* Py_REF_DEBUG */
35# if defined(MS_WIN64)
Martin v. Löwis99b93c22006-03-05 05:33:54 +000036# define PRINT_TOTAL_REFS() fprintf(stderr, "[%Id refs]\n", _Py_RefTotal);
Neal Norwitz4281cef2006-03-04 19:58:13 +000037# else /* ! MS_WIN64 */
38# define PRINT_TOTAL_REFS() fprintf(stderr, "[%ld refs]\n", _Py_RefTotal);
39# endif /* MS_WIN64 */
40#endif
41
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000042extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000043
Guido van Rossum82598051997-03-05 00:20:32 +000044extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000045
Guido van Rossumb73cc041993-11-01 16:28:59 +000046/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000047static void initmain(void);
48static void initsite(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000049static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000050 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000051static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000052 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000053static void err_input(perrdetail *);
54static void initsigs(void);
55static void call_sys_exitfunc(void);
56static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000057extern void _PyUnicode_Init(void);
58extern void _PyUnicode_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000059
Mark Hammond8d98d2c2003-04-19 15:41:53 +000060#ifdef WITH_THREAD
61extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
62extern void _PyGILState_Fini(void);
63#endif /* WITH_THREAD */
64
Guido van Rossum82598051997-03-05 00:20:32 +000065int Py_DebugFlag; /* Needed by parser.c */
66int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000067int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000068int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000069int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000070int Py_FrozenFlag; /* Needed by getpath.c */
Guido van Rossumb16d1972000-05-01 17:55:15 +000071int Py_UnicodeFlag = 0; /* Needed by compile.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000072int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Tim Peters3caca232001-12-06 06:23:26 +000073/* _XXX Py_QnewFlag should go away in 2.3. It's true iff -Qnew is passed,
74 on the command line, and is used in 2.2 by ceval.c to make all "/" divisions
75 true divisions (which they will be in 2.3). */
76int _Py_QnewFlag = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +000077
Mark Hammonda43fd0c2003-02-19 00:33:33 +000078/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000079 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000080*/
Mark Hammondedd07732003-07-15 23:03:55 +000081static PyObject *warnings_module = NULL;
82
83/* Returns a borrowed reference to the 'warnings' module, or NULL.
84 If the module is returned, it is guaranteed to have been obtained
85 without acquiring the import lock
86*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000087PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000088{
89 PyObject *typ, *val, *tb;
90 PyObject *all_modules;
91 /* If we managed to get the module at init time, just use it */
92 if (warnings_module)
93 return warnings_module;
94 /* If it wasn't available at init time, it may be available
95 now in sys.modules (common scenario is frozen apps: import
96 at init time fails, but the frozen init code sets up sys.path
97 correctly, then does an implicit import of warnings for us
98 */
99 /* Save and restore any exceptions */
100 PyErr_Fetch(&typ, &val, &tb);
101
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000102 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000103 if (all_modules) {
104 warnings_module = PyDict_GetItemString(all_modules, "warnings");
105 /* We keep a ref in the global */
106 Py_XINCREF(warnings_module);
107 }
108 PyErr_Restore(typ, val, tb);
109 return warnings_module;
110}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000111
Guido van Rossum25ce5661997-08-02 03:10:38 +0000112static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000113
Thomas Wouters7e474022000-07-16 12:04:32 +0000114/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000115
116int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000117Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000118{
119 return initialized;
120}
121
Guido van Rossum25ce5661997-08-02 03:10:38 +0000122/* Global initializations. Can be undone by Py_Finalize(). Don't
123 call this twice without an intervening Py_Finalize() call. When
124 initializations fail, a fatal error is issued and the function does
125 not return. On return, the first thread and interpreter state have
126 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000127
Guido van Rossum25ce5661997-08-02 03:10:38 +0000128 Locking: you must hold the interpreter lock while calling this.
129 (If the lock has not yet been initialized, that's equivalent to
130 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000131
Guido van Rossum25ce5661997-08-02 03:10:38 +0000132*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000133
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000134static int
135add_flag(int flag, const char *envs)
136{
137 int env = atoi(envs);
138 if (flag < env)
139 flag = env;
140 if (flag < 1)
141 flag = 1;
142 return flag;
143}
144
Guido van Rossuma027efa1997-05-05 20:56:21 +0000145void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000146Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000147{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000148 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000149 PyThreadState *tstate;
150 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000151 char *p;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000152#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
153 char *codeset;
154 char *saved_locale;
155 PyObject *sys_stream, *sys_isatty;
156#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000157 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000158
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000159 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000160 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000161 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000162
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000163 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000164 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000165 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000166 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000167 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000168 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000169
Guido van Rossuma027efa1997-05-05 20:56:21 +0000170 interp = PyInterpreterState_New();
171 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000172 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000173
Guido van Rossuma027efa1997-05-05 20:56:21 +0000174 tstate = PyThreadState_New(interp);
175 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000176 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000177 (void) PyThreadState_Swap(tstate);
178
Guido van Rossum70d893a2001-08-16 08:21:42 +0000179 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000180
Neal Norwitzb2501f42002-12-31 03:42:13 +0000181 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000182 Py_FatalError("Py_Initialize: can't init frames");
183
Neal Norwitzb2501f42002-12-31 03:42:13 +0000184 if (!_PyInt_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000185 Py_FatalError("Py_Initialize: can't init ints");
186
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000187 _PyFloat_Init();
188
Guido van Rossum25ce5661997-08-02 03:10:38 +0000189 interp->modules = PyDict_New();
190 if (interp->modules == NULL)
191 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000192
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000193#ifdef Py_USING_UNICODE
Guido van Rossumc94044c2000-03-10 23:03:54 +0000194 /* Init Unicode implementation; relies on the codec registry */
195 _PyUnicode_Init();
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000196#endif
Guido van Rossumc94044c2000-03-10 23:03:54 +0000197
Barry Warsawf242aa02000-05-25 23:09:49 +0000198 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000199 if (bimod == NULL)
200 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000201 interp->builtins = PyModule_GetDict(bimod);
202 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000203
204 sysmod = _PySys_Init();
205 if (sysmod == NULL)
206 Py_FatalError("Py_Initialize: can't initialize sys");
207 interp->sysdict = PyModule_GetDict(sysmod);
208 Py_INCREF(interp->sysdict);
209 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000210 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000211 PyDict_SetItemString(interp->sysdict, "modules",
212 interp->modules);
213
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000214 _PyImport_Init();
215
Barry Warsawf242aa02000-05-25 23:09:49 +0000216 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000217 _PyExc_Init();
Barry Warsaw5821bc52001-08-13 23:04:56 +0000218 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000219
Barry Warsaw035574d1997-08-29 22:07:17 +0000220 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000221 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000222
Just van Rossum52e14d62002-12-30 22:08:05 +0000223 _PyImportHooks_Init();
224
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000225 if (install_sigs)
226 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000227
228 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000229 if (!Py_NoSiteFlag)
230 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000231
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000232 /* auto-thread-state API, if available */
233#ifdef WITH_THREAD
234 _PyGILState_Init(interp, tstate);
235#endif /* WITH_THREAD */
236
Mark Hammondedd07732003-07-15 23:03:55 +0000237 warnings_module = PyImport_ImportModule("warnings");
238 if (!warnings_module)
239 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000240
241#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
242 /* On Unix, set the file system encoding according to the
243 user's preference, if the CODESET names a well-known
244 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000245 initialized by other means. Also set the encoding of
246 stdin and stdout if these are terminals. */
247
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000248 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000249 setlocale(LC_CTYPE, "");
250 codeset = nl_langinfo(CODESET);
251 if (codeset && *codeset) {
252 PyObject *enc = PyCodec_Encoder(codeset);
253 if (enc) {
254 codeset = strdup(codeset);
255 Py_DECREF(enc);
256 } else {
257 codeset = NULL;
258 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000259 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000260 } else
261 codeset = NULL;
262 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000263 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000264
265 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000266 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000267 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
268 if (!sys_isatty)
269 PyErr_Clear();
270 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
271 if (!PyFile_SetEncoding(sys_stream, codeset))
272 Py_FatalError("Cannot set codeset of stdin");
273 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000274 Py_XDECREF(sys_isatty);
275
276 sys_stream = PySys_GetObject("stdout");
277 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
278 if (!sys_isatty)
279 PyErr_Clear();
280 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
281 if (!PyFile_SetEncoding(sys_stream, codeset))
282 Py_FatalError("Cannot set codeset of stdout");
283 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000284 Py_XDECREF(sys_isatty);
285
286 if (!Py_FileSystemDefaultEncoding)
287 Py_FileSystemDefaultEncoding = codeset;
288 else
289 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000290 }
291#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000292}
293
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000294void
295Py_Initialize(void)
296{
297 Py_InitializeEx(1);
298}
299
300
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000301#ifdef COUNT_ALLOCS
Tim Petersdbd9ba62000-07-09 03:09:57 +0000302extern void dump_counts(void);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000303#endif
304
Guido van Rossum25ce5661997-08-02 03:10:38 +0000305/* Undo the effect of Py_Initialize().
306
307 Beware: if multiple interpreter and/or thread states exist, these
308 are not wiped out; only the current thread and interpreter state
309 are deleted. But since everything else is deleted, those other
310 interpreter and thread states should no longer be used.
311
312 (XXX We should do better, e.g. wipe out all interpreters and
313 threads.)
314
315 Locking: as above.
316
317*/
318
319void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000320Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000321{
322 PyInterpreterState *interp;
323 PyThreadState *tstate;
324
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000325 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000326 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000327
Tim Peters384fd102001-01-21 03:40:37 +0000328 /* The interpreter is still entirely intact at this point, and the
329 * exit funcs may be relying on that. In particular, if some thread
330 * or exit func is still waiting to do an import, the import machinery
331 * expects Py_IsInitialized() to return true. So don't say the
332 * interpreter is uninitialized until after the exit funcs have run.
333 * Note that Threading.py uses an exit func to do a join on all the
334 * threads created thru it, so this also protects pending imports in
335 * the threads created via Threading.
336 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000337 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000338 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000339
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000340 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000341 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000342 interp = tstate->interp;
343
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000344 /* Disable signal handling */
345 PyOS_FiniInterrupts();
346
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000347 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000348 Py_XDECREF(warnings_module);
349 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000350
Guido van Rossume13ddc92003-04-17 17:29:22 +0000351 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000352 * before all modules are destroyed.
353 * XXX If a __del__ or weakref callback is triggered here, and tries to
354 * XXX import a module, bad things can happen, because Python no
355 * XXX longer believes it's initialized.
356 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
357 * XXX is easy to provoke that way. I've also seen, e.g.,
358 * XXX Exception exceptions.ImportError: 'No module named sha'
359 * XXX in <function callback at 0x008F5718> ignored
360 * XXX but I'm unclear on exactly how that one happens. In any case,
361 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000362 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000363 PyGC_Collect();
364
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000365 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000366 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000367
Guido van Rossume13ddc92003-04-17 17:29:22 +0000368 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000369 * new-style class definitions, for example.
370 * XXX This is disabled because it caused too many problems. If
371 * XXX a __del__ or weakref callback triggers here, Python code has
372 * XXX a hard time running, because even the sys module has been
373 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
374 * XXX One symptom is a sequence of information-free messages
375 * XXX coming from threads (if a __del__ or callback is invoked,
376 * XXX other threads can execute too, and any exception they encounter
377 * XXX triggers a comedy of errors as subsystem after subsystem
378 * XXX fails to find what it *expects* to find in sys to help report
379 * XXX the exception and consequent unexpected failures). I've also
380 * XXX seen segfaults then, after adding print statements to the
381 * XXX Python code getting called.
382 */
383#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000384 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000385#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000386
Guido van Rossum1707aad1997-12-08 23:43:45 +0000387 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
388 _PyImport_Fini();
389
390 /* Debugging stuff */
391#ifdef COUNT_ALLOCS
392 dump_counts();
393#endif
394
Neal Norwitz4281cef2006-03-04 19:58:13 +0000395 PRINT_TOTAL_REFS()
Guido van Rossum1707aad1997-12-08 23:43:45 +0000396
Tim Peters9cf25ce2003-04-17 15:21:01 +0000397#ifdef Py_TRACE_REFS
398 /* Display all objects still alive -- this can invoke arbitrary
399 * __repr__ overrides, so requires a mostly-intact interpreter.
400 * Alas, a lot of stuff may still be alive now that will be cleaned
401 * up later.
402 */
Tim Peters269b2a62003-04-17 19:52:29 +0000403 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000404 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000405#endif /* Py_TRACE_REFS */
406
Mark Hammond6cb90292003-04-22 11:18:00 +0000407 /* Cleanup auto-thread-state */
408#ifdef WITH_THREAD
409 _PyGILState_Fini();
410#endif /* WITH_THREAD */
411
Guido van Rossumd922fa42003-04-15 14:10:09 +0000412 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000413 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000414
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000415 /* Now we decref the exception classes. After this point nothing
416 can raise an exception. That's okay, because each Fini() method
417 below has been checked to make sure no exceptions are ever
418 raised.
419 */
420
421 _PyExc_Fini();
422
Guido van Rossumd922fa42003-04-15 14:10:09 +0000423 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000424 PyThreadState_Swap(NULL);
425 PyInterpreterState_Delete(interp);
426
Guido van Rossumd922fa42003-04-15 14:10:09 +0000427 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000428 PyMethod_Fini();
429 PyFrame_Fini();
430 PyCFunction_Fini();
431 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000432 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000433 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000434 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000435 PyInt_Fini();
436 PyFloat_Fini();
437
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000438#ifdef Py_USING_UNICODE
439 /* Cleanup Unicode implementation */
440 _PyUnicode_Fini();
441#endif
442
Guido van Rossumcc283f51997-08-05 02:22:03 +0000443 /* XXX Still allocated:
444 - various static ad-hoc pointers to interned strings
445 - int and float free list blocks
446 - whatever various modules and libraries allocate
447 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000448
449 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000450
Tim Peters269b2a62003-04-17 19:52:29 +0000451#ifdef Py_TRACE_REFS
452 /* Display addresses (& refcnts) of all objects still alive.
453 * An address can be used to find the repr of the object, printed
454 * above by _Py_PrintReferences.
455 */
456 if (Py_GETENV("PYTHONDUMPREFS"))
457 _Py_PrintReferenceAddresses(stderr);
458#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000459#ifdef PYMALLOC_DEBUG
460 if (Py_GETENV("PYTHONMALLOCSTATS"))
461 _PyObject_DebugMallocStats();
462#endif
463
Guido van Rossumcc283f51997-08-05 02:22:03 +0000464 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000465}
466
467/* Create and initialize a new interpreter and thread, and return the
468 new thread. This requires that Py_Initialize() has been called
469 first.
470
471 Unsuccessful initialization yields a NULL pointer. Note that *no*
472 exception information is available even in this case -- the
473 exception information is held in the thread, and there is no
474 thread.
475
476 Locking: as above.
477
478*/
479
480PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000481Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000482{
483 PyInterpreterState *interp;
484 PyThreadState *tstate, *save_tstate;
485 PyObject *bimod, *sysmod;
486
487 if (!initialized)
488 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
489
490 interp = PyInterpreterState_New();
491 if (interp == NULL)
492 return NULL;
493
494 tstate = PyThreadState_New(interp);
495 if (tstate == NULL) {
496 PyInterpreterState_Delete(interp);
497 return NULL;
498 }
499
500 save_tstate = PyThreadState_Swap(tstate);
501
502 /* XXX The following is lax in error checking */
503
504 interp->modules = PyDict_New();
505
506 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
507 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000508 interp->builtins = PyModule_GetDict(bimod);
509 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000510 }
511 sysmod = _PyImport_FindExtension("sys", "sys");
512 if (bimod != NULL && sysmod != NULL) {
513 interp->sysdict = PyModule_GetDict(sysmod);
514 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
527 /* Oops, it didn't work. Undo it all. */
528
529 PyErr_Print();
530 PyThreadState_Clear(tstate);
531 PyThreadState_Swap(save_tstate);
532 PyThreadState_Delete(tstate);
533 PyInterpreterState_Delete(interp);
534
535 return NULL;
536}
537
538/* Delete an interpreter and its last thread. This requires that the
539 given thread state is current, that the thread has no remaining
540 frames, and that it is its interpreter's only remaining thread.
541 It is a fatal error to violate these constraints.
542
543 (Py_Finalize() doesn't have these constraints -- it zaps
544 everything, regardless.)
545
546 Locking: as above.
547
548*/
549
550void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000551Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000552{
553 PyInterpreterState *interp = tstate->interp;
554
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000555 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000556 Py_FatalError("Py_EndInterpreter: thread is not current");
557 if (tstate->frame != NULL)
558 Py_FatalError("Py_EndInterpreter: thread still has a frame");
559 if (tstate != interp->tstate_head || tstate->next != NULL)
560 Py_FatalError("Py_EndInterpreter: not the last thread");
561
562 PyImport_Cleanup();
563 PyInterpreterState_Clear(interp);
564 PyThreadState_Swap(NULL);
565 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000566}
567
568static char *progname = "python";
569
570void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000571Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000572{
573 if (pn && *pn)
574 progname = pn;
575}
576
577char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000578Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000579{
580 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000581}
582
Guido van Rossuma61691e1998-02-06 22:27:24 +0000583static char *default_home = NULL;
584
585void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000586Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000587{
588 default_home = home;
589}
590
591char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000592Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000593{
594 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000595 if (home == NULL && !Py_IgnoreEnvironmentFlag)
596 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000597 return home;
598}
599
Guido van Rossum6135a871995-01-09 17:53:26 +0000600/* Create __main__ module */
601
602static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000603initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000604{
Guido van Rossum82598051997-03-05 00:20:32 +0000605 PyObject *m, *d;
606 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000607 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000608 Py_FatalError("can't create __main__ module");
609 d = PyModule_GetDict(m);
610 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000611 PyObject *bimod = PyImport_ImportModule("__builtin__");
612 if (bimod == NULL ||
613 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000614 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000615 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000616 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000617}
618
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000619/* Import the site module (not into __main__ though) */
620
621static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000622initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000623{
624 PyObject *m, *f;
625 m = PyImport_ImportModule("site");
626 if (m == NULL) {
627 f = PySys_GetObject("stderr");
628 if (Py_VerboseFlag) {
629 PyFile_WriteString(
630 "'import site' failed; traceback:\n", f);
631 PyErr_Print();
632 }
633 else {
634 PyFile_WriteString(
635 "'import site' failed; use -v for traceback\n", f);
636 PyErr_Clear();
637 }
638 }
639 else {
640 Py_DECREF(m);
641 }
642}
643
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000644/* Parse input from a file and execute it */
645
646int
Martin v. Löwis056a69c2006-03-01 16:55:42 +0000647PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000648 PyCompilerFlags *flags)
649{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000650 if (filename == NULL)
651 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000652 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000653 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000654 if (closeit)
655 fclose(fp);
656 return err;
657 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000658 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000659 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000660}
661
662int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000663PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000664{
Guido van Rossum82598051997-03-05 00:20:32 +0000665 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000666 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000667 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000668
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000669 if (flags == NULL) {
670 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000671 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000672 }
Guido van Rossum82598051997-03-05 00:20:32 +0000673 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000674 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000675 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
676 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000677 }
Guido van Rossum82598051997-03-05 00:20:32 +0000678 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000679 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000680 PySys_SetObject("ps2", v = PyString_FromString("... "));
681 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000682 }
683 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000684 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Neal Norwitz4281cef2006-03-04 19:58:13 +0000685 PRINT_TOTAL_REFS()
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000686 if (ret == E_EOF)
687 return 0;
688 /*
689 if (ret == E_NOMEM)
690 return -1;
691 */
692 }
693}
694
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000695/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000696#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000697 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
698 PyPARSE_DONT_IMPLY_DEDENT : 0) \
Neal Norwitz9589ee22006-03-04 19:01:22 +0000699 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
700 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000701
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000702int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000703PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000704{
Guido van Rossum82598051997-03-05 00:20:32 +0000705 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000706 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000707 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000708 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000709 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000710
Guido van Rossum82598051997-03-05 00:20:32 +0000711 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000712 if (v != NULL) {
713 v = PyObject_Str(v);
714 if (v == NULL)
715 PyErr_Clear();
716 else if (PyString_Check(v))
717 ps1 = PyString_AsString(v);
718 }
Guido van Rossum82598051997-03-05 00:20:32 +0000719 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000720 if (w != NULL) {
721 w = PyObject_Str(w);
722 if (w == NULL)
723 PyErr_Clear();
724 else if (PyString_Check(w))
725 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000726 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000727 arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000728 mod = PyParser_ASTFromFile(fp, filename,
729 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000730 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000731 Py_XDECREF(v);
732 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000733 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000734 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000735 if (errcode == E_EOF) {
736 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000737 return E_EOF;
738 }
Guido van Rossum82598051997-03-05 00:20:32 +0000739 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000740 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000741 }
Guido van Rossum82598051997-03-05 00:20:32 +0000742 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000743 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000744 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000745 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000746 }
Guido van Rossum82598051997-03-05 00:20:32 +0000747 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000748 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000749 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000750 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000751 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000752 return -1;
753 }
Guido van Rossum82598051997-03-05 00:20:32 +0000754 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000755 if (Py_FlushLine())
756 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000757 return 0;
758}
759
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000760/* Check whether a file maybe a pyc file: Look at the extension,
761 the file type, and, if we may close it, at the first few bytes. */
762
763static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000764maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000765{
766 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
767 return 1;
768
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000769 /* Only look into the file if we are allowed to close it, since
770 it then should also be seekable. */
771 if (closeit) {
772 /* Read only two bytes of the magic. If the file was opened in
773 text mode, the bytes 3 and 4 of the magic (\r\n) might not
774 be read as they are on disk. */
775 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
776 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000777 /* Mess: In case of -x, the stream is NOT at its start now,
778 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000779 which makes the current stream position formally undefined,
780 and a x-platform nightmare.
781 Unfortunately, we have no direct way to know whether -x
782 was specified. So we use a terrible hack: if the current
783 stream position is not 0, we assume -x was specified, and
784 give up. Bug 132850 on SourceForge spells out the
785 hopelessness of trying anything else (fseek and ftell
786 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000787 */
Tim Peters3e876562001-02-11 04:35:39 +0000788 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000789 if (ftell(fp) == 0) {
790 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000791 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000792 ispyc = 1;
793 rewind(fp);
794 }
Tim Peters3e876562001-02-11 04:35:39 +0000795 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000796 }
797 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000798}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000799
Guido van Rossum0df002c2000-08-27 19:21:52 +0000800int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000801PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000802 PyCompilerFlags *flags)
803{
Guido van Rossum82598051997-03-05 00:20:32 +0000804 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000805 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000806
Guido van Rossum82598051997-03-05 00:20:32 +0000807 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000808 if (m == NULL)
809 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000810 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000811 if (PyDict_GetItemString(d, "__file__") == NULL) {
812 PyObject *f = PyString_FromString(filename);
813 if (f == NULL)
814 return -1;
815 if (PyDict_SetItemString(d, "__file__", f) < 0) {
816 Py_DECREF(f);
817 return -1;
818 }
819 Py_DECREF(f);
820 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000821 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000822 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000823 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000824 if (closeit)
825 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000826 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000827 fprintf(stderr, "python: Can't reopen .pyc file\n");
828 return -1;
829 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000830 /* Turn on optimization if a .pyo file is given */
831 if (strcmp(ext, ".pyo") == 0)
832 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000833 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000834 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000835 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000836 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000837 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000838 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000839 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000840 return -1;
841 }
Guido van Rossum82598051997-03-05 00:20:32 +0000842 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000843 if (Py_FlushLine())
844 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000845 return 0;
846}
847
848int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000849PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000850{
Guido van Rossum82598051997-03-05 00:20:32 +0000851 PyObject *m, *d, *v;
852 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000853 if (m == NULL)
854 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000855 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000856 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000857 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000858 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000859 return -1;
860 }
Guido van Rossum82598051997-03-05 00:20:32 +0000861 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000862 if (Py_FlushLine())
863 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000864 return 0;
865}
866
Barry Warsaw035574d1997-08-29 22:07:17 +0000867static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000868parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
869 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000870{
871 long hold;
872 PyObject *v;
873
874 /* old style errors */
875 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000876 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000877 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000878
879 /* new style errors. `err' is an instance */
880
881 if (! (v = PyObject_GetAttrString(err, "msg")))
882 goto finally;
883 *message = v;
884
885 if (!(v = PyObject_GetAttrString(err, "filename")))
886 goto finally;
887 if (v == Py_None)
888 *filename = NULL;
889 else if (! (*filename = PyString_AsString(v)))
890 goto finally;
891
892 Py_DECREF(v);
893 if (!(v = PyObject_GetAttrString(err, "lineno")))
894 goto finally;
895 hold = PyInt_AsLong(v);
896 Py_DECREF(v);
897 v = NULL;
898 if (hold < 0 && PyErr_Occurred())
899 goto finally;
900 *lineno = (int)hold;
901
902 if (!(v = PyObject_GetAttrString(err, "offset")))
903 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000904 if (v == Py_None) {
905 *offset = -1;
906 Py_DECREF(v);
907 v = NULL;
908 } else {
909 hold = PyInt_AsLong(v);
910 Py_DECREF(v);
911 v = NULL;
912 if (hold < 0 && PyErr_Occurred())
913 goto finally;
914 *offset = (int)hold;
915 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000916
917 if (!(v = PyObject_GetAttrString(err, "text")))
918 goto finally;
919 if (v == Py_None)
920 *text = NULL;
921 else if (! (*text = PyString_AsString(v)))
922 goto finally;
923 Py_DECREF(v);
924 return 1;
925
926finally:
927 Py_XDECREF(v);
928 return 0;
929}
930
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000931void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000932PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000933{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000934 PyErr_PrintEx(1);
935}
936
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000937static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000938print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000939{
940 char *nl;
941 if (offset >= 0) {
942 if (offset > 0 && offset == (int)strlen(text))
943 offset--;
944 for (;;) {
945 nl = strchr(text, '\n');
946 if (nl == NULL || nl-text >= offset)
947 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000948 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000949 text = nl+1;
950 }
951 while (*text == ' ' || *text == '\t') {
952 text++;
953 offset--;
954 }
955 }
956 PyFile_WriteString(" ", f);
957 PyFile_WriteString(text, f);
958 if (*text == '\0' || text[strlen(text)-1] != '\n')
959 PyFile_WriteString("\n", f);
960 if (offset == -1)
961 return;
962 PyFile_WriteString(" ", f);
963 offset--;
964 while (offset > 0) {
965 PyFile_WriteString(" ", f);
966 offset--;
967 }
968 PyFile_WriteString("^\n", f);
969}
970
Guido van Rossum66e8e862001-03-23 17:54:43 +0000971static void
972handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000973{
Neal Norwitz9589ee22006-03-04 19:01:22 +0000974 PyObject *exception, *value, *tb;
975 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +0000976
Guido van Rossum66e8e862001-03-23 17:54:43 +0000977 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000978 if (Py_FlushLine())
979 PyErr_Clear();
980 fflush(stdout);
981 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +0000982 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +0000983 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000984 /* The error code should be in the `code' attribute. */
985 PyObject *code = PyObject_GetAttrString(value, "code");
986 if (code) {
987 Py_DECREF(value);
988 value = code;
989 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +0000990 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000991 }
992 /* If we failed to dig out the 'code' attribute,
993 just let the else clause below print the error. */
994 }
995 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +0000996 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000997 else {
998 PyObject_Print(value, stderr, Py_PRINT_RAW);
999 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001000 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001001 }
Tim Peterscf615b52003-04-19 18:47:02 +00001002 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001003 /* Restore and clear the exception info, in order to properly decref
1004 * the exception, value, and traceback. If we just exit instead,
1005 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1006 * some finalizers from running.
1007 */
Tim Peterscf615b52003-04-19 18:47:02 +00001008 PyErr_Restore(exception, value, tb);
1009 PyErr_Clear();
1010 Py_Exit(exitcode);
1011 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001012}
1013
1014void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001015PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001016{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001017 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001018
1019 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1020 handle_system_exit();
1021 }
Guido van Rossum82598051997-03-05 00:20:32 +00001022 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001023 if (exception == NULL)
1024 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001025 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001026 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001027 return;
Guido van Rossuma61691e1998-02-06 22:27:24 +00001028 if (set_sys_last_vars) {
1029 PySys_SetObject("last_type", exception);
1030 PySys_SetObject("last_value", v);
1031 PySys_SetObject("last_traceback", tb);
1032 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001033 hook = PySys_GetObject("excepthook");
1034 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001035 PyObject *args = PyTuple_Pack(3,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001036 exception, v ? v : Py_None, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001037 PyObject *result = PyEval_CallObject(hook, args);
1038 if (result == NULL) {
1039 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001040 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1041 handle_system_exit();
1042 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001043 PyErr_Fetch(&exception2, &v2, &tb2);
1044 PyErr_NormalizeException(&exception2, &v2, &tb2);
1045 if (Py_FlushLine())
1046 PyErr_Clear();
1047 fflush(stdout);
1048 PySys_WriteStderr("Error in sys.excepthook:\n");
1049 PyErr_Display(exception2, v2, tb2);
1050 PySys_WriteStderr("\nOriginal exception was:\n");
1051 PyErr_Display(exception, v, tb);
Jeremy Hylton07028582001-12-07 15:35:35 +00001052 Py_XDECREF(exception2);
1053 Py_XDECREF(v2);
1054 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001055 }
1056 Py_XDECREF(result);
1057 Py_XDECREF(args);
1058 } else {
1059 PySys_WriteStderr("sys.excepthook is missing\n");
1060 PyErr_Display(exception, v, tb);
1061 }
1062 Py_XDECREF(exception);
1063 Py_XDECREF(v);
1064 Py_XDECREF(tb);
1065}
1066
1067void PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
1068{
1069 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001070 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001071 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001072 if (f == NULL)
1073 fprintf(stderr, "lost sys.stderr\n");
1074 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001075 if (Py_FlushLine())
1076 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001077 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001078 if (tb && tb != Py_None)
1079 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001080 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001081 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001082 {
Guido van Rossum82598051997-03-05 00:20:32 +00001083 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001084 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001085 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001086 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001087 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001088 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001089 else {
1090 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001091 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001092 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001093 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001094 else
Guido van Rossum82598051997-03-05 00:20:32 +00001095 PyFile_WriteString(filename, f);
1096 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001097 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001098 PyFile_WriteString(buf, f);
1099 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001100 if (text != NULL)
1101 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001102 Py_DECREF(value);
1103 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001104 /* Can't be bothered to check all those
1105 PyFile_WriteString() calls */
1106 if (PyErr_Occurred())
1107 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001108 }
1109 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001110 if (err) {
1111 /* Don't do anything else */
1112 }
Brett Cannonbf364092006-03-01 04:25:17 +00001113 else if (PyExceptionClass_Check(exception)) {
1114 char* className = PyExceptionClass_Name(exception);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001115 PyObject* moduleName =
Brett Cannonbf364092006-03-01 04:25:17 +00001116 PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001117
1118 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001119 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001120 else {
1121 char* modstr = PyString_AsString(moduleName);
Brett Cannon2e63b732006-03-02 18:34:57 +00001122 Py_DECREF(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001123 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001124 {
1125 err = PyFile_WriteString(modstr, f);
1126 err += PyFile_WriteString(".", f);
1127 }
1128 }
1129 if (err == 0) {
1130 if (className == NULL)
1131 err = PyFile_WriteString("<unknown>", f);
1132 else
Brett Cannonbf364092006-03-01 04:25:17 +00001133 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001134 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001135 }
1136 else
1137 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001138 if (err == 0 && (value != Py_None)) {
1139 PyObject *s = PyObject_Str(value);
1140 /* only print colon if the str() of the
1141 object is not the empty string
1142 */
1143 if (s == NULL)
1144 err = -1;
1145 else if (!PyString_Check(s) ||
1146 PyString_GET_SIZE(s) != 0)
1147 err = PyFile_WriteString(": ", f);
1148 if (err == 0)
1149 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1150 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001151 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001152 if (err == 0)
1153 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001154 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001155 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001156 /* If an error happened here, don't show it.
1157 XXX This is wrong, but too many callers rely on this behavior. */
1158 if (err != 0)
1159 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001160}
1161
Guido van Rossum82598051997-03-05 00:20:32 +00001162PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001163PyRun_StringFlags(const char *str, int start, PyObject *globals,
1164 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001165{
Neal Norwitze92fba02006-03-04 18:52:26 +00001166 PyObject *ret = NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001167 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001168 mod_ty mod = PyParser_ASTFromString(str, "<string>", start, flags,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001169 arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001170 if (mod != NULL)
1171 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001172 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001173 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001174}
1175
1176PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001177PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001178 PyObject *locals, int closeit, PyCompilerFlags *flags)
1179{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001180 PyObject *ret;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001181 PyArena *arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001182 mod_ty mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001183 flags, NULL, arena);
1184 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001185 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001186 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001187 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001188 if (closeit)
1189 fclose(fp);
Neal Norwitze92fba02006-03-04 18:52:26 +00001190 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001191 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001192 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001193}
1194
Guido van Rossum82598051997-03-05 00:20:32 +00001195static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001196run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001197 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001198{
Guido van Rossum82598051997-03-05 00:20:32 +00001199 PyCodeObject *co;
1200 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001201 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001202 if (co == NULL)
1203 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001204 v = PyEval_EvalCode(co, globals, locals);
1205 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001206 return v;
1207}
1208
Guido van Rossum82598051997-03-05 00:20:32 +00001209static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001210run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
1211 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001212{
Guido van Rossum82598051997-03-05 00:20:32 +00001213 PyCodeObject *co;
1214 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001215 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001216 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001217
Guido van Rossum82598051997-03-05 00:20:32 +00001218 magic = PyMarshal_ReadLongFromFile(fp);
1219 if (magic != PyImport_GetMagicNumber()) {
1220 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001221 "Bad magic number in .pyc file");
1222 return NULL;
1223 }
Guido van Rossum82598051997-03-05 00:20:32 +00001224 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001225 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001226 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001227 if (v == NULL || !PyCode_Check(v)) {
1228 Py_XDECREF(v);
1229 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001230 "Bad code object in .pyc file");
1231 return NULL;
1232 }
Guido van Rossum82598051997-03-05 00:20:32 +00001233 co = (PyCodeObject *)v;
1234 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001235 if (v && flags)
1236 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001237 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001238 return v;
1239}
1240
Guido van Rossum82598051997-03-05 00:20:32 +00001241PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001242Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001243 PyCompilerFlags *flags)
1244{
Guido van Rossum82598051997-03-05 00:20:32 +00001245 PyCodeObject *co;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001246 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001247 mod_ty mod = PyParser_ASTFromString(str, filename, start, flags, arena);
1248 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001249 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001250 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001251 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001252 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001253 PyObject *result = PyAST_mod2obj(mod);
1254 PyArena_Free(arena);
1255 return result;
1256 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001257 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001258 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001259 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001260}
1261
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001262struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001263Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001264{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001265 struct symtable *st;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001266 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001267 mod_ty mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
1268 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001269 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001270 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001271 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001272 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001273 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001274 return st;
1275}
1276
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001277/* Preferred access to parser is through AST. */
1278mod_ty
1279PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001280 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001281{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001282 mod_ty mod;
1283 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001284 node *n = PyParser_ParseStringFlagsFilename(s, filename,
1285 &_PyParser_Grammar, start, &err,
1286 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001287 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001288 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001289 PyNode_Free(n);
1290 return mod;
1291 }
1292 else {
1293 err_input(&err);
1294 return NULL;
1295 }
1296}
1297
1298mod_ty
1299PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001300 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001301 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001302{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001303 mod_ty mod;
1304 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001305 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1306 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001307 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001308 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001309 PyNode_Free(n);
1310 return mod;
1311 }
1312 else {
1313 err_input(&err);
1314 if (errcode)
1315 *errcode = err.error;
1316 return NULL;
1317 }
1318}
1319
Guido van Rossuma110aa61994-08-29 12:50:44 +00001320/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001321
Guido van Rossuma110aa61994-08-29 12:50:44 +00001322node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001323PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001324{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001325 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001326 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1327 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001328 if (n == NULL)
1329 err_input(&err);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001330
Guido van Rossuma110aa61994-08-29 12:50:44 +00001331 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001332}
1333
Guido van Rossuma110aa61994-08-29 12:50:44 +00001334/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001335
Guido van Rossuma110aa61994-08-29 12:50:44 +00001336node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001337PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001338{
Tim Petersfe2127d2001-07-16 05:37:24 +00001339 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001340 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1341 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001342 if (n == NULL)
1343 err_input(&err);
1344 return n;
1345}
1346
1347node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001348PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001349 int start, int flags)
1350{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001351 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001352 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1353 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001354 if (n == NULL)
1355 err_input(&err);
1356 return n;
1357}
1358
1359node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001360PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001361{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001362 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001363}
1364
Guido van Rossum66ebd912003-04-17 16:02:26 +00001365/* May want to move a more generalized form of this to parsetok.c or
1366 even parser modules. */
1367
1368void
1369PyParser_SetError(perrdetail *err)
1370{
1371 err_input(err);
1372}
1373
Guido van Rossuma110aa61994-08-29 12:50:44 +00001374/* Set the error appropriate to the given input error code (see errcode.h) */
1375
1376static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001377err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001378{
Fred Drake85f36392000-07-11 17:53:00 +00001379 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001380 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001381 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001382 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001383 switch (err->error) {
1384 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001385 errtype = PyExc_IndentationError;
1386 if (err->expected == INDENT)
1387 msg = "expected an indented block";
1388 else if (err->token == INDENT)
1389 msg = "unexpected indent";
1390 else if (err->token == DEDENT)
1391 msg = "unexpected unindent";
1392 else {
1393 errtype = PyExc_SyntaxError;
1394 msg = "invalid syntax";
1395 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001396 break;
1397 case E_TOKEN:
1398 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001399 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001400 case E_EOFS:
1401 msg = "EOF while scanning triple-quoted string";
1402 break;
1403 case E_EOLS:
1404 msg = "EOL while scanning single-quoted string";
1405 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001406 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001407 if (!PyErr_Occurred())
1408 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001409 return;
1410 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001411 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001412 return;
1413 case E_EOF:
1414 msg = "unexpected EOF while parsing";
1415 break;
Fred Drake85f36392000-07-11 17:53:00 +00001416 case E_TABSPACE:
1417 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001418 msg = "inconsistent use of tabs and spaces in indentation";
1419 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001420 case E_OVERFLOW:
1421 msg = "expression too long";
1422 break;
Fred Drake85f36392000-07-11 17:53:00 +00001423 case E_DEDENT:
1424 errtype = PyExc_IndentationError;
1425 msg = "unindent does not match any outer indentation level";
1426 break;
1427 case E_TOODEEP:
1428 errtype = PyExc_IndentationError;
1429 msg = "too many levels of indentation";
1430 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001431 case E_DECODE: {
1432 PyObject *type, *value, *tb;
1433 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001434 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001435 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001436 if (u != NULL) {
1437 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001438 }
1439 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001440 if (msg == NULL)
1441 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001442 Py_XDECREF(type);
1443 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001444 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001445 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001446 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001447 case E_LINECONT:
1448 msg = "unexpected character after line continuation character";
1449 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001450 default:
1451 fprintf(stderr, "error=%d\n", err->error);
1452 msg = "unknown parsing error";
1453 break;
1454 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001455 v = Py_BuildValue("(ziiz)", err->filename,
1456 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001457 if (err->text != NULL) {
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001458 PyMem_DEL(err->text);
1459 err->text = NULL;
1460 }
1461 w = NULL;
1462 if (v != NULL)
1463 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001464 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001465 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001466 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001467 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001468}
1469
1470/* Print fatal error message and abort */
1471
1472void
Tim Peters7c321a82002-07-09 02:57:01 +00001473Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001474{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001475 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001476#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001477 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001478 OutputDebugString(msg);
1479 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001480#ifdef _DEBUG
1481 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001482#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001483#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001484 abort();
1485}
1486
1487/* Clean up and exit */
1488
Guido van Rossuma110aa61994-08-29 12:50:44 +00001489#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001490#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001491#endif
1492
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001493#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001494static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001495static int nexitfuncs = 0;
1496
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001497int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001498{
1499 if (nexitfuncs >= NEXITFUNCS)
1500 return -1;
1501 exitfuncs[nexitfuncs++] = func;
1502 return 0;
1503}
1504
Guido van Rossumcc283f51997-08-05 02:22:03 +00001505static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001506call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001507{
Guido van Rossum82598051997-03-05 00:20:32 +00001508 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001509
1510 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001511 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001512 Py_INCREF(exitfunc);
1513 PySys_SetObject("exitfunc", (PyObject *)NULL);
1514 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001515 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001516 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1517 PySys_WriteStderr("Error in sys.exitfunc:\n");
1518 }
Guido van Rossum82598051997-03-05 00:20:32 +00001519 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001520 }
Guido van Rossum82598051997-03-05 00:20:32 +00001521 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001522 }
1523
Guido van Rossum0829c751998-02-28 04:31:39 +00001524 if (Py_FlushLine())
1525 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001526}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001527
Guido van Rossumcc283f51997-08-05 02:22:03 +00001528static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001529call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001530{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001531 while (nexitfuncs > 0)
1532 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001533
1534 fflush(stdout);
1535 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001536}
1537
1538void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001539Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001540{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001541 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001542
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001543 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001544}
1545
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001546static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001547initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001548{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001549#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001550 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001551#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001552#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001553 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001554#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001555#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001556 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001557#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001558 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001559}
1560
Guido van Rossum7433b121997-02-14 19:45:36 +00001561
1562/*
1563 * The file descriptor fd is considered ``interactive'' if either
1564 * a) isatty(fd) is TRUE, or
1565 * b) the -i flag was given, and the filename associated with
1566 * the descriptor is NULL or "<stdin>" or "???".
1567 */
1568int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001569Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001570{
1571 if (isatty((int)fileno(fp)))
1572 return 1;
1573 if (!Py_InteractiveFlag)
1574 return 0;
1575 return (filename == NULL) ||
1576 (strcmp(filename, "<stdin>") == 0) ||
1577 (strcmp(filename, "???") == 0);
1578}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001579
1580
Tim Petersd08e3822003-04-17 15:24:21 +00001581#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001582#if defined(WIN32) && defined(_MSC_VER)
1583
1584/* Stack checking for Microsoft C */
1585
1586#include <malloc.h>
1587#include <excpt.h>
1588
Fred Drakee8de31c2000-08-31 05:38:39 +00001589/*
1590 * Return non-zero when we run out of memory on the stack; zero otherwise.
1591 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001592int
Fred Drake399739f2000-08-31 05:52:44 +00001593PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001594{
1595 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001596 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001597 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001598 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001599 return 0;
1600 } __except (EXCEPTION_EXECUTE_HANDLER) {
1601 /* just ignore all errors */
1602 }
1603 return 1;
1604}
1605
1606#endif /* WIN32 && _MSC_VER */
1607
1608/* Alternate implementations can be added here... */
1609
1610#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001611
1612
1613/* Wrappers around sigaction() or signal(). */
1614
1615PyOS_sighandler_t
1616PyOS_getsig(int sig)
1617{
1618#ifdef HAVE_SIGACTION
1619 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001620 if (sigaction(sig, NULL, &context) == -1)
1621 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001622 return context.sa_handler;
1623#else
1624 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001625/* Special signal handling for the secure CRT in Visual Studio 2005 */
1626#if defined(_MSC_VER) && _MSC_VER >= 1400
1627 switch (sig) {
1628 /* Only these signals are valid */
1629 case SIGINT:
1630 case SIGILL:
1631 case SIGFPE:
1632 case SIGSEGV:
1633 case SIGTERM:
1634 case SIGBREAK:
1635 case SIGABRT:
1636 break;
1637 /* Don't call signal() with other values or it will assert */
1638 default:
1639 return SIG_ERR;
1640 }
1641#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001642 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001643 if (handler != SIG_ERR)
1644 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001645 return handler;
1646#endif
1647}
1648
1649PyOS_sighandler_t
1650PyOS_setsig(int sig, PyOS_sighandler_t handler)
1651{
1652#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001653 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001654 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001655 sigemptyset(&context.sa_mask);
1656 context.sa_flags = 0;
1657 if (sigaction(sig, &context, &ocontext) == -1)
1658 return SIG_ERR;
1659 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001660#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001661 PyOS_sighandler_t oldhandler;
1662 oldhandler = signal(sig, handler);
1663#ifdef HAVE_SIGINTERRUPT
1664 siginterrupt(sig, 1);
1665#endif
1666 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001667#endif
1668}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001669
1670/* Deprecated C API functions still provided for binary compatiblity */
1671
1672#undef PyParser_SimpleParseFile
1673#undef PyParser_SimpleParseString
1674
1675node *
1676PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1677{
1678 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1679}
1680
1681node *
1682PyParser_SimpleParseString(const char *str, int start)
1683{
1684 return PyParser_SimpleParseStringFlags(str, start, 0);
1685}