blob: bbe935297d92318b75c679f2763372521c96b1cb [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
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000032extern char *Py_GetPath(void);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000033
Guido van Rossum82598051997-03-05 00:20:32 +000034extern grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000035
Guido van Rossumb73cc041993-11-01 16:28:59 +000036/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000037static void initmain(void);
38static void initsite(void);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000039static PyObject *run_err_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000040 PyCompilerFlags *, PyArena *arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000041static PyObject *run_mod(mod_ty, const char *, PyObject *, PyObject *,
Neal Norwitzadb69fc2005-12-17 20:54:49 +000042 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000043static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Jeremy Hyltonbc320242001-03-22 02:47:58 +000044 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000045static void err_input(perrdetail *);
46static void initsigs(void);
47static void call_sys_exitfunc(void);
48static void call_ll_exitfuncs(void);
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000049extern void _PyUnicode_Init(void);
50extern void _PyUnicode_Fini(void);
Guido van Rossumc94044c2000-03-10 23:03:54 +000051
Mark Hammond8d98d2c2003-04-19 15:41:53 +000052#ifdef WITH_THREAD
53extern void _PyGILState_Init(PyInterpreterState *, PyThreadState *);
54extern void _PyGILState_Fini(void);
55#endif /* WITH_THREAD */
56
Guido van Rossum82598051997-03-05 00:20:32 +000057int Py_DebugFlag; /* Needed by parser.c */
58int Py_VerboseFlag; /* Needed by import.c */
Guido van Rossum7433b121997-02-14 19:45:36 +000059int Py_InteractiveFlag; /* Needed by Py_FdIsInteractive() below */
Guido van Rossumdcc0c131997-08-29 22:32:42 +000060int Py_NoSiteFlag; /* Suppress 'import site' */
Barry Warsaw3ce09642000-05-02 19:18:59 +000061int Py_UseClassExceptionsFlag = 1; /* Needed by bltinmodule.c: deprecated */
Guido van Rossuma61691e1998-02-06 22:27:24 +000062int Py_FrozenFlag; /* Needed by getpath.c */
Guido van Rossumb16d1972000-05-01 17:55:15 +000063int Py_UnicodeFlag = 0; /* Needed by compile.c */
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +000064int Py_IgnoreEnvironmentFlag; /* e.g. PYTHONPATH, PYTHONHOME */
Tim Peters3caca232001-12-06 06:23:26 +000065/* _XXX Py_QnewFlag should go away in 2.3. It's true iff -Qnew is passed,
66 on the command line, and is used in 2.2 by ceval.c to make all "/" divisions
67 true divisions (which they will be in 2.3). */
68int _Py_QnewFlag = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +000069
Mark Hammonda43fd0c2003-02-19 00:33:33 +000070/* Reference to 'warnings' module, to avoid importing it
Mark Hammondedd07732003-07-15 23:03:55 +000071 on the fly when the import lock may be held. See 683658/771097
Mark Hammonda43fd0c2003-02-19 00:33:33 +000072*/
Mark Hammondedd07732003-07-15 23:03:55 +000073static PyObject *warnings_module = NULL;
74
75/* Returns a borrowed reference to the 'warnings' module, or NULL.
76 If the module is returned, it is guaranteed to have been obtained
77 without acquiring the import lock
78*/
Martin v. Löwisa2c17c52003-08-09 09:47:11 +000079PyObject *PyModule_GetWarningsModule(void)
Mark Hammondedd07732003-07-15 23:03:55 +000080{
81 PyObject *typ, *val, *tb;
82 PyObject *all_modules;
83 /* If we managed to get the module at init time, just use it */
84 if (warnings_module)
85 return warnings_module;
86 /* If it wasn't available at init time, it may be available
87 now in sys.modules (common scenario is frozen apps: import
88 at init time fails, but the frozen init code sets up sys.path
89 correctly, then does an implicit import of warnings for us
90 */
91 /* Save and restore any exceptions */
92 PyErr_Fetch(&typ, &val, &tb);
93
Mark Hammond5f4e8ca2003-07-16 01:54:38 +000094 all_modules = PySys_GetObject("modules");
Mark Hammondedd07732003-07-15 23:03:55 +000095 if (all_modules) {
96 warnings_module = PyDict_GetItemString(all_modules, "warnings");
97 /* We keep a ref in the global */
98 Py_XINCREF(warnings_module);
99 }
100 PyErr_Restore(typ, val, tb);
101 return warnings_module;
102}
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000103
Guido van Rossum25ce5661997-08-02 03:10:38 +0000104static int initialized = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000105
Thomas Wouters7e474022000-07-16 12:04:32 +0000106/* API to access the initialized flag -- useful for esoteric use */
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000107
108int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000109Py_IsInitialized(void)
Guido van Rossume3c0d5e1997-08-22 04:20:13 +0000110{
111 return initialized;
112}
113
Guido van Rossum25ce5661997-08-02 03:10:38 +0000114/* Global initializations. Can be undone by Py_Finalize(). Don't
115 call this twice without an intervening Py_Finalize() call. When
116 initializations fail, a fatal error is issued and the function does
117 not return. On return, the first thread and interpreter state have
118 been created.
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000119
Guido van Rossum25ce5661997-08-02 03:10:38 +0000120 Locking: you must hold the interpreter lock while calling this.
121 (If the lock has not yet been initialized, that's equivalent to
122 having the lock, but you cannot use multiple threads.)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000123
Guido van Rossum25ce5661997-08-02 03:10:38 +0000124*/
Guido van Rossuma027efa1997-05-05 20:56:21 +0000125
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000126static int
127add_flag(int flag, const char *envs)
128{
129 int env = atoi(envs);
130 if (flag < env)
131 flag = env;
132 if (flag < 1)
133 flag = 1;
134 return flag;
135}
136
Guido van Rossuma027efa1997-05-05 20:56:21 +0000137void
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000138Py_InitializeEx(int install_sigs)
Guido van Rossuma027efa1997-05-05 20:56:21 +0000139{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000140 PyInterpreterState *interp;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000141 PyThreadState *tstate;
142 PyObject *bimod, *sysmod;
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000143 char *p;
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000144#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
145 char *codeset;
146 char *saved_locale;
147 PyObject *sys_stream, *sys_isatty;
148#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +0000149 extern void _Py_ReadyTypes(void);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000150
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000151 if (initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000152 return;
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000153 initialized = 1;
Tim Petersd08e3822003-04-17 15:24:21 +0000154
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000155 if ((p = Py_GETENV("PYTHONDEBUG")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000156 Py_DebugFlag = add_flag(Py_DebugFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000157 if ((p = Py_GETENV("PYTHONVERBOSE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000158 Py_VerboseFlag = add_flag(Py_VerboseFlag, p);
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000159 if ((p = Py_GETENV("PYTHONOPTIMIZE")) && *p != '\0')
Guido van Rossum9abaf4d2001-10-12 22:17:56 +0000160 Py_OptimizeFlag = add_flag(Py_OptimizeFlag, p);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000161
Guido van Rossuma027efa1997-05-05 20:56:21 +0000162 interp = PyInterpreterState_New();
163 if (interp == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000164 Py_FatalError("Py_Initialize: can't make first interpreter");
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000165
Guido van Rossuma027efa1997-05-05 20:56:21 +0000166 tstate = PyThreadState_New(interp);
167 if (tstate == NULL)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000168 Py_FatalError("Py_Initialize: can't make first thread");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000169 (void) PyThreadState_Swap(tstate);
170
Guido van Rossum70d893a2001-08-16 08:21:42 +0000171 _Py_ReadyTypes();
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000172
Neal Norwitzb2501f42002-12-31 03:42:13 +0000173 if (!_PyFrame_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000174 Py_FatalError("Py_Initialize: can't init frames");
175
Neal Norwitzb2501f42002-12-31 03:42:13 +0000176 if (!_PyInt_Init())
Neal Norwitzc91ed402002-12-30 22:29:22 +0000177 Py_FatalError("Py_Initialize: can't init ints");
178
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000179 _PyFloat_Init();
180
Guido van Rossum25ce5661997-08-02 03:10:38 +0000181 interp->modules = PyDict_New();
182 if (interp->modules == NULL)
183 Py_FatalError("Py_Initialize: can't make modules dictionary");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000184
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000185#ifdef Py_USING_UNICODE
Guido van Rossumc94044c2000-03-10 23:03:54 +0000186 /* Init Unicode implementation; relies on the codec registry */
187 _PyUnicode_Init();
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000188#endif
Guido van Rossumc94044c2000-03-10 23:03:54 +0000189
Barry Warsawf242aa02000-05-25 23:09:49 +0000190 bimod = _PyBuiltin_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000191 if (bimod == NULL)
192 Py_FatalError("Py_Initialize: can't initialize __builtin__");
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000193 interp->builtins = PyModule_GetDict(bimod);
194 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000195
196 sysmod = _PySys_Init();
197 if (sysmod == NULL)
198 Py_FatalError("Py_Initialize: can't initialize sys");
199 interp->sysdict = PyModule_GetDict(sysmod);
200 Py_INCREF(interp->sysdict);
201 _PyImport_FixupExtension("sys", "sys");
Guido van Rossuma027efa1997-05-05 20:56:21 +0000202 PySys_SetPath(Py_GetPath());
Guido van Rossum25ce5661997-08-02 03:10:38 +0000203 PyDict_SetItemString(interp->sysdict, "modules",
204 interp->modules);
205
Guido van Rossum7c85ab81999-07-08 17:26:56 +0000206 _PyImport_Init();
207
Barry Warsawf242aa02000-05-25 23:09:49 +0000208 /* initialize builtin exceptions */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000209 _PyExc_Init();
Barry Warsaw5821bc52001-08-13 23:04:56 +0000210 _PyImport_FixupExtension("exceptions", "exceptions");
Barry Warsawf242aa02000-05-25 23:09:49 +0000211
Barry Warsaw035574d1997-08-29 22:07:17 +0000212 /* phase 2 of builtins */
Barry Warsaw963b8711997-09-18 16:42:02 +0000213 _PyImport_FixupExtension("__builtin__", "__builtin__");
Barry Warsaw035574d1997-08-29 22:07:17 +0000214
Just van Rossum52e14d62002-12-30 22:08:05 +0000215 _PyImportHooks_Init();
216
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000217 if (install_sigs)
218 initsigs(); /* Signal handling stuff, including initintr() */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000219
220 initmain(); /* Module __main__ */
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000221 if (!Py_NoSiteFlag)
222 initsite(); /* Module site */
Just van Rossum5bfba3a2003-02-25 20:25:12 +0000223
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000224 /* auto-thread-state API, if available */
225#ifdef WITH_THREAD
226 _PyGILState_Init(interp, tstate);
227#endif /* WITH_THREAD */
228
Mark Hammondedd07732003-07-15 23:03:55 +0000229 warnings_module = PyImport_ImportModule("warnings");
230 if (!warnings_module)
231 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000232
233#if defined(Py_USING_UNICODE) && defined(HAVE_LANGINFO_H) && defined(CODESET)
234 /* On Unix, set the file system encoding according to the
235 user's preference, if the CODESET names a well-known
236 Python codec, and Py_FileSystemDefaultEncoding isn't
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000237 initialized by other means. Also set the encoding of
238 stdin and stdout if these are terminals. */
239
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000240 saved_locale = strdup(setlocale(LC_CTYPE, NULL));
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000241 setlocale(LC_CTYPE, "");
242 codeset = nl_langinfo(CODESET);
243 if (codeset && *codeset) {
244 PyObject *enc = PyCodec_Encoder(codeset);
245 if (enc) {
246 codeset = strdup(codeset);
247 Py_DECREF(enc);
248 } else {
249 codeset = NULL;
250 PyErr_Clear();
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000251 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000252 } else
253 codeset = NULL;
254 setlocale(LC_CTYPE, saved_locale);
Martin v. Löwisf56d0152003-11-13 07:43:21 +0000255 free(saved_locale);
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000256
257 if (codeset) {
Michael W. Hudson68debc92003-08-11 12:20:24 +0000258 sys_stream = PySys_GetObject("stdin");
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000259 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
260 if (!sys_isatty)
261 PyErr_Clear();
262 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
263 if (!PyFile_SetEncoding(sys_stream, codeset))
264 Py_FatalError("Cannot set codeset of stdin");
265 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000266 Py_XDECREF(sys_isatty);
267
268 sys_stream = PySys_GetObject("stdout");
269 sys_isatty = PyObject_CallMethod(sys_stream, "isatty", "");
270 if (!sys_isatty)
271 PyErr_Clear();
272 if(sys_isatty && PyObject_IsTrue(sys_isatty)) {
273 if (!PyFile_SetEncoding(sys_stream, codeset))
274 Py_FatalError("Cannot set codeset of stdout");
275 }
Martin v. Löwisa2c17c52003-08-09 09:47:11 +0000276 Py_XDECREF(sys_isatty);
277
278 if (!Py_FileSystemDefaultEncoding)
279 Py_FileSystemDefaultEncoding = codeset;
280 else
281 free(codeset);
Martin v. Löwis73d538b2003-03-05 15:13:47 +0000282 }
283#endif
Guido van Rossum25ce5661997-08-02 03:10:38 +0000284}
285
Martin v. Löwis336e85f2004-08-19 11:31:58 +0000286void
287Py_Initialize(void)
288{
289 Py_InitializeEx(1);
290}
291
292
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000293#ifdef COUNT_ALLOCS
Tim Petersdbd9ba62000-07-09 03:09:57 +0000294extern void dump_counts(void);
Guido van Rossum2edcf0d1998-12-15 16:12:00 +0000295#endif
296
Guido van Rossum25ce5661997-08-02 03:10:38 +0000297/* Undo the effect of Py_Initialize().
298
299 Beware: if multiple interpreter and/or thread states exist, these
300 are not wiped out; only the current thread and interpreter state
301 are deleted. But since everything else is deleted, those other
302 interpreter and thread states should no longer be used.
303
304 (XXX We should do better, e.g. wipe out all interpreters and
305 threads.)
306
307 Locking: as above.
308
309*/
310
311void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000312Py_Finalize(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000313{
314 PyInterpreterState *interp;
315 PyThreadState *tstate;
316
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000317 if (!initialized)
Guido van Rossumaa615051997-08-20 22:40:18 +0000318 return;
Guido van Rossum25ce5661997-08-02 03:10:38 +0000319
Tim Peters384fd102001-01-21 03:40:37 +0000320 /* The interpreter is still entirely intact at this point, and the
321 * exit funcs may be relying on that. In particular, if some thread
322 * or exit func is still waiting to do an import, the import machinery
323 * expects Py_IsInitialized() to return true. So don't say the
324 * interpreter is uninitialized until after the exit funcs have run.
325 * Note that Threading.py uses an exit func to do a join on all the
326 * threads created thru it, so this also protects pending imports in
327 * the threads created via Threading.
328 */
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000329 call_sys_exitfunc();
Tim Peters384fd102001-01-21 03:40:37 +0000330 initialized = 0;
Guido van Rossum4cc462e1998-01-19 22:00:38 +0000331
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000332 /* Get current thread state and interpreter pointer */
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000333 tstate = PyThreadState_GET();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000334 interp = tstate->interp;
335
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000336 /* Disable signal handling */
337 PyOS_FiniInterrupts();
338
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000339 /* drop module references we saved */
Mark Hammondedd07732003-07-15 23:03:55 +0000340 Py_XDECREF(warnings_module);
341 warnings_module = NULL;
Mark Hammonda43fd0c2003-02-19 00:33:33 +0000342
Guido van Rossume13ddc92003-04-17 17:29:22 +0000343 /* Collect garbage. This may call finalizers; it's nice to call these
Tim Peters1d7323e2003-12-01 21:35:27 +0000344 * before all modules are destroyed.
345 * XXX If a __del__ or weakref callback is triggered here, and tries to
346 * XXX import a module, bad things can happen, because Python no
347 * XXX longer believes it's initialized.
348 * XXX Fatal Python error: Interpreter not initialized (version mismatch?)
349 * XXX is easy to provoke that way. I've also seen, e.g.,
350 * XXX Exception exceptions.ImportError: 'No module named sha'
351 * XXX in <function callback at 0x008F5718> ignored
352 * XXX but I'm unclear on exactly how that one happens. In any case,
353 * XXX I haven't seen a real-life report of either of these.
354 */
Guido van Rossume13ddc92003-04-17 17:29:22 +0000355 PyGC_Collect();
356
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000357 /* Destroy all modules */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000358 PyImport_Cleanup();
Guido van Rossum3a44e1b1997-11-03 21:58:47 +0000359
Guido van Rossume13ddc92003-04-17 17:29:22 +0000360 /* Collect final garbage. This disposes of cycles created by
Tim Peters1d7323e2003-12-01 21:35:27 +0000361 * new-style class definitions, for example.
362 * XXX This is disabled because it caused too many problems. If
363 * XXX a __del__ or weakref callback triggers here, Python code has
364 * XXX a hard time running, because even the sys module has been
365 * XXX cleared out (sys.stdout is gone, sys.excepthook is gone, etc).
366 * XXX One symptom is a sequence of information-free messages
367 * XXX coming from threads (if a __del__ or callback is invoked,
368 * XXX other threads can execute too, and any exception they encounter
369 * XXX triggers a comedy of errors as subsystem after subsystem
370 * XXX fails to find what it *expects* to find in sys to help report
371 * XXX the exception and consequent unexpected failures). I've also
372 * XXX seen segfaults then, after adding print statements to the
373 * XXX Python code getting called.
374 */
375#if 0
Guido van Rossume13ddc92003-04-17 17:29:22 +0000376 PyGC_Collect();
Tim Peters1d7323e2003-12-01 21:35:27 +0000377#endif
Guido van Rossume13ddc92003-04-17 17:29:22 +0000378
Guido van Rossum1707aad1997-12-08 23:43:45 +0000379 /* Destroy the database used by _PyImport_{Fixup,Find}Extension */
380 _PyImport_Fini();
381
382 /* Debugging stuff */
383#ifdef COUNT_ALLOCS
384 dump_counts();
385#endif
386
387#ifdef Py_REF_DEBUG
388 fprintf(stderr, "[%ld refs]\n", _Py_RefTotal);
389#endif
390
Tim Peters9cf25ce2003-04-17 15:21:01 +0000391#ifdef Py_TRACE_REFS
392 /* Display all objects still alive -- this can invoke arbitrary
393 * __repr__ overrides, so requires a mostly-intact interpreter.
394 * Alas, a lot of stuff may still be alive now that will be cleaned
395 * up later.
396 */
Tim Peters269b2a62003-04-17 19:52:29 +0000397 if (Py_GETENV("PYTHONDUMPREFS"))
Tim Peters9cf25ce2003-04-17 15:21:01 +0000398 _Py_PrintReferences(stderr);
Tim Peters9cf25ce2003-04-17 15:21:01 +0000399#endif /* Py_TRACE_REFS */
400
Mark Hammond6cb90292003-04-22 11:18:00 +0000401 /* Cleanup auto-thread-state */
402#ifdef WITH_THREAD
403 _PyGILState_Fini();
404#endif /* WITH_THREAD */
405
Guido van Rossumd922fa42003-04-15 14:10:09 +0000406 /* Clear interpreter state */
Barry Warsawf242aa02000-05-25 23:09:49 +0000407 PyInterpreterState_Clear(interp);
Guido van Rossumd922fa42003-04-15 14:10:09 +0000408
Anthony Baxter12b6f6c2005-03-29 13:36:16 +0000409 /* Now we decref the exception classes. After this point nothing
410 can raise an exception. That's okay, because each Fini() method
411 below has been checked to make sure no exceptions are ever
412 raised.
413 */
414
415 _PyExc_Fini();
416
Guido van Rossumd922fa42003-04-15 14:10:09 +0000417 /* Delete current thread */
Barry Warsawf242aa02000-05-25 23:09:49 +0000418 PyThreadState_Swap(NULL);
419 PyInterpreterState_Delete(interp);
420
Guido van Rossumd922fa42003-04-15 14:10:09 +0000421 /* Sundry finalizers */
Guido van Rossumcc283f51997-08-05 02:22:03 +0000422 PyMethod_Fini();
423 PyFrame_Fini();
424 PyCFunction_Fini();
425 PyTuple_Fini();
Raymond Hettingerfb09f0e2004-10-07 03:58:07 +0000426 PyList_Fini();
Raymond Hettingerd7946662005-08-01 21:39:29 +0000427 PySet_Fini();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000428 PyString_Fini();
Guido van Rossumcc283f51997-08-05 02:22:03 +0000429 PyInt_Fini();
430 PyFloat_Fini();
431
Marc-André Lemburg95de5c12002-04-08 08:19:36 +0000432#ifdef Py_USING_UNICODE
433 /* Cleanup Unicode implementation */
434 _PyUnicode_Fini();
435#endif
436
Guido van Rossumcc283f51997-08-05 02:22:03 +0000437 /* XXX Still allocated:
438 - various static ad-hoc pointers to interned strings
439 - int and float free list blocks
440 - whatever various modules and libraries allocate
441 */
Guido van Rossum25ce5661997-08-02 03:10:38 +0000442
443 PyGrammar_RemoveAccelerators(&_PyParser_Grammar);
Guido van Rossumcc283f51997-08-05 02:22:03 +0000444
Tim Peters269b2a62003-04-17 19:52:29 +0000445#ifdef Py_TRACE_REFS
446 /* Display addresses (& refcnts) of all objects still alive.
447 * An address can be used to find the repr of the object, printed
448 * above by _Py_PrintReferences.
449 */
450 if (Py_GETENV("PYTHONDUMPREFS"))
451 _Py_PrintReferenceAddresses(stderr);
452#endif /* Py_TRACE_REFS */
Tim Peters0e871182002-04-13 08:29:14 +0000453#ifdef PYMALLOC_DEBUG
454 if (Py_GETENV("PYTHONMALLOCSTATS"))
455 _PyObject_DebugMallocStats();
456#endif
457
Guido van Rossumcc283f51997-08-05 02:22:03 +0000458 call_ll_exitfuncs();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000459}
460
461/* Create and initialize a new interpreter and thread, and return the
462 new thread. This requires that Py_Initialize() has been called
463 first.
464
465 Unsuccessful initialization yields a NULL pointer. Note that *no*
466 exception information is available even in this case -- the
467 exception information is held in the thread, and there is no
468 thread.
469
470 Locking: as above.
471
472*/
473
474PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000475Py_NewInterpreter(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000476{
477 PyInterpreterState *interp;
478 PyThreadState *tstate, *save_tstate;
479 PyObject *bimod, *sysmod;
480
481 if (!initialized)
482 Py_FatalError("Py_NewInterpreter: call Py_Initialize first");
483
484 interp = PyInterpreterState_New();
485 if (interp == NULL)
486 return NULL;
487
488 tstate = PyThreadState_New(interp);
489 if (tstate == NULL) {
490 PyInterpreterState_Delete(interp);
491 return NULL;
492 }
493
494 save_tstate = PyThreadState_Swap(tstate);
495
496 /* XXX The following is lax in error checking */
497
498 interp->modules = PyDict_New();
499
500 bimod = _PyImport_FindExtension("__builtin__", "__builtin__");
501 if (bimod != NULL) {
Guido van Rossum4a1f39a1997-11-04 19:36:18 +0000502 interp->builtins = PyModule_GetDict(bimod);
503 Py_INCREF(interp->builtins);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000504 }
505 sysmod = _PyImport_FindExtension("sys", "sys");
506 if (bimod != NULL && sysmod != NULL) {
507 interp->sysdict = PyModule_GetDict(sysmod);
508 Py_INCREF(interp->sysdict);
509 PySys_SetPath(Py_GetPath());
510 PyDict_SetItemString(interp->sysdict, "modules",
511 interp->modules);
Martin v. Löwisd7ceb222003-01-22 09:00:38 +0000512 _PyImportHooks_Init();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000513 initmain();
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000514 if (!Py_NoSiteFlag)
515 initsite();
Guido van Rossum25ce5661997-08-02 03:10:38 +0000516 }
517
518 if (!PyErr_Occurred())
519 return tstate;
520
521 /* Oops, it didn't work. Undo it all. */
522
523 PyErr_Print();
524 PyThreadState_Clear(tstate);
525 PyThreadState_Swap(save_tstate);
526 PyThreadState_Delete(tstate);
527 PyInterpreterState_Delete(interp);
528
529 return NULL;
530}
531
532/* Delete an interpreter and its last thread. This requires that the
533 given thread state is current, that the thread has no remaining
534 frames, and that it is its interpreter's only remaining thread.
535 It is a fatal error to violate these constraints.
536
537 (Py_Finalize() doesn't have these constraints -- it zaps
538 everything, regardless.)
539
540 Locking: as above.
541
542*/
543
544void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000545Py_EndInterpreter(PyThreadState *tstate)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000546{
547 PyInterpreterState *interp = tstate->interp;
548
Nicholas Bastine5662ae2004-03-24 22:22:12 +0000549 if (tstate != PyThreadState_GET())
Guido van Rossum25ce5661997-08-02 03:10:38 +0000550 Py_FatalError("Py_EndInterpreter: thread is not current");
551 if (tstate->frame != NULL)
552 Py_FatalError("Py_EndInterpreter: thread still has a frame");
553 if (tstate != interp->tstate_head || tstate->next != NULL)
554 Py_FatalError("Py_EndInterpreter: not the last thread");
555
556 PyImport_Cleanup();
557 PyInterpreterState_Clear(interp);
558 PyThreadState_Swap(NULL);
559 PyInterpreterState_Delete(interp);
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000560}
561
562static char *progname = "python";
563
564void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000565Py_SetProgramName(char *pn)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000566{
567 if (pn && *pn)
568 progname = pn;
569}
570
571char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000572Py_GetProgramName(void)
Guido van Rossumad6dfda1997-07-19 19:17:22 +0000573{
574 return progname;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000575}
576
Guido van Rossuma61691e1998-02-06 22:27:24 +0000577static char *default_home = NULL;
578
579void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000580Py_SetPythonHome(char *home)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000581{
582 default_home = home;
583}
584
585char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000586Py_GetPythonHome(void)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000587{
588 char *home = default_home;
Neil Schemenauer7d4bb9f2001-07-23 16:30:27 +0000589 if (home == NULL && !Py_IgnoreEnvironmentFlag)
590 home = Py_GETENV("PYTHONHOME");
Guido van Rossuma61691e1998-02-06 22:27:24 +0000591 return home;
592}
593
Guido van Rossum6135a871995-01-09 17:53:26 +0000594/* Create __main__ module */
595
596static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000597initmain(void)
Guido van Rossum6135a871995-01-09 17:53:26 +0000598{
Guido van Rossum82598051997-03-05 00:20:32 +0000599 PyObject *m, *d;
600 m = PyImport_AddModule("__main__");
Guido van Rossum6135a871995-01-09 17:53:26 +0000601 if (m == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +0000602 Py_FatalError("can't create __main__ module");
603 d = PyModule_GetDict(m);
604 if (PyDict_GetItemString(d, "__builtins__") == NULL) {
Guido van Rossum858cb731997-11-19 16:15:37 +0000605 PyObject *bimod = PyImport_ImportModule("__builtin__");
606 if (bimod == NULL ||
607 PyDict_SetItemString(d, "__builtins__", bimod) != 0)
Guido van Rossum82598051997-03-05 00:20:32 +0000608 Py_FatalError("can't add __builtins__ to __main__");
Barry Warsaw3d05b1a1999-01-29 21:30:22 +0000609 Py_DECREF(bimod);
Guido van Rossum6135a871995-01-09 17:53:26 +0000610 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000611}
612
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000613/* Import the site module (not into __main__ though) */
614
615static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000616initsite(void)
Guido van Rossumdcc0c131997-08-29 22:32:42 +0000617{
618 PyObject *m, *f;
619 m = PyImport_ImportModule("site");
620 if (m == NULL) {
621 f = PySys_GetObject("stderr");
622 if (Py_VerboseFlag) {
623 PyFile_WriteString(
624 "'import site' failed; traceback:\n", f);
625 PyErr_Print();
626 }
627 else {
628 PyFile_WriteString(
629 "'import site' failed; use -v for traceback\n", f);
630 PyErr_Clear();
631 }
632 }
633 else {
634 Py_DECREF(m);
635 }
636}
637
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000638/* Parse input from a file and execute it */
639
640int
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000641PyRun_AnyFileExFlags(FILE *fp, char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000642 PyCompilerFlags *flags)
643{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000644 if (filename == NULL)
645 filename = "???";
Guido van Rossum0df002c2000-08-27 19:21:52 +0000646 if (Py_FdIsInteractive(fp, filename)) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000647 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
Guido van Rossum0df002c2000-08-27 19:21:52 +0000648 if (closeit)
649 fclose(fp);
650 return err;
651 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000652 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000653 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000654}
655
656int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000657PyRun_InteractiveLoopFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000658{
Guido van Rossum82598051997-03-05 00:20:32 +0000659 PyObject *v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000660 int ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000661 PyCompilerFlags local_flags;
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000662
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000663 if (flags == NULL) {
664 flags = &local_flags;
Tim Peters5ba58662001-07-16 02:29:45 +0000665 local_flags.cf_flags = 0;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000666 }
Guido van Rossum82598051997-03-05 00:20:32 +0000667 v = PySys_GetObject("ps1");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000668 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000669 PySys_SetObject("ps1", v = PyString_FromString(">>> "));
670 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000671 }
Guido van Rossum82598051997-03-05 00:20:32 +0000672 v = PySys_GetObject("ps2");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000673 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000674 PySys_SetObject("ps2", v = PyString_FromString("... "));
675 Py_XDECREF(v);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000676 }
677 for (;;) {
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000678 ret = PyRun_InteractiveOneFlags(fp, filename, flags);
Guido van Rossumaae0d321996-05-22 16:35:33 +0000679#ifdef Py_REF_DEBUG
Guido van Rossum6f9e4331995-03-29 16:57:48 +0000680 fprintf(stderr, "[%ld refs]\n", _Py_RefTotal);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000681#endif
682 if (ret == E_EOF)
683 return 0;
684 /*
685 if (ret == E_NOMEM)
686 return -1;
687 */
688 }
689}
690
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000691/* compute parser flags based on compiler flags */
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000692#define PARSER_FLAGS(flags) \
Thomas Wouters34aa7ba2006-02-28 19:02:24 +0000693 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
694 PyPARSE_DONT_IMPLY_DEDENT : 0) \
695 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
696 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000697
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000698int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000699PyRun_InteractiveOneFlags(FILE *fp, const char *filename, PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000700{
Guido van Rossum82598051997-03-05 00:20:32 +0000701 PyObject *m, *d, *v, *w;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000702 mod_ty mod;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000703 PyArena *arena;
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000704 char *ps1 = "", *ps2 = "";
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000705 int errcode = 0;
Tim Petersfe2127d2001-07-16 05:37:24 +0000706
Guido van Rossum82598051997-03-05 00:20:32 +0000707 v = PySys_GetObject("ps1");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000708 if (v != NULL) {
709 v = PyObject_Str(v);
710 if (v == NULL)
711 PyErr_Clear();
712 else if (PyString_Check(v))
713 ps1 = PyString_AsString(v);
714 }
Guido van Rossum82598051997-03-05 00:20:32 +0000715 w = PySys_GetObject("ps2");
Guido van Rossumddc3fb51997-11-25 20:58:13 +0000716 if (w != NULL) {
717 w = PyObject_Str(w);
718 if (w == NULL)
719 PyErr_Clear();
720 else if (PyString_Check(w))
721 ps2 = PyString_AsString(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000722 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000723 arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000724 mod = PyParser_ASTFromFile(fp, filename,
725 Py_single_input, ps1, ps2,
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000726 flags, &errcode, arena);
Guido van Rossum82598051997-03-05 00:20:32 +0000727 Py_XDECREF(v);
728 Py_XDECREF(w);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000729 if (mod == NULL) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000730 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000731 if (errcode == E_EOF) {
732 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +0000733 return E_EOF;
734 }
Guido van Rossum82598051997-03-05 00:20:32 +0000735 PyErr_Print();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000736 return -1;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000737 }
Guido van Rossum82598051997-03-05 00:20:32 +0000738 m = PyImport_AddModule("__main__");
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000739 if (m == NULL) {
740 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000741 return -1;
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000742 }
Guido van Rossum82598051997-03-05 00:20:32 +0000743 d = PyModule_GetDict(m);
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000744 v = run_mod(mod, filename, d, d, flags, arena);
745 PyArena_Free(arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000746 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000747 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000748 return -1;
749 }
Guido van Rossum82598051997-03-05 00:20:32 +0000750 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000751 if (Py_FlushLine())
752 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000753 return 0;
754}
755
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000756/* Check whether a file maybe a pyc file: Look at the extension,
757 the file type, and, if we may close it, at the first few bytes. */
758
759static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000760maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000761{
762 if (strcmp(ext, ".pyc") == 0 || strcmp(ext, ".pyo") == 0)
763 return 1;
764
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000765 /* Only look into the file if we are allowed to close it, since
766 it then should also be seekable. */
767 if (closeit) {
768 /* Read only two bytes of the magic. If the file was opened in
769 text mode, the bytes 3 and 4 of the magic (\r\n) might not
770 be read as they are on disk. */
771 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
772 unsigned char buf[2];
Tim Peters3e876562001-02-11 04:35:39 +0000773 /* Mess: In case of -x, the stream is NOT at its start now,
774 and ungetc() was used to push back the first newline,
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000775 which makes the current stream position formally undefined,
776 and a x-platform nightmare.
777 Unfortunately, we have no direct way to know whether -x
778 was specified. So we use a terrible hack: if the current
779 stream position is not 0, we assume -x was specified, and
780 give up. Bug 132850 on SourceForge spells out the
781 hopelessness of trying anything else (fseek and ftell
782 don't work predictably x-platform for text-mode files).
Tim Peters3e876562001-02-11 04:35:39 +0000783 */
Tim Peters3e876562001-02-11 04:35:39 +0000784 int ispyc = 0;
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000785 if (ftell(fp) == 0) {
786 if (fread(buf, 1, 2, fp) == 2 &&
Tim Petersd08e3822003-04-17 15:24:21 +0000787 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
Tim Peters6f5a4ef2001-02-17 22:02:34 +0000788 ispyc = 1;
789 rewind(fp);
790 }
Tim Peters3e876562001-02-11 04:35:39 +0000791 return ispyc;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000792 }
793 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000794}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000795
Guido van Rossum0df002c2000-08-27 19:21:52 +0000796int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000797PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000798 PyCompilerFlags *flags)
799{
Guido van Rossum82598051997-03-05 00:20:32 +0000800 PyObject *m, *d, *v;
Martin v. Löwis95292d62002-12-11 14:04:59 +0000801 const char *ext;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000802
Guido van Rossum82598051997-03-05 00:20:32 +0000803 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000804 if (m == NULL)
805 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000806 d = PyModule_GetDict(m);
Fred Drake8ed02042002-10-17 21:24:58 +0000807 if (PyDict_GetItemString(d, "__file__") == NULL) {
808 PyObject *f = PyString_FromString(filename);
809 if (f == NULL)
810 return -1;
811 if (PyDict_SetItemString(d, "__file__", f) < 0) {
812 Py_DECREF(f);
813 return -1;
814 }
815 Py_DECREF(f);
816 }
Guido van Rossumfdef2711994-09-14 13:31:04 +0000817 ext = filename + strlen(filename) - 4;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000818 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000819 /* Try to run a pyc file. First, re-open in binary */
Guido van Rossum0df002c2000-08-27 19:21:52 +0000820 if (closeit)
821 fclose(fp);
Fred Drake8ed02042002-10-17 21:24:58 +0000822 if ((fp = fopen(filename, "rb")) == NULL) {
Guido van Rossumfdef2711994-09-14 13:31:04 +0000823 fprintf(stderr, "python: Can't reopen .pyc file\n");
824 return -1;
825 }
Guido van Rossum2a7f58d1997-04-02 05:28:38 +0000826 /* Turn on optimization if a .pyo file is given */
827 if (strcmp(ext, ".pyo") == 0)
828 Py_OptimizeFlag = 1;
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000829 v = run_pyc_file(fp, filename, d, d, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000830 } else {
Tim Petersd08e3822003-04-17 15:24:21 +0000831 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000832 closeit, flags);
Guido van Rossumfdef2711994-09-14 13:31:04 +0000833 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000834 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000835 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000836 return -1;
837 }
Guido van Rossum82598051997-03-05 00:20:32 +0000838 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000839 if (Py_FlushLine())
840 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000841 return 0;
842}
843
844int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000845PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000846{
Guido van Rossum82598051997-03-05 00:20:32 +0000847 PyObject *m, *d, *v;
848 m = PyImport_AddModule("__main__");
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000849 if (m == NULL)
850 return -1;
Guido van Rossum82598051997-03-05 00:20:32 +0000851 d = PyModule_GetDict(m);
Guido van Rossum393661d2001-08-31 17:40:15 +0000852 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000853 if (v == NULL) {
Guido van Rossum82598051997-03-05 00:20:32 +0000854 PyErr_Print();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000855 return -1;
856 }
Guido van Rossum82598051997-03-05 00:20:32 +0000857 Py_DECREF(v);
Guido van Rossum78a1ed31997-05-22 22:35:04 +0000858 if (Py_FlushLine())
859 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000860 return 0;
861}
862
Barry Warsaw035574d1997-08-29 22:07:17 +0000863static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000864parse_syntax_error(PyObject *err, PyObject **message, const char **filename,
865 int *lineno, int *offset, const char **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000866{
867 long hold;
868 PyObject *v;
869
870 /* old style errors */
871 if (PyTuple_Check(err))
Neal Norwitz78293352002-04-01 01:41:20 +0000872 return PyArg_ParseTuple(err, "O(ziiz)", message, filename,
873 lineno, offset, text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000874
875 /* new style errors. `err' is an instance */
876
877 if (! (v = PyObject_GetAttrString(err, "msg")))
878 goto finally;
879 *message = v;
880
881 if (!(v = PyObject_GetAttrString(err, "filename")))
882 goto finally;
883 if (v == Py_None)
884 *filename = NULL;
885 else if (! (*filename = PyString_AsString(v)))
886 goto finally;
887
888 Py_DECREF(v);
889 if (!(v = PyObject_GetAttrString(err, "lineno")))
890 goto finally;
891 hold = PyInt_AsLong(v);
892 Py_DECREF(v);
893 v = NULL;
894 if (hold < 0 && PyErr_Occurred())
895 goto finally;
896 *lineno = (int)hold;
897
898 if (!(v = PyObject_GetAttrString(err, "offset")))
899 goto finally;
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000900 if (v == Py_None) {
901 *offset = -1;
902 Py_DECREF(v);
903 v = NULL;
904 } else {
905 hold = PyInt_AsLong(v);
906 Py_DECREF(v);
907 v = NULL;
908 if (hold < 0 && PyErr_Occurred())
909 goto finally;
910 *offset = (int)hold;
911 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000912
913 if (!(v = PyObject_GetAttrString(err, "text")))
914 goto finally;
915 if (v == Py_None)
916 *text = NULL;
917 else if (! (*text = PyString_AsString(v)))
918 goto finally;
919 Py_DECREF(v);
920 return 1;
921
922finally:
923 Py_XDECREF(v);
924 return 0;
925}
926
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000927void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000928PyErr_Print(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000929{
Guido van Rossuma61691e1998-02-06 22:27:24 +0000930 PyErr_PrintEx(1);
931}
932
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000933static void
Martin v. Löwis95292d62002-12-11 14:04:59 +0000934print_error_text(PyObject *f, int offset, const char *text)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000935{
936 char *nl;
937 if (offset >= 0) {
938 if (offset > 0 && offset == (int)strlen(text))
939 offset--;
940 for (;;) {
941 nl = strchr(text, '\n');
942 if (nl == NULL || nl-text >= offset)
943 break;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000944 offset -= (int)(nl+1-text);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000945 text = nl+1;
946 }
947 while (*text == ' ' || *text == '\t') {
948 text++;
949 offset--;
950 }
951 }
952 PyFile_WriteString(" ", f);
953 PyFile_WriteString(text, f);
954 if (*text == '\0' || text[strlen(text)-1] != '\n')
955 PyFile_WriteString("\n", f);
956 if (offset == -1)
957 return;
958 PyFile_WriteString(" ", f);
959 offset--;
960 while (offset > 0) {
961 PyFile_WriteString(" ", f);
962 offset--;
963 }
964 PyFile_WriteString("^\n", f);
965}
966
Guido van Rossum66e8e862001-03-23 17:54:43 +0000967static void
968handle_system_exit(void)
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000969{
Guido van Rossum66e8e862001-03-23 17:54:43 +0000970 PyObject *exception, *value, *tb;
Tim Peterscf615b52003-04-19 18:47:02 +0000971 int exitcode = 0;
972
Guido van Rossum66e8e862001-03-23 17:54:43 +0000973 PyErr_Fetch(&exception, &value, &tb);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000974 if (Py_FlushLine())
975 PyErr_Clear();
976 fflush(stdout);
977 if (value == NULL || value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +0000978 goto done;
Brett Cannonbf364092006-03-01 04:25:17 +0000979 if (PyExceptionInstance_Check(value)) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000980 /* The error code should be in the `code' attribute. */
981 PyObject *code = PyObject_GetAttrString(value, "code");
982 if (code) {
983 Py_DECREF(value);
984 value = code;
985 if (value == Py_None)
Tim Peterscf615b52003-04-19 18:47:02 +0000986 goto done;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000987 }
988 /* If we failed to dig out the 'code' attribute,
989 just let the else clause below print the error. */
990 }
991 if (PyInt_Check(value))
Tim Peterscf615b52003-04-19 18:47:02 +0000992 exitcode = (int)PyInt_AsLong(value);
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000993 else {
994 PyObject_Print(value, stderr, Py_PRINT_RAW);
995 PySys_WriteStderr("\n");
Tim Peterscf615b52003-04-19 18:47:02 +0000996 exitcode = 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000997 }
Tim Peterscf615b52003-04-19 18:47:02 +0000998 done:
999 /* Restore and clear the exception info, in order to properly decref
1000 * the exception, value, and traceback. If we just exit instead,
1001 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
1002 * some finalizers from running.
1003 */
1004 PyErr_Restore(exception, value, tb);
1005 PyErr_Clear();
1006 Py_Exit(exitcode);
1007 /* NOTREACHED */
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001008}
1009
1010void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001011PyErr_PrintEx(int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +00001012{
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001013 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001014
1015 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1016 handle_system_exit();
1017 }
Guido van Rossum82598051997-03-05 00:20:32 +00001018 PyErr_Fetch(&exception, &v, &tb);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001019 if (exception == NULL)
1020 return;
Barry Warsaw035574d1997-08-29 22:07:17 +00001021 PyErr_NormalizeException(&exception, &v, &tb);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001022 if (exception == NULL)
Guido van Rossum296b4751997-05-23 00:19:20 +00001023 return;
Guido van Rossuma61691e1998-02-06 22:27:24 +00001024 if (set_sys_last_vars) {
1025 PySys_SetObject("last_type", exception);
1026 PySys_SetObject("last_value", v);
1027 PySys_SetObject("last_traceback", tb);
1028 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001029 hook = PySys_GetObject("excepthook");
1030 if (hook) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001031 PyObject *args = PyTuple_Pack(3,
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001032 exception, v ? v : Py_None, tb ? tb : Py_None);
1033 PyObject *result = PyEval_CallObject(hook, args);
1034 if (result == NULL) {
1035 PyObject *exception2, *v2, *tb2;
Guido van Rossum66e8e862001-03-23 17:54:43 +00001036 if (PyErr_ExceptionMatches(PyExc_SystemExit)) {
1037 handle_system_exit();
1038 }
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001039 PyErr_Fetch(&exception2, &v2, &tb2);
1040 PyErr_NormalizeException(&exception2, &v2, &tb2);
1041 if (Py_FlushLine())
1042 PyErr_Clear();
1043 fflush(stdout);
1044 PySys_WriteStderr("Error in sys.excepthook:\n");
1045 PyErr_Display(exception2, v2, tb2);
1046 PySys_WriteStderr("\nOriginal exception was:\n");
1047 PyErr_Display(exception, v, tb);
Jeremy Hylton07028582001-12-07 15:35:35 +00001048 Py_XDECREF(exception2);
1049 Py_XDECREF(v2);
1050 Py_XDECREF(tb2);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001051 }
1052 Py_XDECREF(result);
1053 Py_XDECREF(args);
1054 } else {
1055 PySys_WriteStderr("sys.excepthook is missing\n");
1056 PyErr_Display(exception, v, tb);
1057 }
1058 Py_XDECREF(exception);
1059 Py_XDECREF(v);
1060 Py_XDECREF(tb);
1061}
1062
1063void PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
1064{
1065 int err = 0;
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001066 PyObject *f = PySys_GetObject("stderr");
Armin Rigo5d2c6832004-03-22 20:16:58 +00001067 Py_INCREF(value);
Guido van Rossum3165fe61992-09-25 21:59:05 +00001068 if (f == NULL)
1069 fprintf(stderr, "lost sys.stderr\n");
1070 else {
Guido van Rossum0829c751998-02-28 04:31:39 +00001071 if (Py_FlushLine())
1072 PyErr_Clear();
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001073 fflush(stdout);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001074 if (tb && tb != Py_None)
1075 err = PyTraceBack_Print(tb, f);
Barry Warsaw36b8f941997-08-26 18:09:48 +00001076 if (err == 0 &&
Armin Rigo5d2c6832004-03-22 20:16:58 +00001077 PyObject_HasAttrString(value, "print_file_and_line"))
Barry Warsaw36b8f941997-08-26 18:09:48 +00001078 {
Guido van Rossum82598051997-03-05 00:20:32 +00001079 PyObject *message;
Martin v. Löwis95292d62002-12-11 14:04:59 +00001080 const char *filename, *text;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001081 int lineno, offset;
Armin Rigo5d2c6832004-03-22 20:16:58 +00001082 if (!parse_syntax_error(value, &message, &filename,
Barry Warsaw035574d1997-08-29 22:07:17 +00001083 &lineno, &offset, &text))
Guido van Rossum82598051997-03-05 00:20:32 +00001084 PyErr_Clear();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001085 else {
1086 char buf[10];
Guido van Rossum82598051997-03-05 00:20:32 +00001087 PyFile_WriteString(" File \"", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001088 if (filename == NULL)
Guido van Rossum82598051997-03-05 00:20:32 +00001089 PyFile_WriteString("<string>", f);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001090 else
Guido van Rossum82598051997-03-05 00:20:32 +00001091 PyFile_WriteString(filename, f);
1092 PyFile_WriteString("\", line ", f);
Jeremy Hylton518ab1c2001-11-28 20:42:20 +00001093 PyOS_snprintf(buf, sizeof(buf), "%d", lineno);
Guido van Rossum82598051997-03-05 00:20:32 +00001094 PyFile_WriteString(buf, f);
1095 PyFile_WriteString("\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +00001096 if (text != NULL)
1097 print_error_text(f, offset, text);
Armin Rigo5d2c6832004-03-22 20:16:58 +00001098 Py_DECREF(value);
1099 value = message;
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001100 /* Can't be bothered to check all those
1101 PyFile_WriteString() calls */
1102 if (PyErr_Occurred())
1103 err = -1;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001104 }
1105 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001106 if (err) {
1107 /* Don't do anything else */
1108 }
Brett Cannonbf364092006-03-01 04:25:17 +00001109 else if (PyExceptionClass_Check(exception)) {
1110 char* className = PyExceptionClass_Name(exception);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001111 PyObject* moduleName =
Brett Cannonbf364092006-03-01 04:25:17 +00001112 PyObject_GetAttrString(exception, "__module__");
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001113
1114 if (moduleName == NULL)
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001115 err = PyFile_WriteString("<unknown>", f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001116 else {
1117 char* modstr = PyString_AsString(moduleName);
Tim Petersd08e3822003-04-17 15:24:21 +00001118 if (modstr && strcmp(modstr, "exceptions"))
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001119 {
1120 err = PyFile_WriteString(modstr, f);
1121 err += PyFile_WriteString(".", f);
1122 }
1123 }
1124 if (err == 0) {
1125 if (className == NULL)
1126 err = PyFile_WriteString("<unknown>", f);
1127 else
Brett Cannonbf364092006-03-01 04:25:17 +00001128 err = PyFile_WriteString(className, f);
Barry Warsaw2f5f6a21997-09-16 21:42:03 +00001129 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001130 }
1131 else
1132 err = PyFile_WriteObject(exception, f, Py_PRINT_RAW);
1133 if (err == 0) {
Armin Rigo5d2c6832004-03-22 20:16:58 +00001134 if (value != Py_None) {
1135 PyObject *s = PyObject_Str(value);
Barry Warsaw035574d1997-08-29 22:07:17 +00001136 /* only print colon if the str() of the
1137 object is not the empty string
1138 */
Guido van Rossumd6bf45b1997-09-05 19:11:53 +00001139 if (s == NULL)
1140 err = -1;
1141 else if (!PyString_Check(s) ||
1142 PyString_GET_SIZE(s) != 0)
Barry Warsaw035574d1997-08-29 22:07:17 +00001143 err = PyFile_WriteString(": ", f);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001144 if (err == 0)
Guido van Rossumd6bf45b1997-09-05 19:11:53 +00001145 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
1146 Py_XDECREF(s);
Guido van Rossum262e1241995-02-07 15:30:45 +00001147 }
Guido van Rossum262e1241995-02-07 15:30:45 +00001148 }
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001149 if (err == 0)
1150 err = PyFile_WriteString("\n", f);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001151 }
Armin Rigo5d2c6832004-03-22 20:16:58 +00001152 Py_DECREF(value);
Guido van Rossum78a1ed31997-05-22 22:35:04 +00001153 /* If an error happened here, don't show it.
1154 XXX This is wrong, but too many callers rely on this behavior. */
1155 if (err != 0)
1156 PyErr_Clear();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001157}
1158
Guido van Rossum82598051997-03-05 00:20:32 +00001159PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001160PyRun_StringFlags(const char *str, int start, PyObject *globals,
1161 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001162{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001163 PyObject *ret;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001164 PyArena *arena = PyArena_New();
1165 mod_ty mod = PyParser_ASTFromString(str, "<string>", start, flags,
1166 arena);
1167 ret = run_err_mod(mod, "<string>", globals, locals, flags, arena);
1168 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001169 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001170}
1171
1172PyObject *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001173PyRun_FileExFlags(FILE *fp, const char *filename, int start, PyObject *globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001174 PyObject *locals, int closeit, PyCompilerFlags *flags)
1175{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001176 PyObject *ret;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001177 PyArena *arena = PyArena_New();
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001178 mod_ty mod = PyParser_ASTFromFile(fp, filename, start, 0, 0,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001179 flags, NULL, arena);
1180 if (mod == NULL) {
1181 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001182 return NULL;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001183 }
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001184 if (closeit)
1185 fclose(fp);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001186 ret = run_err_mod(mod, filename, globals, locals, flags, arena);
1187 PyArena_Free(arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001188 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001189}
1190
Guido van Rossum82598051997-03-05 00:20:32 +00001191static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001192run_err_mod(mod_ty mod, const char *filename, PyObject *globals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001193 PyObject *locals, PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001194{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001195 if (mod == NULL)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001196 return NULL;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001197 return run_mod(mod, filename, globals, locals, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001198}
1199
Guido van Rossum82598051997-03-05 00:20:32 +00001200static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001201run_mod(mod_ty mod, const char *filename, PyObject *globals, PyObject *locals,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001202 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001203{
Guido van Rossum82598051997-03-05 00:20:32 +00001204 PyCodeObject *co;
1205 PyObject *v;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001206 co = PyAST_Compile(mod, filename, flags, arena);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001207 if (co == NULL)
1208 return NULL;
Guido van Rossum82598051997-03-05 00:20:32 +00001209 v = PyEval_EvalCode(co, globals, locals);
1210 Py_DECREF(co);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001211 return v;
1212}
1213
Guido van Rossum82598051997-03-05 00:20:32 +00001214static PyObject *
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001215run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
1216 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001217{
Guido van Rossum82598051997-03-05 00:20:32 +00001218 PyCodeObject *co;
1219 PyObject *v;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001220 long magic;
Fred Drakee8de31c2000-08-31 05:38:39 +00001221 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001222
Guido van Rossum82598051997-03-05 00:20:32 +00001223 magic = PyMarshal_ReadLongFromFile(fp);
1224 if (magic != PyImport_GetMagicNumber()) {
1225 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001226 "Bad magic number in .pyc file");
1227 return NULL;
1228 }
Guido van Rossum82598051997-03-05 00:20:32 +00001229 (void) PyMarshal_ReadLongFromFile(fp);
Tim Petersd9b9ac82001-01-28 00:27:39 +00001230 v = PyMarshal_ReadLastObjectFromFile(fp);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001231 fclose(fp);
Guido van Rossum82598051997-03-05 00:20:32 +00001232 if (v == NULL || !PyCode_Check(v)) {
1233 Py_XDECREF(v);
1234 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumfdef2711994-09-14 13:31:04 +00001235 "Bad code object in .pyc file");
1236 return NULL;
1237 }
Guido van Rossum82598051997-03-05 00:20:32 +00001238 co = (PyCodeObject *)v;
1239 v = PyEval_EvalCode(co, globals, locals);
Jeremy Hyltonb857ba22001-08-10 21:41:33 +00001240 if (v && flags)
1241 flags->cf_flags |= (co->co_flags & PyCF_MASK);
Guido van Rossum82598051997-03-05 00:20:32 +00001242 Py_DECREF(co);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001243 return v;
1244}
1245
Guido van Rossum82598051997-03-05 00:20:32 +00001246PyObject *
Tim Petersd08e3822003-04-17 15:24:21 +00001247Py_CompileStringFlags(const char *str, const char *filename, int start,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001248 PyCompilerFlags *flags)
1249{
Guido van Rossum82598051997-03-05 00:20:32 +00001250 PyCodeObject *co;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001251 PyArena *arena = PyArena_New();
1252 mod_ty mod = PyParser_ASTFromString(str, filename, start, flags, arena);
1253 if (mod == NULL) {
1254 PyArena_Free(arena);
Guido van Rossum5b722181993-03-30 17:46:03 +00001255 return NULL;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001256 }
Martin v. Löwis2b366e42006-02-26 22:12:35 +00001257 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001258 PyObject *result = PyAST_mod2obj(mod);
1259 PyArena_Free(arena);
1260 return result;
1261 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001262 co = PyAST_Compile(mod, filename, flags, arena);
1263 PyArena_Free(arena);
Guido van Rossum82598051997-03-05 00:20:32 +00001264 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001265}
1266
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001267struct symtable *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001268Py_SymtableString(const char *str, const char *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001269{
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001270 struct symtable *st;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001271 PyArena *arena = PyArena_New();
1272 mod_ty mod = PyParser_ASTFromString(str, filename, start, NULL, arena);
1273 if (mod == NULL) {
1274 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001275 return NULL;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001276 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001277 st = PySymtable_Build(mod, filename, 0);
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001278 PyArena_Free(arena);
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001279 return st;
1280}
1281
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001282/* Preferred access to parser is through AST. */
1283mod_ty
1284PyParser_ASTFromString(const char *s, const char *filename, int start,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001285 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001286{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001287 mod_ty mod;
1288 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001289 node *n = PyParser_ParseStringFlagsFilename(s, filename,
1290 &_PyParser_Grammar, start, &err,
1291 PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001292 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001293 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001294 PyNode_Free(n);
1295 return mod;
1296 }
1297 else {
1298 err_input(&err);
1299 return NULL;
1300 }
1301}
1302
1303mod_ty
1304PyParser_ASTFromFile(FILE *fp, const char *filename, int start, char *ps1,
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001305 char *ps2, PyCompilerFlags *flags, int *errcode,
1306 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001307{
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001308 mod_ty mod;
1309 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001310 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1311 start, ps1, ps2, &err, PARSER_FLAGS(flags));
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001312 if (n) {
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001313 mod = PyAST_FromNode(n, flags, filename, arena);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001314 PyNode_Free(n);
1315 return mod;
1316 }
1317 else {
1318 err_input(&err);
1319 if (errcode)
1320 *errcode = err.error;
1321 return NULL;
1322 }
1323}
1324
Guido van Rossuma110aa61994-08-29 12:50:44 +00001325/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001326
Guido van Rossuma110aa61994-08-29 12:50:44 +00001327node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001328PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001329{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001330 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001331 node *n = PyParser_ParseFileFlags(fp, filename, &_PyParser_Grammar,
1332 start, NULL, NULL, &err, flags);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001333 if (n == NULL)
1334 err_input(&err);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001335
Guido van Rossuma110aa61994-08-29 12:50:44 +00001336 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001337}
1338
Guido van Rossuma110aa61994-08-29 12:50:44 +00001339/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001340
Guido van Rossuma110aa61994-08-29 12:50:44 +00001341node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001342PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001343{
Tim Petersfe2127d2001-07-16 05:37:24 +00001344 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001345 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1346 start, &err, flags);
Tim Petersfe2127d2001-07-16 05:37:24 +00001347 if (n == NULL)
1348 err_input(&err);
1349 return n;
1350}
1351
1352node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001353PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Thomas Heller6b17abf2002-07-09 09:23:27 +00001354 int start, int flags)
1355{
Thomas Heller6b17abf2002-07-09 09:23:27 +00001356 perrdetail err;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001357 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1358 &_PyParser_Grammar, start, &err, flags);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001359 if (n == NULL)
1360 err_input(&err);
1361 return n;
1362}
1363
1364node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001365PyParser_SimpleParseStringFilename(const char *str, const char *filename, int start)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001366{
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001367 return PyParser_SimpleParseStringFlagsFilename(str, filename, start, 0);
Thomas Heller6b17abf2002-07-09 09:23:27 +00001368}
1369
Guido van Rossum66ebd912003-04-17 16:02:26 +00001370/* May want to move a more generalized form of this to parsetok.c or
1371 even parser modules. */
1372
1373void
1374PyParser_SetError(perrdetail *err)
1375{
1376 err_input(err);
1377}
1378
Guido van Rossuma110aa61994-08-29 12:50:44 +00001379/* Set the error appropriate to the given input error code (see errcode.h) */
1380
1381static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001382err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001383{
Fred Drake85f36392000-07-11 17:53:00 +00001384 PyObject *v, *w, *errtype;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001385 PyObject* u = NULL;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001386 char *msg = NULL;
Fred Drake85f36392000-07-11 17:53:00 +00001387 errtype = PyExc_SyntaxError;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001388 switch (err->error) {
1389 case E_SYNTAX:
Fred Drake85f36392000-07-11 17:53:00 +00001390 errtype = PyExc_IndentationError;
1391 if (err->expected == INDENT)
1392 msg = "expected an indented block";
1393 else if (err->token == INDENT)
1394 msg = "unexpected indent";
1395 else if (err->token == DEDENT)
1396 msg = "unexpected unindent";
1397 else {
1398 errtype = PyExc_SyntaxError;
1399 msg = "invalid syntax";
1400 }
Guido van Rossuma110aa61994-08-29 12:50:44 +00001401 break;
1402 case E_TOKEN:
1403 msg = "invalid token";
Guido van Rossuma110aa61994-08-29 12:50:44 +00001404 break;
Skip Montanaro118ec702002-08-15 01:20:16 +00001405 case E_EOFS:
1406 msg = "EOF while scanning triple-quoted string";
1407 break;
1408 case E_EOLS:
1409 msg = "EOL while scanning single-quoted string";
1410 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001411 case E_INTR:
Michael W. Hudson30ea2f22004-07-07 17:44:12 +00001412 if (!PyErr_Occurred())
1413 PyErr_SetNone(PyExc_KeyboardInterrupt);
Guido van Rossuma110aa61994-08-29 12:50:44 +00001414 return;
1415 case E_NOMEM:
Guido van Rossum82598051997-03-05 00:20:32 +00001416 PyErr_NoMemory();
Guido van Rossuma110aa61994-08-29 12:50:44 +00001417 return;
1418 case E_EOF:
1419 msg = "unexpected EOF while parsing";
1420 break;
Fred Drake85f36392000-07-11 17:53:00 +00001421 case E_TABSPACE:
1422 errtype = PyExc_TabError;
Guido van Rossum560e8ad1998-04-10 19:43:42 +00001423 msg = "inconsistent use of tabs and spaces in indentation";
1424 break;
Jeremy Hylton94988062000-06-20 19:10:44 +00001425 case E_OVERFLOW:
1426 msg = "expression too long";
1427 break;
Fred Drake85f36392000-07-11 17:53:00 +00001428 case E_DEDENT:
1429 errtype = PyExc_IndentationError;
1430 msg = "unindent does not match any outer indentation level";
1431 break;
1432 case E_TOODEEP:
1433 errtype = PyExc_IndentationError;
1434 msg = "too many levels of indentation";
1435 break;
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001436 case E_DECODE: {
1437 PyObject *type, *value, *tb;
1438 PyErr_Fetch(&type, &value, &tb);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001439 if (value != NULL) {
Martin v. Löwisd35edda2005-08-24 08:39:24 +00001440 u = PyObject_Str(value);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001441 if (u != NULL) {
1442 msg = PyString_AsString(u);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001443 }
1444 }
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001445 if (msg == NULL)
1446 msg = "unknown decode error";
Neal Norwitzdb83eb32005-12-18 05:29:30 +00001447 Py_XDECREF(type);
1448 Py_XDECREF(value);
Neal Norwitz40d37812005-10-02 01:48:49 +00001449 Py_XDECREF(tb);
Martin v. Löwisc2632a52004-07-21 05:35:02 +00001450 break;
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001451 }
Martin v. Löwis4bf108d2005-03-03 11:45:45 +00001452 case E_LINECONT:
1453 msg = "unexpected character after line continuation character";
1454 break;
Guido van Rossuma110aa61994-08-29 12:50:44 +00001455 default:
1456 fprintf(stderr, "error=%d\n", err->error);
1457 msg = "unknown parsing error";
1458 break;
1459 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001460 v = Py_BuildValue("(ziiz)", err->filename,
1461 err->lineno, err->offset, err->text);
1462 if (err->text != NULL) {
1463 PyMem_DEL(err->text);
1464 err->text = NULL;
1465 }
1466 w = NULL;
1467 if (v != NULL)
1468 w = Py_BuildValue("(sO)", msg, v);
Martin v. Löwis00f1e3f2002-08-04 17:29:52 +00001469 Py_XDECREF(u);
Guido van Rossum41318302001-03-23 04:01:07 +00001470 Py_XDECREF(v);
Fred Drake85f36392000-07-11 17:53:00 +00001471 PyErr_SetObject(errtype, w);
Guido van Rossum82598051997-03-05 00:20:32 +00001472 Py_XDECREF(w);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001473}
1474
1475/* Print fatal error message and abort */
1476
1477void
Tim Peters7c321a82002-07-09 02:57:01 +00001478Py_FatalError(const char *msg)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001479{
Guido van Rossum83dd6c31994-09-29 09:38:33 +00001480 fprintf(stderr, "Fatal Python error: %s\n", msg);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001481#ifdef MS_WINDOWS
Guido van Rossum23c94461997-05-22 20:21:30 +00001482 OutputDebugString("Fatal Python error: ");
Guido van Rossuma44823b1995-03-14 15:01:17 +00001483 OutputDebugString(msg);
1484 OutputDebugString("\n");
Guido van Rossum0ba35361998-08-13 13:33:16 +00001485#ifdef _DEBUG
1486 DebugBreak();
Guido van Rossuma44823b1995-03-14 15:01:17 +00001487#endif
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001488#endif /* MS_WINDOWS */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001489 abort();
1490}
1491
1492/* Clean up and exit */
1493
Guido van Rossuma110aa61994-08-29 12:50:44 +00001494#ifdef WITH_THREAD
Guido van Rossum49b56061998-10-01 20:42:43 +00001495#include "pythread.h"
Guido van Rossumf9f2e821992-08-17 08:59:08 +00001496#endif
1497
Guido van Rossum2dcfc961998-10-01 16:01:57 +00001498#define NEXITFUNCS 32
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001499static void (*exitfuncs[NEXITFUNCS])(void);
Guido van Rossum1662dd51994-09-07 14:38:28 +00001500static int nexitfuncs = 0;
1501
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001502int Py_AtExit(void (*func)(void))
Guido van Rossum1662dd51994-09-07 14:38:28 +00001503{
1504 if (nexitfuncs >= NEXITFUNCS)
1505 return -1;
1506 exitfuncs[nexitfuncs++] = func;
1507 return 0;
1508}
1509
Guido van Rossumcc283f51997-08-05 02:22:03 +00001510static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001511call_sys_exitfunc(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001512{
Guido van Rossum82598051997-03-05 00:20:32 +00001513 PyObject *exitfunc = PySys_GetObject("exitfunc");
Guido van Rossum59bff391992-09-03 20:28:00 +00001514
1515 if (exitfunc) {
Fred Drake6a12d8d2001-03-23 17:34:02 +00001516 PyObject *res;
Guido van Rossum82598051997-03-05 00:20:32 +00001517 Py_INCREF(exitfunc);
1518 PySys_SetObject("exitfunc", (PyObject *)NULL);
1519 res = PyEval_CallObject(exitfunc, (PyObject *)NULL);
Guido van Rossum59bff391992-09-03 20:28:00 +00001520 if (res == NULL) {
Ka-Ping Yee26fabb02001-03-23 15:36:41 +00001521 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
1522 PySys_WriteStderr("Error in sys.exitfunc:\n");
1523 }
Guido van Rossum82598051997-03-05 00:20:32 +00001524 PyErr_Print();
Guido van Rossum59bff391992-09-03 20:28:00 +00001525 }
Guido van Rossum82598051997-03-05 00:20:32 +00001526 Py_DECREF(exitfunc);
Guido van Rossum59bff391992-09-03 20:28:00 +00001527 }
1528
Guido van Rossum0829c751998-02-28 04:31:39 +00001529 if (Py_FlushLine())
1530 PyErr_Clear();
Guido van Rossumcc283f51997-08-05 02:22:03 +00001531}
Guido van Rossum1662dd51994-09-07 14:38:28 +00001532
Guido van Rossumcc283f51997-08-05 02:22:03 +00001533static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001534call_ll_exitfuncs(void)
Guido van Rossumcc283f51997-08-05 02:22:03 +00001535{
Guido van Rossum1662dd51994-09-07 14:38:28 +00001536 while (nexitfuncs > 0)
1537 (*exitfuncs[--nexitfuncs])();
Guido van Rossum25ce5661997-08-02 03:10:38 +00001538
1539 fflush(stdout);
1540 fflush(stderr);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001541}
1542
1543void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001544Py_Exit(int sts)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001545{
Guido van Rossumcc283f51997-08-05 02:22:03 +00001546 Py_Finalize();
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001547
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001548 exit(sts);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001549}
1550
Guido van Rossumf1dc5661993-07-05 10:31:29 +00001551static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001552initsigs(void)
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001553{
Guido van Rossuma110aa61994-08-29 12:50:44 +00001554#ifdef SIGPIPE
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001555 PyOS_setsig(SIGPIPE, SIG_IGN);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001556#endif
Guido van Rossum70d893a2001-08-16 08:21:42 +00001557#ifdef SIGXFZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001558 PyOS_setsig(SIGXFZ, SIG_IGN);
Guido van Rossum70d893a2001-08-16 08:21:42 +00001559#endif
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001560#ifdef SIGXFSZ
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001561 PyOS_setsig(SIGXFSZ, SIG_IGN);
Jeremy Hylton1b0bf9b2002-04-23 20:31:01 +00001562#endif
Guido van Rossum82598051997-03-05 00:20:32 +00001563 PyOS_InitInterrupts(); /* May imply initsignal() */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +00001564}
1565
Guido van Rossum7433b121997-02-14 19:45:36 +00001566
1567/*
1568 * The file descriptor fd is considered ``interactive'' if either
1569 * a) isatty(fd) is TRUE, or
1570 * b) the -i flag was given, and the filename associated with
1571 * the descriptor is NULL or "<stdin>" or "???".
1572 */
1573int
Martin v. Löwis95292d62002-12-11 14:04:59 +00001574Py_FdIsInteractive(FILE *fp, const char *filename)
Guido van Rossum7433b121997-02-14 19:45:36 +00001575{
1576 if (isatty((int)fileno(fp)))
1577 return 1;
1578 if (!Py_InteractiveFlag)
1579 return 0;
1580 return (filename == NULL) ||
1581 (strcmp(filename, "<stdin>") == 0) ||
1582 (strcmp(filename, "???") == 0);
1583}
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001584
1585
Tim Petersd08e3822003-04-17 15:24:21 +00001586#if defined(USE_STACKCHECK)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001587#if defined(WIN32) && defined(_MSC_VER)
1588
1589/* Stack checking for Microsoft C */
1590
1591#include <malloc.h>
1592#include <excpt.h>
1593
Fred Drakee8de31c2000-08-31 05:38:39 +00001594/*
1595 * Return non-zero when we run out of memory on the stack; zero otherwise.
1596 */
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001597int
Fred Drake399739f2000-08-31 05:52:44 +00001598PyOS_CheckStack(void)
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001599{
1600 __try {
Tim Peters92e4dd82002-10-05 01:47:34 +00001601 /* alloca throws a stack overflow exception if there's
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001602 not enough space left on the stack */
Tim Peters92e4dd82002-10-05 01:47:34 +00001603 alloca(PYOS_STACK_MARGIN * sizeof(void*));
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001604 return 0;
1605 } __except (EXCEPTION_EXECUTE_HANDLER) {
1606 /* just ignore all errors */
1607 }
1608 return 1;
1609}
1610
1611#endif /* WIN32 && _MSC_VER */
1612
1613/* Alternate implementations can be added here... */
1614
1615#endif /* USE_STACKCHECK */
Guido van Rossum6f256182000-09-16 16:32:19 +00001616
1617
1618/* Wrappers around sigaction() or signal(). */
1619
1620PyOS_sighandler_t
1621PyOS_getsig(int sig)
1622{
1623#ifdef HAVE_SIGACTION
1624 struct sigaction context;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001625 if (sigaction(sig, NULL, &context) == -1)
1626 return SIG_ERR;
Guido van Rossum6f256182000-09-16 16:32:19 +00001627 return context.sa_handler;
1628#else
1629 PyOS_sighandler_t handler;
Martin v. Löwisb45b3152005-11-28 17:34:23 +00001630/* Special signal handling for the secure CRT in Visual Studio 2005 */
1631#if defined(_MSC_VER) && _MSC_VER >= 1400
1632 switch (sig) {
1633 /* Only these signals are valid */
1634 case SIGINT:
1635 case SIGILL:
1636 case SIGFPE:
1637 case SIGSEGV:
1638 case SIGTERM:
1639 case SIGBREAK:
1640 case SIGABRT:
1641 break;
1642 /* Don't call signal() with other values or it will assert */
1643 default:
1644 return SIG_ERR;
1645 }
1646#endif /* _MSC_VER && _MSC_VER >= 1400 */
Guido van Rossum6f256182000-09-16 16:32:19 +00001647 handler = signal(sig, SIG_IGN);
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001648 if (handler != SIG_ERR)
1649 signal(sig, handler);
Guido van Rossum6f256182000-09-16 16:32:19 +00001650 return handler;
1651#endif
1652}
1653
1654PyOS_sighandler_t
1655PyOS_setsig(int sig, PyOS_sighandler_t handler)
1656{
1657#ifdef HAVE_SIGACTION
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001658 struct sigaction context, ocontext;
Guido van Rossum6f256182000-09-16 16:32:19 +00001659 context.sa_handler = handler;
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001660 sigemptyset(&context.sa_mask);
1661 context.sa_flags = 0;
1662 if (sigaction(sig, &context, &ocontext) == -1)
1663 return SIG_ERR;
1664 return ocontext.sa_handler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001665#else
Anthony Baxter9ceaa722004-10-13 14:48:50 +00001666 PyOS_sighandler_t oldhandler;
1667 oldhandler = signal(sig, handler);
1668#ifdef HAVE_SIGINTERRUPT
1669 siginterrupt(sig, 1);
1670#endif
1671 return oldhandler;
Guido van Rossum6f256182000-09-16 16:32:19 +00001672#endif
1673}
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001674
1675/* Deprecated C API functions still provided for binary compatiblity */
1676
1677#undef PyParser_SimpleParseFile
1678#undef PyParser_SimpleParseString
1679
1680node *
1681PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1682{
1683 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
1684}
1685
1686node *
1687PyParser_SimpleParseString(const char *str, int start)
1688{
1689 return PyParser_SimpleParseStringFlags(str, start, 0);
1690}