cpython/Python/traceback.c

282 lines
6.5 KiB
C
Raw Normal View History

1991-02-19 08:39:46 -04:00
1990-12-20 11:06:42 -04:00
/* Traceback implementation */
1997-04-29 15:33:38 -03:00
#include "Python.h"
1990-12-20 11:06:42 -04:00
#include "compile.h"
#include "frameobject.h"
#include "structmember.h"
#include "osdefs.h"
1990-12-20 11:06:42 -04:00
typedef struct _tracebackobject {
1997-04-29 15:33:38 -03:00
PyObject_HEAD
1990-12-20 11:06:42 -04:00
struct _tracebackobject *tb_next;
1997-04-29 15:33:38 -03:00
PyFrameObject *tb_frame;
1990-12-20 11:06:42 -04:00
int tb_lasti;
int tb_lineno;
} tracebackobject;
#define OFF(x) offsetof(tracebackobject, x)
static struct memberlist tb_memberlist[] = {
{"tb_next", T_OBJECT, OFF(tb_next)},
{"tb_frame", T_OBJECT, OFF(tb_frame)},
{"tb_lasti", T_INT, OFF(tb_lasti)},
{"tb_lineno", T_INT, OFF(tb_lineno)},
{NULL} /* Sentinel */
};
1997-04-29 15:33:38 -03:00
static PyObject *
tb_getattr(tracebackobject *tb, char *name)
1990-12-20 11:06:42 -04:00
{
1997-04-29 15:33:38 -03:00
return PyMember_Get((char *)tb, tb_memberlist, name);
1990-12-20 11:06:42 -04:00
}
static void
tb_dealloc(tracebackobject *tb)
1990-12-20 11:06:42 -04:00
{
Py_TRASHCAN_SAFE_BEGIN(tb)
_PyObject_GC_UNTRACK(tb);
1997-04-29 15:33:38 -03:00
Py_XDECREF(tb->tb_next);
Py_XDECREF(tb->tb_frame);
PyObject_GC_Del(tb);
Py_TRASHCAN_SAFE_END(tb)
1990-12-20 11:06:42 -04:00
}
static int
tb_traverse(tracebackobject *tb, visitproc visit, void *arg)
{
int err = 0;
if (tb->tb_next) {
err = visit((PyObject *)tb->tb_next, arg);
if (err)
return err;
}
if (tb->tb_frame)
err = visit((PyObject *)tb->tb_frame, arg);
return err;
}
static void
tb_clear(tracebackobject *tb)
{
Py_XDECREF(tb->tb_next);
Py_XDECREF(tb->tb_frame);
tb->tb_next = NULL;
tb->tb_frame = NULL;
}
PyTypeObject PyTraceBack_Type = {
1997-04-29 15:33:38 -03:00
PyObject_HEAD_INIT(&PyType_Type)
1990-12-20 11:06:42 -04:00
0,
"traceback",
sizeof(tracebackobject),
0,
(destructor)tb_dealloc, /*tp_dealloc*/
1990-12-20 11:06:42 -04:00
0, /*tp_print*/
(getattrfunc)tb_getattr, /*tp_getattr*/
1990-12-20 11:06:42 -04:00
0, /*tp_setattr*/
0, /*tp_compare*/
0, /*tp_repr*/
0, /*tp_as_number*/
0, /*tp_as_sequence*/
0, /*tp_as_mapping*/
0, /* tp_hash */
0, /* tp_call */
0, /* tp_str */
0, /* tp_getattro */
0, /* tp_setattro */
0, /* tp_as_buffer */
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
0, /* tp_doc */
(traverseproc)tb_traverse, /* tp_traverse */
(inquiry)tb_clear, /* tp_clear */
0, /* tp_richcompare */
0, /* tp_weaklistoffset */
0, /* tp_iter */
0, /* tp_iternext */
0, /* tp_methods */
0, /* tp_members */
0, /* tp_getset */
0, /* tp_base */
0, /* tp_dict */
1990-12-20 11:06:42 -04:00
};
static tracebackobject *
newtracebackobject(tracebackobject *next, PyFrameObject *frame, int lasti,
int lineno)
1990-12-20 11:06:42 -04:00
{
tracebackobject *tb;
if ((next != NULL && !PyTraceBack_Check(next)) ||
1997-04-29 15:33:38 -03:00
frame == NULL || !PyFrame_Check(frame)) {
PyErr_BadInternalCall();
1990-12-20 11:06:42 -04:00
return NULL;
}
tb = PyObject_GC_New(tracebackobject, &PyTraceBack_Type);
1990-12-20 11:06:42 -04:00
if (tb != NULL) {
1997-04-29 15:33:38 -03:00
Py_XINCREF(next);
1990-12-20 11:06:42 -04:00
tb->tb_next = next;
1997-04-29 15:33:38 -03:00
Py_XINCREF(frame);
1990-12-20 11:06:42 -04:00
tb->tb_frame = frame;
tb->tb_lasti = lasti;
tb->tb_lineno = lineno;
_PyObject_GC_TRACK(tb);
1990-12-20 11:06:42 -04:00
}
return tb;
}
int
PyTraceBack_Here(PyFrameObject *frame)
1990-12-20 11:06:42 -04:00
{
PyThreadState *tstate = frame->f_tstate;
tracebackobject *oldtb = (tracebackobject *) tstate->curexc_traceback;
tracebackobject *tb = newtracebackobject(oldtb,
frame, frame->f_lasti, frame->f_lineno);
1990-12-20 11:06:42 -04:00
if (tb == NULL)
return -1;
tstate->curexc_traceback = (PyObject *)tb;
Py_XDECREF(oldtb);
1990-12-20 11:06:42 -04:00
return 0;
}
static int
tb_displayline(PyObject *f, char *filename, int lineno, char *name)
1990-12-20 11:06:42 -04:00
{
int err = 0;
1990-12-20 11:06:42 -04:00
FILE *xfp;
char linebuf[2000];
1990-12-20 11:06:42 -04:00
int i;
if (filename == NULL || name == NULL)
return -1;
#ifdef MPW
/* This is needed by MPW's File and Line commands */
#define FMT " File \"%.500s\"; line %d # in %.500s\n"
#else
/* This is needed by Emacs' compile command */
#define FMT " File \"%.500s\", line %d, in %.500s\n"
#endif
1990-12-20 11:06:42 -04:00
xfp = fopen(filename, "r");
if (xfp == NULL) {
/* Search tail of filename in sys.path before giving up */
1997-04-29 15:33:38 -03:00
PyObject *path;
char *tail = strrchr(filename, SEP);
if (tail == NULL)
tail = filename;
else
tail++;
1997-04-29 15:33:38 -03:00
path = PySys_GetObject("path");
if (path != NULL && PyList_Check(path)) {
int npath = PyList_Size(path);
size_t taillen = strlen(tail);
char namebuf[MAXPATHLEN+1];
for (i = 0; i < npath; i++) {
1997-04-29 15:33:38 -03:00
PyObject *v = PyList_GetItem(path, i);
if (v == NULL) {
PyErr_Clear();
break;
}
1997-04-29 15:33:38 -03:00
if (PyString_Check(v)) {
size_t len;
1997-04-29 15:33:38 -03:00
len = PyString_Size(v);
if (len + 1 + taillen >= MAXPATHLEN)
continue; /* Too long */
1997-04-29 15:33:38 -03:00
strcpy(namebuf, PyString_AsString(v));
if (strlen(namebuf) != len)
continue; /* v contains '\0' */
if (len > 0 && namebuf[len-1] != SEP)
namebuf[len++] = SEP;
strcpy(namebuf+len, tail);
xfp = fopen(namebuf, "r");
if (xfp != NULL) {
filename = namebuf;
break;
}
}
}
}
1990-12-20 11:06:42 -04:00
}
2001-11-28 16:42:20 -04:00
PyOS_snprintf(linebuf, sizeof(linebuf), FMT, filename, lineno, name);
err = PyFile_WriteString(linebuf, f);
if (xfp == NULL || err != 0)
return err;
1990-12-20 11:06:42 -04:00
for (i = 0; i < lineno; i++) {
char* pLastChar = &linebuf[sizeof(linebuf)-2];
do {
*pLastChar = '\0';
if (fgets(linebuf, sizeof linebuf, xfp) == NULL)
break;
/* fgets read *something*; if it didn't get as
far as pLastChar, it must have found a newline
or hit the end of the file; if pLastChar is \n,
it obviously found a newline; else we haven't
yet seen a newline, so must continue */
} while (*pLastChar != '\0' && *pLastChar != '\n');
1990-12-20 11:06:42 -04:00
}
if (i == lineno) {
char *p = linebuf;
while (*p == ' ' || *p == '\t' || *p == '\014')
1990-12-20 11:06:42 -04:00
p++;
err = PyFile_WriteString(" ", f);
if (err == 0) {
err = PyFile_WriteString(p, f);
if (err == 0 && strchr(p, '\n') == NULL)
err = PyFile_WriteString("\n", f);
}
1990-12-20 11:06:42 -04:00
}
fclose(xfp);
return err;
1990-12-20 11:06:42 -04:00
}
static int
tb_printinternal(tracebackobject *tb, PyObject *f, int limit)
1990-12-20 11:06:42 -04:00
{
int err = 0;
int depth = 0;
tracebackobject *tb1 = tb;
while (tb1 != NULL) {
depth++;
tb1 = tb1->tb_next;
}
while (tb != NULL && err == 0) {
if (depth <= limit) {
if (Py_OptimizeFlag)
tb->tb_lineno = PyCode_Addr2Line(
tb->tb_frame->f_code, tb->tb_lasti);
err = tb_displayline(f,
1997-04-29 15:33:38 -03:00
PyString_AsString(
tb->tb_frame->f_code->co_filename),
tb->tb_lineno,
1997-04-29 15:33:38 -03:00
PyString_AsString(tb->tb_frame->f_code->co_name));
}
depth--;
1990-12-20 11:06:42 -04:00
tb = tb->tb_next;
if (err == 0)
err = PyErr_CheckSignals();
1990-12-20 11:06:42 -04:00
}
return err;
1990-12-20 11:06:42 -04:00
}
int
PyTraceBack_Print(PyObject *v, PyObject *f)
1990-12-20 11:06:42 -04:00
{
int err;
1997-04-29 15:33:38 -03:00
PyObject *limitv;
int limit = 1000;
1990-12-20 11:06:42 -04:00
if (v == NULL)
return 0;
if (!PyTraceBack_Check(v)) {
1997-04-29 15:33:38 -03:00
PyErr_BadInternalCall();
1990-12-20 11:06:42 -04:00
return -1;
}
1997-04-29 15:33:38 -03:00
limitv = PySys_GetObject("tracebacklimit");
if (limitv && PyInt_Check(limitv)) {
limit = PyInt_AsLong(limitv);
if (limit <= 0)
return 0;
}
err = PyFile_WriteString("Traceback (most recent call last):\n", f);
if (!err)
err = tb_printinternal((tracebackobject *)v, f, limit);
return err;
1990-12-20 11:06:42 -04:00
}