Changeset 8a0c52a in mainline for uspace/lib/usbdev/src/devpoll.c

Timestamp:

2018-01-13T21:15:50Z (6 years ago)

Author:

Petr Manek <petr.manek@…>

Branches:

lfn, master, serial, ticket/834-toolchain-update, topic/msim-upgrade, topic/simplify-dev-export

Children:

91173333

Parents:

71f211f

Message:

usbdev: add polling join mechanism

File:

• uspace/lib/usbdev/src/devpoll.c (modified) (8 diffs)

uspace/lib/usbdev/src/devpoll.c

@@ -47 +47 @@
 #include <errno.h>
 #include <fibril.h>
+#include <fibril_synch.h>
 #include <stdbool.h>
 #include <stdlib.h>
@@ -62 +63 @@
 
         /** Device enpoint mapping to use for polling. */
-        usb_endpoint_mapping_t *polling_mapping;
+        usb_endpoint_mapping_t *ep_mapping;
 
         /** Size of the recieved data. */
@@ -69 +70 @@
         /** Data buffer. */
         uint8_t *buffer;
+
+        /** True if polling is currently in operation. */
+        volatile bool running;
+
+        /** True if polling should terminate as soon as possible. */
+        volatile bool joining;
+
+        /** Synchronization primitives for joining polling end. */
+        fibril_mutex_t guard;
+        fibril_condvar_t cv;
 };
+
+
+static void polling_fini(usb_device_polling_t *polling)
+{
+        /* Free the allocated memory. */
+        free(polling->buffer);
+        free(polling);
+}
 
 
@@ -80 +99 @@
 {
         assert(arg);
-        const usb_device_polling_t *data = arg;
+        usb_device_polling_t *data = arg;
+        data->running = true;
 
         /* Helper to reduce typing. */
         const usb_device_polling_config_t *params = &data->config;
 
-        usb_pipe_t *pipe = &data->polling_mapping->pipe;
+        usb_pipe_t *pipe = &data->ep_mapping->pipe;
 
         if (params->debug > 0) {
                 const usb_endpoint_mapping_t *mapping =
-                    data->polling_mapping;
+                    data->ep_mapping;
                 usb_log_debug("Poll (%p): started polling of `%s' - " \
                     "interface %d (%s,%d,%d), %zuB/%zu.\n",
@@ -136 +156 @@
                         const bool cont = (params->on_error == NULL) ? true :
                             params->on_error(data->dev, rc, params->arg);
-                        if (!cont) {
+                        if (!cont || data->joining) {
                                 /* This is user requested abort, erases failures. */
                                 failed_attempts = 0;
@@ -184 +204 @@
         }
 
-        /* Free the allocated memory. */
-        free(data->buffer);
-        free(data);
+        data->running = false;
+
+        if (data->joining) {
+                /* Notify joiners, if any. */
+                fibril_mutex_lock(&data->guard);
+                fibril_condvar_signal(&data->cv);
+                fibril_mutex_unlock(&data->guard);
+        } else {
+                polling_fini(data);
+        }
 
         return EOK;
@@ -235 +262 @@
         instance->request_size = req_size;
         instance->dev = dev;
-        instance->polling_mapping = epm;
+        instance->ep_mapping = epm;
+        instance->joining = false;
+        fibril_mutex_initialize(&instance->guard);
+        fibril_condvar_initialize(&instance->cv);
 
         /* Copy provided settings. */
@@ -265 +295 @@
 }
 
+int usb_device_poll_join(usb_device_polling_t *polling)
+{
+        int rc;
+        if (!polling)
+                return EBADMEM;
+
+        /* Set the flag */
+        polling->joining = true;
+
+        /* Unregister the pipe. */
+        if ((rc = usb_device_unmap_ep(polling->ep_mapping))) {
+                return rc;
+        }
+
+        /* Wait for the fibril to terminate. */
+        fibril_mutex_lock(&polling->guard);
+        while (polling->running)
+                fibril_condvar_wait(&polling->cv, &polling->guard);
+        fibril_mutex_unlock(&polling->guard);
+
+        /* Free the instance. */
+        polling_fini(polling);
+
+        return EOK;
+}
+
 /**
  * @}