src/web_client.c

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