src/plugin_tc.c

   1 #include "common.h"
   2
   3 #define RRD_TYPE_TC                 "tc"
   4 #define RRD_TYPE_TC_LEN             strlen(RRD_TYPE_TC)
   5
   6 // ----------------------------------------------------------------------------
   7 // /sbin/tc processor
   8 // this requires the script plugins.d/tc-qos-helper.sh
   9
  10 #define TC_LINE_MAX 1024
  11
  12 struct tc_class {
  13     avl avl;
  14
  15     char *id;
  16     uint32_t hash;
  17
  18     char *name;
  19
  20     char *leafid;
  21     uint32_t leaf_hash;
  22
  23     char *parentid;
  24     uint32_t parent_hash;
  25
  26     char hasparent;
  27     char isleaf;
  28     unsigned long long bytes;
  29     unsigned long long packets;
  30     unsigned long long dropped;
  31     unsigned long long overlimits;
  32     unsigned long long requeues;
  33     unsigned long long lended;
  34     unsigned long long borrowed;
  35     unsigned long long giants;
  36     unsigned long long tokens;
  37     unsigned long long ctokens;
  38
  39     RRDDIM *rd_bytes;
  40     RRDDIM *rd_packets;
  41     RRDDIM *rd_dropped;
  42
  43     char name_updated;
  44     char updated;   // updated bytes
  45     int seen;       // seen in the tc list (even without bytes)
  46
  47     struct tc_class *next;
  48     struct tc_class *prev;
  49 };
  50
  51 struct tc_device {
  52     avl avl;
  53
  54     char *id;
  55     uint32_t hash;
  56
  57     char *name;
  58     char *family;
  59
  60     char name_updated;
  61     char family_updated;
  62
  63     char enabled;
  64     char enabled_bytes;
  65     char enabled_packets;
  66     char enabled_dropped;
  67
  68     RRDSET *st_bytes;
  69     RRDSET *st_packets;
  70     RRDSET *st_dropped;
  71
  72     avl_tree classes_index;
  73
  74     struct tc_class *classes;
  75
  76     struct tc_device *next;
  77     struct tc_device *prev;
  78 };
  79
  80
  81 struct tc_device *tc_device_root = NULL;
  82
  83 // ----------------------------------------------------------------------------
  84 // tc_device index
  85
  86 static int tc_device_compare(void* a, void* b) {
  87     if(((struct tc_device *)a)->hash < ((struct tc_device *)b)->hash) return -1;
  88     else if(((struct tc_device *)a)->hash > ((struct tc_device *)b)->hash) return 1;
  89     else return strcmp(((struct tc_device *)a)->id, ((struct tc_device *)b)->id);
  90 }
  91
  92 avl_tree tc_device_root_index = {
  93         NULL,
  94         tc_device_compare
  95 };
  96
  97 #define tc_device_index_add(st) avl_insert(&tc_device_root_index, (avl *)(st))
  98 #define tc_device_index_del(st) avl_remove(&tc_device_root_index, (avl *)(st))
  99
 100 static inline struct tc_device *tc_device_index_find(const char *id, uint32_t hash) {
 101     struct tc_device tmp;
 102     tmp.id = (char *)id;
 103     tmp.hash = (hash)?hash:simple_hash(tmp.id);
 104
 105     return (struct tc_device *)avl_search(&(tc_device_root_index), (avl *)&tmp);
 106 }
 107
 108
 109 // ----------------------------------------------------------------------------
 110 // tc_class index
 111
 112 static int tc_class_compare(void* a, void* b) {
 113     if(((struct tc_class *)a)->hash < ((struct tc_class *)b)->hash) return -1;
 114     else if(((struct tc_class *)a)->hash > ((struct tc_class *)b)->hash) return 1;
 115     else return strcmp(((struct tc_class *)a)->id, ((struct tc_class *)b)->id);
 116 }
 117
 118 #define tc_class_index_add(st, rd) avl_insert(&((st)->classes_index), (avl *)(rd))
 119 #define tc_class_index_del(st, rd) avl_remove(&((st)->classes_index), (avl *)(rd))
 120
 121 static inline struct tc_class *tc_class_index_find(struct tc_device *st, const char *id, uint32_t hash) {
 122     struct tc_class tmp;
 123     tmp.id = (char *)id;
 124     tmp.hash = (hash)?hash:simple_hash(tmp.id);
 125
 126     return (struct tc_class *)avl_search(&(st->classes_index), (avl *) &tmp);
 127 }
 128
 129 // ----------------------------------------------------------------------------
 130
 131 static inline void tc_class_free(struct tc_device *n, struct tc_class *c) {
 132     if(c == n->classes) {
 133         if(c->next)
 134             n->classes = c->next;
 135         else
 136             n->classes = c->prev;
 137     }
 138     if(c->next) c->next->prev = c->prev;
 139     if(c->prev) c->prev->next = c->next;
 140
 141     debug(D_TC_LOOP, "Removing from device '%s' class '%s', parentid '%s', leafid '%s', seen=%d", n->id, c->id, c->parentid?c->parentid:"", c->leafid?c->leafid:"", c->seen);
 142
 143     tc_class_index_del(n, c);
 144
 145     freez(c->id);
 146     freez(c->name);
 147     freez(c->leafid);
 148     freez(c->parentid);
 149     freez(c);
 150 }
 151
 152 static inline void tc_device_classes_cleanup(struct tc_device *d) {
 153     static int cleanup_every = 999;
 154
 155     if(unlikely(cleanup_every > 0)) {
 156         cleanup_every = (int) config_get_number("plugin:tc", "cleanup unused classes every", 60);
 157         if(cleanup_every < 0) cleanup_every = -cleanup_every;
 158     }
 159
 160     d->name_updated = 0;
 161     d->family_updated = 0;
 162
 163     struct tc_class *c = d->classes;
 164     while(c) {
 165         if(unlikely(cleanup_every > 0 && c->seen >= cleanup_every)) {
 166             struct tc_class *nc = c->next;
 167             tc_class_free(d, c);
 168             c = nc;
 169         }
 170         else {
 171             c->updated = 0;
 172             c->name_updated = 0;
 173
 174             c = c->next;
 175         }
 176     }
 177 }
 178
 179 static inline void tc_device_commit(struct tc_device *d) {
 180     static int enable_new_interfaces = -1, enable_bytes = -1, enable_packets = -1, enable_dropped = -1;
 181
 182     if(unlikely(enable_new_interfaces == -1)) {
 183         enable_new_interfaces = config_get_boolean_ondemand("plugin:tc", "enable new interfaces detected at runtime", CONFIG_ONDEMAND_YES);
 184         enable_bytes          = config_get_boolean_ondemand("plugin:tc", "enable traffic charts for all interfaces", CONFIG_ONDEMAND_ONDEMAND);
 185         enable_packets        = config_get_boolean_ondemand("plugin:tc", "enable packets charts for all interfaces", CONFIG_ONDEMAND_ONDEMAND);
 186         enable_dropped        = config_get_boolean_ondemand("plugin:tc", "enable dropped charts for all interfaces", CONFIG_ONDEMAND_ONDEMAND);
 187     }
 188
 189     // we only need to add leaf classes
 190     struct tc_class *c, *x;
 191     unsigned long long bytes_sum = 0, packets_sum = 0, dropped_sum = 0;
 192     int active_classes = 0;
 193
 194     // set all classes
 195     for(c = d->classes ; c ; c = c->next) {
 196         c->isleaf = 1;
 197         c->hasparent = 0;
 198     }
 199
 200     // mark the classes as leafs and parents
 201     for(c = d->classes ; c ; c = c->next) {
 202         if(unlikely(!c->updated)) continue;
 203
 204         for(x = d->classes ; x ; x = x->next) {
 205             if(unlikely(!x->updated)) continue;
 206
 207             if(unlikely(c == x)) continue;
 208
 209             if(x->parentid && (
 210                 (               c->hash      == x->parent_hash && strcmp(c->id,     x->parentid) == 0) ||
 211                 (c->leafid   && c->leaf_hash == x->parent_hash && strcmp(c->leafid, x->parentid) == 0))) {
 212                 // debug(D_TC_LOOP, "TC: In device '%s', class '%s' (leafid: '%s') has as leaf class '%s' (parentid: '%s').", d->name?d->name:d->id, c->name?c->name:c->id, c->leafid?c->leafid:c->id, x->name?x->name:x->id, x->parentid?x->parentid:x->id);
 213                 c->isleaf = 0;
 214                 x->hasparent = 1;
 215             }
 216         }
 217     }
 218
 219     // debugging:
 220     /*
 221     for ( c = d->classes ; c ; c = c->next) {
 222         if(c->isleaf && c->hasparent) debug(D_TC_LOOP, "TC: Device %s, class %s, OK", d->name, c->id);
 223         else debug(D_TC_LOOP, "TC: Device %s, class %s, IGNORE (isleaf: %d, hasparent: %d, parent: %s)", d->name, c->id, c->isleaf, c->hasparent, c->parentid);
 224     }
 225     */
 226
 227     // we need at least a class
 228     for(c = d->classes ; c ; c = c->next) {
 229         // debug(D_TC_LOOP, "TC: Device '%s', class '%s', isLeaf=%d, HasParent=%d, Seen=%d", d->name?d->name:d->id, c->name?c->name:c->id, c->isleaf, c->hasparent, c->seen);
 230         if(!c->updated) continue;
 231         if(c->isleaf && c->hasparent) {
 232             active_classes++;
 233             bytes_sum += c->bytes;
 234             packets_sum += c->packets;
 235             dropped_sum += c->dropped;
 236         }
 237     }
 238
 239     if(unlikely(!active_classes)) {
 240         debug(D_TC_LOOP, "TC: Ignoring TC device '%s'. No leaf classes.", d->name?d->name:d->id);
 241         tc_device_classes_cleanup(d);
 242         return;
 243     }
 244
 245     if(unlikely(d->enabled == -1)) {
 246         char var_name[CONFIG_MAX_NAME + 1];
 247         snprintfz(var_name, CONFIG_MAX_NAME, "qos for %s", d->id);
 248         d->enabled         = config_get_boolean_ondemand("plugin:tc", var_name, enable_new_interfaces);
 249
 250         snprintfz(var_name, CONFIG_MAX_NAME, "traffic chart for %s", d->id);
 251         d->enabled_bytes   = config_get_boolean_ondemand("plugin:tc", var_name, enable_bytes);
 252
 253         snprintfz(var_name, CONFIG_MAX_NAME, "packets chart for %s", d->id);
 254         d->enabled_packets = config_get_boolean_ondemand("plugin:tc", var_name, enable_packets);
 255
 256         snprintfz(var_name, CONFIG_MAX_NAME, "dropped packets chart for %s", d->id);
 257         d->enabled_dropped = config_get_boolean_ondemand("plugin:tc", var_name, enable_dropped);
 258     }
 259
 260     if(likely(d->enabled)) {
 261         // --------------------------------------------------------------------
 262         // bytes
 263
 264         if(d->enabled_bytes == CONFIG_ONDEMAND_YES || (d->enabled_bytes == CONFIG_ONDEMAND_ONDEMAND && bytes_sum)) {
 265             d->enabled_bytes = CONFIG_ONDEMAND_YES;
 266
 267             if(unlikely(!d->st_bytes)) {
 268                 d->st_bytes = rrdset_find_bytype(RRD_TYPE_TC, d->id);
 269                 if(unlikely(!d->st_bytes)) {
 270                     debug(D_TC_LOOP, "TC: Creating new chart for device '%s'", d->name?d->name:d->id);
 271                     d->st_bytes = rrdset_create(RRD_TYPE_TC, d->id, d->name?d->name:d->id, d->family?d->family:d->id, RRD_TYPE_TC ".qos", "Class Usage", "kilobits/s", 7000, rrd_update_every, RRDSET_TYPE_STACKED);
 272                 }
 273             }
 274             else {
 275                 debug(D_TC_LOOP, "TC: Updating chart for device '%s'", d->name?d->name:d->id);
 276                 rrdset_next_plugins(d->st_bytes);
 277
 278                 if(unlikely(d->name_updated && d->name && strcmp(d->id, d->name) != 0)) {
 279                     rrdset_set_name(d->st_bytes, d->name);
 280                     d->name_updated = 0;
 281                 }
 282
 283                 // FIXME
 284                 // update the family
 285             }
 286
 287             for(c = d->classes ; c ; c = c->next) {
 288                 if(unlikely(!c->updated)) continue;
 289
 290                 if(c->isleaf && c->hasparent) {
 291                     c->seen++;
 292
 293                     if(unlikely(!c->rd_bytes)) {
 294                         c->rd_bytes = rrddim_find(d->st_bytes, c->id);
 295                         if(unlikely(!c->rd_bytes)) {
 296                             debug(D_TC_LOOP, "TC: Adding to chart '%s', dimension '%s' (name: '%s')", d->st_bytes->id, c->id, c->name);
 297
 298                             // new class, we have to add it
 299                             c->rd_bytes = rrddim_add(d->st_bytes, c->id, c->name?c->name:c->id, 8, 1024, RRDDIM_INCREMENTAL);
 300                         }
 301                         else debug(D_TC_LOOP, "TC: Updating chart '%s', dimension '%s'", d->st_bytes->id, c->id);
 302                     }
 303
 304                     rrddim_set_by_pointer(d->st_bytes, c->rd_bytes, c->bytes);
 305
 306                     // if it has a name, different to the id
 307                     if(unlikely(c->name_updated && c->name && strcmp(c->id, c->name) != 0)) {
 308                         // update the rrd dimension with the new name
 309                         debug(D_TC_LOOP, "TC: Setting chart '%s', dimension '%s' name to '%s'", d->st_bytes->id, c->rd_bytes->id, c->name);
 310                         rrddim_set_name(d->st_bytes, c->rd_bytes, c->name);
 311                     }
 312                 }
 313             }
 314             rrdset_done(d->st_bytes);
 315         }
 316
 317         // --------------------------------------------------------------------
 318         // packets
 319
 320         if(d->enabled_packets == CONFIG_ONDEMAND_YES || (d->enabled_packets == CONFIG_ONDEMAND_ONDEMAND && packets_sum)) {
 321             d->enabled_packets = CONFIG_ONDEMAND_YES;
 322
 323             if(unlikely(!d->st_packets)) {
 324                 char id[RRD_ID_LENGTH_MAX + 1];
 325                 char name[RRD_ID_LENGTH_MAX + 1];
 326                 snprintfz(id, RRD_ID_LENGTH_MAX, "%s_packets", d->id);
 327                 snprintfz(name, RRD_ID_LENGTH_MAX, "%s_packets", d->name?d->name:d->id);
 328
 329                 d->st_packets = rrdset_find_bytype(RRD_TYPE_TC, id);
 330                 if(unlikely(!d->st_packets)) {
 331                     debug(D_TC_LOOP, "TC: Creating new _packets chart for device '%s'", d->name?d->name:d->id);
 332                     d->st_packets = rrdset_create(RRD_TYPE_TC, id, name, d->family?d->family:d->id, RRD_TYPE_TC ".qos_packets", "Class Packets", "packets/s", 7010, rrd_update_every, RRDSET_TYPE_STACKED);
 333                 }
 334             }
 335             else {
 336                 debug(D_TC_LOOP, "TC: Updating _packets chart for device '%s'", d->name?d->name:d->id);
 337                 rrdset_next_plugins(d->st_packets);
 338
 339                 // FIXME
 340                 // update the family
 341             }
 342
 343             for(c = d->classes ; c ; c = c->next) {
 344                 if(unlikely(!c->updated)) continue;
 345
 346                 if(c->isleaf && c->hasparent) {
 347                     if(unlikely(!c->rd_packets)) {
 348                         c->rd_packets = rrddim_find(d->st_packets, c->id);
 349                         if(unlikely(!c->rd_packets)) {
 350                             debug(D_TC_LOOP, "TC: Adding to chart '%s', dimension '%s' (name: '%s')", d->st_packets->id, c->id, c->name);
 351
 352                             // new class, we have to add it
 353                             c->rd_packets = rrddim_add(d->st_packets, c->id, c->name?c->name:c->id, 1, 1, RRDDIM_INCREMENTAL);
 354                         }
 355                         else debug(D_TC_LOOP, "TC: Updating chart '%s', dimension '%s'", d->st_packets->id, c->id);
 356                     }
 357
 358                     rrddim_set_by_pointer(d->st_packets, c->rd_packets, c->packets);
 359
 360                     // if it has a name, different to the id
 361                     if(unlikely(c->name_updated && c->name && strcmp(c->id, c->name) != 0)) {
 362                         // update the rrd dimension with the new name
 363                         debug(D_TC_LOOP, "TC: Setting chart '%s', dimension '%s' name to '%s'", d->st_packets->id, c->rd_packets->id, c->name);
 364                         rrddim_set_name(d->st_packets, c->rd_packets, c->name);
 365                     }
 366                 }
 367             }
 368             rrdset_done(d->st_packets);
 369         }
 370
 371         // --------------------------------------------------------------------
 372         // dropped
 373
 374         if(d->enabled_dropped == CONFIG_ONDEMAND_YES || (d->enabled_dropped == CONFIG_ONDEMAND_ONDEMAND && dropped_sum)) {
 375             d->enabled_dropped = CONFIG_ONDEMAND_YES;
 376
 377             if(unlikely(!d->st_dropped)) {
 378                 char id[RRD_ID_LENGTH_MAX + 1];
 379                 char name[RRD_ID_LENGTH_MAX + 1];
 380                 snprintfz(id, RRD_ID_LENGTH_MAX, "%s_dropped", d->id);
 381                 snprintfz(name, RRD_ID_LENGTH_MAX, "%s_dropped", d->name?d->name:d->id);
 382
 383                 d->st_dropped = rrdset_find_bytype(RRD_TYPE_TC, id);
 384                 if(unlikely(!d->st_dropped)) {
 385                     debug(D_TC_LOOP, "TC: Creating new _dropped chart for device '%s'", d->name?d->name:d->id);
 386                     d->st_dropped = rrdset_create(RRD_TYPE_TC, id, name, d->family?d->family:d->id, RRD_TYPE_TC ".qos_dropped", "Class Dropped Packets", "packets/s", 7020, rrd_update_every, RRDSET_TYPE_STACKED);
 387                 }
 388             }
 389             else {
 390                 debug(D_TC_LOOP, "TC: Updating _dropped chart for device '%s'", d->name?d->name:d->id);
 391                 rrdset_next_plugins(d->st_dropped);
 392
 393                 // FIXME
 394                 // update the family
 395             }
 396
 397             for(c = d->classes ; c ; c = c->next) {
 398                 if(unlikely(!c->updated)) continue;
 399
 400                 if(c->isleaf && c->hasparent) {
 401                     if(unlikely(!c->rd_dropped)) {
 402                         c->rd_dropped = rrddim_find(d->st_dropped, c->id);
 403                         if(unlikely(!c->rd_dropped)) {
 404                             debug(D_TC_LOOP, "TC: Adding to chart '%s', dimension '%s' (name: '%s')", d->st_dropped->id, c->id, c->name);
 405
 406                             // new class, we have to add it
 407                             c->rd_dropped = rrddim_add(d->st_dropped, c->id, c->name?c->name:c->id, 1, 1, RRDDIM_INCREMENTAL);
 408                         }
 409                         else debug(D_TC_LOOP, "TC: Updating chart '%s', dimension '%s'", d->st_dropped->id, c->id);
 410                     }
 411
 412                     rrddim_set_by_pointer(d->st_dropped, c->rd_dropped, c->dropped);
 413
 414                     // if it has a name, different to the id
 415                     if(unlikely(c->name_updated && c->name && strcmp(c->id, c->name) != 0)) {
 416                         // update the rrd dimension with the new name
 417                         debug(D_TC_LOOP, "TC: Setting chart '%s', dimension '%s' name to '%s'", d->st_dropped->id, c->rd_dropped->id, c->name);
 418                         rrddim_set_name(d->st_dropped, c->rd_dropped, c->name);
 419                     }
 420                 }
 421             }
 422             rrdset_done(d->st_dropped);
 423         }
 424     }
 425
 426     tc_device_classes_cleanup(d);
 427 }
 428
 429 static inline void tc_device_set_class_name(struct tc_device *d, char *id, char *name)
 430 {
 431     struct tc_class *c = tc_class_index_find(d, id, 0);
 432     if(likely(c)) {
 433         freez(c->name);
 434         c->name = NULL;
 435
 436         if(likely(name && *name && strcmp(c->id, name) != 0)) {
 437             debug(D_TC_LOOP, "TC: Setting device '%s', class '%s' name to '%s'", d->id, id, name);
 438             c->name = strdupz(name);
 439             c->name_updated = 1;
 440         }
 441     }
 442 }
 443
 444 static inline void tc_device_set_device_name(struct tc_device *d, char *name) {
 445     freez(d->name);
 446     d->name = NULL;
 447
 448     if(likely(name && *name && strcmp(d->id, name) != 0)) {
 449         debug(D_TC_LOOP, "TC: Setting device '%s' name to '%s'", d->id, name);
 450         d->name = strdupz(name);
 451         d->name_updated = 1;
 452     }
 453 }
 454
 455 static inline void tc_device_set_device_family(struct tc_device *d, char *family) {
 456     freez(d->family);
 457     d->family = NULL;
 458
 459     if(likely(family && *family && strcmp(d->id, family) != 0)) {
 460         debug(D_TC_LOOP, "TC: Setting device '%s' family to '%s'", d->id, family);
 461         d->family = strdupz(family);
 462         d->family_updated = 1;
 463     }
 464     // no need for null termination - it is already null
 465 }
 466
 467 static inline struct tc_device *tc_device_create(char *id)
 468 {
 469     struct tc_device *d = tc_device_index_find(id, 0);
 470
 471     if(!d) {
 472         debug(D_TC_LOOP, "TC: Creating device '%s'", id);
 473
 474         d = callocz(1, sizeof(struct tc_device));
 475
 476         d->id = strdupz(id);
 477         d->hash = simple_hash(d->id);
 478         d->enabled = -1;
 479
 480         avl_init(&d->classes_index, tc_class_compare);
 481         tc_device_index_add(d);
 482
 483         if(!tc_device_root) {
 484             tc_device_root = d;
 485         }
 486         else {
 487             d->next = tc_device_root;
 488             tc_device_root->prev = d;
 489             tc_device_root = d;
 490         }
 491     }
 492
 493     return(d);
 494 }
 495
 496 static inline struct tc_class *tc_class_add(struct tc_device *n, char *id, char *parentid, char *leafid)
 497 {
 498     struct tc_class *c = tc_class_index_find(n, id, 0);
 499
 500     if(!c) {
 501         debug(D_TC_LOOP, "TC: Creating in device '%s', class id '%s', parentid '%s', leafid '%s'", n->id, id, parentid?parentid:"", leafid?leafid:"");
 502
 503         c = callocz(1, sizeof(struct tc_class));
 504
 505         if(n->classes) n->classes->prev = c;
 506         c->next = n->classes;
 507         n->classes = c;
 508
 509         c->id = strdupz(id);
 510         c->hash = simple_hash(c->id);
 511
 512         if(parentid && *parentid) {
 513             c->parentid = strdupz(parentid);
 514             c->parent_hash = simple_hash(c->parentid);
 515         }
 516
 517         if(leafid && *leafid) {
 518             c->leafid = strdupz(leafid);
 519             c->leaf_hash = simple_hash(c->leafid);
 520         }
 521
 522         tc_class_index_add(n, c);
 523     }
 524
 525     c->seen = 1;
 526
 527     return(c);
 528 }
 529
 530 static inline void tc_device_free(struct tc_device *n)
 531 {
 532     if(n->next) n->next->prev = n->prev;
 533     if(n->prev) n->prev->next = n->next;
 534     if(tc_device_root == n) {
 535         if(n->next) tc_device_root = n->next;
 536         else tc_device_root = n->prev;
 537     }
 538
 539     tc_device_index_del(n);
 540
 541     while(n->classes) tc_class_free(n, n->classes);
 542
 543     freez(n->id);
 544     freez(n->name);
 545     freez(n->family);
 546     freez(n);
 547 }
 548
 549 static inline void tc_device_free_all()
 550 {
 551     while(tc_device_root)
 552         tc_device_free(tc_device_root);
 553 }
 554
 555 #define MAX_WORDS 20
 556
 557 static inline int tc_space(char c) {
 558     switch(c) {
 559     case ' ':
 560     case '\t':
 561     case '\r':
 562     case '\n':
 563         return 1;
 564
 565     default:
 566         return 0;
 567     }
 568 }
 569
 570 static inline void tc_split_words(char *str, char **words, int max_words) {
 571     char *s = str;
 572     int i = 0;
 573
 574     // skip all white space
 575     while(tc_space(*s)) s++;
 576
 577     // store the first word
 578     words[i++] = s;
 579
 580     // while we have something
 581     while(*s) {
 582         // if it is a space
 583         if(unlikely(tc_space(*s))) {
 584
 585             // terminate the word
 586             *s++ = '\0';
 587
 588             // skip all white space
 589             while(tc_space(*s)) s++;
 590
 591             // if we reached the end, stop
 592             if(!*s) break;
 593
 594             // store the next word
 595             if(i < max_words) words[i++] = s;
 596             else break;
 597         }
 598         else s++;
 599     }
 600
 601     // terminate the words
 602     while(i < max_words) words[i++] = NULL;
 603 }
 604
 605 pid_t tc_child_pid = 0;
 606 void *tc_main(void *ptr) {
 607     (void)ptr;
 608
 609     info("TC thread created with task id %d", gettid());
 610
 611     if(pthread_setcanceltype(PTHREAD_CANCEL_DEFERRED, NULL) != 0)
 612         error("Cannot set pthread cancel type to DEFERRED.");
 613
 614     if(pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL) != 0)
 615         error("Cannot set pthread cancel state to ENABLE.");
 616
 617     struct rusage thread;
 618     RRDSET *stcpu = NULL, *sttime = NULL;
 619
 620     char buffer[TC_LINE_MAX+1] = "";
 621     char *words[MAX_WORDS] = { NULL };
 622
 623     uint32_t BEGIN_HASH = simple_hash("BEGIN");
 624     uint32_t END_HASH = simple_hash("END");
 625     uint32_t CLASS_HASH = simple_hash("class");
 626     uint32_t SENT_HASH = simple_hash("Sent");
 627     uint32_t LENDED_HASH = simple_hash("lended:");
 628     uint32_t TOKENS_HASH = simple_hash("tokens:");
 629     uint32_t SETDEVICENAME_HASH = simple_hash("SETDEVICENAME");
 630     uint32_t SETDEVICEGROUP_HASH = simple_hash("SETDEVICEGROUP");
 631     uint32_t SETCLASSNAME_HASH = simple_hash("SETCLASSNAME");
 632     uint32_t WORKTIME_HASH = simple_hash("WORKTIME");
 633 #ifdef DETACH_PLUGINS_FROM_NETDATA
 634     uint32_t MYPID_HASH = simple_hash("MYPID");
 635 #endif
 636     uint32_t first_hash;
 637
 638     snprintfz(buffer, TC_LINE_MAX, "%s/tc-qos-helper.sh", config_get("plugins", "plugins directory", PLUGINS_DIR));
 639     char *tc_script = config_get("plugin:tc", "script to run to get tc values", buffer);
 640
 641     for(;1;) {
 642         if(unlikely(netdata_exit)) break;
 643
 644         FILE *fp;
 645         struct tc_device *device = NULL;
 646         struct tc_class *class = NULL;
 647
 648         snprintfz(buffer, TC_LINE_MAX, "exec %s %d", tc_script, rrd_update_every);
 649         debug(D_TC_LOOP, "executing '%s'", buffer);
 650
 651         fp = mypopen(buffer, &tc_child_pid);
 652         if(unlikely(!fp)) {
 653             error("TC: Cannot popen(\"%s\", \"r\").", buffer);
 654             pthread_exit(NULL);
 655             return NULL;
 656         }
 657
 658         while(fgets(buffer, TC_LINE_MAX, fp) != NULL) {
 659             if(unlikely(netdata_exit)) break;
 660
 661             buffer[TC_LINE_MAX] = '\0';
 662             // debug(D_TC_LOOP, "TC: read '%s'", buffer);
 663
 664             tc_split_words(buffer, words, MAX_WORDS);
 665
 666             if(unlikely(!words[0] || !*words[0])) {
 667                 // debug(D_TC_LOOP, "empty line");
 668                 continue;
 669             }
 670             // else debug(D_TC_LOOP, "First word is '%s'", words[0]);
 671
 672             first_hash = simple_hash(words[0]);
 673
 674             if(unlikely(device && first_hash == CLASS_HASH && strcmp(words[0], "class") == 0)) {
 675                 // debug(D_TC_LOOP, "CLASS line on class id='%s', parent='%s', parentid='%s', leaf='%s', leafid='%s'", words[2], words[3], words[4], words[5], words[6]);
 676
 677                 // words[1] : class type
 678                 // words[2] : N:XX
 679                 // words[3] : parent or root
 680                 if(likely(words[1] && words[2] && words[3] && (strcmp(words[3], "parent") == 0 || strcmp(words[3], "root") == 0))) {
 681                     //char *type     = words[1];  // the class: htb, fq_codel, etc
 682
 683                     // we are only interested for HTB classes
 684                     //if(strcmp(type, "htb") != 0) continue;
 685
 686                     char *id       = words[2];  // the class major:minor
 687                     char *parent   = words[3];  // 'parent' or 'root'
 688                     char *parentid = words[4];  // the parent's id
 689                     char *leaf     = words[5];  // 'leaf'
 690                     char *leafid   = words[6];  // leafid
 691
 692                     if(strcmp(parent, "root") == 0) {
 693                         parentid = NULL;
 694                         leafid = NULL;
 695                     }
 696                     else if(!leaf || strcmp(leaf, "leaf") != 0)
 697                         leafid = NULL;
 698
 699                     char leafbuf[20 + 1] = "";
 700                     if(leafid && leafid[strlen(leafid) - 1] == ':') {
 701                         strncpyz(leafbuf, leafid, 20 - 1);
 702                         strcat(leafbuf, "1");
 703                         leafid = leafbuf;
 704                     }
 705
 706                     class = tc_class_add(device, id, parentid, leafid);
 707                 }
 708                 else {
 709                     // clear the last class
 710                     class = NULL;
 711                 }
 712             }
 713             else if(unlikely(first_hash == END_HASH && strcmp(words[0], "END") == 0)) {
 714                 // debug(D_TC_LOOP, "END line");
 715
 716                 if(likely(device)) {
 717                     if(pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, NULL) != 0)
 718                         error("Cannot set pthread cancel state to DISABLE.");
 719
 720                     tc_device_commit(device);
 721                     // tc_device_free(device);
 722
 723                     if(pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL) != 0)
 724                         error("Cannot set pthread cancel state to ENABLE.");
 725                 }
 726
 727                 device = NULL;
 728                 class = NULL;
 729             }
 730             else if(unlikely(first_hash == BEGIN_HASH && strcmp(words[0], "BEGIN") == 0)) {
 731                 // debug(D_TC_LOOP, "BEGIN line on device '%s'", words[1]);
 732
 733                 if(likely(words[1] && *words[1])) {
 734                     device = tc_device_create(words[1]);
 735                 }
 736                 else {
 737                     // tc_device_free(device);
 738                     device = NULL;
 739                 }
 740
 741                 class = NULL;
 742             }
 743             else if(unlikely(device && class && first_hash == SENT_HASH && strcmp(words[0], "Sent") == 0)) {
 744                 // debug(D_TC_LOOP, "SENT line '%s'", words[1]);
 745                 if(likely(words[1] && *words[1])) {
 746                     class->bytes = strtoull(words[1], NULL, 10);
 747                     class->updated = 1;
 748                 }
 749                 else {
 750                     class->updated = 0;
 751                 }
 752
 753                 if(likely(words[3] && *words[3]))
 754                     class->packets = strtoull(words[3], NULL, 10);
 755
 756                 if(likely(words[6] && *words[6]))
 757                     class->dropped = strtoull(words[6], NULL, 10);
 758
 759                 if(likely(words[8] && *words[8]))
 760                     class->overlimits = strtoull(words[8], NULL, 10);
 761
 762                 if(likely(words[10] && *words[10]))
 763                     class->requeues = strtoull(words[8], NULL, 10);
 764             }
 765             else if(unlikely(device && class && class->updated && first_hash == LENDED_HASH && strcmp(words[0], "lended:") == 0)) {
 766                 // debug(D_TC_LOOP, "LENDED line '%s'", words[1]);
 767                 if(likely(words[1] && *words[1]))
 768                     class->lended = strtoull(words[1], NULL, 10);
 769
 770                 if(likely(words[3] && *words[3]))
 771                     class->borrowed = strtoull(words[3], NULL, 10);
 772
 773                 if(likely(words[5] && *words[5]))
 774                     class->giants = strtoull(words[5], NULL, 10);
 775             }
 776             else if(unlikely(device && class && class->updated && first_hash == TOKENS_HASH && strcmp(words[0], "tokens:") == 0)) {
 777                 // debug(D_TC_LOOP, "TOKENS line '%s'", words[1]);
 778                 if(likely(words[1] && *words[1]))
 779                     class->tokens = strtoull(words[1], NULL, 10);
 780
 781                 if(likely(words[3] && *words[3]))
 782                     class->ctokens = strtoull(words[3], NULL, 10);
 783             }
 784             else if(unlikely(device && first_hash == SETDEVICENAME_HASH && strcmp(words[0], "SETDEVICENAME") == 0)) {
 785                 // debug(D_TC_LOOP, "SETDEVICENAME line '%s'", words[1]);
 786                 if(likely(words[1] && *words[1]))
 787                     tc_device_set_device_name(device, words[1]);
 788             }
 789             else if(unlikely(device && first_hash == SETDEVICEGROUP_HASH && strcmp(words[0], "SETDEVICEGROUP") == 0)) {
 790                 // debug(D_TC_LOOP, "SETDEVICEGROUP line '%s'", words[1]);
 791                 if(likely(words[1] && *words[1]))
 792                     tc_device_set_device_family(device, words[1]);
 793             }
 794             else if(unlikely(device && first_hash == SETCLASSNAME_HASH && strcmp(words[0], "SETCLASSNAME") == 0)) {
 795                 // debug(D_TC_LOOP, "SETCLASSNAME line '%s' '%s'", words[1], words[2]);
 796                 char *id    = words[1];
 797                 char *path  = words[2];
 798                 if(likely(id && *id && path && *path))
 799                     tc_device_set_class_name(device, id, path);
 800             }
 801             else if(unlikely(first_hash == WORKTIME_HASH && strcmp(words[0], "WORKTIME") == 0)) {
 802                 // debug(D_TC_LOOP, "WORKTIME line '%s' '%s'", words[1], words[2]);
 803                 getrusage(RUSAGE_THREAD, &thread);
 804
 805                 if(unlikely(!stcpu)) stcpu = rrdset_find("netdata.plugin_tc_cpu");
 806                 if(unlikely(!stcpu)) {
 807                     stcpu = rrdset_create("netdata", "plugin_tc_cpu", NULL, "tc.helper", NULL, "NetData TC CPU usage", "milliseconds/s", 135000, rrd_update_every, RRDSET_TYPE_STACKED);
 808                     rrddim_add(stcpu, "user",  NULL,  1, 1000, RRDDIM_INCREMENTAL);
 809                     rrddim_add(stcpu, "system", NULL, 1, 1000, RRDDIM_INCREMENTAL);
 810                 }
 811                 else rrdset_next(stcpu);
 812
 813                 rrddim_set(stcpu, "user"  , thread.ru_utime.tv_sec * 1000000ULL + thread.ru_utime.tv_usec);
 814                 rrddim_set(stcpu, "system", thread.ru_stime.tv_sec * 1000000ULL + thread.ru_stime.tv_usec);
 815                 rrdset_done(stcpu);
 816
 817                 if(unlikely(!sttime)) stcpu = rrdset_find("netdata.plugin_tc_time");
 818                 if(unlikely(!sttime)) {
 819                     sttime = rrdset_create("netdata", "plugin_tc_time", NULL, "tc.helper", NULL, "NetData TC script execution", "milliseconds/run", 135001, rrd_update_every, RRDSET_TYPE_AREA);
 820                     rrddim_add(sttime, "run_time",  "run time",  1, 1, RRDDIM_ABSOLUTE);
 821                 }
 822                 else rrdset_next(sttime);
 823
 824                 rrddim_set(sttime, "run_time", atoll(words[1]));
 825                 rrdset_done(sttime);
 826
 827             }
 828 #ifdef DETACH_PLUGINS_FROM_NETDATA
 829             else if(unlikely(first_hash == MYPID_HASH && (strcmp(words[0], "MYPID") == 0))) {
 830                 // debug(D_TC_LOOP, "MYPID line '%s'", words[1]);
 831                 char *id = words[1];
 832                 pid_t pid = atol(id);
 833
 834                 if(likely(pid)) tc_child_pid = pid;
 835
 836                 debug(D_TC_LOOP, "TC: Child PID is %d.", tc_child_pid);
 837             }
 838 #endif
 839             //else {
 840             //  debug(D_TC_LOOP, "IGNORED line");
 841             //}
 842         }
 843
 844         // fgets() failed or loop broke
 845         int code = mypclose(fp, tc_child_pid);
 846         tc_child_pid = 0;
 847
 848         if(unlikely(device)) {
 849             // tc_device_free(device);
 850             device = NULL;
 851             class = NULL;
 852         }
 853
 854         if(unlikely(netdata_exit)) {
 855             tc_device_free_all();
 856             pthread_exit(NULL);
 857             return NULL;
 858         }
 859
 860         if(code == 1 || code == 127) {
 861             // 1 = DISABLE
 862             // 127 = cannot even run it
 863             error("TC: tc-qos-helper.sh exited with code %d. Disabling it.", code);
 864
 865             tc_device_free_all();
 866             pthread_exit(NULL);
 867             return NULL;
 868         }
 869
 870         sleep((unsigned int) rrd_update_every);
 871     }
 872
 873     pthread_exit(NULL);
 874     return NULL;
 875 }