Add vstr and its functions.

2 files changed, 220 insertions, 8 deletions

diff --git a/py/misc.h b/py/misc.h
index 9ba80a5c37..c67d9df8f9 100644
--- a/py/misc.h
+++ b/py/misc.h
@@ -58,9 +58,15 @@ void encode_le32(byte *buf, unsigned int i);
 
 /** variable string *********************************************/
 
-/*
-typedef struct _vstr_t vstr_t;
-
+typedef struct _vstr_t {
+    int alloc;
+    int len;
+    char *buf;
+    bool had_error;
+} vstr_t;
+
+void vstr_init(vstr_t *vstr);
+void vstr_clear(vstr_t *vstr);
 vstr_t *vstr_new();
 void vstr_free(vstr_t *vstr);
 void vstr_reset(vstr_t *vstr);
@@ -69,14 +75,14 @@ char *vstr_str(vstr_t *vstr);
 int vstr_len(vstr_t *vstr);
 void vstr_hint_size(vstr_t *vstr, int size);
 char *vstr_add_len(vstr_t *vstr, int len);
+void vstr_add_byte(vstr_t *vstr, byte v);
+void vstr_add_char(vstr_t *vstr, unichar chr);
 void vstr_add_str(vstr_t *vstr, const char *str);
 void vstr_add_strn(vstr_t *vstr, const char *str, int len);
-void vstr_add_byte(vstr_t *vstr, byte v);
-void vstr_add_le16(vstr_t *vstr, unsigned short v);
-void vstr_add_le32(vstr_t *vstr, unsigned int v);
+//void vstr_add_le16(vstr_t *vstr, unsigned short v);
+//void vstr_add_le32(vstr_t *vstr, unsigned int v);
 void vstr_cut_tail(vstr_t *vstr, int len);
-void vstr_printf(vstr_t *vstr, const char *fmt, ...);
-*/
+//void vstr_printf(vstr_t *vstr, const char *fmt, ...);
 
 /** unique string ***********************************************/
 
diff --git a/py/vstr.c b/py/vstr.c
new file mode 100644
index 0000000000..fc7a772452
--- /dev/null
+++ b/py/vstr.c
@@ -0,0 +1,206 @@
+#include <stdio.h>
+#include <stdarg.h>
+#include <string.h>
+#include "misc.h"
+
+// returned value is always at least 1 greater than argument
+#define ROUND_ALLOC(a) (((a) & ((~0) - 7)) + 8)
+
+void vstr_init(vstr_t *vstr) {
+    vstr->alloc = 32;
+    vstr->len = 0;
+    vstr->buf = m_new(char, vstr->alloc);
+    if (vstr->buf == NULL) {
+        m_free(vstr);
+        vstr->had_error = true;
+        return;
+    }
+    vstr->buf[0] = 0;
+    vstr->had_error = false;
+}
+
+void vstr_clear(vstr_t *vstr) {
+    m_free(vstr->buf);
+    vstr->buf = NULL;
+}
+
+vstr_t *vstr_new() {
+    vstr_t *vstr = m_new(vstr_t, 1);
+    if (vstr == NULL) {
+        return NULL;
+    }
+    vstr_init(vstr);
+    return vstr;
+}
+
+void vstr_free(vstr_t *vstr) {
+    if (vstr != NULL) {
+        m_free(vstr->buf);
+        m_free(vstr);
+    }
+}
+
+void vstr_reset(vstr_t *vstr) {
+    vstr->len = 0;
+    vstr->buf[0] = 0;
+    vstr->had_error = false;
+}
+
+bool vstr_had_error(vstr_t *vstr) {
+    return vstr->had_error;
+}
+
+char *vstr_str(vstr_t *vstr) {
+    if (vstr->had_error) {
+        return NULL;
+    }
+    return vstr->buf;
+}
+
+int vstr_len(vstr_t *vstr) {
+    if (vstr->had_error) {
+        return 0;
+    }
+    return vstr->len;
+}
+
+bool vstr_ensure_extra(vstr_t *vstr, int size) {
+    if (vstr->len + size + 1 > vstr->alloc) {
+        int new_alloc = ROUND_ALLOC((vstr->len + size + 1) * 2);
+        char *new_buf = m_renew(char, vstr->buf, new_alloc);
+        if (new_buf == NULL) {
+            vstr->had_error = true;
+            return false;
+        }
+        vstr->alloc = new_alloc;
+        vstr->buf = new_buf;
+    }
+    return true;
+}
+
+void vstr_hint_size(vstr_t *vstr, int size) {
+    // it's not an error if we fail to allocate for the size hint
+    bool er = vstr->had_error;
+    vstr_ensure_extra(vstr, size);
+    vstr->had_error = er;
+}
+
+char *vstr_add_len(vstr_t *vstr, int len) {
+    if (vstr->had_error || !vstr_ensure_extra(vstr, len)) {
+        return NULL;
+    }
+    char *buf = vstr->buf + vstr->len;
+    vstr->len += len;
+    vstr->buf[vstr->len] = 0;
+    return buf;
+}
+
+void vstr_add_byte(vstr_t *vstr, byte v) {
+    byte *buf = (byte*)vstr_add_len(vstr, 1);
+    if (buf == NULL) {
+        return;
+    }
+    buf[0] = v;
+}
+
+void vstr_add_char(vstr_t *vstr, unichar c) {
+    // TODO UNICODE
+    byte *buf = (byte*)vstr_add_len(vstr, 1);
+    if (buf == NULL) {
+        return;
+    }
+    buf[0] = c;
+}
+
+void vstr_add_str(vstr_t *vstr, const char *str) {
+    vstr_add_strn(vstr, str, strlen(str));
+}
+
+void vstr_add_strn(vstr_t *vstr, const char *str, int len) {
+    if (vstr->had_error || !vstr_ensure_extra(vstr, len)) {
+        return;
+    }
+    memmove(vstr->buf + vstr->len, str, len);
+    vstr->len += len;
+    vstr->buf[vstr->len] = 0;
+}
+
+/*
+void vstr_add_le16(vstr_t *vstr, unsigned short v) {
+    byte *buf = (byte*)vstr_add_len(vstr, 2);
+    if (buf == NULL) {
+        return;
+    }
+    encode_le16(buf, v);
+}
+
+void vstr_add_le32(vstr_t *vstr, unsigned int v) {
+    byte *buf = (byte*)vstr_add_len(vstr, 4);
+    if (buf == NULL) {
+        return;
+    }
+    encode_le32(buf, v);
+}
+*/
+
+void vstr_cut_tail(vstr_t *vstr, int len) {
+    if (vstr->had_error) {
+        return;
+    }
+    if (len > vstr->len) {
+        vstr->len = 0;
+    } else {
+        vstr->len -= len;
+    }
+}
+
+/*
+void vstr_printf(vstr_t *vstr, const char *fmt, ...) {
+    if (vstr->had_error || !vstr_ensure_extra(vstr, strlen(fmt))) {
+        return;
+    }
+
+    while (1) {
+        // try to print in the allocated space
+        int size = vstr->alloc - vstr->len;
+        va_list ap;
+        va_start(ap, fmt);
+        int n = vsnprintf(vstr->buf + vstr->len, size, fmt, ap);
+        va_end(ap);
+
+        // if that worked, return
+        if (n > -1 && n < size) {
+            vstr->len += n;
+            return;
+        }
+
+        // else try again with more space
+        if (n > -1) { // glibc 2.1
+            // n + 1 is precisely what is needed
+            if (!vstr_ensure_extra(vstr, n + 1)) {
+                return;
+            }
+        } else { // glibc 2.0
+            // increase to twice the old size
+            if (!vstr_ensure_extra(vstr, size * 2)) {
+                return;
+            }
+        }
+    }
+}
+*/
+
+/** testing *****************************************************/
+
+/*
+int main() {
+    vstr_t *vstr = vstr_new();
+    int i;
+    for (i = 0; i < 10; i++) {
+        vstr_printf(vstr, "%d) this is a test %d %s\n", i, 1234, "'a string'");
+        vstr_add_str(vstr, "-----");
+        vstr_printf(vstr, "this is another test %d %s\n", 1234, "'a second string'");
+        printf("%s", vstr->buf);
+    }
+}
+*/