src/registry.c

   1 #ifdef HAVE_CONFIG_H
   2 #include <config.h>
   3 #endif
   4
   5 #include <uuid/uuid.h>
   6 #include <inttypes.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <ctype.h>
  10 #include <unistd.h>
  11 #include <sys/stat.h>
  12 #include <sys/types.h>
  13 #include <errno.h>
  14 #include <fcntl.h>
  15
  16 #include "log.h"
  17 #include "common.h"
  18 #include "dictionary.h"
  19 #include "appconfig.h"
  20
  21 #include "web_client.h"
  22 #include "rrd.h"
  23 #include "rrd2json.h"
  24 #include "registry.h"
  25
  26
  27 // ----------------------------------------------------------------------------
  28 // TODO
  29 //
  30 // 1. the default tracking cookie expires in 1 year, but the persons are not
  31 //    removed from the db - this means the database only grows - ideally the
  32 //    database should be cleaned in registry_save() for both on-disk and
  33 //    on-memory entries.
  34 //
  35 //    Cleanup:
  36 //    i. Find all the PERSONs that have expired cookie
  37 //    ii. For each of their PERSON_URLs:
  38 //     - decrement the linked MACHINE links
  39 //     - if the linked MACHINE has no other links, remove the linked MACHINE too
  40 //     - remove the PERSON_URL
  41 //
  42 // 2. add protection to prevent abusing the registry by flooding it with
  43 //    requests to fill the memory and crash it.
  44 //
  45 //    Possible protections:
  46 //    - limit the number of URLs per person
  47 //    - limit the number of URLs per machine
  48 //    - limit the number of persons
  49 //    - limit the number of machines
  50 //    - [DONE] limit the size of URLs
  51 //    - [DONE] limit the size of PERSON_URL names
  52 //    - limit the number of requests that add data to the registry,
  53 //      per client IP per hour
  54
  55
  56
  57 #define REGISTRY_URL_FLAGS_DEFAULT 0x00
  58 #define REGISTRY_URL_FLAGS_EXPIRED 0x01
  59
  60 #define DICTIONARY_FLAGS DICTIONARY_FLAG_VALUE_LINK_DONT_CLONE | DICTIONARY_FLAG_NAME_LINK_DONT_CLONE
  61
  62 // ----------------------------------------------------------------------------
  63 // COMMON structures
  64
  65 struct registry {
  66         int enabled;
  67
  68         char machine_guid[36 + 1];
  69
  70         // entries counters / statistics
  71         unsigned long long persons_count;
  72         unsigned long long machines_count;
  73         unsigned long long usages_count;
  74         unsigned long long urls_count;
  75         unsigned long long persons_urls_count;
  76         unsigned long long machines_urls_count;
  77         unsigned long long log_count;
  78
  79         // memory counters / statistics
  80         unsigned long long persons_memory;
  81         unsigned long long machines_memory;
  82         unsigned long long urls_memory;
  83         unsigned long long persons_urls_memory;
  84         unsigned long long machines_urls_memory;
  85
  86         // configuration
  87         unsigned long long save_registry_every_entries;
  88         char *registry_domain;
  89         char *hostname;
  90         char *registry_to_announce;
  91         time_t persons_expiration; // seconds to expire idle persons
  92         int verify_cookies_redirects;
  93
  94         size_t max_url_length;
  95         size_t max_name_length;
  96
  97         // file/path names
  98         char *pathname;
  99         char *db_filename;
 100         char *log_filename;
 101         char *machine_guid_filename;
 102
 103         // open files
 104         FILE *log_fp;
 105
 106         // the database
 107         DICTIONARY *persons;    // dictionary of PERSON *, with key the PERSON.guid
 108         DICTIONARY *machines;   // dictionary of MACHINE *, with key the MACHINE.guid
 109         DICTIONARY *urls;               // dictionary of URL *, with key the URL.url
 110
 111         // concurrency locking
 112         // we keep different locks for different things
 113         // so that many tasks can be completed in parallel
 114         pthread_mutex_t persons_lock;
 115         pthread_mutex_t machines_lock;
 116         pthread_mutex_t urls_lock;
 117         pthread_mutex_t person_urls_lock;
 118         pthread_mutex_t machine_urls_lock;
 119         pthread_mutex_t log_lock;
 120 } registry;
 121
 122
 123 // ----------------------------------------------------------------------------
 124 // URL structures
 125 // Save memory by de-duplicating URLs
 126 // so instead of storing URLs all over the place
 127 // we store them here and we keep pointers elsewhere
 128
 129 struct url {
 130         uint32_t links; // the number of links to this URL - when none is left, we free it
 131         uint16_t len;   // the length of the URL in bytes
 132         char url[1];    // the URL - dynamically allocated to more size
 133 };
 134 typedef struct url URL;
 135
 136
 137 // ----------------------------------------------------------------------------
 138 // MACHINE structures
 139
 140 // For each MACHINE-URL pair we keep this
 141 struct machine_url {
 142         URL *url;                                       // de-duplicated URL
 143 //      DICTIONARY *persons;            // dictionary of PERSON *
 144
 145         uint8_t flags;
 146         uint32_t first_t;                       // the first time we saw this
 147         uint32_t last_t;                        // the last time we saw this
 148         uint32_t usages;                        // how many times this has been accessed
 149 };
 150 typedef struct machine_url MACHINE_URL;
 151
 152 // A machine
 153 struct machine {
 154         char guid[36 + 1];                      // the GUID
 155
 156         uint32_t links;                         // the number of PERSON_URLs linked to this machine
 157
 158         DICTIONARY *urls;                       // MACHINE_URL *
 159
 160         uint32_t first_t;                       // the first time we saw this
 161         uint32_t last_t;                        // the last time we saw this
 162         uint32_t usages;                        // how many times this has been accessed
 163 };
 164 typedef struct machine MACHINE;
 165
 166
 167 // ----------------------------------------------------------------------------
 168 // PERSON structures
 169
 170 // for each PERSON-URL pair we keep this
 171 struct person_url {
 172         URL *url;                                       // de-duplicated URL
 173         MACHINE *machine;                       // link the MACHINE of this URL
 174
 175         uint8_t flags;
 176         uint32_t first_t;                       // the first time we saw this
 177         uint32_t last_t;                        // the last time we saw this
 178         uint32_t usages;                        // how many times this has been accessed
 179
 180         char name[1];                           // the name of the URL, as known by the user
 181                                                                 // dynamically allocated to fit properly
 182 };
 183 typedef struct person_url PERSON_URL;
 184
 185 // A person
 186 struct person {
 187         char guid[36 + 1];                      // the person GUID
 188
 189         DICTIONARY *urls;                       // dictionary of PERSON_URL *
 190
 191         uint32_t first_t;                       // the first time we saw this
 192         uint32_t last_t;                        // the last time we saw this
 193         uint32_t usages;                        // how many times this has been accessed
 194 };
 195 typedef struct person PERSON;
 196
 197
 198 // ----------------------------------------------------------------------------
 199 // REGISTRY concurrency locking
 200
 201 static inline void registry_persons_lock(void) {
 202         pthread_mutex_lock(&registry.persons_lock);
 203 }
 204
 205 static inline void registry_persons_unlock(void) {
 206         pthread_mutex_unlock(&registry.persons_lock);
 207 }
 208
 209 static inline void registry_machines_lock(void) {
 210         pthread_mutex_lock(&registry.machines_lock);
 211 }
 212
 213 static inline void registry_machines_unlock(void) {
 214         pthread_mutex_unlock(&registry.machines_lock);
 215 }
 216
 217 static inline void registry_urls_lock(void) {
 218         pthread_mutex_lock(&registry.urls_lock);
 219 }
 220
 221 static inline void registry_urls_unlock(void) {
 222         pthread_mutex_unlock(&registry.urls_lock);
 223 }
 224
 225 // ideally, we should not lock the whole registry for
 226 // updating a person's urls.
 227 // however, to save the memory required for keeping a
 228 // mutex (40 bytes) per person, we do...
 229 static inline void registry_person_urls_lock(PERSON *p) {
 230         (void)p;
 231         pthread_mutex_lock(&registry.person_urls_lock);
 232 }
 233
 234 static inline void registry_person_urls_unlock(PERSON *p) {
 235         (void)p;
 236         pthread_mutex_unlock(&registry.person_urls_lock);
 237 }
 238
 239 // ideally, we should not lock the whole registry for
 240 // updating a machine's urls.
 241 // however, to save the memory required for keeping a
 242 // mutex (40 bytes) per machine, we do...
 243 static inline void registry_machine_urls_lock(MACHINE *m) {
 244         (void)m;
 245         pthread_mutex_lock(&registry.machine_urls_lock);
 246 }
 247
 248 static inline void registry_machine_urls_unlock(MACHINE *m) {
 249         (void)m;
 250         pthread_mutex_unlock(&registry.machine_urls_lock);
 251 }
 252
 253 static inline void registry_log_lock(void) {
 254         pthread_mutex_lock(&registry.log_lock);
 255 }
 256
 257 static inline void registry_log_unlock(void) {
 258         pthread_mutex_unlock(&registry.log_lock);
 259 }
 260
 261
 262 // ----------------------------------------------------------------------------
 263 // common functions
 264
 265 // parse a GUID and re-generated to be always lower case
 266 // this is used as a protection against the variations of GUIDs
 267 static inline int registry_regenerate_guid(const char *guid, char *result) {
 268         uuid_t uuid;
 269         if(unlikely(uuid_parse(guid, uuid) == -1)) {
 270                 info("Registry: GUID '%s' is not a valid GUID.", guid);
 271                 return -1;
 272         }
 273         else {
 274                 uuid_unparse_lower(uuid, result);
 275
 276 #ifdef NETDATA_INTERNAL_CHECKS
 277                 if(strcmp(guid, result))
 278                         info("Registry: source GUID '%s' and re-generated GUID '%s' differ!", guid, result);
 279 #endif /* NETDATA_INTERNAL_CHECKS */
 280         }
 281
 282         return 0;
 283 }
 284
 285 // make sure the names of the machines / URLs do not contain any tabs
 286 // (which are used as our separator in the database files)
 287 // and are properly trimmed (before and after)
 288 static inline char *registry_fix_machine_name(char *name, size_t *len) {
 289         char *s = name?name:"";
 290
 291         // skip leading spaces
 292         while(*s && isspace(*s)) s++;
 293
 294         // make sure all spaces are a SPACE
 295         char *t = s;
 296         while(*t) {
 297                 if(unlikely(isspace(*t)))
 298                         *t = ' ';
 299
 300                 t++;
 301         }
 302
 303         // remove trailing spaces
 304         while(--t >= s) {
 305                 if(*t == ' ')
 306                         *t = '\0';
 307                 else
 308                         break;
 309         }
 310         t++;
 311
 312         if(likely(len))
 313                 *len = (t - s);
 314
 315         return s;
 316 }
 317
 318 static inline char *registry_fix_url(char *url, size_t *len) {
 319         return registry_fix_machine_name(url, len);
 320 }
 321
 322
 323 // ----------------------------------------------------------------------------
 324 // forward definition of functions
 325
 326 extern PERSON *registry_request_access(char *person_guid, char *machine_guid, char *url, char *name, time_t when);
 327 extern PERSON *registry_request_delete(char *person_guid, char *machine_guid, char *url, char *delete_url, time_t when);
 328
 329
 330 // ----------------------------------------------------------------------------
 331 // URL
 332
 333 static inline URL *registry_url_allocate_nolock(const char *url, size_t urllen) {
 334         // protection from too big URLs
 335         if(urllen > registry.max_url_length)
 336                 urllen = registry.max_url_length;
 337
 338         debug(D_REGISTRY, "Registry: registry_url_allocate_nolock('%s'): allocating %zu bytes", url, sizeof(URL) + urllen);
 339         URL *u = malloc(sizeof(URL) + urllen);
 340         if(!u) fatal("Cannot allocate %zu bytes for URL '%s'", sizeof(URL) + urllen, url);
 341
 342         // a simple strcpy() should do the job
 343         // but I prefer to be safe, since the caller specified urllen
 344         strncpyz(u->url, url, urllen);
 345
 346         u->len = urllen;
 347         u->links = 0;
 348
 349         registry.urls_memory += sizeof(URL) + urllen;
 350
 351         debug(D_REGISTRY, "Registry: registry_url_allocate_nolock('%s'): indexing it", url);
 352         dictionary_set(registry.urls, u->url, u, sizeof(URL));
 353
 354         return u;
 355 }
 356
 357 static inline URL *registry_url_get(const char *url, size_t urllen) {
 358         debug(D_REGISTRY, "Registry: registry_url_get('%s')", url);
 359
 360         registry_urls_lock();
 361
 362         URL *u = dictionary_get(registry.urls, url);
 363         if(!u) {
 364                 u = registry_url_allocate_nolock(url, urllen);
 365                 registry.urls_count++;
 366         }
 367
 368         registry_urls_unlock();
 369
 370         return u;
 371 }
 372
 373 static inline void registry_url_link_nolock(URL *u) {
 374         u->links++;
 375         debug(D_REGISTRY, "Registry: registry_url_link_nolock('%s'): URL has now %u links", u->url, u->links);
 376 }
 377
 378 static inline void registry_url_unlink_nolock(URL *u) {
 379         u->links--;
 380         if(!u->links) {
 381                 debug(D_REGISTRY, "Registry: registry_url_unlink_nolock('%s'): No more links for this URL", u->url);
 382                 dictionary_del(registry.urls, u->url);
 383                 free(u);
 384         }
 385         else
 386                 debug(D_REGISTRY, "Registry: registry_url_unlink_nolock('%s'): URL has %u links left", u->url, u->links);
 387 }
 388
 389
 390 // ----------------------------------------------------------------------------
 391 // MACHINE
 392
 393 static inline MACHINE *registry_machine_find(const char *machine_guid) {
 394         debug(D_REGISTRY, "Registry: registry_machine_find('%s')", machine_guid);
 395         return dictionary_get(registry.machines, machine_guid);
 396 }
 397
 398 static inline MACHINE_URL *registry_machine_url_allocate(MACHINE *m, URL *u, time_t when) {
 399         debug(D_REGISTRY, "registry_machine_link_to_url('%s', '%s'): allocating %zu bytes", m->guid, u->url, sizeof(MACHINE_URL));
 400
 401         MACHINE_URL *mu = malloc(sizeof(MACHINE_URL));
 402         if(!mu) fatal("registry_machine_link_to_url('%s', '%s'): cannot allocate %zu bytes.", m->guid, u->url, sizeof(MACHINE_URL));
 403
 404         // mu->persons = dictionary_create(DICTIONARY_FLAGS);
 405         // dictionary_set(mu->persons, p->guid, p, sizeof(PERSON));
 406
 407         mu->first_t = mu->last_t = when;
 408         mu->usages = 1;
 409         mu->url = u;
 410         mu->flags = REGISTRY_URL_FLAGS_DEFAULT;
 411
 412         registry.machines_urls_memory += sizeof(MACHINE_URL);
 413
 414         debug(D_REGISTRY, "registry_machine_link_to_url('%s', '%s'): indexing URL in machine", m->guid, u->url);
 415         dictionary_set(m->urls, u->url, mu, sizeof(MACHINE_URL));
 416         registry_url_link_nolock(u);
 417
 418         return mu;
 419 }
 420
 421 static inline MACHINE *registry_machine_allocate(const char *machine_guid, time_t when) {
 422         debug(D_REGISTRY, "Registry: registry_machine_allocate('%s'): creating new machine, sizeof(MACHINE)=%zu", machine_guid, sizeof(MACHINE));
 423
 424         MACHINE *m = malloc(sizeof(MACHINE));
 425         if(!m) fatal("Registry: cannot allocate memory for new machine '%s'", machine_guid);
 426
 427         strncpyz(m->guid, machine_guid, 36);
 428
 429         debug(D_REGISTRY, "Registry: registry_machine_allocate('%s'): creating dictionary of urls", machine_guid);
 430         m->urls = dictionary_create(DICTIONARY_FLAGS);
 431
 432         m->first_t = m->last_t = when;
 433         m->usages = 0;
 434
 435         registry.machines_memory += sizeof(MACHINE);
 436
 437         registry.machines_count++;
 438         dictionary_set(registry.machines, m->guid, m, sizeof(MACHINE));
 439
 440         return m;
 441 }
 442
 443 // 1. validate machine GUID
 444 // 2. if it is valid, find it or create it and return it
 445 // 3. if it is not valid, return NULL
 446 static inline MACHINE *registry_machine_get(const char *machine_guid, time_t when) {
 447         MACHINE *m = NULL;
 448
 449         registry_machines_lock();
 450
 451         if(likely(machine_guid && *machine_guid)) {
 452                 // validate it is a GUID
 453                 char buf[36 + 1];
 454                 if(unlikely(registry_regenerate_guid(machine_guid, buf) == -1))
 455                         info("Registry: machine guid '%s' is not a valid guid. Ignoring it.", machine_guid);
 456                 else {
 457                         machine_guid = buf;
 458                         m = registry_machine_find(machine_guid);
 459                         if(!m) m = registry_machine_allocate(machine_guid, when);
 460                 }
 461         }
 462
 463         registry_machines_unlock();
 464
 465         return m;
 466 }
 467
 468
 469 // ----------------------------------------------------------------------------
 470 // PERSON
 471
 472 static inline PERSON *registry_person_find(const char *person_guid) {
 473         debug(D_REGISTRY, "Registry: registry_person_find('%s')", person_guid);
 474         return dictionary_get(registry.persons, person_guid);
 475 }
 476
 477 static inline PERSON_URL *registry_person_url_allocate(PERSON *p, MACHINE *m, URL *u, char *name, size_t namelen, time_t when) {
 478         // protection from too big names
 479         if(namelen > registry.max_name_length)
 480                 namelen = registry.max_name_length;
 481
 482         debug(D_REGISTRY, "registry_person_url_allocate('%s', '%s', '%s'): allocating %zu bytes", p->guid, m->guid, u->url,
 483                   sizeof(PERSON_URL) + namelen);
 484
 485         PERSON_URL *pu = malloc(sizeof(PERSON_URL) + namelen);
 486         if(!pu) fatal("registry_person_url_allocate('%s', '%s', '%s'): cannot allocate %zu bytes.", p->guid, m->guid, u->url, sizeof(PERSON_URL) + namelen);
 487
 488         // a simple strcpy() should do the job
 489         // but I prefer to be safe, since the caller specified urllen
 490         strncpyz(pu->name, name, namelen);
 491
 492         pu->machine = m;
 493         pu->first_t = pu->last_t = when;
 494         pu->usages = 1;
 495         pu->url = u;
 496         pu->flags = REGISTRY_URL_FLAGS_DEFAULT;
 497         m->links++;
 498
 499         registry.persons_urls_memory += sizeof(PERSON_URL) + namelen;
 500
 501         debug(D_REGISTRY, "registry_person_url_allocate('%s', '%s', '%s'): indexing URL in person", p->guid, m->guid, u->url);
 502         dictionary_set(p->urls, u->url, pu, sizeof(PERSON_URL));
 503         registry_url_link_nolock(u);
 504
 505         return pu;
 506 }
 507
 508 static inline PERSON_URL *registry_person_url_reallocate(PERSON *p, MACHINE *m, URL *u, char *name, size_t namelen, time_t when, PERSON_URL *pu) {
 509         // this function is needed to change the name of a PERSON_URL
 510
 511         debug(D_REGISTRY, "registry_person_url_reallocate('%s', '%s', '%s'): allocating %zu bytes", p->guid, m->guid, u->url,
 512                   sizeof(PERSON_URL) + namelen);
 513
 514         PERSON_URL *tpu = registry_person_url_allocate(p, m, u, name, namelen, when);
 515         tpu->first_t = pu->first_t;
 516         tpu->last_t = pu->last_t;
 517         tpu->usages = pu->usages;
 518
 519         // ok, these are a hack - since the registry_person_url_allocate() is
 520         // adding these, we have to subtract them
 521         tpu->machine->links--;
 522         registry.persons_urls_memory -= sizeof(PERSON_URL) + strlen(pu->name);
 523         registry_url_unlink_nolock(u);
 524
 525         free(pu);
 526
 527         return tpu;
 528 }
 529
 530 static inline PERSON *registry_person_allocate(const char *person_guid, time_t when) {
 531         PERSON *p = NULL;
 532
 533         debug(D_REGISTRY, "Registry: registry_person_allocate('%s'): allocating new person, sizeof(PERSON)=%zu", (person_guid)?person_guid:"", sizeof(PERSON));
 534
 535         p = malloc(sizeof(PERSON));
 536         if(!p) fatal("Registry: cannot allocate memory for new person.");
 537
 538         if(!person_guid) {
 539                 for (; ;) {
 540                         uuid_t uuid;
 541                         uuid_generate(uuid);
 542                         uuid_unparse_lower(uuid, p->guid);
 543
 544                         debug(D_REGISTRY, "Registry: Checking if the generated person guid '%s' is unique", p->guid);
 545                         if (!dictionary_get(registry.persons, p->guid)) {
 546                                 debug(D_REGISTRY, "Registry: generated person guid '%s' is unique", p->guid);
 547                                 break;
 548                         }
 549                         else
 550                                 info("Registry: generated person guid '%s' found in the registry. Retrying...", p->guid);
 551                 }
 552         }
 553         else
 554                 strncpyz(p->guid, person_guid, 36);
 555
 556         debug(D_REGISTRY, "Registry: registry_person_allocate('%s'): creating dictionary of urls", p->guid);
 557         p->urls = dictionary_create(DICTIONARY_FLAGS);
 558
 559         p->first_t = p->last_t = when;
 560         p->usages = 0;
 561
 562         registry.persons_memory += sizeof(PERSON);
 563
 564         registry.persons_count++;
 565         dictionary_set(registry.persons, p->guid, p, sizeof(PERSON));
 566
 567         return p;
 568 }
 569
 570
 571 // 1. validate person GUID
 572 // 2. if it is valid, find it
 573 // 3. if it is not valid, create a new one
 574 // 4. return it
 575 static inline PERSON *registry_person_get(const char *person_guid, time_t when) {
 576         PERSON *p = NULL;
 577
 578         registry_persons_lock();
 579
 580         if(person_guid && *person_guid) {
 581                 char buf[36 + 1];
 582                 // validate it is a GUID
 583                 if(unlikely(registry_regenerate_guid(person_guid, buf) == -1))
 584                         info("Registry: person guid '%s' is not a valid guid. Ignoring it.", person_guid);
 585                 else {
 586                         person_guid = buf;
 587                         p = registry_person_find(person_guid);
 588                         if(!p) person_guid = NULL;
 589                 }
 590         }
 591
 592         if(!p) p = registry_person_allocate(NULL, when);
 593
 594         registry_persons_unlock();
 595
 596         return p;
 597 }
 598
 599 // ----------------------------------------------------------------------------
 600 // LINKING OF OBJECTS
 601
 602 static inline PERSON_URL *registry_person_link_to_url(PERSON *p, MACHINE *m, URL *u, char *name, size_t namelen, time_t when) {
 603         debug(D_REGISTRY, "registry_person_link_to_url('%s', '%s', '%s'): searching for URL in person", p->guid, m->guid, u->url);
 604
 605         registry_person_urls_lock(p);
 606
 607         PERSON_URL *pu = dictionary_get(p->urls, u->url);
 608         if(!pu) {
 609                 debug(D_REGISTRY, "registry_person_link_to_url('%s', '%s', '%s'): not found", p->guid, m->guid, u->url);
 610                 pu = registry_person_url_allocate(p, m, u, name, namelen, when);
 611                 registry.persons_urls_count++;
 612         }
 613         else {
 614                 debug(D_REGISTRY, "registry_person_link_to_url('%s', '%s', '%s'): found", p->guid, m->guid, u->url);
 615                 pu->usages++;
 616                 if(likely(pu->last_t < (uint32_t)when)) pu->last_t = when;
 617
 618                 if(pu->machine != m) {
 619                         MACHINE_URL *mu = dictionary_get(pu->machine->urls, u->url);
 620                         if(mu) {
 621                                 info("registry_person_link_to_url('%s', '%s', '%s'): URL switched machines (old was '%s') - expiring it from previous machine.",
 622                                          p->guid, m->guid, u->url, pu->machine->guid);
 623                                 mu->flags |= REGISTRY_URL_FLAGS_EXPIRED;
 624                         }
 625                         else {
 626                                 info("registry_person_link_to_url('%s', '%s', '%s'): URL switched machines (old was '%s') - but the URL is not linked to the old machine.",
 627                                          p->guid, m->guid, u->url, pu->machine->guid);
 628                         }
 629
 630                         pu->machine->links--;
 631                         pu->machine = m;
 632                 }
 633
 634                 if(strcmp(pu->name, name)) {
 635                         // the name of the PERSON_URL has changed !
 636                         pu = registry_person_url_reallocate(p, m, u, name, namelen, when, pu);
 637                 }
 638         }
 639
 640         p->usages++;
 641         if(likely(p->last_t < (uint32_t)when)) p->last_t = when;
 642
 643         if(pu->flags & REGISTRY_URL_FLAGS_EXPIRED) {
 644                 info("registry_person_link_to_url('%s', '%s', '%s'): accessing an expired URL. Re-enabling URL.", p->guid, m->guid, u->url);
 645                 pu->flags &= ~REGISTRY_URL_FLAGS_EXPIRED;
 646         }
 647
 648         registry_person_urls_unlock(p);
 649
 650         return pu;
 651 }
 652
 653 static inline MACHINE_URL *registry_machine_link_to_url(PERSON *p, MACHINE *m, URL *u, time_t when) {
 654         debug(D_REGISTRY, "registry_machine_link_to_url('%s', '%s', '%s'): searching for URL in machine", p->guid, m->guid, u->url);
 655
 656         registry_machine_urls_lock(m);
 657
 658         MACHINE_URL *mu = dictionary_get(m->urls, u->url);
 659         if(!mu) {
 660                 debug(D_REGISTRY, "registry_machine_link_to_url('%s', '%s', '%s'): not found", p->guid, m->guid, u->url);
 661                 mu = registry_machine_url_allocate(m, u, when);
 662                 registry.machines_urls_count++;
 663         }
 664         else {
 665                 debug(D_REGISTRY, "registry_machine_link_to_url('%s', '%s', '%s'): found", p->guid, m->guid, u->url);
 666                 mu->usages++;
 667                 if(likely(mu->last_t < (uint32_t)when)) mu->last_t = when;
 668         }
 669
 670         //debug(D_REGISTRY, "registry_machine_link_to_url('%s', '%s', '%s'): indexing person in machine", p->guid, m->guid, u->url);
 671         //dictionary_set(mu->persons, p->guid, p, sizeof(PERSON));
 672
 673         m->usages++;
 674         if(likely(m->last_t < (uint32_t)when)) m->last_t = when;
 675
 676         if(mu->flags & REGISTRY_URL_FLAGS_EXPIRED) {
 677                 info("registry_machine_link_to_url('%s', '%s', '%s'): accessing an expired URL.", p->guid, m->guid, u->url);
 678                 mu->flags &= ~REGISTRY_URL_FLAGS_EXPIRED;
 679         }
 680
 681         registry_machine_urls_unlock(m);
 682
 683         return mu;
 684 }
 685
 686 // ----------------------------------------------------------------------------
 687 // REGISTRY LOG LOAD/SAVE
 688
 689 static inline int registry_should_save_db(void) {
 690         debug(D_REGISTRY, "log entries %llu, max %llu", registry.log_count, registry.save_registry_every_entries);
 691         return registry.log_count > registry.save_registry_every_entries;
 692 }
 693
 694 static inline void registry_log(const char action, PERSON *p, MACHINE *m, URL *u, char *name) {
 695         if(likely(registry.log_fp)) {
 696                 // we lock only if the file is open
 697                 // to allow replaying the log at registry_log_load()
 698                 registry_log_lock();
 699
 700                 if(unlikely(fprintf(registry.log_fp, "%c\t%08x\t%s\t%s\t%s\t%s\n",
 701                                 action,
 702                                 p->last_t,
 703                                 p->guid,
 704                                 m->guid,
 705                                 name,
 706                                 u->url) < 0))
 707                         error("Registry: failed to save log. Registry data may be lost in case of abnormal restart.");
 708
 709                 // we increase the counter even on failures
 710                 // so that the registry will be saved periodically
 711                 registry.log_count++;
 712
 713                 registry_log_unlock();
 714
 715                 // this must be outside the log_lock(), or a deadlock will happen.
 716                 // registry_save() checks the same inside the log_lock, so only
 717                 // one thread will save the db
 718                 if(unlikely(registry_should_save_db()))
 719                         registry_save();
 720         }
 721 }
 722
 723 static inline int registry_log_open_nolock(void) {
 724         if(registry.log_fp)
 725                 fclose(registry.log_fp);
 726
 727         registry.log_fp = fopen(registry.log_filename, "a");
 728
 729         if(registry.log_fp) {
 730                 if (setvbuf(registry.log_fp, NULL, _IOLBF, 0) != 0)
 731                         error("Cannot set line buffering on registry log file.");
 732                 return 0;
 733         }
 734
 735         error("Cannot open registry log file '%s'. Registry data will be lost in case of netdata or server crash.", registry.log_filename);
 736         return -1;
 737 }
 738
 739 static inline void registry_log_close_nolock(void) {
 740         if(registry.log_fp) {
 741                 fclose(registry.log_fp);
 742                 registry.log_fp = NULL;
 743         }
 744 }
 745
 746 static inline void registry_log_recreate_nolock(void) {
 747         if(registry.log_fp != NULL) {
 748                 registry_log_close_nolock();
 749
 750                 // open it with truncate
 751                 registry.log_fp = fopen(registry.log_filename, "w");
 752                 if(registry.log_fp) fclose(registry.log_fp);
 753                 else error("Cannot truncate registry log '%s'", registry.log_filename);
 754
 755                 registry.log_fp = NULL;
 756
 757                 registry_log_open_nolock();
 758         }
 759 }
 760
 761 int registry_log_load(void) {
 762         char *s, buf[4096 + 1];
 763         size_t line = -1;
 764
 765         // closing the log is required here
 766         // otherwise we will append to it the values we read
 767         registry_log_close_nolock();
 768
 769         debug(D_REGISTRY, "Registry: loading active db from: %s", registry.log_filename);
 770         FILE *fp = fopen(registry.log_filename, "r");
 771         if(!fp)
 772                 error("Registry: cannot open registry file: %s", registry.log_filename);
 773         else {
 774                 line = 0;
 775                 size_t len = 0;
 776                 while ((s = fgets_trim_len(buf, 4096, fp, &len))) {
 777                         line++;
 778
 779                         switch (s[0]) {
 780                                 case 'A': // accesses
 781                                 case 'D': // deletes
 782
 783                                         // verify it is valid
 784                                         if (unlikely(len < 85 || s[1] != '\t' || s[10] != '\t' || s[47] != '\t' || s[84] != '\t')) {
 785                                                 error("Registry: log line %zu is wrong (len = %zu).", line, len);
 786                                                 continue;
 787                                         }
 788                                         s[1] = s[10] = s[47] = s[84] = '\0';
 789
 790                                         // get the variables
 791                                         time_t when = strtoul(&s[2], NULL, 16);
 792                                         char *person_guid = &s[11];
 793                                         char *machine_guid = &s[48];
 794                                         char *name = &s[85];
 795
 796                                         // skip the name to find the url
 797                                         char *url = name;
 798                                         while(*url && *url != '\t') url++;
 799                                         if(!*url) {
 800                                                 error("Registry: log line %zu does not have a url.", line);
 801                                                 continue;
 802                                         }
 803                                         *url++ = '\0';
 804
 805                                         // make sure the person exists
 806                                         // without this, a new person guid will be created
 807                                         PERSON *p = registry_person_find(person_guid);
 808                                         if(!p) p = registry_person_allocate(person_guid, when);
 809
 810                                         if(s[0] == 'A')
 811                                                 registry_request_access(p->guid, machine_guid, url, name, when);
 812                                         else
 813                                                 registry_request_delete(p->guid, machine_guid, url, name, when);
 814
 815                                         break;
 816
 817                                 default:
 818                                         error("Registry: ignoring line %zu of filename '%s': %s.", line, registry.log_filename, s);
 819                                         break;
 820                         }
 821                 }
 822
 823                 fclose(fp);
 824         }
 825
 826         // open the log again
 827         registry_log_open_nolock();
 828
 829         return line;
 830 }
 831
 832
 833 // ----------------------------------------------------------------------------
 834 // REGISTRY REQUESTS
 835
 836 PERSON *registry_request_access(char *person_guid, char *machine_guid, char *url, char *name, time_t when) {
 837         debug(D_REGISTRY, "registry_request_access('%s', '%s', '%s'): NEW REQUEST", (person_guid)?person_guid:"", machine_guid, url);
 838
 839         MACHINE *m = registry_machine_get(machine_guid, when);
 840         if(!m) return NULL;
 841
 842         // make sure the name is valid
 843         size_t namelen;
 844         name = registry_fix_machine_name(name, &namelen);
 845
 846         size_t urllen;
 847         url = registry_fix_url(url, &urllen);
 848
 849         URL *u = registry_url_get(url, urllen);
 850         PERSON *p = registry_person_get(person_guid, when);
 851
 852         registry_person_link_to_url(p, m, u, name, namelen, when);
 853         registry_machine_link_to_url(p, m, u, when);
 854
 855         registry_log('A', p, m, u, name);
 856
 857         registry.usages_count++;
 858         return p;
 859 }
 860
 861 // verify the person, the machine and the URL exist in our DB
 862 PERSON_URL *registry_verify_request(char *person_guid, char *machine_guid, char *url, PERSON **pp, MACHINE **mm) {
 863         char pbuf[36 + 1], mbuf[36 + 1];
 864
 865         if(!person_guid || !*person_guid || !machine_guid || !*machine_guid || !url || !*url) {
 866                 info("Registry Request Verification: invalid request! person: '%s', machine '%s', url '%s'", person_guid?person_guid:"UNSET", machine_guid?machine_guid:"UNSET", url?url:"UNSET");
 867                 return NULL;
 868         }
 869
 870         // normalize the url
 871         url = registry_fix_url(url, NULL);
 872
 873         // make sure the person GUID is valid
 874         if(registry_regenerate_guid(person_guid, pbuf) == -1) {
 875                 info("Registry Request Verification: invalid person GUID, person: '%s', machine '%s', url '%s'", person_guid, machine_guid, url);
 876                 return NULL;
 877         }
 878         person_guid = pbuf;
 879
 880         // make sure the machine GUID is valid
 881         if(registry_regenerate_guid(machine_guid, mbuf) == -1) {
 882                 info("Registry Request Verification: invalid machine GUID, person: '%s', machine '%s', url '%s'", person_guid, machine_guid, url);
 883                 return NULL;
 884         }
 885         machine_guid = mbuf;
 886
 887         // make sure the machine exists
 888         MACHINE *m = registry_machine_find(machine_guid);
 889         if(!m) {
 890                 info("Registry Request Verification: machine not found, person: '%s', machine '%s', url '%s'", person_guid, machine_guid, url);
 891                 return NULL;
 892         }
 893         if(mm) *mm = m;
 894
 895         // make sure the person exist
 896         PERSON *p = registry_person_find(person_guid);
 897         if(!p) {
 898                 info("Registry Request Verification: person not found, person: '%s', machine '%s', url '%s'", person_guid, machine_guid, url);
 899                 return NULL;
 900         }
 901         if(pp) *pp = p;
 902
 903         PERSON_URL *pu = dictionary_get(p->urls, url);
 904         if(!pu) {
 905                 info("Registry Request Verification: URL not found for person, person: '%s', machine '%s', url '%s'", person_guid, machine_guid, url);
 906                 return NULL;
 907         }
 908         return pu;
 909 }
 910
 911 PERSON *registry_request_delete(char *person_guid, char *machine_guid, char *url, char *delete_url, time_t when) {
 912         (void)when;
 913
 914         PERSON *p = NULL;
 915         MACHINE *m = NULL;
 916         PERSON_URL *pu = registry_verify_request(person_guid, machine_guid, url, &p, &m);
 917         if(!pu || !p || !m) return NULL;
 918
 919         // normalize the url
 920         delete_url = registry_fix_url(delete_url, NULL);
 921
 922         // make sure the user is not deleting the url it uses
 923         if(!strcmp(delete_url, pu->url->url)) {
 924                 info("Registry Delete Request: delete URL is the one currently accessed, person: '%s', machine '%s', url '%s', delete url '%s'", p->guid, m->guid, pu->url->url, delete_url);
 925                 return NULL;
 926         }
 927
 928         registry_person_urls_lock(p);
 929
 930         PERSON_URL *dpu = dictionary_get(p->urls, delete_url);
 931         if(!dpu) {
 932                 info("Registry Delete Request: URL not found for person: '%s', machine '%s', url '%s', delete url '%s'", p->guid, m->guid, pu->url->url, delete_url);
 933                 registry_person_urls_unlock(p);
 934                 return NULL;
 935         }
 936
 937         registry_log('D', p, m, pu->url, dpu->url->url);
 938
 939         dictionary_del(p->urls, dpu->url->url);
 940         registry_url_unlink_nolock(dpu->url);
 941         free(dpu);
 942
 943         registry_person_urls_unlock(p);
 944         return p;
 945 }
 946
 947
 948 // a structure to pass to the dictionary_get_all() callback handler
 949 struct machine_request_callback_data {
 950         MACHINE *find_this_machine;
 951         PERSON_URL *result;
 952 };
 953
 954 // the callback function
 955 // this will be run for every PERSON_URL of this PERSON
 956 int machine_request_callback(void *entry, void *data) {
 957         PERSON_URL *mypu = (PERSON_URL *)entry;
 958         struct machine_request_callback_data *myrdata = (struct machine_request_callback_data *)data;
 959
 960         if(mypu->machine == myrdata->find_this_machine) {
 961                 myrdata->result = mypu;
 962                 return -1; // this will also stop the walk through
 963         }
 964
 965         return 0; // continue
 966 }
 967
 968 MACHINE *registry_request_machine(char *person_guid, char *machine_guid, char *url, char *request_machine, time_t when) {
 969         (void)when;
 970
 971         char mbuf[36 + 1];
 972
 973         PERSON *p = NULL;
 974         MACHINE *m = NULL;
 975         PERSON_URL *pu = registry_verify_request(person_guid, machine_guid, url, &p, &m);
 976         if(!pu || !p || !m) return NULL;
 977
 978         // make sure the machine GUID is valid
 979         if(registry_regenerate_guid(request_machine, mbuf) == -1) {
 980                 info("Registry Machine URLs request: invalid machine GUID, person: '%s', machine '%s', url '%s', request machine '%s'", p->guid, m->guid, pu->url->url, request_machine);
 981                 return NULL;
 982         }
 983         request_machine = mbuf;
 984
 985         // make sure the machine exists
 986         m = registry_machine_find(request_machine);
 987         if(!m) {
 988                 info("Registry Machine URLs request: machine not found, person: '%s', machine '%s', url '%s', request machine '%s'", p->guid, m->guid, pu->url->url, request_machine);
 989                 return NULL;
 990         }
 991
 992         // Verify the user has in the past accessed this machine
 993         // We will walk through the PERSON_URLs to find the machine
 994         // linking to our machine
 995
 996         // a structure to pass to the dictionary_get_all() callback handler
 997         struct machine_request_callback_data rdata = { m, NULL };
 998
 999         // request a walk through on the dictionary
1000         // no need for locking here, the underlying dictionary has its own
1001         dictionary_get_all(p->urls, machine_request_callback, &rdata);
1002
1003         if(rdata.result)
1004                 return m;
1005
1006         return NULL;
1007 }
1008
1009
1010 // ----------------------------------------------------------------------------
1011 // REGISTRY JSON generation
1012
1013 #define REGISTRY_STATUS_OK "ok"
1014 #define REGISTRY_STATUS_FAILED "failed"
1015 #define REGISTRY_STATUS_DISABLED "disabled"
1016
1017 int registry_verify_cookies_redirects(void) {
1018         return registry.verify_cookies_redirects;
1019 }
1020
1021 const char *registry_to_announce(void) {
1022         return registry.registry_to_announce;
1023 }
1024
1025 void registry_set_cookie(struct web_client *w, const char *guid) {
1026         char edate[100];
1027         time_t et = time(NULL) + registry.persons_expiration;
1028         struct tm etmbuf, *etm = gmtime_r(&et, &etmbuf);
1029         strftime(edate, sizeof(edate), "%a, %d %b %Y %H:%M:%S %Z", etm);
1030
1031         snprintfz(w->cookie1, COOKIE_MAX, NETDATA_REGISTRY_COOKIE_NAME "=%s; Expires=%s", guid, edate);
1032
1033         if(registry.registry_domain && registry.registry_domain[0])
1034                 snprintfz(w->cookie2, COOKIE_MAX, NETDATA_REGISTRY_COOKIE_NAME "=%s; Domain=%s; Expires=%s", guid, registry.registry_domain, edate);
1035 }
1036
1037 static inline void registry_set_person_cookie(struct web_client *w, PERSON *p) {
1038         registry_set_cookie(w, p->guid);
1039 }
1040
1041 static inline void registry_json_header(struct web_client *w, const char *action, const char *status) {
1042         buffer_flush(w->response.data);
1043         w->response.data->contenttype = CT_APPLICATION_JSON;
1044         buffer_sprintf(w->response.data, "{\n\t\"action\": \"%s\",\n\t\"status\": \"%s\",\n\t\"hostname\": \"%s\",\n\t\"machine_guid\": \"%s\"",
1045                                    action, status, registry.hostname, registry.machine_guid);
1046 }
1047
1048 static inline void registry_json_footer(struct web_client *w) {
1049         buffer_strcat(w->response.data, "\n}\n");
1050 }
1051
1052 int registry_request_hello_json(struct web_client *w) {
1053         registry_json_header(w, "hello", REGISTRY_STATUS_OK);
1054
1055         buffer_sprintf(w->response.data, ",\n\t\"registry\": \"%s\"",
1056                                    registry.registry_to_announce);
1057
1058         registry_json_footer(w);
1059         return 200;
1060 }
1061
1062 static inline int registry_json_disabled(struct web_client *w, const char *action) {
1063         registry_json_header(w, action, REGISTRY_STATUS_DISABLED);
1064
1065         buffer_sprintf(w->response.data, ",\n\t\"registry\": \"%s\"",
1066                                    registry.registry_to_announce);
1067
1068         registry_json_footer(w);
1069         return 200;
1070 }
1071
1072 // structure used be the callbacks below
1073 struct registry_json_walk_person_urls_callback {
1074         PERSON *p;
1075         MACHINE *m;
1076         struct web_client *w;
1077         int count;
1078 };
1079
1080 // callback for rendering PERSON_URLs
1081 static inline int registry_json_person_url_callback(void *entry, void *data) {
1082         PERSON_URL *pu = (PERSON_URL *)entry;
1083         struct registry_json_walk_person_urls_callback *c = (struct registry_json_walk_person_urls_callback *)data;
1084         struct web_client *w = c->w;
1085
1086         if(unlikely(c->count++))
1087                 buffer_strcat(w->response.data, ",");
1088
1089         buffer_sprintf(w->response.data, "\n\t\t[ \"%s\", \"%s\", %u000, %u, \"%s\" ]",
1090                                    pu->machine->guid, pu->url->url, pu->last_t, pu->usages, pu->name);
1091
1092         return 1;
1093 }
1094
1095 // callback for rendering MACHINE_URLs
1096 static inline int registry_json_machine_url_callback(void *entry, void *data) {
1097         MACHINE_URL *mu = (MACHINE_URL *)entry;
1098         struct registry_json_walk_person_urls_callback *c = (struct registry_json_walk_person_urls_callback *)data;
1099         struct web_client *w = c->w;
1100         MACHINE *m = c->m;
1101
1102         if(unlikely(c->count++))
1103                 buffer_strcat(w->response.data, ",");
1104
1105         buffer_sprintf(w->response.data, "\n\t\t[ \"%s\", \"%s\", %u000, %u ]",
1106                                    m->guid, mu->url->url, mu->last_t, mu->usages);
1107
1108         return 1;
1109 }
1110
1111
1112 // the main method for registering an access
1113 int registry_request_access_json(struct web_client *w, char *person_guid, char *machine_guid, char *url, char *name, time_t when) {
1114         if(!registry.enabled)
1115                 return registry_json_disabled(w, "access");
1116
1117         PERSON *p = registry_request_access(person_guid, machine_guid, url, name, when);
1118         if(!p) {
1119                 registry_json_header(w, "access", REGISTRY_STATUS_FAILED);
1120                 registry_json_footer(w);
1121                 return 412;
1122         }
1123
1124         // set the cookie
1125         registry_set_person_cookie(w, p);
1126
1127         // generate the response
1128         registry_json_header(w, "access", REGISTRY_STATUS_OK);
1129
1130         buffer_sprintf(w->response.data, ",\n\t\"person_guid\": \"%s\",\n\t\"urls\": [", p->guid);
1131         struct registry_json_walk_person_urls_callback c = { p, NULL, w, 0 };
1132         dictionary_get_all(p->urls, registry_json_person_url_callback, &c);
1133         buffer_strcat(w->response.data, "\n\t]\n");
1134
1135         registry_json_footer(w);
1136         return 200;
1137 }
1138
1139 // the main method for deleting a URL from a person
1140 int registry_request_delete_json(struct web_client *w, char *person_guid, char *machine_guid, char *url, char *delete_url, time_t when) {
1141         if(!registry.enabled)
1142                 return registry_json_disabled(w, "delete");
1143
1144         PERSON *p = registry_request_delete(person_guid, machine_guid, url, delete_url, when);
1145         if(!p) {
1146                 registry_json_header(w, "delete", REGISTRY_STATUS_FAILED);
1147                 registry_json_footer(w);
1148                 return 412;
1149         }
1150
1151         // generate the response
1152         registry_json_header(w, "delete", REGISTRY_STATUS_OK);
1153         registry_json_footer(w);
1154         return 200;
1155 }
1156
1157 // the main method for searching the URLs of a netdata
1158 int registry_request_search_json(struct web_client *w, char *person_guid, char *machine_guid, char *url, char *request_machine, time_t when) {
1159         if(!registry.enabled)
1160                 return registry_json_disabled(w, "search");
1161
1162         MACHINE *m = registry_request_machine(person_guid, machine_guid, url, request_machine, when);
1163         if(!m) {
1164                 registry_json_header(w, "search", REGISTRY_STATUS_FAILED);
1165                 registry_json_footer(w);
1166                 return 404;
1167         }
1168
1169         registry_json_header(w, "search", REGISTRY_STATUS_OK);
1170
1171         buffer_strcat(w->response.data, ",\n\t\"urls\": [");
1172         struct registry_json_walk_person_urls_callback c = { NULL, m, w, 0 };
1173         dictionary_get_all(m->urls, registry_json_machine_url_callback, &c);
1174         buffer_strcat(w->response.data, "\n\t]\n");
1175
1176         registry_json_footer(w);
1177         return 200;
1178 }
1179
1180 // structure used be the callbacks below
1181 struct registry_person_url_callback_verify_machine_exists_data {
1182         MACHINE *m;
1183         int count;
1184 };
1185
1186 int registry_person_url_callback_verify_machine_exists(void *entry, void *data) {
1187         struct registry_person_url_callback_verify_machine_exists_data *d = (struct registry_person_url_callback_verify_machine_exists_data *)data;
1188         PERSON_URL *pu = (PERSON_URL *)entry;
1189         MACHINE *m = d->m;
1190
1191         if(pu->machine == m)
1192                 d->count++;
1193
1194         return 0;
1195 }
1196
1197 // the main method for switching user identity
1198 int registry_request_switch_json(struct web_client *w, char *person_guid, char *machine_guid, char *url, char *new_person_guid, time_t when) {
1199         (void)url;
1200         (void)when;
1201
1202         if(!registry.enabled)
1203                 return registry_json_disabled(w, "switch");
1204
1205         PERSON *op = registry_person_find(person_guid);
1206         if(!op) {
1207                 registry_json_header(w, "switch", REGISTRY_STATUS_FAILED);
1208                 registry_json_footer(w);
1209                 return 430;
1210         }
1211
1212         PERSON *np = registry_person_find(new_person_guid);
1213         if(!np) {
1214                 registry_json_header(w, "switch", REGISTRY_STATUS_FAILED);
1215                 registry_json_footer(w);
1216                 return 431;
1217         }
1218
1219         MACHINE *m = registry_machine_find(machine_guid);
1220         if(!m) {
1221                 registry_json_header(w, "switch", REGISTRY_STATUS_FAILED);
1222                 registry_json_footer(w);
1223                 return 432;
1224         }
1225
1226         struct registry_person_url_callback_verify_machine_exists_data data = { m, 0 };
1227
1228         // verify the old person has access to this machine
1229         dictionary_get_all(op->urls, registry_person_url_callback_verify_machine_exists, &data);
1230         if(!data.count) {
1231                 registry_json_header(w, "switch", REGISTRY_STATUS_FAILED);
1232                 registry_json_footer(w);
1233                 return 433;
1234         }
1235
1236         // verify the new person has access to this machine
1237         data.count = 0;
1238         dictionary_get_all(np->urls, registry_person_url_callback_verify_machine_exists, &data);
1239         if(!data.count) {
1240                 registry_json_header(w, "switch", REGISTRY_STATUS_FAILED);
1241                 registry_json_footer(w);
1242                 return 434;
1243         }
1244
1245         // set the cookie of the new person
1246         // the user just switched identity
1247         registry_set_person_cookie(w, np);
1248
1249         // generate the response
1250         registry_json_header(w, "switch", REGISTRY_STATUS_OK);
1251         buffer_sprintf(w->response.data, ",\n\t\"person_guid\": \"%s\"", np->guid);
1252         registry_json_footer(w);
1253         return 200;
1254 }
1255
1256
1257 // ----------------------------------------------------------------------------
1258 // REGISTRY THIS MACHINE UNIQUE ID
1259
1260 char *registry_get_this_machine_guid(void) {
1261         if(likely(registry.machine_guid[0]))
1262                 return registry.machine_guid;
1263
1264         // read it from disk
1265         int fd = open(registry.machine_guid_filename, O_RDONLY);
1266         if(fd != -1) {
1267                 char buf[36 + 1];
1268                 if(read(fd, buf, 36) != 36)
1269                         error("Failed to read machine GUID from '%s'", registry.machine_guid_filename);
1270                 else {
1271                         buf[36] = '\0';
1272                         if(registry_regenerate_guid(buf, registry.machine_guid) == -1) {
1273                                 error("Failed to validate machine GUID '%s' from '%s'. Ignoring it - this might mean this netdata will appear as duplicate in the registry.",
1274                                           buf, registry.machine_guid_filename);
1275
1276                                 registry.machine_guid[0] = '\0';
1277                         }
1278                 }
1279                 close(fd);
1280         }
1281
1282         // generate a new one?
1283         if(!registry.machine_guid[0]) {
1284                 uuid_t uuid;
1285
1286                 uuid_generate_time(uuid);
1287                 uuid_unparse_lower(uuid, registry.machine_guid);
1288                 registry.machine_guid[36] = '\0';
1289
1290                 // save it
1291                 fd = open(registry.machine_guid_filename, O_WRONLY|O_CREAT|O_TRUNC, 444);
1292                 if(fd == -1)
1293                         fatal("Cannot create unique machine id file '%s'. Please fix this.", registry.machine_guid_filename);
1294
1295                 if(write(fd, registry.machine_guid, 36) != 36)
1296                         fatal("Cannot write the unique machine id file '%s'. Please fix this.", registry.machine_guid_filename);
1297
1298                 close(fd);
1299         }
1300
1301         return registry.machine_guid;
1302 }
1303
1304
1305 // ----------------------------------------------------------------------------
1306 // REGISTRY LOAD/SAVE
1307
1308 int registry_machine_save_url(void *entry, void *file) {
1309         MACHINE_URL *mu = entry;
1310         FILE *fp = file;
1311
1312         debug(D_REGISTRY, "Registry: registry_machine_save_url('%s')", mu->url->url);
1313
1314         int ret = fprintf(fp, "V\t%08x\t%08x\t%08x\t%02x\t%s\n",
1315                         mu->first_t,
1316                         mu->last_t,
1317                         mu->usages,
1318                         mu->flags,
1319                         mu->url->url
1320         );
1321
1322         // error handling is done at registry_save()
1323
1324         return ret;
1325 }
1326
1327 int registry_machine_save(void *entry, void *file) {
1328         MACHINE *m = entry;
1329         FILE *fp = file;
1330
1331         debug(D_REGISTRY, "Registry: registry_machine_save('%s')", m->guid);
1332
1333         int ret = fprintf(fp, "M\t%08x\t%08x\t%08x\t%s\n",
1334                         m->first_t,
1335                         m->last_t,
1336                         m->usages,
1337                         m->guid
1338         );
1339
1340         if(ret >= 0) {
1341                 int ret2 = dictionary_get_all(m->urls, registry_machine_save_url, fp);
1342                 if(ret2 < 0) return ret2;
1343                 ret += ret2;
1344         }
1345
1346         // error handling is done at registry_save()
1347
1348         return ret;
1349 }
1350
1351 static inline int registry_person_save_url(void *entry, void *file) {
1352         PERSON_URL *pu = entry;
1353         FILE *fp = file;
1354
1355         debug(D_REGISTRY, "Registry: registry_person_save_url('%s')", pu->url->url);
1356
1357         int ret = fprintf(fp, "U\t%08x\t%08x\t%08x\t%02x\t%s\t%s\t%s\n",
1358                         pu->first_t,
1359                         pu->last_t,
1360                         pu->usages,
1361                         pu->flags,
1362                         pu->machine->guid,
1363                         pu->name,
1364                         pu->url->url
1365         );
1366
1367         // error handling is done at registry_save()
1368
1369         return ret;
1370 }
1371
1372 static inline int registry_person_save(void *entry, void *file) {
1373         PERSON *p = entry;
1374         FILE *fp = file;
1375
1376         debug(D_REGISTRY, "Registry: registry_person_save('%s')", p->guid);
1377
1378         int ret = fprintf(fp, "P\t%08x\t%08x\t%08x\t%s\n",
1379                         p->first_t,
1380                         p->last_t,
1381                         p->usages,
1382                         p->guid
1383         );
1384
1385         if(ret >= 0) {
1386                 int ret2 = dictionary_get_all(p->urls, registry_person_save_url, fp);
1387                 if (ret2 < 0) return ret2;
1388                 ret += ret2;
1389         }
1390
1391         // error handling is done at registry_save()
1392
1393         return ret;
1394 }
1395
1396 int registry_save(void) {
1397         if(!registry.enabled) return -1;
1398
1399         // make sure the log is not updated
1400         registry_log_lock();
1401
1402         if(unlikely(!registry_should_save_db())) {
1403                 registry_log_unlock();
1404                 return -2;
1405         }
1406
1407         char tmp_filename[FILENAME_MAX + 1];
1408         char old_filename[FILENAME_MAX + 1];
1409
1410         snprintfz(old_filename, FILENAME_MAX, "%s.old", registry.db_filename);
1411         snprintfz(tmp_filename, FILENAME_MAX, "%s.tmp", registry.db_filename);
1412
1413         debug(D_REGISTRY, "Registry: Creating file '%s'", tmp_filename);
1414         FILE *fp = fopen(tmp_filename, "w");
1415         if(!fp) {
1416                 error("Registry: Cannot create file: %s", tmp_filename);
1417                 registry_log_unlock();
1418                 return -1;
1419         }
1420
1421         // dictionary_get_all() has its own locking, so this is safe to do
1422
1423         debug(D_REGISTRY, "Saving all machines");
1424         int bytes1 = dictionary_get_all(registry.machines, registry_machine_save, fp);
1425         if(bytes1 < 0) {
1426                 error("Registry: Cannot save registry machines - return value %d", bytes1);
1427                 fclose(fp);
1428                 registry_log_unlock();
1429                 return bytes1;
1430         }
1431         debug(D_REGISTRY, "Registry: saving machines took %d bytes", bytes1);
1432
1433         debug(D_REGISTRY, "Saving all persons");
1434         int bytes2 = dictionary_get_all(registry.persons, registry_person_save, fp);
1435         if(bytes2 < 0) {
1436                 error("Registry: Cannot save registry persons - return value %d", bytes2);
1437                 fclose(fp);
1438                 registry_log_unlock();
1439                 return bytes2;
1440         }
1441         debug(D_REGISTRY, "Registry: saving persons took %d bytes", bytes2);
1442
1443         // save the totals
1444         fprintf(fp, "T\t%016llx\t%016llx\t%016llx\t%016llx\t%016llx\t%016llx\n",
1445                         registry.persons_count,
1446                         registry.machines_count,
1447                         registry.usages_count + 1, // this is required - it is lost on db rotation
1448                         registry.urls_count,
1449                         registry.persons_urls_count,
1450                         registry.machines_urls_count
1451         );
1452
1453         fclose(fp);
1454
1455         errno = 0;
1456
1457         // remove the .old db
1458         debug(D_REGISTRY, "Registry: Removing old db '%s'", old_filename);
1459         if(unlink(old_filename) == -1 && errno != ENOENT)
1460                 error("Registry: cannot remove old registry file '%s'", old_filename);
1461
1462         // rename the db to .old
1463         debug(D_REGISTRY, "Registry: Link current db '%s' to .old: '%s'", registry.db_filename, old_filename);
1464         if(link(registry.db_filename, old_filename) == -1 && errno != ENOENT)
1465                 error("Registry: cannot move file '%s' to '%s'. Saving registry DB failed!", tmp_filename, registry.db_filename);
1466
1467         else {
1468                 // remove the database (it is saved in .old)
1469                 debug(D_REGISTRY, "Registry: removing db '%s'", registry.db_filename);
1470                 if (unlink(registry.db_filename) == -1 && errno != ENOENT)
1471                         error("Registry: cannot remove old registry file '%s'", registry.db_filename);
1472
1473                 // move the .tmp to make it active
1474                 debug(D_REGISTRY, "Registry: linking tmp db '%s' to active db '%s'", tmp_filename, registry.db_filename);
1475                 if (link(tmp_filename, registry.db_filename) == -1) {
1476                         error("Registry: cannot move file '%s' to '%s'. Saving registry DB failed!", tmp_filename,
1477                                   registry.db_filename);
1478
1479                         // move the .old back
1480                         debug(D_REGISTRY, "Registry: linking old db '%s' to active db '%s'", old_filename, registry.db_filename);
1481                         if(link(old_filename, registry.db_filename) == -1)
1482                                 error("Registry: cannot move file '%s' to '%s'. Recovering the old registry DB failed!", old_filename, registry.db_filename);
1483                 }
1484                 else {
1485                         debug(D_REGISTRY, "Registry: removing tmp db '%s'", tmp_filename);
1486                         if(unlink(tmp_filename) == -1)
1487                                 error("Registry: cannot remove tmp registry file '%s'", tmp_filename);
1488
1489                         // it has been moved successfully
1490                         // discard the current registry log
1491                         registry_log_recreate_nolock();
1492
1493                         registry.log_count = 0;
1494                 }
1495         }
1496
1497         // continue operations
1498         registry_log_unlock();
1499
1500         return -1;
1501 }
1502
1503 static inline size_t registry_load(void) {
1504         char *s, buf[4096 + 1];
1505         PERSON *p = NULL;
1506         MACHINE *m = NULL;
1507         URL *u = NULL;
1508         size_t line = 0;
1509
1510         debug(D_REGISTRY, "Registry: loading active db from: '%s'", registry.db_filename);
1511         FILE *fp = fopen(registry.db_filename, "r");
1512         if(!fp) {
1513                 error("Registry: cannot open registry file: '%s'", registry.db_filename);
1514                 return 0;
1515         }
1516
1517         size_t len = 0;
1518         buf[4096] = '\0';
1519         while((s = fgets_trim_len(buf, 4096, fp, &len))) {
1520                 line++;
1521
1522                 debug(D_REGISTRY, "Registry: read line %zu to length %zu: %s", line, len, s);
1523                 switch(*s) {
1524                         case 'T': // totals
1525                                 if(unlikely(len != 103 || s[1] != '\t' || s[18] != '\t' || s[35] != '\t' || s[52] != '\t' || s[69] != '\t' || s[86] != '\t' || s[103] != '\0')) {
1526                                         error("Registry totals line %zu is wrong (len = %zu).", line, len);
1527                                         continue;
1528                                 }
1529                                 registry.persons_count = strtoull(&s[2], NULL, 16);
1530                                 registry.machines_count = strtoull(&s[19], NULL, 16);
1531                                 registry.usages_count = strtoull(&s[36], NULL, 16);
1532                                 registry.urls_count = strtoull(&s[53], NULL, 16);
1533                                 registry.persons_urls_count = strtoull(&s[70], NULL, 16);
1534                                 registry.machines_urls_count = strtoull(&s[87], NULL, 16);
1535                                 break;
1536
1537                         case 'P': // person
1538                                 m = NULL;
1539                                 // verify it is valid
1540                                 if(unlikely(len != 65 || s[1] != '\t' || s[10] != '\t' || s[19] != '\t' || s[28] != '\t' || s[65] != '\0')) {
1541                                         error("Registry person line %zu is wrong (len = %zu).", line, len);
1542                                         continue;
1543                                 }
1544
1545                                 s[1] = s[10] = s[19] = s[28] = '\0';
1546                                 p = registry_person_allocate(&s[29], strtoul(&s[2], NULL, 16));
1547                                 p->last_t = strtoul(&s[11], NULL, 16);
1548                                 p->usages = strtoul(&s[20], NULL, 16);
1549                                 debug(D_REGISTRY, "Registry loaded person '%s', first: %u, last: %u, usages: %u", p->guid, p->first_t, p->last_t, p->usages);
1550                                 break;
1551
1552                         case 'M': // machine
1553                                 p = NULL;
1554                                 // verify it is valid
1555                                 if(unlikely(len != 65 || s[1] != '\t' || s[10] != '\t' || s[19] != '\t' || s[28] != '\t' || s[65] != '\0')) {
1556                                         error("Registry person line %zu is wrong (len = %zu).", line, len);
1557                                         continue;
1558                                 }
1559
1560                                 s[1] = s[10] = s[19] = s[28] = '\0';
1561                                 m = registry_machine_allocate(&s[29], strtoul(&s[2], NULL, 16));
1562                                 m->last_t = strtoul(&s[11], NULL, 16);
1563                                 m->usages = strtoul(&s[20], NULL, 16);
1564                                 debug(D_REGISTRY, "Registry loaded machine '%s', first: %u, last: %u, usages: %u", m->guid, m->first_t, m->last_t, m->usages);
1565                                 break;
1566
1567                         case 'U': // person URL
1568                                 if(unlikely(!p)) {
1569                                         error("Registry: ignoring line %zu, no person loaded: %s", line, s);
1570                                         continue;
1571                                 }
1572
1573                                 // verify it is valid
1574                                 if(len < 69 || s[1] != '\t' || s[10] != '\t' || s[19] != '\t' || s[28] != '\t' || s[31] != '\t' || s[68] != '\t') {
1575                                         error("Registry person URL line %zu is wrong (len = %zu).", line, len);
1576                                         continue;
1577                                 }
1578
1579                                 s[1] = s[10] = s[19] = s[28] = s[31] = s[68] = '\0';
1580
1581                                 // skip the name to find the url
1582                                 char *url = &s[69];
1583                                 while(*url && *url != '\t') url++;
1584                                 if(!*url) {
1585                                         error("Registry person URL line %zu does not have a url.", line);
1586                                         continue;
1587                                 }
1588                                 *url++ = '\0';
1589
1590                                 u = registry_url_allocate_nolock(url, strlen(url));
1591
1592                                 time_t first_t = strtoul(&s[2], NULL, 16);
1593
1594                                 m = registry_machine_find(&s[32]);
1595                                 if(!m) m = registry_machine_allocate(&s[32], first_t);
1596
1597                                 PERSON_URL *pu = registry_person_url_allocate(p, m, u, &s[69], strlen(&s[69]), first_t);
1598                                 pu->last_t = strtoul(&s[11], NULL, 16);
1599                                 pu->usages = strtoul(&s[20], NULL, 16);
1600                                 pu->flags = strtoul(&s[29], NULL, 16);
1601                                 debug(D_REGISTRY, "Registry loaded person URL '%s' with name '%s' of machine '%s', first: %u, last: %u, usages: %u, flags: %02x", u->url, pu->name, m->guid, pu->first_t, pu->last_t, pu->usages, pu->flags);
1602                                 break;
1603
1604                         case 'V': // machine URL
1605                                 if(unlikely(!m)) {
1606                                         error("Registry: ignoring line %zu, no machine loaded: %s", line, s);
1607                                         continue;
1608                                 }
1609
1610                                 // verify it is valid
1611                                 if(len < 32 || s[1] != '\t' || s[10] != '\t' || s[19] != '\t' || s[28] != '\t' || s[31] != '\t') {
1612                                         error("Registry person URL line %zu is wrong (len = %zu).", line, len);
1613                                         continue;
1614                                 }
1615
1616                                 s[1] = s[10] = s[19] = s[28] = s[31] = '\0';
1617                                 u = registry_url_allocate_nolock(&s[32], strlen(&s[32]));
1618
1619                                 MACHINE_URL *mu = registry_machine_url_allocate(m, u, strtoul(&s[2], NULL, 16));
1620                                 mu->last_t = strtoul(&s[11], NULL, 16);
1621                                 mu->usages = strtoul(&s[20], NULL, 16);
1622                                 mu->flags = strtoul(&s[29], NULL, 16);
1623                                 debug(D_REGISTRY, "Registry loaded machine URL '%s', machine '%s', first: %u, last: %u, usages: %u, flags: %02x", u->url, m->guid, mu->first_t, mu->last_t, mu->usages, mu->flags);
1624                                 break;
1625
1626                         default:
1627                                 error("Registry: ignoring line %zu of filename '%s': %s.", line, registry.db_filename, s);
1628                                 break;
1629                 }
1630         }
1631         fclose(fp);
1632
1633         return line;
1634 }
1635
1636 // ----------------------------------------------------------------------------
1637 // REGISTRY
1638
1639 int registry_init(void) {
1640         char filename[FILENAME_MAX + 1];
1641
1642         // registry enabled?
1643         registry.enabled = config_get_boolean("registry", "enabled", 0);
1644
1645         if(mkdir(VARLIB_DIR, 0755) == -1 && errno != EEXIST)
1646                 error("Cannot create directory '" VARLIB_DIR "'");
1647
1648         // pathnames
1649         registry.pathname = config_get("registry", "registry db directory", VARLIB_DIR "/registry");
1650         if(mkdir(registry.pathname, 0755) == -1 && errno != EEXIST) {
1651                 error("Cannot create directory '%s'. Registry disabled.", registry.pathname);
1652                 registry.enabled = 0;
1653                 return -1;
1654         }
1655
1656         // filenames
1657         snprintfz(filename, FILENAME_MAX, "%s/netdata.public.unique.id", registry.pathname);
1658         registry.machine_guid_filename = config_get("registry", "netdata unique id file", filename);
1659         registry_get_this_machine_guid();
1660
1661         snprintfz(filename, FILENAME_MAX, "%s/registry.db", registry.pathname);
1662         registry.db_filename = config_get("registry", "registry db file", filename);
1663
1664         snprintfz(filename, FILENAME_MAX, "%s/registry-log.db", registry.pathname);
1665         registry.log_filename = config_get("registry", "registry log file", filename);
1666
1667         // configuration options
1668         registry.save_registry_every_entries = config_get_number("registry", "registry save db every new entries", 1000000);
1669         registry.persons_expiration = config_get_number("registry", "registry expire idle persons days", 365) * 86400;
1670         registry.registry_domain = config_get("registry", "registry domain", "");
1671         registry.registry_to_announce = config_get("registry", "registry to announce", "https://registry.my-netdata.io");
1672         registry.hostname = config_get("registry", "registry hostname", config_get("global", "hostname", hostname));
1673         registry.verify_cookies_redirects = config_get_boolean("registry", "verify browser cookies support", 1);
1674
1675         registry.max_url_length = config_get_number("registry", "max URL length", 1024);
1676         if(registry.max_url_length < 10) {
1677                 registry.max_url_length = 10;
1678                 config_set_number("registry", "max URL length", registry.max_url_length);
1679         }
1680
1681         registry.max_name_length = config_get_number("registry", "max URL name length", 50);
1682         if(registry.max_name_length < 10) {
1683                 registry.max_name_length = 10;
1684                 config_set_number("registry", "max URL name length", registry.max_name_length);
1685         }
1686
1687         // initialize entries counters
1688         registry.persons_count = 0;
1689         registry.machines_count = 0;
1690         registry.usages_count = 0;
1691         registry.urls_count = 0;
1692         registry.persons_urls_count = 0;
1693         registry.machines_urls_count = 0;
1694
1695         // initialize memory counters
1696         registry.persons_memory = 0;
1697         registry.machines_memory = 0;
1698         registry.urls_memory = 0;
1699         registry.persons_urls_memory = 0;
1700         registry.machines_urls_memory = 0;
1701
1702         // initialize locks
1703         pthread_mutex_init(&registry.persons_lock, NULL);
1704         pthread_mutex_init(&registry.machines_lock, NULL);
1705         pthread_mutex_init(&registry.urls_lock, NULL);
1706         pthread_mutex_init(&registry.person_urls_lock, NULL);
1707         pthread_mutex_init(&registry.machine_urls_lock, NULL);
1708
1709         // create dictionaries
1710         registry.persons = dictionary_create(DICTIONARY_FLAGS);
1711         registry.machines = dictionary_create(DICTIONARY_FLAGS);
1712         registry.urls = dictionary_create(DICTIONARY_FLAGS);
1713
1714         // load the registry database
1715         if(registry.enabled) {
1716                 registry_log_open_nolock();
1717                 registry_load();
1718                 registry_log_load();
1719         }
1720
1721         return 0;
1722 }
1723
1724 void registry_free(void) {
1725         if(!registry.enabled) return;
1726
1727         // we need to destroy the dictionaries ourselves
1728         // since the dictionaries use memory we allocated
1729
1730         while(registry.persons->values_index.root) {
1731                 PERSON *p = ((NAME_VALUE *)registry.persons->values_index.root)->value;
1732
1733                 // fprintf(stderr, "\nPERSON: '%s', first: %u, last: %u, usages: %u\n", p->guid, p->first_t, p->last_t, p->usages);
1734
1735                 while(p->urls->values_index.root) {
1736                         PERSON_URL *pu = ((NAME_VALUE *)p->urls->values_index.root)->value;
1737
1738                         // fprintf(stderr, "\tURL: '%s', first: %u, last: %u, usages: %u, flags: 0x%02x\n", pu->url->url, pu->first_t, pu->last_t, pu->usages, pu->flags);
1739
1740                         debug(D_REGISTRY, "Registry: deleting url '%s' from person '%s'", pu->url->url, p->guid);
1741                         dictionary_del(p->urls, pu->url->url);
1742
1743                         debug(D_REGISTRY, "Registry: unlinking url '%s' from person", pu->url->url);
1744                         registry_url_unlink_nolock(pu->url);
1745
1746                         debug(D_REGISTRY, "Registry: freeing person url");
1747                         free(pu);
1748                 }
1749
1750                 debug(D_REGISTRY, "Registry: deleting person '%s' from persons registry", p->guid);
1751                 dictionary_del(registry.persons, p->guid);
1752
1753                 debug(D_REGISTRY, "Registry: destroying URL dictionary of person '%s'", p->guid);
1754                 dictionary_destroy(p->urls);
1755
1756                 debug(D_REGISTRY, "Registry: freeing person '%s'", p->guid);
1757                 free(p);
1758         }
1759
1760         while(registry.machines->values_index.root) {
1761                 MACHINE *m = ((NAME_VALUE *)registry.machines->values_index.root)->value;
1762
1763                 // fprintf(stderr, "\nMACHINE: '%s', first: %u, last: %u, usages: %u\n", m->guid, m->first_t, m->last_t, m->usages);
1764
1765                 while(m->urls->values_index.root) {
1766                         MACHINE_URL *mu = ((NAME_VALUE *)m->urls->values_index.root)->value;
1767
1768                         // fprintf(stderr, "\tURL: '%s', first: %u, last: %u, usages: %u, flags: 0x%02x\n", mu->url->url, mu->first_t, mu->last_t, mu->usages, mu->flags);
1769
1770                         //debug(D_REGISTRY, "Registry: destroying persons dictionary from url '%s'", mu->url->url);
1771                         //dictionary_destroy(mu->persons);
1772
1773                         debug(D_REGISTRY, "Registry: deleting url '%s' from person '%s'", mu->url->url, m->guid);
1774                         dictionary_del(m->urls, mu->url->url);
1775
1776                         debug(D_REGISTRY, "Registry: unlinking url '%s' from machine", mu->url->url);
1777                         registry_url_unlink_nolock(mu->url);
1778
1779                         debug(D_REGISTRY, "Registry: freeing machine url");
1780                         free(mu);
1781                 }
1782
1783                 debug(D_REGISTRY, "Registry: deleting machine '%s' from machines registry", m->guid);
1784                 dictionary_del(registry.machines, m->guid);
1785
1786                 debug(D_REGISTRY, "Registry: destroying URL dictionary of machine '%s'", m->guid);
1787                 dictionary_destroy(m->urls);
1788
1789                 debug(D_REGISTRY, "Registry: freeing machine '%s'", m->guid);
1790                 free(m);
1791         }
1792
1793         // and free the memory of remaining dictionary structures
1794
1795         debug(D_REGISTRY, "Registry: destroying persons dictionary");
1796         dictionary_destroy(registry.persons);
1797
1798         debug(D_REGISTRY, "Registry: destroying machines dictionary");
1799         dictionary_destroy(registry.machines);
1800
1801         debug(D_REGISTRY, "Registry: destroying urls dictionary");
1802         dictionary_destroy(registry.urls);
1803 }
1804
1805 // ----------------------------------------------------------------------------
1806 // STATISTICS
1807
1808 void registry_statistics(void) {
1809         if(!registry.enabled) return;
1810
1811         static RRDSET *sts = NULL, *stc = NULL, *stm = NULL;
1812
1813         if(!sts) sts = rrdset_find("netdata.registry_sessions");
1814         if(!sts) {
1815                 sts = rrdset_create("netdata", "registry_sessions", NULL, "registry", NULL, "NetData Registry Sessions", "session", 131000, rrd_update_every, RRDSET_TYPE_LINE);
1816
1817                 rrddim_add(sts, "sessions",  NULL,  1, 1, RRDDIM_ABSOLUTE);
1818         }
1819         else rrdset_next(sts);
1820
1821         rrddim_set(sts, "sessions", registry.usages_count);
1822         rrdset_done(sts);
1823
1824         // ------------------------------------------------------------------------
1825
1826         if(!stc) stc = rrdset_find("netdata.registry_entries");
1827         if(!stc) {
1828                 stc = rrdset_create("netdata", "registry_entries", NULL, "registry", NULL, "NetData Registry Entries", "entries", 131100, rrd_update_every, RRDSET_TYPE_LINE);
1829
1830                 rrddim_add(stc, "persons",        NULL,  1, 1, RRDDIM_ABSOLUTE);
1831                 rrddim_add(stc, "machines",       NULL,  1, 1, RRDDIM_ABSOLUTE);
1832                 rrddim_add(stc, "urls",           NULL,  1, 1, RRDDIM_ABSOLUTE);
1833                 rrddim_add(stc, "persons_urls",   NULL,  1, 1, RRDDIM_ABSOLUTE);
1834                 rrddim_add(stc, "machines_urls",  NULL,  1, 1, RRDDIM_ABSOLUTE);
1835         }
1836         else rrdset_next(stc);
1837
1838         rrddim_set(stc, "persons",       registry.persons_count);
1839         rrddim_set(stc, "machines",      registry.machines_count);
1840         rrddim_set(stc, "urls",          registry.urls_count);
1841         rrddim_set(stc, "persons_urls",  registry.persons_urls_count);
1842         rrddim_set(stc, "machines_urls", registry.machines_urls_count);
1843         rrdset_done(stc);
1844
1845         // ------------------------------------------------------------------------
1846
1847         if(!stm) stm = rrdset_find("netdata.registry_mem");
1848         if(!stm) {
1849                 stm = rrdset_create("netdata", "registry_mem", NULL, "registry", NULL, "NetData Registry Memory", "KB", 131300, rrd_update_every, RRDSET_TYPE_STACKED);
1850
1851                 rrddim_add(stm, "persons",        NULL,  1, 1024, RRDDIM_ABSOLUTE);
1852                 rrddim_add(stm, "machines",       NULL,  1, 1024, RRDDIM_ABSOLUTE);
1853                 rrddim_add(stm, "urls",           NULL,  1, 1024, RRDDIM_ABSOLUTE);
1854                 rrddim_add(stm, "persons_urls",   NULL,  1, 1024, RRDDIM_ABSOLUTE);
1855                 rrddim_add(stm, "machines_urls",  NULL,  1, 1024, RRDDIM_ABSOLUTE);
1856         }
1857         else rrdset_next(stm);
1858
1859         rrddim_set(stm, "persons",       registry.persons_memory + registry.persons_count * sizeof(NAME_VALUE) + sizeof(DICTIONARY));
1860         rrddim_set(stm, "machines",      registry.machines_memory + registry.machines_count * sizeof(NAME_VALUE) + sizeof(DICTIONARY));
1861         rrddim_set(stm, "urls",          registry.urls_memory + registry.urls_count * sizeof(NAME_VALUE) + sizeof(DICTIONARY));
1862         rrddim_set(stm, "persons_urls",  registry.persons_urls_memory + registry.persons_count * sizeof(DICTIONARY) + registry.persons_urls_count * sizeof(NAME_VALUE));
1863         rrddim_set(stm, "machines_urls", registry.machines_urls_memory + registry.machines_count * sizeof(DICTIONARY) + registry.machines_urls_count * sizeof(NAME_VALUE));
1864         rrdset_done(stm);
1865 }