]> sigrok.org Git - libsigrok.git/blobdiff - session.c
Add struct sr_session parameter to all session source backends.
[libsigrok.git] / session.c
index 266207259ca3c82e9c42c76208af3ae8afb88880..4c6095e00481777234c7255cb6789b1c9acb2896 100644 (file)
--- a/session.c
+++ b/session.c
 #include "libsigrok.h"
 #include "libsigrok-internal.h"
 
-/* Message logging helpers with driver-specific prefix string. */
-#define DRIVER_LOG_DOMAIN "session: "
-#define sr_log(l, s, args...) sr_log(l, DRIVER_LOG_DOMAIN s, ## args)
-#define sr_spew(s, args...) sr_spew(DRIVER_LOG_DOMAIN s, ## args)
-#define sr_dbg(s, args...) sr_dbg(DRIVER_LOG_DOMAIN s, ## args)
-#define sr_info(s, args...) sr_info(DRIVER_LOG_DOMAIN s, ## args)
-#define sr_warn(s, args...) sr_warn(DRIVER_LOG_DOMAIN s, ## args)
-#define sr_err(s, args...) sr_err(DRIVER_LOG_DOMAIN s, ## args)
+/** @cond PRIVATE */
+#define LOG_PREFIX "session"
+/** @endcond */
 
 /**
  * @file
@@ -50,7 +45,7 @@
 
 struct source {
        int timeout;
-       sr_receive_data_callback_t cb;
+       sr_receive_data_callback cb;
        void *cb_data;
 
        /* This is used to keep track of the object (fd, pollfd or channel) which is
@@ -60,75 +55,95 @@ struct source {
 };
 
 struct datafeed_callback {
-       sr_datafeed_callback_t cb;
+       sr_datafeed_callback cb;
        void *cb_data;
 };
 
-/* There can only be one session at a time. */
-/* 'session' is not static, it's used elsewhere (via 'extern'). */
-struct sr_session *session;
+/* There can currently only be one session at a time. */
+/* 'sr_current_session' is not static, it's used elsewhere (via 'extern'). */
+struct sr_session *sr_current_session;
 
 /**
  * Create a new session.
+ * Currently, there can be only one session at a time within the same process.
  *
- * @todo Should it use the file-global "session" variable or take an argument?
- *       The same question applies to all the other session functions.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_BUG A session exists already.
  *
- * @return A pointer to the newly allocated session, or NULL upon errors.
+ * @since 0.4.0
  */
-SR_API struct sr_session *sr_session_new(void)
+SR_API int sr_session_new(struct sr_session **session)
 {
-       if (!(session = g_try_malloc0(sizeof(struct sr_session)))) {
-               sr_err("Session malloc failed.");
-               return NULL;
+       if (sr_current_session) {
+               sr_err("%s: session already exists", __func__);
+               return SR_ERR_BUG;
        }
 
-       session->source_timeout = -1;
-       session->abort_session = FALSE;
-       g_mutex_init(&session->stop_mutex);
+       sr_current_session = g_malloc0(sizeof(struct sr_session));
+
+       sr_current_session->source_timeout = -1;
+       sr_current_session->running = FALSE;
+       sr_current_session->abort_session = FALSE;
+       g_mutex_init(&sr_current_session->stop_mutex);
 
-       return session;
+       *session = sr_current_session;
+
+       return SR_OK;
 }
 
 /**
- * Destroy the current session.
- *
+ * Destroy a session.
  * This frees up all memory used by the session.
  *
- * @return SR_OK upon success, SR_ERR_BUG if no session exists.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid session passed.
+ *
+ * @since 0.4.0
  */
-SR_API int sr_session_destroy(void)
+SR_API int sr_session_destroy(struct sr_session *session)
 {
        if (!session) {
                sr_err("%s: session was NULL", __func__);
-               return SR_ERR_BUG;
+               return SR_ERR_ARG;
        }
 
-       sr_session_dev_remove_all();
-
-       /* TODO: Error checks needed? */
-
+       sr_session_dev_remove_all(session);
        g_mutex_clear(&session->stop_mutex);
+       if (session->trigger)
+               sr_trigger_free(session->trigger);
 
        g_free(session);
-       session = NULL;
+
+       if (session == sr_current_session)
+               sr_current_session = NULL;
 
        return SR_OK;
 }
 
 /**
- * Remove all the devices from the current session.
+ * Remove all the devices from a session.
  *
  * The session itself (i.e., the struct sr_session) is not free'd and still
  * exists after this function returns.
  *
- * @return SR_OK upon success, SR_ERR_BUG if no session exists.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_BUG Invalid session passed.
+ *
+ * @since 0.4.0
  */
-SR_API int sr_session_dev_remove_all(void)
+SR_API int sr_session_dev_remove_all(struct sr_session *session)
 {
+       struct sr_dev_inst *sdi;
+       GSList *l;
+
        if (!session) {
                sr_err("%s: session was NULL", __func__);
-               return SR_ERR_BUG;
+               return SR_ERR_ARG;
+       }
+
+       for (l = session->devs; l; l = l->next) {
+               sdi = (struct sr_dev_inst *) l->data;
+               sdi->session = NULL;
        }
 
        g_slist_free(session->devs);
@@ -138,16 +153,21 @@ SR_API int sr_session_dev_remove_all(void)
 }
 
 /**
- * Add a device instance to the current session.
+ * Add a device instance to a session.
  *
- * @param sdi The device instance to add to the current session. Must not
+ * @param sdi The device instance to add to a session. Must not
  *            be NULL. Also, sdi->driver and sdi->driver->dev_open must
  *            not be NULL.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid argument.
+ *
+ * @since 0.4.0
  */
-SR_API int sr_session_dev_add(const struct sr_dev_inst *sdi)
+SR_API int sr_session_dev_add(struct sr_session *session,
+               struct sr_dev_inst *sdi)
 {
+       int ret;
 
        if (!sdi) {
                sr_err("%s: sdi was NULL", __func__);
@@ -156,7 +176,14 @@ SR_API int sr_session_dev_add(const struct sr_dev_inst *sdi)
 
        if (!session) {
                sr_err("%s: session was NULL", __func__);
-               return SR_ERR_BUG;
+               return SR_ERR_ARG;
+       }
+
+       /* If sdi->session is not NULL, the device is already in this or
+        * another session. */
+       if (sdi->session) {
+               sr_err("%s: already assigned to session", __func__);
+               return SR_ERR_ARG;
        }
 
        /* If sdi->driver is NULL, this is a virtual device. */
@@ -165,6 +192,7 @@ SR_API int sr_session_dev_add(const struct sr_dev_inst *sdi)
                       "a virtual device; continuing", __func__);
                /* Just add the device, don't run dev_open(). */
                session->devs = g_slist_append(session->devs, (gpointer)sdi);
+               sdi->session = session;
                return SR_OK;
        }
 
@@ -175,20 +203,69 @@ SR_API int sr_session_dev_add(const struct sr_dev_inst *sdi)
        }
 
        session->devs = g_slist_append(session->devs, (gpointer)sdi);
+       sdi->session = session;
+
+       if (session->running) {
+               /* Adding a device to a running session. Commit settings
+                * and start acquisition on that device now. */
+               if ((ret = sr_config_commit(sdi)) != SR_OK) {
+                       sr_err("Failed to commit device settings before "
+                              "starting acquisition in running session (%s)",
+                              sr_strerror(ret));
+                       return ret;
+               }
+               if ((ret = sdi->driver->dev_acquisition_start(sdi,
+                                               (void *)sdi)) != SR_OK) {
+                       sr_err("Failed to start acquisition of device in "
+                              "running session (%s)", sr_strerror(ret));
+                       return ret;
+               }
+       }
 
        return SR_OK;
 }
 
 /**
- * Remove all datafeed callbacks in the current session.
+ * List all device instances attached to a session.
+ *
+ * @param devlist A pointer where the device instance list will be
+ *                stored on return. If no devices are in the session,
+ *                this will be NULL. Each element in the list points
+ *                to a struct sr_dev_inst *.
+ *                The list must be freed by the caller, but not the
+ *                elements pointed to.
+ *
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid argument.
  *
- * @return SR_OK upon success, SR_ERR_BUG if no session exists.
+ * @since 0.4.0
  */
-SR_API int sr_session_datafeed_callback_remove_all(void)
+SR_API int sr_session_dev_list(struct sr_session *session, GSList **devlist)
+{
+       if (!session)
+               return SR_ERR_ARG;
+
+       if (!devlist)
+               return SR_ERR_ARG;
+
+       *devlist = g_slist_copy(session->devs);
+
+       return SR_OK;
+}
+
+/**
+ * Remove all datafeed callbacks in a session.
+ *
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid session passed.
+ *
+ * @since 0.4.0
+ */
+SR_API int sr_session_datafeed_callback_remove_all(struct sr_session *session)
 {
        if (!session) {
                sr_err("%s: session was NULL", __func__);
-               return SR_ERR_BUG;
+               return SR_ERR_ARG;
        }
 
        g_slist_free_full(session->datafeed_callbacks, g_free);
@@ -198,15 +275,19 @@ SR_API int sr_session_datafeed_callback_remove_all(void)
 }
 
 /**
- * Add a datafeed callback to the current session.
+ * Add a datafeed callback to a session.
  *
  * @param cb Function to call when a chunk of data is received.
  *           Must not be NULL.
  * @param cb_data Opaque pointer passed in by the caller.
  *
- * @return SR_OK upon success, SR_ERR_BUG if no session exists.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_BUG No session exists.
+ *
+ * @since 0.3.0
  */
-SR_API int sr_session_datafeed_callback_add(sr_datafeed_callback_t cb, void *cb_data)
+SR_API int sr_session_datafeed_callback_add(struct sr_session *session,
+               sr_datafeed_callback cb, void *cb_data)
 {
        struct datafeed_callback *cb_struct;
 
@@ -232,76 +313,152 @@ SR_API int sr_session_datafeed_callback_add(sr_datafeed_callback_t cb, void *cb_
        return SR_OK;
 }
 
-static int sr_session_run_poll(void)
+SR_API struct sr_trigger *sr_session_trigger_get(struct sr_session *session)
 {
+       return session->trigger;
+}
+
+SR_API int sr_session_trigger_set(struct sr_session *session, struct sr_trigger *trig)
+{
+       session->trigger = trig;
+
+       return SR_OK;
+}
+
+/**
+ * Call every device in the current session's callback.
+ *
+ * For sessions not driven by select loops such as sr_session_run(),
+ * but driven by another scheduler, this can be used to poll the devices
+ * from within that scheduler.
+ *
+ * @param block If TRUE, this call will wait for any of the session's
+ *              sources to fire an event on the file descriptors, or
+ *              any of their timeouts to activate. In other words, this
+ *              can be used as a select loop.
+ *              If FALSE, all sources have their callback run, regardless
+ *              of file descriptor or timeout status.
+ *
+ * @retval SR_OK Success.
+ * @retval SR_ERR Error occured.
+ */
+static int sr_session_iteration(gboolean block)
+{
+       struct sr_session *session = sr_current_session;
        unsigned int i;
        int ret;
 
-       while (session->num_sources > 0) {
-               ret = g_poll(session->pollfds, session->num_sources,
-                               session->source_timeout);
-               for (i = 0; i < session->num_sources; i++) {
-                       if (session->pollfds[i].revents > 0 || (ret == 0
-                               && session->source_timeout == session->sources[i].timeout)) {
-                               /*
-                                * Invoke the source's callback on an event,
-                                * or if the poll timed out and this source
-                                * asked for that timeout.
-                                */
-                               if (!session->sources[i].cb(session->pollfds[i].fd,
-                                               session->pollfds[i].revents,
-                                               session->sources[i].cb_data))
-                                       sr_session_source_remove(session->sources[i].poll_object);
-                       }
+       ret = g_poll(session->pollfds, session->num_sources,
+                       block ? session->source_timeout : 0);
+       for (i = 0; i < session->num_sources; i++) {
+               if (session->pollfds[i].revents > 0 || (ret == 0
+                       && session->source_timeout == session->sources[i].timeout)) {
                        /*
-                        * We want to take as little time as possible to stop
-                        * the session if we have been told to do so. Therefore,
-                        * we check the flag after processing every source, not
-                        * just once per main event loop.
+                        * Invoke the source's callback on an event,
+                        * or if the poll timed out and this source
+                        * asked for that timeout.
                         */
-                       g_mutex_lock(&session->stop_mutex);
-                       if (session->abort_session)
-                               sr_session_stop_sync();
-                       g_mutex_unlock(&session->stop_mutex);
+                       if (!session->sources[i].cb(session->pollfds[i].fd,
+                                       session->pollfds[i].revents,
+                                       session->sources[i].cb_data))
+                               sr_session_source_remove(session,
+                                               session->sources[i].poll_object);
+               }
+               /*
+                * We want to take as little time as possible to stop
+                * the session if we have been told to do so. Therefore,
+                * we check the flag after processing every source, not
+                * just once per main event loop.
+                */
+               g_mutex_lock(&session->stop_mutex);
+               if (session->abort_session) {
+                       sr_session_stop_sync(session);
+                       /* But once is enough. */
+                       session->abort_session = FALSE;
                }
+               g_mutex_unlock(&session->stop_mutex);
        }
 
        return SR_OK;
 }
 
+
+static int verify_trigger(struct sr_trigger *trigger)
+{
+       struct sr_trigger_stage *stage;
+       struct sr_trigger_match *match;
+       GSList *l, *m;
+
+       if (!trigger->stages) {
+               sr_err("No trigger stages defined.");
+               return SR_ERR;
+       }
+
+       sr_spew("Checking trigger:");
+       for (l = trigger->stages; l; l = l->next) {
+               stage = l->data;
+               if (!stage->matches) {
+                       sr_err("Stage %d has no matches defined.", stage->stage);
+                       return SR_ERR;
+               }
+               for (m = stage->matches; m; m = m->next) {
+                       match = m->data;
+                       if (!match->channel) {
+                               sr_err("Stage %d match has no channel.", stage->stage);
+                               return SR_ERR;
+                       }
+                       if (!match->match) {
+                               sr_err("Stage %d match is not defined.", stage->stage);
+                               return SR_ERR;
+                       }
+                       sr_spew("Stage %d match on channel %s, match %d", stage->stage,
+                                       match->channel->name, match->match);
+               }
+       }
+
+       return SR_OK;
+}
 /**
  * Start a session.
  *
- * There can only be one session at a time.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid session passed.
  *
- * @return SR_OK upon success, SR_ERR upon errors.
+ * @since 0.4.0
  */
-SR_API int sr_session_start(void)
+SR_API int sr_session_start(struct sr_session *session)
 {
        struct sr_dev_inst *sdi;
        GSList *l;
        int ret;
 
        if (!session) {
-               sr_err("%s: session was NULL; a session must be "
-                      "created before starting it.", __func__);
-               return SR_ERR_BUG;
+               sr_err("%s: session was NULL", __func__);
+               return SR_ERR_ARG;
        }
 
        if (!session->devs) {
                sr_err("%s: session->devs was NULL; a session "
                       "cannot be started without devices.", __func__);
-               return SR_ERR_BUG;
+               return SR_ERR_ARG;
        }
 
+       if (session->trigger && verify_trigger(session->trigger) != SR_OK)
+               return SR_ERR;
+
        sr_info("Starting.");
 
        ret = SR_OK;
        for (l = session->devs; l; l = l->next) {
                sdi = l->data;
+               if ((ret = sr_config_commit(sdi)) != SR_OK) {
+                       sr_err("Failed to commit device settings before "
+                              "starting acquisition (%s)", sr_strerror(ret));
+                       break;
+               }
                if ((ret = sdi->driver->dev_acquisition_start(sdi, sdi)) != SR_OK) {
                        sr_err("%s: could not start an acquisition "
-                              "(%d)", __func__, ret);
+                              "(%s)", __func__, sr_strerror(ret));
                        break;
                }
        }
@@ -312,24 +469,27 @@ SR_API int sr_session_start(void)
 }
 
 /**
- * Run the session.
+ * Run a session.
+ *
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid session passed.
  *
- * @return SR_OK upon success, SR_ERR_BUG upon errors.
+ * @since 0.4.0
  */
-SR_API int sr_session_run(void)
+SR_API int sr_session_run(struct sr_session *session)
 {
        if (!session) {
-               sr_err("%s: session was NULL; a session must be "
-                      "created first, before running it.", __func__);
-               return SR_ERR_BUG;
+               sr_err("%s: session was NULL", __func__);
+               return SR_ERR_ARG;
        }
 
        if (!session->devs) {
                /* TODO: Actually the case? */
                sr_err("%s: session->devs was NULL; a session "
                       "cannot be run without devices.", __func__);
-               return SR_ERR_BUG;
+               return SR_ERR_ARG;
        }
+       session->running = TRUE;
 
        sr_info("Running.");
 
@@ -340,31 +500,35 @@ SR_API int sr_session_run(void)
                        session->sources[0].cb(-1, 0, session->sources[0].cb_data);
        } else {
                /* Real sources, use g_poll() main loop. */
-               sr_session_run_poll();
+               while (session->num_sources)
+                       sr_session_iteration(TRUE);
        }
 
        return SR_OK;
 }
 
 /**
- * Stop the current session.
+ * Stop a session.
  *
- * The current session is stopped immediately, with all acquisition sessions
- * being stopped and hardware drivers cleaned up.
+ * The session is stopped immediately, with all acquisition sessions stopped
+ * and hardware drivers cleaned up.
  *
  * This must be called from within the session thread, to prevent freeing
  * resources that the session thread will try to use.
  *
- * @return SR_OK upon success, SR_ERR_BUG if no session exists.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid session passed.
+ *
+ * @private
  */
-SR_PRIV int sr_session_stop_sync(void)
+SR_PRIV int sr_session_stop_sync(struct sr_session *session)
 {
        struct sr_dev_inst *sdi;
        GSList *l;
 
        if (!session) {
                sr_err("%s: session was NULL", __func__);
-               return SR_ERR_BUG;
+               return SR_ERR_ARG;
        }
 
        sr_info("Stopping.");
@@ -376,33 +540,28 @@ SR_PRIV int sr_session_stop_sync(void)
                                sdi->driver->dev_acquisition_stop(sdi, sdi);
                }
        }
-
-       /*
-        * Some sources may not be necessarily associated with a device.
-        * Those sources may still be present even after stopping all devices.
-        * We need to make sure all sources are removed, or we risk running the
-        * session in an infinite loop.
-        */
-       while (session->num_sources)
-               sr_session_source_remove(session->sources[0].poll_object);
+       session->running = FALSE;
 
        return SR_OK;
 }
 
 /**
- * Stop the current session.
+ * Stop a session.
  *
- * The current session is stopped immediately, with all acquisition sessions
- * being stopped and hardware drivers cleaned up.
+ * The session is stopped immediately, with all acquisition sessions being
+ * stopped and hardware drivers cleaned up.
  *
  * If the session is run in a separate thread, this function will not block
  * until the session is finished executing. It is the caller's responsibility
  * to wait for the session thread to return before assuming that the session is
  * completely decommissioned.
  *
- * @return SR_OK upon success, SR_ERR_BUG if no session exists.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid session passed.
+ *
+ * @since 0.4.0
  */
-SR_API int sr_session_stop(void)
+SR_API int sr_session_stop(struct sr_session *session)
 {
        if (!session) {
                sr_err("%s: session was NULL", __func__);
@@ -438,8 +597,8 @@ static void datafeed_dump(const struct sr_datafeed_packet *packet)
                break;
        case SR_DF_LOGIC:
                logic = packet->payload;
-               sr_dbg("bus: Received SR_DF_LOGIC packet (%" PRIu64 " bytes).",
-                      logic->length);
+               sr_dbg("bus: Received SR_DF_LOGIC packet (%" PRIu64 " bytes, "
+                      "unitsize = %d).", logic->length, logic->unitsize);
                break;
        case SR_DF_ANALOG:
                analog = packet->payload;
@@ -469,7 +628,8 @@ static void datafeed_dump(const struct sr_datafeed_packet *packet)
  * @param sdi TODO.
  * @param packet The datafeed packet to send to the session bus.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid argument.
  *
  * @private
  */
@@ -489,7 +649,7 @@ SR_PRIV int sr_session_send(const struct sr_dev_inst *sdi,
                return SR_ERR_ARG;
        }
 
-       for (l = session->datafeed_callbacks; l; l = l->next) {
+       for (l = sr_current_session->datafeed_callbacks; l; l = l->next) {
                if (sr_log_loglevel_get() >= SR_LOG_DBG)
                        datafeed_dump(packet);
                cb_struct = l->data;
@@ -503,16 +663,18 @@ SR_PRIV int sr_session_send(const struct sr_dev_inst *sdi,
  * Add an event source for a file descriptor.
  *
  * @param pollfd The GPollFD.
- * @param timeout Max time to wait before the callback is called, ignored if 0.
+ * @param[in] timeout Max time to wait before the callback is called,
+ *              ignored if 0.
  * @param cb Callback function to add. Must not be NULL.
  * @param cb_data Data for the callback function. Can be NULL.
  * @param poll_object TODO.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments, or
- *         SR_ERR_MALLOC upon memory allocation errors.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid argument.
+ * @retval SR_ERR_MALLOC Memory allocation error.
  */
-static int _sr_session_source_add(GPollFD *pollfd, int timeout,
-       sr_receive_data_callback_t cb, void *cb_data, gintptr poll_object)
+static int _sr_session_source_add(struct sr_session *session, GPollFD *pollfd,
+               int timeout, sr_receive_data_callback cb, void *cb_data, gintptr poll_object)
 {
        struct source *new_sources, *s;
        GPollFD *new_pollfds;
@@ -563,18 +725,23 @@ static int _sr_session_source_add(GPollFD *pollfd, int timeout,
  * @param cb Callback function to add. Must not be NULL.
  * @param cb_data Data for the callback function. Can be NULL.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments, or
- *         SR_ERR_MALLOC upon memory allocation errors.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid argument.
+ * @retval SR_ERR_MALLOC Memory allocation error.
+ *
+ * @since 0.3.0
  */
-SR_API int sr_session_source_add(int fd, int events, int timeout,
-               sr_receive_data_callback_t cb, void *cb_data)
+SR_API int sr_session_source_add(struct sr_session *session, int fd,
+               int events, int timeout, sr_receive_data_callback cb, void *cb_data)
 {
        GPollFD p;
 
+       (void) session;
+
        p.fd = fd;
        p.events = events;
 
-       return _sr_session_source_add(&p, timeout, cb, cb_data, (gintptr)fd);
+       return _sr_session_source_add(session, &p, timeout, cb, cb_data, (gintptr)fd);
 }
 
 /**
@@ -585,13 +752,19 @@ SR_API int sr_session_source_add(int fd, int events, int timeout,
  * @param cb Callback function to add. Must not be NULL.
  * @param cb_data Data for the callback function. Can be NULL.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments, or
- *         SR_ERR_MALLOC upon memory allocation errors.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid argument.
+ * @retval SR_ERR_MALLOC Memory allocation error.
+ *
+ * @since 0.3.0
  */
-SR_API int sr_session_source_add_pollfd(GPollFD *pollfd, int timeout,
-               sr_receive_data_callback_t cb, void *cb_data)
+SR_API int sr_session_source_add_pollfd(struct sr_session *session,
+               GPollFD *pollfd, int timeout, sr_receive_data_callback cb,
+               void *cb_data)
 {
-       return _sr_session_source_add(pollfd, timeout, cb,
+       (void) session;
+
+       return _sr_session_source_add(session, pollfd, timeout, cb,
                                      cb_data, (gintptr)pollfd);
 }
 
@@ -604,14 +777,20 @@ SR_API int sr_session_source_add_pollfd(GPollFD *pollfd, int timeout,
  * @param cb Callback function to add. Must not be NULL.
  * @param cb_data Data for the callback function. Can be NULL.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments, or
- *         SR_ERR_MALLOC upon memory allocation errors.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid argument.
+ * @retval SR_ERR_MALLOC Memory allocation error.
+ *
+ * @since 0.3.0
  */
-SR_API int sr_session_source_add_channel(GIOChannel *channel, int events,
-               int timeout, sr_receive_data_callback_t cb, void *cb_data)
+SR_API int sr_session_source_add_channel(struct sr_session *session,
+               GIOChannel *channel, int events, int timeout,
+               sr_receive_data_callback cb, void *cb_data)
 {
        GPollFD p;
 
+       (void) session;
+
 #ifdef _WIN32
        g_io_channel_win32_make_pollfd(channel, events, &p);
 #else
@@ -619,7 +798,7 @@ SR_API int sr_session_source_add_channel(GIOChannel *channel, int events,
        p.events = events;
 #endif
 
-       return _sr_session_source_add(&p, timeout, cb, cb_data, (gintptr)channel);
+       return _sr_session_source_add(session, &p, timeout, cb, cb_data, (gintptr)channel);
 }
 
 /**
@@ -627,13 +806,14 @@ SR_API int sr_session_source_add_channel(GIOChannel *channel, int events,
  *
  * @todo Add more error checks and logging.
  *
- * @param channel The channel for which the source should be removed.
+ * @param poll_object The channel for which the source should be removed.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments, or
- *         SR_ERR_MALLOC upon memory allocation errors, SR_ERR_BUG upon
- *         internal errors.
+ * @retval SR_OK Success
+ * @retval SR_ERR_ARG Invalid arguments
+ * @retval SR_ERR_MALLOC Memory allocation error
+ * @retval SR_ERR_BUG Internal error
  */
-static int _sr_session_source_remove(gintptr poll_object)
+static int _sr_session_source_remove(struct sr_session *session, gintptr poll_object)
 {
        struct source *new_sources;
        GPollFD *new_pollfds;
@@ -685,13 +865,18 @@ static int _sr_session_source_remove(gintptr poll_object)
  *
  * @param fd The file descriptor for which the source should be removed.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments, or
- *         SR_ERR_MALLOC upon memory allocation errors, SR_ERR_BUG upon
- *         internal errors.
+ * @retval SR_OK Success
+ * @retval SR_ERR_ARG Invalid argument
+ * @retval SR_ERR_MALLOC Memory allocation error.
+ * @retval SR_ERR_BUG Internal error.
+ *
+ * @since 0.3.0
  */
-SR_API int sr_session_source_remove(int fd)
+SR_API int sr_session_source_remove(struct sr_session *session, int fd)
 {
-       return _sr_session_source_remove((gintptr)fd);
+       (void) session;
+
+       return _sr_session_source_remove(session, (gintptr)fd);
 }
 
 /**
@@ -702,10 +887,15 @@ SR_API int sr_session_source_remove(int fd)
  * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments, or
  *         SR_ERR_MALLOC upon memory allocation errors, SR_ERR_BUG upon
  *         internal errors.
+ *
+ * @since 0.2.0
  */
-SR_API int sr_session_source_remove_pollfd(GPollFD *pollfd)
+SR_API int sr_session_source_remove_pollfd(struct sr_session *session,
+               GPollFD *pollfd)
 {
-       return _sr_session_source_remove((gintptr)pollfd);
+       (void) session;
+
+       return _sr_session_source_remove(session, (gintptr)pollfd);
 }
 
 /**
@@ -713,13 +903,19 @@ SR_API int sr_session_source_remove_pollfd(GPollFD *pollfd)
  *
  * @param channel The channel for which the source should be removed.
  *
- * @return SR_OK upon success, SR_ERR_ARG upon invalid arguments, or
- *         SR_ERR_MALLOC upon memory allocation errors, SR_ERR_BUG upon
- *         internal errors.
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Invalid argument.
+ * @retval SR_ERR_MALLOC Memory allocation error.
+ * @return SR_ERR_BUG Internal error.
+ *
+ * @since 0.2.0
  */
-SR_API int sr_session_source_remove_channel(GIOChannel *channel)
+SR_API int sr_session_source_remove_channel(struct sr_session *session,
+               GIOChannel *channel)
 {
-       return _sr_session_source_remove((gintptr)channel);
+       (void) session;
+
+       return _sr_session_source_remove(session, (gintptr)channel);
 }
 
 /** @} */