From e5619a3635dcec97afa2cddb0dc0344466324f6d Mon Sep 17 00:00:00 2001
From: Pixel <pixel@nobis-crew.org>
Date: Sun, 13 Nov 2011 19:34:58 +0100
Subject: Fixing a few bugs with Buffer - yes, it indeed does need a few more
 unit tests....

---
 src/Buffer.cc | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/Buffer.cc')

diff --git a/src/Buffer.cc b/src/Buffer.cc
index e029df5..f276ec0 100644
--- a/src/Buffer.cc
+++ b/src/Buffer.cc
@@ -38,12 +38,16 @@ ssize_t Balau::Buffer::write(const void * buf, size_t count) throw (GeneralExcep
     if (endBlock > oldEndBlock) {
         m_buffer = (uint8_t *) realloc(m_buffer, endBlock * s_blockSize);
         memset(m_buffer + oldEndBlock * s_blockSize, 0, (endBlock - oldEndBlock) * s_blockSize);
+        m_numBlocks = endBlock;
     }
 
     memcpy(m_buffer + cursor, buf, count);
 
     wseek(cursor + count);
 
+    if (m_bufSize < end)
+        m_bufSize = end;
+
     return count;
 }
 
-- 
cgit v1.2.3