1 /* 2 * Copyright (C) 2010 Google Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 15 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 16 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 17 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 18 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 19 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 20 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 21 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 #ifndef ArenaTestHelpers_h 27 #define ArenaTestHelpers_h 28 29 #include "PODArena.h" 30 #include <gtest/gtest.h> 31 #include <wtf/NotFound.h> 32 #include <wtf/Vector.h> 33 34 namespace WebCore { 35 namespace ArenaTestHelpers { 36 37 // An allocator for the PODArena which tracks the regions which have 38 // been allocated. 39 class TrackedAllocator : public PODArena::FastMallocAllocator { 40 public: create()41 static PassRefPtr<TrackedAllocator> create() 42 { 43 return adoptRef(new TrackedAllocator); 44 } 45 allocate(size_t size)46 virtual void* allocate(size_t size) 47 { 48 void* result = PODArena::FastMallocAllocator::allocate(size); 49 m_allocatedRegions.append(result); 50 return result; 51 } 52 free(void * ptr)53 virtual void free(void* ptr) 54 { 55 size_t slot = m_allocatedRegions.find(ptr); 56 ASSERT_NE(slot, notFound); 57 m_allocatedRegions.remove(slot); 58 PODArena::FastMallocAllocator::free(ptr); 59 } 60 isEmpty()61 bool isEmpty() const 62 { 63 return !numRegions(); 64 } 65 numRegions()66 int numRegions() const 67 { 68 return m_allocatedRegions.size(); 69 } 70 71 private: TrackedAllocator()72 TrackedAllocator() { } 73 Vector<void*> m_allocatedRegions; 74 }; 75 76 } // namespace ArenaTestHelpers 77 } // namespace WebCore 78 79 #endif // ArenaTestHelpers_h 80