GPL headers: Use correct project name.
[libsigrokdecode.git] / exception.c
1 /*
2  * This file is part of the libsigrokdecode 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 /** @private */
28 SRD_PRIV void srd_exception_catch(const char *format, ...)
29 {
30         PyObject *etype, *evalue, *etb, *py_str;
31         PyTracebackObject *py_tb;
32         GString *msg;
33         va_list args;
34         char *ename, *str, *tracestr;
35
36         if (!PyErr_Occurred())
37                 /* Nothing is wrong. */
38                 return;
39
40         PyErr_Fetch(&etype, &evalue, &etb);
41         PyErr_NormalizeException(&etype, &evalue, &etb);
42
43         if (!(py_str = PyObject_Str(evalue))) {
44                 /* Shouldn't happen. */
45                 srd_dbg("Failed to convert exception value to string.");
46                 return;
47         }
48
49         msg = g_string_sized_new(128);
50         va_start(args, format);
51         g_string_vprintf(msg, format, args);
52         va_end(args);
53
54         /* Can be NULL. */
55         if (evalue)
56                 ename = (char *)Py_TYPE(evalue)->tp_name;
57         else
58                 ename = "(unknown exception)";
59
60         /* Send the exception error message(s) to srd_err(). */
61         py_str_as_str(py_str, &str);
62         g_string_append(msg, ename);
63         g_string_append(msg, ": ");
64         g_string_append(msg, str);
65         Py_DecRef(py_str);
66         srd_err(msg->str);
67
68         /* Send a more precise error location to srd_dbg(), if we have it. */
69         if (etb && etb != Py_None) {
70                 tracestr = NULL;
71                 py_tb = (PyTracebackObject *)etb;
72                 py_str = PyUnicode_FromFormat("%U:%d in %U",
73                                         py_tb->tb_frame->f_code->co_filename,
74                                         py_tb->tb_frame->f_lineno,
75                                         py_tb->tb_frame->f_code->co_name);
76                 py_str_as_str(py_str, &tracestr);
77                 Py_DecRef(py_str);
78                 g_string_printf(msg, "%s in %s: %s", ename, tracestr, str);
79                 srd_dbg(msg->str);
80                 g_free(tracestr);
81         }
82         g_free(str);
83         g_string_free(msg, TRUE);
84
85         Py_XDECREF(etype);
86         Py_XDECREF(evalue);
87         Py_XDECREF(etb);
88
89         /* Just in case. */
90         PyErr_Clear();
91 }