type_decoder: Fixup memory leak in Decoder.put() (meta, python)
[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 <config.h>
21 #include "libsigrokdecode-internal.h" /* First, so we avoid a _POSIX_C_SOURCE warning. */
22 #include "libsigrokdecode.h"
23 #include <stdarg.h>
24 #include <glib.h>
25
26 static char *py_stringify(PyObject *py_obj)
27 {
28         PyObject *py_str, *py_bytes;
29         char *str = NULL;
30
31         /* Note: Caller already ran PyGILState_Ensure(). */
32
33         if (!py_obj)
34                 return NULL;
35
36         py_str = PyObject_Str(py_obj);
37         if (!py_str || !PyUnicode_Check(py_str))
38                 goto cleanup;
39
40         py_bytes = PyUnicode_AsUTF8String(py_str);
41         if (!py_bytes)
42                 goto cleanup;
43
44         str = g_strdup(PyBytes_AsString(py_bytes));
45         Py_DECREF(py_bytes);
46
47 cleanup:
48         Py_XDECREF(py_str);
49         if (!str) {
50                 PyErr_Clear();
51                 srd_dbg("Failed to stringify object.");
52         }
53         return str;
54 }
55
56 static char *py_get_string_attr(PyObject *py_obj, const char *attr)
57 {
58         PyObject *py_str, *py_bytes;
59         char *str = NULL;
60
61         /* Note: Caller already ran PyGILState_Ensure(). */
62
63         if (!py_obj)
64                 return NULL;
65
66         py_str = PyObject_GetAttrString(py_obj, attr);
67         if (!py_str || !PyUnicode_Check(py_str))
68                 goto cleanup;
69
70         py_bytes = PyUnicode_AsUTF8String(py_str);
71         if (!py_bytes)
72                 goto cleanup;
73
74         str = g_strdup(PyBytes_AsString(py_bytes));
75         Py_DECREF(py_bytes);
76
77 cleanup:
78         Py_XDECREF(py_str);
79         if (!str) {
80                 PyErr_Clear();
81                 srd_dbg("Failed to get object attribute %s.", attr);
82         }
83         return str;
84 }
85
86 /** @private */
87 SRD_PRIV void srd_exception_catch(const char *format, ...)
88 {
89         int i, ret;
90         va_list args;
91         PyObject *py_etype, *py_evalue, *py_etraceback;
92         PyObject *py_mod, *py_func, *py_tracefmt;
93         char *msg, *etype_name, *evalue_str, *outstr;
94         const char *etype_name_fallback;
95         PyGILState_STATE gstate;
96         GString *s;
97
98         py_etype = py_evalue = py_etraceback = py_mod = py_func = NULL;
99
100         va_start(args, format);
101         msg = g_strdup_vprintf(format, args);
102         va_end(args);
103
104         gstate = PyGILState_Ensure();
105
106         PyErr_Fetch(&py_etype, &py_evalue, &py_etraceback);
107         if (!py_etype) {
108                 /* No current exception, so just print the message. */
109                 srd_err("%s.", msg);
110                 goto cleanup;
111         }
112         PyErr_NormalizeException(&py_etype, &py_evalue, &py_etraceback);
113
114         etype_name = py_get_string_attr(py_etype, "__name__");
115         evalue_str = py_stringify(py_evalue);
116         etype_name_fallback = (etype_name) ? etype_name : "(unknown exception)";
117
118         if (evalue_str)
119                 srd_err("%s: %s: %s", etype_name_fallback, msg, evalue_str);
120         else
121                 srd_err("%s: %s.", etype_name_fallback, msg);
122
123         g_free(evalue_str);
124         g_free(etype_name);
125
126         /* If there is no traceback object, we are done. */
127         if (!py_etraceback)
128                 goto cleanup;
129
130         py_mod = py_import_by_name("traceback");
131         if (!py_mod)
132                 goto cleanup;
133
134         py_func = PyObject_GetAttrString(py_mod, "format_exception");
135         if (!py_func || !PyCallable_Check(py_func))
136                 goto cleanup;
137
138         /* Call into Python to format the stack trace. */
139         py_tracefmt = PyObject_CallFunctionObjArgs(py_func,
140                         py_etype, py_evalue, py_etraceback, NULL);
141         if (!py_tracefmt || !PyList_Check(py_tracefmt))
142                 goto cleanup;
143
144         s = g_string_sized_new(128);
145         for (i = 0; i < PyList_Size(py_tracefmt); i++) {
146                 ret = py_listitem_as_str(py_tracefmt, i, &outstr);
147                 if (ret == 0) {
148                         s = g_string_append(s, outstr);
149                         g_free(outstr);
150                 }
151         }
152         srd_err("%s", s->str);
153         g_string_free(s, TRUE);
154
155         Py_DECREF(py_tracefmt);
156
157 cleanup:
158         Py_XDECREF(py_func);
159         Py_XDECREF(py_mod);
160         Py_XDECREF(py_etraceback);
161         Py_XDECREF(py_evalue);
162         Py_XDECREF(py_etype);
163
164         /* Just in case. */
165         PyErr_Clear();
166
167         PyGILState_Release(gstate);
168
169         g_free(msg);
170 }