1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright 2016, Olivier MATZ <zer0@droids-corp.org>
5 #define _GNU_SOURCE /* for vasprintf */
12 #include <ecoli_malloc.h>
13 #include <ecoli_string.h>
14 #include <ecoli_log.h>
15 #include <ecoli_test.h>
17 EC_LOG_TYPE_REGISTER(log);
19 static ec_log_t ec_log_fct = ec_log_default_cb;
20 static void *ec_log_opaque;
24 enum ec_log_level level;
27 static struct ec_log_type *log_types;
28 static size_t log_types_len;
29 static enum ec_log_level global_level = EC_LOG_WARNING;
31 int ec_log_level_set(enum ec_log_level level)
33 if (level < 0 || level > EC_LOG_DEBUG)
40 enum ec_log_level ec_log_level_get(void)
45 int ec_log_default_cb(int type, enum ec_log_level level, void *opaque,
50 if (level > ec_log_level_get())
53 if (fprintf(stderr, "[%d] %-12s %s", level, ec_log_name(type), str) < 0)
59 int ec_log_fct_register(ec_log_t usr_log, void *opaque)
63 if (usr_log == NULL) {
64 ec_log_fct = ec_log_default_cb;
68 ec_log_opaque = opaque;
75 ec_log_lookup(const char *name)
79 for (i = 0; i < log_types_len; i++) {
80 if (log_types[i].name == NULL)
82 if (strcmp(name, log_types[i].name) == 0)
90 ec_log_type_register(const char *name)
92 struct ec_log_type *new_types;
96 id = ec_log_lookup(name);
100 new_types = ec_realloc(log_types,
101 sizeof(*new_types) * (log_types_len + 1));
102 if (new_types == NULL)
103 return -1; /* errno is set */
104 log_types = new_types;
106 copy = ec_strdup(name);
108 return -1; /* errno is set */
110 id = log_types_len++;
111 log_types[id].name = copy;
112 log_types[id].level = EC_LOG_DEBUG;
118 ec_log_name(int type)
120 if (type < 0 || (unsigned int)type >= log_types_len)
122 return log_types[type].name;
125 int ec_vlog(int type, enum ec_log_level level, const char *format, va_list ap)
130 /* don't use ec_vasprintf here, because it will call
131 * ec_malloc(), then ec_log(), ec_vasprintf()...
132 * -> stack overflow */
133 ret = vasprintf(&s, format, ap);
137 ret = ec_log_fct(type, level, ec_log_opaque, s);
143 int ec_log(int type, enum ec_log_level level, const char *format, ...)
148 va_start(ap, format);
149 ret = ec_vlog(type, level, format, ap);
155 /* LCOV_EXCL_START */
157 log_cb(int type, enum ec_log_level level, void *opaque, const char *str)
167 static int ec_log_testcase(void)
169 ec_log_t prev_log_cb;
178 prev_log_cb = ec_log_fct;
179 prev_opaque = ec_log_opaque;
181 ret = ec_log_fct_register(log_cb, &check_cb);
182 testres |= EC_TEST_CHECK(ret == 0,
183 "cannot register log function\n");
184 EC_LOG(LOG_ERR, "test\n");
185 testres |= EC_TEST_CHECK(check_cb == 1,
186 "log callback was not invoked\n");
187 logtype = ec_log_lookup("dsdedesdes");
188 testres |= EC_TEST_CHECK(logtype == -1,
189 "lookup invalid name should return -1");
190 logtype = ec_log_lookup("log");
191 logname = ec_log_name(logtype);
192 testres |= EC_TEST_CHECK(logname != NULL &&
193 !strcmp(logname, "log"),
194 "cannot get log name\n");
195 logname = ec_log_name(-1);
196 testres |= EC_TEST_CHECK(logname != NULL &&
197 !strcmp(logname, "unknown"),
198 "cannot get invalid log name\n");
199 logname = ec_log_name(34324);
200 testres |= EC_TEST_CHECK(logname != NULL &&
201 !strcmp(logname, "unknown"),
202 "cannot get invalid log name\n");
203 level = ec_log_level_get();
204 ret = ec_log_level_set(2);
205 testres |= EC_TEST_CHECK(ret == 0 && ec_log_level_get() == 2,
206 "cannot set log level\n");
207 ret = ec_log_level_set(10);
208 testres |= EC_TEST_CHECK(ret != 0,
209 "should not be able to set log level\n");
211 ret = ec_log_fct_register(NULL, NULL);
212 ec_log_level_set(LOG_DEBUG);
213 EC_LOG(LOG_DEBUG, "test log\n");
214 ec_log_level_set(LOG_INFO);
215 EC_LOG(LOG_DEBUG, "test log (not displayed)\n");
216 ec_log_level_set(level);
218 ec_log_fct = prev_log_cb;
219 ec_log_opaque = prev_opaque;
225 static struct ec_test ec_log_test = {
227 .test = ec_log_testcase,
230 EC_TEST_REGISTER(ec_log_test);