src/powerdns.c

   1 /**
   2  * collectd - src/powerdns.c
   3  * Copyright (C) 2007-2008  C-Ware, Inc.
   4  * Copyright (C) 2008       Florian Forster
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU General Public License as published by the
   8  * Free Software Foundation; only version 2 of the License is applicable.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  18  *
  19  * Author:
  20  *   Luke Heberling <lukeh at c-ware.com>
  21  *   Florian Forster <octo at verplant.org>
  22  *
  23  * DESCRIPTION
  24  *   Queries a PowerDNS control socket for statistics
  25  **/
  26
  27 #include "collectd.h"
  28 #include "common.h"
  29 #include "plugin.h"
  30 #include "configfile.h"
  31 #include "utils_llist.h"
  32
  33 #include <sys/stat.h>
  34 #include <unistd.h>
  35 #include <stdlib.h>
  36 #include <stdio.h>
  37 #include <string.h>
  38 #include <errno.h>
  39 #include <sys/types.h>
  40 #include <sys/socket.h>
  41 #include <sys/un.h>
  42
  43 #ifndef UNIX_PATH_MAX
  44 # define UNIX_PATH_MAX sizeof (((struct sockaddr_un *)0)->sun_path)
  45 #endif
  46 #define FUNC_ERROR(func) do { char errbuf[1024]; ERROR ("powerdns plugin: %s failed: %s", func, sstrerror (errno, errbuf, sizeof (errbuf))); } while (0)
  47
  48 #define SERVER_SOCKET  LOCALSTATEDIR"/run/pdns.controlsocket"
  49 #define SERVER_COMMAND "SHOW * \n"
  50
  51 #define RECURSOR_SOCKET  LOCALSTATEDIR"/run/pdns_recursor.controlsocket"
  52 #define RECURSOR_COMMAND "get noerror-answers nxdomain-answers " \
  53   "servfail-answers sys-msec user-msec qa-latency cache-entries cache-hits " \
  54   "cache-misses questions\n"
  55
  56 struct list_item_s;
  57 typedef struct list_item_s list_item_t;
  58
  59 struct list_item_s
  60 {
  61   enum
  62   {
  63     SRV_AUTHORITATIVE,
  64     SRV_RECURSOR
  65   } server_type;
  66   int (*func) (list_item_t *item);
  67   char *instance;
  68
  69   char **fields;
  70   int fields_num;
  71   char *command;
  72
  73   struct sockaddr_un sockaddr;
  74   int socktype;
  75 };
  76
  77 struct statname_lookup_s
  78 {
  79   char *name;
  80   char *type;
  81   char *type_instance;
  82 };
  83 typedef struct statname_lookup_s statname_lookup_t;
  84
  85 /* Description of statistics returned by the recursor: {{{
  86 all-outqueries      counts the number of outgoing UDP queries since starting
  87 answers0-1          counts the number of queries answered within 1 milisecond
  88 answers100-1000     counts the number of queries answered within 1 second
  89 answers10-100       counts the number of queries answered within 100 miliseconds
  90 answers1-10         counts the number of queries answered within 10 miliseconds
  91 answers-slow        counts the number of queries answered after 1 second
  92 cache-entries       shows the number of entries in the cache
  93 cache-hits          counts the number of cache hits since starting
  94 cache-misses        counts the number of cache misses since starting
  95 chain-resends       number of queries chained to existing outstanding query
  96 client-parse-errors counts number of client packets that could not be parsed
  97 concurrent-queries  shows the number of MThreads currently running
  98 dlg-only-drops      number of records dropped because of delegation only setting
  99 negcache-entries    shows the number of entries in the Negative answer cache
 100 noerror-answers     counts the number of times it answered NOERROR since starting
 101 nsspeeds-entries    shows the number of entries in the NS speeds map
 102 nsset-invalidations number of times an nsset was dropped because it no longer worked
 103 nxdomain-answers    counts the number of times it answered NXDOMAIN since starting
 104 outgoing-timeouts   counts the number of timeouts on outgoing UDP queries since starting
 105 qa-latency          shows the current latency average
 106 questions           counts all End-user initiated queries with the RD bit set
 107 resource-limits     counts number of queries that could not be performed because of resource limits
 108 server-parse-errors counts number of server replied packets that could not be parsed
 109 servfail-answers    counts the number of times it answered SERVFAIL since starting
 110 spoof-prevents      number of times PowerDNS considered itself spoofed, and dropped the data
 111 sys-msec            number of CPU milliseconds spent in 'system' mode
 112 tcp-client-overflow number of times an IP address was denied TCP access because it already had too many connections
 113 tcp-outqueries      counts the number of outgoing TCP queries since starting
 114 tcp-questions       counts all incoming TCP queries (since starting)
 115 throttled-out       counts the number of throttled outgoing UDP queries since starting
 116 throttle-entries    shows the number of entries in the throttle map
 117 unauthorized-tcp    number of TCP questions denied because of allow-from restrictions
 118 unauthorized-udp    number of UDP questions denied because of allow-from restrictions
 119 unexpected-packets  number of answers from remote servers that were unexpected (might point to spoofing)
 120 uptime              number of seconds process has been running (since 3.1.5)
 121 user-msec           number of CPU milliseconds spent in 'user' mode
 122 }}} */
 123
 124 const char* const default_server_fields[] = /* {{{ */
 125 {
 126   "latency"
 127   "packetcache-hit",
 128   "packetcache-miss",
 129   "packetcache-size",
 130   "query-cache-hit",
 131   "query-cache-miss",
 132   "recursing-answers",
 133   "recursing-questions",
 134   "tcp-answers",
 135   "tcp-queries",
 136   "udp-answers",
 137   "udp-queries",
 138 }; /* }}} */
 139 int default_server_fields_num = STATIC_ARRAY_SIZE (default_server_fields);
 140
 141 statname_lookup_t lookup_table[] = /* {{{ */
 142 {
 143   /*********************
 144    * Server statistics *
 145    *********************/
 146   /* Questions */
 147   {"recursing-questions",    "dns_question", "recurse"},
 148   {"tcp-queries",            "dns_question", "tcp"},
 149   {"udp-queries",            "dns_question", "udp"},
 150
 151   /* Answers */
 152   {"recursing-answers",      "dns_answer",   "recurse"},
 153   {"tcp-answers",            "dns_answer",   "tcp"},
 154   {"udp-answers",            "dns_answer",   "udp"},
 155
 156   /* Cache stuff */
 157   {"packetcache-hit",        "cache_result", "packet-hit"},
 158   {"packetcache-miss",       "cache_result", "packet-miss"},
 159   {"packetcache-size",       "cache_size",   "packet"},
 160   {"query-cache-hit",        "cache_result", "query-hit"},
 161   {"query-cache-miss",       "cache_result", "query-miss"},
 162
 163   /* Latency */
 164   {"latency",                "latency",      NULL},
 165
 166   /* Other stuff.. */
 167   {"corrupt-packets",        "ipt_packets",  "corrupt"},
 168   {"deferred-cache-inserts", "counter",      "cache-deferred_insert"},
 169   {"deferred-cache-lookup",  "counter",      "cache-deferred_lookup"},
 170   {"qsize-a",                "cache_size",   "answers"},
 171   {"qsize-q",                "cache_size",   "questions"},
 172   {"servfail-packets",       "ipt_packets",  "servfail"},
 173   {"timedout-packets",       "ipt_packets",  "timeout"},
 174   {"udp4-answers",           "dns_answer",   "udp4"},
 175   {"udp4-queries",           "dns_question", "queries-udp4"},
 176   {"udp6-answers",           "dns_answer",   "udp6"},
 177   {"udp6-queries",           "dns_question", "queries-udp6"},
 178
 179   /***********************
 180    * Recursor statistics *
 181    ***********************/
 182   /* Answers by return code */
 183   {"noerror-answers",     "dns_rcode",    "NOERROR"},
 184   {"nxdomain-answers",    "dns_rcode",    "NXDOMAIN"},
 185   {"servfail-answers",    "dns_rcode",    "SERVFAIL"},
 186
 187   /* CPU utilization */
 188   {"sys-msec",            "cpu",          "system"},
 189   {"user-msec",           "cpu",          "user"},
 190
 191   /* Question-to-answer latency */
 192   {"qa-latency",          "latency",      NULL},
 193
 194   /* Cache */
 195   {"cache-entries",       "cache_size",   NULL},
 196   {"cache-hits",          "cache_result", "hit"},
 197   {"cache-misses",        "cache_result", "miss"},
 198
 199   /* Total number of questions.. */
 200   {"questions",           "dns_qtype",    "total"},
 201
 202   /* All the other stuff.. */
 203   {"all-outqueries",      "dns_question", "outgoing"},
 204   {"answers0-1",          "dns_answer",   "0_1"},
 205   {"answers1-10",         "dns_answer",   "1_10"},
 206   {"answers10-100",       "dns_answer",   "10_100"},
 207   {"answers100-1000",     "dns_answer",   "100_1000"},
 208   {"answers-slow",        "dns_answer",   "slow"},
 209   {"chain-resends",       "dns_question", "chained"},
 210   {"client-parse-errors", "counter",      "drops-client_parse_error"},
 211   {"concurrent-queries",  "dns_question", "concurrent"},
 212   {"dlg-only-drops",      "counter",      "drops-delegation_only"},
 213   {"negcache-entries",    "cache_size",   "negative"},
 214   {"nsspeeds-entries",    "gauge",        "entries-ns_speeds"},
 215   {"nsset-invalidations", "counter",      "ns_set_invalidation"},
 216   {"outgoing-timeouts",   "counter",      "drops-timeout_outgoing"},
 217   {"resource-limits",     "counter",      "drops-resource_limit"},
 218   {"server-parse-errors", "counter",      "drops-server_parse_error"},
 219   {"spoof-prevents",      "counter",      "drops-spoofed"},
 220   {"tcp-client-overflow", "counter",      "denied-client_overflow_tcp"},
 221   {"tcp-outqueries",      "dns_question", "outgoing-tcp"},
 222   {"tcp-questions",       "dns_question", "incoming-tcp"},
 223   {"throttled-out",       "dns_question", "outgoing-throttled"},
 224   {"throttle-entries",    "gauge",        "entries-throttle"},
 225   {"unauthorized-tcp",    "counter",      "denied-unauthorized_tcp"},
 226   {"unauthorized-udp",    "counter",      "denied-unauthorized_udp"},
 227   {"unexpected-packets",  "dns_answer",   "unexpected"}
 228   /* {"uptime", "", ""} */
 229 }; /* }}} */
 230 int lookup_table_length = STATIC_ARRAY_SIZE (lookup_table);
 231
 232 static llist_t *list = NULL;
 233
 234 #define PDNS_LOCAL_SOCKPATH LOCALSTATEDIR"/run/"PACKAGE_NAME"-powerdns"
 235 static char *local_sockpath = NULL;
 236
 237 /* TODO: Do this before 4.4:
 238  * - Recursor:
 239  *   - Complete list of known pdns -> collectd mappings.
 240  * - Update the collectd.conf(5) manpage.
 241  *
 242  * -octo
 243  */
 244
 245 /* <http://doc.powerdns.com/recursor-stats.html> */
 246 static void submit (const char *plugin_instance, /* {{{ */
 247     const char *pdns_type, const char *value)
 248 {
 249   value_list_t vl = VALUE_LIST_INIT;
 250   value_t values[1];
 251
 252   const char *type = NULL;
 253   const char *type_instance = NULL;
 254   const data_set_t *ds;
 255
 256   int i;
 257
 258   for (i = 0; i < lookup_table_length; i++)
 259     if (strcmp (lookup_table[i].name, pdns_type) == 0)
 260       break;
 261
 262   if (lookup_table[i].type == NULL)
 263     return;
 264
 265   if (i >= lookup_table_length)
 266   {
 267     INFO ("powerdns plugin: submit: Not found in lookup table: %s = %s;",
 268         pdns_type, value);
 269     return;
 270   }
 271
 272   type = lookup_table[i].type;
 273   type_instance = lookup_table[i].type_instance;
 274
 275   ds = plugin_get_ds (type);
 276   if (ds == NULL)
 277   {
 278     ERROR ("powerdns plugin: The lookup table returned type `%s', "
 279         "but I cannot find it via `plugin_get_ds'.",
 280         type);
 281     return;
 282   }
 283
 284   if (ds->ds_num != 1)
 285   {
 286     ERROR ("powerdns plugin: type `%s' has %i data sources, "
 287         "but I can only handle one.",
 288         type, ds->ds_num);
 289     return;
 290   }
 291
 292   if (0 != parse_value (value, &values[0], ds->ds[0].type))
 293   {
 294     ERROR ("powerdns plugin: Cannot convert `%s' "
 295         "to a number.", value);
 296     return;
 297   }
 298
 299   vl.values = values;
 300   vl.values_len = 1;
 301   sstrncpy (vl.host, hostname_g, sizeof (vl.host));
 302   sstrncpy (vl.plugin, "powerdns", sizeof (vl.plugin));
 303   sstrncpy (vl.type, type, sizeof (vl.type));
 304   if (type_instance != NULL)
 305     sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
 306   sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
 307
 308   plugin_dispatch_values (&vl);
 309 } /* }}} static void submit */
 310
 311 static int powerdns_get_data_dgram (list_item_t *item, /* {{{ */
 312     char **ret_buffer,
 313     size_t *ret_buffer_size)
 314 {
 315   int sd;
 316   int status;
 317
 318   char temp[4096];
 319   char *buffer = NULL;
 320   size_t buffer_size = 0;
 321
 322   struct sockaddr_un sa_unix;
 323
 324   sd = socket (PF_UNIX, item->socktype, 0);
 325   if (sd < 0)
 326   {
 327     FUNC_ERROR ("socket");
 328     return (-1);
 329   }
 330
 331   memset (&sa_unix, 0, sizeof (sa_unix));
 332   sa_unix.sun_family = AF_UNIX;
 333   sstrncpy (sa_unix.sun_path,
 334       (local_sockpath != NULL) ? local_sockpath : PDNS_LOCAL_SOCKPATH,
 335       sizeof (sa_unix.sun_path));
 336
 337   status = unlink (sa_unix.sun_path);
 338   if ((status != 0) && (errno != ENOENT))
 339   {
 340     FUNC_ERROR ("unlink");
 341     close (sd);
 342     return (-1);
 343   }
 344
 345   do /* while (0) */
 346   {
 347     /* We need to bind to a specific path, because this is a datagram socket
 348      * and otherwise the daemon cannot answer. */
 349     status = bind (sd, (struct sockaddr *) &sa_unix, sizeof (sa_unix));
 350     if (status != 0)
 351     {
 352       FUNC_ERROR ("bind");
 353       break;
 354     }
 355
 356     /* Make the socket writeable by the daemon.. */
 357     status = chmod (sa_unix.sun_path, 0666);
 358     if (status != 0)
 359     {
 360       FUNC_ERROR ("chmod");
 361       break;
 362     }
 363
 364     struct timeval timeout;
 365     timeout.tv_sec=2;
 366     if (timeout.tv_sec < interval_g * 3 / 4)
 367       timeout.tv_sec = interval_g * 3 / 4;
 368     timeout.tv_usec=0;
 369     status = setsockopt (sd, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof (timeout));
 370     if (status != 0)
 371     {
 372       FUNC_ERROR ("setsockopt");
 373       break;
 374     }
 375
 376     status = connect (sd, (struct sockaddr *) &item->sockaddr,
 377         sizeof (item->sockaddr));
 378     if (status != 0)
 379     {
 380       FUNC_ERROR ("connect");
 381       break;
 382     }
 383
 384     status = send (sd, item->command, strlen (item->command), 0);
 385     if (status < 0)
 386     {
 387       FUNC_ERROR ("send");
 388       break;
 389     }
 390
 391     status = recv (sd, temp, sizeof (temp), /* flags = */ 0);
 392     if (status < 0)
 393     {
 394       FUNC_ERROR ("recv");
 395       break;
 396     }
 397     buffer_size = status + 1;
 398     status = 0;
 399   } while (0);
 400
 401   close (sd);
 402   unlink (sa_unix.sun_path);
 403
 404   if (status != 0)
 405     return (-1);
 406
 407   assert (buffer_size > 0);
 408   buffer = (char *) malloc (buffer_size);
 409   if (buffer == NULL)
 410   {
 411     FUNC_ERROR ("malloc");
 412     return (-1);
 413   }
 414
 415   memcpy (buffer, temp, buffer_size - 1);
 416   buffer[buffer_size - 1] = 0;
 417
 418   *ret_buffer = buffer;
 419   *ret_buffer_size = buffer_size;
 420
 421   return (0);
 422 } /* }}} int powerdns_get_data_dgram */
 423
 424 static int powerdns_get_data_stream (list_item_t *item, /* {{{ */
 425     char **ret_buffer,
 426     size_t *ret_buffer_size)
 427 {
 428   int sd;
 429   int status;
 430
 431   char temp[4096];
 432   char *buffer = NULL;
 433   size_t buffer_size = 0;
 434
 435   sd = socket (PF_UNIX, item->socktype, 0);
 436   if (sd < 0)
 437   {
 438     FUNC_ERROR ("socket");
 439     return (-1);
 440   }
 441
 442   struct timeval timeout;
 443   timeout.tv_sec=5;
 444   timeout.tv_usec=0;
 445   status = setsockopt (sd, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof (timeout));
 446
 447   status = connect (sd, (struct sockaddr *) &item->sockaddr,
 448       sizeof (item->sockaddr));
 449   if (status != 0)
 450   {
 451     FUNC_ERROR ("connect");
 452     close (sd);
 453     return (-1);
 454   }
 455
 456   /* strlen + 1, because we need to send the terminating NULL byte, too. */
 457   status = send (sd, item->command, strlen (item->command) + 1,
 458       /* flags = */ 0);
 459   if (status < 0)
 460   {
 461     FUNC_ERROR ("send");
 462     close (sd);
 463     return (-1);
 464   }
 465
 466   while (42)
 467   {
 468     char *buffer_new;
 469
 470     status = recv (sd, temp, sizeof (temp), /* flags = */ 0);
 471     if (status < 0)
 472     {
 473       FUNC_ERROR ("recv");
 474       break;
 475     }
 476     else if (status == 0)
 477       break;
 478
 479     buffer_new = (char *) realloc (buffer, buffer_size + status + 1);
 480     if (buffer_new == NULL)
 481     {
 482       FUNC_ERROR ("realloc");
 483       status = -1;
 484       break;
 485     }
 486     buffer = buffer_new;
 487
 488     memcpy (buffer + buffer_size, temp, status);
 489     buffer_size += status;
 490     buffer[buffer_size] = 0;
 491   } /* while (42) */
 492   close (sd);
 493   sd = -1;
 494
 495   if (status < 0)
 496   {
 497     sfree (buffer);
 498   }
 499   else
 500   {
 501     assert (status == 0);
 502     *ret_buffer = buffer;
 503     *ret_buffer_size = buffer_size;
 504   }
 505
 506   return (status);
 507 } /* }}} int powerdns_get_data_stream */
 508
 509 static int powerdns_get_data (list_item_t *item, char **ret_buffer,
 510     size_t *ret_buffer_size)
 511 {
 512   if (item->socktype == SOCK_DGRAM)
 513     return (powerdns_get_data_dgram (item, ret_buffer, ret_buffer_size));
 514   else if (item->socktype == SOCK_STREAM)
 515     return (powerdns_get_data_stream (item, ret_buffer, ret_buffer_size));
 516   else
 517   {
 518     ERROR ("powerdns plugin: Unknown socket type: %i", (int) item->socktype);
 519     return (-1);
 520   }
 521 } /* int powerdns_get_data */
 522
 523 static int powerdns_read_server (list_item_t *item) /* {{{ */
 524 {
 525   char *buffer = NULL;
 526   size_t buffer_size = 0;
 527   int status;
 528
 529   char *dummy;
 530   char *saveptr;
 531
 532   char *key;
 533   char *value;
 534
 535   const char* const *fields;
 536   int fields_num;
 537
 538   if (item->command == NULL)
 539     item->command = strdup (SERVER_COMMAND);
 540   if (item->command == NULL)
 541   {
 542     ERROR ("powerdns plugin: strdup failed.");
 543     return (-1);
 544   }
 545
 546   status = powerdns_get_data (item, &buffer, &buffer_size);
 547   if (status != 0)
 548     return (-1);
 549
 550   if (item->fields_num != 0)
 551   {
 552     fields = (const char* const *) item->fields;
 553     fields_num = item->fields_num;
 554   }
 555   else
 556   {
 557     fields = default_server_fields;
 558     fields_num = default_server_fields_num;
 559   }
 560
 561   assert (fields != NULL);
 562   assert (fields_num > 0);
 563
 564   /* corrupt-packets=0,deferred-cache-inserts=0,deferred-cache-lookup=0,latency=0,packetcache-hit=0,packetcache-miss=0,packetcache-size=0,qsize-q=0,query-cache-hit=0,query-cache-miss=0,recursing-answers=0,recursing-questions=0,servfail-packets=0,tcp-answers=0,tcp-queries=0,timedout-packets=0,udp-answers=0,udp-queries=0,udp4-answers=0,udp4-queries=0,udp6-answers=0,udp6-queries=0, */
 565   dummy = buffer;
 566   saveptr = NULL;
 567   while ((key = strtok_r (dummy, ",", &saveptr)) != NULL)
 568   {
 569     int i;
 570
 571     dummy = NULL;
 572
 573     value = strchr (key, '=');
 574     if (value == NULL)
 575       break;
 576
 577     *value = '\0';
 578     value++;
 579
 580     if (value[0] == '\0')
 581       continue;
 582
 583     /* Check if this item was requested. */
 584     for (i = 0; i < fields_num; i++)
 585       if (strcasecmp (key, fields[i]) == 0)
 586         break;
 587     if (i >= fields_num)
 588       continue;
 589
 590     submit (item->instance, key, value);
 591   } /* while (strtok_r) */
 592
 593   sfree (buffer);
 594
 595   return (0);
 596 } /* }}} int powerdns_read_server */
 597
 598 /*
 599  * powerdns_update_recursor_command
 600  *
 601  * Creates a string that holds the command to be sent to the recursor. This
 602  * string is stores in the `command' member of the `list_item_t' passed to the
 603  * function. This function is called by `powerdns_read_recursor'.
 604  */
 605 static int powerdns_update_recursor_command (list_item_t *li) /* {{{ */
 606 {
 607   char buffer[4096];
 608   int status;
 609
 610   if (li == NULL)
 611     return (0);
 612
 613   if (li->fields_num < 1)
 614   {
 615     sstrncpy (buffer, RECURSOR_COMMAND, sizeof (buffer));
 616   }
 617   else
 618   {
 619     sstrncpy (buffer, "get ", sizeof (buffer));
 620     status = strjoin (&buffer[strlen("get ")], sizeof (buffer) - strlen ("get "),
 621         li->fields, li->fields_num,
 622         /* seperator = */ " ");
 623     if (status < 0)
 624     {
 625       ERROR ("powerdns plugin: strjoin failed.");
 626       return (-1);
 627     }
 628     buffer[sizeof (buffer) - 1] = 0;
 629     int i = strlen (buffer);
 630     if (i < sizeof (buffer) - 2)
 631     {
 632       buffer[i++] = ' ';
 633       buffer[i++] = '\n';
 634       buffer[i++] = '\0';
 635     }
 636   }
 637
 638   buffer[sizeof (buffer) - 1] = 0;
 639   li->command = strdup (buffer);
 640   if (li->command == NULL)
 641   {
 642     ERROR ("powerdns plugin: strdup failed.");
 643     return (-1);
 644   }
 645
 646   return (0);
 647 } /* }}} int powerdns_update_recursor_command */
 648
 649 static int powerdns_read_recursor (list_item_t *item) /* {{{ */
 650 {
 651   char *buffer = NULL;
 652   size_t buffer_size = 0;
 653   int status;
 654
 655   char *dummy;
 656
 657   char *keys_list;
 658   char *key;
 659   char *key_saveptr;
 660   char *value;
 661   char *value_saveptr;
 662
 663   if (item->command == NULL)
 664   {
 665     status = powerdns_update_recursor_command (item);
 666     if (status != 0)
 667     {
 668       ERROR ("powerdns plugin: powerdns_update_recursor_command failed.");
 669       return (-1);
 670     }
 671
 672     DEBUG ("powerdns plugin: powerdns_read_recursor: item->command = %s;",
 673         item->command);
 674   }
 675   assert (item->command != NULL);
 676
 677   status = powerdns_get_data (item, &buffer, &buffer_size);
 678   if (status != 0)
 679   {
 680     ERROR ("powerdns plugin: powerdns_get_data failed.");
 681     return (-1);
 682   }
 683
 684   keys_list = strdup (item->command);
 685   if (keys_list == NULL)
 686   {
 687     FUNC_ERROR ("strdup");
 688     sfree (buffer);
 689     return (-1);
 690   }
 691
 692   key_saveptr = NULL;
 693   value_saveptr = NULL;
 694
 695   /* Skip the `get' at the beginning */
 696   strtok_r (keys_list, " \t", &key_saveptr);
 697
 698   dummy = buffer;
 699   while ((value = strtok_r (dummy, " \t\n\r", &value_saveptr)) != NULL)
 700   {
 701     dummy = NULL;
 702
 703     key = strtok_r (NULL, " \t", &key_saveptr);
 704     if (key == NULL)
 705       break;
 706
 707     submit (item->instance, key, value);
 708   } /* while (strtok_r) */
 709
 710   sfree (buffer);
 711   sfree (keys_list);
 712
 713   return (0);
 714 } /* }}} int powerdns_read_recursor */
 715
 716 static int powerdns_config_add_string (const char *name, /* {{{ */
 717     char **dest,
 718     oconfig_item_t *ci)
 719 {
 720   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
 721   {
 722     WARNING ("powerdns plugin: `%s' needs exactly one string argument.",
 723         name);
 724     return (-1);
 725   }
 726
 727   sfree (*dest);
 728   *dest = strdup (ci->values[0].value.string);
 729   if (*dest == NULL)
 730     return (-1);
 731
 732   return (0);
 733 } /* }}} int powerdns_config_add_string */
 734
 735 static int powerdns_config_add_collect (list_item_t *li, /* {{{ */
 736     oconfig_item_t *ci)
 737 {
 738   int i;
 739   char **temp;
 740
 741   if (ci->values_num < 1)
 742   {
 743     WARNING ("powerdns plugin: The `Collect' option needs "
 744         "at least one argument.");
 745     return (-1);
 746   }
 747
 748   for (i = 0; i < ci->values_num; i++)
 749     if (ci->values[i].type != OCONFIG_TYPE_STRING)
 750     {
 751       WARNING ("powerdns plugin: Only string arguments are allowed to "
 752           "the `Collect' option.");
 753       return (-1);
 754     }
 755
 756   temp = (char **) realloc (li->fields,
 757       sizeof (char *) * (li->fields_num + ci->values_num));
 758   if (temp == NULL)
 759   {
 760     WARNING ("powerdns plugin: realloc failed.");
 761     return (-1);
 762   }
 763   li->fields = temp;
 764
 765   for (i = 0; i < ci->values_num; i++)
 766   {
 767     li->fields[li->fields_num] = strdup (ci->values[i].value.string);
 768     if (li->fields[li->fields_num] == NULL)
 769     {
 770       WARNING ("powerdns plugin: strdup failed.");
 771       continue;
 772     }
 773     li->fields_num++;
 774   }
 775
 776   /* Invalidate a previously computed command */
 777   sfree (li->command);
 778
 779   return (0);
 780 } /* }}} int powerdns_config_add_collect */
 781
 782 static int powerdns_config_add_server (oconfig_item_t *ci) /* {{{ */
 783 {
 784   char *socket_temp;
 785
 786   list_item_t *item;
 787   int status;
 788   int i;
 789
 790   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
 791   {
 792     WARNING ("powerdns plugin: `%s' needs exactly one string argument.",
 793         ci->key);
 794     return (-1);
 795   }
 796
 797   item = (list_item_t *) malloc (sizeof (list_item_t));
 798   if (item == NULL)
 799   {
 800     ERROR ("powerdns plugin: malloc failed.");
 801     return (-1);
 802   }
 803   memset (item, '\0', sizeof (list_item_t));
 804
 805   item->instance = strdup (ci->values[0].value.string);
 806   if (item->instance == NULL)
 807   {
 808     ERROR ("powerdns plugin: strdup failed.");
 809     sfree (item);
 810     return (-1);
 811   }
 812
 813   /*
 814    * Set default values for the members of list_item_t
 815    */
 816   if (strcasecmp ("Server", ci->key) == 0)
 817   {
 818     item->server_type = SRV_AUTHORITATIVE;
 819     item->func = powerdns_read_server;
 820     item->socktype = SOCK_STREAM;
 821     socket_temp = strdup (SERVER_SOCKET);
 822   }
 823   else if (strcasecmp ("Recursor", ci->key) == 0)
 824   {
 825     item->server_type = SRV_RECURSOR;
 826     item->func = powerdns_read_recursor;
 827     item->socktype = SOCK_DGRAM;
 828     socket_temp = strdup (RECURSOR_SOCKET);
 829   }
 830   else
 831   {
 832     /* We must never get here.. */
 833     assert (0);
 834     return (-1);
 835   }
 836
 837   status = 0;
 838   for (i = 0; i < ci->children_num; i++)
 839   {
 840     oconfig_item_t *option = ci->children + i;
 841
 842     if (strcasecmp ("Collect", option->key) == 0)
 843       status = powerdns_config_add_collect (item, option);
 844     else if (strcasecmp ("Socket", option->key) == 0)
 845       status = powerdns_config_add_string ("Socket", &socket_temp, option);
 846     else
 847     {
 848       ERROR ("powerdns plugin: Option `%s' not allowed here.", option->key);
 849       status = -1;
 850     }
 851
 852     if (status != 0)
 853       break;
 854   }
 855
 856   while (status == 0)
 857   {
 858     llentry_t *e;
 859
 860     if (socket_temp == NULL)
 861     {
 862       ERROR ("powerdns plugin: socket_temp == NULL.");
 863       status = -1;
 864       break;
 865     }
 866
 867     item->sockaddr.sun_family = AF_UNIX;
 868     sstrncpy (item->sockaddr.sun_path, socket_temp,
 869       sizeof (item->sockaddr.sun_path));
 870
 871     e = llentry_create (item->instance, item);
 872     if (e == NULL)
 873     {
 874       ERROR ("powerdns plugin: llentry_create failed.");
 875       status = -1;
 876       break;
 877     }
 878     llist_append (list, e);
 879
 880     break;
 881   }
 882
 883   if (status != 0)
 884   {
 885     sfree (item);
 886     return (-1);
 887   }
 888
 889   DEBUG ("powerdns plugin: Add server: instance = %s;", item->instance);
 890
 891   return (0);
 892 } /* }}} int powerdns_config_add_server */
 893
 894 static int powerdns_config (oconfig_item_t *ci) /* {{{ */
 895 {
 896   int i;
 897
 898   DEBUG ("powerdns plugin: powerdns_config (ci = %p);", (void *) ci);
 899
 900   if (list == NULL)
 901   {
 902     list = llist_create ();
 903
 904     if (list == NULL)
 905     {
 906       ERROR ("powerdns plugin: `llist_create' failed.");
 907       return (-1);
 908     }
 909   }
 910
 911   for (i = 0; i < ci->children_num; i++)
 912   {
 913     oconfig_item_t *option = ci->children + i;
 914
 915     if ((strcasecmp ("Server", option->key) == 0)
 916         || (strcasecmp ("Recursor", option->key) == 0))
 917       powerdns_config_add_server (option);
 918     else if (strcasecmp ("LocalSocket", option->key) == 0)
 919     {
 920       if ((option->values_num != 1) || (option->values[0].type != OCONFIG_TYPE_STRING))
 921       {
 922         WARNING ("powerdns plugin: `%s' needs exactly one string argument.", option->key);
 923       }
 924       else
 925       {
 926         char *temp = strdup (option->values[0].value.string);
 927         if (temp == NULL)
 928           return (1);
 929         sfree (local_sockpath);
 930         local_sockpath = temp;
 931       }
 932     }
 933     else
 934     {
 935       ERROR ("powerdns plugin: Option `%s' not allowed here.", option->key);
 936     }
 937   } /* for (i = 0; i < ci->children_num; i++) */
 938
 939   return (0);
 940 } /* }}} int powerdns_config */
 941
 942 static int powerdns_read (void)
 943 {
 944   llentry_t *e;
 945
 946   for (e = llist_head (list); e != NULL; e = e->next)
 947   {
 948     list_item_t *item = e->value;
 949     item->func (item);
 950   }
 951
 952   return (0);
 953 } /* static int powerdns_read */
 954
 955 static int powerdns_shutdown (void)
 956 {
 957   llentry_t *e;
 958
 959   if (list == NULL)
 960     return (0);
 961
 962   for (e = llist_head (list); e != NULL; e = e->next)
 963   {
 964     list_item_t *item = (list_item_t *) e->value;
 965     e->value = NULL;
 966
 967     sfree (item->instance);
 968     sfree (item->command);
 969     sfree (item);
 970   }
 971
 972   llist_destroy (list);
 973   list = NULL;
 974
 975   return (0);
 976 } /* static int powerdns_shutdown */
 977
 978 void module_register (void)
 979 {
 980   plugin_register_complex_config ("powerdns", powerdns_config);
 981   plugin_register_read ("powerdns", powerdns_read);
 982   plugin_register_shutdown ("powerdns", powerdns_shutdown );
 983 } /* void module_register */
 984
 985 /* vim: set sw=2 sts=2 ts=8 fdm=marker : */