gmond plugin: Update copyright notice.
[collectd.git] / src / gmond.c
1 /**
2  * collectd - src/gmond.c
3  * Copyright (C) 2009  Florian octo Forster
4  *
5  * This program is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License as published by the
7  * Free Software Foundation; only version 2 of the License is applicable.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License along
15  * with this program; if not, write to the Free Software Foundation, Inc.,
16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
17  *
18  * Authors:
19  *   Florian octo Forster <octo at verplant.org>
20  **/
21
22 #include "collectd.h"
23 #include "plugin.h"
24 #include "common.h"
25 #include "configfile.h"
26 #include "utils_avltree.h"
27
28 #if HAVE_PTHREAD_H
29 # include <pthread.h>
30 #endif
31 #if HAVE_SYS_SOCKET_H
32 # include <sys/socket.h>
33 #endif
34 #if HAVE_NETDB_H
35 # include <netdb.h>
36 #endif
37 #if HAVE_NETINET_IN_H
38 # include <netinet/in.h>
39 #endif
40 #if HAVE_ARPA_INET_H
41 # include <arpa/inet.h>
42 #endif
43 #if HAVE_POLL_H
44 # include <poll.h>
45 #endif
46
47 #include <gm_protocol.h>
48
49 #ifndef IPV6_ADD_MEMBERSHIP
50 # ifdef IPV6_JOIN_GROUP
51 #  define IPV6_ADD_MEMBERSHIP IPV6_JOIN_GROUP
52 # else
53 #  error "Neither IP_ADD_MEMBERSHIP nor IPV6_JOIN_GROUP is defined"
54 # endif
55 #endif /* !IP_ADD_MEMBERSHIP */
56
57 #ifdef GANGLIA_MAX_MESSAGE_LEN
58 # define BUFF_SIZE GANGLIA_MAX_MESSAGE_LEN
59 #else
60 # define BUFF_SIZE 1400
61 #endif
62
63 struct socket_entry_s
64 {
65   int                     fd;
66   struct sockaddr_storage addr;
67   socklen_t               addrlen;
68 };
69 typedef struct socket_entry_s socket_entry_t;
70
71 struct staging_entry_s
72 {
73   char key[2 * DATA_MAX_NAME_LEN];
74   value_list_t vl;
75   int flags;
76 };
77 typedef struct staging_entry_s staging_entry_t;
78
79 struct metric_map_s
80 {
81   char *ganglia_name;
82   char *type;
83   char *type_instance;
84   char *ds_name;
85   int   ds_type;
86   int   ds_index;
87 };
88 typedef struct metric_map_s metric_map_t;
89
90 #define MC_RECEIVE_GROUP_DEFAULT "239.2.11.71"
91 static char          *mc_receive_group = NULL;
92 #define MC_RECEIVE_PORT_DEFAULT "8649"
93 static char          *mc_receive_port = NULL;
94
95 static struct pollfd *mc_receive_sockets = NULL;
96 static size_t         mc_receive_sockets_num = 0;
97
98 static socket_entry_t  *mc_send_sockets = NULL;
99 static size_t           mc_send_sockets_num = 0;
100 static pthread_mutex_t  mc_send_sockets_lock = PTHREAD_MUTEX_INITIALIZER;
101
102 static int            mc_receive_thread_loop    = 0;
103 static int            mc_receive_thread_running = 0;
104 static pthread_t      mc_receive_thread_id;
105
106 static metric_map_t metric_map_default[] =
107 { /*---------------+-------------+-----------+-------------+------+-----*
108    * ganglia_name  ! type        ! type_inst ! data_source ! type ! idx *
109    *---------------+-------------+-----------+-------------+------+-----*/
110   { "load_one",     "load",       "",         "shortterm",     -1,   -1 },
111   { "load_five",    "load",       "",         "midterm",       -1,   -1 },
112   { "load_fifteen", "load",       "",         "longterm",      -1,   -1 },
113   { "cpu_user",     "cpu",        "user",     "value",         -1,   -1 },
114   { "cpu_system",   "cpu",        "system",   "value",         -1,   -1 },
115   { "cpu_idle",     "cpu",        "idle",     "value",         -1,   -1 },
116   { "cpu_nice",     "cpu",        "nice",     "value",         -1,   -1 },
117   { "cpu_wio",      "cpu",        "wait",     "value",         -1,   -1 },
118   { "mem_free",     "memory",     "free",     "value",         -1,   -1 },
119   { "mem_shared",   "memory",     "shared",   "value",         -1,   -1 },
120   { "mem_buffers",  "memory",     "buffered", "value",         -1,   -1 },
121   { "mem_cached",   "memory",     "cached",   "value",         -1,   -1 },
122   { "mem_total",    "memory",     "total",    "value",         -1,   -1 },
123   { "bytes_in",     "if_octets",  "",         "rx",            -1,   -1 },
124   { "bytes_out",    "if_octets",  "",         "tx",            -1,   -1 },
125   { "pkts_in",      "if_packets", "",         "rx",            -1,   -1 },
126   { "pkts_out",     "if_packets", "",         "tx",            -1,   -1 }
127 };
128 static size_t metric_map_len_default = STATIC_ARRAY_SIZE (metric_map_default);
129
130 static metric_map_t *metric_map = NULL;
131 static size_t        metric_map_len = 0;
132
133 static c_avl_tree_t   *staging_tree;
134 static pthread_mutex_t staging_lock = PTHREAD_MUTEX_INITIALIZER;
135
136 static metric_map_t *metric_lookup (const char *key) /* {{{ */
137 {
138   metric_map_t *map;
139   size_t map_len;
140   size_t i;
141
142   /* Search the user-supplied table first.. */
143   map = metric_map;
144   map_len = metric_map_len;
145   for (i = 0; i < map_len; i++)
146     if (strcmp (map[i].ganglia_name, key) == 0)
147       break;
148
149   /* .. and fall back to the built-in table if nothing is found. */
150   if (i >= map_len)
151   {
152     map = metric_map_default;
153     map_len = metric_map_len_default;
154
155     for (i = 0; i < map_len; i++)
156       if (strcmp (map[i].ganglia_name, key) == 0)
157         break;
158   }
159
160   if (i >= map_len)
161     return (NULL);
162
163   /* Look up the DS type and ds_index. */
164   if ((map[i].ds_type < 0) || (map[i].ds_index < 0)) /* {{{ */
165   {
166     const data_set_t *ds;
167
168     ds = plugin_get_ds (map[i].type);
169     if (ds == NULL)
170     {
171       WARNING ("gmond plugin: Type not defined: %s", map[i].type);
172       return (NULL);
173     }
174
175     if ((map[i].ds_name == NULL) && (ds->ds_num != 1))
176     {
177       WARNING ("gmond plugin: No data source name defined for metric %s, "
178           "but type %s has more than one data source.",
179           map[i].ganglia_name, map[i].type);
180       return (NULL);
181     }
182
183     if (map[i].ds_name == NULL)
184     {
185       map[i].ds_index = 0;
186     }
187     else
188     {
189       int j;
190
191       for (j = 0; j < ds->ds_num; j++)
192         if (strcasecmp (ds->ds[j].name, map[i].ds_name) == 0)
193           break;
194
195       if (j >= ds->ds_num)
196       {
197         WARNING ("gmond plugin: There is no data source "
198             "named `%s' in type `%s'.",
199             map[i].ds_name, ds->type);
200         return (NULL);
201       }
202       map[i].ds_index = j;
203     }
204
205     map[i].ds_type = ds->ds[map[i].ds_index].type;
206   } /* }}} if ((map[i].ds_type < 0) || (map[i].ds_index < 0)) */
207
208   return (map + i);
209 } /* }}} metric_map_t *metric_lookup */
210
211 static int create_sockets (socket_entry_t **ret_sockets, /* {{{ */
212     size_t *ret_sockets_num,
213     const char *node, const char *service, int listen)
214 {
215   struct addrinfo  ai_hints;
216   struct addrinfo *ai_list;
217   struct addrinfo *ai_ptr;
218   int              ai_return;
219
220   socket_entry_t *sockets;
221   size_t          sockets_num;
222
223   int status;
224     
225   sockets     = *ret_sockets;
226   sockets_num = *ret_sockets_num;
227
228   memset (&ai_hints, 0, sizeof (ai_hints));
229   ai_hints.ai_flags    = 0;
230 #ifdef AI_PASSIVE
231   ai_hints.ai_flags |= AI_PASSIVE;
232 #endif
233 #ifdef AI_ADDRCONFIG
234   ai_hints.ai_flags |= AI_ADDRCONFIG;
235 #endif
236   ai_hints.ai_family   = AF_UNSPEC;
237   ai_hints.ai_socktype = SOCK_DGRAM;
238   ai_hints.ai_protocol = IPPROTO_UDP;
239
240   ai_return = getaddrinfo (node, service, &ai_hints, &ai_list);
241   if (ai_return != 0)
242   {
243     char errbuf[1024];
244     ERROR ("gmond plugin: getaddrinfo (%s, %s) failed: %s",
245         (node == NULL) ? "(null)" : node,
246         (service == NULL) ? "(null)" : service,
247         (ai_return == EAI_SYSTEM)
248         ? sstrerror (errno, errbuf, sizeof (errbuf))
249         : gai_strerror (ai_return));
250     return (-1);
251   }
252
253   for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next) /* {{{ */
254   {
255     socket_entry_t *tmp;
256
257     tmp = realloc (sockets, (sockets_num + 1) * sizeof (*sockets));
258     if (tmp == NULL)
259     {
260       ERROR ("gmond plugin: realloc failed.");
261       continue;
262     }
263     sockets = tmp;
264
265     sockets[sockets_num].fd = socket (ai_ptr->ai_family, ai_ptr->ai_socktype,
266         ai_ptr->ai_protocol);
267     if (sockets[sockets_num].fd < 0)
268     {
269       char errbuf[1024];
270       ERROR ("gmond plugin: socket failed: %s",
271           sstrerror (errno, errbuf, sizeof (errbuf)));
272       continue;
273     }
274
275     assert (sizeof (sockets[sockets_num].addr) >= ai_ptr->ai_addrlen);
276     memcpy (&sockets[sockets_num].addr, ai_ptr->ai_addr, ai_ptr->ai_addrlen);
277     sockets[sockets_num].addrlen = ai_ptr->ai_addrlen;
278
279     /* Sending socket: Open only one socket and don't bind it. */
280     if (listen == 0)
281     {
282       sockets_num++;
283       break;
284     }
285     else
286     {
287       int yes = 1;
288
289       setsockopt (sockets[sockets_num].fd, SOL_SOCKET, SO_REUSEADDR,
290           (void *) &yes, sizeof (yes));
291     }
292
293     status = bind (sockets[sockets_num].fd, ai_ptr->ai_addr, ai_ptr->ai_addrlen);
294     if (status != 0)
295     {
296       char errbuf[1024];
297       ERROR ("gmond plugin: bind failed: %s",
298           sstrerror (errno, errbuf, sizeof (errbuf)));
299       close (sockets[sockets_num].fd);
300       continue;
301     }
302
303     if (ai_ptr->ai_family == AF_INET)
304     {
305       struct sockaddr_in *addr;
306       struct ip_mreq mreq;
307       int loop;
308
309       addr = (struct sockaddr_in *) ai_ptr->ai_addr;
310
311       if (!IN_MULTICAST (ntohl (addr->sin_addr.s_addr)))
312       {
313         sockets_num++;
314         continue;
315       }
316
317       loop = 1;
318       setsockopt (sockets[sockets_num].fd, IPPROTO_IP, IP_MULTICAST_LOOP,
319           (void *) &loop, sizeof (loop));
320
321       memset (&mreq, 0, sizeof (mreq));
322       mreq.imr_multiaddr.s_addr = addr->sin_addr.s_addr;
323       mreq.imr_interface.s_addr = htonl (INADDR_ANY);
324       setsockopt (sockets[sockets_num].fd, IPPROTO_IP, IP_ADD_MEMBERSHIP,
325           (void *) &mreq, sizeof (mreq));
326     } /* if (ai_ptr->ai_family == AF_INET) */
327     else if (ai_ptr->ai_family == AF_INET6)
328     {
329       struct sockaddr_in6 *addr;
330       struct ipv6_mreq mreq;
331       int loop;
332
333       addr = (struct sockaddr_in6 *) ai_ptr->ai_addr;
334
335       if (!IN6_IS_ADDR_MULTICAST (&addr->sin6_addr))
336       {
337         sockets_num++;
338         continue;
339       }
340
341       loop = 1;
342       setsockopt (sockets[sockets_num].fd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP,
343           (void *) &loop, sizeof (loop));
344
345       memset (&mreq, 0, sizeof (mreq));
346       memcpy (&mreq.ipv6mr_multiaddr,
347           &addr->sin6_addr, sizeof (addr->sin6_addr));
348       mreq.ipv6mr_interface = 0; /* any */
349       setsockopt (sockets[sockets_num].fd, IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP,
350           (void *) &mreq, sizeof (mreq));
351     } /* if (ai_ptr->ai_family == AF_INET6) */
352
353     sockets_num++;
354   } /* }}} for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next) */
355
356   freeaddrinfo (ai_list);
357
358   if ((*ret_sockets_num) >= sockets_num)
359     return (-1);
360
361   *ret_sockets = sockets;
362   *ret_sockets_num = sockets_num;
363   return (0);
364 } /* }}} int create_sockets */
365
366 static int request_meta_data (const char *host, const char *name) /* {{{ */
367 {
368   Ganglia_metadata_msg msg;
369   char buffer[BUFF_SIZE];
370   unsigned int buffer_size;
371   XDR xdr;
372   size_t i;
373
374   memset (&msg, 0, sizeof (msg));
375
376   msg.id = gmetadata_request;
377   msg.Ganglia_metadata_msg_u.grequest.metric_id.host = strdup (host);
378   msg.Ganglia_metadata_msg_u.grequest.metric_id.name = strdup (name);
379
380   if ((msg.Ganglia_metadata_msg_u.grequest.metric_id.host == NULL)
381       || (msg.Ganglia_metadata_msg_u.grequest.metric_id.name == NULL))
382   {
383     sfree (msg.Ganglia_metadata_msg_u.grequest.metric_id.host);
384     sfree (msg.Ganglia_metadata_msg_u.grequest.metric_id.name);
385     return (-1);
386   }
387
388   memset (buffer, 0, sizeof (buffer));
389   xdrmem_create (&xdr, buffer, sizeof (buffer), XDR_ENCODE);
390
391   if (!xdr_Ganglia_metadata_msg (&xdr, &msg))
392   {
393     sfree (msg.Ganglia_metadata_msg_u.grequest.metric_id.host);
394     sfree (msg.Ganglia_metadata_msg_u.grequest.metric_id.name);
395     return (-1);
396   }
397
398   buffer_size = xdr_getpos (&xdr);
399
400   DEBUG ("gmond plugin: Requesting meta data for %s/%s.",
401       host, name);
402
403   pthread_mutex_lock (&mc_send_sockets_lock);
404   for (i = 0; i < mc_send_sockets_num; i++)
405     sendto (mc_send_sockets[i].fd, buffer, (size_t) buffer_size,
406         /* flags = */ 0,
407         (struct sockaddr *) &mc_send_sockets[i].addr,
408         mc_send_sockets[i].addrlen);
409   pthread_mutex_unlock (&mc_send_sockets_lock);
410
411   sfree (msg.Ganglia_metadata_msg_u.grequest.metric_id.host);
412   sfree (msg.Ganglia_metadata_msg_u.grequest.metric_id.name);
413   return (0);
414 } /* }}} int request_meta_data */
415
416 static staging_entry_t *staging_entry_get (const char *host, /* {{{ */
417     const char *name,
418     const char *type, const char *type_instance,
419     int values_len)
420 {
421   char key[2 * DATA_MAX_NAME_LEN];
422   staging_entry_t *se;
423   int status;
424
425   if (staging_tree == NULL)
426     return (NULL);
427
428   ssnprintf (key, sizeof (key), "%s/%s/%s", host, type,
429       (type_instance != NULL) ? type_instance : "");
430
431   se = NULL;
432   status = c_avl_get (staging_tree, key, (void *) &se);
433   if (status == 0)
434     return (se);
435
436   /* insert new entry */
437   se = (staging_entry_t *) malloc (sizeof (*se));
438   if (se == NULL)
439     return (NULL);
440   memset (se, 0, sizeof (*se));
441
442   sstrncpy (se->key, key, sizeof (se->key));
443   se->flags = 0;
444
445   se->vl.values = (value_t *) calloc (values_len, sizeof (*se->vl.values));
446   if (se->vl.values == NULL)
447   {
448     sfree (se);
449     return (NULL);
450   }
451   se->vl.values_len = values_len;
452
453   se->vl.time = 0;
454   se->vl.interval = 0;
455   sstrncpy (se->vl.host, host, sizeof (se->vl.host));
456   sstrncpy (se->vl.plugin, "gmond", sizeof (se->vl.plugin));
457   sstrncpy (se->vl.type, type, sizeof (se->vl.type));
458   if (type_instance != NULL)
459     sstrncpy (se->vl.type_instance, type_instance,
460         sizeof (se->vl.type_instance));
461
462   status = c_avl_insert (staging_tree, se->key, se);
463   if (status != 0)
464   {
465     ERROR ("gmond plugin: c_avl_insert failed.");
466     sfree (se->vl.values);
467     sfree (se);
468     return (NULL);
469   }
470
471   return (se);
472 } /* }}} staging_entry_t *staging_entry_get */
473
474 static int staging_entry_submit (const char *host, const char *name, /* {{{ */
475     staging_entry_t *se)
476 {
477   value_list_t vl;
478   value_t values[se->vl.values_len];
479
480   if (se->vl.interval == 0)
481   {
482     /* No meta data has been received for this metric yet. */
483     se->flags = 0;
484     pthread_mutex_unlock (&staging_lock);
485     request_meta_data (host, name);
486     return (0);
487   }
488
489   se->flags = 0;
490
491   memcpy (values, se->vl.values, sizeof (values));
492   memcpy (&vl, &se->vl, sizeof (vl));
493
494   /* Unlock before calling `plugin_dispatch_values'.. */
495   pthread_mutex_unlock (&staging_lock);
496
497   vl.values = values;
498
499   plugin_dispatch_values (&vl);
500
501   return (0);
502 } /* }}} int staging_entry_submit */
503
504 static int staging_entry_update (const char *host, const char *name, /* {{{ */
505     const char *type, const char *type_instance,
506     int ds_index, int ds_type, value_t value)
507 {
508   const data_set_t *ds;
509   staging_entry_t *se;
510
511   ds = plugin_get_ds (type);
512   if (ds == NULL)
513   {
514     ERROR ("gmond plugin: Looking up type %s failed.", type);
515     return (-1);
516   }
517
518   if (ds->ds_num <= ds_index)
519   {
520     ERROR ("gmond plugin: Invalid index %i: %s has only %i data source(s).",
521         ds_index, ds->type, ds->ds_num);
522     return (-1);
523   }
524
525   pthread_mutex_lock (&staging_lock);
526
527   se = staging_entry_get (host, name, type, type_instance, ds->ds_num);
528   if (se == NULL)
529   {
530     pthread_mutex_unlock (&staging_lock);
531     ERROR ("gmond plugin: staging_entry_get failed.");
532     return (-1);
533   }
534   if (se->vl.values_len != ds->ds_num)
535   {
536     pthread_mutex_unlock (&staging_lock);
537     return (-1);
538   }
539
540   if (ds_type == DS_TYPE_COUNTER)
541     se->vl.values[ds_index].counter += value.counter;
542   else if (ds_type == DS_TYPE_GAUGE)
543     se->vl.values[ds_index].gauge = value.gauge;
544   se->flags |= (0x01 << ds_index);
545
546   /* Check if all values have been set and submit if so. */
547   if (se->flags == ((0x01 << se->vl.values_len) - 1))
548   {
549     /* `staging_lock' is unlocked in `staging_entry_submit'. */
550     staging_entry_submit (host, name, se);
551   }
552   else
553   {
554     pthread_mutex_unlock (&staging_lock);
555   }
556
557   return (0);
558 } /* }}} int staging_entry_update */
559
560 static int mc_handle_value_msg (Ganglia_value_msg *msg) /* {{{ */
561 {
562   const char *host;
563   const char *name;
564   metric_map_t *map;
565
566   value_t value_counter;
567   value_t value_gauge;
568
569   /* Fill in `host', `name', `value_counter', and `value_gauge' according to
570    * the value type, or return with an error. */
571   switch (msg->id) /* {{{ */
572   {
573     case gmetric_uint:
574     {
575       Ganglia_gmetric_uint msg_uint;
576
577       msg_uint = msg->Ganglia_value_msg_u.gu_int;
578
579       host = msg_uint.metric_id.host;
580       name = msg_uint.metric_id.name;
581       value_counter.counter = (counter_t) msg_uint.ui;
582       value_gauge.gauge = (gauge_t) msg_uint.ui;
583       break;
584     }
585
586     case gmetric_string:
587     {
588       Ganglia_gmetric_string msg_string;
589       char *endptr;
590
591       msg_string = msg->Ganglia_value_msg_u.gstr;
592
593       host = msg_string.metric_id.host;
594       name = msg_string.metric_id.name;
595
596       endptr = NULL;
597       errno = 0;
598       value_counter.counter = (counter_t) strtoll (msg_string.str,
599           &endptr, /* base = */ 0);
600       if ((endptr == msg_string.str) || (errno != 0))
601         value_counter.counter = -1;
602
603       endptr = NULL;
604       errno = 0;
605       value_gauge.gauge = (gauge_t) strtod (msg_string.str, &endptr);
606       if ((endptr == msg_string.str) || (errno != 0))
607         value_gauge.gauge = NAN;
608
609       break;
610     }
611
612     case gmetric_float:
613     {
614       Ganglia_gmetric_float msg_float;
615
616       msg_float = msg->Ganglia_value_msg_u.gf;
617
618       host = msg_float.metric_id.host;
619       name = msg_float.metric_id.name;
620       value_counter.counter = (counter_t) msg_float.f;
621       value_gauge.gauge = (gauge_t) msg_float.f;
622       break;
623     }
624
625     case gmetric_double:
626     {
627       Ganglia_gmetric_double msg_double;
628
629       msg_double = msg->Ganglia_value_msg_u.gd;
630
631       host = msg_double.metric_id.host;
632       name = msg_double.metric_id.name;
633       value_counter.counter = (counter_t) msg_double.d;
634       value_gauge.gauge = (gauge_t) msg_double.d;
635       break;
636     }
637     default:
638       DEBUG ("gmond plugin: Value type not handled: %i", msg->id);
639       return (-1);
640   } /* }}} switch (msg->id) */
641
642   assert (host != NULL);
643   assert (name != NULL);
644
645   map = metric_lookup (name);
646   if (map != NULL)
647     return (staging_entry_update (host, name,
648           map->type, map->type_instance,
649           map->ds_index, map->ds_type,
650           (map->ds_type == DS_TYPE_COUNTER) ? value_counter : value_gauge));
651
652   DEBUG ("gmond plugin: Cannot find a translation for %s.", name);
653   return (-1);
654 } /* }}} int mc_handle_value_msg */
655
656 static int mc_handle_metadata_msg (Ganglia_metadata_msg *msg) /* {{{ */
657 {
658   switch (msg->id)
659   {
660     case gmetadata_full:
661     {
662       Ganglia_metadatadef msg_meta;
663       staging_entry_t *se;
664       const data_set_t *ds;
665       metric_map_t *map;
666
667       msg_meta = msg->Ganglia_metadata_msg_u.gfull;
668
669       if (msg_meta.metric.tmax <= 0)
670         return (-1);
671
672       map = metric_lookup (msg_meta.metric_id.name);
673       if (map == NULL)
674       {
675         DEBUG ("gmond plugin: Not handling meta data %s.",
676             msg_meta.metric_id.name);
677         return (0);
678       }
679
680       ds = plugin_get_ds (map->type);
681       if (ds == NULL)
682       {
683         WARNING ("gmond plugin: Could not find data set %s.", map->type);
684         return (-1);
685       }
686
687       DEBUG ("gmond plugin: Received meta data for %s/%s.",
688           msg_meta.metric_id.host, msg_meta.metric_id.name);
689
690       pthread_mutex_lock (&staging_lock);
691       se = staging_entry_get (msg_meta.metric_id.host,
692           msg_meta.metric_id.name,
693           map->type, map->type_instance,
694           ds->ds_num);
695       if (se != NULL)
696         se->vl.interval = (int) msg_meta.metric.tmax;
697       pthread_mutex_unlock (&staging_lock);
698
699       if (se == NULL)
700       {
701         ERROR ("gmond plugin: staging_entry_get failed.");
702         return (-1);
703       }
704
705       break;
706     }
707
708     default:
709     {
710       return (-1);
711     }
712   }
713
714   return (0);
715 } /* }}} int mc_handle_metadata_msg */
716
717 static int mc_handle_metric (void *buffer, size_t buffer_size) /* {{{ */
718 {
719   XDR xdr;
720   Ganglia_msg_formats format;
721
722   xdrmem_create (&xdr, buffer, buffer_size, XDR_DECODE);
723
724   xdr_Ganglia_msg_formats (&xdr, &format);
725   xdr_setpos (&xdr, 0);
726
727   switch (format)
728   {
729     case gmetric_ushort:
730     case gmetric_short:
731     case gmetric_int:
732     case gmetric_uint:
733     case gmetric_string:
734     case gmetric_float:
735     case gmetric_double:
736     {
737       Ganglia_value_msg msg;
738
739       memset (&msg, 0, sizeof (msg));
740       if (xdr_Ganglia_value_msg (&xdr, &msg))
741         mc_handle_value_msg (&msg);
742       break;
743     }
744
745     case gmetadata_full:
746     case gmetadata_request:
747     {
748       Ganglia_metadata_msg msg;
749       memset (&msg, 0, sizeof (msg));
750       if (xdr_Ganglia_metadata_msg (&xdr, &msg))
751         mc_handle_metadata_msg (&msg);
752       break;
753     }
754
755     default:
756       DEBUG ("gmond plugin: Unknown format: %i", format);
757       return (-1);
758   } /* switch (format) */
759
760
761   return (0);
762 } /* }}} int mc_handle_metric */
763
764 static int mc_handle_socket (struct pollfd *p) /* {{{ */
765 {
766   char buffer[BUFF_SIZE];
767   ssize_t buffer_size;
768
769   if ((p->revents & (POLLIN | POLLPRI)) == 0)
770   {
771     p->revents = 0;
772     return (-1);
773   }
774
775   buffer_size = recv (p->fd, buffer, sizeof (buffer), /* flags = */ 0);
776   if (buffer_size <= 0)
777   {
778     char errbuf[1024];
779     ERROR ("gmond plugin: recv failed: %s",
780         sstrerror (errno, errbuf, sizeof (errbuf)));
781     p->revents = 0;
782     return (-1);
783   }
784
785   mc_handle_metric (buffer, (size_t) buffer_size);
786   return (0);
787 } /* }}} int mc_handle_socket */
788
789 static void *mc_receive_thread (void *arg) /* {{{ */
790 {
791   socket_entry_t *mc_receive_socket_entries;
792   int status;
793   size_t i;
794
795   mc_receive_socket_entries = NULL;
796   status = create_sockets (&mc_receive_socket_entries, &mc_receive_sockets_num,
797       (mc_receive_group != NULL) ? mc_receive_group : MC_RECEIVE_GROUP_DEFAULT,
798       (mc_receive_port != NULL) ? mc_receive_port : MC_RECEIVE_PORT_DEFAULT,
799       /* listen = */ 1);
800   if (status != 0)
801   {
802     ERROR ("gmond plugin: create_sockets failed.");
803     return ((void *) -1);
804   }
805
806   mc_receive_sockets = (struct pollfd *) calloc (mc_receive_sockets_num,
807       sizeof (*mc_receive_sockets));
808   if (mc_receive_sockets == NULL)
809   {
810     ERROR ("gmond plugin: calloc failed.");
811     for (i = 0; i < mc_receive_sockets_num; i++)
812       close (mc_receive_socket_entries[i].fd);
813     free (mc_receive_socket_entries);
814     mc_receive_socket_entries = NULL;
815     mc_receive_sockets_num = 0;
816     return ((void *) -1);
817   }
818
819   for (i = 0; i < mc_receive_sockets_num; i++)
820   {
821     mc_receive_sockets[i].fd = mc_receive_socket_entries[i].fd;
822     mc_receive_sockets[i].events = POLLIN | POLLPRI;
823     mc_receive_sockets[i].revents = 0;
824   }
825
826   while (mc_receive_thread_loop != 0)
827   {
828     status = poll (mc_receive_sockets, mc_receive_sockets_num, -1);
829     if (status <= 0)
830     {
831       char errbuf[1024];
832       if (errno == EINTR)
833         continue;
834       ERROR ("gmond plugin: poll failed: %s",
835           sstrerror (errno, errbuf, sizeof (errbuf)));
836       break;
837     }
838
839     for (i = 0; i < mc_receive_sockets_num; i++)
840     {
841       if (mc_receive_sockets[i].revents != 0)
842         mc_handle_socket (mc_receive_sockets + i);
843     }
844   } /* while (mc_receive_thread_loop != 0) */
845
846   return ((void *) 0);
847 } /* }}} void *mc_receive_thread */
848
849 static int mc_receive_thread_start (void) /* {{{ */
850 {
851   int status;
852
853   if (mc_receive_thread_running != 0)
854     return (-1);
855
856   mc_receive_thread_loop = 1;
857
858   status = pthread_create (&mc_receive_thread_id, /* attr = */ NULL,
859       mc_receive_thread, /* args = */ NULL);
860   if (status != 0)
861   {
862     ERROR ("gmond plugin: Starting receive thread failed.");
863     mc_receive_thread_loop = 0;
864     return (-1);
865   }
866
867   mc_receive_thread_running = 1;
868   return (0);
869 } /* }}} int start_receive_thread */
870
871 static int mc_receive_thread_stop (void) /* {{{ */
872 {
873   if (mc_receive_thread_running == 0)
874     return (-1);
875
876   mc_receive_thread_loop = 0;
877
878   INFO ("gmond plugin: Stopping receive thread.");
879   pthread_kill (mc_receive_thread_id, SIGTERM);
880   pthread_join (mc_receive_thread_id, /* return value = */ NULL);
881   memset (&mc_receive_thread_id, 0, sizeof (mc_receive_thread_id));
882
883   mc_receive_thread_running = 0;
884
885   return (0);
886 } /* }}} int mc_receive_thread_stop */
887
888 /* 
889  * Config:
890  *
891  * <Plugin gmond>
892  *   MCReceiveFrom "239.2.11.71" "8649"
893  *   <Metric "load_one">
894  *     Type "load"
895  *     [TypeInstance "foo"]
896  *     [DataSource "bar"]
897  *   </Metric>
898  * </Plugin>
899  */
900 static int gmond_config_set_string (oconfig_item_t *ci, char **str) /* {{{ */
901 {
902   char *tmp;
903
904   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
905   {
906     WARNING ("gmond plugin: The `%s' option needs "
907         "exactly one string argument.", ci->key);
908     return (-1);
909   }
910
911   tmp = strdup (ci->values[0].value.string);
912   if (tmp == NULL)
913   {
914     ERROR ("gmond plugin: strdup failed.");
915     return (-1);
916   }
917
918   sfree (*str);
919   *str = tmp;
920   return (0);
921 } /* }}} int gmond_config_set_string */
922
923 static int gmond_config_add_metric (oconfig_item_t *ci) /* {{{ */
924 {
925   metric_map_t *map;
926   int i;
927
928   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
929   {
930     WARNING ("gmond plugin: `Metric' blocks need "
931         "exactly one string argument.");
932     return (-1);
933   }
934
935   map = realloc (metric_map, (metric_map_len + 1) * sizeof (*metric_map));
936   if (map == NULL)
937   {
938     ERROR ("gmond plugin: realloc failed.");
939     return (-1);
940   }
941   metric_map = map;
942   map = metric_map + metric_map_len;
943
944   memset (map, 0, sizeof (*map));
945   map->type = NULL;
946   map->type_instance = NULL;
947   map->ds_name = NULL;
948   map->ds_type = -1;
949   map->ds_index = -1;
950
951   map->ganglia_name = strdup (ci->values[0].value.string);
952   if (map->ganglia_name == NULL)
953   {
954     ERROR ("gmond plugin: strdup failed.");
955     return (-1);
956   }
957
958   for (i = 0; i < ci->children_num; i++)
959   {
960     oconfig_item_t *child = ci->children + i;
961     if (strcasecmp ("Type", child->key) == 0)
962       gmond_config_set_string (child, &map->type);
963     else if (strcasecmp ("TypeInstance", child->key) == 0)
964       gmond_config_set_string (child, &map->type_instance);
965     else if (strcasecmp ("DataSource", child->key) == 0)
966       gmond_config_set_string (child, &map->ds_name);
967     else
968     {
969       WARNING ("gmond plugin: Unknown configuration option `%s' ignored.",
970           child->key);
971     }
972   }
973
974   if (map->type == NULL)
975   {
976     ERROR ("gmond plugin: No type is set for metric %s.",
977         map->ganglia_name);
978     sfree (map->ganglia_name);
979     sfree (map->type_instance);
980     return (-1);
981   }
982
983   metric_map_len++;
984   return (0);
985 } /* }}} int gmond_config_add_metric */
986
987 static int gmond_config_set_address (oconfig_item_t *ci, /* {{{ */
988     char **ret_addr, char **ret_port)
989 {
990   char *addr;
991   char *port;
992
993   if ((ci->values_num != 1) && (ci->values_num != 2))
994   {
995     WARNING ("gmond plugin: The `%s' config option needs "
996         "one or two string arguments.",
997         ci->key);
998     return (-1);
999   }
1000   if ((ci->values[0].type != OCONFIG_TYPE_STRING)
1001       || ((ci->values_num == 2)
1002         && (ci->values[1].type != OCONFIG_TYPE_STRING)))
1003   {
1004     WARNING ("gmond plugin: The `%s' config option needs "
1005         "one or two string arguments.",
1006         ci->key);
1007     return (-1);
1008   }
1009
1010   addr = strdup (ci->values[0].value.string);
1011   if (ci->values_num == 2)
1012     port = strdup (ci->values[1].value.string);
1013   else
1014     port = NULL;
1015
1016   if ((addr == NULL) || ((ci->values_num == 2) && (port == NULL)))
1017   {
1018     ERROR ("gmond plugin: strdup failed.");
1019     sfree (addr);
1020     sfree (port);
1021     return (-1);
1022   }
1023
1024   sfree (*ret_addr);
1025   sfree (*ret_port);
1026
1027   *ret_addr = addr;
1028   *ret_port = port;
1029
1030   return (0);
1031 } /* }}} int gmond_config_set_address */
1032
1033 static int gmond_config (oconfig_item_t *ci) /* {{{ */
1034 {
1035   int i;
1036
1037   for (i = 0; i < ci->children_num; i++)
1038   {
1039     oconfig_item_t *child = ci->children + i;
1040     if (strcasecmp ("MCReceiveFrom", child->key) == 0)
1041       gmond_config_set_address (child, &mc_receive_group, &mc_receive_port);
1042     else if (strcasecmp ("Metric", child->key) == 0)
1043       gmond_config_add_metric (child);
1044     else
1045     {
1046       WARNING ("gmond plugin: Unknown configuration option `%s' ignored.",
1047           child->key);
1048     }
1049   }
1050
1051   return (0);
1052 } /* }}} int gmond_config */
1053
1054 static int gmond_init (void) /* {{{ */
1055 {
1056   create_sockets (&mc_send_sockets, &mc_send_sockets_num,
1057       (mc_receive_group != NULL) ? mc_receive_group : MC_RECEIVE_GROUP_DEFAULT,
1058       (mc_receive_port != NULL) ? mc_receive_port : MC_RECEIVE_PORT_DEFAULT,
1059       /* listen = */ 0);
1060
1061   staging_tree = c_avl_create ((void *) strcmp);
1062   if (staging_tree == NULL)
1063   {
1064     ERROR ("gmond plugin: c_avl_create failed.");
1065     return (-1);
1066   }
1067
1068   mc_receive_thread_start ();
1069
1070   return (0);
1071 } /* }}} int gmond_init */
1072
1073 static int gmond_shutdown (void) /* {{{ */
1074 {
1075   size_t i;
1076
1077   mc_receive_thread_stop ();
1078
1079   pthread_mutex_lock (&mc_send_sockets_lock);
1080   for (i = 0; i < mc_send_sockets_num; i++)
1081   {
1082     close (mc_send_sockets[i].fd);
1083     mc_send_sockets[i].fd = -1;
1084   }
1085   sfree (mc_send_sockets);
1086   mc_send_sockets_num = 0;
1087   pthread_mutex_unlock (&mc_send_sockets_lock);
1088
1089
1090   return (0);
1091 } /* }}} int gmond_shutdown */
1092
1093 void module_register (void)
1094 {
1095   plugin_register_complex_config ("gmond", gmond_config);
1096   plugin_register_init ("gmond", gmond_init);
1097   plugin_register_shutdown ("gmond", gmond_shutdown);
1098 }
1099
1100 /* vim: set sw=2 sts=2 et fdm=marker : */