X-Git-Url: http://sigrok.org/gitweb/?a=blobdiff_plain;f=decoder.c;h=25ca4767c6961c17536717c473204d6e98ce20b0;hb=af21ec73e2bf2a6aaec758b090902e7cf3ab96ad;hp=6ad4037eb5828a5489ecc4a61e3386e644a9681c;hpb=e9dd2fea0514df9316a12bae191b810811fc9f01;p=libsigrokdecode.git

diff --git a/decoder.c b/decoder.c
index 6ad4037..25ca476 100644
--- a/decoder.c
+++ b/decoder.c
@@ -39,7 +39,7 @@
 
 /** @cond PRIVATE */
 
-/* The list of protocol decoders. */
+/* The list of loaded protocol decoders. */
 static GSList *pd_list = NULL;
 
 /* srd.c */
@@ -64,7 +64,7 @@ static gboolean srd_check_init(void)
 }
 
 /**
- * Returns the list of supported/loaded protocol decoders.
+ * Returns the list of loaded protocol decoders.
  *
  * This is a GSList of pointers to struct srd_decoder items.
  *
@@ -586,6 +586,29 @@ static int check_method(PyObject *py_dec, const char *mod_name,
 	return SRD_OK;
 }
 
+/**
+ * Get the API version of the specified decoder.
+ *
+ * @param d The decoder to use. Must not be NULL.
+ *
+ * @return The API version of the decoder, or 0 upon errors.
+ */
+SRD_PRIV long srd_decoder_apiver(const struct srd_decoder *d)
+{
+	PyObject *py_apiver;
+	long apiver;
+
+	if (!d)
+		return 0;
+
+	py_apiver = PyObject_GetAttrString(d->py_dec, "api_version");
+	apiver = (py_apiver && PyLong_Check(py_apiver))
+			? PyLong_AsLong(py_apiver) : 0;
+	Py_XDECREF(py_apiver);
+
+	return apiver;
+}
+
 /**
  * Load a protocol decoder module into the embedded Python interpreter.
  *
@@ -597,7 +620,7 @@ static int check_method(PyObject *py_dec, const char *mod_name,
  */
 SRD_API int srd_decoder_load(const char *module_name)
 {
-	PyObject *py_basedec, *py_apiver;
+	PyObject *py_basedec;
 	struct srd_decoder *d;
 	long apiver;
 	int is_subclass;
@@ -648,11 +671,7 @@ SRD_API int srd_decoder_load(const char *module_name)
 	 * Check that this decoder has the correct PD API version.
 	 * PDs of different API versions are incompatible and cannot work.
 	 */
-	py_apiver = PyObject_GetAttrString(d->py_dec, "api_version");
-	apiver = (py_apiver && PyLong_Check(py_apiver))
-			? PyLong_AsLong(py_apiver) : 0;
-	Py_XDECREF(py_apiver);
-
+	apiver = srd_decoder_apiver(d);
 	if (apiver != 2) {
 		srd_exception_catch("Only PDs of API version 2 are supported");
 		goto err_out;
@@ -704,7 +723,7 @@ SRD_API int srd_decoder_load(const char *module_name)
 	if (get_binary_classes(d) != SRD_OK)
 		goto err_out;
 
-	/* Append it to the list of supported/loaded decoders. */
+	/* Append it to the list of loaded decoders. */
 	pd_list = g_slist_append(pd_list, d);
 
 	return SRD_OK;
@@ -787,6 +806,9 @@ SRD_API int srd_decoder_unload(struct srd_decoder *dec)
 		srd_inst_free_all(sess, NULL);
 	}
 
+	/* Remove the PD from the list of loaded decoders. */
+	pd_list = g_slist_remove(pd_list, dec);
+
 	decoder_free(dec);
 
 	return SRD_OK;