1/*
2 * Copyright (C) 2017 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#pragma once
27
28#include "IsoConfig.h"
29#include "Mutex.h"
30
31namespace bmalloc {
32
33template<typename Config> class IsoHeapImpl;
34
35namespace api {
36
37// You have to declare IsoHeaps this way:
38//
39// static IsoHeap<type> myTypeHeap;
40//
41// It's not valid to create an IsoHeap except in static storage.
42
43template<typename Type>
44struct IsoHeap {
45 typedef IsoConfig<sizeof(Type)> Config;
46
47 void* allocate();
48 void* tryAllocate();
49 void deallocate(void* p);
50
51 void scavenge();
52
53 bool isInitialized();
54
55 unsigned allocatorOffset() { return m_allocatorOffsetPlusOne - 1; }
56 void setAllocatorOffset(unsigned value) { m_allocatorOffsetPlusOne = value + 1; }
57
58 unsigned deallocatorOffset() { return m_deallocatorOffsetPlusOne - 1; }
59 void setDeallocatorOffset(unsigned value) { m_deallocatorOffsetPlusOne = value + 1; }
60
61 IsoHeapImpl<Config>& impl();
62
63 Mutex m_initializationLock;
64 unsigned m_allocatorOffsetPlusOne;
65 unsigned m_deallocatorOffsetPlusOne;
66 IsoHeapImpl<Config>* m_impl;
67};
68
69// Use this together with MAKE_BISO_MALLOCED_IMPL.
70#define MAKE_BISO_MALLOCED(isoType, exportMacro) \
71public: \
72 static exportMacro ::bmalloc::api::IsoHeap<isoType>& bisoHeap(); \
73 \
74 void* operator new(size_t, void* p) { return p; } \
75 void* operator new[](size_t, void* p) { return p; } \
76 \
77 exportMacro void* operator new(size_t size);\
78 exportMacro void operator delete(void* p);\
79 \
80 void* operator new[](size_t size) = delete; \
81 void operator delete[](void* p) = delete; \
82private: \
83typedef int __makeBisoMallocedMacroSemicolonifier
84
85} } // namespace bmalloc::api
86