X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=src%2Fcore%2Fstore.c;h=e5cd886aae31be9bb1d35be92e9f42b7d729e4b5;hp=162a3fc2d7c736d97c5bd406e609a130f6ff2228;hb=bef7167f0dc1fd405e35d5cbffb3c0820945a9ea;hpb=d170cd34bb8b4f135ec41c622e6a09ff83ff1011 diff --git a/src/core/store.c b/src/core/store.c index 162a3fc..e5cd886 100644 --- a/src/core/store.c +++ b/src/core/store.c @@ -1,5 +1,5 @@ /* - * syscollector - src/core/store.c + * SysDB - src/core/store.c * Copyright (C) 2012 Sebastian 'tokkee' Harl * All rights reserved. * @@ -25,7 +25,7 @@ * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ -#include "syscollector.h" +#include "sysdb.h" #include "core/store.h" #include "utils/llist.h" #include "utils/string.h" @@ -40,11 +40,22 @@ #include +/* + * private data types + */ + +/* type used for looking up named objects */ +typedef struct { + sdb_object_t parent; + const char *obj_name; +} sdb_store_lookup_obj_t; +#define SDB_STORE_LOOKUP_OBJ_INIT { SDB_OBJECT_INIT, NULL } + /* * private variables */ -static sc_llist_t *host_list = NULL; +static sdb_llist_t *host_list = NULL; static pthread_rwlock_t host_lock = PTHREAD_RWLOCK_INITIALIZER; /* @@ -52,119 +63,176 @@ static pthread_rwlock_t host_lock = PTHREAD_RWLOCK_INITIALIZER; */ static int -sc_store_obj_cmp_by_name(const sc_object_t *a, const sc_object_t *b) +sdb_store_obj_cmp_by_name(const sdb_object_t *a, const sdb_object_t *b) { - const sc_store_obj_t *h1 = (const sc_store_obj_t *)a; - const sc_store_obj_t *h2 = (const sc_store_obj_t *)b; + const sdb_store_obj_t *h1 = (const sdb_store_obj_t *)a; + const sdb_store_obj_t *h2 = (const sdb_store_obj_t *)b; assert(h1 && h2); return strcasecmp(h1->name, h2->name); -} /* sc_store_obj_cmp_by_name */ +} /* sdb_store_obj_cmp_by_name */ + +static int +sdb_cmp_store_obj_with_name(const sdb_object_t *a, const sdb_object_t *b) +{ + const sdb_store_obj_t *obj = (const sdb_store_obj_t *)a; + const sdb_store_lookup_obj_t *lookup = (const sdb_store_lookup_obj_t *)b; + + assert(obj && lookup); + return strcasecmp(obj->name, lookup->obj_name); +} /* sdb_cmp_store_obj_with_name */ static int -sc_host_init(sc_object_t *obj, va_list ap) +sdb_host_init(sdb_object_t *obj, va_list ap) { - char *name = va_arg(ap, char *); + const char *name = va_arg(ap, const char *); - SC_HOST(obj)->host_name = strdup(name); - if (! SC_HOST(obj)->host_name) + SDB_HOST(obj)->host_name = strdup(name); + if (! SDB_HOST(obj)->host_name) return -1; - SC_HOST(obj)->host_last_update = sc_gettime(); + SDB_HOST(obj)->host_last_update = sdb_gettime(); /* ignore errors -> last_update will be updated later */ - SC_HOST(obj)->services = sc_llist_create(); - if (! SC_HOST(obj)->services) + SDB_HOST(obj)->attributes = sdb_llist_create(); + if (! SDB_HOST(obj)->attributes) + return -1; + SDB_HOST(obj)->services = sdb_llist_create(); + if (! SDB_HOST(obj)->services) return -1; return 0; -} /* sc_host_init */ +} /* sdb_host_init */ static void -sc_host_destroy(sc_object_t *obj) +sdb_host_destroy(sdb_object_t *obj) { assert(obj); - if (SC_HOST(obj)->host_name) - free(SC_HOST(obj)->host_name); + if (SDB_HOST(obj)->host_name) + free(SDB_HOST(obj)->host_name); - if (SC_HOST(obj)->services) - sc_llist_destroy(SC_HOST(obj)->services); -} /* sc_host_destroy */ + if (SDB_HOST(obj)->attributes) + sdb_llist_destroy(SDB_HOST(obj)->attributes); + if (SDB_HOST(obj)->services) + sdb_llist_destroy(SDB_HOST(obj)->services); +} /* sdb_host_destroy */ static int -sc_svc_init(sc_object_t *obj, va_list ap) +sdb_attr_init(sdb_object_t *obj, va_list ap) { - char *hostname = va_arg(ap, char *); - char *name = va_arg(ap, char *); + const char *hostname = va_arg(ap, const char *); + const char *name = va_arg(ap, const char *); + const char *value = va_arg(ap, const char *); + + SDB_ATTR(obj)->hostname = strdup(hostname); + SDB_ATTR(obj)->attr_name = strdup(name); + SDB_ATTR(obj)->attr_value = strdup(value); + if ((! SDB_ATTR(obj)->hostname) + || (! SDB_ATTR(obj)->attr_name) || (! SDB_ATTR(obj)->attr_value)) + return -1; - SC_SVC(obj)->hostname = strdup(hostname); - SC_SVC(obj)->svc_name = strdup(name); - if ((! SC_SVC(obj)->hostname) || (! SC_SVC(obj)->svc_name)) + SDB_ATTR(obj)->attr_last_update = sdb_gettime(); + return 0; +} /* sdb_attr_init */ + +static void +sdb_attr_destroy(sdb_object_t *obj) +{ + assert(obj); + + if (SDB_ATTR(obj)->hostname) + free(SDB_ATTR(obj)->hostname); + if (SDB_ATTR(obj)->attr_name) + free(SDB_ATTR(obj)->attr_name); + if (SDB_ATTR(obj)->attr_value) + free(SDB_ATTR(obj)->attr_value); +} /* sdb_attr_destroy */ + +static int +sdb_svc_init(sdb_object_t *obj, va_list ap) +{ + const char *hostname = va_arg(ap, const char *); + const char *name = va_arg(ap, const char *); + + SDB_SVC(obj)->hostname = strdup(hostname); + SDB_SVC(obj)->svc_name = strdup(name); + if ((! SDB_SVC(obj)->hostname) || (! SDB_SVC(obj)->svc_name)) return -1; - SC_SVC(obj)->svc_last_update = sc_gettime(); + SDB_SVC(obj)->svc_last_update = sdb_gettime(); /* ignore errors -> last_update will be updated later */ return 0; -} /* sc_svc_init */ +} /* sdb_svc_init */ static void -sc_svc_destroy(sc_object_t *obj) +sdb_svc_destroy(sdb_object_t *obj) { assert(obj); - if (SC_SVC(obj)->hostname) - free(SC_SVC(obj)->hostname); - if (SC_SVC(obj)->svc_name) - free(SC_SVC(obj)->svc_name); -} /* sc_svc_destroy */ + if (SDB_SVC(obj)->hostname) + free(SDB_SVC(obj)->hostname); + if (SDB_SVC(obj)->svc_name) + free(SDB_SVC(obj)->svc_name); +} /* sdb_svc_destroy */ /* * public API */ -sc_host_t * -sc_host_create(char *name) +sdb_host_t * +sdb_host_create(const char *name) { - sc_object_t *obj; + sdb_object_t *obj; if (! name) return NULL; - obj = sc_object_create(sizeof(sc_host_t), sc_host_init, - sc_host_destroy, name); + obj = sdb_object_create(sizeof(sdb_host_t), sdb_host_init, + sdb_host_destroy, name); if (! obj) return NULL; - return SC_HOST(obj); -} /* sc_host_create */ + return SDB_HOST(obj); +} /* sdb_host_create */ -sc_host_t * -sc_host_clone(const sc_host_t *host) +sdb_host_t * +sdb_host_clone(const sdb_host_t *host) { - sc_host_t *clone; + sdb_host_t *clone; - clone = sc_host_create(host->host_name); + clone = sdb_host_create(host->host_name); if (! clone) return NULL; + /* make sure these are initialized; else sdb_object_deref() might access + * arbitrary memory in case of an error */ + clone->services = clone->attributes = NULL; + + if (host->attributes) { + clone->attributes = sdb_llist_clone(host->attributes); + if (! clone->attributes) { + sdb_object_deref(SDB_OBJ(clone)); + return NULL; + } + } + clone->host_last_update = host->host_last_update; if (host->services) { - clone->services = sc_llist_clone(host->services); + clone->services = sdb_llist_clone(host->services); if (! clone->services) { - sc_object_deref(SC_OBJ(clone)); + sdb_object_deref(SDB_OBJ(clone)); return NULL; } } - else - clone->services = NULL; return clone; -} /* sc_host_clone */ +} /* sdb_host_clone */ int -sc_store_host(const sc_host_t *host) +sdb_store_host(const sdb_host_t *host) { - sc_time_t last_update; + sdb_store_lookup_obj_t lookup = SDB_STORE_LOOKUP_OBJ_INIT; + sdb_time_t last_update; - sc_host_t *old; + sdb_host_t *old; int status = 0; if ((! host) || (! host->host_name)) @@ -172,19 +240,20 @@ sc_store_host(const sc_host_t *host) last_update = host->host_last_update; if (last_update <= 0) - last_update = sc_gettime(); + last_update = sdb_gettime(); pthread_rwlock_wrlock(&host_lock); if (! host_list) { - if (! (host_list = sc_llist_create())) { + if (! (host_list = sdb_llist_create())) { pthread_rwlock_unlock(&host_lock); return -1; } } - old = SC_HOST(sc_llist_search(host_list, (const sc_object_t *)host, - sc_store_obj_cmp_by_name)); + lookup.obj_name = host->host_name; + old = SDB_HOST(sdb_llist_search(host_list, (const sdb_object_t *)&lookup, + sdb_cmp_store_obj_with_name)); if (old) { if (old->host_last_update > last_update) { @@ -200,91 +269,206 @@ sc_store_host(const sc_host_t *host) } } else { - sc_host_t *new = sc_host_clone(host); + sdb_host_t *new = sdb_host_clone(host); if (! new) { char errbuf[1024]; fprintf(stderr, "store: Failed to clone host object: %s\n", - sc_strerror(errno, errbuf, sizeof(errbuf))); + sdb_strerror(errno, errbuf, sizeof(errbuf))); + pthread_rwlock_unlock(&host_lock); return -1; } + if (! new->attributes) { + if (! (new->attributes = sdb_llist_create())) { + char errbuf[1024]; + fprintf(stderr, "store: Failed to initialize " + "host object '%s': %s\n", host->host_name, + sdb_strerror(errno, errbuf, sizeof(errbuf))); + sdb_object_deref(SDB_OBJ(new)); + pthread_rwlock_unlock(&host_lock); + return -1; + } + } + if (! new->services) { - if (! (new->services = sc_llist_create())) { + if (! (new->services = sdb_llist_create())) { char errbuf[1024]; fprintf(stderr, "store: Failed to initialize " "host object '%s': %s\n", host->host_name, - sc_strerror(errno, errbuf, sizeof(errbuf))); - sc_object_deref(SC_OBJ(new)); + sdb_strerror(errno, errbuf, sizeof(errbuf))); + sdb_object_deref(SDB_OBJ(new)); + pthread_rwlock_unlock(&host_lock); return -1; } } - status = sc_llist_insert_sorted(host_list, SC_OBJ(new), - sc_store_obj_cmp_by_name); + status = sdb_llist_insert_sorted(host_list, SDB_OBJ(new), + sdb_store_obj_cmp_by_name); /* pass control to the list or destroy in case of an error */ - sc_object_deref(SC_OBJ(new)); + sdb_object_deref(SDB_OBJ(new)); } pthread_rwlock_unlock(&host_lock); return status; -} /* sc_store_host */ +} /* sdb_store_host */ -const sc_host_t * -sc_store_get_host(char *name) +const sdb_host_t * +sdb_store_get_host(const char *name) { - sc_host_t tmp = SC_HOST_INIT; - sc_host_t *host; + sdb_store_lookup_obj_t lookup = SDB_STORE_LOOKUP_OBJ_INIT; + sdb_host_t *host; if (! name) return NULL; - tmp.host_name = name; - host = SC_HOST(sc_llist_search(host_list, (const sc_object_t *)&tmp, - sc_store_obj_cmp_by_name)); + lookup.obj_name = name; + host = SDB_HOST(sdb_llist_search(host_list, (const sdb_object_t *)&lookup, + sdb_cmp_store_obj_with_name)); if (! host) return NULL; return host; -} /* sc_store_get_host */ +} /* sdb_store_get_host */ + +sdb_attribute_t * +sdb_attribute_create(const char *hostname, + const char *name, const char *value) +{ + sdb_object_t *obj; + + if ((! hostname) || (! name) || (! value)) + return NULL; -sc_service_t * -sc_service_create(char *hostname, char *name) + obj = sdb_object_create(sizeof(sdb_attribute_t), sdb_attr_init, + sdb_attr_destroy, hostname, name, value); + if (! obj) + return NULL; + return SDB_ATTR(obj); +} /* sdb_attribute_create */ + +sdb_attribute_t * +sdb_attribute_clone(const sdb_attribute_t *attr) { - sc_object_t *obj; + sdb_attribute_t *clone; + + clone = sdb_attribute_create(attr->hostname, + attr->attr_name, attr->attr_value); + if (! clone) + return NULL; + + clone->attr_last_update = attr->attr_last_update; + return clone; +} /* sdb_attribute_clone */ + +int +sdb_store_attribute(const sdb_attribute_t *attr) +{ + sdb_store_lookup_obj_t lookup = SDB_STORE_LOOKUP_OBJ_INIT; + sdb_host_t *host; + + sdb_attribute_t *old; + + sdb_time_t last_update; + + int status = 0; + + if (! attr) + return -1; + + last_update = attr->attr_last_update; + if (last_update <= 0) + last_update = sdb_gettime(); + + if (! host_list) + return -1; + + pthread_rwlock_wrlock(&host_lock); + + lookup.obj_name = attr->hostname; + host = SDB_HOST(sdb_llist_search(host_list, (const sdb_object_t *)&lookup, + sdb_cmp_store_obj_with_name)); + + if (! host) { + pthread_rwlock_unlock(&host_lock); + return -1; + } + + lookup.obj_name = attr->attr_name; + old = SDB_ATTR(sdb_llist_search(host->attributes, + (const sdb_object_t *)&lookup, + sdb_cmp_store_obj_with_name)); + + if (old) { + if (old->host_last_update > last_update) { + fprintf(stderr, "store: Cannot update attribute '%s/%s' - " + "value too old (%"PRIscTIME" < %"PRIscTIME")\n", + attr->hostname, attr->attr_name, last_update, + old->host_last_update); + status = 1; + } + else { + old->attr_last_update = last_update; + } + } + else { + sdb_attribute_t *new = sdb_attribute_clone(attr); + if (! new) { + char errbuf[1024]; + fprintf(stderr, "store: Failed to clone attribute object: %s\n", + sdb_strerror(errno, errbuf, sizeof(errbuf))); + pthread_rwlock_unlock(&host_lock); + return -1; + } + + status = sdb_llist_insert_sorted(host->attributes, SDB_OBJ(new), + sdb_store_obj_cmp_by_name); + + /* pass control to the list or destroy in case of an error */ + sdb_object_deref(SDB_OBJ(new)); + } + + pthread_rwlock_unlock(&host_lock); + return status; +} /* sdb_store_attribute */ + +sdb_service_t * +sdb_service_create(const char *hostname, const char *name) +{ + sdb_object_t *obj; if ((! hostname) || (! name)) return NULL; - obj = sc_object_create(sizeof(sc_service_t), sc_svc_init, - sc_svc_destroy, hostname, name); + obj = sdb_object_create(sizeof(sdb_service_t), sdb_svc_init, + sdb_svc_destroy, hostname, name); if (! obj) return NULL; - return SC_SVC(obj); -} /* sc_service_create */ + return SDB_SVC(obj); +} /* sdb_service_create */ -sc_service_t * -sc_service_clone(const sc_service_t *svc) +sdb_service_t * +sdb_service_clone(const sdb_service_t *svc) { - sc_service_t *clone; + sdb_service_t *clone; - clone = sc_service_create(svc->hostname, svc->svc_name); + clone = sdb_service_create(svc->hostname, svc->svc_name); if (! clone) return NULL; clone->svc_last_update = svc->svc_last_update; return clone; -} /* sc_service_clone */ +} /* sdb_service_clone */ int -sc_store_service(const sc_service_t *svc) +sdb_store_service(const sdb_service_t *svc) { - sc_host_t tmp = SC_HOST_INIT; - sc_host_t *host; + sdb_store_lookup_obj_t lookup = SDB_STORE_LOOKUP_OBJ_INIT; + sdb_host_t *host; - sc_service_t *old; + sdb_service_t *old; - sc_time_t last_update; + sdb_time_t last_update; int status = 0; @@ -293,22 +477,25 @@ sc_store_service(const sc_service_t *svc) last_update = svc->svc_last_update; if (last_update <= 0) - last_update = sc_gettime(); + last_update = sdb_gettime(); if (! host_list) return -1; pthread_rwlock_wrlock(&host_lock); - tmp.host_name = svc->hostname; - host = SC_HOST(sc_llist_search(host_list, (const sc_object_t *)&tmp, - sc_store_obj_cmp_by_name)); + lookup.obj_name = svc->hostname; + host = SDB_HOST(sdb_llist_search(host_list, (const sdb_object_t *)&lookup, + sdb_cmp_store_obj_with_name)); - if (! host) + if (! host) { + pthread_rwlock_unlock(&host_lock); return -1; + } - old = SC_SVC(sc_llist_search(host->services, (const sc_object_t *)svc, - sc_store_obj_cmp_by_name)); + lookup.obj_name = svc->svc_name; + old = SDB_SVC(sdb_llist_search(host->services, (const sdb_object_t *)&lookup, + sdb_cmp_store_obj_with_name)); if (old) { if (old->host_last_update > last_update) { @@ -323,66 +510,71 @@ sc_store_service(const sc_service_t *svc) } } else { - sc_service_t *new = sc_service_clone(svc); + sdb_service_t *new = sdb_service_clone(svc); if (! new) { char errbuf[1024]; fprintf(stderr, "store: Failed to clone service object: %s\n", - sc_strerror(errno, errbuf, sizeof(errbuf))); + sdb_strerror(errno, errbuf, sizeof(errbuf))); + pthread_rwlock_unlock(&host_lock); return -1; } - status = sc_llist_insert_sorted(host->services, SC_OBJ(new), - sc_store_obj_cmp_by_name); + status = sdb_llist_insert_sorted(host->services, SDB_OBJ(new), + sdb_store_obj_cmp_by_name); /* pass control to the list or destroy in case of an error */ - sc_object_deref(SC_OBJ(new)); + sdb_object_deref(SDB_OBJ(new)); } pthread_rwlock_unlock(&host_lock); return status; -} /* sc_store_service */ +} /* sdb_store_service */ -const sc_service_t * -sc_store_get_service(const sc_host_t *host, char *name) +const sdb_service_t * +sdb_store_get_service(const sdb_host_t *host, const char *name) { - sc_service_t tmp = SC_SVC_INIT; - sc_service_t *svc; + sdb_store_lookup_obj_t lookup = SDB_STORE_LOOKUP_OBJ_INIT; + sdb_service_t *svc; if ((! host) || (! name)) return NULL; - tmp.svc_name = name; - svc = SC_SVC(sc_llist_search(host->services, (const sc_object_t *)&tmp, - sc_store_obj_cmp_by_name)); + lookup.obj_name = name; + svc = SDB_SVC(sdb_llist_search(host->services, + (const sdb_object_t *)&lookup, + sdb_cmp_store_obj_with_name)); if (! svc) return NULL; return svc; -} /* sc_store_get_service */ +} /* sdb_store_get_service */ int -sc_store_dump(FILE *fh) +sdb_store_dump(FILE *fh) { - sc_llist_iter_t *host_iter; + sdb_llist_iter_t *host_iter; if (! fh) return -1; pthread_rwlock_rdlock(&host_lock); - host_iter = sc_llist_get_iter(host_list); - if (! host_iter) + host_iter = sdb_llist_get_iter(host_list); + if (! host_iter) { + pthread_rwlock_unlock(&host_lock); return -1; + } - while (sc_llist_iter_has_next(host_iter)) { - sc_host_t *host = SC_HOST(sc_llist_iter_get_next(host_iter)); - sc_llist_iter_t *svc_iter; + while (sdb_llist_iter_has_next(host_iter)) { + sdb_host_t *host = SDB_HOST(sdb_llist_iter_get_next(host_iter)); + sdb_llist_iter_t *svc_iter; + sdb_llist_iter_t *attr_iter; char time_str[64]; assert(host); - if (! sc_strftime(time_str, sizeof(time_str), + if (! sdb_strftime(time_str, sizeof(time_str), "%F %T %z", host->host_last_update)) snprintf(time_str, sizeof(time_str), ""); time_str[sizeof(time_str) - 1] = '\0'; @@ -390,20 +582,43 @@ sc_store_dump(FILE *fh) fprintf(fh, "Host '%s' (last updated: %s):\n", host->host_name, time_str); - svc_iter = sc_llist_get_iter(host->services); + attr_iter = sdb_llist_get_iter(host->attributes); + if (! attr_iter) { + char errbuf[1024]; + fprintf(fh, "Failed to retrieve attributes: %s\n", + sdb_strerror(errno, errbuf, sizeof(errbuf))); + continue; + } + + while (sdb_llist_iter_has_next(attr_iter)) { + sdb_attribute_t *attr = SDB_ATTR(sdb_llist_iter_get_next(attr_iter)); + assert(attr); + + if (! sdb_strftime(time_str, sizeof(time_str), + "%F %T %z", attr->attr_last_update)) + snprintf(time_str, sizeof(time_str), ""); + time_str[sizeof(time_str) - 1] = '\0'; + + fprintf(fh, "\tAttribute '%s' -> '%s' (last updated: %s)\n", + attr->attr_name, attr->attr_value, time_str); + } + + sdb_llist_iter_destroy(attr_iter); + + svc_iter = sdb_llist_get_iter(host->services); if (! svc_iter) { char errbuf[1024]; fprintf(fh, "Failed to retrieve services: %s\n", - sc_strerror(errno, errbuf, sizeof(errbuf))); + sdb_strerror(errno, errbuf, sizeof(errbuf))); continue; } - while (sc_llist_iter_has_next(svc_iter)) { - sc_service_t *svc = SC_SVC(sc_llist_iter_get_next(svc_iter)); + while (sdb_llist_iter_has_next(svc_iter)) { + sdb_service_t *svc = SDB_SVC(sdb_llist_iter_get_next(svc_iter)); assert(svc); - if (! sc_strftime(time_str, sizeof(time_str), - "%F %T %z", host->host_last_update)) + if (! sdb_strftime(time_str, sizeof(time_str), + "%F %T %z", svc->svc_last_update)) snprintf(time_str, sizeof(time_str), ""); time_str[sizeof(time_str) - 1] = '\0'; @@ -411,12 +626,13 @@ sc_store_dump(FILE *fh) svc->svc_name, time_str); } - sc_llist_iter_destroy(svc_iter); + sdb_llist_iter_destroy(svc_iter); } - sc_llist_iter_destroy(host_iter); + sdb_llist_iter_destroy(host_iter); + pthread_rwlock_unlock(&host_lock); return 0; -} /* sc_store_dump */ +} /* sdb_store_dump */ /* vim: set tw=78 sw=4 ts=4 noexpandtab : */