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
buf_output_init(struct buf_output * out)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
buf_output_free(struct buf_output * out)18 void buf_output_free(struct buf_output *out)
19 {
20 free(out->buf);
21 }
22
buf_output_add(struct buf_output * out,const char * buf,size_t len)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
buf_output_flush(struct buf_output * out)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