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