blob: cb0e3b02e163ad8cff13c11894c3294d3cf838d8 [file] [log] [blame]
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001
Eric Snowc7ec9982017-05-23 23:00:52 -07002/* Top level execution of Python code (including in __main__) */
3
4/* To help control the interfaces between the startup, execution and
5 * shutdown code, the phases are split across separate modules (boostrap,
6 * pythonrun, shutdown)
7 */
8
9/* TODO: Cull includes following phase split */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000010
Guido van Rossum82598051997-03-05 00:20:32 +000011#include "Python.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +000012
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000013#include "Python-ast.h"
Victor Stinner3bb183d2018-11-22 18:38:38 +010014#undef Yield /* undefine macro conflicting with <winbase.h> */
Victor Stinner4f98f462020-04-15 04:01:58 +020015
Victor Stinnere5014be2020-04-14 17:52:15 +020016#include "pycore_interp.h" // PyInterpreterState.importlib
Victor Stinner4f98f462020-04-15 04:01:58 +020017#include "pycore_object.h" // _PyDebug_PrintTotalRefs()
18#include "pycore_pyerrors.h" // _PyErr_Fetch
19#include "pycore_pylifecycle.h" // _Py_UnhandledKeyboardInterrupt
Victor Stinnere5014be2020-04-14 17:52:15 +020020#include "pycore_pystate.h" // _PyInterpreterState_GET()
Victor Stinner4f98f462020-04-15 04:01:58 +020021#include "pycore_sysmodule.h" // _PySys_Audit()
Guido van Rossum1984f1e1992-08-04 12:41:02 +000022
Victor Stinner4f98f462020-04-15 04:01:58 +020023#include "node.h" // node
24#include "token.h" // INDENT
25#include "parsetok.h" // perrdetail
26#include "errcode.h" // E_EOF
27#include "code.h" // PyCodeObject
28#include "symtable.h" // PySymtable_BuildObject()
29#include "ast.h" // PyAST_FromNodeObject()
30#include "marshal.h" // PyMarshal_ReadLongFromFile()
31
Lysandros Nikolaouebebb642020-04-23 18:36:06 +030032#include "pegen_interface.h" // PyPegen_ASTFrom*
Pablo Galindoc5fc1562020-04-22 23:29:27 +010033
Victor Stinner4f98f462020-04-15 04:01:58 +020034#ifdef MS_WINDOWS
35# include "malloc.h" // alloca()
Thomas Wouters0e3f5912006-08-11 14:57:12 +000036#endif
Guido van Rossuma9e7dc11992-10-18 18:53:57 +000037
Benjamin Peterson80a50ac2009-01-02 21:24:04 +000038#ifdef MS_WINDOWS
Victor Stinner4f98f462020-04-15 04:01:58 +020039# undef BYTE
40# include "windows.h"
Benjamin Peterson80a50ac2009-01-02 21:24:04 +000041#endif
Martin v. Löwis5c88d812009-01-02 20:47:48 +000042
Guido van Rossuma44823b1995-03-14 15:01:17 +000043
Victor Stinnerbd303c12013-11-07 23:07:29 +010044_Py_IDENTIFIER(builtins);
Victor Stinner09054372013-11-06 22:41:44 +010045_Py_IDENTIFIER(excepthook);
Victor Stinner3f36a572013-11-12 21:39:02 +010046_Py_IDENTIFIER(flush);
Victor Stinnerbd303c12013-11-07 23:07:29 +010047_Py_IDENTIFIER(last_traceback);
Victor Stinner09054372013-11-06 22:41:44 +010048_Py_IDENTIFIER(last_type);
49_Py_IDENTIFIER(last_value);
Victor Stinnerbd303c12013-11-07 23:07:29 +010050_Py_IDENTIFIER(ps1);
51_Py_IDENTIFIER(ps2);
52_Py_IDENTIFIER(stdin);
53_Py_IDENTIFIER(stdout);
54_Py_IDENTIFIER(stderr);
Victor Stinnerefa7a0e2013-11-07 12:37:56 +010055_Py_static_string(PyId_string, "<string>");
Victor Stinner09054372013-11-06 22:41:44 +010056
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000057#ifdef __cplusplus
58extern "C" {
Neal Norwitz4281cef2006-03-04 19:58:13 +000059#endif
60
Vinay Sajip0b60f642019-10-15 08:26:12 +010061extern Py_EXPORTED_SYMBOL grammar _PyParser_Grammar; /* From graminit.c */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000062
Guido van Rossumb73cc041993-11-01 16:28:59 +000063/* Forward */
Amaury Forgeot d'Arc7fedbe52008-04-10 21:03:09 +000064static void flush_io(void);
Victor Stinner95701bd2013-11-06 18:41:07 +010065static PyObject *run_mod(mod_ty, PyObject *, PyObject *, PyObject *,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000066 PyCompilerFlags *, PyArena *);
Martin v. Löwis95292d62002-12-11 14:04:59 +000067static PyObject *run_pyc_file(FILE *, const char *, PyObject *, PyObject *,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000068 PyCompilerFlags *);
Tim Petersdbd9ba62000-07-09 03:09:57 +000069static void err_input(perrdetail *);
Victor Stinner7f2fee32011-04-05 00:39:01 +020070static void err_free(perrdetail *);
xdegayee0582a32017-11-12 16:50:48 +010071static int PyRun_InteractiveOneObjectEx(FILE *, PyObject *, PyCompilerFlags *);
Guido van Rossumce3a72a2007-10-19 23:16:50 +000072
Guido van Rossum1984f1e1992-08-04 12:41:02 +000073/* Parse input from a file and execute it */
Guido van Rossum1984f1e1992-08-04 12:41:02 +000074int
Thomas Wouters4d70c3d2006-06-08 14:42:34 +000075PyRun_AnyFileExFlags(FILE *fp, const char *filename, int closeit,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000076 PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +000077{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000078 if (filename == NULL)
79 filename = "???";
80 if (Py_FdIsInteractive(fp, filename)) {
81 int err = PyRun_InteractiveLoopFlags(fp, filename, flags);
82 if (closeit)
83 fclose(fp);
84 return err;
85 }
86 else
87 return PyRun_SimpleFileExFlags(fp, filename, closeit, flags);
Guido van Rossum1984f1e1992-08-04 12:41:02 +000088}
89
90int
Victor Stinner95701bd2013-11-06 18:41:07 +010091PyRun_InteractiveLoopFlags(FILE *fp, const char *filename_str, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +000092{
Victor Stinner95701bd2013-11-06 18:41:07 +010093 PyObject *filename, *v;
94 int ret, err;
Victor Stinner37d66d72019-06-13 02:16:41 +020095 PyCompilerFlags local_flags = _PyCompilerFlags_INIT;
xdegayee0582a32017-11-12 16:50:48 +010096 int nomem_count = 0;
Victor Stinner25420fe2017-11-20 18:12:22 -080097#ifdef Py_REF_DEBUG
Victor Stinnerda7933e2020-04-13 03:04:28 +020098 int show_ref_count = _Py_GetConfig()->show_ref_count;
Victor Stinner25420fe2017-11-20 18:12:22 -080099#endif
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000100
Victor Stinner95701bd2013-11-06 18:41:07 +0100101 filename = PyUnicode_DecodeFSDefault(filename_str);
102 if (filename == NULL) {
103 PyErr_Print();
104 return -1;
105 }
106
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000107 if (flags == NULL) {
108 flags = &local_flags;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000109 }
Victor Stinner09054372013-11-06 22:41:44 +0100110 v = _PySys_GetObjectId(&PyId_ps1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000111 if (v == NULL) {
Victor Stinner09054372013-11-06 22:41:44 +0100112 _PySys_SetObjectId(&PyId_ps1, v = PyUnicode_FromString(">>> "));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000113 Py_XDECREF(v);
114 }
Victor Stinner09054372013-11-06 22:41:44 +0100115 v = _PySys_GetObjectId(&PyId_ps2);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000116 if (v == NULL) {
Victor Stinner09054372013-11-06 22:41:44 +0100117 _PySys_SetObjectId(&PyId_ps2, v = PyUnicode_FromString("... "));
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000118 Py_XDECREF(v);
119 }
xdegayee0582a32017-11-12 16:50:48 +0100120 err = 0;
121 do {
122 ret = PyRun_InteractiveOneObjectEx(fp, filename, flags);
123 if (ret == -1 && PyErr_Occurred()) {
124 /* Prevent an endless loop after multiple consecutive MemoryErrors
125 * while still allowing an interactive command to fail with a
126 * MemoryError. */
127 if (PyErr_ExceptionMatches(PyExc_MemoryError)) {
128 if (++nomem_count > 16) {
129 PyErr_Clear();
130 err = -1;
131 break;
132 }
133 } else {
134 nomem_count = 0;
135 }
136 PyErr_Print();
137 flush_io();
138 } else {
139 nomem_count = 0;
140 }
Eric Snowdae02762017-09-14 00:35:58 -0700141#ifdef Py_REF_DEBUG
Victor Stinner25420fe2017-11-20 18:12:22 -0800142 if (show_ref_count) {
Eric Snowdae02762017-09-14 00:35:58 -0700143 _PyDebug_PrintTotalRefs();
Victor Stinner25420fe2017-11-20 18:12:22 -0800144 }
Eric Snowdae02762017-09-14 00:35:58 -0700145#endif
xdegayee0582a32017-11-12 16:50:48 +0100146 } while (ret != E_EOF);
Victor Stinner95701bd2013-11-06 18:41:07 +0100147 Py_DECREF(filename);
148 return err;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000149}
150
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000151/* compute parser flags based on compiler flags */
Benjamin Petersonf5b52242009-03-02 23:31:26 +0000152static int PARSER_FLAGS(PyCompilerFlags *flags)
153{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000154 int parser_flags = 0;
155 if (!flags)
156 return 0;
157 if (flags->cf_flags & PyCF_DONT_IMPLY_DEDENT)
158 parser_flags |= PyPARSE_DONT_IMPLY_DEDENT;
159 if (flags->cf_flags & PyCF_IGNORE_COOKIE)
160 parser_flags |= PyPARSE_IGNORE_COOKIE;
161 if (flags->cf_flags & CO_FUTURE_BARRY_AS_BDFL)
162 parser_flags |= PyPARSE_BARRY_AS_BDFL;
Guido van Rossumdcfcd142019-01-31 03:40:27 -0800163 if (flags->cf_flags & PyCF_TYPE_COMMENTS)
164 parser_flags |= PyPARSE_TYPE_COMMENTS;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000165 return parser_flags;
Benjamin Petersonf5b52242009-03-02 23:31:26 +0000166}
Neil Schemenauerc24ea082002-03-22 23:53:36 +0000167
Thomas Wouters89f507f2006-12-13 04:49:30 +0000168#if 0
169/* Keep an example of flags with future keyword support. */
170#define PARSER_FLAGS(flags) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000171 ((flags) ? ((((flags)->cf_flags & PyCF_DONT_IMPLY_DEDENT) ? \
172 PyPARSE_DONT_IMPLY_DEDENT : 0) \
173 | ((flags)->cf_flags & CO_FUTURE_WITH_STATEMENT ? \
174 PyPARSE_WITH_IS_KEYWORD : 0)) : 0)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000175#endif
176
xdegayee0582a32017-11-12 16:50:48 +0100177/* A PyRun_InteractiveOneObject() auxiliary function that does not print the
178 * error on failure. */
179static int
180PyRun_InteractiveOneObjectEx(FILE *fp, PyObject *filename,
181 PyCompilerFlags *flags)
Jeremy Hylton9f324e92001-03-01 22:59:14 +0000182{
Victor Stinner95701bd2013-11-06 18:41:07 +0100183 PyObject *m, *d, *v, *w, *oenc = NULL, *mod_name;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000184 mod_ty mod;
185 PyArena *arena;
Serhiy Storchaka85b0f5b2016-11-20 10:16:47 +0200186 const char *ps1 = "", *ps2 = "", *enc = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000187 int errcode = 0;
Victor Stinner1def7752020-04-23 03:03:24 +0200188 int use_peg = _PyInterpreterState_GET()->config._use_peg_parser;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200189 _Py_IDENTIFIER(encoding);
Victor Stinner95701bd2013-11-06 18:41:07 +0100190 _Py_IDENTIFIER(__main__);
191
192 mod_name = _PyUnicode_FromId(&PyId___main__); /* borrowed */
193 if (mod_name == NULL) {
Victor Stinner95701bd2013-11-06 18:41:07 +0100194 return -1;
195 }
Tim Petersfe2127d2001-07-16 05:37:24 +0000196
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000197 if (fp == stdin) {
Benjamin Petersonfe1b22a2013-04-29 10:23:08 -0400198 /* Fetch encoding from sys.stdin if possible. */
Victor Stinnerbd303c12013-11-07 23:07:29 +0100199 v = _PySys_GetObjectId(&PyId_stdin);
Benjamin Petersonfe1b22a2013-04-29 10:23:08 -0400200 if (v && v != Py_None) {
201 oenc = _PyObject_GetAttrId(v, &PyId_encoding);
202 if (oenc)
Serhiy Storchaka06515832016-11-20 09:13:07 +0200203 enc = PyUnicode_AsUTF8(oenc);
Benjamin Petersonfe1b22a2013-04-29 10:23:08 -0400204 if (!enc)
205 PyErr_Clear();
206 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000207 }
Victor Stinner09054372013-11-06 22:41:44 +0100208 v = _PySys_GetObjectId(&PyId_ps1);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000209 if (v != NULL) {
210 v = PyObject_Str(v);
211 if (v == NULL)
212 PyErr_Clear();
Victor Stinner386fe712010-05-19 00:34:15 +0000213 else if (PyUnicode_Check(v)) {
Serhiy Storchaka06515832016-11-20 09:13:07 +0200214 ps1 = PyUnicode_AsUTF8(v);
Victor Stinner386fe712010-05-19 00:34:15 +0000215 if (ps1 == NULL) {
216 PyErr_Clear();
217 ps1 = "";
218 }
219 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000220 }
Victor Stinner09054372013-11-06 22:41:44 +0100221 w = _PySys_GetObjectId(&PyId_ps2);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000222 if (w != NULL) {
223 w = PyObject_Str(w);
224 if (w == NULL)
225 PyErr_Clear();
Victor Stinner386fe712010-05-19 00:34:15 +0000226 else if (PyUnicode_Check(w)) {
Serhiy Storchaka06515832016-11-20 09:13:07 +0200227 ps2 = PyUnicode_AsUTF8(w);
Victor Stinner386fe712010-05-19 00:34:15 +0000228 if (ps2 == NULL) {
229 PyErr_Clear();
230 ps2 = "";
231 }
232 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000233 }
234 arena = PyArena_New();
235 if (arena == NULL) {
236 Py_XDECREF(v);
237 Py_XDECREF(w);
238 Py_XDECREF(oenc);
239 return -1;
240 }
Pablo Galindoc5fc1562020-04-22 23:29:27 +0100241
242 if (use_peg) {
243 mod = PyPegen_ASTFromFileObject(fp, filename, Py_single_input,
Pablo Galindo2b74c832020-04-27 18:02:07 +0100244 enc, ps1, ps2, flags, &errcode, arena);
Pablo Galindoc5fc1562020-04-22 23:29:27 +0100245 }
246 else {
247 mod = PyParser_ASTFromFileObject(fp, filename, enc,
248 Py_single_input, ps1, ps2,
249 flags, &errcode, arena);
250 }
251
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000252 Py_XDECREF(v);
253 Py_XDECREF(w);
254 Py_XDECREF(oenc);
255 if (mod == NULL) {
256 PyArena_Free(arena);
257 if (errcode == E_EOF) {
258 PyErr_Clear();
259 return E_EOF;
260 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000261 return -1;
262 }
Victor Stinner95701bd2013-11-06 18:41:07 +0100263 m = PyImport_AddModuleObject(mod_name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000264 if (m == NULL) {
265 PyArena_Free(arena);
266 return -1;
267 }
268 d = PyModule_GetDict(m);
269 v = run_mod(mod, filename, d, d, flags, arena);
270 PyArena_Free(arena);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000271 if (v == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000272 return -1;
273 }
274 Py_DECREF(v);
Antoine Pitrou9845c7e2014-05-11 13:42:17 +0200275 flush_io();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000276 return 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000277}
278
Victor Stinner95701bd2013-11-06 18:41:07 +0100279int
xdegayee0582a32017-11-12 16:50:48 +0100280PyRun_InteractiveOneObject(FILE *fp, PyObject *filename, PyCompilerFlags *flags)
281{
282 int res;
283
284 res = PyRun_InteractiveOneObjectEx(fp, filename, flags);
285 if (res == -1) {
286 PyErr_Print();
287 flush_io();
288 }
289 return res;
290}
291
292int
Victor Stinner95701bd2013-11-06 18:41:07 +0100293PyRun_InteractiveOneFlags(FILE *fp, const char *filename_str, PyCompilerFlags *flags)
294{
295 PyObject *filename;
296 int res;
297
298 filename = PyUnicode_DecodeFSDefault(filename_str);
299 if (filename == NULL) {
300 PyErr_Print();
301 return -1;
302 }
303 res = PyRun_InteractiveOneObject(fp, filename, flags);
304 Py_DECREF(filename);
305 return res;
306}
307
308
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000309/* Check whether a file maybe a pyc file: Look at the extension,
310 the file type, and, if we may close it, at the first few bytes. */
311
312static int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000313maybe_pyc_file(FILE *fp, const char* filename, const char* ext, int closeit)
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000314{
Brett Cannonf299abd2015-04-13 14:21:02 -0400315 if (strcmp(ext, ".pyc") == 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000316 return 1;
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000317
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000318 /* Only look into the file if we are allowed to close it, since
319 it then should also be seekable. */
320 if (closeit) {
321 /* Read only two bytes of the magic. If the file was opened in
322 text mode, the bytes 3 and 4 of the magic (\r\n) might not
323 be read as they are on disk. */
324 unsigned int halfmagic = PyImport_GetMagicNumber() & 0xFFFF;
325 unsigned char buf[2];
326 /* Mess: In case of -x, the stream is NOT at its start now,
327 and ungetc() was used to push back the first newline,
328 which makes the current stream position formally undefined,
329 and a x-platform nightmare.
330 Unfortunately, we have no direct way to know whether -x
331 was specified. So we use a terrible hack: if the current
332 stream position is not 0, we assume -x was specified, and
333 give up. Bug 132850 on SourceForge spells out the
334 hopelessness of trying anything else (fseek and ftell
335 don't work predictably x-platform for text-mode files).
336 */
337 int ispyc = 0;
338 if (ftell(fp) == 0) {
339 if (fread(buf, 1, 2, fp) == 2 &&
340 ((unsigned int)buf[1]<<8 | buf[0]) == halfmagic)
341 ispyc = 1;
342 rewind(fp);
343 }
344 return ispyc;
345 }
346 return 0;
Tim Petersd08e3822003-04-17 15:24:21 +0000347}
Martin v. Löwisbe4c0f52001-01-04 20:30:56 +0000348
Antoine Pitrou32d483c2013-07-30 21:01:23 +0200349static int
350set_main_loader(PyObject *d, const char *filename, const char *loader_name)
Nick Coghlan85e729e2012-07-15 18:09:52 +1000351{
Eric Snow32439d62015-05-02 19:15:18 -0600352 PyObject *filename_obj, *bootstrap, *loader_type = NULL, *loader;
Nick Coghlanb7a58942012-07-15 23:21:08 +1000353 int result = 0;
Andrew Svetlov90c0eb22012-11-01 14:51:14 +0200354
355 filename_obj = PyUnicode_DecodeFSDefault(filename);
356 if (filename_obj == NULL)
357 return -1;
Victor Stinner81a7be32020-04-14 15:14:01 +0200358 PyInterpreterState *interp = _PyInterpreterState_GET();
Eric Snow32439d62015-05-02 19:15:18 -0600359 bootstrap = PyObject_GetAttrString(interp->importlib,
360 "_bootstrap_external");
361 if (bootstrap != NULL) {
362 loader_type = PyObject_GetAttrString(bootstrap, loader_name);
363 Py_DECREF(bootstrap);
364 }
Nick Coghlan3f94cbf2012-07-15 19:10:39 +1000365 if (loader_type == NULL) {
Andrew Svetlov90c0eb22012-11-01 14:51:14 +0200366 Py_DECREF(filename_obj);
Nick Coghlan3f94cbf2012-07-15 19:10:39 +1000367 return -1;
368 }
Andrew Svetlov90c0eb22012-11-01 14:51:14 +0200369 loader = PyObject_CallFunction(loader_type, "sN", "__main__", filename_obj);
Nick Coghlanb7a58942012-07-15 23:21:08 +1000370 Py_DECREF(loader_type);
371 if (loader == NULL) {
Nick Coghlan85e729e2012-07-15 18:09:52 +1000372 return -1;
373 }
Nick Coghlanb7a58942012-07-15 23:21:08 +1000374 if (PyDict_SetItemString(d, "__loader__", loader) < 0) {
375 result = -1;
376 }
Nick Coghlan85e729e2012-07-15 18:09:52 +1000377 Py_DECREF(loader);
Nick Coghlanb7a58942012-07-15 23:21:08 +1000378 return result;
Nick Coghlan85e729e2012-07-15 18:09:52 +1000379}
380
381int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000382PyRun_SimpleFileExFlags(FILE *fp, const char *filename, int closeit,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000383 PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +0000384{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000385 PyObject *m, *d, *v;
386 const char *ext;
Hynek Schlawack5c6b3e22012-11-07 09:02:24 +0100387 int set_file_name = 0, ret = -1;
Victor Stinner0fcab4a2011-01-04 12:59:15 +0000388 size_t len;
Guido van Rossumfdef2711994-09-14 13:31:04 +0000389
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000390 m = PyImport_AddModule("__main__");
391 if (m == NULL)
392 return -1;
Hynek Schlawack5c6b3e22012-11-07 09:02:24 +0100393 Py_INCREF(m);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000394 d = PyModule_GetDict(m);
395 if (PyDict_GetItemString(d, "__file__") == NULL) {
396 PyObject *f;
Victor Stinner4c7c8c32010-10-16 13:14:10 +0000397 f = PyUnicode_DecodeFSDefault(filename);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000398 if (f == NULL)
Hynek Schlawack5c6b3e22012-11-07 09:02:24 +0100399 goto done;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000400 if (PyDict_SetItemString(d, "__file__", f) < 0) {
401 Py_DECREF(f);
Hynek Schlawack5c6b3e22012-11-07 09:02:24 +0100402 goto done;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000403 }
Barry Warsaw916048d2011-09-20 14:45:44 -0400404 if (PyDict_SetItemString(d, "__cached__", Py_None) < 0) {
405 Py_DECREF(f);
Hynek Schlawack5c6b3e22012-11-07 09:02:24 +0100406 goto done;
Barry Warsaw916048d2011-09-20 14:45:44 -0400407 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000408 set_file_name = 1;
409 Py_DECREF(f);
410 }
411 len = strlen(filename);
412 ext = filename + len - (len > 4 ? 4 : 0);
413 if (maybe_pyc_file(fp, filename, ext, closeit)) {
Christian Heimes04ac4c12012-09-11 15:47:28 +0200414 FILE *pyc_fp;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000415 /* Try to run a pyc file. First, re-open in binary */
416 if (closeit)
417 fclose(fp);
Victor Stinnerdaf45552013-08-28 00:53:59 +0200418 if ((pyc_fp = _Py_fopen(filename, "rb")) == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000419 fprintf(stderr, "python: Can't reopen .pyc file\n");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000420 goto done;
421 }
Nick Coghlan85e729e2012-07-15 18:09:52 +1000422
423 if (set_main_loader(d, filename, "SourcelessFileLoader") < 0) {
424 fprintf(stderr, "python: failed to set __main__.__loader__\n");
425 ret = -1;
Christian Heimes04ac4c12012-09-11 15:47:28 +0200426 fclose(pyc_fp);
Nick Coghlan85e729e2012-07-15 18:09:52 +1000427 goto done;
428 }
Christian Heimes04ac4c12012-09-11 15:47:28 +0200429 v = run_pyc_file(pyc_fp, filename, d, d, flags);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000430 } else {
Nick Coghlan85e729e2012-07-15 18:09:52 +1000431 /* When running from stdin, leave __main__.__loader__ alone */
432 if (strcmp(filename, "<stdin>") != 0 &&
433 set_main_loader(d, filename, "SourceFileLoader") < 0) {
434 fprintf(stderr, "python: failed to set __main__.__loader__\n");
435 ret = -1;
436 goto done;
437 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000438 v = PyRun_FileExFlags(fp, filename, Py_file_input, d, d,
439 closeit, flags);
440 }
441 flush_io();
442 if (v == NULL) {
Zackery Spytzd8cba5d2018-07-03 13:47:22 -0600443 Py_CLEAR(m);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000444 PyErr_Print();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000445 goto done;
446 }
447 Py_DECREF(v);
448 ret = 0;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000449 done:
INADA Naoki82daa602018-11-29 20:01:27 +0900450 if (set_file_name) {
451 if (PyDict_DelItemString(d, "__file__")) {
452 PyErr_Clear();
453 }
454 if (PyDict_DelItemString(d, "__cached__")) {
455 PyErr_Clear();
456 }
457 }
Zackery Spytzd8cba5d2018-07-03 13:47:22 -0600458 Py_XDECREF(m);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000459 return ret;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000460}
461
462int
Martin v. Löwis95292d62002-12-11 14:04:59 +0000463PyRun_SimpleStringFlags(const char *command, PyCompilerFlags *flags)
Guido van Rossum393661d2001-08-31 17:40:15 +0000464{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000465 PyObject *m, *d, *v;
466 m = PyImport_AddModule("__main__");
467 if (m == NULL)
468 return -1;
469 d = PyModule_GetDict(m);
470 v = PyRun_StringFlags(command, Py_file_input, d, d, flags);
471 if (v == NULL) {
472 PyErr_Print();
473 return -1;
474 }
475 Py_DECREF(v);
476 return 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000477}
478
Barry Warsaw035574d1997-08-29 22:07:17 +0000479static int
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100480parse_syntax_error(PyObject *err, PyObject **message, PyObject **filename,
Ammar Askar20fe5322020-06-03 07:34:55 +0000481 Py_ssize_t *lineno, Py_ssize_t *offset, PyObject **text)
Barry Warsaw035574d1997-08-29 22:07:17 +0000482{
Ammar Askar20fe5322020-06-03 07:34:55 +0000483 Py_ssize_t hold;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000484 PyObject *v;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200485 _Py_IDENTIFIER(msg);
486 _Py_IDENTIFIER(filename);
487 _Py_IDENTIFIER(lineno);
488 _Py_IDENTIFIER(offset);
489 _Py_IDENTIFIER(text);
Barry Warsaw035574d1997-08-29 22:07:17 +0000490
Benjamin Peterson80d50422012-04-03 00:30:38 -0400491 *message = NULL;
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100492 *filename = NULL;
Benjamin Peterson80d50422012-04-03 00:30:38 -0400493
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000494 /* new style errors. `err' is an instance */
Benjamin Peterson0a9a6362012-04-03 00:35:36 -0400495 *message = _PyObject_GetAttrId(err, &PyId_msg);
Benjamin Peterson80d50422012-04-03 00:30:38 -0400496 if (!*message)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000497 goto finally;
Barry Warsaw035574d1997-08-29 22:07:17 +0000498
Benjamin Peterson0a9a6362012-04-03 00:35:36 -0400499 v = _PyObject_GetAttrId(err, &PyId_filename);
Benjamin Peterson80d50422012-04-03 00:30:38 -0400500 if (!v)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000501 goto finally;
Benjamin Peterson80d50422012-04-03 00:30:38 -0400502 if (v == Py_None) {
503 Py_DECREF(v);
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100504 *filename = _PyUnicode_FromId(&PyId_string);
505 if (*filename == NULL)
506 goto finally;
507 Py_INCREF(*filename);
Benjamin Peterson80d50422012-04-03 00:30:38 -0400508 }
509 else {
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100510 *filename = v;
Benjamin Peterson80d50422012-04-03 00:30:38 -0400511 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000512
Benjamin Peterson0a9a6362012-04-03 00:35:36 -0400513 v = _PyObject_GetAttrId(err, &PyId_lineno);
Benjamin Peterson80d50422012-04-03 00:30:38 -0400514 if (!v)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000515 goto finally;
Ammar Askar20fe5322020-06-03 07:34:55 +0000516 hold = PyLong_AsSsize_t(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000517 Py_DECREF(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000518 if (hold < 0 && PyErr_Occurred())
519 goto finally;
Serhiy Storchaka56f6e762015-09-06 21:25:30 +0300520 *lineno = hold;
Barry Warsaw035574d1997-08-29 22:07:17 +0000521
Benjamin Peterson0a9a6362012-04-03 00:35:36 -0400522 v = _PyObject_GetAttrId(err, &PyId_offset);
Benjamin Peterson80d50422012-04-03 00:30:38 -0400523 if (!v)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000524 goto finally;
525 if (v == Py_None) {
526 *offset = -1;
527 Py_DECREF(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000528 } else {
Ammar Askar20fe5322020-06-03 07:34:55 +0000529 hold = PyLong_AsSsize_t(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000530 Py_DECREF(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000531 if (hold < 0 && PyErr_Occurred())
532 goto finally;
Serhiy Storchaka56f6e762015-09-06 21:25:30 +0300533 *offset = hold;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000534 }
Barry Warsaw035574d1997-08-29 22:07:17 +0000535
Benjamin Peterson0a9a6362012-04-03 00:35:36 -0400536 v = _PyObject_GetAttrId(err, &PyId_text);
Benjamin Peterson80d50422012-04-03 00:30:38 -0400537 if (!v)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000538 goto finally;
Benjamin Peterson80d50422012-04-03 00:30:38 -0400539 if (v == Py_None) {
540 Py_DECREF(v);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000541 *text = NULL;
Benjamin Peterson80d50422012-04-03 00:30:38 -0400542 }
543 else {
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100544 *text = v;
Benjamin Peterson80d50422012-04-03 00:30:38 -0400545 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000546 return 1;
Barry Warsaw035574d1997-08-29 22:07:17 +0000547
548finally:
Benjamin Peterson80d50422012-04-03 00:30:38 -0400549 Py_XDECREF(*message);
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100550 Py_XDECREF(*filename);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000551 return 0;
Barry Warsaw035574d1997-08-29 22:07:17 +0000552}
553
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000554static void
Ammar Askar20fe5322020-06-03 07:34:55 +0000555print_error_text(PyObject *f, Py_ssize_t offset, PyObject *text_obj)
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000556{
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700557 /* Convert text to a char pointer; return if error */
558 const char *text = PyUnicode_AsUTF8(text_obj);
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100559 if (text == NULL)
560 return;
561
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700562 /* Convert offset from 1-based to 0-based */
563 offset--;
564
565 /* Strip leading whitespace from text, adjusting offset as we go */
566 while (*text == ' ' || *text == '\t' || *text == '\f') {
567 text++;
568 offset--;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000569 }
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700570
571 /* Calculate text length excluding trailing newline */
572 Py_ssize_t len = strlen(text);
573 if (len > 0 && text[len-1] == '\n') {
574 len--;
575 }
576
577 /* Clip offset to at most len */
578 if (offset > len) {
579 offset = len;
580 }
581
582 /* Skip past newlines embedded in text */
583 for (;;) {
584 const char *nl = strchr(text, '\n');
585 if (nl == NULL) {
586 break;
587 }
588 Py_ssize_t inl = nl - text;
Ammar Askar20fe5322020-06-03 07:34:55 +0000589 if (inl >= offset) {
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700590 break;
591 }
592 inl += 1;
593 text += inl;
594 len -= inl;
595 offset -= (int)inl;
596 }
597
598 /* Print text */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000599 PyFile_WriteString(" ", f);
600 PyFile_WriteString(text, f);
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700601
602 /* Make sure there's a newline at the end */
603 if (text[len] != '\n') {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000604 PyFile_WriteString("\n", f);
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700605 }
606
607 /* Don't print caret if it points to the left of the text */
608 if (offset < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000609 return;
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700610
611 /* Write caret line */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000612 PyFile_WriteString(" ", f);
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700613 while (--offset >= 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000614 PyFile_WriteString(" ", f);
Guido van Rossum15bc9ab2020-05-14 19:22:48 -0700615 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000616 PyFile_WriteString("^\n", f);
Jeremy Hylton9f1b9932001-02-28 07:07:43 +0000617}
618
Tim Peterscf615b52003-04-19 18:47:02 +0000619
Victor Stinner12083282019-05-17 23:05:29 +0200620int
621_Py_HandleSystemExit(int *exitcode_p)
622{
Victor Stinnerda7933e2020-04-13 03:04:28 +0200623 int inspect = _Py_GetConfig()->inspect;
Victor Stinnerc96be812019-05-14 17:34:56 +0200624 if (inspect) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000625 /* Don't exit if -i flag was given. This flag is set to 0
626 * when entering interactive mode for inspecting. */
Victor Stinner12083282019-05-17 23:05:29 +0200627 return 0;
Victor Stinnerc96be812019-05-14 17:34:56 +0200628 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000629
Victor Stinner12083282019-05-17 23:05:29 +0200630 if (!PyErr_ExceptionMatches(PyExc_SystemExit)) {
631 return 0;
632 }
633
634 PyObject *exception, *value, *tb;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000635 PyErr_Fetch(&exception, &value, &tb);
Victor Stinner12083282019-05-17 23:05:29 +0200636
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000637 fflush(stdout);
Victor Stinner12083282019-05-17 23:05:29 +0200638
639 int exitcode = 0;
640 if (value == NULL || value == Py_None) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000641 goto done;
Victor Stinner12083282019-05-17 23:05:29 +0200642 }
643
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000644 if (PyExceptionInstance_Check(value)) {
645 /* The error code should be in the `code' attribute. */
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200646 _Py_IDENTIFIER(code);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200647 PyObject *code = _PyObject_GetAttrId(value, &PyId_code);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000648 if (code) {
649 Py_DECREF(value);
650 value = code;
651 if (value == Py_None)
652 goto done;
653 }
654 /* If we failed to dig out the 'code' attribute,
655 just let the else clause below print the error. */
656 }
Victor Stinner12083282019-05-17 23:05:29 +0200657
658 if (PyLong_Check(value)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000659 exitcode = (int)PyLong_AsLong(value);
Victor Stinner12083282019-05-17 23:05:29 +0200660 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000661 else {
Victor Stinnerbd303c12013-11-07 23:07:29 +0100662 PyObject *sys_stderr = _PySys_GetObjectId(&PyId_stderr);
Nick Coghland979e432014-02-09 10:43:21 +1000663 /* We clear the exception here to avoid triggering the assertion
664 * in PyObject_Str that ensures it won't silently lose exception
665 * details.
666 */
667 PyErr_Clear();
Victor Stinner7126dbc2010-05-21 23:45:42 +0000668 if (sys_stderr != NULL && sys_stderr != Py_None) {
669 PyFile_WriteObject(value, sys_stderr, Py_PRINT_RAW);
670 } else {
671 PyObject_Print(value, stderr, Py_PRINT_RAW);
672 fflush(stderr);
673 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000674 PySys_WriteStderr("\n");
675 exitcode = 1;
676 }
Victor Stinner12083282019-05-17 23:05:29 +0200677
Tim Peterscf615b52003-04-19 18:47:02 +0000678 done:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000679 /* Restore and clear the exception info, in order to properly decref
680 * the exception, value, and traceback. If we just exit instead,
681 * these leak, which confuses PYTHONDUMPREFS output, and may prevent
682 * some finalizers from running.
683 */
684 PyErr_Restore(exception, value, tb);
685 PyErr_Clear();
Victor Stinner12083282019-05-17 23:05:29 +0200686 *exitcode_p = exitcode;
687 return 1;
Ka-Ping Yee26fabb02001-03-23 15:36:41 +0000688}
689
Victor Stinner12083282019-05-17 23:05:29 +0200690
691static void
692handle_system_exit(void)
693{
694 int exitcode;
695 if (_Py_HandleSystemExit(&exitcode)) {
696 Py_Exit(exitcode);
697 }
698}
699
700
Victor Stinner438a12d2019-05-24 17:01:38 +0200701static void
702_PyErr_PrintEx(PyThreadState *tstate, int set_sys_last_vars)
Guido van Rossuma61691e1998-02-06 22:27:24 +0000703{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000704 PyObject *exception, *v, *tb, *hook;
Guido van Rossum66e8e862001-03-23 17:54:43 +0000705
Victor Stinner12083282019-05-17 23:05:29 +0200706 handle_system_exit();
707
Victor Stinner438a12d2019-05-24 17:01:38 +0200708 _PyErr_Fetch(tstate, &exception, &v, &tb);
709 if (exception == NULL) {
710 goto done;
711 }
712
713 _PyErr_NormalizeException(tstate, &exception, &v, &tb);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000714 if (tb == NULL) {
715 tb = Py_None;
716 Py_INCREF(tb);
717 }
718 PyException_SetTraceback(v, tb);
Victor Stinner438a12d2019-05-24 17:01:38 +0200719 if (exception == NULL) {
720 goto done;
721 }
722
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000723 /* Now we know v != NULL too */
724 if (set_sys_last_vars) {
xdegaye66caacf2017-10-23 18:08:41 +0200725 if (_PySys_SetObjectId(&PyId_last_type, exception) < 0) {
Victor Stinner438a12d2019-05-24 17:01:38 +0200726 _PyErr_Clear(tstate);
xdegaye66caacf2017-10-23 18:08:41 +0200727 }
728 if (_PySys_SetObjectId(&PyId_last_value, v) < 0) {
Victor Stinner438a12d2019-05-24 17:01:38 +0200729 _PyErr_Clear(tstate);
xdegaye66caacf2017-10-23 18:08:41 +0200730 }
731 if (_PySys_SetObjectId(&PyId_last_traceback, tb) < 0) {
Victor Stinner438a12d2019-05-24 17:01:38 +0200732 _PyErr_Clear(tstate);
xdegaye66caacf2017-10-23 18:08:41 +0200733 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000734 }
Victor Stinner09054372013-11-06 22:41:44 +0100735 hook = _PySys_GetObjectId(&PyId_excepthook);
Victor Stinner1c1e68c2020-03-27 15:11:45 +0100736 if (_PySys_Audit(tstate, "sys.excepthook", "OOOO", hook ? hook : Py_None,
737 exception, v, tb) < 0) {
Steve Dowerbea33f52019-11-28 08:46:11 -0800738 if (PyErr_ExceptionMatches(PyExc_RuntimeError)) {
739 PyErr_Clear();
740 goto done;
741 }
742 _PyErr_WriteUnraisableMsg("in audit hook", NULL);
743 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000744 if (hook) {
Victor Stinner71cb64a2016-08-20 00:57:43 +0200745 PyObject* stack[3];
746 PyObject *result;
747
748 stack[0] = exception;
749 stack[1] = v;
750 stack[2] = tb;
Victor Stinner559bb6a2016-08-22 22:48:54 +0200751 result = _PyObject_FastCall(hook, stack, 3);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000752 if (result == NULL) {
Victor Stinner12083282019-05-17 23:05:29 +0200753 handle_system_exit();
754
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000755 PyObject *exception2, *v2, *tb2;
Victor Stinner438a12d2019-05-24 17:01:38 +0200756 _PyErr_Fetch(tstate, &exception2, &v2, &tb2);
757 _PyErr_NormalizeException(tstate, &exception2, &v2, &tb2);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000758 /* It should not be possible for exception2 or v2
759 to be NULL. However PyErr_Display() can't
760 tolerate NULLs, so just be safe. */
761 if (exception2 == NULL) {
762 exception2 = Py_None;
763 Py_INCREF(exception2);
764 }
765 if (v2 == NULL) {
766 v2 = Py_None;
767 Py_INCREF(v2);
768 }
769 fflush(stdout);
770 PySys_WriteStderr("Error in sys.excepthook:\n");
771 PyErr_Display(exception2, v2, tb2);
772 PySys_WriteStderr("\nOriginal exception was:\n");
773 PyErr_Display(exception, v, tb);
774 Py_DECREF(exception2);
775 Py_DECREF(v2);
776 Py_XDECREF(tb2);
777 }
778 Py_XDECREF(result);
Victor Stinner438a12d2019-05-24 17:01:38 +0200779 }
780 else {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000781 PySys_WriteStderr("sys.excepthook is missing\n");
782 PyErr_Display(exception, v, tb);
783 }
Victor Stinner438a12d2019-05-24 17:01:38 +0200784
785done:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000786 Py_XDECREF(exception);
787 Py_XDECREF(v);
788 Py_XDECREF(tb);
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +0000789}
790
Victor Stinner438a12d2019-05-24 17:01:38 +0200791void
792_PyErr_Print(PyThreadState *tstate)
793{
794 _PyErr_PrintEx(tstate, 1);
795}
796
797void
798PyErr_PrintEx(int set_sys_last_vars)
799{
800 PyThreadState *tstate = _PyThreadState_GET();
801 _PyErr_PrintEx(tstate, set_sys_last_vars);
802}
803
804void
805PyErr_Print(void)
806{
807 PyErr_PrintEx(1);
808}
809
Benjamin Petersone6528212008-07-15 15:32:09 +0000810static void
811print_exception(PyObject *f, PyObject *value)
812{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000813 int err = 0;
814 PyObject *type, *tb;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200815 _Py_IDENTIFIER(print_file_and_line);
Benjamin Petersone6528212008-07-15 15:32:09 +0000816
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000817 if (!PyExceptionInstance_Check(value)) {
Victor Stinner52ce3b02013-12-09 02:10:08 +0100818 err = PyFile_WriteString("TypeError: print_exception(): Exception expected for value, ", f);
819 err += PyFile_WriteString(Py_TYPE(value)->tp_name, f);
820 err += PyFile_WriteString(" found\n", f);
821 if (err)
822 PyErr_Clear();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000823 return;
824 }
Benjamin Peterson26582602008-08-23 20:08:07 +0000825
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000826 Py_INCREF(value);
827 fflush(stdout);
828 type = (PyObject *) Py_TYPE(value);
829 tb = PyException_GetTraceback(value);
830 if (tb && tb != Py_None)
831 err = PyTraceBack_Print(tb, f);
832 if (err == 0 &&
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200833 _PyObject_HasAttrId(value, &PyId_print_file_and_line))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000834 {
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100835 PyObject *message, *filename, *text;
Ammar Askar20fe5322020-06-03 07:34:55 +0000836 Py_ssize_t lineno, offset;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000837 if (!parse_syntax_error(value, &message, &filename,
838 &lineno, &offset, &text))
839 PyErr_Clear();
840 else {
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100841 PyObject *line;
842
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000843 Py_DECREF(value);
844 value = message;
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100845
Ammar Askar20fe5322020-06-03 07:34:55 +0000846 line = PyUnicode_FromFormat(" File \"%S\", line %zd\n",
Victor Stinnerefa7a0e2013-11-07 12:37:56 +0100847 filename, lineno);
848 Py_DECREF(filename);
849 if (line != NULL) {
850 PyFile_WriteObject(line, f, Py_PRINT_RAW);
851 Py_DECREF(line);
852 }
853
854 if (text != NULL) {
855 print_error_text(f, offset, text);
856 Py_DECREF(text);
857 }
858
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000859 /* Can't be bothered to check all those
860 PyFile_WriteString() calls */
861 if (PyErr_Occurred())
862 err = -1;
863 }
864 }
865 if (err) {
866 /* Don't do anything else */
867 }
868 else {
869 PyObject* moduleName;
Serhiy Storchakaceeef102018-06-15 11:09:43 +0300870 const char *className;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200871 _Py_IDENTIFIER(__module__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000872 assert(PyExceptionClass_Check(type));
873 className = PyExceptionClass_Name(type);
874 if (className != NULL) {
Serhiy Storchakaceeef102018-06-15 11:09:43 +0300875 const char *dot = strrchr(className, '.');
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000876 if (dot != NULL)
877 className = dot+1;
878 }
Benjamin Petersone6528212008-07-15 15:32:09 +0000879
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200880 moduleName = _PyObject_GetAttrId(type, &PyId___module__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000881 if (moduleName == NULL || !PyUnicode_Check(moduleName))
882 {
Victor Stinner13b21bd2011-05-26 14:25:13 +0200883 Py_XDECREF(moduleName);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000884 err = PyFile_WriteString("<unknown>", f);
885 }
886 else {
Serhiy Storchakaf5894dd2016-11-16 15:40:39 +0200887 if (!_PyUnicode_EqualToASCIIId(moduleName, &PyId_builtins))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000888 {
Victor Stinner937114f2013-11-07 00:12:30 +0100889 err = PyFile_WriteObject(moduleName, f, Py_PRINT_RAW);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000890 err += PyFile_WriteString(".", f);
891 }
892 Py_DECREF(moduleName);
893 }
894 if (err == 0) {
895 if (className == NULL)
896 err = PyFile_WriteString("<unknown>", f);
897 else
898 err = PyFile_WriteString(className, f);
899 }
900 }
901 if (err == 0 && (value != Py_None)) {
902 PyObject *s = PyObject_Str(value);
903 /* only print colon if the str() of the
904 object is not the empty string
905 */
Martin Panter3263f682016-02-28 03:16:11 +0000906 if (s == NULL) {
907 PyErr_Clear();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000908 err = -1;
Martin Panter3263f682016-02-28 03:16:11 +0000909 PyFile_WriteString(": <exception str() failed>", f);
910 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000911 else if (!PyUnicode_Check(s) ||
Victor Stinnere251d6d2011-11-20 19:20:00 +0100912 PyUnicode_GetLength(s) != 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000913 err = PyFile_WriteString(": ", f);
914 if (err == 0)
915 err = PyFile_WriteObject(s, f, Py_PRINT_RAW);
916 Py_XDECREF(s);
917 }
918 /* try to write a newline in any case */
Martin Panter3263f682016-02-28 03:16:11 +0000919 if (err < 0) {
920 PyErr_Clear();
921 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000922 err += PyFile_WriteString("\n", f);
923 Py_XDECREF(tb);
924 Py_DECREF(value);
925 /* If an error happened here, don't show it.
926 XXX This is wrong, but too many callers rely on this behavior. */
927 if (err != 0)
928 PyErr_Clear();
Benjamin Petersone6528212008-07-15 15:32:09 +0000929}
930
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200931static const char cause_message[] =
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000932 "\nThe above exception was the direct cause "
933 "of the following exception:\n\n";
Benjamin Petersone6528212008-07-15 15:32:09 +0000934
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200935static const char context_message[] =
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000936 "\nDuring handling of the above exception, "
937 "another exception occurred:\n\n";
Benjamin Petersone6528212008-07-15 15:32:09 +0000938
939static void
940print_exception_recursive(PyObject *f, PyObject *value, PyObject *seen)
941{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000942 int err = 0, res;
943 PyObject *cause, *context;
Benjamin Petersone6528212008-07-15 15:32:09 +0000944
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000945 if (seen != NULL) {
946 /* Exception chaining */
Zane Bitterde860732017-10-17 17:29:39 -0400947 PyObject *value_id = PyLong_FromVoidPtr(value);
948 if (value_id == NULL || PySet_Add(seen, value_id) == -1)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000949 PyErr_Clear();
950 else if (PyExceptionInstance_Check(value)) {
Zane Bitterde860732017-10-17 17:29:39 -0400951 PyObject *check_id = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000952 cause = PyException_GetCause(value);
953 context = PyException_GetContext(value);
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700954 if (cause) {
Zane Bitterde860732017-10-17 17:29:39 -0400955 check_id = PyLong_FromVoidPtr(cause);
956 if (check_id == NULL) {
957 res = -1;
958 } else {
959 res = PySet_Contains(seen, check_id);
960 Py_DECREF(check_id);
961 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000962 if (res == -1)
963 PyErr_Clear();
964 if (res == 0) {
965 print_exception_recursive(
966 f, cause, seen);
967 err |= PyFile_WriteString(
968 cause_message, f);
969 }
970 }
Benjamin Petersond5a1c442012-05-14 22:09:31 -0700971 else if (context &&
972 !((PyBaseExceptionObject *)value)->suppress_context) {
Zane Bitterde860732017-10-17 17:29:39 -0400973 check_id = PyLong_FromVoidPtr(context);
974 if (check_id == NULL) {
975 res = -1;
976 } else {
977 res = PySet_Contains(seen, check_id);
978 Py_DECREF(check_id);
979 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000980 if (res == -1)
981 PyErr_Clear();
982 if (res == 0) {
983 print_exception_recursive(
984 f, context, seen);
985 err |= PyFile_WriteString(
986 context_message, f);
987 }
988 }
989 Py_XDECREF(context);
990 Py_XDECREF(cause);
991 }
Zane Bitterde860732017-10-17 17:29:39 -0400992 Py_XDECREF(value_id);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000993 }
994 print_exception(f, value);
995 if (err != 0)
996 PyErr_Clear();
Benjamin Petersone6528212008-07-15 15:32:09 +0000997}
998
Thomas Wouters477c8d52006-05-27 19:21:47 +0000999void
Victor Stinnercd590a72019-05-28 00:39:52 +02001000_PyErr_Display(PyObject *file, PyObject *exception, PyObject *value, PyObject *tb)
Ka-Ping Yeeb5c51322001-03-23 02:46:52 +00001001{
Victor Stinnercd590a72019-05-28 00:39:52 +02001002 assert(file != NULL && file != Py_None);
1003
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001004 PyObject *seen;
Antoine Pitrou24201d42013-10-13 21:53:13 +02001005 if (PyExceptionInstance_Check(value)
1006 && tb != NULL && PyTraceBack_Check(tb)) {
1007 /* Put the traceback on the exception, otherwise it won't get
1008 displayed. See issue #18776. */
1009 PyObject *cur_tb = PyException_GetTraceback(value);
1010 if (cur_tb == NULL)
1011 PyException_SetTraceback(value, tb);
1012 else
1013 Py_DECREF(cur_tb);
1014 }
Victor Stinnercd590a72019-05-28 00:39:52 +02001015
1016 /* We choose to ignore seen being possibly NULL, and report
1017 at least the main exception (it could be a MemoryError).
1018 */
1019 seen = PySet_New(NULL);
1020 if (seen == NULL) {
1021 PyErr_Clear();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001022 }
Victor Stinnercd590a72019-05-28 00:39:52 +02001023 print_exception_recursive(file, value, seen);
1024 Py_XDECREF(seen);
Victor Stinnera85a1d32019-05-28 16:01:17 +02001025
1026 /* Call file.flush() */
Jeroen Demeyer762f93f2019-07-08 10:19:25 +02001027 PyObject *res = _PyObject_CallMethodIdNoArgs(file, &PyId_flush);
Victor Stinnera85a1d32019-05-28 16:01:17 +02001028 if (!res) {
1029 /* Silently ignore file.flush() error */
1030 PyErr_Clear();
1031 }
1032 else {
1033 Py_DECREF(res);
1034 }
Victor Stinnercd590a72019-05-28 00:39:52 +02001035}
1036
1037void
1038PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
1039{
1040 PyObject *file = _PySys_GetObjectId(&PyId_stderr);
1041 if (file == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001042 _PyObject_Dump(value);
1043 fprintf(stderr, "lost sys.stderr\n");
Victor Stinnercd590a72019-05-28 00:39:52 +02001044 return;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001045 }
Victor Stinnercd590a72019-05-28 00:39:52 +02001046 if (file == Py_None) {
1047 return;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001048 }
Victor Stinnercd590a72019-05-28 00:39:52 +02001049
1050 _PyErr_Display(file, exception, value, tb);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001051}
1052
Guido van Rossum82598051997-03-05 00:20:32 +00001053PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001054PyRun_StringFlags(const char *str, int start, PyObject *globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001055 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001056{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001057 PyObject *ret = NULL;
1058 mod_ty mod;
Victor Stinner95701bd2013-11-06 18:41:07 +01001059 PyArena *arena;
Victor Stinner95701bd2013-11-06 18:41:07 +01001060 PyObject *filename;
Victor Stinner1def7752020-04-23 03:03:24 +02001061 int use_peg = _PyInterpreterState_GET()->config._use_peg_parser;
Victor Stinner95701bd2013-11-06 18:41:07 +01001062
1063 filename = _PyUnicode_FromId(&PyId_string); /* borrowed */
1064 if (filename == NULL)
1065 return NULL;
1066
1067 arena = PyArena_New();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001068 if (arena == NULL)
1069 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001070
Pablo Galindoc5fc1562020-04-22 23:29:27 +01001071 if (use_peg) {
1072 mod = PyPegen_ASTFromStringObject(str, filename, start, flags, arena);
1073 }
1074 else {
1075 mod = PyParser_ASTFromStringObject(str, filename, start, flags, arena);
1076 }
1077
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001078 if (mod != NULL)
Victor Stinner95701bd2013-11-06 18:41:07 +01001079 ret = run_mod(mod, filename, globals, locals, flags, arena);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001080 PyArena_Free(arena);
1081 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001082}
1083
1084PyObject *
Victor Stinner95701bd2013-11-06 18:41:07 +01001085PyRun_FileExFlags(FILE *fp, const char *filename_str, int start, PyObject *globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001086 PyObject *locals, int closeit, PyCompilerFlags *flags)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001087{
Victor Stinner95701bd2013-11-06 18:41:07 +01001088 PyObject *ret = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001089 mod_ty mod;
Victor Stinner95701bd2013-11-06 18:41:07 +01001090 PyArena *arena = NULL;
1091 PyObject *filename;
Victor Stinner1def7752020-04-23 03:03:24 +02001092 int use_peg = _PyInterpreterState_GET()->config._use_peg_parser;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001093
Victor Stinner95701bd2013-11-06 18:41:07 +01001094 filename = PyUnicode_DecodeFSDefault(filename_str);
1095 if (filename == NULL)
1096 goto exit;
1097
1098 arena = PyArena_New();
1099 if (arena == NULL)
1100 goto exit;
1101
Pablo Galindoc5fc1562020-04-22 23:29:27 +01001102 if (use_peg) {
1103 mod = PyPegen_ASTFromFileObject(fp, filename, start, NULL, NULL, NULL,
Pablo Galindo2b74c832020-04-27 18:02:07 +01001104 flags, NULL, arena);
Pablo Galindoc5fc1562020-04-22 23:29:27 +01001105 }
1106 else {
1107 mod = PyParser_ASTFromFileObject(fp, filename, NULL, start, 0, 0,
1108 flags, NULL, arena);
1109 }
1110
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001111 if (closeit)
1112 fclose(fp);
1113 if (mod == NULL) {
Victor Stinner95701bd2013-11-06 18:41:07 +01001114 goto exit;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001115 }
1116 ret = run_mod(mod, filename, globals, locals, flags, arena);
Victor Stinner95701bd2013-11-06 18:41:07 +01001117
1118exit:
1119 Py_XDECREF(filename);
1120 if (arena != NULL)
1121 PyArena_Free(arena);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001122 return ret;
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001123}
1124
Guido van Rossum6c193fa2007-12-05 05:14:58 +00001125static void
1126flush_io(void)
1127{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001128 PyObject *f, *r;
1129 PyObject *type, *value, *traceback;
Amaury Forgeot d'Arc9ed77352008-04-04 23:25:27 +00001130
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001131 /* Save the current exception */
1132 PyErr_Fetch(&type, &value, &traceback);
Amaury Forgeot d'Arc9ed77352008-04-04 23:25:27 +00001133
Victor Stinnerbd303c12013-11-07 23:07:29 +01001134 f = _PySys_GetObjectId(&PyId_stderr);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001135 if (f != NULL) {
Jeroen Demeyer762f93f2019-07-08 10:19:25 +02001136 r = _PyObject_CallMethodIdNoArgs(f, &PyId_flush);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001137 if (r)
1138 Py_DECREF(r);
1139 else
1140 PyErr_Clear();
1141 }
Victor Stinnerbd303c12013-11-07 23:07:29 +01001142 f = _PySys_GetObjectId(&PyId_stdout);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001143 if (f != NULL) {
Jeroen Demeyer762f93f2019-07-08 10:19:25 +02001144 r = _PyObject_CallMethodIdNoArgs(f, &PyId_flush);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001145 if (r)
1146 Py_DECREF(r);
1147 else
1148 PyErr_Clear();
1149 }
Amaury Forgeot d'Arc9ed77352008-04-04 23:25:27 +00001150
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001151 PyErr_Restore(type, value, traceback);
Guido van Rossum6c193fa2007-12-05 05:14:58 +00001152}
1153
Guido van Rossum82598051997-03-05 00:20:32 +00001154static PyObject *
Victor Stinner1c1e68c2020-03-27 15:11:45 +01001155run_eval_code_obj(PyThreadState *tstate, PyCodeObject *co, PyObject *globals, PyObject *locals)
Gregory P. Smith38f11cc2019-02-16 12:57:40 -08001156{
1157 PyObject *v;
Gregory P. Smithd9bc5432019-02-20 17:35:54 -08001158 /*
1159 * We explicitly re-initialize _Py_UnhandledKeyboardInterrupt every eval
1160 * _just in case_ someone is calling into an embedded Python where they
1161 * don't care about an uncaught KeyboardInterrupt exception (why didn't they
1162 * leave config.install_signal_handlers set to 0?!?) but then later call
1163 * Py_Main() itself (which _checks_ this flag and dies with a signal after
1164 * its interpreter exits). We don't want a previous embedded interpreter's
1165 * uncaught exception to trigger an unexplained signal exit from a future
1166 * Py_Main() based one.
1167 */
1168 _Py_UnhandledKeyboardInterrupt = 0;
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001169
1170 /* Set globals['__builtins__'] if it doesn't exist */
1171 if (globals != NULL && PyDict_GetItemString(globals, "__builtins__") == NULL) {
Victor Stinner1c1e68c2020-03-27 15:11:45 +01001172 if (PyDict_SetItemString(globals, "__builtins__",
1173 tstate->interp->builtins) < 0) {
Victor Stinner9ef5dca2019-05-16 17:38:16 +02001174 return NULL;
1175 }
1176 }
1177
Gregory P. Smith38f11cc2019-02-16 12:57:40 -08001178 v = PyEval_EvalCode((PyObject*)co, globals, locals);
Victor Stinner1c1e68c2020-03-27 15:11:45 +01001179 if (!v && _PyErr_Occurred(tstate) == PyExc_KeyboardInterrupt) {
Gregory P. Smith38f11cc2019-02-16 12:57:40 -08001180 _Py_UnhandledKeyboardInterrupt = 1;
1181 }
1182 return v;
1183}
1184
1185static PyObject *
Victor Stinner95701bd2013-11-06 18:41:07 +01001186run_mod(mod_ty mod, PyObject *filename, PyObject *globals, PyObject *locals,
1187 PyCompilerFlags *flags, PyArena *arena)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001188{
Victor Stinner1c1e68c2020-03-27 15:11:45 +01001189 PyThreadState *tstate = _PyThreadState_GET();
1190 PyCodeObject *co = PyAST_CompileObject(mod, filename, flags, -1, arena);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001191 if (co == NULL)
1192 return NULL;
Steve Dowerb82e17e2019-05-23 08:45:22 -07001193
Victor Stinner1c1e68c2020-03-27 15:11:45 +01001194 if (_PySys_Audit(tstate, "exec", "O", co) < 0) {
Steve Dowerb82e17e2019-05-23 08:45:22 -07001195 Py_DECREF(co);
1196 return NULL;
1197 }
1198
Victor Stinner1c1e68c2020-03-27 15:11:45 +01001199 PyObject *v = run_eval_code_obj(tstate, co, globals, locals);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001200 Py_DECREF(co);
1201 return v;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001202}
1203
Guido van Rossum82598051997-03-05 00:20:32 +00001204static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001205run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001206 PyObject *locals, PyCompilerFlags *flags)
Guido van Rossumfdef2711994-09-14 13:31:04 +00001207{
Victor Stinner1c1e68c2020-03-27 15:11:45 +01001208 PyThreadState *tstate = _PyThreadState_GET();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001209 PyCodeObject *co;
1210 PyObject *v;
1211 long magic;
1212 long PyImport_GetMagicNumber(void);
Guido van Rossumfdef2711994-09-14 13:31:04 +00001213
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001214 magic = PyMarshal_ReadLongFromFile(fp);
1215 if (magic != PyImport_GetMagicNumber()) {
Victor Stinner5200f552015-03-18 13:56:25 +01001216 if (!PyErr_Occurred())
1217 PyErr_SetString(PyExc_RuntimeError,
1218 "Bad magic number in .pyc file");
Zackery Spytzea737752018-06-23 21:15:24 -06001219 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001220 }
Benjamin Peterson42aa93b2017-12-09 10:26:52 -08001221 /* Skip the rest of the header. */
1222 (void) PyMarshal_ReadLongFromFile(fp);
Antoine Pitrou5136ac02012-01-13 18:52:16 +01001223 (void) PyMarshal_ReadLongFromFile(fp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001224 (void) PyMarshal_ReadLongFromFile(fp);
Zackery Spytzea737752018-06-23 21:15:24 -06001225 if (PyErr_Occurred()) {
1226 goto error;
1227 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001228 v = PyMarshal_ReadLastObjectFromFile(fp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001229 if (v == NULL || !PyCode_Check(v)) {
1230 Py_XDECREF(v);
1231 PyErr_SetString(PyExc_RuntimeError,
1232 "Bad code object in .pyc file");
Zackery Spytzea737752018-06-23 21:15:24 -06001233 goto error;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001234 }
Zackery Spytzea737752018-06-23 21:15:24 -06001235 fclose(fp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001236 co = (PyCodeObject *)v;
Victor Stinner1c1e68c2020-03-27 15:11:45 +01001237 v = run_eval_code_obj(tstate, co, globals, locals);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001238 if (v && flags)
1239 flags->cf_flags |= (co->co_flags & PyCF_MASK);
1240 Py_DECREF(co);
1241 return v;
Zackery Spytzea737752018-06-23 21:15:24 -06001242error:
1243 fclose(fp);
1244 return NULL;
Guido van Rossumfdef2711994-09-14 13:31:04 +00001245}
1246
Guido van Rossum82598051997-03-05 00:20:32 +00001247PyObject *
Victor Stinner14e461d2013-08-26 22:28:21 +02001248Py_CompileStringObject(const char *str, PyObject *filename, int start,
1249 PyCompilerFlags *flags, int optimize)
Jeremy Hyltonbc320242001-03-22 02:47:58 +00001250{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 PyCodeObject *co;
1252 mod_ty mod;
Victor Stinner1def7752020-04-23 03:03:24 +02001253 int use_peg = _PyInterpreterState_GET()->config._use_peg_parser;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001254 PyArena *arena = PyArena_New();
1255 if (arena == NULL)
1256 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001257
Pablo Galindoc5fc1562020-04-22 23:29:27 +01001258 if (use_peg) {
1259 mod = PyPegen_ASTFromStringObject(str, filename, start, flags, arena);
1260 }
1261 else {
1262 mod = PyParser_ASTFromStringObject(str, filename, start, flags, arena);
1263 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001264 if (mod == NULL) {
1265 PyArena_Free(arena);
1266 return NULL;
1267 }
1268 if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
1269 PyObject *result = PyAST_mod2obj(mod);
1270 PyArena_Free(arena);
1271 return result;
1272 }
Victor Stinner14e461d2013-08-26 22:28:21 +02001273 co = PyAST_CompileObject(mod, filename, flags, optimize, arena);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001274 PyArena_Free(arena);
1275 return (PyObject *)co;
Guido van Rossum5b722181993-03-30 17:46:03 +00001276}
1277
Victor Stinner14e461d2013-08-26 22:28:21 +02001278PyObject *
1279Py_CompileStringExFlags(const char *str, const char *filename_str, int start,
1280 PyCompilerFlags *flags, int optimize)
1281{
1282 PyObject *filename, *co;
1283 filename = PyUnicode_DecodeFSDefault(filename_str);
1284 if (filename == NULL)
1285 return NULL;
1286 co = Py_CompileStringObject(str, filename, start, flags, optimize);
1287 Py_DECREF(filename);
1288 return co;
1289}
1290
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00001291/* For use in Py_LIMITED_API */
1292#undef Py_CompileString
1293PyObject *
1294PyCompileString(const char *str, const char *filename, int start)
1295{
1296 return Py_CompileStringFlags(str, filename, start, NULL);
1297}
1298
Dino Viehland41540692019-05-28 16:21:17 -07001299const char *
1300_Py_SourceAsString(PyObject *cmd, const char *funcname, const char *what, PyCompilerFlags *cf, PyObject **cmd_copy)
1301{
1302 const char *str;
1303 Py_ssize_t size;
1304 Py_buffer view;
1305
1306 *cmd_copy = NULL;
1307 if (PyUnicode_Check(cmd)) {
1308 cf->cf_flags |= PyCF_IGNORE_COOKIE;
1309 str = PyUnicode_AsUTF8AndSize(cmd, &size);
1310 if (str == NULL)
1311 return NULL;
1312 }
1313 else if (PyBytes_Check(cmd)) {
1314 str = PyBytes_AS_STRING(cmd);
1315 size = PyBytes_GET_SIZE(cmd);
1316 }
1317 else if (PyByteArray_Check(cmd)) {
1318 str = PyByteArray_AS_STRING(cmd);
1319 size = PyByteArray_GET_SIZE(cmd);
1320 }
1321 else if (PyObject_GetBuffer(cmd, &view, PyBUF_SIMPLE) == 0) {
1322 /* Copy to NUL-terminated buffer. */
1323 *cmd_copy = PyBytes_FromStringAndSize(
1324 (const char *)view.buf, view.len);
1325 PyBuffer_Release(&view);
1326 if (*cmd_copy == NULL) {
1327 return NULL;
1328 }
1329 str = PyBytes_AS_STRING(*cmd_copy);
1330 size = PyBytes_GET_SIZE(*cmd_copy);
1331 }
1332 else {
1333 PyErr_Format(PyExc_TypeError,
1334 "%s() arg 1 must be a %s object",
1335 funcname, what);
1336 return NULL;
1337 }
1338
1339 if (strlen(str) != (size_t)size) {
1340 PyErr_SetString(PyExc_ValueError,
1341 "source code string cannot contain null bytes");
1342 Py_CLEAR(*cmd_copy);
1343 return NULL;
1344 }
1345 return str;
1346}
1347
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001348struct symtable *
Victor Stinner14e461d2013-08-26 22:28:21 +02001349Py_SymtableStringObject(const char *str, PyObject *filename, int start)
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001350{
Victor Stinner37d66d72019-06-13 02:16:41 +02001351 PyCompilerFlags flags = _PyCompilerFlags_INIT;
Dino Viehland41540692019-05-28 16:21:17 -07001352 return _Py_SymtableStringObjectFlags(str, filename, start, &flags);
1353}
1354
1355struct symtable *
1356_Py_SymtableStringObjectFlags(const char *str, PyObject *filename, int start, PyCompilerFlags *flags)
1357{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001358 struct symtable *st;
1359 mod_ty mod;
Victor Stinner1def7752020-04-23 03:03:24 +02001360 int use_peg = _PyInterpreterState_GET()->config._use_peg_parser;
Victor Stinner14e461d2013-08-26 22:28:21 +02001361 PyArena *arena;
1362
1363 arena = PyArena_New();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001364 if (arena == NULL)
1365 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001366
Pablo Galindoc5fc1562020-04-22 23:29:27 +01001367 if (use_peg) {
1368 mod = PyPegen_ASTFromStringObject(str, filename, start, flags, arena);
1369 }
1370 else {
1371 mod = PyParser_ASTFromStringObject(str, filename, start, flags, arena);
1372 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001373 if (mod == NULL) {
1374 PyArena_Free(arena);
1375 return NULL;
1376 }
Victor Stinner14e461d2013-08-26 22:28:21 +02001377 st = PySymtable_BuildObject(mod, filename, 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001378 PyArena_Free(arena);
1379 return st;
Jeremy Hylton4b38da62001-02-02 18:19:15 +00001380}
1381
Victor Stinner14e461d2013-08-26 22:28:21 +02001382struct symtable *
1383Py_SymtableString(const char *str, const char *filename_str, int start)
1384{
1385 PyObject *filename;
1386 struct symtable *st;
1387
1388 filename = PyUnicode_DecodeFSDefault(filename_str);
1389 if (filename == NULL)
1390 return NULL;
1391 st = Py_SymtableStringObject(str, filename, start);
1392 Py_DECREF(filename);
1393 return st;
1394}
1395
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001396/* Preferred access to parser is through AST. */
1397mod_ty
Victor Stinner14e461d2013-08-26 22:28:21 +02001398PyParser_ASTFromStringObject(const char *s, PyObject *filename, int start,
1399 PyCompilerFlags *flags, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001400{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001401 mod_ty mod;
Victor Stinner37d66d72019-06-13 02:16:41 +02001402 PyCompilerFlags localflags = _PyCompilerFlags_INIT;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001403 perrdetail err;
1404 int iflags = PARSER_FLAGS(flags);
Guido van Rossum495da292019-03-07 12:38:08 -08001405 if (flags && flags->cf_feature_version < 7)
1406 iflags |= PyPARSE_ASYNC_HACKS;
Christian Heimes4d6ec852008-03-26 22:34:47 +00001407
Victor Stinner14e461d2013-08-26 22:28:21 +02001408 node *n = PyParser_ParseStringObject(s, filename,
1409 &_PyParser_Grammar, start, &err,
1410 &iflags);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001411 if (flags == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001412 flags = &localflags;
1413 }
1414 if (n) {
1415 flags->cf_flags |= iflags & PyCF_MASK;
Victor Stinner14e461d2013-08-26 22:28:21 +02001416 mod = PyAST_FromNodeObject(n, flags, filename, arena);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001417 PyNode_Free(n);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001418 }
1419 else {
1420 err_input(&err);
Victor Stinner7f2fee32011-04-05 00:39:01 +02001421 mod = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001422 }
Victor Stinner7f2fee32011-04-05 00:39:01 +02001423 err_free(&err);
1424 return mod;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001425}
1426
1427mod_ty
Victor Stinner14e461d2013-08-26 22:28:21 +02001428PyParser_ASTFromString(const char *s, const char *filename_str, int start,
1429 PyCompilerFlags *flags, PyArena *arena)
1430{
1431 PyObject *filename;
1432 mod_ty mod;
1433 filename = PyUnicode_DecodeFSDefault(filename_str);
1434 if (filename == NULL)
1435 return NULL;
1436 mod = PyParser_ASTFromStringObject(s, filename, start, flags, arena);
1437 Py_DECREF(filename);
1438 return mod;
1439}
1440
1441mod_ty
1442PyParser_ASTFromFileObject(FILE *fp, PyObject *filename, const char* enc,
Serhiy Storchakaef1585e2015-12-25 20:01:53 +02001443 int start, const char *ps1,
1444 const char *ps2, PyCompilerFlags *flags, int *errcode,
Victor Stinner14e461d2013-08-26 22:28:21 +02001445 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001446{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001447 mod_ty mod;
Victor Stinner37d66d72019-06-13 02:16:41 +02001448 PyCompilerFlags localflags = _PyCompilerFlags_INIT;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001449 perrdetail err;
1450 int iflags = PARSER_FLAGS(flags);
Christian Heimes4d6ec852008-03-26 22:34:47 +00001451
Victor Stinner14e461d2013-08-26 22:28:21 +02001452 node *n = PyParser_ParseFileObject(fp, filename, enc,
1453 &_PyParser_Grammar,
1454 start, ps1, ps2, &err, &iflags);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001455 if (flags == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001456 flags = &localflags;
1457 }
1458 if (n) {
1459 flags->cf_flags |= iflags & PyCF_MASK;
Victor Stinner14e461d2013-08-26 22:28:21 +02001460 mod = PyAST_FromNodeObject(n, flags, filename, arena);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001461 PyNode_Free(n);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001462 }
1463 else {
1464 err_input(&err);
1465 if (errcode)
1466 *errcode = err.error;
Victor Stinner7f2fee32011-04-05 00:39:01 +02001467 mod = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001468 }
Victor Stinner7f2fee32011-04-05 00:39:01 +02001469 err_free(&err);
1470 return mod;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001471}
1472
Victor Stinner14e461d2013-08-26 22:28:21 +02001473mod_ty
1474PyParser_ASTFromFile(FILE *fp, const char *filename_str, const char* enc,
Serhiy Storchakaef1585e2015-12-25 20:01:53 +02001475 int start, const char *ps1,
1476 const char *ps2, PyCompilerFlags *flags, int *errcode,
Victor Stinner14e461d2013-08-26 22:28:21 +02001477 PyArena *arena)
1478{
1479 mod_ty mod;
1480 PyObject *filename;
1481 filename = PyUnicode_DecodeFSDefault(filename_str);
1482 if (filename == NULL)
1483 return NULL;
1484 mod = PyParser_ASTFromFileObject(fp, filename, enc, start, ps1, ps2,
1485 flags, errcode, arena);
1486 Py_DECREF(filename);
1487 return mod;
1488}
1489
Guido van Rossuma110aa61994-08-29 12:50:44 +00001490/* Simplified interface to parsefile -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001491
Guido van Rossuma110aa61994-08-29 12:50:44 +00001492node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001493PyParser_SimpleParseFileFlags(FILE *fp, const char *filename, int start, int flags)
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001494{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001495 perrdetail err;
1496 node *n = PyParser_ParseFileFlags(fp, filename, NULL,
1497 &_PyParser_Grammar,
1498 start, NULL, NULL, &err, flags);
1499 if (n == NULL)
1500 err_input(&err);
Victor Stinner7f2fee32011-04-05 00:39:01 +02001501 err_free(&err);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001502
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001503 return n;
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001504}
1505
Guido van Rossuma110aa61994-08-29 12:50:44 +00001506/* Simplified interface to parsestring -- return node or set exception */
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001507
Guido van Rossuma110aa61994-08-29 12:50:44 +00001508node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001509PyParser_SimpleParseStringFlags(const char *str, int start, int flags)
Tim Petersfe2127d2001-07-16 05:37:24 +00001510{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001511 perrdetail err;
1512 node *n = PyParser_ParseStringFlags(str, &_PyParser_Grammar,
1513 start, &err, flags);
1514 if (n == NULL)
1515 err_input(&err);
Victor Stinner7f2fee32011-04-05 00:39:01 +02001516 err_free(&err);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001517 return n;
Tim Petersfe2127d2001-07-16 05:37:24 +00001518}
1519
1520node *
Martin v. Löwis95292d62002-12-11 14:04:59 +00001521PyParser_SimpleParseStringFlagsFilename(const char *str, const char *filename,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001522 int start, int flags)
Thomas Heller6b17abf2002-07-09 09:23:27 +00001523{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001524 perrdetail err;
1525 node *n = PyParser_ParseStringFlagsFilename(str, filename,
1526 &_PyParser_Grammar, start, &err, flags);
1527 if (n == NULL)
1528 err_input(&err);
Victor Stinner7f2fee32011-04-05 00:39:01 +02001529 err_free(&err);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001530 return n;
Thomas Heller6b17abf2002-07-09 09:23:27 +00001531}
1532
Guido van Rossum66ebd912003-04-17 16:02:26 +00001533/* May want to move a more generalized form of this to parsetok.c or
1534 even parser modules. */
1535
1536void
Victor Stinner7f2fee32011-04-05 00:39:01 +02001537PyParser_ClearError(perrdetail *err)
1538{
1539 err_free(err);
1540}
1541
1542void
Guido van Rossum66ebd912003-04-17 16:02:26 +00001543PyParser_SetError(perrdetail *err)
1544{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001545 err_input(err);
Guido van Rossum66ebd912003-04-17 16:02:26 +00001546}
1547
Victor Stinner7f2fee32011-04-05 00:39:01 +02001548static void
1549err_free(perrdetail *err)
1550{
1551 Py_CLEAR(err->filename);
1552}
1553
Guido van Rossuma110aa61994-08-29 12:50:44 +00001554/* Set the error appropriate to the given input error code (see errcode.h) */
1555
1556static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001557err_input(perrdetail *err)
Guido van Rossuma110aa61994-08-29 12:50:44 +00001558{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001559 PyObject *v, *w, *errtype, *errtext;
1560 PyObject *msg_obj = NULL;
Serhiy Storchakae2f92de2017-11-11 13:06:26 +02001561 const char *msg = NULL;
Serhiy Storchaka65fd0592014-01-21 22:26:52 +02001562 int offset = err->offset;
Victor Stinner4c7c8c32010-10-16 13:14:10 +00001563
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001564 errtype = PyExc_SyntaxError;
1565 switch (err->error) {
1566 case E_ERROR:
Serhiy Storchaka993030a2018-07-12 00:17:53 +03001567 goto cleanup;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001568 case E_SYNTAX:
1569 errtype = PyExc_IndentationError;
1570 if (err->expected == INDENT)
1571 msg = "expected an indented block";
1572 else if (err->token == INDENT)
1573 msg = "unexpected indent";
1574 else if (err->token == DEDENT)
1575 msg = "unexpected unindent";
Serhiy Storchakaaba24ff2018-07-23 23:41:11 +03001576 else if (err->expected == NOTEQUAL) {
1577 errtype = PyExc_SyntaxError;
1578 msg = "with Barry as BDFL, use '<>' instead of '!='";
1579 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001580 else {
1581 errtype = PyExc_SyntaxError;
1582 msg = "invalid syntax";
1583 }
1584 break;
1585 case E_TOKEN:
1586 msg = "invalid token";
1587 break;
1588 case E_EOFS:
1589 msg = "EOF while scanning triple-quoted string literal";
1590 break;
1591 case E_EOLS:
1592 msg = "EOL while scanning string literal";
1593 break;
1594 case E_INTR:
1595 if (!PyErr_Occurred())
1596 PyErr_SetNone(PyExc_KeyboardInterrupt);
1597 goto cleanup;
1598 case E_NOMEM:
1599 PyErr_NoMemory();
1600 goto cleanup;
1601 case E_EOF:
1602 msg = "unexpected EOF while parsing";
1603 break;
1604 case E_TABSPACE:
1605 errtype = PyExc_TabError;
1606 msg = "inconsistent use of tabs and spaces in indentation";
1607 break;
1608 case E_OVERFLOW:
1609 msg = "expression too long";
1610 break;
1611 case E_DEDENT:
1612 errtype = PyExc_IndentationError;
1613 msg = "unindent does not match any outer indentation level";
1614 break;
1615 case E_TOODEEP:
1616 errtype = PyExc_IndentationError;
1617 msg = "too many levels of indentation";
1618 break;
1619 case E_DECODE: {
1620 PyObject *type, *value, *tb;
1621 PyErr_Fetch(&type, &value, &tb);
1622 msg = "unknown decode error";
1623 if (value != NULL)
1624 msg_obj = PyObject_Str(value);
1625 Py_XDECREF(type);
1626 Py_XDECREF(value);
1627 Py_XDECREF(tb);
1628 break;
1629 }
1630 case E_LINECONT:
1631 msg = "unexpected character after line continuation character";
1632 break;
Martin v. Löwis47383402007-08-15 07:32:56 +00001633
Meador Ingefa21bf02012-01-19 01:08:41 -06001634 case E_BADSINGLE:
1635 msg = "multiple statements found while compiling a single statement";
1636 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001637 default:
1638 fprintf(stderr, "error=%d\n", err->error);
1639 msg = "unknown parsing error";
1640 break;
1641 }
1642 /* err->text may not be UTF-8 in case of decoding errors.
1643 Explicitly convert to an object. */
1644 if (!err->text) {
1645 errtext = Py_None;
1646 Py_INCREF(Py_None);
1647 } else {
Serhiy Storchaka65fd0592014-01-21 22:26:52 +02001648 errtext = PyUnicode_DecodeUTF8(err->text, err->offset,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001649 "replace");
Serhiy Storchaka65fd0592014-01-21 22:26:52 +02001650 if (errtext != NULL) {
1651 Py_ssize_t len = strlen(err->text);
1652 offset = (int)PyUnicode_GET_LENGTH(errtext);
1653 if (len != err->offset) {
1654 Py_DECREF(errtext);
1655 errtext = PyUnicode_DecodeUTF8(err->text, len,
1656 "replace");
1657 }
1658 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001659 }
Victor Stinner7f2fee32011-04-05 00:39:01 +02001660 v = Py_BuildValue("(OiiN)", err->filename,
Serhiy Storchaka65fd0592014-01-21 22:26:52 +02001661 err->lineno, offset, errtext);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001662 if (v != NULL) {
1663 if (msg_obj)
1664 w = Py_BuildValue("(OO)", msg_obj, v);
1665 else
1666 w = Py_BuildValue("(sO)", msg, v);
1667 } else
1668 w = NULL;
1669 Py_XDECREF(v);
1670 PyErr_SetObject(errtype, w);
1671 Py_XDECREF(w);
Georg Brandl3dbca812008-07-23 16:10:53 +00001672cleanup:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001673 Py_XDECREF(msg_obj);
1674 if (err->text != NULL) {
1675 PyObject_FREE(err->text);
1676 err->text = NULL;
1677 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001678}
1679
Fredrik Lundh2f15b252000-08-27 19:15:31 +00001680
Zachary Warec4821d62014-11-21 23:35:12 -06001681#if defined(USE_STACKCHECK)
1682#if defined(WIN32) && defined(_MSC_VER)
1683
1684/* Stack checking for Microsoft C */
1685
1686#include <malloc.h>
1687#include <excpt.h>
1688
1689/*
1690 * Return non-zero when we run out of memory on the stack; zero otherwise.
1691 */
1692int
1693PyOS_CheckStack(void)
1694{
1695 __try {
1696 /* alloca throws a stack overflow exception if there's
1697 not enough space left on the stack */
1698 alloca(PYOS_STACK_MARGIN * sizeof(void*));
1699 return 0;
1700 } __except (GetExceptionCode() == STATUS_STACK_OVERFLOW ?
1701 EXCEPTION_EXECUTE_HANDLER :
1702 EXCEPTION_CONTINUE_SEARCH) {
1703 int errcode = _resetstkoflw();
1704 if (errcode == 0)
1705 {
1706 Py_FatalError("Could not reset the stack!");
1707 }
1708 }
1709 return 1;
1710}
1711
1712#endif /* WIN32 && _MSC_VER */
1713
1714/* Alternate implementations can be added here... */
1715
1716#endif /* USE_STACKCHECK */
1717
Martin Panter46f50722016-05-26 05:35:26 +00001718/* Deprecated C API functions still provided for binary compatibility */
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001719
1720#undef PyParser_SimpleParseFile
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001721PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001722PyParser_SimpleParseFile(FILE *fp, const char *filename, int start)
1723{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001724 return PyParser_SimpleParseFileFlags(fp, filename, start, 0);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001725}
1726
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001727#undef PyParser_SimpleParseString
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001728PyAPI_FUNC(node *)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001729PyParser_SimpleParseString(const char *str, int start)
1730{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001731 return PyParser_SimpleParseStringFlags(str, start, 0);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001732}
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001733
1734#undef PyRun_AnyFile
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001735PyAPI_FUNC(int)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001736PyRun_AnyFile(FILE *fp, const char *name)
1737{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001738 return PyRun_AnyFileExFlags(fp, name, 0, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001739}
1740
1741#undef PyRun_AnyFileEx
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001742PyAPI_FUNC(int)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001743PyRun_AnyFileEx(FILE *fp, const char *name, int closeit)
1744{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001745 return PyRun_AnyFileExFlags(fp, name, closeit, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001746}
1747
1748#undef PyRun_AnyFileFlags
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001749PyAPI_FUNC(int)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001750PyRun_AnyFileFlags(FILE *fp, const char *name, PyCompilerFlags *flags)
1751{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001752 return PyRun_AnyFileExFlags(fp, name, 0, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001753}
1754
1755#undef PyRun_File
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001756PyAPI_FUNC(PyObject *)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001757PyRun_File(FILE *fp, const char *p, int s, PyObject *g, PyObject *l)
1758{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001759 return PyRun_FileExFlags(fp, p, s, g, l, 0, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001760}
1761
1762#undef PyRun_FileEx
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001763PyAPI_FUNC(PyObject *)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001764PyRun_FileEx(FILE *fp, const char *p, int s, PyObject *g, PyObject *l, int c)
1765{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001766 return PyRun_FileExFlags(fp, p, s, g, l, c, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001767}
1768
1769#undef PyRun_FileFlags
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001770PyAPI_FUNC(PyObject *)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001771PyRun_FileFlags(FILE *fp, const char *p, int s, PyObject *g, PyObject *l,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001772 PyCompilerFlags *flags)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001773{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001774 return PyRun_FileExFlags(fp, p, s, g, l, 0, flags);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001775}
1776
1777#undef PyRun_SimpleFile
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001778PyAPI_FUNC(int)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001779PyRun_SimpleFile(FILE *f, const char *p)
1780{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001781 return PyRun_SimpleFileExFlags(f, p, 0, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001782}
1783
1784#undef PyRun_SimpleFileEx
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001785PyAPI_FUNC(int)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001786PyRun_SimpleFileEx(FILE *f, const char *p, int c)
1787{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001788 return PyRun_SimpleFileExFlags(f, p, c, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001789}
1790
1791
1792#undef PyRun_String
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001793PyAPI_FUNC(PyObject *)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001794PyRun_String(const char *str, int s, PyObject *g, PyObject *l)
1795{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001796 return PyRun_StringFlags(str, s, g, l, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001797}
1798
1799#undef PyRun_SimpleString
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001800PyAPI_FUNC(int)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001801PyRun_SimpleString(const char *s)
1802{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001803 return PyRun_SimpleStringFlags(s, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001804}
1805
1806#undef Py_CompileString
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001807PyAPI_FUNC(PyObject *)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001808Py_CompileString(const char *str, const char *p, int s)
1809{
Georg Brandl8334fd92010-12-04 10:26:46 +00001810 return Py_CompileStringExFlags(str, p, s, NULL, -1);
1811}
1812
1813#undef Py_CompileStringFlags
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001814PyAPI_FUNC(PyObject *)
Georg Brandl8334fd92010-12-04 10:26:46 +00001815Py_CompileStringFlags(const char *str, const char *p, int s,
1816 PyCompilerFlags *flags)
1817{
1818 return Py_CompileStringExFlags(str, p, s, flags, -1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001819}
1820
1821#undef PyRun_InteractiveOne
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001822PyAPI_FUNC(int)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001823PyRun_InteractiveOne(FILE *f, const char *p)
1824{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001825 return PyRun_InteractiveOneFlags(f, p, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001826}
1827
1828#undef PyRun_InteractiveLoop
Benjamin Peterson62161ce2019-11-04 21:34:14 -08001829PyAPI_FUNC(int)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001830PyRun_InteractiveLoop(FILE *f, const char *p)
1831{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001832 return PyRun_InteractiveLoopFlags(f, p, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001833}
1834
1835#ifdef __cplusplus
1836}
1837#endif