#include "Python.h"

#include "pycore_ast.h" // asdl_seq_*

#include "pycore_call.h" // _PyObject_CallMethodFormat()

#include "pycore_compile.h" // _PyAST_Optimize

#include "pycore_fileutils.h" // _Py_BEGIN_SUPPRESS_IPH

#include "pycore_frame.h" // _PyFrame_GetCode()

#include "pycore_interp.h" // PyInterpreterState.gc

#include "pycore_parser.h" // _PyParser_ASTFromString

#include "pycore_pyarena.h" // _PyArena_Free()

#include "pycore_pyerrors.h" // _PyErr_GetRaisedException()

#include "pycore_pystate.h" // _PyThreadState_GET()

#include "pycore_traceback.h" // EXCEPTION_TB_HEADER

#include "../Parser/pegen.h" // _PyPegen_byte_offset_to_character_offset()

#include "frameobject.h" // PyFrame_New()

#include "structmember.h" // PyMemberDef

#include "osdefs.h" // SEP

#ifdef HAVE_FCNTL_H

# include <fcntl.h>

#define OFF(x) offsetof(PyTracebackObject, x)

#define PUTS(fd, str) _Py_write_noraise(fd, str, (int)strlen(str))

#define MAX_STRING_LENGTH 500

#define MAX_FRAME_DEPTH 100

#define MAX_NTHREADS 100

extern char* _PyTokenizer_FindEncodingFilename(int, PyObject *);

#include "clinic/traceback.c.h"

static PyObject *

tb_create_raw(PyTracebackObject *next, PyFrameObject *frame, int lasti,

int lineno)

{

PyTracebackObject *tb;

if ((next != NULL && !PyTraceBack_Check(next)) ||

frame == NULL || !PyFrame_Check(frame)) {

PyErr_BadInternalCall();

return NULL;

}

tb = PyObject_GC_New(PyTracebackObject, &PyTraceBack_Type);

if (tb != NULL) {

tb->tb_next = (PyTracebackObject*)Py_XNewRef(next);

tb->tb_frame = (PyFrameObject*)Py_XNewRef(frame);

tb->tb_lasti = lasti;

tb->tb_lineno = lineno;

PyObject_GC_Track(tb);

}

return (PyObject *)tb;

}

static PyObject *

tb_new_impl(PyTypeObject *type, PyObject *tb_next, PyFrameObject *tb_frame,

int tb_lasti, int tb_lineno)

{

if (tb_next == Py_None) {

tb_next = NULL;

} else if (!PyTraceBack_Check(tb_next)) {

return PyErr_Format(PyExc_TypeError,

"expected traceback object or None, got '%s'",

Py_TYPE(tb_next)->tp_name);

}

return tb_create_raw((PyTracebackObject *)tb_next, tb_frame, tb_lasti,

tb_lineno);

}

static PyObject *

tb_dir(PyTracebackObject *self, PyObject *Py_UNUSED(ignored))

{

return Py_BuildValue("[ssss]", "tb_frame", "tb_next",

"tb_lasti", "tb_lineno");

}

static PyObject *

tb_next_get(PyTracebackObject *self, void *Py_UNUSED(_))

{

PyObject* ret = (PyObject*)self->tb_next;

if (!ret) {

ret = Py_None;

}

return Py_NewRef(ret);

}

static int

tb_next_set(PyTracebackObject *self, PyObject *new_next, void *Py_UNUSED(_))

{

if (!new_next) {

PyErr_Format(PyExc_TypeError, "can't delete tb_next attribute");

return -1;

}

/* We accept None or a traceback object, and map None -> NULL (inverse of

if (new_next == Py_None) {

new_next = NULL;

} else if (!PyTraceBack_Check(new_next)) {

PyErr_Format(PyExc_TypeError,

"expected traceback object, got '%s'",

Py_TYPE(new_next)->tp_name);

return -1;

}

/* Check for loops */

PyTracebackObject *cursor = (PyTracebackObject *)new_next;

while (cursor) {

if (cursor == self) {

PyErr_Format(PyExc_ValueError, "traceback loop detected");

return -1;

}

cursor = cursor->tb_next;

}

Py_XSETREF(self->tb_next, (PyTracebackObject *)Py_XNewRef(new_next));

return 0;

}

static PyMethodDef tb_methods[] = {

{"__dir__", _PyCFunction_CAST(tb_dir), METH_NOARGS},

{NULL, NULL, 0, NULL},

};

static PyMemberDef tb_memberlist[] = {

{"tb_frame", T_OBJECT, OFF(tb_frame), READONLY|PY_AUDIT_READ},

{"tb_lasti", T_INT, OFF(tb_lasti), READONLY},

{"tb_lineno", T_INT, OFF(tb_lineno), READONLY},

{NULL} /* Sentinel */

};

static PyGetSetDef tb_getsetters[] = {

{"tb_next", (getter)tb_next_get, (setter)tb_next_set, NULL, NULL},

{NULL} /* Sentinel */

};

static void

tb_dealloc(PyTracebackObject *tb)

{

PyObject_GC_UnTrack(tb);

Py_TRASHCAN_BEGIN(tb, tb_dealloc)

Py_XDECREF(tb->tb_next);

Py_XDECREF(tb->tb_frame);

PyObject_GC_Del(tb);

Py_TRASHCAN_END

}

static int

tb_traverse(PyTracebackObject *tb, visitproc visit, void *arg)

{

Py_VISIT(tb->tb_next);

Py_VISIT(tb->tb_frame);

return 0;

}

static int

tb_clear(PyTracebackObject *tb)

{

Py_CLEAR(tb->tb_next);

Py_CLEAR(tb->tb_frame);

return 0;

}

PyTypeObject PyTraceBack_Type = {

PyVarObject_HEAD_INIT(&PyType_Type, 0)

"traceback",

sizeof(PyTracebackObject),

0,

(destructor)tb_dealloc, /*tp_dealloc*/

0, /*tp_vectorcall_offset*/

0, /*tp_getattr*/

0, /*tp_setattr*/

0, /*tp_as_async*/

0, /*tp_repr*/

0, /*tp_as_number*/

0, /*tp_as_sequence*/

0, /*tp_as_mapping*/

0, /* tp_hash */

0, /* tp_call */

0, /* tp_str */

PyObject_GenericGetAttr, /* tp_getattro */

0, /* tp_setattro */

0, /* tp_as_buffer */

Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */

tb_new__doc__, /* tp_doc */

(traverseproc)tb_traverse, /* tp_traverse */

(inquiry)tb_clear, /* tp_clear */

0, /* tp_richcompare */

0, /* tp_weaklistoffset */

0, /* tp_iter */

0, /* tp_iternext */

tb_methods, /* tp_methods */

tb_memberlist, /* tp_members */

tb_getsetters, /* tp_getset */

0, /* tp_base */

0, /* tp_dict */

0, /* tp_descr_get */

0, /* tp_descr_set */

0, /* tp_dictoffset */

0, /* tp_init */

0, /* tp_alloc */

tb_new, /* tp_new */

};

_PyTraceBack_FromFrame(PyObject *tb_next, PyFrameObject *frame)

{

assert(tb_next == NULL || PyTraceBack_Check(tb_next));

assert(frame != NULL);

int addr = _PyInterpreterFrame_LASTI(frame->f_frame) * sizeof(_Py_CODEUNIT);

return tb_create_raw((PyTracebackObject *)tb_next, frame, addr,

PyFrame_GetLineNumber(frame));

}

PyTraceBack_Here(PyFrameObject *frame)

{

PyObject *exc = PyErr_GetRaisedException();

assert(PyExceptionInstance_Check(exc));

PyObject *tb = PyException_GetTraceback(exc);

PyObject *newtb = _PyTraceBack_FromFrame(tb, frame);

Py_XDECREF(tb);

if (newtb == NULL) {

_PyErr_ChainExceptions1(exc);

return -1;

}

PyException_SetTraceback(exc, newtb);

Py_XDECREF(newtb);

PyErr_SetRaisedException(exc);

return 0;

}

void _PyTraceback_Add(const char *funcname, const char *filename, int lineno)

{

PyObject *globals;

PyCodeObject *code;

PyFrameObject *frame;

PyThreadState *tstate = _PyThreadState_GET();

/* Save and clear the current exception. Python functions must not be

PyObject *exc = _PyErr_GetRaisedException(tstate);

globals = PyDict_New();

if (!globals)

goto error;

code = PyCode_NewEmpty(filename, funcname, lineno);

if (!code) {

Py_DECREF(globals);

goto error;

}

frame = PyFrame_New(tstate, code, globals, NULL);

Py_DECREF(globals);

Py_DECREF(code);

if (!frame)

goto error;

frame->f_lineno = lineno;

_PyErr_SetRaisedException(tstate, exc);

PyTraceBack_Here(frame);

Py_DECREF(frame);

return;

error:

_PyErr_ChainExceptions1(exc);

}

static PyObject *

_Py_FindSourceFile(PyObject *filename, char* namebuf, size_t namelen, PyObject *io)

{

Py_ssize_t i;

PyObject *binary;

PyObject *v;

Py_ssize_t npath;

size_t taillen;

PyObject *syspath;

PyObject *path;

const char* tail;

PyObject *filebytes;

const char* filepath;

Py_ssize_t len;

PyObject* result;

PyObject *open = NULL;

filebytes = PyUnicode_EncodeFSDefault(filename);

if (filebytes == NULL) {

PyErr_Clear();

return NULL;

}

filepath = PyBytes_AS_STRING(filebytes);

/* Search tail of filename in sys.path before giving up */

tail = strrchr(filepath, SEP);

if (tail == NULL)

tail = filepath;

else

tail++;

taillen = strlen(tail);

PyThreadState *tstate = _PyThreadState_GET();

syspath = _PySys_GetAttr(tstate, &_Py_ID(path));

if (syspath == NULL || !PyList_Check(syspath))

goto error;

npath = PyList_Size(syspath);

open = PyObject_GetAttr(io, &_Py_ID(open));

for (i = 0; i < npath; i++) {

v = PyList_GetItem(syspath, i);

if (v == NULL) {

PyErr_Clear();

break;

}

if (!PyUnicode_Check(v))

continue;

path = PyUnicode_EncodeFSDefault(v);

if (path == NULL) {

PyErr_Clear();

continue;

}

len = PyBytes_GET_SIZE(path);

if (len + 1 + (Py_ssize_t)taillen >= (Py_ssize_t)namelen - 1) {

Py_DECREF(path);

continue; /* Too long */

}

strcpy(namebuf, PyBytes_AS_STRING(path));

Py_DECREF(path);

if (strlen(namebuf) != (size_t)len)

continue; /* v contains '\0' */

if (len > 0 && namebuf[len-1] != SEP)

namebuf[len++] = SEP;

strcpy(namebuf+len, tail);

binary = _PyObject_CallMethodFormat(tstate, open, "ss", namebuf, "rb");

if (binary != NULL) {

result = binary;

goto finally;

}

PyErr_Clear();

}

goto error;

error:

result = NULL;

finally:

Py_XDECREF(open);

Py_DECREF(filebytes);

return result;

}

_Py_WriteIndent(int indent, PyObject *f)

{

char buf[11] = " ";

assert(strlen(buf) == 10);

while (indent > 0) {

if (indent < 10) {

buf[indent] = '\0';

}

if (PyFile_WriteString(buf, f) < 0) {

return -1;

}

indent -= 10;

}

return 0;

}

_Py_WriteIndentedMargin(int indent, const char *margin, PyObject *f)

{

if (_Py_WriteIndent(indent, f) < 0) {

return -1;

}

if (margin) {

if (PyFile_WriteString(margin, f) < 0) {

return -1;

}

}

return 0;

}

static int

display_source_line_with_margin(PyObject *f, PyObject *filename, int lineno, int indent,

int margin_indent, const char *margin,

int *truncation, PyObject **line)

{

int fd;

int i;

char *found_encoding;

const char *encoding;

PyObject *io;

PyObject *binary;

PyObject *fob = NULL;

PyObject *lineobj = NULL;

PyObject *res;

char buf[MAXPATHLEN+1];

int kind;

const void *data;

/* open the file */

if (filename == NULL)

return 0;

/* Do not attempt to open things like <string> or <stdin> */

assert(PyUnicode_Check(filename));

if (PyUnicode_READ_CHAR(filename, 0) == '<') {

Py_ssize_t len = PyUnicode_GET_LENGTH(filename);

if (len > 0 && PyUnicode_READ_CHAR(filename, len - 1) == '>') {

return 0;

}

}

io = PyImport_ImportModule("io");

if (io == NULL) {

return -1;

}

binary = _PyObject_CallMethod(io, &_Py_ID(open), "Os", filename, "rb");

if (binary == NULL) {

PyErr_Clear();

binary = _Py_FindSourceFile(filename, buf, sizeof(buf), io);

if (binary == NULL) {

Py_DECREF(io);

return -1;

}

}

/* use the right encoding to decode the file as unicode */

fd = PyObject_AsFileDescriptor(binary);

if (fd < 0) {

Py_DECREF(io);

Py_DECREF(binary);

return 0;

}

found_encoding = _PyTokenizer_FindEncodingFilename(fd, filename);

if (found_encoding == NULL)

PyErr_Clear();

encoding = (found_encoding != NULL) ? found_encoding : "utf-8";

/* Reset position */

if (lseek(fd, 0, SEEK_SET) == (off_t)-1) {

Py_DECREF(io);

Py_DECREF(binary);

PyMem_Free(found_encoding);

return 0;

}

fob = _PyObject_CallMethod(io, &_Py_ID(TextIOWrapper),

"Os", binary, encoding);

Py_DECREF(io);

PyMem_Free(found_encoding);

if (fob == NULL) {

PyErr_Clear();

res = PyObject_CallMethodNoArgs(binary, &_Py_ID(close));

Py_DECREF(binary);

if (res)

Py_DECREF(res);

else

PyErr_Clear();

return 0;

}

Py_DECREF(binary);

/* get the line number lineno */

for (i = 0; i < lineno; i++) {

Py_XDECREF(lineobj);

lineobj = PyFile_GetLine(fob, -1);

if (!lineobj) {

PyErr_Clear();

break;

}

}

res = PyObject_CallMethodNoArgs(fob, &_Py_ID(close));

if (res) {

Py_DECREF(res);

}

else {

PyErr_Clear();

}

Py_DECREF(fob);

if (!lineobj || !PyUnicode_Check(lineobj)) {

Py_XDECREF(lineobj);

return -1;

}

if (line) {

*line = Py_NewRef(lineobj);

}

/* remove the indentation of the line */

kind = PyUnicode_KIND(lineobj);

data = PyUnicode_DATA(lineobj);

for (i=0; i < PyUnicode_GET_LENGTH(lineobj); i++) {

Py_UCS4 ch = PyUnicode_READ(kind, data, i);

if (ch != ' ' && ch != '\t' && ch != '\014')

break;

}

if (i) {

PyObject *truncated;

truncated = PyUnicode_Substring(lineobj, i, PyUnicode_GET_LENGTH(lineobj));

if (truncated) {

Py_SETREF(lineobj, truncated);

} else {

PyErr_Clear();

}

}

if (truncation != NULL) {

*truncation = i - indent;

}

if (_Py_WriteIndentedMargin(margin_indent, margin, f) < 0) {

goto error;

}

/* Write some spaces before the line */

if (_Py_WriteIndent(indent, f) < 0) {

goto error;

}

/* finally display the line */

if (PyFile_WriteObject(lineobj, f, Py_PRINT_RAW) < 0) {

goto error;

}

if (PyFile_WriteString("\n", f) < 0) {

goto error;

}

Py_DECREF(lineobj);

return 0;

error:

Py_DECREF(lineobj);

return -1;

}

_Py_DisplaySourceLine(PyObject *f, PyObject *filename, int lineno, int indent,

int *truncation, PyObject **line)

{

return display_source_line_with_margin(f, filename, lineno, indent, 0,

NULL, truncation, line);

}

#define IS_WHITESPACE(c) (((c) == ' ') || ((c) == '\t') || ((c) == '\f'))

static int

extract_anchors_from_expr(const char *segment_str, expr_ty expr, Py_ssize_t *left_anchor, Py_ssize_t *right_anchor,

char** primary_error_char, char** secondary_error_char)

{

switch (expr->kind) {

case BinOp_kind: {

expr_ty left = expr->v.BinOp.left;

expr_ty right = expr->v.BinOp.right;

for (int i = left->end_col_offset; i < right->col_offset; i++) {

if (IS_WHITESPACE(segment_str[i])) {

continue;

}

*left_anchor = i;

*right_anchor = i + 1;

// Check whether if this a two-character operator (e.g //)

if (i + 1 < right->col_offset && !IS_WHITESPACE(segment_str[i + 1])) {

++*right_anchor;

}

// Set the error characters

*primary_error_char = "~";

*secondary_error_char = "^";

break;

}

return 1;

}

case Subscript_kind: {

*left_anchor = expr->v.Subscript.value->end_col_offset;

*right_anchor = expr->v.Subscript.slice->end_col_offset + 1;

// Set the error characters

*primary_error_char = "~";

*secondary_error_char = "^";

return 1;

}

default:

return 0;

}

}

static int

extract_anchors_from_stmt(const char *segment_str, stmt_ty statement, Py_ssize_t *left_anchor, Py_ssize_t *right_anchor,

char** primary_error_char, char** secondary_error_char)

{

switch (statement->kind) {

case Expr_kind: {

return extract_anchors_from_expr(segment_str, statement->v.Expr.value, left_anchor, right_anchor,

primary_error_char, secondary_error_char);

}

default:

return 0;

}

}

static int

extract_anchors_from_line(PyObject *filename, PyObject *line,

Py_ssize_t start_offset, Py_ssize_t end_offset,

Py_ssize_t *left_anchor, Py_ssize_t *right_anchor,

char** primary_error_char, char** secondary_error_char)

{

int res = -1;

PyArena *arena = NULL;

PyObject *segment = PyUnicode_Substring(line, start_offset, end_offset);

if (!segment) {

goto done;

}

const char *segment_str = PyUnicode_AsUTF8(segment);

if (!segment_str) {

goto done;

}

arena = _PyArena_New();

if (!arena) {

goto done;

}

PyCompilerFlags flags = _PyCompilerFlags_INIT;

mod_ty module = _PyParser_ASTFromString(segment_str, filename, Py_file_input,

&flags, arena);

if (!module) {

goto done;

}

if (!_PyAST_Optimize(module, arena, _Py_GetConfig()->optimization_level, 0)) {

goto done;

}

assert(module->kind == Module_kind);

if (asdl_seq_LEN(module->v.Module.body) == 1) {

stmt_ty statement = asdl_seq_GET(module->v.Module.body, 0);

res = extract_anchors_from_stmt(segment_str, statement, left_anchor, right_anchor,

primary_error_char, secondary_error_char);

} else {

res = 0;

}

done:

if (res > 0) {

// Normalize the AST offsets to byte offsets and adjust them with the

// start of the actual line (instead of the source code segment).

assert(segment != NULL);

assert(*left_anchor >= 0);

assert(*right_anchor >= 0);

*left_anchor = _PyPegen_byte_offset_to_character_offset(segment, *left_anchor) + start_offset;

*right_anchor = _PyPegen_byte_offset_to_character_offset(segment, *right_anchor) + start_offset;

}

Py_XDECREF(segment);

if (arena) {

_PyArena_Free(arena);

}

return res;

}

#define _TRACEBACK_SOURCE_LINE_INDENT 4

static inline int

ignore_source_errors(void) {

if (PyErr_Occurred()) {

if (PyErr_ExceptionMatches(PyExc_KeyboardInterrupt)) {

return -1;

}

PyErr_Clear();

}

return 0;

}

static inline int

print_error_location_carets(PyObject *f, int offset, Py_ssize_t start_offset, Py_ssize_t end_offset,

Py_ssize_t right_start_offset, Py_ssize_t left_end_offset,

const char *primary, const char *secondary) {

int special_chars = (left_end_offset != -1 || right_start_offset != -1);

const char *str;

while (++offset <= end_offset) {

if (offset <= start_offset) {

str = " ";

} else if (special_chars && left_end_offset < offset && offset <= right_start_offset) {

str = secondary;

} else {

str = primary;

}

if (PyFile_WriteString(str, f) < 0) {

return -1;

}

}

if (PyFile_WriteString("\n", f) < 0) {

return -1;

}

return 0;

}

static int

tb_displayline(PyTracebackObject* tb, PyObject *f, PyObject *filename, int lineno,

PyFrameObject *frame, PyObject *name, int margin_indent, const char *margin)

{

if (filename == NULL || name == NULL) {

return -1;

}

if (_Py_WriteIndentedMargin(margin_indent, margin, f) < 0) {

return -1;

}

PyObject *line = PyUnicode_FromFormat(" File \"%U\", line %d, in %U\n",

filename, lineno, name);

if (line == NULL) {

return -1;

}

int res = PyFile_WriteObject(line, f, Py_PRINT_RAW);

Py_DECREF(line);

if (res < 0) {

return -1;

}

int err = 0;

int truncation = _TRACEBACK_SOURCE_LINE_INDENT;

PyObject* source_line = NULL;

int rc = display_source_line_with_margin(

f, filename, lineno, _TRACEBACK_SOURCE_LINE_INDENT,

margin_indent, margin, &truncation, &source_line);

if (rc != 0 || !source_line) {

/* ignore errors since we can't report them, can we? */

err = ignore_source_errors();

goto done;

}

int code_offset = tb->tb_lasti;

PyCodeObject* code = _PyFrame_GetCode(frame->f_frame);

const Py_ssize_t source_line_len = PyUnicode_GET_LENGTH(source_line);

int start_line;

int end_line;

int start_col_byte_offset;

int end_col_byte_offset;

if (!PyCode_Addr2Location(code, code_offset, &start_line, &start_col_byte_offset,

&end_line, &end_col_byte_offset)) {

goto done;

}

if (start_line < 0 || end_line < 0

|| start_col_byte_offset < 0

|| end_col_byte_offset < 0)

{

goto done;

}

// When displaying errors, we will use the following generic structure:

//

// ERROR LINE ERROR LINE ERROR LINE ERROR LINE ERROR LINE ERROR LINE ERROR LINE

// ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^~~~~~~~~~~~~~~~~~~~

// | |-> left_end_offset | |-> end_offset

// |-> start_offset |-> right_start_offset

//

// In general we will only have (start_offset, end_offset) but we can gather more information

// by analyzing the AST of the text between *start_offset* and *end_offset*. If this succeeds

// we could get *left_end_offset* and *right_start_offset* and some selection of characters for

// the different ranges (primary_error_char and secondary_error_char). If we cannot obtain the

// AST information or we cannot identify special ranges within it, then left_end_offset and

// right_end_offset will be set to -1.

//

// To keep the column indicators pertinent, they are not shown when the primary character

// spans the whole line.

// Convert the utf-8 byte offset to the actual character offset so we print the right number of carets.

assert(source_line);

Py_ssize_t start_offset = _PyPegen_byte_offset_to_character_offset(source_line, start_col_byte_offset);

if (start_offset < 0) {

err = ignore_source_errors() < 0;

goto done;

}

Py_ssize_t end_offset = _PyPegen_byte_offset_to_character_offset(source_line, end_col_byte_offset);

if (end_offset < 0) {

err = ignore_source_errors() < 0;

goto done;

}

Py_ssize_t left_end_offset = -1;

Py_ssize_t right_start_offset = -1;

char *primary_error_char = "^";

char *secondary_error_char = primary_error_char;

if (start_line == end_line) {

int res = extract_anchors_from_line(filename, source_line, start_offset, end_offset,

&left_end_offset, &right_start_offset,

&primary_error_char, &secondary_error_char);

if (res < 0 && ignore_source_errors() < 0) {

goto done;

}

}

else {

// If this is a multi-line expression, then we will highlight until

// the last non-whitespace character.

const char *source_line_str = PyUnicode_AsUTF8(source_line);

if (!source_line_str) {

goto done;

}

Py_ssize_t i = source_line_len;

while (--i >= 0) {

if (!IS_WHITESPACE(source_line_str[i])) {

break;

}

}

end_offset = i + 1;

}

// Elide indicators if primary char spans the frame line

Py_ssize_t stripped_line_len = source_line_len - truncation - _TRACEBACK_SOURCE_LINE_INDENT;

bool has_secondary_ranges = (left_end_offset != -1 || right_start_offset != -1);

if (end_offset - start_offset == stripped_line_len && !has_secondary_ranges) {

goto done;

}

if (_Py_WriteIndentedMargin(margin_indent, margin, f) < 0) {

err = -1;

goto done;

}

if (print_error_location_carets(f, truncation, start_offset, end_offset,

right_start_offset, left_end_offset,

primary_error_char, secondary_error_char) < 0) {

err = -1;

goto done;

}

done:

Py_XDECREF(source_line);

return err;

}

static const int TB_RECURSIVE_CUTOFF = 3; // Also hardcoded in traceback.py.

static int

tb_print_line_repeated(PyObject *f, long cnt)

{

cnt -= TB_RECURSIVE_CUTOFF;

PyObject *line = PyUnicode_FromFormat(

(cnt > 1)

? " [Previous line repeated %ld more times]\n"

: " [Previous line repeated %ld more time]\n",

cnt);

if (line == NULL) {

return -1;

}

int err = PyFile_WriteObject(line, f, Py_PRINT_RAW);

Py_DECREF(line);

return err;

}

static int

tb_printinternal(PyTracebackObject *tb, PyObject *f, long limit,

int indent, const char *margin)

{

PyCodeObject *code = NULL;

Py_ssize_t depth = 0;

PyObject *last_file = NULL;

int last_line = -1;

PyObject *last_name = NULL;

long cnt = 0;

PyTracebackObject *tb1 = tb;

while (tb1 != NULL) {

depth++;

tb1 = tb1->tb_next;

}

while (tb != NULL && depth > limit) {

depth--;

tb = tb->tb_next;

}

while (tb != NULL) {

code = PyFrame_GetCode(tb->tb_frame);

if (last_file == NULL ||

code->co_filename != last_file ||

last_line == -1 || tb->tb_lineno != last_line ||

last_name == NULL || code->co_name != last_name) {

if (cnt > TB_RECURSIVE_CUTOFF) {

if (tb_print_line_repeated(f, cnt) < 0) {

goto error;

}

}

last_file = code->co_filename;

last_line = tb->tb_lineno;

last_name = code->co_name;

cnt = 0;

}

cnt++;

if (cnt <= TB_RECURSIVE_CUTOFF) {

if (tb_displayline(tb, f, code->co_filename, tb->tb_lineno,

tb->tb_frame, code->co_name, indent, margin) < 0) {

goto error;

}

if (PyErr_CheckSignals() < 0) {

goto error;

}

}

Py_CLEAR(code);

tb = tb->tb_next;

}

if (cnt > TB_RECURSIVE_CUTOFF) {

if (tb_print_line_repeated(f, cnt) < 0) {

goto error;

}

}

return 0;

error:

Py_XDECREF(code);

return -1;

}

#define PyTraceBack_LIMIT 1000

_PyTraceBack_Print_Indented(PyObject *v, int indent, const char *margin,

const char *header_margin, const char *header, PyObject *f)

{

PyObject *limitv;

long limit = PyTraceBack_LIMIT;

if (v == NULL) {

return 0;

}

if (!PyTraceBack_Check(v)) {

PyErr_BadInternalCall();

return -1;

}

limitv = PySys_GetObject("tracebacklimit");

if (limitv && PyLong_Check(limitv)) {

int overflow;

limit = PyLong_AsLongAndOverflow(limitv, &overflow);

if (overflow > 0) {

limit = LONG_MAX;

}

else if (limit <= 0) {

return 0;

}

}

if (_Py_WriteIndentedMargin(indent, header_margin, f) < 0) {