X-Git-Url: https://git.octo.it/?p=collection4.git;a=blobdiff_plain;f=src%2Fdp_rrdtool.c;h=d7ad5ba612360bb05a3853602176ab7594b8ef55;hp=6db02059d6a95ebfe87fd17b2a0b98e5bd6f8eeb;hb=c4c851ce35ad88eff7f843dbced132f55b8a2f6c;hpb=9e7f6711d2cebfec79c8038c51037ba79810323c

diff --git a/src/dp_rrdtool.c b/src/dp_rrdtool.c
index 6db0205..d7ad5ba 100644
--- a/src/dp_rrdtool.c
+++ b/src/dp_rrdtool.c
@@ -21,12 +21,7 @@
  *   Florian octo Forster <ff at octo.it>
  **/
 
-#include "graph_types.h"
-#include "graph_ident.h"
-#include "data_provider.h"
-#include "filesystem.h"
-#include "oconfig.h"
-#include "common.h"
+#include "config.h"
 
 #include <stdlib.h>
 #include <stdio.h>
@@ -37,6 +32,16 @@
 
 #include <rrd.h>
 
+#include "graph_types.h"
+#include "graph_ident.h"
+#include "data_provider.h"
+#include "filesystem.h"
+#include "oconfig.h"
+#include "common.h"
+
+#include <fcgiapp.h>
+#include <fcgi_stdio.h>
+
 struct dp_rrdtool_s
 {
   char *data_dir;
@@ -52,23 +57,27 @@ struct dp_get_idents_data_s
 typedef struct dp_get_idents_data_s dp_get_idents_data_t;
 
 static int scan_type_cb (__attribute__((unused)) const char *base_dir,
-    const char *sub_dir, void *ud)
+    const char *file, void *ud)
 { /* {{{ */
   dp_get_idents_data_t *data = ud;
-  size_t sub_dir_len;
+  size_t file_len;
   char type_copy[1024];
+  size_t type_copy_len;
   char *type_inst;
 
-  sub_dir_len = strlen (sub_dir);
-  if (sub_dir_len < 5)
+  file_len = strlen (file);
+  if (file_len < 5)
     return (0);
 
   /* Ignore files that don't end in ".rrd". */
-  if (strcasecmp (".rrd", sub_dir + (sub_dir_len - 4)) != 0)
+  if (strcasecmp (".rrd", file + (file_len - 4)) != 0)
     return (0);
 
-  strncpy (type_copy, sub_dir, sizeof (type_copy));
-  type_copy[sub_dir_len - 4] = 0;
+  strncpy (type_copy, file, sizeof (type_copy));
+  type_copy_len = file_len - 4;
+  if (type_copy_len > (sizeof (type_copy) - 1))
+    type_copy_len = sizeof (type_copy) - 1;
+  type_copy[type_copy_len] = 0;
 
   type_inst = strchr (type_copy, '-');
   if (type_inst != NULL)
@@ -217,7 +226,8 @@ static int get_ident_ds_names (void *priv, graph_ident_t *ident,
   info = rrd_info (rrd_argc, rrd_argv);
   if (info == NULL)
   {
-    printf ("%s: rrd_info (%s) failed.\n", __func__, file);
+    fprintf (stderr, "%s: rrd_info (%s) failed.\n", __func__, file);
+    fflush (stderr);
     return (-1);
   }
 
@@ -234,13 +244,13 @@ static int get_ident_ds_names (void *priv, graph_ident_t *ident,
       continue;
 
     keylen = strlen (ptr->key);
-    if (keylen < strlen ("ds[?].index"))
+    if (keylen < strlen ("ds[?].type"))
       continue;
 
-    dslen = keylen - strlen ("ds[].index");
+    dslen = keylen - strlen ("ds[].type");
     assert (dslen >= 1);
 
-    if (strcmp ("].index", ptr->key + (strlen ("ds[") + dslen)) != 0)
+    if (strcmp ("].type", ptr->key + (strlen ("ds[") + dslen)) != 0)
       continue;
 
     ds = malloc (dslen + 1);
@@ -284,6 +294,9 @@ static int get_ident_data (void *priv,
   unsigned long data_index;
   unsigned long data_length;
 
+  dp_data_point_t *dp = NULL;
+  size_t dp_num = 0;
+
   status = ident_to_rrdfile (ident, config, filename, sizeof (filename));
   if (status != 0)
     return (status);
@@ -308,6 +321,7 @@ static int get_ident_data (void *priv,
     free (ds_namv[i]);            \
   free (ds_namv);                 \
   free (data);                    \
+  free (dp);                      \
   return (ret_status);            \
 } while (0)
 
@@ -321,21 +335,24 @@ static int get_ident_data (void *priv,
   /* Number of data points returned. */
   data_length = (rrd_end - rrd_start) / step;
 
+  dp_num = (size_t) data_length;
+  dp = calloc (dp_num, sizeof (*dp));
+  if (dp == NULL)
+    BAIL_OUT (ENOMEM);
+
   for (data_index = 0; data_index < data_length; data_index++)
   {
-    dp_data_point_t dp;
     unsigned long index = (ds_count * data_index) + ds_index;
 
-    memset (&dp, 0, sizeof (dp));
-    dp.time.tv_sec = rrd_start + (data_index * step);
-    dp.time.tv_nsec = 0;
-    dp.value = (double) data[index];
-
-    status = (*cb) (ident, ds_name, &dp, ud);
-    if (status != 0)
-      BAIL_OUT (status);
+    dp[data_index].time.tv_sec = rrd_start + (data_index * step);
+    dp[data_index].time.tv_nsec = 0;
+    dp[data_index].value = (double) data[index];
   }
 
+  status = (*cb) (ident, ds_name, dp, dp_num, ud);
+  if (status != 0)
+    BAIL_OUT (status);
+
   BAIL_OUT (0);
 #undef BAIL_OUT
 } /* }}} int get_ident_data */