From c9cb319a703d2af8962711e730986e606a2870b2 Mon Sep 17 00:00:00 2001
From: dreamsourcelabTAI <tzz@dreamsourcelab.com>
Date: Fri, 17 Jun 2022 14:39:38 +0800
Subject: [PATCH] add: when decode task end, try to call sub decoder end
 function

---
 libsigrokdecode4DSL/session.c | 39 +++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/libsigrokdecode4DSL/session.c b/libsigrokdecode4DSL/session.c
index a48ddf30..45b867a0 100755
--- a/libsigrokdecode4DSL/session.c
+++ b/libsigrokdecode4DSL/session.c
@@ -441,9 +441,48 @@ SRD_API int srd_session_end(struct srd_session *sess, char **error)
 				return SRD_ERR_PYTHON;
 			}
 		}
+
+		if (di->next_di != NULL){
+			srd_call_sub_decoder_end(di, error);
+		}
 	}
 
 	PyGILState_Release(gstate);
+	return SRD_OK;
+}
+
+
+SRD_API int srd_call_sub_decoder_end(struct srd_decoder_inst *di, char **error)
+{
+	assert(di && di->next_di);
+
+	GSList *l;
+	struct srd_decoder_inst *sub_dec;
+	PyGILState_STATE gstate;
+	PyObject *py_res; 
+
+	for (l = di->next_di; l; l = l->next){
+		sub_dec = l->data;
+
+		if (PyObject_HasAttrString(sub_dec->py_inst, "end"))
+		{
+			py_res = PyObject_CallMethod(sub_dec->py_inst, "end", NULL);
+
+			if (!py_res)
+			{ 
+				srd_exception_catch(error, "Protocol decoder instance %s",
+									sub_dec->inst_id);
+				PyGILState_Release(gstate);
+				return SRD_ERR_PYTHON;
+			}
+		}
+
+		//next level decoder
+		if (sub_dec->next_di != NULL){
+			if (srd_call_sub_decoder_end(sub_dec, error) != SRD_OK)
+				return SRD_ERR_PYTHON;
+		}
+	}
 
 	return SRD_OK;
 }