Fix MemoryPoolAllocator::Clear() to clear user-buffer

Tencent · May 14, 2015 · 0691502 · 0691502
1 parent 1034587
commit 0691502
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/include/rapidjson/allocators.h b/include/rapidjson/allocators.h
@@ -143,11 +143,13 @@ class MemoryPoolAllocator {
 
     //! Deallocates all memory chunks, excluding the user-supplied buffer.
     void Clear() {
-        while(chunkHead_ != 0 && chunkHead_ != userBuffer_) {
+        while (chunkHead_ && chunkHead_ != userBuffer_) {
             ChunkHeader* next = chunkHead_->next;
             baseAllocator_->Free(chunkHead_);
             chunkHead_ = next;
         }
+        if (chunkHead_ && chunkHead_ == userBuffer_)
+            chunkHead_->size = 0; // Clear user buffer
     }
 
     //! Computes the total capacity of allocated memory chunks.