src/graph_instance.c

   1 #include <stdlib.h>
   2 #include <string.h>
   3 #include <errno.h>
   4 #include <time.h>
   5
   6 #include "graph_instance.h"
   7 #include "graph_ident.h"
   8 #include "graph_list.h"
   9 #include "common.h"
  10 #include "utils_params.h"
  11
  12 #include <fcgiapp.h>
  13 #include <fcgi_stdio.h>
  14
  15 struct graph_instance_s /* {{{ */
  16 {
  17   graph_ident_t *select;
  18
  19   graph_ident_t **files;
  20   size_t files_num;
  21
  22   graph_instance_t *next;
  23 }; /* }}} struct graph_instance_s */
  24
  25 struct def_callback_data_s
  26 {
  27   graph_instance_t *inst;
  28   str_array_t *args;
  29 };
  30 typedef struct def_callback_data_s def_callback_data_t;
  31
  32 /*
  33  * Private functions
  34  */
  35 /* Create one DEF for each data source in the file. Called by
  36  * "inst_get_default_defs" for each file. */
  37 static graph_def_t *ident_get_default_defs (graph_config_t *cfg, /* {{{ */
  38     graph_ident_t *ident, graph_def_t *def_head)
  39 {
  40   graph_def_t *defs = NULL;
  41   char *file;
  42   char **dses = NULL;
  43   size_t dses_num = 0;
  44   int status;
  45   size_t i;
  46
  47   if ((cfg == NULL) || (ident == NULL))
  48     return (def_head);
  49
  50   file = ident_to_file (ident);
  51   if (file == NULL)
  52   {
  53     fprintf (stderr, "ident_get_default_defs: ident_to_file failed\n");
  54     return (def_head);
  55   }
  56
  57   status = ds_list_from_rrd_file (file, &dses_num, &dses);
  58   if (status != 0)
  59   {
  60     free (file);
  61     return (def_head);
  62   }
  63
  64   for (i = 0; i < dses_num; i++)
  65   {
  66     graph_def_t *def;
  67
  68     def = def_search (def_head, ident, dses[i]);
  69     if (def != NULL)
  70       continue;
  71
  72     def = def_create (cfg, ident, dses[i]);
  73     if (def == NULL)
  74       continue;
  75
  76     if (defs == NULL)
  77       defs = def;
  78     else
  79       def_append (defs, def);
  80
  81     free (dses[i]);
  82   }
  83
  84   free (dses);
  85   free (file);
  86
  87   return (defs);
  88 } /* }}} int ident_get_default_defs */
  89
  90 /* Create one or more DEFs for each file in the graph instance. The number
  91  * depends on the number of data sources in each of the files. Called from
  92  * "inst_get_rrdargs" if no DEFs are available from the configuration.
  93  * */
  94 static graph_def_t *inst_get_default_defs (graph_config_t *cfg, /* {{{ */
  95     graph_instance_t *inst)
  96 {
  97   graph_def_t *defs = NULL;
  98   size_t i;
  99
 100   if ((cfg == NULL) || (inst == NULL))
 101     return (NULL);
 102
 103   for (i = 0; i < inst->files_num; i++)
 104   {
 105     graph_def_t *def;
 106
 107     def = ident_get_default_defs (cfg, inst->files[i], defs);
 108     if (def == NULL)
 109       continue;
 110
 111     if (defs == NULL)
 112       defs = def;
 113     else
 114       def_append (defs, def);
 115   }
 116
 117   return (defs);
 118 } /* }}} graph_def_t *inst_get_default_defs */
 119
 120 /* Called with each DEF in turn. Calls "def_get_rrdargs" with every appropriate
 121  * file / DEF pair. */
 122 static int gl_instance_get_rrdargs_cb (graph_def_t *def, void *user_data) /* {{{ */
 123 {
 124   def_callback_data_t *data = user_data;
 125   graph_instance_t *inst = data->inst;
 126   str_array_t *args = data->args;
 127
 128   size_t i;
 129
 130   for (i = 0; i < inst->files_num; i++)
 131   {
 132     if (!def_matches (def, inst->files[i]))
 133       continue;
 134
 135     def_get_rrdargs (def, inst->files[i], args);
 136   }
 137
 138   return (0);
 139 } /* }}} int gl_instance_get_rrdargs_cb */
 140
 141 static const char *get_part_from_param (const char *prim_key, /* {{{ */
 142     const char *sec_key)
 143 {
 144   const char *val;
 145
 146   val = param (prim_key);
 147   if (val != NULL)
 148     return (val);
 149
 150   return (param (sec_key));
 151 } /* }}} const char *get_part_from_param */
 152
 153 /*
 154  * Public functions
 155  */
 156 graph_instance_t *inst_create (graph_config_t *cfg, /* {{{ */
 157     const graph_ident_t *ident)
 158 {
 159   graph_instance_t *i;
 160   graph_ident_t *selector;
 161
 162   if ((cfg == NULL) || (ident == NULL))
 163     return (NULL);
 164
 165   i = malloc (sizeof (*i));
 166   if (i == NULL)
 167     return (NULL);
 168   memset (i, 0, sizeof (*i));
 169
 170   selector = graph_get_selector (cfg);
 171   if (selector == NULL)
 172   {
 173     fprintf (stderr, "inst_create: graph_get_selector failed\n");
 174     free (i);
 175     return (NULL);
 176   }
 177
 178   i->select = ident_copy_with_selector (selector, ident,
 179       IDENT_FLAG_REPLACE_ANY);
 180   if (i->select == NULL)
 181   {
 182     fprintf (stderr, "inst_create: ident_copy_with_selector failed\n");
 183     ident_destroy (selector);
 184     free (i);
 185     return (NULL);
 186   }
 187
 188   ident_destroy (selector);
 189
 190   i->files = NULL;
 191   i->files_num = 0;
 192
 193   i->next = NULL;
 194
 195   return (i);
 196 } /* }}} graph_instance_t *inst_create */
 197
 198 void inst_destroy (graph_instance_t *inst) /* {{{ */
 199 {
 200   graph_instance_t *next;
 201   size_t i;
 202
 203   if (inst == NULL)
 204     return;
 205
 206   next = inst->next;
 207
 208   ident_destroy (inst->select);
 209
 210   for (i = 0; i < inst->files_num; i++)
 211     ident_destroy (inst->files[i]);
 212   free (inst->files);
 213
 214   free (inst);
 215
 216   inst_destroy (next);
 217 } /* }}} void inst_destroy */
 218
 219 int inst_add_file (graph_instance_t *inst, /* {{{ */
 220     const graph_ident_t *file)
 221 {
 222   graph_ident_t **tmp;
 223
 224   tmp = realloc (inst->files, sizeof (*inst->files) * (inst->files_num + 1));
 225   if (tmp == NULL)
 226     return (ENOMEM);
 227   inst->files = tmp;
 228
 229   inst->files[inst->files_num] = ident_clone (file);
 230   if (inst->files[inst->files_num] == NULL)
 231     return (ENOMEM);
 232
 233   inst->files_num++;
 234
 235   return (0);
 236 } /* }}} int inst_add_file */
 237
 238 graph_instance_t *inst_get_selected (graph_config_t *cfg) /* {{{ */
 239 {
 240   const char *host = get_part_from_param ("inst_host", "host");
 241   const char *plugin = get_part_from_param ("inst_plugin", "plugin");
 242   const char *plugin_instance = get_part_from_param ("inst_plugin_instance", "plugin_instance");
 243   const char *type = get_part_from_param ("inst_type", "type");
 244   const char *type_instance = get_part_from_param ("inst_type_instance", "type_instance");
 245   graph_ident_t *ident;
 246   graph_instance_t *inst;
 247
 248   if (cfg == NULL)
 249     cfg = gl_graph_get_selected ();
 250
 251   if (cfg == NULL)
 252   {
 253     DEBUG ("inst_get_selected: cfg == NULL;\n");
 254     return (NULL);
 255   }
 256
 257   if ((host == NULL)
 258       || (plugin == NULL) || (plugin_instance == NULL)
 259       || (type == NULL) || (type_instance == NULL))
 260   {
 261     DEBUG ("inst_get_selected: A parameter is NULL.\n");
 262     return (NULL);
 263   }
 264
 265   ident = ident_create (host, plugin, plugin_instance, type, type_instance);
 266
 267   for (inst = graph_get_instances (cfg); inst != NULL; inst = inst->next)
 268   {
 269     if (ident_compare (ident, inst->select) != 0)
 270       continue;
 271
 272     ident_destroy (ident);
 273     return (inst);
 274   }
 275
 276   DEBUG ("inst_get_selected: No match found.\n");
 277   ident_destroy (ident);
 278   return (NULL);
 279 } /* }}} graph_instance_t *inst_get_selected */
 280
 281 int inst_get_rrdargs (graph_config_t *cfg, /* {{{ */
 282     graph_instance_t *inst,
 283     str_array_t *args)
 284 {
 285   def_callback_data_t data = { inst, args };
 286   graph_def_t *defs;
 287   int status;
 288
 289   if ((cfg == NULL) || (inst == NULL) || (args == NULL))
 290     return (EINVAL);
 291
 292   status = graph_get_rrdargs (cfg, inst, args);
 293   if (status != 0)
 294     return (status);
 295
 296   defs = graph_get_defs (cfg);
 297   if (defs == NULL)
 298   {
 299     defs = inst_get_default_defs (cfg, inst);
 300
 301     if (defs == NULL)
 302       return (-1);
 303
 304     status = def_foreach (defs, gl_instance_get_rrdargs_cb, &data);
 305
 306     def_destroy (defs);
 307   }
 308   else
 309   {
 310     status = def_foreach (defs, gl_instance_get_rrdargs_cb, &data);
 311   }
 312
 313   return (status);
 314 } /* }}} int inst_get_rrdargs */
 315
 316 graph_ident_t *inst_get_selector (graph_instance_t *inst) /* {{{ */
 317 {
 318   if (inst == NULL)
 319     return (NULL);
 320
 321   return (ident_clone (inst->select));
 322 } /* }}} graph_ident_t *inst_get_selector */
 323
 324 int inst_get_params (graph_config_t *cfg, graph_instance_t *inst, /* {{{ */
 325     char *buffer, size_t buffer_size)
 326 {
 327   graph_ident_t *cfg_select;
 328
 329   if ((cfg == NULL) || (inst == NULL)
 330       || (buffer == NULL) || (buffer_size < 1))
 331     return (EINVAL);
 332
 333   cfg_select = graph_get_selector (cfg);
 334   if (cfg_select == NULL)
 335   {
 336     fprintf (stderr, "inst_get_params: graph_get_selector failed");
 337     return (-1);
 338   }
 339
 340   buffer[0] = 0;
 341
 342 #define COPY_FIELD(field) do {                                  \
 343   const char *cfg_f  = ident_get_##field (cfg_select);          \
 344   const char *inst_f = ident_get_##field (inst->select);        \
 345   if (strcmp (cfg_f, inst_f) == 0)                              \
 346   {                                                             \
 347     strlcat (buffer, #field, buffer_size);                      \
 348     strlcat (buffer, "=", buffer_size);                         \
 349     strlcat (buffer, cfg_f, buffer_size);                       \
 350   }                                                             \
 351   else                                                          \
 352   {                                                             \
 353     strlcat (buffer, "graph_", buffer_size);                    \
 354     strlcat (buffer, #field, buffer_size);                      \
 355     strlcat (buffer, "=", buffer_size);                         \
 356     strlcat (buffer, cfg_f, buffer_size);                       \
 357     strlcat (buffer, ";", buffer_size);                         \
 358     strlcat (buffer, "inst_", buffer_size);                     \
 359     strlcat (buffer, #field, buffer_size);                      \
 360     strlcat (buffer, "=", buffer_size);                         \
 361     strlcat (buffer, inst_f, buffer_size);                      \
 362   }                                                             \
 363 } while (0)
 364
 365   COPY_FIELD(host);
 366   strlcat (buffer, ";", buffer_size);
 367   COPY_FIELD(plugin);
 368   strlcat (buffer, ";", buffer_size);
 369   COPY_FIELD(plugin_instance);
 370   strlcat (buffer, ";", buffer_size);
 371   COPY_FIELD(type);
 372   strlcat (buffer, ";", buffer_size);
 373   COPY_FIELD(type_instance);
 374
 375 #undef COPY_FIELD
 376
 377   ident_destroy (cfg_select);
 378
 379   return (0);
 380 } /* }}} int inst_get_params */
 381
 382 int inst_append (graph_instance_t *head, graph_instance_t *inst) /* {{{ */
 383 {
 384   graph_instance_t *ptr;
 385
 386   if ((head == NULL) || (inst == NULL))
 387     return (EINVAL);
 388
 389   ptr = head;
 390   while (ptr->next != NULL)
 391     ptr = ptr->next;
 392
 393   ptr->next = inst;
 394
 395   return (0);
 396 } /* }}} int inst_append */
 397
 398 int inst_foreach (graph_instance_t *inst, /* {{{ */
 399                 inst_callback_t cb, void *user_data)
 400 {
 401   graph_instance_t *ptr;
 402
 403   if ((inst == NULL) || (cb == NULL))
 404     return (EINVAL);
 405
 406   for (ptr = inst; ptr != NULL; ptr = ptr->next)
 407   {
 408     int status;
 409
 410     status = (*cb) (ptr, user_data);
 411     if (status != 0)
 412       return (status);
 413   }
 414
 415   return (0);
 416 } /* }}} int inst_foreach */
 417
 418 graph_instance_t *inst_find_matching (graph_instance_t *inst, /* {{{ */
 419     const graph_ident_t *ident)
 420 {
 421   graph_instance_t *ptr;
 422
 423   if ((inst == NULL) || (ident == NULL))
 424     return (NULL);
 425
 426   for (ptr = inst; ptr != NULL; ptr = ptr->next)
 427     if (ident_matches (ptr->select, ident))
 428       return (ptr);
 429
 430   return (NULL);
 431 } /* }}} graph_instance_t *inst_find_matching */
 432
 433 int inst_describe (graph_config_t *cfg, graph_instance_t *inst, /* {{{ */
 434     char *buffer, size_t buffer_size)
 435 {
 436   graph_ident_t *cfg_select;
 437
 438   if ((cfg == NULL) || (inst == NULL)
 439       || (buffer == NULL) || (buffer_size < 2))
 440     return (EINVAL);
 441
 442   cfg_select = graph_get_selector (cfg);
 443   if (cfg_select == NULL)
 444   {
 445     fprintf (stderr, "inst_describe: graph_get_selector failed\n");
 446     return (-1);
 447   }
 448
 449   buffer[0] = 0;
 450
 451 #define CHECK_FIELD(field) do {                                              \
 452   if (IS_ANY (ident_get_##field (cfg_select)))                               \
 453   {                                                                          \
 454     if (buffer[0] != 0)                                                      \
 455       strlcat (buffer, "/", buffer_size);                                    \
 456     strlcat (buffer, ident_get_##field (inst->select), buffer_size);         \
 457   }                                                                          \
 458 } while (0)
 459
 460   CHECK_FIELD (host);
 461   CHECK_FIELD (plugin);
 462   CHECK_FIELD (plugin_instance);
 463   CHECK_FIELD (type);
 464   CHECK_FIELD (type_instance);
 465
 466 #undef CHECK_FIELD
 467
 468   if (buffer[0] == 0)
 469     strlcat (buffer, "default", buffer_size);
 470
 471   ident_destroy (cfg_select);
 472
 473   return (0);
 474 } /* }}} int inst_describe */
 475
 476 time_t inst_get_mtime (graph_instance_t *inst) /* {{{ */
 477 {
 478   size_t i;
 479   time_t mtime;
 480
 481   if (inst == NULL)
 482     return (0);
 483
 484   mtime = 0;
 485   for (i = 0; i < inst->files_num; i++)
 486   {
 487     time_t tmp;
 488
 489     tmp = ident_get_mtime (inst->files[i]);
 490     if (mtime < tmp)
 491       mtime = tmp;
 492   }
 493
 494   return (mtime);
 495 } /* }}} time_t inst_get_mtime */
 496
 497 /* vim: set sw=2 sts=2 et fdm=marker : */