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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
|
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <pthread.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <netinet/tcp.h>
#include <malloc.h>
#include "common.h"
#include "log.h"
#include "appconfig.h"
#include "url.h"
#include "web_buffer.h"
#include "web_client.h"
#include "web_server.h"
#include "global_statistics.h"
#include "rrd.h"
#include "rrd2json.h"
int listen_backlog = LISTEN_BACKLOG;
int listen_fd = -1;
int listen_port = LISTEN_PORT;
static void log_allocations(void)
{
static int mem = 0;
struct mallinfo mi;
mi = mallinfo();
if(mi.uordblks > mem) {
int clients = 0;
struct web_client *w;
for(w = web_clients; w ; w = w->next) clients++;
info("Allocated memory increased from %d to %d (increased by %d bytes). There are %d web clients connected.", mem, mi.uordblks, mi.uordblks - mem, clients);
mem = mi.uordblks;
}
}
int create_listen_socket4(int port, int listen_backlog)
{
int sock;
int sockopt = 1;
struct sockaddr_in name;
debug(D_LISTENER, "IPv4 creating new listening socket on port %d", port);
sock = socket(AF_INET, SOCK_STREAM, 0);
if(sock < 0) {
error("IPv4 socket() failed.");
return -1;
}
/* avoid "address already in use" */
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void*)&sockopt, sizeof(sockopt));
memset(&name, 0, sizeof(struct sockaddr_in));
name.sin_family = AF_INET;
name.sin_port = htons (port);
name.sin_addr.s_addr = htonl (INADDR_ANY);
if(bind (sock, (struct sockaddr *) &name, sizeof (name)) < 0) {
close(sock);
error("IPv4 bind() failed.");
return -1;
}
if(listen(sock, listen_backlog) < 0) {
close(sock);
fatal("IPv4 listen() failed.");
return -1;
}
debug(D_LISTENER, "IPv4 listening port %d created", port);
return sock;
}
int create_listen_socket6(int port, int listen_backlog)
{
int sock = -1;
int sockopt = 1;
struct sockaddr_in6 name;
debug(D_LISTENER, "IPv6 creating new listening socket on port %d", port);
sock = socket(AF_INET6, SOCK_STREAM, 0);
if (sock < 0) {
error("IPv6 socket() failed.");
return -1;
}
/* avoid "address already in use" */
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void*)&sockopt, sizeof(sockopt));
memset(&name, 0, sizeof(struct sockaddr_in6));
name.sin6_family = AF_INET6;
name.sin6_port = htons ((uint16_t) port);
name.sin6_addr = in6addr_any;
name.sin6_scope_id = 0;
if (bind (sock, (struct sockaddr *) &name, sizeof (name)) < 0) {
close(sock);
error("IPv6 bind() failed.");
return -1;
}
if (listen(sock, listen_backlog) < 0) {
close(sock);
fatal("IPv6 listen() failed.");
return -1;
}
debug(D_LISTENER, "IPv6 listening port %d created", port);
return sock;
}
// --------------------------------------------------------------------------------------
// the main socket listener
// 1. it accepts new incoming requests on our port
// 2. creates a new web_client for each connection received
// 3. spawns a new pthread to serve the client (this is optimal for keep-alive clients)
// 4. cleans up old web_clients that their pthreads have been exited
void *socket_listen_main(void *ptr)
{
if(ptr) { ; }
info("WEB SERVER thread created with task id %d", gettid());
struct web_client *w;
struct timeval tv;
int retval;
if(ptr) { ; }
if(pthread_setcanceltype(PTHREAD_CANCEL_DEFERRED, NULL) != 0)
error("Cannot set pthread cancel type to DEFERRED.");
if(pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL) != 0)
error("Cannot set pthread cancel state to ENABLE.");
web_client_timeout = (int) config_get_number("global", "disconnect idle web clients after seconds", DEFAULT_DISCONNECT_IDLE_WEB_CLIENTS_AFTER_SECONDS);
web_enable_gzip = config_get_boolean("global", "enable web responses gzip compression", web_enable_gzip);
if(listen_fd < 0) fatal("LISTENER: Listen socket is not ready.");
fd_set ifds, ofds, efds;
int fdmax = listen_fd;
FD_ZERO (&ifds);
FD_ZERO (&ofds);
FD_ZERO (&efds);
for(;;) {
tv.tv_sec = 0;
tv.tv_usec = 200000;
if(listen_fd >= 0) {
FD_SET(listen_fd, &ifds);
FD_SET(listen_fd, &efds);
}
// debug(D_WEB_CLIENT, "LISTENER: Waiting...");
retval = select(fdmax+1, &ifds, &ofds, &efds, &tv);
if(retval == -1) {
error("LISTENER: select() failed.");
continue;
}
else if(retval) {
// check for new incoming connections
if(FD_ISSET(listen_fd, &ifds)) {
w = web_client_create(listen_fd);
if(unlikely(!w)) {
// no need for error log - web_client_create already logged the error
continue;
}
if(pthread_create(&w->thread, NULL, web_client_main, w) != 0) {
error("%llu: failed to create new thread for web client.");
w->obsolete = 1;
}
else if(pthread_detach(w->thread) != 0) {
error("%llu: Cannot request detach of newly created web client thread.", w->id);
w->obsolete = 1;
}
}
else debug(D_WEB_CLIENT, "LISTENER: select() didn't do anything.");
}
//else {
// debug(D_WEB_CLIENT, "LISTENER: select() timeout.");
//}
// cleanup unused clients
for(w = web_clients; w ; w = w?w->next:NULL) {
if(w->obsolete) {
debug(D_WEB_CLIENT, "%llu: Removing client.", w->id);
// pthread_join(w->thread, NULL);
w = web_client_free(w);
log_allocations();
}
}
}
error("LISTENER: exit!");
if(listen_fd >= 0) close(listen_fd);
exit(2);
return NULL;
}
|
