1 /*
2  * Copyright (C) 2011 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef ART_COMPILER_UTILS_JNI_MACRO_ASSEMBLER_H_
18 #define ART_COMPILER_UTILS_JNI_MACRO_ASSEMBLER_H_
19 
20 #include <vector>
21 
22 #include <android-base/logging.h>
23 
24 #include "arch/instruction_set.h"
25 #include "base/arena_allocator.h"
26 #include "base/arena_object.h"
27 #include "base/array_ref.h"
28 #include "base/enums.h"
29 #include "base/macros.h"
30 #include "managed_register.h"
31 #include "offsets.h"
32 
33 namespace art {
34 
35 class ArenaAllocator;
36 class DebugFrameOpCodeWriterForAssembler;
37 class InstructionSetFeatures;
38 class MemoryRegion;
39 class JNIMacroLabel;
40 
41 enum class JNIMacroUnaryCondition {
42   kZero,
43   kNotZero
44 };
45 
46 template <PointerSize kPointerSize>
47 class JNIMacroAssembler : public DeletableArenaObject<kArenaAllocAssembler> {
48  public:
49   static std::unique_ptr<JNIMacroAssembler<kPointerSize>> Create(
50       ArenaAllocator* allocator,
51       InstructionSet instruction_set,
52       const InstructionSetFeatures* instruction_set_features = nullptr);
53 
54   // Finalize the code; emit slow paths, fixup branches, add literal pool, etc.
55   virtual void FinalizeCode() = 0;
56 
57   // Size of generated code
58   virtual size_t CodeSize() const = 0;
59 
60   // Copy instructions out of assembly buffer into the given region of memory
61   virtual void FinalizeInstructions(const MemoryRegion& region) = 0;
62 
63   // Emit code that will create an activation on the stack
64   virtual void BuildFrame(size_t frame_size,
65                           ManagedRegister method_reg,
66                           ArrayRef<const ManagedRegister> callee_save_regs,
67                           const ManagedRegisterEntrySpills& entry_spills) = 0;
68 
69   // Emit code that will remove an activation from the stack
70   //
71   // Argument `may_suspend` must be `true` if the compiled method may be
72   // suspended during its execution (otherwise `false`, if it is impossible
73   // to suspend during its execution).
74   virtual void RemoveFrame(size_t frame_size,
75                            ArrayRef<const ManagedRegister> callee_save_regs,
76                            bool may_suspend) = 0;
77 
78   virtual void IncreaseFrameSize(size_t adjust) = 0;
79   virtual void DecreaseFrameSize(size_t adjust) = 0;
80 
81   // Store routines
82   virtual void Store(FrameOffset offs, ManagedRegister src, size_t size) = 0;
83   virtual void StoreRef(FrameOffset dest, ManagedRegister src) = 0;
84   virtual void StoreRawPtr(FrameOffset dest, ManagedRegister src) = 0;
85 
86   virtual void StoreImmediateToFrame(FrameOffset dest, uint32_t imm, ManagedRegister scratch) = 0;
87 
88   virtual void StoreStackOffsetToThread(ThreadOffset<kPointerSize> thr_offs,
89                                         FrameOffset fr_offs,
90                                         ManagedRegister scratch) = 0;
91 
92   virtual void StoreStackPointerToThread(ThreadOffset<kPointerSize> thr_offs) = 0;
93 
94   virtual void StoreSpanning(FrameOffset dest,
95                              ManagedRegister src,
96                              FrameOffset in_off,
97                              ManagedRegister scratch) = 0;
98 
99   // Load routines
100   virtual void Load(ManagedRegister dest, FrameOffset src, size_t size) = 0;
101 
102   virtual void LoadFromThread(ManagedRegister dest,
103                               ThreadOffset<kPointerSize> src,
104                               size_t size) = 0;
105 
106   virtual void LoadRef(ManagedRegister dest, FrameOffset src) = 0;
107   // If unpoison_reference is true and kPoisonReference is true, then we negate the read reference.
108   virtual void LoadRef(ManagedRegister dest,
109                        ManagedRegister base,
110                        MemberOffset offs,
111                        bool unpoison_reference) = 0;
112 
113   virtual void LoadRawPtr(ManagedRegister dest, ManagedRegister base, Offset offs) = 0;
114 
115   virtual void LoadRawPtrFromThread(ManagedRegister dest, ThreadOffset<kPointerSize> offs) = 0;
116 
117   // Copying routines
118   virtual void Move(ManagedRegister dest, ManagedRegister src, size_t size) = 0;
119 
120   virtual void CopyRawPtrFromThread(FrameOffset fr_offs,
121                                     ThreadOffset<kPointerSize> thr_offs,
122                                     ManagedRegister scratch) = 0;
123 
124   virtual void CopyRawPtrToThread(ThreadOffset<kPointerSize> thr_offs,
125                                   FrameOffset fr_offs,
126                                   ManagedRegister scratch) = 0;
127 
128   virtual void CopyRef(FrameOffset dest, FrameOffset src, ManagedRegister scratch) = 0;
129 
130   virtual void Copy(FrameOffset dest, FrameOffset src, ManagedRegister scratch, size_t size) = 0;
131 
132   virtual void Copy(FrameOffset dest,
133                     ManagedRegister src_base,
134                     Offset src_offset,
135                     ManagedRegister scratch,
136                     size_t size) = 0;
137 
138   virtual void Copy(ManagedRegister dest_base,
139                     Offset dest_offset,
140                     FrameOffset src,
141                     ManagedRegister scratch,
142                     size_t size) = 0;
143 
144   virtual void Copy(FrameOffset dest,
145                     FrameOffset src_base,
146                     Offset src_offset,
147                     ManagedRegister scratch,
148                     size_t size) = 0;
149 
150   virtual void Copy(ManagedRegister dest,
151                     Offset dest_offset,
152                     ManagedRegister src,
153                     Offset src_offset,
154                     ManagedRegister scratch,
155                     size_t size) = 0;
156 
157   virtual void Copy(FrameOffset dest,
158                     Offset dest_offset,
159                     FrameOffset src,
160                     Offset src_offset,
161                     ManagedRegister scratch,
162                     size_t size) = 0;
163 
164   virtual void MemoryBarrier(ManagedRegister scratch) = 0;
165 
166   // Sign extension
167   virtual void SignExtend(ManagedRegister mreg, size_t size) = 0;
168 
169   // Zero extension
170   virtual void ZeroExtend(ManagedRegister mreg, size_t size) = 0;
171 
172   // Exploit fast access in managed code to Thread::Current()
173   virtual void GetCurrentThread(ManagedRegister tr) = 0;
174   virtual void GetCurrentThread(FrameOffset dest_offset, ManagedRegister scratch) = 0;
175 
176   // Set up out_reg to hold a Object** into the handle scope, or to be null if the
177   // value is null and null_allowed. in_reg holds a possibly stale reference
178   // that can be used to avoid loading the handle scope entry to see if the value is
179   // null.
180   virtual void CreateHandleScopeEntry(ManagedRegister out_reg,
181                                       FrameOffset handlescope_offset,
182                                       ManagedRegister in_reg,
183                                       bool null_allowed) = 0;
184 
185   // Set up out_off to hold a Object** into the handle scope, or to be null if the
186   // value is null and null_allowed.
187   virtual void CreateHandleScopeEntry(FrameOffset out_off,
188                                       FrameOffset handlescope_offset,
189                                       ManagedRegister scratch,
190                                       bool null_allowed) = 0;
191 
192   // src holds a handle scope entry (Object**) load this into dst
193   virtual void LoadReferenceFromHandleScope(ManagedRegister dst, ManagedRegister src) = 0;
194 
195   // Heap::VerifyObject on src. In some cases (such as a reference to this) we
196   // know that src may not be null.
197   virtual void VerifyObject(ManagedRegister src, bool could_be_null) = 0;
198   virtual void VerifyObject(FrameOffset src, bool could_be_null) = 0;
199 
200   // Call to address held at [base+offset]
201   virtual void Call(ManagedRegister base, Offset offset, ManagedRegister scratch) = 0;
202   virtual void Call(FrameOffset base, Offset offset, ManagedRegister scratch) = 0;
203   virtual void CallFromThread(ThreadOffset<kPointerSize> offset, ManagedRegister scratch) = 0;
204 
205   // Generate code to check if Thread::Current()->exception_ is non-null
206   // and branch to a ExceptionSlowPath if it is.
207   virtual void ExceptionPoll(ManagedRegister scratch, size_t stack_adjust) = 0;
208 
209   // Create a new label that can be used with Jump/Bind calls.
210   virtual std::unique_ptr<JNIMacroLabel> CreateLabel() = 0;
211   // Emit an unconditional jump to the label.
212   virtual void Jump(JNIMacroLabel* label) = 0;
213   // Emit a conditional jump to the label by applying a unary condition test to the register.
214   virtual void Jump(JNIMacroLabel* label, JNIMacroUnaryCondition cond, ManagedRegister test) = 0;
215   // Code at this offset will serve as the target for the Jump call.
216   virtual void Bind(JNIMacroLabel* label) = 0;
217 
~JNIMacroAssembler()218   virtual ~JNIMacroAssembler() {}
219 
220   /**
221    * @brief Buffer of DWARF's Call Frame Information opcodes.
222    * @details It is used by debuggers and other tools to unwind the call stack.
223    */
224   virtual DebugFrameOpCodeWriterForAssembler& cfi() = 0;
225 
SetEmitRunTimeChecksInDebugMode(bool value)226   void SetEmitRunTimeChecksInDebugMode(bool value) {
227     emit_run_time_checks_in_debug_mode_ = value;
228   }
229 
230  protected:
JNIMacroAssembler()231   JNIMacroAssembler() {}
232 
233   // Should run-time checks be emitted in debug mode?
234   bool emit_run_time_checks_in_debug_mode_ = false;
235 };
236 
237 // A "Label" class used with the JNIMacroAssembler
238 // allowing one to use branches (jumping from one place to another).
239 //
240 // This is just an interface, so every platform must provide
241 // its own implementation of it.
242 //
243 // It is only safe to use a label created
244 // via JNIMacroAssembler::CreateLabel with that same macro assembler.
245 class JNIMacroLabel {
246  public:
247   virtual ~JNIMacroLabel() = 0;
248 
249   const InstructionSet isa_;
250  protected:
JNIMacroLabel(InstructionSet isa)251   explicit JNIMacroLabel(InstructionSet isa) : isa_(isa) {}
252 };
253 
~JNIMacroLabel()254 inline JNIMacroLabel::~JNIMacroLabel() {
255   // Compulsory definition for a pure virtual destructor
256   // to avoid linking errors.
257 }
258 
259 template <typename T, PointerSize kPointerSize>
260 class JNIMacroAssemblerFwd : public JNIMacroAssembler<kPointerSize> {
261  public:
FinalizeCode()262   void FinalizeCode() override {
263     asm_.FinalizeCode();
264   }
265 
CodeSize()266   size_t CodeSize() const override {
267     return asm_.CodeSize();
268   }
269 
FinalizeInstructions(const MemoryRegion & region)270   void FinalizeInstructions(const MemoryRegion& region) override {
271     asm_.FinalizeInstructions(region);
272   }
273 
cfi()274   DebugFrameOpCodeWriterForAssembler& cfi() override {
275     return asm_.cfi();
276   }
277 
278  protected:
JNIMacroAssemblerFwd(ArenaAllocator * allocator)279   explicit JNIMacroAssemblerFwd(ArenaAllocator* allocator) : asm_(allocator) {}
280 
281   T asm_;
282 };
283 
284 template <typename Self, typename PlatformLabel, InstructionSet kIsa>
285 class JNIMacroLabelCommon : public JNIMacroLabel {
286  public:
Cast(JNIMacroLabel * label)287   static Self* Cast(JNIMacroLabel* label) {
288     CHECK(label != nullptr);
289     CHECK_EQ(kIsa, label->isa_);
290 
291     return reinterpret_cast<Self*>(label);
292   }
293 
294  protected:
AsPlatformLabel()295   PlatformLabel* AsPlatformLabel() {
296     return &label_;
297   }
298 
JNIMacroLabelCommon()299   JNIMacroLabelCommon() : JNIMacroLabel(kIsa) {
300   }
301 
~JNIMacroLabelCommon()302   ~JNIMacroLabelCommon() override {}
303 
304  private:
305   PlatformLabel label_;
306 };
307 
308 }  // namespace art
309 
310 #endif  // ART_COMPILER_UTILS_JNI_MACRO_ASSEMBLER_H_
311