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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#include "tcp_server.h"
#include "log.h"
#include "net.h"
#include "signal.h"
#include <pthread.h>
#include <stdlib.h>
#include <unistd.h>
int tcp_server_create(struct tcp_server *server, const char *port)
{
server->fd = net_bind(port);
if (server->fd < 0)
return server->fd;
return 0;
}
void tcp_server_destroy(const struct tcp_server *server)
{
check_errno(close(server->fd), "close");
}
struct child_context {
int fd;
tcp_server_conn_handler handler;
void *arg;
};
static void *connection_thread(void *_ctx)
{
struct child_context *ctx = (struct child_context *)_ctx;
int ret = 0;
ret = signal_block_child();
if (ret < 0)
goto close;
ctx->handler(ctx->fd, ctx->arg);
close:
check_errno(close(ctx->fd), "close");
free(ctx);
return NULL;
}
int tcp_server_accept(const struct tcp_server *server, tcp_server_conn_handler handler, void *arg)
{
struct child_context *ctx;
pthread_attr_t child_attr;
sigset_t old_mask;
pthread_t child;
int conn_fd, ret = 0;
ret = net_accept(server->fd);
if (ret < 0)
return ret;
conn_fd = ret;
ctx = malloc(sizeof(*ctx));
if (!ctx) {
print_errno("malloc");
ret = -1;
goto close_conn;
}
*ctx = (struct child_context){conn_fd, handler, arg};
ret = pthread_attr_init(&child_attr);
if (ret) {
pthread_print_errno(ret, "pthread_attr_init");
goto free_ctx;
}
ret = pthread_attr_setdetachstate(&child_attr, PTHREAD_CREATE_DETACHED);
if (ret) {
pthread_print_errno(ret, "pthread_attr_setdetachstate");
goto destroy_attr;
}
ret = signal_block_parent(&old_mask);
if (ret < 0)
goto destroy_attr;
ret = pthread_create(&child, &child_attr, connection_thread, ctx);
if (ret) {
pthread_print_errno(ret, "pthread_create");
goto restore_mask;
}
signal_set(&old_mask, NULL);
pthread_check(pthread_attr_destroy(&child_attr), "pthread_attr_destroy");
return ret;
restore_mask:
signal_set(&old_mask, NULL);
destroy_attr:
pthread_check(pthread_attr_destroy(&child_attr), "pthread_attr_destroy");
free_ctx:
free(ctx);
close_conn:
check_errno(close(conn_fd), "close");
return ret;
}
|
