libsanitizer/sanitizer_common/sanitizer_allocator_internal.h

   1 //===-- sanitizer_allocator_internal.h --------------------------*- C++ -*-===//
   2 //
   3 // This file is distributed under the University of Illinois Open Source
   4 // License. See LICENSE.TXT for details.
   5 //
   6 //===----------------------------------------------------------------------===//
   7 //
   8 // This allocator is used inside run-times.
   9 //
  10 //===----------------------------------------------------------------------===//
  11
  12 #ifndef SANITIZER_ALLOCATOR_INTERNAL_H
  13 #define SANITIZER_ALLOCATOR_INTERNAL_H
  14
  15 #include "sanitizer_allocator.h"
  16 #include "sanitizer_internal_defs.h"
  17
  18 namespace __sanitizer {
  19
  20 // FIXME: Check if we may use even more compact size class map for internal
  21 // purposes.
  22 typedef CompactSizeClassMap InternalSizeClassMap;
  23
  24 static const uptr kInternalAllocatorSpace = 0;
  25 static const u64 kInternalAllocatorSize = SANITIZER_MMAP_RANGE_SIZE;
  26 static const uptr kInternalAllocatorRegionSizeLog = 20;
  27 #if SANITIZER_WORDSIZE == 32
  28 static const uptr kInternalAllocatorNumRegions =
  29     kInternalAllocatorSize >> kInternalAllocatorRegionSizeLog;
  30 typedef FlatByteMap<kInternalAllocatorNumRegions> ByteMap;
  31 #else
  32 static const uptr kInternalAllocatorNumRegions =
  33     kInternalAllocatorSize >> kInternalAllocatorRegionSizeLog;
  34 typedef TwoLevelByteMap<(kInternalAllocatorNumRegions >> 12), 1 << 12> ByteMap;
  35 #endif
  36 typedef SizeClassAllocator32<
  37     kInternalAllocatorSpace, kInternalAllocatorSize, 0, InternalSizeClassMap,
  38     kInternalAllocatorRegionSizeLog, ByteMap> PrimaryInternalAllocator;
  39
  40 typedef SizeClassAllocatorLocalCache<PrimaryInternalAllocator>
  41     InternalAllocatorCache;
  42
  43 typedef CombinedAllocator<PrimaryInternalAllocator, InternalAllocatorCache,
  44                           LargeMmapAllocator<> > InternalAllocator;
  45
  46 void *InternalAlloc(uptr size, InternalAllocatorCache *cache = nullptr,
  47                     uptr alignment = 0);
  48 void *InternalRealloc(void *p, uptr size,
  49                       InternalAllocatorCache *cache = nullptr);
  50 void *InternalCalloc(uptr countr, uptr size,
  51                      InternalAllocatorCache *cache = nullptr);
  52 void InternalFree(void *p, InternalAllocatorCache *cache = nullptr);
  53 InternalAllocator *internal_allocator();
  54
  55 enum InternalAllocEnum {
  56   INTERNAL_ALLOC
  57 };
  58
  59 } // namespace __sanitizer
  60
  61 inline void *operator new(__sanitizer::operator_new_size_type size,
  62                           __sanitizer::InternalAllocEnum) {
  63   return __sanitizer::InternalAlloc(size);
  64 }
  65
  66 #endif // SANITIZER_ALLOCATOR_INTERNAL_H