[FEATURE] In ByteBuffer: added reference counter

Author: hasherezade

parser/ByteBuffer.cpp

@@ -3,7 +3,7 @@
 
 ByteBuffer::ByteBuffer(bufsize_t v_size, bufsize_t v_padding)
     : content(nullptr), contentSize(0), padding(v_padding),
-    originalSize(0)
+    originalSize(0), m_refs(0)
 {
     if (v_size == 0) throw BufferException("Zero size requested");
 
@@ -14,7 +14,7 @@ ByteBuffer::ByteBuffer(bufsize_t v_size, bufsize_t v_padding)
 
 ByteBuffer::ByteBuffer(BYTE *v_content, bufsize_t v_size, bufsize_t v_padding)
     : content(nullptr), contentSize(0), padding(v_padding),
-    originalSize(0)
+    originalSize(0), m_refs(0)
 {
     if (v_size == 0) throw BufferException("Zero size requested");
 
@@ -28,7 +28,7 @@ ByteBuffer::ByteBuffer(BYTE *v_content, bufsize_t v_size, bufsize_t v_padding)
 
 ByteBuffer::ByteBuffer(AbstractByteBuffer *v_parent, offset_t v_offset, bufsize_t v_size, bufsize_t v_padding)
     : content(NULL), contentSize(0), padding(0),
-    originalSize(0)
+    originalSize(0), m_refs(0)
 {
     if (!v_parent) throw BufferException("Cannot make subBuffer for NULL buffer!");
     if (!v_size) throw BufferException("Cannot make 0 size buffer!");

parser/include/bearparser/ByteBuffer.h

@@ -10,6 +10,22 @@ class ByteBuffer : public AbstractByteBuffer
     ByteBuffer(BYTE *v_content, bufsize_t v_size, bufsize_t padding = DEFAULT_PADDING);
     ByteBuffer(AbstractByteBuffer *sourceBuf, offset_t offset, bufsize_t size, bufsize_t padding = DEFAULT_PADDING);
 
+    static bool release(ByteBuffer* buf)
+    {
+        if (!buf) return false;
+        if (buf->m_refs == 0) {
+            delete buf;
+            return true;
+        }
+        buf->m_refs--;
+        return false;
+    }
+
+    void addRef()
+    {
+        m_refs++;
+    }
+
     virtual ~ByteBuffer();
 
     virtual bufsize_t getContentSize() { return contentSize; }
@@ -26,5 +42,7 @@ class ByteBuffer : public AbstractByteBuffer
     bufsize_t padding;
 
     bufsize_t originalSize;
+
+    size_t m_refs;
 };