X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=src%2Futils%2Ferror.c;h=f40e31ada656f24e4719783bbe279cf49cfb7be2;hp=6bb17d5821f5ea68259606cbc6bcd8312d55b04e;hb=56b97a180a53aecbfe9f7162b8ece3faae973cf9;hpb=b7cd97981cd4d5534cd0437d27564899047f0ac7

diff --git a/src/utils/error.c b/src/utils/error.c
index 6bb17d5..f40e31a 100644
--- a/src/utils/error.c
+++ b/src/utils/error.c
@@ -25,7 +25,12 @@
  * 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>
 
@@ -41,10 +46,10 @@
 
 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