src/web_client.c

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