Commit | Line | Data |
---|---|---|
ac9b9101 JA |
1 | #include <unistd.h> |
2 | #include <fcntl.h> | |
3 | #include <string.h> | |
4 | #include <stdarg.h> | |
e46d8091 | 5 | #include <syslog.h> |
3c39a379 | 6 | |
ac9b9101 | 7 | #include "fio.h" |
3c39a379 | 8 | |
7d64aa48 JA |
9 | #define LOG_START_SZ 512 |
10 | ||
830814d4 JA |
11 | size_t log_info_buf(const char *buf, size_t len) |
12 | { | |
3dc3aa41 TK |
13 | /* |
14 | * buf could be NULL (not just ""). | |
15 | */ | |
16 | if (!buf) | |
17 | return 0; | |
18 | ||
408e0b90 JA |
19 | if (is_backend) { |
20 | size_t ret = fio_server_text_output(FIO_LOG_INFO, buf, len); | |
21 | if (ret != -1) | |
22 | return ret; | |
23 | } | |
24 | ||
25 | if (log_syslog) { | |
830814d4 JA |
26 | syslog(LOG_INFO, "%s", buf); |
27 | return len; | |
28 | } else | |
29 | return fwrite(buf, len, 1, f_out); | |
30 | } | |
31 | ||
fc97d6ae | 32 | static size_t valist_to_buf(char **buffer, const char *fmt, va_list src_args) |
e46d8091 | 33 | { |
7d64aa48 | 34 | size_t len, cur = LOG_START_SZ; |
fc97d6ae | 35 | va_list args; |
7d64aa48 JA |
36 | |
37 | do { | |
fc97d6ae | 38 | *buffer = calloc(1, cur); |
e5f9a813 RE |
39 | if (!*buffer) |
40 | return 0; | |
fc97d6ae JA |
41 | |
42 | va_copy(args, src_args); | |
43 | len = vsnprintf(*buffer, cur, fmt, args); | |
44 | va_end(args); | |
7d64aa48 | 45 | |
c78e8496 | 46 | if (len < cur) |
7d64aa48 | 47 | break; |
734f38cb | 48 | |
c78e8496 | 49 | cur = len + 1; |
fc97d6ae | 50 | free(*buffer); |
7d64aa48 | 51 | } while (1); |
734f38cb | 52 | |
fc97d6ae | 53 | return len; |
e46d8091 JA |
54 | } |
55 | ||
e5f9a813 RE |
56 | /* allocate buffer, fill with prefix string followed by vararg string */ |
57 | static size_t prevalist_to_buf(char **buffer, const char *pre, int prelen, | |
58 | const char *fmt, va_list src_args) | |
59 | { | |
60 | size_t len, cur = LOG_START_SZ; | |
61 | va_list args; | |
62 | ||
63 | do { | |
64 | *buffer = calloc(1, cur); | |
65 | if (!*buffer) | |
66 | return 0; | |
67 | ||
68 | va_copy(args, src_args); | |
69 | memcpy(*buffer, pre, prelen); | |
70 | len = prelen + vsnprintf(*buffer + prelen, cur - prelen, fmt, args); | |
71 | va_end(args); | |
72 | ||
73 | if (len < cur) | |
74 | break; | |
75 | ||
76 | cur = len + 1; | |
77 | free(*buffer); | |
78 | } while (1); | |
79 | ||
80 | return len; | |
81 | } | |
82 | ||
fc97d6ae | 83 | size_t log_valist(const char *fmt, va_list args) |
3c39a379 | 84 | { |
7d64aa48 | 85 | char *buffer; |
fc97d6ae | 86 | size_t len; |
7d64aa48 | 87 | |
fc97d6ae JA |
88 | len = valist_to_buf(&buffer, fmt, args); |
89 | len = log_info_buf(buffer, len); | |
90 | free(buffer); | |
f29b25a3 | 91 | |
fc97d6ae JA |
92 | return len; |
93 | } | |
7d64aa48 | 94 | |
e5f9a813 RE |
95 | /* add prefix for the specified type in front of the valist */ |
96 | void log_prevalist(int type, const char *fmt, va_list args) | |
97 | { | |
98 | char pre[32]; | |
99 | char *buffer; | |
100 | size_t len; | |
df4bf117 | 101 | int prelen; |
e5f9a813 RE |
102 | pid_t pid; |
103 | ||
104 | pid = gettid(); | |
105 | if (fio_debug_jobp && *fio_debug_jobp != -1U | |
106 | && pid != *fio_debug_jobp) | |
107 | return; | |
108 | ||
109 | prelen = snprintf(pre, sizeof pre, "%-8s %-5u ", debug_levels[type].name, (int) pid); | |
df4bf117 JA |
110 | if (prelen > 0) { |
111 | len = prevalist_to_buf(&buffer, pre, prelen, fmt, args); | |
112 | len = log_info_buf(buffer, len); | |
113 | free(buffer); | |
114 | } | |
e5f9a813 RE |
115 | } |
116 | ||
fc97d6ae JA |
117 | size_t log_info(const char *format, ...) |
118 | { | |
119 | va_list args; | |
120 | size_t ret; | |
7d64aa48 | 121 | |
fc97d6ae JA |
122 | va_start(args, format); |
123 | ret = log_valist(format, args); | |
124 | va_end(args); | |
7d64aa48 | 125 | |
fc97d6ae | 126 | return ret; |
f29b25a3 JA |
127 | } |
128 | ||
5768cc2b | 129 | size_t __log_buf(struct buf_output *buf, const char *format, ...) |
a666cab8 | 130 | { |
7d64aa48 | 131 | char *buffer; |
a666cab8 | 132 | va_list args; |
fc97d6ae | 133 | size_t len; |
a666cab8 | 134 | |
fc97d6ae JA |
135 | va_start(args, format); |
136 | len = valist_to_buf(&buffer, format, args); | |
137 | va_end(args); | |
7d64aa48 | 138 | |
fc97d6ae | 139 | len = buf_output_add(buf, buffer, len); |
7d64aa48 JA |
140 | free(buffer); |
141 | ||
fc97d6ae | 142 | return len; |
a666cab8 JA |
143 | } |
144 | ||
fdd5f15f VKF |
145 | int log_info_flush(void) |
146 | { | |
147 | if (is_backend || log_syslog) | |
148 | return 0; | |
149 | ||
150 | return fflush(f_out); | |
151 | } | |
152 | ||
5768cc2b | 153 | size_t log_err(const char *format, ...) |
fb71fbd7 | 154 | { |
fc97d6ae | 155 | size_t ret, len; |
7d64aa48 | 156 | char *buffer; |
ac9b9101 | 157 | va_list args; |
f29b25a3 | 158 | |
fc97d6ae JA |
159 | va_start(args, format); |
160 | len = valist_to_buf(&buffer, format, args); | |
161 | va_end(args); | |
fb71fbd7 | 162 | |
408e0b90 | 163 | if (is_backend) { |
7d64aa48 | 164 | ret = fio_server_text_output(FIO_LOG_ERR, buffer, len); |
408e0b90 | 165 | if (ret != -1) |
7d64aa48 | 166 | goto done; |
408e0b90 JA |
167 | } |
168 | ||
169 | if (log_syslog) { | |
13755d94 | 170 | syslog(LOG_INFO, "%s", buffer); |
7d64aa48 | 171 | ret = len; |
13755d94 | 172 | } else { |
7d64aa48 | 173 | if (f_err != stderr) |
1f39e555 | 174 | ret = fwrite(buffer, len, 1, stderr); |
fb71fbd7 | 175 | |
7d64aa48 | 176 | ret = fwrite(buffer, len, 1, f_err); |
f29b25a3 | 177 | } |
7d64aa48 JA |
178 | |
179 | done: | |
180 | free(buffer); | |
181 | return ret; | |
e3cedca7 | 182 | } |
3ad04399 JA |
183 | |
184 | const char *log_get_level(int level) | |
185 | { | |
186 | static const char *levels[] = { "Unknown", "Debug", "Info", "Error", | |
187 | "Unknown" }; | |
188 | ||
189 | if (level >= FIO_LOG_NR) | |
190 | level = FIO_LOG_NR; | |
191 | ||
192 | return levels[level]; | |
193 | } |