src/web_client.c

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