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