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
76
77
78
79
80
81
82
83
84
85
86
|
#include <stdlib.h>
#include <string.h>
#include "buffer.h"
#include "util.h"
#define BUF_SIZE 1024
void buffer_init(Buffer *buf) {
memset(buf, 0, sizeof *buf);
}
bool buffer_grow(Buffer *buf, size_t size) {
if (size < BUF_SIZE)
size = BUF_SIZE;
if (buf->size < size) {
if (buf->size > 0)
size *= 2;
char *data = realloc(buf->data, size);
if (!data)
return false;
buf->size = size;
buf->data = data;
}
return true;
}
void buffer_truncate(Buffer *buf) {
buf->len = 0;
}
void buffer_release(Buffer *buf) {
if (!buf)
return;
free(buf->data);
buffer_init(buf);
}
bool buffer_put(Buffer *buf, const void *data, size_t len) {
if (!buffer_grow(buf, len))
return false;
memmove(buf->data, data, len);
buf->len = len;
return true;
}
bool buffer_put0(Buffer *buf, const char *data) {
return buffer_put(buf, data, strlen(data)+1);
}
bool buffer_insert(Buffer *buf, size_t pos, const void *data, size_t len) {
if (pos > buf->len)
return false;
if (!buffer_grow(buf, buf->len + len))
return false;
memmove(buf->data + pos + len, buf->data + pos, buf->len - pos);
memcpy(buf->data + pos, data, len);
buf->len += len;
return true;
}
bool buffer_insert0(Buffer *buf, size_t pos, const char *data) {
if (pos == 0)
return buffer_prepend0(buf, data);
if (pos == buf->len)
return buffer_append0(buf, data);
return buffer_insert(buf, pos, data, strlen(data));
}
bool buffer_append(Buffer *buf, const void *data, size_t len) {
return buffer_insert(buf, buf->len, data, len);
}
bool buffer_append0(Buffer *buf, const char *data) {
if (buf->len > 0 && buf->data[buf->len-1] == '\0')
buf->len--;
return buffer_append(buf, data, strlen(data)) && buffer_append(buf, "\0", 1);
}
bool buffer_prepend(Buffer *buf, const void *data, size_t len) {
return buffer_insert(buf, 0, data, len);
}
bool buffer_prepend0(Buffer *buf, const char *data) {
return buffer_prepend(buf, data, strlen(data) + (buf->len == 0));
}
|
