]> arthur.barton.de Git - netdata.git/blob - src/web_client.c
rename chart fields to avoid conflicts with backends; fixes #1962
[netdata.git] / src / web_client.c
1 #include "common.h"
2
3 #define INITIAL_WEB_DATA_LENGTH 16384
4 #define WEB_REQUEST_LENGTH 16384
5 #define TOO_BIG_REQUEST 16384
6
7 int web_client_timeout = DEFAULT_DISCONNECT_IDLE_WEB_CLIENTS_AFTER_SECONDS;
8 int respect_web_browser_do_not_track_policy = 0;
9 char *web_x_frame_options = NULL;
10
11 #ifdef NETDATA_WITH_ZLIB
12 int web_enable_gzip = 1, web_gzip_level = 3, web_gzip_strategy = Z_DEFAULT_STRATEGY;
13 #endif /* NETDATA_WITH_ZLIB */
14
15 struct web_client *web_clients = NULL;
16 unsigned long long web_clients_count = 0;
17
18 static inline int web_client_crock_socket(struct web_client *w) {
19 #ifdef TCP_CORK
20     if(likely(!w->tcp_cork && w->ofd != -1)) {
21         w->tcp_cork = 1;
22         if(unlikely(setsockopt(w->ofd, IPPROTO_TCP, TCP_CORK, (char *) &w->tcp_cork, sizeof(int)) != 0)) {
23             error("%llu: failed to enable TCP_CORK on socket.", w->id);
24             w->tcp_cork = 0;
25             return -1;
26         }
27     }
28 #endif /* TCP_CORK */
29
30     return 0;
31 }
32
33 static inline int web_client_uncrock_socket(struct web_client *w) {
34 #ifdef TCP_CORK
35     if(likely(w->tcp_cork && w->ofd != -1)) {
36         w->tcp_cork = 0;
37         if(unlikely(setsockopt(w->ofd, IPPROTO_TCP, TCP_CORK, (char *) &w->tcp_cork, sizeof(int)) != 0)) {
38             error("%llu: failed to disable TCP_CORK on socket.", w->id);
39             w->tcp_cork = 1;
40             return -1;
41         }
42     }
43 #endif /* TCP_CORK */
44
45     return 0;
46 }
47
48 struct web_client *web_client_create(int listener) {
49     struct web_client *w;
50
51     w = callocz(1, sizeof(struct web_client));
52     w->id = ++web_clients_count;
53     w->mode = WEB_CLIENT_MODE_NORMAL;
54
55     {
56         struct sockaddr *sadr;
57         socklen_t addrlen;
58
59         sadr = (struct sockaddr*) &w->clientaddr;
60         addrlen = sizeof(w->clientaddr);
61
62         w->ifd = accept4(listener, sadr, &addrlen, SOCK_NONBLOCK);
63         if (w->ifd == -1) {
64             error("%llu: Cannot accept new incoming connection.", w->id);
65             freez(w);
66             return NULL;
67         }
68         w->ofd = w->ifd;
69
70         if(getnameinfo(sadr, addrlen, w->client_ip, NI_MAXHOST, w->client_port, NI_MAXSERV, NI_NUMERICHOST | NI_NUMERICSERV) != 0) {
71             error("Cannot getnameinfo() on received client connection.");
72             strncpyz(w->client_ip,   "UNKNOWN", NI_MAXHOST);
73             strncpyz(w->client_port, "UNKNOWN", NI_MAXSERV);
74         }
75         w->client_ip[NI_MAXHOST]   = '\0';
76         w->client_port[NI_MAXSERV] = '\0';
77
78         switch(sadr->sa_family) {
79         case AF_INET:
80             debug(D_WEB_CLIENT_ACCESS, "%llu: New IPv4 web client from %s port %s on socket %d.", w->id, w->client_ip, w->client_port, w->ifd);
81             break;
82
83         case AF_INET6:
84             if(strncmp(w->client_ip, "::ffff:", 7) == 0) {
85                 memmove(w->client_ip, &w->client_ip[7], strlen(&w->client_ip[7]) + 1);
86                 debug(D_WEB_CLIENT_ACCESS, "%llu: New IPv4 web client from %s port %s on socket %d.", w->id, w->client_ip, w->client_port, w->ifd);
87             }
88             else
89                 debug(D_WEB_CLIENT_ACCESS, "%llu: New IPv6 web client from %s port %s on socket %d.", w->id, w->client_ip, w->client_port, w->ifd);
90             break;
91
92         default:
93             debug(D_WEB_CLIENT_ACCESS, "%llu: New UNKNOWN web client from %s port %s on socket %d.", w->id, w->client_ip, w->client_port, w->ifd);
94             break;
95         }
96
97         int flag = 1;
98         if(setsockopt(w->ofd, IPPROTO_TCP, TCP_NODELAY, (char *) &flag, sizeof(int)) != 0)
99             error("%llu: failed to enable TCP_NODELAY on socket.", w->id);
100
101         flag = 1;
102         if(setsockopt(w->ifd, SOL_SOCKET, SO_KEEPALIVE, (char *) &flag, sizeof(int)) != 0)
103             error("%llu: Cannot set SO_KEEPALIVE on socket.", w->id);
104     }
105
106     w->response.data = buffer_create(INITIAL_WEB_DATA_LENGTH);
107     w->response.header = buffer_create(HTTP_RESPONSE_HEADER_SIZE);
108     w->response.header_output = buffer_create(HTTP_RESPONSE_HEADER_SIZE);
109     w->origin[0] = '*';
110     w->wait_receive = 1;
111
112     if(web_clients) web_clients->prev = w;
113     w->next = web_clients;
114     web_clients = w;
115
116     web_client_connected();
117
118     return(w);
119 }
120
121 void web_client_reset(struct web_client *w) {
122     web_client_uncrock_socket(w);
123
124     debug(D_WEB_CLIENT, "%llu: Resetting client.", w->id);
125
126     if(likely(w->last_url[0])) {
127         struct timeval tv;
128         now_realtime_timeval(&tv);
129
130         size_t size = (w->mode == WEB_CLIENT_MODE_FILECOPY)?w->response.rlen:w->response.data->len;
131         size_t sent = size;
132 #ifdef NETDATA_WITH_ZLIB
133         if(likely(w->response.zoutput)) sent = (size_t)w->response.zstream.total_out;
134 #endif
135
136         // --------------------------------------------------------------------
137         // global statistics
138
139         finished_web_request_statistics(dt_usec(&tv, &w->tv_in),
140                                         w->stats_received_bytes,
141                                         w->stats_sent_bytes,
142                                         size,
143                                         sent);
144
145         w->stats_received_bytes = 0;
146         w->stats_sent_bytes = 0;
147
148
149         // --------------------------------------------------------------------
150         // access log
151
152         log_access("%llu: (sent/all = %zu/%zu bytes %0.0f%%, prep/sent/total = %0.2f/%0.2f/%0.2f ms) %s: %d '%s'",
153                    w->id,
154                    sent, size, -((size > 0) ? ((size - sent) / (double) size * 100.0) : 0.0),
155                    dt_usec(&w->tv_ready, &w->tv_in) / 1000.0,
156                    dt_usec(&tv, &w->tv_ready) / 1000.0,
157                    dt_usec(&tv, &w->tv_in) / 1000.0,
158                    (w->mode == WEB_CLIENT_MODE_FILECOPY) ? "filecopy" : ((w->mode == WEB_CLIENT_MODE_OPTIONS)
159                                                                          ? "options" : "data"),
160                    w->response.code,
161                    w->last_url
162         );
163     }
164
165     if(unlikely(w->mode == WEB_CLIENT_MODE_FILECOPY)) {
166         if(w->ifd != w->ofd) {
167             debug(D_WEB_CLIENT, "%llu: Closing filecopy input file descriptor %d.", w->id, w->ifd);
168             if(w->ifd != -1) close(w->ifd);
169             w->ifd = w->ofd;
170         }
171     }
172
173     w->last_url[0] = '\0';
174     w->cookie1[0] = '\0';
175     w->cookie2[0] = '\0';
176     w->origin[0] = '*';
177     w->origin[1] = '\0';
178
179     w->mode = WEB_CLIENT_MODE_NORMAL;
180
181     w->tcp_cork = 0;
182     w->donottrack = 0;
183     w->tracking_required = 0;
184     w->keepalive = 0;
185     w->decoded_url[0] = '\0';
186
187     buffer_reset(w->response.header_output);
188     buffer_reset(w->response.header);
189     buffer_reset(w->response.data);
190     w->response.rlen = 0;
191     w->response.sent = 0;
192     w->response.code = 0;
193
194     w->wait_receive = 1;
195     w->wait_send = 0;
196
197     w->response.zoutput = 0;
198
199     // if we had enabled compression, release it
200 #ifdef NETDATA_WITH_ZLIB
201     if(w->response.zinitialized) {
202         debug(D_DEFLATE, "%llu: Freeing compression resources.", w->id);
203         deflateEnd(&w->response.zstream);
204         w->response.zsent = 0;
205         w->response.zhave = 0;
206         w->response.zstream.avail_in = 0;
207         w->response.zstream.avail_out = 0;
208         w->response.zstream.total_in = 0;
209         w->response.zstream.total_out = 0;
210         w->response.zinitialized = 0;
211     }
212 #endif // NETDATA_WITH_ZLIB
213 }
214
215 struct web_client *web_client_free(struct web_client *w) {
216     web_client_reset(w);
217
218     struct web_client *n = w->next;
219     if(w == web_clients) web_clients = n;
220
221     debug(D_WEB_CLIENT_ACCESS, "%llu: Closing web client from %s port %s.", w->id, w->client_ip, w->client_port);
222
223     if(w->prev) w->prev->next = w->next;
224     if(w->next) w->next->prev = w->prev;
225     buffer_free(w->response.header_output);
226     buffer_free(w->response.header);
227     buffer_free(w->response.data);
228     if(w->ifd != -1) close(w->ifd);
229     if(w->ofd != -1 && w->ofd != w->ifd) close(w->ofd);
230     freez(w);
231
232     web_client_disconnected();
233
234     return(n);
235 }
236
237 uid_t web_files_uid(void) {
238     static char *web_owner = NULL;
239     static uid_t owner_uid = 0;
240
241     if(unlikely(!web_owner)) {
242         web_owner = config_get(CONFIG_SECTION_WEB, "web files owner", config_get(CONFIG_SECTION_GLOBAL, "run as user", ""));
243         if(!web_owner || !*web_owner)
244             owner_uid = geteuid();
245         else {
246             // getpwnam() is not thread safe,
247             // but we have called this function once
248             // while single threaded
249             struct passwd *pw = getpwnam(web_owner);
250             if(!pw) {
251                 error("User '%s' is not present. Ignoring option.", web_owner);
252                 owner_uid = geteuid();
253             }
254             else {
255                 debug(D_WEB_CLIENT, "Web files owner set to %s.", web_owner);
256                 owner_uid = pw->pw_uid;
257             }
258         }
259     }
260
261     return(owner_uid);
262 }
263
264 gid_t web_files_gid(void) {
265     static char *web_group = NULL;
266     static gid_t owner_gid = 0;
267
268     if(unlikely(!web_group)) {
269         web_group = config_get(CONFIG_SECTION_WEB, "web files group", config_get(CONFIG_SECTION_WEB, "web files owner", ""));
270         if(!web_group || !*web_group)
271             owner_gid = getegid();
272         else {
273             // getgrnam() is not thread safe,
274             // but we have called this function once
275             // while single threaded
276             struct group *gr = getgrnam(web_group);
277             if(!gr) {
278                 error("Group '%s' is not present. Ignoring option.", web_group);
279                 owner_gid = getegid();
280             }
281             else {
282                 debug(D_WEB_CLIENT, "Web files group set to %s.", web_group);
283                 owner_gid = gr->gr_gid;
284             }
285         }
286     }
287
288     return(owner_gid);
289 }
290
291 int mysendfile(struct web_client *w, char *filename) {
292     debug(D_WEB_CLIENT, "%llu: Looking for file '%s/%s'", w->id, netdata_configured_web_dir, filename);
293
294     // skip leading slashes
295     while (*filename == '/') filename++;
296
297     // if the filename contain known paths, skip them
298     if(strncmp(filename, WEB_PATH_FILE "/", strlen(WEB_PATH_FILE) + 1) == 0)
299         filename = &filename[strlen(WEB_PATH_FILE) + 1];
300
301     char *s;
302     for(s = filename; *s ;s++) {
303         if( !isalnum(*s) && *s != '/' && *s != '.' && *s != '-' && *s != '_') {
304             debug(D_WEB_CLIENT_ACCESS, "%llu: File '%s' is not acceptable.", w->id, filename);
305             w->response.data->contenttype = CT_TEXT_HTML;
306             buffer_sprintf(w->response.data, "Filename contains invalid characters: ");
307             buffer_strcat_htmlescape(w->response.data, filename);
308             return 400;
309         }
310     }
311
312     // if the filename contains a .. refuse to serve it
313     if(strstr(filename, "..") != 0) {
314         debug(D_WEB_CLIENT_ACCESS, "%llu: File '%s' is not acceptable.", w->id, filename);
315         w->response.data->contenttype = CT_TEXT_HTML;
316         buffer_strcat(w->response.data, "Relative filenames are not supported: ");
317         buffer_strcat_htmlescape(w->response.data, filename);
318         return 400;
319     }
320
321     // access the file
322     char webfilename[FILENAME_MAX + 1];
323     snprintfz(webfilename, FILENAME_MAX, "%s/%s", netdata_configured_web_dir, filename);
324
325     // check if the file exists
326     struct stat stat;
327     if(lstat(webfilename, &stat) != 0) {
328         debug(D_WEB_CLIENT_ACCESS, "%llu: File '%s' is not found.", w->id, webfilename);
329         w->response.data->contenttype = CT_TEXT_HTML;
330         buffer_strcat(w->response.data, "File does not exist, or is not accessible: ");
331         buffer_strcat_htmlescape(w->response.data, webfilename);
332         return 404;
333     }
334
335     // check if the file is owned by expected user
336     if(stat.st_uid != web_files_uid()) {
337         error("%llu: File '%s' is owned by user %u (expected user %u). Access Denied.", w->id, webfilename, stat.st_uid, web_files_uid());
338         w->response.data->contenttype = CT_TEXT_HTML;
339         buffer_strcat(w->response.data, "Access to file is not permitted: ");
340         buffer_strcat_htmlescape(w->response.data, webfilename);
341         return 403;
342     }
343
344     // check if the file is owned by expected group
345     if(stat.st_gid != web_files_gid()) {
346         error("%llu: File '%s' is owned by group %u (expected group %u). Access Denied.", w->id, webfilename, stat.st_gid, web_files_gid());
347         w->response.data->contenttype = CT_TEXT_HTML;
348         buffer_strcat(w->response.data, "Access to file is not permitted: ");
349         buffer_strcat_htmlescape(w->response.data, webfilename);
350         return 403;
351     }
352
353     if((stat.st_mode & S_IFMT) == S_IFDIR) {
354         snprintfz(webfilename, FILENAME_MAX, "%s/index.html", filename);
355         return mysendfile(w, webfilename);
356     }
357
358     if((stat.st_mode & S_IFMT) != S_IFREG) {
359         error("%llu: File '%s' is not a regular file. Access Denied.", w->id, webfilename);
360         w->response.data->contenttype = CT_TEXT_HTML;
361         buffer_strcat(w->response.data, "Access to file is not permitted: ");
362         buffer_strcat_htmlescape(w->response.data, webfilename);
363         return 403;
364     }
365
366     // open the file
367     w->ifd = open(webfilename, O_NONBLOCK, O_RDONLY);
368     if(w->ifd == -1) {
369         w->ifd = w->ofd;
370
371         if(errno == EBUSY || errno == EAGAIN) {
372             error("%llu: File '%s' is busy, sending 307 Moved Temporarily to force retry.", w->id, webfilename);
373             w->response.data->contenttype = CT_TEXT_HTML;
374             buffer_sprintf(w->response.header, "Location: /" WEB_PATH_FILE "/%s\r\n", filename);
375             buffer_strcat(w->response.data, "File is currently busy, please try again later: ");
376             buffer_strcat_htmlescape(w->response.data, webfilename);
377             return 307;
378         }
379         else {
380             error("%llu: Cannot open file '%s'.", w->id, webfilename);
381             w->response.data->contenttype = CT_TEXT_HTML;
382             buffer_strcat(w->response.data, "Cannot open file: ");
383             buffer_strcat_htmlescape(w->response.data, webfilename);
384             return 404;
385         }
386     }
387     if(fcntl(w->ifd, F_SETFL, O_NONBLOCK) < 0)
388         error("%llu: Cannot set O_NONBLOCK on file '%s'.", w->id, webfilename);
389
390     // pick a Content-Type for the file
391          if(strstr(filename, ".html") != NULL)  w->response.data->contenttype = CT_TEXT_HTML;
392     else if(strstr(filename, ".js")   != NULL)  w->response.data->contenttype = CT_APPLICATION_X_JAVASCRIPT;
393     else if(strstr(filename, ".css")  != NULL)  w->response.data->contenttype = CT_TEXT_CSS;
394     else if(strstr(filename, ".xml")  != NULL)  w->response.data->contenttype = CT_TEXT_XML;
395     else if(strstr(filename, ".xsl")  != NULL)  w->response.data->contenttype = CT_TEXT_XSL;
396     else if(strstr(filename, ".txt")  != NULL)  w->response.data->contenttype = CT_TEXT_PLAIN;
397     else if(strstr(filename, ".svg")  != NULL)  w->response.data->contenttype = CT_IMAGE_SVG_XML;
398     else if(strstr(filename, ".ttf")  != NULL)  w->response.data->contenttype = CT_APPLICATION_X_FONT_TRUETYPE;
399     else if(strstr(filename, ".otf")  != NULL)  w->response.data->contenttype = CT_APPLICATION_X_FONT_OPENTYPE;
400     else if(strstr(filename, ".woff2")!= NULL)  w->response.data->contenttype = CT_APPLICATION_FONT_WOFF2;
401     else if(strstr(filename, ".woff") != NULL)  w->response.data->contenttype = CT_APPLICATION_FONT_WOFF;
402     else if(strstr(filename, ".eot")  != NULL)  w->response.data->contenttype = CT_APPLICATION_VND_MS_FONTOBJ;
403     else if(strstr(filename, ".png")  != NULL)  w->response.data->contenttype = CT_IMAGE_PNG;
404     else if(strstr(filename, ".jpg")  != NULL)  w->response.data->contenttype = CT_IMAGE_JPG;
405     else if(strstr(filename, ".jpeg") != NULL)  w->response.data->contenttype = CT_IMAGE_JPG;
406     else if(strstr(filename, ".gif")  != NULL)  w->response.data->contenttype = CT_IMAGE_GIF;
407     else if(strstr(filename, ".bmp")  != NULL)  w->response.data->contenttype = CT_IMAGE_BMP;
408     else if(strstr(filename, ".ico")  != NULL)  w->response.data->contenttype = CT_IMAGE_XICON;
409     else if(strstr(filename, ".icns") != NULL)  w->response.data->contenttype = CT_IMAGE_ICNS;
410     else w->response.data->contenttype = CT_APPLICATION_OCTET_STREAM;
411
412     debug(D_WEB_CLIENT_ACCESS, "%llu: Sending file '%s' (%ld bytes, ifd %d, ofd %d).", w->id, webfilename, stat.st_size, w->ifd, w->ofd);
413
414     w->mode = WEB_CLIENT_MODE_FILECOPY;
415     w->wait_receive = 1;
416     w->wait_send = 0;
417     buffer_flush(w->response.data);
418     w->response.rlen = stat.st_size;
419 #ifdef __APPLE__
420     w->response.data->date = stat.st_mtimespec.tv_sec;
421 #else
422     w->response.data->date = stat.st_mtim.tv_sec;
423 #endif /* __APPLE__ */
424     buffer_cacheable(w->response.data);
425
426     return 200;
427 }
428
429
430 #ifdef NETDATA_WITH_ZLIB
431 void web_client_enable_deflate(struct web_client *w, int gzip) {
432     if(unlikely(w->response.zinitialized)) {
433         debug(D_DEFLATE, "%llu: Compression has already be initialized for this client.", w->id);
434         return;
435     }
436
437     if(unlikely(w->response.sent)) {
438         error("%llu: Cannot enable compression in the middle of a conversation.", w->id);
439         return;
440     }
441
442     w->response.zstream.zalloc = Z_NULL;
443     w->response.zstream.zfree = Z_NULL;
444     w->response.zstream.opaque = Z_NULL;
445
446     w->response.zstream.next_in = (Bytef *)w->response.data->buffer;
447     w->response.zstream.avail_in = 0;
448     w->response.zstream.total_in = 0;
449
450     w->response.zstream.next_out = w->response.zbuffer;
451     w->response.zstream.avail_out = 0;
452     w->response.zstream.total_out = 0;
453
454     w->response.zstream.zalloc = Z_NULL;
455     w->response.zstream.zfree = Z_NULL;
456     w->response.zstream.opaque = Z_NULL;
457
458 //  if(deflateInit(&w->response.zstream, Z_DEFAULT_COMPRESSION) != Z_OK) {
459 //      error("%llu: Failed to initialize zlib. Proceeding without compression.", w->id);
460 //      return;
461 //  }
462
463     // Select GZIP compression: windowbits = 15 + 16 = 31
464     if(deflateInit2(&w->response.zstream, web_gzip_level, Z_DEFLATED, 15 + ((gzip)?16:0), 8, web_gzip_strategy) != Z_OK) {
465         error("%llu: Failed to initialize zlib. Proceeding without compression.", w->id);
466         return;
467     }
468
469     w->response.zsent = 0;
470     w->response.zoutput = 1;
471     w->response.zinitialized = 1;
472
473     debug(D_DEFLATE, "%llu: Initialized compression.", w->id);
474 }
475 #endif // NETDATA_WITH_ZLIB
476
477 void buffer_data_options2string(BUFFER *wb, uint32_t options) {
478     int count = 0;
479
480     if(options & RRDR_OPTION_NONZERO) {
481         if(count++) buffer_strcat(wb, " ");
482         buffer_strcat(wb, "nonzero");
483     }
484
485     if(options & RRDR_OPTION_REVERSED) {
486         if(count++) buffer_strcat(wb, " ");
487         buffer_strcat(wb, "flip");
488     }
489
490     if(options & RRDR_OPTION_JSON_WRAP) {
491         if(count++) buffer_strcat(wb, " ");
492         buffer_strcat(wb, "jsonwrap");
493     }
494
495     if(options & RRDR_OPTION_MIN2MAX) {
496         if(count++) buffer_strcat(wb, " ");
497         buffer_strcat(wb, "min2max");
498     }
499
500     if(options & RRDR_OPTION_MILLISECONDS) {
501         if(count++) buffer_strcat(wb, " ");
502         buffer_strcat(wb, "ms");
503     }
504
505     if(options & RRDR_OPTION_ABSOLUTE) {
506         if(count++) buffer_strcat(wb, " ");
507         buffer_strcat(wb, "absolute");
508     }
509
510     if(options & RRDR_OPTION_SECONDS) {
511         if(count++) buffer_strcat(wb, " ");
512         buffer_strcat(wb, "seconds");
513     }
514
515     if(options & RRDR_OPTION_NULL2ZERO) {
516         if(count++) buffer_strcat(wb, " ");
517         buffer_strcat(wb, "null2zero");
518     }
519
520     if(options & RRDR_OPTION_OBJECTSROWS) {
521         if(count++) buffer_strcat(wb, " ");
522         buffer_strcat(wb, "objectrows");
523     }
524
525     if(options & RRDR_OPTION_GOOGLE_JSON) {
526         if(count++) buffer_strcat(wb, " ");
527         buffer_strcat(wb, "google_json");
528     }
529
530     if(options & RRDR_OPTION_PERCENTAGE) {
531         if(count++) buffer_strcat(wb, " ");
532         buffer_strcat(wb, "percentage");
533     }
534
535     if(options & RRDR_OPTION_NOT_ALIGNED) {
536         if(count++) buffer_strcat(wb, " ");
537         buffer_strcat(wb, "unaligned");
538     }
539 }
540
541 const char *group_method2string(int group) {
542     switch(group) {
543         case GROUP_UNDEFINED:
544             return "";
545
546         case GROUP_AVERAGE:
547             return "average";
548
549         case GROUP_MIN:
550             return "min";
551
552         case GROUP_MAX:
553             return "max";
554
555         case GROUP_SUM:
556             return "sum";
557
558         case GROUP_INCREMENTAL_SUM:
559             return "incremental-sum";
560
561         default:
562             return "unknown-group-method";
563     }
564 }
565
566 static inline int check_host_and_call(RRDHOST *host, struct web_client *w, char *url, int (*func)(RRDHOST *, struct web_client *, char *)) {
567     if(unlikely(host->rrd_memory_mode == RRD_MEMORY_MODE_NONE)) {
568         buffer_flush(w->response.data);
569         buffer_strcat(w->response.data, "This host does not maintain a database");
570         return 400;
571     }
572
573     return func(host, w, url);
574 }
575
576 int web_client_api_request(RRDHOST *host, struct web_client *w, char *url)
577 {
578     // get the api version
579     char *tok = mystrsep(&url, "/?&");
580     if(tok && *tok) {
581         debug(D_WEB_CLIENT, "%llu: Searching for API version '%s'.", w->id, tok);
582         if(strcmp(tok, "v1") == 0)
583             return web_client_api_request_v1(host, w, url);
584         else {
585             buffer_flush(w->response.data);
586             w->response.data->contenttype = CT_TEXT_HTML;
587             buffer_strcat(w->response.data, "Unsupported API version: ");
588             buffer_strcat_htmlescape(w->response.data, tok);
589             return 404;
590         }
591     }
592     else {
593         buffer_flush(w->response.data);
594         buffer_sprintf(w->response.data, "Which API version?");
595         return 400;
596     }
597 }
598
599 const char *web_content_type_to_string(uint8_t contenttype) {
600     switch(contenttype) {
601         case CT_TEXT_HTML:
602             return "text/html; charset=utf-8";
603
604         case CT_APPLICATION_XML:
605             return "application/xml; charset=utf-8";
606
607         case CT_APPLICATION_JSON:
608             return "application/json; charset=utf-8";
609
610         case CT_APPLICATION_X_JAVASCRIPT:
611             return "application/x-javascript; charset=utf-8";
612
613         case CT_TEXT_CSS:
614             return "text/css; charset=utf-8";
615
616         case CT_TEXT_XML:
617             return "text/xml; charset=utf-8";
618
619         case CT_TEXT_XSL:
620             return "text/xsl; charset=utf-8";
621
622         case CT_APPLICATION_OCTET_STREAM:
623             return "application/octet-stream";
624
625         case CT_IMAGE_SVG_XML:
626             return "image/svg+xml";
627
628         case CT_APPLICATION_X_FONT_TRUETYPE:
629             return "application/x-font-truetype";
630
631         case CT_APPLICATION_X_FONT_OPENTYPE:
632             return "application/x-font-opentype";
633
634         case CT_APPLICATION_FONT_WOFF:
635             return "application/font-woff";
636
637         case CT_APPLICATION_FONT_WOFF2:
638             return "application/font-woff2";
639
640         case CT_APPLICATION_VND_MS_FONTOBJ:
641             return "application/vnd.ms-fontobject";
642
643         case CT_IMAGE_PNG:
644             return "image/png";
645
646         case CT_IMAGE_JPG:
647             return "image/jpeg";
648
649         case CT_IMAGE_GIF:
650             return "image/gif";
651
652         case CT_IMAGE_XICON:
653             return "image/x-icon";
654
655         case CT_IMAGE_BMP:
656             return "image/bmp";
657
658         case CT_IMAGE_ICNS:
659             return "image/icns";
660
661         case CT_PROMETHEUS:
662             return "text/plain; version=0.0.4";
663
664         default:
665         case CT_TEXT_PLAIN:
666             return "text/plain; charset=utf-8";
667     }
668 }
669
670
671 const char *web_response_code_to_string(int code) {
672     switch(code) {
673         case 200:
674             return "OK";
675
676         case 307:
677             return "Temporary Redirect";
678
679         case 400:
680             return "Bad Request";
681
682         case 403:
683             return "Forbidden";
684
685         case 404:
686             return "Not Found";
687
688         case 412:
689             return "Preconditions Failed";
690
691         default:
692             if(code >= 100 && code < 200)
693                 return "Informational";
694
695             if(code >= 200 && code < 300)
696                 return "Successful";
697
698             if(code >= 300 && code < 400)
699                 return "Redirection";
700
701             if(code >= 400 && code < 500)
702                 return "Bad Request";
703
704             if(code >= 500 && code < 600)
705                 return "Server Error";
706
707             return "Undefined Error";
708     }
709 }
710
711 static inline char *http_header_parse(struct web_client *w, char *s) {
712     static uint32_t hash_origin = 0, hash_connection = 0, hash_accept_encoding = 0, hash_donottrack = 0;
713
714     if(unlikely(!hash_origin)) {
715         hash_origin = simple_uhash("Origin");
716         hash_connection = simple_uhash("Connection");
717         hash_accept_encoding = simple_uhash("Accept-Encoding");
718         hash_donottrack = simple_uhash("DNT");
719     }
720
721     char *e = s;
722
723     // find the :
724     while(*e && *e != ':') e++;
725     if(!*e) return e;
726
727     // get the name
728     *e = '\0';
729
730     // find the value
731     char *v = e + 1, *ve;
732
733     // skip leading spaces from value
734     while(*v == ' ') v++;
735     ve = v;
736
737     // find the \r
738     while(*ve && *ve != '\r') ve++;
739     if(!*ve || ve[1] != '\n') {
740         *e = ':';
741         return ve;
742     }
743
744     // terminate the value
745     *ve = '\0';
746
747     // fprintf(stderr, "HEADER: '%s' = '%s'\n", s, v);
748     uint32_t hash = simple_uhash(s);
749
750     if(hash == hash_origin && !strcasecmp(s, "Origin"))
751         strncpyz(w->origin, v, ORIGIN_MAX);
752
753     else if(hash == hash_connection && !strcasecmp(s, "Connection")) {
754         if(strcasestr(v, "keep-alive"))
755             w->keepalive = 1;
756     }
757     else if(respect_web_browser_do_not_track_policy && hash == hash_donottrack && !strcasecmp(s, "DNT")) {
758         if(*v == '0') w->donottrack = 0;
759         else if(*v == '1') w->donottrack = 1;
760     }
761 #ifdef NETDATA_WITH_ZLIB
762     else if(hash == hash_accept_encoding && !strcasecmp(s, "Accept-Encoding")) {
763         if(web_enable_gzip) {
764             if(strcasestr(v, "gzip"))
765                 web_client_enable_deflate(w, 1);
766             //
767             // does not seem to work
768             // else if(strcasestr(v, "deflate"))
769             //  web_client_enable_deflate(w, 0);
770         }
771     }
772 #endif /* NETDATA_WITH_ZLIB */
773
774     *e = ':';
775     *ve = '\r';
776     return ve;
777 }
778
779 // http_request_validate()
780 // returns:
781 // = 0 : all good, process the request
782 // > 0 : request is not supported
783 // < 0 : request is incomplete - wait for more data
784
785 typedef enum http_validation {
786     HTTP_VALIDATION_OK,
787     HTTP_VALIDATION_NOT_SUPPORTED,
788     HTTP_VALIDATION_INCOMPLETE
789 } HTTP_VALIDATION;
790
791 static inline HTTP_VALIDATION http_request_validate(struct web_client *w) {
792     char *s = w->response.data->buffer, *encoded_url = NULL;
793
794     // is is a valid request?
795     if(!strncmp(s, "GET ", 4)) {
796         encoded_url = s = &s[4];
797         w->mode = WEB_CLIENT_MODE_NORMAL;
798     }
799     else if(!strncmp(s, "OPTIONS ", 8)) {
800         encoded_url = s = &s[8];
801         w->mode = WEB_CLIENT_MODE_OPTIONS;
802     }
803     else if(!strncmp(s, "STREAM ", 7)) {
804         encoded_url = s = &s[7];
805         w->mode = WEB_CLIENT_MODE_STREAM;
806     }
807     else {
808         w->wait_receive = 0;
809         return HTTP_VALIDATION_NOT_SUPPORTED;
810     }
811
812     // find the SPACE + "HTTP/"
813     while(*s) {
814         // find the next space
815         while (*s && *s != ' ') s++;
816
817         // is it SPACE + "HTTP/" ?
818         if(*s && !strncmp(s, " HTTP/", 6)) break;
819         else s++;
820     }
821
822     // incomplete requests
823     if(unlikely(!*s)) {
824         w->wait_receive = 1;
825         return HTTP_VALIDATION_INCOMPLETE;
826     }
827
828     // we have the end of encoded_url - remember it
829     char *ue = s;
830
831     // make sure we have complete request
832     // complete requests contain: \r\n\r\n
833     while(*s) {
834         // find a line feed
835         while(*s && *s++ != '\r');
836
837         // did we reach the end?
838         if(unlikely(!*s)) break;
839
840         // is it \r\n ?
841         if(likely(*s++ == '\n')) {
842
843             // is it again \r\n ? (header end)
844             if(unlikely(*s == '\r' && s[1] == '\n')) {
845                 // a valid complete HTTP request found
846
847                 *ue = '\0';
848                 url_decode_r(w->decoded_url, encoded_url, URL_MAX + 1);
849                 *ue = ' ';
850                 
851                 // copy the URL - we are going to overwrite parts of it
852                 // FIXME -- we should avoid it
853                 strncpyz(w->last_url, w->decoded_url, URL_MAX);
854
855                 w->wait_receive = 0;
856                 return HTTP_VALIDATION_OK;
857             }
858
859             // another header line
860             s = http_header_parse(w, s);
861         }
862     }
863
864     // incomplete request
865     w->wait_receive = 1;
866     return HTTP_VALIDATION_INCOMPLETE;
867 }
868
869 static inline void web_client_send_http_header(struct web_client *w) {
870     if(unlikely(w->response.code != 200))
871         buffer_no_cacheable(w->response.data);
872
873     // set a proper expiration date, if not already set
874     if(unlikely(!w->response.data->expires)) {
875         if(w->response.data->options & WB_CONTENT_NO_CACHEABLE)
876             w->response.data->expires = w->tv_ready.tv_sec + localhost->rrd_update_every;
877         else
878             w->response.data->expires = w->tv_ready.tv_sec + 86400;
879     }
880
881     // prepare the HTTP response header
882     debug(D_WEB_CLIENT, "%llu: Generating HTTP header with response %d.", w->id, w->response.code);
883
884     const char *content_type_string = web_content_type_to_string(w->response.data->contenttype);
885     const char *code_msg = web_response_code_to_string(w->response.code);
886
887     // prepare the last modified and expiration dates
888     char date[32], edate[32];
889     {
890         struct tm tmbuf, *tm;
891
892         tm = gmtime_r(&w->response.data->date, &tmbuf);
893         strftime(date, sizeof(date), "%a, %d %b %Y %H:%M:%S %Z", tm);
894
895         tm = gmtime_r(&w->response.data->expires, &tmbuf);
896         strftime(edate, sizeof(edate), "%a, %d %b %Y %H:%M:%S %Z", tm);
897     }
898
899     buffer_sprintf(w->response.header_output,
900             "HTTP/1.1 %d %s\r\n"
901                     "Connection: %s\r\n"
902                     "Server: NetData Embedded HTTP Server\r\n"
903                     "Access-Control-Allow-Origin: %s\r\n"
904                     "Access-Control-Allow-Credentials: true\r\n"
905                     "Content-Type: %s\r\n"
906                     "Date: %s\r\n"
907                    , w->response.code, code_msg
908                    , w->keepalive?"keep-alive":"close"
909                    , w->origin
910                    , content_type_string
911                    , date
912     );
913
914     if(unlikely(web_x_frame_options))
915         buffer_sprintf(w->response.header_output, "X-Frame-Options: %s\r\n", web_x_frame_options);
916
917     if(w->cookie1[0] || w->cookie2[0]) {
918         if(w->cookie1[0]) {
919             buffer_sprintf(w->response.header_output,
920                     "Set-Cookie: %s\r\n",
921                     w->cookie1);
922         }
923
924         if(w->cookie2[0]) {
925             buffer_sprintf(w->response.header_output,
926                     "Set-Cookie: %s\r\n",
927                     w->cookie2);
928         }
929
930         if(respect_web_browser_do_not_track_policy)
931             buffer_sprintf(w->response.header_output,
932                     "Tk: T;cookies\r\n");
933     }
934     else {
935         if(respect_web_browser_do_not_track_policy) {
936             if(w->tracking_required)
937                 buffer_sprintf(w->response.header_output,
938                         "Tk: T;cookies\r\n");
939             else
940                 buffer_sprintf(w->response.header_output,
941                         "Tk: N\r\n");
942         }
943     }
944
945     if(w->mode == WEB_CLIENT_MODE_OPTIONS) {
946         buffer_strcat(w->response.header_output,
947                 "Access-Control-Allow-Methods: GET, OPTIONS\r\n"
948                         "Access-Control-Allow-Headers: accept, x-requested-with, origin, content-type, cookie, pragma, cache-control\r\n"
949                         "Access-Control-Max-Age: 1209600\r\n" // 86400 * 14
950         );
951     }
952     else {
953         buffer_sprintf(w->response.header_output,
954                 "Cache-Control: %s\r\n"
955                         "Expires: %s\r\n",
956                 (w->response.data->options & WB_CONTENT_NO_CACHEABLE)?"no-cache":"public",
957                 edate);
958     }
959
960     // copy a possibly available custom header
961     if(unlikely(buffer_strlen(w->response.header)))
962         buffer_strcat(w->response.header_output, buffer_tostring(w->response.header));
963
964     // headers related to the transfer method
965     if(likely(w->response.zoutput)) {
966         buffer_strcat(w->response.header_output,
967                 "Content-Encoding: gzip\r\n"
968                         "Transfer-Encoding: chunked\r\n"
969         );
970     }
971     else {
972         if(likely((w->response.data->len || w->response.rlen))) {
973             // we know the content length, put it
974             buffer_sprintf(w->response.header_output, "Content-Length: %zu\r\n", w->response.data->len? w->response.data->len: w->response.rlen);
975         }
976         else {
977             // we don't know the content length, disable keep-alive
978             w->keepalive = 0;
979         }
980     }
981
982     // end of HTTP header
983     buffer_strcat(w->response.header_output, "\r\n");
984
985     // sent the HTTP header
986     debug(D_WEB_DATA, "%llu: Sending response HTTP header of size %zu: '%s'"
987           , w->id
988           , buffer_strlen(w->response.header_output)
989           , buffer_tostring(w->response.header_output)
990     );
991
992     web_client_crock_socket(w);
993
994     ssize_t bytes = send(w->ofd, buffer_tostring(w->response.header_output), buffer_strlen(w->response.header_output), 0);
995     if(bytes != (ssize_t) buffer_strlen(w->response.header_output)) {
996         if(bytes > 0)
997             w->stats_sent_bytes += bytes;
998
999         debug(D_WEB_CLIENT, "%llu: HTTP Header failed to be sent (I sent %zu bytes but the system sent %zd bytes). Closing web client."
1000               , w->id
1001               , buffer_strlen(w->response.header_output)
1002               , bytes);
1003
1004         WEB_CLIENT_IS_DEAD(w);
1005         return;
1006     }
1007     else
1008         w->stats_sent_bytes += bytes;
1009 }
1010
1011 static inline int web_client_process_url(RRDHOST *host, struct web_client *w, char *url);
1012
1013 static inline int web_client_switch_host(RRDHOST *host, struct web_client *w, char *url) {
1014     static uint32_t hash_localhost = 0;
1015
1016     if(unlikely(!hash_localhost)) {
1017         hash_localhost = simple_hash("localhost");
1018     }
1019
1020     if(host != localhost) {
1021         buffer_flush(w->response.data);
1022         buffer_strcat(w->response.data, "Nesting of hosts is not allowed.");
1023         return 400;
1024     }
1025
1026     char *tok = mystrsep(&url, "/?&");
1027     if(tok && *tok) {
1028         debug(D_WEB_CLIENT, "%llu: Searching for host with name '%s'.", w->id, tok);
1029
1030         // copy the URL, we need it to serve files
1031         w->last_url[0] = '/';
1032         if(url && *url) strncpyz(&w->last_url[1], url, URL_MAX - 1);
1033         else w->last_url[1] = '\0';
1034
1035         uint32_t hash = simple_hash(tok);
1036
1037         host = rrdhost_find_by_hostname(tok, hash);
1038         if(!host) host = rrdhost_find_by_guid(tok, hash);
1039
1040         if(host) return web_client_process_url(host, w, url);
1041     }
1042
1043     buffer_flush(w->response.data);
1044     w->response.data->contenttype = CT_TEXT_HTML;
1045     buffer_strcat(w->response.data, "This netdata does not maintain a database for host: ");
1046     buffer_strcat_htmlescape(w->response.data, tok?tok:"");
1047     return 404;
1048 }
1049
1050 static inline int web_client_process_url(RRDHOST *host, struct web_client *w, char *url) {
1051     static uint32_t
1052             hash_api = 0,
1053             hash_netdata_conf = 0,
1054             hash_data = 0,
1055             hash_datasource = 0,
1056             hash_graph = 0,
1057             hash_list = 0,
1058             hash_all_json = 0,
1059             hash_host = 0;
1060
1061 #ifdef NETDATA_INTERNAL_CHECKS
1062     static uint32_t hash_exit = 0, hash_debug = 0, hash_mirror = 0;
1063 #endif
1064
1065     if(unlikely(!hash_api)) {
1066         hash_api = simple_hash("api");
1067         hash_netdata_conf = simple_hash("netdata.conf");
1068         hash_data = simple_hash(WEB_PATH_DATA);
1069         hash_datasource = simple_hash(WEB_PATH_DATASOURCE);
1070         hash_graph = simple_hash(WEB_PATH_GRAPH);
1071         hash_list = simple_hash("list");
1072         hash_all_json = simple_hash("all.json");
1073         hash_host = simple_hash("host");
1074 #ifdef NETDATA_INTERNAL_CHECKS
1075         hash_exit = simple_hash("exit");
1076         hash_debug = simple_hash("debug");
1077         hash_mirror = simple_hash("mirror");
1078 #endif
1079     }
1080
1081     char *tok = mystrsep(&url, "/?");
1082     if(likely(tok && *tok)) {
1083         uint32_t hash = simple_hash(tok);
1084         debug(D_WEB_CLIENT, "%llu: Processing command '%s'.", w->id, tok);
1085
1086         if(unlikely(hash == hash_api && strcmp(tok, "api") == 0)) {                           // current API
1087             debug(D_WEB_CLIENT_ACCESS, "%llu: API request ...", w->id);
1088             return check_host_and_call(host, w, url, web_client_api_request);
1089         }
1090         else if(unlikely(hash == hash_host && strcmp(tok, "host") == 0)) {                    // host switching
1091             debug(D_WEB_CLIENT_ACCESS, "%llu: host switch request ...", w->id);
1092             return web_client_switch_host(host, w, url);
1093         }
1094         else if(unlikely(hash == hash_data && strcmp(tok, WEB_PATH_DATA) == 0)) {             // old API "data"
1095             debug(D_WEB_CLIENT_ACCESS, "%llu: old API data request...", w->id);
1096             return check_host_and_call(host, w, url, web_client_api_old_data_request_json);
1097         }
1098         else if(unlikely(hash == hash_datasource && strcmp(tok, WEB_PATH_DATASOURCE) == 0)) { // old API "datasource"
1099             debug(D_WEB_CLIENT_ACCESS, "%llu: old API datasource request...", w->id);
1100             return check_host_and_call(host, w, url, web_client_api_old_data_request_jsonp);
1101         }
1102         else if(unlikely(hash == hash_graph && strcmp(tok, WEB_PATH_GRAPH) == 0)) {           // old API "graph"
1103             debug(D_WEB_CLIENT_ACCESS, "%llu: old API graph request...", w->id);
1104             return check_host_and_call(host, w, url, web_client_api_old_graph_request);
1105         }
1106         else if(unlikely(hash == hash_list && strcmp(tok, "list") == 0)) {                    // old API "list"
1107             debug(D_WEB_CLIENT_ACCESS, "%llu: old API list request...", w->id);
1108             return check_host_and_call(host, w, url, web_client_api_old_list_request);
1109         }
1110         else if(unlikely(hash == hash_all_json && strcmp(tok, "all.json") == 0)) {            // old API "all.json"
1111             debug(D_WEB_CLIENT_ACCESS, "%llu: old API all.json request...", w->id);
1112             return check_host_and_call(host, w, url, web_client_api_old_all_json);
1113         }
1114         else if(unlikely(hash == hash_netdata_conf && strcmp(tok, "netdata.conf") == 0)) {    // netdata.conf
1115             debug(D_WEB_CLIENT_ACCESS, "%llu: generating netdata.conf ...", w->id);
1116             w->response.data->contenttype = CT_TEXT_PLAIN;
1117             buffer_flush(w->response.data);
1118             config_generate(w->response.data, 0);
1119             return 200;
1120         }
1121 #ifdef NETDATA_INTERNAL_CHECKS
1122         else if(unlikely(hash == hash_exit && strcmp(tok, "exit") == 0)) {
1123             w->response.data->contenttype = CT_TEXT_PLAIN;
1124             buffer_flush(w->response.data);
1125
1126             if(!netdata_exit)
1127                 buffer_strcat(w->response.data, "ok, will do...");
1128             else
1129                 buffer_strcat(w->response.data, "I am doing it already");
1130
1131             error("web request to exit received.");
1132             netdata_cleanup_and_exit(0);
1133             return 200;
1134         }
1135         else if(unlikely(hash == hash_debug && strcmp(tok, "debug") == 0)) {
1136             buffer_flush(w->response.data);
1137
1138             // get the name of the data to show
1139             tok = mystrsep(&url, "/?&");
1140             if(tok && *tok) {
1141                 debug(D_WEB_CLIENT, "%llu: Searching for RRD data with name '%s'.", w->id, tok);
1142
1143                 // do we have such a data set?
1144                 RRDSET *st = rrdset_find_byname(host, tok);
1145                 if(!st) st = rrdset_find(host, tok);
1146                 if(!st) {
1147                     w->response.data->contenttype = CT_TEXT_HTML;
1148                     buffer_strcat(w->response.data, "Chart is not found: ");
1149                     buffer_strcat_htmlescape(w->response.data, tok);
1150                     debug(D_WEB_CLIENT_ACCESS, "%llu: %s is not found.", w->id, tok);
1151                     return 404;
1152                 }
1153
1154                 debug_flags |= D_RRD_STATS;
1155
1156                 if(rrdset_flag_check(st, RRDSET_FLAG_DEBUG))
1157                     rrdset_flag_clear(st, RRDSET_FLAG_DEBUG);
1158                 else
1159                     rrdset_flag_set(st, RRDSET_FLAG_DEBUG);
1160
1161                 w->response.data->contenttype = CT_TEXT_HTML;
1162                 buffer_sprintf(w->response.data, "Chart has now debug %s: ", rrdset_flag_check(st, RRDSET_FLAG_DEBUG)?"enabled":"disabled");
1163                 buffer_strcat_htmlescape(w->response.data, tok);
1164                 debug(D_WEB_CLIENT_ACCESS, "%llu: debug for %s is %s.", w->id, tok, rrdset_flag_check(st, RRDSET_FLAG_DEBUG)?"enabled":"disabled");
1165                 return 200;
1166             }
1167
1168             buffer_flush(w->response.data);
1169             buffer_strcat(w->response.data, "debug which chart?\r\n");
1170             return 400;
1171         }
1172         else if(unlikely(hash == hash_mirror && strcmp(tok, "mirror") == 0)) {
1173             debug(D_WEB_CLIENT_ACCESS, "%llu: Mirroring...", w->id);
1174
1175             // replace the zero bytes with spaces
1176             buffer_char_replace(w->response.data, '\0', ' ');
1177
1178             // just leave the buffer as is
1179             // it will be copied back to the client
1180
1181             return 200;
1182         }
1183 #endif  /* NETDATA_INTERNAL_CHECKS */
1184     }
1185
1186     char filename[FILENAME_MAX+1];
1187     url = filename;
1188     strncpyz(filename, w->last_url, FILENAME_MAX);
1189     tok = mystrsep(&url, "?");
1190     buffer_flush(w->response.data);
1191     return mysendfile(w, (tok && *tok)?tok:"/");
1192 }
1193
1194 void web_client_process_request(struct web_client *w) {
1195
1196     // start timing us
1197     now_realtime_timeval(&w->tv_in);
1198
1199     switch(http_request_validate(w)) {
1200         case HTTP_VALIDATION_OK:
1201             switch(w->mode) {
1202                 case WEB_CLIENT_MODE_STREAM:
1203                     w->response.code = rrdpush_receiver_thread_spawn(localhost, w, w->decoded_url);
1204                     return;
1205
1206                 case WEB_CLIENT_MODE_OPTIONS:
1207                     w->response.data->contenttype = CT_TEXT_PLAIN;
1208                     buffer_flush(w->response.data);
1209                     buffer_strcat(w->response.data, "OK");
1210                     w->response.code = 200;
1211                     break;
1212
1213                 case WEB_CLIENT_MODE_FILECOPY:
1214                 case WEB_CLIENT_MODE_NORMAL:
1215                     w->response.code = web_client_process_url(localhost, w, w->decoded_url);
1216                     break;
1217             }
1218             break;
1219
1220         case HTTP_VALIDATION_INCOMPLETE:
1221             if(w->response.data->len > TOO_BIG_REQUEST) {
1222                 strcpy(w->last_url, "too big request");
1223
1224                 debug(D_WEB_CLIENT_ACCESS, "%llu: Received request is too big (%zu bytes).", w->id, w->response.data->len);
1225
1226                 buffer_flush(w->response.data);
1227                 buffer_sprintf(w->response.data, "Received request is too big  (%zu bytes).\r\n", w->response.data->len);
1228                 w->response.code = 400;
1229             }
1230             else {
1231                 // wait for more data
1232                 return;
1233             }
1234             break;
1235
1236         case HTTP_VALIDATION_NOT_SUPPORTED:
1237             debug(D_WEB_CLIENT_ACCESS, "%llu: Cannot understand '%s'.", w->id, w->response.data->buffer);
1238
1239             buffer_flush(w->response.data);
1240             buffer_strcat(w->response.data, "I don't understand you...\r\n");
1241             w->response.code = 400;
1242             break;
1243     }
1244
1245     // keep track of the time we done processing
1246     now_realtime_timeval(&w->tv_ready);
1247
1248     w->response.sent = 0;
1249
1250     // set a proper last modified date
1251     if(unlikely(!w->response.data->date))
1252         w->response.data->date = w->tv_ready.tv_sec;
1253
1254     web_client_send_http_header(w);
1255
1256     // enable sending immediately if we have data
1257     if(w->response.data->len) w->wait_send = 1;
1258     else w->wait_send = 0;
1259
1260     switch(w->mode) {
1261         case WEB_CLIENT_MODE_STREAM:
1262             debug(D_WEB_CLIENT, "%llu: STREAM done.", w->id);
1263             break;
1264
1265         case WEB_CLIENT_MODE_OPTIONS:
1266             debug(D_WEB_CLIENT, "%llu: Done preparing the OPTIONS response. Sending data (%zu bytes) to client.", w->id, w->response.data->len);
1267             break;
1268
1269         case WEB_CLIENT_MODE_NORMAL:
1270             debug(D_WEB_CLIENT, "%llu: Done preparing the response. Sending data (%zu bytes) to client.", w->id, w->response.data->len);
1271             break;
1272
1273         case WEB_CLIENT_MODE_FILECOPY:
1274             if(w->response.rlen) {
1275                 debug(D_WEB_CLIENT, "%llu: Done preparing the response. Will be sending data file of %zu bytes to client.", w->id, w->response.rlen);
1276                 w->wait_receive = 1;
1277
1278                 /*
1279                 // utilize the kernel sendfile() for copying the file to the socket.
1280                 // this block of code can be commented, without anything missing.
1281                 // when it is commented, the program will copy the data using async I/O.
1282                 {
1283                     long len = sendfile(w->ofd, w->ifd, NULL, w->response.data->rbytes);
1284                     if(len != w->response.data->rbytes)
1285                         error("%llu: sendfile() should copy %ld bytes, but copied %ld. Falling back to manual copy.", w->id, w->response.data->rbytes, len);
1286                     else
1287                         web_client_reset(w);
1288                 }
1289                 */
1290             }
1291             else
1292                 debug(D_WEB_CLIENT, "%llu: Done preparing the response. Will be sending an unknown amount of bytes to client.", w->id);
1293             break;
1294
1295         default:
1296             fatal("%llu: Unknown client mode %u.", w->id, w->mode);
1297             break;
1298     }
1299 }
1300
1301 ssize_t web_client_send_chunk_header(struct web_client *w, size_t len)
1302 {
1303     debug(D_DEFLATE, "%llu: OPEN CHUNK of %zu bytes (hex: %zx).", w->id, len, len);
1304     char buf[24];
1305     sprintf(buf, "%zX\r\n", len);
1306     
1307     ssize_t bytes = send(w->ofd, buf, strlen(buf), 0);
1308     if(bytes > 0) {
1309         debug(D_DEFLATE, "%llu: Sent chunk header %zd bytes.", w->id, bytes);
1310         w->stats_sent_bytes += bytes;
1311     }
1312
1313     else if(bytes == 0) {
1314         debug(D_WEB_CLIENT, "%llu: Did not send chunk header to the client.", w->id);
1315         WEB_CLIENT_IS_DEAD(w);
1316     }
1317     else {
1318         debug(D_WEB_CLIENT, "%llu: Failed to send chunk header to client.", w->id);
1319         WEB_CLIENT_IS_DEAD(w);
1320     }
1321
1322     return bytes;
1323 }
1324
1325 ssize_t web_client_send_chunk_close(struct web_client *w)
1326 {
1327     //debug(D_DEFLATE, "%llu: CLOSE CHUNK.", w->id);
1328
1329     ssize_t bytes = send(w->ofd, "\r\n", 2, 0);
1330     if(bytes > 0) {
1331         debug(D_DEFLATE, "%llu: Sent chunk suffix %zd bytes.", w->id, bytes);
1332         w->stats_sent_bytes += bytes;
1333     }
1334
1335     else if(bytes == 0) {
1336         debug(D_WEB_CLIENT, "%llu: Did not send chunk suffix to the client.", w->id);
1337         WEB_CLIENT_IS_DEAD(w);
1338     }
1339     else {
1340         debug(D_WEB_CLIENT, "%llu: Failed to send chunk suffix to client.", w->id);
1341         WEB_CLIENT_IS_DEAD(w);
1342     }
1343
1344     return bytes;
1345 }
1346
1347 ssize_t web_client_send_chunk_finalize(struct web_client *w)
1348 {
1349     //debug(D_DEFLATE, "%llu: FINALIZE CHUNK.", w->id);
1350
1351     ssize_t bytes = send(w->ofd, "\r\n0\r\n\r\n", 7, 0);
1352     if(bytes > 0) {
1353         debug(D_DEFLATE, "%llu: Sent chunk suffix %zd bytes.", w->id, bytes);
1354         w->stats_sent_bytes += bytes;
1355     }
1356
1357     else if(bytes == 0) {
1358         debug(D_WEB_CLIENT, "%llu: Did not send chunk finalize suffix to the client.", w->id);
1359         WEB_CLIENT_IS_DEAD(w);
1360     }
1361     else {
1362         debug(D_WEB_CLIENT, "%llu: Failed to send chunk finalize suffix to client.", w->id);
1363         WEB_CLIENT_IS_DEAD(w);
1364     }
1365
1366     return bytes;
1367 }
1368
1369 #ifdef NETDATA_WITH_ZLIB
1370 ssize_t web_client_send_deflate(struct web_client *w)
1371 {
1372     ssize_t len = 0, t = 0;
1373
1374     // when using compression,
1375     // w->response.sent is the amount of bytes passed through compression
1376
1377     debug(D_DEFLATE, "%llu: web_client_send_deflate(): w->response.data->len = %zu, w->response.sent = %zu, w->response.zhave = %zu, w->response.zsent = %zu, w->response.zstream.avail_in = %u, w->response.zstream.avail_out = %u, w->response.zstream.total_in = %lu, w->response.zstream.total_out = %lu.",
1378         w->id, w->response.data->len, w->response.sent, w->response.zhave, w->response.zsent, w->response.zstream.avail_in, w->response.zstream.avail_out, w->response.zstream.total_in, w->response.zstream.total_out);
1379
1380     if(w->response.data->len - w->response.sent == 0 && w->response.zstream.avail_in == 0 && w->response.zhave == w->response.zsent && w->response.zstream.avail_out != 0) {
1381         // there is nothing to send
1382
1383         debug(D_WEB_CLIENT, "%llu: Out of output data.", w->id);
1384
1385         // finalize the chunk
1386         if(w->response.sent != 0) {
1387             t = web_client_send_chunk_finalize(w);
1388             if(t < 0) return t;
1389         }
1390
1391         if(w->mode == WEB_CLIENT_MODE_FILECOPY && w->wait_receive && w->response.rlen && w->response.rlen > w->response.data->len) {
1392             // we have to wait, more data will come
1393             debug(D_WEB_CLIENT, "%llu: Waiting for more data to become available.", w->id);
1394             w->wait_send = 0;
1395             return t;
1396         }
1397
1398         if(unlikely(!w->keepalive)) {
1399             debug(D_WEB_CLIENT, "%llu: Closing (keep-alive is not enabled). %zu bytes sent.", w->id, w->response.sent);
1400             WEB_CLIENT_IS_DEAD(w);
1401             return t;
1402         }
1403
1404         // reset the client
1405         web_client_reset(w);
1406         debug(D_WEB_CLIENT, "%llu: Done sending all data on socket.", w->id);
1407         return t;
1408     }
1409
1410     if(w->response.zhave == w->response.zsent) {
1411         // compress more input data
1412
1413         // close the previous open chunk
1414         if(w->response.sent != 0) {
1415             t = web_client_send_chunk_close(w);
1416             if(t < 0) return t;
1417         }
1418
1419         debug(D_DEFLATE, "%llu: Compressing %zu new bytes starting from %zu (and %u left behind).", w->id, (w->response.data->len - w->response.sent), w->response.sent, w->response.zstream.avail_in);
1420
1421         // give the compressor all the data not passed through the compressor yet
1422         if(w->response.data->len > w->response.sent) {
1423             w->response.zstream.next_in = (Bytef *)&w->response.data->buffer[w->response.sent - w->response.zstream.avail_in];
1424             w->response.zstream.avail_in += (uInt) (w->response.data->len - w->response.sent);
1425         }
1426
1427         // reset the compressor output buffer
1428         w->response.zstream.next_out = w->response.zbuffer;
1429         w->response.zstream.avail_out = ZLIB_CHUNK;
1430
1431         // ask for FINISH if we have all the input
1432         int flush = Z_SYNC_FLUSH;
1433         if(w->mode == WEB_CLIENT_MODE_NORMAL
1434             || (w->mode == WEB_CLIENT_MODE_FILECOPY && !w->wait_receive && w->response.data->len == w->response.rlen)) {
1435             flush = Z_FINISH;
1436             debug(D_DEFLATE, "%llu: Requesting Z_FINISH, if possible.", w->id);
1437         }
1438         else {
1439             debug(D_DEFLATE, "%llu: Requesting Z_SYNC_FLUSH.", w->id);
1440         }
1441
1442         // compress
1443         if(deflate(&w->response.zstream, flush) == Z_STREAM_ERROR) {
1444             error("%llu: Compression failed. Closing down client.", w->id);
1445             web_client_reset(w);
1446             return(-1);
1447         }
1448
1449         w->response.zhave = ZLIB_CHUNK - w->response.zstream.avail_out;
1450         w->response.zsent = 0;
1451
1452         // keep track of the bytes passed through the compressor
1453         w->response.sent = w->response.data->len;
1454
1455         debug(D_DEFLATE, "%llu: Compression produced %zu bytes.", w->id, w->response.zhave);
1456
1457         // open a new chunk
1458         ssize_t t2 = web_client_send_chunk_header(w, w->response.zhave);
1459         if(t2 < 0) return t2;
1460         t += t2;
1461     }
1462     
1463     debug(D_WEB_CLIENT, "%llu: Sending %zu bytes of data (+%zd of chunk header).", w->id, w->response.zhave - w->response.zsent, t);
1464
1465     len = send(w->ofd, &w->response.zbuffer[w->response.zsent], (size_t) (w->response.zhave - w->response.zsent), MSG_DONTWAIT);
1466     if(len > 0) {
1467         w->stats_sent_bytes += len;
1468         w->response.zsent += len;
1469         len += t;
1470         debug(D_WEB_CLIENT, "%llu: Sent %zd bytes.", w->id, len);
1471     }
1472     else if(len == 0) {
1473         debug(D_WEB_CLIENT, "%llu: Did not send any bytes to the client (zhave = %zu, zsent = %zu, need to send = %zu).",
1474             w->id, w->response.zhave, w->response.zsent, w->response.zhave - w->response.zsent);
1475
1476         WEB_CLIENT_IS_DEAD(w);
1477     }
1478     else {
1479         debug(D_WEB_CLIENT, "%llu: Failed to send data to client.", w->id);
1480         WEB_CLIENT_IS_DEAD(w);
1481     }
1482
1483     return(len);
1484 }
1485 #endif // NETDATA_WITH_ZLIB
1486
1487 ssize_t web_client_send(struct web_client *w) {
1488 #ifdef NETDATA_WITH_ZLIB
1489     if(likely(w->response.zoutput)) return web_client_send_deflate(w);
1490 #endif // NETDATA_WITH_ZLIB
1491
1492     ssize_t bytes;
1493
1494     if(unlikely(w->response.data->len - w->response.sent == 0)) {
1495         // there is nothing to send
1496
1497         debug(D_WEB_CLIENT, "%llu: Out of output data.", w->id);
1498
1499         // there can be two cases for this
1500         // A. we have done everything
1501         // B. we temporarily have nothing to send, waiting for the buffer to be filled by ifd
1502
1503         if(w->mode == WEB_CLIENT_MODE_FILECOPY && w->wait_receive && w->response.rlen && w->response.rlen > w->response.data->len) {
1504             // we have to wait, more data will come
1505             debug(D_WEB_CLIENT, "%llu: Waiting for more data to become available.", w->id);
1506             w->wait_send = 0;
1507             return 0;
1508         }
1509
1510         if(unlikely(!w->keepalive)) {
1511             debug(D_WEB_CLIENT, "%llu: Closing (keep-alive is not enabled). %zu bytes sent.", w->id, w->response.sent);
1512             WEB_CLIENT_IS_DEAD(w);
1513             return 0;
1514         }
1515
1516         web_client_reset(w);
1517         debug(D_WEB_CLIENT, "%llu: Done sending all data on socket. Waiting for next request on the same socket.", w->id);
1518         return 0;
1519     }
1520
1521     bytes = send(w->ofd, &w->response.data->buffer[w->response.sent], w->response.data->len - w->response.sent, MSG_DONTWAIT);
1522     if(likely(bytes > 0)) {
1523         w->stats_sent_bytes += bytes;
1524         w->response.sent += bytes;
1525         debug(D_WEB_CLIENT, "%llu: Sent %zd bytes.", w->id, bytes);
1526     }
1527     else if(likely(bytes == 0)) {
1528         debug(D_WEB_CLIENT, "%llu: Did not send any bytes to the client.", w->id);
1529         WEB_CLIENT_IS_DEAD(w);
1530     }
1531     else {
1532         debug(D_WEB_CLIENT, "%llu: Failed to send data to client.", w->id);
1533         WEB_CLIENT_IS_DEAD(w);
1534     }
1535
1536     return(bytes);
1537 }
1538
1539 ssize_t web_client_receive(struct web_client *w)
1540 {
1541     // do we have any space for more data?
1542     buffer_need_bytes(w->response.data, WEB_REQUEST_LENGTH);
1543
1544     ssize_t left = w->response.data->size - w->response.data->len;
1545     ssize_t bytes;
1546
1547     if(unlikely(w->mode == WEB_CLIENT_MODE_FILECOPY))
1548         bytes = read(w->ifd, &w->response.data->buffer[w->response.data->len], (size_t) (left - 1));
1549     else
1550         bytes = recv(w->ifd, &w->response.data->buffer[w->response.data->len], (size_t) (left - 1), MSG_DONTWAIT);
1551
1552     if(likely(bytes > 0)) {
1553         if(w->mode != WEB_CLIENT_MODE_FILECOPY)
1554             w->stats_received_bytes += bytes;
1555
1556         size_t old = w->response.data->len;
1557         w->response.data->len += bytes;
1558         w->response.data->buffer[w->response.data->len] = '\0';
1559
1560         debug(D_WEB_CLIENT, "%llu: Received %zd bytes.", w->id, bytes);
1561         debug(D_WEB_DATA, "%llu: Received data: '%s'.", w->id, &w->response.data->buffer[old]);
1562
1563         if(w->mode == WEB_CLIENT_MODE_FILECOPY) {
1564             w->wait_send = 1;
1565
1566             if(w->response.rlen && w->response.data->len >= w->response.rlen)
1567                 w->wait_receive = 0;
1568         }
1569     }
1570     else if(likely(bytes == 0)) {
1571         debug(D_WEB_CLIENT, "%llu: Out of input data.", w->id);
1572
1573         // if we cannot read, it means we have an error on input.
1574         // if however, we are copying a file from ifd to ofd, we should not return an error.
1575         // in this case, the error should be generated when the file has been sent to the client.
1576
1577         if(w->mode == WEB_CLIENT_MODE_FILECOPY) {
1578             // we are copying data from ifd to ofd
1579             // let it finish copying...
1580             w->wait_receive = 0;
1581
1582             debug(D_WEB_CLIENT, "%llu: Read the whole file.", w->id);
1583             if(w->ifd != w->ofd) close(w->ifd);
1584             w->ifd = w->ofd;
1585         }
1586         else {
1587             debug(D_WEB_CLIENT, "%llu: failed to receive data.", w->id);
1588             WEB_CLIENT_IS_DEAD(w);
1589         }
1590     }
1591     else {
1592         debug(D_WEB_CLIENT, "%llu: receive data failed.", w->id);
1593         WEB_CLIENT_IS_DEAD(w);
1594     }
1595
1596     return(bytes);
1597 }
1598
1599
1600 // --------------------------------------------------------------------------------------
1601 // the thread of a single client
1602
1603 // 1. waits for input and output, using async I/O
1604 // 2. it processes HTTP requests
1605 // 3. it generates HTTP responses
1606 // 4. it copies data from input to output if mode is FILECOPY
1607
1608 void *web_client_main(void *ptr)
1609 {
1610     if(pthread_setcanceltype(PTHREAD_CANCEL_DEFERRED, NULL) != 0)
1611         error("Cannot set pthread cancel type to DEFERRED.");
1612
1613     if(pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL) != 0)
1614         error("Cannot set pthread cancel state to ENABLE.");
1615
1616     struct web_client *w = ptr;
1617     struct pollfd fds[2], *ifd, *ofd;
1618     int retval, timeout;
1619     nfds_t fdmax = 0;
1620
1621     log_access("%llu: %s port %s connected on thread task id %d", w->id, w->client_ip, w->client_port, gettid());
1622
1623     for(;;) {
1624         if(unlikely(netdata_exit)) break;
1625
1626         if(unlikely(w->dead)) {
1627             debug(D_WEB_CLIENT, "%llu: client is dead.", w->id);
1628             break;
1629         }
1630         else if(unlikely(!w->wait_receive && !w->wait_send)) {
1631             debug(D_WEB_CLIENT, "%llu: client is not set for neither receiving nor sending data.", w->id);
1632             break;
1633         }
1634
1635         if(unlikely(w->ifd < 0 || w->ofd < 0)) {
1636             error("%llu: invalid file descriptor, ifd = %d, ofd = %d (required 0 <= fd", w->id, w->ifd, w->ofd);
1637             break;
1638         }
1639
1640         if(w->ifd == w->ofd) {
1641             fds[0].fd = w->ifd;
1642             fds[0].events = 0;
1643             fds[0].revents = 0;
1644
1645             if(w->wait_receive) fds[0].events |= POLLIN;
1646             if(w->wait_send)    fds[0].events |= POLLOUT;
1647
1648             fds[1].fd = -1;
1649             fds[1].events = 0;
1650             fds[1].revents = 0;
1651
1652             ifd = ofd = &fds[0];
1653
1654             fdmax = 1;
1655         }
1656         else {
1657             fds[0].fd = w->ifd;
1658             fds[0].events = 0;
1659             fds[0].revents = 0;
1660             if(w->wait_receive) fds[0].events |= POLLIN;
1661             ifd = &fds[0];
1662
1663             fds[1].fd = w->ofd;
1664             fds[1].events = 0;
1665             fds[1].revents = 0;
1666             if(w->wait_send)    fds[1].events |= POLLOUT;
1667             ofd = &fds[1];
1668
1669             fdmax = 2;
1670         }
1671
1672         debug(D_WEB_CLIENT, "%llu: Waiting socket async I/O for %s %s", w->id, w->wait_receive?"INPUT":"", w->wait_send?"OUTPUT":"");
1673         errno = 0;
1674         timeout = web_client_timeout * 1000;
1675         retval = poll(fds, fdmax, timeout);
1676
1677         if(unlikely(netdata_exit)) break;
1678
1679         if(unlikely(retval == -1)) {
1680             if(errno == EAGAIN || errno == EINTR) {
1681                 debug(D_WEB_CLIENT, "%llu: EAGAIN received.", w->id);
1682                 continue;
1683             }
1684
1685             debug(D_WEB_CLIENT, "%llu: LISTENER: poll() failed (input fd = %d, output fd = %d). Closing client.", w->id, w->ifd, w->ofd);
1686             break;
1687         }
1688         else if(unlikely(!retval)) {
1689             debug(D_WEB_CLIENT, "%llu: Timeout while waiting socket async I/O for %s %s", w->id, w->wait_receive?"INPUT":"", w->wait_send?"OUTPUT":"");
1690             break;
1691         }
1692
1693         if(unlikely(netdata_exit)) break;
1694
1695         int used = 0;
1696         if(w->wait_send && ofd->revents & POLLOUT) {
1697             used++;
1698             if(web_client_send(w) < 0) {
1699                 debug(D_WEB_CLIENT, "%llu: Cannot send data to client. Closing client.", w->id);
1700                 break;
1701             }
1702         }
1703
1704         if(unlikely(netdata_exit)) break;
1705
1706         if(w->wait_receive && (ifd->revents & POLLIN || ifd->revents & POLLPRI)) {
1707             used++;
1708             if(web_client_receive(w) < 0) {
1709                 debug(D_WEB_CLIENT, "%llu: Cannot receive data from client. Closing client.", w->id);
1710                 break;
1711             }
1712
1713             if(w->mode == WEB_CLIENT_MODE_NORMAL) {
1714                 debug(D_WEB_CLIENT, "%llu: Attempting to process received data.", w->id);
1715                 web_client_process_request(w);
1716
1717                 // if the sockets are closed, may have transferred this client
1718                 // to plugins.d
1719                 if(unlikely(w->mode == WEB_CLIENT_MODE_STREAM))
1720                     break;
1721             }
1722         }
1723
1724         if(unlikely(!used)) {
1725             debug(D_WEB_CLIENT_ACCESS, "%llu: Received error on socket.", w->id);
1726             break;
1727         }
1728     }
1729
1730     web_client_reset(w);
1731
1732     log_access("%llu: %s port %s disconnected from thread task id %d", w->id, w->client_ip, w->client_port, gettid());
1733     debug(D_WEB_CLIENT, "%llu: done...", w->id);
1734
1735     // close the sockets/files now
1736     // to free file descriptors
1737     if(w->ifd == w->ofd) {
1738         if(w->ifd != -1) close(w->ifd);
1739     }
1740     else {
1741         if(w->ifd != -1) close(w->ifd);
1742         if(w->ofd != -1) close(w->ofd);
1743     }
1744     w->ifd = -1;
1745     w->ofd = -1;
1746
1747     w->obsolete = 1;
1748
1749     pthread_exit(NULL);
1750     return NULL;
1751 }