summaryrefslogtreecommitdiffstats
path: root/common/poller.c
diff options
context:
space:
mode:
Diffstat (limited to 'common/poller.c')
-rw-r--r--common/poller.c58
1 files changed, 58 insertions, 0 deletions
diff --git a/common/poller.c b/common/poller.c
index 0583a53..496636d 100644
--- a/common/poller.c
+++ b/common/poller.c
@@ -11,24 +11,82 @@
#include <module.h>
#include <param.h>
#include <poller.h>
+#include <clock.h>
static LIST_HEAD(poller_list);
static int poller_active;
int poller_register(struct poller_struct *poller)
{
+ if (poller->registered)
+ return -EBUSY;
+
list_add_tail(&poller->list, &poller_list);
+ poller->registered = 1;
return 0;
}
int poller_unregister(struct poller_struct *poller)
{
+ if (!poller->registered)
+ return -ENODEV;
+
list_del(&poller->list);
+ poller->registered = 0;
return 0;
}
+static void poller_async_callback(struct poller_struct *poller)
+{
+ struct poller_async *pa = container_of(poller, struct poller_async, poller);
+
+ if (get_time_ns() < pa->end)
+ return;
+
+ pa->fn(pa->ctx);
+ pa->fn = NULL;
+ poller_unregister(&pa->poller);
+}
+
+/*
+ * Cancel an outstanding asynchronous function call
+ *
+ * @pa the poller that has been scheduled
+ *
+ * Cancel an outstanding function call. Returns 0 if the call
+ * has actually been cancelled or -ENODEV when the call wasn't
+ * scheduled.
+ *
+ */
+int poller_async_cancel(struct poller_async *pa)
+{
+ return poller_unregister(&pa->poller);
+}
+
+/*
+ * Call a function asynchronously
+ *
+ * @pa the poller to be used
+ * @delay The delay in nanoseconds
+ * @fn The function to call
+ * @ctx context pointer passed to the function
+ *
+ * This calls the passed function after a delay of delay_ns. Returns
+ * a pointer which can be used as a cookie to cancel a scheduled call.
+ */
+int poller_call_async(struct poller_async *pa, uint64_t delay_ns,
+ void (*fn)(void *), void *ctx)
+{
+ pa->ctx = ctx;
+ pa->poller.func = poller_async_callback;
+ pa->end = get_time_ns() + delay_ns;
+ pa->fn = fn;
+
+ return poller_register(&pa->poller);
+}
+
void poller_call(void)
{
struct poller_struct *poller, *tmp;