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
35namespace bmalloc {
36
37class Deallocator;
38
39// Per-cache object allocator.
40
41class Allocator {
42public:
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
52private:
53    void* allocateFastCase(SmallAllocator&);
54
55    void* allocateMedium(size_t);
56    void* allocateLarge(size_t);
57    void* allocateXLarge(size_t);
58
59    void log(SmallAllocator&);
60    void log(MediumAllocator&);
61
62    void processSmallAllocatorLog();
63    void processMediumAllocatorLog();
64
65    Deallocator& m_deallocator;
66
67    std::array<SmallAllocator, smallMax / alignment> m_smallAllocators;
68    MediumAllocator m_mediumAllocator;
69
70    FixedVector<std::pair<SmallLine*, unsigned char>, smallAllocatorLogCapacity> m_smallAllocatorLog;
71    FixedVector<std::pair<MediumLine*, unsigned char>, mediumAllocatorLogCapacity> m_mediumAllocatorLog;
72};
73
74inline 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
87inline 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
98