Merge branch 'collectd-5.4'
[collectd.git] / src / snmp.c
1 /**
2  * collectd - src/snmp.c
3  * Copyright (C) 2007-2012  Florian octo Forster
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a
6  * copy of this software and associated documentation files (the "Software"),
7  * to deal in the Software without restriction, including without limitation
8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9  * and/or sell copies of the Software, and to permit persons to whom the
10  * Software is furnished to do so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be included in
13  * all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21  * DEALINGS IN THE SOFTWARE.
22  *
23  * Authors:
24  *   Florian octo Forster <octo at collectd.org>
25  **/
26
27 #include "collectd.h"
28 #include "common.h"
29 #include "plugin.h"
30 #include "utils_complain.h"
31
32 #include <pthread.h>
33
34 #include <net-snmp/net-snmp-config.h>
35 #include <net-snmp/net-snmp-includes.h>
36
37 #include <fnmatch.h>
38
39 /*
40  * Private data structes
41  */
42 struct oid_s
43 {
44   oid oid[MAX_OID_LEN];
45   size_t oid_len;
46 };
47 typedef struct oid_s oid_t;
48
49 union instance_u
50 {
51   char  string[DATA_MAX_NAME_LEN];
52   oid_t oid;
53 };
54 typedef union instance_u instance_t;
55
56 struct data_definition_s
57 {
58   char *name; /* used to reference this from the `Collect' option */
59   char *type; /* used to find the data_set */
60   _Bool is_table;
61   instance_t instance;
62   char *instance_prefix;
63   oid_t *values;
64   int values_len;
65   double scale;
66   double shift;
67   struct data_definition_s *next;
68   char **ignores;
69   size_t ignores_len;
70   int invert_match;
71 };
72 typedef struct data_definition_s data_definition_t;
73
74 struct host_definition_s
75 {
76   char *name;
77   char *address;
78   int version;
79
80   /* snmpv1/2 options */
81   char *community;
82
83   /* snmpv3 security options */
84   char *username;
85   oid *auth_protocol;
86   size_t auth_protocol_len;
87   char *auth_passphrase;
88   oid *priv_protocol;
89   size_t priv_protocol_len;
90   char *priv_passphrase;
91   int security_level;
92   char *context;
93
94   void *sess_handle;
95   c_complain_t complaint;
96   cdtime_t interval;
97   data_definition_t **data_list;
98   int data_list_len;
99 };
100 typedef struct host_definition_s host_definition_t;
101
102 /* These two types are used to cache values in `csnmp_read_table' to handle
103  * gaps in tables. */
104 struct csnmp_list_instances_s
105 {
106   oid_t suffix;
107   char instance[DATA_MAX_NAME_LEN];
108   struct csnmp_list_instances_s *next;
109 };
110 typedef struct csnmp_list_instances_s csnmp_list_instances_t;
111
112 struct csnmp_table_values_s
113 {
114   oid_t suffix;
115   value_t value;
116   struct csnmp_table_values_s *next;
117 };
118 typedef struct csnmp_table_values_s csnmp_table_values_t;
119
120 /*
121  * Private variables
122  */
123 static data_definition_t *data_head = NULL;
124
125 /*
126  * Prototypes
127  */
128 static int csnmp_read_host (user_data_t *ud);
129
130 /*
131  * Private functions
132  */
133 static void csnmp_oid_init (oid_t *dst, oid const *src, size_t n)
134 {
135   assert (n <= STATIC_ARRAY_SIZE (dst->oid));
136   memcpy (dst->oid, src, sizeof (*src) * n);
137   dst->oid_len = n;
138 }
139
140 static int csnmp_oid_compare (oid_t const *left, oid_t const *right)
141 {
142   return (snmp_oid_compare (left->oid, left->oid_len,
143         right->oid, right->oid_len));
144 }
145
146 static int csnmp_oid_suffix (oid_t *dst, oid_t const *src,
147     oid_t const *root)
148 {
149   /* Make sure "src" is in "root"s subtree. */
150   if (src->oid_len <= root->oid_len)
151     return (EINVAL);
152   if (snmp_oid_ncompare (root->oid, root->oid_len,
153         src->oid, src->oid_len,
154         /* n = */ root->oid_len) != 0)
155     return (EINVAL);
156
157   memset (dst, 0, sizeof (*dst));
158   dst->oid_len = src->oid_len - root->oid_len;
159   memcpy (dst->oid, &src->oid[root->oid_len],
160       dst->oid_len * sizeof (dst->oid[0]));
161   return (0);
162 }
163
164 static int csnmp_oid_to_string (char *buffer, size_t buffer_size,
165     oid_t const *o)
166 {
167   char oid_str[MAX_OID_LEN][16];
168   char *oid_str_ptr[MAX_OID_LEN];
169   size_t i;
170
171   for (i = 0; i < o->oid_len; i++)
172   {
173     ssnprintf (oid_str[i], sizeof (oid_str[i]), "%lu", (unsigned long) o->oid[i]);
174     oid_str_ptr[i] = oid_str[i];
175   }
176
177   return (strjoin (buffer, buffer_size,
178         oid_str_ptr, o->oid_len, /* separator = */ "."));
179 }
180
181 static void csnmp_host_close_session (host_definition_t *host) /* {{{ */
182 {
183   if (host->sess_handle == NULL)
184     return;
185
186   snmp_sess_close (host->sess_handle);
187   host->sess_handle = NULL;
188 } /* }}} void csnmp_host_close_session */
189
190 static void csnmp_host_definition_destroy (void *arg) /* {{{ */
191 {
192   host_definition_t *hd;
193
194   hd = arg;
195
196   if (hd == NULL)
197     return;
198
199   if (hd->name != NULL)
200   {
201     DEBUG ("snmp plugin: Destroying host definition for host `%s'.",
202         hd->name);
203   }
204
205   csnmp_host_close_session (hd);
206
207   sfree (hd->name);
208   sfree (hd->address);
209   sfree (hd->community);
210   sfree (hd->username);
211   sfree (hd->auth_passphrase);
212   sfree (hd->priv_passphrase);
213   sfree (hd->context);
214   sfree (hd->data_list);
215
216   sfree (hd);
217 } /* }}} void csnmp_host_definition_destroy */
218
219 /* Many functions to handle the configuration. {{{ */
220 /* First there are many functions which do configuration stuff. It's a big
221  * bloated and messy, I'm afraid. */
222
223 /*
224  * Callgraph for the config stuff:
225  *  csnmp_config
226  *  +-> call_snmp_init_once
227  *  +-> csnmp_config_add_data
228  *  !   +-> csnmp_config_add_data_instance
229  *  !   +-> csnmp_config_add_data_instance_prefix
230  *  !   +-> csnmp_config_add_data_values
231  *  +-> csnmp_config_add_host
232  *      +-> csnmp_config_add_host_version
233  *      +-> csnmp_config_add_host_collect
234  *      +-> csnmp_config_add_host_auth_protocol
235  *      +-> csnmp_config_add_host_priv_protocol
236  *      +-> csnmp_config_add_host_security_level
237  */
238 static void call_snmp_init_once (void)
239 {
240   static int have_init = 0;
241
242   if (have_init == 0)
243     init_snmp (PACKAGE_NAME);
244   have_init = 1;
245 } /* void call_snmp_init_once */
246
247 static int csnmp_config_add_data_instance (data_definition_t *dd, oconfig_item_t *ci)
248 {
249   char buffer[DATA_MAX_NAME_LEN];
250   int status;
251
252   status = cf_util_get_string_buffer(ci, buffer, sizeof(buffer));
253   if (status != 0)
254     return status;
255
256   if (dd->is_table)
257   {
258     /* Instance is an OID */
259     dd->instance.oid.oid_len = MAX_OID_LEN;
260
261     if (!read_objid (buffer,
262           dd->instance.oid.oid, &dd->instance.oid.oid_len))
263     {
264       ERROR ("snmp plugin: read_objid (%s) failed.", buffer);
265       return (-1);
266     }
267   }
268   else
269   {
270     /* Instance is a simple string */
271     sstrncpy (dd->instance.string, buffer,
272         sizeof (dd->instance.string));
273   }
274
275   return (0);
276 } /* int csnmp_config_add_data_instance */
277
278 static int csnmp_config_add_data_instance_prefix (data_definition_t *dd,
279     oconfig_item_t *ci)
280 {
281   int status;
282
283   if (!dd->is_table)
284   {
285     WARNING ("snmp plugin: data %s: InstancePrefix is ignored when `Table' "
286         "is set to `false'.", dd->name);
287     return (-1);
288   }
289
290   status = cf_util_get_string(ci, &dd->instance_prefix);
291   return status;
292 } /* int csnmp_config_add_data_instance_prefix */
293
294 static int csnmp_config_add_data_values (data_definition_t *dd, oconfig_item_t *ci)
295 {
296   int i;
297
298   if (ci->values_num < 1)
299   {
300     WARNING ("snmp plugin: `Values' needs at least one argument.");
301     return (-1);
302   }
303
304   for (i = 0; i < ci->values_num; i++)
305     if (ci->values[i].type != OCONFIG_TYPE_STRING)
306     {
307       WARNING ("snmp plugin: `Values' needs only string argument.");
308       return (-1);
309     }
310
311   sfree (dd->values);
312   dd->values_len = 0;
313   dd->values = (oid_t *) malloc (sizeof (oid_t) * ci->values_num);
314   if (dd->values == NULL)
315     return (-1);
316   dd->values_len = ci->values_num;
317
318   for (i = 0; i < ci->values_num; i++)
319   {
320     dd->values[i].oid_len = MAX_OID_LEN;
321
322     if (NULL == snmp_parse_oid (ci->values[i].value.string,
323           dd->values[i].oid, &dd->values[i].oid_len))
324     {
325       ERROR ("snmp plugin: snmp_parse_oid (%s) failed.",
326           ci->values[i].value.string);
327       free (dd->values);
328       dd->values = NULL;
329       dd->values_len = 0;
330       return (-1);
331     }
332   }
333
334   return (0);
335 } /* int csnmp_config_add_data_instance */
336
337 static int csnmp_config_add_data_blacklist(data_definition_t *dd, oconfig_item_t *ci)
338 {
339   int i;
340
341   if (ci->values_num < 1)
342     return (0);
343
344   for (i = 0; i < ci->values_num; i++)
345   {
346     if (ci->values[i].type != OCONFIG_TYPE_STRING)
347     {
348       WARNING ("snmp plugin: `Ignore' needs only string argument.");
349       return (-1);
350     }
351   }
352
353   dd->ignores_len = 0;
354   dd->ignores = NULL;
355
356   for (i = 0; i < ci->values_num; ++i)
357   {
358     if (strarray_add(&(dd->ignores), &(dd->ignores_len), ci->values[i].value.string) != 0)
359     {
360       ERROR("snmp plugin: Can't allocate memory");
361       strarray_free(dd->ignores, dd->ignores_len);
362       return (ENOMEM); 
363     }
364   }
365   return 0;
366 } /* int csnmp_config_add_data_blacklist */
367
368 static int csnmp_config_add_data_blacklist_match_inverted(data_definition_t *dd, oconfig_item_t *ci)
369 {
370   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_BOOLEAN))
371   {
372     WARNING ("snmp plugin: `InvertMatch' needs exactly one boolean argument.");
373     return (-1);
374   }
375
376   dd->invert_match = ci->values[0].value.boolean ? 1 : 0;
377
378   return (0);
379 } /* int csnmp_config_add_data_blacklist_match_inverted */
380
381 static int csnmp_config_add_data (oconfig_item_t *ci)
382 {
383   data_definition_t *dd;
384   int status = 0;
385   int i;
386
387   dd = (data_definition_t *) malloc (sizeof (data_definition_t));
388   if (dd == NULL)
389     return (-1);
390   memset (dd, '\0', sizeof (data_definition_t));
391
392   status = cf_util_get_string(ci, &dd->name);
393   if (status != 0)
394   {
395     free (dd);
396     return (-1);
397   }
398
399   dd->scale = 1.0;
400   dd->shift = 0.0;
401
402   for (i = 0; i < ci->children_num; i++)
403   {
404     oconfig_item_t *option = ci->children + i;
405     status = 0;
406
407     if (strcasecmp ("Type", option->key) == 0)
408       status = cf_util_get_string(option, &dd->type);
409     else if (strcasecmp ("Table", option->key) == 0)
410       status = cf_util_get_boolean(option, &dd->is_table);
411     else if (strcasecmp ("Instance", option->key) == 0)
412       status = csnmp_config_add_data_instance (dd, option);
413     else if (strcasecmp ("InstancePrefix", option->key) == 0)
414       status = csnmp_config_add_data_instance_prefix (dd, option);
415     else if (strcasecmp ("Values", option->key) == 0)
416       status = csnmp_config_add_data_values (dd, option);
417     else if (strcasecmp ("Shift", option->key) == 0)
418       status = cf_util_get_double(option, &dd->shift);
419     else if (strcasecmp ("Scale", option->key) == 0)
420       status = cf_util_get_double(option, &dd->scale);
421     else if (strcasecmp ("Ignore", option->key) == 0)
422       status = csnmp_config_add_data_blacklist(dd, option);
423     else if (strcasecmp ("InvertMatch", option->key) == 0)
424       status = csnmp_config_add_data_blacklist_match_inverted(dd, option);
425     else
426     {
427       WARNING ("snmp plugin: Option `%s' not allowed here.", option->key);
428       status = -1;
429     }
430
431     if (status != 0)
432       break;
433   } /* for (ci->children) */
434
435   while (status == 0)
436   {
437     if (dd->type == NULL)
438     {
439       WARNING ("snmp plugin: `Type' not given for data `%s'", dd->name);
440       status = -1;
441       break;
442     }
443     if (dd->values == NULL)
444     {
445       WARNING ("snmp plugin: No `Value' given for data `%s'", dd->name);
446       status = -1;
447       break;
448     }
449
450     break;
451   } /* while (status == 0) */
452
453   if (status != 0)
454   {
455     sfree (dd->name);
456     sfree (dd->instance_prefix);
457     sfree (dd->values);
458     sfree (dd->ignores);
459     sfree (dd);
460     return (-1);
461   }
462
463   DEBUG ("snmp plugin: dd = { name = %s, type = %s, is_table = %s, values_len = %i }",
464       dd->name, dd->type, (dd->is_table != 0) ? "true" : "false", dd->values_len);
465
466   if (data_head == NULL)
467     data_head = dd;
468   else
469   {
470     data_definition_t *last;
471     last = data_head;
472     while (last->next != NULL)
473       last = last->next;
474     last->next = dd;
475   }
476
477   return (0);
478 } /* int csnmp_config_add_data */
479
480 static int csnmp_config_add_host_version (host_definition_t *hd, oconfig_item_t *ci)
481 {
482   int version;
483
484   if ((ci->values_num != 1)
485       || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
486   {
487     WARNING ("snmp plugin: The `Version' config option needs exactly one number argument.");
488     return (-1);
489   }
490
491   version = (int) ci->values[0].value.number;
492   if ((version < 1) || (version > 3))
493   {
494     WARNING ("snmp plugin: `Version' must either be `1', `2', or `3'.");
495     return (-1);
496   }
497
498   hd->version = version;
499
500   return (0);
501 } /* int csnmp_config_add_host_address */
502
503 static int csnmp_config_add_host_collect (host_definition_t *host,
504     oconfig_item_t *ci)
505 {
506   data_definition_t *data;
507   data_definition_t **data_list;
508   int data_list_len;
509   int i;
510
511   if (ci->values_num < 1)
512   {
513     WARNING ("snmp plugin: `Collect' needs at least one argument.");
514     return (-1);
515   }
516
517   for (i = 0; i < ci->values_num; i++)
518     if (ci->values[i].type != OCONFIG_TYPE_STRING)
519     {
520       WARNING ("snmp plugin: All arguments to `Collect' must be strings.");
521       return (-1);
522     }
523
524   data_list_len = host->data_list_len + ci->values_num;
525   data_list = (data_definition_t **) realloc (host->data_list,
526       sizeof (data_definition_t *) * data_list_len);
527   if (data_list == NULL)
528     return (-1);
529   host->data_list = data_list;
530
531   for (i = 0; i < ci->values_num; i++)
532   {
533     for (data = data_head; data != NULL; data = data->next)
534       if (strcasecmp (ci->values[i].value.string, data->name) == 0)
535         break;
536
537     if (data == NULL)
538     {
539       WARNING ("snmp plugin: No such data configured: `%s'",
540           ci->values[i].value.string);
541       continue;
542     }
543
544     DEBUG ("snmp plugin: Collect: host = %s, data[%i] = %s;",
545         host->name, host->data_list_len, data->name);
546
547     host->data_list[host->data_list_len] = data;
548     host->data_list_len++;
549   } /* for (values_num) */
550
551   return (0);
552 } /* int csnmp_config_add_host_collect */
553
554 static int csnmp_config_add_host_auth_protocol (host_definition_t *hd, oconfig_item_t *ci)
555 {
556   char buffer[4];
557   int status;
558
559   status = cf_util_get_string_buffer(ci, buffer, sizeof(buffer));
560   if (status != 0)
561     return status;
562
563   if (strcasecmp("MD5", buffer) == 0) {
564     hd->auth_protocol = usmHMACMD5AuthProtocol;
565     hd->auth_protocol_len = sizeof(usmHMACMD5AuthProtocol)/sizeof(oid);
566   }
567   else if (strcasecmp("SHA", buffer) == 0) {
568     hd->auth_protocol = usmHMACSHA1AuthProtocol;
569     hd->auth_protocol_len = sizeof(usmHMACSHA1AuthProtocol)/sizeof(oid);
570   }
571   else
572   {
573     WARNING ("snmp plugin: The `AuthProtocol' config option must be `MD5' or `SHA'.");
574     return (-1);
575   }
576
577   DEBUG ("snmp plugin: host = %s; host->auth_protocol = %s;",
578       hd->name, hd->auth_protocol == usmHMACMD5AuthProtocol ? "MD5" : "SHA");
579
580   return (0);
581 } /* int csnmp_config_add_host_auth_protocol */
582
583 static int csnmp_config_add_host_priv_protocol (host_definition_t *hd, oconfig_item_t *ci)
584 {
585   char buffer[4];
586   int status;
587
588   status = cf_util_get_string_buffer(ci, buffer, sizeof(buffer));
589   if (status != 0)
590     return status;
591
592   if (strcasecmp("AES", buffer) == 0)
593   {
594     hd->priv_protocol = usmAESPrivProtocol;
595     hd->priv_protocol_len = sizeof(usmAESPrivProtocol)/sizeof(oid);
596   }
597   else if (strcasecmp("DES", buffer) == 0) {
598     hd->priv_protocol = usmDESPrivProtocol;
599     hd->priv_protocol_len = sizeof(usmDESPrivProtocol)/sizeof(oid);
600   }
601   else
602   {
603     WARNING ("snmp plugin: The `PrivProtocol' config option must be `AES' or `DES'.");
604     return (-1);
605   }
606
607   DEBUG ("snmp plugin: host = %s; host->priv_protocol = %s;",
608       hd->name, hd->priv_protocol == usmAESPrivProtocol ? "AES" : "DES");
609
610   return (0);
611 } /* int csnmp_config_add_host_priv_protocol */
612
613 static int csnmp_config_add_host_security_level (host_definition_t *hd, oconfig_item_t *ci)
614 {
615   char buffer[16];
616   int status;
617
618   status = cf_util_get_string_buffer(ci, buffer, sizeof(buffer));
619   if (status != 0)
620     return status;
621
622   if (strcasecmp("noAuthNoPriv", buffer) == 0)
623     hd->security_level = SNMP_SEC_LEVEL_NOAUTH;
624   else if (strcasecmp("authNoPriv", buffer) == 0)
625     hd->security_level = SNMP_SEC_LEVEL_AUTHNOPRIV;
626   else if (strcasecmp("authPriv", buffer) == 0)
627     hd->security_level = SNMP_SEC_LEVEL_AUTHPRIV;
628   else
629   {
630     WARNING ("snmp plugin: The `SecurityLevel' config option must be `noAuthNoPriv', `authNoPriv', or `authPriv'.");
631     return (-1);
632   }
633
634   DEBUG ("snmp plugin: host = %s; host->security_level = %d;",
635       hd->name, hd->security_level);
636
637   return (0);
638 } /* int csnmp_config_add_host_security_level */
639
640 static int csnmp_config_add_host (oconfig_item_t *ci)
641 {
642   host_definition_t *hd;
643   int status = 0;
644   int i;
645
646   /* Registration stuff. */
647   char cb_name[DATA_MAX_NAME_LEN];
648   user_data_t cb_data;
649   struct timespec cb_interval;
650
651   hd = (host_definition_t *) malloc (sizeof (host_definition_t));
652   if (hd == NULL)
653     return (-1);
654   memset (hd, '\0', sizeof (host_definition_t));
655   hd->version = 2;
656   C_COMPLAIN_INIT (&hd->complaint);
657
658   status = cf_util_get_string(ci, &hd->name);
659   if (status != 0)
660     return status;
661
662   hd->sess_handle = NULL;
663   hd->interval = 0;
664
665   for (i = 0; i < ci->children_num; i++)
666   {
667     oconfig_item_t *option = ci->children + i;
668     status = 0;
669
670     if (strcasecmp ("Address", option->key) == 0)
671       status = cf_util_get_string(option, &hd->address);
672     else if (strcasecmp ("Community", option->key) == 0)
673       status = cf_util_get_string(option, &hd->community);
674     else if (strcasecmp ("Version", option->key) == 0)
675       status = csnmp_config_add_host_version (hd, option);
676     else if (strcasecmp ("Collect", option->key) == 0)
677       csnmp_config_add_host_collect (hd, option);
678     else if (strcasecmp ("Interval", option->key) == 0)
679       cf_util_get_cdtime (option, &hd->interval);
680     else if (strcasecmp ("Username", option->key) == 0)
681       status = cf_util_get_string(option, &hd->username);
682     else if (strcasecmp ("AuthProtocol", option->key) == 0)
683       status = csnmp_config_add_host_auth_protocol (hd, option);
684     else if (strcasecmp ("PrivacyProtocol", option->key) == 0)
685       status = csnmp_config_add_host_priv_protocol (hd, option);
686     else if (strcasecmp ("AuthPassphrase", option->key) == 0)
687       status = cf_util_get_string(option, &hd->auth_passphrase);
688     else if (strcasecmp ("PrivacyPassphrase", option->key) == 0)
689       status = cf_util_get_string(option, &hd->priv_passphrase);
690     else if (strcasecmp ("SecurityLevel", option->key) == 0)
691       status = csnmp_config_add_host_security_level (hd, option);
692     else if (strcasecmp ("Context", option->key) == 0)
693       status = cf_util_get_string(option, &hd->context);
694     else
695     {
696       WARNING ("snmp plugin: csnmp_config_add_host: Option `%s' not allowed here.", option->key);
697       status = -1;
698     }
699
700     if (status != 0)
701       break;
702   } /* for (ci->children) */
703
704   while (status == 0)
705   {
706     if (hd->address == NULL)
707     {
708       WARNING ("snmp plugin: `Address' not given for host `%s'", hd->name);
709       status = -1;
710       break;
711     }
712     if (hd->community == NULL && hd->version < 3)
713     {
714       WARNING ("snmp plugin: `Community' not given for host `%s'", hd->name);
715       status = -1;
716       break;
717     }
718     if (hd->version == 3)
719     {
720       if (hd->username == NULL)
721       {
722         WARNING ("snmp plugin: `Username' not given for host `%s'", hd->name);
723         status = -1;
724         break;
725       }
726       if (hd->security_level == 0)
727       {
728         WARNING ("snmp plugin: `SecurityLevel' not given for host `%s'", hd->name);
729         status = -1;
730         break;
731       }
732       if (hd->security_level == SNMP_SEC_LEVEL_AUTHNOPRIV || hd->security_level == SNMP_SEC_LEVEL_AUTHPRIV)
733       {
734         if (hd->auth_protocol == NULL)
735         {
736           WARNING ("snmp plugin: `AuthProtocol' not given for host `%s'", hd->name);
737           status = -1;
738           break;
739         }
740         if (hd->auth_passphrase == NULL)
741         {
742           WARNING ("snmp plugin: `AuthPassphrase' not given for host `%s'", hd->name);
743           status = -1;
744           break;
745         }
746       }
747       if (hd->security_level == SNMP_SEC_LEVEL_AUTHPRIV)
748       {
749         if (hd->priv_protocol == NULL)
750         {
751           WARNING ("snmp plugin: `PrivacyProtocol' not given for host `%s'", hd->name);
752           status = -1;
753           break;
754         }
755         if (hd->priv_passphrase == NULL)
756         {
757           WARNING ("snmp plugin: `PrivacyPassphrase' not given for host `%s'", hd->name);
758           status = -1;
759           break;
760         }
761       }
762     }
763
764     break;
765   } /* while (status == 0) */
766
767   if (status != 0)
768   {
769     csnmp_host_definition_destroy (hd);
770     return (-1);
771   }
772
773   DEBUG ("snmp plugin: hd = { name = %s, address = %s, community = %s, version = %i }",
774       hd->name, hd->address, hd->community, hd->version);
775
776   ssnprintf (cb_name, sizeof (cb_name), "snmp-%s", hd->name);
777
778   memset (&cb_data, 0, sizeof (cb_data));
779   cb_data.data = hd;
780   cb_data.free_func = csnmp_host_definition_destroy;
781
782   CDTIME_T_TO_TIMESPEC (hd->interval, &cb_interval);
783
784   status = plugin_register_complex_read (/* group = */ NULL, cb_name,
785       csnmp_read_host, /* interval = */ &cb_interval,
786       /* user_data = */ &cb_data);
787   if (status != 0)
788   {
789     ERROR ("snmp plugin: Registering complex read function failed.");
790     csnmp_host_definition_destroy (hd);
791     return (-1);
792   }
793
794   return (0);
795 } /* int csnmp_config_add_host */
796
797 static int csnmp_config (oconfig_item_t *ci)
798 {
799   int i;
800
801   call_snmp_init_once ();
802
803   for (i = 0; i < ci->children_num; i++)
804   {
805     oconfig_item_t *child = ci->children + i;
806     if (strcasecmp ("Data", child->key) == 0)
807       csnmp_config_add_data (child);
808     else if (strcasecmp ("Host", child->key) == 0)
809       csnmp_config_add_host (child);
810     else
811     {
812       WARNING ("snmp plugin: Ignoring unknown config option `%s'.", child->key);
813     }
814   } /* for (ci->children) */
815
816   return (0);
817 } /* int csnmp_config */
818
819 /* }}} End of the config stuff. Now the interesting part begins */
820
821 static void csnmp_host_open_session (host_definition_t *host)
822 {
823   struct snmp_session sess;
824   int error;
825
826   if (host->sess_handle != NULL)
827     csnmp_host_close_session (host);
828
829   snmp_sess_init (&sess);
830   sess.peername = host->address;
831   switch (host->version)
832   {
833     case 1:
834       sess.version = SNMP_VERSION_1;
835       break;
836     case 3:
837       sess.version = SNMP_VERSION_3;
838       break;
839     default:
840       sess.version = SNMP_VERSION_2c;
841       break;
842   }
843
844   if (host->version == 3)
845   {
846     sess.securityName = host->username;
847     sess.securityNameLen = strlen (host->username);
848     sess.securityLevel = host->security_level;
849
850     if (sess.securityLevel == SNMP_SEC_LEVEL_AUTHNOPRIV || sess.securityLevel == SNMP_SEC_LEVEL_AUTHPRIV)
851     {
852       sess.securityAuthProto = host->auth_protocol;
853       sess.securityAuthProtoLen = host->auth_protocol_len;
854       sess.securityAuthKeyLen = USM_AUTH_KU_LEN;
855       error = generate_Ku (sess.securityAuthProto,
856             sess.securityAuthProtoLen,
857             (u_char *) host->auth_passphrase,
858             strlen(host->auth_passphrase),
859             sess.securityAuthKey,
860             &sess.securityAuthKeyLen);
861       if (error != SNMPERR_SUCCESS) {
862         ERROR ("snmp plugin: host %s: Error generating Ku from auth_passphrase. (Error %d)", host->name, error);
863       }
864     }
865
866     if (sess.securityLevel == SNMP_SEC_LEVEL_AUTHPRIV)
867     {
868       sess.securityPrivProto = host->priv_protocol;
869       sess.securityPrivProtoLen = host->priv_protocol_len;
870       sess.securityPrivKeyLen = USM_PRIV_KU_LEN;
871       error = generate_Ku (sess.securityAuthProto,
872             sess.securityAuthProtoLen,
873             (u_char *) host->priv_passphrase,
874             strlen(host->priv_passphrase),
875             sess.securityPrivKey,
876             &sess.securityPrivKeyLen);
877       if (error != SNMPERR_SUCCESS) {
878         ERROR ("snmp plugin: host %s: Error generating Ku from priv_passphrase. (Error %d)", host->name, error);
879       }
880     }
881
882     if (host->context != NULL)
883     {
884       sess.contextName = host->context;
885       sess.contextNameLen = strlen (host->context);
886     }
887   }
888   else /* SNMPv1/2 "authenticates" with community string */
889   {
890     sess.community = (u_char *) host->community;
891     sess.community_len = strlen (host->community);
892   }
893
894   /* snmp_sess_open will copy the `struct snmp_session *'. */
895   host->sess_handle = snmp_sess_open (&sess);
896
897   if (host->sess_handle == NULL)
898   {
899     char *errstr = NULL;
900
901     snmp_error (&sess, NULL, NULL, &errstr);
902
903     ERROR ("snmp plugin: host %s: snmp_sess_open failed: %s",
904         host->name, (errstr == NULL) ? "Unknown problem" : errstr);
905     sfree (errstr);
906   }
907 } /* void csnmp_host_open_session */
908
909 /* TODO: Check if negative values wrap around. Problem: negative temperatures. */
910 static value_t csnmp_value_list_to_value (struct variable_list *vl, int type,
911     double scale, double shift,
912     const char *host_name, const char *data_name)
913 {
914   value_t ret;
915   uint64_t tmp_unsigned = 0;
916   int64_t tmp_signed = 0;
917   _Bool defined = 1;
918   /* Set to true when the original SNMP type appears to have been signed. */
919   _Bool prefer_signed = 0;
920
921   if ((vl->type == ASN_INTEGER)
922       || (vl->type == ASN_UINTEGER)
923       || (vl->type == ASN_COUNTER)
924 #ifdef ASN_TIMETICKS
925       || (vl->type == ASN_TIMETICKS)
926 #endif
927       || (vl->type == ASN_GAUGE))
928   {
929     tmp_unsigned = (uint32_t) *vl->val.integer;
930     tmp_signed = (int32_t) *vl->val.integer;
931
932     if ((vl->type == ASN_INTEGER)
933         || (vl->type == ASN_GAUGE))
934       prefer_signed = 1;
935
936     DEBUG ("snmp plugin: Parsed int32 value is %"PRIu64".", tmp_unsigned);
937   }
938   else if (vl->type == ASN_COUNTER64)
939   {
940     tmp_unsigned = (uint32_t) vl->val.counter64->high;
941     tmp_unsigned = tmp_unsigned << 32;
942     tmp_unsigned += (uint32_t) vl->val.counter64->low;
943     tmp_signed = (int64_t) tmp_unsigned;
944     DEBUG ("snmp plugin: Parsed int64 value is %"PRIu64".", tmp_unsigned);
945   }
946   else if (vl->type == ASN_OCTET_STR)
947   {
948     /* We'll handle this later.. */
949   }
950   else
951   {
952     char oid_buffer[1024];
953
954     memset (oid_buffer, 0, sizeof (oid_buffer));
955     snprint_objid (oid_buffer, sizeof (oid_buffer) - 1,
956         vl->name, vl->name_length);
957
958 #ifdef ASN_NULL
959     if (vl->type == ASN_NULL)
960       INFO ("snmp plugin: OID \"%s\" is undefined (type ASN_NULL)",
961           oid_buffer);
962     else
963 #endif
964       WARNING ("snmp plugin: I don't know the ASN type #%i "
965                "(OID: \"%s\", data block \"%s\", host block \"%s\")",
966           (int) vl->type, oid_buffer,
967           (data_name != NULL) ? data_name : "UNKNOWN",
968           (host_name != NULL) ? host_name : "UNKNOWN");
969
970     defined = 0;
971   }
972
973   if (vl->type == ASN_OCTET_STR)
974   {
975     int status = -1;
976
977     if (vl->val.string != NULL)
978     {
979       char string[64];
980       size_t string_length;
981
982       string_length = sizeof (string) - 1;
983       if (vl->val_len < string_length)
984         string_length = vl->val_len;
985
986       /* The strings we get from the Net-SNMP library may not be null
987        * terminated. That is why we're using `memcpy' here and not `strcpy'.
988        * `string_length' is set to `vl->val_len' which holds the length of the
989        * string.  -octo */
990       memcpy (string, vl->val.string, string_length);
991       string[string_length] = 0;
992
993       status = parse_value (string, &ret, type);
994       if (status != 0)
995       {
996         ERROR ("snmp plugin: csnmp_value_list_to_value: Parsing string as %s failed: %s",
997             DS_TYPE_TO_STRING (type), string);
998       }
999     }
1000
1001     if (status != 0)
1002     {
1003       switch (type)
1004       {
1005         case DS_TYPE_COUNTER:
1006         case DS_TYPE_DERIVE:
1007         case DS_TYPE_ABSOLUTE:
1008           memset (&ret, 0, sizeof (ret));
1009           break;
1010
1011         case DS_TYPE_GAUGE:
1012           ret.gauge = NAN;
1013           break;
1014
1015         default:
1016           ERROR ("snmp plugin: csnmp_value_list_to_value: Unknown "
1017               "data source type: %i.", type);
1018           ret.gauge = NAN;
1019       }
1020     }
1021   } /* if (vl->type == ASN_OCTET_STR) */
1022   else if (type == DS_TYPE_COUNTER)
1023   {
1024     ret.counter = tmp_unsigned;
1025   }
1026   else if (type == DS_TYPE_GAUGE)
1027   {
1028     if (!defined)
1029       ret.gauge = NAN;
1030     else if (prefer_signed)
1031       ret.gauge = (scale * tmp_signed) + shift;
1032     else
1033       ret.gauge = (scale * tmp_unsigned) + shift;
1034   }
1035   else if (type == DS_TYPE_DERIVE)
1036   {
1037     if (prefer_signed)
1038       ret.derive = (derive_t) tmp_signed;
1039     else
1040       ret.derive = (derive_t) tmp_unsigned;
1041   }
1042   else if (type == DS_TYPE_ABSOLUTE)
1043   {
1044     ret.absolute = (absolute_t) tmp_unsigned;
1045   }
1046   else
1047   {
1048     ERROR ("snmp plugin: csnmp_value_list_to_value: Unknown data source "
1049         "type: %i.", type);
1050     ret.gauge = NAN;
1051   }
1052
1053   return (ret);
1054 } /* value_t csnmp_value_list_to_value */
1055
1056 static int csnmp_strvbcopy_hexstring (char *dst, /* {{{ */
1057     const struct variable_list *vb, size_t dst_size)
1058 {
1059   char *buffer_ptr;
1060   size_t buffer_free;
1061   size_t i;
1062
1063   buffer_ptr = dst;
1064   buffer_free = dst_size;
1065
1066   for (i = 0; i < vb->val_len; i++)
1067   {
1068     int status;
1069
1070     status = snprintf (buffer_ptr, buffer_free,
1071         (i == 0) ? "%02x" : ":%02x", (unsigned int) vb->val.bitstring[i]);
1072
1073     if (status >= buffer_free)
1074     {
1075       buffer_ptr += (buffer_free - 1);
1076       *buffer_ptr = 0;
1077       return (dst_size + (buffer_free - status));
1078     }
1079     else /* if (status < buffer_free) */
1080     {
1081       buffer_ptr += status;
1082       buffer_free -= status;
1083     }
1084   }
1085
1086   return ((int) (dst_size - buffer_free));
1087 } /* }}} int csnmp_strvbcopy_hexstring */
1088
1089 static int csnmp_strvbcopy (char *dst, /* {{{ */
1090     const struct variable_list *vb, size_t dst_size)
1091 {
1092   char *src;
1093   size_t num_chars;
1094   size_t i;
1095
1096   if (vb->type == ASN_OCTET_STR)
1097     src = (char *) vb->val.string;
1098   else if (vb->type == ASN_BIT_STR)
1099     src = (char *) vb->val.bitstring;
1100   else
1101   {
1102     dst[0] = 0;
1103     return (EINVAL);
1104   }
1105
1106   num_chars = dst_size - 1;
1107   if (num_chars > vb->val_len)
1108     num_chars = vb->val_len;
1109
1110   for (i = 0; i < num_chars; i++)
1111   {
1112     /* Check for control characters. */
1113     if ((unsigned char)src[i] < 32)
1114       return (csnmp_strvbcopy_hexstring (dst, vb, dst_size));
1115     dst[i] = src[i];
1116   }
1117   dst[num_chars] = 0;
1118
1119   return ((int) vb->val_len);
1120 } /* }}} int csnmp_strvbcopy */
1121
1122 static int csnmp_instance_list_add (csnmp_list_instances_t **head,
1123     csnmp_list_instances_t **tail,
1124     const struct snmp_pdu *res,
1125     const host_definition_t *hd, const data_definition_t *dd)
1126 {
1127   csnmp_list_instances_t *il;
1128   struct variable_list *vb;
1129   oid_t vb_name;
1130   int status;
1131   uint32_t i;
1132   uint32_t is_matched;
1133
1134   /* Set vb on the last variable */
1135   for (vb = res->variables;
1136       (vb != NULL) && (vb->next_variable != NULL);
1137       vb = vb->next_variable)
1138     /* do nothing */;
1139   if (vb == NULL)
1140     return (-1);
1141
1142   csnmp_oid_init (&vb_name, vb->name, vb->name_length);
1143
1144   il = malloc (sizeof (*il));
1145   if (il == NULL)
1146   {
1147     ERROR ("snmp plugin: malloc failed.");
1148     return (-1);
1149   }
1150   memset (il, 0, sizeof (*il));
1151   il->next = NULL;
1152
1153   status = csnmp_oid_suffix (&il->suffix, &vb_name, &dd->instance.oid);
1154   if (status != 0)
1155   {
1156     sfree (il);
1157     return (status);
1158   }
1159
1160   /* Get instance name */
1161   if ((vb->type == ASN_OCTET_STR) || (vb->type == ASN_BIT_STR))
1162   {
1163     char *ptr;
1164
1165     csnmp_strvbcopy (il->instance, vb, sizeof (il->instance));
1166     is_matched = 0;
1167     for (i = 0; i < dd->ignores_len; i++)
1168     {
1169       status = fnmatch(dd->ignores[i], il->instance, 0);
1170       if (status == 0)
1171       {
1172         if (dd->invert_match == 0)
1173         {
1174           sfree(il);
1175           return 0;
1176         }
1177         else
1178         {
1179           is_matched = 1;
1180           break;
1181         }
1182       }
1183     }
1184     if (dd->invert_match != 0 && is_matched == 0)
1185     {
1186       sfree(il);
1187       return 0;
1188     }
1189     for (ptr = il->instance; *ptr != '\0'; ptr++)
1190     {
1191       if ((*ptr > 0) && (*ptr < 32))
1192         *ptr = ' ';
1193       else if (*ptr == '/')
1194         *ptr = '_';
1195     }
1196     DEBUG ("snmp plugin: il->instance = `%s';", il->instance);
1197   }
1198   else
1199   {
1200     value_t val = csnmp_value_list_to_value (vb, DS_TYPE_COUNTER,
1201         /* scale = */ 1.0, /* shift = */ 0.0, hd->name, dd->name);
1202     ssnprintf (il->instance, sizeof (il->instance),
1203         "%llu", val.counter);
1204   }
1205
1206   /* TODO: Debugging output */
1207
1208   if (*head == NULL)
1209     *head = il;
1210   else
1211     (*tail)->next = il;
1212   *tail = il;
1213
1214   return (0);
1215 } /* int csnmp_instance_list_add */
1216
1217 static int csnmp_dispatch_table (host_definition_t *host, data_definition_t *data,
1218     csnmp_list_instances_t *instance_list,
1219     csnmp_table_values_t **value_table)
1220 {
1221   const data_set_t *ds;
1222   value_list_t vl = VALUE_LIST_INIT;
1223
1224   csnmp_list_instances_t *instance_list_ptr;
1225   csnmp_table_values_t **value_table_ptr;
1226
1227   int i;
1228   _Bool have_more;
1229   oid_t current_suffix;
1230
1231   ds = plugin_get_ds (data->type);
1232   if (!ds)
1233   {
1234     ERROR ("snmp plugin: DataSet `%s' not defined.", data->type);
1235     return (-1);
1236   }
1237   assert (ds->ds_num == data->values_len);
1238
1239   instance_list_ptr = instance_list;
1240
1241   value_table_ptr = malloc (sizeof (*value_table_ptr) * data->values_len);
1242   if (value_table_ptr == NULL)
1243     return (-1);
1244   for (i = 0; i < data->values_len; i++)
1245     value_table_ptr[i] = value_table[i];
1246
1247   vl.values_len = ds->ds_num;
1248   vl.values = malloc (sizeof (*vl.values) * vl.values_len);
1249   if (vl.values == NULL)
1250   {
1251     ERROR ("snmp plugin: malloc failed.");
1252     sfree (value_table_ptr);
1253     return (-1);
1254   }
1255
1256   sstrncpy (vl.host, host->name, sizeof (vl.host));
1257   sstrncpy (vl.plugin, "snmp", sizeof (vl.plugin));
1258
1259   vl.interval = host->interval;
1260
1261   have_more = 1;
1262   memset (&current_suffix, 0, sizeof (current_suffix));
1263   while (have_more)
1264   {
1265     _Bool suffix_skipped = 0;
1266
1267     /* Determine next suffix to handle. */
1268     if (instance_list != NULL)
1269     {
1270       if (instance_list_ptr == NULL)
1271       {
1272         have_more = 0;
1273         continue;
1274       }
1275
1276       memcpy (&current_suffix, &instance_list_ptr->suffix, sizeof (current_suffix));
1277     }
1278     else /* no instance configured */
1279     {
1280       csnmp_table_values_t *ptr = value_table_ptr[0];
1281       if (ptr == NULL)
1282       {
1283         have_more = 0;
1284         continue;
1285       }
1286
1287       memcpy (&current_suffix, &ptr->suffix, sizeof (current_suffix));
1288     }
1289
1290     /* Update all the value_table_ptr to point at the entry with the same
1291      * trailing partial OID */
1292     for (i = 0; i < data->values_len; i++)
1293     {
1294       while ((value_table_ptr[i] != NULL)
1295           && (csnmp_oid_compare (&value_table_ptr[i]->suffix, &current_suffix) < 0))
1296         value_table_ptr[i] = value_table_ptr[i]->next;
1297
1298       if (value_table_ptr[i] == NULL)
1299       {
1300         have_more = 0;
1301         break;
1302       }
1303       else if (csnmp_oid_compare (&value_table_ptr[i]->suffix, &current_suffix) > 0)
1304       {
1305         /* This suffix is missing in the subtree. Indicate this with the
1306          * "suffix_skipped" flag and try the next instance / suffix. */
1307         suffix_skipped = 1;
1308         break;
1309       }
1310     } /* for (i = 0; i < columns; i++) */
1311
1312     if (!have_more)
1313       break;
1314
1315     /* Matching the values failed. Start from the beginning again. */
1316     if (suffix_skipped)
1317     {
1318       if (instance_list != NULL)
1319         instance_list_ptr = instance_list_ptr->next;
1320       else
1321         value_table_ptr[0] = value_table_ptr[0]->next;
1322
1323       continue;
1324     }
1325
1326     /* if we reach this line, all value_table_ptr[i] are non-NULL and are set
1327      * to the same subid. instance_list_ptr is either NULL or points to the
1328      * same subid, too. */
1329 #if COLLECT_DEBUG
1330     for (i = 1; i < data->values_len; i++)
1331     {
1332       assert (value_table_ptr[i] != NULL);
1333       assert (csnmp_oid_compare (&value_table_ptr[i-1]->suffix,
1334             &value_table_ptr[i]->suffix) == 0);
1335     }
1336     assert ((instance_list_ptr == NULL)
1337         || (csnmp_oid_compare (&instance_list_ptr->suffix,
1338             &value_table_ptr[0]->suffix) == 0));
1339 #endif
1340
1341     sstrncpy (vl.type, data->type, sizeof (vl.type));
1342
1343     {
1344       char temp[DATA_MAX_NAME_LEN];
1345
1346       if (instance_list_ptr == NULL)
1347         csnmp_oid_to_string (temp, sizeof (temp), &current_suffix);
1348       else
1349         sstrncpy (temp, instance_list_ptr->instance, sizeof (temp));
1350
1351       if (data->instance_prefix == NULL)
1352         sstrncpy (vl.type_instance, temp, sizeof (vl.type_instance));
1353       else
1354         ssnprintf (vl.type_instance, sizeof (vl.type_instance), "%s%s",
1355             data->instance_prefix, temp);
1356     }
1357
1358     for (i = 0; i < data->values_len; i++)
1359       vl.values[i] = value_table_ptr[i]->value;
1360
1361     /* If we get here `vl.type_instance' and all `vl.values' have been set */
1362     plugin_dispatch_values (&vl);
1363
1364     if (instance_list != NULL)
1365       instance_list_ptr = instance_list_ptr->next;
1366     else
1367       value_table_ptr[0] = value_table_ptr[0]->next;
1368   } /* while (have_more) */
1369
1370   sfree (vl.values);
1371   sfree (value_table_ptr);
1372
1373   return (0);
1374 } /* int csnmp_dispatch_table */
1375
1376 static int csnmp_read_table (host_definition_t *host, data_definition_t *data)
1377 {
1378   struct snmp_pdu *req;
1379   struct snmp_pdu *res;
1380   struct variable_list *vb;
1381
1382   const data_set_t *ds;
1383
1384   uint32_t oid_list_len = (uint32_t) (data->values_len + 1);
1385   /* Holds the last OID returned by the device. We use this in the GETNEXT
1386    * request to proceed. */
1387   oid_t oid_list[oid_list_len];
1388   /* Set to false when an OID has left its subtree so we don't re-request it
1389    * again. */
1390   _Bool oid_list_todo[oid_list_len];
1391
1392   int status;
1393   int i;
1394   uint32_t j;
1395
1396   /* `value_list_head' and `value_list_tail' implement a linked list for each
1397    * value. `instance_list_head' and `instance_list_tail' implement a linked list of
1398    * instance names. This is used to jump gaps in the table. */
1399   csnmp_list_instances_t *instance_list_head;
1400   csnmp_list_instances_t *instance_list_tail;
1401   csnmp_table_values_t **value_list_head;
1402   csnmp_table_values_t **value_list_tail;
1403
1404   DEBUG ("snmp plugin: csnmp_read_table (host = %s, data = %s)",
1405       host->name, data->name);
1406
1407   if (host->sess_handle == NULL)
1408   {
1409     DEBUG ("snmp plugin: csnmp_read_table: host->sess_handle == NULL");
1410     return (-1);
1411   }
1412
1413   ds = plugin_get_ds (data->type);
1414   if (!ds)
1415   {
1416     ERROR ("snmp plugin: DataSet `%s' not defined.", data->type);
1417     return (-1);
1418   }
1419
1420   if (ds->ds_num != data->values_len)
1421   {
1422     ERROR ("snmp plugin: DataSet `%s' requires %i values, but config talks about %i",
1423         data->type, ds->ds_num, data->values_len);
1424     return (-1);
1425   }
1426
1427   /* We need a copy of all the OIDs, because GETNEXT will destroy them. */
1428   memcpy (oid_list, data->values, data->values_len * sizeof (oid_t));
1429   if (data->instance.oid.oid_len > 0)
1430     memcpy (oid_list + data->values_len, &data->instance.oid, sizeof (oid_t));
1431   else /* no InstanceFrom option specified. */
1432     oid_list_len--;
1433
1434   for (j = 0; j < oid_list_len; j++)
1435     oid_list_todo[j] = 1;
1436
1437   /* We're going to construct n linked lists, one for each "value".
1438    * value_list_head will contain pointers to the heads of these linked lists,
1439    * value_list_tail will contain pointers to the tail of the lists. */
1440   value_list_head = calloc (data->values_len, sizeof (*value_list_head));
1441   value_list_tail = calloc (data->values_len, sizeof (*value_list_tail));
1442   if ((value_list_head == NULL) || (value_list_tail == NULL))
1443   {
1444     ERROR ("snmp plugin: csnmp_read_table: calloc failed.");
1445     sfree (value_list_head);
1446     sfree (value_list_tail);
1447     return (-1);
1448   }
1449
1450   instance_list_head = NULL;
1451   instance_list_tail = NULL;
1452
1453   status = 0;
1454   while (status == 0)
1455   {
1456     int oid_list_todo_num;
1457
1458     req = snmp_pdu_create (SNMP_MSG_GETNEXT);
1459     if (req == NULL)
1460     {
1461       ERROR ("snmp plugin: snmp_pdu_create failed.");
1462       status = -1;
1463       break;
1464     }
1465
1466     oid_list_todo_num = 0;
1467     for (j = 0; j < oid_list_len; j++)
1468     {
1469       /* Do not rerequest already finished OIDs */
1470       if (!oid_list_todo[j])
1471         continue;
1472       oid_list_todo_num++;
1473       snmp_add_null_var (req, oid_list[j].oid, oid_list[j].oid_len);
1474     }
1475
1476     if (oid_list_todo_num == 0)
1477     {
1478       /* The request is still empty - so we are finished */
1479       DEBUG ("snmp plugin: all variables have left their subtree");
1480       status = 0;
1481       break;
1482     }
1483
1484     res = NULL;
1485     status = snmp_sess_synch_response (host->sess_handle, req, &res);
1486     if ((status != STAT_SUCCESS) || (res == NULL))
1487     {
1488       char *errstr = NULL;
1489
1490       snmp_sess_error (host->sess_handle, NULL, NULL, &errstr);
1491
1492       c_complain (LOG_ERR, &host->complaint,
1493           "snmp plugin: host %s: snmp_sess_synch_response failed: %s",
1494           host->name, (errstr == NULL) ? "Unknown problem" : errstr);
1495
1496       if (res != NULL)
1497         snmp_free_pdu (res);
1498       res = NULL;
1499
1500       sfree (errstr);
1501       csnmp_host_close_session (host);
1502
1503       status = -1;
1504       break;
1505     }
1506
1507     status = 0;
1508     assert (res != NULL);
1509     c_release (LOG_INFO, &host->complaint,
1510         "snmp plugin: host %s: snmp_sess_synch_response successful.",
1511         host->name);
1512
1513     vb = res->variables;
1514     if (vb == NULL)
1515     {
1516       status = -1;
1517       break;
1518     }
1519
1520     for (vb = res->variables, i = 0; (vb != NULL); vb = vb->next_variable, i++)
1521     {
1522       /* Calculate value index from todo list */
1523       while (!oid_list_todo[i] && (i < oid_list_len))
1524         i++;
1525
1526       /* An instance is configured and the res variable we process is the
1527        * instance value (last index) */
1528       if ((data->instance.oid.oid_len > 0) && (i == data->values_len))
1529       {
1530         if ((vb->type == SNMP_ENDOFMIBVIEW)
1531             || (snmp_oid_ncompare (data->instance.oid.oid,
1532                 data->instance.oid.oid_len,
1533                 vb->name, vb->name_length,
1534                 data->instance.oid.oid_len) != 0))
1535         {
1536           DEBUG ("snmp plugin: host = %s; data = %s; Instance left its subtree.",
1537               host->name, data->name);
1538           oid_list_todo[i] = 0;
1539           continue;
1540         }
1541
1542         /* Allocate a new `csnmp_list_instances_t', insert the instance name and
1543          * add it to the list */
1544         if (csnmp_instance_list_add (&instance_list_head, &instance_list_tail,
1545               res, host, data) != 0)
1546         {
1547           ERROR ("snmp plugin: csnmp_instance_list_add failed.");
1548           status = -1;
1549           break;
1550         }
1551       }
1552       else /* The variable we are processing is a normal value */
1553       {
1554         csnmp_table_values_t *vt;
1555         oid_t vb_name;
1556         oid_t suffix;
1557         int ret;
1558
1559         csnmp_oid_init (&vb_name, vb->name, vb->name_length);
1560
1561         /* Calculate the current suffix. This is later used to check that the
1562          * suffix is increasing. This also checks if we left the subtree */
1563         ret = csnmp_oid_suffix (&suffix, &vb_name, data->values + i);
1564         if (ret != 0)
1565         {
1566           DEBUG ("snmp plugin: host = %s; data = %s; i = %i; "
1567               "Value probably left its subtree.",
1568               host->name, data->name, i);
1569           oid_list_todo[i] = 0;
1570           continue;
1571         }
1572
1573         /* Make sure the OIDs returned by the agent are increasing. Otherwise our
1574          * table matching algorithm will get confused. */
1575         if ((value_list_tail[i] != NULL)
1576             && (csnmp_oid_compare (&suffix, &value_list_tail[i]->suffix) <= 0))
1577         {
1578           DEBUG ("snmp plugin: host = %s; data = %s; i = %i; "
1579               "Suffix is not increasing.",
1580               host->name, data->name, i);
1581           oid_list_todo[i] = 0;
1582           continue;
1583         }
1584
1585         vt = malloc (sizeof (*vt));
1586         if (vt == NULL)
1587         {
1588           ERROR ("snmp plugin: malloc failed.");
1589           status = -1;
1590           break;
1591         }
1592         memset (vt, 0, sizeof (*vt));
1593
1594         vt->value = csnmp_value_list_to_value (vb, ds->ds[i].type,
1595             data->scale, data->shift, host->name, data->name);
1596         memcpy (&vt->suffix, &suffix, sizeof (vt->suffix));
1597         vt->next = NULL;
1598
1599         if (value_list_tail[i] == NULL)
1600           value_list_head[i] = vt;
1601         else
1602           value_list_tail[i]->next = vt;
1603         value_list_tail[i] = vt;
1604       }
1605
1606       /* Copy OID to oid_list[i] */
1607       memcpy (oid_list[i].oid, vb->name, sizeof (oid) * vb->name_length);
1608       oid_list[i].oid_len = vb->name_length;
1609
1610     } /* for (vb = res->variables ...) */
1611
1612     if (res != NULL)
1613       snmp_free_pdu (res);
1614     res = NULL;
1615   } /* while (status == 0) */
1616
1617   if (res != NULL)
1618     snmp_free_pdu (res);
1619   res = NULL;
1620
1621   if (req != NULL)
1622     snmp_free_pdu (req);
1623   req = NULL;
1624
1625   if (status == 0)
1626     csnmp_dispatch_table (host, data, instance_list_head, value_list_head);
1627
1628   /* Free all allocated variables here */
1629   while (instance_list_head != NULL)
1630   {
1631     csnmp_list_instances_t *next = instance_list_head->next;
1632     sfree (instance_list_head);
1633     instance_list_head = next;
1634   }
1635
1636   for (i = 0; i < data->values_len; i++)
1637   {
1638     while (value_list_head[i] != NULL)
1639     {
1640       csnmp_table_values_t *next = value_list_head[i]->next;
1641       sfree (value_list_head[i]);
1642       value_list_head[i] = next;
1643     }
1644   }
1645
1646   sfree (value_list_head);
1647   sfree (value_list_tail);
1648
1649   return (0);
1650 } /* int csnmp_read_table */
1651
1652 static int csnmp_read_value (host_definition_t *host, data_definition_t *data)
1653 {
1654   struct snmp_pdu *req;
1655   struct snmp_pdu *res;
1656   struct variable_list *vb;
1657
1658   const data_set_t *ds;
1659   value_list_t vl = VALUE_LIST_INIT;
1660
1661   int status;
1662   int i;
1663
1664   DEBUG ("snmp plugin: csnmp_read_value (host = %s, data = %s)",
1665       host->name, data->name);
1666
1667   if (host->sess_handle == NULL)
1668   {
1669     DEBUG ("snmp plugin: csnmp_read_table: host->sess_handle == NULL");
1670     return (-1);
1671   }
1672
1673   ds = plugin_get_ds (data->type);
1674   if (!ds)
1675   {
1676     ERROR ("snmp plugin: DataSet `%s' not defined.", data->type);
1677     return (-1);
1678   }
1679
1680   if (ds->ds_num != data->values_len)
1681   {
1682     ERROR ("snmp plugin: DataSet `%s' requires %i values, but config talks about %i",
1683         data->type, ds->ds_num, data->values_len);
1684     return (-1);
1685   }
1686
1687   vl.values_len = ds->ds_num;
1688   vl.values = (value_t *) malloc (sizeof (value_t) * vl.values_len);
1689   if (vl.values == NULL)
1690     return (-1);
1691   for (i = 0; i < vl.values_len; i++)
1692   {
1693     if (ds->ds[i].type == DS_TYPE_COUNTER)
1694       vl.values[i].counter = 0;
1695     else
1696       vl.values[i].gauge = NAN;
1697   }
1698
1699   sstrncpy (vl.host, host->name, sizeof (vl.host));
1700   sstrncpy (vl.plugin, "snmp", sizeof (vl.plugin));
1701   sstrncpy (vl.type, data->type, sizeof (vl.type));
1702   sstrncpy (vl.type_instance, data->instance.string, sizeof (vl.type_instance));
1703
1704   vl.interval = host->interval;
1705
1706   req = snmp_pdu_create (SNMP_MSG_GET);
1707   if (req == NULL)
1708   {
1709     ERROR ("snmp plugin: snmp_pdu_create failed.");
1710     sfree (vl.values);
1711     return (-1);
1712   }
1713
1714   for (i = 0; i < data->values_len; i++)
1715     snmp_add_null_var (req, data->values[i].oid, data->values[i].oid_len);
1716
1717   res = NULL;
1718   status = snmp_sess_synch_response (host->sess_handle, req, &res);
1719
1720   if ((status != STAT_SUCCESS) || (res == NULL))
1721   {
1722     char *errstr = NULL;
1723
1724     snmp_sess_error (host->sess_handle, NULL, NULL, &errstr);
1725     ERROR ("snmp plugin: host %s: snmp_sess_synch_response failed: %s",
1726         host->name, (errstr == NULL) ? "Unknown problem" : errstr);
1727
1728     if (res != NULL)
1729       snmp_free_pdu (res);
1730     res = NULL;
1731
1732     sfree (errstr);
1733     csnmp_host_close_session (host);
1734
1735     return (-1);
1736   }
1737
1738
1739   for (vb = res->variables; vb != NULL; vb = vb->next_variable)
1740   {
1741 #if COLLECT_DEBUG
1742     char buffer[1024];
1743     snprint_variable (buffer, sizeof (buffer),
1744         vb->name, vb->name_length, vb);
1745     DEBUG ("snmp plugin: Got this variable: %s", buffer);
1746 #endif /* COLLECT_DEBUG */
1747
1748     for (i = 0; i < data->values_len; i++)
1749       if (snmp_oid_compare (data->values[i].oid, data->values[i].oid_len,
1750             vb->name, vb->name_length) == 0)
1751         vl.values[i] = csnmp_value_list_to_value (vb, ds->ds[i].type,
1752             data->scale, data->shift, host->name, data->name);
1753   } /* for (res->variables) */
1754
1755   if (res != NULL)
1756     snmp_free_pdu (res);
1757   res = NULL;
1758
1759   DEBUG ("snmp plugin: -> plugin_dispatch_values (&vl);");
1760   plugin_dispatch_values (&vl);
1761   sfree (vl.values);
1762
1763   return (0);
1764 } /* int csnmp_read_value */
1765
1766 static int csnmp_read_host (user_data_t *ud)
1767 {
1768   host_definition_t *host;
1769   cdtime_t time_start;
1770   cdtime_t time_end;
1771   int status;
1772   int success;
1773   int i;
1774
1775   host = ud->data;
1776
1777   if (host->interval == 0)
1778     host->interval = plugin_get_interval ();
1779
1780   time_start = cdtime ();
1781
1782   if (host->sess_handle == NULL)
1783     csnmp_host_open_session (host);
1784
1785   if (host->sess_handle == NULL)
1786     return (-1);
1787
1788   success = 0;
1789   for (i = 0; i < host->data_list_len; i++)
1790   {
1791     data_definition_t *data = host->data_list[i];
1792
1793     if (data->is_table)
1794       status = csnmp_read_table (host, data);
1795     else
1796       status = csnmp_read_value (host, data);
1797
1798     if (status == 0)
1799       success++;
1800   }
1801
1802   time_end = cdtime ();
1803   if ((time_end - time_start) > host->interval)
1804   {
1805     WARNING ("snmp plugin: Host `%s' should be queried every %.3f "
1806         "seconds, but reading all values takes %.3f seconds.",
1807         host->name,
1808         CDTIME_T_TO_DOUBLE (host->interval),
1809         CDTIME_T_TO_DOUBLE (time_end - time_start));
1810   }
1811
1812   if (success == 0)
1813     return (-1);
1814
1815   return (0);
1816 } /* int csnmp_read_host */
1817
1818 static int csnmp_init (void)
1819 {
1820   call_snmp_init_once ();
1821
1822   return (0);
1823 } /* int csnmp_init */
1824
1825 static int csnmp_shutdown (void)
1826 {
1827   data_definition_t *data_this;
1828   data_definition_t *data_next;
1829
1830   /* When we get here, the read threads have been stopped and all the
1831    * `host_definition_t' will be freed. */
1832   DEBUG ("snmp plugin: Destroying all data definitions.");
1833
1834   data_this = data_head;
1835   data_head = NULL;
1836   while (data_this != NULL)
1837   {
1838     data_next = data_this->next;
1839
1840     sfree (data_this->name);
1841     sfree (data_this->type);
1842     sfree (data_this->values);
1843     sfree (data_this->ignores);
1844     sfree (data_this);
1845
1846     data_this = data_next;
1847   }
1848
1849   return (0);
1850 } /* int csnmp_shutdown */
1851
1852 void module_register (void)
1853 {
1854   plugin_register_complex_config ("snmp", csnmp_config);
1855   plugin_register_init ("snmp", csnmp_init);
1856   plugin_register_shutdown ("snmp", csnmp_shutdown);
1857 } /* void module_register */
1858
1859 /*
1860  * vim: shiftwidth=2 softtabstop=2 tabstop=8 fdm=marker
1861  */