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 
11 void buf_output_init(struct buf_output *out)
12 {
13 	out->max_buflen = 0;
14 	out->buflen = 0;
15 	out->buf = NULL;
16 }
17 
18 void buf_output_free(struct buf_output *out)
19 {
20 	free(out->buf);
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 }
43 
44 size_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