X-Git-Url: https://sigrok.org/gitweb/?p=libsigrokdecode.git;a=blobdiff_plain;f=type_decoder.c;h=6c6eab6b22f584cbcbb0324431cd53e7a0e988fd;hp=0a92a45df7b1510c98ede2c226779cf8d667c435;hb=819e508972da02a0dcff7f81178f0283546a9afd;hpb=487890c822762d9886dfd022ed599c9909ceaab9

diff --git a/type_decoder.c b/type_decoder.c
index 0a92a45..6c6eab6 100644
--- a/type_decoder.c
+++ b/type_decoder.c
@@ -1089,6 +1089,22 @@ static PyObject *Decoder_wait(PyObject *self, PyObject *args)
 		/* Signal the main thread that we handled all samples. */
 		g_cond_signal(&di->handled_all_samples_cond);
 
+		/*
+		 * When EOF was provided externally, communicate the
+		 * Python EOFError exception to .decode() and return
+		 * from the .wait() method call. This is motivated by
+		 * the use of Python context managers, so that .decode()
+		 * methods can "close" incompletely accumulated data
+		 * when the sample data is exhausted.
+		 */
+		if (di->communicate_eof) {
+			srd_dbg("%s: %s: Raising EOF from wait().",
+				di->inst_id, __func__);
+			g_mutex_unlock(&di->data_mutex);
+			PyErr_SetString(PyExc_EOFError, "samples exhausted");
+			goto err;
+		}
+
 		/*
 		 * When termination of wait() and decode() was requested,
 		 * then exit the loop after releasing the mutex.