Skip to content

Commit 68f59a9

Browse files
dpgeorgedpgeorge
committed
Add vstr and its functions.
1 parent df4b4f3 commit 68f59a9

2 files changed

Lines changed: 220 additions & 8 deletions

File tree

‎py/misc.h‎

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,15 @@ void encode_le32(byte *buf, unsigned int i);
5858

5959
/** variable string *********************************************/
6060

61-
/*
62-
typedef struct _vstr_t vstr_t;
63-
61+
typedef struct _vstr_t {
62+
int alloc;
63+
int len;
64+
char *buf;
65+
bool had_error;
66+
} vstr_t;
67+
68+
void vstr_init(vstr_t *vstr);
69+
void vstr_clear(vstr_t *vstr);
6470
vstr_t *vstr_new();
6571
void vstr_free(vstr_t *vstr);
6672
void vstr_reset(vstr_t *vstr);
@@ -69,14 +75,14 @@ char *vstr_str(vstr_t *vstr);
6975
int vstr_len(vstr_t *vstr);
7076
void vstr_hint_size(vstr_t *vstr, int size);
7177
char *vstr_add_len(vstr_t *vstr, int len);
78+
void vstr_add_byte(vstr_t *vstr, byte v);
79+
void vstr_add_char(vstr_t *vstr, unichar chr);
7280
void vstr_add_str(vstr_t *vstr, const char *str);
7381
void vstr_add_strn(vstr_t *vstr, const char *str, int len);
74-
void vstr_add_byte(vstr_t *vstr, byte v);
75-
void vstr_add_le16(vstr_t *vstr, unsigned short v);
76-
void vstr_add_le32(vstr_t *vstr, unsigned int v);
82+
//void vstr_add_le16(vstr_t *vstr, unsigned short v);
83+
//void vstr_add_le32(vstr_t *vstr, unsigned int v);
7784
void vstr_cut_tail(vstr_t *vstr, int len);
78-
void vstr_printf(vstr_t *vstr, const char *fmt, ...);
79-
*/
85+
//void vstr_printf(vstr_t *vstr, const char *fmt, ...);
8086

8187
/** unique string ***********************************************/
8288

‎py/vstr.c‎

Lines changed: 206 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,206 @@
1+
#include <stdio.h>
2+
#include <stdarg.h>
3+
#include <string.h>
4+
#include "misc.h"
5+
6+
// returned value is always at least 1 greater than argument
7+
#define ROUND_ALLOC(a) (((a) & ((~0) - 7)) + 8)
8+
9+
void vstr_init(vstr_t *vstr) {
10+
vstr->alloc = 32;
11+
vstr->len = 0;
12+
vstr->buf = m_new(char, vstr->alloc);
13+
if (vstr->buf == NULL) {
14+
m_free(vstr);
15+
vstr->had_error = true;
16+
return;
17+
}
18+
vstr->buf[0] = 0;
19+
vstr->had_error = false;
20+
}
21+
22+
void vstr_clear(vstr_t *vstr) {
23+
m_free(vstr->buf);
24+
vstr->buf = NULL;
25+
}
26+
27+
vstr_t *vstr_new() {
28+
vstr_t *vstr = m_new(vstr_t, 1);
29+
if (vstr == NULL) {
30+
return NULL;
31+
}
32+
vstr_init(vstr);
33+
return vstr;
34+
}
35+
36+
void vstr_free(vstr_t *vstr) {
37+
if (vstr != NULL) {
38+
m_free(vstr->buf);
39+
m_free(vstr);
40+
}
41+
}
42+
43+
void vstr_reset(vstr_t *vstr) {
44+
vstr->len = 0;
45+
vstr->buf[0] = 0;
46+
vstr->had_error = false;
47+
}
48+
49+
bool vstr_had_error(vstr_t *vstr) {
50+
return vstr->had_error;
51+
}
52+
53+
char *vstr_str(vstr_t *vstr) {
54+
if (vstr->had_error) {
55+
return NULL;
56+
}
57+
return vstr->buf;
58+
}
59+
60+
int vstr_len(vstr_t *vstr) {
61+
if (vstr->had_error) {
62+
return 0;
63+
}
64+
return vstr->len;
65+
}
66+
67+
bool vstr_ensure_extra(vstr_t *vstr, int size) {
68+
if (vstr->len + size + 1 > vstr->alloc) {
69+
int new_alloc = ROUND_ALLOC((vstr->len + size + 1) * 2);
70+
char *new_buf = m_renew(char, vstr->buf, new_alloc);
71+
if (new_buf == NULL) {
72+
vstr->had_error = true;
73+
return false;
74+
}
75+
vstr->alloc = new_alloc;
76+
vstr->buf = new_buf;
77+
}
78+
return true;
79+
}
80+
81+
void vstr_hint_size(vstr_t *vstr, int size) {
82+
// it's not an error if we fail to allocate for the size hint
83+
bool er = vstr->had_error;
84+
vstr_ensure_extra(vstr, size);
85+
vstr->had_error = er;
86+
}
87+
88+
char *vstr_add_len(vstr_t *vstr, int len) {
89+
if (vstr->had_error || !vstr_ensure_extra(vstr, len)) {
90+
return NULL;
91+
}
92+
char *buf = vstr->buf + vstr->len;
93+
vstr->len += len;
94+
vstr->buf[vstr->len] = 0;
95+
return buf;
96+
}
97+
98+
void vstr_add_byte(vstr_t *vstr, byte v) {
99+
byte *buf = (byte*)vstr_add_len(vstr, 1);
100+
if (buf == NULL) {
101+
return;
102+
}
103+
buf[0] = v;
104+
}
105+
106+
void vstr_add_char(vstr_t *vstr, unichar c) {
107+
// TODO UNICODE
108+
byte *buf = (byte*)vstr_add_len(vstr, 1);
109+
if (buf == NULL) {
110+
return;
111+
}
112+
buf[0] = c;
113+
}
114+
115+
void vstr_add_str(vstr_t *vstr, const char *str) {
116+
vstr_add_strn(vstr, str, strlen(str));
117+
}
118+
119+
void vstr_add_strn(vstr_t *vstr, const char *str, int len) {
120+
if (vstr->had_error || !vstr_ensure_extra(vstr, len)) {
121+
return;
122+
}
123+
memmove(vstr->buf + vstr->len, str, len);
124+
vstr->len += len;
125+
vstr->buf[vstr->len] = 0;
126+
}
127+
128+
/*
129+
void vstr_add_le16(vstr_t *vstr, unsigned short v) {
130+
byte *buf = (byte*)vstr_add_len(vstr, 2);
131+
if (buf == NULL) {
132+
return;
133+
}
134+
encode_le16(buf, v);
135+
}
136+
137+
void vstr_add_le32(vstr_t *vstr, unsigned int v) {
138+
byte *buf = (byte*)vstr_add_len(vstr, 4);
139+
if (buf == NULL) {
140+
return;
141+
}
142+
encode_le32(buf, v);
143+
}
144+
*/
145+
146+
void vstr_cut_tail(vstr_t *vstr, int len) {
147+
if (vstr->had_error) {
148+
return;
149+
}
150+
if (len > vstr->len) {
151+
vstr->len = 0;
152+
} else {
153+
vstr->len -= len;
154+
}
155+
}
156+
157+
/*
158+
void vstr_printf(vstr_t *vstr, const char *fmt, ...) {
159+
if (vstr->had_error || !vstr_ensure_extra(vstr, strlen(fmt))) {
160+
return;
161+
}
162+
163+
while (1) {
164+
// try to print in the allocated space
165+
int size = vstr->alloc - vstr->len;
166+
va_list ap;
167+
va_start(ap, fmt);
168+
int n = vsnprintf(vstr->buf + vstr->len, size, fmt, ap);
169+
va_end(ap);
170+
171+
// if that worked, return
172+
if (n > -1 && n < size) {
173+
vstr->len += n;
174+
return;
175+
}
176+
177+
// else try again with more space
178+
if (n > -1) { // glibc 2.1
179+
// n + 1 is precisely what is needed
180+
if (!vstr_ensure_extra(vstr, n + 1)) {
181+
return;
182+
}
183+
} else { // glibc 2.0
184+
// increase to twice the old size
185+
if (!vstr_ensure_extra(vstr, size * 2)) {
186+
return;
187+
}
188+
}
189+
}
190+
}
191+
*/
192+
193+
/** testing *****************************************************/
194+
195+
/*
196+
int main() {
197+
vstr_t *vstr = vstr_new();
198+
int i;
199+
for (i = 0; i < 10; i++) {
200+
vstr_printf(vstr, "%d) this is a test %d %s\n", i, 1234, "'a string'");
201+
vstr_add_str(vstr, "-----");
202+
vstr_printf(vstr, "this is another test %d %s\n", 1234, "'a second string'");
203+
printf("%s", vstr->buf);
204+
}
205+
}
206+
*/

0 commit comments

Comments
 (0)