src/{lib,}rrdc.c: Make `rrdd_connect' use the default path if path == NULL.

[rrdd.git] / src / librrdc.c

diff --git a/src/librrdc.c b/src/librrdc.c
index c0ede89..93d9420 100644 (file)
--- a/src/librrdc.c
+++ b/src/librrdc.c
@@ -22,17 +22,50 @@
 #include "rrdd.h"
 #include "rrdc.h"
 
-#define SOCK_TEMPLATE "/tmp/rrdc.sock.XXXXXX"
-
 static pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;
 static int sd;
 static FILE *sh;
 
+static int buffer_add_string (const char *str,
+    char **buffer_ret, size_t *buffer_size_ret)
+{
+  size_t str_size;
+
+  str_size = strlen (str) + 1;
+
+  if (*buffer_size_ret < str_size)
+    return (-1);
+
+  memcpy (*buffer_ret, str, str_size);
+  *buffer_ret += str_size;
+  *buffer_size_ret -= str_size;
+
+  return (0);
+} /* int buffer_add_string */
+
+static int buffer_add_value (const char *value,
+    char **buffer_ret, size_t *buffer_size_ret)
+{
+  char temp[4096];
+
+  if (strncmp (value, "N:", 2) == 0)
+    snprintf (temp, sizeof (temp), "%lu:%s",
+        (unsigned long) time (NULL), value + 2);
+  else
+    strncpy (temp, value, sizeof (temp));
+  temp[sizeof (temp) - 1] = 0;
+
+  return (buffer_add_string (temp, buffer_ret, buffer_size_ret));
+} /* int buffer_add_value */
+
 int rrdd_connect (const char *path)
 {
   struct sockaddr_un sa;
   int status;
 
+  if (path == NULL)
+    path = RRDD_SOCK_PATH;
+
   pthread_mutex_lock (&lock);
 
   if (sh != NULL)
@@ -103,17 +136,49 @@ int rrdd_disconnect (void)
 int rrdd_update (const char *filename, int values_num,
                const char * const *values)
 {
+  char buffer[4096];
+  char *buffer_ptr;
+  size_t buffer_size;
+  int status;
+  int i;
+
+  memset (buffer, 0, sizeof (buffer));
+  buffer_ptr = &buffer[0];
+  buffer_size = sizeof (buffer) - 1;
+
+  buffer_add_string ("update", &buffer_ptr, &buffer_size);
+  buffer_add_string (filename, &buffer_ptr, &buffer_size);
+  for (i = 0; i < values_num; i++)
+    buffer_add_value (values[i], &buffer_ptr, &buffer_size);
+
   pthread_mutex_lock (&lock);
 
   if (sh == NULL)
   {
     pthread_mutex_unlock (&lock);
-    return (-1);
+    return (ENOTCONN);
+  }
+
+  status = write (sd, buffer, sizeof (buffer) - buffer_size);
+
+  status = read (sd, buffer, sizeof (buffer));
+  if (status < 0)
+  {
+    status = errno;
+    pthread_mutex_unlock (&lock);
+    return (status);
+  }
+  else if (status == 0)
+  {
+    pthread_mutex_unlock (&lock);
+    return (ENODATA);
   }
 
+  status = atoi (buffer);
+
   pthread_mutex_unlock (&lock);
 
-  return (0);
+  return (status);
 } /* int rrd_update_daemon */
 
 /*