1 /*
2 * Copyright (c) 2018, 2024, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 *
23 */
24
25 #ifndef SHARE_GC_SHARED_MEMALLOCATOR_HPP
26 #define SHARE_GC_SHARED_MEMALLOCATOR_HPP
27
28 #include "memory/memRegion.hpp"
29 #include "oops/oopsHierarchy.hpp"
30 #include "runtime/javaThread.hpp"
31 #include "utilities/exceptions.hpp"
32 #include "utilities/globalDefinitions.hpp"
33 #include "utilities/macros.hpp"
34
35 // These fascilities are used for allocating, and initializing newly allocated objects.
36
37 class MemAllocator: StackObj {
38 protected:
39 class Allocation;
40
41 Thread* const _thread;
42 Klass* const _klass;
43 const size_t _word_size;
44
45 // Allocate from the current thread's TLAB, without taking a new TLAB (no safepoint).
46 HeapWord* mem_allocate_inside_tlab_fast() const;
47
48 private:
49 // Allocate in a TLAB. Could allocate a new TLAB, and therefore potentially safepoint.
50 HeapWord* mem_allocate_inside_tlab_slow(Allocation& allocation) const;
51
52 // Allocate outside a TLAB. Could safepoint.
53 HeapWord* mem_allocate_outside_tlab(Allocation& allocation) const;
54
55 protected:
56 MemAllocator(Klass* klass, size_t word_size, Thread* thread)
57 : _thread(thread),
58 _klass(klass),
59 _word_size(word_size)
60 {
61 assert(_thread == Thread::current(), "must be");
62 }
63
64 // Initialization provided by subclasses.
65 virtual oop initialize(HeapWord* mem) const = 0;
66
67 // This function clears the memory of the object.
68 void mem_clear(HeapWord* mem) const;
69
70 // This finish constructing an oop by installing the mark word and the Klass* pointer
71 // last. At the point when the Klass pointer is initialized, this is a constructed object
72 // that must be parseable as an oop by concurrent collectors.
73 oop finish(HeapWord* mem) const;
74
75 // Raw memory allocation. This will try to do a TLAB allocation, and otherwise fall
76 // back to calling CollectedHeap::mem_allocate().
77 HeapWord* mem_allocate(Allocation& allocation) const;
78
79 public:
80 // Allocate and fully construct the object, and perform various instrumentation. Could safepoint.
81 oop allocate() const;
82 };
83
84 class ObjAllocator: public MemAllocator {
85 public:
86 ObjAllocator(Klass* klass, size_t word_size, Thread* thread = Thread::current())
87 : MemAllocator(klass, word_size, thread) {}
88
89 virtual oop initialize(HeapWord* mem) const;
90 };
91
92 class ObjArrayAllocator: public MemAllocator {
93 protected:
94 const int _length;
95 const bool _do_zero;
96
97 void mem_zap_start_padding(HeapWord* mem) const PRODUCT_RETURN;
98 void mem_zap_end_padding(HeapWord* mem) const PRODUCT_RETURN;
99
100 public:
101 ObjArrayAllocator(Klass* klass, size_t word_size, int length, bool do_zero,
102 Thread* thread = Thread::current())
103 : MemAllocator(klass, word_size, thread),
104 _length(length),
105 _do_zero(do_zero) {}
106
107 virtual oop initialize(HeapWord* mem) const;
108 };
109
110 class ClassAllocator: public MemAllocator {
111 size_t _base_size;
112 public:
113 ClassAllocator(Klass* klass, size_t word_size, size_t base_size, Thread* thread = Thread::current())
114 : MemAllocator(klass, word_size, thread),
115 _base_size(base_size) {}
116
117 virtual oop initialize(HeapWord* mem) const;
118 };
119
120 // Manages a scope where a failed heap allocation results in
121 // suppression of JVMTI "resource exhausted" events and
122 // throwing a shared, backtrace-less OOME instance.
123 // Used for OOMEs that will not be propagated to user code.
124 class InternalOOMEMark: public StackObj {
125 private:
126 bool _outer;
127 JavaThread* _thread;
128
129 public:
130 explicit InternalOOMEMark(JavaThread* thread) {
131 assert(thread != nullptr, "nullptr is not supported");
132 _outer = thread->is_in_internal_oome_mark();
133 thread->set_is_in_internal_oome_mark(true);
134 _thread = thread;
135 }
136
137 ~InternalOOMEMark() {
138 // Check that only InternalOOMEMark sets
139 // JavaThread::_is_in_internal_oome_mark
140 assert(_thread->is_in_internal_oome_mark(), "must be");
141 _thread->set_is_in_internal_oome_mark(_outer);
142 }
143
144 JavaThread* thread() const { return _thread; }
145 };
146
147 #endif // SHARE_GC_SHARED_MEMALLOCATOR_HPP