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
34namespace WebCore {
35namespace ArenaTestHelpers {
36
37// An allocator for the PODArena which tracks the regions which have
38// been allocated.
39class TrackedAllocator : public PODArena::FastMallocAllocator {
40public:
41    static PassRefPtr<TrackedAllocator> create()
42    {
43        return adoptRef(new TrackedAllocator);
44    }
45
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
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
61    bool isEmpty() const
62    {
63        return !numRegions();
64    }
65
66    int numRegions() const
67    {
68        return m_allocatedRegions.size();
69    }
70
71private:
72    TrackedAllocator() { }
73    Vector<void*> m_allocatedRegions;
74};
75
76} // namespace ArenaTestHelpers
77} // namespace WebCore
78
79#endif // ArenaTestHelpers_h
80