diff options
Diffstat (limited to 'common/poller.c')
| -rw-r--r-- | common/poller.c | 26 |
1 files changed, 22 insertions, 4 deletions
diff --git a/common/poller.c b/common/poller.c index 496636d1d7..32795b641f 100644 --- a/common/poller.c +++ b/common/poller.c @@ -32,6 +32,7 @@ int poller_unregister(struct poller_struct *poller) if (!poller->registered) return -ENODEV; + list_del(&poller->list); poller->registered = 0; @@ -42,12 +43,14 @@ static void poller_async_callback(struct poller_struct *poller) { struct poller_async *pa = container_of(poller, struct poller_async, poller); + if (!pa->active) + return; + if (get_time_ns() < pa->end) return; + pa->active = 0; pa->fn(pa->ctx); - pa->fn = NULL; - poller_unregister(&pa->poller); } /* @@ -62,7 +65,9 @@ static void poller_async_callback(struct poller_struct *poller) */ int poller_async_cancel(struct poller_async *pa) { - return poller_unregister(&pa->poller); + pa->active = 0; + + return 0; } /* @@ -80,13 +85,26 @@ 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; + pa->active = 1; + + return 0; +} + +int poller_async_register(struct poller_async *pa) +{ + pa->poller.func = poller_async_callback; + pa->active = 0; return poller_register(&pa->poller); } +int poller_async_unregister(struct poller_async *pa) +{ + return poller_unregister(&pa->poller); +} + void poller_call(void) { struct poller_struct *poller, *tmp; |