src/daemon/common.c

   1 /**
   2  * collectd - src/common.c
   3  * Copyright (C) 2005-2014  Florian octo Forster
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9  * and/or sell copies of the Software, and to permit persons to whom the
  10  * Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in
  13  * all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *   Florian octo Forster <octo at collectd.org>
  25  *   Niki W. Waibel <niki.waibel@gmx.net>
  26  *   Sebastian Harl <sh at tokkee.org>
  27  *   Michał Mirosław <mirq-linux at rere.qmqm.pl>
  28 **/
  29
  30 #if HAVE_CONFIG_H
  31 # include "config.h"
  32 #endif
  33
  34 #include "collectd.h"
  35
  36 #include "common.h"
  37 #include "plugin.h"
  38 #include "utils_cache.h"
  39
  40 #ifdef HAVE_MATH_H
  41 # include <math.h>
  42 #endif
  43
  44 /* for getaddrinfo */
  45 #include <sys/types.h>
  46 #include <netdb.h>
  47
  48 #include <poll.h>
  49
  50 #if HAVE_NETINET_IN_H
  51 # include <netinet/in.h>
  52 #endif
  53
  54 #if HAVE_NETINET_TCP_H
  55 # include <netinet/tcp.h>
  56 #endif
  57
  58 /* for ntohl and htonl */
  59 #if HAVE_ARPA_INET_H
  60 # include <arpa/inet.h>
  61 #endif
  62
  63 #ifdef HAVE_LIBKSTAT
  64 extern kstat_ctl_t *kc;
  65 #endif
  66
  67 #if !HAVE_GETPWNAM_R
  68 static pthread_mutex_t getpwnam_r_lock = PTHREAD_MUTEX_INITIALIZER;
  69 #endif
  70
  71 #if !HAVE_STRERROR_R
  72 static pthread_mutex_t strerror_r_lock = PTHREAD_MUTEX_INITIALIZER;
  73 #endif
  74
  75 char *sstrncpy (char *dest, const char *src, size_t n)
  76 {
  77         strncpy (dest, src, n);
  78         dest[n - 1] = '\0';
  79
  80         return (dest);
  81 } /* char *sstrncpy */
  82
  83 int ssnprintf (char *dest, size_t n, const char *format, ...)
  84 {
  85         int ret = 0;
  86         va_list ap;
  87
  88         va_start (ap, format);
  89         ret = vsnprintf (dest, n, format, ap);
  90         dest[n - 1] = '\0';
  91         va_end (ap);
  92
  93         return (ret);
  94 } /* int ssnprintf */
  95
  96 char *ssnprintf_alloc (char const *format, ...) /* {{{ */
  97 {
  98         char static_buffer[1024] = "";
  99         char *alloc_buffer;
 100         size_t alloc_buffer_size;
 101         int status;
 102         va_list ap;
 103
 104         /* Try printing into the static buffer. In many cases it will be
 105          * sufficiently large and we can simply return a strdup() of this
 106          * buffer. */
 107         va_start (ap, format);
 108         status = vsnprintf (static_buffer, sizeof (static_buffer), format, ap);
 109         va_end (ap);
 110         if (status < 0)
 111                 return (NULL);
 112
 113         /* "status" does not include the null byte. */
 114         alloc_buffer_size = (size_t) (status + 1);
 115         if (alloc_buffer_size <= sizeof (static_buffer))
 116                 return (strdup (static_buffer));
 117
 118         /* Allocate a buffer large enough to hold the string. */
 119         alloc_buffer = calloc (1, alloc_buffer_size);
 120         if (alloc_buffer == NULL)
 121                 return (NULL);
 122
 123         /* Print again into this new buffer. */
 124         va_start (ap, format);
 125         status = vsnprintf (alloc_buffer, alloc_buffer_size, format, ap);
 126         va_end (ap);
 127         if (status < 0)
 128         {
 129                 sfree (alloc_buffer);
 130                 return (NULL);
 131         }
 132
 133         return (alloc_buffer);
 134 } /* }}} char *ssnprintf_alloc */
 135
 136 char *sstrdup (const char *s)
 137 {
 138         char *r;
 139         size_t sz;
 140
 141         if (s == NULL)
 142                 return (NULL);
 143
 144         /* Do not use `strdup' here, because it's not specified in POSIX. It's
 145          * ``only'' an XSI extension. */
 146         sz = strlen (s) + 1;
 147         r = malloc (sz);
 148         if (r == NULL)
 149         {
 150                 ERROR ("sstrdup: Out of memory.");
 151                 exit (3);
 152         }
 153         memcpy (r, s, sizeof (char) * sz);
 154
 155         return (r);
 156 } /* char *sstrdup */
 157
 158 /* Even though Posix requires "strerror_r" to return an "int",
 159  * some systems (e.g. the GNU libc) return a "char *" _and_
 160  * ignore the second argument ... -tokkee */
 161 char *sstrerror (int errnum, char *buf, size_t buflen)
 162 {
 163         buf[0] = '\0';
 164
 165 #if !HAVE_STRERROR_R
 166         {
 167                 char *temp;
 168
 169                 pthread_mutex_lock (&strerror_r_lock);
 170
 171                 temp = strerror (errnum);
 172                 sstrncpy (buf, temp, buflen);
 173
 174                 pthread_mutex_unlock (&strerror_r_lock);
 175         }
 176 /* #endif !HAVE_STRERROR_R */
 177
 178 #elif STRERROR_R_CHAR_P
 179         {
 180                 char *temp;
 181                 temp = strerror_r (errnum, buf, buflen);
 182                 if (buf[0] == '\0')
 183                 {
 184                         if ((temp != NULL) && (temp != buf) && (temp[0] != '\0'))
 185                                 sstrncpy (buf, temp, buflen);
 186                         else
 187                                 sstrncpy (buf, "strerror_r did not return "
 188                                                 "an error message", buflen);
 189                 }
 190         }
 191 /* #endif STRERROR_R_CHAR_P */
 192
 193 #else
 194         if (strerror_r (errnum, buf, buflen) != 0)
 195         {
 196                 ssnprintf (buf, buflen, "Error #%i; "
 197                                 "Additionally, strerror_r failed.",
 198                                 errnum);
 199         }
 200 #endif /* STRERROR_R_CHAR_P */
 201
 202         return (buf);
 203 } /* char *sstrerror */
 204
 205 void *smalloc (size_t size)
 206 {
 207         void *r;
 208
 209         if ((r = malloc (size)) == NULL)
 210         {
 211                 ERROR ("Not enough memory.");
 212                 exit (3);
 213         }
 214
 215         return (r);
 216 } /* void *smalloc */
 217
 218 #if 0
 219 void sfree (void **ptr)
 220 {
 221         if (ptr == NULL)
 222                 return;
 223
 224         if (*ptr != NULL)
 225                 free (*ptr);
 226
 227         *ptr = NULL;
 228 }
 229 #endif
 230
 231 ssize_t sread (int fd, void *buf, size_t count)
 232 {
 233         char    *ptr;
 234         size_t   nleft;
 235         ssize_t  status;
 236
 237         ptr   = (char *) buf;
 238         nleft = count;
 239
 240         while (nleft > 0)
 241         {
 242                 status = read (fd, (void *) ptr, nleft);
 243
 244                 if ((status < 0) && ((errno == EAGAIN) || (errno == EINTR)))
 245                         continue;
 246
 247                 if (status < 0)
 248                         return (status);
 249
 250                 if (status == 0)
 251                 {
 252                         DEBUG ("Received EOF from fd %i. "
 253                                         "Closing fd and returning error.",
 254                                         fd);
 255                         close (fd);
 256                         return (-1);
 257                 }
 258
 259                 assert ((0 > status) || (nleft >= (size_t)status));
 260
 261                 nleft = nleft - ((size_t) status);
 262                 ptr   = ptr   + ((size_t) status);
 263         }
 264
 265         return (0);
 266 }
 267
 268
 269 ssize_t swrite (int fd, const void *buf, size_t count)
 270 {
 271         const char *ptr;
 272         size_t      nleft;
 273         ssize_t     status;
 274         struct      pollfd pfd;
 275
 276         ptr   = (const char *) buf;
 277         nleft = count;
 278
 279         if (fd < 0)
 280                 return (-1);
 281
 282         /* checking for closed peer connection */
 283         pfd.fd = fd;
 284         pfd.events = POLLIN | POLLHUP;
 285         pfd.revents = 0;
 286         if (poll(&pfd, 1, 0) > 0) {
 287                 char buffer[32];
 288                 if (recv(fd, buffer, sizeof(buffer), MSG_PEEK | MSG_DONTWAIT) == 0) {
 289                         // if recv returns zero (even though poll() said there is data to be read),
 290                         // that means the connection has been closed
 291                         return -1;
 292                 }
 293         }
 294
 295         while (nleft > 0)
 296         {
 297                 status = write (fd, (const void *) ptr, nleft);
 298
 299                 if ((status < 0) && ((errno == EAGAIN) || (errno == EINTR)))
 300                         continue;
 301
 302                 if (status < 0)
 303                         return (status);
 304
 305                 nleft = nleft - ((size_t) status);
 306                 ptr   = ptr   + ((size_t) status);
 307         }
 308
 309         return (0);
 310 }
 311
 312 int strsplit (char *string, char **fields, size_t size)
 313 {
 314         size_t i;
 315         char *ptr;
 316         char *saveptr;
 317
 318         i = 0;
 319         ptr = string;
 320         saveptr = NULL;
 321         while ((fields[i] = strtok_r (ptr, " \t\r\n", &saveptr)) != NULL)
 322         {
 323                 ptr = NULL;
 324                 i++;
 325
 326                 if (i >= size)
 327                         break;
 328         }
 329
 330         return ((int) i);
 331 }
 332
 333 int strjoin (char *buffer, size_t buffer_size,
 334                 char **fields, size_t fields_num,
 335                 const char *sep)
 336 {
 337         size_t avail;
 338         char *ptr;
 339         size_t sep_len;
 340
 341         if ((buffer_size < 1) || (fields_num == 0))
 342                 return (-1);
 343
 344         memset (buffer, 0, buffer_size);
 345         ptr = buffer;
 346         avail = buffer_size - 1;
 347
 348         sep_len = 0;
 349         if (sep != NULL)
 350                 sep_len = strlen (sep);
 351
 352         for (size_t i = 0; i < fields_num; i++)
 353         {
 354                 size_t field_len;
 355
 356                 if ((i > 0) && (sep_len > 0))
 357                 {
 358                         if (avail < sep_len)
 359                                 return (-1);
 360
 361                         memcpy (ptr, sep, sep_len);
 362                         ptr += sep_len;
 363                         avail -= sep_len;
 364                 }
 365
 366                 field_len = strlen (fields[i]);
 367                 if (avail < field_len)
 368                         return (-1);
 369
 370                 memcpy (ptr, fields[i], field_len);
 371                 ptr += field_len;
 372                 avail -= field_len;
 373         }
 374
 375         assert (buffer[buffer_size - 1] == 0);
 376         return ((int) strlen (buffer));
 377 }
 378
 379 int escape_string (char *buffer, size_t buffer_size)
 380 {
 381   char *temp;
 382   size_t j;
 383
 384   /* Check if we need to escape at all first */
 385   temp = strpbrk (buffer, " \t\"\\");
 386   if (temp == NULL)
 387     return (0);
 388
 389   if (buffer_size < 3)
 390     return (EINVAL);
 391
 392   temp = calloc (1, buffer_size);
 393   if (temp == NULL)
 394     return (ENOMEM);
 395
 396   temp[0] = '"';
 397   j = 1;
 398
 399   for (size_t i = 0; i < buffer_size; i++)
 400   {
 401     if (buffer[i] == 0)
 402     {
 403       break;
 404     }
 405     else if ((buffer[i] == '"') || (buffer[i] == '\\'))
 406     {
 407       if (j > (buffer_size - 4))
 408         break;
 409       temp[j] = '\\';
 410       temp[j + 1] = buffer[i];
 411       j += 2;
 412     }
 413     else
 414     {
 415       if (j > (buffer_size - 3))
 416         break;
 417       temp[j] = buffer[i];
 418       j++;
 419     }
 420   }
 421
 422   assert ((j + 1) < buffer_size);
 423   temp[j] = '"';
 424   temp[j + 1] = 0;
 425
 426   sstrncpy (buffer, temp, buffer_size);
 427   sfree (temp);
 428   return (0);
 429 } /* int escape_string */
 430
 431 int strunescape (char *buf, size_t buf_len)
 432 {
 433         for (size_t i = 0; (i < buf_len) && (buf[i] != '\0'); ++i)
 434         {
 435                 if (buf[i] != '\\')
 436                         continue;
 437
 438                 if (((i + 1) >= buf_len) || (buf[i + 1] == 0)) {
 439                         ERROR ("string unescape: backslash found at end of string.");
 440                         /* Ensure null-byte at the end of the buffer. */
 441                         buf[i] = 0;
 442                         return (-1);
 443                 }
 444
 445                 switch (buf[i + 1]) {
 446                         case 't':
 447                                 buf[i] = '\t';
 448                                 break;
 449                         case 'n':
 450                                 buf[i] = '\n';
 451                                 break;
 452                         case 'r':
 453                                 buf[i] = '\r';
 454                                 break;
 455                         default:
 456                                 buf[i] = buf[i + 1];
 457                                 break;
 458                 }
 459
 460                 /* Move everything after the position one position to the left.
 461                  * Add a null-byte as last character in the buffer. */
 462                 memmove (buf + i + 1, buf + i + 2, buf_len - i - 2);
 463                 buf[buf_len - 1] = 0;
 464         }
 465         return (0);
 466 } /* int strunescape */
 467
 468 size_t strstripnewline (char *buffer)
 469 {
 470         size_t buffer_len = strlen (buffer);
 471
 472         while (buffer_len > 0)
 473         {
 474                 if ((buffer[buffer_len - 1] != '\n')
 475                                 && (buffer[buffer_len - 1] != '\r'))
 476                         break;
 477                 buffer_len--;
 478                 buffer[buffer_len] = 0;
 479         }
 480
 481         return (buffer_len);
 482 } /* size_t strstripnewline */
 483
 484 int escape_slashes (char *buffer, size_t buffer_size)
 485 {
 486         size_t buffer_len;
 487
 488         buffer_len = strlen (buffer);
 489
 490         if (buffer_len <= 1)
 491         {
 492                 if (strcmp ("/", buffer) == 0)
 493                 {
 494                         if (buffer_size < 5)
 495                                 return (-1);
 496                         sstrncpy (buffer, "root", buffer_size);
 497                 }
 498                 return (0);
 499         }
 500
 501         /* Move one to the left */
 502         if (buffer[0] == '/')
 503         {
 504                 memmove (buffer, buffer + 1, buffer_len);
 505                 buffer_len--;
 506         }
 507
 508         for (size_t i = 0; i < buffer_len; i++)
 509         {
 510                 if (buffer[i] == '/')
 511                         buffer[i] = '_';
 512         }
 513
 514         return (0);
 515 } /* int escape_slashes */
 516
 517 void replace_special (char *buffer, size_t buffer_size)
 518 {
 519         for (size_t i = 0; i < buffer_size; i++)
 520         {
 521                 if (buffer[i] == 0)
 522                         return;
 523                 if ((!isalnum ((int) buffer[i])) && (buffer[i] != '-'))
 524                         buffer[i] = '_';
 525         }
 526 } /* void replace_special */
 527
 528 int timeval_cmp (struct timeval tv0, struct timeval tv1, struct timeval *delta)
 529 {
 530         struct timeval *larger;
 531         struct timeval *smaller;
 532
 533         int status;
 534
 535         NORMALIZE_TIMEVAL (tv0);
 536         NORMALIZE_TIMEVAL (tv1);
 537
 538         if ((tv0.tv_sec == tv1.tv_sec) && (tv0.tv_usec == tv1.tv_usec))
 539         {
 540                 if (delta != NULL) {
 541                         delta->tv_sec  = 0;
 542                         delta->tv_usec = 0;
 543                 }
 544                 return (0);
 545         }
 546
 547         if ((tv0.tv_sec < tv1.tv_sec)
 548                         || ((tv0.tv_sec == tv1.tv_sec) && (tv0.tv_usec < tv1.tv_usec)))
 549         {
 550                 larger  = &tv1;
 551                 smaller = &tv0;
 552                 status  = -1;
 553         }
 554         else
 555         {
 556                 larger  = &tv0;
 557                 smaller = &tv1;
 558                 status  = 1;
 559         }
 560
 561         if (delta != NULL) {
 562                 delta->tv_sec = larger->tv_sec - smaller->tv_sec;
 563
 564                 if (smaller->tv_usec <= larger->tv_usec)
 565                         delta->tv_usec = larger->tv_usec - smaller->tv_usec;
 566                 else
 567                 {
 568                         --delta->tv_sec;
 569                         delta->tv_usec = 1000000 + larger->tv_usec - smaller->tv_usec;
 570                 }
 571         }
 572
 573         assert ((delta == NULL)
 574                         || ((0 <= delta->tv_usec) && (delta->tv_usec < 1000000)));
 575
 576         return (status);
 577 } /* int timeval_cmp */
 578
 579 int check_create_dir (const char *file_orig)
 580 {
 581         struct stat statbuf;
 582
 583         char  file_copy[512];
 584         char  dir[512];
 585         int   dir_len = 512;
 586         char *fields[16];
 587         int   fields_num;
 588         char *ptr;
 589         char *saveptr;
 590         int   last_is_file = 1;
 591         int   path_is_absolute = 0;
 592         size_t len;
 593
 594         /*
 595          * Sanity checks first
 596          */
 597         if (file_orig == NULL)
 598                 return (-1);
 599
 600         if ((len = strlen (file_orig)) < 1)
 601                 return (-1);
 602         else if (len >= sizeof (file_copy))
 603                 return (-1);
 604
 605         /*
 606          * If `file_orig' ends in a slash the last component is a directory,
 607          * otherwise it's a file. Act accordingly..
 608          */
 609         if (file_orig[len - 1] == '/')
 610                 last_is_file = 0;
 611         if (file_orig[0] == '/')
 612                 path_is_absolute = 1;
 613
 614         /*
 615          * Create a copy for `strtok_r' to destroy
 616          */
 617         sstrncpy (file_copy, file_orig, sizeof (file_copy));
 618
 619         /*
 620          * Break into components. This will eat up several slashes in a row and
 621          * remove leading and trailing slashes..
 622          */
 623         ptr = file_copy;
 624         saveptr = NULL;
 625         fields_num = 0;
 626         while ((fields[fields_num] = strtok_r (ptr, "/", &saveptr)) != NULL)
 627         {
 628                 ptr = NULL;
 629                 fields_num++;
 630
 631                 if (fields_num >= 16)
 632                         break;
 633         }
 634
 635         /*
 636          * For each component, do..
 637          */
 638         for (int i = 0; i < (fields_num - last_is_file); i++)
 639         {
 640                 /*
 641                  * Do not create directories that start with a dot. This
 642                  * prevents `../../' attacks and other likely malicious
 643                  * behavior.
 644                  */
 645                 if (fields[i][0] == '.')
 646                 {
 647                         ERROR ("Cowardly refusing to create a directory that "
 648                                         "begins with a `.' (dot): `%s'", file_orig);
 649                         return (-2);
 650                 }
 651
 652                 /*
 653                  * Join the components together again
 654                  */
 655                 dir[0] = '/';
 656                 if (strjoin (dir + path_is_absolute, (size_t) (dir_len - path_is_absolute),
 657                                         fields, (size_t) (i + 1), "/") < 0)
 658                 {
 659                         ERROR ("strjoin failed: `%s', component #%i", file_orig, i);
 660                         return (-1);
 661                 }
 662
 663                 while (42) {
 664                         if ((stat (dir, &statbuf) == -1)
 665                                         && (lstat (dir, &statbuf) == -1))
 666                         {
 667                                 if (errno == ENOENT)
 668                                 {
 669                                         if (mkdir (dir, S_IRWXU | S_IRWXG | S_IRWXO) == 0)
 670                                                 break;
 671
 672                                         /* this might happen, if a different thread created
 673                                          * the directory in the meantime
 674                                          * => call stat() again to check for S_ISDIR() */
 675                                         if (EEXIST == errno)
 676                                                 continue;
 677
 678                                         char errbuf[1024];
 679                                         ERROR ("check_create_dir: mkdir (%s): %s", dir,
 680                                                         sstrerror (errno,
 681                                                                 errbuf, sizeof (errbuf)));
 682                                         return (-1);
 683                                 }
 684                                 else
 685                                 {
 686                                         char errbuf[1024];
 687                                         ERROR ("check_create_dir: stat (%s): %s", dir,
 688                                                         sstrerror (errno, errbuf,
 689                                                                 sizeof (errbuf)));
 690                                         return (-1);
 691                                 }
 692                         }
 693                         else if (!S_ISDIR (statbuf.st_mode))
 694                         {
 695                                 ERROR ("check_create_dir: `%s' exists but is not "
 696                                                 "a directory!", dir);
 697                                 return (-1);
 698                         }
 699                         break;
 700                 }
 701         }
 702
 703         return (0);
 704 } /* check_create_dir */
 705
 706 #ifdef HAVE_LIBKSTAT
 707 int get_kstat (kstat_t **ksp_ptr, char *module, int instance, char *name)
 708 {
 709         char ident[128];
 710
 711         *ksp_ptr = NULL;
 712
 713         if (kc == NULL)
 714                 return (-1);
 715
 716         ssnprintf (ident, sizeof (ident), "%s,%i,%s", module, instance, name);
 717
 718         *ksp_ptr = kstat_lookup (kc, module, instance, name);
 719         if (*ksp_ptr == NULL)
 720         {
 721                 ERROR ("get_kstat: Cound not find kstat %s", ident);
 722                 return (-1);
 723         }
 724
 725         if ((*ksp_ptr)->ks_type != KSTAT_TYPE_NAMED)
 726         {
 727                 ERROR ("get_kstat: kstat %s has wrong type", ident);
 728                 *ksp_ptr = NULL;
 729                 return (-1);
 730         }
 731
 732 #ifdef assert
 733         assert (*ksp_ptr != NULL);
 734         assert ((*ksp_ptr)->ks_type == KSTAT_TYPE_NAMED);
 735 #endif
 736
 737         if (kstat_read (kc, *ksp_ptr, NULL) == -1)
 738         {
 739                 ERROR ("get_kstat: kstat %s could not be read", ident);
 740                 return (-1);
 741         }
 742
 743         if ((*ksp_ptr)->ks_type != KSTAT_TYPE_NAMED)
 744         {
 745                 ERROR ("get_kstat: kstat %s has wrong type", ident);
 746                 return (-1);
 747         }
 748
 749         return (0);
 750 }
 751
 752 long long get_kstat_value (kstat_t *ksp, char *name)
 753 {
 754         kstat_named_t *kn;
 755         long long retval = -1LL;
 756
 757         if (ksp == NULL)
 758         {
 759                 ERROR ("get_kstat_value (\"%s\"): ksp is NULL.", name);
 760                 return (-1LL);
 761         }
 762         else if (ksp->ks_type != KSTAT_TYPE_NAMED)
 763         {
 764                 ERROR ("get_kstat_value (\"%s\"): ksp->ks_type (%#x) "
 765                                 "is not KSTAT_TYPE_NAMED (%#x).",
 766                                 name,
 767                                 (unsigned int) ksp->ks_type,
 768                                 (unsigned int) KSTAT_TYPE_NAMED);
 769                 return (-1LL);
 770         }
 771
 772         if ((kn = (kstat_named_t *) kstat_data_lookup (ksp, name)) == NULL)
 773                 return (-1LL);
 774
 775         if (kn->data_type == KSTAT_DATA_INT32)
 776                 retval = (long long) kn->value.i32;
 777         else if (kn->data_type == KSTAT_DATA_UINT32)
 778                 retval = (long long) kn->value.ui32;
 779         else if (kn->data_type == KSTAT_DATA_INT64)
 780                 retval = (long long) kn->value.i64; /* According to ANSI C99 `long long' must hold at least 64 bits */
 781         else if (kn->data_type == KSTAT_DATA_UINT64)
 782                 retval = (long long) kn->value.ui64; /* XXX: Might overflow! */
 783         else
 784                 WARNING ("get_kstat_value: Not a numeric value: %s", name);
 785
 786         return (retval);
 787 }
 788 #endif /* HAVE_LIBKSTAT */
 789
 790 #ifndef HAVE_HTONLL
 791 unsigned long long ntohll (unsigned long long n)
 792 {
 793 #if BYTE_ORDER == BIG_ENDIAN
 794         return (n);
 795 #else
 796         return (((unsigned long long) ntohl (n)) << 32) + ntohl (n >> 32);
 797 #endif
 798 } /* unsigned long long ntohll */
 799
 800 unsigned long long htonll (unsigned long long n)
 801 {
 802 #if BYTE_ORDER == BIG_ENDIAN
 803         return (n);
 804 #else
 805         return (((unsigned long long) htonl (n)) << 32) + htonl (n >> 32);
 806 #endif
 807 } /* unsigned long long htonll */
 808 #endif /* HAVE_HTONLL */
 809
 810 #if FP_LAYOUT_NEED_NOTHING
 811 /* Well, we need nothing.. */
 812 /* #endif FP_LAYOUT_NEED_NOTHING */
 813
 814 #elif FP_LAYOUT_NEED_ENDIANFLIP || FP_LAYOUT_NEED_INTSWAP
 815 # if FP_LAYOUT_NEED_ENDIANFLIP
 816 #  define FP_CONVERT(A) ((((uint64_t)(A) & 0xff00000000000000LL) >> 56) | \
 817                          (((uint64_t)(A) & 0x00ff000000000000LL) >> 40) | \
 818                          (((uint64_t)(A) & 0x0000ff0000000000LL) >> 24) | \
 819                          (((uint64_t)(A) & 0x000000ff00000000LL) >> 8)  | \
 820                          (((uint64_t)(A) & 0x00000000ff000000LL) << 8)  | \
 821                          (((uint64_t)(A) & 0x0000000000ff0000LL) << 24) | \
 822                          (((uint64_t)(A) & 0x000000000000ff00LL) << 40) | \
 823                          (((uint64_t)(A) & 0x00000000000000ffLL) << 56))
 824 # else
 825 #  define FP_CONVERT(A) ((((uint64_t)(A) & 0xffffffff00000000LL) >> 32) | \
 826                          (((uint64_t)(A) & 0x00000000ffffffffLL) << 32))
 827 # endif
 828
 829 double ntohd (double d)
 830 {
 831         union
 832         {
 833                 uint8_t  byte[8];
 834                 uint64_t integer;
 835                 double   floating;
 836         } ret;
 837
 838         ret.floating = d;
 839
 840         /* NAN in x86 byte order */
 841         if ((ret.byte[0] == 0x00) && (ret.byte[1] == 0x00)
 842                         && (ret.byte[2] == 0x00) && (ret.byte[3] == 0x00)
 843                         && (ret.byte[4] == 0x00) && (ret.byte[5] == 0x00)
 844                         && (ret.byte[6] == 0xf8) && (ret.byte[7] == 0x7f))
 845         {
 846                 return (NAN);
 847         }
 848         else
 849         {
 850                 uint64_t tmp;
 851
 852                 tmp = ret.integer;
 853                 ret.integer = FP_CONVERT (tmp);
 854                 return (ret.floating);
 855         }
 856 } /* double ntohd */
 857
 858 double htond (double d)
 859 {
 860         union
 861         {
 862                 uint8_t  byte[8];
 863                 uint64_t integer;
 864                 double   floating;
 865         } ret;
 866
 867         if (isnan (d))
 868         {
 869                 ret.byte[0] = ret.byte[1] = ret.byte[2] = ret.byte[3] = 0x00;
 870                 ret.byte[4] = ret.byte[5] = 0x00;
 871                 ret.byte[6] = 0xf8;
 872                 ret.byte[7] = 0x7f;
 873                 return (ret.floating);
 874         }
 875         else
 876         {
 877                 uint64_t tmp;
 878
 879                 ret.floating = d;
 880                 tmp = FP_CONVERT (ret.integer);
 881                 ret.integer = tmp;
 882                 return (ret.floating);
 883         }
 884 } /* double htond */
 885 #endif /* FP_LAYOUT_NEED_ENDIANFLIP || FP_LAYOUT_NEED_INTSWAP */
 886
 887 int format_name (char *ret, int ret_len,
 888                 const char *hostname,
 889                 const char *plugin, const char *plugin_instance,
 890                 const char *type, const char *type_instance)
 891 {
 892   char *buffer;
 893   size_t buffer_size;
 894
 895   buffer = ret;
 896   buffer_size = (size_t) ret_len;
 897
 898 #define APPEND(str) do {                                               \
 899   size_t l = strlen (str);                                             \
 900   if (l >= buffer_size)                                                \
 901     return (ENOBUFS);                                                  \
 902   memcpy (buffer, (str), l);                                           \
 903   buffer += l; buffer_size -= l;                                       \
 904 } while (0)
 905
 906   assert (plugin != NULL);
 907   assert (type != NULL);
 908
 909   APPEND (hostname);
 910   APPEND ("/");
 911   APPEND (plugin);
 912   if ((plugin_instance != NULL) && (plugin_instance[0] != 0))
 913   {
 914     APPEND ("-");
 915     APPEND (plugin_instance);
 916   }
 917   APPEND ("/");
 918   APPEND (type);
 919   if ((type_instance != NULL) && (type_instance[0] != 0))
 920   {
 921     APPEND ("-");
 922     APPEND (type_instance);
 923   }
 924   assert (buffer_size > 0);
 925   buffer[0] = 0;
 926
 927 #undef APPEND
 928   return (0);
 929 } /* int format_name */
 930
 931 int format_values (char *ret, size_t ret_len, /* {{{ */
 932                 const data_set_t *ds, const value_list_t *vl,
 933                 _Bool store_rates)
 934 {
 935         size_t offset = 0;
 936         int status;
 937         gauge_t *rates = NULL;
 938
 939         assert (0 == strcmp (ds->type, vl->type));
 940
 941         memset (ret, 0, ret_len);
 942
 943 #define BUFFER_ADD(...) do { \
 944         status = ssnprintf (ret + offset, ret_len - offset, \
 945                         __VA_ARGS__); \
 946         if (status < 1) \
 947         { \
 948                 sfree (rates); \
 949                 return (-1); \
 950         } \
 951         else if (((size_t) status) >= (ret_len - offset)) \
 952         { \
 953                 sfree (rates); \
 954                 return (-1); \
 955         } \
 956         else \
 957                 offset += ((size_t) status); \
 958 } while (0)
 959
 960         BUFFER_ADD ("%.3f", CDTIME_T_TO_DOUBLE (vl->time));
 961
 962         for (size_t i = 0; i < ds->ds_num; i++)
 963         {
 964                 if (ds->ds[i].type == DS_TYPE_GAUGE)
 965                         BUFFER_ADD (":"GAUGE_FORMAT, vl->values[i].gauge);
 966                 else if (store_rates)
 967                 {
 968                         if (rates == NULL)
 969                                 rates = uc_get_rate (ds, vl);
 970                         if (rates == NULL)
 971                         {
 972                                 WARNING ("format_values: uc_get_rate failed.");
 973                                 return (-1);
 974                         }
 975                         BUFFER_ADD (":"GAUGE_FORMAT, rates[i]);
 976                 }
 977                 else if (ds->ds[i].type == DS_TYPE_COUNTER)
 978                         BUFFER_ADD (":%llu", vl->values[i].counter);
 979                 else if (ds->ds[i].type == DS_TYPE_DERIVE)
 980                         BUFFER_ADD (":%"PRIi64, vl->values[i].derive);
 981                 else if (ds->ds[i].type == DS_TYPE_ABSOLUTE)
 982                         BUFFER_ADD (":%"PRIu64, vl->values[i].absolute);
 983                 else
 984                 {
 985                         ERROR ("format_values: Unknown data source type: %i",
 986                                         ds->ds[i].type);
 987                         sfree (rates);
 988                         return (-1);
 989                 }
 990         } /* for ds->ds_num */
 991
 992 #undef BUFFER_ADD
 993
 994         sfree (rates);
 995         return (0);
 996 } /* }}} int format_values */
 997
 998 int parse_identifier (char *str, char **ret_host,
 999                 char **ret_plugin, char **ret_plugin_instance,
1000                 char **ret_type, char **ret_type_instance)
1001 {
1002         char *hostname = NULL;
1003         char *plugin = NULL;
1004         char *plugin_instance = NULL;
1005         char *type = NULL;
1006         char *type_instance = NULL;
1007
1008         hostname = str;
1009         if (hostname == NULL)
1010                 return (-1);
1011
1012         plugin = strchr (hostname, '/');
1013         if (plugin == NULL)
1014                 return (-1);
1015         *plugin = '\0'; plugin++;
1016
1017         type = strchr (plugin, '/');
1018         if (type == NULL)
1019                 return (-1);
1020         *type = '\0'; type++;
1021
1022         plugin_instance = strchr (plugin, '-');
1023         if (plugin_instance != NULL)
1024         {
1025                 *plugin_instance = '\0';
1026                 plugin_instance++;
1027         }
1028
1029         type_instance = strchr (type, '-');
1030         if (type_instance != NULL)
1031         {
1032                 *type_instance = '\0';
1033                 type_instance++;
1034         }
1035
1036         *ret_host = hostname;
1037         *ret_plugin = plugin;
1038         *ret_plugin_instance = plugin_instance;
1039         *ret_type = type;
1040         *ret_type_instance = type_instance;
1041         return (0);
1042 } /* int parse_identifier */
1043
1044 int parse_identifier_vl (const char *str, value_list_t *vl) /* {{{ */
1045 {
1046         char str_copy[6 * DATA_MAX_NAME_LEN];
1047         char *host = NULL;
1048         char *plugin = NULL;
1049         char *plugin_instance = NULL;
1050         char *type = NULL;
1051         char *type_instance = NULL;
1052         int status;
1053
1054         if ((str == NULL) || (vl == NULL))
1055                 return (EINVAL);
1056
1057         sstrncpy (str_copy, str, sizeof (str_copy));
1058
1059         status = parse_identifier (str_copy, &host,
1060                         &plugin, &plugin_instance,
1061                         &type, &type_instance);
1062         if (status != 0)
1063                 return (status);
1064
1065         sstrncpy (vl->host, host, sizeof (vl->host));
1066         sstrncpy (vl->plugin, plugin, sizeof (vl->plugin));
1067         sstrncpy (vl->plugin_instance,
1068                         (plugin_instance != NULL) ? plugin_instance : "",
1069                         sizeof (vl->plugin_instance));
1070         sstrncpy (vl->type, type, sizeof (vl->type));
1071         sstrncpy (vl->type_instance,
1072                         (type_instance != NULL) ? type_instance : "",
1073                         sizeof (vl->type_instance));
1074
1075         return (0);
1076 } /* }}} int parse_identifier_vl */
1077
1078 int parse_value (const char *value_orig, value_t *ret_value, int ds_type)
1079 {
1080   char *value;
1081   char *endptr = NULL;
1082   size_t value_len;
1083
1084   if (value_orig == NULL)
1085     return (EINVAL);
1086
1087   value = strdup (value_orig);
1088   if (value == NULL)
1089     return (ENOMEM);
1090   value_len = strlen (value);
1091
1092   while ((value_len > 0) && isspace ((int) value[value_len - 1]))
1093   {
1094     value[value_len - 1] = 0;
1095     value_len--;
1096   }
1097
1098   switch (ds_type)
1099   {
1100     case DS_TYPE_COUNTER:
1101       ret_value->counter = (counter_t) strtoull (value, &endptr, 0);
1102       break;
1103
1104     case DS_TYPE_GAUGE:
1105       ret_value->gauge = (gauge_t) strtod (value, &endptr);
1106       break;
1107
1108     case DS_TYPE_DERIVE:
1109       ret_value->derive = (derive_t) strtoll (value, &endptr, 0);
1110       break;
1111
1112     case DS_TYPE_ABSOLUTE:
1113       ret_value->absolute = (absolute_t) strtoull (value, &endptr, 0);
1114       break;
1115
1116     default:
1117       sfree (value);
1118       ERROR ("parse_value: Invalid data source type: %i.", ds_type);
1119       return -1;
1120   }
1121
1122   if (value == endptr) {
1123     ERROR ("parse_value: Failed to parse string as %s: %s.",
1124         DS_TYPE_TO_STRING (ds_type), value);
1125     sfree (value);
1126     return -1;
1127   }
1128   else if ((NULL != endptr) && ('\0' != *endptr))
1129     INFO ("parse_value: Ignoring trailing garbage \"%s\" after %s value. "
1130         "Input string was \"%s\".",
1131         endptr, DS_TYPE_TO_STRING (ds_type), value_orig);
1132
1133   sfree (value);
1134   return 0;
1135 } /* int parse_value */
1136
1137 int parse_values (char *buffer, value_list_t *vl, const data_set_t *ds)
1138 {
1139         size_t i;
1140         char *dummy;
1141         char *ptr;
1142         char *saveptr;
1143
1144         if ((buffer == NULL) || (vl == NULL) || (ds == NULL))
1145                 return EINVAL;
1146
1147         i = 0;
1148         dummy = buffer;
1149         saveptr = NULL;
1150         vl->time = 0;
1151         while ((ptr = strtok_r (dummy, ":", &saveptr)) != NULL)
1152         {
1153                 dummy = NULL;
1154
1155                 if (i >= vl->values_len)
1156                 {
1157                         /* Make sure i is invalid. */
1158                         i = 0;
1159                         break;
1160                 }
1161
1162                 if (vl->time == 0)
1163                 {
1164                         if (strcmp ("N", ptr) == 0)
1165                                 vl->time = cdtime ();
1166                         else
1167                         {
1168                                 char *endptr = NULL;
1169                                 double tmp;
1170
1171                                 errno = 0;
1172                                 tmp = strtod (ptr, &endptr);
1173                                 if ((errno != 0)                    /* Overflow */
1174                                                 || (endptr == ptr)  /* Invalid string */
1175                                                 || (endptr == NULL) /* This should not happen */
1176                                                 || (*endptr != 0))  /* Trailing chars */
1177                                         return (-1);
1178
1179                                 vl->time = DOUBLE_TO_CDTIME_T (tmp);
1180                         }
1181
1182                         continue;
1183                 }
1184
1185                 if ((strcmp ("U", ptr) == 0) && (ds->ds[i].type == DS_TYPE_GAUGE))
1186                         vl->values[i].gauge = NAN;
1187                 else if (0 != parse_value (ptr, &vl->values[i], ds->ds[i].type))
1188                         return -1;
1189
1190                 i++;
1191         } /* while (strtok_r) */
1192
1193         if ((ptr != NULL) || (i == 0))
1194                 return (-1);
1195         return (0);
1196 } /* int parse_values */
1197
1198 #if !HAVE_GETPWNAM_R
1199 int getpwnam_r (const char *name, struct passwd *pwbuf, char *buf,
1200                 size_t buflen, struct passwd **pwbufp)
1201 {
1202         int status = 0;
1203         struct passwd *pw;
1204
1205         memset (pwbuf, '\0', sizeof (struct passwd));
1206
1207         pthread_mutex_lock (&getpwnam_r_lock);
1208
1209         do
1210         {
1211                 pw = getpwnam (name);
1212                 if (pw == NULL)
1213                 {
1214                         status = (errno != 0) ? errno : ENOENT;
1215                         break;
1216                 }
1217
1218 #define GETPWNAM_COPY_MEMBER(member) \
1219                 if (pw->member != NULL) \
1220                 { \
1221                         int len = strlen (pw->member); \
1222                         if (len >= buflen) \
1223                         { \
1224                                 status = ENOMEM; \
1225                                 break; \
1226                         } \
1227                         sstrncpy (buf, pw->member, buflen); \
1228                         pwbuf->member = buf; \
1229                         buf    += (len + 1); \
1230                         buflen -= (len + 1); \
1231                 }
1232                 GETPWNAM_COPY_MEMBER(pw_name);
1233                 GETPWNAM_COPY_MEMBER(pw_passwd);
1234                 GETPWNAM_COPY_MEMBER(pw_gecos);
1235                 GETPWNAM_COPY_MEMBER(pw_dir);
1236                 GETPWNAM_COPY_MEMBER(pw_shell);
1237
1238                 pwbuf->pw_uid = pw->pw_uid;
1239                 pwbuf->pw_gid = pw->pw_gid;
1240
1241                 if (pwbufp != NULL)
1242                         *pwbufp = pwbuf;
1243         } while (0);
1244
1245         pthread_mutex_unlock (&getpwnam_r_lock);
1246
1247         return (status);
1248 } /* int getpwnam_r */
1249 #endif /* !HAVE_GETPWNAM_R */
1250
1251 int notification_init (notification_t *n, int severity, const char *message,
1252                 const char *host,
1253                 const char *plugin, const char *plugin_instance,
1254                 const char *type, const char *type_instance)
1255 {
1256         memset (n, '\0', sizeof (notification_t));
1257
1258         n->severity = severity;
1259
1260         if (message != NULL)
1261                 sstrncpy (n->message, message, sizeof (n->message));
1262         if (host != NULL)
1263                 sstrncpy (n->host, host, sizeof (n->host));
1264         if (plugin != NULL)
1265                 sstrncpy (n->plugin, plugin, sizeof (n->plugin));
1266         if (plugin_instance != NULL)
1267                 sstrncpy (n->plugin_instance, plugin_instance,
1268                                 sizeof (n->plugin_instance));
1269         if (type != NULL)
1270                 sstrncpy (n->type, type, sizeof (n->type));
1271         if (type_instance != NULL)
1272                 sstrncpy (n->type_instance, type_instance,
1273                                 sizeof (n->type_instance));
1274
1275         return (0);
1276 } /* int notification_init */
1277
1278 int walk_directory (const char *dir, dirwalk_callback_f callback,
1279                 void *user_data, int include_hidden)
1280 {
1281         struct dirent *ent;
1282         DIR *dh;
1283         int success;
1284         int failure;
1285
1286         success = 0;
1287         failure = 0;
1288
1289         if ((dh = opendir (dir)) == NULL)
1290         {
1291                 char errbuf[1024];
1292                 ERROR ("walk_directory: Cannot open '%s': %s", dir,
1293                                 sstrerror (errno, errbuf, sizeof (errbuf)));
1294                 return -1;
1295         }
1296
1297         while ((ent = readdir (dh)) != NULL)
1298         {
1299                 int status;
1300
1301                 if (include_hidden)
1302                 {
1303                         if ((strcmp (".", ent->d_name) == 0)
1304                                         || (strcmp ("..", ent->d_name) == 0))
1305                                 continue;
1306                 }
1307                 else /* if (!include_hidden) */
1308                 {
1309                         if (ent->d_name[0]=='.')
1310                                 continue;
1311                 }
1312
1313                 status = (*callback) (dir, ent->d_name, user_data);
1314                 if (status != 0)
1315                         failure++;
1316                 else
1317                         success++;
1318         }
1319
1320         closedir (dh);
1321
1322         if ((success == 0) && (failure > 0))
1323                 return (-1);
1324         return (0);
1325 }
1326
1327 ssize_t read_file_contents (const char *filename, char *buf, size_t bufsize)
1328 {
1329         FILE *fh;
1330         ssize_t ret;
1331
1332         fh = fopen (filename, "r");
1333         if (fh == NULL)
1334                 return (-1);
1335
1336         ret = (ssize_t) fread (buf, 1, bufsize, fh);
1337         if ((ret == 0) && (ferror (fh) != 0))
1338         {
1339                 ERROR ("read_file_contents: Reading file \"%s\" failed.",
1340                                 filename);
1341                 ret = -1;
1342         }
1343
1344         fclose(fh);
1345         return (ret);
1346 }
1347
1348 counter_t counter_diff (counter_t old_value, counter_t new_value)
1349 {
1350         counter_t diff;
1351
1352         if (old_value > new_value)
1353         {
1354                 if (old_value <= 4294967295U)
1355                         diff = (4294967295U - old_value) + new_value + 1;
1356                 else
1357                         diff = (18446744073709551615ULL - old_value) + new_value + 1;
1358         }
1359         else
1360         {
1361                 diff = new_value - old_value;
1362         }
1363
1364         return (diff);
1365 } /* counter_t counter_diff */
1366
1367 int rate_to_value (value_t *ret_value, gauge_t rate, /* {{{ */
1368                 rate_to_value_state_t *state,
1369                 int ds_type, cdtime_t t)
1370 {
1371         gauge_t delta_gauge;
1372         cdtime_t delta_t;
1373
1374         if (ds_type == DS_TYPE_GAUGE)
1375         {
1376                 state->last_value.gauge = rate;
1377                 state->last_time = t;
1378
1379                 *ret_value = state->last_value;
1380                 return (0);
1381         }
1382
1383         /* Counter and absolute can't handle negative rates. Reset "last time"
1384          * to zero, so that the next valid rate will re-initialize the
1385          * structure. */
1386         if ((rate < 0.0)
1387                         && ((ds_type == DS_TYPE_COUNTER)
1388                                 || (ds_type == DS_TYPE_ABSOLUTE)))
1389         {
1390                 memset (state, 0, sizeof (*state));
1391                 return (EINVAL);
1392         }
1393
1394         /* Another invalid state: The time is not increasing. */
1395         if (t <= state->last_time)
1396         {
1397                 memset (state, 0, sizeof (*state));
1398                 return (EINVAL);
1399         }
1400
1401         delta_t = t - state->last_time;
1402         delta_gauge = (rate * CDTIME_T_TO_DOUBLE (delta_t)) + state->residual;
1403
1404         /* Previous value is invalid. */
1405         if (state->last_time == 0) /* {{{ */
1406         {
1407                 if (ds_type == DS_TYPE_DERIVE)
1408                 {
1409                         state->last_value.derive = (derive_t) rate;
1410                         state->residual = rate - ((gauge_t) state->last_value.derive);
1411                 }
1412                 else if (ds_type == DS_TYPE_COUNTER)
1413                 {
1414                         state->last_value.counter = (counter_t) rate;
1415                         state->residual = rate - ((gauge_t) state->last_value.counter);
1416                 }
1417                 else if (ds_type == DS_TYPE_ABSOLUTE)
1418                 {
1419                         state->last_value.absolute = (absolute_t) rate;
1420                         state->residual = rate - ((gauge_t) state->last_value.absolute);
1421                 }
1422                 else
1423                 {
1424                         assert (23 == 42);
1425                 }
1426
1427                 state->last_time = t;
1428                 return (EAGAIN);
1429         } /* }}} */
1430
1431         if (ds_type == DS_TYPE_DERIVE)
1432         {
1433                 derive_t delta_derive = (derive_t) delta_gauge;
1434
1435                 state->last_value.derive += delta_derive;
1436                 state->residual = delta_gauge - ((gauge_t) delta_derive);
1437         }
1438         else if (ds_type == DS_TYPE_COUNTER)
1439         {
1440                 counter_t delta_counter = (counter_t) delta_gauge;
1441
1442                 state->last_value.counter += delta_counter;
1443                 state->residual = delta_gauge - ((gauge_t) delta_counter);
1444         }
1445         else if (ds_type == DS_TYPE_ABSOLUTE)
1446         {
1447                 absolute_t delta_absolute = (absolute_t) delta_gauge;
1448
1449                 state->last_value.absolute = delta_absolute;
1450                 state->residual = delta_gauge - ((gauge_t) delta_absolute);
1451         }
1452         else
1453         {
1454                 assert (23 == 42);
1455         }
1456
1457         state->last_time = t;
1458         *ret_value = state->last_value;
1459         return (0);
1460 } /* }}} value_t rate_to_value */
1461
1462 int value_to_rate (gauge_t *ret_rate, /* {{{ */
1463                 value_t value, int ds_type, cdtime_t t, value_to_rate_state_t *state)
1464 {
1465         gauge_t interval;
1466
1467         /* Another invalid state: The time is not increasing. */
1468         if (t <= state->last_time)
1469         {
1470                 memset (state, 0, sizeof (*state));
1471                 return (EINVAL);
1472         }
1473
1474         interval = CDTIME_T_TO_DOUBLE(t - state->last_time);
1475
1476         /* Previous value is invalid. */
1477         if (state->last_time == 0)
1478         {
1479                 state->last_value = value;
1480                 state->last_time = t;
1481                 return (EAGAIN);
1482         }
1483
1484         switch (ds_type) {
1485         case DS_TYPE_DERIVE: {
1486                 derive_t diff = value.derive - state->last_value.derive;
1487                 *ret_rate = ((gauge_t) diff) / ((gauge_t) interval);
1488                 break;
1489         }
1490         case DS_TYPE_GAUGE: {
1491                 *ret_rate = value.gauge;
1492                 break;
1493         }
1494         case DS_TYPE_COUNTER: {
1495                 counter_t diff = counter_diff (state->last_value.counter, value.counter);
1496                 *ret_rate = ((gauge_t) diff) / ((gauge_t) interval);
1497                 break;
1498         }
1499         case DS_TYPE_ABSOLUTE: {
1500                 absolute_t diff = value.absolute;
1501                 *ret_rate = ((gauge_t) diff) / ((gauge_t) interval);
1502                 break;
1503         }
1504         default:
1505                 return EINVAL;
1506         }
1507
1508         state->last_value = value;
1509         state->last_time = t;
1510         return (0);
1511 } /* }}} value_t rate_to_value */
1512
1513 int service_name_to_port_number (const char *service_name)
1514 {
1515         struct addrinfo *ai_list;
1516         int status;
1517         int service_number;
1518
1519         if (service_name == NULL)
1520                 return (-1);
1521
1522         struct addrinfo ai_hints = {
1523                 .ai_family = AF_UNSPEC
1524         };
1525
1526         status = getaddrinfo (/* node = */ NULL, service_name,
1527                         &ai_hints, &ai_list);
1528         if (status != 0)
1529         {
1530                 ERROR ("service_name_to_port_number: getaddrinfo failed: %s",
1531                                 gai_strerror (status));
1532                 return (-1);
1533         }
1534
1535         service_number = -1;
1536         for (struct addrinfo *ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
1537         {
1538                 if (ai_ptr->ai_family == AF_INET)
1539                 {
1540                         struct sockaddr_in *sa;
1541
1542                         sa = (void *) ai_ptr->ai_addr;
1543                         service_number = (int) ntohs (sa->sin_port);
1544                 }
1545                 else if (ai_ptr->ai_family == AF_INET6)
1546                 {
1547                         struct sockaddr_in6 *sa;
1548
1549                         sa = (void *) ai_ptr->ai_addr;
1550                         service_number = (int) ntohs (sa->sin6_port);
1551                 }
1552
1553                 if ((service_number > 0) && (service_number <= 65535))
1554                         break;
1555         }
1556
1557         freeaddrinfo (ai_list);
1558
1559         if ((service_number > 0) && (service_number <= 65535))
1560                 return (service_number);
1561         return (-1);
1562 } /* int service_name_to_port_number */
1563
1564 void set_sock_opts (int sockfd) /* {{{ */
1565 {
1566         int status;
1567         int socktype;
1568
1569         socklen_t socklen = sizeof (socklen_t);
1570         int so_keepalive = 1;
1571         int tcp_keepidle = ((CDTIME_T_TO_MS(plugin_get_interval()) - 1) / 100 + 1);
1572         int tcp_keepintvl = ((CDTIME_T_TO_MS(plugin_get_interval()) - 1) / 1000 + 1);
1573
1574         status = getsockopt (sockfd, SOL_SOCKET, SO_TYPE, &socktype, &socklen);
1575         if (status != 0)
1576         {
1577                 WARNING ("set_sock_opts: failed to determine socket type");
1578                 return;
1579         }
1580
1581         if (socktype == SOCK_STREAM)
1582         {
1583                 status = setsockopt(sockfd, SOL_SOCKET, SO_KEEPALIVE,
1584                                 &so_keepalive, sizeof (so_keepalive));
1585                 if (status != 0)
1586                         WARNING ("set_sock_opts: failed to set socket keepalive flag");
1587
1588 #ifdef TCP_KEEPIDLE
1589                 status = setsockopt(sockfd, IPPROTO_TCP, TCP_KEEPIDLE,
1590                                 &tcp_keepidle, sizeof (tcp_keepidle));
1591                 if (status != 0)
1592                         WARNING ("set_sock_opts: failed to set socket tcp keepalive time");
1593 #endif
1594
1595 #ifdef TCP_KEEPINTVL
1596                 status = setsockopt(sockfd, IPPROTO_TCP, TCP_KEEPINTVL,
1597                                 &tcp_keepintvl, sizeof (tcp_keepintvl));
1598                 if (status != 0)
1599                         WARNING ("set_sock_opts: failed to set socket tcp keepalive interval");
1600 #endif
1601         }
1602 } /* }}} void set_sock_opts */
1603
1604 int strtoderive (const char *string, derive_t *ret_value) /* {{{ */
1605 {
1606         derive_t tmp;
1607         char *endptr;
1608
1609         if ((string == NULL) || (ret_value == NULL))
1610                 return (EINVAL);
1611
1612         errno = 0;
1613         endptr = NULL;
1614         tmp = (derive_t) strtoll (string, &endptr, /* base = */ 0);
1615         if ((endptr == string) || (errno != 0))
1616                 return (-1);
1617
1618         *ret_value = tmp;
1619         return (0);
1620 } /* }}} int strtoderive */
1621
1622 int strtogauge (const char *string, gauge_t *ret_value) /* {{{ */
1623 {
1624         gauge_t tmp;
1625         char *endptr = NULL;
1626
1627         if ((string == NULL) || (ret_value == NULL))
1628                 return (EINVAL);
1629
1630         errno = 0;
1631         endptr = NULL;
1632         tmp = (gauge_t) strtod (string, &endptr);
1633         if (errno != 0)
1634                 return (errno);
1635         else if ((endptr == NULL) || (*endptr != 0))
1636                 return (EINVAL);
1637
1638         *ret_value = tmp;
1639         return (0);
1640 } /* }}} int strtogauge */
1641
1642 int strarray_add (char ***ret_array, size_t *ret_array_len, char const *str) /* {{{ */
1643 {
1644         char **array;
1645         size_t array_len = *ret_array_len;
1646
1647         if (str == NULL)
1648                 return (EINVAL);
1649
1650         array = realloc (*ret_array,
1651             (array_len + 1) * sizeof (*array));
1652         if (array == NULL)
1653                 return (ENOMEM);
1654         *ret_array = array;
1655
1656         array[array_len] = strdup (str);
1657         if (array[array_len] == NULL)
1658                 return (ENOMEM);
1659
1660         array_len++;
1661         *ret_array_len = array_len;
1662         return (0);
1663 } /* }}} int strarray_add */
1664
1665 void strarray_free (char **array, size_t array_len) /* {{{ */
1666 {
1667         for (size_t i = 0; i < array_len; i++)
1668                 sfree (array[i]);
1669         sfree (array);
1670 } /* }}} void strarray_free */