1 /*
2 * SysDB - src/utils/error.c
3 * Copyright (C) 2013 Sebastian 'tokkee' Harl <sh@tokkee.org>
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
17 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR
19 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
28 #include "utils/error.h"
30 #include <pthread.h>
32 #include <stdarg.h>
33 #include <stdio.h>
34 #include <string.h>
36 #include <stdlib.h>
38 /*
39 * private data types
40 */
42 typedef struct {
43 int prio;
44 char msg[SDB_MAX_ERROR];
45 _Bool finalized;
46 } sdb_error_ctx_t;
47 #define SDB_ERROR_INIT { -1, "", 1 }
49 /*
50 * private variables
51 */
53 static sdb_error_ctx_t default_error_ctx = SDB_ERROR_INIT;
55 static pthread_key_t error_ctx_key;
56 static _Bool error_ctx_key_initialized = 0;
58 /*
59 * private helper functions
60 */
62 static void
63 sdb_error_ctx_destructor(void *ctx)
64 {
65 if (! ctx)
66 return;
67 free(ctx);
68 } /* sdb_error_ctx_destructor */
70 static void
71 sdb_error_ctx_init(void)
72 {
73 if (error_ctx_key_initialized)
74 return;
76 pthread_key_create(&error_ctx_key, sdb_error_ctx_destructor);
77 error_ctx_key_initialized = 1;
78 } /* sdb_error_init */
80 static sdb_error_ctx_t *
81 sdb_error_ctx_create(void)
82 {
83 sdb_error_ctx_t *ctx;
85 ctx = malloc(sizeof(*ctx));
86 if (! ctx)
87 return NULL;
89 *ctx = default_error_ctx;
91 if (! error_ctx_key_initialized)
92 sdb_error_ctx_init();
93 pthread_setspecific(error_ctx_key, ctx);
94 return ctx;
95 } /* sdb_error_ctx_create */
97 static sdb_error_ctx_t *
98 sdb_error_get_ctx(void)
99 {
100 sdb_error_ctx_t *ctx;
102 if (! error_ctx_key_initialized)
103 sdb_error_ctx_init();
104 ctx = pthread_getspecific(error_ctx_key);
106 if (! ctx)
107 ctx = sdb_error_ctx_create();
108 if (! ctx)
109 return NULL;
110 return ctx;
111 } /* sdb_error_get_ctx */
113 static int
114 sdb_error_clear(void)
115 {
116 sdb_error_ctx_t *ctx;
118 ctx = sdb_error_get_ctx();
119 if (! ctx)
120 return -1;
122 ctx->prio = -1;
123 ctx->msg[0] = '\0';
124 ctx->finalized = 1;
125 return 0;
126 } /* sdb_error_clear */
128 static int
129 sdb_error_vappend(int prio, const char *fmt, va_list ap)
130 {
131 sdb_error_ctx_t *ctx;
132 size_t len;
134 ctx = sdb_error_get_ctx();
135 if (! ctx)
136 return -1;
138 len = strlen(ctx->msg);
139 if (len >= sizeof(ctx->msg))
140 return 0; /* nothing written */
142 if (prio >= 0)
143 ctx->prio = prio;
145 ctx->finalized = 0;
146 return vsnprintf(ctx->msg + len, sizeof(ctx->msg) - len, fmt, ap);
147 } /* sdb_error_vappend */
149 static int
150 sdb_error_log(void)
151 {
152 sdb_error_ctx_t *ctx;
153 int ret;
155 ctx = sdb_error_get_ctx();
156 if (! ctx)
157 return -1;
159 if (ctx->finalized)
160 return 0;
162 ret = fprintf(stderr, "[%s] %s\n",
163 SDB_LOG_PRIO_TO_STRING(ctx->prio), ctx->msg);
164 ctx->finalized = 1;
165 return ret;
166 } /* sdb_error_log */
168 /*
169 * public API
170 */
172 int
173 sdb_error_set(int prio, const char *fmt, ...)
174 {
175 va_list ap;
176 int ret;
178 if (sdb_error_clear())
179 return -1;
181 va_start(ap, fmt);
182 ret = sdb_error_vappend(prio, fmt, ap);
183 va_end(ap);
185 sdb_error_log();
186 return ret;
187 } /* sdb_error_set */
189 int
190 sdb_error_start(int prio, const char *fmt, ...)
191 {
192 va_list ap;
193 int ret;
195 if (sdb_error_clear())
196 return -1;
198 va_start(ap, fmt);
199 ret = sdb_error_vappend(prio, fmt, ap);
200 va_end(ap);
202 return ret;
203 } /* sdb_error_start */
205 int
206 sdb_error_append(const char *fmt, ...)
207 {
208 va_list ap;
209 int ret;
211 va_start(ap, fmt);
212 ret = sdb_error_vappend(/* dont change prio */ -1, fmt, ap);
213 va_end(ap);
215 return ret;
216 } /* sdb_error_append */
218 int
219 sdb_error_finish(void)
220 {
221 return sdb_error_log();
222 } /* sdb_error_finish */
224 const char *
225 sdb_error_get(void)
226 {
227 sdb_error_ctx_t *ctx;
229 ctx = sdb_error_get_ctx();
230 if (! ctx)
231 return "success";
232 return ctx->msg;
233 } /* sdb_error_get */
235 int
236 sdb_error_get_prio(void)
237 {
238 sdb_error_ctx_t *ctx;
240 ctx = sdb_error_get_ctx();
241 if (! ctx)
242 return -1;
243 return ctx->prio;
244 } /* sdb_error_get_prio */
246 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */