acceptor.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

#include <stdbool.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/epoll.h>

#include "acceptor.h"
#include "util.h"
#include "connection.h"
#include "evloop.h"

struct conn_acceptor {
    struct tsocket *sock;
    evloop_t *el;
};
typedef struct conn_acceptor conn_acceptor_t;

static void conn_acceptor_accept(conn_acceptor_t *ca)
{
    struct tsocket *conn_sock = tsocket_accept(ca->sock);
    if (conn_sock == NULL) {
        perror("socket accept");
        return;
    }

    if (setblocking(conn_sock->fd, false) == -1) {
        perror("setblocking");
        tsocket_destroy(conn_sock);
        return;
    }

    event_t *conn_ev = connection_create_event(conn_sock);

    if (evloop_add(ca->el, conn_ev, EPOLLIN | EPOLLET) == -1) {
        perror("eventloop add fd: conn_sock");
        return;
    }

    printf("New client fd %d, ip: %s, port: %d\n", 
            conn_sock->fd, conn_sock->addr, conn_sock->port);
}

static void conn_acceptor_destroy(conn_acceptor_t *ca)
{
    tsocket_destroy(ca->sock);
    free(ca);
}

event_t *conn_acceptor_create_event(struct tsocket *sock, evloop_t *el)
{
    conn_acceptor_t *ca = malloc(sizeof(*ca));
    ca->sock = sock;
    ca->el = el;

    struct event *ev = event_create(ca, sock->fd,
            (evloop_process_func_t) conn_acceptor_accept,
            (evloop_destroy_func_t) conn_acceptor_destroy);
    return ev;
}