src/snmp.c

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