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