commit 8cd72bdf92fc8d8d735e332295e9582ae20bbaef
author Damien George <damien.p.george@gmail.com> Mon Mar 31 17:10:59 2014 +0100
committer Damien George <damien.p.george@gmail.com> Mon Mar 31 17:10:59 2014 +0100
tree 2f40feb82e3e18d98b4d626ffeb0f068bdd074fc
parent 43e92cfb521db55fb2d0001388c0e3a532405b50

py: Fix vstr_init for case that alloc = 0.

py/vstr.c

diff --git a/py/vstr.c b/py/vstr.c
index 032acc6..4fc4326 100644
--- a/py/vstr.c
+++ b/py/vstr.c
@@ -10,6 +10,10 @@
 #define ROUND_ALLOC(a) (((a) & ((~0) - 7)) + 8)
 
 void vstr_init(vstr_t *vstr, int alloc) {
+    if (alloc < 2) {
+        // need at least 1 byte for the null byte at the end
+        alloc = 2;
+    }
     vstr->alloc = alloc;
     vstr->len = 0;
     vstr->buf = m_new(char, vstr->alloc);