1

     2

#include "sock_tcp.h"

     3

     4

#include <stdlib.h>

     5

#include <sys/types.h>

     6

#include <sys/socket.h>

     7

#include <netdb.h>

     8

#include <unistd.h>

     9

#include <string.h>

    10

#include <err.h>

    11

    12

/*

    13

 * Our sock_stream_type.methods.read implementation

    14

 */

    15

static int sock_tcp_read (struct sock_stream *base_sock, void *buf, size_t len)

    16

{

    17

    struct sock_tcp *sock = SOCK_FROM_BASE(base_sock, struct sock_tcp);

    18

    19

    return read(sock->fd, buf, len);

    20

}

    21

    22

/*

    23

 * Our sock_stream_type.methods.write implementation

    24

 */

    25

static int sock_tcp_write (struct sock_stream *base_sock, const void *buf, size_t len)

    26

{

    27

    struct sock_tcp *sock = SOCK_FROM_BASE(base_sock, struct sock_tcp);

    28

    29

    return write(sock->fd, buf, len);

    30

}

    31

    32

/*

    36

 */

    37

struct sock_stream_type sock_tcp_type = {

    38

    .methods.read   = &sock_tcp_read,

    39

    .methods.write  = &sock_tcp_write,

    40

};

    41

    43

{

    44

    struct sock_tcp *sock;

    45

    46

    // alloc

    47

    if ((sock = calloc(1, sizeof(*sock))) == NULL)

    48

        errx(1, "calloc");

    52

    59

    // initialize

    60

    sock->fd = fd;

    61

    62

    // done

    64

}

    65

    67

{

    68

    struct addrinfo hints, *res, *r;

    70

    71

    // hints

    72

    memset(&hints, 0, sizeof(hints));

    73

    hints.ai_family = AF_UNSPEC;

    74

    hints.ai_socktype = SOCK_STREAM;

    75

    76

    // resolve

    78

        errx(1, "getaddrinfo: %s", gai_strerror(_err));

    79

    80

    // use

    81

    for (r = res; r; r = r->ai_next) {

    82

        // XXX: wrong

    84

            err(1, "socket");

    85

    87

            err(1, "connect");

    88

    89

        break;

    90

    }

    91

    92

    // ensure we got some valid socket

    94

        errx(1, "no valid socket");

    95

    98

}

    99