1 #include "tcp_internal.h"

     2 #include "sock_internal.h"

     3 

     4 #include <unistd.h>

     5 

     6 /*

     7  * Our transport_type

     8  */

     9 const struct transport_type tcp_transport_type = {

    10     .base_type = {

    11         .parent             = &transport_fd_type.base_type,

    12     },

    13     .methods                = {

    14         .read               = transport_fd__read,

    15         .write              = transport_fd__write,

    16         .events             = transport_fd__events,

    17         .deinit             = transport_fd__deinit,

    18     },

    19 };

    20 

    21 void tcp_transport_init (struct tcp_transport *trans, evutil_socket_t sock)

    22 {

    23     struct event_base *ev_base = _sock_stream_ctx.ev_base;

    24 

    25     transport_fd_init(&trans->base_fd, ev_base, sock);

    26 }

    27 

    28 err_t tcp_transport_create (struct tcp_transport **trans_ptr, const struct transport_info *info, evutil_socket_t sock, error_t *err)

    29 {

    30     struct tcp_transport *trans;

    31 

    32     // alloc

    33     if ((trans = calloc(1, sizeof(*trans))) == NULL)

    34         JUMP_SET_ERROR(err, ERR_MEM);

    35 

    36     // init transport

    37     transport_init(&trans->base_fd.base, &tcp_transport_type, info);

    38 

    39     // init ourselves

    40     tcp_transport_init(trans, sock);

    41 

    42     // setup the socket?

    43     if (sock >= 0) {

    44         // make it non-blocking

    45         if ((ERROR_CODE(err) = transport_fd_nonblock(&trans->base_fd, true)))

    46             goto error;

    47     }

    48 

    49     // ok

    50     *trans_ptr = trans;

    51 

    52     return SUCCESS;

    53 

    54 error:

    55     // cleanup

    56     if (trans)

    57         tcp_transport_deinit(trans);

    58     else

    59         EVUTIL_CLOSESOCKET(sock);

    60 

    61     return ERROR_CODE(err);

    62 }

    63 

    64 err_t tcp_transport_connected (struct tcp_transport *trans, error_t *err)

    65 {

    66     // set up for default transport event-based operation

    67     if ((ERROR_CODE(err) = transport_fd_defaults(&trans->base_fd)))

    68         return ERROR_CODE(err);

    69 

    70     // ok

    71     transport_connected(&trans->base_fd.base, NULL, false);

    72 

    73     return SUCCESS;

    74 }

    75 

    76 void tcp_transport_deinit (struct tcp_transport *trans)

    77 {

    78     transport_fd_deinit(&trans->base_fd);

    79 }

    80 

    81 void tcp_transport_destroy (struct tcp_transport *trans)

    82 {

    83     tcp_transport_deinit(trans);

    84 

    85     free(trans);

    86 }