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