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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#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"
#include "server.h"
struct conn_acceptor {
tsocket_t *sock;
evloop_t **evloops;
int nevloops;
connection_callback_func_t conn_callback_func;
};
conn_acceptor_t *conn_acceptor_create(tsocket_t *sock, evloop_t **evloops, int nevloops)
{
conn_acceptor_t *acceptor = malloc(sizeof(*acceptor));
if (!acceptor)
return NULL;
acceptor->sock = sock;
acceptor->evloops = evloops;
acceptor->nevloops = nevloops;
acceptor->conn_callback_func = NULL;
return acceptor;
}
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_event_create(conn_sock, ca->conn_callback_func);
// TODO: Use a better way to decide which sub reactor to use.
int ind = conn_sock->fd % ca->nevloops;
if (evloop_add(ca->evloops[ind], 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);
}
void conn_acceptor_destroy(conn_acceptor_t *ca)
{
tsocket_destroy(ca->sock);
free(ca);
}
event_t *conn_acceptor_event_create(conn_acceptor_t *acceptor)
{
event_t *ev = event_create(acceptor, acceptor->sock->fd,
(evloop_process_func_t) conn_acceptor_accept,
(evloop_destroy_func_t) conn_acceptor_destroy);
return ev;
}
void conn_acceptor_on_connect(conn_acceptor_t *acceptor, connection_callback_func_t func)
{
acceptor->conn_callback_func = func;
}
|