X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Flibsystemd%2Fsd-event%2Fsd-event.c;h=a2935abd64903ed79d3b86a3fd271e9923a5db31;hb=f7f53e9e6ebb9f656d880f5e779fc174a1d983bf;hp=70770fc7ad51e1c9f640eb6e899a14b2b417beb0;hpb=59bc1fd776b5270bcabe3c536852afd0c892fbe1;p=elogind.git diff --git a/src/libsystemd/sd-event/sd-event.c b/src/libsystemd/sd-event/sd-event.c index 70770fc7a..a2935abd6 100644 --- a/src/libsystemd/sd-event/sd-event.c +++ b/src/libsystemd/sd-event/sd-event.c @@ -33,6 +33,7 @@ #include "time-util.h" #include "missing.h" #include "set.h" +#include "list.h" #include "sd-event.h" @@ -42,6 +43,7 @@ typedef enum EventSourceType { SOURCE_IO, SOURCE_TIME_REALTIME, + SOURCE_TIME_BOOTTIME, SOURCE_TIME_MONOTONIC, SOURCE_TIME_REALTIME_ALARM, SOURCE_TIME_BOOTTIME_ALARM, @@ -51,11 +53,11 @@ typedef enum EventSourceType { SOURCE_POST, SOURCE_EXIT, SOURCE_WATCHDOG, - _SOUFCE_EVENT_SOURCE_TYPE_MAX, + _SOURCE_EVENT_SOURCE_TYPE_MAX, _SOURCE_EVENT_SOURCE_TYPE_INVALID = -1 } EventSourceType; -#define EVENT_SOURCE_IS_TIME(t) IN_SET((t), SOURCE_TIME_REALTIME, SOURCE_TIME_MONOTONIC, SOURCE_TIME_REALTIME_ALARM, SOURCE_TIME_BOOTTIME_ALARM) +#define EVENT_SOURCE_IS_TIME(t) IN_SET((t), SOURCE_TIME_REALTIME, SOURCE_TIME_BOOTTIME, SOURCE_TIME_MONOTONIC, SOURCE_TIME_REALTIME_ALARM, SOURCE_TIME_BOOTTIME_ALARM) struct sd_event_source { unsigned n_ref; @@ -64,10 +66,13 @@ struct sd_event_source { void *userdata; sd_event_handler_t prepare; + char *name; + EventSourceType type:5; int enabled:3; bool pending:1; bool dispatching:1; + bool floating:1; int64_t priority; unsigned pending_index; @@ -75,6 +80,8 @@ struct sd_event_source { unsigned pending_iteration; unsigned prepare_iteration; + LIST_FIELDS(sd_event_source, sources); + union { struct { sd_event_io_handler_t callback; @@ -126,6 +133,8 @@ struct clock_data { Prioq *earliest; Prioq *latest; usec_t next; + + bool needs_rearm:1; }; struct sd_event { @@ -138,10 +147,11 @@ struct sd_event { Prioq *pending; Prioq *prepare; - /* timerfd_create() only supports these four clocks so far. We + /* timerfd_create() only supports these five clocks so far. We * can add support for more clocks when the kernel learns to * deal with them, too. */ struct clock_data realtime; + struct clock_data boottime; struct clock_data monotonic; struct clock_data realtime_alarm; struct clock_data boottime_alarm; @@ -177,8 +187,12 @@ struct sd_event { usec_t watchdog_last, watchdog_period; unsigned n_sources; + + LIST_HEAD(sd_event_source, sources); }; +static void source_disconnect(sd_event_source *s); + static int pending_prioq_compare(const void *a, const void *b) { const sd_event_source *x = a, *y = b; @@ -349,7 +363,16 @@ static void free_clock_data(struct clock_data *d) { } static void event_free(sd_event *e) { + sd_event_source *s; + assert(e); + + while ((s = e->sources)) { + assert(s->floating); + source_disconnect(s); + sd_event_source_unref(s); + } + assert(e->n_sources == 0); if (e->default_event_ptr) @@ -360,6 +383,7 @@ static void event_free(sd_event *e) { safe_close(e->watchdog_fd); free_clock_data(&e->realtime); + free_clock_data(&e->boottime); free_clock_data(&e->monotonic); free_clock_data(&e->realtime_alarm); free_clock_data(&e->boottime_alarm); @@ -386,10 +410,10 @@ _public_ int sd_event_new(sd_event** ret) { return -ENOMEM; e->n_ref = 1; - e->signal_fd = e->watchdog_fd = e->epoll_fd = e->realtime.fd = e->monotonic.fd = e->realtime_alarm.fd = e->boottime_alarm.fd = -1; - e->realtime.next = e->monotonic.next = e->realtime_alarm.next = e->boottime_alarm.next = (usec_t) -1; + e->signal_fd = e->watchdog_fd = e->epoll_fd = e->realtime.fd = e->boottime.fd = e->monotonic.fd = e->realtime_alarm.fd = e->boottime_alarm.fd = -1; + e->realtime.next = e->boottime.next = e->monotonic.next = e->realtime_alarm.next = e->boottime_alarm.next = USEC_INFINITY; e->original_pid = getpid(); - e->perturb = (usec_t) -1; + e->perturb = USEC_INFINITY; assert_se(sigemptyset(&e->sigset) == 0); @@ -500,6 +524,9 @@ static clockid_t event_source_type_to_clock(EventSourceType t) { case SOURCE_TIME_REALTIME: return CLOCK_REALTIME; + case SOURCE_TIME_BOOTTIME: + return CLOCK_BOOTTIME; + case SOURCE_TIME_MONOTONIC: return CLOCK_MONOTONIC; @@ -521,6 +548,9 @@ static EventSourceType clock_to_event_source_type(clockid_t clock) { case CLOCK_REALTIME: return SOURCE_TIME_REALTIME; + case CLOCK_BOOTTIME: + return SOURCE_TIME_BOOTTIME; + case CLOCK_MONOTONIC: return SOURCE_TIME_MONOTONIC; @@ -543,6 +573,9 @@ static struct clock_data* event_get_clock_data(sd_event *e, EventSourceType t) { case SOURCE_TIME_REALTIME: return &e->realtime; + case SOURCE_TIME_BOOTTIME: + return &e->boottime; + case SOURCE_TIME_MONOTONIC: return &e->monotonic; @@ -557,86 +590,104 @@ static struct clock_data* event_get_clock_data(sd_event *e, EventSourceType t) { } } -static void source_free(sd_event_source *s) { +static void source_disconnect(sd_event_source *s) { + sd_event *event; + assert(s); - if (s->event) { - assert(s->event->n_sources > 0); + if (!s->event) + return; - switch (s->type) { + assert(s->event->n_sources > 0); - case SOURCE_IO: - if (s->io.fd >= 0) - source_io_unregister(s); + switch (s->type) { - break; + case SOURCE_IO: + if (s->io.fd >= 0) + source_io_unregister(s); - case SOURCE_TIME_REALTIME: - case SOURCE_TIME_MONOTONIC: - case SOURCE_TIME_REALTIME_ALARM: - case SOURCE_TIME_BOOTTIME_ALARM: { - struct clock_data *d; + break; - d = event_get_clock_data(s->event, s->type); - assert(d); + case SOURCE_TIME_REALTIME: + case SOURCE_TIME_BOOTTIME: + case SOURCE_TIME_MONOTONIC: + case SOURCE_TIME_REALTIME_ALARM: + case SOURCE_TIME_BOOTTIME_ALARM: { + struct clock_data *d; - prioq_remove(d->earliest, s, &s->time.earliest_index); - prioq_remove(d->latest, s, &s->time.latest_index); - break; + d = event_get_clock_data(s->event, s->type); + assert(d); + + prioq_remove(d->earliest, s, &s->time.earliest_index); + prioq_remove(d->latest, s, &s->time.latest_index); + d->needs_rearm = true; + break; + } + + case SOURCE_SIGNAL: + if (s->signal.sig > 0) { + if (s->signal.sig != SIGCHLD || s->event->n_enabled_child_sources == 0) + assert_se(sigdelset(&s->event->sigset, s->signal.sig) == 0); + + if (s->event->signal_sources) + s->event->signal_sources[s->signal.sig] = NULL; } - case SOURCE_SIGNAL: - if (s->signal.sig > 0) { - if (s->signal.sig != SIGCHLD || s->event->n_enabled_child_sources == 0) - assert_se(sigdelset(&s->event->sigset, s->signal.sig) == 0); + break; - if (s->event->signal_sources) - s->event->signal_sources[s->signal.sig] = NULL; + case SOURCE_CHILD: + if (s->child.pid > 0) { + if (s->enabled != SD_EVENT_OFF) { + assert(s->event->n_enabled_child_sources > 0); + s->event->n_enabled_child_sources--; } - break; + if (!s->event->signal_sources || !s->event->signal_sources[SIGCHLD]) + assert_se(sigdelset(&s->event->sigset, SIGCHLD) == 0); - case SOURCE_CHILD: - if (s->child.pid > 0) { - if (s->enabled != SD_EVENT_OFF) { - assert(s->event->n_enabled_child_sources > 0); - s->event->n_enabled_child_sources--; - } + hashmap_remove(s->event->child_sources, INT_TO_PTR(s->child.pid)); + } - if (!s->event->signal_sources || !s->event->signal_sources[SIGCHLD]) - assert_se(sigdelset(&s->event->sigset, SIGCHLD) == 0); + break; - hashmap_remove(s->event->child_sources, INT_TO_PTR(s->child.pid)); - } + case SOURCE_DEFER: + /* nothing */ + break; - break; + case SOURCE_POST: + set_remove(s->event->post_sources, s); + break; - case SOURCE_DEFER: - /* nothing */ - break; + case SOURCE_EXIT: + prioq_remove(s->event->exit, s, &s->exit.prioq_index); + break; - case SOURCE_POST: - set_remove(s->event->post_sources, s); - break; + default: + assert_not_reached("Wut? I shouldn't exist."); + } - case SOURCE_EXIT: - prioq_remove(s->event->exit, s, &s->exit.prioq_index); - break; + if (s->pending) + prioq_remove(s->event->pending, s, &s->pending_index); - default: - assert_not_reached("Wut? I shouldn't exist."); - } + if (s->prepare) + prioq_remove(s->event->prepare, s, &s->prepare_index); - if (s->pending) - prioq_remove(s->event->pending, s, &s->pending_index); + event = s->event; - if (s->prepare) - prioq_remove(s->event->prepare, s, &s->prepare_index); + s->type = _SOURCE_EVENT_SOURCE_TYPE_INVALID; + s->event = NULL; + LIST_REMOVE(sources, event->sources, s); + event->n_sources--; - s->event->n_sources--; - sd_event_unref(s->event); - } + if (!s->floating) + sd_event_unref(event); +} +static void source_free(sd_event_source *s) { + assert(s); + + source_disconnect(s); + free(s->name); free(s); } @@ -670,12 +721,13 @@ static int source_set_pending(sd_event_source *s, bool b) { prioq_reshuffle(d->earliest, s, &s->time.earliest_index); prioq_reshuffle(d->latest, s, &s->time.latest_index); + d->needs_rearm = true; } return 0; } -static sd_event_source *source_new(sd_event *e, EventSourceType type) { +static sd_event_source *source_new(sd_event *e, bool floating, EventSourceType type) { sd_event_source *s; assert(e); @@ -685,10 +737,15 @@ static sd_event_source *source_new(sd_event *e, EventSourceType type) { return NULL; s->n_ref = 1; - s->event = sd_event_ref(e); + s->event = e; + s->floating = floating; s->type = type; s->pending_index = s->prepare_index = PRIOQ_IDX_NULL; + if (!floating) + sd_event_ref(e); + + LIST_PREPEND(sources, e->sources, s); e->n_sources ++; return s; @@ -709,11 +766,10 @@ _public_ int sd_event_add_io( assert_return(fd >= 0, -EINVAL); assert_return(!(events & ~(EPOLLIN|EPOLLOUT|EPOLLRDHUP|EPOLLPRI|EPOLLERR|EPOLLHUP|EPOLLET)), -EINVAL); assert_return(callback, -EINVAL); - assert_return(ret, -EINVAL); assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); assert_return(!event_pid_changed(e), -ECHILD); - s = source_new(e, SOURCE_IO); + s = source_new(e, !ret, SOURCE_IO); if (!s) return -ENOMEM; @@ -726,10 +782,12 @@ _public_ int sd_event_add_io( r = source_io_register(s, s->enabled, events); if (r < 0) { source_free(s); - return -errno; + return r; } - *ret = s; + if (ret) + *ret = s; + return 0; } @@ -745,7 +803,7 @@ static void initialize_perturb(sd_event *e) { bit. Here, we calculate a perturbation usec offset from the boot ID. */ - if (_likely_(e->perturb != (usec_t) -1)) + if (_likely_(e->perturb != USEC_INFINITY)) return; if (sd_id128_get_boot(&bootid) >= 0) @@ -798,7 +856,6 @@ _public_ int sd_event_add_time( int r; assert_return(e, -EINVAL); - assert_return(ret, -EINVAL); assert_return(usec != (uint64_t) -1, -EINVAL); assert_return(accuracy != (uint64_t) -1, -EINVAL); assert_return(callback, -EINVAL); @@ -829,7 +886,7 @@ _public_ int sd_event_add_time( return r; } - s = source_new(e, type); + s = source_new(e, !ret, type); if (!s) return -ENOMEM; @@ -840,6 +897,8 @@ _public_ int sd_event_add_time( s->userdata = userdata; s->enabled = SD_EVENT_ONESHOT; + d->needs_rearm = true; + r = prioq_put(d->earliest, s, &s->time.earliest_index); if (r < 0) goto fail; @@ -848,7 +907,9 @@ _public_ int sd_event_add_time( if (r < 0) goto fail; - *ret = s; + if (ret) + *ret = s; + return 0; fail: @@ -906,7 +967,6 @@ _public_ int sd_event_add_signal( assert_return(e, -EINVAL); assert_return(sig > 0, -EINVAL); assert_return(sig < _NSIG, -EINVAL); - assert_return(ret, -EINVAL); assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); assert_return(!event_pid_changed(e), -ECHILD); @@ -927,7 +987,7 @@ _public_ int sd_event_add_signal( } else if (e->signal_sources[sig]) return -EBUSY; - s = source_new(e, SOURCE_SIGNAL); + s = source_new(e, !ret, SOURCE_SIGNAL); if (!s) return -ENOMEM; @@ -947,7 +1007,9 @@ _public_ int sd_event_add_signal( } } - *ret = s; + if (ret) + *ret = s; + return 0; } @@ -967,7 +1029,6 @@ _public_ int sd_event_add_child( assert_return(!(options & ~(WEXITED|WSTOPPED|WCONTINUED)), -EINVAL); assert_return(options != 0, -EINVAL); assert_return(callback, -EINVAL); - assert_return(ret, -EINVAL); assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); assert_return(!event_pid_changed(e), -ECHILD); @@ -978,7 +1039,7 @@ _public_ int sd_event_add_child( if (hashmap_contains(e->child_sources, INT_TO_PTR(pid))) return -EBUSY; - s = source_new(e, SOURCE_CHILD); + s = source_new(e, !ret, SOURCE_CHILD); if (!s) return -ENOMEM; @@ -1002,13 +1063,15 @@ _public_ int sd_event_add_child( r = event_update_signal_fd(e); if (r < 0) { source_free(s); - return -errno; + return r; } } e->need_process_child = true; - *ret = s; + if (ret) + *ret = s; + return 0; } @@ -1023,11 +1086,10 @@ _public_ int sd_event_add_defer( assert_return(e, -EINVAL); assert_return(callback, -EINVAL); - assert_return(ret, -EINVAL); assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); assert_return(!event_pid_changed(e), -ECHILD); - s = source_new(e, SOURCE_DEFER); + s = source_new(e, !ret, SOURCE_DEFER); if (!s) return -ENOMEM; @@ -1041,7 +1103,9 @@ _public_ int sd_event_add_defer( return r; } - *ret = s; + if (ret) + *ret = s; + return 0; } @@ -1056,7 +1120,6 @@ _public_ int sd_event_add_post( assert_return(e, -EINVAL); assert_return(callback, -EINVAL); - assert_return(ret, -EINVAL); assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); assert_return(!event_pid_changed(e), -ECHILD); @@ -1064,7 +1127,7 @@ _public_ int sd_event_add_post( if (r < 0) return r; - s = source_new(e, SOURCE_POST); + s = source_new(e, !ret, SOURCE_POST); if (!s) return -ENOMEM; @@ -1078,7 +1141,9 @@ _public_ int sd_event_add_post( return r; } - *ret = s; + if (ret) + *ret = s; + return 0; } @@ -1093,7 +1158,6 @@ _public_ int sd_event_add_exit( assert_return(e, -EINVAL); assert_return(callback, -EINVAL); - assert_return(ret, -EINVAL); assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); assert_return(!event_pid_changed(e), -ECHILD); @@ -1103,7 +1167,7 @@ _public_ int sd_event_add_exit( return -ENOMEM; } - s = source_new(e, SOURCE_EXIT); + s = source_new(e, !ret, SOURCE_EXIT); if (!s) return -ENOMEM; @@ -1118,7 +1182,9 @@ _public_ int sd_event_add_exit( return r; } - *ret = s; + if (ret) + *ret = s; + return 0; } @@ -1151,6 +1217,8 @@ _public_ sd_event_source* sd_event_source_unref(sd_event_source *s) { if (s->dispatching) { if (s->type == SOURCE_IO) source_io_unregister(s); + + source_disconnect(s); } else source_free(s); } @@ -1158,6 +1226,32 @@ _public_ sd_event_source* sd_event_source_unref(sd_event_source *s) { return NULL; } +_public_ int sd_event_source_set_name(sd_event_source *s, const char *name) { + char *new_name = NULL; + + assert_return(s, -EINVAL); + + if (name) { + new_name = strdup(name); + if (!new_name) + return -ENOMEM; + } + + free(s->name); + s->name = new_name; + + return 0; +} + +_public_ int sd_event_source_get_name(sd_event_source *s, const char **name) { + assert_return(s, -EINVAL); + assert_return(name, -EINVAL); + + *name = s->name; + + return 0; +} + _public_ sd_event *sd_event_source_get_event(sd_event_source *s) { assert_return(s, NULL); @@ -1236,7 +1330,8 @@ _public_ int sd_event_source_set_io_events(sd_event_source *s, uint32_t events) assert_return(s->event->state != SD_EVENT_FINISHED, -ESTALE); assert_return(!event_pid_changed(s->event), -ECHILD); - if (s->io.events == events) + /* edge-triggered updates are never skipped, so we can reset edges */ + if (s->io.events == events && !(events & EPOLLET)) return 0; if (s->enabled != SD_EVENT_OFF) { @@ -1313,9 +1408,13 @@ _public_ int sd_event_source_set_enabled(sd_event_source *s, int m) { assert_return(s, -EINVAL); assert_return(m == SD_EVENT_OFF || m == SD_EVENT_ON || m == SD_EVENT_ONESHOT, -EINVAL); - assert_return(s->event->state != SD_EVENT_FINISHED, -ESTALE); assert_return(!event_pid_changed(s->event), -ECHILD); + /* If we are dead anyway, we are fine with turning off + * sources, but everything else needs to fail. */ + if (s->event->state == SD_EVENT_FINISHED) + return m == SD_EVENT_OFF ? 0 : -ESTALE; + if (s->enabled == m) return 0; @@ -1332,6 +1431,7 @@ _public_ int sd_event_source_set_enabled(sd_event_source *s, int m) { break; case SOURCE_TIME_REALTIME: + case SOURCE_TIME_BOOTTIME: case SOURCE_TIME_MONOTONIC: case SOURCE_TIME_REALTIME_ALARM: case SOURCE_TIME_BOOTTIME_ALARM: { @@ -1343,6 +1443,7 @@ _public_ int sd_event_source_set_enabled(sd_event_source *s, int m) { prioq_reshuffle(d->earliest, s, &s->time.earliest_index); prioq_reshuffle(d->latest, s, &s->time.latest_index); + d->needs_rearm = true; break; } @@ -1394,6 +1495,7 @@ _public_ int sd_event_source_set_enabled(sd_event_source *s, int m) { break; case SOURCE_TIME_REALTIME: + case SOURCE_TIME_BOOTTIME: case SOURCE_TIME_MONOTONIC: case SOURCE_TIME_REALTIME_ALARM: case SOURCE_TIME_BOOTTIME_ALARM: { @@ -1405,6 +1507,7 @@ _public_ int sd_event_source_set_enabled(sd_event_source *s, int m) { prioq_reshuffle(d->earliest, s, &s->time.earliest_index); prioq_reshuffle(d->latest, s, &s->time.latest_index); + d->needs_rearm = true; break; } @@ -1482,6 +1585,7 @@ _public_ int sd_event_source_set_time(sd_event_source *s, uint64_t usec) { prioq_reshuffle(d->earliest, s, &s->time.earliest_index); prioq_reshuffle(d->latest, s, &s->time.latest_index); + d->needs_rearm = true; return 0; } @@ -1516,6 +1620,7 @@ _public_ int sd_event_source_set_time_accuracy(sd_event_source *s, uint64_t usec assert(d); prioq_reshuffle(d->latest, s, &s->time.latest_index); + d->needs_rearm = true; return 0; } @@ -1679,13 +1784,18 @@ static int event_arm_timer( assert(e); assert(d); + if (!d->needs_rearm) + return 0; + else + d->needs_rearm = false; + a = prioq_peek(d->earliest); if (!a || a->enabled == SD_EVENT_OFF) { if (d->fd < 0) return 0; - if (d->next == (usec_t) -1) + if (d->next == USEC_INFINITY) return 0; /* disarm */ @@ -1693,7 +1803,7 @@ static int event_arm_timer( if (r < 0) return r; - d->next = (usec_t) -1; + d->next = USEC_INFINITY; return 0; } @@ -1761,7 +1871,7 @@ static int flush_timer(sd_event *e, int fd, uint32_t events, usec_t *next) { return -EIO; if (next) - *next = (usec_t) -1; + *next = USEC_INFINITY; return 0; } @@ -1791,6 +1901,7 @@ static int process_timer( prioq_reshuffle(d->earliest, s, &s->time.earliest_index); prioq_reshuffle(d->latest, s, &s->time.latest_index); + d->needs_rearm = true; } return 0; @@ -1868,14 +1979,13 @@ static int process_signal(sd_event *e, uint32_t events) { int r; assert(e); - assert(e->signal_sources); assert_return(events == EPOLLIN, -EIO); for (;;) { struct signalfd_siginfo si; ssize_t ss; - sd_event_source *s; + sd_event_source *s = NULL; ss = read(e->signal_fd, &si, sizeof(si)); if (ss < 0) { @@ -1890,16 +2000,19 @@ static int process_signal(sd_event *e, uint32_t events) { read_one = true; - s = e->signal_sources[si.ssi_signo]; if (si.ssi_signo == SIGCHLD) { r = process_child(e); if (r < 0) return r; - if (r > 0 || !s) + if (r > 0) continue; - } else - if (!s) - return -EIO; + } + + if (e->signal_sources) + s = e->signal_sources[si.ssi_signo]; + + if (!s) + continue; s->signal.siginfo = si; r = source_set_pending(s, true); @@ -1952,6 +2065,7 @@ static int source_dispatch(sd_event_source *s) { break; case SOURCE_TIME_REALTIME: + case SOURCE_TIME_BOOTTIME: case SOURCE_TIME_MONOTONIC: case SOURCE_TIME_REALTIME_ALARM: case SOURCE_TIME_BOOTTIME_ALARM: @@ -1991,7 +2105,7 @@ static int source_dispatch(sd_event_source *s) { break; case SOURCE_WATCHDOG: - case _SOUFCE_EVENT_SOURCE_TYPE_MAX: + case _SOURCE_EVENT_SOURCE_TYPE_MAX: case _SOURCE_EVENT_SOURCE_TYPE_INVALID: assert_not_reached("Wut? I shouldn't exist."); } @@ -2125,11 +2239,8 @@ static int process_watchdog(sd_event *e) { return arm_watchdog(e); } -_public_ int sd_event_run(sd_event *e, uint64_t timeout) { - struct epoll_event *ev_queue; - unsigned ev_queue_max; - sd_event_source *p; - int r, i, m; +_public_ int sd_event_prepare(sd_event *e) { + int r; assert_return(e, -EINVAL); assert_return(!event_pid_changed(e), -ECHILD); @@ -2137,34 +2248,64 @@ _public_ int sd_event_run(sd_event *e, uint64_t timeout) { assert_return(e->state == SD_EVENT_PASSIVE, -EBUSY); if (e->exit_requested) - return dispatch_exit(e); + goto pending; - sd_event_ref(e); e->iteration++; - e->state = SD_EVENT_RUNNING; r = event_prepare(e); if (r < 0) - goto finish; + return r; r = event_arm_timer(e, &e->realtime); if (r < 0) - goto finish; + return r; + + r = event_arm_timer(e, &e->boottime); + if (r < 0) + return r; r = event_arm_timer(e, &e->monotonic); if (r < 0) - goto finish; + return r; r = event_arm_timer(e, &e->realtime_alarm); if (r < 0) - goto finish; + return r; r = event_arm_timer(e, &e->boottime_alarm); if (r < 0) - goto finish; + return r; if (event_next_pending(e) || e->need_process_child) - timeout = 0; + goto pending; + + e->state = SD_EVENT_PREPARED; + + return 0; + +pending: + e->state = SD_EVENT_PREPARED; + r = sd_event_wait(e, 0); + if (r == 0) + e->state = SD_EVENT_PREPARED; + + return r; +} + +_public_ int sd_event_wait(sd_event *e, uint64_t timeout) { + struct epoll_event *ev_queue; + unsigned ev_queue_max; + int r, m, i; + + assert_return(e, -EINVAL); + assert_return(!event_pid_changed(e), -ECHILD); + assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); + assert_return(e->state == SD_EVENT_PREPARED, -EBUSY); + + if (e->exit_requested) { + e->state = SD_EVENT_PENDING; + return 1; + } ev_queue_max = CLAMP(e->n_sources, 1U, EPOLL_QUEUE_MAX); ev_queue = newa(struct epoll_event, ev_queue_max); @@ -2172,7 +2313,13 @@ _public_ int sd_event_run(sd_event *e, uint64_t timeout) { m = epoll_wait(e->epoll_fd, ev_queue, ev_queue_max, timeout == (uint64_t) -1 ? -1 : (int) ((timeout + USEC_PER_MSEC - 1) / USEC_PER_MSEC)); if (m < 0) { - r = errno == EAGAIN || errno == EINTR ? 1 : -errno; + if (errno == EINTR) { + e->state = SD_EVENT_PENDING; + return 1; + } + + r = -errno; + goto finish; } @@ -2183,6 +2330,8 @@ _public_ int sd_event_run(sd_event *e, uint64_t timeout) { if (ev_queue[i].data.ptr == INT_TO_PTR(SOURCE_TIME_REALTIME)) r = flush_timer(e, e->realtime.fd, ev_queue[i].events, &e->realtime.next); + else if (ev_queue[i].data.ptr == INT_TO_PTR(SOURCE_TIME_BOOTTIME)) + r = flush_timer(e, e->boottime.fd, ev_queue[i].events, &e->boottime.next); else if (ev_queue[i].data.ptr == INT_TO_PTR(SOURCE_TIME_MONOTONIC)) r = flush_timer(e, e->monotonic.fd, ev_queue[i].events, &e->monotonic.next); else if (ev_queue[i].data.ptr == INT_TO_PTR(SOURCE_TIME_REALTIME_ALARM)) @@ -2208,6 +2357,10 @@ _public_ int sd_event_run(sd_event *e, uint64_t timeout) { if (r < 0) goto finish; + r = process_timer(e, e->timestamp_boottime, &e->boottime); + if (r < 0) + goto finish; + r = process_timer(e, e->timestamp.monotonic, &e->monotonic); if (r < 0) goto finish; @@ -2226,21 +2379,71 @@ _public_ int sd_event_run(sd_event *e, uint64_t timeout) { goto finish; } - p = event_next_pending(e); - if (!p) { - r = 1; - goto finish; + if (event_next_pending(e)) { + e->state = SD_EVENT_PENDING; + + return 1; } - r = source_dispatch(p); + r = 0; finish: e->state = SD_EVENT_PASSIVE; - sd_event_unref(e); return r; } +_public_ int sd_event_dispatch(sd_event *e) { + sd_event_source *p; + int r; + + assert_return(e, -EINVAL); + assert_return(!event_pid_changed(e), -ECHILD); + assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); + assert_return(e->state == SD_EVENT_PENDING, -EBUSY); + + if (e->exit_requested) + return dispatch_exit(e); + + p = event_next_pending(e); + if (p) { + sd_event_ref(e); + + e->state = SD_EVENT_RUNNING; + r = source_dispatch(p); + e->state = SD_EVENT_PASSIVE; + + sd_event_unref(e); + + return r; + } + + e->state = SD_EVENT_PASSIVE; + + return 1; +} + +_public_ int sd_event_run(sd_event *e, uint64_t timeout) { + int r; + + assert_return(e, -EINVAL); + assert_return(!event_pid_changed(e), -ECHILD); + assert_return(e->state != SD_EVENT_FINISHED, -ESTALE); + assert_return(e->state == SD_EVENT_PASSIVE, -EBUSY); + + r = sd_event_prepare(e); + if (r > 0) + return sd_event_dispatch(e); + else if (r < 0) + return r; + + r = sd_event_wait(e, timeout); + if (r > 0) + return sd_event_dispatch(e); + else + return r; +} + _public_ int sd_event_loop(sd_event *e) { int r; @@ -2263,6 +2466,14 @@ finish: return r; } +_public_ int sd_event_get_fd(sd_event *e) { + + assert_return(e, -EINVAL); + assert_return(!event_pid_changed(e), -ECHILD); + + return e->epoll_fd; +} + _public_ int sd_event_get_state(sd_event *e) { assert_return(e, -EINVAL); assert_return(!event_pid_changed(e), -ECHILD); @@ -2313,6 +2524,7 @@ _public_ int sd_event_now(sd_event *e, clockid_t clock, uint64_t *usec) { *usec = e->timestamp.monotonic; break; + case CLOCK_BOOTTIME: case CLOCK_BOOTTIME_ALARM: *usec = e->timestamp_boottime; break;