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