src/librrdc.c

   1 /**
   2  * collectd - src/rrdc.c
   3  * Copyright (C) 2008 Florian octo Forster
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; only version 2 of the License is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Florian octo Forster <octo at verplant.org>
  20  **/
  21
  22 #include "rrdd.h"
  23 #include "rrdc.h"
  24
  25 static pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;
  26 static int sd;
  27 static FILE *sh;
  28
  29 static int buffer_add_string (const char *str,
  30     char **buffer_ret, size_t *buffer_size_ret)
  31 {
  32   size_t str_size;
  33
  34   str_size = strlen (str) + 1;
  35
  36   if (*buffer_size_ret < str_size)
  37     return (-1);
  38
  39   memcpy (*buffer_ret, str, str_size);
  40   *buffer_ret += str_size;
  41   *buffer_size_ret -= str_size;
  42
  43   return (0);
  44 } /* int buffer_add_string */
  45
  46 static int buffer_add_value (const char *value,
  47     char **buffer_ret, size_t *buffer_size_ret)
  48 {
  49   char temp[4096];
  50
  51   if (strncmp (value, "N:", 2) == 0)
  52     snprintf (temp, sizeof (temp), "%lu:%s",
  53         (unsigned long) time (NULL), value + 2);
  54   else
  55     strncpy (temp, value, sizeof (temp));
  56   temp[sizeof (temp) - 1] = 0;
  57
  58   return (buffer_add_string (temp, buffer_ret, buffer_size_ret));
  59 } /* int buffer_add_value */
  60
  61 int rrdd_connect (const char *path)
  62 {
  63   struct sockaddr_un sa;
  64   int status;
  65
  66   if (path == NULL)
  67     path = RRDD_SOCK_PATH;
  68
  69   pthread_mutex_lock (&lock);
  70
  71   if (sh != NULL)
  72   {
  73     pthread_mutex_unlock (&lock);
  74     return (0);
  75   }
  76
  77   sd = socket (PF_UNIX, SOCK_STREAM, /* protocol = */ 0);
  78   if (sd < 0)
  79   {
  80     status = errno;
  81     pthread_mutex_unlock (&lock);
  82     return (status);
  83   }
  84
  85   memset (&sa, 0, sizeof (sa));
  86   sa.sun_family = AF_UNIX;
  87   strncpy (sa.sun_path, path, sizeof (sa.sun_path) - 1);
  88
  89   status = connect (sd, (struct sockaddr *) &sa, sizeof (sa));
  90   if (status != 0)
  91   {
  92     status = errno;
  93     pthread_mutex_unlock (&lock);
  94     return (status);
  95   }
  96
  97   sh = fdopen (sd, "w+");
  98   if (sh == NULL)
  99   {
 100     status = errno;
 101     close (sd);
 102     sd = -1;
 103     pthread_mutex_unlock (&lock);
 104     return (status);
 105   }
 106
 107   pthread_mutex_unlock (&lock);
 108
 109   return (0);
 110 } /* int rrdd_connect */
 111
 112 int rrdd_disconnect (void)
 113 {
 114   int status;
 115
 116   pthread_mutex_lock (&lock);
 117
 118   if (sh == NULL)
 119   {
 120     pthread_mutex_unlock (&lock);
 121     return (-1);
 122   }
 123
 124   status = fclose (sh);
 125   if (status != 0)
 126     status = errno;
 127
 128   sh = NULL;
 129   sd = -1;
 130
 131   pthread_mutex_unlock (&lock);
 132
 133   return (status);
 134 } /* int rrdd_disconnect */
 135
 136 int rrdd_update (const char *filename, int values_num,
 137                 const char * const *values)
 138 {
 139   char buffer[4096];
 140   char *buffer_ptr;
 141   size_t buffer_size;
 142   int status;
 143   int i;
 144
 145   memset (buffer, 0, sizeof (buffer));
 146   buffer_ptr = &buffer[0];
 147   buffer_size = sizeof (buffer) - 1;
 148
 149   buffer_add_string ("update", &buffer_ptr, &buffer_size);
 150   buffer_add_string (filename, &buffer_ptr, &buffer_size);
 151   for (i = 0; i < values_num; i++)
 152     buffer_add_value (values[i], &buffer_ptr, &buffer_size);
 153
 154   pthread_mutex_lock (&lock);
 155
 156   if (sh == NULL)
 157   {
 158     pthread_mutex_unlock (&lock);
 159     return (ENOTCONN);
 160   }
 161
 162   status = write (sd, buffer, sizeof (buffer) - buffer_size);
 163
 164   status = read (sd, buffer, sizeof (buffer));
 165   if (status < 0)
 166   {
 167     status = errno;
 168     pthread_mutex_unlock (&lock);
 169     return (status);
 170   }
 171   else if (status == 0)
 172   {
 173     pthread_mutex_unlock (&lock);
 174     return (ENODATA);
 175   }
 176
 177   status = atoi (buffer);
 178
 179   pthread_mutex_unlock (&lock);
 180
 181   return (status);
 182 } /* int rrd_update_daemon */
 183
 184 /*
 185  * vim: set sw=2 sts=2 ts=8 et fdm=marker :
 186  */