terom@177: #include "tcp_internal.h"
terom@177: #include "log.h"
terom@177: 
terom@177: /*
terom@177:  * Our transport methods
terom@177:  */
terom@177: static void tcp_client__deinit (transport_t *transport)
terom@177: {
terom@177:     struct tcp_client *client = transport_check(transport, &tcp_client_type);
terom@177:     
terom@177:     // proxy
terom@177:     tcp_client_deinit(client);
terom@177: }
terom@177: 
terom@177: /*
terom@177:  * Our transport_type
terom@177:  */
terom@177: const struct transport_type tcp_client_type = {
terom@177:     .base_type = {
terom@177:         .parent             = &tcp_transport_type.base_type,
terom@177:     },
terom@177:     .methods                = {
terom@177:         .read               = transport_fd__read,
terom@177:         .write              = transport_fd__write,
terom@177:         .events             = transport_fd__events,
terom@177:         .deinit             = tcp_client__deinit,
terom@177:     },
terom@177: };
terom@177: 
terom@177: /*
terom@177:  * Forward-declare 
terom@177:  */
terom@177: static void tcp_client_on_connect (struct transport_fd *fd, short what, void *arg);
terom@177: 
terom@177: /*
terom@177:  * Function implementations
terom@177:  */
terom@177: void tcp_client_init (struct tcp_client *client)
terom@177: {
terom@177:     tcp_transport_init(&client->base_trans, -1);
terom@177:     
terom@177:     resolve_result_init(&client->rr);
terom@177: }
terom@177: 
terom@177: /*
terom@177:  * Start connecting to the given address in a non-blocking fashion. Returns any errors that immediately crop up,
terom@177:  * otherwise eventually calls tcp_client_connect_done().
terom@177:  */
terom@177: static err_t tcp_client_connect_addr (struct tcp_client *client, struct addrinfo *addr, error_t *err)
terom@177: {
terom@177:     struct transport_fd *_fd = &client->base_trans.base_fd;
terom@177:     int ret;
terom@177:     evutil_socket_t sock;
terom@177:     err_t tmp;
terom@177: 
terom@177:     // first, create the socket
terom@177:     if ((sock = tcp_sock_create(addr, err)) < 0)
terom@177:         return ERROR_CODE(err);
terom@177: 
terom@177:     // set it as our sock
terom@177:     if ((ERROR_CODE(err) = transport_fd_set(_fd, sock)))
terom@177:         goto error;
terom@177: 
terom@177:     // then, set it up as nonblocking
terom@177:     if ((ERROR_CODE(err) = transport_fd_nonblock(_fd, true)))
terom@177:         goto error;
terom@177: 
terom@177:     // then, initiate the connect
terom@177:     if ((ret = connect(sock, addr->ai_addr, addr->ai_addrlen)) < 0 && errno != EINPROGRESS) 
terom@177:         JUMP_SET_ERROR_ERRNO(err, ERR_CONNECT);
terom@177:     
terom@177:     if (ret < 0) {
terom@177:         // ok, connect started, setup our completion callback
terom@177:         if ((ERROR_CODE(err) = transport_fd_setup(_fd, tcp_client_on_connect, client)))
terom@177:             goto error;
terom@177:     
terom@177:         // enable for write
terom@177:         if ((ERROR_CODE(err) = transport_fd_enable(_fd, TRANSPORT_WRITE)))
terom@177:             goto error;
terom@177: 
terom@177:     } else {
terom@177:         // oops... blocking connect - fail to avoid confusion
terom@177:         // XXX: come up with a better error name to use
terom@177:         // XXX: support non-async connects as well
terom@177:         JUMP_SET_ERROR_EXTRA(err, ERR_CONNECT, EINPROGRESS);
terom@177:     }
terom@177:     
terom@177:     // ok
terom@177:     return SUCCESS;
terom@177: 
terom@177: error:
terom@177:     // close the stuff we did open
terom@177:     if ((tmp = transport_fd_close(_fd)))
terom@177:         log_warn("error closing socket after connect error: %s", error_name(tmp));
terom@177: 
terom@177:     return ERROR_CODE(err);
terom@177: }
terom@177: 
terom@177: 
terom@177: /*
terom@177:  * Attempt to connect to the next addrinfo, or the next one, if that fails, etc.
terom@177:  *
terom@177:  * This does not call transport_connected().
terom@177:  */
terom@177: static err_t tcp_client_connect_continue (struct tcp_client *client, error_t *err)
terom@177: {
terom@177:     struct addrinfo *addr;
terom@177: 
terom@177:     // try and connect to each one until we find one that works
terom@177:     while ((addr = resolve_result_next(&client->rr))) {
terom@177:         // attempt to start connect
terom@177:         if (tcp_client_connect_addr(client, addr, err) == SUCCESS)
terom@177:             break;
terom@177: 
terom@177:         // log a warning on the failed connect
terom@177:         log_warn_error(err, "%s", resolve_addr_text(addr));
terom@177:     }
terom@177:     
terom@177: 
terom@177:     if (addr)
terom@177:         // we succesfully did a tcp_client_connect_addr on valid address
terom@177:         return SUCCESS;
terom@177: 
terom@177:     else
terom@177:         // all of the connect_async_addr's failed, return the last error
terom@177:         return ERROR_CODE(err);
terom@177: }
terom@177: 
terom@177: /*
terom@177:  * Cleanup our resolver state and any connect callbacks after a connect
terom@177:  */
terom@177: static void tcp_client_connect_cleanup (struct tcp_client *client)
terom@177: {
terom@177:     // drop the resolver stuff
terom@177:     resolve_result_deinit(&client->rr);
terom@177:     
terom@177:     // remove our event handler
terom@177:     transport_fd_clear(&client->base_trans.base_fd);
terom@177: }
terom@177: 
terom@177: /*
terom@177:  * Our async connect operation has completed, clean up, set up state for event-based operation with user callbacks, and
terom@177:  * invoke transport_connected().
terom@177:  *
terom@177:  * The given \a err should be NULL for successful completion, or the error for failures.
terom@177:  */
terom@177: static void tcp_client_connect_done (struct tcp_client *client, error_t *conn_err)
terom@177: {
terom@177:     error_t err;
terom@177: 
terom@177:     // cleanup
terom@177:     tcp_client_connect_cleanup(client);
terom@177: 
terom@177:     if (conn_err)
terom@177:         JUMP_SET_ERROR_INFO(&err, conn_err);
terom@177: 
terom@177:     // let the transport handle the rest
terom@177:     if (tcp_transport_connected(&client->base_trans, &err))
terom@177:         goto error;
terom@177:     
terom@177:     // ok
terom@177:     return;
terom@177: 
terom@177: error:    
terom@177:     // pass the error on to transport
terom@177:     transport_connected(&client->base_trans.base_fd.base, &err, false);
terom@177: }
terom@177: 
terom@177: /*
terom@177:  * Our async connect callback
terom@177:  */
terom@177: static void tcp_client_on_connect (struct transport_fd *fd, short what, void *arg)
terom@177: {
terom@177:     struct tcp_client *client = arg;
terom@177:     error_t err;
terom@177:     err_t tmp;
terom@177: 
terom@177:     // XXX: timeouts
terom@177:     (void) what;
terom@177:     
terom@177:     // read socket error code
terom@177:     if (tcp_sock_error(client->base_trans.base_fd.fd, &err))
terom@177:         goto error;
terom@177: 
terom@177:     // did the connect fail?
terom@177:     if (ERROR_EXTRA(&err))
terom@177:         JUMP_SET_ERROR(&err, ERR_CONNECT);
terom@177:     
terom@177:     // done, success
terom@177:     return tcp_client_connect_done(client, NULL);
terom@177: 
terom@177: error:
terom@177:     // close the socket
terom@177:     if ((tmp = transport_fd_close(fd)))
terom@177:         log_warn("error closing socket after connect error: %s", error_name(tmp));
terom@177: 
terom@177:     // log a warning
terom@177:     log_warn_error(&err, "connect to %s failed", "???");
terom@177: 
terom@177:     // try the next one or fail completely
terom@177:     if (tcp_client_connect_continue(client, &err))
terom@177:         tcp_client_connect_done(client, &err);
terom@177: }
terom@177: 
terom@177: err_t tcp_client_connect_async (struct tcp_client *client, const char *hostname, const char *service, error_t *err)
terom@177: {
terom@177:     // do the resolving
terom@177:     if (resolve_addr(&client->rr, hostname, service, SOCK_STREAM, 0, err))
terom@177:         return ERROR_CODE(err);
terom@177: 
terom@177:     // start connecting with the first result
terom@177:     if (tcp_client_connect_continue(client, err))
terom@177:         goto error;
terom@177: 
terom@177:     // ok
terom@177:     return SUCCESS;
terom@177: 
terom@177: error:
terom@177:     // cleanup
terom@177:     resolve_result_deinit(&client->rr);
terom@177:     
terom@177:     return ERROR_CODE(err);
terom@177: }
terom@177: 
terom@177: void tcp_client_deinit (struct tcp_client *client)
terom@177: {
terom@177:     // cleanup our stuff
terom@177:     resolve_result_deinit(&client->rr);
terom@177:     
terom@177:     // deinit lower transport
terom@177:     tcp_transport_deinit(&client->base_trans);
terom@177: }
terom@177: 
terom@177: /*
terom@177:  * Deinit and free, not using the transport interface
terom@177:  */
terom@177: static void tcp_client_destroy (struct tcp_client *client)
terom@177: {
terom@177:     tcp_client_deinit(client);
terom@177: 
terom@177:     free(client);
terom@177: }
terom@177: 
terom@177: /*
terom@177:  * Public interface
terom@177:  */
terom@177: err_t tcp_connect (const struct transport_info *info, transport_t **transport_ptr, 
terom@177:         const char *host, const char *service, error_t *err)
terom@177: {
terom@177:     struct tcp_client *client;
terom@177:  
terom@177:     // alloc
terom@177:     if ((client = calloc(1, sizeof(*client))) == NULL)
terom@177:         return ERR_CALLOC;
terom@177: 
terom@177:     // init transport
terom@177:     transport_init(&client->base_trans.base_fd.base, &tcp_client_type, info);
terom@177:     
terom@177:     // init our state
terom@177:     tcp_client_init(client);
terom@177:  
terom@177:     // begin connect
terom@177:     if (tcp_client_connect_async(client, host, service, err))
terom@177:         goto error;
terom@177: 
terom@177:     // good
terom@177:     *transport_ptr = &client->base_trans.base_fd.base;
terom@177: 
terom@177:     return 0;
terom@177: 
terom@177: error:
terom@177:     // cleanup
terom@177:     tcp_client_destroy(client);
terom@177:         
terom@177:     // return error code
terom@177:     return ERROR_CODE(err);
terom@177: }
terom@177: