Code

plugin: Make sdb_plugin_info_t public.
[sysdb.git] / src / utils / error.c
index 6bb17d5821f5ea68259606cbc6bcd8312d55b04e..f40e31ada656f24e4719783bbe279cf49cfb7be2 100644 (file)
  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
+#if HAVE_CONFIG_H
+#      include "config.h"
+#endif /* HAVE_CONFIG_H */
+
 #include "utils/error.h"
+#include "utils/strbuf.h"
 
 #include <pthread.h>
 
 
 typedef struct {
        int   prio;
-       char  msg[SDB_MAX_ERROR];
+       sdb_strbuf_t *msg;
        _Bool logged;
 } sdb_error_ctx_t;
-#define SDB_ERROR_INIT { -1, "", 1 }
+#define SDB_ERROR_INIT { -1, NULL, 1 }
 
 /*
  * private variables
@@ -55,15 +60,21 @@ static sdb_error_ctx_t default_error_ctx = SDB_ERROR_INIT;
 static pthread_key_t error_ctx_key;
 static _Bool         error_ctx_key_initialized = 0;
 
+static int (*logger)(int prio, const char *msg) = NULL;
+
 /*
  * private helper functions
  */
 
 static void
-sdb_error_ctx_destructor(void *ctx)
+sdb_error_ctx_destructor(void *p)
 {
+       sdb_error_ctx_t *ctx = p;
+
        if (! ctx)
                return;
+
+       sdb_strbuf_destroy(ctx->msg);
        free(ctx);
 } /* sdb_error_ctx_destructor */
 
@@ -87,6 +98,11 @@ sdb_error_ctx_create(void)
                return NULL;
 
        *ctx = default_error_ctx;
+       ctx->msg = sdb_strbuf_create(64);
+       if (! ctx->msg) {
+               free(ctx);
+               return NULL;
+       }
 
        if (! error_ctx_key_initialized)
                sdb_error_ctx_init();
@@ -111,7 +127,7 @@ sdb_error_get_ctx(void)
 } /* sdb_error_get_ctx */
 
 static int
-sdb_error_clear(void)
+sdb_error_vprintf(const char *fmt, va_list ap)
 {
        sdb_error_ctx_t *ctx;
 
@@ -119,28 +135,21 @@ sdb_error_clear(void)
        if (! ctx)
                return -1;
 
-       ctx->prio = -1;
-       ctx->msg[0] = '\0';
-       ctx->logged = 1;
-       return 0;
-} /* sdb_error_clear */
+       ctx->logged = 0;
+       return (int)sdb_strbuf_vsprintf(ctx->msg, fmt, ap);
+} /* sdb_error_vprintf */
 
 static int
 sdb_error_vappend(const char *fmt, va_list ap)
 {
        sdb_error_ctx_t *ctx;
-       size_t len;
 
        ctx = sdb_error_get_ctx();
        if (! ctx)
                return -1;
 
-       len = strlen(ctx->msg);
-       if (len >= sizeof(ctx->msg))
-               return 0; /* nothing written */
-
        ctx->logged = 0;
-       return vsnprintf(ctx->msg + len, sizeof(ctx->msg) - len, fmt, ap);
+       return (int)sdb_strbuf_vappend(ctx->msg, fmt, ap);
 } /* sdb_error_vappend */
 
 static int
@@ -159,8 +168,12 @@ sdb_do_log(int prio)
        if (ctx->logged)
                return 0;
 
-       ret = fprintf(stderr, "[%s] %s\n",
-                       SDB_LOG_PRIO_TO_STRING(prio), ctx->msg);
+       if (logger)
+               ret = logger(prio, sdb_strbuf_string(ctx->msg));
+       else
+               ret = fprintf(stderr, "[%s] %s\n", SDB_LOG_PRIO_TO_STRING(prio),
+                               sdb_strbuf_string(ctx->msg));
+
        ctx->logged = 1;
        return ret;
 } /* sdb_do_log */
@@ -169,20 +182,24 @@ sdb_do_log(int prio)
  * public API
  */
 
+void
+sdb_error_set_logger(int (*f)(int, const char *))
+{
+       logger = f;
+} /* sdb_error_set_logger */
+
 int
 sdb_log(int prio, const char *fmt, ...)
 {
        va_list ap;
        int ret;
 
-       if (sdb_error_clear())
-               return -1;
-
        va_start(ap, fmt);
-       ret = sdb_error_vappend(fmt, ap);
+       ret = sdb_error_vprintf(fmt, ap);
        va_end(ap);
 
-       sdb_do_log(prio);
+       if (ret > 0)
+               sdb_do_log(prio);
        return ret;
 } /* sdb_log */
 
@@ -192,11 +209,8 @@ sdb_error_set(const char *fmt, ...)
        va_list ap;
        int ret;
 
-       if (sdb_error_clear())
-               return -1;
-
        va_start(ap, fmt);
-       ret = sdb_error_vappend(fmt, ap);
+       ret = sdb_error_vprintf(fmt, ap);
        va_end(ap);
 
        return ret;
@@ -215,6 +229,19 @@ sdb_error_append(const char *fmt, ...)
        return ret;
 } /* sdb_error_append */
 
+int
+sdb_error_chomp(void)
+{
+       sdb_error_ctx_t *ctx;
+
+       ctx = sdb_error_get_ctx();
+       if (! ctx)
+               return -1;
+
+       sdb_strbuf_chomp(ctx->msg);
+       return 0;
+} /* sdb_error_chomp */
+
 int
 sdb_error_log(int prio)
 {
@@ -229,7 +256,7 @@ sdb_error_get(void)
        ctx = sdb_error_get_ctx();
        if (! ctx)
                return "success";
-       return ctx->msg;
+       return sdb_strbuf_string(ctx->msg);
 } /* sdb_error_get */
 
 int