X-Git-Url: https://sigrok.org/gitweb/?p=libsigrok.git;a=blobdiff_plain;f=session_driver.c;h=fd86ced1655a39298e190b4027766501a0dc991d;hp=fe0db3e6e0acbbf0ad5b808e4321b11fe8a48d35;hb=0812c40e361c9a75f3b4ef318a57ce8ba0479fa1;hpb=2c1a012ed243c39dc4ec1d24eede97e046224877

diff --git a/session_driver.c b/session_driver.c
index fe0db3e6..fd86ced1 100644
--- a/session_driver.c
+++ b/session_driver.c
@@ -53,6 +53,8 @@ static const int hwcaps[] = {
 	SR_CONF_SAMPLERATE,
 };
 
+extern struct sr_session *sr_current_session;
+
 static int receive_data(int fd, int revents, void *cb_data)
 {
 	struct sr_dev_inst *sdi;
@@ -160,7 +162,7 @@ static int receive_data(int fd, int revents, void *cb_data)
 	if (!got_data) {
 		packet.type = SR_DF_END;
 		sr_session_send(cb_data, &packet);
-		sr_session_source_remove(-1);
+		sr_session_source_remove(sr_current_session, -1);
 	}
 
 	return TRUE;
@@ -311,7 +313,7 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
 	std_session_send_df_header(cb_data, LOG_PREFIX);
 
 	/* freewheeling source */
-	sr_session_source_add(-1, 0, 0, receive_data, cb_data);
+	sr_session_source_add(sr_current_session, -1, 0, 0, receive_data, cb_data);
 
 	return SR_OK;
 }