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