blob: 7b1f26439bc0188517c92cf5f2946fe4a1bc07d4 [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 */
Neal Norwitzf2e0c452006-03-06 23:31:27 +000038# define PRINT_TOTAL_REFS() fprintf(stderr, "[%ld refs]\n", \
39 Py_SAFE_DOWNCAST(_Py_RefTotal, Py_ssize_t, long));
Neal Norwitz4281cef2006-03-04 19:58:13 +000040# endif /* MS_WIN64 */
41#endif
42
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000043extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000044
Guido van Rossum82598051997-03-05 00:20:32 +000045extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000046
Guido van Rossumb73cc041993-11-01 16:28:59 +000047/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000048static void initmain(void);
49static void initsite(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000050static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000051 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000052static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000053 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000054static void err_input(perrdetail *);
55static void initsigs(void);
56static void call_sys_exitfunc(void);
57static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000058extern void _PyUnicode_Init(void);
59extern void _PyUnicode_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000060
Mark Hammond8d98d2c2003-04-19 15:41:53 +000061#ifdef WITH_THREAD
62extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
63extern void _PyGILState_Fini(void);
64#endif /* WITH_THREAD */
65
Guido van Rossum82598051997-03-05 00:20:32 +000066int Py_DebugFlag; /* Needed by parser.c */
67int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000068int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000069int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000070int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000071int Py_FrozenFlag; /* Needed by getpath.c */
Guido van Rossumb16d1972000-05-01 17:55:15 +000072int Py_UnicodeFlag = 0; /* Needed by compile.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000073int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Tim Peters3caca232001-12-06 06:23:26 +000074/* _XXX Py_QnewFlag should go away in 2.3. It's true iff -Qnew is passed,
75 on the command line, and is used in 2.2 by ceval.c to make all "/" divisions
76 true divisions (which they will be in 2.3). */
77int _Py_QnewFlag = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +000078
Mark Hammonda43fd0c2003-02-19 00:33:33 +000079/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000080 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000081*/
Mark Hammondedd07732003-07-15 23:03:55 +000082static PyObject *warnings_module = NULL;
83
84/* Returns a borrowed reference to the 'warnings' module, or NULL.
85 If the module is returned, it is guaranteed to have been obtained
86 without acquiring the import lock
87*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000088PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000089{
90 PyObject *typ, *val, *tb;
91 PyObject *all_modules;
92 /* If we managed to get the module at init time, just use it */
93 if (warnings_module)
94 return warnings_module;
95 /* If it wasn't available at init time, it may be available
96 now in sys.modules (common scenario is frozen apps: import
97 at init time fails, but the frozen init code sets up sys.path
98 correctly, then does an implicit import of warnings for us
99 */
100 /* Save and restore any exceptions */
101 PyErr_Fetch(&typ, &val, &tb);
102
Mark Hammond5f4e8ca2003-07-16 01:54:38 +0000103 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +0000104 if (all_modules) {
105 warnings_module = PyDict_GetItemString(all_modules, "warnings");
106 /* We keep a ref in the global */
107 Py_XINCREF(warnings_module);
108 }
109 PyErr_Restore(typ, val, tb);
110 return warnings_module;
111}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000112
Guido van Rossum25ce5661997-08-02 03:10:38 +0000113static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000114
Thomas Wouters7e474022000-07-16 12:04:32 +0000115/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000116
117int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000118Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000119{
120 return initialized;
121}
122
Guido van Rossum25ce5661997-08-02 03:10:38 +0000123/* Global initializations. Can be undone by Py_Finalize(). Don't
124 call this twice without an intervening Py_Finalize() call. When
125 initializations fail, a fatal error is issued and the function does
126 not return. On return, the first thread and interpreter state have
127 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000128
Guido van Rossum25ce5661997-08-02 03:10:38 +0000129 Locking: you must hold the interpreter lock while calling this.
130 (If the lock has not yet been initialized, that's equivalent to
131 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000132
Guido van Rossum25ce5661997-08-02 03:10:38 +0000133*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000134
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000135static int
136add_flag(int flag, const char *envs)
137{
138 int env = atoi(envs);
139 if (flag < env)
140 flag = env;
141 if (flag < 1)
142 flag = 1;
143 return flag;
144}
145
Guido van Rossuma027efa1997-05-05 20:56:21 +0000146void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000147Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000148{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000149 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000150 PyThreadState *tstate;
151 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000152 char *p;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000153#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
154 char *codeset;
155 char *saved_locale;
156 PyObject *sys_stream, *sys_isatty;
157#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000158 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000159
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000160 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000161 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000162 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000163
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000164 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000165 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000166 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000167 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000168 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000169 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000170
Guido van Rossuma027efa1997-05-05 20:56:21 +0000171 interp = PyInterpreterState_New();
172 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000173 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000174
Guido van Rossuma027efa1997-05-05 20:56:21 +0000175 tstate = PyThreadState_New(interp);
176 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000177 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000178 (void) PyThreadState_Swap(tstate);
179
Guido van Rossum70d893a2001-08-16 08:21:42 +0000180 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000181
Neal Norwitzb2501f42002-12-31 03:42:13 +0000182 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000183 Py_FatalError("Py_Initialize: can't init frames");
184
Neal Norwitzb2501f42002-12-31 03:42:13 +0000185 if (!_PyInt_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000186 Py_FatalError("Py_Initialize: can't init ints");
187
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000188 _PyFloat_Init();
189
Guido van Rossum25ce5661997-08-02 03:10:38 +0000190 interp->modules = PyDict_New();
191 if (interp->modules == NULL)
192 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000193
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000194#ifdef Py_USING_UNICODE
Guido van Rossumc94044c2000-03-10 23:03:54 +0000195 /* Init Unicode implementation; relies on the codec registry */
196 _PyUnicode_Init();
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000197#endif
Guido van Rossumc94044c2000-03-10 23:03:54 +0000198
Barry Warsawf242aa02000-05-25 23:09:49 +0000199 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000200 if (bimod == NULL)
201 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000202 interp->builtins = PyModule_GetDict(bimod);
203 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000204
205 sysmod = _PySys_Init();
206 if (sysmod == NULL)
207 Py_FatalError("Py_Initialize: can't initialize sys");
208 interp->sysdict = PyModule_GetDict(sysmod);
209 Py_INCREF(interp->sysdict);
210 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000211 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000212 PyDict_SetItemString(interp->sysdict, "modules",
213 interp->modules);
214
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000215 _PyImport_Init();
216
Barry Warsawf242aa02000-05-25 23:09:49 +0000217 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000218 _PyExc_Init();
Barry Warsaw5821bc52001-08-13 23:04:56 +0000219 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000220
Barry Warsaw035574d1997-08-29 22:07:17 +0000221 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000222 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000223
Just van Rossum52e14d62002-12-30 22:08:05 +0000224 _PyImportHooks_Init();
225
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000226 if (install_sigs)
227 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000228
229 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000230 if (!Py_NoSiteFlag)
231 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000232
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000233 /* auto-thread-state API, if available */
234#ifdef WITH_THREAD
235 _PyGILState_Init(interp, tstate);
236#endif /* WITH_THREAD */
237
Mark Hammondedd07732003-07-15 23:03:55 +0000238 warnings_module = PyImport_ImportModule("warnings");
239 if (!warnings_module)
240 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000241
242#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
243 /* On Unix, set the file system encoding according to the
244 user's preference, if the CODESET names a well-known
245 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000246 initialized by other means. Also set the encoding of
247 stdin and stdout if these are terminals. */
248
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000249 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000250 setlocale(LC_CTYPE, "");
251 codeset = nl_langinfo(CODESET);
252 if (codeset && *codeset) {
253 PyObject *enc = PyCodec_Encoder(codeset);
254 if (enc) {
255 codeset = strdup(codeset);
256 Py_DECREF(enc);
257 } else {
258 codeset = NULL;
259 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000260 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000261 } else
262 codeset = NULL;
263 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000264 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000265
266 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000267 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000268 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
269 if (!sys_isatty)
270 PyErr_Clear();
271 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
272 if (!PyFile_SetEncoding(sys_stream, codeset))
273 Py_FatalError("Cannot set codeset of stdin");
274 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000275 Py_XDECREF(sys_isatty);
276
277 sys_stream = PySys_GetObject("stdout");
278 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
279 if (!sys_isatty)
280 PyErr_Clear();
281 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
282 if (!PyFile_SetEncoding(sys_stream, codeset))
283 Py_FatalError("Cannot set codeset of stdout");
284 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000285 Py_XDECREF(sys_isatty);
286
287 if (!Py_FileSystemDefaultEncoding)
288 Py_FileSystemDefaultEncoding = codeset;
289 else
290 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000291 }
292#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000293}
294
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000295void
296Py_Initialize(void)
297{
298 Py_InitializeEx(1);
299}
300
301
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000302#ifdef COUNT_ALLOCS
Tim Petersdbd9ba62000-07-09 03:09:57 +0000303extern void dump_counts(void);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000304#endif
305
Guido van Rossum25ce5661997-08-02 03:10:38 +0000306/* Undo the effect of Py_Initialize().
307
308 Beware: if multiple interpreter and/or thread states exist, these
309 are not wiped out; only the current thread and interpreter state
310 are deleted. But since everything else is deleted, those other
311 interpreter and thread states should no longer be used.
312
313 (XXX We should do better, e.g. wipe out all interpreters and
314 threads.)
315
316 Locking: as above.
317
318*/
319
320void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000321Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000322{
323 PyInterpreterState *interp;
324 PyThreadState *tstate;
325
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000326 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000327 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000328
Tim Peters384fd102001-01-21 03:40:37 +0000329 /* The interpreter is still entirely intact at this point, and the
330 * exit funcs may be relying on that. In particular, if some thread
331 * or exit func is still waiting to do an import, the import machinery
332 * expects Py_IsInitialized() to return true. So don't say the
333 * interpreter is uninitialized until after the exit funcs have run.
334 * Note that Threading.py uses an exit func to do a join on all the
335 * threads created thru it, so this also protects pending imports in
336 * the threads created via Threading.
337 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000338 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000339 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000340
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000341 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000342 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000343 interp = tstate->interp;
344
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000345 /* Disable signal handling */
346 PyOS_FiniInterrupts();
347
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000348 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000349 Py_XDECREF(warnings_module);
350 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000351
Guido van Rossume13ddc92003-04-17 17:29:22 +0000352 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000353 * before all modules are destroyed.
354 * XXX If a __del__ or weakref callback is triggered here, and tries to
355 * XXX import a module, bad things can happen, because Python no
356 * XXX longer believes it's initialized.
357 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
358 * XXX is easy to provoke that way. I've also seen, e.g.,
359 * XXX Exception exceptions.ImportError: 'No module named sha'
360 * XXX in <function callback at 0x008F5718> ignored
361 * XXX but I'm unclear on exactly how that one happens. In any case,
362 * XXX I haven't seen a real-life report of either of these.
Neal Norwitz9589ee22006-03-04 19:01:22 +0000363 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000364 PyGC_Collect();
365
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000366 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000367 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000368
Guido van Rossume13ddc92003-04-17 17:29:22 +0000369 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000370 * new-style class definitions, for example.
371 * XXX This is disabled because it caused too many problems. If
372 * XXX a __del__ or weakref callback triggers here, Python code has
373 * XXX a hard time running, because even the sys module has been
374 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
375 * XXX One symptom is a sequence of information-free messages
376 * XXX coming from threads (if a __del__ or callback is invoked,
377 * XXX other threads can execute too, and any exception they encounter
378 * XXX triggers a comedy of errors as subsystem after subsystem
379 * XXX fails to find what it *expects* to find in sys to help report
380 * XXX the exception and consequent unexpected failures). I've also
381 * XXX seen segfaults then, after adding print statements to the
382 * XXX Python code getting called.
383 */
384#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000385 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000386#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000387
Guido van Rossum1707aad1997-12-08 23:43:45 +0000388 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
389 _PyImport_Fini();
390
391 /* Debugging stuff */
392#ifdef COUNT_ALLOCS
393 dump_counts();
394#endif
395
Neal Norwitz4281cef2006-03-04 19:58:13 +0000396 PRINT_TOTAL_REFS()
Guido van Rossum1707aad1997-12-08 23:43:45 +0000397
Tim Peters9cf25ce2003-04-17 15:21:01 +0000398#ifdef Py_TRACE_REFS
399 /* Display all objects still alive -- this can invoke arbitrary
400 * __repr__ overrides, so requires a mostly-intact interpreter.
401 * Alas, a lot of stuff may still be alive now that will be cleaned
402 * up later.
403 */
Tim Peters269b2a62003-04-17 19:52:29 +0000404 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000405 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000406#endif /* Py_TRACE_REFS */
407
Mark Hammond6cb90292003-04-22 11:18:00 +0000408 /* Cleanup auto-thread-state */
409#ifdef WITH_THREAD
410 _PyGILState_Fini();
411#endif /* WITH_THREAD */
412
Guido van Rossumd922fa42003-04-15 14:10:09 +0000413 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000414 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000415
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000416 /* Now we decref the exception classes. After this point nothing
417 can raise an exception. That's okay, because each Fini() method
418 below has been checked to make sure no exceptions are ever
419 raised.
420 */
421
422 _PyExc_Fini();
423
Guido van Rossumd922fa42003-04-15 14:10:09 +0000424 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000425 PyThreadState_Swap(NULL);
426 PyInterpreterState_Delete(interp);
427
Guido van Rossumd922fa42003-04-15 14:10:09 +0000428 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000429 PyMethod_Fini();
430 PyFrame_Fini();
431 PyCFunction_Fini();
432 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000433 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000434 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000435 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000436 PyInt_Fini();
437 PyFloat_Fini();
438
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000439#ifdef Py_USING_UNICODE
440 /* Cleanup Unicode implementation */
441 _PyUnicode_Fini();
442#endif
443
Guido van Rossumcc283f51997-08-05 02:22:03 +0000444 /* XXX Still allocated:
445 - various static ad-hoc pointers to interned strings
446 - int and float free list blocks
447 - whatever various modules and libraries allocate
448 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000449
450 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000451
Tim Peters269b2a62003-04-17 19:52:29 +0000452#ifdef Py_TRACE_REFS
453 /* Display addresses (& refcnts) of all objects still alive.
454 * An address can be used to find the repr of the object, printed
455 * above by _Py_PrintReferences.
456 */
457 if (Py_GETENV("PYTHONDUMPREFS"))
458 _Py_PrintReferenceAddresses(stderr);
459#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000460#ifdef PYMALLOC_DEBUG
461 if (Py_GETENV("PYTHONMALLOCSTATS"))
462 _PyObject_DebugMallocStats();
463#endif
464
Guido van Rossumcc283f51997-08-05 02:22:03 +0000465 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000466}
467
468/* Create and initialize a new interpreter and thread, and return the
469 new thread. This requires that Py_Initialize() has been called
470 first.
471
472 Unsuccessful initialization yields a NULL pointer. Note that *no*
473 exception information is available even in this case -- the
474 exception information is held in the thread, and there is no
475 thread.
476
477 Locking: as above.
478
479*/
480
481PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000482Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000483{
484 PyInterpreterState *interp;
485 PyThreadState *tstate, *save_tstate;
486 PyObject *bimod, *sysmod;
487
488 if (!initialized)
489 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
490
491 interp = PyInterpreterState_New();
492 if (interp == NULL)
493 return NULL;
494
495 tstate = PyThreadState_New(interp);
496 if (tstate == NULL) {
497 PyInterpreterState_Delete(interp);
498 return NULL;
499 }
500
501 save_tstate = PyThreadState_Swap(tstate);
502
503 /* XXX The following is lax in error checking */
504
505 interp->modules = PyDict_New();
506
507 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
508 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000509 interp->builtins = PyModule_GetDict(bimod);
510 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000511 }
512 sysmod = _PyImport_FindExtension("sys", "sys");
513 if (bimod != NULL && sysmod != NULL) {
514 interp->sysdict = PyModule_GetDict(sysmod);
515 Py_INCREF(interp->sysdict);
516 PySys_SetPath(Py_GetPath());
517 PyDict_SetItemString(interp->sysdict, "modules",
518 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000519 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000520 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000521 if (!Py_NoSiteFlag)
522 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000523 }
524
525 if (!PyErr_Occurred())
526 return tstate;
527
528 /* 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
Martin v. Löwis056a69c2006-03-01 16:55:42 +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);
Neal Norwitz4281cef2006-03-04 19:58:13 +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) ? \
699 PyPARSE_DONT_IMPLY_DEDENT : 0) \
Neal Norwitz9589ee22006-03-04 19:01:22 +0000700 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
701 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000702
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000703int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000704PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000705{
Guido van Rossum82598051997-03-05 00:20:32 +0000706 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000707 mod_ty mod;
Neal Norwitz9589ee22006-03-04 19:01:22 +0000708 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000709 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000710 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000711
Guido van Rossum82598051997-03-05 00:20:32 +0000712 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000713 if (v != NULL) {
714 v = PyObject_Str(v);
715 if (v == NULL)
716 PyErr_Clear();
717 else if (PyString_Check(v))
718 ps1 = PyString_AsString(v);
719 }
Guido van Rossum82598051997-03-05 00:20:32 +0000720 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000721 if (w != NULL) {
722 w = PyObject_Str(w);
723 if (w == NULL)
724 PyErr_Clear();
725 else if (PyString_Check(w))
726 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000727 }
Neal Norwitz9589ee22006-03-04 19:01:22 +0000728 arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000729 mod = PyParser_ASTFromFile(fp, filename,
730 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000731 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000732 Py_XDECREF(v);
733 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000734 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000735 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000736 if (errcode == E_EOF) {
737 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000738 return E_EOF;
739 }
Guido van Rossum82598051997-03-05 00:20:32 +0000740 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000741 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000742 }
Guido van Rossum82598051997-03-05 00:20:32 +0000743 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000744 if (m == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +0000745 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000746 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000747 }
Guido van Rossum82598051997-03-05 00:20:32 +0000748 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000749 v = run_mod(mod, filename, d, d, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +0000750 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000751 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000752 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000753 return -1;
754 }
Guido van Rossum82598051997-03-05 00:20:32 +0000755 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000756 if (Py_FlushLine())
757 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000758 return 0;
759}
760
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000761/* Check whether a file maybe a pyc file: Look at the extension,
762 the file type, and, if we may close it, at the first few bytes. */
763
764static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000765maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000766{
767 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
768 return 1;
769
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000770 /* Only look into the file if we are allowed to close it, since
771 it then should also be seekable. */
772 if (closeit) {
773 /* Read only two bytes of the magic. If the file was opened in
774 text mode, the bytes 3 and 4 of the magic (\r\n) might not
775 be read as they are on disk. */
776 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
777 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000778 /* Mess: In case of -x, the stream is NOT at its start now,
779 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000780 which makes the current stream position formally undefined,
781 and a x-platform nightmare.
782 Unfortunately, we have no direct way to know whether -x
783 was specified. So we use a terrible hack: if the current
784 stream position is not 0, we assume -x was specified, and
785 give up. Bug 132850 on SourceForge spells out the
786 hopelessness of trying anything else (fseek and ftell
787 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000788 */
Tim Peters3e876562001-02-11 04:35:39 +0000789 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000790 if (ftell(fp) == 0) {
791 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000792 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000793 ispyc = 1;
794 rewind(fp);
795 }
Tim Peters3e876562001-02-11 04:35:39 +0000796 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000797 }
798 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000799}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000800
Guido van Rossum0df002c2000-08-27 19:21:52 +0000801int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000802PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000803 PyCompilerFlags *flags)
804{
Guido van Rossum82598051997-03-05 00:20:32 +0000805 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000806 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000807
Guido van Rossum82598051997-03-05 00:20:32 +0000808 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000809 if (m == NULL)
810 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000811 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000812 if (PyDict_GetItemString(d, "__file__") == NULL) {
813 PyObject *f = PyString_FromString(filename);
814 if (f == NULL)
815 return -1;
816 if (PyDict_SetItemString(d, "__file__", f) < 0) {
817 Py_DECREF(f);
818 return -1;
819 }
820 Py_DECREF(f);
821 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000822 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000823 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000824 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000825 if (closeit)
826 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000827 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000828 fprintf(stderr, "python: Can't reopen .pyc file\n");
829 return -1;
830 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000831 /* Turn on optimization if a .pyo file is given */
832 if (strcmp(ext, ".pyo") == 0)
833 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000834 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000835 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000836 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000837 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000838 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000839 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000840 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000841 return -1;
842 }
Guido van Rossum82598051997-03-05 00:20:32 +0000843 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000844 if (Py_FlushLine())
845 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000846 return 0;
847}
848
849int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000850PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000851{
Guido van Rossum82598051997-03-05 00:20:32 +0000852 PyObject *m, *d, *v;
853 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000854 if (m == NULL)
855 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000856 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000857 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000858 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000859 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000860 return -1;
861 }
Guido van Rossum82598051997-03-05 00:20:32 +0000862 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000863 if (Py_FlushLine())
864 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000865 return 0;
866}
867
Barry Warsaw035574d1997-08-29 22:07:17 +0000868static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000869parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
870 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000871{
872 long hold;
873 PyObject *v;
874
875 /* old style errors */
876 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000877 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
Neal Norwitz9589ee22006-03-04 19:01:22 +0000878 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000879
880 /* new style errors. `err' is an instance */
881
882 if (! (v = PyObject_GetAttrString(err, "msg")))
883 goto finally;
884 *message = v;
885
886 if (!(v = PyObject_GetAttrString(err, "filename")))
887 goto finally;
888 if (v == Py_None)
889 *filename = NULL;
890 else if (! (*filename = PyString_AsString(v)))
891 goto finally;
892
893 Py_DECREF(v);
894 if (!(v = PyObject_GetAttrString(err, "lineno")))
895 goto finally;
896 hold = PyInt_AsLong(v);
897 Py_DECREF(v);
898 v = NULL;
899 if (hold < 0 && PyErr_Occurred())
900 goto finally;
901 *lineno = (int)hold;
902
903 if (!(v = PyObject_GetAttrString(err, "offset")))
904 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000905 if (v == Py_None) {
906 *offset = -1;
907 Py_DECREF(v);
908 v = NULL;
909 } else {
910 hold = PyInt_AsLong(v);
911 Py_DECREF(v);
912 v = NULL;
913 if (hold < 0 && PyErr_Occurred())
914 goto finally;
915 *offset = (int)hold;
916 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000917
918 if (!(v = PyObject_GetAttrString(err, "text")))
919 goto finally;
920 if (v == Py_None)
921 *text = NULL;
922 else if (! (*text = PyString_AsString(v)))
923 goto finally;
924 Py_DECREF(v);
925 return 1;
926
927finally:
928 Py_XDECREF(v);
929 return 0;
930}
931
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000932void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000933PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000934{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000935 PyErr_PrintEx(1);
936}
937
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000938static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000939print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000940{
941 char *nl;
942 if (offset >= 0) {
943 if (offset > 0 && offset == (int)strlen(text))
944 offset--;
945 for (;;) {
946 nl = strchr(text, '\n');
947 if (nl == NULL || nl-text >= offset)
948 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000949 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000950 text = nl+1;
951 }
952 while (*text == ' ' || *text == '\t') {
953 text++;
954 offset--;
955 }
956 }
957 PyFile_WriteString(" ", f);
958 PyFile_WriteString(text, f);
959 if (*text == '\0' || text[strlen(text)-1] != '\n')
960 PyFile_WriteString("\n", f);
961 if (offset == -1)
962 return;
963 PyFile_WriteString(" ", f);
964 offset--;
965 while (offset > 0) {
966 PyFile_WriteString(" ", f);
967 offset--;
968 }
969 PyFile_WriteString("^\n", f);
970}
971
Guido van Rossum66e8e862001-03-23 17:54:43 +0000972static void
973handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000974{
Neal Norwitz9589ee22006-03-04 19:01:22 +0000975 PyObject *exception, *value, *tb;
976 int exitcode = 0;
Tim Peterscf615b52003-04-19 18:47:02 +0000977
Guido van Rossum66e8e862001-03-23 17:54:43 +0000978 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000979 if (Py_FlushLine())
980 PyErr_Clear();
981 fflush(stdout);
982 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +0000983 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +0000984 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000985 /* The error code should be in the `code' attribute. */
986 PyObject *code = PyObject_GetAttrString(value, "code");
987 if (code) {
988 Py_DECREF(value);
989 value = code;
990 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +0000991 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000992 }
993 /* If we failed to dig out the 'code' attribute,
994 just let the else clause below print the error. */
995 }
996 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +0000997 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000998 else {
999 PyObject_Print(value, stderr, Py_PRINT_RAW);
1000 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +00001001 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001002 }
Tim Peterscf615b52003-04-19 18:47:02 +00001003 done:
Neal Norwitz9589ee22006-03-04 19:01:22 +00001004 /* Restore and clear the exception info, in order to properly decref
1005 * the exception, value, and traceback. If we just exit instead,
1006 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1007 * some finalizers from running.
1008 */
Tim Peterscf615b52003-04-19 18:47:02 +00001009 PyErr_Restore(exception, value, tb);
1010 PyErr_Clear();
1011 Py_Exit(exitcode);
1012 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001013}
1014
1015void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001016PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001017{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001018 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001019
1020 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1021 handle_system_exit();
1022 }
Guido van Rossum82598051997-03-05 00:20:32 +00001023 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001024 if (exception == NULL)
1025 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001026 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001027 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001028 return;
Guido van Rossum9d785502006-03-07 18:31:44 +00001029 /* Now we know v != NULL too */
Guido van Rossuma61691e1998-02-06 22:27:24 +00001030 if (set_sys_last_vars) {
1031 PySys_SetObject("last_type", exception);
1032 PySys_SetObject("last_value", v);
1033 PySys_SetObject("last_traceback", tb);
1034 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001035 hook = PySys_GetObject("excepthook");
1036 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001037 PyObject *args = PyTuple_Pack(3,
Guido van Rossum9d785502006-03-07 18:31:44 +00001038 exception, v, tb ? tb : Py_None);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001039 PyObject *result = PyEval_CallObject(hook, args);
1040 if (result == NULL) {
1041 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001042 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1043 handle_system_exit();
1044 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001045 PyErr_Fetch(&exception2, &v2, &tb2);
1046 PyErr_NormalizeException(&exception2, &v2, &tb2);
1047 if (Py_FlushLine())
1048 PyErr_Clear();
1049 fflush(stdout);
1050 PySys_WriteStderr("Error in sys.excepthook:\n");
1051 PyErr_Display(exception2, v2, tb2);
1052 PySys_WriteStderr("\nOriginal exception was:\n");
1053 PyErr_Display(exception, v, tb);
Jeremy Hylton07028582001-12-07 15:35:35 +00001054 Py_XDECREF(exception2);
1055 Py_XDECREF(v2);
1056 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001057 }
1058 Py_XDECREF(result);
1059 Py_XDECREF(args);
1060 } else {
1061 PySys_WriteStderr("sys.excepthook is missing\n");
1062 PyErr_Display(exception, v, tb);
1063 }
1064 Py_XDECREF(exception);
1065 Py_XDECREF(v);
1066 Py_XDECREF(tb);
1067}
1068
1069void PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
1070{
1071 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001072 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001073 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001074 if (f == NULL)
1075 fprintf(stderr, "lost sys.stderr\n");
1076 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001077 if (Py_FlushLine())
1078 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001079 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001080 if (tb && tb != Py_None)
1081 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001082 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001083 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001084 {
Guido van Rossum82598051997-03-05 00:20:32 +00001085 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001086 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001087 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001088 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001089 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001090 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001091 else {
1092 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001093 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001094 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001095 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001096 else
Guido van Rossum82598051997-03-05 00:20:32 +00001097 PyFile_WriteString(filename, f);
1098 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001099 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001100 PyFile_WriteString(buf, f);
1101 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001102 if (text != NULL)
1103 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001104 Py_DECREF(value);
1105 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001106 /* Can't be bothered to check all those
1107 PyFile_WriteString() calls */
1108 if (PyErr_Occurred())
1109 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001110 }
1111 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001112 if (err) {
1113 /* Don't do anything else */
1114 }
Brett Cannonbf364092006-03-01 04:25:17 +00001115 else if (PyExceptionClass_Check(exception)) {
1116 char* className = PyExceptionClass_Name(exception);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001117 PyObject* moduleName =
Brett Cannonbf364092006-03-01 04:25:17 +00001118 PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001119
1120 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001121 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001122 else {
1123 char* modstr = PyString_AsString(moduleName);
Brett Cannon2e63b732006-03-02 18:34:57 +00001124 Py_DECREF(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001125 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001126 {
1127 err = PyFile_WriteString(modstr, f);
1128 err += PyFile_WriteString(".", f);
1129 }
1130 }
1131 if (err == 0) {
1132 if (className == NULL)
1133 err = PyFile_WriteString("<unknown>", f);
1134 else
Brett Cannonbf364092006-03-01 04:25:17 +00001135 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001136 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001137 }
1138 else
1139 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
Brett Cannon2e63b732006-03-02 18:34:57 +00001140 if (err == 0 && (value != Py_None)) {
1141 PyObject *s = PyObject_Str(value);
1142 /* only print colon if the str() of the
1143 object is not the empty string
1144 */
1145 if (s == NULL)
1146 err = -1;
1147 else if (!PyString_Check(s) ||
1148 PyString_GET_SIZE(s) != 0)
1149 err = PyFile_WriteString(": ", f);
1150 if (err == 0)
1151 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1152 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001153 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001154 if (err == 0)
1155 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001156 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001157 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001158 /* If an error happened here, don't show it.
1159 XXX This is wrong, but too many callers rely on this behavior. */
1160 if (err != 0)
1161 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001162}
1163
Guido van Rossum82598051997-03-05 00:20:32 +00001164PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001165PyRun_StringFlags(const char *str, int start, PyObject *globals,
1166 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001167{
Neal Norwitze92fba02006-03-04 18:52:26 +00001168 PyObject *ret = NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001169 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001170 mod_ty mod = PyParser_ASTFromString(str, "<string>", start, flags,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001171 arena);
Neal Norwitze92fba02006-03-04 18:52:26 +00001172 if (mod != NULL)
1173 ret = run_mod(mod, "<string>", globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001174 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001175 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001176}
1177
1178PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001179PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001180 PyObject *locals, int closeit, PyCompilerFlags *flags)
1181{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001182 PyObject *ret;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001183 PyArena *arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001184 mod_ty mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001185 flags, NULL, arena);
1186 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001187 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001188 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001189 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001190 if (closeit)
1191 fclose(fp);
Neal Norwitze92fba02006-03-04 18:52:26 +00001192 ret = run_mod(mod, filename, globals, locals, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001193 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001194 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001195}
1196
Guido van Rossum82598051997-03-05 00:20:32 +00001197static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001198run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001199 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001200{
Guido van Rossum82598051997-03-05 00:20:32 +00001201 PyCodeObject *co;
1202 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001203 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001204 if (co == NULL)
1205 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001206 v = PyEval_EvalCode(co, globals, locals);
1207 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001208 return v;
1209}
1210
Guido van Rossum82598051997-03-05 00:20:32 +00001211static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001212run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
1213 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001214{
Guido van Rossum82598051997-03-05 00:20:32 +00001215 PyCodeObject *co;
1216 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001217 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001218 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001219
Guido van Rossum82598051997-03-05 00:20:32 +00001220 magic = PyMarshal_ReadLongFromFile(fp);
1221 if (magic != PyImport_GetMagicNumber()) {
1222 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001223 "Bad magic number in .pyc file");
1224 return NULL;
1225 }
Guido van Rossum82598051997-03-05 00:20:32 +00001226 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001227 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001228 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001229 if (v == NULL || !PyCode_Check(v)) {
1230 Py_XDECREF(v);
1231 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001232 "Bad code object in .pyc file");
1233 return NULL;
1234 }
Guido van Rossum82598051997-03-05 00:20:32 +00001235 co = (PyCodeObject *)v;
1236 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001237 if (v && flags)
1238 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001239 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001240 return v;
1241}
1242
Guido van Rossum82598051997-03-05 00:20:32 +00001243PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001244Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001245 PyCompilerFlags *flags)
1246{
Guido van Rossum82598051997-03-05 00:20:32 +00001247 PyCodeObject *co;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001248 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001249 mod_ty mod = PyParser_ASTFromString(str, filename, start, flags, arena);
1250 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001251 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001252 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001253 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001254 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001255 PyObject *result = PyAST_mod2obj(mod);
1256 PyArena_Free(arena);
1257 return result;
1258 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001259 co = PyAST_Compile(mod, filename, flags, arena);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001260 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001261 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001262}
1263
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001264struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001265Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001266{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001267 struct symtable *st;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001268 PyArena *arena = PyArena_New();
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001269 mod_ty mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
1270 if (mod == NULL) {
Neal Norwitz9589ee22006-03-04 19:01:22 +00001271 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001272 return NULL;
Neal Norwitz9589ee22006-03-04 19:01:22 +00001273 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001274 st = PySymtable_Build(mod, filename, 0);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001275 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001276 return st;
1277}
1278
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001279/* Preferred access to parser is through AST. */
1280mod_ty
1281PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001282 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001283{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001284 mod_ty mod;
1285 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001286 node *n = PyParser_ParseStringFlagsFilename(s, filename,
1287 &_PyParser_Grammar, start, &err,
1288 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001289 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001290 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001291 PyNode_Free(n);
1292 return mod;
1293 }
1294 else {
1295 err_input(&err);
1296 return NULL;
1297 }
1298}
1299
1300mod_ty
1301PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001302 char *ps2, PyCompilerFlags *flags, int *errcode,
Neal Norwitz9589ee22006-03-04 19:01:22 +00001303 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001304{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001305 mod_ty mod;
1306 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001307 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1308 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001309 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001310 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001311 PyNode_Free(n);
1312 return mod;
1313 }
1314 else {
1315 err_input(&err);
1316 if (errcode)
1317 *errcode = err.error;
1318 return NULL;
1319 }
1320}
1321
Guido van Rossuma110aa61994-08-29 12:50:44 +00001322/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001323
Guido van Rossuma110aa61994-08-29 12:50:44 +00001324node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001325PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001326{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001327 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001328 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1329 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001330 if (n == NULL)
1331 err_input(&err);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001332
Guido van Rossuma110aa61994-08-29 12:50:44 +00001333 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001334}
1335
Guido van Rossuma110aa61994-08-29 12:50:44 +00001336/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001337
Guido van Rossuma110aa61994-08-29 12:50:44 +00001338node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001339PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001340{
Tim Petersfe2127d2001-07-16 05:37:24 +00001341 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001342 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1343 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001344 if (n == NULL)
1345 err_input(&err);
1346 return n;
1347}
1348
1349node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001350PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001351 int start, int flags)
1352{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001353 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001354 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1355 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001356 if (n == NULL)
1357 err_input(&err);
1358 return n;
1359}
1360
1361node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001362PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001363{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001364 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001365}
1366
Guido van Rossum66ebd912003-04-17 16:02:26 +00001367/* May want to move a more generalized form of this to parsetok.c or
1368 even parser modules. */
1369
1370void
1371PyParser_SetError(perrdetail *err)
1372{
1373 err_input(err);
1374}
1375
Guido van Rossuma110aa61994-08-29 12:50:44 +00001376/* Set the error appropriate to the given input error code (see errcode.h) */
1377
1378static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001379err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001380{
Fred Drake85f36392000-07-11 17:53:00 +00001381 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001382 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001383 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001384 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001385 switch (err->error) {
1386 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001387 errtype = PyExc_IndentationError;
1388 if (err->expected == INDENT)
1389 msg = "expected an indented block";
1390 else if (err->token == INDENT)
1391 msg = "unexpected indent";
1392 else if (err->token == DEDENT)
1393 msg = "unexpected unindent";
1394 else {
1395 errtype = PyExc_SyntaxError;
1396 msg = "invalid syntax";
1397 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001398 break;
1399 case E_TOKEN:
1400 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001401 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001402 case E_EOFS:
1403 msg = "EOF while scanning triple-quoted string";
1404 break;
1405 case E_EOLS:
1406 msg = "EOL while scanning single-quoted string";
1407 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001408 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001409 if (!PyErr_Occurred())
1410 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001411 return;
1412 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001413 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001414 return;
1415 case E_EOF:
1416 msg = "unexpected EOF while parsing";
1417 break;
Fred Drake85f36392000-07-11 17:53:00 +00001418 case E_TABSPACE:
1419 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001420 msg = "inconsistent use of tabs and spaces in indentation";
1421 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001422 case E_OVERFLOW:
1423 msg = "expression too long";
1424 break;
Fred Drake85f36392000-07-11 17:53:00 +00001425 case E_DEDENT:
1426 errtype = PyExc_IndentationError;
1427 msg = "unindent does not match any outer indentation level";
1428 break;
1429 case E_TOODEEP:
1430 errtype = PyExc_IndentationError;
1431 msg = "too many levels of indentation";
1432 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001433 case E_DECODE: {
1434 PyObject *type, *value, *tb;
1435 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001436 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001437 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001438 if (u != NULL) {
1439 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001440 }
1441 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001442 if (msg == NULL)
1443 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001444 Py_XDECREF(type);
1445 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001446 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001447 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001448 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001449 case E_LINECONT:
1450 msg = "unexpected character after line continuation character";
1451 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001452 default:
1453 fprintf(stderr, "error=%d\n", err->error);
1454 msg = "unknown parsing error";
1455 break;
1456 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001457 v = Py_BuildValue("(ziiz)", err->filename,
1458 err->lineno, err->offset, err->text);
Neal Norwitz9589ee22006-03-04 19:01:22 +00001459 if (err->text != NULL) {
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001460 PyMem_DEL(err->text);
1461 err->text = NULL;
1462 }
1463 w = NULL;
1464 if (v != NULL)
1465 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001466 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001467 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001468 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001469 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001470}
1471
1472/* Print fatal error message and abort */
1473
1474void
Tim Peters7c321a82002-07-09 02:57:01 +00001475Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001476{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001477 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001478#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001479 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001480 OutputDebugString(msg);
1481 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001482#ifdef _DEBUG
1483 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001484#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001485#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001486 abort();
1487}
1488
1489/* Clean up and exit */
1490
Guido van Rossuma110aa61994-08-29 12:50:44 +00001491#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001492#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001493#endif
1494
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001495#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001496static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001497static int nexitfuncs = 0;
1498
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001499int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001500{
1501 if (nexitfuncs >= NEXITFUNCS)
1502 return -1;
1503 exitfuncs[nexitfuncs++] = func;
1504 return 0;
1505}
1506
Guido van Rossumcc283f51997-08-05 02:22:03 +00001507static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001508call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001509{
Guido van Rossum82598051997-03-05 00:20:32 +00001510 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001511
1512 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001513 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001514 Py_INCREF(exitfunc);
1515 PySys_SetObject("exitfunc", (PyObject *)NULL);
1516 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001517 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001518 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1519 PySys_WriteStderr("Error in sys.exitfunc:\n");
1520 }
Guido van Rossum82598051997-03-05 00:20:32 +00001521 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001522 }
Guido van Rossum82598051997-03-05 00:20:32 +00001523 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001524 }
1525
Guido van Rossum0829c751998-02-28 04:31:39 +00001526 if (Py_FlushLine())
1527 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001528}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001529
Guido van Rossumcc283f51997-08-05 02:22:03 +00001530static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001531call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001532{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001533 while (nexitfuncs > 0)
1534 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001535
1536 fflush(stdout);
1537 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001538}
1539
1540void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001541Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001542{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001543 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001544
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001545 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001546}
1547
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001548static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001549initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001550{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001551#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001552 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001553#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001554#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001555 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001556#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001557#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001558 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001559#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001560 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001561}
1562
Guido van Rossum7433b121997-02-14 19:45:36 +00001563
1564/*
1565 * The file descriptor fd is considered ``interactive'' if either
1566 * a) isatty(fd) is TRUE, or
1567 * b) the -i flag was given, and the filename associated with
1568 * the descriptor is NULL or "<stdin>" or "???".
1569 */
1570int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001571Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001572{
1573 if (isatty((int)fileno(fp)))
1574 return 1;
1575 if (!Py_InteractiveFlag)
1576 return 0;
1577 return (filename == NULL) ||
1578 (strcmp(filename, "<stdin>") == 0) ||
1579 (strcmp(filename, "???") == 0);
1580}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001581
1582
Tim Petersd08e3822003-04-17 15:24:21 +00001583#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001584#if defined(WIN32) && defined(_MSC_VER)
1585
1586/* Stack checking for Microsoft C */
1587
1588#include <malloc.h>
1589#include <excpt.h>
1590
Fred Drakee8de31c2000-08-31 05:38:39 +00001591/*
1592 * Return non-zero when we run out of memory on the stack; zero otherwise.
1593 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001594int
Fred Drake399739f2000-08-31 05:52:44 +00001595PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001596{
1597 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001598 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001599 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001600 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001601 return 0;
1602 } __except (EXCEPTION_EXECUTE_HANDLER) {
1603 /* just ignore all errors */
1604 }
1605 return 1;
1606}
1607
1608#endif /* WIN32 && _MSC_VER */
1609
1610/* Alternate implementations can be added here... */
1611
1612#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001613
1614
1615/* Wrappers around sigaction() or signal(). */
1616
1617PyOS_sighandler_t
1618PyOS_getsig(int sig)
1619{
1620#ifdef HAVE_SIGACTION
1621 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001622 if (sigaction(sig, NULL, &context) == -1)
1623 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001624 return context.sa_handler;
1625#else
1626 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001627/* Special signal handling for the secure CRT in Visual Studio 2005 */
1628#if defined(_MSC_VER) && _MSC_VER >= 1400
1629 switch (sig) {
1630 /* Only these signals are valid */
1631 case SIGINT:
1632 case SIGILL:
1633 case SIGFPE:
1634 case SIGSEGV:
1635 case SIGTERM:
1636 case SIGBREAK:
1637 case SIGABRT:
1638 break;
1639 /* Don't call signal() with other values or it will assert */
1640 default:
1641 return SIG_ERR;
1642 }
1643#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001644 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001645 if (handler != SIG_ERR)
1646 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001647 return handler;
1648#endif
1649}
1650
1651PyOS_sighandler_t
1652PyOS_setsig(int sig, PyOS_sighandler_t handler)
1653{
1654#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001655 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001656 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001657 sigemptyset(&context.sa_mask);
1658 context.sa_flags = 0;
1659 if (sigaction(sig, &context, &ocontext) == -1)
1660 return SIG_ERR;
1661 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001662#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001663 PyOS_sighandler_t oldhandler;
1664 oldhandler = signal(sig, handler);
1665#ifdef HAVE_SIGINTERRUPT
1666 siginterrupt(sig, 1);
1667#endif
1668 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001669#endif
1670}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001671
1672/* Deprecated C API functions still provided for binary compatiblity */
1673
1674#undef PyParser_SimpleParseFile
1675#undef PyParser_SimpleParseString
1676
1677node *
1678PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1679{
1680 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1681}
1682
1683node *
1684PyParser_SimpleParseString(const char *str, int start)
1685{
1686 return PyParser_SimpleParseStringFlags(str, start, 0);
1687}