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