X-Git-Url: https://sigrok.org/gitweb/?p=libsigrokdecode.git;a=blobdiff_plain;f=util.c;h=764474ba2a6e461814e53b574cd87dbecfb81dd9;hp=6164aaa91e87d95c398ab3bdb5c5b25dd3f51b40;hb=bcd4e47e8cc688d2a04c6fbfde1e0a354405f769;hpb=73191416cc24884ffab5f21cc3405c7757884fba diff --git a/util.c b/util.c index 6164aaa..764474b 100644 --- a/util.c +++ b/util.c @@ -1,8 +1,8 @@ /* - * This file is part of the sigrok project. + * This file is part of the libsigrokdecode project. * * Copyright (C) 2010 Uwe Hermann - * Copyright (C) 2011 Bert Vermeulen + * Copyright (C) 2012 Bert Vermeulen * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by @@ -18,63 +18,179 @@ * along with this program. If not, see . */ +#include "libsigrokdecode-internal.h" /* First, so we avoid a _POSIX_C_SOURCE warning. */ +#include "libsigrokdecode.h" #include "config.h" -#include "sigrokdecode.h" /* First, so we avoid a _POSIX_C_SOURCE warning. */ - /** - * Helper function to get the value of a python object's attribute, - * returned as a newly allocated char *. + * Get the value of a Python object's attribute, returned as a newly + * allocated char *. * * @param py_obj The object to probe. - * @param key Name of the attribute to retrieve. + * @param attr Name of the attribute to retrieve. * @param outstr ptr to char * storage to be filled in. * * @return SRD_OK upon success, a (negative) error code otherwise. * The 'outstr' argument points to a malloc()ed string upon success. + * + * @private */ -int h_str(PyObject *py_obj, const char *key, char **outstr) +SRD_PRIV int py_attr_as_str(const PyObject *py_obj, const char *attr, + char **outstr) { - PyObject *py_str, *py_encstr; - char *str; + PyObject *py_str; + int ret; + + if (!PyObject_HasAttrString((PyObject *)py_obj, attr)) { + srd_dbg("%s object has no attribute '%s'.", + Py_TYPE(py_obj)->tp_name, attr); + return SRD_ERR_PYTHON; + } + + if (!(py_str = PyObject_GetAttrString((PyObject *)py_obj, attr))) { + srd_exception_catch(""); + return SRD_ERR_PYTHON; + } + + if (!PyUnicode_Check(py_str)) { + srd_dbg("%s attribute should be a string, but is a %s.", + attr, Py_TYPE(py_str)->tp_name); + Py_DecRef(py_str); + return SRD_ERR_PYTHON; + } + + ret = py_str_as_str(py_str, outstr); + Py_DecRef(py_str); + + return ret; +} + +/** + * Get the value of a Python dictionary item, returned as a newly + * allocated char *. + * + * @param py_obj The dictionary to probe. + * @param key Key of the item to retrieve. + * @param outstr Pointer to char * storage to be filled in. + * + * @return SRD_OK upon success, a (negative) error code otherwise. + * The 'outstr' argument points to a malloc()ed string upon success. + * + * @private + */ +SRD_PRIV int py_dictitem_as_str(const PyObject *py_obj, const char *key, + char **outstr) +{ + PyObject *py_value; + int ret; + + if (!PyDict_Check((PyObject *)py_obj)) { + srd_dbg("Object is a %s, not a dictionary.", + Py_TYPE((PyObject *)py_obj)->tp_name); + return SRD_ERR_PYTHON; + } + + if (!(py_value = PyDict_GetItemString((PyObject *)py_obj, key))) { + srd_dbg("Dictionary has no attribute '%s'.", key); + return SRD_ERR_PYTHON; + } + + if (!PyUnicode_Check(py_value)) { + srd_dbg("Dictionary value for %s should be a string, but is " + "a %s.", key, Py_TYPE(py_value)->tp_name); + return SRD_ERR_PYTHON; + } + + ret = py_str_as_str(py_value, outstr); + + return ret; +} + +/** + * Get the value of a Python unicode string object, returned as a newly + * allocated char *. + * + * @param py_str The unicode string object. + * @param outstr ptr to char * storage to be filled in. + * + * @return SRD_OK upon success, a (negative) error code otherwise. + * The 'outstr' argument points to a malloc()ed string upon success. + * + * @private + */ +SRD_PRIV int py_str_as_str(const PyObject *py_str, char **outstr) +{ + PyObject *py_encstr; int ret; + char *str; - py_str = py_encstr = NULL; + py_encstr = NULL; str = NULL; ret = SRD_OK; - if (!(py_str = PyObject_GetAttrString(py_obj, (char *)key))) { - /* TODO: log level 4 debug message */ + if (!PyUnicode_Check((PyObject *)py_str)) { + srd_dbg("Object is a %s, not a string object.", + Py_TYPE((PyObject *)py_str)->tp_name); ret = SRD_ERR_PYTHON; goto err_out; } - if (!(py_encstr = PyUnicode_AsEncodedString(py_str, "utf-8", NULL))) { - /* TODO: log level 4 debug message */ + if (!(py_encstr = PyUnicode_AsEncodedString((PyObject *)py_str, + "utf-8", NULL))) { ret = SRD_ERR_PYTHON; goto err_out; } if (!(str = PyBytes_AS_STRING(py_encstr))) { - /* TODO: log level 4 debug message */ ret = SRD_ERR_PYTHON; goto err_out; } - if (!(*outstr = g_strdup(str))) { - /* TODO: log level 4 debug message */ - ret = SRD_ERR_MALLOC; - goto err_out; - } + *outstr = g_strdup(str); err_out: - if (py_str) - Py_XDECREF(py_str); if (py_encstr) Py_XDECREF(py_encstr); - if (PyErr_Occurred()) - PyErr_Print(); + if (PyErr_Occurred()) { + srd_exception_catch("string conversion failed"); + } return ret; } +/** + * Convert a Python list of unicode strings to a NULL-terminated UTF8-encoded + * char * array. The caller must g_free() each string when finished. + * + * @param py_strlist The list object. + * @param outstr ptr to char ** storage to be filled in. + * + * @return SRD_OK upon success, a (negative) error code otherwise. + * The 'outstr' argument points to a g_malloc()ed char** upon success. + * + * @private + */ +SRD_PRIV int py_strseq_to_char(const PyObject *py_strseq, char ***outstr) +{ + PyObject *py_str; + int list_len, i; + char **out, *str; + + list_len = PySequence_Size((PyObject *)py_strseq); + if (!(out = g_try_malloc(sizeof(char *) * (list_len + 1)))) { + srd_err("Failed to g_malloc() 'out'."); + return SRD_ERR_MALLOC; + } + for (i = 0; i < list_len; i++) { + if (!(py_str = PyUnicode_AsEncodedString( + PySequence_GetItem((PyObject *)py_strseq, i), "utf-8", NULL))) + return SRD_ERR_PYTHON; + if (!(str = PyBytes_AS_STRING(py_str))) + return SRD_ERR_PYTHON; + out[i] = g_strdup(str); + } + out[i] = NULL; + *outstr = out; + + return SRD_OK; +}