1 files changed, 16 insertions, 9 deletions
diff --git a/src/std/buffer.c b/src/std/buffer.c
index 343173a..b0099bb 100644
--- a/src/std/buffer.c
+++ b/src/std/buffer.c
@@ -18,24 +18,25 @@ void tsl_buffer_deinit(TslBuffer *self) {
 
 static int tsl_buffer_ensure_capacity(TslBuffer *self, size_t new_size) {
     void *new_ptr;
+    size_t new_capacity = self->capacity;
     if(new_size <= self->capacity)
         return 1;
+
+    if(new_capacity == 0)
+        new_capacity = 8;
     
-    if(self->capacity != 0) {
-        size_t new_capacity = self->capacity;
-        while(new_capacity < new_size) {
-            new_capacity <<= 1;
-        }
-        new_size = new_capacity;
+    while(new_capacity < new_size) {
+        new_capacity <<= 1;
     }
-    new_ptr = realloc(self->data, new_size);
+
+    new_ptr = realloc(self->data, new_capacity);
     if(!new_ptr) {
         fprintf(stderr, "Error: buffer append failed. Reason: out of memory\n");
         return 0;
     }
 
     self->data = new_ptr;
-    self->capacity = new_size;
+    self->capacity = new_capacity;
     return 1;
 }
 
@@ -47,9 +48,15 @@ int tsl_buffer_append(TslBuffer *self, const void *data, size_t size) {
     return 1;
 }
 
-void tsl_buffer_pop(TslBuffer *self, size_t size) {
+void* tsl_buffer_pop(TslBuffer *self, size_t size) {
     assert(self->size >= size);
     self->size -= size;
+    return (char*)self->data + self->size;
+}
+
+void* tsl_buffer_top(TslBuffer *self, size_t data_size) {
+    tsl_buffer_pop(self, data_size);
+    return (char*)self->data + self->size;
 }
 
 void* tsl_buffer_begin(TslBuffer *self) {