exception.c

   1 /*
   2  * This file is part of the sigrok project.
   3  *
   4  * Copyright (C) 2012 Bert Vermeulen <bert@biot.com>
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #include "sigrokdecode.h" /* First, so we avoid a _POSIX_C_SOURCE warning. */
  21 #include "sigrokdecode-internal.h"
  22 #include "config.h"
  23 #include <stdarg.h>
  24 #include <glib.h>
  25 #include <frameobject.h> /* Python header not pulled in by default. */
  26
  27 SRD_PRIV void srd_exception_catch(const char *format, ...)
  28 {
  29         PyObject *etype, *evalue, *etb, *py_str;
  30         PyTracebackObject *py_tb;
  31         GString *msg;
  32         va_list args;
  33         char *ename, *str, *tracestr;
  34
  35         if (!PyErr_Occurred())
  36                 /* Nothing is wrong. */
  37                 return;
  38
  39         PyErr_Fetch(&etype, &evalue, &etb);
  40         PyErr_NormalizeException(&etype, &evalue, &etb);
  41
  42         if (!(py_str = PyObject_Str(evalue))) {
  43                 /* Shouldn't happen. */
  44                 srd_dbg("Failed to convert exception value to string.");
  45                 return;
  46         }
  47
  48         msg = g_string_sized_new(128);
  49         va_start(args, format);
  50         g_string_vprintf(msg, format, args);
  51         va_end(args);
  52
  53         /* Can be NULL. */
  54         if (evalue)
  55                 ename = (char *)Py_TYPE(evalue)->tp_name;
  56         else
  57                 ename = "(unknown exception)";
  58
  59         /* Send the exception error message(s) to srd_err(). */
  60         py_str_as_str(py_str, &str);
  61         g_string_append(msg, ename);
  62         g_string_append(msg, ": ");
  63         g_string_append(msg, str);
  64         Py_DecRef(py_str);
  65         srd_err(msg->str);
  66
  67         /* Send a more precise error location to srd_dbg(), if we have it. */
  68         if (etb && etb != Py_None) {
  69                 tracestr = NULL;
  70                 py_tb = (PyTracebackObject *)etb;
  71                 py_str = PyUnicode_FromFormat("%U:%d in %U",
  72                                         py_tb->tb_frame->f_code->co_filename,
  73                                         py_tb->tb_frame->f_lineno,
  74                                         py_tb->tb_frame->f_code->co_name);
  75                 py_str_as_str(py_str, &tracestr);
  76                 Py_DecRef(py_str);
  77                 g_string_printf(msg, "%s in %s: %s", ename, tracestr, str);
  78                 srd_dbg(msg->str);
  79                 g_free(tracestr);
  80         }
  81         g_free(str);
  82         g_string_free(msg, TRUE);
  83
  84         Py_XDECREF(etype);
  85         Py_XDECREF(evalue);
  86         Py_XDECREF(etb);
  87
  88         /* Just in case. */
  89         PyErr_Clear();
  90 }