e61f4435c5e2144ce46fdc8cf88b6da2dbd8e3f0
[WebKit-https.git] / Source / bmalloc / bmalloc / BumpAllocator.h
1 /*
2  * Copyright (C) 2014 Apple 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  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #ifndef BumpAllocator_h
27 #define BumpAllocator_h
28
29 #include "BAssert.h"
30 #include "BumpRange.h"
31 #include "ObjectType.h"
32
33 namespace bmalloc {
34
35 // Helper object for allocating small objects.
36
37 class BumpAllocator {
38 public:
39     BumpAllocator();
40     void init(size_t);
41     
42     size_t size() { return m_size; }
43     
44     bool isNull() { return !m_ptr; }
45     void clear();
46
47     bool canAllocate() { return !!m_remaining; }
48     void* allocate();
49
50     void refill(const BumpRange&);
51
52 private:
53     void validate(void*);
54
55     char* m_ptr;
56     unsigned short m_size;
57     unsigned short m_remaining;
58 };
59
60 inline BumpAllocator::BumpAllocator()
61     : m_ptr()
62     , m_size()
63     , m_remaining()
64 {
65 }
66
67 inline void BumpAllocator::init(size_t size)
68 {
69     m_ptr = nullptr;
70     m_size = size;
71     m_remaining = 0;
72 }
73
74 inline void BumpAllocator::validate(void* ptr)
75 {
76     UNUSED(ptr);
77     if (m_size <= smallMax) {
78         BASSERT(isSmall(ptr));
79         return;
80     }
81     
82     BASSERT(m_size <= smallMax);
83     BASSERT(isSmall(ptr));
84 }
85
86 inline void* BumpAllocator::allocate()
87 {
88     BASSERT(m_remaining);
89
90     --m_remaining;
91     char* result = m_ptr;
92     m_ptr += m_size;
93     validate(result);
94     return result;
95 }
96
97 inline void BumpAllocator::refill(const BumpRange& bumpRange)
98 {
99     BASSERT(!canAllocate());
100     m_ptr = bumpRange.begin;
101     m_remaining = bumpRange.objectCount;
102 }
103
104 inline void BumpAllocator::clear()
105 {
106     m_ptr = nullptr;
107     m_remaining = 0;
108 }
109
110 } // namespace bmalloc
111
112 #endif // BumpAllocator_h