File: log.c

package info (click to toggle)
fcft 3.3.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 3,248 kB
  • sloc: ansic: 8,160; python: 115; sh: 44; makefile: 4
file content (160 lines) | stat: -rw-r--r-- 4,737 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
#include "log.h"

#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include <string.h>
#include <stdbool.h>
#include <errno.h>
#include <assert.h>
#include <unistd.h>

#include <syslog.h>

#include <fcft/fcft.h>

static bool colorize = false;
static bool do_syslog = false;
static enum fcft_log_class log_level = FCFT_LOG_CLASS_NONE;

void
fcft_log_init(enum fcft_log_colorize _colorize, bool _do_syslog,
              enum fcft_log_class _log_level)
{
    /* Don't use colors if NO_COLOR is defined and not empty */
    const char *no_color_str = getenv("NO_COLOR");
    const bool no_color = no_color_str != NULL && no_color_str[0] != '\0';

    colorize = _colorize == FCFT_LOG_COLORIZE_NEVER ? false
        : _colorize == FCFT_LOG_COLORIZE_ALWAYS ? true
        : !no_color && isatty(STDERR_FILENO);
    do_syslog = _do_syslog;
    log_level = _log_level;
}

static void
_log(enum fcft_log_class log_class, const char *module, const char *file, int lineno,
     const char *fmt, int sys_errno, va_list va)
{
    if (log_class > log_level)
        return;

    const char *class = "abcd";
    int class_clr = 0;
    switch (log_class) {
    case FCFT_LOG_CLASS_NONE:     assert(false); class = "none"; class_clr = 31; break;
    case FCFT_LOG_CLASS_ERROR:    class = " err"; class_clr = 31; break;
    case FCFT_LOG_CLASS_WARNING:  class = "warn"; class_clr = 33; break;
    case FCFT_LOG_CLASS_INFO:     class = "info"; class_clr = 97; break;
    case FCFT_LOG_CLASS_DEBUG:    class = " dbg"; class_clr = 36; break;
    }

    char clr[16];
    snprintf(clr, sizeof(clr), "\033[%dm", class_clr);
    fprintf(stderr, "%s%s%s: ", colorize ? clr : "", class, colorize ? "\033[0m" : "");

    if (colorize)
        fprintf(stderr, "\033[2m");
    fprintf(stderr, "%s:%d: ", file, lineno);
    if (colorize)
        fprintf(stderr, "\033[0m");

    vfprintf(stderr, fmt, va);

    if (sys_errno != 0)
        fprintf(stderr, ": %s", strerror(sys_errno));

    fprintf(stderr, "\n");
}

static void
_sys_log(enum fcft_log_class log_class, const char *module,
         const char *file __attribute__((unused)),
         int lineno __attribute__((unused)),
         const char *fmt, int sys_errno, va_list va)
{
    if (!do_syslog)
        return;

    if (log_class > log_level)
        return;

    /* Map our log level to syslog's level */
    int level = -1;
    switch (log_class) {
    case FCFT_LOG_CLASS_NONE:     assert(false); level = LOG_ERR; break;
    case FCFT_LOG_CLASS_ERROR:    level = LOG_ERR; break;
    case FCFT_LOG_CLASS_WARNING:  level = LOG_WARNING; break;
    case FCFT_LOG_CLASS_INFO:     level = LOG_INFO; break;
    case FCFT_LOG_CLASS_DEBUG:    level = LOG_DEBUG; break;
    }

    assert(level != -1);

    const char *sys_err = sys_errno != 0 ? strerror(sys_errno) : NULL;

    va_list va2;
    va_copy(va2, va);

    /* Calculate required size of buffer holding the entire log message */
    int required_len = 0;
    required_len += strlen(module) + 2;  /* "%s: " */
    required_len += vsnprintf(NULL, 0, fmt, va2); va_end(va2);

    if (sys_errno != 0)
        required_len += strlen(sys_err) + 2; /* ": %s" */

    /* Format the msg */
    char *msg = malloc(required_len + 1);
    int idx = 0;

    idx += snprintf(&msg[idx], required_len + 1 - idx, "%s: ", module);
    idx += vsnprintf(&msg[idx], required_len + 1 - idx, fmt, va);

    if (sys_errno != 0) {
        snprintf(
            &msg[idx], required_len + 1 - idx, ": %s", strerror(sys_errno));
    }

    syslog(level, "%s", msg);
    free(msg);
}

void
fcft_log_msg(enum fcft_log_class log_class, const char *module,
        const char *file, int lineno, const char *fmt, ...)
{
    va_list ap1, ap2;
    va_start(ap1, fmt);
    va_copy(ap2, ap1);
    _log(log_class, module, file, lineno, fmt, 0, ap1);
    _sys_log(log_class, module, file, lineno, fmt, 0, ap2);
    va_end(ap1);
    va_end(ap2);
}

void fcft_log_errno(enum fcft_log_class log_class, const char *module,
                    const char *file, int lineno,
                    const char *fmt, ...)
{
    va_list ap1, ap2;
    va_start(ap1, fmt);
    va_copy(ap2, ap1);
    _log(log_class, module, file, lineno, fmt, errno, ap1);
    _sys_log(log_class, module, file, lineno, fmt, errno, ap2);
    va_end(ap1);
    va_end(ap2);
}

void fcft_log_errno_provided(enum fcft_log_class log_class, const char *module,
                             const char *file, int lineno, int _errno,
                             const char *fmt, ...)
{
    va_list ap1, ap2;
    va_start(ap1, fmt);
    va_copy(ap2, ap1);
    _log(log_class, module, file, lineno, fmt, _errno, ap1);
    _sys_log(log_class, module, file, lineno, fmt, _errno, ap2);
    va_end(ap1);
    va_end(ap2);
}