bmalloc should segregate medium-sized objects by line like it does for small-sized...
[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 <array>
34
35 namespace bmalloc {
36
37 class Deallocator;
38
39 // Per-cache object allocator.
40
41 class Allocator {
42 public:
43     Allocator(Deallocator&);
44     ~Allocator();
45
46     void* allocate(size_t);
47     bool allocateFastCase(size_t, void*&);
48     void* allocateSlowCase(size_t);
49     
50     void scavenge();
51
52 private:
53     void* allocateFastCase(SmallAllocator&);
54
55     void* allocateMedium(size_t);
56     void* allocateLarge(size_t);
57     void* allocateXLarge(size_t);
58     
59     void retire(SmallAllocator&);
60     void retire(MediumAllocator&);
61
62     void processSmallAllocatorLog();
63     void processMediumAllocatorLog();
64
65     Deallocator& m_deallocator;
66
67     std::array<SmallAllocator, smallMax / alignment> m_smallAllocators;
68     std::array<MediumAllocator, mediumMax / alignment> m_mediumAllocators;
69
70     FixedVector<std::pair<SmallLine*, unsigned char>, smallAllocatorLogCapacity> m_smallAllocatorLog;
71     FixedVector<std::pair<MediumLine*, unsigned char>, mediumAllocatorLogCapacity> m_mediumAllocatorLog;
72 };
73
74 inline bool Allocator::allocateFastCase(size_t size, void*& object)
75 {
76     if (size > smallMax)
77         return false;
78
79     SmallAllocator& allocator = m_smallAllocators[smallSizeClassFor(size)];
80     if (!allocator.canAllocate())
81         return false;
82
83     object = allocator.allocate();
84     return true;
85 }
86
87 inline void* Allocator::allocate(size_t size)
88 {
89     void* object;
90     if (!allocateFastCase(size, object))
91         return allocateSlowCase(size);
92     return object;
93 }
94
95 } // namespace bmalloc
96
97 #endif // Allocator_h