Merge branch 's3_crypto' of github.com:hualongfeng/fio
[fio.git] / log.c
1 #include <unistd.h>
2 #include <string.h>
3 #include <stdarg.h>
4 #include <syslog.h>
5
6 #include "fio.h"
7 #include "oslib/asprintf.h"
8
9 size_t log_info_buf(const char *buf, size_t len)
10 {
11         /*
12          * buf could be NULL (not just "").
13          */
14         if (!buf)
15                 return 0;
16
17         if (is_backend) {
18                 ssize_t ret = fio_server_text_output(FIO_LOG_INFO, buf, len);
19                 if (ret != -1)
20                         return ret;
21         }
22
23         if (log_syslog) {
24                 syslog(LOG_INFO, "%s", buf);
25                 return len;
26         } else
27                 return fwrite(buf, len, 1, f_out);
28 }
29
30 size_t log_valist(const char *fmt, va_list args)
31 {
32         char *buffer;
33         int len;
34
35         len = vasprintf(&buffer, fmt, args);
36         if (len < 0)
37                 return 0;
38         len = log_info_buf(buffer, len);
39         free(buffer);
40
41         return len;
42 }
43
44 /* add prefix for the specified type in front of the valist */
45 #ifdef FIO_INC_DEBUG
46 void log_prevalist(int type, const char *fmt, va_list args)
47 {
48         char *buf1, *buf2;
49         int len;
50         pid_t pid;
51
52         pid = gettid();
53         if (fio_debug_jobp && *fio_debug_jobp != -1U
54             && pid != *fio_debug_jobp)
55                 return;
56
57         len = vasprintf(&buf1, fmt, args);
58         if (len < 0)
59                 return;
60         len = asprintf(&buf2, "%-8s %-5u %s", debug_levels[type].name,
61                        (int) pid, buf1);
62         free(buf1);
63         if (len < 0)
64                 return;
65         log_info_buf(buf2, len);
66         free(buf2);
67 }
68 #endif
69
70 ssize_t log_info(const char *format, ...)
71 {
72         va_list args;
73         ssize_t ret;
74
75         va_start(args, format);
76         ret = log_valist(format, args);
77         va_end(args);
78
79         return ret;
80 }
81
82 size_t __log_buf(struct buf_output *buf, const char *format, ...)
83 {
84         char *buffer;
85         va_list args;
86         int len;
87
88         va_start(args, format);
89         len = vasprintf(&buffer, format, args);
90         va_end(args);
91         if (len < 0)
92                 return 0;
93         len = buf_output_add(buf, buffer, len);
94         free(buffer);
95
96         return len;
97 }
98
99 int log_info_flush(void)
100 {
101         if (is_backend || log_syslog)
102                 return 0;
103
104         return fflush(f_out);
105 }
106
107 ssize_t log_err(const char *format, ...)
108 {
109         ssize_t ret;
110         int len;
111         char *buffer;
112         va_list args;
113
114         va_start(args, format);
115         len = vasprintf(&buffer, format, args);
116         va_end(args);
117         if (len < 0)
118                 return len;
119
120         if (is_backend) {
121                 ret = fio_server_text_output(FIO_LOG_ERR, buffer, len);
122                 if (ret != -1)
123                         goto done;
124         }
125
126         if (log_syslog) {
127                 syslog(LOG_INFO, "%s", buffer);
128                 ret = len;
129         } else {
130                 if (f_err != stderr)
131                         ret = fwrite(buffer, len, 1, stderr);
132
133                 ret = fwrite(buffer, len, 1, f_err);
134         }
135
136 done:
137         free(buffer);
138         return ret;
139 }
140
141 const char *log_get_level(int level)
142 {
143         static const char *levels[] = { "Unknown", "Debug", "Info", "Error",
144                                                 "Unknown" };
145
146         if (level >= FIO_LOG_NR)
147                 level = FIO_LOG_NR;
148
149         return levels[level];
150 }