]> arthur.barton.de Git - netdata.git/blob - src/web_client.c
added more fping alarms; added the ability to have alarms that never send CLEAR notif...
[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 web_donotrack_comply = 0;
9
10 #ifdef NETDATA_WITH_ZLIB
11 int web_enable_gzip = 1, web_gzip_level = 3, web_gzip_strategy = Z_DEFAULT_STRATEGY;
12 #endif /* NETDATA_WITH_ZLIB */
13
14 struct web_client *web_clients = NULL;
15 unsigned long long web_clients_count = 0;
16
17 static inline int web_client_crock_socket(struct web_client *w) {
18 #ifdef TCP_CORK
19     if(likely(!w->tcp_cork && w->ofd != -1)) {
20         w->tcp_cork = 1;
21         if(unlikely(setsockopt(w->ofd, IPPROTO_TCP, TCP_CORK, (char *) &w->tcp_cork, sizeof(int)) != 0)) {
22             error("%llu: failed to enable TCP_CORK on socket.", w->id);
23             w->tcp_cork = 0;
24             return -1;
25         }
26     }
27 #endif /* TCP_CORK */
28
29     return 0;
30 }
31
32 static inline int web_client_uncrock_socket(struct web_client *w) {
33 #ifdef TCP_CORK
34     if(likely(w->tcp_cork && w->ofd != -1)) {
35         w->tcp_cork = 0;
36         if(unlikely(setsockopt(w->ofd, IPPROTO_TCP, TCP_CORK, (char *) &w->tcp_cork, sizeof(int)) != 0)) {
37             error("%llu: failed to disable TCP_CORK on socket.", w->id);
38             w->tcp_cork = 1;
39             return -1;
40         }
41     }
42 #endif /* TCP_CORK */
43
44     return 0;
45 }
46
47 struct web_client *web_client_create(int listener)
48 {
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     if(w->response.header_output) buffer_free(w->response.header_output);
226     if(w->response.header) buffer_free(w->response.header);
227     if(w->response.data) 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("global", "web files owner", config_get("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("global", "web files group", config_get("global", "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 {
293     static char *web_dir = NULL;
294
295     // initialize our static data
296     if(unlikely(!web_dir)) web_dir = config_get("global", "web files directory", WEB_DIR);
297
298     debug(D_WEB_CLIENT, "%llu: Looking for file '%s/%s'", w->id, web_dir, filename);
299
300     // skip leading slashes
301     while (*filename == '/') filename++;
302
303     // if the filename contain known paths, skip them
304     if(strncmp(filename, WEB_PATH_FILE "/", strlen(WEB_PATH_FILE) + 1) == 0)
305         filename = &filename[strlen(WEB_PATH_FILE) + 1];
306
307     char *s;
308     for(s = filename; *s ;s++) {
309         if( !isalnum(*s) && *s != '/' && *s != '.' && *s != '-' && *s != '_') {
310             debug(D_WEB_CLIENT_ACCESS, "%llu: File '%s' is not acceptable.", w->id, filename);
311             buffer_sprintf(w->response.data, "Filename contains invalid characters: ");
312             buffer_strcat_htmlescape(w->response.data, filename);
313             return 400;
314         }
315     }
316
317     // if the filename contains a .. refuse to serve it
318     if(strstr(filename, "..") != 0) {
319         debug(D_WEB_CLIENT_ACCESS, "%llu: File '%s' is not acceptable.", w->id, filename);
320         buffer_strcat(w->response.data, "Relative filenames are not supported: ");
321         buffer_strcat_htmlescape(w->response.data, filename);
322         return 400;
323     }
324
325     // access the file
326     char webfilename[FILENAME_MAX + 1];
327     snprintfz(webfilename, FILENAME_MAX, "%s/%s", web_dir, filename);
328
329     // check if the file exists
330     struct stat stat;
331     if(lstat(webfilename, &stat) != 0) {
332         debug(D_WEB_CLIENT_ACCESS, "%llu: File '%s' is not found.", w->id, webfilename);
333         buffer_strcat(w->response.data, "File does not exist, or is not accessible: ");
334         buffer_strcat_htmlescape(w->response.data, webfilename);
335         return 404;
336     }
337
338     // check if the file is owned by expected user
339     if(stat.st_uid != web_files_uid()) {
340         error("%llu: File '%s' is owned by user %u (expected user %u). Access Denied.", w->id, webfilename, stat.st_uid, web_files_uid());
341         buffer_strcat(w->response.data, "Access to file is not permitted: ");
342         buffer_strcat_htmlescape(w->response.data, webfilename);
343         return 403;
344     }
345
346     // check if the file is owned by expected group
347     if(stat.st_gid != web_files_gid()) {
348         error("%llu: File '%s' is owned by group %u (expected group %u). Access Denied.", w->id, webfilename, stat.st_gid, web_files_gid());
349         buffer_strcat(w->response.data, "Access to file is not permitted: ");
350         buffer_strcat_htmlescape(w->response.data, webfilename);
351         return 403;
352     }
353
354     if((stat.st_mode & S_IFMT) == S_IFDIR) {
355         snprintfz(webfilename, FILENAME_MAX, "%s/index.html", filename);
356         return mysendfile(w, webfilename);
357     }
358
359     if((stat.st_mode & S_IFMT) != S_IFREG) {
360         error("%llu: File '%s' is not a regular file. Access Denied.", w->id, webfilename);
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             buffer_sprintf(w->response.header, "Location: /" WEB_PATH_FILE "/%s\r\n", filename);
374             buffer_strcat(w->response.data, "File is currently busy, please try again later: ");
375             buffer_strcat_htmlescape(w->response.data, webfilename);
376             return 307;
377         }
378         else {
379             error("%llu: Cannot open file '%s'.", w->id, webfilename);
380             buffer_strcat(w->response.data, "Cannot open file: ");
381             buffer_strcat_htmlescape(w->response.data, webfilename);
382             return 404;
383         }
384     }
385     if(fcntl(w->ifd, F_SETFL, O_NONBLOCK) < 0)
386         error("%llu: Cannot set O_NONBLOCK on file '%s'.", w->id, webfilename);
387
388     // pick a Content-Type for the file
389          if(strstr(filename, ".html") != NULL)  w->response.data->contenttype = CT_TEXT_HTML;
390     else if(strstr(filename, ".js")   != NULL)  w->response.data->contenttype = CT_APPLICATION_X_JAVASCRIPT;
391     else if(strstr(filename, ".css")  != NULL)  w->response.data->contenttype = CT_TEXT_CSS;
392     else if(strstr(filename, ".xml")  != NULL)  w->response.data->contenttype = CT_TEXT_XML;
393     else if(strstr(filename, ".xsl")  != NULL)  w->response.data->contenttype = CT_TEXT_XSL;
394     else if(strstr(filename, ".txt")  != NULL)  w->response.data->contenttype = CT_TEXT_PLAIN;
395     else if(strstr(filename, ".svg")  != NULL)  w->response.data->contenttype = CT_IMAGE_SVG_XML;
396     else if(strstr(filename, ".ttf")  != NULL)  w->response.data->contenttype = CT_APPLICATION_X_FONT_TRUETYPE;
397     else if(strstr(filename, ".otf")  != NULL)  w->response.data->contenttype = CT_APPLICATION_X_FONT_OPENTYPE;
398     else if(strstr(filename, ".woff2")!= NULL)  w->response.data->contenttype = CT_APPLICATION_FONT_WOFF2;
399     else if(strstr(filename, ".woff") != NULL)  w->response.data->contenttype = CT_APPLICATION_FONT_WOFF;
400     else if(strstr(filename, ".eot")  != NULL)  w->response.data->contenttype = CT_APPLICATION_VND_MS_FONTOBJ;
401     else if(strstr(filename, ".png")  != NULL)  w->response.data->contenttype = CT_IMAGE_PNG;
402     else if(strstr(filename, ".jpg")  != NULL)  w->response.data->contenttype = CT_IMAGE_JPG;
403     else if(strstr(filename, ".jpeg") != NULL)  w->response.data->contenttype = CT_IMAGE_JPG;
404     else if(strstr(filename, ".gif")  != NULL)  w->response.data->contenttype = CT_IMAGE_GIF;
405     else if(strstr(filename, ".bmp")  != NULL)  w->response.data->contenttype = CT_IMAGE_BMP;
406     else if(strstr(filename, ".ico")  != NULL)  w->response.data->contenttype = CT_IMAGE_XICON;
407     else if(strstr(filename, ".icns") != NULL)  w->response.data->contenttype = CT_IMAGE_ICNS;
408     else w->response.data->contenttype = CT_APPLICATION_OCTET_STREAM;
409
410     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);
411
412     w->mode = WEB_CLIENT_MODE_FILECOPY;
413     w->wait_receive = 1;
414     w->wait_send = 0;
415     buffer_flush(w->response.data);
416     w->response.rlen = stat.st_size;
417 #ifdef __APPLE__
418     w->response.data->date = stat.st_mtimespec.tv_sec;
419 #else
420     w->response.data->date = stat.st_mtim.tv_sec;
421 #endif /* __APPLE__ */
422     buffer_cacheable(w->response.data);
423
424     return 200;
425 }
426
427
428 #ifdef NETDATA_WITH_ZLIB
429 void web_client_enable_deflate(struct web_client *w, int gzip) {
430     if(unlikely(w->response.zinitialized)) {
431         debug(D_DEFLATE, "%llu: Compression has already be initialized for this client.", w->id);
432         return;
433     }
434
435     if(unlikely(w->response.sent)) {
436         error("%llu: Cannot enable compression in the middle of a conversation.", w->id);
437         return;
438     }
439
440     w->response.zstream.zalloc = Z_NULL;
441     w->response.zstream.zfree = Z_NULL;
442     w->response.zstream.opaque = Z_NULL;
443
444     w->response.zstream.next_in = (Bytef *)w->response.data->buffer;
445     w->response.zstream.avail_in = 0;
446     w->response.zstream.total_in = 0;
447
448     w->response.zstream.next_out = w->response.zbuffer;
449     w->response.zstream.avail_out = 0;
450     w->response.zstream.total_out = 0;
451
452     w->response.zstream.zalloc = Z_NULL;
453     w->response.zstream.zfree = Z_NULL;
454     w->response.zstream.opaque = Z_NULL;
455
456 //  if(deflateInit(&w->response.zstream, Z_DEFAULT_COMPRESSION) != Z_OK) {
457 //      error("%llu: Failed to initialize zlib. Proceeding without compression.", w->id);
458 //      return;
459 //  }
460
461     // Select GZIP compression: windowbits = 15 + 16 = 31
462     if(deflateInit2(&w->response.zstream, web_gzip_level, Z_DEFLATED, 15 + ((gzip)?16:0), 8, web_gzip_strategy) != Z_OK) {
463         error("%llu: Failed to initialize zlib. Proceeding without compression.", w->id);
464         return;
465     }
466
467     w->response.zsent = 0;
468     w->response.zoutput = 1;
469     w->response.zinitialized = 1;
470
471     debug(D_DEFLATE, "%llu: Initialized compression.", w->id);
472 }
473 #endif // NETDATA_WITH_ZLIB
474
475 void buffer_data_options2string(BUFFER *wb, uint32_t options) {
476     int count = 0;
477
478     if(options & RRDR_OPTION_NONZERO) {
479         if(count++) buffer_strcat(wb, " ");
480         buffer_strcat(wb, "nonzero");
481     }
482
483     if(options & RRDR_OPTION_REVERSED) {
484         if(count++) buffer_strcat(wb, " ");
485         buffer_strcat(wb, "flip");
486     }
487
488     if(options & RRDR_OPTION_JSON_WRAP) {
489         if(count++) buffer_strcat(wb, " ");
490         buffer_strcat(wb, "jsonwrap");
491     }
492
493     if(options & RRDR_OPTION_MIN2MAX) {
494         if(count++) buffer_strcat(wb, " ");
495         buffer_strcat(wb, "min2max");
496     }
497
498     if(options & RRDR_OPTION_MILLISECONDS) {
499         if(count++) buffer_strcat(wb, " ");
500         buffer_strcat(wb, "ms");
501     }
502
503     if(options & RRDR_OPTION_ABSOLUTE) {
504         if(count++) buffer_strcat(wb, " ");
505         buffer_strcat(wb, "absolute");
506     }
507
508     if(options & RRDR_OPTION_SECONDS) {
509         if(count++) buffer_strcat(wb, " ");
510         buffer_strcat(wb, "seconds");
511     }
512
513     if(options & RRDR_OPTION_NULL2ZERO) {
514         if(count++) buffer_strcat(wb, " ");
515         buffer_strcat(wb, "null2zero");
516     }
517
518     if(options & RRDR_OPTION_OBJECTSROWS) {
519         if(count++) buffer_strcat(wb, " ");
520         buffer_strcat(wb, "objectrows");
521     }
522
523     if(options & RRDR_OPTION_GOOGLE_JSON) {
524         if(count++) buffer_strcat(wb, " ");
525         buffer_strcat(wb, "google_json");
526     }
527
528     if(options & RRDR_OPTION_PERCENTAGE) {
529         if(count++) buffer_strcat(wb, " ");
530         buffer_strcat(wb, "percentage");
531     }
532
533     if(options & RRDR_OPTION_NOT_ALIGNED) {
534         if(count++) buffer_strcat(wb, " ");
535         buffer_strcat(wb, "unaligned");
536     }
537 }
538
539 uint32_t web_client_api_request_v1_data_options(char *o)
540 {
541     uint32_t ret = 0x00000000;
542     char *tok;
543
544     while(o && *o && (tok = mystrsep(&o, ", |"))) {
545         if(!*tok) continue;
546
547         if(!strcmp(tok, "nonzero"))
548             ret |= RRDR_OPTION_NONZERO;
549         else if(!strcmp(tok, "flip") || !strcmp(tok, "reversed") || !strcmp(tok, "reverse"))
550             ret |= RRDR_OPTION_REVERSED;
551         else if(!strcmp(tok, "jsonwrap"))
552             ret |= RRDR_OPTION_JSON_WRAP;
553         else if(!strcmp(tok, "min2max"))
554             ret |= RRDR_OPTION_MIN2MAX;
555         else if(!strcmp(tok, "ms") || !strcmp(tok, "milliseconds"))
556             ret |= RRDR_OPTION_MILLISECONDS;
557         else if(!strcmp(tok, "abs") || !strcmp(tok, "absolute") || !strcmp(tok, "absolute_sum") || !strcmp(tok, "absolute-sum"))
558             ret |= RRDR_OPTION_ABSOLUTE;
559         else if(!strcmp(tok, "seconds"))
560             ret |= RRDR_OPTION_SECONDS;
561         else if(!strcmp(tok, "null2zero"))
562             ret |= RRDR_OPTION_NULL2ZERO;
563         else if(!strcmp(tok, "objectrows"))
564             ret |= RRDR_OPTION_OBJECTSROWS;
565         else if(!strcmp(tok, "google_json"))
566             ret |= RRDR_OPTION_GOOGLE_JSON;
567         else if(!strcmp(tok, "percentage"))
568             ret |= RRDR_OPTION_PERCENTAGE;
569         else if(!strcmp(tok, "unaligned"))
570             ret |= RRDR_OPTION_NOT_ALIGNED;
571     }
572
573     return ret;
574 }
575
576 uint32_t web_client_api_request_v1_data_format(char *name)
577 {
578     if(!strcmp(name, DATASOURCE_FORMAT_DATATABLE_JSON)) // datatable
579         return DATASOURCE_DATATABLE_JSON;
580
581     else if(!strcmp(name, DATASOURCE_FORMAT_DATATABLE_JSONP)) // datasource
582         return DATASOURCE_DATATABLE_JSONP;
583
584     else if(!strcmp(name, DATASOURCE_FORMAT_JSON)) // json
585         return DATASOURCE_JSON;
586
587     else if(!strcmp(name, DATASOURCE_FORMAT_JSONP)) // jsonp
588         return DATASOURCE_JSONP;
589
590     else if(!strcmp(name, DATASOURCE_FORMAT_SSV)) // ssv
591         return DATASOURCE_SSV;
592
593     else if(!strcmp(name, DATASOURCE_FORMAT_CSV)) // csv
594         return DATASOURCE_CSV;
595
596     else if(!strcmp(name, DATASOURCE_FORMAT_TSV) || !strcmp(name, "tsv-excel")) // tsv
597         return DATASOURCE_TSV;
598
599     else if(!strcmp(name, DATASOURCE_FORMAT_HTML)) // html
600         return DATASOURCE_HTML;
601
602     else if(!strcmp(name, DATASOURCE_FORMAT_JS_ARRAY)) // array
603         return DATASOURCE_JS_ARRAY;
604
605     else if(!strcmp(name, DATASOURCE_FORMAT_SSV_COMMA)) // ssvcomma
606         return DATASOURCE_SSV_COMMA;
607
608     else if(!strcmp(name, DATASOURCE_FORMAT_CSV_JSON_ARRAY)) // csvjsonarray
609         return DATASOURCE_CSV_JSON_ARRAY;
610
611     return DATASOURCE_JSON;
612 }
613
614 uint32_t web_client_api_request_v1_data_google_format(char *name)
615 {
616     if(!strcmp(name, "json"))
617         return DATASOURCE_DATATABLE_JSONP;
618
619     else if(!strcmp(name, "html"))
620         return DATASOURCE_HTML;
621
622     else if(!strcmp(name, "csv"))
623         return DATASOURCE_CSV;
624
625     else if(!strcmp(name, "tsv-excel"))
626         return DATASOURCE_TSV;
627
628     return DATASOURCE_JSON;
629 }
630
631 const char *group_method2string(int group) {
632     switch(group) {
633         case GROUP_UNDEFINED:
634             return "";
635
636         case GROUP_AVERAGE:
637             return "average";
638
639         case GROUP_MIN:
640             return "min";
641
642         case GROUP_MAX:
643             return "max";
644
645         case GROUP_SUM:
646             return "sum";
647
648         case GROUP_INCREMENTAL_SUM:
649             return "incremental-sum";
650
651         default:
652             return "unknown-group-method";
653     }
654 }
655
656 int web_client_api_request_v1_data_group(char *name, int def)
657 {
658     if(!strcmp(name, "average"))
659         return GROUP_AVERAGE;
660
661     else if(!strcmp(name, "min"))
662         return GROUP_MIN;
663
664     else if(!strcmp(name, "max"))
665         return GROUP_MAX;
666
667     else if(!strcmp(name, "sum"))
668         return GROUP_SUM;
669
670     else if(!strcmp(name, "incremental-sum"))
671         return GROUP_INCREMENTAL_SUM;
672
673     return def;
674 }
675
676 int web_client_api_request_v1_alarms(struct web_client *w, char *url)
677 {
678     int all = 0;
679
680     while(url) {
681         char *value = mystrsep(&url, "?&");
682         if (!value || !*value) continue;
683
684         if(!strcmp(value, "all")) all = 1;
685         else if(!strcmp(value, "active")) all = 0;
686     }
687
688     buffer_flush(w->response.data);
689     w->response.data->contenttype = CT_APPLICATION_JSON;
690     health_alarms2json(&localhost, w->response.data, all);
691     return 200;
692 }
693
694 int web_client_api_request_v1_alarm_log(struct web_client *w, char *url)
695 {
696     uint32_t after = 0;
697
698     while(url) {
699         char *value = mystrsep(&url, "?&");
700         if (!value || !*value) continue;
701
702         char *name = mystrsep(&value, "=");
703         if(!name || !*name) continue;
704         if(!value || !*value) continue;
705
706         if(!strcmp(name, "after")) after = strtoul(value, NULL, 0);
707     }
708
709     buffer_flush(w->response.data);
710     w->response.data->contenttype = CT_APPLICATION_JSON;
711     health_alarm_log2json(&localhost, w->response.data, after);
712     return 200;
713 }
714
715 int web_client_api_request_single_chart(struct web_client *w, char *url, void callback(RRDSET *st, BUFFER *buf))
716 {
717     int ret = 400;
718     char *chart = NULL;
719
720     buffer_flush(w->response.data);
721
722     while(url) {
723         char *value = mystrsep(&url, "?&");
724         if(!value || !*value) continue;
725
726         char *name = mystrsep(&value, "=");
727         if(!name || !*name) continue;
728         if(!value || !*value) continue;
729
730         // name and value are now the parameters
731         // they are not null and not empty
732
733         if(!strcmp(name, "chart")) chart = value;
734         //else {
735         /// buffer_sprintf(w->response.data, "Unknown parameter '%s' in request.", name);
736         //  goto cleanup;
737         //}
738     }
739
740     if(!chart || !*chart) {
741         buffer_sprintf(w->response.data, "No chart id is given at the request.");
742         goto cleanup;
743     }
744
745     RRDSET *st = rrdset_find(chart);
746     if(!st) st = rrdset_find_byname(chart);
747     if(!st) {
748         buffer_strcat(w->response.data, "Chart is not found: ");
749         buffer_strcat_htmlescape(w->response.data, chart);
750         ret = 404;
751         goto cleanup;
752     }
753
754     w->response.data->contenttype = CT_APPLICATION_JSON;
755     callback(st, w->response.data);
756     return 200;
757
758     cleanup:
759     return ret;
760 }
761
762 int web_client_api_request_v1_alarm_variables(struct web_client *w, char *url)
763 {
764     return web_client_api_request_single_chart(w, url, health_api_v1_chart_variables2json);
765 }
766
767 int web_client_api_request_v1_charts(struct web_client *w, char *url)
768 {
769     (void)url;
770
771     buffer_flush(w->response.data);
772     w->response.data->contenttype = CT_APPLICATION_JSON;
773     rrd_stats_api_v1_charts(w->response.data);
774     return 200;
775 }
776
777 int web_client_api_request_v1_allmetrics(struct web_client *w, char *url)
778 {
779     int format = ALLMETRICS_SHELL;
780
781     while(url) {
782         char *value = mystrsep(&url, "?&");
783         if (!value || !*value) continue;
784
785         char *name = mystrsep(&value, "=");
786         if(!name || !*name) continue;
787         if(!value || !*value) continue;
788
789         if(!strcmp(name, "format")) {
790             if(!strcmp(value, ALLMETRICS_FORMAT_SHELL))
791                 format = ALLMETRICS_SHELL;
792             else if(!strcmp(value, ALLMETRICS_FORMAT_PROMETHEUS))
793                 format = ALLMETRICS_PROMETHEUS;
794             else
795                 format = 0;
796         }
797     }
798
799     buffer_flush(w->response.data);
800     buffer_no_cacheable(w->response.data);
801
802     switch(format) {
803         case ALLMETRICS_SHELL:
804             w->response.data->contenttype = CT_TEXT_PLAIN;
805             rrd_stats_api_v1_charts_allmetrics_shell(w->response.data);
806             return 200;
807
808         case ALLMETRICS_PROMETHEUS:
809             w->response.data->contenttype = CT_PROMETHEUS;
810             rrd_stats_api_v1_charts_allmetrics_prometheus(w->response.data);
811             return 200;
812
813         default:
814             w->response.data->contenttype = CT_TEXT_PLAIN;
815             buffer_strcat(w->response.data, "Which format? Only '" ALLMETRICS_FORMAT_SHELL "' and '" ALLMETRICS_FORMAT_PROMETHEUS "' is currently supported.");
816             return 400;
817     }
818 }
819
820 int web_client_api_request_v1_chart(struct web_client *w, char *url)
821 {
822     return web_client_api_request_single_chart(w, url, rrd_stats_api_v1_chart);
823 }
824
825 int web_client_api_request_v1_badge(struct web_client *w, char *url) {
826     int ret = 400;
827     buffer_flush(w->response.data);
828
829     BUFFER *dimensions = NULL;
830     
831     const char *chart = NULL
832             , *before_str = NULL
833             , *after_str = NULL
834             , *points_str = NULL
835             , *multiply_str = NULL
836             , *divide_str = NULL
837             , *label = NULL
838             , *units = NULL
839             , *label_color = NULL
840             , *value_color = NULL
841             , *refresh_str = NULL
842             , *precision_str = NULL
843             , *alarm = NULL;
844
845     int group = GROUP_AVERAGE;
846     uint32_t options = 0x00000000;
847
848     while(url) {
849         char *value = mystrsep(&url, "/?&");
850         if(!value || !*value) continue;
851
852         char *name = mystrsep(&value, "=");
853         if(!name || !*name) continue;
854         if(!value || !*value) continue;
855
856         debug(D_WEB_CLIENT, "%llu: API v1 badge.svg query param '%s' with value '%s'", w->id, name, value);
857
858         // name and value are now the parameters
859         // they are not null and not empty
860
861         if(!strcmp(name, "chart")) chart = value;
862         else if(!strcmp(name, "dimension") || !strcmp(name, "dim") || !strcmp(name, "dimensions") || !strcmp(name, "dims")) {
863             if(!dimensions)
864                 dimensions = buffer_create(100);
865
866             buffer_strcat(dimensions, "|");
867             buffer_strcat(dimensions, value);
868         }
869         else if(!strcmp(name, "after")) after_str = value;
870         else if(!strcmp(name, "before")) before_str = value;
871         else if(!strcmp(name, "points")) points_str = value;
872         else if(!strcmp(name, "group")) {
873             group = web_client_api_request_v1_data_group(value, GROUP_AVERAGE);
874         }
875         else if(!strcmp(name, "options")) {
876             options |= web_client_api_request_v1_data_options(value);
877         }
878         else if(!strcmp(name, "label")) label = value;
879         else if(!strcmp(name, "units")) units = value;
880         else if(!strcmp(name, "label_color")) label_color = value;
881         else if(!strcmp(name, "value_color")) value_color = value;
882         else if(!strcmp(name, "multiply")) multiply_str = value;
883         else if(!strcmp(name, "divide")) divide_str = value;
884         else if(!strcmp(name, "refresh")) refresh_str = value;
885         else if(!strcmp(name, "precision")) precision_str = value;
886         else if(!strcmp(name, "alarm")) alarm = value;
887     }
888
889     if(!chart || !*chart) {
890         buffer_no_cacheable(w->response.data);
891         buffer_sprintf(w->response.data, "No chart id is given at the request.");
892         goto cleanup;
893     }
894
895     RRDSET *st = rrdset_find(chart);
896     if(!st) st = rrdset_find_byname(chart);
897     if(!st) {
898         buffer_no_cacheable(w->response.data);
899         buffer_svg(w->response.data, "chart not found", NAN, "", NULL, NULL, -1);
900         ret = 200;
901         goto cleanup;
902     }
903
904     RRDCALC *rc = NULL;
905     if(alarm) {
906         rc = rrdcalc_find(st, alarm);
907         if (!rc) {
908             buffer_no_cacheable(w->response.data);
909             buffer_svg(w->response.data, "alarm not found", NAN, "", NULL, NULL, -1);
910             ret = 200;
911             goto cleanup;
912         }
913     }
914
915     long long multiply  = (multiply_str  && *multiply_str )?str2l(multiply_str):1;
916     long long divide    = (divide_str    && *divide_str   )?str2l(divide_str):1;
917     long long before    = (before_str    && *before_str   )?str2l(before_str):0;
918     long long after     = (after_str     && *after_str    )?str2l(after_str):-st->update_every;
919     int       points    = (points_str    && *points_str   )?str2i(points_str):1;
920     int       precision = (precision_str && *precision_str)?str2i(precision_str):-1;
921
922     if(!multiply) multiply = 1;
923     if(!divide) divide = 1;
924
925     int refresh = 0;
926     if(refresh_str && *refresh_str) {
927         if(!strcmp(refresh_str, "auto")) {
928             if(rc) refresh = rc->update_every;
929             else if(options & RRDR_OPTION_NOT_ALIGNED)
930                 refresh = st->update_every;
931             else {
932                 refresh = (int)(before - after);
933                 if(refresh < 0) refresh = -refresh;
934             }
935         }
936         else {
937             refresh = str2i(refresh_str);
938             if(refresh < 0) refresh = -refresh;
939         }
940     }
941
942     if(!label) {
943         if(alarm) {
944             char *s = (char *)alarm;
945             while(*s) {
946                 if(*s == '_') *s = ' ';
947                 s++;
948             }
949             label = alarm;
950         }
951         else if(dimensions) {
952             const char *dim = buffer_tostring(dimensions);
953             if(*dim == '|') dim++;
954             label = dim;
955         }
956         else
957             label = st->name;
958     }
959     if(!units) {
960         if(alarm) {
961             if(rc->units)
962                 units = rc->units;
963             else
964                 units = "";
965         }
966         else if(options & RRDR_OPTION_PERCENTAGE)
967             units = "%";
968         else
969             units = st->units;
970     }
971
972     debug(D_WEB_CLIENT, "%llu: API command 'badge.svg' for chart '%s', alarm '%s', dimensions '%s', after '%lld', before '%lld', points '%d', group '%d', options '0x%08x'"
973             , w->id
974             , chart
975             , alarm?alarm:""
976             , (dimensions)?buffer_tostring(dimensions):""
977             , after
978             , before
979             , points
980             , group
981             , options
982             );
983
984     if(rc) {
985         if (refresh > 0) {
986             buffer_sprintf(w->response.header, "Refresh: %d\r\n", refresh);
987             w->response.data->expires = now_realtime_sec() + refresh;
988         }
989         else buffer_no_cacheable(w->response.data);
990
991         if(!value_color) {
992             switch(rc->status) {
993                 case RRDCALC_STATUS_CRITICAL:
994                     value_color = "red";
995                     break;
996
997                 case RRDCALC_STATUS_WARNING:
998                     value_color = "orange";
999                     break;
1000
1001                 case RRDCALC_STATUS_CLEAR:
1002                     value_color = "brightgreen";
1003                     break;
1004
1005                 case RRDCALC_STATUS_UNDEFINED:
1006                     value_color = "lightgrey";
1007                     break;
1008
1009                 case RRDCALC_STATUS_UNINITIALIZED:
1010                     value_color = "#000";
1011                     break;
1012
1013                 default:
1014                     value_color = "grey";
1015                     break;
1016             }
1017         }
1018
1019         buffer_svg(w->response.data,
1020                 label,
1021                 (isnan(rc->value)||isinf(rc->value)) ? rc->value : rc->value * multiply / divide,
1022                 units,
1023                 label_color,
1024                 value_color,
1025                 precision);
1026         ret = 200;
1027     }
1028     else {
1029         time_t latest_timestamp = 0;
1030         int value_is_null = 1;
1031         calculated_number n = NAN;
1032         ret = 500;
1033
1034         // if the collected value is too old, don't calculate its value
1035         if (rrdset_last_entry_t(st) >= (now_realtime_sec() - (st->update_every * st->gap_when_lost_iterations_above)))
1036             ret = rrd2value(st,
1037                             w->response.data,
1038                             &n,
1039                             (dimensions) ? buffer_tostring(dimensions) : NULL,
1040                             points,
1041                             after,
1042                             before,
1043                             group,
1044                             options,
1045                             NULL,
1046                             &latest_timestamp,
1047                             &value_is_null);
1048
1049         // if the value cannot be calculated, show empty badge
1050         if (ret != 200) {
1051             buffer_no_cacheable(w->response.data);
1052             value_is_null = 1;
1053             n = 0;
1054             ret = 200;
1055         }
1056         else if (refresh > 0) {
1057             buffer_sprintf(w->response.header, "Refresh: %d\r\n", refresh);
1058             w->response.data->expires = now_realtime_sec() + refresh;
1059         }
1060         else buffer_no_cacheable(w->response.data);
1061
1062         // render the badge
1063         buffer_svg(w->response.data,
1064                 label,
1065                 (value_is_null)?NAN:(n * multiply / divide),
1066                 units,
1067                 label_color,
1068                 value_color,
1069                 precision);
1070     }
1071
1072 cleanup:
1073     if(dimensions)
1074         buffer_free(dimensions);
1075     return ret;
1076 }
1077
1078 // returns the HTTP code
1079 int web_client_api_request_v1_data(struct web_client *w, char *url)
1080 {
1081     debug(D_WEB_CLIENT, "%llu: API v1 data with URL '%s'", w->id, url);
1082
1083     int ret = 400;
1084     BUFFER *dimensions = NULL;
1085
1086     buffer_flush(w->response.data);
1087
1088     char    *google_version = "0.6",
1089             *google_reqId = "0",
1090             *google_sig = "0",
1091             *google_out = "json",
1092             *responseHandler = NULL,
1093             *outFileName = NULL;
1094
1095     time_t last_timestamp_in_data = 0, google_timestamp = 0;
1096
1097     char *chart = NULL
1098             , *before_str = NULL
1099             , *after_str = NULL
1100             , *points_str = NULL;
1101
1102     int group = GROUP_AVERAGE;
1103     uint32_t format = DATASOURCE_JSON;
1104     uint32_t options = 0x00000000;
1105
1106     while(url) {
1107         char *value = mystrsep(&url, "?&");
1108         if(!value || !*value) continue;
1109
1110         char *name = mystrsep(&value, "=");
1111         if(!name || !*name) continue;
1112         if(!value || !*value) continue;
1113
1114         debug(D_WEB_CLIENT, "%llu: API v1 data query param '%s' with value '%s'", w->id, name, value);
1115
1116         // name and value are now the parameters
1117         // they are not null and not empty
1118
1119         if(!strcmp(name, "chart")) chart = value;
1120         else if(!strcmp(name, "dimension") || !strcmp(name, "dim") || !strcmp(name, "dimensions") || !strcmp(name, "dims")) {
1121             if(!dimensions) dimensions = buffer_create(100);
1122             buffer_strcat(dimensions, "|");
1123             buffer_strcat(dimensions, value);
1124         }
1125         else if(!strcmp(name, "after")) after_str = value;
1126         else if(!strcmp(name, "before")) before_str = value;
1127         else if(!strcmp(name, "points")) points_str = value;
1128         else if(!strcmp(name, "group")) {
1129             group = web_client_api_request_v1_data_group(value, GROUP_AVERAGE);
1130         }
1131         else if(!strcmp(name, "format")) {
1132             format = web_client_api_request_v1_data_format(value);
1133         }
1134         else if(!strcmp(name, "options")) {
1135             options |= web_client_api_request_v1_data_options(value);
1136         }
1137         else if(!strcmp(name, "callback")) {
1138             responseHandler = value;
1139         }
1140         else if(!strcmp(name, "filename")) {
1141             outFileName = value;
1142         }
1143         else if(!strcmp(name, "tqx")) {
1144             // parse Google Visualization API options
1145             // https://developers.google.com/chart/interactive/docs/dev/implementing_data_source
1146             char *tqx_name, *tqx_value;
1147
1148             while(value) {
1149                 tqx_value = mystrsep(&value, ";");
1150                 if(!tqx_value || !*tqx_value) continue;
1151
1152                 tqx_name = mystrsep(&tqx_value, ":");
1153                 if(!tqx_name || !*tqx_name) continue;
1154                 if(!tqx_value || !*tqx_value) continue;
1155
1156                 if(!strcmp(tqx_name, "version"))
1157                     google_version = tqx_value;
1158                 else if(!strcmp(tqx_name, "reqId"))
1159                     google_reqId = tqx_value;
1160                 else if(!strcmp(tqx_name, "sig")) {
1161                     google_sig = tqx_value;
1162                     google_timestamp = strtoul(google_sig, NULL, 0);
1163                 }
1164                 else if(!strcmp(tqx_name, "out")) {
1165                     google_out = tqx_value;
1166                     format = web_client_api_request_v1_data_google_format(google_out);
1167                 }
1168                 else if(!strcmp(tqx_name, "responseHandler"))
1169                     responseHandler = tqx_value;
1170                 else if(!strcmp(tqx_name, "outFileName"))
1171                     outFileName = tqx_value;
1172             }
1173         }
1174     }
1175
1176     if(!chart || !*chart) {
1177         buffer_sprintf(w->response.data, "No chart id is given at the request.");
1178         goto cleanup;
1179     }
1180
1181     RRDSET *st = rrdset_find(chart);
1182     if(!st) st = rrdset_find_byname(chart);
1183     if(!st) {
1184         buffer_strcat(w->response.data, "Chart is not found: ");
1185         buffer_strcat_htmlescape(w->response.data, chart);
1186         ret = 404;
1187         goto cleanup;
1188     }
1189
1190     long long before = (before_str && *before_str)?str2l(before_str):0;
1191     long long after  = (after_str  && *after_str) ?str2l(after_str):0;
1192     int       points = (points_str && *points_str)?str2i(points_str):0;
1193
1194     debug(D_WEB_CLIENT, "%llu: API command 'data' for chart '%s', dimensions '%s', after '%lld', before '%lld', points '%d', group '%d', format '%u', options '0x%08x'"
1195             , w->id
1196             , chart
1197             , (dimensions)?buffer_tostring(dimensions):""
1198             , after
1199             , before
1200             , points
1201             , group
1202             , format
1203             , options
1204             );
1205
1206     if(outFileName && *outFileName) {
1207         buffer_sprintf(w->response.header, "Content-Disposition: attachment; filename=\"%s\"\r\n", outFileName);
1208         debug(D_WEB_CLIENT, "%llu: generating outfilename header: '%s'", w->id, outFileName);
1209     }
1210
1211     if(format == DATASOURCE_DATATABLE_JSONP) {
1212         if(responseHandler == NULL)
1213             responseHandler = "google.visualization.Query.setResponse";
1214
1215         debug(D_WEB_CLIENT_ACCESS, "%llu: GOOGLE JSON/JSONP: version = '%s', reqId = '%s', sig = '%s', out = '%s', responseHandler = '%s', outFileName = '%s'",
1216                 w->id, google_version, google_reqId, google_sig, google_out, responseHandler, outFileName
1217             );
1218
1219         buffer_sprintf(w->response.data,
1220             "%s({version:'%s',reqId:'%s',status:'ok',sig:'%ld',table:",
1221             responseHandler, google_version, google_reqId, st->last_updated.tv_sec);
1222     }
1223     else if(format == DATASOURCE_JSONP) {
1224         if(responseHandler == NULL)
1225             responseHandler = "callback";
1226
1227         buffer_strcat(w->response.data, responseHandler);
1228         buffer_strcat(w->response.data, "(");
1229     }
1230
1231     ret = rrd2format(st, w->response.data, dimensions, format, points, after, before, group, options, &last_timestamp_in_data);
1232
1233     if(format == DATASOURCE_DATATABLE_JSONP) {
1234         if(google_timestamp < last_timestamp_in_data)
1235             buffer_strcat(w->response.data, "});");
1236
1237         else {
1238             // the client already has the latest data
1239             buffer_flush(w->response.data);
1240             buffer_sprintf(w->response.data,
1241                 "%s({version:'%s',reqId:'%s',status:'error',errors:[{reason:'not_modified',message:'Data not modified'}]});",
1242                 responseHandler, google_version, google_reqId);
1243         }
1244     }
1245     else if(format == DATASOURCE_JSONP)
1246         buffer_strcat(w->response.data, ");");
1247
1248 cleanup:
1249     if(dimensions) buffer_free(dimensions);
1250     return ret;
1251 }
1252
1253
1254 int web_client_api_request_v1_registry(struct web_client *w, char *url)
1255 {
1256     static uint32_t hash_action = 0, hash_access = 0, hash_hello = 0, hash_delete = 0, hash_search = 0,
1257             hash_switch = 0, hash_machine = 0, hash_url = 0, hash_name = 0, hash_delete_url = 0, hash_for = 0,
1258             hash_to = 0 /*, hash_redirects = 0 */;
1259
1260     if(unlikely(!hash_action)) {
1261         hash_action = simple_hash("action");
1262         hash_access = simple_hash("access");
1263         hash_hello = simple_hash("hello");
1264         hash_delete = simple_hash("delete");
1265         hash_search = simple_hash("search");
1266         hash_switch = simple_hash("switch");
1267         hash_machine = simple_hash("machine");
1268         hash_url = simple_hash("url");
1269         hash_name = simple_hash("name");
1270         hash_delete_url = simple_hash("delete_url");
1271         hash_for = simple_hash("for");
1272         hash_to = simple_hash("to");
1273 /*
1274         hash_redirects = simple_hash("redirects");
1275 */
1276     }
1277
1278     char person_guid[GUID_LEN + 1] = "";
1279
1280     debug(D_WEB_CLIENT, "%llu: API v1 registry with URL '%s'", w->id, url);
1281
1282     // FIXME
1283     // The browser may send multiple cookies with our id
1284     
1285     char *cookie = strstr(w->response.data->buffer, NETDATA_REGISTRY_COOKIE_NAME "=");
1286     if(cookie)
1287         strncpyz(person_guid, &cookie[sizeof(NETDATA_REGISTRY_COOKIE_NAME)], 36);
1288
1289     char action = '\0';
1290     char *machine_guid = NULL,
1291             *machine_url = NULL,
1292             *url_name = NULL,
1293             *search_machine_guid = NULL,
1294             *delete_url = NULL,
1295             *to_person_guid = NULL;
1296 /*
1297     int redirects = 0;
1298 */
1299
1300     while(url) {
1301         char *value = mystrsep(&url, "?&");
1302         if (!value || !*value) continue;
1303
1304         char *name = mystrsep(&value, "=");
1305         if (!name || !*name) continue;
1306         if (!value || !*value) continue;
1307
1308         debug(D_WEB_CLIENT, "%llu: API v1 registry query param '%s' with value '%s'", w->id, name, value);
1309
1310         uint32_t hash = simple_hash(name);
1311
1312         if(hash == hash_action && !strcmp(name, "action")) {
1313             uint32_t vhash = simple_hash(value);
1314
1315             if(vhash == hash_access && !strcmp(value, "access")) action = 'A';
1316             else if(vhash == hash_hello && !strcmp(value, "hello")) action = 'H';
1317             else if(vhash == hash_delete && !strcmp(value, "delete")) action = 'D';
1318             else if(vhash == hash_search && !strcmp(value, "search")) action = 'S';
1319             else if(vhash == hash_switch && !strcmp(value, "switch")) action = 'W';
1320 #ifdef NETDATA_INTERNAL_CHECKS
1321             else error("unknown registry action '%s'", value);
1322 #endif /* NETDATA_INTERNAL_CHECKS */
1323         }
1324 /*
1325         else if(hash == hash_redirects && !strcmp(name, "redirects"))
1326             redirects = atoi(value);
1327 */
1328         else if(hash == hash_machine && !strcmp(name, "machine"))
1329             machine_guid = value;
1330
1331         else if(hash == hash_url && !strcmp(name, "url"))
1332             machine_url = value;
1333
1334         else if(action == 'A') {
1335             if(hash == hash_name && !strcmp(name, "name"))
1336                 url_name = value;
1337         }
1338         else if(action == 'D') {
1339             if(hash == hash_delete_url && !strcmp(name, "delete_url"))
1340                 delete_url = value;
1341         }
1342         else if(action == 'S') {
1343             if(hash == hash_for && !strcmp(name, "for"))
1344                 search_machine_guid = value;
1345         }
1346         else if(action == 'W') {
1347             if(hash == hash_to && !strcmp(name, "to"))
1348                 to_person_guid = value;
1349         }
1350 #ifdef NETDATA_INTERNAL_CHECKS
1351         else error("unused registry URL parameter '%s' with value '%s'", name, value);
1352 #endif /* NETDATA_INTERNAL_CHECKS */
1353     }
1354
1355     if(web_donotrack_comply && w->donottrack) {
1356         buffer_flush(w->response.data);
1357         buffer_sprintf(w->response.data, "Your web browser is sending 'DNT: 1' (Do Not Track). The registry requires persistent cookies on your browser to work.");
1358         return 400;
1359     }
1360
1361     if(action == 'A' && (!machine_guid || !machine_url || !url_name)) {
1362         error("Invalid registry request - access requires these parameters: machine ('%s'), url ('%s'), name ('%s')",
1363                 machine_guid?machine_guid:"UNSET", machine_url?machine_url:"UNSET", url_name?url_name:"UNSET");
1364         buffer_flush(w->response.data);
1365         buffer_strcat(w->response.data, "Invalid registry Access request.");
1366         return 400;
1367     }
1368     else if(action == 'D' && (!machine_guid || !machine_url || !delete_url)) {
1369         error("Invalid registry request - delete requires these parameters: machine ('%s'), url ('%s'), delete_url ('%s')",
1370                 machine_guid?machine_guid:"UNSET", machine_url?machine_url:"UNSET", delete_url?delete_url:"UNSET");
1371         buffer_flush(w->response.data);
1372         buffer_strcat(w->response.data, "Invalid registry Delete request.");
1373         return 400;
1374     }
1375     else if(action == 'S' && (!machine_guid || !machine_url || !search_machine_guid)) {
1376         error("Invalid registry request - search requires these parameters: machine ('%s'), url ('%s'), for ('%s')",
1377                 machine_guid?machine_guid:"UNSET", machine_url?machine_url:"UNSET", search_machine_guid?search_machine_guid:"UNSET");
1378         buffer_flush(w->response.data);
1379         buffer_strcat(w->response.data, "Invalid registry Search request.");
1380         return 400;
1381     }
1382     else if(action == 'W' && (!machine_guid || !machine_url || !to_person_guid)) {
1383         error("Invalid registry request - switching identity requires these parameters: machine ('%s'), url ('%s'), to ('%s')",
1384                 machine_guid?machine_guid:"UNSET", machine_url?machine_url:"UNSET", to_person_guid?to_person_guid:"UNSET");
1385         buffer_flush(w->response.data);
1386         buffer_strcat(w->response.data, "Invalid registry Switch request.");
1387         return 400;
1388     }
1389
1390     switch(action) {
1391         case 'A':
1392             w->tracking_required = 1;
1393             return registry_request_access_json(w, person_guid, machine_guid, machine_url, url_name, now_realtime_sec());
1394
1395         case 'D':
1396             w->tracking_required = 1;
1397             return registry_request_delete_json(w, person_guid, machine_guid, machine_url, delete_url, now_realtime_sec());
1398
1399         case 'S':
1400             w->tracking_required = 1;
1401             return registry_request_search_json(w, person_guid, machine_guid, machine_url, search_machine_guid, now_realtime_sec());
1402
1403         case 'W':
1404             w->tracking_required = 1;
1405             return registry_request_switch_json(w, person_guid, machine_guid, machine_url, to_person_guid, now_realtime_sec());
1406
1407         case 'H':
1408             return registry_request_hello_json(w);
1409
1410         default:
1411             buffer_flush(w->response.data);
1412             buffer_strcat(w->response.data, "Invalid registry request - you need to set an action: hello, access, delete, search");
1413             return 400;
1414     }
1415 }
1416
1417 int web_client_api_request_v1(struct web_client *w, char *url) {
1418     static uint32_t hash_data = 0, hash_chart = 0, hash_charts = 0, hash_registry = 0, hash_badge = 0, hash_alarms = 0, hash_alarm_log = 0, hash_alarm_variables = 0, hash_raw = 0;
1419
1420     if(unlikely(hash_data == 0)) {
1421         hash_data = simple_hash("data");
1422         hash_chart = simple_hash("chart");
1423         hash_charts = simple_hash("charts");
1424         hash_registry = simple_hash("registry");
1425         hash_badge = simple_hash("badge.svg");
1426         hash_alarms = simple_hash("alarms");
1427         hash_alarm_log = simple_hash("alarm_log");
1428         hash_alarm_variables = simple_hash("alarm_variables");
1429         hash_raw = simple_hash("allmetrics");
1430     }
1431
1432     // get the command
1433     char *tok = mystrsep(&url, "/?&");
1434     if(tok && *tok) {
1435         debug(D_WEB_CLIENT, "%llu: Searching for API v1 command '%s'.", w->id, tok);
1436         uint32_t hash = simple_hash(tok);
1437
1438         if(hash == hash_data && !strcmp(tok, "data"))
1439             return web_client_api_request_v1_data(w, url);
1440
1441         else if(hash == hash_chart && !strcmp(tok, "chart"))
1442             return web_client_api_request_v1_chart(w, url);
1443
1444         else if(hash == hash_charts && !strcmp(tok, "charts"))
1445             return web_client_api_request_v1_charts(w, url);
1446
1447         else if(hash == hash_registry && !strcmp(tok, "registry"))
1448             return web_client_api_request_v1_registry(w, url);
1449
1450         else if(hash == hash_badge && !strcmp(tok, "badge.svg"))
1451             return web_client_api_request_v1_badge(w, url);
1452
1453         else if(hash == hash_alarms && !strcmp(tok, "alarms"))
1454             return web_client_api_request_v1_alarms(w, url);
1455
1456         else if(hash == hash_alarm_log && !strcmp(tok, "alarm_log"))
1457             return web_client_api_request_v1_alarm_log(w, url);
1458
1459         else if(hash == hash_alarm_variables && !strcmp(tok, "alarm_variables"))
1460             return web_client_api_request_v1_alarm_variables(w, url);
1461
1462         else if(hash == hash_raw && !strcmp(tok, "allmetrics"))
1463             return web_client_api_request_v1_allmetrics(w, url);
1464
1465         else {
1466             buffer_flush(w->response.data);
1467             buffer_strcat(w->response.data, "Unsupported v1 API command: ");
1468             buffer_strcat_htmlescape(w->response.data, tok);
1469             return 404;
1470         }
1471     }
1472     else {
1473         buffer_flush(w->response.data);
1474         buffer_sprintf(w->response.data, "Which API v1 command?");
1475         return 400;
1476     }
1477 }
1478
1479 int web_client_api_request(struct web_client *w, char *url)
1480 {
1481     // get the api version
1482     char *tok = mystrsep(&url, "/?&");
1483     if(tok && *tok) {
1484         debug(D_WEB_CLIENT, "%llu: Searching for API version '%s'.", w->id, tok);
1485         if(strcmp(tok, "v1") == 0)
1486             return web_client_api_request_v1(w, url);
1487         else {
1488             buffer_flush(w->response.data);
1489             buffer_strcat(w->response.data, "Unsupported API version: ");
1490             buffer_strcat_htmlescape(w->response.data, tok);
1491             return 404;
1492         }
1493     }
1494     else {
1495         buffer_flush(w->response.data);
1496         buffer_sprintf(w->response.data, "Which API version?");
1497         return 400;
1498     }
1499 }
1500
1501 int web_client_api_old_data_request(struct web_client *w, char *url, int datasource_type)
1502 {
1503     if(!url || !*url) {
1504         buffer_flush(w->response.data);
1505         buffer_sprintf(w->response.data, "Incomplete request.");
1506         return 400;
1507     }
1508
1509     RRDSET *st = NULL;
1510
1511     char *args = strchr(url, '?');
1512     if(args) {
1513         *args='\0';
1514         args = &args[1];
1515     }
1516
1517     // get the name of the data to show
1518     char *tok = mystrsep(&url, "/");
1519     if(!tok) tok = "";
1520
1521     // do we have such a data set?
1522     if(*tok) {
1523         debug(D_WEB_CLIENT, "%llu: Searching for RRD data with name '%s'.", w->id, tok);
1524         st = rrdset_find_byname(tok);
1525         if(!st) st = rrdset_find(tok);
1526     }
1527
1528     if(!st) {
1529         // we don't have it
1530         // try to send a file with that name
1531         buffer_flush(w->response.data);
1532         return(mysendfile(w, tok));
1533     }
1534
1535     // we have it
1536     debug(D_WEB_CLIENT, "%llu: Found RRD data with name '%s'.", w->id, tok);
1537
1538     // how many entries does the client want?
1539     int lines = rrd_default_history_entries;
1540     int group_count = 1;
1541     time_t after = 0, before = 0;
1542     int group_method = GROUP_AVERAGE;
1543     int nonzero = 0;
1544
1545     if(url) {
1546         // parse the lines required
1547         tok = mystrsep(&url, "/");
1548         if(tok) lines = str2i(tok);
1549         if(lines < 1) lines = 1;
1550     }
1551     if(url) {
1552         // parse the group count required
1553         tok = mystrsep(&url, "/");
1554         if(tok && *tok) group_count = str2i(tok);
1555         if(group_count < 1) group_count = 1;
1556         //if(group_count > save_history / 20) group_count = save_history / 20;
1557     }
1558     if(url) {
1559         // parse the grouping method required
1560         tok = mystrsep(&url, "/");
1561         if(tok && *tok) {
1562             if(strcmp(tok, "max") == 0) group_method = GROUP_MAX;
1563             else if(strcmp(tok, "average") == 0) group_method = GROUP_AVERAGE;
1564             else if(strcmp(tok, "sum") == 0) group_method = GROUP_SUM;
1565             else debug(D_WEB_CLIENT, "%llu: Unknown group method '%s'", w->id, tok);
1566         }
1567     }
1568     if(url) {
1569         // parse after time
1570         tok = mystrsep(&url, "/");
1571         if(tok && *tok) after = str2ul(tok);
1572         if(after < 0) after = 0;
1573     }
1574     if(url) {
1575         // parse before time
1576         tok = mystrsep(&url, "/");
1577         if(tok && *tok) before = str2ul(tok);
1578         if(before < 0) before = 0;
1579     }
1580     if(url) {
1581         // parse nonzero
1582         tok = mystrsep(&url, "/");
1583         if(tok && *tok && strcmp(tok, "nonzero") == 0) nonzero = 1;
1584     }
1585
1586     w->response.data->contenttype = CT_APPLICATION_JSON;
1587     buffer_flush(w->response.data);
1588
1589     char *google_version = "0.6";
1590     char *google_reqId = "0";
1591     char *google_sig = "0";
1592     char *google_out = "json";
1593     char *google_responseHandler = "google.visualization.Query.setResponse";
1594     char *google_outFileName = NULL;
1595     time_t last_timestamp_in_data = 0;
1596     if(datasource_type == DATASOURCE_DATATABLE_JSON || datasource_type == DATASOURCE_DATATABLE_JSONP) {
1597
1598         w->response.data->contenttype = CT_APPLICATION_X_JAVASCRIPT;
1599
1600         while(args) {
1601             tok = mystrsep(&args, "&");
1602             if(tok && *tok) {
1603                 char *name = mystrsep(&tok, "=");
1604                 if(name && *name && strcmp(name, "tqx") == 0) {
1605                     char *key = mystrsep(&tok, ":");
1606                     char *value = mystrsep(&tok, ";");
1607                     if(key && value && *key && *value) {
1608                         if(strcmp(key, "version") == 0)
1609                             google_version = value;
1610
1611                         else if(strcmp(key, "reqId") == 0)
1612                             google_reqId = value;
1613
1614                         else if(strcmp(key, "sig") == 0)
1615                             google_sig = value;
1616
1617                         else if(strcmp(key, "out") == 0)
1618                             google_out = value;
1619
1620                         else if(strcmp(key, "responseHandler") == 0)
1621                             google_responseHandler = value;
1622
1623                         else if(strcmp(key, "outFileName") == 0)
1624                             google_outFileName = value;
1625                     }
1626                 }
1627             }
1628         }
1629
1630         debug(D_WEB_CLIENT_ACCESS, "%llu: GOOGLE JSONP: version = '%s', reqId = '%s', sig = '%s', out = '%s', responseHandler = '%s', outFileName = '%s'",
1631             w->id, google_version, google_reqId, google_sig, google_out, google_responseHandler, google_outFileName
1632             );
1633
1634         if(datasource_type == DATASOURCE_DATATABLE_JSONP) {
1635             last_timestamp_in_data = strtoul(google_sig, NULL, 0);
1636
1637             // check the client wants json
1638             if(strcmp(google_out, "json") != 0) {
1639                 buffer_sprintf(w->response.data,
1640                     "%s({version:'%s',reqId:'%s',status:'error',errors:[{reason:'invalid_query',message:'output format is not supported',detailed_message:'the format %s requested is not supported by netdata.'}]});",
1641                     google_responseHandler, google_version, google_reqId, google_out);
1642                     return 200;
1643             }
1644         }
1645     }
1646
1647     if(datasource_type == DATASOURCE_DATATABLE_JSONP) {
1648         buffer_sprintf(w->response.data,
1649             "%s({version:'%s',reqId:'%s',status:'ok',sig:'%ld',table:",
1650             google_responseHandler, google_version, google_reqId, st->last_updated.tv_sec);
1651     }
1652
1653     debug(D_WEB_CLIENT_ACCESS, "%llu: Sending RRD data '%s' (id %s, %d lines, %d group, %d group_method, %ld after, %ld before).",
1654         w->id, st->name, st->id, lines, group_count, group_method, after, before);
1655
1656     time_t timestamp_in_data = rrd_stats_json(datasource_type, st, w->response.data, lines, group_count, group_method, (unsigned long)after, (unsigned long)before, nonzero);
1657
1658     if(datasource_type == DATASOURCE_DATATABLE_JSONP) {
1659         if(timestamp_in_data > last_timestamp_in_data)
1660             buffer_strcat(w->response.data, "});");
1661
1662         else {
1663             // the client already has the latest data
1664             buffer_flush(w->response.data);
1665             buffer_sprintf(w->response.data,
1666                 "%s({version:'%s',reqId:'%s',status:'error',errors:[{reason:'not_modified',message:'Data not modified'}]});",
1667                 google_responseHandler, google_version, google_reqId);
1668         }
1669     }
1670
1671     return 200;
1672 }
1673
1674 const char *web_content_type_to_string(uint8_t contenttype) {
1675     switch(contenttype) {
1676         case CT_TEXT_HTML:
1677             return "text/html; charset=utf-8";
1678
1679         case CT_APPLICATION_XML:
1680             return "application/xml; charset=utf-8";
1681
1682         case CT_APPLICATION_JSON:
1683             return "application/json; charset=utf-8";
1684
1685         case CT_APPLICATION_X_JAVASCRIPT:
1686             return "application/x-javascript; charset=utf-8";
1687
1688         case CT_TEXT_CSS:
1689             return "text/css; charset=utf-8";
1690
1691         case CT_TEXT_XML:
1692             return "text/xml; charset=utf-8";
1693
1694         case CT_TEXT_XSL:
1695             return "text/xsl; charset=utf-8";
1696
1697         case CT_APPLICATION_OCTET_STREAM:
1698             return "application/octet-stream";
1699
1700         case CT_IMAGE_SVG_XML:
1701             return "image/svg+xml";
1702
1703         case CT_APPLICATION_X_FONT_TRUETYPE:
1704             return "application/x-font-truetype";
1705
1706         case CT_APPLICATION_X_FONT_OPENTYPE:
1707             return "application/x-font-opentype";
1708
1709         case CT_APPLICATION_FONT_WOFF:
1710             return "application/font-woff";
1711
1712         case CT_APPLICATION_FONT_WOFF2:
1713             return "application/font-woff2";
1714
1715         case CT_APPLICATION_VND_MS_FONTOBJ:
1716             return "application/vnd.ms-fontobject";
1717
1718         case CT_IMAGE_PNG:
1719             return "image/png";
1720
1721         case CT_IMAGE_JPG:
1722             return "image/jpeg";
1723
1724         case CT_IMAGE_GIF:
1725             return "image/gif";
1726
1727         case CT_IMAGE_XICON:
1728             return "image/x-icon";
1729
1730         case CT_IMAGE_BMP:
1731             return "image/bmp";
1732
1733         case CT_IMAGE_ICNS:
1734             return "image/icns";
1735
1736         case CT_PROMETHEUS:
1737             return "text/plain; version=0.0.4";
1738
1739         default:
1740         case CT_TEXT_PLAIN:
1741             return "text/plain; charset=utf-8";
1742     }
1743 }
1744
1745
1746 const char *web_response_code_to_string(int code) {
1747     switch(code) {
1748         case 200:
1749             return "OK";
1750
1751         case 307:
1752             return "Temporary Redirect";
1753
1754         case 400:
1755             return "Bad Request";
1756
1757         case 403:
1758             return "Forbidden";
1759
1760         case 404:
1761             return "Not Found";
1762
1763         case 412:
1764             return "Preconditions Failed";
1765
1766         default:
1767             if(code >= 100 && code < 200)
1768                 return "Informational";
1769
1770             if(code >= 200 && code < 300)
1771                 return "Successful";
1772
1773             if(code >= 300 && code < 400)
1774                 return "Redirection";
1775
1776             if(code >= 400 && code < 500)
1777                 return "Bad Request";
1778
1779             if(code >= 500 && code < 600)
1780                 return "Server Error";
1781
1782             return "Undefined Error";
1783     }
1784 }
1785
1786 static inline char *http_header_parse(struct web_client *w, char *s) {
1787     static uint32_t hash_origin = 0, hash_connection = 0, hash_accept_encoding = 0, hash_donottrack = 0;
1788
1789     if(unlikely(!hash_origin)) {
1790         hash_origin = simple_uhash("Origin");
1791         hash_connection = simple_uhash("Connection");
1792         hash_accept_encoding = simple_uhash("Accept-Encoding");
1793         hash_donottrack = simple_uhash("DNT");
1794     }
1795
1796     char *e = s;
1797
1798     // find the :
1799     while(*e && *e != ':') e++;
1800     if(!*e) return e;
1801
1802     // get the name
1803     *e = '\0';
1804
1805     // find the value
1806     char *v = e + 1, *ve;
1807
1808     // skip leading spaces from value
1809     while(*v == ' ') v++;
1810     ve = v;
1811
1812     // find the \r
1813     while(*ve && *ve != '\r') ve++;
1814     if(!*ve || ve[1] != '\n') {
1815         *e = ':';
1816         return ve;
1817     }
1818
1819     // terminate the value
1820     *ve = '\0';
1821
1822     // fprintf(stderr, "HEADER: '%s' = '%s'\n", s, v);
1823     uint32_t hash = simple_uhash(s);
1824
1825     if(hash == hash_origin && !strcasecmp(s, "Origin"))
1826         strncpyz(w->origin, v, ORIGIN_MAX);
1827
1828     else if(hash == hash_connection && !strcasecmp(s, "Connection")) {
1829         if(strcasestr(v, "keep-alive"))
1830             w->keepalive = 1;
1831     }
1832     else if(web_donotrack_comply && hash == hash_donottrack && !strcasecmp(s, "DNT")) {
1833         if(*v == '0') w->donottrack = 0;
1834         else if(*v == '1') w->donottrack = 1;
1835     }
1836 #ifdef NETDATA_WITH_ZLIB
1837     else if(hash == hash_accept_encoding && !strcasecmp(s, "Accept-Encoding")) {
1838         if(web_enable_gzip) {
1839             if(strcasestr(v, "gzip"))
1840                 web_client_enable_deflate(w, 1);
1841             //
1842             // does not seem to work
1843             // else if(strcasestr(v, "deflate"))
1844             //  web_client_enable_deflate(w, 0);
1845         }
1846     }
1847 #endif /* NETDATA_WITH_ZLIB */
1848
1849     *e = ':';
1850     *ve = '\r';
1851     return ve;
1852 }
1853
1854 // http_request_validate()
1855 // returns:
1856 // = 0 : all good, process the request
1857 // > 0 : request is not supported
1858 // < 0 : request is incomplete - wait for more data
1859
1860 static inline int http_request_validate(struct web_client *w) {
1861     char *s = w->response.data->buffer, *encoded_url = NULL;
1862
1863     // is is a valid request?
1864     if(!strncmp(s, "GET ", 4)) {
1865         encoded_url = s = &s[4];
1866         w->mode = WEB_CLIENT_MODE_NORMAL;
1867     }
1868     else if(!strncmp(s, "OPTIONS ", 8)) {
1869         encoded_url = s = &s[8];
1870         w->mode = WEB_CLIENT_MODE_OPTIONS;
1871     }
1872     else {
1873         w->wait_receive = 0;
1874         return 1;
1875     }
1876
1877     // find the SPACE + "HTTP/"
1878     while(*s) {
1879         // find the next space
1880         while (*s && *s != ' ') s++;
1881
1882         // is it SPACE + "HTTP/" ?
1883         if(*s && !strncmp(s, " HTTP/", 6)) break;
1884         else s++;
1885     }
1886
1887     // incomplete requests
1888     if(unlikely(!*s)) {
1889         w->wait_receive = 1;
1890         return -2;
1891     }
1892
1893     // we have the end of encoded_url - remember it
1894     char *ue = s;
1895
1896     // make sure we have complete request
1897     // complete requests contain: \r\n\r\n
1898     while(*s) {
1899         // find a line feed
1900         while(*s && *s++ != '\r');
1901
1902         // did we reach the end?
1903         if(unlikely(!*s)) break;
1904
1905         // is it \r\n ?
1906         if(likely(*s++ == '\n')) {
1907
1908             // is it again \r\n ? (header end)
1909             if(unlikely(*s == '\r' && s[1] == '\n')) {
1910                 // a valid complete HTTP request found
1911
1912                 *ue = '\0';
1913                 url_decode_r(w->decoded_url, encoded_url, URL_MAX + 1);
1914                 *ue = ' ';
1915                 
1916                 // copy the URL - we are going to overwrite parts of it
1917                 // FIXME -- we should avoid it
1918                 strncpyz(w->last_url, w->decoded_url, URL_MAX);
1919
1920                 w->wait_receive = 0;
1921                 return 0;
1922             }
1923
1924             // another header line
1925             s = http_header_parse(w, s);
1926         }
1927     }
1928
1929     // incomplete request
1930     w->wait_receive = 1;
1931     return -3;
1932 }
1933
1934 void web_client_process(struct web_client *w) {
1935     static uint32_t
1936             hash_api = 0,
1937             hash_netdata_conf = 0,
1938             hash_data = 0,
1939             hash_datasource = 0,
1940             hash_graph = 0,
1941             hash_list = 0,
1942             hash_all_json = 0;
1943
1944 #ifdef NETDATA_INTERNAL_CHECKS
1945     static uint32_t hash_exit = 0, hash_debug = 0, hash_mirror = 0;
1946 #endif
1947
1948     // start timing us
1949     now_realtime_timeval(&w->tv_in);
1950
1951     if(unlikely(!hash_api)) {
1952         hash_api = simple_hash("api");
1953         hash_netdata_conf = simple_hash("netdata.conf");
1954         hash_data = simple_hash(WEB_PATH_DATA);
1955         hash_datasource = simple_hash(WEB_PATH_DATASOURCE);
1956         hash_graph = simple_hash(WEB_PATH_GRAPH);
1957         hash_list = simple_hash("list");
1958         hash_all_json = simple_hash("all.json");
1959 #ifdef NETDATA_INTERNAL_CHECKS
1960         hash_exit = simple_hash("exit");
1961         hash_debug = simple_hash("debug");
1962         hash_mirror = simple_hash("mirror");
1963 #endif
1964     }
1965
1966     int code = 500;
1967     ssize_t bytes;
1968
1969     int what_to_do = http_request_validate(w);
1970
1971     // wait for more data
1972     if(what_to_do < 0) {
1973         if(w->response.data->len > TOO_BIG_REQUEST) {
1974             strcpy(w->last_url, "too big request");
1975
1976             debug(D_WEB_CLIENT_ACCESS, "%llu: Received request is too big (%zu bytes).", w->id, w->response.data->len);
1977
1978             code = 400;
1979             buffer_flush(w->response.data);
1980             buffer_sprintf(w->response.data, "Received request is too big  (%zu bytes).\r\n", w->response.data->len);
1981         }
1982         else {
1983             // wait for more data
1984             return;
1985         }
1986     }
1987     else if(what_to_do > 0) {
1988         // strcpy(w->last_url, "not a valid request");
1989
1990         debug(D_WEB_CLIENT_ACCESS, "%llu: Cannot understand '%s'.", w->id, w->response.data->buffer);
1991
1992         code = 500;
1993         buffer_flush(w->response.data);
1994         buffer_strcat(w->response.data, "I don't understand you...\r\n");
1995     }
1996     else { // what_to_do == 0
1997         if(w->mode == WEB_CLIENT_MODE_OPTIONS) {
1998             code = 200;
1999             w->response.data->contenttype = CT_TEXT_PLAIN;
2000             buffer_flush(w->response.data);
2001             buffer_strcat(w->response.data, "OK");
2002         }
2003         else {
2004             char *url = w->decoded_url;
2005             char *tok = mystrsep(&url, "/?");
2006             if(tok && *tok) {
2007                 uint32_t hash = simple_hash(tok);
2008                 debug(D_WEB_CLIENT, "%llu: Processing command '%s'.", w->id, tok);
2009
2010                 if(hash == hash_api && strcmp(tok, "api") == 0) {
2011                     // the client is requesting api access
2012                     code = web_client_api_request(w, url);
2013                 }
2014                 else if(hash == hash_netdata_conf && strcmp(tok, "netdata.conf") == 0) {
2015                     code = 200;
2016                     debug(D_WEB_CLIENT_ACCESS, "%llu: Sending netdata.conf ...", w->id);
2017
2018                     w->response.data->contenttype = CT_TEXT_PLAIN;
2019                     buffer_flush(w->response.data);
2020                     generate_config(w->response.data, 0);
2021                 }
2022                 else if(hash == hash_data && strcmp(tok, WEB_PATH_DATA) == 0) { // "data"
2023                     // the client is requesting rrd data -- OLD API
2024                     code = web_client_api_old_data_request(w, url, DATASOURCE_JSON);
2025                 }
2026                 else if(hash == hash_datasource && strcmp(tok, WEB_PATH_DATASOURCE) == 0) { // "datasource"
2027                     // the client is requesting google datasource -- OLD API
2028                     code = web_client_api_old_data_request(w, url, DATASOURCE_DATATABLE_JSONP);
2029                 }
2030                 else if(hash == hash_graph && strcmp(tok, WEB_PATH_GRAPH) == 0) { // "graph"
2031                     // the client is requesting an rrd graph -- OLD API
2032
2033                     // get the name of the data to show
2034                     tok = mystrsep(&url, "/?&");
2035                     if(tok && *tok) {
2036                         debug(D_WEB_CLIENT, "%llu: Searching for RRD data with name '%s'.", w->id, tok);
2037
2038                         // do we have such a data set?
2039                         RRDSET *st = rrdset_find_byname(tok);
2040                         if(!st) st = rrdset_find(tok);
2041                         if(!st) {
2042                             // we don't have it
2043                             // try to send a file with that name
2044                             buffer_flush(w->response.data);
2045                             code = mysendfile(w, tok);
2046                         }
2047                         else {
2048                             code = 200;
2049                             debug(D_WEB_CLIENT_ACCESS, "%llu: Sending %s.json of RRD_STATS...", w->id, st->name);
2050                             w->response.data->contenttype = CT_APPLICATION_JSON;
2051                             buffer_flush(w->response.data);
2052                             rrd_stats_graph_json(st, url, w->response.data);
2053                         }
2054                     }
2055                     else {
2056                         code = 400;
2057                         buffer_flush(w->response.data);
2058                         buffer_strcat(w->response.data, "Graph name?\r\n");
2059                     }
2060                 }
2061                 else if(hash == hash_list && strcmp(tok, "list") == 0) {
2062                     // OLD API
2063                     code = 200;
2064
2065                     debug(D_WEB_CLIENT_ACCESS, "%llu: Sending list of RRD_STATS...", w->id);
2066
2067                     buffer_flush(w->response.data);
2068                     RRDSET *st = localhost.rrdset_root;
2069
2070                     for ( ; st ; st = st->next )
2071                         buffer_sprintf(w->response.data, "%s\n", st->name);
2072                 }
2073                 else if(hash == hash_all_json && strcmp(tok, "all.json") == 0) {
2074                     // OLD API
2075                     code = 200;
2076                     debug(D_WEB_CLIENT_ACCESS, "%llu: Sending JSON list of all monitors of RRD_STATS...", w->id);
2077
2078                     w->response.data->contenttype = CT_APPLICATION_JSON;
2079                     buffer_flush(w->response.data);
2080                     rrd_stats_all_json(w->response.data);
2081                 }
2082 #ifdef NETDATA_INTERNAL_CHECKS
2083                 else if(hash == hash_exit && strcmp(tok, "exit") == 0) {
2084                     code = 200;
2085                     w->response.data->contenttype = CT_TEXT_PLAIN;
2086                     buffer_flush(w->response.data);
2087
2088                     if(!netdata_exit)
2089                         buffer_strcat(w->response.data, "ok, will do...");
2090                     else
2091                         buffer_strcat(w->response.data, "I am doing it already");
2092
2093                     error("web request to exit received.");
2094                     netdata_cleanup_and_exit(0);
2095                 }
2096                 else if(hash == hash_debug && strcmp(tok, "debug") == 0) {
2097                     buffer_flush(w->response.data);
2098
2099                     // get the name of the data to show
2100                     tok = mystrsep(&url, "/?&");
2101                     if(tok && *tok) {
2102                         debug(D_WEB_CLIENT, "%llu: Searching for RRD data with name '%s'.", w->id, tok);
2103
2104                         // do we have such a data set?
2105                         RRDSET *st = rrdset_find_byname(tok);
2106                         if(!st) st = rrdset_find(tok);
2107                         if(!st) {
2108                             code = 404;
2109                             buffer_strcat(w->response.data, "Chart is not found: ");
2110                             buffer_strcat_htmlescape(w->response.data, tok);
2111                             debug(D_WEB_CLIENT_ACCESS, "%llu: %s is not found.", w->id, tok);
2112                         }
2113                         else {
2114                             code = 200;
2115                             debug_flags |= D_RRD_STATS;
2116                             st->debug = !st->debug;
2117                             buffer_sprintf(w->response.data, "Chart has now debug %s: ", st->debug?"enabled":"disabled");
2118                             buffer_strcat_htmlescape(w->response.data, tok);
2119                             debug(D_WEB_CLIENT_ACCESS, "%llu: debug for %s is %s.", w->id, tok, st->debug?"enabled":"disabled");
2120                         }
2121                     }
2122                     else {
2123                         code = 500;
2124                         buffer_flush(w->response.data);
2125                         buffer_strcat(w->response.data, "debug which chart?\r\n");
2126                     }
2127                 }
2128                 else if(hash == hash_mirror && strcmp(tok, "mirror") == 0) {
2129                     code = 200;
2130
2131                     debug(D_WEB_CLIENT_ACCESS, "%llu: Mirroring...", w->id);
2132
2133                     // replace the zero bytes with spaces
2134                     buffer_char_replace(w->response.data, '\0', ' ');
2135
2136                     // just leave the buffer as is
2137                     // it will be copied back to the client
2138                 }
2139 #endif  /* NETDATA_INTERNAL_CHECKS */
2140                 else {
2141                     char filename[FILENAME_MAX+1];
2142                     url = filename;
2143                     strncpyz(filename, w->last_url, FILENAME_MAX);
2144                     tok = mystrsep(&url, "?");
2145                     buffer_flush(w->response.data);
2146                     code = mysendfile(w, (tok && *tok)?tok:"/");
2147                 }
2148             }
2149             else {
2150                 char filename[FILENAME_MAX+1];
2151                 url = filename;
2152                 strncpyz(filename, w->last_url, FILENAME_MAX);
2153                 tok = mystrsep(&url, "?");
2154                 buffer_flush(w->response.data);
2155                 code = mysendfile(w, (tok && *tok)?tok:"/");
2156             }
2157         }
2158     }
2159
2160     now_realtime_timeval(&w->tv_ready);
2161     w->response.sent = 0;
2162     w->response.code = code;
2163
2164     // set a proper last modified date
2165     if(unlikely(!w->response.data->date))
2166         w->response.data->date = w->tv_ready.tv_sec;
2167
2168     if(unlikely(code != 200))
2169         buffer_no_cacheable(w->response.data);
2170
2171     // set a proper expiration date, if not already set
2172     if(unlikely(!w->response.data->expires)) {
2173         if(w->response.data->options & WB_CONTENT_NO_CACHEABLE)
2174             w->response.data->expires = w->tv_ready.tv_sec + rrd_update_every;
2175         else
2176             w->response.data->expires = w->tv_ready.tv_sec + 86400;
2177     }
2178
2179     // prepare the HTTP response header
2180     debug(D_WEB_CLIENT, "%llu: Generating HTTP header with response %d.", w->id, code);
2181
2182     const char *content_type_string = web_content_type_to_string(w->response.data->contenttype);
2183     const char *code_msg = web_response_code_to_string(code);
2184
2185     // prepare the last modified and expiration dates
2186     char date[32], edate[32];
2187     {
2188         struct tm tmbuf, *tm;
2189
2190         tm = gmtime_r(&w->response.data->date, &tmbuf);
2191         strftime(date, sizeof(date), "%a, %d %b %Y %H:%M:%S %Z", tm);
2192
2193         tm = gmtime_r(&w->response.data->expires, &tmbuf);
2194         strftime(edate, sizeof(edate), "%a, %d %b %Y %H:%M:%S %Z", tm);
2195     }
2196
2197     buffer_sprintf(w->response.header_output,
2198         "HTTP/1.1 %d %s\r\n"
2199         "Connection: %s\r\n"
2200         "Server: NetData Embedded HTTP Server\r\n"
2201         "Access-Control-Allow-Origin: %s\r\n"
2202         "Access-Control-Allow-Credentials: true\r\n"
2203         "Content-Type: %s\r\n"
2204         "Date: %s\r\n"
2205         , code, code_msg
2206         , w->keepalive?"keep-alive":"close"
2207         , w->origin
2208         , content_type_string
2209         , date
2210         );
2211
2212     if(w->cookie1[0] || w->cookie2[0]) {
2213         if(w->cookie1[0]) {
2214             buffer_sprintf(w->response.header_output,
2215                "Set-Cookie: %s\r\n",
2216                w->cookie1);
2217         }
2218
2219         if(w->cookie2[0]) {
2220             buffer_sprintf(w->response.header_output,
2221                "Set-Cookie: %s\r\n",
2222                w->cookie2);
2223         }
2224
2225         if(web_donotrack_comply)
2226             buffer_sprintf(w->response.header_output,
2227                "Tk: T;cookies\r\n");
2228     }
2229     else {
2230         if(web_donotrack_comply) {
2231             if(w->tracking_required)
2232                 buffer_sprintf(w->response.header_output,
2233                    "Tk: T;cookies\r\n");
2234             else
2235                 buffer_sprintf(w->response.header_output,
2236                    "Tk: N\r\n");
2237         }
2238     }
2239
2240     if(w->mode == WEB_CLIENT_MODE_OPTIONS) {
2241         buffer_strcat(w->response.header_output,
2242             "Access-Control-Allow-Methods: GET, OPTIONS\r\n"
2243             "Access-Control-Allow-Headers: accept, x-requested-with, origin, content-type, cookie, pragma, cache-control\r\n"
2244             "Access-Control-Max-Age: 1209600\r\n" // 86400 * 14
2245             );
2246     }
2247     else {
2248         buffer_sprintf(w->response.header_output,
2249             "Cache-Control: %s\r\n"
2250             "Expires: %s\r\n",
2251             (w->response.data->options & WB_CONTENT_NO_CACHEABLE)?"no-cache":"public",
2252             edate);
2253     }
2254
2255     // copy a possibly available custom header
2256     if(unlikely(buffer_strlen(w->response.header)))
2257         buffer_strcat(w->response.header_output, buffer_tostring(w->response.header));
2258
2259     // headers related to the transfer method
2260     if(likely(w->response.zoutput)) {
2261         buffer_strcat(w->response.header_output,
2262             "Content-Encoding: gzip\r\n"
2263             "Transfer-Encoding: chunked\r\n"
2264             );
2265     }
2266     else {
2267         if(likely((w->response.data->len || w->response.rlen))) {
2268             // we know the content length, put it
2269             buffer_sprintf(w->response.header_output, "Content-Length: %zu\r\n", w->response.data->len? w->response.data->len: w->response.rlen);
2270         }
2271         else {
2272             // we don't know the content length, disable keep-alive
2273             w->keepalive = 0;
2274         }
2275     }
2276
2277     // end of HTTP header
2278     buffer_strcat(w->response.header_output, "\r\n");
2279
2280     // sent the HTTP header
2281     debug(D_WEB_DATA, "%llu: Sending response HTTP header of size %zu: '%s'"
2282             , w->id
2283             , buffer_strlen(w->response.header_output)
2284             , buffer_tostring(w->response.header_output)
2285             );
2286
2287     web_client_crock_socket(w);
2288
2289     bytes = send(w->ofd, buffer_tostring(w->response.header_output), buffer_strlen(w->response.header_output), 0);
2290     if(bytes != (ssize_t) buffer_strlen(w->response.header_output)) {
2291         if(bytes > 0)
2292             w->stats_sent_bytes += bytes;
2293
2294         debug(D_WEB_CLIENT, "%llu: HTTP Header failed to be sent (I sent %zu bytes but the system sent %zd bytes). Closing web client."
2295             , w->id
2296             , buffer_strlen(w->response.header_output)
2297             , bytes);
2298
2299         WEB_CLIENT_IS_DEAD(w);
2300         return;
2301     }
2302     else 
2303         w->stats_sent_bytes += bytes;
2304
2305     // enable sending immediately if we have data
2306     if(w->response.data->len) w->wait_send = 1;
2307     else w->wait_send = 0;
2308
2309     // pretty logging
2310     switch(w->mode) {
2311         case WEB_CLIENT_MODE_OPTIONS:
2312             debug(D_WEB_CLIENT, "%llu: Done preparing the OPTIONS response. Sending data (%zu bytes) to client.", w->id, w->response.data->len);
2313             break;
2314
2315         case WEB_CLIENT_MODE_NORMAL:
2316             debug(D_WEB_CLIENT, "%llu: Done preparing the response. Sending data (%zu bytes) to client.", w->id, w->response.data->len);
2317             break;
2318
2319         case WEB_CLIENT_MODE_FILECOPY:
2320             if(w->response.rlen) {
2321                 debug(D_WEB_CLIENT, "%llu: Done preparing the response. Will be sending data file of %zu bytes to client.", w->id, w->response.rlen);
2322                 w->wait_receive = 1;
2323
2324                 /*
2325                 // utilize the kernel sendfile() for copying the file to the socket.
2326                 // this block of code can be commented, without anything missing.
2327                 // when it is commented, the program will copy the data using async I/O.
2328                 {
2329                     long len = sendfile(w->ofd, w->ifd, NULL, w->response.data->rbytes);
2330                     if(len != w->response.data->rbytes)
2331                         error("%llu: sendfile() should copy %ld bytes, but copied %ld. Falling back to manual copy.", w->id, w->response.data->rbytes, len);
2332                     else
2333                         web_client_reset(w);
2334                 }
2335                 */
2336             }
2337             else
2338                 debug(D_WEB_CLIENT, "%llu: Done preparing the response. Will be sending an unknown amount of bytes to client.", w->id);
2339             break;
2340
2341         default:
2342             fatal("%llu: Unknown client mode %d.", w->id, w->mode);
2343             break;
2344     }
2345 }
2346
2347 ssize_t web_client_send_chunk_header(struct web_client *w, size_t len)
2348 {
2349     debug(D_DEFLATE, "%llu: OPEN CHUNK of %zu bytes (hex: %zx).", w->id, len, len);
2350     char buf[24];
2351     sprintf(buf, "%zX\r\n", len);
2352     
2353     ssize_t bytes = send(w->ofd, buf, strlen(buf), 0);
2354     if(bytes > 0) {
2355         debug(D_DEFLATE, "%llu: Sent chunk header %zd bytes.", w->id, bytes);
2356         w->stats_sent_bytes += bytes;
2357     }
2358
2359     else if(bytes == 0) {
2360         debug(D_WEB_CLIENT, "%llu: Did not send chunk header to the client.", w->id);
2361         WEB_CLIENT_IS_DEAD(w);
2362     }
2363     else {
2364         debug(D_WEB_CLIENT, "%llu: Failed to send chunk header to client.", w->id);
2365         WEB_CLIENT_IS_DEAD(w);
2366     }
2367
2368     return bytes;
2369 }
2370
2371 ssize_t web_client_send_chunk_close(struct web_client *w)
2372 {
2373     //debug(D_DEFLATE, "%llu: CLOSE CHUNK.", w->id);
2374
2375     ssize_t bytes = send(w->ofd, "\r\n", 2, 0);
2376     if(bytes > 0) {
2377         debug(D_DEFLATE, "%llu: Sent chunk suffix %zd bytes.", w->id, bytes);
2378         w->stats_sent_bytes += bytes;
2379     }
2380
2381     else if(bytes == 0) {
2382         debug(D_WEB_CLIENT, "%llu: Did not send chunk suffix to the client.", w->id);
2383         WEB_CLIENT_IS_DEAD(w);
2384     }
2385     else {
2386         debug(D_WEB_CLIENT, "%llu: Failed to send chunk suffix to client.", w->id);
2387         WEB_CLIENT_IS_DEAD(w);
2388     }
2389
2390     return bytes;
2391 }
2392
2393 ssize_t web_client_send_chunk_finalize(struct web_client *w)
2394 {
2395     //debug(D_DEFLATE, "%llu: FINALIZE CHUNK.", w->id);
2396
2397     ssize_t bytes = send(w->ofd, "\r\n0\r\n\r\n", 7, 0);
2398     if(bytes > 0) {
2399         debug(D_DEFLATE, "%llu: Sent chunk suffix %zd bytes.", w->id, bytes);
2400         w->stats_sent_bytes += bytes;
2401     }
2402
2403     else if(bytes == 0) {
2404         debug(D_WEB_CLIENT, "%llu: Did not send chunk finalize suffix to the client.", w->id);
2405         WEB_CLIENT_IS_DEAD(w);
2406     }
2407     else {
2408         debug(D_WEB_CLIENT, "%llu: Failed to send chunk finalize suffix to client.", w->id);
2409         WEB_CLIENT_IS_DEAD(w);
2410     }
2411
2412     return bytes;
2413 }
2414
2415 #ifdef NETDATA_WITH_ZLIB
2416 ssize_t web_client_send_deflate(struct web_client *w)
2417 {
2418     ssize_t len = 0, t = 0;
2419
2420     // when using compression,
2421     // w->response.sent is the amount of bytes passed through compression
2422
2423     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.",
2424         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);
2425
2426     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) {
2427         // there is nothing to send
2428
2429         debug(D_WEB_CLIENT, "%llu: Out of output data.", w->id);
2430
2431         // finalize the chunk
2432         if(w->response.sent != 0) {
2433             t = web_client_send_chunk_finalize(w);
2434             if(t < 0) return t;
2435         }
2436
2437         if(w->mode == WEB_CLIENT_MODE_FILECOPY && w->wait_receive && w->response.rlen && w->response.rlen > w->response.data->len) {
2438             // we have to wait, more data will come
2439             debug(D_WEB_CLIENT, "%llu: Waiting for more data to become available.", w->id);
2440             w->wait_send = 0;
2441             return t;
2442         }
2443
2444         if(unlikely(!w->keepalive)) {
2445             debug(D_WEB_CLIENT, "%llu: Closing (keep-alive is not enabled). %zu bytes sent.", w->id, w->response.sent);
2446             WEB_CLIENT_IS_DEAD(w);
2447             return t;
2448         }
2449
2450         // reset the client
2451         web_client_reset(w);
2452         debug(D_WEB_CLIENT, "%llu: Done sending all data on socket.", w->id);
2453         return t;
2454     }
2455
2456     if(w->response.zhave == w->response.zsent) {
2457         // compress more input data
2458
2459         // close the previous open chunk
2460         if(w->response.sent != 0) {
2461             t = web_client_send_chunk_close(w);
2462             if(t < 0) return t;
2463         }
2464
2465         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);
2466
2467         // give the compressor all the data not passed through the compressor yet
2468         if(w->response.data->len > w->response.sent) {
2469             w->response.zstream.next_in = (Bytef *)&w->response.data->buffer[w->response.sent - w->response.zstream.avail_in];
2470             w->response.zstream.avail_in += (uInt) (w->response.data->len - w->response.sent);
2471         }
2472
2473         // reset the compressor output buffer
2474         w->response.zstream.next_out = w->response.zbuffer;
2475         w->response.zstream.avail_out = ZLIB_CHUNK;
2476
2477         // ask for FINISH if we have all the input
2478         int flush = Z_SYNC_FLUSH;
2479         if(w->mode == WEB_CLIENT_MODE_NORMAL
2480             || (w->mode == WEB_CLIENT_MODE_FILECOPY && !w->wait_receive && w->response.data->len == w->response.rlen)) {
2481             flush = Z_FINISH;
2482             debug(D_DEFLATE, "%llu: Requesting Z_FINISH, if possible.", w->id);
2483         }
2484         else {
2485             debug(D_DEFLATE, "%llu: Requesting Z_SYNC_FLUSH.", w->id);
2486         }
2487
2488         // compress
2489         if(deflate(&w->response.zstream, flush) == Z_STREAM_ERROR) {
2490             error("%llu: Compression failed. Closing down client.", w->id);
2491             web_client_reset(w);
2492             return(-1);
2493         }
2494
2495         w->response.zhave = ZLIB_CHUNK - w->response.zstream.avail_out;
2496         w->response.zsent = 0;
2497
2498         // keep track of the bytes passed through the compressor
2499         w->response.sent = w->response.data->len;
2500
2501         debug(D_DEFLATE, "%llu: Compression produced %zu bytes.", w->id, w->response.zhave);
2502
2503         // open a new chunk
2504         ssize_t t2 = web_client_send_chunk_header(w, w->response.zhave);
2505         if(t2 < 0) return t2;
2506         t += t2;
2507     }
2508     
2509     debug(D_WEB_CLIENT, "%llu: Sending %zu bytes of data (+%zd of chunk header).", w->id, w->response.zhave - w->response.zsent, t);
2510
2511     len = send(w->ofd, &w->response.zbuffer[w->response.zsent], (size_t) (w->response.zhave - w->response.zsent), MSG_DONTWAIT);
2512     if(len > 0) {
2513         w->stats_sent_bytes += len;
2514         w->response.zsent += len;
2515         len += t;
2516         debug(D_WEB_CLIENT, "%llu: Sent %zd bytes.", w->id, len);
2517     }
2518     else if(len == 0) {
2519         debug(D_WEB_CLIENT, "%llu: Did not send any bytes to the client (zhave = %zu, zsent = %zu, need to send = %zu).",
2520             w->id, w->response.zhave, w->response.zsent, w->response.zhave - w->response.zsent);
2521
2522         WEB_CLIENT_IS_DEAD(w);
2523     }
2524     else {
2525         debug(D_WEB_CLIENT, "%llu: Failed to send data to client.", w->id);
2526         WEB_CLIENT_IS_DEAD(w);
2527     }
2528
2529     return(len);
2530 }
2531 #endif // NETDATA_WITH_ZLIB
2532
2533 ssize_t web_client_send(struct web_client *w) {
2534 #ifdef NETDATA_WITH_ZLIB
2535     if(likely(w->response.zoutput)) return web_client_send_deflate(w);
2536 #endif // NETDATA_WITH_ZLIB
2537
2538     ssize_t bytes;
2539
2540     if(unlikely(w->response.data->len - w->response.sent == 0)) {
2541         // there is nothing to send
2542
2543         debug(D_WEB_CLIENT, "%llu: Out of output data.", w->id);
2544
2545         // there can be two cases for this
2546         // A. we have done everything
2547         // B. we temporarily have nothing to send, waiting for the buffer to be filled by ifd
2548
2549         if(w->mode == WEB_CLIENT_MODE_FILECOPY && w->wait_receive && w->response.rlen && w->response.rlen > w->response.data->len) {
2550             // we have to wait, more data will come
2551             debug(D_WEB_CLIENT, "%llu: Waiting for more data to become available.", w->id);
2552             w->wait_send = 0;
2553             return 0;
2554         }
2555
2556         if(unlikely(!w->keepalive)) {
2557             debug(D_WEB_CLIENT, "%llu: Closing (keep-alive is not enabled). %zu bytes sent.", w->id, w->response.sent);
2558             WEB_CLIENT_IS_DEAD(w);
2559             return 0;
2560         }
2561
2562         web_client_reset(w);
2563         debug(D_WEB_CLIENT, "%llu: Done sending all data on socket. Waiting for next request on the same socket.", w->id);
2564         return 0;
2565     }
2566
2567     bytes = send(w->ofd, &w->response.data->buffer[w->response.sent], w->response.data->len - w->response.sent, MSG_DONTWAIT);
2568     if(likely(bytes > 0)) {
2569         w->stats_sent_bytes += bytes;
2570         w->response.sent += bytes;
2571         debug(D_WEB_CLIENT, "%llu: Sent %zd bytes.", w->id, bytes);
2572     }
2573     else if(likely(bytes == 0)) {
2574         debug(D_WEB_CLIENT, "%llu: Did not send any bytes to the client.", w->id);
2575         WEB_CLIENT_IS_DEAD(w);
2576     }
2577     else {
2578         debug(D_WEB_CLIENT, "%llu: Failed to send data to client.", w->id);
2579         WEB_CLIENT_IS_DEAD(w);
2580     }
2581
2582     return(bytes);
2583 }
2584
2585 ssize_t web_client_receive(struct web_client *w)
2586 {
2587     // do we have any space for more data?
2588     buffer_need_bytes(w->response.data, WEB_REQUEST_LENGTH);
2589
2590     ssize_t left = w->response.data->size - w->response.data->len;
2591     ssize_t bytes;
2592
2593     if(unlikely(w->mode == WEB_CLIENT_MODE_FILECOPY))
2594         bytes = read(w->ifd, &w->response.data->buffer[w->response.data->len], (size_t) (left - 1));
2595     else
2596         bytes = recv(w->ifd, &w->response.data->buffer[w->response.data->len], (size_t) (left - 1), MSG_DONTWAIT);
2597
2598     if(likely(bytes > 0)) {
2599         if(w->mode != WEB_CLIENT_MODE_FILECOPY)
2600             w->stats_received_bytes += bytes;
2601
2602         size_t old = w->response.data->len;
2603         w->response.data->len += bytes;
2604         w->response.data->buffer[w->response.data->len] = '\0';
2605
2606         debug(D_WEB_CLIENT, "%llu: Received %zd bytes.", w->id, bytes);
2607         debug(D_WEB_DATA, "%llu: Received data: '%s'.", w->id, &w->response.data->buffer[old]);
2608
2609         if(w->mode == WEB_CLIENT_MODE_FILECOPY) {
2610             w->wait_send = 1;
2611
2612             if(w->response.rlen && w->response.data->len >= w->response.rlen)
2613                 w->wait_receive = 0;
2614         }
2615     }
2616     else if(likely(bytes == 0)) {
2617         debug(D_WEB_CLIENT, "%llu: Out of input data.", w->id);
2618
2619         // if we cannot read, it means we have an error on input.
2620         // if however, we are copying a file from ifd to ofd, we should not return an error.
2621         // in this case, the error should be generated when the file has been sent to the client.
2622
2623         if(w->mode == WEB_CLIENT_MODE_FILECOPY) {
2624             // we are copying data from ifd to ofd
2625             // let it finish copying...
2626             w->wait_receive = 0;
2627
2628             debug(D_WEB_CLIENT, "%llu: Read the whole file.", w->id);
2629             if(w->ifd != w->ofd) close(w->ifd);
2630             w->ifd = w->ofd;
2631         }
2632         else {
2633             debug(D_WEB_CLIENT, "%llu: failed to receive data.", w->id);
2634             WEB_CLIENT_IS_DEAD(w);
2635         }
2636     }
2637     else {
2638         debug(D_WEB_CLIENT, "%llu: receive data failed.", w->id);
2639         WEB_CLIENT_IS_DEAD(w);
2640     }
2641
2642     return(bytes);
2643 }
2644
2645
2646 // --------------------------------------------------------------------------------------
2647 // the thread of a single client
2648
2649 // 1. waits for input and output, using async I/O
2650 // 2. it processes HTTP requests
2651 // 3. it generates HTTP responses
2652 // 4. it copies data from input to output if mode is FILECOPY
2653
2654 void *web_client_main(void *ptr)
2655 {
2656     if(pthread_setcanceltype(PTHREAD_CANCEL_DEFERRED, NULL) != 0)
2657         error("Cannot set pthread cancel type to DEFERRED.");
2658
2659     if(pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL) != 0)
2660         error("Cannot set pthread cancel state to ENABLE.");
2661
2662     struct web_client *w = ptr;
2663     struct pollfd fds[2], *ifd, *ofd;
2664     int retval, fdmax = 0, timeout;
2665
2666     log_access("%llu: %s port %s connected on thread task id %d", w->id, w->client_ip, w->client_port, gettid());
2667
2668     for(;;) {
2669         if(unlikely(w->dead)) {
2670             debug(D_WEB_CLIENT, "%llu: client is dead.", w->id);
2671             break;
2672         }
2673         else if(unlikely(!w->wait_receive && !w->wait_send)) {
2674             debug(D_WEB_CLIENT, "%llu: client is not set for neither receiving nor sending data.", w->id);
2675             break;
2676         }
2677
2678         if(unlikely(w->ifd < 0 || w->ofd < 0)) {
2679             error("%llu: invalid file descriptor, ifd = %d, ofd = %d (required 0 <= fd", w->id, w->ifd, w->ofd);
2680             break;
2681         }
2682
2683         if(w->ifd == w->ofd) {
2684             fds[0].fd = w->ifd;
2685             fds[0].events = 0;
2686             fds[0].revents = 0;
2687
2688             if(w->wait_receive) fds[0].events |= POLLIN;
2689             if(w->wait_send)    fds[0].events |= POLLOUT;
2690
2691             fds[1].fd = -1;
2692             fds[1].events = 0;
2693             fds[1].revents = 0;
2694
2695             ifd = ofd = &fds[0];
2696
2697             fdmax = 1;
2698         }
2699         else {
2700             fds[0].fd = w->ifd;
2701             fds[0].events = 0;
2702             fds[0].revents = 0;
2703             if(w->wait_receive) fds[0].events |= POLLIN;
2704             ifd = &fds[0];
2705
2706             fds[1].fd = w->ofd;
2707             fds[1].events = 0;
2708             fds[1].revents = 0;
2709             if(w->wait_send)    fds[1].events |= POLLOUT;
2710             ofd = &fds[1];
2711
2712             fdmax = 2;
2713         }
2714
2715         debug(D_WEB_CLIENT, "%llu: Waiting socket async I/O for %s %s", w->id, w->wait_receive?"INPUT":"", w->wait_send?"OUTPUT":"");
2716         errno = 0;
2717         timeout = web_client_timeout * 1000;
2718         retval = poll(fds, fdmax, timeout);
2719
2720         if(unlikely(retval == -1)) {
2721             if(errno == EAGAIN || errno == EINTR) {
2722                 debug(D_WEB_CLIENT, "%llu: EAGAIN received.", w->id);
2723                 continue;
2724             }
2725
2726             debug(D_WEB_CLIENT, "%llu: LISTENER: poll() failed (input fd = %d, output fd = %d). Closing client.", w->id, w->ifd, w->ofd);
2727             break;
2728         }
2729         else if(unlikely(!retval)) {
2730             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":"");
2731             break;
2732         }
2733
2734         int used = 0;
2735         if(w->wait_send && ofd->revents & POLLOUT) {
2736             used++;
2737             if(web_client_send(w) < 0) {
2738                 debug(D_WEB_CLIENT, "%llu: Cannot send data to client. Closing client.", w->id);
2739                 break;
2740             }
2741         }
2742
2743         if(w->wait_receive && (ifd->revents & POLLIN || ifd->revents & POLLPRI)) {
2744             used++;
2745             if(web_client_receive(w) < 0) {
2746                 debug(D_WEB_CLIENT, "%llu: Cannot receive data from client. Closing client.", w->id);
2747                 break;
2748             }
2749
2750             if(w->mode == WEB_CLIENT_MODE_NORMAL) {
2751                 debug(D_WEB_CLIENT, "%llu: Attempting to process received data.", w->id);
2752                 web_client_process(w);
2753             }
2754         }
2755
2756         if(unlikely(!used)) {
2757             debug(D_WEB_CLIENT_ACCESS, "%llu: Received error on socket.", w->id);
2758             break;
2759         }
2760     }
2761
2762     web_client_reset(w);
2763
2764     log_access("%llu: %s port %s disconnected from thread task id %d", w->id, w->client_ip, w->client_port, gettid());
2765     debug(D_WEB_CLIENT, "%llu: done...", w->id);
2766
2767     // close the sockets/files now
2768     // to free file descriptors
2769     if(w->ifd == w->ofd) {
2770         if(w->ifd != -1) close(w->ifd);
2771     }
2772     else {
2773         if(w->ifd != -1) close(w->ifd);
2774         if(w->ofd != -1) close(w->ofd);
2775     }
2776     w->ifd = -1;
2777     w->ofd = -1;
2778
2779     w->obsolete = 1;
2780
2781     pthread_exit(NULL);
2782     return NULL;
2783 }