log: Modify the implementation such that it uses asprintf()
[fio.git] / log.c
diff --git a/log.c b/log.c
index a112a31ac8fa065b23544acdd7fc0883606ef27b..7e8b13b31a84c906a09db6fce5044ef8e051fcae 100644 (file)
--- a/log.c
+++ b/log.c
-#include <stdio.h>
-#include <stdlib.h>
-#include "list.h"
-#include "fio.h"
+#include <unistd.h>
+#include <fcntl.h>
+#include <string.h>
+#include <stdarg.h>
+#include <syslog.h>
 
-void write_iolog_put(struct thread_data *td, struct io_u *io_u)
-{
-       fprintf(td->iolog_f, "%d,%llu,%u\n", io_u->ddir, io_u->offset, io_u->buflen);
-}
+#include "fio.h"
+#include "oslib/asprintf.h"
 
-int read_iolog_get(struct thread_data *td, struct io_u *io_u)
+size_t log_info_buf(const char *buf, size_t len)
 {
-       struct io_piece *ipo;
-
-       if (!list_empty(&td->io_log_list)) {
-               ipo = list_entry(td->io_log_list.next, struct io_piece, list);
-               list_del(&ipo->list);
-               io_u->offset = ipo->offset;
-               io_u->buflen = ipo->len;
-               io_u->ddir = ipo->ddir;
-               io_u->file = ipo->file;
-               free(ipo);
+       /*
+        * 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;
        }
 
-       return 1;
+       if (log_syslog) {
+               syslog(LOG_INFO, "%s", buf);
+               return len;
+       } else
+               return fwrite(buf, len, 1, f_out);
 }
 
-void prune_io_piece_log(struct thread_data *td)
+size_t log_valist(const char *fmt, va_list args)
 {
-       struct io_piece *ipo;
+       char *buffer;
+       int len;
 
-       while (!list_empty(&td->io_hist_list)) {
-               ipo = list_entry(td->io_hist_list.next, struct io_piece, list);
+       len = vasprintf(&buffer, fmt, args);
+       if (len < 0)
+               return 0;
+       len = log_info_buf(buffer, len);
+       free(buffer);
 
-               list_del(&ipo->list);
-               free(ipo);
-       }
+       return len;
 }
 
-/*
- * log a succesful write, so we can unwind the log for verify
- */
-void log_io_piece(struct thread_data *td, struct io_u *io_u)
+/* add prefix for the specified type in front of the valist */
+void log_prevalist(int type, const char *fmt, va_list args)
 {
-       struct io_piece *ipo = malloc(sizeof(struct io_piece));
-       struct list_head *entry;
+       char *buf1, *buf2;
+       int len;
+       pid_t pid;
 
-       INIT_LIST_HEAD(&ipo->list);
-       ipo->file = io_u->file;
-       ipo->offset = io_u->offset;
-       ipo->len = io_u->buflen;
-
-       /*
-        * for random io where the writes extend the file, it will typically
-        * be laid out with the block scattered as written. it's faster to
-        * read them in in that order again, so don't sort
-        */
-       if (td->sequential || !td->overwrite) {
-               list_add_tail(&ipo->list, &td->io_hist_list);
+       pid = gettid();
+       if (fio_debug_jobp && *fio_debug_jobp != -1U
+           && pid != *fio_debug_jobp)
                return;
-       }
-
-       /*
-        * for random io, sort the list so verify will run faster
-        */
-       entry = &td->io_hist_list;
-       while ((entry = entry->prev) != &td->io_hist_list) {
-               struct io_piece *__ipo = list_entry(entry, struct io_piece, list);
 
-               if (__ipo->offset < ipo->offset)
-                       break;
-       }
-
-       list_add(&ipo->list, entry);
-}
-
-void write_iolog_close(struct thread_data *td)
-{
-       fflush(td->iolog_f);
-       fclose(td->iolog_f);
-       free(td->iolog_buf);
+       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);
 }
 
-/*
- * Open a stored log and read in the entries.
- */
-static int init_iolog_read(struct thread_data *td)
+size_t log_info(const char *format, ...)
 {
-       unsigned long long offset;
-       unsigned int bytes;
-       char *str, *p;
-       FILE *f;
-       int rw, reads, writes;
-
-       f = fopen(td->read_iolog_file, "r");
-       if (!f) {
-               perror("fopen read iolog");
-               return 1;
-       }
-
-       /*
-        * Read in the read iolog and store it, reuse the infrastructure
-        * for doing verifications.
-        */
-       str = malloc(4096);
-       reads = writes = 0;
-       while ((p = fgets(str, 4096, f)) != NULL) {
-               struct io_piece *ipo;
-
-               if (sscanf(p, "%d,%llu,%u", &rw, &offset, &bytes) != 3) {
-                       log_err("bad iolog: %s\n", p);
-                       continue;
-               }
-               if (rw == DDIR_READ)
-                       reads++;
-               else if (rw == DDIR_WRITE)
-                       writes++;
-               else {
-                       log_err("bad ddir: %d\n", rw);
-                       continue;
-               }
-
-               ipo = malloc(sizeof(*ipo));
-               INIT_LIST_HEAD(&ipo->list);
-               ipo->offset = offset;
-               ipo->len = bytes;
-               ipo->ddir = rw;
-               if (bytes > td->max_bs[rw])
-                       td->max_bs[rw] = bytes;
-               list_add_tail(&ipo->list, &td->io_log_list);
-       }
+       va_list args;
+       size_t ret;
 
-       free(str);
-       fclose(f);
+       va_start(args, format);
+       ret = log_valist(format, args);
+       va_end(args);
 
-       if (!reads && !writes)
-               return 1;
-       else if (reads && !writes)
-               td->ddir = DDIR_READ;
-       else if (!reads && writes)
-               td->ddir = DDIR_READ;
-       else
-               td->iomix = 1;
-
-       return 0;
+       return ret;
 }
 
-/*
- * Setup a log for storing io patterns.
- */
-static int init_iolog_write(struct thread_data *td)
+size_t __log_buf(struct buf_output *buf, const char *format, ...)
 {
-       FILE *f;
-
-       f = fopen(td->write_iolog_file, "w+");
-       if (!f) {
-               perror("fopen write iolog");
-               return 1;
-       }
+       char *buffer;
+       va_list args;
+       int len;
+
+       va_start(args, format);
+       len = vasprintf(&buffer, format, args);
+       va_end(args);
+       if (len < 0)
+               return 0;
+       len = buf_output_add(buf, buffer, len);
+       free(buffer);
 
-       /*
-        * That's it for writing, setup a log buffer and we're done.
-         */
-       td->iolog_f = f;
-       td->iolog_buf = malloc(8192);
-       setvbuf(f, td->iolog_buf, _IOFBF, 8192);
-       return 0;
+       return len;
 }
 
-int init_iolog(struct thread_data *td)
+int log_info_flush(void)
 {
-       int ret = 0;
-
-       if (td->io_ops->flags & FIO_CPUIO)
+       if (is_backend || log_syslog)
                return 0;
 
-       if (td->read_iolog_file)
-               ret = init_iolog_read(td);
-       else if (td->write_iolog_file)
-               ret = init_iolog_write(td);
-
-       return 0;
+       return fflush(f_out);
 }
 
-int setup_rate(struct thread_data *td)
+size_t log_err(const char *format, ...)
 {
-       int nr_reads_per_sec;
-
-       if (!td->rate)
-               return 0;
-
-       if (td->rate < td->ratemin) {
-               log_err("min rate larger than nominal rate\n");
-               return -1;
+       size_t ret;
+       int len;
+       char *buffer;
+       va_list args;
+
+       va_start(args, format);
+       len = vasprintf(&buffer, format, args);
+       va_end(args);
+       if (len < 0)
+               return len;
+
+       if (is_backend) {
+               ret = fio_server_text_output(FIO_LOG_ERR, buffer, len);
+               if (ret != -1)
+                       goto done;
        }
 
-       nr_reads_per_sec = (td->rate * 1024) / td->min_bs[DDIR_READ];
-       td->rate_usec_cycle = 1000000 / nr_reads_per_sec;
-       td->rate_pending_usleep = 0;
-       return 0;
-}
+       if (log_syslog) {
+               syslog(LOG_INFO, "%s", buffer);
+               ret = len;
+       } else {
+               if (f_err != stderr)
+                       ret = fwrite(buffer, len, 1, stderr);
 
-void setup_log(struct io_log **log)
-{
-       struct io_log *l = malloc(sizeof(*l));
+               ret = fwrite(buffer, len, 1, f_err);
+       }
 
-       l->nr_samples = 0;
-       l->max_samples = 1024;
-       l->log = malloc(l->max_samples * sizeof(struct io_sample));
-       *log = l;
+done:
+       free(buffer);
+       return ret;
 }
 
-void finish_log(struct thread_data *td, struct io_log *log, const char *name)
+const char *log_get_level(int level)
 {
-       char file_name[256];
-       FILE *f;
-       unsigned int i;
-
-       snprintf(file_name, 200, "client%d_%s.log", td->thread_number, name);
-       f = fopen(file_name, "w");
-       if (!f) {
-               perror("fopen log");
-               return;
-       }
+       static const char *levels[] = { "Unknown", "Debug", "Info", "Error",
+                                               "Unknown" };
 
-       for (i = 0; i < log->nr_samples; i++)
-               fprintf(f, "%lu, %lu, %u\n", log->log[i].time, log->log[i].val, log->log[i].ddir);
+       if (level >= FIO_LOG_NR)
+               level = FIO_LOG_NR;
 
-       fclose(f);
-       free(log->log);
-       free(log);
+       return levels[level];
 }