summaryrefslogtreecommitdiff
path: root/daemon/rrd-update.c
diff options
context:
space:
mode:
Diffstat (limited to 'daemon/rrd-update.c')
-rw-r--r--daemon/rrd-update.c134
1 files changed, 134 insertions, 0 deletions
diff --git a/daemon/rrd-update.c b/daemon/rrd-update.c
new file mode 100644
index 0000000..b4be997
--- /dev/null
+++ b/daemon/rrd-update.c
@@ -0,0 +1,134 @@
+/*
+ * Copyright (c) 2005, Nate Nielsen
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * * Redistributions of source code must retain the above
+ * copyright notice, this list of conditions and the
+ * following disclaimer.
+ * * Redistributions in binary form must reproduce the
+ * above copyright notice, this list of conditions and
+ * the following disclaimer in the documentation and/or
+ * other materials provided with the distribution.
+ * * The names of contributors to this software may not be
+ * used to endorse or promote products derived from this
+ * software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+ * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
+ * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
+ * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
+ * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
+ * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
+ * DAMAGE.
+ *
+ *
+ * CONTRIBUTORS
+ * Nate Nielsen <nielsen@memberwebs.com>
+ *
+ */
+
+#define _GNU_SOURCE
+
+#include "usuals.h"
+#include <errno.h>
+#include <syslog.h>
+#include <unistd.h>
+
+#include "stringx.h"
+#include "rrdbotd.h"
+
+#define MAX_NUMLEN 40
+
+void rb_rrd_update(rb_poller *poll)
+{
+ char buf[MAX_NUMLEN];
+ const char* argv[5];
+ char* template;
+ char* items;
+ int r, tlen, ilen;
+ rb_item *it;
+
+ if(!poll->items)
+ return;
+
+ tlen = 0;
+ ilen = 3;
+
+ for(it = poll->items; it; it = it->next)
+ {
+ tlen += strlen(it->rrdfield) + 1;
+ ilen += 40;
+ }
+
+ template = (char*)calloc(tlen, sizeof(char));
+ items = (char*)calloc(ilen, sizeof(char));
+
+ if(!template || !items)
+ {
+ if(items)
+ free(items);
+ if(template)
+ free(template);
+ rb_messagex(LOG_CRIT, "out of memory");
+ return;
+ }
+
+ /* Put in the right time */
+ snprintf(items, ilen, "%lld:", (poll->last_polled / 1000L));
+
+ /* Build the rest of the arguments */
+ for(it = poll->items; it; it = it->next)
+ {
+ if(it != poll->items)
+ {
+ strlcat(template, ":", tlen);
+ strlcat(items, ":", ilen);
+ }
+
+ strlcat(template, it->rrdfield, tlen);
+
+ if(it->value == RB_UNKNOWN)
+ strlcat(items, "U", ilen);
+ else
+ {
+ snprintf(buf, MAX_NUMLEN, "%.4lf", it->value);
+ buf[MAX_NUMLEN - 1] = 0;
+ strlcat(items, buf, ilen);
+ }
+ }
+
+ /* Always have to clear before calling rrdtool. klunky :( */
+ optind = 0;
+ opterr = 0;
+
+ /* TODO: We need support for @ to specify when these values occurred */
+
+ argv[0] = "rrdupdate";
+ argv[1] = poll->rrdname;
+ argv[2] = "-t";
+ argv[3] = template;
+ argv[4] = items;
+
+ rb_messagex(LOG_DEBUG, "updating RRD file: %s", poll->rrdname);
+ rb_messagex(LOG_DEBUG, "> template: %s", template);
+ rb_messagex(LOG_DEBUG, "> values: %s", items);
+
+ rrd_clear_error();
+ r = rrd_update(5, argv);
+
+ if(r != 0)
+ rb_messagex(LOG_ERR, "couldn't update rrd file: %s: %s",
+ poll->rrdname, rrd_get_error());
+
+ free(template);
+ free(items);
+}