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