lib/output_buffer: harden buf_output_free() and kill buf_output_clear()
[fio.git] / lib / output_buffer.c
1 #include <stdio.h>
2 #include <string.h>
3 #include <stdlib.h>
4
5 #include "output_buffer.h"
6 #include "../minmax.h"
7
8 #define BUF_INC 1024
9
10 void buf_output_init(struct buf_output *out)
11 {
12         out->max_buflen = 0;
13         out->buflen = 0;
14         out->buf = NULL;
15 }
16
17 void buf_output_free(struct buf_output *out)
18 {
19         free(out->buf);
20         buf_output_init(out);
21 }
22
23 size_t buf_output_add(struct buf_output *out, const char *buf, size_t len)
24 {
25         if (out->max_buflen - out->buflen < len) {
26                 size_t need = len - (out->max_buflen - out->buflen);
27                 size_t old_max = out->max_buflen;
28
29                 need = max((size_t) BUF_INC, need);
30                 out->max_buflen += need;
31                 out->buf = realloc(out->buf, out->max_buflen);
32
33                 old_max = max(old_max, out->buflen + len);
34                 if (old_max + need > out->max_buflen)
35                         need = out->max_buflen - old_max;
36                 memset(&out->buf[old_max], 0, need);
37         }
38
39         memcpy(&out->buf[out->buflen], buf, len);
40         out->buflen += len;
41         return len;
42 }