srd: Cosmetics, whitespace, coding-style fixes.
[libsigrokdecode.git] / 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 void catch_exception(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 }