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 "BumpAllocator.h"
  30 #include "FixedVector.h"
  31 #include "MediumAllocator.h"
  32 #include "Sizes.h"
  33 #include "SmallLine.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     void* allocateFastCase(BumpAllocator&);
  55
  56     void* allocateMedium(size_t);
  57     void* allocateLarge(size_t);
  58     void* allocateXLarge(size_t);
  59
  60     Deallocator& m_deallocator;
  61
  62     std::array<BumpAllocator, smallMax / alignment> m_smallAllocators;
  63     std::array<BumpAllocator, mediumMax / alignment> m_mediumAllocators;
  64 };
  65
  66 inline bool Allocator::allocateFastCase(size_t size, void*& object)
  67 {
  68     if (size > smallMax)
  69         return false;
  70
  71     BumpAllocator& allocator = m_smallAllocators[smallSizeClassFor(size)];
  72     if (!allocator.canAllocate())
  73         return false;
  74
  75     object = allocator.allocate();
  76     return true;
  77 }
  78
  79 inline void* Allocator::allocate(size_t size)
  80 {
  81     void* object;
  82     if (!allocateFastCase(size, object))
  83         return allocateSlowCase(size);
  84     return object;
  85 }
  86
  87 } // namespace bmalloc
  88
  89 #endif // Allocator_h