bmalloc: Added an XSmall line size
[WebKit-https.git] / Source / bmalloc / bmalloc / Allocator.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 Allocator_h
27 #define Allocator_h
28
29 #include "FixedVector.h"
30 #include "MediumAllocator.h"
31 #include "Sizes.h"
32 #include "SmallAllocator.h"
33 #include "XSmallAllocator.h"
34 #include <array>
35
36 namespace bmalloc {
37
38 class Deallocator;
39
40 // Per-cache object allocator.
41
42 class Allocator {
43 public:
44     Allocator(Deallocator&);
45     ~Allocator();
46
47     void* allocate(size_t);
48     bool allocateFastCase(size_t, void*&);
49     void* allocateSlowCase(size_t);
50     
51     void scavenge();
52
53 private:
54     XSmallAllocator& xSmallAllocatorFor(size_t);
55     SmallAllocator& smallAllocatorFor(size_t);
56     
57     void* allocateFastCase(SmallAllocator&);
58
59     void* allocateMedium(size_t);
60     void* allocateLarge(size_t);
61     void* allocateXLarge(size_t);
62     
63     void log(XSmallAllocator&);
64     void log(SmallAllocator&);
65     void log(MediumAllocator&);
66
67     void processXSmallAllocatorLog();
68     void processSmallAllocatorLog();
69     void processMediumAllocatorLog();
70
71     Deallocator& m_deallocator;
72
73     std::array<XSmallAllocator, xSmallMax / alignment> m_xSmallAllocators;
74     std::array<SmallAllocator, smallMax / alignment> m_smallAllocators;
75     MediumAllocator m_mediumAllocator;
76
77     FixedVector<std::pair<XSmallLine*, unsigned char>, xSmallAllocatorLogCapacity> m_xSmallAllocatorLog;
78     FixedVector<std::pair<SmallLine*, unsigned char>, smallAllocatorLogCapacity> m_smallAllocatorLog;
79     FixedVector<std::pair<MediumLine*, unsigned char>, mediumAllocatorLogCapacity> m_mediumAllocatorLog;
80 };
81
82 inline XSmallAllocator& Allocator::xSmallAllocatorFor(size_t size)
83 {
84     size_t index = mask((size - 1ul) / alignment, m_xSmallAllocators.size() - 1);
85     return m_xSmallAllocators[index];
86 }
87
88 inline SmallAllocator& Allocator::smallAllocatorFor(size_t size)
89 {
90     size_t index = mask((size - 1ul) / alignment, m_smallAllocators.size() - 1);
91     return m_smallAllocators[index];
92 }
93
94 inline bool Allocator::allocateFastCase(size_t size, void*& object)
95 {
96     if (size <= xSmallMax) {
97         XSmallAllocator& allocator = xSmallAllocatorFor(size);
98         if (!allocator.canAllocate())
99             return false;
100         
101         object = allocator.allocate();
102         return true;
103     }
104     
105     if (size <= smallMax) {
106         SmallAllocator& allocator = smallAllocatorFor(size);
107         if (!allocator.canAllocate())
108             return false;
109
110         object = allocator.allocate();
111         return true;
112     }
113     
114     return false;
115 }
116
117 inline void* Allocator::allocate(size_t size)
118 {
119     void* object;
120     if (!allocateFastCase(size, object))
121         return allocateSlowCase(size);
122     return object;
123 }
124
125 } // namespace bmalloc
126
127 #endif // Allocator_h