]> arthur.barton.de Git - netdata.git/blobdiff - src/rrd.c
Merge pull request #528 from paulfantom/master
[netdata.git] / src / rrd.c
index 379b1c8b8d642b203b20a7760550e5aed6c4b0a7..28efff0a15074992287fba30a053c58eb9974da9 100644 (file)
--- a/src/rrd.c
+++ b/src/rrd.c
@@ -18,6 +18,7 @@
 #include "log.h"
 #include "appconfig.h"
 
+#include "main.h"
 #include "rrd.h"
 
 #define RRD_DEFAULT_GAP_INTERPOLATIONS 1
@@ -42,8 +43,6 @@ int rrd_memory_mode = RRD_MEMORY_MODE_SAVE;
 // ----------------------------------------------------------------------------
 // RRDSET index
 
-static int rrdset_iterator(avl *a) { if(a) {}; return 0; }
-
 static int rrdset_compare(void* a, void* b) {
        if(((RRDSET *)a)->hash < ((RRDSET *)b)->hash) return -1;
        else if(((RRDSET *)a)->hash > ((RRDSET *)b)->hash) return 1;
@@ -59,13 +58,11 @@ avl_tree_lock rrdset_root_index = {
 #define rrdset_index_del(st) avl_remove_lock(&rrdset_root_index, (avl *)(st))
 
 static RRDSET *rrdset_index_find(const char *id, uint32_t hash) {
-       RRDSET *result = NULL, tmp;
-       strncpy(tmp.id, id, RRD_ID_LENGTH_MAX);
-       tmp.id[RRD_ID_LENGTH_MAX] = '\0';
+       RRDSET tmp;
+       strncpyz(tmp.id, id, RRD_ID_LENGTH_MAX);
        tmp.hash = (hash)?hash:simple_hash(tmp.id);
 
-       avl_search_lock(&(rrdset_root_index), (avl *) &tmp, rrdset_iterator, (avl **) &result);
-       return result;
+       return (RRDSET *)avl_search_lock(&(rrdset_root_index), (avl *) &tmp);
 }
 
 // ----------------------------------------------------------------------------
@@ -73,8 +70,6 @@ static RRDSET *rrdset_index_find(const char *id, uint32_t hash) {
 
 #define rrdset_from_avlname(avlname_ptr) ((RRDSET *)((avlname_ptr) - offsetof(RRDSET, avlname)))
 
-static int rrdset_iterator_name(avl *a) { if(a) {}; return 0; }
-
 static int rrdset_compare_name(void* a, void* b) {
        RRDSET *A = rrdset_from_avlname(a);
        RRDSET *B = rrdset_from_avlname(b);
@@ -105,7 +100,7 @@ static RRDSET *rrdset_index_find_name(const char *name, uint32_t hash) {
        tmp.hash_name = (hash)?hash:simple_hash(tmp.name);
 
        // fprintf(stderr, "SEARCHING: %s\n", name);
-       avl_search_lock(&(rrdset_root_index_name), (avl *) (&(tmp.avlname)), rrdset_iterator_name, (avl **) &result);
+       result = avl_search_lock(&(rrdset_root_index_name), (avl *) (&(tmp.avlname)));
        if(result) {
                RRDSET *st = rrdset_from_avlname(result);
                if(strcmp(st->magic, RRDSET_MAGIC))
@@ -122,8 +117,6 @@ static RRDSET *rrdset_index_find_name(const char *name, uint32_t hash) {
 // ----------------------------------------------------------------------------
 // RRDDIM index
 
-static int rrddim_iterator(avl *a) { if(a) {}; return 0; }
-
 static int rrddim_compare(void* a, void* b) {
        if(((RRDDIM *)a)->hash < ((RRDDIM *)b)->hash) return -1;
        else if(((RRDDIM *)a)->hash > ((RRDDIM *)b)->hash) return 1;
@@ -134,13 +127,11 @@ static int rrddim_compare(void* a, void* b) {
 #define rrddim_index_del(st,rd ) avl_remove_lock(&((st)->dimensions_index), (avl *)(rd))
 
 static RRDDIM *rrddim_index_find(RRDSET *st, const char *id, uint32_t hash) {
-       RRDDIM *result = NULL, tmp;
-       strncpy(tmp.id, id, RRD_ID_LENGTH_MAX);
-       tmp.id[RRD_ID_LENGTH_MAX] = '\0';
+       RRDDIM tmp;
+       strncpyz(tmp.id, id, RRD_ID_LENGTH_MAX);
        tmp.hash = (hash)?hash:simple_hash(tmp.id);
 
-       avl_search_lock(&(st->dimensions_index), (avl *) &tmp, rrddim_iterator, (avl **) &result);
-       return result;
+       return (RRDDIM *)avl_search_lock(&(st->dimensions_index), (avl *) &tmp);
 }
 
 // ----------------------------------------------------------------------------
@@ -212,8 +203,8 @@ int rrd_memory_mode_id(const char *name)
 
 int rrddim_algorithm_id(const char *name)
 {
-       if(strcmp(name, RRDDIM_ABSOLUTE_NAME) == 0)                     return RRDDIM_ABSOLUTE;
        if(strcmp(name, RRDDIM_INCREMENTAL_NAME) == 0)                  return RRDDIM_INCREMENTAL;
+       if(strcmp(name, RRDDIM_ABSOLUTE_NAME) == 0)                     return RRDDIM_ABSOLUTE;
        if(strcmp(name, RRDDIM_PCENT_OVER_ROW_TOTAL_NAME) == 0)                 return RRDDIM_PCENT_OVER_ROW_TOTAL;
        if(strcmp(name, RRDDIM_PCENT_OVER_DIFF_TOTAL_NAME) == 0)        return RRDDIM_PCENT_OVER_DIFF_TOTAL;
        return RRDDIM_ABSOLUTE;
@@ -245,15 +236,18 @@ const char *rrddim_algorithm_name(int chart_type)
 // ----------------------------------------------------------------------------
 // chart names
 
-char *rrdset_strncpy_name(char *to, const char *from, int length)
+char *rrdset_strncpyz_name(char *to, const char *from, size_t length)
 {
-       int i;
-       for(i = 0; i < length && from[i] ;i++) {
-               if(from[i] == '.' || isalpha(from[i]) || isdigit(from[i])) to[i] = from[i];
-               else to[i] = '_';
+       char c, *p = to;
+
+       while (length-- && (c = *from++)) {
+               if(c != '.' && !isalnum(c))
+                       c = '_';
+
+               *p++ = c;
        }
-       if(i < length) to[i] = '\0';
-       to[length - 1] = '\0';
+
+       *p = '\0';
 
        return to;
 }
@@ -267,8 +261,8 @@ void rrdset_set_name(RRDSET *st, const char *name)
        char b[CONFIG_MAX_VALUE + 1];
        char n[RRD_ID_LENGTH_MAX + 1];
 
-       snprintf(n, RRD_ID_LENGTH_MAX, "%s.%s", st->type, name);
-       rrdset_strncpy_name(b, n, CONFIG_MAX_VALUE);
+       snprintfz(n, RRD_ID_LENGTH_MAX, "%s.%s", st->type, name);
+       rrdset_strncpyz_name(b, n, CONFIG_MAX_VALUE);
        st->name = config_get(st->id, "name", b);
        st->hash_name = simple_hash(st->name);
 
@@ -287,9 +281,9 @@ char *rrdset_cache_dir(const char *id)
 
        char b[FILENAME_MAX + 1];
        char n[FILENAME_MAX + 1];
-       rrdset_strncpy_name(b, id, FILENAME_MAX);
+       rrdset_strncpyz_name(b, id, FILENAME_MAX);
 
-       snprintf(n, FILENAME_MAX, "%s/%s", cache_dir, b);
+       snprintfz(n, FILENAME_MAX, "%s/%s", cache_dir, b);
        ret = config_get(id, "cache directory", n);
 
        if(rrd_memory_mode == RRD_MEMORY_MODE_MAP || rrd_memory_mode == RRD_MEMORY_MODE_SAVE) {
@@ -341,7 +335,7 @@ RRDSET *rrdset_create(const char *type, const char *id, const char *name, const
        char fullfilename[FILENAME_MAX + 1];
        RRDSET *st = NULL;
 
-       snprintf(fullid, RRD_ID_LENGTH_MAX, "%s.%s", type, id);
+       snprintfz(fullid, RRD_ID_LENGTH_MAX, "%s.%s", type, id);
 
        st = rrdset_find(fullid);
        if(st) {
@@ -361,7 +355,7 @@ RRDSET *rrdset_create(const char *type, const char *id, const char *name, const
 
        debug(D_RRD_CALLS, "Creating RRD_STATS for '%s.%s'.", type, id);
 
-       snprintf(fullfilename, FILENAME_MAX, "%s/main.db", cache_dir);
+       snprintfz(fullfilename, FILENAME_MAX, "%s/main.db", cache_dir);
        if(rrd_memory_mode != RRD_MEMORY_MODE_RAM) st = (RRDSET *)mymmap(fullfilename, size, ((rrd_memory_mode == RRD_MEMORY_MODE_MAP)?MAP_SHARED:MAP_PRIVATE), 0);
        if(st) {
                if(strcmp(st->magic, RRDSET_MAGIC) != 0) {
@@ -453,7 +447,7 @@ RRDSET *rrdset_create(const char *type, const char *id, const char *name, const
 
        {
                char varvalue[CONFIG_MAX_VALUE + 1];
-               snprintf(varvalue, CONFIG_MAX_VALUE, "%s (%s)", title?title:"", st->name);
+               snprintfz(varvalue, CONFIG_MAX_VALUE, "%s (%s)", title?title:"", st->name);
                st->title = config_get(st->id, "title", varvalue);
        }
 
@@ -478,8 +472,8 @@ RRDDIM *rrddim_add(RRDSET *st, const char *id, const char *name, long multiplier
 
        debug(D_RRD_CALLS, "Adding dimension '%s/%s'.", st->id, id);
 
-       rrdset_strncpy_name(filename, id, FILENAME_MAX);
-       snprintf(fullfilename, FILENAME_MAX, "%s/%s.db", st->cache_dir, filename);
+       rrdset_strncpyz_name(filename, id, FILENAME_MAX);
+       snprintfz(fullfilename, FILENAME_MAX, "%s/%s.db", st->cache_dir, filename);
        if(rrd_memory_mode != RRD_MEMORY_MODE_RAM) rd = (RRDDIM *)mymmap(fullfilename, size, ((rrd_memory_mode == RRD_MEMORY_MODE_MAP)?MAP_SHARED:MAP_PRIVATE), 1);
        if(rd) {
                struct timeval now;
@@ -551,19 +545,19 @@ RRDDIM *rrddim_add(RRDSET *st, const char *id, const char *name, long multiplier
 
        strcpy(rd->magic, RRDDIMENSION_MAGIC);
        strcpy(rd->cache_filename, fullfilename);
-       strncpy(rd->id, id, RRD_ID_LENGTH_MAX);
+       strncpyz(rd->id, id, RRD_ID_LENGTH_MAX);
        rd->hash = simple_hash(rd->id);
 
-       snprintf(varname, CONFIG_MAX_NAME, "dim %s name", rd->id);
+       snprintfz(varname, CONFIG_MAX_NAME, "dim %s name", rd->id);
        rd->name = config_get(st->id, varname, (name && *name)?name:rd->id);
 
-       snprintf(varname, CONFIG_MAX_NAME, "dim %s algorithm", rd->id);
+       snprintfz(varname, CONFIG_MAX_NAME, "dim %s algorithm", rd->id);
        rd->algorithm = rrddim_algorithm_id(config_get(st->id, varname, rrddim_algorithm_name(algorithm)));
 
-       snprintf(varname, CONFIG_MAX_NAME, "dim %s multiplier", rd->id);
+       snprintfz(varname, CONFIG_MAX_NAME, "dim %s multiplier", rd->id);
        rd->multiplier = config_get_number(st->id, varname, multiplier);
 
-       snprintf(varname, CONFIG_MAX_NAME, "dim %s divisor", rd->id);
+       snprintfz(varname, CONFIG_MAX_NAME, "dim %s divisor", rd->id);
        rd->divisor = config_get_number(st->id, varname, divisor);
        if(!rd->divisor) rd->divisor = 1;
 
@@ -594,7 +588,7 @@ void rrddim_set_name(RRDSET *st, RRDDIM *rd, const char *name)
        debug(D_RRD_CALLS, "rrddim_set_name() %s.%s", st->name, rd->name);
 
        char varname[CONFIG_MAX_NAME + 1];
-       snprintf(varname, CONFIG_MAX_NAME, "dim %s name", rd->id);
+       snprintfz(varname, CONFIG_MAX_NAME, "dim %s name", rd->id);
        config_set_default(st->id, varname, name);
 }
 
@@ -668,12 +662,8 @@ void rrdset_free_all(void)
        info("Memory cleanup completed...");
 }
 
-void rrdset_save_all(void)
-{
-       debug(D_RRD_CALLS, "rrdset_save_all()");
-
-       // let it log a few error messages
-       error_log_limit_reset();
+void rrdset_save_all(void) {
+       info("Saving database...");
 
        RRDSET *st;
        RRDDIM *rd;
@@ -714,12 +704,10 @@ RRDSET *rrdset_find_bytype(const char *type, const char *id)
 
        char buf[RRD_ID_LENGTH_MAX + 1];
 
-       strncpy(buf, type, RRD_ID_LENGTH_MAX - 1);
-       buf[RRD_ID_LENGTH_MAX - 1] = '\0';
+       strncpyz(buf, type, RRD_ID_LENGTH_MAX - 1);
        strcat(buf, ".");
        int len = (int) strlen(buf);
-       strncpy(&buf[len], id, (size_t) (RRD_ID_LENGTH_MAX - len));
-       buf[RRD_ID_LENGTH_MAX] = '\0';
+       strncpyz(&buf[len], id, (size_t) (RRD_ID_LENGTH_MAX - len));
 
        return(rrdset_find(buf));
 }
@@ -823,6 +811,8 @@ void rrdset_next_plugins(RRDSET *st)
 
 unsigned long long rrdset_done(RRDSET *st)
 {
+       if(unlikely(netdata_exit)) return 0;
+
        debug(D_RRD_CALLS, "rrdset_done() for chart %s", st->name);
 
        RRDDIM *rd, *last;
@@ -886,7 +876,7 @@ unsigned long long rrdset_done(RRDSET *st)
 
        // check if we will re-write the entire data set
        if(unlikely(usecdiff(&st->last_collected_time, &st->last_updated) > st->update_every * st->entries * 1000000ULL)) {
-               info("%s: too old data (last updated at %u.%u, last collected at %u.%u). Reseting it. Will not store the next entry.", st->name, st->last_updated.tv_sec, st->last_updated.tv_usec, st->last_collected_time.tv_sec, st->last_collected_time.tv_usec);
+               info("%s: too old data (last updated at %zu.%zu, last collected at %zu.%zu). Reseting it. Will not store the next entry.", st->name, st->last_updated.tv_sec, st->last_updated.tv_usec, st->last_collected_time.tv_sec, st->last_collected_time.tv_usec);
                rrdset_reset(st);
 
                st->usec_since_last_update = st->update_every * 1000000ULL;