1#include <stdio.h>
2#include <string.h>
3#include <stdlib.h>
4
5#include "output_buffer.h"
6#include "../log.h"
7#include "../minmax.h"
8
9#define BUF_INC	1024
10
11void buf_output_init(struct buf_output *out)
12{
13	out->max_buflen = 0;
14	out->buflen = 0;
15	out->buf = NULL;
16}
17
18void buf_output_free(struct buf_output *out)
19{
20	free(out->buf);
21}
22
23size_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}
43
44size_t buf_output_flush(struct buf_output *out)
45{
46	size_t ret = 0;
47
48	if (out->buflen) {
49		ret = log_info_buf(out->buf, out->buflen);
50		memset(out->buf, 0, out->max_buflen);
51		out->buflen = 0;
52	}
53
54	return ret;
55}
56