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