blob: 4ba4de1458bc562e89b38551b79209c3e44c4334 [file] [log] [blame]
Bert Vermeulenb2c19612011-12-04 10:33:02 +01001/*
2 * This file is part of the sigrok project.
3 *
4 * Copyright (C) 2010 Uwe Hermann <uwe@hermann-uwe.de>
Bert Vermeulenbc5f5a42012-01-05 03:31:36 +01005 * Copyright (C) 2012 Bert Vermeulen <bert@biot.com>
Bert Vermeulenb2c19612011-12-04 10:33:02 +01006 *
7 * This program is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation, either version 3 of the License, or
10 * (at your option) any later version.
11 *
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with this program. If not, see <http://www.gnu.org/licenses/>.
19 */
20
Uwe Hermann73191412012-01-03 19:56:01 +010021#include "sigrokdecode.h" /* First, so we avoid a _POSIX_C_SOURCE warning. */
Bert Vermeulen7ce77752012-01-10 00:25:16 +010022#include "sigrokdecode-internal.h"
23#include "config.h"
Bert Vermeulenb2c19612011-12-04 10:33:02 +010024#include <glib.h>
Bert Vermeulen1aef2f92011-12-15 03:31:31 +010025#include <inttypes.h>
Bert Vermeulen0bdadba2012-01-17 03:37:34 +010026#include <stdlib.h>
Bert Vermeulenb2c19612011-12-04 10:33:02 +010027
Bert Vermeulend906d3f2012-01-22 02:51:49 +010028/* List of decoder instances. */
Bert Vermeulen7ce77752012-01-10 00:25:16 +010029static GSList *di_list = NULL;
Bert Vermeulend906d3f2012-01-22 02:51:49 +010030
Uwe Hermann361fdca2012-03-15 22:00:24 +010031/* List of frontend callbacks to receive decoder output. */
Bert Vermeulen983cb0f2012-01-08 03:20:12 +010032static GSList *callbacks = NULL;
33
Uwe Hermannc9bfccc2012-02-08 22:39:30 +010034/* decoder.c */
Uwe Hermann55c3c5f2012-02-09 19:11:53 +010035extern SRD_PRIV GSList *pd_list;
Bert Vermeulenb2c19612011-12-04 10:33:02 +010036
Uwe Hermannc9bfccc2012-02-08 22:39:30 +010037/* module_sigrokdecode.c */
Uwe Hermann55c3c5f2012-02-09 19:11:53 +010038extern SRD_PRIV PyMODINIT_FUNC PyInit_sigrokdecode(void);
Bert Vermeulenb2c19612011-12-04 10:33:02 +010039
Uwe Hermannc9bfccc2012-02-08 22:39:30 +010040/* type_logic.c */
Uwe Hermann55c3c5f2012-02-09 19:11:53 +010041extern SRD_PRIV PyTypeObject srd_logic_type;
Bert Vermeulenb2c19612011-12-04 10:33:02 +010042
Bert Vermeulenb2c19612011-12-04 10:33:02 +010043/**
44 * Initialize libsigrokdecode.
45 *
46 * This initializes the Python interpreter, and creates and initializes
Uwe Hermann361fdca2012-03-15 22:00:24 +010047 * a "sigrokdecode" Python module.
Bert Vermeulenb2c19612011-12-04 10:33:02 +010048 *
49 * Then, it searches for sigrok protocol decoder files (*.py) in the
50 * "decoders" subdirectory of the the sigrok installation directory.
51 * All decoders that are found are loaded into memory and added to an
Bert Vermeulen9d122fd2012-02-14 03:43:28 +010052 * internal list of decoders, which can be queried via srd_decoders_list().
Bert Vermeulenb2c19612011-12-04 10:33:02 +010053 *
54 * The caller is responsible for calling the clean-up function srd_exit(),
55 * which will properly shut down libsigrokdecode and free its allocated memory.
56 *
Bert Vermeulen7ce77752012-01-10 00:25:16 +010057 * Multiple calls to srd_init(), without calling srd_exit() in between,
Bert Vermeulenb2c19612011-12-04 10:33:02 +010058 * are not allowed.
59 *
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +010060 * @param path Path to an extra directory containing protocol decoders
Uwe Hermann361fdca2012-03-15 22:00:24 +010061 * which will be added to the Python sys.path, or NULL.
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +010062 *
Bert Vermeulenb2c19612011-12-04 10:33:02 +010063 * @return SRD_OK upon success, a (negative) error code otherwise.
64 * Upon Python errors, return SRD_ERR_PYTHON. If the sigrok decoders
65 * directory cannot be accessed, return SRD_ERR_DECODERS_DIR.
66 * If not enough memory could be allocated, return SRD_ERR_MALLOC.
67 */
Uwe Hermannabeeed82012-03-16 15:12:54 +010068SRD_API int srd_init(const char *path)
Bert Vermeulenb2c19612011-12-04 10:33:02 +010069{
70 int ret;
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +010071 char *env_path;
Bert Vermeulenb2c19612011-12-04 10:33:02 +010072
Uwe Hermann7a1712c2012-01-26 01:15:10 +010073 srd_dbg("Initializing libsigrokdecode.");
Bert Vermeulend906d3f2012-01-22 02:51:49 +010074
75 /* Add our own module to the list of built-in modules. */
Bert Vermeulenbc5f5a42012-01-05 03:31:36 +010076 PyImport_AppendInittab("sigrokdecode", PyInit_sigrokdecode);
Bert Vermeulen5f802ec2011-12-27 22:15:53 +010077
Uwe Hermann511e2122012-02-10 09:29:38 +010078 /* Initialize the Python interpreter. */
Bert Vermeulenb2c19612011-12-04 10:33:02 +010079 Py_Initialize();
80
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +010081 /* Installed decoders. */
82 if ((ret = add_modulepath(DECODERS_DIR)) != SRD_OK) {
Bert Vermeulenb2c19612011-12-04 10:33:02 +010083 Py_Finalize();
84 return ret;
85 }
86
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +010087 /* Path specified by the user. */
88 if (path) {
89 if ((ret = add_modulepath(path)) != SRD_OK) {
90 Py_Finalize();
91 return ret;
92 }
93 }
94
95 /* Environment variable overrides everything, for debugging. */
96 if ((env_path = getenv("SIGROKDECODE_DIR"))) {
Bert Vermeulen5154cad2012-02-28 02:01:41 +010097 if ((ret = add_modulepath(env_path)) != SRD_OK) {
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +010098 Py_Finalize();
99 return ret;
100 }
101 }
102
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100103 return SRD_OK;
104}
105
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100106/**
107 * Shutdown libsigrokdecode.
108 *
109 * This frees all the memory allocated for protocol decoders and shuts down
110 * the Python interpreter.
111 *
112 * This function should only be called if there was a (successful!) invocation
113 * of srd_init() before. Calling this function multiple times in a row, without
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100114 * any successful srd_init() calls in between, is not allowed.
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100115 *
116 * @return SRD_OK upon success, a (negative) error code otherwise.
117 */
Uwe Hermann55c3c5f2012-02-09 19:11:53 +0100118SRD_API int srd_exit(void)
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100119{
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100120 srd_dbg("Exiting libsigrokdecode.");
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100121
Uwe Hermann8ad6e502012-03-16 15:25:53 +0100122 srd_decoder_unload_all();
Bert Vermeulene5080882011-12-07 09:56:49 +0100123 g_slist_free(pd_list);
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100124
125 /* Py_Finalize() returns void, any finalization errors are ignored. */
126 Py_Finalize();
127
128 return SRD_OK;
129}
130
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100131/**
Uwe Hermann9d9fcb32012-01-23 19:58:06 +0100132 * Add an additional search directory for the protocol decoders.
133 *
134 * The specified directory is prepended (not appended!) to Python's sys.path,
135 * in order to search for sigrok protocol decoders in the specified
136 * directories first, and in the generic Python module directories (and in
137 * the current working directory) last. This avoids conflicts if there are
138 * Python modules which have the same name as a sigrok protocol decoder in
139 * sys.path or in the current working directory.
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100140 *
Uwe Hermann361fdca2012-03-15 22:00:24 +0100141 * @param path Path to the directory containing protocol decoders which shall
142 * be added to the Python sys.path, or NULL.
Uwe Hermann9d9fcb32012-01-23 19:58:06 +0100143 *
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +0100144 * @return SRD_OK upon success, a (negative) error code otherwise.
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100145 */
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +0100146SRD_PRIV int add_modulepath(const char *path)
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100147{
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +0100148 PyObject *py_cur_path, *py_item;
149 GString *new_path;
150 int wc_len, i;
151 wchar_t *wc_new_path;
152 char *item;
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100153
Uwe Hermann8ad6e502012-03-16 15:25:53 +0100154 srd_dbg("Adding '%s' to module path.", path);
Uwe Hermann639c5682012-01-19 00:21:00 +0100155
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +0100156 new_path = g_string_sized_new(256);
157 g_string_assign(new_path, g_strdup(path));
158 py_cur_path = PySys_GetObject("path");
159 for (i = 0; i < PyList_Size(py_cur_path); i++) {
160 g_string_append(new_path, g_strdup(G_SEARCHPATH_SEPARATOR_S));
161 py_item = PyList_GetItem(py_cur_path, i);
162 if (!PyUnicode_Check(py_item))
163 /* Shouldn't happen. */
164 continue;
165 if (py_str_as_str(py_item, &item) != SRD_OK)
166 continue;
167 g_string_append(new_path, item);
168 }
Uwe Hermann639c5682012-01-19 00:21:00 +0100169
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +0100170 /* Convert to wide chars. */
171 wc_len = sizeof(wchar_t) * (new_path->len + 1);
172 if (!(wc_new_path = g_try_malloc(wc_len))) {
173 srd_dbg("malloc failed");
174 return SRD_ERR_MALLOC;
175 }
176 mbstowcs(wc_new_path, new_path->str, wc_len);
177 PySys_SetPath(wc_new_path);
178 g_string_free(new_path, TRUE);
179 g_free(wc_new_path);
Uwe Hermann639c5682012-01-19 00:21:00 +0100180
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +0100181//#ifdef _WIN32
182// gchar **splitted;
183//
184// /*
185// * On Windows/MinGW, Python's sys.path needs entries of the form
186// * 'C:\\foo\\bar' instead of '/foo/bar'.
187// */
188//
189// splitted = g_strsplit(DECODERS_DIR, "/", 0);
190// path = g_build_pathv("\\\\", splitted);
191// g_strfreev(splitted);
192//#else
193// path = g_strdup(DECODERS_DIR);
194//#endif
Uwe Hermann639c5682012-01-19 00:21:00 +0100195
Bert Vermeulen65e1c7d2012-02-12 04:07:22 +0100196 return SRD_OK;
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100197}
198
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100199/**
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100200 * Set options in a decoder instance.
201 *
Uwe Hermann361fdca2012-03-15 22:00:24 +0100202 * Handled options are removed from the hash.
203 *
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100204 * @param di Decoder instance.
205 * @param options A GHashTable of options to set.
206 *
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100207 * @return SRD_OK upon success, a (negative) error code otherwise.
208 */
Bert Vermeulen9d122fd2012-02-14 03:43:28 +0100209SRD_API int srd_inst_options_set(struct srd_decoder_inst *di,
Uwe Hermann361fdca2012-03-15 22:00:24 +0100210 GHashTable *options)
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100211{
212 PyObject *py_dec_options, *py_dec_optkeys, *py_di_options, *py_optval;
213 PyObject *py_optlist, *py_classval;
214 Py_UNICODE *py_ustr;
215 unsigned long long int val_ull;
216 int num_optkeys, ret, size, i;
217 char *key, *value;
218
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100219 if (!PyObject_HasAttrString(di->decoder->py_dec, "options")) {
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100220 /* Decoder has no options. */
Bert Vermeulene431d9c2012-01-18 22:59:14 +0100221 if (g_hash_table_size(options) == 0) {
222 /* No options provided. */
223 return SRD_OK;
224 } else {
225 srd_err("Protocol decoder has no options.");
226 return SRD_ERR_ARG;
227 }
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100228 return SRD_OK;
Bert Vermeulene431d9c2012-01-18 22:59:14 +0100229 }
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100230
231 ret = SRD_ERR_PYTHON;
232 key = NULL;
233 py_dec_options = py_dec_optkeys = py_di_options = py_optval = NULL;
234 py_optlist = py_classval = NULL;
235 py_dec_options = PyObject_GetAttrString(di->decoder->py_dec, "options");
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100236
237 /* All of these are synthesized objects, so they're good. */
238 py_dec_optkeys = PyDict_Keys(py_dec_options);
239 num_optkeys = PyList_Size(py_dec_optkeys);
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100240 if (!(py_di_options = PyObject_GetAttrString(di->py_inst, "options")))
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100241 goto err_out;
242 for (i = 0; i < num_optkeys; i++) {
243 /* Get the default class value for this option. */
244 py_str_as_str(PyList_GetItem(py_dec_optkeys, i), &key);
245 if (!(py_optlist = PyDict_GetItemString(py_dec_options, key)))
246 goto err_out;
247 if (!(py_classval = PyList_GetItem(py_optlist, 1)))
248 goto err_out;
Bert Vermeulen130ef082012-01-19 00:00:02 +0100249 if (!PyUnicode_Check(py_classval) && !PyLong_Check(py_classval)) {
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100250 srd_err("Options of type %s are not yet supported.",
251 Py_TYPE(py_classval)->tp_name);
Bert Vermeulen130ef082012-01-19 00:00:02 +0100252 goto err_out;
253 }
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100254
255 if ((value = g_hash_table_lookup(options, key))) {
256 /* An override for this option was provided. */
257 if (PyUnicode_Check(py_classval)) {
258 if (!(py_optval = PyUnicode_FromString(value))) {
259 /* Some UTF-8 encoding error. */
260 PyErr_Clear();
261 goto err_out;
262 }
263 } else if (PyLong_Check(py_classval)) {
264 if (!(py_optval = PyLong_FromString(value, NULL, 0))) {
265 /* ValueError Exception */
266 PyErr_Clear();
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100267 srd_err("Option %s has invalid value "
268 "%s: expected integer.",
269 key, value);
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100270 goto err_out;
271 }
272 }
273 g_hash_table_remove(options, key);
274 } else {
275 /* Use the class default for this option. */
276 if (PyUnicode_Check(py_classval)) {
277 /* Make a brand new copy of the string. */
278 py_ustr = PyUnicode_AS_UNICODE(py_classval);
279 size = PyUnicode_GET_SIZE(py_classval);
280 py_optval = PyUnicode_FromUnicode(py_ustr, size);
281 } else if (PyLong_Check(py_classval)) {
282 /* Make a brand new copy of the integer. */
283 val_ull = PyLong_AsUnsignedLongLong(py_classval);
284 if (val_ull == (unsigned long long)-1) {
285 /* OverFlowError exception */
286 PyErr_Clear();
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100287 srd_err("Invalid integer value for %s: "
288 "expected integer.", key);
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100289 goto err_out;
290 }
291 if (!(py_optval = PyLong_FromUnsignedLongLong(val_ull)))
292 goto err_out;
293 }
294 }
295
Uwe Hermann361fdca2012-03-15 22:00:24 +0100296 /*
297 * If we got here, py_optval holds a known good new reference
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100298 * to the instance option to set.
299 */
300 if (PyDict_SetItemString(py_di_options, key, py_optval) == -1)
301 goto err_out;
302 }
303
304 ret = SRD_OK;
305
306err_out:
307 Py_XDECREF(py_optlist);
308 Py_XDECREF(py_di_options);
309 Py_XDECREF(py_dec_optkeys);
310 Py_XDECREF(py_dec_options);
Uwe Hermann6f858312012-01-28 21:26:26 +0100311 g_free(key);
Bert Vermeulen2086c682012-01-23 04:51:33 +0100312 if (PyErr_Occurred())
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100313 catch_exception("Stray exception in srd_inst_set_options().");
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100314
315 return ret;
316}
317
Bert Vermeulen9d122fd2012-02-14 03:43:28 +0100318/* Helper GComparefunc for g_slist_find_custom() in srd_inst_probes_set() */
Uwe Hermannabeeed82012-03-16 15:12:54 +0100319static gint compare_probe_id(const struct srd_probe *a, const char *probe_id)
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100320{
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100321 return strcmp(a->id, probe_id);
322}
323
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100324/**
325 * Set probes in a decoder instance.
326 *
327 * @param di Decoder instance.
328 * @param probes A GHashTable of probes to set. Key is probe name, value is
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100329 * the probe number. Samples passed to this instance will be
330 * arranged in this order.
Bert Vermeulen12243c22012-02-12 14:18:32 +0100331 *
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100332 * @return SRD_OK upon success, a (negative) error code otherwise.
333 */
Bert Vermeulen9d122fd2012-02-14 03:43:28 +0100334SRD_API int srd_inst_probes_set(struct srd_decoder_inst *di,
Uwe Hermann361fdca2012-03-15 22:00:24 +0100335 GHashTable *new_probes)
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100336{
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100337 GList *l;
338 GSList *sl;
339 struct srd_probe *p;
340 int *new_probemap, new_probenum;
Bert Vermeulenbe873262012-01-23 19:34:23 +0100341 char *probe_id, *probenum_str;
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100342
Bert Vermeulence46bee2012-01-27 08:21:58 +0100343 srd_dbg("set probes called for instance %s with list of %d probes",
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100344 di->inst_id, g_hash_table_size(new_probes));
Bert Vermeulence46bee2012-01-27 08:21:58 +0100345
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100346 if (g_hash_table_size(new_probes) == 0)
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100347 /* No probes provided. */
348 return SRD_OK;
349
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100350 if (di->dec_num_probes == 0) {
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100351 /* Decoder has no probes. */
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100352 srd_err("Protocol decoder %s has no probes to define.",
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100353 di->decoder->name);
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100354 return SRD_ERR_ARG;
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100355 }
356
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100357 new_probemap = NULL;
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100358
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100359 if (!(new_probemap = g_try_malloc(sizeof(int) * di->dec_num_probes))) {
Uwe Hermanna61ece22012-02-10 00:06:58 +0100360 srd_err("Failed to g_malloc() new probe map.");
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100361 return SRD_ERR_MALLOC;
362 }
363
364 for (l = g_hash_table_get_keys(new_probes); l; l = l->next) {
365 probe_id = l->data;
Bert Vermeulenbe873262012-01-23 19:34:23 +0100366 probenum_str = g_hash_table_lookup(new_probes, probe_id);
367 if (!probenum_str) {
368 /* Probe name was specified without a value. */
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100369 srd_err("No probe number was specified for %s.",
370 probe_id);
Bert Vermeulenbe873262012-01-23 19:34:23 +0100371 g_free(new_probemap);
372 return SRD_ERR_ARG;
373 }
374 new_probenum = strtol(probenum_str, NULL, 10);
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100375 if (!(sl = g_slist_find_custom(di->decoder->probes, probe_id,
376 (GCompareFunc)compare_probe_id))) {
377 /* Fall back on optional probes. */
Bert Vermeulendcdf4882012-02-01 00:07:55 +0100378 if (!(sl = g_slist_find_custom(di->decoder->opt_probes,
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100379 probe_id, (GCompareFunc) compare_probe_id))) {
380 srd_err("Protocol decoder %s has no probe "
381 "'%s'.", di->decoder->name, probe_id);
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100382 g_free(new_probemap);
383 return SRD_ERR_ARG;
384 }
385 }
386 p = sl->data;
387 new_probemap[p->order] = new_probenum;
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100388 srd_dbg("setting probe mapping for %d = probe %d", p->order,
389 new_probenum);
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100390 }
391 g_free(di->dec_probemap);
392 di->dec_probemap = new_probemap;
393
394 return SRD_OK;
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100395}
396
397/**
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100398 * Create a new protocol decoder instance.
399 *
Uwe Hermann42a85ed2012-01-10 20:00:25 +0100400 * @param id Decoder 'id' field.
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100401 * @param options GHashtable of options which override the defaults set in
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100402 * the decoder class.
Bert Vermeulen12243c22012-02-12 14:18:32 +0100403 *
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100404 * @return Pointer to a newly allocated struct srd_decoder_inst, or
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100405 * NULL in case of failure.
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100406 */
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100407SRD_API struct srd_decoder_inst *srd_inst_new(const char *decoder_id,
Uwe Hermann361fdca2012-03-15 22:00:24 +0100408 GHashTable *options)
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100409{
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100410 int i;
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100411 struct srd_decoder *dec;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100412 struct srd_decoder_inst *di;
413 char *inst_id;
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100414
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100415 srd_dbg("Creating new %s instance.", decoder_id);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100416
Bert Vermeulen9d122fd2012-02-14 03:43:28 +0100417 if (!(dec = srd_decoder_get_by_id(decoder_id))) {
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100418 srd_err("Protocol decoder %s not found.", decoder_id);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100419 return NULL;
420 }
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100421
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100422 if (!(di = g_try_malloc0(sizeof(struct srd_decoder_inst)))) {
Uwe Hermanna61ece22012-02-10 00:06:58 +0100423 srd_err("Failed to g_malloc() instance.");
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100424 return NULL;
425 }
426
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100427 inst_id = g_hash_table_lookup(options, "id");
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100428 di->decoder = dec;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100429 di->inst_id = g_strdup(inst_id ? inst_id : decoder_id);
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100430 g_hash_table_remove(options, "id");
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100431
Uwe Hermann361fdca2012-03-15 22:00:24 +0100432 /*
433 * Prepare a default probe map, where samples come in the
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100434 * order in which the decoder class defined them.
435 */
436 di->dec_num_probes = g_slist_length(di->decoder->probes) +
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100437 g_slist_length(di->decoder->opt_probes);
Bert Vermeulen19a90ba2012-01-21 19:45:04 +0100438 if (di->dec_num_probes) {
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100439 if (!(di->dec_probemap =
440 g_try_malloc(sizeof(int) * di->dec_num_probes))) {
Uwe Hermanna61ece22012-02-10 00:06:58 +0100441 srd_err("Failed to g_malloc() probe map.");
Bert Vermeulen19a90ba2012-01-21 19:45:04 +0100442 g_free(di);
443 return NULL;
444 }
445 for (i = 0; i < di->dec_num_probes; i++)
446 di->dec_probemap[i] = i;
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100447 }
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100448
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100449 /* Create a new instance of this decoder class. */
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100450 if (!(di->py_inst = PyObject_CallObject(dec->py_dec, NULL))) {
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100451 if (PyErr_Occurred())
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100452 catch_exception("failed to create %s instance: ",
453 decoder_id);
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100454 g_free(di->dec_probemap);
455 g_free(di);
456 return NULL;
457 }
458
Bert Vermeulen9d122fd2012-02-14 03:43:28 +0100459 if (srd_inst_options_set(di, options) != SRD_OK) {
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100460 g_free(di->dec_probemap);
Bert Vermeulen0bdadba2012-01-17 03:37:34 +0100461 g_free(di);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100462 return NULL;
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100463 }
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100464
465 /* Instance takes input from a frontend by default. */
Bert Vermeulene5080882011-12-07 09:56:49 +0100466 di_list = g_slist_append(di_list, di);
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100467
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100468 return di;
469}
470
Bert Vermeulen582c8472012-02-12 14:55:20 +0100471/**
472 * Stack a decoder instance on top of another.
473 *
474 * @param di_from The instance to move.
475 * @param di_to The instance on top of which di_from will be stacked.
476 *
477 * @return SRD_OK upon success, a (negative) error code otherwise.
478 */
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100479SRD_API int srd_inst_stack(struct srd_decoder_inst *di_from,
Uwe Hermannabeeed82012-03-16 15:12:54 +0100480 struct srd_decoder_inst *di_to)
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100481{
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100482 if (!di_from || !di_to) {
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100483 srd_err("Invalid from/to instance pair.");
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100484 return SRD_ERR_ARG;
485 }
486
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100487 if (g_slist_find(di_list, di_to)) {
488 /* Remove from the unstacked list. */
489 di_list = g_slist_remove(di_list, di_to);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100490 }
491
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100492 /* Stack on top of source di. */
493 di_from->next_di = g_slist_append(di_from->next_di, di_to);
494
495 return SRD_OK;
496}
497
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100498/**
Uwe Hermann361fdca2012-03-15 22:00:24 +0100499 * Find a decoder instance by its instance ID.
500 *
501 * Only the bottom level of instances are searched -- instances already stacked
502 * on top of another one will not be found.
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100503 *
Uwe Hermanned2306a2012-03-18 14:16:50 +0100504 * @param inst_id The instance ID to be found.
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100505 *
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100506 * @return Pointer to struct srd_decoder_inst, or NULL if not found.
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100507 */
Uwe Hermannabeeed82012-03-16 15:12:54 +0100508SRD_API struct srd_decoder_inst *srd_inst_find_by_id(const char *inst_id)
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100509{
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100510 GSList *l;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100511 struct srd_decoder_inst *tmp, *di;
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100512
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100513 di = NULL;
514 for (l = di_list; l; l = l->next) {
515 tmp = l->data;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100516 if (!strcmp(tmp->inst_id, inst_id)) {
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100517 di = tmp;
518 break;
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100519 }
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100520 }
521
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100522 return di;
523}
524
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100525/**
Uwe Hermann361fdca2012-03-15 22:00:24 +0100526 * Find a decoder instance by its Python object.
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100527 *
Uwe Hermann361fdca2012-03-15 22:00:24 +0100528 * I.e. find that instance's instantiation of the sigrokdecode.Decoder class.
529 * This will recurse to find the instance anywhere in the stack tree.
530 *
531 * @param stack Pointer to a GSList of struct srd_decoder_inst, indicating the
532 * stack to search. To start searching at the bottom level of
533 * decoder instances, pass NULL.
Uwe Hermann511e2122012-02-10 09:29:38 +0100534 * @param obj The Python class instantiation.
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100535 *
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100536 * @return Pointer to struct srd_decoder_inst, or NULL if not found.
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100537 */
Uwe Hermannabeeed82012-03-16 15:12:54 +0100538SRD_PRIV struct srd_decoder_inst *srd_inst_find_by_obj(const GSList *stack,
539 const PyObject *obj)
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100540{
Uwe Hermannabeeed82012-03-16 15:12:54 +0100541// TODO?
542 const GSList *l;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100543 struct srd_decoder_inst *tmp, *di;
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100544
545 di = NULL;
546 for (l = stack ? stack : di_list; di == NULL && l != NULL; l = l->next) {
547 tmp = l->data;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100548 if (tmp->py_inst == obj)
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100549 di = tmp;
550 else if (tmp->next_di)
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100551 di = srd_inst_find_by_obj(tmp->next_di, obj);
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100552 }
553
554 return di;
555}
556
Bert Vermeulenee4ed222012-02-12 13:47:07 +0100557SRD_PRIV int srd_inst_start(struct srd_decoder_inst *di, PyObject *args)
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100558{
559 PyObject *py_name, *py_res;
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100560 GSList *l;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100561 struct srd_decoder_inst *next_di;
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100562
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100563 srd_dbg("Calling start() method on protocol decoder instance %s.",
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100564 di->inst_id);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100565
566 if (!(py_name = PyUnicode_FromString("start"))) {
Uwe Hermann511e2122012-02-10 09:29:38 +0100567 srd_err("Unable to build Python object for 'start'.");
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100568 catch_exception("Protocol decoder instance %s: ",
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100569 di->inst_id);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100570 return SRD_ERR_PYTHON;
571 }
572
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100573 if (!(py_res = PyObject_CallMethodObjArgs(di->py_inst,
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100574 py_name, args, NULL))) {
575 catch_exception("Protocol decoder instance %s: ",
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100576 di->inst_id);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100577 return SRD_ERR_PYTHON;
578 }
579
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100580 Py_DecRef(py_res);
581 Py_DecRef(py_name);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100582
Uwe Hermann361fdca2012-03-15 22:00:24 +0100583 /*
584 * Start all the PDs stacked on top of this one. Pass along the
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100585 * metadata all the way from the bottom PD, even though it's only
586 * applicable to logic data for now.
587 */
588 for (l = di->next_di; l; l = l->next) {
589 next_di = l->data;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100590 srd_inst_start(next_di, args);
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100591 }
592
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100593 return SRD_OK;
594}
595
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100596/**
597 * Run the specified decoder function.
598 *
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100599 * @param start_samplenum The starting sample number for the buffer's sample
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100600 * set, relative to the start of capture.
601 * @param di The decoder instance to call. Must not be NULL.
602 * @param inbuf The buffer to decode. Must not be NULL.
603 * @param inbuflen Length of the buffer. Must be > 0.
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100604 *
605 * @return SRD_OK upon success, a (negative) error code otherwise.
606 */
Bert Vermeulen12243c22012-02-12 14:18:32 +0100607SRD_PRIV int srd_inst_decode(uint64_t start_samplenum,
Uwe Hermannabeeed82012-03-16 15:12:54 +0100608 const struct srd_decoder_inst *di,
609 const uint8_t *inbuf, uint64_t inbuflen)
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100610{
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100611 PyObject *py_res;
Bert Vermeulenbc5f5a42012-01-05 03:31:36 +0100612 srd_logic *logic;
Bert Vermeulen86528292012-01-15 23:20:39 +0100613 uint64_t end_samplenum;
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100614
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100615 srd_dbg("Calling decode() on instance %s with %d bytes starting "
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100616 "at sample %d.", di->inst_id, inbuflen, start_samplenum);
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100617
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100618 /* Return an error upon unusable input. */
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100619 if (!di) {
620 srd_dbg("empty decoder instance");
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100621 return SRD_ERR_ARG;
622 }
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100623 if (!inbuf) {
624 srd_dbg("NULL buffer pointer");
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100625 return SRD_ERR_ARG;
626 }
627 if (inbuflen == 0) {
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100628 srd_dbg("empty buffer");
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100629 return SRD_ERR_ARG;
630 }
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100631
Uwe Hermann361fdca2012-03-15 22:00:24 +0100632 /*
633 * Create new srd_logic object. Each iteration around the PD's loop
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100634 * will fill one sample into this object.
635 */
Bert Vermeulenbc5f5a42012-01-05 03:31:36 +0100636 logic = PyObject_New(srd_logic, &srd_logic_type);
637 Py_INCREF(logic);
Uwe Hermannabeeed82012-03-16 15:12:54 +0100638 logic->di = (struct srd_decoder_inst *)di;
Bert Vermeulen86528292012-01-15 23:20:39 +0100639 logic->start_samplenum = start_samplenum;
Bert Vermeulenbc5f5a42012-01-05 03:31:36 +0100640 logic->itercnt = 0;
Uwe Hermannabeeed82012-03-16 15:12:54 +0100641 logic->inbuf = (uint8_t *)inbuf;
Bert Vermeulenbc5f5a42012-01-05 03:31:36 +0100642 logic->inbuflen = inbuflen;
643 logic->sample = PyList_New(2);
644 Py_INCREF(logic->sample);
645
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100646 Py_IncRef(di->py_inst);
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100647 end_samplenum = start_samplenum + inbuflen / di->data_unitsize;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100648 if (!(py_res = PyObject_CallMethod(di->py_inst, "decode",
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100649 "KKO", logic->start_samplenum,
650 end_samplenum, logic))) {
651 catch_exception("Protocol decoder instance %s: ",
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100652 di->inst_id);
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100653 return SRD_ERR_PYTHON; /* TODO: More specific error? */
654 }
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100655 Py_DecRef(py_res);
Bert Vermeulenbc5f5a42012-01-05 03:31:36 +0100656
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100657 return SRD_OK;
658}
659
Bert Vermeulen12243c22012-02-12 14:18:32 +0100660SRD_PRIV void srd_inst_free(struct srd_decoder_inst *di)
Bert Vermeulenfa12a212012-01-31 23:48:10 +0100661{
662 GSList *l;
663 struct srd_pd_output *pdo;
664
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100665 srd_dbg("Freeing instance %s", di->inst_id);
Bert Vermeulenfa12a212012-01-31 23:48:10 +0100666
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100667 Py_DecRef(di->py_inst);
668 g_free(di->inst_id);
Bert Vermeulenfa12a212012-01-31 23:48:10 +0100669 g_free(di->dec_probemap);
670 g_slist_free(di->next_di);
671 for (l = di->pd_output; l; l = l->next) {
672 pdo = l->data;
673 g_free(pdo->proto_id);
674 g_free(pdo);
675 }
676 g_slist_free(di->pd_output);
Bert Vermeulenfa12a212012-01-31 23:48:10 +0100677}
678
Bert Vermeulen12243c22012-02-12 14:18:32 +0100679SRD_PRIV void srd_inst_free_all(GSList *stack)
Bert Vermeulenfa12a212012-01-31 23:48:10 +0100680{
681 GSList *l;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100682 struct srd_decoder_inst *di;
Bert Vermeulenfa12a212012-01-31 23:48:10 +0100683
684 di = NULL;
685 for (l = stack ? stack : di_list; di == NULL && l != NULL; l = l->next) {
686 di = l->data;
687 if (di->next_di)
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100688 srd_inst_free_all(di->next_di);
689 srd_inst_free(di);
Bert Vermeulenfa12a212012-01-31 23:48:10 +0100690 }
691 if (!stack) {
692 g_slist_free(di_list);
693 di_list = NULL;
694 }
Bert Vermeulenfa12a212012-01-31 23:48:10 +0100695}
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100696
Bert Vermeulen582c8472012-02-12 14:55:20 +0100697/**
698 * Start a decoding session.
699 *
700 * Decoders, instances and stack must have been prepared beforehand.
701 *
702 * @param num_probes The number of probes which the incoming feed will contain.
703 * @param unitsize The number of bytes per sample in the incoming feed.
Uwe Hermann361fdca2012-03-15 22:00:24 +0100704 * @param samplerate The samplerate of the incoming feed.
Bert Vermeulen582c8472012-02-12 14:55:20 +0100705 *
706 * @return SRD_OK upon success, a (negative) error code otherwise.
707 */
Uwe Hermann55c3c5f2012-02-09 19:11:53 +0100708SRD_API int srd_session_start(int num_probes, int unitsize, uint64_t samplerate)
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100709{
710 PyObject *args;
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100711 GSList *d;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100712 struct srd_decoder_inst *di;
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100713 int ret;
714
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100715 srd_dbg("Calling start() on all instances with %d probes, "
716 "unitsize %d samplerate %d.", num_probes, unitsize, samplerate);
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100717
Uwe Hermann361fdca2012-03-15 22:00:24 +0100718 /*
719 * Currently only one item of metadata is passed along to decoders,
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100720 * samplerate. This can be extended as needed.
721 */
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100722 if (!(args = Py_BuildValue("{s:l}", "samplerate", (long)samplerate))) {
Uwe Hermann511e2122012-02-10 09:29:38 +0100723 srd_err("Unable to build Python object for metadata.");
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100724 return SRD_ERR_PYTHON;
725 }
726
727 /* Run the start() method on all decoders receiving frontend data. */
728 for (d = di_list; d; d = d->next) {
729 di = d->data;
Bert Vermeulenf38ec282012-01-20 22:25:42 +0100730 di->data_num_probes = num_probes;
731 di->data_unitsize = unitsize;
732 di->data_samplerate = samplerate;
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100733 if ((ret = srd_inst_start(di, args) != SRD_OK))
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100734 break;
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100735 }
736
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100737 Py_DecRef(args);
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100738
Bert Vermeulen2072ae02012-01-25 01:49:35 +0100739 return ret;
Bert Vermeulen7ce77752012-01-10 00:25:16 +0100740}
741
Bert Vermeulen582c8472012-02-12 14:55:20 +0100742/**
Uwe Hermann29dad0a2012-03-20 17:58:25 +0100743 * Send a chunk of logic sample data to a running decoder session.
Bert Vermeulen582c8472012-02-12 14:55:20 +0100744 *
745 * @param start_samplenum The sample number of the first sample in this chunk.
746 * @param inbuf Pointer to sample data.
747 * @param inbuf Length in bytes of the buffer.
748 *
749 * @return SRD_OK upon success, a (negative) error code otherwise.
750 */
Uwe Hermann29dad0a2012-03-20 17:58:25 +0100751SRD_API int srd_session_send(uint64_t start_samplenum, const uint8_t *inbuf,
Uwe Hermann55c3c5f2012-02-09 19:11:53 +0100752 uint64_t inbuflen)
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100753{
754 GSList *d;
755 int ret;
756
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100757 srd_dbg("Calling decode() on all instances with starting sample "
Uwe Hermannc9bfccc2012-02-08 22:39:30 +0100758 "number %" PRIu64 ", %" PRIu64 " bytes at 0x%p",
759 start_samplenum, inbuflen, inbuf);
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100760
Bert Vermeulene5080882011-12-07 09:56:49 +0100761 for (d = di_list; d; d = d->next) {
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100762 if ((ret = srd_inst_decode(start_samplenum, d->data, inbuf,
Uwe Hermann361fdca2012-03-15 22:00:24 +0100763 inbuflen)) != SRD_OK)
Bert Vermeulenb2c19612011-12-04 10:33:02 +0100764 return ret;
765 }
766
767 return SRD_OK;
768}
769
Bert Vermeulen582c8472012-02-12 14:55:20 +0100770/**
771 * Register a decoder output callback function.
772 *
773 * The function will be called when a protocol decoder sends output back
774 * to the PD controller (except for Python objects, which only go up the
775 * stack).
776 *
777 * @param output_type The output type this callback will receive. Only one
Uwe Hermann4c1d0672012-02-29 21:56:24 +0100778 * callback per output type can be registered.
Uwe Hermann361fdca2012-03-15 22:00:24 +0100779 * @param cb The function to call. Must not be NULL.
780 * @param cb_data Private data for the callback function. Can be NULL.
Bert Vermeulen582c8472012-02-12 14:55:20 +0100781 */
Uwe Hermannae53d0a2012-02-11 22:38:17 +0100782SRD_API int srd_register_callback(int output_type,
Uwe Hermann41a5d962012-02-29 22:32:34 +0100783 srd_pd_output_callback_t cb, void *cb_data)
Bert Vermeulen4fadb122012-01-22 03:29:22 +0100784{
785 struct srd_pd_callback *pd_cb;
786
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100787 srd_dbg("Registering new callback for output type %d.", output_type);
Bert Vermeulen4fadb122012-01-22 03:29:22 +0100788
Uwe Hermanna61ece22012-02-10 00:06:58 +0100789 if (!(pd_cb = g_try_malloc(sizeof(struct srd_pd_callback)))) {
790 srd_err("Failed to g_malloc() struct srd_pd_callback.");
Bert Vermeulen4fadb122012-01-22 03:29:22 +0100791 return SRD_ERR_MALLOC;
Uwe Hermanna61ece22012-02-10 00:06:58 +0100792 }
Bert Vermeulen4fadb122012-01-22 03:29:22 +0100793
794 pd_cb->output_type = output_type;
Uwe Hermann4c1d0672012-02-29 21:56:24 +0100795 pd_cb->cb = cb;
Uwe Hermann41a5d962012-02-29 22:32:34 +0100796 pd_cb->cb_data = cb_data;
Bert Vermeulen4fadb122012-01-22 03:29:22 +0100797 callbacks = g_slist_append(callbacks, pd_cb);
798
799 return SRD_OK;
800}
801
Bert Vermeulen582c8472012-02-12 14:55:20 +0100802SRD_PRIV void *srd_find_callback(int output_type)
Bert Vermeulen4fadb122012-01-22 03:29:22 +0100803{
804 GSList *l;
805 struct srd_pd_callback *pd_cb;
806 void *(cb);
807
808 cb = NULL;
809 for (l = callbacks; l; l = l->next) {
810 pd_cb = l->data;
811 if (pd_cb->output_type == output_type) {
Uwe Hermann4c1d0672012-02-29 21:56:24 +0100812 cb = pd_cb->cb;
Bert Vermeulen4fadb122012-01-22 03:29:22 +0100813 break;
814 }
815 }
816
817 return cb;
818}
819
Uwe Hermann511e2122012-02-10 09:29:38 +0100820/* This is the backend function to Python sigrokdecode.add() call. */
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100821SRD_PRIV int pd_add(struct srd_decoder_inst *di, int output_type,
Uwe Hermannabeeed82012-03-16 15:12:54 +0100822 const char *proto_id)
Bert Vermeulene5080882011-12-07 09:56:49 +0100823{
Bert Vermeulene5080882011-12-07 09:56:49 +0100824 struct srd_pd_output *pdo;
Bert Vermeulene5080882011-12-07 09:56:49 +0100825
Uwe Hermann7a1712c2012-01-26 01:15:10 +0100826 srd_dbg("Instance %s creating new output type %d for %s.",
Bert Vermeulena8b72b02012-02-11 18:06:56 +0100827 di->inst_id, output_type, proto_id);
Bert Vermeulend906d3f2012-01-22 02:51:49 +0100828
Uwe Hermanna61ece22012-02-10 00:06:58 +0100829 if (!(pdo = g_try_malloc(sizeof(struct srd_pd_output)))) {
830 srd_err("Failed to g_malloc() struct srd_pd_output.");
Bert Vermeulene5080882011-12-07 09:56:49 +0100831 return -1;
Uwe Hermanna61ece22012-02-10 00:06:58 +0100832 }
Bert Vermeulene5080882011-12-07 09:56:49 +0100833
Bert Vermeulenb2315462012-01-09 00:13:03 +0100834 /* pdo_id is just a simple index, nothing is deleted from this list anyway. */
Bert Vermeulen15969942012-01-07 02:50:14 +0100835 pdo->pdo_id = g_slist_length(di->pd_output);
Bert Vermeulene5080882011-12-07 09:56:49 +0100836 pdo->output_type = output_type;
Bert Vermeulenb3b2cae2012-01-24 02:02:03 +0100837 pdo->di = di;
Uwe Hermann94d43b32012-01-10 20:44:22 +0100838 pdo->proto_id = g_strdup(proto_id);
Bert Vermeulene5080882011-12-07 09:56:49 +0100839 di->pd_output = g_slist_append(di->pd_output, pdo);
840
Bert Vermeulen15969942012-01-07 02:50:14 +0100841 return pdo->pdo_id;
Bert Vermeulene5080882011-12-07 09:56:49 +0100842}