1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
#include <inttypes.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "iobuf.h"
#define GROWLEN (1 << 14)
struct io_buffer
buf_init(size_t size)
{
struct io_buffer iob;
iob.off = 0;
iob.len = size;
iob.error = (iob.buf = malloc(size)) == NULL ? 1 : 0;
return iob;
}
void
buf_free(struct io_buffer *iob)
{
if (iob->buf != NULL)
free(iob->buf);
}
int
buf_grow(struct io_buffer *iob, size_t addlen)
{
if (iob->error)
return iob->error;
char *nbuf = realloc(iob->buf, iob->len + addlen);
if (nbuf == NULL) {
iob->error = 1;
return 0;
} else {
iob->buf = nbuf;
iob->len += addlen;
return 1;
}
}
int
buf_print(struct io_buffer *iob, const char *fmt, ...)
{
if (iob->error)
return 0;
int np;
va_list ap;
va_start(ap, fmt);
np = vsnprintf(NULL, 0, fmt, ap);
va_end(ap);
if (np + 1 > iob->len - iob->off)
if (!buf_grow(iob, (1 + (np + 1) / GROWLEN) * GROWLEN))
return 0;
va_start(ap, fmt);
vsnprintf(iob->buf + iob->off, np + 1, fmt, ap);
va_end(ap);
iob->off += np;
return 1;
}
int
buf_write(struct io_buffer *iob, const void *buf, size_t len)
{
if (iob->error)
return 0;
if (len > iob->len - iob->off)
if (!buf_grow(iob, (1 + len / GROWLEN) * GROWLEN))
return 0;
bcopy(buf, iob->buf + iob->off, len);
iob->off += len;
return 1;
}
|