Allow a number of extra bytes in the allocForXXX() methods.

committer: mfx <mfx> 1042575213 +0000

src/mem.cpp

@@ -90,18 +90,20 @@ void MemBuffer::alloc(unsigned size)
 }
 
 
-void MemBuffer::allocForCompression(unsigned uncompressed_size)
+void MemBuffer::allocForCompression(unsigned uncompressed_size, unsigned extra)
 {
     assert((int)uncompressed_size > 0);
-    alloc(uncompressed_size + uncompressed_size/8 + 256, 0);
+    assert((int)extra >= 0);
+    alloc(uncompressed_size + uncompressed_size/8 + 256 + extra, 0);
 }
 
 
-void MemBuffer::allocForUncompression(unsigned uncompressed_size)
+void MemBuffer::allocForUncompression(unsigned uncompressed_size, unsigned extra)
 {
     assert((int)uncompressed_size > 0);
-    //alloc(uncompressed_size + 3 + 512, 0);  // 512 safety bytes
+    assert((int)extra >= 0);
-    alloc(uncompressed_size + 3, 0);  // 3 bytes for asm_fast decompresion
+    //alloc(uncompressed_size + 3 + 512 + extra, 0);  // 512 safety bytes
+    alloc(uncompressed_size + 3 + extra, 0);  // 3 bytes for asm_fast decompresion
 }
 
 

src/mem.h

@@ -41,8 +41,8 @@ public:
     ~MemBuffer();
 
     void alloc(unsigned size);
-    void allocForCompression(unsigned uncompressed_size);
+    void allocForCompression(unsigned uncompressed_size, unsigned extra=0);
-    void allocForUncompression(unsigned uncompressed_size);
+    void allocForUncompression(unsigned uncompressed_size, unsigned extra=0);
 
     void dealloc();