diff options
| -rw-r--r-- | liberty.c | 139 |
1 files changed, 89 insertions, 50 deletions
@@ -3916,7 +3916,7 @@ test_run (struct test *self) // --- Connector --------------------------------------------------------------- -#ifdef LIBERTY_WANT_POLLER +#if defined LIBERTY_WANT_POLLER && defined LIBERTY_WANT_ASYNC // This is a helper that tries to establish a connection with any address on // a given list. Sadly it also introduces a bit of a callback hell. @@ -3924,10 +3924,14 @@ test_run (struct test *self) struct connector_target { LIST_HEADER (struct connector_target) + struct connector *connector; ///< Parent connector char *hostname; ///< Target hostname or address char *service; ///< Target service name or port + struct async *getaddrinfo_event; ///< Address resolution + struct error *getaddrinfo_error; ///< Address resolution error + struct addrinfo *results; ///< Resolved target struct addrinfo *iter; ///< Current endpoint }; @@ -3935,13 +3939,18 @@ struct connector_target static struct connector_target * connector_target_new (void) { - struct connector_target *self = xmalloc (sizeof *self); + struct connector_target *self = xcalloc (1, sizeof *self); return self; } static void connector_target_destroy (struct connector_target *self) { + if (self->getaddrinfo_event) + async_cancel (self->getaddrinfo_event); + if (self->getaddrinfo_error) + error_free (self->getaddrinfo_error); + free (self->hostname); free (self->service); freeaddrinfo (self->results); @@ -3952,6 +3961,7 @@ connector_target_destroy (struct connector_target *self) struct connector { + struct poller *poller; ///< Poller int socket; ///< Socket FD for the connection struct poller_fd connected_event; ///< We've connected or failed struct connector_target *targets; ///< Targets @@ -3983,15 +3993,18 @@ connector_notify_connecting (struct connector *self, return; const char *real_host = target->hostname; - - // We don't really need this, so we can let it quietly fail char buf[NI_MAXHOST]; - int err = getnameinfo (gai_iter->ai_addr, gai_iter->ai_addrlen, - buf, sizeof buf, NULL, 0, NI_NUMERICHOST); - if (err) - LOG_FUNC_FAILURE ("getnameinfo", gai_strerror (err)); - else - real_host = buf; + + if (gai_iter) + { + // We don't really need this, so we can let it quietly fail + int err = getnameinfo (gai_iter->ai_addr, gai_iter->ai_addrlen, + buf, sizeof buf, NULL, 0, NI_NUMERICHOST); + if (err) + LOG_FUNC_FAILURE ("getnameinfo", gai_strerror (err)); + else + real_host = buf; + } char *address = format_host_port_pair (real_host, target->service); self->on_connecting (self->user_data, address); @@ -3999,13 +4012,6 @@ connector_notify_connecting (struct connector *self, } static void -connector_notify_error (struct connector *self, const char *error) -{ - if (self->on_error) - self->on_error (self->user_data, error); -} - -static void connector_notify_connected (struct connector *self, int fd) { set_blocking (fd, true); @@ -4016,23 +4022,38 @@ static void connector_prepare_next (struct connector *self) { struct connector_target *target = self->targets; - if (!(target->iter = target->iter->ai_next)) + if (!target->iter || !(target->iter = target->iter->ai_next)) { LIST_UNLINK_WITH_TAIL (self->targets, self->targets_t, target); connector_target_destroy (target); } } +static void connector_handle_error (struct connector *self, const char *error); + +/// See if there's any target remaining at all -- it can however either still +/// be waiting for address resolution to finish, or have already failed +static bool +connector_check_target (struct connector *self, struct connector_target *target) +{ + if (!target) + self->on_failure (self->user_data); + else if (target->getaddrinfo_error) + { + connector_notify_connecting (self, target, NULL); + connector_handle_error (self, target->getaddrinfo_error->message); + } + else if (target->results) + return true; + return false; +} + static void connector_step (struct connector *self) { struct connector_target *target = self->targets; - if (!target) - { - // Total failure, none of the targets has succeeded - self->on_failure (self->user_data); + if (!connector_check_target (self, target)) return; - } struct addrinfo *gai_iter = target->iter; hard_assert (gai_iter != NULL); @@ -4043,10 +4064,7 @@ connector_step (struct connector *self) gai_iter->ai_socktype, gai_iter->ai_protocol); if (fd == -1) { - connector_notify_error (self, strerror (errno)); - - connector_prepare_next (self); - connector_step (self); + connector_handle_error (self, strerror (errno)); return; } @@ -4066,15 +4084,22 @@ connector_step (struct connector *self) } else { - connector_notify_error (self, strerror (errno)); + connector_handle_error (self, strerror (errno)); xclose (fd); - - connector_prepare_next (self); - connector_step (self); } } static void +connector_handle_error (struct connector *self, const char *error) +{ + if (self->on_error) + self->on_error (self->user_data, error); + + connector_prepare_next (self); + connector_step (self); +} + +static void connector_on_ready (const struct pollfd *pfd, struct connector *self) { // See http://cr.yp.to/docs/connect.html if this doesn't work. @@ -4087,14 +4112,11 @@ connector_on_ready (const struct pollfd *pfd, struct connector *self) if (error) { - connector_notify_error (self, strerror (error)); - poller_fd_reset (&self->connected_event); xclose (self->socket); self->socket = -1; - connector_prepare_next (self); - connector_step (self); + connector_handle_error (self, strerror (error)); } else { @@ -4108,6 +4130,7 @@ static void connector_init (struct connector *self, struct poller *poller) { memset (self, 0, sizeof *self); + self->poller = poller; self->socket = -1; poller_fd_init (&self->connected_event, poller, self->socket); self->connected_event.user_data = self; @@ -4125,34 +4148,50 @@ connector_free (struct connector *self) connector_target_destroy (iter); } -static bool -connector_add_target (struct connector *self, - const char *hostname, const char *service, struct error **e) +static void +connector_on_getaddrinfo (int err, struct addrinfo *results, void *user_data) { - struct addrinfo hints, *results; - memset (&hints, 0, sizeof hints); - hints.ai_socktype = SOCK_STREAM; + struct connector_target *self = user_data; - // TODO: even this should be done asynchronously, most likely in - // a thread pool, similarly to how libuv does it - int err = getaddrinfo (hostname, service, &hints, &results); if (err) { - error_set (e, "%s: %s", "getaddrinfo", gai_strerror (err)); - return false; + error_set (&self->getaddrinfo_error, + "%s: %s", "getaddrinfo", gai_strerror (err)); } + self->results = self->iter = results; + self->getaddrinfo_event = NULL; + + // We've been waiting for this address to be resolved + if (self == self->connector->targets) + connector_step (self->connector); +} + +/// Connection will be attempted asynchronously once you add any target +static void +connector_add_target (struct connector *self, + const char *hostname, const char *service) +{ struct connector_target *target = connector_target_new (); + target->connector = self; target->hostname = xstrdup (hostname); target->service = xstrdup (service); - target->results = results; - target->iter = target->results; + + struct addrinfo hints; + memset (&hints, 0, sizeof hints); + hints.ai_socktype = SOCK_STREAM; + + struct async_getaddrinfo *gai = async_getaddrinfo + (&self->poller->common.async, hostname, service, &hints); + + gai->dispatcher = connector_on_getaddrinfo; + gai->user_data = target; + target->getaddrinfo_event = &gai->async; LIST_APPEND_WITH_TAIL (self->targets, self->targets_t, target); - return true; } -#endif // LIBERTY_WANT_POLLER +#endif // defined LIBERTY_WANT_POLLER && defined LIBERTY_WANT_ASYNC // --- Advanced configuration -------------------------------------------------- |