X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=log.c;h=95351d57d96ca4fd0ea97e43d38352f9b7b3dde1;hp=a26dee8775cb751eb786cb562cfafbcf4decdca5;hb=fc97d6ae95fadbd1c57197d5f744f3353e41cd57;hpb=de26d51f3e909997c9a500bf9f15a1bd5c18bb49 diff --git a/log.c b/log.c index a26dee87..95351d57 100644 --- a/log.c +++ b/log.c @@ -6,92 +6,128 @@ #include "fio.h" -int log_valist(const char *str, va_list args) +#define LOG_START_SZ 512 + +size_t log_info_buf(const char *buf, size_t len) { - char buffer[1024]; - size_t len; + /* + * buf could be NULL (not just ""). + */ + if (!buf) + return 0; + + if (is_backend) { + size_t ret = fio_server_text_output(FIO_LOG_INFO, buf, len); + if (ret != -1) + return ret; + } - len = vsnprintf(buffer, sizeof(buffer), str, args); + if (log_syslog) { + syslog(LOG_INFO, "%s", buf); + return len; + } else + return fwrite(buf, len, 1, f_out); +} - if (log_syslog) - syslog(LOG_INFO, "%s", buffer); - else - len = fwrite(buffer, len, 1, f_out); +static size_t valist_to_buf(char **buffer, const char *fmt, va_list src_args) +{ + size_t len, cur = LOG_START_SZ; + va_list args; + + do { + *buffer = calloc(1, cur); + + va_copy(args, src_args); + len = vsnprintf(*buffer, cur, fmt, args); + va_end(args); + + if (len < cur) + break; + + cur = len + 1; + free(*buffer); + } while (1); return len; } -int log_local_buf(const char *buf, size_t len) +size_t log_valist(const char *fmt, va_list args) { - if (log_syslog) - syslog(LOG_INFO, "%s", buf); - else - len = fwrite(buf, len, 1, f_out); + char *buffer; + size_t len; + + len = valist_to_buf(&buffer, fmt, args); + len = log_info_buf(buffer, len); + free(buffer); return len; } -int log_local(const char *format, ...) +size_t log_info(const char *format, ...) { - char buffer[1024]; va_list args; - size_t len; + size_t ret; va_start(args, format); - len = vsnprintf(buffer, sizeof(buffer), format, args); + ret = log_valist(format, args); va_end(args); - if (log_syslog) - syslog(LOG_INFO, "%s", buffer); - else - len = fwrite(buffer, len, 1, f_out); - - return len; + return ret; } -int log_info(const char *format, ...) +size_t __log_buf(struct buf_output *buf, const char *format, ...) { - char buffer[1024]; + char *buffer; va_list args; size_t len; va_start(args, format); - len = vsnprintf(buffer, sizeof(buffer), format, args); + len = valist_to_buf(&buffer, format, args); va_end(args); - if (is_backend) - return fio_server_text_output(FIO_LOG_INFO, buffer, len); - else if (log_syslog) { - syslog(LOG_INFO, "%s", buffer); - return len; - } else - return fwrite(buffer, len, 1, f_out); + len = buf_output_add(buf, buffer, len); + free(buffer); + + return len; +} + +int log_info_flush(void) +{ + if (is_backend || log_syslog) + return 0; + + return fflush(f_out); } -int log_err(const char *format, ...) +size_t log_err(const char *format, ...) { - char buffer[1024]; + size_t ret, len; + char *buffer; va_list args; - size_t len; va_start(args, format); - len = vsnprintf(buffer, sizeof(buffer), format, args); + len = valist_to_buf(&buffer, format, args); va_end(args); - if (is_backend) - return fio_server_text_output(FIO_LOG_ERR, buffer, len); - else if (log_syslog) { + if (is_backend) { + ret = fio_server_text_output(FIO_LOG_ERR, buffer, len); + if (ret != -1) + goto done; + } + + if (log_syslog) { syslog(LOG_INFO, "%s", buffer); - return len; + ret = len; } else { - if (f_err != stderr) { - int fio_unused ret; - + if (f_err != stderr) ret = fwrite(buffer, len, 1, stderr); - } - return fwrite(buffer, len, 1, f_err); + ret = fwrite(buffer, len, 1, f_err); } + +done: + free(buffer); + return ret; } const char *log_get_level(int level)