src/network.c

   1 /**
   2  * collectd - src/network.c
   3  * Copyright (C) 2005-2013  Florian octo Forster
   4  * Copyright (C) 2009       Aman Gupta
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU Lesser General Public License as published by
   8  * the Free Software Foundation; only version 2.1 of the License is
   9  * applicable.
  10  *
  11  * This program is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  * Authors:
  21  *   Florian octo Forster <octo at collectd.org>
  22  *   Aman Gupta <aman at tmm1.net>
  23  **/
  24
  25 #define _DEFAULT_SOURCE
  26 #define _BSD_SOURCE /* For struct ip_mreq */
  27
  28 #include "collectd.h"
  29 #include "plugin.h"
  30 #include "common.h"
  31 #include "configfile.h"
  32 #include "utils_fbhash.h"
  33 #include "utils_avltree.h"
  34 #include "utils_cache.h"
  35 #include "utils_complain.h"
  36
  37 #include "network.h"
  38
  39 #if HAVE_PTHREAD_H
  40 # include <pthread.h>
  41 #endif
  42 #if HAVE_SYS_SOCKET_H
  43 # include <sys/socket.h>
  44 #endif
  45 #if HAVE_NETDB_H
  46 # include <netdb.h>
  47 #endif
  48 #if HAVE_NETINET_IN_H
  49 # include <netinet/in.h>
  50 #endif
  51 #if HAVE_ARPA_INET_H
  52 # include <arpa/inet.h>
  53 #endif
  54 #if HAVE_POLL_H
  55 # include <poll.h>
  56 #endif
  57 #if HAVE_NET_IF_H
  58 # include <net/if.h>
  59 #endif
  60
  61 #if HAVE_LIBGCRYPT
  62 # include <pthread.h>
  63 # if defined __APPLE__
  64 /* default xcode compiler throws warnings even when deprecated functionality
  65  * is not used. -Werror breaks the build because of erroneous warnings.
  66  * http://stackoverflow.com/questions/10556299/compiler-warnings-with-libgcrypt-v1-5-0/12830209#12830209
  67  */
  68 #  pragma GCC diagnostic ignored "-Wdeprecated-declarations"
  69 # endif
  70 /* FreeBSD's copy of libgcrypt extends the existing GCRYPT_NO_DEPRECATED
  71  * to properly hide all deprecated functionality.
  72  * http://svnweb.freebsd.org/ports/head/security/libgcrypt/files/patch-src__gcrypt.h.in
  73  */
  74 # define GCRYPT_NO_DEPRECATED
  75 # include <gcrypt.h>
  76 # if defined __APPLE__
  77 /* Re enable deprecation warnings */
  78 #  pragma GCC diagnostic warning "-Wdeprecated-declarations"
  79 # endif
  80 # if GCRYPT_VERSION_NUMBER < 0x010600
  81 GCRY_THREAD_OPTION_PTHREAD_IMPL;
  82 # endif
  83 #endif
  84
  85 #ifndef IPV6_ADD_MEMBERSHIP
  86 # ifdef IPV6_JOIN_GROUP
  87 #  define IPV6_ADD_MEMBERSHIP IPV6_JOIN_GROUP
  88 # else
  89 #  error "Neither IP_ADD_MEMBERSHIP nor IPV6_JOIN_GROUP is defined"
  90 # endif
  91 #endif /* !IP_ADD_MEMBERSHIP */
  92
  93 /*
  94  * Maximum size required for encryption / signing:
  95  *
  96  *    42 bytes for the encryption header
  97  * +  64 bytes for the username
  98  * -----------
  99  * = 106 bytes
 100  */
 101 #define BUFF_SIG_SIZE 106
 102
 103 /*
 104  * Private data types
 105  */
 106 #define SECURITY_LEVEL_NONE     0
 107 #if HAVE_LIBGCRYPT
 108 # define SECURITY_LEVEL_SIGN    1
 109 # define SECURITY_LEVEL_ENCRYPT 2
 110 #endif
 111 struct sockent_client
 112 {
 113         int fd;
 114         struct sockaddr_storage *addr;
 115         socklen_t                addrlen;
 116 #if HAVE_LIBGCRYPT
 117         int security_level;
 118         char *username;
 119         char *password;
 120         gcry_cipher_hd_t cypher;
 121         unsigned char password_hash[32];
 122 #endif
 123         cdtime_t next_resolve_reconnect;
 124         cdtime_t resolve_interval;
 125 };
 126
 127 struct sockent_server
 128 {
 129         int *fd;
 130         size_t fd_num;
 131 #if HAVE_LIBGCRYPT
 132         int security_level;
 133         char *auth_file;
 134         fbhash_t *userdb;
 135         gcry_cipher_hd_t cypher;
 136 #endif
 137 };
 138
 139 typedef struct sockent
 140 {
 141 #define SOCKENT_TYPE_CLIENT 1
 142 #define SOCKENT_TYPE_SERVER 2
 143         int type;
 144
 145         char *node;
 146         char *service;
 147         int interface;
 148
 149         union
 150         {
 151                 struct sockent_client client;
 152                 struct sockent_server server;
 153         } data;
 154
 155         struct sockent *next;
 156 } sockent_t;
 157
 158 /*                      1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 3 3
 159  *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 160  * +-------+-----------------------+-------------------------------+
 161  * ! Ver.  !                       ! Length                        !
 162  * +-------+-----------------------+-------------------------------+
 163  */
 164 struct part_header_s
 165 {
 166         uint16_t type;
 167         uint16_t length;
 168 };
 169 typedef struct part_header_s part_header_t;
 170
 171 /*                      1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 3 3
 172  *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 173  * +-------------------------------+-------------------------------+
 174  * ! Type                          ! Length                        !
 175  * +-------------------------------+-------------------------------+
 176  * : (Length - 4) Bytes                                            :
 177  * +---------------------------------------------------------------+
 178  */
 179 struct part_string_s
 180 {
 181         part_header_t *head;
 182         char *value;
 183 };
 184 typedef struct part_string_s part_string_t;
 185
 186 /*                      1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 3 3
 187  *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 188  * +-------------------------------+-------------------------------+
 189  * ! Type                          ! Length                        !
 190  * +-------------------------------+-------------------------------+
 191  * : (Length - 4 == 2 || 4 || 8) Bytes                             :
 192  * +---------------------------------------------------------------+
 193  */
 194 struct part_number_s
 195 {
 196         part_header_t *head;
 197         uint64_t *value;
 198 };
 199 typedef struct part_number_s part_number_t;
 200
 201 /*                      1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 3 3
 202  *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 203  * +-------------------------------+-------------------------------+
 204  * ! Type                          ! Length                        !
 205  * +-------------------------------+---------------+---------------+
 206  * ! Num of values                 ! Type0         ! Type1         !
 207  * +-------------------------------+---------------+---------------+
 208  * ! Value0                                                        !
 209  * !                                                               !
 210  * +---------------------------------------------------------------+
 211  * ! Value1                                                        !
 212  * !                                                               !
 213  * +---------------------------------------------------------------+
 214  */
 215 struct part_values_s
 216 {
 217         part_header_t *head;
 218         uint16_t *num_values;
 219         uint8_t  *values_types;
 220         value_t  *values;
 221 };
 222 typedef struct part_values_s part_values_t;
 223
 224 /*                      1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 3 3
 225  *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 226  * +-------------------------------+-------------------------------+
 227  * ! Type                          ! Length                        !
 228  * +-------------------------------+-------------------------------+
 229  * ! Hash (Bits   0 -  31)                                         !
 230  * : :                                                             :
 231  * ! Hash (Bits 224 - 255)                                         !
 232  * +---------------------------------------------------------------+
 233  */
 234 /* Minimum size */
 235 #define PART_SIGNATURE_SHA256_SIZE 36
 236 struct part_signature_sha256_s
 237 {
 238   part_header_t head;
 239   unsigned char hash[32];
 240   char *username;
 241 };
 242 typedef struct part_signature_sha256_s part_signature_sha256_t;
 243
 244 /*                      1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 3 3
 245  *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 246  * +-------------------------------+-------------------------------+
 247  * ! Type                          ! Length                        !
 248  * +-------------------------------+-------------------------------+
 249  * ! Original length               ! Padding (0 - 15 bytes)        !
 250  * +-------------------------------+-------------------------------+
 251  * ! Hash (Bits   0 -  31)                                         !
 252  * : :                                                             :
 253  * ! Hash (Bits 128 - 159)                                         !
 254  * +---------------------------------------------------------------+
 255  */
 256 /* Minimum size */
 257 #define PART_ENCRYPTION_AES256_SIZE 42
 258 struct part_encryption_aes256_s
 259 {
 260   part_header_t head;
 261   uint16_t username_length;
 262   char *username;
 263   unsigned char iv[16];
 264   /* <encrypted> */
 265   unsigned char hash[20];
 266   /*   <payload /> */
 267   /* </encrypted> */
 268 };
 269 typedef struct part_encryption_aes256_s part_encryption_aes256_t;
 270
 271 struct receive_list_entry_s
 272 {
 273   char *data;
 274   int  data_len;
 275   int  fd;
 276   struct receive_list_entry_s *next;
 277 };
 278 typedef struct receive_list_entry_s receive_list_entry_t;
 279
 280 /*
 281  * Private variables
 282  */
 283 static int network_config_ttl = 0;
 284 /* Ethernet - (IPv6 + UDP) = 1500 - (40 + 8) = 1452 */
 285 static size_t network_config_packet_size = 1452;
 286 static int network_config_forward = 0;
 287 static int network_config_stats = 0;
 288
 289 static sockent_t *sending_sockets = NULL;
 290
 291 static receive_list_entry_t *receive_list_head = NULL;
 292 static receive_list_entry_t *receive_list_tail = NULL;
 293 static pthread_mutex_t       receive_list_lock = PTHREAD_MUTEX_INITIALIZER;
 294 static pthread_cond_t        receive_list_cond = PTHREAD_COND_INITIALIZER;
 295 static uint64_t              receive_list_length = 0;
 296
 297 static sockent_t     *listen_sockets = NULL;
 298 static struct pollfd *listen_sockets_pollfd = NULL;
 299 static size_t         listen_sockets_num = 0;
 300
 301 /* The receive and dispatch threads will run as long as `listen_loop' is set to
 302  * zero. */
 303 static int       listen_loop = 0;
 304 static int       receive_thread_running = 0;
 305 static pthread_t receive_thread_id;
 306 static int       dispatch_thread_running = 0;
 307 static pthread_t dispatch_thread_id;
 308
 309 /* Buffer in which to-be-sent network packets are constructed. */
 310 static char            *send_buffer;
 311 static char            *send_buffer_ptr;
 312 static int              send_buffer_fill;
 313 static cdtime_t         send_buffer_last_update;
 314 static value_list_t     send_buffer_vl = VALUE_LIST_STATIC;
 315 static pthread_mutex_t  send_buffer_lock = PTHREAD_MUTEX_INITIALIZER;
 316
 317 /* XXX: These counters are incremented from one place only. The spot in which
 318  * the values are incremented is either only reachable by one thread (the
 319  * dispatch thread, for example) or locked by some lock (send_buffer_lock for
 320  * example). Only if neither is true, the stats_lock is acquired. The counters
 321  * are always read without holding a lock in the hope that writing 8 bytes to
 322  * memory is an atomic operation. */
 323 static derive_t stats_octets_rx  = 0;
 324 static derive_t stats_octets_tx  = 0;
 325 static derive_t stats_packets_rx = 0;
 326 static derive_t stats_packets_tx = 0;
 327 static derive_t stats_values_dispatched = 0;
 328 static derive_t stats_values_not_dispatched = 0;
 329 static derive_t stats_values_sent = 0;
 330 static derive_t stats_values_not_sent = 0;
 331 static pthread_mutex_t stats_lock = PTHREAD_MUTEX_INITIALIZER;
 332
 333 /*
 334  * Private functions
 335  */
 336 static _Bool check_receive_okay (const value_list_t *vl) /* {{{ */
 337 {
 338   uint64_t time_sent = 0;
 339   int status;
 340
 341   status = uc_meta_data_get_unsigned_int (vl,
 342       "network:time_sent", &time_sent);
 343
 344   /* This is a value we already sent. Don't allow it to be received again in
 345    * order to avoid looping. */
 346   if ((status == 0) && (time_sent >= ((uint64_t) vl->time)))
 347     return (0);
 348
 349   return (1);
 350 } /* }}} _Bool check_receive_okay */
 351
 352 static _Bool check_send_okay (const value_list_t *vl) /* {{{ */
 353 {
 354   _Bool received = 0;
 355   int status;
 356
 357   if (network_config_forward != 0)
 358     return (1);
 359
 360   if (vl->meta == NULL)
 361     return (1);
 362
 363   status = meta_data_get_boolean (vl->meta, "network:received", &received);
 364   if (status == -ENOENT)
 365     return (1);
 366   else if (status != 0)
 367   {
 368     ERROR ("network plugin: check_send_okay: meta_data_get_boolean failed "
 369         "with status %i.", status);
 370     return (1);
 371   }
 372
 373   /* By default, only *send* value lists that were not *received* by the
 374    * network plugin. */
 375   return (!received);
 376 } /* }}} _Bool check_send_okay */
 377
 378 static _Bool check_notify_received (const notification_t *n) /* {{{ */
 379 {
 380   notification_meta_t *ptr;
 381
 382   for (ptr = n->meta; ptr != NULL; ptr = ptr->next)
 383     if ((strcmp ("network:received", ptr->name) == 0)
 384         && (ptr->type == NM_TYPE_BOOLEAN))
 385       return ((_Bool) ptr->nm_value.nm_boolean);
 386
 387   return (0);
 388 } /* }}} _Bool check_notify_received */
 389
 390 static _Bool check_send_notify_okay (const notification_t *n) /* {{{ */
 391 {
 392   static c_complain_t complain_forwarding = C_COMPLAIN_INIT_STATIC;
 393   _Bool received = 0;
 394
 395   if (n->meta == NULL)
 396     return (1);
 397
 398   received = check_notify_received (n);
 399
 400   if (network_config_forward && received)
 401   {
 402     c_complain_once (LOG_ERR, &complain_forwarding,
 403         "network plugin: A notification has been received via the network "
 404         "and forwarding is enabled. Forwarding of notifications is currently "
 405         "not supported, because there is not loop-deteciton available. "
 406         "Please contact the collectd mailing list if you need this "
 407         "feature.");
 408   }
 409
 410   /* By default, only *send* value lists that were not *received* by the
 411    * network plugin. */
 412   return (!received);
 413 } /* }}} _Bool check_send_notify_okay */
 414
 415 static int network_dispatch_values (value_list_t *vl, /* {{{ */
 416     const char *username)
 417 {
 418   int status;
 419
 420   if ((vl->time <= 0)
 421       || (strlen (vl->host) <= 0)
 422       || (strlen (vl->plugin) <= 0)
 423       || (strlen (vl->type) <= 0))
 424     return (-EINVAL);
 425
 426   if (!check_receive_okay (vl))
 427   {
 428 #if COLLECT_DEBUG
 429     char name[6*DATA_MAX_NAME_LEN];
 430     FORMAT_VL (name, sizeof (name), vl);
 431     name[sizeof (name) - 1] = 0;
 432     DEBUG ("network plugin: network_dispatch_values: "
 433         "NOT dispatching %s.", name);
 434 #endif
 435     stats_values_not_dispatched++;
 436     return (0);
 437   }
 438
 439   assert (vl->meta == NULL);
 440
 441   vl->meta = meta_data_create ();
 442   if (vl->meta == NULL)
 443   {
 444     ERROR ("network plugin: meta_data_create failed.");
 445     return (-ENOMEM);
 446   }
 447
 448   status = meta_data_add_boolean (vl->meta, "network:received", 1);
 449   if (status != 0)
 450   {
 451     ERROR ("network plugin: meta_data_add_boolean failed.");
 452     meta_data_destroy (vl->meta);
 453     vl->meta = NULL;
 454     return (status);
 455   }
 456
 457   if (username != NULL)
 458   {
 459     status = meta_data_add_string (vl->meta, "network:username", username);
 460     if (status != 0)
 461     {
 462       ERROR ("network plugin: meta_data_add_string failed.");
 463       meta_data_destroy (vl->meta);
 464       vl->meta = NULL;
 465       return (status);
 466     }
 467   }
 468
 469   plugin_dispatch_values (vl);
 470   stats_values_dispatched++;
 471
 472   meta_data_destroy (vl->meta);
 473   vl->meta = NULL;
 474
 475   return (0);
 476 } /* }}} int network_dispatch_values */
 477
 478 static int network_dispatch_notification (notification_t *n) /* {{{ */
 479 {
 480   int status;
 481
 482   assert (n->meta == NULL);
 483
 484   status = plugin_notification_meta_add_boolean (n, "network:received", 1);
 485   if (status != 0)
 486   {
 487     ERROR ("network plugin: plugin_notification_meta_add_boolean failed.");
 488     plugin_notification_meta_free (n->meta);
 489     n->meta = NULL;
 490     return (status);
 491   }
 492
 493   status = plugin_dispatch_notification (n);
 494
 495   plugin_notification_meta_free (n->meta);
 496   n->meta = NULL;
 497
 498   return (status);
 499 } /* }}} int network_dispatch_notification */
 500
 501 #if HAVE_LIBGCRYPT
 502 static void network_init_gcrypt (void) /* {{{ */
 503 {
 504   /* http://lists.gnupg.org/pipermail/gcrypt-devel/2003-August/000458.html
 505    * Because you can't know in a library whether another library has
 506    * already initialized the library */
 507   if (gcry_control (GCRYCTL_ANY_INITIALIZATION_P))
 508     return;
 509
 510  /* http://www.gnupg.org/documentation/manuals/gcrypt/Multi_002dThreading.html
 511   * To ensure thread-safety, it's important to set GCRYCTL_SET_THREAD_CBS
 512   * *before* initalizing Libgcrypt with gcry_check_version(), which itself must
 513   * be called before any other gcry_* function. GCRYCTL_ANY_INITIALIZATION_P
 514   * above doesn't count, as it doesn't implicitly initalize Libgcrypt.
 515   *
 516   * tl;dr: keep all these gry_* statements in this exact order please. */
 517 # if GCRYPT_VERSION_NUMBER < 0x010600
 518   gcry_control (GCRYCTL_SET_THREAD_CBS, &gcry_threads_pthread);
 519 # endif
 520   gcry_check_version (NULL);
 521   gcry_control (GCRYCTL_INIT_SECMEM, 32768);
 522   gcry_control (GCRYCTL_INITIALIZATION_FINISHED);
 523 } /* }}} void network_init_gcrypt */
 524
 525 static gcry_cipher_hd_t network_get_aes256_cypher (sockent_t *se, /* {{{ */
 526     const void *iv, size_t iv_size, const char *username)
 527 {
 528   gcry_error_t err;
 529   gcry_cipher_hd_t *cyper_ptr;
 530   unsigned char password_hash[32];
 531
 532   if (se->type == SOCKENT_TYPE_CLIENT)
 533   {
 534           cyper_ptr = &se->data.client.cypher;
 535           memcpy (password_hash, se->data.client.password_hash,
 536                           sizeof (password_hash));
 537   }
 538   else
 539   {
 540           char *secret;
 541
 542           cyper_ptr = &se->data.server.cypher;
 543
 544           if (username == NULL)
 545                   return (NULL);
 546
 547           secret = fbh_get (se->data.server.userdb, username);
 548           if (secret == NULL)
 549                   return (NULL);
 550
 551           gcry_md_hash_buffer (GCRY_MD_SHA256,
 552                           password_hash,
 553                           secret, strlen (secret));
 554
 555           sfree (secret);
 556   }
 557
 558   if (*cyper_ptr == NULL)
 559   {
 560     err = gcry_cipher_open (cyper_ptr,
 561         GCRY_CIPHER_AES256, GCRY_CIPHER_MODE_OFB, /* flags = */ 0);
 562     if (err != 0)
 563     {
 564       ERROR ("network plugin: gcry_cipher_open returned: %s",
 565           gcry_strerror (err));
 566       *cyper_ptr = NULL;
 567       return (NULL);
 568     }
 569   }
 570   else
 571   {
 572     gcry_cipher_reset (*cyper_ptr);
 573   }
 574   assert (*cyper_ptr != NULL);
 575
 576   err = gcry_cipher_setkey (*cyper_ptr,
 577       password_hash, sizeof (password_hash));
 578   if (err != 0)
 579   {
 580     ERROR ("network plugin: gcry_cipher_setkey returned: %s",
 581         gcry_strerror (err));
 582     gcry_cipher_close (*cyper_ptr);
 583     *cyper_ptr = NULL;
 584     return (NULL);
 585   }
 586
 587   err = gcry_cipher_setiv (*cyper_ptr, iv, iv_size);
 588   if (err != 0)
 589   {
 590     ERROR ("network plugin: gcry_cipher_setkey returned: %s",
 591         gcry_strerror (err));
 592     gcry_cipher_close (*cyper_ptr);
 593     *cyper_ptr = NULL;
 594     return (NULL);
 595   }
 596
 597   return (*cyper_ptr);
 598 } /* }}} int network_get_aes256_cypher */
 599 #endif /* HAVE_LIBGCRYPT */
 600
 601 static int write_part_values (char **ret_buffer, int *ret_buffer_len,
 602                 const data_set_t *ds, const value_list_t *vl)
 603 {
 604         char *packet_ptr;
 605         int packet_len;
 606         int num_values;
 607
 608         part_header_t pkg_ph;
 609         uint16_t      pkg_num_values;
 610         uint8_t      *pkg_values_types;
 611         value_t      *pkg_values;
 612
 613         int offset;
 614         int i;
 615
 616         num_values = vl->values_len;
 617         packet_len = sizeof (part_header_t) + sizeof (uint16_t)
 618                 + (num_values * sizeof (uint8_t))
 619                 + (num_values * sizeof (value_t));
 620
 621         if (*ret_buffer_len < packet_len)
 622                 return (-1);
 623
 624         pkg_values_types = (uint8_t *) malloc (num_values * sizeof (uint8_t));
 625         if (pkg_values_types == NULL)
 626         {
 627                 ERROR ("network plugin: write_part_values: malloc failed.");
 628                 return (-1);
 629         }
 630
 631         pkg_values = (value_t *) malloc (num_values * sizeof (value_t));
 632         if (pkg_values == NULL)
 633         {
 634                 free (pkg_values_types);
 635                 ERROR ("network plugin: write_part_values: malloc failed.");
 636                 return (-1);
 637         }
 638
 639         pkg_ph.type = htons (TYPE_VALUES);
 640         pkg_ph.length = htons (packet_len);
 641
 642         pkg_num_values = htons ((uint16_t) vl->values_len);
 643
 644         for (i = 0; i < num_values; i++)
 645         {
 646                 pkg_values_types[i] = (uint8_t) ds->ds[i].type;
 647                 switch (ds->ds[i].type)
 648                 {
 649                         case DS_TYPE_COUNTER:
 650                                 pkg_values[i].counter = htonll (vl->values[i].counter);
 651                                 break;
 652
 653                         case DS_TYPE_GAUGE:
 654                                 pkg_values[i].gauge = htond (vl->values[i].gauge);
 655                                 break;
 656
 657                         case DS_TYPE_DERIVE:
 658                                 pkg_values[i].derive = htonll (vl->values[i].derive);
 659                                 break;
 660
 661                         case DS_TYPE_ABSOLUTE:
 662                                 pkg_values[i].absolute = htonll (vl->values[i].absolute);
 663                                 break;
 664
 665                         default:
 666                                 free (pkg_values_types);
 667                                 free (pkg_values);
 668                                 ERROR ("network plugin: write_part_values: "
 669                                                 "Unknown data source type: %i",
 670                                                 ds->ds[i].type);
 671                                 return (-1);
 672                 } /* switch (ds->ds[i].type) */
 673         } /* for (num_values) */
 674
 675         /*
 676          * Use `memcpy' to write everything to the buffer, because the pointer
 677          * may be unaligned and some architectures, such as SPARC, can't handle
 678          * that.
 679          */
 680         packet_ptr = *ret_buffer;
 681         offset = 0;
 682         memcpy (packet_ptr + offset, &pkg_ph, sizeof (pkg_ph));
 683         offset += sizeof (pkg_ph);
 684         memcpy (packet_ptr + offset, &pkg_num_values, sizeof (pkg_num_values));
 685         offset += sizeof (pkg_num_values);
 686         memcpy (packet_ptr + offset, pkg_values_types, num_values * sizeof (uint8_t));
 687         offset += num_values * sizeof (uint8_t);
 688         memcpy (packet_ptr + offset, pkg_values, num_values * sizeof (value_t));
 689         offset += num_values * sizeof (value_t);
 690
 691         assert (offset == packet_len);
 692
 693         *ret_buffer = packet_ptr + packet_len;
 694         *ret_buffer_len -= packet_len;
 695
 696         free (pkg_values_types);
 697         free (pkg_values);
 698
 699         return (0);
 700 } /* int write_part_values */
 701
 702 static int write_part_number (char **ret_buffer, int *ret_buffer_len,
 703                 int type, uint64_t value)
 704 {
 705         char *packet_ptr;
 706         int packet_len;
 707
 708         part_header_t pkg_head;
 709         uint64_t pkg_value;
 710
 711         int offset;
 712
 713         packet_len = sizeof (pkg_head) + sizeof (pkg_value);
 714
 715         if (*ret_buffer_len < packet_len)
 716                 return (-1);
 717
 718         pkg_head.type = htons (type);
 719         pkg_head.length = htons (packet_len);
 720         pkg_value = htonll (value);
 721
 722         packet_ptr = *ret_buffer;
 723         offset = 0;
 724         memcpy (packet_ptr + offset, &pkg_head, sizeof (pkg_head));
 725         offset += sizeof (pkg_head);
 726         memcpy (packet_ptr + offset, &pkg_value, sizeof (pkg_value));
 727         offset += sizeof (pkg_value);
 728
 729         assert (offset == packet_len);
 730
 731         *ret_buffer = packet_ptr + packet_len;
 732         *ret_buffer_len -= packet_len;
 733
 734         return (0);
 735 } /* int write_part_number */
 736
 737 static int write_part_string (char **ret_buffer, int *ret_buffer_len,
 738                 int type, const char *str, int str_len)
 739 {
 740         char *buffer;
 741         int buffer_len;
 742
 743         uint16_t pkg_type;
 744         uint16_t pkg_length;
 745
 746         int offset;
 747
 748         buffer_len = 2 * sizeof (uint16_t) + str_len + 1;
 749         if (*ret_buffer_len < buffer_len)
 750                 return (-1);
 751
 752         pkg_type = htons (type);
 753         pkg_length = htons (buffer_len);
 754
 755         buffer = *ret_buffer;
 756         offset = 0;
 757         memcpy (buffer + offset, (void *) &pkg_type, sizeof (pkg_type));
 758         offset += sizeof (pkg_type);
 759         memcpy (buffer + offset, (void *) &pkg_length, sizeof (pkg_length));
 760         offset += sizeof (pkg_length);
 761         memcpy (buffer + offset, str, str_len);
 762         offset += str_len;
 763         memset (buffer + offset, '\0', 1);
 764         offset += 1;
 765
 766         assert (offset == buffer_len);
 767
 768         *ret_buffer = buffer + buffer_len;
 769         *ret_buffer_len -= buffer_len;
 770
 771         return (0);
 772 } /* int write_part_string */
 773
 774 static int parse_part_values (void **ret_buffer, size_t *ret_buffer_len,
 775                 value_t **ret_values, int *ret_num_values)
 776 {
 777         char *buffer = *ret_buffer;
 778         size_t buffer_len = *ret_buffer_len;
 779
 780         uint16_t tmp16;
 781         size_t exp_size;
 782         int   i;
 783
 784         uint16_t pkg_length;
 785         uint16_t pkg_type;
 786         uint16_t pkg_numval;
 787
 788         uint8_t *pkg_types;
 789         value_t *pkg_values;
 790
 791         if (buffer_len < 15)
 792         {
 793                 NOTICE ("network plugin: packet is too short: "
 794                                 "buffer_len = %zu", buffer_len);
 795                 return (-1);
 796         }
 797
 798         memcpy ((void *) &tmp16, buffer, sizeof (tmp16));
 799         buffer += sizeof (tmp16);
 800         pkg_type = ntohs (tmp16);
 801
 802         memcpy ((void *) &tmp16, buffer, sizeof (tmp16));
 803         buffer += sizeof (tmp16);
 804         pkg_length = ntohs (tmp16);
 805
 806         memcpy ((void *) &tmp16, buffer, sizeof (tmp16));
 807         buffer += sizeof (tmp16);
 808         pkg_numval = ntohs (tmp16);
 809
 810         assert (pkg_type == TYPE_VALUES);
 811
 812         exp_size = 3 * sizeof (uint16_t)
 813                 + pkg_numval * (sizeof (uint8_t) + sizeof (value_t));
 814         if (buffer_len < exp_size)
 815         {
 816                 WARNING ("network plugin: parse_part_values: "
 817                                 "Packet too short: "
 818                                 "Chunk of size %zu expected, "
 819                                 "but buffer has only %zu bytes left.",
 820                                 exp_size, buffer_len);
 821                 return (-1);
 822         }
 823
 824         if (pkg_length != exp_size)
 825         {
 826                 WARNING ("network plugin: parse_part_values: "
 827                                 "Length and number of values "
 828                                 "in the packet don't match.");
 829                 return (-1);
 830         }
 831
 832         pkg_types = (uint8_t *) malloc (pkg_numval * sizeof (uint8_t));
 833         pkg_values = (value_t *) malloc (pkg_numval * sizeof (value_t));
 834         if ((pkg_types == NULL) || (pkg_values == NULL))
 835         {
 836                 sfree (pkg_types);
 837                 sfree (pkg_values);
 838                 ERROR ("network plugin: parse_part_values: malloc failed.");
 839                 return (-1);
 840         }
 841
 842         memcpy ((void *) pkg_types, (void *) buffer, pkg_numval * sizeof (uint8_t));
 843         buffer += pkg_numval * sizeof (uint8_t);
 844         memcpy ((void *) pkg_values, (void *) buffer, pkg_numval * sizeof (value_t));
 845         buffer += pkg_numval * sizeof (value_t);
 846
 847         for (i = 0; i < pkg_numval; i++)
 848         {
 849                 switch (pkg_types[i])
 850                 {
 851                   case DS_TYPE_COUNTER:
 852                     pkg_values[i].counter = (counter_t) ntohll (pkg_values[i].counter);
 853                     break;
 854
 855                   case DS_TYPE_GAUGE:
 856                     pkg_values[i].gauge = (gauge_t) ntohd (pkg_values[i].gauge);
 857                     break;
 858
 859                   case DS_TYPE_DERIVE:
 860                     pkg_values[i].derive = (derive_t) ntohll (pkg_values[i].derive);
 861                     break;
 862
 863                   case DS_TYPE_ABSOLUTE:
 864                     pkg_values[i].absolute = (absolute_t) ntohll (pkg_values[i].absolute);
 865                     break;
 866
 867                   default:
 868                     NOTICE ("network plugin: parse_part_values: "
 869                         "Don't know how to handle data source type %"PRIu8,
 870                         pkg_types[i]);
 871                     sfree (pkg_types);
 872                     sfree (pkg_values);
 873                     return (-1);
 874                 } /* switch (pkg_types[i]) */
 875         }
 876
 877         *ret_buffer     = buffer;
 878         *ret_buffer_len = buffer_len - pkg_length;
 879         *ret_num_values = pkg_numval;
 880         *ret_values     = pkg_values;
 881
 882         sfree (pkg_types);
 883
 884         return (0);
 885 } /* int parse_part_values */
 886
 887 static int parse_part_number (void **ret_buffer, size_t *ret_buffer_len,
 888                 uint64_t *value)
 889 {
 890         char *buffer = *ret_buffer;
 891         size_t buffer_len = *ret_buffer_len;
 892
 893         uint16_t tmp16;
 894         uint64_t tmp64;
 895         size_t exp_size = 2 * sizeof (uint16_t) + sizeof (uint64_t);
 896
 897         uint16_t pkg_length;
 898
 899         if (buffer_len < exp_size)
 900         {
 901                 WARNING ("network plugin: parse_part_number: "
 902                                 "Packet too short: "
 903                                 "Chunk of size %zu expected, "
 904                                 "but buffer has only %zu bytes left.",
 905                                 exp_size, buffer_len);
 906                 return (-1);
 907         }
 908
 909         memcpy ((void *) &tmp16, buffer, sizeof (tmp16));
 910         buffer += sizeof (tmp16);
 911         /* pkg_type = ntohs (tmp16); */
 912
 913         memcpy ((void *) &tmp16, buffer, sizeof (tmp16));
 914         buffer += sizeof (tmp16);
 915         pkg_length = ntohs (tmp16);
 916
 917         memcpy ((void *) &tmp64, buffer, sizeof (tmp64));
 918         buffer += sizeof (tmp64);
 919         *value = ntohll (tmp64);
 920
 921         *ret_buffer = buffer;
 922         *ret_buffer_len = buffer_len - pkg_length;
 923
 924         return (0);
 925 } /* int parse_part_number */
 926
 927 static int parse_part_string (void **ret_buffer, size_t *ret_buffer_len,
 928                 char *output, size_t const output_len)
 929 {
 930         char *buffer = *ret_buffer;
 931         size_t buffer_len = *ret_buffer_len;
 932
 933         uint16_t tmp16;
 934         size_t const header_size = 2 * sizeof (uint16_t);
 935
 936         uint16_t pkg_length;
 937         size_t payload_size;
 938
 939         if (output_len <= 0)
 940                 return (EINVAL);
 941
 942         if (buffer_len < header_size)
 943         {
 944                 WARNING ("network plugin: parse_part_string: "
 945                                 "Packet too short: "
 946                                 "Chunk of at least size %zu expected, "
 947                                 "but buffer has only %zu bytes left.",
 948                                 header_size, buffer_len);
 949                 return (-1);
 950         }
 951
 952         memcpy ((void *) &tmp16, buffer, sizeof (tmp16));
 953         buffer += sizeof (tmp16);
 954         /* pkg_type = ntohs (tmp16); */
 955
 956         memcpy ((void *) &tmp16, buffer, sizeof (tmp16));
 957         buffer += sizeof (tmp16);
 958         pkg_length = ntohs (tmp16);
 959         payload_size = ((size_t) pkg_length) - header_size;
 960
 961         /* Check that packet fits in the input buffer */
 962         if (pkg_length > buffer_len)
 963         {
 964                 WARNING ("network plugin: parse_part_string: "
 965                                 "Packet too big: "
 966                                 "Chunk of size %"PRIu16" received, "
 967                                 "but buffer has only %zu bytes left.",
 968                                 pkg_length, buffer_len);
 969                 return (-1);
 970         }
 971
 972         /* Check that pkg_length is in the valid range */
 973         if (pkg_length <= header_size)
 974         {
 975                 WARNING ("network plugin: parse_part_string: "
 976                                 "Packet too short: "
 977                                 "Header claims this packet is only %hu "
 978                                 "bytes long.", pkg_length);
 979                 return (-1);
 980         }
 981
 982         /* Check that the package data fits into the output buffer.
 983          * The previous if-statement ensures that:
 984          * `pkg_length > header_size' */
 985         if (output_len < payload_size)
 986         {
 987                 WARNING ("network plugin: parse_part_string: "
 988                                 "Buffer too small: "
 989                                 "Output buffer holds %zu bytes, "
 990                                 "which is too small to hold the received "
 991                                 "%zu byte string.",
 992                                 output_len, payload_size);
 993                 return (-1);
 994         }
 995
 996         /* All sanity checks successfull, let's copy the data over */
 997         memcpy ((void *) output, (void *) buffer, payload_size);
 998         buffer += payload_size;
 999
1000         /* For some very weird reason '\0' doesn't do the trick on SPARC in
1001          * this statement. */
1002         if (output[payload_size - 1] != 0)
1003         {
1004                 WARNING ("network plugin: parse_part_string: "
1005                                 "Received string does not end "
1006                                 "with a NULL-byte.");
1007                 return (-1);
1008         }
1009
1010         *ret_buffer = buffer;
1011         *ret_buffer_len = buffer_len - pkg_length;
1012
1013         return (0);
1014 } /* int parse_part_string */
1015
1016 /* Forward declaration: parse_part_sign_sha256 and parse_part_encr_aes256 call
1017  * parse_packet and vice versa. */
1018 #define PP_SIGNED    0x01
1019 #define PP_ENCRYPTED 0x02
1020 static int parse_packet (sockent_t *se,
1021                 void *buffer, size_t buffer_size, int flags,
1022                 const char *username);
1023
1024 #define BUFFER_READ(p,s) do { \
1025   memcpy ((p), buffer + buffer_offset, (s)); \
1026   buffer_offset += (s); \
1027 } while (0)
1028
1029 #if HAVE_LIBGCRYPT
1030 static int parse_part_sign_sha256 (sockent_t *se, /* {{{ */
1031     void **ret_buffer, size_t *ret_buffer_len, int flags)
1032 {
1033   static c_complain_t complain_no_users = C_COMPLAIN_INIT_STATIC;
1034
1035   char *buffer;
1036   size_t buffer_len;
1037   size_t buffer_offset;
1038
1039   size_t username_len;
1040   char *secret;
1041
1042   part_signature_sha256_t pss;
1043   uint16_t pss_head_length;
1044   char hash[sizeof (pss.hash)];
1045
1046   gcry_md_hd_t hd;
1047   gcry_error_t err;
1048   unsigned char *hash_ptr;
1049
1050   buffer = *ret_buffer;
1051   buffer_len = *ret_buffer_len;
1052   buffer_offset = 0;
1053
1054   if (se->data.server.userdb == NULL)
1055   {
1056     c_complain (LOG_NOTICE, &complain_no_users,
1057         "network plugin: Received signed network packet but can't verify it "
1058         "because no user DB has been configured. Will accept it.");
1059     return (0);
1060   }
1061
1062   /* Check if the buffer has enough data for this structure. */
1063   if (buffer_len <= PART_SIGNATURE_SHA256_SIZE)
1064     return (-ENOMEM);
1065
1066   /* Read type and length header */
1067   BUFFER_READ (&pss.head.type, sizeof (pss.head.type));
1068   BUFFER_READ (&pss.head.length, sizeof (pss.head.length));
1069   pss_head_length = ntohs (pss.head.length);
1070
1071   /* Check if the `pss_head_length' is within bounds. */
1072   if ((pss_head_length <= PART_SIGNATURE_SHA256_SIZE)
1073       || (pss_head_length > buffer_len))
1074   {
1075     ERROR ("network plugin: HMAC-SHA-256 with invalid length received.");
1076     return (-1);
1077   }
1078
1079   /* Copy the hash. */
1080   BUFFER_READ (pss.hash, sizeof (pss.hash));
1081
1082   /* Calculate username length (without null byte) and allocate memory */
1083   username_len = pss_head_length - PART_SIGNATURE_SHA256_SIZE;
1084   pss.username = malloc (username_len + 1);
1085   if (pss.username == NULL)
1086     return (-ENOMEM);
1087
1088   /* Read the username */
1089   BUFFER_READ (pss.username, username_len);
1090   pss.username[username_len] = 0;
1091
1092   assert (buffer_offset == pss_head_length);
1093
1094   /* Query the password */
1095   secret = fbh_get (se->data.server.userdb, pss.username);
1096   if (secret == NULL)
1097   {
1098     ERROR ("network plugin: Unknown user: %s", pss.username);
1099     sfree (pss.username);
1100     return (-ENOENT);
1101   }
1102
1103   /* Create a hash device and check the HMAC */
1104   hd = NULL;
1105   err = gcry_md_open (&hd, GCRY_MD_SHA256, GCRY_MD_FLAG_HMAC);
1106   if (err != 0)
1107   {
1108     ERROR ("network plugin: Creating HMAC-SHA-256 object failed: %s",
1109         gcry_strerror (err));
1110     sfree (secret);
1111     sfree (pss.username);
1112     return (-1);
1113   }
1114
1115   err = gcry_md_setkey (hd, secret, strlen (secret));
1116   if (err != 0)
1117   {
1118     ERROR ("network plugin: gcry_md_setkey failed: %s", gcry_strerror (err));
1119     gcry_md_close (hd);
1120     sfree (secret);
1121     sfree (pss.username);
1122     return (-1);
1123   }
1124
1125   gcry_md_write (hd,
1126       buffer     + PART_SIGNATURE_SHA256_SIZE,
1127       buffer_len - PART_SIGNATURE_SHA256_SIZE);
1128   hash_ptr = gcry_md_read (hd, GCRY_MD_SHA256);
1129   if (hash_ptr == NULL)
1130   {
1131     ERROR ("network plugin: gcry_md_read failed.");
1132     gcry_md_close (hd);
1133     sfree (secret);
1134     sfree (pss.username);
1135     return (-1);
1136   }
1137   memcpy (hash, hash_ptr, sizeof (hash));
1138
1139   /* Clean up */
1140   gcry_md_close (hd);
1141   hd = NULL;
1142
1143   if (memcmp (pss.hash, hash, sizeof (pss.hash)) != 0)
1144   {
1145     WARNING ("network plugin: Verifying HMAC-SHA-256 signature failed: "
1146         "Hash mismatch.");
1147   }
1148   else
1149   {
1150     parse_packet (se, buffer + buffer_offset, buffer_len - buffer_offset,
1151         flags | PP_SIGNED, pss.username);
1152   }
1153
1154   sfree (secret);
1155   sfree (pss.username);
1156
1157   *ret_buffer = buffer + buffer_len;
1158   *ret_buffer_len = 0;
1159
1160   return (0);
1161 } /* }}} int parse_part_sign_sha256 */
1162 /* #endif HAVE_LIBGCRYPT */
1163
1164 #else /* if !HAVE_LIBGCRYPT */
1165 static int parse_part_sign_sha256 (sockent_t *se, /* {{{ */
1166     void **ret_buffer, size_t *ret_buffer_size, int flags)
1167 {
1168   static int warning_has_been_printed = 0;
1169
1170   char *buffer;
1171   size_t buffer_size;
1172   size_t buffer_offset;
1173   uint16_t part_len;
1174
1175   part_signature_sha256_t pss;
1176
1177   buffer = *ret_buffer;
1178   buffer_size = *ret_buffer_size;
1179   buffer_offset = 0;
1180
1181   if (buffer_size <= PART_SIGNATURE_SHA256_SIZE)
1182     return (-ENOMEM);
1183
1184   BUFFER_READ (&pss.head.type, sizeof (pss.head.type));
1185   BUFFER_READ (&pss.head.length, sizeof (pss.head.length));
1186   part_len = ntohs (pss.head.length);
1187
1188   if ((part_len <= PART_SIGNATURE_SHA256_SIZE)
1189       || (part_len > buffer_size))
1190     return (-EINVAL);
1191
1192   if (warning_has_been_printed == 0)
1193   {
1194     WARNING ("network plugin: Received signed packet, but the network "
1195         "plugin was not linked with libgcrypt, so I cannot "
1196         "verify the signature. The packet will be accepted.");
1197     warning_has_been_printed = 1;
1198   }
1199
1200   parse_packet (se, buffer + part_len, buffer_size - part_len, flags,
1201       /* username = */ NULL);
1202
1203   *ret_buffer = buffer + buffer_size;
1204   *ret_buffer_size = 0;
1205
1206   return (0);
1207 } /* }}} int parse_part_sign_sha256 */
1208 #endif /* !HAVE_LIBGCRYPT */
1209
1210 #if HAVE_LIBGCRYPT
1211 static int parse_part_encr_aes256 (sockent_t *se, /* {{{ */
1212                 void **ret_buffer, size_t *ret_buffer_len,
1213                 int flags)
1214 {
1215   char  *buffer = *ret_buffer;
1216   size_t buffer_len = *ret_buffer_len;
1217   size_t payload_len;
1218   size_t part_size;
1219   size_t buffer_offset;
1220   uint16_t username_len;
1221   part_encryption_aes256_t pea;
1222   unsigned char hash[sizeof (pea.hash)];
1223
1224   gcry_cipher_hd_t cypher;
1225   gcry_error_t err;
1226
1227   /* Make sure at least the header if available. */
1228   if (buffer_len <= PART_ENCRYPTION_AES256_SIZE)
1229   {
1230     NOTICE ("network plugin: parse_part_encr_aes256: "
1231         "Discarding short packet.");
1232     return (-1);
1233   }
1234
1235   buffer_offset = 0;
1236
1237   /* Copy the unencrypted information into `pea'. */
1238   BUFFER_READ (&pea.head.type, sizeof (pea.head.type));
1239   BUFFER_READ (&pea.head.length, sizeof (pea.head.length));
1240
1241   /* Check the `part size'. */
1242   part_size = ntohs (pea.head.length);
1243   if ((part_size <= PART_ENCRYPTION_AES256_SIZE)
1244       || (part_size > buffer_len))
1245   {
1246     NOTICE ("network plugin: parse_part_encr_aes256: "
1247         "Discarding part with invalid size.");
1248     return (-1);
1249   }
1250
1251   /* Read the username */
1252   BUFFER_READ (&username_len, sizeof (username_len));
1253   username_len = ntohs (username_len);
1254
1255   if ((username_len <= 0)
1256       || (username_len > (part_size - (PART_ENCRYPTION_AES256_SIZE + 1))))
1257   {
1258     NOTICE ("network plugin: parse_part_encr_aes256: "
1259         "Discarding part with invalid username length.");
1260     return (-1);
1261   }
1262
1263   assert (username_len > 0);
1264   pea.username = malloc (username_len + 1);
1265   if (pea.username == NULL)
1266     return (-ENOMEM);
1267   BUFFER_READ (pea.username, username_len);
1268   pea.username[username_len] = 0;
1269
1270   /* Last but not least, the initialization vector */
1271   BUFFER_READ (pea.iv, sizeof (pea.iv));
1272
1273   /* Make sure we are at the right position */
1274   assert (buffer_offset == (username_len +
1275         PART_ENCRYPTION_AES256_SIZE - sizeof (pea.hash)));
1276
1277   cypher = network_get_aes256_cypher (se, pea.iv, sizeof (pea.iv),
1278       pea.username);
1279   if (cypher == NULL)
1280   {
1281     sfree (pea.username);
1282     return (-1);
1283   }
1284
1285   payload_len = part_size - (PART_ENCRYPTION_AES256_SIZE + username_len);
1286   assert (payload_len > 0);
1287
1288   /* Decrypt the packet in-place */
1289   err = gcry_cipher_decrypt (cypher,
1290       buffer    + buffer_offset,
1291       part_size - buffer_offset,
1292       /* in = */ NULL, /* in len = */ 0);
1293   if (err != 0)
1294   {
1295     sfree (pea.username);
1296     ERROR ("network plugin: gcry_cipher_decrypt returned: %s",
1297         gcry_strerror (err));
1298     return (-1);
1299   }
1300
1301   /* Read the hash */
1302   BUFFER_READ (pea.hash, sizeof (pea.hash));
1303
1304   /* Make sure we're at the right position - again */
1305   assert (buffer_offset == (username_len + PART_ENCRYPTION_AES256_SIZE));
1306   assert (buffer_offset == (part_size - payload_len));
1307
1308   /* Check hash sum */
1309   memset (hash, 0, sizeof (hash));
1310   gcry_md_hash_buffer (GCRY_MD_SHA1, hash,
1311       buffer + buffer_offset, payload_len);
1312   if (memcmp (hash, pea.hash, sizeof (hash)) != 0)
1313   {
1314     sfree (pea.username);
1315     ERROR ("network plugin: Decryption failed: Checksum mismatch.");
1316     return (-1);
1317   }
1318
1319   parse_packet (se, buffer + buffer_offset, payload_len,
1320       flags | PP_ENCRYPTED, pea.username);
1321
1322   /* XXX: Free pea.username?!? */
1323
1324   /* Update return values */
1325   *ret_buffer =     buffer     + part_size;
1326   *ret_buffer_len = buffer_len - part_size;
1327
1328   sfree (pea.username);
1329
1330   return (0);
1331 } /* }}} int parse_part_encr_aes256 */
1332 /* #endif HAVE_LIBGCRYPT */
1333
1334 #else /* if !HAVE_LIBGCRYPT */
1335 static int parse_part_encr_aes256 (sockent_t *se, /* {{{ */
1336     void **ret_buffer, size_t *ret_buffer_size, int flags)
1337 {
1338   static int warning_has_been_printed = 0;
1339
1340   char *buffer;
1341   size_t buffer_size;
1342   size_t buffer_offset;
1343
1344   part_header_t ph;
1345   size_t ph_length;
1346
1347   buffer = *ret_buffer;
1348   buffer_size = *ret_buffer_size;
1349   buffer_offset = 0;
1350
1351   /* parse_packet assures this minimum size. */
1352   assert (buffer_size >= (sizeof (ph.type) + sizeof (ph.length)));
1353
1354   BUFFER_READ (&ph.type, sizeof (ph.type));
1355   BUFFER_READ (&ph.length, sizeof (ph.length));
1356   ph_length = ntohs (ph.length);
1357
1358   if ((ph_length <= PART_ENCRYPTION_AES256_SIZE)
1359       || (ph_length > buffer_size))
1360   {
1361     ERROR ("network plugin: AES-256 encrypted part "
1362         "with invalid length received.");
1363     return (-1);
1364   }
1365
1366   if (warning_has_been_printed == 0)
1367   {
1368     WARNING ("network plugin: Received encrypted packet, but the network "
1369         "plugin was not linked with libgcrypt, so I cannot "
1370         "decrypt it. The part will be discarded.");
1371     warning_has_been_printed = 1;
1372   }
1373
1374   *ret_buffer += ph_length;
1375   *ret_buffer_size -= ph_length;
1376
1377   return (0);
1378 } /* }}} int parse_part_encr_aes256 */
1379 #endif /* !HAVE_LIBGCRYPT */
1380
1381 #undef BUFFER_READ
1382
1383 static int parse_packet (sockent_t *se, /* {{{ */
1384                 void *buffer, size_t buffer_size, int flags,
1385                 const char *username)
1386 {
1387         int status;
1388
1389         value_list_t vl = VALUE_LIST_INIT;
1390         notification_t n;
1391
1392 #if HAVE_LIBGCRYPT
1393         int packet_was_signed = (flags & PP_SIGNED);
1394         int packet_was_encrypted = (flags & PP_ENCRYPTED);
1395         int printed_ignore_warning = 0;
1396 #endif /* HAVE_LIBGCRYPT */
1397
1398
1399         memset (&vl, '\0', sizeof (vl));
1400         memset (&n, '\0', sizeof (n));
1401         status = 0;
1402
1403         while ((status == 0) && (0 < buffer_size)
1404                         && ((unsigned int) buffer_size > sizeof (part_header_t)))
1405         {
1406                 uint16_t pkg_length;
1407                 uint16_t pkg_type;
1408
1409                 memcpy ((void *) &pkg_type,
1410                                 (void *) buffer,
1411                                 sizeof (pkg_type));
1412                 memcpy ((void *) &pkg_length,
1413                                 (void *) (buffer + sizeof (pkg_type)),
1414                                 sizeof (pkg_length));
1415
1416                 pkg_length = ntohs (pkg_length);
1417                 pkg_type = ntohs (pkg_type);
1418
1419                 if (pkg_length > buffer_size)
1420                         break;
1421                 /* Ensure that this loop terminates eventually */
1422                 if (pkg_length < (2 * sizeof (uint16_t)))
1423                         break;
1424
1425                 if (pkg_type == TYPE_ENCR_AES256)
1426                 {
1427                         status = parse_part_encr_aes256 (se,
1428                                         &buffer, &buffer_size, flags);
1429                         if (status != 0)
1430                         {
1431                                 ERROR ("network plugin: Decrypting AES256 "
1432                                                 "part failed "
1433                                                 "with status %i.", status);
1434                                 break;
1435                         }
1436                 }
1437 #if HAVE_LIBGCRYPT
1438                 else if ((se->data.server.security_level == SECURITY_LEVEL_ENCRYPT)
1439                                 && (packet_was_encrypted == 0))
1440                 {
1441                         if (printed_ignore_warning == 0)
1442                         {
1443                                 INFO ("network plugin: Unencrypted packet or "
1444                                                 "part has been ignored.");
1445                                 printed_ignore_warning = 1;
1446                         }
1447                         buffer = ((char *) buffer) + pkg_length;
1448                         continue;
1449                 }
1450 #endif /* HAVE_LIBGCRYPT */
1451                 else if (pkg_type == TYPE_SIGN_SHA256)
1452                 {
1453                         status = parse_part_sign_sha256 (se,
1454                                         &buffer, &buffer_size, flags);
1455                         if (status != 0)
1456                         {
1457                                 ERROR ("network plugin: Verifying HMAC-SHA-256 "
1458                                                 "signature failed "
1459                                                 "with status %i.", status);
1460                                 break;
1461                         }
1462                 }
1463 #if HAVE_LIBGCRYPT
1464                 else if ((se->data.server.security_level == SECURITY_LEVEL_SIGN)
1465                                 && (packet_was_encrypted == 0)
1466                                 && (packet_was_signed == 0))
1467                 {
1468                         if (printed_ignore_warning == 0)
1469                         {
1470                                 INFO ("network plugin: Unsigned packet or "
1471                                                 "part has been ignored.");
1472                                 printed_ignore_warning = 1;
1473                         }
1474                         buffer = ((char *) buffer) + pkg_length;
1475                         continue;
1476                 }
1477 #endif /* HAVE_LIBGCRYPT */
1478                 else if (pkg_type == TYPE_VALUES)
1479                 {
1480                         status = parse_part_values (&buffer, &buffer_size,
1481                                         &vl.values, &vl.values_len);
1482                         if (status != 0)
1483                                 break;
1484
1485                         network_dispatch_values (&vl, username);
1486
1487                         sfree (vl.values);
1488                 }
1489                 else if (pkg_type == TYPE_TIME)
1490                 {
1491                         uint64_t tmp = 0;
1492                         status = parse_part_number (&buffer, &buffer_size,
1493                                         &tmp);
1494                         if (status == 0)
1495                         {
1496                                 vl.time = TIME_T_TO_CDTIME_T (tmp);
1497                                 n.time  = TIME_T_TO_CDTIME_T (tmp);
1498                         }
1499                 }
1500                 else if (pkg_type == TYPE_TIME_HR)
1501                 {
1502                         uint64_t tmp = 0;
1503                         status = parse_part_number (&buffer, &buffer_size,
1504                                         &tmp);
1505                         if (status == 0)
1506                         {
1507                                 vl.time = (cdtime_t) tmp;
1508                                 n.time  = (cdtime_t) tmp;
1509                         }
1510                 }
1511                 else if (pkg_type == TYPE_INTERVAL)
1512                 {
1513                         uint64_t tmp = 0;
1514                         status = parse_part_number (&buffer, &buffer_size,
1515                                         &tmp);
1516                         if (status == 0)
1517                                 vl.interval = TIME_T_TO_CDTIME_T (tmp);
1518                 }
1519                 else if (pkg_type == TYPE_INTERVAL_HR)
1520                 {
1521                         uint64_t tmp = 0;
1522                         status = parse_part_number (&buffer, &buffer_size,
1523                                         &tmp);
1524                         if (status == 0)
1525                                 vl.interval = (cdtime_t) tmp;
1526                 }
1527                 else if (pkg_type == TYPE_HOST)
1528                 {
1529                         status = parse_part_string (&buffer, &buffer_size,
1530                                         vl.host, sizeof (vl.host));
1531                         if (status == 0)
1532                                 sstrncpy (n.host, vl.host, sizeof (n.host));
1533                 }
1534                 else if (pkg_type == TYPE_PLUGIN)
1535                 {
1536                         status = parse_part_string (&buffer, &buffer_size,
1537                                         vl.plugin, sizeof (vl.plugin));
1538                         if (status == 0)
1539                                 sstrncpy (n.plugin, vl.plugin,
1540                                                 sizeof (n.plugin));
1541                 }
1542                 else if (pkg_type == TYPE_PLUGIN_INSTANCE)
1543                 {
1544                         status = parse_part_string (&buffer, &buffer_size,
1545                                         vl.plugin_instance,
1546                                         sizeof (vl.plugin_instance));
1547                         if (status == 0)
1548                                 sstrncpy (n.plugin_instance,
1549                                                 vl.plugin_instance,
1550                                                 sizeof (n.plugin_instance));
1551                 }
1552                 else if (pkg_type == TYPE_TYPE)
1553                 {
1554                         status = parse_part_string (&buffer, &buffer_size,
1555                                         vl.type, sizeof (vl.type));
1556                         if (status == 0)
1557                                 sstrncpy (n.type, vl.type, sizeof (n.type));
1558                 }
1559                 else if (pkg_type == TYPE_TYPE_INSTANCE)
1560                 {
1561                         status = parse_part_string (&buffer, &buffer_size,
1562                                         vl.type_instance,
1563                                         sizeof (vl.type_instance));
1564                         if (status == 0)
1565                                 sstrncpy (n.type_instance, vl.type_instance,
1566                                                 sizeof (n.type_instance));
1567                 }
1568                 else if (pkg_type == TYPE_MESSAGE)
1569                 {
1570                         status = parse_part_string (&buffer, &buffer_size,
1571                                         n.message, sizeof (n.message));
1572
1573                         if (status != 0)
1574                         {
1575                                 /* do nothing */
1576                         }
1577                         else if ((n.severity != NOTIF_FAILURE)
1578                                         && (n.severity != NOTIF_WARNING)
1579                                         && (n.severity != NOTIF_OKAY))
1580                         {
1581                                 INFO ("network plugin: "
1582                                                 "Ignoring notification with "
1583                                                 "unknown severity %i.",
1584                                                 n.severity);
1585                         }
1586                         else if (n.time <= 0)
1587                         {
1588                                 INFO ("network plugin: "
1589                                                 "Ignoring notification with "
1590                                                 "time == 0.");
1591                         }
1592                         else if (strlen (n.message) <= 0)
1593                         {
1594                                 INFO ("network plugin: "
1595                                                 "Ignoring notification with "
1596                                                 "an empty message.");
1597                         }
1598                         else
1599                         {
1600                                 network_dispatch_notification (&n);
1601                         }
1602                 }
1603                 else if (pkg_type == TYPE_SEVERITY)
1604                 {
1605                         uint64_t tmp = 0;
1606                         status = parse_part_number (&buffer, &buffer_size,
1607                                         &tmp);
1608                         if (status == 0)
1609                                 n.severity = (int) tmp;
1610                 }
1611                 else
1612                 {
1613                         DEBUG ("network plugin: parse_packet: Unknown part"
1614                                         " type: 0x%04hx", pkg_type);
1615                         buffer = ((char *) buffer) + pkg_length;
1616                 }
1617         } /* while (buffer_size > sizeof (part_header_t)) */
1618
1619         if (status == 0 && buffer_size > 0)
1620                 WARNING ("network plugin: parse_packet: Received truncated "
1621                                 "packet, try increasing `MaxPacketSize'");
1622
1623         return (status);
1624 } /* }}} int parse_packet */
1625
1626 static void free_sockent_client (struct sockent_client *sec) /* {{{ */
1627 {
1628   if (sec->fd >= 0)
1629   {
1630     close (sec->fd);
1631     sec->fd = -1;
1632   }
1633   sfree (sec->addr);
1634 #if HAVE_LIBGCRYPT
1635   sfree (sec->username);
1636   sfree (sec->password);
1637   if (sec->cypher != NULL)
1638     gcry_cipher_close (sec->cypher);
1639 #endif
1640 } /* }}} void free_sockent_client */
1641
1642 static void free_sockent_server (struct sockent_server *ses) /* {{{ */
1643 {
1644   size_t i;
1645
1646   for (i = 0; i < ses->fd_num; i++)
1647   {
1648     if (ses->fd[i] >= 0)
1649     {
1650       close (ses->fd[i]);
1651       ses->fd[i] = -1;
1652     }
1653   }
1654
1655   sfree (ses->fd);
1656 #if HAVE_LIBGCRYPT
1657   sfree (ses->auth_file);
1658   fbh_destroy (ses->userdb);
1659   if (ses->cypher != NULL)
1660     gcry_cipher_close (ses->cypher);
1661 #endif
1662 } /* }}} void free_sockent_server */
1663
1664 static void sockent_destroy (sockent_t *se) /* {{{ */
1665 {
1666   sockent_t *next;
1667
1668   DEBUG ("network plugin: sockent_destroy (se = %p);", (void *) se);
1669
1670   while (se != NULL)
1671   {
1672     next = se->next;
1673
1674     sfree (se->node);
1675     sfree (se->service);
1676
1677     if (se->type == SOCKENT_TYPE_CLIENT)
1678       free_sockent_client (&se->data.client);
1679     else
1680       free_sockent_server (&se->data.server);
1681
1682     sfree (se);
1683     se = next;
1684   }
1685 } /* }}} void sockent_destroy */
1686
1687 /*
1688  * int network_set_ttl
1689  *
1690  * Set the `IP_MULTICAST_TTL', `IP_TTL', `IPV6_MULTICAST_HOPS' or
1691  * `IPV6_UNICAST_HOPS', depending on which option is applicable.
1692  *
1693  * The `struct addrinfo' is used to destinguish between unicast and multicast
1694  * sockets.
1695  */
1696 static int network_set_ttl (const sockent_t *se, const struct addrinfo *ai)
1697 {
1698         DEBUG ("network plugin: network_set_ttl: network_config_ttl = %i;",
1699                         network_config_ttl);
1700
1701         assert (se->type == SOCKENT_TYPE_CLIENT);
1702
1703         if ((network_config_ttl < 1) || (network_config_ttl > 255))
1704                 return (-1);
1705
1706         if (ai->ai_family == AF_INET)
1707         {
1708                 struct sockaddr_in *addr = (struct sockaddr_in *) ai->ai_addr;
1709                 int optname;
1710
1711                 if (IN_MULTICAST (ntohl (addr->sin_addr.s_addr)))
1712                         optname = IP_MULTICAST_TTL;
1713                 else
1714                         optname = IP_TTL;
1715
1716                 if (setsockopt (se->data.client.fd, IPPROTO_IP, optname,
1717                                         &network_config_ttl,
1718                                         sizeof (network_config_ttl)) != 0)
1719                 {
1720                         char errbuf[1024];
1721                         ERROR ("network plugin: setsockopt (ipv4-ttl): %s",
1722                                         sstrerror (errno, errbuf, sizeof (errbuf)));
1723                         return (-1);
1724                 }
1725         }
1726         else if (ai->ai_family == AF_INET6)
1727         {
1728                 /* Useful example: http://gsyc.escet.urjc.es/~eva/IPv6-web/examples/mcast.html */
1729                 struct sockaddr_in6 *addr = (struct sockaddr_in6 *) ai->ai_addr;
1730                 int optname;
1731
1732                 if (IN6_IS_ADDR_MULTICAST (&addr->sin6_addr))
1733                         optname = IPV6_MULTICAST_HOPS;
1734                 else
1735                         optname = IPV6_UNICAST_HOPS;
1736
1737                 if (setsockopt (se->data.client.fd, IPPROTO_IPV6, optname,
1738                                         &network_config_ttl,
1739                                         sizeof (network_config_ttl)) != 0)
1740                 {
1741                         char errbuf[1024];
1742                         ERROR ("network plugin: setsockopt(ipv6-ttl): %s",
1743                                         sstrerror (errno, errbuf,
1744                                                 sizeof (errbuf)));
1745                         return (-1);
1746                 }
1747         }
1748
1749         return (0);
1750 } /* int network_set_ttl */
1751
1752 static int network_set_interface (const sockent_t *se, const struct addrinfo *ai) /* {{{ */
1753 {
1754         DEBUG ("network plugin: network_set_interface: interface index = %i;",
1755                         se->interface);
1756
1757         assert (se->type == SOCKENT_TYPE_CLIENT);
1758
1759         if (ai->ai_family == AF_INET)
1760         {
1761                 struct sockaddr_in *addr = (struct sockaddr_in *) ai->ai_addr;
1762
1763                 if (IN_MULTICAST (ntohl (addr->sin_addr.s_addr)))
1764                 {
1765 #if HAVE_STRUCT_IP_MREQN_IMR_IFINDEX
1766                         /* If possible, use the "ip_mreqn" structure which has
1767                          * an "interface index" member. Using the interface
1768                          * index is preferred here, because of its similarity
1769                          * to the way IPv6 handles this. Unfortunately, it
1770                          * appears not to be portable. */
1771                         struct ip_mreqn mreq;
1772
1773                         memset (&mreq, 0, sizeof (mreq));
1774                         mreq.imr_multiaddr.s_addr = addr->sin_addr.s_addr;
1775                         mreq.imr_address.s_addr = ntohl (INADDR_ANY);
1776                         mreq.imr_ifindex = se->interface;
1777 #else
1778                         struct ip_mreq mreq;
1779
1780                         memset (&mreq, 0, sizeof (mreq));
1781                         mreq.imr_multiaddr.s_addr = addr->sin_addr.s_addr;
1782                         mreq.imr_interface.s_addr = ntohl (INADDR_ANY);
1783 #endif
1784
1785                         if (setsockopt (se->data.client.fd, IPPROTO_IP, IP_MULTICAST_IF,
1786                                                 &mreq, sizeof (mreq)) != 0)
1787                         {
1788                                 char errbuf[1024];
1789                                 ERROR ("network plugin: setsockopt (ipv4-multicast-if): %s",
1790                                                 sstrerror (errno, errbuf, sizeof (errbuf)));
1791                                 return (-1);
1792                         }
1793
1794                         return (0);
1795                 }
1796         }
1797         else if (ai->ai_family == AF_INET6)
1798         {
1799                 struct sockaddr_in6 *addr = (struct sockaddr_in6 *) ai->ai_addr;
1800
1801                 if (IN6_IS_ADDR_MULTICAST (&addr->sin6_addr))
1802                 {
1803                         if (setsockopt (se->data.client.fd, IPPROTO_IPV6, IPV6_MULTICAST_IF,
1804                                                 &se->interface,
1805                                                 sizeof (se->interface)) != 0)
1806                         {
1807                                 char errbuf[1024];
1808                                 ERROR ("network plugin: setsockopt (ipv6-multicast-if): %s",
1809                                                 sstrerror (errno, errbuf,
1810                                                         sizeof (errbuf)));
1811                                 return (-1);
1812                         }
1813
1814                         return (0);
1815                 }
1816         }
1817
1818         /* else: Not a multicast interface. */
1819         if (se->interface != 0)
1820         {
1821 #if defined(HAVE_IF_INDEXTONAME) && HAVE_IF_INDEXTONAME && defined(SO_BINDTODEVICE)
1822                 char interface_name[IFNAMSIZ];
1823
1824                 if (if_indextoname (se->interface, interface_name) == NULL)
1825                         return (-1);
1826
1827                 DEBUG ("network plugin: Binding socket to interface %s", interface_name);
1828
1829                 if (setsockopt (se->data.client.fd, SOL_SOCKET, SO_BINDTODEVICE,
1830                                         interface_name,
1831                                         sizeof(interface_name)) == -1 )
1832                 {
1833                         char errbuf[1024];
1834                         ERROR ("network plugin: setsockopt (bind-if): %s",
1835                                         sstrerror (errno, errbuf, sizeof (errbuf)));
1836                         return (-1);
1837                 }
1838 /* #endif HAVE_IF_INDEXTONAME && SO_BINDTODEVICE */
1839
1840 #else
1841                 WARNING ("network plugin: Cannot set the interface on a unicast "
1842                         "socket because "
1843 # if !defined(SO_BINDTODEVICE)
1844                         "the \"SO_BINDTODEVICE\" socket option "
1845 # else
1846                         "the \"if_indextoname\" function "
1847 # endif
1848                         "is not available on your system.");
1849 #endif
1850
1851         }
1852
1853         return (0);
1854 } /* }}} network_set_interface */
1855
1856 static int network_bind_socket (int fd, const struct addrinfo *ai, const int interface_idx)
1857 {
1858 #if KERNEL_SOLARIS
1859         char loop   = 0;
1860 #else
1861         int loop = 0;
1862 #endif
1863         int yes  = 1;
1864
1865         /* allow multiple sockets to use the same PORT number */
1866         if (setsockopt (fd, SOL_SOCKET, SO_REUSEADDR,
1867                                 &yes, sizeof(yes)) == -1) {
1868                 char errbuf[1024];
1869                 ERROR ("network plugin: setsockopt (reuseaddr): %s",
1870                                 sstrerror (errno, errbuf, sizeof (errbuf)));
1871                 return (-1);
1872         }
1873
1874         DEBUG ("fd = %i; calling `bind'", fd);
1875
1876         if (bind (fd, ai->ai_addr, ai->ai_addrlen) == -1)
1877         {
1878                 char errbuf[1024];
1879                 ERROR ("bind: %s",
1880                                 sstrerror (errno, errbuf, sizeof (errbuf)));
1881                 return (-1);
1882         }
1883
1884         if (ai->ai_family == AF_INET)
1885         {
1886                 struct sockaddr_in *addr = (struct sockaddr_in *) ai->ai_addr;
1887                 if (IN_MULTICAST (ntohl (addr->sin_addr.s_addr)))
1888                 {
1889 #if HAVE_STRUCT_IP_MREQN_IMR_IFINDEX
1890                         struct ip_mreqn mreq;
1891 #else
1892                         struct ip_mreq mreq;
1893 #endif
1894
1895                         DEBUG ("fd = %i; IPv4 multicast address found", fd);
1896
1897                         mreq.imr_multiaddr.s_addr = addr->sin_addr.s_addr;
1898 #if HAVE_STRUCT_IP_MREQN_IMR_IFINDEX
1899                         /* Set the interface using the interface index if
1900                          * possible (available). Unfortunately, the struct
1901                          * ip_mreqn is not portable. */
1902                         mreq.imr_address.s_addr = ntohl (INADDR_ANY);
1903                         mreq.imr_ifindex = interface_idx;
1904 #else
1905                         mreq.imr_interface.s_addr = ntohl (INADDR_ANY);
1906 #endif
1907
1908                         if (setsockopt (fd, IPPROTO_IP, IP_MULTICAST_LOOP,
1909                                                 &loop, sizeof (loop)) == -1)
1910                         {
1911                                 char errbuf[1024];
1912                                 ERROR ("network plugin: setsockopt (multicast-loop): %s",
1913                                                 sstrerror (errno, errbuf,
1914                                                         sizeof (errbuf)));
1915                                 return (-1);
1916                         }
1917
1918                         if (setsockopt (fd, IPPROTO_IP, IP_ADD_MEMBERSHIP,
1919                                                 &mreq, sizeof (mreq)) == -1)
1920                         {
1921                                 char errbuf[1024];
1922                                 ERROR ("network plugin: setsockopt (add-membership): %s",
1923                                                 sstrerror (errno, errbuf,
1924                                                         sizeof (errbuf)));
1925                                 return (-1);
1926                         }
1927
1928                         return (0);
1929                 }
1930         }
1931         else if (ai->ai_family == AF_INET6)
1932         {
1933                 /* Useful example: http://gsyc.escet.urjc.es/~eva/IPv6-web/examples/mcast.html */
1934                 struct sockaddr_in6 *addr = (struct sockaddr_in6 *) ai->ai_addr;
1935                 if (IN6_IS_ADDR_MULTICAST (&addr->sin6_addr))
1936                 {
1937                         struct ipv6_mreq mreq;
1938
1939                         DEBUG ("fd = %i; IPv6 multicast address found", fd);
1940
1941                         memcpy (&mreq.ipv6mr_multiaddr,
1942                                         &addr->sin6_addr,
1943                                         sizeof (addr->sin6_addr));
1944
1945                         /* http://developer.apple.com/documentation/Darwin/Reference/ManPages/man4/ip6.4.html
1946                          * ipv6mr_interface may be set to zeroes to
1947                          * choose the default multicast interface or to
1948                          * the index of a particular multicast-capable
1949                          * interface if the host is multihomed.
1950                          * Membership is associ-associated with a
1951                          * single interface; programs running on
1952                          * multihomed hosts may need to join the same
1953                          * group on more than one interface.*/
1954                         mreq.ipv6mr_interface = interface_idx;
1955
1956                         if (setsockopt (fd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP,
1957                                                 &loop, sizeof (loop)) == -1)
1958                         {
1959                                 char errbuf[1024];
1960                                 ERROR ("network plugin: setsockopt (ipv6-multicast-loop): %s",
1961                                                 sstrerror (errno, errbuf,
1962                                                         sizeof (errbuf)));
1963                                 return (-1);
1964                         }
1965
1966                         if (setsockopt (fd, IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP,
1967                                                 &mreq, sizeof (mreq)) == -1)
1968                         {
1969                                 char errbuf[1024];
1970                                 ERROR ("network plugin: setsockopt (ipv6-add-membership): %s",
1971                                                 sstrerror (errno, errbuf,
1972                                                         sizeof (errbuf)));
1973                                 return (-1);
1974                         }
1975
1976                         return (0);
1977                 }
1978         }
1979
1980 #if defined(HAVE_IF_INDEXTONAME) && HAVE_IF_INDEXTONAME && defined(SO_BINDTODEVICE)
1981         /* if a specific interface was set, bind the socket to it. But to avoid
1982          * possible problems with multicast routing, only do that for non-multicast
1983          * addresses */
1984         if (interface_idx != 0)
1985         {
1986                 char interface_name[IFNAMSIZ];
1987
1988                 if (if_indextoname (interface_idx, interface_name) == NULL)
1989                         return (-1);
1990
1991                 DEBUG ("fd = %i; Binding socket to interface %s", fd, interface_name);
1992
1993                 if (setsockopt (fd, SOL_SOCKET, SO_BINDTODEVICE,
1994                                         interface_name,
1995                                         sizeof(interface_name)) == -1 )
1996                 {
1997                         char errbuf[1024];
1998                         ERROR ("network plugin: setsockopt (bind-if): %s",
1999                                         sstrerror (errno, errbuf, sizeof (errbuf)));
2000                         return (-1);
2001                 }
2002         }
2003 #endif /* HAVE_IF_INDEXTONAME && SO_BINDTODEVICE */
2004
2005         return (0);
2006 } /* int network_bind_socket */
2007
2008 /* Initialize a sockent structure. `type' must be either `SOCKENT_TYPE_CLIENT'
2009  * or `SOCKENT_TYPE_SERVER' */
2010 static sockent_t *sockent_create (int type) /* {{{ */
2011 {
2012         sockent_t *se;
2013
2014         if ((type != SOCKENT_TYPE_CLIENT) && (type != SOCKENT_TYPE_SERVER))
2015                 return (NULL);
2016
2017         se = malloc (sizeof (*se));
2018         if (se == NULL)
2019                 return (NULL);
2020         memset (se, 0, sizeof (*se));
2021
2022         se->type = type;
2023         se->node = NULL;
2024         se->service = NULL;
2025         se->interface = 0;
2026         se->next = NULL;
2027
2028         if (type == SOCKENT_TYPE_SERVER)
2029         {
2030                 se->data.server.fd = NULL;
2031 #if HAVE_LIBGCRYPT
2032                 se->data.server.security_level = SECURITY_LEVEL_NONE;
2033                 se->data.server.auth_file = NULL;
2034                 se->data.server.userdb = NULL;
2035                 se->data.server.cypher = NULL;
2036 #endif
2037         }
2038         else
2039         {
2040                 se->data.client.fd = -1;
2041                 se->data.client.addr = NULL;
2042                 se->data.client.resolve_interval = 0;
2043                 se->data.client.next_resolve_reconnect = 0;
2044 #if HAVE_LIBGCRYPT
2045                 se->data.client.security_level = SECURITY_LEVEL_NONE;
2046                 se->data.client.username = NULL;
2047                 se->data.client.password = NULL;
2048                 se->data.client.cypher = NULL;
2049 #endif
2050         }
2051
2052         return (se);
2053 } /* }}} sockent_t *sockent_create */
2054
2055 static int sockent_init_crypto (sockent_t *se) /* {{{ */
2056 {
2057 #if HAVE_LIBGCRYPT /* {{{ */
2058         if (se->type == SOCKENT_TYPE_CLIENT)
2059         {
2060                 if (se->data.client.security_level > SECURITY_LEVEL_NONE)
2061                 {
2062                         network_init_gcrypt ();
2063
2064                         if ((se->data.client.username == NULL)
2065                                         || (se->data.client.password == NULL))
2066                         {
2067                                 ERROR ("network plugin: Client socket with "
2068                                                 "security requested, but no "
2069                                                 "credentials are configured.");
2070                                 return (-1);
2071                         }
2072                         gcry_md_hash_buffer (GCRY_MD_SHA256,
2073                                         se->data.client.password_hash,
2074                                         se->data.client.password,
2075                                         strlen (se->data.client.password));
2076                 }
2077         }
2078         else /* (se->type == SOCKENT_TYPE_SERVER) */
2079         {
2080                 if (se->data.server.security_level > SECURITY_LEVEL_NONE)
2081                 {
2082                         network_init_gcrypt ();
2083
2084                         if (se->data.server.auth_file == NULL)
2085                         {
2086                                 ERROR ("network plugin: Server socket with "
2087                                                 "security requested, but no "
2088                                                 "password file is configured.");
2089                                 return (-1);
2090                         }
2091                 }
2092                 if (se->data.server.auth_file != NULL)
2093                 {
2094                         se->data.server.userdb = fbh_create (se->data.server.auth_file);
2095                         if (se->data.server.userdb == NULL)
2096                         {
2097                                 ERROR ("network plugin: Reading password file "
2098                                                 "`%s' failed.",
2099                                                 se->data.server.auth_file);
2100                                 if (se->data.server.security_level > SECURITY_LEVEL_NONE)
2101                                         return (-1);
2102                         }
2103                 }
2104         }
2105 #endif /* }}} HAVE_LIBGCRYPT */
2106
2107         return (0);
2108 } /* }}} int sockent_init_crypto */
2109
2110 static int sockent_client_disconnect (sockent_t *se) /* {{{ */
2111 {
2112         struct sockent_client *client;
2113
2114         if ((se == NULL) || (se->type != SOCKENT_TYPE_CLIENT))
2115                 return (EINVAL);
2116
2117         client = &se->data.client;
2118         if (client->fd >= 0) /* connected */
2119         {
2120                 close (client->fd);
2121                 client->fd = -1;
2122         }
2123
2124         sfree (client->addr);
2125         client->addrlen = 0;
2126
2127         return (0);
2128 } /* }}} int sockent_client_disconnect */
2129
2130 static int sockent_client_connect (sockent_t *se) /* {{{ */
2131 {
2132         static c_complain_t complaint = C_COMPLAIN_INIT_STATIC;
2133
2134         struct sockent_client *client;
2135         struct addrinfo  ai_hints;
2136         struct addrinfo *ai_list = NULL, *ai_ptr;
2137         int status;
2138         _Bool reconnect = 0;
2139         cdtime_t now;
2140
2141         if ((se == NULL) || (se->type != SOCKENT_TYPE_CLIENT))
2142                 return (EINVAL);
2143
2144         client = &se->data.client;
2145
2146         now = cdtime ();
2147         if (client->resolve_interval != 0 && client->next_resolve_reconnect < now) {
2148                 DEBUG("network plugin: Reconnecting socket, resolve_interval = %lf, next_resolve_reconnect = %lf",
2149                         CDTIME_T_TO_DOUBLE(client->resolve_interval), CDTIME_T_TO_DOUBLE(client->next_resolve_reconnect));
2150                 reconnect = 1;
2151         }
2152
2153         if (client->fd >= 0 && !reconnect) /* already connected and not stale*/
2154                 return (0);
2155
2156         memset (&ai_hints, 0, sizeof (ai_hints));
2157 #ifdef AI_ADDRCONFIG
2158         ai_hints.ai_flags |= AI_ADDRCONFIG;
2159 #endif
2160         ai_hints.ai_family   = AF_UNSPEC;
2161         ai_hints.ai_socktype = SOCK_DGRAM;
2162         ai_hints.ai_protocol = IPPROTO_UDP;
2163
2164         status = getaddrinfo (se->node,
2165                         (se->service != NULL) ? se->service : NET_DEFAULT_PORT,
2166                         &ai_hints, &ai_list);
2167         if (status != 0)
2168         {
2169                 c_complain (LOG_ERR, &complaint,
2170                                 "network plugin: getaddrinfo (%s, %s) failed: %s",
2171                                 (se->node == NULL) ? "(null)" : se->node,
2172                                 (se->service == NULL) ? "(null)" : se->service,
2173                                 gai_strerror (status));
2174                 return (-1);
2175         }
2176         else
2177         {
2178                 c_release (LOG_NOTICE, &complaint,
2179                                 "network plugin: Successfully resolved \"%s\".",
2180                                 se->node);
2181         }
2182
2183         for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
2184         {
2185                 if (client->fd >= 0) /* when we reconnect */
2186                         sockent_client_disconnect(se);
2187
2188                 client->fd = socket (ai_ptr->ai_family,
2189                                 ai_ptr->ai_socktype,
2190                                 ai_ptr->ai_protocol);
2191                 if (client->fd < 0)
2192                 {
2193                         char errbuf[1024];
2194                         ERROR ("network plugin: socket(2) failed: %s",
2195                                         sstrerror (errno, errbuf,
2196                                                 sizeof (errbuf)));
2197                         continue;
2198                 }
2199
2200                 client->addr = malloc (sizeof (*client->addr));
2201                 if (client->addr == NULL)
2202                 {
2203                         ERROR ("network plugin: malloc failed.");
2204                         close (client->fd);
2205                         client->fd = -1;
2206                         continue;
2207                 }
2208
2209                 memset (client->addr, 0, sizeof (*client->addr));
2210                 assert (sizeof (*client->addr) >= ai_ptr->ai_addrlen);
2211                 memcpy (client->addr, ai_ptr->ai_addr, ai_ptr->ai_addrlen);
2212                 client->addrlen = ai_ptr->ai_addrlen;
2213
2214                 network_set_ttl (se, ai_ptr);
2215                 network_set_interface (se, ai_ptr);
2216
2217                 /* We don't open more than one write-socket per
2218                  * node/service pair.. */
2219                 break;
2220         }
2221
2222         freeaddrinfo (ai_list);
2223         if (client->fd < 0)
2224                 return (-1);
2225
2226         if (client->resolve_interval > 0)
2227                 client->next_resolve_reconnect = now + client->resolve_interval;
2228         return (0);
2229 } /* }}} int sockent_client_connect */
2230
2231 /* Open the file descriptors for a initialized sockent structure. */
2232 static int sockent_server_listen (sockent_t *se) /* {{{ */
2233 {
2234         struct addrinfo  ai_hints;
2235         struct addrinfo *ai_list, *ai_ptr;
2236         int              status;
2237
2238         const char *node;
2239         const char *service;
2240
2241         if (se == NULL)
2242                 return (-1);
2243
2244         node = se->node;
2245         service = se->service;
2246
2247         if (service == NULL)
2248           service = NET_DEFAULT_PORT;
2249
2250         DEBUG ("network plugin: sockent_server_listen: node = %s; service = %s;",
2251             node, service);
2252
2253         memset (&ai_hints, 0, sizeof (ai_hints));
2254         ai_hints.ai_flags  = 0;
2255 #ifdef AI_PASSIVE
2256         ai_hints.ai_flags |= AI_PASSIVE;
2257 #endif
2258 #ifdef AI_ADDRCONFIG
2259         ai_hints.ai_flags |= AI_ADDRCONFIG;
2260 #endif
2261         ai_hints.ai_family   = AF_UNSPEC;
2262         ai_hints.ai_socktype = SOCK_DGRAM;
2263         ai_hints.ai_protocol = IPPROTO_UDP;
2264
2265         status = getaddrinfo (node, service, &ai_hints, &ai_list);
2266         if (status != 0)
2267         {
2268                 ERROR ("network plugin: getaddrinfo (%s, %s) failed: %s",
2269                                 (se->node == NULL) ? "(null)" : se->node,
2270                                 (se->service == NULL) ? "(null)" : se->service,
2271                                 gai_strerror (status));
2272                 return (-1);
2273         }
2274
2275         for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
2276         {
2277                 int *tmp;
2278
2279                 tmp = realloc (se->data.server.fd,
2280                                 sizeof (*tmp) * (se->data.server.fd_num + 1));
2281                 if (tmp == NULL)
2282                 {
2283                         ERROR ("network plugin: realloc failed.");
2284                         continue;
2285                 }
2286                 se->data.server.fd = tmp;
2287                 tmp = se->data.server.fd + se->data.server.fd_num;
2288
2289                 *tmp = socket (ai_ptr->ai_family, ai_ptr->ai_socktype,
2290                                 ai_ptr->ai_protocol);
2291                 if (*tmp < 0)
2292                 {
2293                         char errbuf[1024];
2294                         ERROR ("network plugin: socket(2) failed: %s",
2295                                         sstrerror (errno, errbuf,
2296                                                 sizeof (errbuf)));
2297                         continue;
2298                 }
2299
2300                 status = network_bind_socket (*tmp, ai_ptr, se->interface);
2301                 if (status != 0)
2302                 {
2303                         close (*tmp);
2304                         *tmp = -1;
2305                         continue;
2306                 }
2307
2308                 se->data.server.fd_num++;
2309                 continue;
2310         } /* for (ai_list) */
2311
2312         freeaddrinfo (ai_list);
2313
2314         if (se->data.server.fd_num <= 0)
2315                 return (-1);
2316         return (0);
2317 } /* }}} int sockent_server_listen */
2318
2319 /* Add a sockent to the global list of sockets */
2320 static int sockent_add (sockent_t *se) /* {{{ */
2321 {
2322         sockent_t *last_ptr;
2323
2324         if (se == NULL)
2325                 return (-1);
2326
2327         if (se->type == SOCKENT_TYPE_SERVER)
2328         {
2329                 struct pollfd *tmp;
2330                 size_t i;
2331
2332                 tmp = realloc (listen_sockets_pollfd,
2333                                 sizeof (*tmp) * (listen_sockets_num
2334                                         + se->data.server.fd_num));
2335                 if (tmp == NULL)
2336                 {
2337                         ERROR ("network plugin: realloc failed.");
2338                         return (-1);
2339                 }
2340                 listen_sockets_pollfd = tmp;
2341                 tmp = listen_sockets_pollfd + listen_sockets_num;
2342
2343                 for (i = 0; i < se->data.server.fd_num; i++)
2344                 {
2345                         memset (tmp + i, 0, sizeof (*tmp));
2346                         tmp[i].fd = se->data.server.fd[i];
2347                         tmp[i].events = POLLIN | POLLPRI;
2348                         tmp[i].revents = 0;
2349                 }
2350
2351                 listen_sockets_num += se->data.server.fd_num;
2352
2353                 if (listen_sockets == NULL)
2354                 {
2355                         listen_sockets = se;
2356                         return (0);
2357                 }
2358                 last_ptr = listen_sockets;
2359         }
2360         else /* if (se->type == SOCKENT_TYPE_CLIENT) */
2361         {
2362                 if (sending_sockets == NULL)
2363                 {
2364                         sending_sockets = se;
2365                         return (0);
2366                 }
2367                 last_ptr = sending_sockets;
2368         }
2369
2370         while (last_ptr->next != NULL)
2371                 last_ptr = last_ptr->next;
2372         last_ptr->next = se;
2373
2374         return (0);
2375 } /* }}} int sockent_add */
2376
2377 static void *dispatch_thread (void __attribute__((unused)) *arg) /* {{{ */
2378 {
2379   while (42)
2380   {
2381     receive_list_entry_t *ent;
2382     sockent_t *se;
2383
2384     /* Lock and wait for more data to come in */
2385     pthread_mutex_lock (&receive_list_lock);
2386     while ((listen_loop == 0)
2387         && (receive_list_head == NULL))
2388       pthread_cond_wait (&receive_list_cond, &receive_list_lock);
2389
2390     /* Remove the head entry and unlock */
2391     ent = receive_list_head;
2392     if (ent != NULL)
2393       receive_list_head = ent->next;
2394     receive_list_length--;
2395     pthread_mutex_unlock (&receive_list_lock);
2396
2397     /* Check whether we are supposed to exit. We do NOT check `listen_loop'
2398      * because we dispatch all missing packets before shutting down. */
2399     if (ent == NULL)
2400       break;
2401
2402     /* Look for the correct `sockent_t' */
2403     se = listen_sockets;
2404     while (se != NULL)
2405     {
2406       size_t i;
2407
2408       for (i = 0; i < se->data.server.fd_num; i++)
2409         if (se->data.server.fd[i] == ent->fd)
2410           break;
2411
2412       if (i < se->data.server.fd_num)
2413         break;
2414
2415       se = se->next;
2416     }
2417
2418     if (se == NULL)
2419     {
2420       ERROR ("network plugin: Got packet from FD %i, but can't "
2421           "find an appropriate socket entry.",
2422           ent->fd);
2423       sfree (ent->data);
2424       sfree (ent);
2425       continue;
2426     }
2427
2428     parse_packet (se, ent->data, ent->data_len, /* flags = */ 0,
2429         /* username = */ NULL);
2430     sfree (ent->data);
2431     sfree (ent);
2432   } /* while (42) */
2433
2434   return (NULL);
2435 } /* }}} void *dispatch_thread */
2436
2437 static int network_receive (void) /* {{{ */
2438 {
2439         char buffer[network_config_packet_size];
2440         int  buffer_len;
2441
2442         int i;
2443         int status;
2444
2445         receive_list_entry_t *private_list_head;
2446         receive_list_entry_t *private_list_tail;
2447         uint64_t              private_list_length;
2448
2449         assert (listen_sockets_num > 0);
2450
2451         private_list_head = NULL;
2452         private_list_tail = NULL;
2453         private_list_length = 0;
2454
2455         while (listen_loop == 0)
2456         {
2457                 status = poll (listen_sockets_pollfd, listen_sockets_num, -1);
2458
2459                 if (status <= 0)
2460                 {
2461                         char errbuf[1024];
2462                         if (errno == EINTR)
2463                                 continue;
2464                         ERROR ("poll failed: %s",
2465                                         sstrerror (errno, errbuf, sizeof (errbuf)));
2466                         return (-1);
2467                 }
2468
2469                 for (i = 0; (i < listen_sockets_num) && (status > 0); i++)
2470                 {
2471                         receive_list_entry_t *ent;
2472
2473                         if ((listen_sockets_pollfd[i].revents
2474                                                 & (POLLIN | POLLPRI)) == 0)
2475                                 continue;
2476                         status--;
2477
2478                         buffer_len = recv (listen_sockets_pollfd[i].fd,
2479                                         buffer, sizeof (buffer),
2480                                         0 /* no flags */);
2481                         if (buffer_len < 0)
2482                         {
2483                                 char errbuf[1024];
2484                                 ERROR ("recv failed: %s",
2485                                                 sstrerror (errno, errbuf,
2486                                                         sizeof (errbuf)));
2487                                 return (-1);
2488                         }
2489
2490                         stats_octets_rx += ((uint64_t) buffer_len);
2491                         stats_packets_rx++;
2492
2493                         /* TODO: Possible performance enhancement: Do not free
2494                          * these entries in the dispatch thread but put them in
2495                          * another list, so we don't have to allocate more and
2496                          * more of these structures. */
2497                         ent = malloc (sizeof (receive_list_entry_t));
2498                         if (ent == NULL)
2499                         {
2500                                 ERROR ("network plugin: malloc failed.");
2501                                 return (-1);
2502                         }
2503                         memset (ent, 0, sizeof (receive_list_entry_t));
2504                         ent->data = malloc (network_config_packet_size);
2505                         if (ent->data == NULL)
2506                         {
2507                                 sfree (ent);
2508                                 ERROR ("network plugin: malloc failed.");
2509                                 return (-1);
2510                         }
2511                         ent->fd = listen_sockets_pollfd[i].fd;
2512                         ent->next = NULL;
2513
2514                         memcpy (ent->data, buffer, buffer_len);
2515                         ent->data_len = buffer_len;
2516
2517                         if (private_list_head == NULL)
2518                                 private_list_head = ent;
2519                         else
2520                                 private_list_tail->next = ent;
2521                         private_list_tail = ent;
2522                         private_list_length++;
2523
2524                         /* Do not block here. Blocking here has led to
2525                          * insufficient performance in the past. */
2526                         if (pthread_mutex_trylock (&receive_list_lock) == 0)
2527                         {
2528                                 assert (((receive_list_head == NULL) && (receive_list_length == 0))
2529                                                 || ((receive_list_head != NULL) && (receive_list_length != 0)));
2530
2531                                 if (receive_list_head == NULL)
2532                                         receive_list_head = private_list_head;
2533                                 else
2534                                         receive_list_tail->next = private_list_head;
2535                                 receive_list_tail = private_list_tail;
2536                                 receive_list_length += private_list_length;
2537
2538                                 pthread_cond_signal (&receive_list_cond);
2539                                 pthread_mutex_unlock (&receive_list_lock);
2540
2541                                 private_list_head = NULL;
2542                                 private_list_tail = NULL;
2543                                 private_list_length = 0;
2544                         }
2545                 } /* for (listen_sockets_pollfd) */
2546         } /* while (listen_loop == 0) */
2547
2548         /* Make sure everything is dispatched before exiting. */
2549         if (private_list_head != NULL)
2550         {
2551                 pthread_mutex_lock (&receive_list_lock);
2552
2553                 if (receive_list_head == NULL)
2554                         receive_list_head = private_list_head;
2555                 else
2556                         receive_list_tail->next = private_list_head;
2557                 receive_list_tail = private_list_tail;
2558                 receive_list_length += private_list_length;
2559
2560                 private_list_head = NULL;
2561                 private_list_tail = NULL;
2562                 private_list_length = 0;
2563
2564                 pthread_cond_signal (&receive_list_cond);
2565                 pthread_mutex_unlock (&receive_list_lock);
2566         }
2567
2568         return (0);
2569 } /* }}} int network_receive */
2570
2571 static void *receive_thread (void __attribute__((unused)) *arg)
2572 {
2573         return (network_receive () ? (void *) 1 : (void *) 0);
2574 } /* void *receive_thread */
2575
2576 static void network_init_buffer (void)
2577 {
2578         memset (send_buffer, 0, network_config_packet_size);
2579         send_buffer_ptr = send_buffer;
2580         send_buffer_fill = 0;
2581         send_buffer_last_update = 0;
2582
2583         memset (&send_buffer_vl, 0, sizeof (send_buffer_vl));
2584 } /* int network_init_buffer */
2585
2586 static void networt_send_buffer_plain (sockent_t *se, /* {{{ */
2587                 const char *buffer, size_t buffer_size)
2588 {
2589         int status;
2590
2591         while (42)
2592         {
2593                 status = sockent_client_connect (se);
2594                 if (status != 0)
2595                         return;
2596
2597                 status = sendto (se->data.client.fd, buffer, buffer_size,
2598                                 /* flags = */ 0,
2599                                 (struct sockaddr *) se->data.client.addr,
2600                                 se->data.client.addrlen);
2601                 if (status < 0)
2602                 {
2603                         char errbuf[1024];
2604
2605                         if ((errno == EINTR) || (errno == EAGAIN))
2606                                 continue;
2607
2608                         ERROR ("network plugin: sendto failed: %s. Closing sending socket.",
2609                                         sstrerror (errno, errbuf, sizeof (errbuf)));
2610                         sockent_client_disconnect (se);
2611                         return;
2612                 }
2613
2614                 break;
2615         } /* while (42) */
2616 } /* }}} void networt_send_buffer_plain */
2617
2618 #if HAVE_LIBGCRYPT
2619 #define BUFFER_ADD(p,s) do { \
2620   memcpy (buffer + buffer_offset, (p), (s)); \
2621   buffer_offset += (s); \
2622 } while (0)
2623
2624 static void networt_send_buffer_signed (sockent_t *se, /* {{{ */
2625                 const char *in_buffer, size_t in_buffer_size)
2626 {
2627   part_signature_sha256_t ps;
2628   char buffer[BUFF_SIG_SIZE + in_buffer_size];
2629   size_t buffer_offset;
2630   size_t username_len;
2631
2632   gcry_md_hd_t hd;
2633   gcry_error_t err;
2634   unsigned char *hash;
2635
2636   hd = NULL;
2637   err = gcry_md_open (&hd, GCRY_MD_SHA256, GCRY_MD_FLAG_HMAC);
2638   if (err != 0)
2639   {
2640     ERROR ("network plugin: Creating HMAC object failed: %s",
2641         gcry_strerror (err));
2642     return;
2643   }
2644
2645   err = gcry_md_setkey (hd, se->data.client.password,
2646       strlen (se->data.client.password));
2647   if (err != 0)
2648   {
2649     ERROR ("network plugin: gcry_md_setkey failed: %s",
2650         gcry_strerror (err));
2651     gcry_md_close (hd);
2652     return;
2653   }
2654
2655   username_len = strlen (se->data.client.username);
2656   if (username_len > (BUFF_SIG_SIZE - PART_SIGNATURE_SHA256_SIZE))
2657   {
2658     ERROR ("network plugin: Username too long: %s",
2659         se->data.client.username);
2660     return;
2661   }
2662
2663   memcpy (buffer + PART_SIGNATURE_SHA256_SIZE,
2664       se->data.client.username, username_len);
2665   memcpy (buffer + PART_SIGNATURE_SHA256_SIZE + username_len,
2666       in_buffer, in_buffer_size);
2667
2668   /* Initialize the `ps' structure. */
2669   memset (&ps, 0, sizeof (ps));
2670   ps.head.type = htons (TYPE_SIGN_SHA256);
2671   ps.head.length = htons (PART_SIGNATURE_SHA256_SIZE + username_len);
2672
2673   /* Calculate the hash value. */
2674   gcry_md_write (hd, buffer + PART_SIGNATURE_SHA256_SIZE,
2675       username_len + in_buffer_size);
2676   hash = gcry_md_read (hd, GCRY_MD_SHA256);
2677   if (hash == NULL)
2678   {
2679     ERROR ("network plugin: gcry_md_read failed.");
2680     gcry_md_close (hd);
2681     return;
2682   }
2683   memcpy (ps.hash, hash, sizeof (ps.hash));
2684
2685   /* Add the header */
2686   buffer_offset = 0;
2687
2688   BUFFER_ADD (&ps.head.type, sizeof (ps.head.type));
2689   BUFFER_ADD (&ps.head.length, sizeof (ps.head.length));
2690   BUFFER_ADD (ps.hash, sizeof (ps.hash));
2691
2692   assert (buffer_offset == PART_SIGNATURE_SHA256_SIZE);
2693
2694   gcry_md_close (hd);
2695   hd = NULL;
2696
2697   buffer_offset = PART_SIGNATURE_SHA256_SIZE + username_len + in_buffer_size;
2698   networt_send_buffer_plain (se, buffer, buffer_offset);
2699 } /* }}} void networt_send_buffer_signed */
2700
2701 static void networt_send_buffer_encrypted (sockent_t *se, /* {{{ */
2702                 const char *in_buffer, size_t in_buffer_size)
2703 {
2704   part_encryption_aes256_t pea;
2705   char buffer[BUFF_SIG_SIZE + in_buffer_size];
2706   size_t buffer_size;
2707   size_t buffer_offset;
2708   size_t header_size;
2709   size_t username_len;
2710   gcry_error_t err;
2711   gcry_cipher_hd_t cypher;
2712
2713   /* Initialize the header fields */
2714   memset (&pea, 0, sizeof (pea));
2715   pea.head.type = htons (TYPE_ENCR_AES256);
2716
2717   pea.username = se->data.client.username;
2718
2719   username_len = strlen (pea.username);
2720   if ((PART_ENCRYPTION_AES256_SIZE + username_len) > BUFF_SIG_SIZE)
2721   {
2722     ERROR ("network plugin: Username too long: %s", pea.username);
2723     return;
2724   }
2725
2726   buffer_size = PART_ENCRYPTION_AES256_SIZE + username_len + in_buffer_size;
2727   header_size = PART_ENCRYPTION_AES256_SIZE + username_len
2728     - sizeof (pea.hash);
2729
2730   assert (buffer_size <= sizeof (buffer));
2731   DEBUG ("network plugin: networt_send_buffer_encrypted: "
2732       "buffer_size = %zu;", buffer_size);
2733
2734   pea.head.length = htons ((uint16_t) (PART_ENCRYPTION_AES256_SIZE
2735         + username_len + in_buffer_size));
2736   pea.username_length = htons ((uint16_t) username_len);
2737
2738   /* Chose a random initialization vector. */
2739   gcry_randomize ((void *) &pea.iv, sizeof (pea.iv), GCRY_STRONG_RANDOM);
2740
2741   /* Create hash of the payload */
2742   gcry_md_hash_buffer (GCRY_MD_SHA1, pea.hash, in_buffer, in_buffer_size);
2743
2744   /* Initialize the buffer */
2745   buffer_offset = 0;
2746   memset (buffer, 0, sizeof (buffer));
2747
2748
2749   BUFFER_ADD (&pea.head.type, sizeof (pea.head.type));
2750   BUFFER_ADD (&pea.head.length, sizeof (pea.head.length));
2751   BUFFER_ADD (&pea.username_length, sizeof (pea.username_length));
2752   BUFFER_ADD (pea.username, username_len);
2753   BUFFER_ADD (pea.iv, sizeof (pea.iv));
2754   assert (buffer_offset == header_size);
2755   BUFFER_ADD (pea.hash, sizeof (pea.hash));
2756   BUFFER_ADD (in_buffer, in_buffer_size);
2757
2758   assert (buffer_offset == buffer_size);
2759
2760   cypher = network_get_aes256_cypher (se, pea.iv, sizeof (pea.iv),
2761       se->data.client.password);
2762   if (cypher == NULL)
2763     return;
2764
2765   /* Encrypt the buffer in-place */
2766   err = gcry_cipher_encrypt (cypher,
2767       buffer      + header_size,
2768       buffer_size - header_size,
2769       /* in = */ NULL, /* in len = */ 0);
2770   if (err != 0)
2771   {
2772     ERROR ("network plugin: gcry_cipher_encrypt returned: %s",
2773         gcry_strerror (err));
2774     return;
2775   }
2776
2777   /* Send it out without further modifications */
2778   networt_send_buffer_plain (se, buffer, buffer_size);
2779 } /* }}} void networt_send_buffer_encrypted */
2780 #undef BUFFER_ADD
2781 #endif /* HAVE_LIBGCRYPT */
2782
2783 static void network_send_buffer (char *buffer, size_t buffer_len) /* {{{ */
2784 {
2785   sockent_t *se;
2786
2787   DEBUG ("network plugin: network_send_buffer: buffer_len = %zu", buffer_len);
2788
2789   for (se = sending_sockets; se != NULL; se = se->next)
2790   {
2791 #if HAVE_LIBGCRYPT
2792     if (se->data.client.security_level == SECURITY_LEVEL_ENCRYPT)
2793       networt_send_buffer_encrypted (se, buffer, buffer_len);
2794     else if (se->data.client.security_level == SECURITY_LEVEL_SIGN)
2795       networt_send_buffer_signed (se, buffer, buffer_len);
2796     else /* if (se->data.client.security_level == SECURITY_LEVEL_NONE) */
2797 #endif /* HAVE_LIBGCRYPT */
2798       networt_send_buffer_plain (se, buffer, buffer_len);
2799   } /* for (sending_sockets) */
2800 } /* }}} void network_send_buffer */
2801
2802 static int add_to_buffer (char *buffer, int buffer_size, /* {{{ */
2803                 value_list_t *vl_def,
2804                 const data_set_t *ds, const value_list_t *vl)
2805 {
2806         char *buffer_orig = buffer;
2807
2808         if (strcmp (vl_def->host, vl->host) != 0)
2809         {
2810                 if (write_part_string (&buffer, &buffer_size, TYPE_HOST,
2811                                         vl->host, strlen (vl->host)) != 0)
2812                         return (-1);
2813                 sstrncpy (vl_def->host, vl->host, sizeof (vl_def->host));
2814         }
2815
2816         if (vl_def->time != vl->time)
2817         {
2818                 if (write_part_number (&buffer, &buffer_size, TYPE_TIME_HR,
2819                                         (uint64_t) vl->time))
2820                         return (-1);
2821                 vl_def->time = vl->time;
2822         }
2823
2824         if (vl_def->interval != vl->interval)
2825         {
2826                 if (write_part_number (&buffer, &buffer_size, TYPE_INTERVAL_HR,
2827                                         (uint64_t) vl->interval))
2828                         return (-1);
2829                 vl_def->interval = vl->interval;
2830         }
2831
2832         if (strcmp (vl_def->plugin, vl->plugin) != 0)
2833         {
2834                 if (write_part_string (&buffer, &buffer_size, TYPE_PLUGIN,
2835                                         vl->plugin, strlen (vl->plugin)) != 0)
2836                         return (-1);
2837                 sstrncpy (vl_def->plugin, vl->plugin, sizeof (vl_def->plugin));
2838         }
2839
2840         if (strcmp (vl_def->plugin_instance, vl->plugin_instance) != 0)
2841         {
2842                 if (write_part_string (&buffer, &buffer_size, TYPE_PLUGIN_INSTANCE,
2843                                         vl->plugin_instance,
2844                                         strlen (vl->plugin_instance)) != 0)
2845                         return (-1);
2846                 sstrncpy (vl_def->plugin_instance, vl->plugin_instance, sizeof (vl_def->plugin_instance));
2847         }
2848
2849         if (strcmp (vl_def->type, vl->type) != 0)
2850         {
2851                 if (write_part_string (&buffer, &buffer_size, TYPE_TYPE,
2852                                         vl->type, strlen (vl->type)) != 0)
2853                         return (-1);
2854                 sstrncpy (vl_def->type, ds->type, sizeof (vl_def->type));
2855         }
2856
2857         if (strcmp (vl_def->type_instance, vl->type_instance) != 0)
2858         {
2859                 if (write_part_string (&buffer, &buffer_size, TYPE_TYPE_INSTANCE,
2860                                         vl->type_instance,
2861                                         strlen (vl->type_instance)) != 0)
2862                         return (-1);
2863                 sstrncpy (vl_def->type_instance, vl->type_instance, sizeof (vl_def->type_instance));
2864         }
2865
2866         if (write_part_values (&buffer, &buffer_size, ds, vl) != 0)
2867                 return (-1);
2868
2869         return (buffer - buffer_orig);
2870 } /* }}} int add_to_buffer */
2871
2872 static void flush_buffer (void)
2873 {
2874         DEBUG ("network plugin: flush_buffer: send_buffer_fill = %i",
2875                         send_buffer_fill);
2876
2877         network_send_buffer (send_buffer, (size_t) send_buffer_fill);
2878
2879         stats_octets_tx += ((uint64_t) send_buffer_fill);
2880         stats_packets_tx++;
2881
2882         network_init_buffer ();
2883 }
2884
2885 static int network_write (const data_set_t *ds, const value_list_t *vl,
2886                 user_data_t __attribute__((unused)) *user_data)
2887 {
2888         int status;
2889
2890         if (!check_send_okay (vl))
2891         {
2892 #if COLLECT_DEBUG
2893           char name[6*DATA_MAX_NAME_LEN];
2894           FORMAT_VL (name, sizeof (name), vl);
2895           name[sizeof (name) - 1] = 0;
2896           DEBUG ("network plugin: network_write: "
2897               "NOT sending %s.", name);
2898 #endif
2899           /* Counter is not protected by another lock and may be reached by
2900            * multiple threads */
2901           pthread_mutex_lock (&stats_lock);
2902           stats_values_not_sent++;
2903           pthread_mutex_unlock (&stats_lock);
2904           return (0);
2905         }
2906
2907         uc_meta_data_add_unsigned_int (vl,
2908             "network:time_sent", (uint64_t) vl->time);
2909
2910         pthread_mutex_lock (&send_buffer_lock);
2911
2912         status = add_to_buffer (send_buffer_ptr,
2913                         network_config_packet_size - (send_buffer_fill + BUFF_SIG_SIZE),
2914                         &send_buffer_vl,
2915                         ds, vl);
2916         if (status >= 0)
2917         {
2918                 /* status == bytes added to the buffer */
2919                 send_buffer_fill += status;
2920                 send_buffer_ptr  += status;
2921                 send_buffer_last_update = cdtime();
2922
2923                 stats_values_sent++;
2924         }
2925         else
2926         {
2927                 flush_buffer ();
2928
2929                 status = add_to_buffer (send_buffer_ptr,
2930                                 network_config_packet_size - (send_buffer_fill + BUFF_SIG_SIZE),
2931                                 &send_buffer_vl,
2932                                 ds, vl);
2933
2934                 if (status >= 0)
2935                 {
2936                         send_buffer_fill += status;
2937                         send_buffer_ptr  += status;
2938
2939                         stats_values_sent++;
2940                 }
2941         }
2942
2943         if (status < 0)
2944         {
2945                 ERROR ("network plugin: Unable to append to the "
2946                                 "buffer for some weird reason");
2947         }
2948         else if ((network_config_packet_size - send_buffer_fill) < 15)
2949         {
2950                 flush_buffer ();
2951         }
2952
2953         pthread_mutex_unlock (&send_buffer_lock);
2954
2955         return ((status < 0) ? -1 : 0);
2956 } /* int network_write */
2957
2958 static int network_config_set_boolean (const oconfig_item_t *ci, /* {{{ */
2959     int *retval)
2960 {
2961   if ((ci->values_num != 1)
2962       || ((ci->values[0].type != OCONFIG_TYPE_BOOLEAN)
2963         && (ci->values[0].type != OCONFIG_TYPE_STRING)))
2964   {
2965     ERROR ("network plugin: The `%s' config option needs "
2966         "exactly one boolean argument.", ci->key);
2967     return (-1);
2968   }
2969
2970   if (ci->values[0].type == OCONFIG_TYPE_BOOLEAN)
2971   {
2972     if (ci->values[0].value.boolean)
2973       *retval = 1;
2974     else
2975       *retval = 0;
2976   }
2977   else
2978   {
2979     char *str = ci->values[0].value.string;
2980
2981     if (IS_TRUE (str))
2982       *retval = 1;
2983     else if (IS_FALSE (str))
2984       *retval = 0;
2985     else
2986     {
2987       ERROR ("network plugin: Cannot parse string value `%s' of the `%s' "
2988           "option as boolean value.",
2989           str, ci->key);
2990       return (-1);
2991     }
2992   }
2993
2994   return (0);
2995 } /* }}} int network_config_set_boolean */
2996
2997 static int network_config_set_ttl (const oconfig_item_t *ci) /* {{{ */
2998 {
2999   int tmp;
3000   if ((ci->values_num != 1)
3001       || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
3002   {
3003     WARNING ("network plugin: The `TimeToLive' config option needs exactly "
3004         "one numeric argument.");
3005     return (-1);
3006   }
3007
3008   tmp = (int) ci->values[0].value.number;
3009   if ((tmp > 0) && (tmp <= 255))
3010     network_config_ttl = tmp;
3011   else {
3012     WARNING ("network plugin: The `TimeToLive' must be between 1 and 255.");
3013     return (-1);
3014   }
3015
3016   return (0);
3017 } /* }}} int network_config_set_ttl */
3018
3019 static int network_config_set_interface (const oconfig_item_t *ci, /* {{{ */
3020     int *interface)
3021 {
3022   if ((ci->values_num != 1)
3023       || (ci->values[0].type != OCONFIG_TYPE_STRING))
3024   {
3025     WARNING ("network plugin: The `Interface' config option needs exactly "
3026         "one string argument.");
3027     return (-1);
3028   }
3029
3030   if (interface == NULL)
3031     return (-1);
3032
3033   *interface = if_nametoindex (ci->values[0].value.string);
3034
3035   return (0);
3036 } /* }}} int network_config_set_interface */
3037
3038 static int network_config_set_buffer_size (const oconfig_item_t *ci) /* {{{ */
3039 {
3040   int tmp;
3041   if ((ci->values_num != 1)
3042       || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
3043   {
3044     WARNING ("network plugin: The `MaxPacketSize' config option needs exactly "
3045         "one numeric argument.");
3046     return (-1);
3047   }
3048
3049   tmp = (int) ci->values[0].value.number;
3050   if ((tmp >= 1024) && (tmp <= 65535))
3051     network_config_packet_size = tmp;
3052
3053   return (0);
3054 } /* }}} int network_config_set_buffer_size */
3055
3056 #if HAVE_LIBGCRYPT
3057 static int network_config_set_string (const oconfig_item_t *ci, /* {{{ */
3058     char **ret_string)
3059 {
3060   char *tmp;
3061   if ((ci->values_num != 1)
3062       || (ci->values[0].type != OCONFIG_TYPE_STRING))
3063   {
3064     WARNING ("network plugin: The `%s' config option needs exactly "
3065         "one string argument.", ci->key);
3066     return (-1);
3067   }
3068
3069   tmp = strdup (ci->values[0].value.string);
3070   if (tmp == NULL)
3071     return (-1);
3072
3073   sfree (*ret_string);
3074   *ret_string = tmp;
3075
3076   return (0);
3077 } /* }}} int network_config_set_string */
3078 #endif /* HAVE_LIBGCRYPT */
3079
3080 #if HAVE_LIBGCRYPT
3081 static int network_config_set_security_level (oconfig_item_t *ci, /* {{{ */
3082     int *retval)
3083 {
3084   char *str;
3085   if ((ci->values_num != 1)
3086       || (ci->values[0].type != OCONFIG_TYPE_STRING))
3087   {
3088     WARNING ("network plugin: The `SecurityLevel' config option needs exactly "
3089         "one string argument.");
3090     return (-1);
3091   }
3092
3093   str = ci->values[0].value.string;
3094   if (strcasecmp ("Encrypt", str) == 0)
3095     *retval = SECURITY_LEVEL_ENCRYPT;
3096   else if (strcasecmp ("Sign", str) == 0)
3097     *retval = SECURITY_LEVEL_SIGN;
3098   else if (strcasecmp ("None", str) == 0)
3099     *retval = SECURITY_LEVEL_NONE;
3100   else
3101   {
3102     WARNING ("network plugin: Unknown security level: %s.", str);
3103     return (-1);
3104   }
3105
3106   return (0);
3107 } /* }}} int network_config_set_security_level */
3108 #endif /* HAVE_LIBGCRYPT */
3109
3110 static int network_config_add_listen (const oconfig_item_t *ci) /* {{{ */
3111 {
3112   sockent_t *se;
3113   int status;
3114   int i;
3115
3116   if ((ci->values_num < 1) || (ci->values_num > 2)
3117       || (ci->values[0].type != OCONFIG_TYPE_STRING)
3118       || ((ci->values_num > 1) && (ci->values[1].type != OCONFIG_TYPE_STRING)))
3119   {
3120     ERROR ("network plugin: The `%s' config option needs "
3121         "one or two string arguments.", ci->key);
3122     return (-1);
3123   }
3124
3125   se = sockent_create (SOCKENT_TYPE_SERVER);
3126   if (se == NULL)
3127   {
3128     ERROR ("network plugin: sockent_create failed.");
3129     return (-1);
3130   }
3131
3132   se->node = strdup (ci->values[0].value.string);
3133   if (ci->values_num >= 2)
3134     se->service = strdup (ci->values[1].value.string);
3135
3136   for (i = 0; i < ci->children_num; i++)
3137   {
3138     oconfig_item_t *child = ci->children + i;
3139
3140 #if HAVE_LIBGCRYPT
3141     if (strcasecmp ("AuthFile", child->key) == 0)
3142       network_config_set_string (child, &se->data.server.auth_file);
3143     else if (strcasecmp ("SecurityLevel", child->key) == 0)
3144       network_config_set_security_level (child,
3145           &se->data.server.security_level);
3146     else
3147 #endif /* HAVE_LIBGCRYPT */
3148     if (strcasecmp ("Interface", child->key) == 0)
3149       network_config_set_interface (child,
3150           &se->interface);
3151     else
3152     {
3153       WARNING ("network plugin: Option `%s' is not allowed here.",
3154           child->key);
3155     }
3156   }
3157
3158 #if HAVE_LIBGCRYPT
3159   if ((se->data.server.security_level > SECURITY_LEVEL_NONE)
3160       && (se->data.server.auth_file == NULL))
3161   {
3162     ERROR ("network plugin: A security level higher than `none' was "
3163         "requested, but no AuthFile option was given. Cowardly refusing to "
3164         "open this socket!");
3165     sockent_destroy (se);
3166     return (-1);
3167   }
3168 #endif /* HAVE_LIBGCRYPT */
3169
3170   status = sockent_init_crypto (se);
3171   if (status != 0)
3172   {
3173     ERROR ("network plugin: network_config_add_listen: sockent_init_crypto() failed.");
3174     sockent_destroy (se);
3175     return (-1);
3176   }
3177
3178   status = sockent_server_listen (se);
3179   if (status != 0)
3180   {
3181     ERROR ("network plugin: network_config_add_server: sockent_server_listen failed.");
3182     sockent_destroy (se);
3183     return (-1);
3184   }
3185
3186   status = sockent_add (se);
3187   if (status != 0)
3188   {
3189     ERROR ("network plugin: network_config_add_listen: sockent_add failed.");
3190     sockent_destroy (se);
3191     return (-1);
3192   }
3193
3194   return (0);
3195 } /* }}} int network_config_add_listen */
3196
3197 static int network_config_add_server (const oconfig_item_t *ci) /* {{{ */
3198 {
3199   sockent_t *se;
3200   int status;
3201   int i;
3202
3203   if ((ci->values_num < 1) || (ci->values_num > 2)
3204       || (ci->values[0].type != OCONFIG_TYPE_STRING)
3205       || ((ci->values_num > 1) && (ci->values[1].type != OCONFIG_TYPE_STRING)))
3206   {
3207     ERROR ("network plugin: The `%s' config option needs "
3208         "one or two string arguments.", ci->key);
3209     return (-1);
3210   }
3211
3212   se = sockent_create (SOCKENT_TYPE_CLIENT);
3213   if (se == NULL)
3214   {
3215     ERROR ("network plugin: sockent_create failed.");
3216     return (-1);
3217   }
3218
3219   se->node = strdup (ci->values[0].value.string);
3220   if (ci->values_num >= 2)
3221     se->service = strdup (ci->values[1].value.string);
3222
3223   for (i = 0; i < ci->children_num; i++)
3224   {
3225     oconfig_item_t *child = ci->children + i;
3226
3227 #if HAVE_LIBGCRYPT
3228     if (strcasecmp ("Username", child->key) == 0)
3229       network_config_set_string (child, &se->data.client.username);
3230     else if (strcasecmp ("Password", child->key) == 0)
3231       network_config_set_string (child, &se->data.client.password);
3232     else if (strcasecmp ("SecurityLevel", child->key) == 0)
3233       network_config_set_security_level (child,
3234           &se->data.client.security_level);
3235     else
3236 #endif /* HAVE_LIBGCRYPT */
3237     if (strcasecmp ("Interface", child->key) == 0)
3238       network_config_set_interface (child,
3239           &se->interface);
3240                 else if (strcasecmp ("ResolveInterval", child->key) == 0)
3241                         cf_util_get_cdtime(child, &se->data.client.resolve_interval);
3242     else
3243     {
3244       WARNING ("network plugin: Option `%s' is not allowed here.",
3245           child->key);
3246     }
3247   }
3248
3249 #if HAVE_LIBGCRYPT
3250   if ((se->data.client.security_level > SECURITY_LEVEL_NONE)
3251       && ((se->data.client.username == NULL)
3252         || (se->data.client.password == NULL)))
3253   {
3254     ERROR ("network plugin: A security level higher than `none' was "
3255         "requested, but no Username or Password option was given. "
3256         "Cowardly refusing to open this socket!");
3257     sockent_destroy (se);
3258     return (-1);
3259   }
3260 #endif /* HAVE_LIBGCRYPT */
3261
3262   status = sockent_init_crypto (se);
3263   if (status != 0)
3264   {
3265     ERROR ("network plugin: network_config_add_server: sockent_init_crypto() failed.");
3266     sockent_destroy (se);
3267     return (-1);
3268   }
3269
3270   /* No call to sockent_client_connect() here -- it is called from
3271    * networt_send_buffer_plain(). */
3272
3273   status = sockent_add (se);
3274   if (status != 0)
3275   {
3276     ERROR ("network plugin: network_config_add_server: sockent_add failed.");
3277     sockent_destroy (se);
3278     return (-1);
3279   }
3280
3281   return (0);
3282 } /* }}} int network_config_add_server */
3283
3284 static int network_config (oconfig_item_t *ci) /* {{{ */
3285 {
3286   int i;
3287
3288   /* The options need to be applied first */
3289   for (i = 0; i < ci->children_num; i++)
3290   {
3291     oconfig_item_t *child = ci->children + i;
3292     if (strcasecmp ("TimeToLive", child->key) == 0)
3293       network_config_set_ttl (child);
3294   }
3295
3296   for (i = 0; i < ci->children_num; i++)
3297   {
3298     oconfig_item_t *child = ci->children + i;
3299
3300     if (strcasecmp ("Listen", child->key) == 0)
3301       network_config_add_listen (child);
3302     else if (strcasecmp ("Server", child->key) == 0)
3303       network_config_add_server (child);
3304     else if (strcasecmp ("TimeToLive", child->key) == 0) {
3305       /* Handled earlier */
3306     }
3307     else if (strcasecmp ("MaxPacketSize", child->key) == 0)
3308       network_config_set_buffer_size (child);
3309     else if (strcasecmp ("Forward", child->key) == 0)
3310       network_config_set_boolean (child, &network_config_forward);
3311     else if (strcasecmp ("ReportStats", child->key) == 0)
3312       network_config_set_boolean (child, &network_config_stats);
3313     else
3314     {
3315       WARNING ("network plugin: Option `%s' is not allowed here.",
3316           child->key);
3317     }
3318   }
3319
3320   return (0);
3321 } /* }}} int network_config */
3322
3323 static int network_notification (const notification_t *n,
3324     user_data_t __attribute__((unused)) *user_data)
3325 {
3326   char  buffer[network_config_packet_size];
3327   char *buffer_ptr = buffer;
3328   int   buffer_free = sizeof (buffer);
3329   int   status;
3330
3331   if (!check_send_notify_okay (n))
3332     return (0);
3333
3334   memset (buffer, 0, sizeof (buffer));
3335
3336   status = write_part_number (&buffer_ptr, &buffer_free, TYPE_TIME_HR,
3337       (uint64_t) n->time);
3338   if (status != 0)
3339     return (-1);
3340
3341   status = write_part_number (&buffer_ptr, &buffer_free, TYPE_SEVERITY,
3342       (uint64_t) n->severity);
3343   if (status != 0)
3344     return (-1);
3345
3346   if (strlen (n->host) > 0)
3347   {
3348     status = write_part_string (&buffer_ptr, &buffer_free, TYPE_HOST,
3349         n->host, strlen (n->host));
3350     if (status != 0)
3351       return (-1);
3352   }
3353
3354   if (strlen (n->plugin) > 0)
3355   {
3356     status = write_part_string (&buffer_ptr, &buffer_free, TYPE_PLUGIN,
3357         n->plugin, strlen (n->plugin));
3358     if (status != 0)
3359       return (-1);
3360   }
3361
3362   if (strlen (n->plugin_instance) > 0)
3363   {
3364     status = write_part_string (&buffer_ptr, &buffer_free,
3365         TYPE_PLUGIN_INSTANCE,
3366         n->plugin_instance, strlen (n->plugin_instance));
3367     if (status != 0)
3368       return (-1);
3369   }
3370
3371   if (strlen (n->type) > 0)
3372   {
3373     status = write_part_string (&buffer_ptr, &buffer_free, TYPE_TYPE,
3374         n->type, strlen (n->type));
3375     if (status != 0)
3376       return (-1);
3377   }
3378
3379   if (strlen (n->type_instance) > 0)
3380   {
3381     status = write_part_string (&buffer_ptr, &buffer_free, TYPE_TYPE_INSTANCE,
3382         n->type_instance, strlen (n->type_instance));
3383     if (status != 0)
3384       return (-1);
3385   }
3386
3387   status = write_part_string (&buffer_ptr, &buffer_free, TYPE_MESSAGE,
3388       n->message, strlen (n->message));
3389   if (status != 0)
3390     return (-1);
3391
3392   network_send_buffer (buffer, sizeof (buffer) - buffer_free);
3393
3394   return (0);
3395 } /* int network_notification */
3396
3397 static int network_shutdown (void)
3398 {
3399         sockent_t *se;
3400
3401         listen_loop++;
3402
3403         /* Kill the listening thread */
3404         if (receive_thread_running != 0)
3405         {
3406                 INFO ("network plugin: Stopping receive thread.");
3407                 pthread_kill (receive_thread_id, SIGTERM);
3408                 pthread_join (receive_thread_id, NULL /* no return value */);
3409                 memset (&receive_thread_id, 0, sizeof (receive_thread_id));
3410                 receive_thread_running = 0;
3411         }
3412
3413         /* Shutdown the dispatching thread */
3414         if (dispatch_thread_running != 0)
3415         {
3416                 INFO ("network plugin: Stopping dispatch thread.");
3417                 pthread_mutex_lock (&receive_list_lock);
3418                 pthread_cond_broadcast (&receive_list_cond);
3419                 pthread_mutex_unlock (&receive_list_lock);
3420                 pthread_join (dispatch_thread_id, /* ret = */ NULL);
3421                 dispatch_thread_running = 0;
3422         }
3423
3424         sockent_destroy (listen_sockets);
3425
3426         if (send_buffer_fill > 0)
3427                 flush_buffer ();
3428
3429         sfree (send_buffer);
3430
3431         for (se = sending_sockets; se != NULL; se = se->next)
3432                 sockent_client_disconnect (se);
3433         sockent_destroy (sending_sockets);
3434
3435         plugin_unregister_config ("network");
3436         plugin_unregister_init ("network");
3437         plugin_unregister_write ("network");
3438         plugin_unregister_shutdown ("network");
3439
3440         return (0);
3441 } /* int network_shutdown */
3442
3443 static int network_stats_read (void) /* {{{ */
3444 {
3445         derive_t copy_octets_rx;
3446         derive_t copy_octets_tx;
3447         derive_t copy_packets_rx;
3448         derive_t copy_packets_tx;
3449         derive_t copy_values_dispatched;
3450         derive_t copy_values_not_dispatched;
3451         derive_t copy_values_sent;
3452         derive_t copy_values_not_sent;
3453         derive_t copy_receive_list_length;
3454         value_list_t vl = VALUE_LIST_INIT;
3455         value_t values[2];
3456
3457         copy_octets_rx = stats_octets_rx;
3458         copy_octets_tx = stats_octets_tx;
3459         copy_packets_rx = stats_packets_rx;
3460         copy_packets_tx = stats_packets_tx;
3461         copy_values_dispatched = stats_values_dispatched;
3462         copy_values_not_dispatched = stats_values_not_dispatched;
3463         copy_values_sent = stats_values_sent;
3464         copy_values_not_sent = stats_values_not_sent;
3465         copy_receive_list_length = receive_list_length;
3466
3467         /* Initialize `vl' */
3468         vl.values = values;
3469         vl.values_len = 2;
3470         vl.time = 0;
3471         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
3472         sstrncpy (vl.plugin, "network", sizeof (vl.plugin));
3473
3474         /* Octets received / sent */
3475         vl.values[0].derive = (derive_t) copy_octets_rx;
3476         vl.values[1].derive = (derive_t) copy_octets_tx;
3477         sstrncpy (vl.type, "if_octets", sizeof (vl.type));
3478         plugin_dispatch_values (&vl);
3479
3480         /* Packets received / send */
3481         vl.values[0].derive = (derive_t) copy_packets_rx;
3482         vl.values[1].derive = (derive_t) copy_packets_tx;
3483         sstrncpy (vl.type, "if_packets", sizeof (vl.type));
3484         plugin_dispatch_values (&vl);
3485
3486         /* Values (not) dispatched and (not) send */
3487         sstrncpy (vl.type, "total_values", sizeof (vl.type));
3488         vl.values_len = 1;
3489
3490         vl.values[0].derive = (derive_t) copy_values_dispatched;
3491         sstrncpy (vl.type_instance, "dispatch-accepted",
3492                         sizeof (vl.type_instance));
3493         plugin_dispatch_values (&vl);
3494
3495         vl.values[0].derive = (derive_t) copy_values_not_dispatched;
3496         sstrncpy (vl.type_instance, "dispatch-rejected",
3497                         sizeof (vl.type_instance));
3498         plugin_dispatch_values (&vl);
3499
3500         vl.values[0].derive = (derive_t) copy_values_sent;
3501         sstrncpy (vl.type_instance, "send-accepted",
3502                         sizeof (vl.type_instance));
3503         plugin_dispatch_values (&vl);
3504
3505         vl.values[0].derive = (derive_t) copy_values_not_sent;
3506         sstrncpy (vl.type_instance, "send-rejected",
3507                         sizeof (vl.type_instance));
3508         plugin_dispatch_values (&vl);
3509
3510         /* Receive queue length */
3511         vl.values[0].gauge = (gauge_t) copy_receive_list_length;
3512         sstrncpy (vl.type, "queue_length", sizeof (vl.type));
3513         vl.type_instance[0] = 0;
3514         plugin_dispatch_values (&vl);
3515
3516         return (0);
3517 } /* }}} int network_stats_read */
3518
3519 static int network_init (void)
3520 {
3521         static _Bool have_init = 0;
3522
3523         /* Check if we were already initialized. If so, just return - there's
3524          * nothing more to do (for now, that is). */
3525         if (have_init)
3526                 return (0);
3527         have_init = 1;
3528
3529 #if HAVE_LIBGCRYPT
3530         network_init_gcrypt ();
3531 #endif
3532
3533         if (network_config_stats != 0)
3534                 plugin_register_read ("network", network_stats_read);
3535
3536         plugin_register_shutdown ("network", network_shutdown);
3537
3538         send_buffer = malloc (network_config_packet_size);
3539         if (send_buffer == NULL)
3540         {
3541                 ERROR ("network plugin: malloc failed.");
3542                 return (-1);
3543         }
3544         network_init_buffer ();
3545
3546         /* setup socket(s) and so on */
3547         if (sending_sockets != NULL)
3548         {
3549                 plugin_register_write ("network", network_write,
3550                                 /* user_data = */ NULL);
3551                 plugin_register_notification ("network", network_notification,
3552                                 /* user_data = */ NULL);
3553         }
3554
3555         /* If no threads need to be started, return here. */
3556         if ((listen_sockets_num == 0)
3557                         || ((dispatch_thread_running != 0)
3558                                 && (receive_thread_running != 0)))
3559                 return (0);
3560
3561         if (dispatch_thread_running == 0)
3562         {
3563                 int status;
3564                 status = plugin_thread_create (&dispatch_thread_id,
3565                                 NULL /* no attributes */,
3566                                 dispatch_thread,
3567                                 NULL /* no argument */);
3568                 if (status != 0)
3569                 {
3570                         char errbuf[1024];
3571                         ERROR ("network: pthread_create failed: %s",
3572                                         sstrerror (errno, errbuf,
3573                                                 sizeof (errbuf)));
3574                 }
3575                 else
3576                 {
3577                         dispatch_thread_running = 1;
3578                 }
3579         }
3580
3581         if (receive_thread_running == 0)
3582         {
3583                 int status;
3584                 status = plugin_thread_create (&receive_thread_id,
3585                                 NULL /* no attributes */,
3586                                 receive_thread,
3587                                 NULL /* no argument */);
3588                 if (status != 0)
3589                 {
3590                         char errbuf[1024];
3591                         ERROR ("network: pthread_create failed: %s",
3592                                         sstrerror (errno, errbuf,
3593                                                 sizeof (errbuf)));
3594                 }
3595                 else
3596                 {
3597                         receive_thread_running = 1;
3598                 }
3599         }
3600
3601         return (0);
3602 } /* int network_init */
3603
3604 /*
3605  * The flush option of the network plugin cannot flush individual identifiers.
3606  * All the values are added to a buffer and sent when the buffer is full, the
3607  * requested value may or may not be in there, it's not worth finding out. We
3608  * just send the buffer if `flush'  is called - if the requested value was in
3609  * there, good. If not, well, then there is nothing to flush.. -octo
3610  */
3611 static int network_flush (cdtime_t timeout,
3612                 __attribute__((unused)) const char *identifier,
3613                 __attribute__((unused)) user_data_t *user_data)
3614 {
3615         pthread_mutex_lock (&send_buffer_lock);
3616
3617         if (send_buffer_fill > 0)
3618         {
3619                 if (timeout > 0)
3620                 {
3621                         cdtime_t now = cdtime ();
3622                         if ((send_buffer_last_update + timeout) > now)
3623                         {
3624                                 pthread_mutex_unlock (&send_buffer_lock);
3625                                 return (0);
3626                         }
3627                 }
3628                 flush_buffer ();
3629         }
3630         pthread_mutex_unlock (&send_buffer_lock);
3631
3632         return (0);
3633 } /* int network_flush */
3634
3635 void module_register (void)
3636 {
3637         plugin_register_complex_config ("network", network_config);
3638         plugin_register_init   ("network", network_init);
3639         plugin_register_flush   ("network", network_flush,
3640                         /* user_data = */ NULL);
3641 } /* void module_register */
3642
3643 /* vim: set fdm=marker : */