log: Modify the implementation such that it uses asprintf()

[fio.git] / log.c

diff --git a/log.c b/log.c
index d481edf2fd17efb711d454047f625e56b1aebe49..7e8b13b31a84c906a09db6fce5044ef8e051fcae 100644 (file)
--- a/log.c
+++ b/log.c
@@ -5,97 +5,136 @@
 #include <syslog.h>
 
 #include "fio.h"
+#include "oslib/asprintf.h"
 
-int log_valist(const char *str, va_list args)
+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);
-       len = min(len, sizeof(buffer) - 1);
+       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);
+size_t log_valist(const char *fmt, va_list args)
+{
+       char *buffer;
+       int len;
+
+       len = vasprintf(&buffer, fmt, args);
+       if (len < 0)
+               return 0;
+       len = log_info_buf(buffer, len);
+       free(buffer);
 
        return len;
 }
 
-int log_local_buf(const char *buf, size_t len)
+/* add prefix for the specified type in front of the valist */
+void log_prevalist(int type, const char *fmt, va_list args)
 {
-       if (log_syslog)
-               syslog(LOG_INFO, "%s", buf);
-       else
-               len = fwrite(buf, len, 1, f_out);
-
-       return len;
+       char *buf1, *buf2;
+       int len;
+       pid_t pid;
+
+       pid = gettid();
+       if (fio_debug_jobp && *fio_debug_jobp != -1U
+           && pid != *fio_debug_jobp)
+               return;
+
+       len = vasprintf(&buf1, fmt, args);
+       if (len < 0)
+               return;
+       len = asprintf(&buf2, "%-8s %-5u %s", debug_levels[type].name,
+                      (int) pid, buf1);
+       free(buf1);
+       if (len < 0)
+               return;
+       len = log_info_buf(buf2, len);
+       free(buf2);
 }
 
-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);
-       len = min(len, sizeof(buffer) - 1);
 
-       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;
+       int len;
 
        va_start(args, format);
-       len = vsnprintf(buffer, sizeof(buffer), format, args);
+       len = vasprintf(&buffer, format, args);
        va_end(args);
-       len = min(len, sizeof(buffer) - 1);
+       if (len < 0)
+               return 0;
+       len = buf_output_add(buf, buffer, len);
+       free(buffer);
 
-       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);
+       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;
+       int len;
+       char *buffer;
        va_list args;
-       size_t len;
 
        va_start(args, format);
-       len = vsnprintf(buffer, sizeof(buffer), format, args);
+       len = vasprintf(&buffer, format, args);
        va_end(args);
-       len = min(len, sizeof(buffer) - 1);
+       if (len < 0)
+               return len;
+
+       if (is_backend) {
+               ret = fio_server_text_output(FIO_LOG_ERR, buffer, len);
+               if (ret != -1)
+                       goto done;
+       }
 
-       if (is_backend)
-               return fio_server_text_output(FIO_LOG_ERR, buffer, len);
-       else if (log_syslog) {
+       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)