plpy_elog.c File Reference

#include "postgres.h"
#include "lib/stringinfo.h"
#include "plpy_elog.h"
#include "plpy_main.h"
#include "plpy_procedure.h"
#include "plpy_util.h"

Include dependency graph for plpy_elog.c:

Go to the source code of this file.

Functions
static void	PLy_traceback (PyObject *e, PyObject *v, PyObject *tb, char *volatile *xmsg, char *volatile *tbmsg, int *tb_depth)
static void	PLy_get_spi_error_data (PyObject *exc, int *sqlerrcode, char **detail, char **hint, char **query, int *position, char **schema_name, char **table_name, char **column_name, char **datatype_name, char **constraint_name)
static void	PLy_get_error_data (PyObject *exc, int *sqlerrcode, char **detail, char **hint, char **schema_name, char **table_name, char **column_name, char **datatype_name, char **constraint_name)
static char *	get_source_line (const char *src, int lineno)
static void	get_string_attr (PyObject *obj, char *attrname, char **str)
static bool	set_string_attr (PyObject *obj, char *attrname, char *str)
void	PLy_elog_impl (int elevel, const char *fmt,...)
static void	PLy_get_sqlerrcode (PyObject *exc, int *sqlerrcode)
void	PLy_exception_set (PyObject *exc, const char *fmt,...)
void	PLy_exception_set_plural (PyObject *exc, const char *fmt_singular, const char *fmt_plural, unsigned long n,...)
void	PLy_exception_set_with_details (PyObject *excclass, ErrorData *edata)

Variables
PyObject *	PLy_exc_error = NULL
PyObject *	PLy_exc_fatal = NULL
PyObject *	PLy_exc_spi_error = NULL

Function Documentation

get_source_line()

static char * get_source_line ( const char *  src, int  lineno  )

static

Definition at line 448 of file plpy_elog.c.

{
    const char *s = NULL;
    const char *next = src;
    int         current = 0;
 
    /* sanity check */
    if (lineno <= 0)
        return NULL;
 
    while (current < lineno)
    {
        s = next;
        next = strchr(s + 1, '\n');
        current++;
        if (next == NULL)
            break;
    }
 
    if (current != lineno)
        return NULL;
 
    while (*s && isspace((unsigned char) *s))
        s++;
 
    if (next == NULL)
        return pstrdup(s);
 
    /*
     * Sanity check, next < s if the line was all-whitespace, which should
     * never happen if Python reported a frame created on that line, but check
     * anyway.
     */
    if (next < s)
        return NULL;
 
    return pnstrdup(s, next - s);
}

References fb(), next, pnstrdup(), and pstrdup().

Referenced by PLy_traceback().

get_string_attr()

static void get_string_attr ( PyObject *  obj, char *  attrname, char **  str  )

static

Definition at line 580 of file plpy_elog.c.

{
    PyObject   *val;
 
    val = PyObject_GetAttrString(obj, attrname);
    if (val != NULL && val != Py_None)
    {
        *str = PLyUnicode_AsString(val);
    }
    Py_XDECREF(val);
}

References fb(), PLyUnicode_AsString(), str, and val.

Referenced by PLy_get_error_data().

PLy_elog_impl()

void PLy_elog_impl	(	int	elevel,
		const char *	fmt,
			...
	)

Definition at line 44 of file plpy_elog.c.

{
    int         save_errno = errno;
    char       *volatile xmsg = NULL;
    char       *volatile tbmsg = NULL;
    int         tb_depth;
    StringInfoData emsg;
    PyObject   *exc,
               *val,
               *tb;
 
    /* If we'll need emsg, must initialize it before entering PG_TRY */
    if (fmt)
        initStringInfo(&emsg);
 
    PyErr_Fetch(&exc, &val, &tb);
 
    /* Use a PG_TRY block to ensure we release the PyObjects just acquired */
    PG_TRY();
    {
        const char *primary = NULL;
        int         sqlerrcode = 0;
        char       *detail = NULL;
        char       *hint = NULL;
        char       *query = NULL;
        int         position = 0;
        char       *schema_name = NULL;
        char       *table_name = NULL;
        char       *column_name = NULL;
        char       *datatype_name = NULL;
        char       *constraint_name = NULL;
 
        if (exc != NULL)
        {
            PyErr_NormalizeException(&exc, &val, &tb);
 
            if (PyErr_GivenExceptionMatches(val, PLy_exc_spi_error))
                PLy_get_spi_error_data(val, &sqlerrcode,
                                       &detail, &hint, &query, &position,
                                       &schema_name, &table_name, &column_name,
                                       &datatype_name, &constraint_name);
            else if (PyErr_GivenExceptionMatches(val, PLy_exc_error))
                PLy_get_error_data(val, &sqlerrcode, &detail, &hint,
                                   &schema_name, &table_name, &column_name,
                                   &datatype_name, &constraint_name);
            else if (PyErr_GivenExceptionMatches(val, PLy_exc_fatal))
                elevel = FATAL;
        }
 
        PLy_traceback(exc, val, tb,
                      &xmsg, &tbmsg, &tb_depth);
 
        if (fmt)
        {
            for (;;)
            {
                va_list     ap;
                int         needed;
 
                errno = save_errno;
                va_start(ap, fmt);
                needed = appendStringInfoVA(&emsg, dgettext(TEXTDOMAIN, fmt), ap);
                va_end(ap);
                if (needed == 0)
                    break;
                enlargeStringInfo(&emsg, needed);
            }
            primary = emsg.data;
 
            /* If there's an exception message, it goes in the detail. */
            if (xmsg)
                detail = xmsg;
        }
        else
        {
            if (xmsg)
                primary = xmsg;
        }
 
        ereport(elevel,
                (errcode(sqlerrcode ? sqlerrcode : ERRCODE_EXTERNAL_ROUTINE_EXCEPTION),
                 errmsg_internal("%s", primary ? primary : "no exception data"),
                 (detail) ? errdetail_internal("%s", detail) : 0,
                 (tb_depth > 0 && tbmsg) ? errcontext("%s", tbmsg) : 0,
                 (hint) ? errhint("%s", hint) : 0,
                 (query) ? internalerrquery(query) : 0,
                 (position) ? internalerrposition(position) : 0,
                 (schema_name) ? err_generic_string(PG_DIAG_SCHEMA_NAME,
                                                    schema_name) : 0,
                 (table_name) ? err_generic_string(PG_DIAG_TABLE_NAME,
                                                   table_name) : 0,
                 (column_name) ? err_generic_string(PG_DIAG_COLUMN_NAME,
                                                    column_name) : 0,
                 (datatype_name) ? err_generic_string(PG_DIAG_DATATYPE_NAME,
                                                      datatype_name) : 0,
                 (constraint_name) ? err_generic_string(PG_DIAG_CONSTRAINT_NAME,
                                                        constraint_name) : 0));
    }
    PG_FINALLY();
    {
        Py_XDECREF(exc);
        Py_XDECREF(val);
        Py_XDECREF(tb);
        /* For neatness' sake, also release our string buffers */
        if (fmt)
            pfree(emsg.data);
        if (xmsg)
            pfree(xmsg);
        if (tbmsg)
            pfree(tbmsg);
    }
    PG_END_TRY();
}

References appendStringInfoVA(), dgettext, enlargeStringInfo(), ereport, err_generic_string(), errcode(), errcontext, errdetail_internal(), errhint(), errmsg_internal(), FATAL, fb(), initStringInfo(), internalerrposition(), internalerrquery(), pfree(), PG_DIAG_COLUMN_NAME, PG_DIAG_CONSTRAINT_NAME, PG_DIAG_DATATYPE_NAME, PG_DIAG_SCHEMA_NAME, PG_DIAG_TABLE_NAME, PG_END_TRY, PG_FINALLY, PG_TRY, PLy_exc_error, PLy_exc_fatal, PLy_exc_spi_error, PLy_get_error_data(), PLy_get_spi_error_data(), PLy_traceback(), TEXTDOMAIN, and val.

Referenced by _PG_init().

PLy_exception_set()

void PLy_exception_set	(	PyObject *	exc,
		const char *	fmt,
			...
	)

Definition at line 490 of file plpy_elog.c.

{
    char        buf[1024];
    va_list     ap;
 
    va_start(ap, fmt);
    vsnprintf(buf, sizeof(buf), dgettext(TEXTDOMAIN, fmt), ap);
    va_end(ap);
 
    PyErr_SetString(exc, buf);
}

References buf, dgettext, fb(), TEXTDOMAIN, and vsnprintf.

Referenced by PLy_cursor(), PLy_cursor_close(), PLy_cursor_fetch(), PLy_cursor_iternext(), PLy_cursor_plan(), PLy_output(), PLy_result_colnames(), PLy_result_coltypes(), PLy_result_coltypmods(), PLy_spi_execute(), PLy_spi_execute_plan(), PLy_spi_execute_query(), PLy_spi_prepare(), PLy_subtransaction_enter(), and PLy_subtransaction_exit().

PLy_exception_set_plural()

void PLy_exception_set_plural	(	PyObject *	exc,
		const char *	fmt_singular,
		const char *	fmt_plural,
		unsigned long	n,
			...
	)

Definition at line 504 of file plpy_elog.c.

{
    char        buf[1024];
    va_list     ap;
 
    va_start(ap, n);
    vsnprintf(buf, sizeof(buf),
              dngettext(TEXTDOMAIN, fmt_singular, fmt_plural, n),
              ap);
    va_end(ap);
 
    PyErr_SetString(exc, buf);
}

References buf, dngettext, fb(), TEXTDOMAIN, and vsnprintf.

Referenced by PLy_cursor_plan(), and PLy_spi_execute_plan().

PLy_exception_set_with_details()

void PLy_exception_set_with_details	(	PyObject *	excclass,
		ErrorData *	edata
	)

Definition at line 522 of file plpy_elog.c.

{
    PyObject   *args = NULL;
    PyObject   *error = NULL;
 
    args = Py_BuildValue("(s)", edata->message);
    if (!args)
        goto failure;
 
    /* create a new exception with the error message as the parameter */
    error = PyObject_CallObject(excclass, args);
    if (!error)
        goto failure;
 
    if (!set_string_attr(error, "sqlstate",
                         unpack_sql_state(edata->sqlerrcode)))
        goto failure;
 
    if (!set_string_attr(error, "detail", edata->detail))
        goto failure;
 
    if (!set_string_attr(error, "hint", edata->hint))
        goto failure;
 
    if (!set_string_attr(error, "query", edata->internalquery))
        goto failure;
 
    if (!set_string_attr(error, "schema_name", edata->schema_name))
        goto failure;
 
    if (!set_string_attr(error, "table_name", edata->table_name))
        goto failure;
 
    if (!set_string_attr(error, "column_name", edata->column_name))
        goto failure;
 
    if (!set_string_attr(error, "datatype_name", edata->datatype_name))
        goto failure;
 
    if (!set_string_attr(error, "constraint_name", edata->constraint_name))
        goto failure;
 
    PyErr_SetObject(excclass, error);
 
    Py_DECREF(args);
    Py_DECREF(error);
 
    return;
 
failure:
    Py_XDECREF(args);
    Py_XDECREF(error);
 
    elog(ERROR, "could not convert error to Python exception");
}

References elog, ERROR, error(), fb(), set_string_attr(), and unpack_sql_state().

Referenced by PLy_output().

PLy_get_error_data()

static void PLy_get_error_data ( PyObject *  exc, int *  sqlerrcode, char **  detail, char **  hint, char **  schema_name, char **  table_name, char **  column_name, char **  datatype_name, char **  constraint_name  )

static

Definition at line 430 of file plpy_elog.c.

{
    PLy_get_sqlerrcode(exc, sqlerrcode);
    get_string_attr(exc, "detail", detail);
    get_string_attr(exc, "hint", hint);
    get_string_attr(exc, "schema_name", schema_name);
    get_string_attr(exc, "table_name", table_name);
    get_string_attr(exc, "column_name", column_name);
    get_string_attr(exc, "datatype_name", datatype_name);
    get_string_attr(exc, "constraint_name", constraint_name);
}

References get_string_attr(), and PLy_get_sqlerrcode().

Referenced by PLy_elog_impl().

PLy_get_spi_error_data()

static void PLy_get_spi_error_data ( PyObject *  exc, int *  sqlerrcode, char **  detail, char **  hint, char **  query, int *  position, char **  schema_name, char **  table_name, char **  column_name, char **  datatype_name, char **  constraint_name  )

static

Definition at line 389 of file plpy_elog.c.

{
    PyObject   *spidata;
 
    spidata = PyObject_GetAttrString(exc, "spidata");
 
    if (spidata != NULL)
    {
        PyArg_ParseTuple(spidata, "izzzizzzzz",
                         sqlerrcode, detail, hint, query, position,
                         schema_name, table_name, column_name,
                         datatype_name, constraint_name);
    }
    else
    {
        /*
         * If there's no spidata, at least set the sqlerrcode. This can happen
         * if someone explicitly raises a SPI exception from Python code.
         */
        PLy_get_sqlerrcode(exc, sqlerrcode);
    }
 
    Py_XDECREF(spidata);
}

References fb(), and PLy_get_sqlerrcode().

Referenced by PLy_elog_impl().

PLy_get_sqlerrcode()

static void PLy_get_sqlerrcode ( PyObject *  exc, int *  sqlerrcode  )

static

Definition at line 361 of file plpy_elog.c.

{
    PyObject   *sqlstate;
    char       *buffer;
 
    sqlstate = PyObject_GetAttrString(exc, "sqlstate");
    if (sqlstate == NULL)
        return;
 
    buffer = PLyUnicode_AsString(sqlstate);
    if (strlen(buffer) == 5 &&
        strspn(buffer, "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ") == 5)
    {
        *sqlerrcode = MAKE_SQLSTATE(buffer[0], buffer[1], buffer[2],
                                    buffer[3], buffer[4]);
    }
 
    Py_DECREF(sqlstate);
}

References fb(), MAKE_SQLSTATE, and PLyUnicode_AsString().

Referenced by PLy_get_error_data(), and PLy_get_spi_error_data().

PLy_traceback()

static void PLy_traceback ( PyObject *  e, PyObject *  v, PyObject *  tb, char *volatile *  xmsg, char *volatile *  tbmsg, int *  tb_depth  )

static

Definition at line 166 of file plpy_elog.c.

{
    PyObject   *volatile e_type_o = NULL;
    PyObject   *volatile e_module_o = NULL;
    PyObject   *volatile vob = NULL;
    StringInfoData tbstr;
 
    /*
     * if no exception, return nulls
     */
    if (e == NULL)
    {
        *xmsg = NULL;
        *tbmsg = NULL;
        *tb_depth = 0;
 
        return;
    }
 
    /*
     * Format the exception and its value and put it in xmsg.
     */
    PG_TRY();
    {
        char       *e_type_s = NULL;
        char       *e_module_s = NULL;
        const char *vstr;
        StringInfoData xstr;
 
        e_type_o = PyObject_GetAttrString(e, "__name__");
        e_module_o = PyObject_GetAttrString(e, "__module__");
        if (e_type_o)
            e_type_s = PLyUnicode_AsString(e_type_o);
        if (e_module_o)
            e_module_s = PLyUnicode_AsString(e_module_o);
 
        if (v && ((vob = PyObject_Str(v)) != NULL))
            vstr = PLyUnicode_AsString(vob);
        else
            vstr = "unknown";
 
        initStringInfo(&xstr);
        if (!e_type_s || !e_module_s)
        {
            /* shouldn't happen */
            appendStringInfoString(&xstr, "unrecognized exception");
        }
        /* mimics behavior of traceback.format_exception_only */
        else if (strcmp(e_module_s, "builtins") == 0
                 || strcmp(e_module_s, "__main__") == 0
                 || strcmp(e_module_s, "exceptions") == 0)
            appendStringInfoString(&xstr, e_type_s);
        else
            appendStringInfo(&xstr, "%s.%s", e_module_s, e_type_s);
        appendStringInfo(&xstr, ": %s", vstr);
 
        *xmsg = xstr.data;
    }
    PG_FINALLY();
    {
        Py_XDECREF(e_type_o);
        Py_XDECREF(e_module_o);
        Py_XDECREF(vob);
    }
    PG_END_TRY();
 
    /*
     * Now format the traceback and put it in tbmsg.
     */
    *tb_depth = 0;
    initStringInfo(&tbstr);
    /* Mimic Python traceback reporting as close as possible. */
    appendStringInfoString(&tbstr, "Traceback (most recent call last):");
    while (tb != NULL && tb != Py_None)
    {
        PyObject   *volatile frame = NULL;
        PyObject   *volatile code = NULL;
        PyObject   *volatile name = NULL;
        PyObject   *volatile lineno = NULL;
        PyObject   *volatile filename = NULL;
 
        PG_TRY();
        {
            lineno = PyObject_GetAttrString(tb, "tb_lineno");
            if (lineno == NULL)
                elog(ERROR, "could not get line number from Python traceback");
 
            frame = PyObject_GetAttrString(tb, "tb_frame");
            if (frame == NULL)
                elog(ERROR, "could not get frame from Python traceback");
 
            code = PyObject_GetAttrString(frame, "f_code");
            if (code == NULL)
                elog(ERROR, "could not get code object from Python frame");
 
            name = PyObject_GetAttrString(code, "co_name");
            if (name == NULL)
                elog(ERROR, "could not get function name from Python code object");
 
            filename = PyObject_GetAttrString(code, "co_filename");
            if (filename == NULL)
                elog(ERROR, "could not get file name from Python code object");
 
            /* The first frame always points at <module>, skip it. */
            if (*tb_depth > 0)
            {
                PLyExecutionContext *exec_ctx = PLy_current_execution_context();
                char       *proname;
                char       *fname;
                char       *line;
                char       *plain_filename;
                long        plain_lineno;
 
                /*
                 * The second frame points at the internal function, but to
                 * mimic Python error reporting we want to say <module>.
                 */
                if (*tb_depth == 1)
                    fname = "<module>";
                else
                    fname = PLyUnicode_AsString(name);
 
                proname = PLy_procedure_name(exec_ctx->curr_proc);
                plain_filename = PLyUnicode_AsString(filename);
                plain_lineno = PyLong_AsLong(lineno);
 
                if (proname == NULL)
                    appendStringInfo(&tbstr, "\n  PL/Python anonymous code block, line %ld, in %s",
                                     plain_lineno - 1, fname);
                else
                    appendStringInfo(&tbstr, "\n  PL/Python function \"%s\", line %ld, in %s",
                                     proname, plain_lineno - 1, fname);
 
                /*
                 * function code object was compiled with "<string>" as the
                 * filename
                 */
                if (exec_ctx->curr_proc && plain_filename != NULL &&
                    strcmp(plain_filename, "<string>") == 0)
                {
                    /*
                     * If we know the current procedure, append the exact line
                     * from the source, again mimicking Python's traceback.py
                     * module behavior.  We could store the already line-split
                     * source to avoid splitting it every time, but producing
                     * a traceback is not the most important scenario to
                     * optimize for.  But we do not go as far as traceback.py
                     * in reading the source of imported modules.
                     */
                    line = get_source_line(exec_ctx->curr_proc->src, plain_lineno);
                    if (line)
                    {
                        appendStringInfo(&tbstr, "\n    %s", line);
                        pfree(line);
                    }
                }
            }
        }
        PG_FINALLY();
        {
            Py_XDECREF(frame);
            Py_XDECREF(code);
            Py_XDECREF(name);
            Py_XDECREF(lineno);
            Py_XDECREF(filename);
        }
        PG_END_TRY();
 
        /* Advance to the next frame. */
        tb = PyObject_GetAttrString(tb, "tb_next");
        if (tb == NULL)
            elog(ERROR, "could not traverse Python traceback");
 
        /*
         * Release the refcount that PyObject_GetAttrString acquired on the
         * next frame object.  We don't need it, because our caller has a
         * refcount on the first frame object and the frame objects each have
         * a refcount on the next one.  If we tried to hold this refcount
         * longer, it would greatly complicate cleanup in the event of a
         * failure in the above PG_TRY block.
         */
        Py_DECREF(tb);
 
        (*tb_depth)++;
    }
 
    /* Return the traceback. */
    *tbmsg = tbstr.data;
}

References appendStringInfo(), appendStringInfoString(), elog, ERROR, fb(), filename, get_source_line(), initStringInfo(), name, pfree(), PG_END_TRY, PG_FINALLY, PG_TRY, PLy_current_execution_context(), PLy_procedure_name(), PLyUnicode_AsString(), and proname.

Referenced by PLy_elog_impl().

set_string_attr()

static bool set_string_attr ( PyObject *  obj, char *  attrname, char *  str  )

static

Definition at line 596 of file plpy_elog.c.

{
    int         result;
    PyObject   *val;
 
    if (str != NULL)
    {
        val = PLyUnicode_FromString(str);
        if (!val)
            return false;
    }
    else
    {
        val = Py_None;
        Py_INCREF(Py_None);
    }
 
    result = PyObject_SetAttrString(obj, attrname, val);
    Py_DECREF(val);
 
    return result != -1;
}

References fb(), PLyUnicode_FromString(), str, and val.

Referenced by PLy_exception_set_with_details().

Variable Documentation

PLy_exc_error

PyObject* PLy_exc_error = NULL

Definition at line 15 of file plpy_elog.c.

Referenced by PLy_add_exceptions(), PLy_cursor(), PLy_elog_impl(), PLy_output(), PLy_result_colnames(), PLy_result_coltypes(), PLy_result_coltypmods(), and PLy_spi_execute().

PLy_exc_fatal

PyObject* PLy_exc_fatal = NULL

Definition at line 16 of file plpy_elog.c.

Referenced by PLy_add_exceptions(), and PLy_elog_impl().

PLy_exc_spi_error

PyObject* PLy_exc_spi_error = NULL

Definition at line 17 of file plpy_elog.c.

Referenced by PLy_add_exceptions(), PLy_commit(), PLy_elog_impl(), PLy_rollback(), PLy_spi_execute_plan(), PLy_spi_execute_query(), and PLy_spi_subtransaction_abort().