src/web_client.c

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