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