22 #include "libsigrokdecode-internal.h"
90 static GSList *di_list = NULL;
93 static GSList *callbacks = NULL;
102 extern PyMODINIT_FUNC PyInit_sigrokdecode(
void);
105 extern SRD_PRIV PyTypeObject srd_logic_type;
141 srd_dbg(
"Initializing libsigrokdecode.");
144 PyImport_AppendInittab(
"sigrokdecode", PyInit_sigrokdecode);
150 if ((ret = srd_decoder_searchpath_add(DECODERS_DIR)) !=
SRD_OK) {
157 if ((ret = srd_decoder_searchpath_add(path)) !=
SRD_OK) {
164 if ((env_path = getenv(
"SIGROKDECODE_DIR"))) {
165 if ((ret = srd_decoder_searchpath_add(env_path)) !=
SRD_OK) {
190 srd_dbg(
"Exiting libsigrokdecode.");
193 g_slist_free(pd_list);
221 SRD_PRIV int srd_decoder_searchpath_add(
const char *path)
223 PyObject *py_cur_path, *py_item;
226 wchar_t *wc_new_path;
229 srd_dbg(
"Adding '%s' to module path.", path);
231 new_path = g_string_sized_new(256);
232 g_string_assign(new_path, g_strdup(path));
233 py_cur_path = PySys_GetObject(
"path");
234 for (i = 0; i < PyList_Size(py_cur_path); i++) {
235 g_string_append(new_path, g_strdup(G_SEARCHPATH_SEPARATOR_S));
236 py_item = PyList_GetItem(py_cur_path, i);
237 if (!PyUnicode_Check(py_item))
240 if (py_str_as_str(py_item, &item) !=
SRD_OK)
242 g_string_append(new_path, item);
246 wc_len =
sizeof(wchar_t) * (new_path->len + 1);
247 if (!(wc_new_path = g_try_malloc(wc_len))) {
248 srd_dbg(
"malloc failed");
251 mbstowcs(wc_new_path, new_path->str, wc_len);
252 PySys_SetPath(wc_new_path);
253 g_string_free(new_path, TRUE);
299 PyObject *py_dec_options, *py_dec_optkeys, *py_di_options, *py_optval;
300 PyObject *py_optlist, *py_classval;
303 unsigned long long int val_ull;
305 int num_optkeys, ret, size, i;
309 if (!PyObject_HasAttrString(di->
decoder->
py_dec,
"options")) {
311 if (g_hash_table_size(options) == 0) {
315 srd_err(
"Protocol decoder has no options.");
323 py_dec_options = py_dec_optkeys = py_di_options = py_optval = NULL;
324 py_optlist = py_classval = NULL;
325 py_dec_options = PyObject_GetAttrString(di->
decoder->
py_dec,
"options");
328 py_dec_optkeys = PyDict_Keys(py_dec_options);
329 num_optkeys = PyList_Size(py_dec_optkeys);
330 if (!(py_di_options = PyObject_GetAttrString(di->
py_inst,
"options")))
332 for (i = 0; i < num_optkeys; i++) {
334 py_str_as_str(PyList_GetItem(py_dec_optkeys, i), &key);
335 if (!(py_optlist = PyDict_GetItemString(py_dec_options, key)))
337 if (!(py_classval = PyList_GetItem(py_optlist, 1)))
339 if (!PyUnicode_Check(py_classval) && !PyLong_Check(py_classval)) {
340 srd_err(
"Options of type %s are not yet supported.",
341 Py_TYPE(py_classval)->tp_name);
345 if ((value = g_hash_table_lookup(options, key))) {
346 dbg = g_variant_print(value, TRUE);
347 srd_dbg(
"got option '%s' = %s", key, dbg);
350 if (PyUnicode_Check(py_classval)) {
351 if (!g_variant_is_of_type(value, G_VARIANT_TYPE_STRING)) {
352 srd_err(
"Option '%s' requires a string value.", key);
355 val_str = g_variant_get_string(value, NULL);
356 if (!(py_optval = PyUnicode_FromString(val_str))) {
359 srd_err(
"Option '%s' requires a UTF-8 string value.", key);
362 }
else if (PyLong_Check(py_classval)) {
363 if (!g_variant_is_of_type(value, G_VARIANT_TYPE_INT64)) {
364 srd_err(
"Option '%s' requires an integer value.", key);
367 val_int = g_variant_get_int64(value);
368 if (!(py_optval = PyLong_FromLong(val_int))) {
371 srd_err(
"Option '%s' has invalid integer value.", key);
375 g_hash_table_remove(options, key);
378 if (PyUnicode_Check(py_classval)) {
380 py_ustr = PyUnicode_AS_UNICODE(py_classval);
381 size = PyUnicode_GET_SIZE(py_classval);
382 py_optval = PyUnicode_FromUnicode(py_ustr, size);
383 }
else if (PyLong_Check(py_classval)) {
385 val_ull = PyLong_AsUnsignedLongLong(py_classval);
386 if (val_ull == (
unsigned long long)-1) {
389 srd_err(
"Invalid integer value for %s: "
390 "expected integer.", key);
393 if (!(py_optval = PyLong_FromUnsignedLongLong(val_ull)))
402 if (PyDict_SetItemString(py_di_options, key, py_optval) == -1)
409 Py_XDECREF(py_optlist);
410 Py_XDECREF(py_di_options);
411 Py_XDECREF(py_dec_optkeys);
412 Py_XDECREF(py_dec_options);
414 if (PyErr_Occurred())
415 srd_exception_catch(
"Stray exception in srd_inst_option_set().");
421 static gint compare_probe_id(
const struct srd_probe *a,
const char *probe_id)
423 return strcmp(a->
id, probe_id);
442 GHashTable *new_probes)
448 int *new_probemap, new_probenum;
450 int i, num_required_probes;
452 srd_dbg(
"set probes called for instance %s with list of %d probes",
453 di->
inst_id, g_hash_table_size(new_probes));
455 if (g_hash_table_size(new_probes) == 0)
461 srd_err(
"Protocol decoder %s has no probes to define.",
468 if (!(new_probemap = g_try_malloc(
sizeof(
int) * di->
dec_num_probes))) {
469 srd_err(
"Failed to g_malloc() new probe map.");
478 new_probemap[i] = -1;
480 for (l = g_hash_table_get_keys(new_probes); l; l = l->next) {
482 probe_val= g_hash_table_lookup(new_probes, probe_id);
483 if (!g_variant_is_of_type(probe_val, G_VARIANT_TYPE_INT32)) {
485 srd_err(
"No probe number was specified for %s.",
487 g_free(new_probemap);
490 new_probenum = g_variant_get_int32(probe_val);
491 if (!(sl = g_slist_find_custom(di->
decoder->
probes, probe_id,
492 (GCompareFunc)compare_probe_id))) {
495 probe_id, (GCompareFunc) compare_probe_id))) {
496 srd_err(
"Protocol decoder %s has no probe "
498 g_free(new_probemap);
503 new_probemap[p->
order] = new_probenum;
504 srd_dbg(
"Setting probe mapping: %s (index %d) = probe %d.",
505 p->
id, p->
order, new_probenum);
508 srd_dbg(
"Final probe map:");
511 srd_dbg(
" - index %d = probe %d (%s)", i, new_probemap[i],
512 (i < num_required_probes) ?
"required" :
"optional");
541 srd_dbg(
"Creating new %s instance.", decoder_id);
544 srd_err(
"Protocol decoder %s not found.", decoder_id);
549 srd_err(
"Failed to g_malloc() instance.");
555 inst_id = g_hash_table_lookup(options,
"id");
556 di->
inst_id = g_strdup(inst_id ? inst_id : decoder_id);
557 g_hash_table_remove(options,
"id");
559 di->
inst_id = g_strdup(decoder_id);
570 srd_err(
"Failed to g_malloc() probe map.");
579 if (!(di->
py_inst = PyObject_CallObject(dec->
py_dec, NULL))) {
580 if (PyErr_Occurred())
581 srd_exception_catch(
"failed to create %s instance: ",
595 di_list = g_slist_append(di_list, di);
613 if (!di_from || !di_to) {
614 srd_err(
"Invalid from/to instance pair.");
618 if (g_slist_find(di_list, di_to)) {
620 di_list = g_slist_remove(di_list, di_to);
647 for (l = di_list; l; l = l->next) {
649 if (!strcmp(tmp->
inst_id, inst_id)) {
682 for (l = stack ? stack : di_list; di == NULL && l != NULL; l = l->next) {
687 di = srd_inst_find_by_obj(tmp->
next_di, obj);
696 PyObject *py_name, *py_res;
700 srd_dbg(
"Calling start() method on protocol decoder instance %s.",
703 if (!(py_name = PyUnicode_FromString(
"start"))) {
704 srd_err(
"Unable to build Python object for 'start'.");
705 srd_exception_catch(
"Protocol decoder instance %s: ",
710 if (!(py_res = PyObject_CallMethodObjArgs(di->
py_inst,
711 py_name, args, NULL))) {
712 srd_exception_catch(
"Protocol decoder instance %s: ",
725 for (l = di->
next_di; l; l = l->next) {
727 srd_inst_start(next_di, args);
748 SRD_PRIV int srd_inst_decode(uint64_t start_samplenum,
754 uint64_t end_samplenum;
756 srd_dbg(
"Calling decode() on instance %s with %d bytes starting "
757 "at sample %d.", di->
inst_id, inbuflen, start_samplenum);
761 srd_dbg(
"empty decoder instance");
765 srd_dbg(
"NULL buffer pointer");
769 srd_dbg(
"empty buffer");
777 logic = PyObject_New(
srd_logic, &srd_logic_type);
782 logic->
inbuf = (uint8_t *)inbuf;
784 logic->
sample = PyList_New(2);
788 end_samplenum = start_samplenum + inbuflen / di->
data_unitsize;
789 if (!(py_res = PyObject_CallMethod(di->
py_inst,
"decode",
791 end_samplenum, logic))) {
792 srd_exception_catch(
"Protocol decoder instance %s: ",
807 srd_dbg(
"Freeing instance %s", di->
inst_id);
813 for (l = di->
pd_output; l; l = l->next) {
822 SRD_PRIV void srd_inst_free_all(GSList *stack)
828 for (l = stack ? stack : di_list; di == NULL && l != NULL; l = l->next) {
831 srd_inst_free_all(di->
next_di);
835 g_slist_free(di_list);
872 srd_dbg(
"Calling start() on all instances with %d probes, "
873 "unitsize %d samplerate %d.", num_probes, unitsize, samplerate);
879 if (!(args = Py_BuildValue(
"{s:l}",
"samplerate", (
long)samplerate))) {
880 srd_err(
"Unable to build Python object for metadata.");
885 for (d = di_list; d; d = d->next) {
890 if ((ret = srd_inst_start(di, args)) !=
SRD_OK)
916 srd_dbg(
"Calling decode() on all instances with starting sample "
917 "number %" PRIu64
", %" PRIu64
" bytes at 0x%p",
918 start_samplenum, inbuflen, inbuf);
920 for (d = di_list; d; d = d->next) {
921 if ((ret = srd_inst_decode(start_samplenum, d->data, inbuf,
948 srd_dbg(
"Registering new callback for output type %d.", output_type);
951 srd_err(
"Failed to g_malloc() struct srd_pd_callback.");
958 callbacks = g_slist_append(callbacks, pd_cb);
971 for (l = callbacks; l; l = l->next) {
985 int output_type,
const char *proto_id)
989 srd_dbg(
"Instance %s creating new output type %d for %s.",
990 di->
inst_id, output_type, proto_id);
993 srd_err(
"Failed to g_malloc() struct srd_pd_output.");
1001 pdo->
proto_id = g_strdup(proto_id);