output_buffer: only realloc once, and memset just what we need
[fio.git] / lib / output_buffer.c
index 145272a00d60e4ed067ac404c1a1966a4bb98668..c1fdfc95f610aca10ece35ebdc2dac437c01af66 100644 (file)
@@ -4,15 +4,15 @@
 
 #include "output_buffer.h"
 #include "../log.h"
+#include "../minmax.h"
 
 #define BUF_INC        1024
 
-void buf_output_init(struct buf_output *out, int index)
+void buf_output_init(struct buf_output *out)
 {
-       out->max_buflen = BUF_INC;
-       out->buf = malloc(out->max_buflen);
-       memset(out->buf, 0, out->max_buflen);
+       out->max_buflen = 0;
        out->buflen = 0;
+       out->buf = NULL;
 }
 
 void buf_output_free(struct buf_output *out)
@@ -22,12 +22,18 @@ void buf_output_free(struct buf_output *out)
 
 size_t buf_output_add(struct buf_output *out, const char *buf, size_t len)
 {
-       while (out->max_buflen - out->buflen < len) {
-               size_t newlen = out->max_buflen + BUF_INC - out->buflen;
+       if (out->max_buflen - out->buflen < len) {
+               size_t need = len - (out->max_buflen - out->buflen);
+               size_t old_max = out->max_buflen;
 
-               out->max_buflen += BUF_INC;
+               need = max((size_t) BUF_INC, need);
+               out->max_buflen += need;
                out->buf = realloc(out->buf, out->max_buflen);
-               memset(&out->buf[out->buflen], 0, newlen);
+
+               old_max = max(old_max, out->buflen + len);
+               if (old_max + need > out->max_buflen)
+                       need = out->max_buflen - old_max;
+               memset(&out->buf[old_max], 0, need);
        }
 
        memcpy(&out->buf[out->buflen], buf, len);
@@ -40,7 +46,7 @@ size_t buf_output_flush(struct buf_output *out)
        size_t ret = 0;
 
        if (out->buflen) {
-               ret = log_local_buf(out->buf, out->buflen);
+               ret = log_info_buf(out->buf, out->buflen);
                memset(out->buf, 0, out->max_buflen);
                out->buflen = 0;
        }