2012-12-04 15:54:41 +08:00
|
|
|
//===-- sanitizer_allocator.h -----------------------------------*- C++ -*-===//
|
|
|
|
//
|
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//
|
|
|
|
// Specialized memory allocator for ThreadSanitizer, MemorySanitizer, etc.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#ifndef SANITIZER_ALLOCATOR_H
|
|
|
|
#define SANITIZER_ALLOCATOR_H
|
|
|
|
|
|
|
|
#include "sanitizer_internal_defs.h"
|
|
|
|
#include "sanitizer_common.h"
|
|
|
|
#include "sanitizer_libc.h"
|
|
|
|
#include "sanitizer_list.h"
|
|
|
|
#include "sanitizer_mutex.h"
|
2013-01-14 16:23:34 +08:00
|
|
|
#include "sanitizer_lfstack.h"
|
2012-12-04 15:54:41 +08:00
|
|
|
|
|
|
|
namespace __sanitizer {
|
|
|
|
|
2012-12-24 21:41:07 +08:00
|
|
|
// SizeClassMap maps allocation sizes into size classes and back.
|
|
|
|
// Class 0 corresponds to size 0.
|
2013-02-07 23:04:06 +08:00
|
|
|
// Classes 1 - 16 correspond to sizes 16 to 256 (size = class_id * 16).
|
2012-12-24 21:41:07 +08:00
|
|
|
// Next 8 classes: 256 + i * 32 (i = 1 to 8).
|
2013-02-07 23:04:06 +08:00
|
|
|
// Next 8 classes: 512 + i * 64 (i = 1 to 8).
|
2012-12-24 21:41:07 +08:00
|
|
|
// ...
|
|
|
|
// Next 8 classes: 2^k + i * 2^(k-3) (i = 1 to 8).
|
|
|
|
// Last class corresponds to kMaxSize = 1 << kMaxSizeLog.
|
|
|
|
//
|
|
|
|
// This structure of the size class map gives us:
|
|
|
|
// - Efficient table-free class-to-size and size-to-class functions.
|
|
|
|
// - Difference between two consequent size classes is betweed 12% and 6%
|
|
|
|
//
|
|
|
|
// This class also gives a hint to a thread-caching allocator about the amount
|
|
|
|
// of chunks that need to be cached per-thread:
|
|
|
|
// - kMaxNumCached is the maximal number of chunks per size class.
|
|
|
|
// - (1 << kMaxBytesCachedLog) is the maximal number of bytes per size class.
|
|
|
|
//
|
|
|
|
// Part of output of SizeClassMap::Print():
|
2013-02-07 23:04:06 +08:00
|
|
|
// c00 => s: 0 diff: +0 00% l 0 cached: 0 0; id 0
|
|
|
|
// c01 => s: 16 diff: +16 00% l 4 cached: 256 4096; id 1
|
|
|
|
// c02 => s: 32 diff: +16 100% l 5 cached: 256 8192; id 2
|
|
|
|
// c03 => s: 48 diff: +16 50% l 5 cached: 256 12288; id 3
|
|
|
|
// c04 => s: 64 diff: +16 33% l 6 cached: 256 16384; id 4
|
|
|
|
// c05 => s: 80 diff: +16 25% l 6 cached: 256 20480; id 5
|
|
|
|
// c06 => s: 96 diff: +16 20% l 6 cached: 256 24576; id 6
|
|
|
|
// c07 => s: 112 diff: +16 16% l 6 cached: 256 28672; id 7
|
2012-12-24 21:41:07 +08:00
|
|
|
//
|
2013-02-07 23:04:06 +08:00
|
|
|
// c08 => s: 128 diff: +16 14% l 7 cached: 256 32768; id 8
|
|
|
|
// c09 => s: 144 diff: +16 12% l 7 cached: 256 36864; id 9
|
|
|
|
// c10 => s: 160 diff: +16 11% l 7 cached: 256 40960; id 10
|
|
|
|
// c11 => s: 176 diff: +16 10% l 7 cached: 256 45056; id 11
|
|
|
|
// c12 => s: 192 diff: +16 09% l 7 cached: 256 49152; id 12
|
|
|
|
// c13 => s: 208 diff: +16 08% l 7 cached: 256 53248; id 13
|
|
|
|
// c14 => s: 224 diff: +16 07% l 7 cached: 256 57344; id 14
|
|
|
|
// c15 => s: 240 diff: +16 07% l 7 cached: 256 61440; id 15
|
2012-12-24 21:41:07 +08:00
|
|
|
//
|
2013-02-07 23:04:06 +08:00
|
|
|
// c16 => s: 256 diff: +16 06% l 8 cached: 256 65536; id 16
|
|
|
|
// c17 => s: 288 diff: +32 12% l 8 cached: 227 65376; id 17
|
|
|
|
// c18 => s: 320 diff: +32 11% l 8 cached: 204 65280; id 18
|
|
|
|
// c19 => s: 352 diff: +32 10% l 8 cached: 186 65472; id 19
|
|
|
|
// c20 => s: 384 diff: +32 09% l 8 cached: 170 65280; id 20
|
|
|
|
// c21 => s: 416 diff: +32 08% l 8 cached: 157 65312; id 21
|
|
|
|
// c22 => s: 448 diff: +32 07% l 8 cached: 146 65408; id 22
|
|
|
|
// c23 => s: 480 diff: +32 07% l 8 cached: 136 65280; id 23
|
2012-12-24 21:41:07 +08:00
|
|
|
//
|
2013-02-07 23:04:06 +08:00
|
|
|
// c24 => s: 512 diff: +32 06% l 9 cached: 128 65536; id 24
|
|
|
|
// c25 => s: 576 diff: +64 12% l 9 cached: 113 65088; id 25
|
|
|
|
// c26 => s: 640 diff: +64 11% l 9 cached: 102 65280; id 26
|
|
|
|
// c27 => s: 704 diff: +64 10% l 9 cached: 93 65472; id 27
|
|
|
|
// c28 => s: 768 diff: +64 09% l 9 cached: 85 65280; id 28
|
|
|
|
// c29 => s: 832 diff: +64 08% l 9 cached: 78 64896; id 29
|
|
|
|
// c30 => s: 896 diff: +64 07% l 9 cached: 73 65408; id 30
|
|
|
|
// c31 => s: 960 diff: +64 07% l 9 cached: 68 65280; id 31
|
2012-12-24 21:41:07 +08:00
|
|
|
//
|
2013-02-07 23:04:06 +08:00
|
|
|
// c32 => s: 1024 diff: +64 06% l 10 cached: 64 65536; id 32
|
2012-12-04 15:54:41 +08:00
|
|
|
|
2013-01-15 17:02:20 +08:00
|
|
|
template <uptr kMaxSizeLog, uptr kMaxNumCachedT, uptr kMaxBytesCachedLog,
|
2013-01-12 00:41:19 +08:00
|
|
|
uptr kMinBatchClassT>
|
2012-12-24 21:41:07 +08:00
|
|
|
class SizeClassMap {
|
2013-02-07 23:04:06 +08:00
|
|
|
static const uptr kMinSizeLog = 4;
|
2012-12-24 21:41:07 +08:00
|
|
|
static const uptr kMidSizeLog = kMinSizeLog + 4;
|
|
|
|
static const uptr kMinSize = 1 << kMinSizeLog;
|
|
|
|
static const uptr kMidSize = 1 << kMidSizeLog;
|
|
|
|
static const uptr kMidClass = kMidSize / kMinSize;
|
|
|
|
static const uptr S = 3;
|
|
|
|
static const uptr M = (1 << S) - 1;
|
|
|
|
|
|
|
|
public:
|
2013-01-15 17:02:20 +08:00
|
|
|
static const uptr kMaxNumCached = kMaxNumCachedT;
|
2013-01-12 00:41:19 +08:00
|
|
|
struct TransferBatch {
|
|
|
|
TransferBatch *next;
|
|
|
|
uptr count;
|
|
|
|
void *batch[kMaxNumCached];
|
|
|
|
};
|
|
|
|
|
|
|
|
static const uptr kMinBatchClass = kMinBatchClassT;
|
2012-12-24 21:41:07 +08:00
|
|
|
static const uptr kMaxSize = 1 << kMaxSizeLog;
|
|
|
|
static const uptr kNumClasses =
|
|
|
|
kMidClass + ((kMaxSizeLog - kMidSizeLog) << S) + 1;
|
|
|
|
COMPILER_CHECK(kNumClasses >= 32 && kNumClasses <= 256);
|
|
|
|
static const uptr kNumClassesRounded =
|
|
|
|
kNumClasses == 32 ? 32 :
|
|
|
|
kNumClasses <= 64 ? 64 :
|
|
|
|
kNumClasses <= 128 ? 128 : 256;
|
2012-12-04 15:54:41 +08:00
|
|
|
|
|
|
|
static uptr Size(uptr class_id) {
|
2012-12-24 21:41:07 +08:00
|
|
|
if (class_id <= kMidClass)
|
|
|
|
return kMinSize * class_id;
|
|
|
|
class_id -= kMidClass;
|
|
|
|
uptr t = kMidSize << (class_id >> S);
|
|
|
|
return t + (t >> S) * (class_id & M);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
2012-12-24 21:41:07 +08:00
|
|
|
|
2012-12-04 15:54:41 +08:00
|
|
|
static uptr ClassID(uptr size) {
|
2012-12-24 21:41:07 +08:00
|
|
|
if (size <= kMidSize)
|
|
|
|
return (size + kMinSize - 1) >> kMinSizeLog;
|
|
|
|
if (size > kMaxSize) return 0;
|
2013-02-08 20:02:00 +08:00
|
|
|
uptr l = MostSignificantSetBitIndex(size);
|
2012-12-24 21:41:07 +08:00
|
|
|
uptr hbits = (size >> (l - S)) & M;
|
|
|
|
uptr lbits = size & ((1 << (l - S)) - 1);
|
|
|
|
uptr l1 = l - kMidSizeLog;
|
|
|
|
return kMidClass + (l1 << S) + hbits + (lbits > 0);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static uptr MaxCached(uptr class_id) {
|
2012-12-24 21:41:07 +08:00
|
|
|
if (class_id == 0) return 0;
|
|
|
|
uptr n = (1UL << kMaxBytesCachedLog) / Size(class_id);
|
2013-02-04 16:07:45 +08:00
|
|
|
return Max<uptr>(1, Min(kMaxNumCached, n));
|
2012-12-24 21:41:07 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static void Print() {
|
|
|
|
uptr prev_s = 0;
|
|
|
|
uptr total_cached = 0;
|
|
|
|
for (uptr i = 0; i < kNumClasses; i++) {
|
|
|
|
uptr s = Size(i);
|
|
|
|
if (s >= kMidSize / 2 && (s & (s - 1)) == 0)
|
|
|
|
Printf("\n");
|
|
|
|
uptr d = s - prev_s;
|
|
|
|
uptr p = prev_s ? (d * 100 / prev_s) : 0;
|
2013-03-01 16:49:14 +08:00
|
|
|
uptr l = s ? MostSignificantSetBitIndex(s) : 0;
|
2012-12-24 21:41:07 +08:00
|
|
|
uptr cached = MaxCached(i) * s;
|
|
|
|
Printf("c%02zd => s: %zd diff: +%zd %02zd%% l %zd "
|
|
|
|
"cached: %zd %zd; id %zd\n",
|
|
|
|
i, Size(i), d, p, l, MaxCached(i), cached, ClassID(s));
|
|
|
|
total_cached += cached;
|
|
|
|
prev_s = s;
|
|
|
|
}
|
|
|
|
Printf("Total cached: %zd\n", total_cached);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void Validate() {
|
|
|
|
for (uptr c = 1; c < kNumClasses; c++) {
|
|
|
|
// Printf("Validate: c%zd\n", c);
|
|
|
|
uptr s = Size(c);
|
2013-02-26 20:59:06 +08:00
|
|
|
CHECK_NE(s, 0U);
|
2012-12-24 21:41:07 +08:00
|
|
|
CHECK_EQ(ClassID(s), c);
|
|
|
|
if (c != kNumClasses - 1)
|
|
|
|
CHECK_EQ(ClassID(s + 1), c + 1);
|
|
|
|
CHECK_EQ(ClassID(s - 1), c);
|
|
|
|
if (c)
|
|
|
|
CHECK_GT(Size(c), Size(c-1));
|
|
|
|
}
|
|
|
|
CHECK_EQ(ClassID(kMaxSize + 1), 0);
|
|
|
|
|
|
|
|
for (uptr s = 1; s <= kMaxSize; s++) {
|
|
|
|
uptr c = ClassID(s);
|
|
|
|
// Printf("s%zd => c%zd\n", s, c);
|
|
|
|
CHECK_LT(c, kNumClasses);
|
|
|
|
CHECK_GE(Size(c), s);
|
|
|
|
if (c > 0)
|
|
|
|
CHECK_LT(Size(c-1), s);
|
|
|
|
}
|
2012-12-04 15:54:41 +08:00
|
|
|
|
2013-01-12 00:41:19 +08:00
|
|
|
// TransferBatch for kMinBatchClass must fit into the block itself.
|
|
|
|
const uptr batch_size = sizeof(TransferBatch)
|
|
|
|
- sizeof(void*) // NOLINT
|
|
|
|
* (kMaxNumCached - MaxCached(kMinBatchClass));
|
|
|
|
CHECK_LE(batch_size, Size(kMinBatchClass));
|
|
|
|
// TransferBatch for kMinBatchClass-1 must not fit into the block itself.
|
|
|
|
const uptr batch_size1 = sizeof(TransferBatch)
|
|
|
|
- sizeof(void*) // NOLINT
|
|
|
|
* (kMaxNumCached - MaxCached(kMinBatchClass - 1));
|
|
|
|
CHECK_GT(batch_size1, Size(kMinBatchClass - 1));
|
|
|
|
}
|
2012-12-04 15:54:41 +08:00
|
|
|
};
|
|
|
|
|
2013-02-07 23:04:06 +08:00
|
|
|
typedef SizeClassMap<17, 256, 16, FIRST_32_SECOND_64(25, 28)>
|
2013-01-12 00:41:19 +08:00
|
|
|
DefaultSizeClassMap;
|
2013-02-07 23:04:06 +08:00
|
|
|
typedef SizeClassMap<17, 64, 14, FIRST_32_SECOND_64(17, 20)>
|
2013-01-12 00:41:19 +08:00
|
|
|
CompactSizeClassMap;
|
|
|
|
template<class SizeClassAllocator> struct SizeClassAllocatorLocalCache;
|
2012-12-04 15:54:41 +08:00
|
|
|
|
2013-01-24 17:08:03 +08:00
|
|
|
// Memory allocator statistics
|
|
|
|
enum AllocatorStat {
|
|
|
|
AllocatorStatMalloced,
|
|
|
|
AllocatorStatFreed,
|
|
|
|
AllocatorStatMmapped,
|
|
|
|
AllocatorStatUnmapped,
|
|
|
|
AllocatorStatCount
|
|
|
|
};
|
|
|
|
|
|
|
|
typedef u64 AllocatorStatCounters[AllocatorStatCount];
|
|
|
|
|
|
|
|
// Per-thread stats, live in per-thread cache.
|
|
|
|
class AllocatorStats {
|
|
|
|
public:
|
|
|
|
void Init() {
|
|
|
|
internal_memset(this, 0, sizeof(*this));
|
|
|
|
}
|
|
|
|
|
|
|
|
void Add(AllocatorStat i, u64 v) {
|
|
|
|
v += atomic_load(&stats_[i], memory_order_relaxed);
|
|
|
|
atomic_store(&stats_[i], v, memory_order_relaxed);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Set(AllocatorStat i, u64 v) {
|
|
|
|
atomic_store(&stats_[i], v, memory_order_relaxed);
|
|
|
|
}
|
|
|
|
|
|
|
|
u64 Get(AllocatorStat i) const {
|
|
|
|
return atomic_load(&stats_[i], memory_order_relaxed);
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
friend class AllocatorGlobalStats;
|
|
|
|
AllocatorStats *next_;
|
|
|
|
AllocatorStats *prev_;
|
|
|
|
atomic_uint64_t stats_[AllocatorStatCount];
|
|
|
|
};
|
|
|
|
|
|
|
|
// Global stats, used for aggregation and querying.
|
|
|
|
class AllocatorGlobalStats : public AllocatorStats {
|
|
|
|
public:
|
|
|
|
void Init() {
|
|
|
|
internal_memset(this, 0, sizeof(*this));
|
|
|
|
next_ = this;
|
|
|
|
prev_ = this;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Register(AllocatorStats *s) {
|
|
|
|
SpinMutexLock l(&mu_);
|
|
|
|
s->next_ = next_;
|
|
|
|
s->prev_ = this;
|
|
|
|
next_->prev_ = s;
|
|
|
|
next_ = s;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Unregister(AllocatorStats *s) {
|
|
|
|
SpinMutexLock l(&mu_);
|
|
|
|
s->prev_->next_ = s->next_;
|
|
|
|
s->next_->prev_ = s->prev_;
|
|
|
|
for (int i = 0; i < AllocatorStatCount; i++)
|
|
|
|
Add(AllocatorStat(i), s->Get(AllocatorStat(i)));
|
|
|
|
}
|
|
|
|
|
|
|
|
void Get(AllocatorStatCounters s) const {
|
|
|
|
internal_memset(s, 0, AllocatorStatCount * sizeof(u64));
|
|
|
|
SpinMutexLock l(&mu_);
|
|
|
|
const AllocatorStats *stats = this;
|
|
|
|
for (;;) {
|
|
|
|
for (int i = 0; i < AllocatorStatCount; i++)
|
|
|
|
s[i] += stats->Get(AllocatorStat(i));
|
|
|
|
stats = stats->next_;
|
|
|
|
if (stats == this)
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
mutable SpinMutex mu_;
|
|
|
|
};
|
|
|
|
|
2012-12-12 22:32:18 +08:00
|
|
|
// Allocators call these callbacks on mmap/munmap.
|
|
|
|
struct NoOpMapUnmapCallback {
|
|
|
|
void OnMap(uptr p, uptr size) const { }
|
|
|
|
void OnUnmap(uptr p, uptr size) const { }
|
|
|
|
};
|
|
|
|
|
2012-12-05 18:09:15 +08:00
|
|
|
// SizeClassAllocator64 -- allocator for 64-bit address space.
|
|
|
|
//
|
|
|
|
// Space: a portion of address space of kSpaceSize bytes starting at
|
|
|
|
// a fixed address (kSpaceBeg). Both constants are powers of two and
|
|
|
|
// kSpaceBeg is kSpaceSize-aligned.
|
2012-12-13 13:05:11 +08:00
|
|
|
// At the beginning the entire space is mprotect-ed, then small parts of it
|
|
|
|
// are mapped on demand.
|
2012-12-05 18:09:15 +08:00
|
|
|
//
|
|
|
|
// Region: a part of Space dedicated to a single size class.
|
|
|
|
// There are kNumClasses Regions of equal size.
|
|
|
|
//
|
|
|
|
// UserChunk: a piece of memory returned to user.
|
|
|
|
// MetaChunk: kMetadataSize bytes of metadata associated with a UserChunk.
|
|
|
|
//
|
|
|
|
// A Region looks like this:
|
|
|
|
// UserChunk1 ... UserChunkN <gap> MetaChunkN ... MetaChunk1
|
|
|
|
template <const uptr kSpaceBeg, const uptr kSpaceSize,
|
2012-12-12 22:32:18 +08:00
|
|
|
const uptr kMetadataSize, class SizeClassMap,
|
|
|
|
class MapUnmapCallback = NoOpMapUnmapCallback>
|
2012-12-05 18:09:15 +08:00
|
|
|
class SizeClassAllocator64 {
|
|
|
|
public:
|
2013-01-12 00:41:19 +08:00
|
|
|
typedef typename SizeClassMap::TransferBatch Batch;
|
|
|
|
typedef SizeClassAllocator64<kSpaceBeg, kSpaceSize, kMetadataSize,
|
|
|
|
SizeClassMap, MapUnmapCallback> ThisT;
|
|
|
|
typedef SizeClassAllocatorLocalCache<ThisT> AllocatorCache;
|
|
|
|
|
2012-12-05 18:09:15 +08:00
|
|
|
void Init() {
|
2012-12-13 13:05:11 +08:00
|
|
|
CHECK_EQ(kSpaceBeg,
|
|
|
|
reinterpret_cast<uptr>(Mprotect(kSpaceBeg, kSpaceSize)));
|
2012-12-12 22:32:18 +08:00
|
|
|
MapWithCallback(kSpaceEnd, AdditionalSize());
|
|
|
|
}
|
|
|
|
|
|
|
|
void MapWithCallback(uptr beg, uptr size) {
|
2012-12-13 13:36:00 +08:00
|
|
|
CHECK_EQ(beg, reinterpret_cast<uptr>(MmapFixedOrDie(beg, size)));
|
2012-12-12 22:32:18 +08:00
|
|
|
MapUnmapCallback().OnMap(beg, size);
|
|
|
|
}
|
|
|
|
|
|
|
|
void UnmapWithCallback(uptr beg, uptr size) {
|
|
|
|
MapUnmapCallback().OnUnmap(beg, size);
|
|
|
|
UnmapOrDie(reinterpret_cast<void *>(beg), size);
|
2012-12-05 18:09:15 +08:00
|
|
|
}
|
|
|
|
|
2012-12-25 17:40:20 +08:00
|
|
|
static bool CanAllocate(uptr size, uptr alignment) {
|
2012-12-05 18:09:15 +08:00
|
|
|
return size <= SizeClassMap::kMaxSize &&
|
|
|
|
alignment <= SizeClassMap::kMaxSize;
|
|
|
|
}
|
|
|
|
|
2013-01-29 01:29:50 +08:00
|
|
|
NOINLINE Batch* AllocateBatch(AllocatorStats *stat, AllocatorCache *c,
|
2013-01-24 17:08:03 +08:00
|
|
|
uptr class_id) {
|
2012-12-05 18:09:15 +08:00
|
|
|
CHECK_LT(class_id, kNumClasses);
|
|
|
|
RegionInfo *region = GetRegionInfo(class_id);
|
2013-01-14 16:23:34 +08:00
|
|
|
Batch *b = region->free_list.Pop();
|
|
|
|
if (b == 0)
|
2013-01-24 17:08:03 +08:00
|
|
|
b = PopulateFreeList(stat, c, class_id, region);
|
2013-01-14 16:51:08 +08:00
|
|
|
region->n_allocated += b->count;
|
2013-01-12 00:41:19 +08:00
|
|
|
return b;
|
2012-12-05 18:09:15 +08:00
|
|
|
}
|
|
|
|
|
2013-01-29 01:29:50 +08:00
|
|
|
NOINLINE void DeallocateBatch(AllocatorStats *stat, uptr class_id, Batch *b) {
|
2012-12-05 18:09:15 +08:00
|
|
|
RegionInfo *region = GetRegionInfo(class_id);
|
2013-01-14 16:23:34 +08:00
|
|
|
region->free_list.Push(b);
|
2013-01-14 16:51:08 +08:00
|
|
|
region->n_freed += b->count;
|
2012-12-05 18:09:15 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static bool PointerIsMine(void *p) {
|
|
|
|
return reinterpret_cast<uptr>(p) / kSpaceSize == kSpaceBeg / kSpaceSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
static uptr GetSizeClass(void *p) {
|
2012-12-24 21:41:07 +08:00
|
|
|
return (reinterpret_cast<uptr>(p) / kRegionSize) % kNumClassesRounded;
|
2012-12-05 18:09:15 +08:00
|
|
|
}
|
|
|
|
|
2012-12-19 16:32:50 +08:00
|
|
|
void *GetBlockBegin(void *p) {
|
2012-12-05 18:09:15 +08:00
|
|
|
uptr class_id = GetSizeClass(p);
|
|
|
|
uptr size = SizeClassMap::Size(class_id);
|
|
|
|
uptr chunk_idx = GetChunkIdx((uptr)p, size);
|
|
|
|
uptr reg_beg = (uptr)p & ~(kRegionSize - 1);
|
2012-12-21 21:38:05 +08:00
|
|
|
uptr beg = chunk_idx * size;
|
|
|
|
uptr next_beg = beg + size;
|
2012-12-19 16:32:50 +08:00
|
|
|
RegionInfo *region = GetRegionInfo(class_id);
|
2012-12-21 21:38:05 +08:00
|
|
|
if (region->mapped_user >= next_beg)
|
|
|
|
return reinterpret_cast<void*>(reg_beg + beg);
|
2012-12-19 16:32:50 +08:00
|
|
|
return 0;
|
2012-12-05 18:09:15 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static uptr GetActuallyAllocatedSize(void *p) {
|
|
|
|
CHECK(PointerIsMine(p));
|
|
|
|
return SizeClassMap::Size(GetSizeClass(p));
|
|
|
|
}
|
|
|
|
|
|
|
|
uptr ClassID(uptr size) { return SizeClassMap::ClassID(size); }
|
|
|
|
|
|
|
|
void *GetMetaData(void *p) {
|
|
|
|
uptr class_id = GetSizeClass(p);
|
|
|
|
uptr size = SizeClassMap::Size(class_id);
|
|
|
|
uptr chunk_idx = GetChunkIdx(reinterpret_cast<uptr>(p), size);
|
|
|
|
return reinterpret_cast<void*>(kSpaceBeg + (kRegionSize * (class_id + 1)) -
|
|
|
|
(1 + chunk_idx) * kMetadataSize);
|
|
|
|
}
|
|
|
|
|
|
|
|
uptr TotalMemoryUsed() {
|
|
|
|
uptr res = 0;
|
|
|
|
for (uptr i = 0; i < kNumClasses; i++)
|
|
|
|
res += GetRegionInfo(i)->allocated_user;
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test-only.
|
|
|
|
void TestOnlyUnmap() {
|
2012-12-12 22:32:18 +08:00
|
|
|
UnmapWithCallback(kSpaceBeg, kSpaceSize + AdditionalSize());
|
2012-12-05 18:09:15 +08:00
|
|
|
}
|
|
|
|
|
2012-12-27 22:09:19 +08:00
|
|
|
void PrintStats() {
|
|
|
|
uptr total_mapped = 0;
|
|
|
|
uptr n_allocated = 0;
|
|
|
|
uptr n_freed = 0;
|
|
|
|
for (uptr class_id = 1; class_id < kNumClasses; class_id++) {
|
|
|
|
RegionInfo *region = GetRegionInfo(class_id);
|
|
|
|
total_mapped += region->mapped_user;
|
|
|
|
n_allocated += region->n_allocated;
|
|
|
|
n_freed += region->n_freed;
|
|
|
|
}
|
|
|
|
Printf("Stats: SizeClassAllocator64: %zdM mapped in %zd allocations; "
|
|
|
|
"remains %zd\n",
|
|
|
|
total_mapped >> 20, n_allocated, n_allocated - n_freed);
|
|
|
|
for (uptr class_id = 1; class_id < kNumClasses; class_id++) {
|
|
|
|
RegionInfo *region = GetRegionInfo(class_id);
|
|
|
|
if (region->mapped_user == 0) continue;
|
|
|
|
Printf(" %02zd (%zd): total: %zd K allocs: %zd remains: %zd\n",
|
|
|
|
class_id,
|
|
|
|
SizeClassMap::Size(class_id),
|
|
|
|
region->mapped_user >> 10,
|
|
|
|
region->n_allocated,
|
|
|
|
region->n_allocated - region->n_freed);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-02-07 19:40:03 +08:00
|
|
|
// ForceLock() and ForceUnlock() are needed to implement Darwin malloc zone
|
|
|
|
// introspection API.
|
|
|
|
void ForceLock() {
|
|
|
|
for (uptr i = 0; i < kNumClasses; i++) {
|
|
|
|
GetRegionInfo(i)->mutex.Lock();
|
2013-02-07 22:58:04 +08:00
|
|
|
}
|
2013-02-07 19:40:03 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void ForceUnlock() {
|
|
|
|
for (int i = (int)kNumClasses - 1; i >= 0; i--) {
|
|
|
|
GetRegionInfo(i)->mutex.Unlock();
|
2013-02-07 22:58:04 +08:00
|
|
|
}
|
2013-02-07 19:40:03 +08:00
|
|
|
}
|
|
|
|
|
2012-12-05 18:09:15 +08:00
|
|
|
typedef SizeClassMap SizeClassMapT;
|
2012-12-24 21:41:07 +08:00
|
|
|
static const uptr kNumClasses = SizeClassMap::kNumClasses;
|
|
|
|
static const uptr kNumClassesRounded = SizeClassMap::kNumClassesRounded;
|
2012-12-05 18:09:15 +08:00
|
|
|
|
|
|
|
private:
|
2012-12-24 21:41:07 +08:00
|
|
|
static const uptr kRegionSize = kSpaceSize / kNumClassesRounded;
|
2012-12-12 22:32:18 +08:00
|
|
|
static const uptr kSpaceEnd = kSpaceBeg + kSpaceSize;
|
2012-12-05 18:09:15 +08:00
|
|
|
COMPILER_CHECK(kSpaceBeg % kSpaceSize == 0);
|
|
|
|
// kRegionSize must be >= 2^32.
|
|
|
|
COMPILER_CHECK((kRegionSize) >= (1ULL << (SANITIZER_WORDSIZE / 2)));
|
|
|
|
// Populate the free list with at most this number of bytes at once
|
|
|
|
// or with one element if its size is greater.
|
2013-01-12 00:41:19 +08:00
|
|
|
static const uptr kPopulateSize = 1 << 14;
|
2012-12-21 21:38:05 +08:00
|
|
|
// Call mmap for user memory with at least this size.
|
2013-01-23 22:07:17 +08:00
|
|
|
static const uptr kUserMapSize = 1 << 16;
|
2012-12-21 21:38:05 +08:00
|
|
|
// Call mmap for metadata memory with at least this size.
|
|
|
|
static const uptr kMetaMapSize = 1 << 16;
|
2012-12-05 18:09:15 +08:00
|
|
|
|
|
|
|
struct RegionInfo {
|
2013-01-14 15:51:39 +08:00
|
|
|
BlockingMutex mutex;
|
2013-01-14 16:23:34 +08:00
|
|
|
LFStack<Batch> free_list;
|
2012-12-05 18:09:15 +08:00
|
|
|
uptr allocated_user; // Bytes allocated for user memory.
|
|
|
|
uptr allocated_meta; // Bytes allocated for metadata.
|
2012-12-13 13:05:11 +08:00
|
|
|
uptr mapped_user; // Bytes mapped for user memory.
|
|
|
|
uptr mapped_meta; // Bytes mapped for metadata.
|
2012-12-27 22:09:19 +08:00
|
|
|
uptr n_allocated, n_freed; // Just stats.
|
2012-12-05 18:09:15 +08:00
|
|
|
};
|
2012-12-13 13:05:11 +08:00
|
|
|
COMPILER_CHECK(sizeof(RegionInfo) >= kCacheLineSize);
|
2012-12-05 18:09:15 +08:00
|
|
|
|
|
|
|
static uptr AdditionalSize() {
|
2012-12-27 22:09:19 +08:00
|
|
|
return RoundUpTo(sizeof(RegionInfo) * kNumClassesRounded,
|
|
|
|
GetPageSizeCached());
|
2012-12-05 18:09:15 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
RegionInfo *GetRegionInfo(uptr class_id) {
|
|
|
|
CHECK_LT(class_id, kNumClasses);
|
|
|
|
RegionInfo *regions = reinterpret_cast<RegionInfo*>(kSpaceBeg + kSpaceSize);
|
|
|
|
return ®ions[class_id];
|
|
|
|
}
|
|
|
|
|
|
|
|
static uptr GetChunkIdx(uptr chunk, uptr size) {
|
|
|
|
u32 offset = chunk % kRegionSize;
|
|
|
|
// Here we divide by a non-constant. This is costly.
|
|
|
|
// We require that kRegionSize is at least 2^32 so that offset is 32-bit.
|
|
|
|
// We save 2x by using 32-bit div, but may need to use a 256-way switch.
|
|
|
|
return offset / (u32)size;
|
|
|
|
}
|
|
|
|
|
2013-01-29 01:29:50 +08:00
|
|
|
NOINLINE Batch* PopulateFreeList(AllocatorStats *stat, AllocatorCache *c,
|
2013-01-24 17:08:03 +08:00
|
|
|
uptr class_id, RegionInfo *region) {
|
2013-01-14 16:23:34 +08:00
|
|
|
BlockingMutexLock l(®ion->mutex);
|
|
|
|
Batch *b = region->free_list.Pop();
|
|
|
|
if (b)
|
|
|
|
return b;
|
2012-12-05 18:09:15 +08:00
|
|
|
uptr size = SizeClassMap::Size(class_id);
|
2013-01-12 00:41:19 +08:00
|
|
|
uptr count = size < kPopulateSize ? SizeClassMap::MaxCached(class_id) : 1;
|
2012-12-05 18:09:15 +08:00
|
|
|
uptr beg_idx = region->allocated_user;
|
2013-01-12 00:41:19 +08:00
|
|
|
uptr end_idx = beg_idx + count * size;
|
2012-12-05 18:09:15 +08:00
|
|
|
uptr region_beg = kSpaceBeg + kRegionSize * class_id;
|
2012-12-19 14:51:45 +08:00
|
|
|
if (end_idx + size > region->mapped_user) {
|
2012-12-13 13:05:11 +08:00
|
|
|
// Do the mmap for the user memory.
|
2012-12-21 21:38:05 +08:00
|
|
|
uptr map_size = kUserMapSize;
|
|
|
|
while (end_idx + size > region->mapped_user + map_size)
|
|
|
|
map_size += kUserMapSize;
|
2012-12-24 21:41:07 +08:00
|
|
|
CHECK_GE(region->mapped_user + map_size, end_idx);
|
2012-12-21 21:38:05 +08:00
|
|
|
MapWithCallback(region_beg + region->mapped_user, map_size);
|
2013-01-24 17:08:03 +08:00
|
|
|
stat->Add(AllocatorStatMmapped, map_size);
|
2012-12-21 21:38:05 +08:00
|
|
|
region->mapped_user += map_size;
|
2012-12-13 13:05:11 +08:00
|
|
|
}
|
2013-01-14 18:49:11 +08:00
|
|
|
uptr total_count = (region->mapped_user - beg_idx - size)
|
|
|
|
/ size / count * count;
|
|
|
|
region->allocated_meta += total_count * kMetadataSize;
|
2012-12-13 13:05:11 +08:00
|
|
|
if (region->allocated_meta > region->mapped_meta) {
|
2012-12-21 21:38:05 +08:00
|
|
|
uptr map_size = kMetaMapSize;
|
|
|
|
while (region->allocated_meta > region->mapped_meta + map_size)
|
|
|
|
map_size += kMetaMapSize;
|
2012-12-13 13:05:11 +08:00
|
|
|
// Do the mmap for the metadata.
|
2012-12-21 21:38:05 +08:00
|
|
|
CHECK_GE(region->mapped_meta + map_size, region->allocated_meta);
|
2012-12-13 13:05:11 +08:00
|
|
|
MapWithCallback(region_beg + kRegionSize -
|
2012-12-21 21:38:05 +08:00
|
|
|
region->mapped_meta - map_size, map_size);
|
|
|
|
region->mapped_meta += map_size;
|
2012-12-13 13:05:11 +08:00
|
|
|
}
|
2012-12-21 21:38:05 +08:00
|
|
|
CHECK_LE(region->allocated_meta, region->mapped_meta);
|
2012-12-05 18:09:15 +08:00
|
|
|
if (region->allocated_user + region->allocated_meta > kRegionSize) {
|
|
|
|
Printf("Out of memory. Dying.\n");
|
|
|
|
Printf("The process has exhausted %zuMB for size class %zu.\n",
|
|
|
|
kRegionSize / 1024 / 1024, size);
|
|
|
|
Die();
|
|
|
|
}
|
2013-01-14 18:49:11 +08:00
|
|
|
for (;;) {
|
|
|
|
if (class_id < SizeClassMap::kMinBatchClass)
|
|
|
|
b = (Batch*)c->Allocate(this, SizeClassMap::ClassID(sizeof(Batch)));
|
|
|
|
else
|
|
|
|
b = (Batch*)(region_beg + beg_idx);
|
|
|
|
b->count = count;
|
|
|
|
for (uptr i = 0; i < count; i++)
|
|
|
|
b->batch[i] = (void*)(region_beg + beg_idx + i * size);
|
|
|
|
region->allocated_user += count * size;
|
|
|
|
CHECK_LE(region->allocated_user, region->mapped_user);
|
|
|
|
beg_idx += count * size;
|
|
|
|
if (beg_idx + count * size + size > region->mapped_user)
|
|
|
|
break;
|
|
|
|
region->free_list.Push(b);
|
|
|
|
}
|
2013-01-14 16:23:34 +08:00
|
|
|
return b;
|
2012-12-05 18:09:15 +08:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2012-12-06 20:49:28 +08:00
|
|
|
// SizeClassAllocator32 -- allocator for 32-bit address space.
|
|
|
|
// This allocator can theoretically be used on 64-bit arch, but there it is less
|
|
|
|
// efficient than SizeClassAllocator64.
|
|
|
|
//
|
|
|
|
// [kSpaceBeg, kSpaceBeg + kSpaceSize) is the range of addresses which can
|
|
|
|
// be returned by MmapOrDie().
|
|
|
|
//
|
|
|
|
// Region:
|
|
|
|
// a result of a single call to MmapAlignedOrDie(kRegionSize, kRegionSize).
|
|
|
|
// Since the regions are aligned by kRegionSize, there are exactly
|
|
|
|
// kNumPossibleRegions possible regions in the address space and so we keep
|
2012-12-12 22:32:18 +08:00
|
|
|
// an u8 array possible_regions[kNumPossibleRegions] to store the size classes.
|
2012-12-06 20:49:28 +08:00
|
|
|
// 0 size class means the region is not used by the allocator.
|
|
|
|
//
|
|
|
|
// One Region is used to allocate chunks of a single size class.
|
|
|
|
// A Region looks like this:
|
|
|
|
// UserChunk1 .. UserChunkN <gap> MetaChunkN .. MetaChunk1
|
|
|
|
//
|
|
|
|
// In order to avoid false sharing the objects of this class should be
|
|
|
|
// chache-line aligned.
|
|
|
|
template <const uptr kSpaceBeg, const u64 kSpaceSize,
|
2012-12-12 22:32:18 +08:00
|
|
|
const uptr kMetadataSize, class SizeClassMap,
|
|
|
|
class MapUnmapCallback = NoOpMapUnmapCallback>
|
2012-12-06 20:49:28 +08:00
|
|
|
class SizeClassAllocator32 {
|
|
|
|
public:
|
2013-01-12 00:41:19 +08:00
|
|
|
typedef typename SizeClassMap::TransferBatch Batch;
|
|
|
|
typedef SizeClassAllocator32<kSpaceBeg, kSpaceSize, kMetadataSize,
|
|
|
|
SizeClassMap, MapUnmapCallback> ThisT;
|
|
|
|
typedef SizeClassAllocatorLocalCache<ThisT> AllocatorCache;
|
|
|
|
|
2012-12-06 20:49:28 +08:00
|
|
|
void Init() {
|
2012-12-12 22:32:18 +08:00
|
|
|
state_ = reinterpret_cast<State *>(MapWithCallback(sizeof(State)));
|
|
|
|
}
|
|
|
|
|
|
|
|
void *MapWithCallback(uptr size) {
|
|
|
|
size = RoundUpTo(size, GetPageSizeCached());
|
|
|
|
void *res = MmapOrDie(size, "SizeClassAllocator32");
|
|
|
|
MapUnmapCallback().OnMap((uptr)res, size);
|
|
|
|
return res;
|
|
|
|
}
|
2013-01-24 17:08:03 +08:00
|
|
|
|
2012-12-12 22:32:18 +08:00
|
|
|
void UnmapWithCallback(uptr beg, uptr size) {
|
|
|
|
MapUnmapCallback().OnUnmap(beg, size);
|
|
|
|
UnmapOrDie(reinterpret_cast<void *>(beg), size);
|
2012-12-06 20:49:28 +08:00
|
|
|
}
|
|
|
|
|
2012-12-25 17:40:20 +08:00
|
|
|
static bool CanAllocate(uptr size, uptr alignment) {
|
2012-12-06 20:49:28 +08:00
|
|
|
return size <= SizeClassMap::kMaxSize &&
|
|
|
|
alignment <= SizeClassMap::kMaxSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
void *GetMetaData(void *p) {
|
|
|
|
CHECK(PointerIsMine(p));
|
|
|
|
uptr mem = reinterpret_cast<uptr>(p);
|
|
|
|
uptr beg = ComputeRegionBeg(mem);
|
|
|
|
uptr size = SizeClassMap::Size(GetSizeClass(p));
|
|
|
|
u32 offset = mem - beg;
|
|
|
|
uptr n = offset / (u32)size; // 32-bit division
|
|
|
|
uptr meta = (beg + kRegionSize) - (n + 1) * kMetadataSize;
|
2012-12-06 21:13:58 +08:00
|
|
|
return reinterpret_cast<void*>(meta);
|
2012-12-06 20:49:28 +08:00
|
|
|
}
|
|
|
|
|
2013-01-29 01:29:50 +08:00
|
|
|
NOINLINE Batch* AllocateBatch(AllocatorStats *stat, AllocatorCache *c,
|
2013-01-24 17:08:03 +08:00
|
|
|
uptr class_id) {
|
2013-01-12 00:41:19 +08:00
|
|
|
CHECK_LT(class_id, kNumClasses);
|
2012-12-06 22:27:32 +08:00
|
|
|
SizeClassInfo *sci = GetSizeClassInfo(class_id);
|
|
|
|
SpinMutexLock l(&sci->mutex);
|
2013-01-12 00:41:19 +08:00
|
|
|
if (sci->free_list.empty())
|
2013-01-24 17:08:03 +08:00
|
|
|
PopulateFreeList(stat, c, sci, class_id);
|
2012-12-06 22:27:32 +08:00
|
|
|
CHECK(!sci->free_list.empty());
|
2013-01-12 00:41:19 +08:00
|
|
|
Batch *b = sci->free_list.front();
|
|
|
|
sci->free_list.pop_front();
|
|
|
|
return b;
|
2012-12-06 22:27:32 +08:00
|
|
|
}
|
|
|
|
|
2013-01-29 01:29:50 +08:00
|
|
|
NOINLINE void DeallocateBatch(AllocatorStats *stat, uptr class_id, Batch *b) {
|
2013-01-12 00:41:19 +08:00
|
|
|
CHECK_LT(class_id, kNumClasses);
|
2012-12-06 22:27:32 +08:00
|
|
|
SizeClassInfo *sci = GetSizeClassInfo(class_id);
|
|
|
|
SpinMutexLock l(&sci->mutex);
|
2013-01-12 00:41:19 +08:00
|
|
|
sci->free_list.push_front(b);
|
2012-12-06 22:27:32 +08:00
|
|
|
}
|
|
|
|
|
2012-12-06 20:49:28 +08:00
|
|
|
bool PointerIsMine(void *p) {
|
2012-12-24 21:41:07 +08:00
|
|
|
return GetSizeClass(p) != 0;
|
2012-12-06 20:49:28 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
uptr GetSizeClass(void *p) {
|
2012-12-24 21:41:07 +08:00
|
|
|
return state_->possible_regions[ComputeRegionId(reinterpret_cast<uptr>(p))];
|
2012-12-06 20:49:28 +08:00
|
|
|
}
|
|
|
|
|
2012-12-06 21:13:58 +08:00
|
|
|
void *GetBlockBegin(void *p) {
|
|
|
|
CHECK(PointerIsMine(p));
|
|
|
|
uptr mem = reinterpret_cast<uptr>(p);
|
|
|
|
uptr beg = ComputeRegionBeg(mem);
|
|
|
|
uptr size = SizeClassMap::Size(GetSizeClass(p));
|
|
|
|
u32 offset = mem - beg;
|
|
|
|
u32 n = offset / (u32)size; // 32-bit division
|
|
|
|
uptr res = beg + (n * (u32)size);
|
|
|
|
return reinterpret_cast<void*>(res);
|
|
|
|
}
|
|
|
|
|
2012-12-06 20:49:28 +08:00
|
|
|
uptr GetActuallyAllocatedSize(void *p) {
|
|
|
|
CHECK(PointerIsMine(p));
|
|
|
|
return SizeClassMap::Size(GetSizeClass(p));
|
|
|
|
}
|
|
|
|
|
2012-12-06 22:27:32 +08:00
|
|
|
uptr ClassID(uptr size) { return SizeClassMap::ClassID(size); }
|
|
|
|
|
2012-12-06 20:49:28 +08:00
|
|
|
uptr TotalMemoryUsed() {
|
|
|
|
// No need to lock here.
|
|
|
|
uptr res = 0;
|
|
|
|
for (uptr i = 0; i < kNumPossibleRegions; i++)
|
2012-12-12 22:32:18 +08:00
|
|
|
if (state_->possible_regions[i])
|
2012-12-06 20:49:28 +08:00
|
|
|
res += kRegionSize;
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
void TestOnlyUnmap() {
|
|
|
|
for (uptr i = 0; i < kNumPossibleRegions; i++)
|
2012-12-12 22:32:18 +08:00
|
|
|
if (state_->possible_regions[i])
|
|
|
|
UnmapWithCallback((i * kRegionSize), kRegionSize);
|
|
|
|
UnmapWithCallback(reinterpret_cast<uptr>(state_), sizeof(State));
|
2012-12-06 20:49:28 +08:00
|
|
|
}
|
|
|
|
|
2013-02-07 19:40:03 +08:00
|
|
|
// ForceLock() and ForceUnlock() are needed to implement Darwin malloc zone
|
|
|
|
// introspection API.
|
|
|
|
void ForceLock() {
|
2013-02-07 20:00:40 +08:00
|
|
|
for (uptr i = 0; i < kNumClasses; i++) {
|
2013-02-07 19:40:03 +08:00
|
|
|
GetSizeClassInfo(i)->mutex.Lock();
|
2013-02-07 22:58:04 +08:00
|
|
|
}
|
2013-02-07 19:40:03 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void ForceUnlock() {
|
|
|
|
for (int i = kNumClasses - 1; i >= 0; i--) {
|
|
|
|
GetSizeClassInfo(i)->mutex.Unlock();
|
2013-02-07 22:58:04 +08:00
|
|
|
}
|
2013-02-07 19:40:03 +08:00
|
|
|
}
|
|
|
|
|
2012-12-27 22:09:19 +08:00
|
|
|
void PrintStats() {
|
|
|
|
}
|
|
|
|
|
2012-12-06 20:49:28 +08:00
|
|
|
typedef SizeClassMap SizeClassMapT;
|
2012-12-24 21:41:07 +08:00
|
|
|
static const uptr kNumClasses = SizeClassMap::kNumClasses;
|
2012-12-07 17:40:17 +08:00
|
|
|
|
2012-12-06 20:49:28 +08:00
|
|
|
private:
|
|
|
|
static const uptr kRegionSizeLog = SANITIZER_WORDSIZE == 64 ? 24 : 20;
|
|
|
|
static const uptr kRegionSize = 1 << kRegionSizeLog;
|
|
|
|
static const uptr kNumPossibleRegions = kSpaceSize / kRegionSize;
|
|
|
|
|
|
|
|
struct SizeClassInfo {
|
|
|
|
SpinMutex mutex;
|
2013-01-12 00:41:19 +08:00
|
|
|
IntrusiveList<Batch> free_list;
|
|
|
|
char padding[kCacheLineSize - sizeof(uptr) - sizeof(IntrusiveList<Batch>)];
|
2012-12-06 20:49:28 +08:00
|
|
|
};
|
|
|
|
COMPILER_CHECK(sizeof(SizeClassInfo) == kCacheLineSize);
|
|
|
|
|
|
|
|
uptr ComputeRegionId(uptr mem) {
|
|
|
|
uptr res = mem >> kRegionSizeLog;
|
|
|
|
CHECK_LT(res, kNumPossibleRegions);
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
uptr ComputeRegionBeg(uptr mem) {
|
|
|
|
return mem & ~(kRegionSize - 1);
|
|
|
|
}
|
|
|
|
|
2013-01-24 17:08:03 +08:00
|
|
|
uptr AllocateRegion(AllocatorStats *stat, uptr class_id) {
|
2012-12-06 20:49:28 +08:00
|
|
|
CHECK_LT(class_id, kNumClasses);
|
|
|
|
uptr res = reinterpret_cast<uptr>(MmapAlignedOrDie(kRegionSize, kRegionSize,
|
|
|
|
"SizeClassAllocator32"));
|
2012-12-12 22:32:18 +08:00
|
|
|
MapUnmapCallback().OnMap(res, kRegionSize);
|
2013-01-24 17:08:03 +08:00
|
|
|
stat->Add(AllocatorStatMmapped, kRegionSize);
|
2012-12-06 20:49:28 +08:00
|
|
|
CHECK_EQ(0U, (res & (kRegionSize - 1)));
|
2012-12-12 22:32:18 +08:00
|
|
|
CHECK_EQ(0U, state_->possible_regions[ComputeRegionId(res)]);
|
2012-12-24 21:41:07 +08:00
|
|
|
state_->possible_regions[ComputeRegionId(res)] = class_id;
|
2012-12-06 20:49:28 +08:00
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
SizeClassInfo *GetSizeClassInfo(uptr class_id) {
|
|
|
|
CHECK_LT(class_id, kNumClasses);
|
2012-12-12 22:32:18 +08:00
|
|
|
return &state_->size_class_info_array[class_id];
|
2012-12-06 20:49:28 +08:00
|
|
|
}
|
|
|
|
|
2013-01-24 17:08:03 +08:00
|
|
|
void PopulateFreeList(AllocatorStats *stat, AllocatorCache *c,
|
|
|
|
SizeClassInfo *sci, uptr class_id) {
|
2012-12-06 20:49:28 +08:00
|
|
|
uptr size = SizeClassMap::Size(class_id);
|
2013-01-24 17:08:03 +08:00
|
|
|
uptr reg = AllocateRegion(stat, class_id);
|
2012-12-06 20:49:28 +08:00
|
|
|
uptr n_chunks = kRegionSize / (size + kMetadataSize);
|
2013-01-15 17:02:20 +08:00
|
|
|
uptr max_count = SizeClassMap::MaxCached(class_id);
|
2013-01-12 00:41:19 +08:00
|
|
|
Batch *b = 0;
|
|
|
|
for (uptr i = reg; i < reg + n_chunks * size; i += size) {
|
|
|
|
if (b == 0) {
|
|
|
|
if (class_id < SizeClassMap::kMinBatchClass)
|
|
|
|
b = (Batch*)c->Allocate(this, SizeClassMap::ClassID(sizeof(Batch)));
|
|
|
|
else
|
|
|
|
b = (Batch*)i;
|
|
|
|
b->count = 0;
|
|
|
|
}
|
|
|
|
b->batch[b->count++] = (void*)i;
|
2013-01-15 17:02:20 +08:00
|
|
|
if (b->count == max_count) {
|
2013-01-12 00:41:19 +08:00
|
|
|
sci->free_list.push_back(b);
|
|
|
|
b = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (b)
|
|
|
|
sci->free_list.push_back(b);
|
2012-12-06 20:49:28 +08:00
|
|
|
}
|
|
|
|
|
2012-12-12 22:32:18 +08:00
|
|
|
struct State {
|
|
|
|
u8 possible_regions[kNumPossibleRegions];
|
|
|
|
SizeClassInfo size_class_info_array[kNumClasses];
|
|
|
|
};
|
|
|
|
State *state_;
|
2012-12-06 20:49:28 +08:00
|
|
|
};
|
|
|
|
|
2013-01-10 21:06:35 +08:00
|
|
|
// Objects of this type should be used as local caches for SizeClassAllocator64
|
|
|
|
// or SizeClassAllocator32. Since the typical use of this class is to have one
|
|
|
|
// object per thread in TLS, is has to be POD.
|
2012-12-04 22:15:17 +08:00
|
|
|
template<class SizeClassAllocator>
|
2012-12-04 15:54:41 +08:00
|
|
|
struct SizeClassAllocatorLocalCache {
|
2012-12-04 21:59:22 +08:00
|
|
|
typedef SizeClassAllocator Allocator;
|
2012-12-04 22:15:17 +08:00
|
|
|
static const uptr kNumClasses = SizeClassAllocator::kNumClasses;
|
2013-01-15 17:02:20 +08:00
|
|
|
|
2013-01-24 17:08:03 +08:00
|
|
|
void Init(AllocatorGlobalStats *s) {
|
|
|
|
stats_.Init();
|
|
|
|
if (s)
|
|
|
|
s->Register(&stats_);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Destroy(SizeClassAllocator *allocator, AllocatorGlobalStats *s) {
|
|
|
|
Drain(allocator);
|
|
|
|
if (s)
|
|
|
|
s->Unregister(&stats_);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void *Allocate(SizeClassAllocator *allocator, uptr class_id) {
|
2012-12-24 21:41:07 +08:00
|
|
|
CHECK_NE(class_id, 0UL);
|
2012-12-04 15:54:41 +08:00
|
|
|
CHECK_LT(class_id, kNumClasses);
|
2013-01-24 17:08:03 +08:00
|
|
|
stats_.Add(AllocatorStatMalloced, SizeClassMap::Size(class_id));
|
2013-01-12 00:41:19 +08:00
|
|
|
PerClass *c = &per_class_[class_id];
|
2013-01-15 17:02:20 +08:00
|
|
|
if (UNLIKELY(c->count == 0))
|
|
|
|
Refill(allocator, class_id);
|
|
|
|
void *res = c->batch[--c->count];
|
|
|
|
PREFETCH(c->batch[c->count - 1]);
|
2012-12-04 15:54:41 +08:00
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Deallocate(SizeClassAllocator *allocator, uptr class_id, void *p) {
|
2012-12-24 21:41:07 +08:00
|
|
|
CHECK_NE(class_id, 0UL);
|
2012-12-04 15:54:41 +08:00
|
|
|
CHECK_LT(class_id, kNumClasses);
|
2013-01-24 17:08:03 +08:00
|
|
|
stats_.Add(AllocatorStatFreed, SizeClassMap::Size(class_id));
|
2013-01-12 00:41:19 +08:00
|
|
|
PerClass *c = &per_class_[class_id];
|
2013-01-15 17:02:20 +08:00
|
|
|
if (UNLIKELY(c->count == c->max_count))
|
|
|
|
Drain(allocator, class_id);
|
|
|
|
c->batch[c->count++] = p;
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void Drain(SizeClassAllocator *allocator) {
|
2013-01-15 17:02:20 +08:00
|
|
|
for (uptr class_id = 0; class_id < kNumClasses; class_id++) {
|
|
|
|
PerClass *c = &per_class_[class_id];
|
|
|
|
while (c->count > 0)
|
|
|
|
Drain(allocator, class_id);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// private:
|
|
|
|
typedef typename SizeClassAllocator::SizeClassMapT SizeClassMap;
|
2013-01-12 00:41:19 +08:00
|
|
|
typedef typename SizeClassMap::TransferBatch Batch;
|
|
|
|
struct PerClass {
|
2013-01-15 17:02:20 +08:00
|
|
|
uptr count;
|
|
|
|
uptr max_count;
|
|
|
|
void *batch[2 * SizeClassMap::kMaxNumCached];
|
2013-01-12 00:41:19 +08:00
|
|
|
};
|
|
|
|
PerClass per_class_[kNumClasses];
|
2013-01-24 17:08:03 +08:00
|
|
|
AllocatorStats stats_;
|
2013-01-15 17:02:20 +08:00
|
|
|
|
|
|
|
void InitCache() {
|
|
|
|
if (per_class_[0].max_count)
|
|
|
|
return;
|
|
|
|
for (uptr i = 0; i < kNumClasses; i++) {
|
|
|
|
PerClass *c = &per_class_[i];
|
|
|
|
c->max_count = 2 * SizeClassMap::MaxCached(i);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-01-29 01:29:50 +08:00
|
|
|
NOINLINE void Refill(SizeClassAllocator *allocator, uptr class_id) {
|
2013-01-15 17:02:20 +08:00
|
|
|
InitCache();
|
|
|
|
PerClass *c = &per_class_[class_id];
|
2013-01-24 17:08:03 +08:00
|
|
|
Batch *b = allocator->AllocateBatch(&stats_, this, class_id);
|
|
|
|
CHECK_GT(b->count, 0);
|
2013-01-15 17:02:20 +08:00
|
|
|
for (uptr i = 0; i < b->count; i++)
|
|
|
|
c->batch[i] = b->batch[i];
|
|
|
|
c->count = b->count;
|
|
|
|
if (class_id < SizeClassMap::kMinBatchClass)
|
|
|
|
Deallocate(allocator, SizeClassMap::ClassID(sizeof(Batch)), b);
|
|
|
|
}
|
|
|
|
|
2013-01-29 01:29:50 +08:00
|
|
|
NOINLINE void Drain(SizeClassAllocator *allocator, uptr class_id) {
|
2013-01-15 17:02:20 +08:00
|
|
|
InitCache();
|
|
|
|
PerClass *c = &per_class_[class_id];
|
|
|
|
Batch *b;
|
|
|
|
if (class_id < SizeClassMap::kMinBatchClass)
|
|
|
|
b = (Batch*)Allocate(allocator, SizeClassMap::ClassID(sizeof(Batch)));
|
|
|
|
else
|
|
|
|
b = (Batch*)c->batch[0];
|
|
|
|
uptr cnt = Min(c->max_count / 2, c->count);
|
|
|
|
for (uptr i = 0; i < cnt; i++) {
|
|
|
|
b->batch[i] = c->batch[i];
|
|
|
|
c->batch[i] = c->batch[i + c->max_count / 2];
|
|
|
|
}
|
|
|
|
b->count = cnt;
|
|
|
|
c->count -= cnt;
|
2013-01-24 17:08:03 +08:00
|
|
|
allocator->DeallocateBatch(&stats_, class_id, b);
|
2013-01-15 17:02:20 +08:00
|
|
|
}
|
2012-12-04 15:54:41 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
// This class can (de)allocate only large chunks of memory using mmap/unmap.
|
|
|
|
// The main purpose of this allocator is to cover large and rare allocation
|
|
|
|
// sizes not covered by more efficient allocators (e.g. SizeClassAllocator64).
|
2012-12-12 22:32:18 +08:00
|
|
|
template <class MapUnmapCallback = NoOpMapUnmapCallback>
|
2012-12-04 15:54:41 +08:00
|
|
|
class LargeMmapAllocator {
|
|
|
|
public:
|
|
|
|
void Init() {
|
|
|
|
internal_memset(this, 0, sizeof(*this));
|
|
|
|
page_size_ = GetPageSizeCached();
|
|
|
|
}
|
2013-01-11 19:15:48 +08:00
|
|
|
|
2013-01-24 17:08:03 +08:00
|
|
|
void *Allocate(AllocatorStats *stat, uptr size, uptr alignment) {
|
2012-12-04 15:54:41 +08:00
|
|
|
CHECK(IsPowerOfTwo(alignment));
|
|
|
|
uptr map_size = RoundUpMapSize(size);
|
|
|
|
if (alignment > page_size_)
|
|
|
|
map_size += alignment;
|
|
|
|
if (map_size < size) return 0; // Overflow.
|
|
|
|
uptr map_beg = reinterpret_cast<uptr>(
|
|
|
|
MmapOrDie(map_size, "LargeMmapAllocator"));
|
2012-12-12 22:32:18 +08:00
|
|
|
MapUnmapCallback().OnMap(map_beg, map_size);
|
2012-12-04 15:54:41 +08:00
|
|
|
uptr map_end = map_beg + map_size;
|
|
|
|
uptr res = map_beg + page_size_;
|
|
|
|
if (res & (alignment - 1)) // Align.
|
|
|
|
res += alignment - (res & (alignment - 1));
|
|
|
|
CHECK_EQ(0, res & (alignment - 1));
|
|
|
|
CHECK_LE(res + size, map_end);
|
|
|
|
Header *h = GetHeader(res);
|
|
|
|
h->size = size;
|
|
|
|
h->map_beg = map_beg;
|
|
|
|
h->map_size = map_size;
|
2013-02-08 20:02:00 +08:00
|
|
|
uptr size_log = MostSignificantSetBitIndex(map_size);
|
2013-01-10 21:38:38 +08:00
|
|
|
CHECK_LT(size_log, ARRAY_SIZE(stats.by_size_log));
|
2012-12-04 15:54:41 +08:00
|
|
|
{
|
|
|
|
SpinMutexLock l(&mutex_);
|
2012-12-24 22:35:14 +08:00
|
|
|
uptr idx = n_chunks_++;
|
2012-12-25 15:50:35 +08:00
|
|
|
CHECK_LT(idx, kMaxNumChunks);
|
2012-12-24 22:35:14 +08:00
|
|
|
h->chunk_idx = idx;
|
|
|
|
chunks_[idx] = h;
|
2012-12-27 22:09:19 +08:00
|
|
|
stats.n_allocs++;
|
|
|
|
stats.currently_allocated += map_size;
|
|
|
|
stats.max_allocated = Max(stats.max_allocated, stats.currently_allocated);
|
2013-01-10 21:38:38 +08:00
|
|
|
stats.by_size_log[size_log]++;
|
2013-01-24 17:08:03 +08:00
|
|
|
stat->Add(AllocatorStatMalloced, map_size);
|
|
|
|
stat->Add(AllocatorStatMmapped, map_size);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
return reinterpret_cast<void*>(res);
|
|
|
|
}
|
|
|
|
|
2013-01-24 17:08:03 +08:00
|
|
|
void Deallocate(AllocatorStats *stat, void *p) {
|
2012-12-04 15:54:41 +08:00
|
|
|
Header *h = GetHeader(p);
|
|
|
|
{
|
|
|
|
SpinMutexLock l(&mutex_);
|
2012-12-24 22:35:14 +08:00
|
|
|
uptr idx = h->chunk_idx;
|
|
|
|
CHECK_EQ(chunks_[idx], h);
|
|
|
|
CHECK_LT(idx, n_chunks_);
|
|
|
|
chunks_[idx] = chunks_[n_chunks_ - 1];
|
|
|
|
chunks_[idx]->chunk_idx = idx;
|
|
|
|
n_chunks_--;
|
2012-12-27 22:09:19 +08:00
|
|
|
stats.n_frees++;
|
|
|
|
stats.currently_allocated -= h->map_size;
|
2013-01-24 17:08:03 +08:00
|
|
|
stat->Add(AllocatorStatFreed, h->map_size);
|
|
|
|
stat->Add(AllocatorStatUnmapped, h->map_size);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
2012-12-12 22:32:18 +08:00
|
|
|
MapUnmapCallback().OnUnmap(h->map_beg, h->map_size);
|
2012-12-04 15:54:41 +08:00
|
|
|
UnmapOrDie(reinterpret_cast<void*>(h->map_beg), h->map_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
uptr TotalMemoryUsed() {
|
|
|
|
SpinMutexLock l(&mutex_);
|
|
|
|
uptr res = 0;
|
2012-12-24 22:35:14 +08:00
|
|
|
for (uptr i = 0; i < n_chunks_; i++) {
|
|
|
|
Header *h = chunks_[i];
|
|
|
|
CHECK_EQ(h->chunk_idx, i);
|
|
|
|
res += RoundUpMapSize(h->size);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool PointerIsMine(void *p) {
|
2012-12-18 22:56:38 +08:00
|
|
|
return GetBlockBegin(p) != 0;
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
uptr GetActuallyAllocatedSize(void *p) {
|
2012-12-18 22:56:38 +08:00
|
|
|
return RoundUpTo(GetHeader(p)->size, page_size_);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// At least page_size_/2 metadata bytes is available.
|
|
|
|
void *GetMetaData(void *p) {
|
2012-12-25 15:50:35 +08:00
|
|
|
// Too slow: CHECK_EQ(p, GetBlockBegin(p));
|
|
|
|
CHECK(IsAligned(reinterpret_cast<uptr>(p), page_size_));
|
2012-12-04 15:54:41 +08:00
|
|
|
return GetHeader(p) + 1;
|
|
|
|
}
|
|
|
|
|
2012-12-18 22:56:38 +08:00
|
|
|
void *GetBlockBegin(void *ptr) {
|
|
|
|
uptr p = reinterpret_cast<uptr>(ptr);
|
2012-12-04 15:54:41 +08:00
|
|
|
SpinMutexLock l(&mutex_);
|
2012-12-24 22:35:14 +08:00
|
|
|
uptr nearest_chunk = 0;
|
|
|
|
// Cache-friendly linear search.
|
|
|
|
for (uptr i = 0; i < n_chunks_; i++) {
|
|
|
|
uptr ch = reinterpret_cast<uptr>(chunks_[i]);
|
|
|
|
if (p < ch) continue; // p is at left to this chunk, skip it.
|
|
|
|
if (p - ch < p - nearest_chunk)
|
|
|
|
nearest_chunk = ch;
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
2012-12-24 22:35:14 +08:00
|
|
|
if (!nearest_chunk)
|
|
|
|
return 0;
|
|
|
|
Header *h = reinterpret_cast<Header *>(nearest_chunk);
|
|
|
|
CHECK_GE(nearest_chunk, h->map_beg);
|
|
|
|
CHECK_LT(nearest_chunk, h->map_beg + h->map_size);
|
|
|
|
CHECK_LE(nearest_chunk, p);
|
|
|
|
if (h->map_beg + h->map_size < p)
|
|
|
|
return 0;
|
|
|
|
return GetUser(h);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
|
2012-12-27 22:09:19 +08:00
|
|
|
void PrintStats() {
|
|
|
|
Printf("Stats: LargeMmapAllocator: allocated %zd times, "
|
2013-01-10 21:38:38 +08:00
|
|
|
"remains %zd (%zd K) max %zd M; by size logs: ",
|
2012-12-27 22:09:19 +08:00
|
|
|
stats.n_allocs, stats.n_allocs - stats.n_frees,
|
|
|
|
stats.currently_allocated >> 10, stats.max_allocated >> 20);
|
2013-01-10 21:38:38 +08:00
|
|
|
for (uptr i = 0; i < ARRAY_SIZE(stats.by_size_log); i++) {
|
|
|
|
uptr c = stats.by_size_log[i];
|
|
|
|
if (!c) continue;
|
|
|
|
Printf("%zd:%zd; ", i, c);
|
|
|
|
}
|
|
|
|
Printf("\n");
|
2012-12-27 22:09:19 +08:00
|
|
|
}
|
|
|
|
|
2013-02-07 19:40:03 +08:00
|
|
|
// ForceLock() and ForceUnlock() are needed to implement Darwin malloc zone
|
|
|
|
// introspection API.
|
|
|
|
void ForceLock() {
|
|
|
|
mutex_.Lock();
|
|
|
|
}
|
|
|
|
|
|
|
|
void ForceUnlock() {
|
|
|
|
mutex_.Unlock();
|
|
|
|
}
|
|
|
|
|
2012-12-04 15:54:41 +08:00
|
|
|
private:
|
2012-12-24 22:35:14 +08:00
|
|
|
static const int kMaxNumChunks = 1 << FIRST_32_SECOND_64(15, 18);
|
2012-12-04 15:54:41 +08:00
|
|
|
struct Header {
|
|
|
|
uptr map_beg;
|
|
|
|
uptr map_size;
|
|
|
|
uptr size;
|
2012-12-24 22:35:14 +08:00
|
|
|
uptr chunk_idx;
|
2012-12-04 15:54:41 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
Header *GetHeader(uptr p) {
|
|
|
|
CHECK_EQ(p % page_size_, 0);
|
|
|
|
return reinterpret_cast<Header*>(p - page_size_);
|
|
|
|
}
|
|
|
|
Header *GetHeader(void *p) { return GetHeader(reinterpret_cast<uptr>(p)); }
|
|
|
|
|
|
|
|
void *GetUser(Header *h) {
|
|
|
|
CHECK_EQ((uptr)h % page_size_, 0);
|
|
|
|
return reinterpret_cast<void*>(reinterpret_cast<uptr>(h) + page_size_);
|
|
|
|
}
|
|
|
|
|
|
|
|
uptr RoundUpMapSize(uptr size) {
|
|
|
|
return RoundUpTo(size, page_size_) + page_size_;
|
|
|
|
}
|
|
|
|
|
|
|
|
uptr page_size_;
|
2012-12-24 22:35:14 +08:00
|
|
|
Header *chunks_[kMaxNumChunks];
|
|
|
|
uptr n_chunks_;
|
2012-12-27 22:09:19 +08:00
|
|
|
struct Stats {
|
2013-01-10 21:38:38 +08:00
|
|
|
uptr n_allocs, n_frees, currently_allocated, max_allocated, by_size_log[64];
|
2012-12-27 22:09:19 +08:00
|
|
|
} stats;
|
2012-12-04 15:54:41 +08:00
|
|
|
SpinMutex mutex_;
|
|
|
|
};
|
|
|
|
|
|
|
|
// This class implements a complete memory allocator by using two
|
|
|
|
// internal allocators:
|
|
|
|
// PrimaryAllocator is efficient, but may not allocate some sizes (alignments).
|
|
|
|
// When allocating 2^x bytes it should return 2^x aligned chunk.
|
|
|
|
// PrimaryAllocator is used via a local AllocatorCache.
|
|
|
|
// SecondaryAllocator can allocate anything, but is not efficient.
|
|
|
|
template <class PrimaryAllocator, class AllocatorCache,
|
|
|
|
class SecondaryAllocator> // NOLINT
|
|
|
|
class CombinedAllocator {
|
|
|
|
public:
|
|
|
|
void Init() {
|
|
|
|
primary_.Init();
|
|
|
|
secondary_.Init();
|
2013-01-24 17:08:03 +08:00
|
|
|
stats_.Init();
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void *Allocate(AllocatorCache *cache, uptr size, uptr alignment,
|
|
|
|
bool cleared = false) {
|
|
|
|
// Returning 0 on malloc(0) may break a lot of code.
|
|
|
|
if (size == 0)
|
|
|
|
size = 1;
|
|
|
|
if (size + alignment < size)
|
|
|
|
return 0;
|
|
|
|
if (alignment > 8)
|
|
|
|
size = RoundUpTo(size, alignment);
|
|
|
|
void *res;
|
2013-01-11 19:15:48 +08:00
|
|
|
if (primary_.CanAllocate(size, alignment))
|
|
|
|
res = cache->Allocate(&primary_, primary_.ClassID(size));
|
|
|
|
else
|
2013-01-24 17:08:03 +08:00
|
|
|
res = secondary_.Allocate(&stats_, size, alignment);
|
2012-12-04 15:54:41 +08:00
|
|
|
if (alignment > 8)
|
|
|
|
CHECK_EQ(reinterpret_cast<uptr>(res) & (alignment - 1), 0);
|
|
|
|
if (cleared && res)
|
|
|
|
internal_memset(res, 0, size);
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Deallocate(AllocatorCache *cache, void *p) {
|
|
|
|
if (!p) return;
|
|
|
|
if (primary_.PointerIsMine(p))
|
|
|
|
cache->Deallocate(&primary_, primary_.GetSizeClass(p), p);
|
|
|
|
else
|
2013-01-24 17:08:03 +08:00
|
|
|
secondary_.Deallocate(&stats_, p);
|
2012-12-04 15:54:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void *Reallocate(AllocatorCache *cache, void *p, uptr new_size,
|
|
|
|
uptr alignment) {
|
|
|
|
if (!p)
|
|
|
|
return Allocate(cache, new_size, alignment);
|
|
|
|
if (!new_size) {
|
|
|
|
Deallocate(cache, p);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
CHECK(PointerIsMine(p));
|
|
|
|
uptr old_size = GetActuallyAllocatedSize(p);
|
|
|
|
uptr memcpy_size = Min(new_size, old_size);
|
|
|
|
void *new_p = Allocate(cache, new_size, alignment);
|
|
|
|
if (new_p)
|
|
|
|
internal_memcpy(new_p, p, memcpy_size);
|
|
|
|
Deallocate(cache, p);
|
|
|
|
return new_p;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool PointerIsMine(void *p) {
|
|
|
|
if (primary_.PointerIsMine(p))
|
|
|
|
return true;
|
|
|
|
return secondary_.PointerIsMine(p);
|
|
|
|
}
|
|
|
|
|
2012-12-26 12:52:07 +08:00
|
|
|
bool FromPrimary(void *p) {
|
|
|
|
return primary_.PointerIsMine(p);
|
|
|
|
}
|
|
|
|
|
2012-12-04 15:54:41 +08:00
|
|
|
void *GetMetaData(void *p) {
|
|
|
|
if (primary_.PointerIsMine(p))
|
|
|
|
return primary_.GetMetaData(p);
|
|
|
|
return secondary_.GetMetaData(p);
|
|
|
|
}
|
|
|
|
|
|
|
|
void *GetBlockBegin(void *p) {
|
|
|
|
if (primary_.PointerIsMine(p))
|
|
|
|
return primary_.GetBlockBegin(p);
|
|
|
|
return secondary_.GetBlockBegin(p);
|
|
|
|
}
|
|
|
|
|
|
|
|
uptr GetActuallyAllocatedSize(void *p) {
|
|
|
|
if (primary_.PointerIsMine(p))
|
|
|
|
return primary_.GetActuallyAllocatedSize(p);
|
|
|
|
return secondary_.GetActuallyAllocatedSize(p);
|
|
|
|
}
|
|
|
|
|
|
|
|
uptr TotalMemoryUsed() {
|
|
|
|
return primary_.TotalMemoryUsed() + secondary_.TotalMemoryUsed();
|
|
|
|
}
|
|
|
|
|
|
|
|
void TestOnlyUnmap() { primary_.TestOnlyUnmap(); }
|
|
|
|
|
2013-01-24 17:08:03 +08:00
|
|
|
void InitCache(AllocatorCache *cache) {
|
|
|
|
cache->Init(&stats_);
|
|
|
|
}
|
|
|
|
|
|
|
|
void DestroyCache(AllocatorCache *cache) {
|
|
|
|
cache->Destroy(&primary_, &stats_);
|
|
|
|
}
|
|
|
|
|
2012-12-04 15:54:41 +08:00
|
|
|
void SwallowCache(AllocatorCache *cache) {
|
|
|
|
cache->Drain(&primary_);
|
|
|
|
}
|
|
|
|
|
2013-01-24 17:08:03 +08:00
|
|
|
void GetStats(AllocatorStatCounters s) const {
|
|
|
|
stats_.Get(s);
|
|
|
|
}
|
|
|
|
|
2012-12-27 22:09:19 +08:00
|
|
|
void PrintStats() {
|
|
|
|
primary_.PrintStats();
|
|
|
|
secondary_.PrintStats();
|
|
|
|
}
|
|
|
|
|
2013-02-07 19:40:03 +08:00
|
|
|
// ForceLock() and ForceUnlock() are needed to implement Darwin malloc zone
|
|
|
|
// introspection API.
|
|
|
|
void ForceLock() {
|
|
|
|
primary_.ForceLock();
|
|
|
|
secondary_.ForceLock();
|
|
|
|
}
|
|
|
|
|
|
|
|
void ForceUnlock() {
|
|
|
|
secondary_.ForceUnlock();
|
|
|
|
primary_.ForceUnlock();
|
|
|
|
}
|
|
|
|
|
2012-12-04 15:54:41 +08:00
|
|
|
private:
|
|
|
|
PrimaryAllocator primary_;
|
|
|
|
SecondaryAllocator secondary_;
|
2013-01-24 17:08:03 +08:00
|
|
|
AllocatorGlobalStats stats_;
|
2012-12-04 15:54:41 +08:00
|
|
|
};
|
|
|
|
|
2013-01-25 19:46:22 +08:00
|
|
|
// Returns true if calloc(size, n) should return 0 due to overflow in size*n.
|
|
|
|
bool CallocShouldReturnNullDueToOverflow(uptr size, uptr n);
|
|
|
|
|
2012-12-04 15:54:41 +08:00
|
|
|
} // namespace __sanitizer
|
|
|
|
|
|
|
|
#endif // SANITIZER_ALLOCATOR_H
|
|
|
|
|