forked from OSchip/llvm-project
76 lines
2.1 KiB
C++
76 lines
2.1 KiB
C++
//===-- scudo_allocator_combined.h ------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// Scudo Combined Allocator, dispatches allocation & deallocation requests to
|
|
/// the Primary or the Secondary backend allocators.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef SCUDO_ALLOCATOR_COMBINED_H_
|
|
#define SCUDO_ALLOCATOR_COMBINED_H_
|
|
|
|
#ifndef SCUDO_ALLOCATOR_H_
|
|
# error "This file must be included inside scudo_allocator.h."
|
|
#endif
|
|
|
|
class CombinedAllocator {
|
|
public:
|
|
using PrimaryAllocator = PrimaryT;
|
|
using SecondaryAllocator = SecondaryT;
|
|
using AllocatorCache = typename PrimaryAllocator::AllocatorCache;
|
|
void init(s32 ReleaseToOSIntervalMs) {
|
|
Primary.Init(ReleaseToOSIntervalMs);
|
|
Secondary.Init();
|
|
Stats.Init();
|
|
}
|
|
|
|
// Primary allocations are always MinAlignment aligned, and as such do not
|
|
// require an Alignment parameter.
|
|
void *allocatePrimary(AllocatorCache *Cache, uptr ClassId) {
|
|
return Cache->Allocate(&Primary, ClassId);
|
|
}
|
|
|
|
// Secondary allocations do not require a Cache, but do require an Alignment
|
|
// parameter.
|
|
void *allocateSecondary(uptr Size, uptr Alignment) {
|
|
return Secondary.Allocate(&Stats, Size, Alignment);
|
|
}
|
|
|
|
void deallocatePrimary(AllocatorCache *Cache, void *Ptr, uptr ClassId) {
|
|
Cache->Deallocate(&Primary, ClassId, Ptr);
|
|
}
|
|
|
|
void deallocateSecondary(void *Ptr) {
|
|
Secondary.Deallocate(&Stats, Ptr);
|
|
}
|
|
|
|
void initCache(AllocatorCache *Cache) {
|
|
Cache->Init(&Stats);
|
|
}
|
|
|
|
void destroyCache(AllocatorCache *Cache) {
|
|
Cache->Destroy(&Primary, &Stats);
|
|
}
|
|
|
|
void getStats(AllocatorStatCounters StatType) const {
|
|
Stats.Get(StatType);
|
|
}
|
|
|
|
void printStats() {
|
|
Primary.PrintStats();
|
|
Secondary.PrintStats();
|
|
}
|
|
|
|
private:
|
|
PrimaryAllocator Primary;
|
|
SecondaryAllocator Secondary;
|
|
AllocatorGlobalStats Stats;
|
|
};
|
|
|
|
#endif // SCUDO_ALLOCATOR_COMBINED_H_
|