src/graph_{ident,instance}.[ch]: Implement getting the mtime from instances.
[collection4.git] / src / graph_ident.c
1 #include <stdlib.h>
2 #include <stdio.h>
3 #include <unistd.h>
4 #include <string.h>
5 #include <strings.h>
6 #include <errno.h>
7 #include <limits.h> /* PATH_MAX */
8 #include <sys/types.h>
9 #include <sys/stat.h>
10
11 #include "graph_ident.h"
12 #include "common.h"
13 #include "filesystem.h"
14
15 #include <fcgiapp.h>
16 #include <fcgi_stdio.h>
17
18 /*
19  * Data types
20  */
21 struct graph_ident_s /* {{{ */
22 {
23   char *host;
24   char *plugin;
25   char *plugin_instance;
26   char *type;
27   char *type_instance;
28 }; /* }}} struct graph_ident_s */
29
30 /*
31  * Private functions
32  */
33 static char *part_copy_with_selector (const char *selector, /* {{{ */
34     const char *part, unsigned int flags)
35 {
36   if ((selector == NULL) || (part == NULL))
37     return (NULL);
38
39   if ((flags & IDENT_FLAG_REPLACE_ANY) && IS_ANY (part))
40     return (NULL);
41
42   if ((flags & IDENT_FLAG_REPLACE_ALL) && IS_ALL (part))
43     return (NULL);
44
45   /* Replace the ANY and ALL flags if requested and if the selecter actually
46    * *is* that flag. */
47   if (IS_ANY (selector))
48   {
49     if (flags & IDENT_FLAG_REPLACE_ANY)
50       return (strdup (part));
51     else
52       return (strdup (selector));
53   }
54
55   if (IS_ALL (selector))
56   {
57     if (flags & IDENT_FLAG_REPLACE_ALL)
58       return (strdup (part));
59     else
60       return (strdup (selector));
61   }
62
63   if (strcmp (selector, part) != 0)
64     return (NULL);
65
66   /* Otherwise (no replacement), return a copy of the selector. */
67   return (strdup (selector));
68 } /* }}} char *part_copy_with_selector */
69
70 static _Bool part_matches (const char *selector, /* {{{ */
71     const char *part)
72 {
73   if ((selector == NULL) && (part == NULL))
74     return (1);
75
76   if (selector == NULL) /* && (part != NULL) */
77     return (0);
78
79   if (IS_ANY(selector) || IS_ALL(selector))
80     return (1);
81
82   if (part == NULL) /* && (selector != NULL) */
83     return (0);
84
85   if (strcmp (selector, part) == 0)
86     return (1);
87
88   return (0);
89 } /* }}} _Bool part_matches */
90
91 /*
92  * Public functions
93  */
94 graph_ident_t *ident_create (const char *host, /* {{{ */
95     const char *plugin, const char *plugin_instance,
96     const char *type, const char *type_instance)
97 {
98   graph_ident_t *ret;
99
100   if ((host == NULL)
101       || (plugin == NULL) || (plugin_instance == NULL)
102       || (type == NULL) || (type_instance == NULL))
103     return (NULL);
104
105   ret = malloc (sizeof (*ret));
106   if (ret == NULL)
107     return (NULL);
108   memset (ret, 0, sizeof (*ret));
109
110   ret->host = NULL;
111   ret->host = NULL;
112   ret->plugin = NULL;
113   ret->plugin_instance = NULL;
114   ret->type = NULL;
115   ret->type_instance = NULL;
116
117 #define COPY_PART(p) do {        \
118   ret->p = strdup (p);           \
119   if (ret->p == NULL)            \
120   {                              \
121     free (ret->host);            \
122     free (ret->plugin);          \
123     free (ret->plugin_instance); \
124     free (ret->type);            \
125     free (ret->type_instance);   \
126     free (ret);                  \
127     return (NULL);               \
128   }                              \
129 } while (0)
130
131   COPY_PART(host);
132   COPY_PART(plugin);
133   COPY_PART(plugin_instance);
134   COPY_PART(type);
135   COPY_PART(type_instance);
136
137 #undef COPY_PART
138
139   return (ret);
140 } /* }}} graph_ident_t *ident_create */
141
142 graph_ident_t *ident_clone (const graph_ident_t *ident) /* {{{ */
143 {
144   return (ident_create (ident->host,
145         ident->plugin, ident->plugin_instance,
146         ident->type, ident->type_instance));
147 } /* }}} graph_ident_t *ident_clone */
148
149 graph_ident_t *ident_copy_with_selector (const graph_ident_t *selector, /* {{{ */
150     const graph_ident_t *ident, unsigned int flags)
151 {
152   graph_ident_t *ret;
153
154   if ((selector == NULL) || (ident == NULL))
155     return (NULL);
156
157   ret = malloc (sizeof (*ret));
158   if (ret == NULL)
159     return (NULL);
160   memset (ret, 0, sizeof (*ret));
161   ret->host = NULL;
162   ret->plugin = NULL;
163   ret->plugin_instance = NULL;
164   ret->type = NULL;
165   ret->type_instance = NULL;
166
167 #define COPY_PART(p) do {                                  \
168   ret->p = part_copy_with_selector (selector->p, ident->p, flags); \
169   if (ret->p == NULL)                                      \
170   {                                                        \
171     free (ret->host);                                      \
172     free (ret->plugin);                                    \
173     free (ret->plugin_instance);                           \
174     free (ret->type);                                      \
175     free (ret->type_instance);                             \
176     return (NULL);                                         \
177   }                                                        \
178 } while (0)
179
180   COPY_PART (host);
181   COPY_PART (plugin);
182   COPY_PART (plugin_instance);
183   COPY_PART (type);
184   COPY_PART (type_instance);
185
186 #undef COPY_PART
187
188   return (ret);
189 } /* }}} graph_ident_t *ident_copy_with_selector */
190
191 void ident_destroy (graph_ident_t *ident) /* {{{ */
192 {
193   if (ident == NULL)
194     return;
195
196   free (ident->host);
197   free (ident->plugin);
198   free (ident->plugin_instance);
199   free (ident->type);
200   free (ident->type_instance);
201
202   free (ident);
203 } /* }}} void ident_destroy */
204
205 /* ident_get_* methods {{{ */
206 const char *ident_get_host (graph_ident_t *ident) /* {{{ */
207 {
208   if (ident == NULL)
209     return (NULL);
210
211   return (ident->host);
212 } /* }}} char *ident_get_host */
213
214 const char *ident_get_plugin (graph_ident_t *ident) /* {{{ */
215 {
216   if (ident == NULL)
217     return (NULL);
218
219   return (ident->plugin);
220 } /* }}} char *ident_get_plugin */
221
222 const char *ident_get_plugin_instance (graph_ident_t *ident) /* {{{ */
223 {
224   if (ident == NULL)
225     return (NULL);
226
227   return (ident->plugin_instance);
228 } /* }}} char *ident_get_plugin_instance */
229
230 const char *ident_get_type (graph_ident_t *ident) /* {{{ */
231 {
232   if (ident == NULL)
233     return (NULL);
234
235   return (ident->type);
236 } /* }}} char *ident_get_type */
237
238 const char *ident_get_type_instance (graph_ident_t *ident) /* {{{ */
239 {
240   if (ident == NULL)
241     return (NULL);
242
243   return (ident->type_instance);
244 } /* }}} char *ident_get_type_instance */
245 /* }}} ident_get_* methods */
246
247 /* ident_set_* methods {{{ */
248 int ident_set_host (graph_ident_t *ident, const char *host) /* {{{ */
249 {
250   char *tmp;
251
252   if ((ident == NULL) || (host == NULL))
253     return (EINVAL);
254
255   tmp = strdup (host);
256   if (tmp == NULL)
257     return (ENOMEM);
258
259   free (ident->host);
260   ident->host = tmp;
261
262   return (0);
263 } /* }}} int ident_set_host */
264
265 int ident_set_plugin (graph_ident_t *ident, const char *plugin) /* {{{ */
266 {
267   char *tmp;
268
269   if ((ident == NULL) || (plugin == NULL))
270     return (EINVAL);
271
272   tmp = strdup (plugin);
273   if (tmp == NULL)
274     return (ENOMEM);
275
276   free (ident->plugin);
277   ident->plugin = tmp;
278
279   return (0);
280 } /* }}} int ident_set_plugin */
281
282 int ident_set_plugin_instance (graph_ident_t *ident, const char *plugin_instance) /* {{{ */
283 {
284   char *tmp;
285
286   if ((ident == NULL) || (plugin_instance == NULL))
287     return (EINVAL);
288
289   tmp = strdup (plugin_instance);
290   if (tmp == NULL)
291     return (ENOMEM);
292
293   free (ident->plugin_instance);
294   ident->plugin_instance = tmp;
295
296   return (0);
297 } /* }}} int ident_set_plugin_instance */
298
299 int ident_set_type (graph_ident_t *ident, const char *type) /* {{{ */
300 {
301   char *tmp;
302
303   if ((ident == NULL) || (type == NULL))
304     return (EINVAL);
305
306   tmp = strdup (type);
307   if (tmp == NULL)
308     return (ENOMEM);
309
310   free (ident->type);
311   ident->type = tmp;
312
313   return (0);
314 } /* }}} int ident_set_type */
315
316 int ident_set_type_instance (graph_ident_t *ident, const char *type_instance) /* {{{ */
317 {
318   char *tmp;
319
320   if ((ident == NULL) || (type_instance == NULL))
321     return (EINVAL);
322
323   tmp = strdup (type_instance);
324   if (tmp == NULL)
325     return (ENOMEM);
326
327   free (ident->type_instance);
328   ident->type_instance = tmp;
329
330   return (0);
331 } /* }}} int ident_set_type_instance */
332
333 /* }}} ident_set_* methods */
334
335 int ident_compare (const graph_ident_t *i0, /* {{{ */
336     const graph_ident_t *i1)
337 {
338   int status;
339
340 #define COMPARE_PART(p) do {       \
341   status = strcmp (i0->p, i1->p);  \
342   if (status != 0)                 \
343     return (status);               \
344 } while (0)
345
346   COMPARE_PART (host);
347   COMPARE_PART (plugin);
348   COMPARE_PART (plugin_instance);
349   COMPARE_PART (type);
350   COMPARE_PART (type_instance);
351
352 #undef COMPARE_PART
353
354   return (0);
355 } /* }}} int ident_compare */
356
357 _Bool ident_matches (const graph_ident_t *selector, /* {{{ */
358     const graph_ident_t *ident)
359 {
360   if ((selector == NULL) && (ident == NULL))
361     return (0);
362   else if (selector == NULL)
363     return (-1);
364   else if (ident == NULL)
365     return (1);
366
367   if (!part_matches (selector->host, ident->host))
368     return (0);
369
370   if (!part_matches (selector->plugin, ident->plugin))
371     return (0);
372
373   if (!part_matches (selector->plugin_instance, ident->plugin_instance))
374     return (0);
375
376   if (!part_matches (selector->type, ident->type))
377     return (0);
378
379   if (!part_matches (selector->type_instance, ident->type_instance))
380     return (0);
381
382   return (1);
383 } /* }}} _Bool ident_matches */
384
385 char *ident_to_string (const graph_ident_t *ident) /* {{{ */
386 {
387   char buffer[PATH_MAX];
388
389   buffer[0] = 0;
390
391   strlcat (buffer, ident->host, sizeof (buffer));
392   strlcat (buffer, "/", sizeof (buffer));
393   strlcat (buffer, ident->plugin, sizeof (buffer));
394   if (ident->plugin_instance[0] != 0)
395   {
396     strlcat (buffer, "-", sizeof (buffer));
397     strlcat (buffer, ident->plugin_instance, sizeof (buffer));
398   }
399   strlcat (buffer, "/", sizeof (buffer));
400   strlcat (buffer, ident->type, sizeof (buffer));
401   if (ident->type_instance[0] != 0)
402   {
403     strlcat (buffer, "-", sizeof (buffer));
404     strlcat (buffer, ident->type_instance, sizeof (buffer));
405   }
406
407   return (strdup (buffer));
408 } /* }}} char *ident_to_string */
409
410 char *ident_to_file (const graph_ident_t *ident) /* {{{ */
411 {
412   char buffer[PATH_MAX];
413
414   buffer[0] = 0;
415
416   strlcat (buffer, DATA_DIR, sizeof (buffer));
417   strlcat (buffer, "/", sizeof (buffer));
418
419   strlcat (buffer, ident->host, sizeof (buffer));
420   strlcat (buffer, "/", sizeof (buffer));
421   strlcat (buffer, ident->plugin, sizeof (buffer));
422   if (ident->plugin_instance[0] != 0)
423   {
424     strlcat (buffer, "-", sizeof (buffer));
425     strlcat (buffer, ident->plugin_instance, sizeof (buffer));
426   }
427   strlcat (buffer, "/", sizeof (buffer));
428   strlcat (buffer, ident->type, sizeof (buffer));
429   if (ident->type_instance[0] != 0)
430   {
431     strlcat (buffer, "-", sizeof (buffer));
432     strlcat (buffer, ident->type_instance, sizeof (buffer));
433   }
434
435   strlcat (buffer, ".rrd", sizeof (buffer));
436
437   return (strdup (buffer));
438 } /* }}} char *ident_to_file */
439
440 char *ident_to_json (const graph_ident_t *ident) /* {{{ */
441 {
442   char buffer[4096];
443
444   buffer[0] = 0;
445
446   strlcat (buffer, "{\"host\":\"", sizeof (buffer));
447   strlcat (buffer, ident->host, sizeof (buffer));
448   strlcat (buffer, "\",\"plugin\":\"", sizeof (buffer));
449   strlcat (buffer, ident->plugin, sizeof (buffer));
450   strlcat (buffer, "\",\"plugin_instance\":\"", sizeof (buffer));
451   strlcat (buffer, ident->plugin_instance, sizeof (buffer));
452   strlcat (buffer, "\",\"type\":\"", sizeof (buffer));
453   strlcat (buffer, ident->type, sizeof (buffer));
454   strlcat (buffer, "\",\"type_instance\":\"", sizeof (buffer));
455   strlcat (buffer, ident->type_instance, sizeof (buffer));
456   strlcat (buffer, "\"}", sizeof (buffer));
457
458   return (strdup (buffer));
459 } /* }}} char *ident_to_json */
460
461 time_t ident_get_mtime (const graph_ident_t *ident) /* {{{ */
462 {
463   char *file;
464   struct stat statbuf;
465   int status;
466
467   if (ident == NULL)
468     return (0);
469
470   file = ident_to_file (ident);
471   if (file == NULL)
472     return (0);
473
474   memset (&statbuf, 0, sizeof (statbuf));
475   status = stat (file, &statbuf);
476   if (status != 0)
477   {
478     fprintf (stderr, "ident_get_mtime: stat'ing file \"%s\" failed: %s\n",
479         file, strerror (errno));
480     return (0);
481   }
482
483   free (file);
484   return (statbuf.st_mtime);
485 } /* }}} time_t ident_get_mtime */
486
487 /* vim: set sw=2 sts=2 et fdm=marker : */
488