blob: 4d360158a251d9b319bb59db28d3404b6ccc7a62 [file] [log] [blame]
Andreas Gampe878d58c2015-01-15 23:24:00 -08001/*
2 * Copyright (C) 2015 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#include "intrinsics_arm64.h"
18
Serban Constantinescu579885a2015-02-22 20:51:33 +000019#include "arch/arm64/instruction_set_features_arm64.h"
Mathieu Chartiere401d142015-04-22 13:56:20 -070020#include "art_method.h"
Andreas Gampe878d58c2015-01-15 23:24:00 -080021#include "code_generator_arm64.h"
22#include "common_arm64.h"
23#include "entrypoints/quick/quick_entrypoints.h"
24#include "intrinsics.h"
Andreas Gampec6ea7d02017-02-01 16:46:28 -080025#include "lock_word.h"
Andreas Gampe878d58c2015-01-15 23:24:00 -080026#include "mirror/array-inl.h"
Andreas Gampec15a2f42017-04-21 12:09:39 -070027#include "mirror/object_array-inl.h"
Andreas Gampec6ea7d02017-02-01 16:46:28 -080028#include "mirror/reference.h"
Vladimir Markoe39f14f2017-02-10 15:44:25 +000029#include "mirror/string-inl.h"
Andreas Gampec6ea7d02017-02-01 16:46:28 -080030#include "scoped_thread_state_change-inl.h"
31#include "thread-inl.h"
Andreas Gampe878d58c2015-01-15 23:24:00 -080032#include "utils/arm64/assembler_arm64.h"
Andreas Gampe878d58c2015-01-15 23:24:00 -080033
Scott Wakeling97c72b72016-06-24 16:19:36 +010034using namespace vixl::aarch64; // NOLINT(build/namespaces)
Andreas Gampe878d58c2015-01-15 23:24:00 -080035
Artem Serovaf4e42a2016-08-08 15:11:24 +010036// TODO(VIXL): Make VIXL compile with -Wshadow.
Scott Wakeling97c72b72016-06-24 16:19:36 +010037#pragma GCC diagnostic push
38#pragma GCC diagnostic ignored "-Wshadow"
Artem Serovaf4e42a2016-08-08 15:11:24 +010039#include "aarch64/disasm-aarch64.h"
40#include "aarch64/macro-assembler-aarch64.h"
Scott Wakeling97c72b72016-06-24 16:19:36 +010041#pragma GCC diagnostic pop
Andreas Gampe878d58c2015-01-15 23:24:00 -080042
43namespace art {
44
45namespace arm64 {
46
47using helpers::DRegisterFrom;
48using helpers::FPRegisterFrom;
49using helpers::HeapOperand;
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +000050using helpers::LocationFrom;
Scott Wakeling9ee23f42015-07-23 10:44:35 +010051using helpers::OperandFrom;
Andreas Gampe878d58c2015-01-15 23:24:00 -080052using helpers::RegisterFrom;
53using helpers::SRegisterFrom;
54using helpers::WRegisterFrom;
55using helpers::XRegisterFrom;
xueliang.zhong49924c92016-03-03 10:52:51 +000056using helpers::InputRegisterAt;
Scott Wakeling1f36f412016-04-21 11:13:45 +010057using helpers::OutputRegister;
Andreas Gampe878d58c2015-01-15 23:24:00 -080058
Andreas Gampe878d58c2015-01-15 23:24:00 -080059namespace {
60
61ALWAYS_INLINE inline MemOperand AbsoluteHeapOperandFrom(Location location, size_t offset = 0) {
62 return MemOperand(XRegisterFrom(location), offset);
63}
64
65} // namespace
66
Scott Wakeling97c72b72016-06-24 16:19:36 +010067MacroAssembler* IntrinsicCodeGeneratorARM64::GetVIXLAssembler() {
Alexandre Rames087930f2016-08-02 13:45:28 +010068 return codegen_->GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -080069}
70
71ArenaAllocator* IntrinsicCodeGeneratorARM64::GetAllocator() {
72 return codegen_->GetGraph()->GetArena();
73}
74
Alexandre Rames087930f2016-08-02 13:45:28 +010075#define __ codegen->GetVIXLAssembler()->
Andreas Gampe878d58c2015-01-15 23:24:00 -080076
77static void MoveFromReturnRegister(Location trg,
78 Primitive::Type type,
79 CodeGeneratorARM64* codegen) {
80 if (!trg.IsValid()) {
81 DCHECK(type == Primitive::kPrimVoid);
82 return;
83 }
84
85 DCHECK_NE(type, Primitive::kPrimVoid);
86
Jeff Hao848f70a2014-01-15 13:49:50 -080087 if (Primitive::IsIntegralType(type) || type == Primitive::kPrimNot) {
Andreas Gampe878d58c2015-01-15 23:24:00 -080088 Register trg_reg = RegisterFrom(trg, type);
89 Register res_reg = RegisterFrom(ARM64ReturnLocation(type), type);
90 __ Mov(trg_reg, res_reg, kDiscardForSameWReg);
91 } else {
92 FPRegister trg_reg = FPRegisterFrom(trg, type);
93 FPRegister res_reg = FPRegisterFrom(ARM64ReturnLocation(type), type);
94 __ Fmov(trg_reg, res_reg);
95 }
96}
97
Roland Levillainec525fc2015-04-28 15:50:20 +010098static void MoveArguments(HInvoke* invoke, CodeGeneratorARM64* codegen) {
Roland Levillain2d27c8e2015-04-28 15:48:45 +010099 InvokeDexCallingConventionVisitorARM64 calling_convention_visitor;
Roland Levillainec525fc2015-04-28 15:50:20 +0100100 IntrinsicVisitor::MoveArguments(invoke, codegen, &calling_convention_visitor);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800101}
102
103// Slow-path for fallback (calling the managed code to handle the intrinsic) in an intrinsified
104// call. This will copy the arguments into the positions for a regular call.
105//
106// Note: The actual parameters are required to be in the locations given by the invoke's location
107// summary. If an intrinsic modifies those locations before a slowpath call, they must be
108// restored!
109class IntrinsicSlowPathARM64 : public SlowPathCodeARM64 {
110 public:
David Srbecky9cd6d372016-02-09 15:24:47 +0000111 explicit IntrinsicSlowPathARM64(HInvoke* invoke)
112 : SlowPathCodeARM64(invoke), invoke_(invoke) { }
Andreas Gampe878d58c2015-01-15 23:24:00 -0800113
114 void EmitNativeCode(CodeGenerator* codegen_in) OVERRIDE {
115 CodeGeneratorARM64* codegen = down_cast<CodeGeneratorARM64*>(codegen_in);
116 __ Bind(GetEntryLabel());
117
Nicolas Geoffraya8ac9132015-03-13 16:36:36 +0000118 SaveLiveRegisters(codegen, invoke_->GetLocations());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800119
Roland Levillainec525fc2015-04-28 15:50:20 +0100120 MoveArguments(invoke_, codegen);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800121
Artem Serov914d7a82017-02-07 14:33:49 +0000122 {
123 // Ensure that between the BLR (emitted by Generate*Call) and RecordPcInfo there
124 // are no pools emitted.
125 vixl::EmissionCheckScope guard(codegen->GetVIXLAssembler(), kInvokeCodeMarginSizeInBytes);
126 if (invoke_->IsInvokeStaticOrDirect()) {
127 codegen->GenerateStaticOrDirectCall(invoke_->AsInvokeStaticOrDirect(),
128 LocationFrom(kArtMethodRegister));
129 } else {
130 codegen->GenerateVirtualCall(invoke_->AsInvokeVirtual(), LocationFrom(kArtMethodRegister));
131 }
132 codegen->RecordPcInfo(invoke_, invoke_->GetDexPc(), this);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800133 }
134
135 // Copy the result back to the expected output.
136 Location out = invoke_->GetLocations()->Out();
137 if (out.IsValid()) {
138 DCHECK(out.IsRegister()); // TODO: Replace this when we support output in memory.
139 DCHECK(!invoke_->GetLocations()->GetLiveRegisters()->ContainsCoreRegister(out.reg()));
140 MoveFromReturnRegister(out, invoke_->GetType(), codegen);
141 }
142
Nicolas Geoffraya8ac9132015-03-13 16:36:36 +0000143 RestoreLiveRegisters(codegen, invoke_->GetLocations());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800144 __ B(GetExitLabel());
145 }
146
Alexandre Rames9931f312015-06-19 14:47:01 +0100147 const char* GetDescription() const OVERRIDE { return "IntrinsicSlowPathARM64"; }
148
Andreas Gampe878d58c2015-01-15 23:24:00 -0800149 private:
150 // The instruction where this slow path is happening.
151 HInvoke* const invoke_;
152
153 DISALLOW_COPY_AND_ASSIGN(IntrinsicSlowPathARM64);
154};
155
Roland Levillain0b671c02016-08-19 12:02:34 +0100156// Slow path implementing the SystemArrayCopy intrinsic copy loop with read barriers.
157class ReadBarrierSystemArrayCopySlowPathARM64 : public SlowPathCodeARM64 {
158 public:
159 ReadBarrierSystemArrayCopySlowPathARM64(HInstruction* instruction, Location tmp)
160 : SlowPathCodeARM64(instruction), tmp_(tmp) {
161 DCHECK(kEmitCompilerReadBarrier);
162 DCHECK(kUseBakerReadBarrier);
163 }
164
165 void EmitNativeCode(CodeGenerator* codegen_in) OVERRIDE {
166 CodeGeneratorARM64* codegen = down_cast<CodeGeneratorARM64*>(codegen_in);
167 LocationSummary* locations = instruction_->GetLocations();
168 DCHECK(locations->CanCall());
169 DCHECK(instruction_->IsInvokeStaticOrDirect())
170 << "Unexpected instruction in read barrier arraycopy slow path: "
171 << instruction_->DebugName();
172 DCHECK(instruction_->GetLocations()->Intrinsified());
173 DCHECK_EQ(instruction_->AsInvoke()->GetIntrinsic(), Intrinsics::kSystemArrayCopy);
174
175 const int32_t element_size = Primitive::ComponentSize(Primitive::kPrimNot);
176
177 Register src_curr_addr = XRegisterFrom(locations->GetTemp(0));
178 Register dst_curr_addr = XRegisterFrom(locations->GetTemp(1));
179 Register src_stop_addr = XRegisterFrom(locations->GetTemp(2));
180 Register tmp_reg = WRegisterFrom(tmp_);
181
182 __ Bind(GetEntryLabel());
183 vixl::aarch64::Label slow_copy_loop;
184 __ Bind(&slow_copy_loop);
185 __ Ldr(tmp_reg, MemOperand(src_curr_addr, element_size, PostIndex));
186 codegen->GetAssembler()->MaybeUnpoisonHeapReference(tmp_reg);
187 // TODO: Inline the mark bit check before calling the runtime?
188 // tmp_reg = ReadBarrier::Mark(tmp_reg);
189 // No need to save live registers; it's taken care of by the
190 // entrypoint. Also, there is no need to update the stack mask,
191 // as this runtime call will not trigger a garbage collection.
192 // (See ReadBarrierMarkSlowPathARM64::EmitNativeCode for more
193 // explanations.)
194 DCHECK_NE(tmp_.reg(), LR);
195 DCHECK_NE(tmp_.reg(), WSP);
196 DCHECK_NE(tmp_.reg(), WZR);
197 // IP0 is used internally by the ReadBarrierMarkRegX entry point
198 // as a temporary (and not preserved). It thus cannot be used by
199 // any live register in this slow path.
200 DCHECK_NE(LocationFrom(src_curr_addr).reg(), IP0);
201 DCHECK_NE(LocationFrom(dst_curr_addr).reg(), IP0);
202 DCHECK_NE(LocationFrom(src_stop_addr).reg(), IP0);
203 DCHECK_NE(tmp_.reg(), IP0);
204 DCHECK(0 <= tmp_.reg() && tmp_.reg() < kNumberOfWRegisters) << tmp_.reg();
Roland Levillain9cc0ea82017-03-16 11:25:59 +0000205 // TODO: Load the entrypoint once before the loop, instead of
206 // loading it at every iteration.
Roland Levillain0b671c02016-08-19 12:02:34 +0100207 int32_t entry_point_offset =
208 CodeGenerator::GetReadBarrierMarkEntryPointsOffset<kArm64PointerSize>(tmp_.reg());
209 // This runtime call does not require a stack map.
210 codegen->InvokeRuntimeWithoutRecordingPcInfo(entry_point_offset, instruction_, this);
211 codegen->GetAssembler()->MaybePoisonHeapReference(tmp_reg);
212 __ Str(tmp_reg, MemOperand(dst_curr_addr, element_size, PostIndex));
213 __ Cmp(src_curr_addr, src_stop_addr);
214 __ B(&slow_copy_loop, ne);
215 __ B(GetExitLabel());
216 }
217
218 const char* GetDescription() const OVERRIDE { return "ReadBarrierSystemArrayCopySlowPathARM64"; }
219
220 private:
221 Location tmp_;
222
223 DISALLOW_COPY_AND_ASSIGN(ReadBarrierSystemArrayCopySlowPathARM64);
224};
Andreas Gampe878d58c2015-01-15 23:24:00 -0800225#undef __
226
227bool IntrinsicLocationsBuilderARM64::TryDispatch(HInvoke* invoke) {
228 Dispatch(invoke);
229 LocationSummary* res = invoke->GetLocations();
Roland Levillain22ccc3a2015-11-24 13:10:05 +0000230 if (res == nullptr) {
231 return false;
232 }
Roland Levillain22ccc3a2015-11-24 13:10:05 +0000233 return res->Intrinsified();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800234}
235
236#define __ masm->
237
238static void CreateFPToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
239 LocationSummary* locations = new (arena) LocationSummary(invoke,
240 LocationSummary::kNoCall,
241 kIntrinsified);
242 locations->SetInAt(0, Location::RequiresFpuRegister());
243 locations->SetOut(Location::RequiresRegister());
244}
245
246static void CreateIntToFPLocations(ArenaAllocator* arena, HInvoke* invoke) {
247 LocationSummary* locations = new (arena) LocationSummary(invoke,
248 LocationSummary::kNoCall,
249 kIntrinsified);
250 locations->SetInAt(0, Location::RequiresRegister());
251 locations->SetOut(Location::RequiresFpuRegister());
252}
253
Scott Wakeling97c72b72016-06-24 16:19:36 +0100254static void MoveFPToInt(LocationSummary* locations, bool is64bit, MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800255 Location input = locations->InAt(0);
256 Location output = locations->Out();
257 __ Fmov(is64bit ? XRegisterFrom(output) : WRegisterFrom(output),
258 is64bit ? DRegisterFrom(input) : SRegisterFrom(input));
259}
260
Scott Wakeling97c72b72016-06-24 16:19:36 +0100261static void MoveIntToFP(LocationSummary* locations, bool is64bit, MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800262 Location input = locations->InAt(0);
263 Location output = locations->Out();
264 __ Fmov(is64bit ? DRegisterFrom(output) : SRegisterFrom(output),
265 is64bit ? XRegisterFrom(input) : WRegisterFrom(input));
266}
267
268void IntrinsicLocationsBuilderARM64::VisitDoubleDoubleToRawLongBits(HInvoke* invoke) {
269 CreateFPToIntLocations(arena_, invoke);
270}
271void IntrinsicLocationsBuilderARM64::VisitDoubleLongBitsToDouble(HInvoke* invoke) {
272 CreateIntToFPLocations(arena_, invoke);
273}
274
275void IntrinsicCodeGeneratorARM64::VisitDoubleDoubleToRawLongBits(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000276 MoveFPToInt(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800277}
278void IntrinsicCodeGeneratorARM64::VisitDoubleLongBitsToDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000279 MoveIntToFP(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800280}
281
282void IntrinsicLocationsBuilderARM64::VisitFloatFloatToRawIntBits(HInvoke* invoke) {
283 CreateFPToIntLocations(arena_, invoke);
284}
285void IntrinsicLocationsBuilderARM64::VisitFloatIntBitsToFloat(HInvoke* invoke) {
286 CreateIntToFPLocations(arena_, invoke);
287}
288
289void IntrinsicCodeGeneratorARM64::VisitFloatFloatToRawIntBits(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000290 MoveFPToInt(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800291}
292void IntrinsicCodeGeneratorARM64::VisitFloatIntBitsToFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000293 MoveIntToFP(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800294}
295
296static void CreateIntToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
297 LocationSummary* locations = new (arena) LocationSummary(invoke,
298 LocationSummary::kNoCall,
299 kIntrinsified);
300 locations->SetInAt(0, Location::RequiresRegister());
301 locations->SetOut(Location::RequiresRegister(), Location::kNoOutputOverlap);
302}
303
304static void GenReverseBytes(LocationSummary* locations,
305 Primitive::Type type,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100306 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800307 Location in = locations->InAt(0);
308 Location out = locations->Out();
309
310 switch (type) {
311 case Primitive::kPrimShort:
312 __ Rev16(WRegisterFrom(out), WRegisterFrom(in));
313 __ Sxth(WRegisterFrom(out), WRegisterFrom(out));
314 break;
315 case Primitive::kPrimInt:
316 case Primitive::kPrimLong:
317 __ Rev(RegisterFrom(out, type), RegisterFrom(in, type));
318 break;
319 default:
320 LOG(FATAL) << "Unexpected size for reverse-bytes: " << type;
321 UNREACHABLE();
322 }
323}
324
325void IntrinsicLocationsBuilderARM64::VisitIntegerReverseBytes(HInvoke* invoke) {
326 CreateIntToIntLocations(arena_, invoke);
327}
328
329void IntrinsicCodeGeneratorARM64::VisitIntegerReverseBytes(HInvoke* invoke) {
330 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimInt, GetVIXLAssembler());
331}
332
333void IntrinsicLocationsBuilderARM64::VisitLongReverseBytes(HInvoke* invoke) {
334 CreateIntToIntLocations(arena_, invoke);
335}
336
337void IntrinsicCodeGeneratorARM64::VisitLongReverseBytes(HInvoke* invoke) {
338 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimLong, GetVIXLAssembler());
339}
340
341void IntrinsicLocationsBuilderARM64::VisitShortReverseBytes(HInvoke* invoke) {
342 CreateIntToIntLocations(arena_, invoke);
343}
344
345void IntrinsicCodeGeneratorARM64::VisitShortReverseBytes(HInvoke* invoke) {
346 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimShort, GetVIXLAssembler());
347}
348
Aart Bik7b565022016-01-28 14:36:22 -0800349static void CreateIntIntToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
350 LocationSummary* locations = new (arena) LocationSummary(invoke,
351 LocationSummary::kNoCall,
352 kIntrinsified);
353 locations->SetInAt(0, Location::RequiresRegister());
354 locations->SetInAt(1, Location::RequiresRegister());
355 locations->SetOut(Location::RequiresRegister(), Location::kNoOutputOverlap);
356}
357
Scott Wakeling611d3392015-07-10 11:42:06 +0100358static void GenNumberOfLeadingZeros(LocationSummary* locations,
359 Primitive::Type type,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100360 MacroAssembler* masm) {
Scott Wakeling611d3392015-07-10 11:42:06 +0100361 DCHECK(type == Primitive::kPrimInt || type == Primitive::kPrimLong);
362
363 Location in = locations->InAt(0);
364 Location out = locations->Out();
365
366 __ Clz(RegisterFrom(out, type), RegisterFrom(in, type));
367}
368
369void IntrinsicLocationsBuilderARM64::VisitIntegerNumberOfLeadingZeros(HInvoke* invoke) {
370 CreateIntToIntLocations(arena_, invoke);
371}
372
373void IntrinsicCodeGeneratorARM64::VisitIntegerNumberOfLeadingZeros(HInvoke* invoke) {
374 GenNumberOfLeadingZeros(invoke->GetLocations(), Primitive::kPrimInt, GetVIXLAssembler());
375}
376
377void IntrinsicLocationsBuilderARM64::VisitLongNumberOfLeadingZeros(HInvoke* invoke) {
378 CreateIntToIntLocations(arena_, invoke);
379}
380
381void IntrinsicCodeGeneratorARM64::VisitLongNumberOfLeadingZeros(HInvoke* invoke) {
382 GenNumberOfLeadingZeros(invoke->GetLocations(), Primitive::kPrimLong, GetVIXLAssembler());
383}
384
Scott Wakeling9ee23f42015-07-23 10:44:35 +0100385static void GenNumberOfTrailingZeros(LocationSummary* locations,
386 Primitive::Type type,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100387 MacroAssembler* masm) {
Scott Wakeling9ee23f42015-07-23 10:44:35 +0100388 DCHECK(type == Primitive::kPrimInt || type == Primitive::kPrimLong);
389
390 Location in = locations->InAt(0);
391 Location out = locations->Out();
392
393 __ Rbit(RegisterFrom(out, type), RegisterFrom(in, type));
394 __ Clz(RegisterFrom(out, type), RegisterFrom(out, type));
395}
396
397void IntrinsicLocationsBuilderARM64::VisitIntegerNumberOfTrailingZeros(HInvoke* invoke) {
398 CreateIntToIntLocations(arena_, invoke);
399}
400
401void IntrinsicCodeGeneratorARM64::VisitIntegerNumberOfTrailingZeros(HInvoke* invoke) {
402 GenNumberOfTrailingZeros(invoke->GetLocations(), Primitive::kPrimInt, GetVIXLAssembler());
403}
404
405void IntrinsicLocationsBuilderARM64::VisitLongNumberOfTrailingZeros(HInvoke* invoke) {
406 CreateIntToIntLocations(arena_, invoke);
407}
408
409void IntrinsicCodeGeneratorARM64::VisitLongNumberOfTrailingZeros(HInvoke* invoke) {
410 GenNumberOfTrailingZeros(invoke->GetLocations(), Primitive::kPrimLong, GetVIXLAssembler());
411}
412
Andreas Gampe878d58c2015-01-15 23:24:00 -0800413static void GenReverse(LocationSummary* locations,
414 Primitive::Type type,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100415 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800416 DCHECK(type == Primitive::kPrimInt || type == Primitive::kPrimLong);
417
418 Location in = locations->InAt(0);
419 Location out = locations->Out();
420
421 __ Rbit(RegisterFrom(out, type), RegisterFrom(in, type));
422}
423
424void IntrinsicLocationsBuilderARM64::VisitIntegerReverse(HInvoke* invoke) {
425 CreateIntToIntLocations(arena_, invoke);
426}
427
428void IntrinsicCodeGeneratorARM64::VisitIntegerReverse(HInvoke* invoke) {
429 GenReverse(invoke->GetLocations(), Primitive::kPrimInt, GetVIXLAssembler());
430}
431
432void IntrinsicLocationsBuilderARM64::VisitLongReverse(HInvoke* invoke) {
433 CreateIntToIntLocations(arena_, invoke);
434}
435
436void IntrinsicCodeGeneratorARM64::VisitLongReverse(HInvoke* invoke) {
437 GenReverse(invoke->GetLocations(), Primitive::kPrimLong, GetVIXLAssembler());
438}
439
Scott Wakeling97c72b72016-06-24 16:19:36 +0100440static void GenBitCount(HInvoke* instr, Primitive::Type type, MacroAssembler* masm) {
Roland Levillainfa3912e2016-04-01 18:21:55 +0100441 DCHECK(Primitive::IsIntOrLongType(type)) << type;
442 DCHECK_EQ(instr->GetType(), Primitive::kPrimInt);
443 DCHECK_EQ(Primitive::PrimitiveKind(instr->InputAt(0)->GetType()), type);
xueliang.zhong49924c92016-03-03 10:52:51 +0000444
xueliang.zhong49924c92016-03-03 10:52:51 +0000445 UseScratchRegisterScope temps(masm);
446
Nicolas Geoffray457413a2016-03-04 11:10:17 +0000447 Register src = InputRegisterAt(instr, 0);
Roland Levillainfa3912e2016-04-01 18:21:55 +0100448 Register dst = RegisterFrom(instr->GetLocations()->Out(), type);
449 FPRegister fpr = (type == Primitive::kPrimLong) ? temps.AcquireD() : temps.AcquireS();
xueliang.zhong49924c92016-03-03 10:52:51 +0000450
451 __ Fmov(fpr, src);
Nicolas Geoffray457413a2016-03-04 11:10:17 +0000452 __ Cnt(fpr.V8B(), fpr.V8B());
453 __ Addv(fpr.B(), fpr.V8B());
xueliang.zhong49924c92016-03-03 10:52:51 +0000454 __ Fmov(dst, fpr);
455}
456
457void IntrinsicLocationsBuilderARM64::VisitLongBitCount(HInvoke* invoke) {
458 CreateIntToIntLocations(arena_, invoke);
459}
460
461void IntrinsicCodeGeneratorARM64::VisitLongBitCount(HInvoke* invoke) {
Roland Levillainfa3912e2016-04-01 18:21:55 +0100462 GenBitCount(invoke, Primitive::kPrimLong, GetVIXLAssembler());
xueliang.zhong49924c92016-03-03 10:52:51 +0000463}
464
465void IntrinsicLocationsBuilderARM64::VisitIntegerBitCount(HInvoke* invoke) {
466 CreateIntToIntLocations(arena_, invoke);
467}
468
469void IntrinsicCodeGeneratorARM64::VisitIntegerBitCount(HInvoke* invoke) {
Roland Levillainfa3912e2016-04-01 18:21:55 +0100470 GenBitCount(invoke, Primitive::kPrimInt, GetVIXLAssembler());
xueliang.zhong49924c92016-03-03 10:52:51 +0000471}
472
Andreas Gampe878d58c2015-01-15 23:24:00 -0800473static void CreateFPToFPLocations(ArenaAllocator* arena, HInvoke* invoke) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800474 LocationSummary* locations = new (arena) LocationSummary(invoke,
475 LocationSummary::kNoCall,
476 kIntrinsified);
477 locations->SetInAt(0, Location::RequiresFpuRegister());
478 locations->SetOut(Location::RequiresFpuRegister(), Location::kNoOutputOverlap);
479}
480
Scott Wakeling97c72b72016-06-24 16:19:36 +0100481static void MathAbsFP(LocationSummary* locations, bool is64bit, MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800482 Location in = locations->InAt(0);
483 Location out = locations->Out();
484
485 FPRegister in_reg = is64bit ? DRegisterFrom(in) : SRegisterFrom(in);
486 FPRegister out_reg = is64bit ? DRegisterFrom(out) : SRegisterFrom(out);
487
488 __ Fabs(out_reg, in_reg);
489}
490
491void IntrinsicLocationsBuilderARM64::VisitMathAbsDouble(HInvoke* invoke) {
492 CreateFPToFPLocations(arena_, invoke);
493}
494
495void IntrinsicCodeGeneratorARM64::VisitMathAbsDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000496 MathAbsFP(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800497}
498
499void IntrinsicLocationsBuilderARM64::VisitMathAbsFloat(HInvoke* invoke) {
500 CreateFPToFPLocations(arena_, invoke);
501}
502
503void IntrinsicCodeGeneratorARM64::VisitMathAbsFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000504 MathAbsFP(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800505}
506
507static void CreateIntToInt(ArenaAllocator* arena, HInvoke* invoke) {
508 LocationSummary* locations = new (arena) LocationSummary(invoke,
509 LocationSummary::kNoCall,
510 kIntrinsified);
511 locations->SetInAt(0, Location::RequiresRegister());
512 locations->SetOut(Location::RequiresRegister(), Location::kNoOutputOverlap);
513}
514
515static void GenAbsInteger(LocationSummary* locations,
516 bool is64bit,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100517 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800518 Location in = locations->InAt(0);
519 Location output = locations->Out();
520
521 Register in_reg = is64bit ? XRegisterFrom(in) : WRegisterFrom(in);
522 Register out_reg = is64bit ? XRegisterFrom(output) : WRegisterFrom(output);
523
524 __ Cmp(in_reg, Operand(0));
525 __ Cneg(out_reg, in_reg, lt);
526}
527
528void IntrinsicLocationsBuilderARM64::VisitMathAbsInt(HInvoke* invoke) {
529 CreateIntToInt(arena_, invoke);
530}
531
532void IntrinsicCodeGeneratorARM64::VisitMathAbsInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000533 GenAbsInteger(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800534}
535
536void IntrinsicLocationsBuilderARM64::VisitMathAbsLong(HInvoke* invoke) {
537 CreateIntToInt(arena_, invoke);
538}
539
540void IntrinsicCodeGeneratorARM64::VisitMathAbsLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000541 GenAbsInteger(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800542}
543
544static void GenMinMaxFP(LocationSummary* locations,
545 bool is_min,
546 bool is_double,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100547 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800548 Location op1 = locations->InAt(0);
549 Location op2 = locations->InAt(1);
550 Location out = locations->Out();
551
552 FPRegister op1_reg = is_double ? DRegisterFrom(op1) : SRegisterFrom(op1);
553 FPRegister op2_reg = is_double ? DRegisterFrom(op2) : SRegisterFrom(op2);
554 FPRegister out_reg = is_double ? DRegisterFrom(out) : SRegisterFrom(out);
555 if (is_min) {
556 __ Fmin(out_reg, op1_reg, op2_reg);
557 } else {
558 __ Fmax(out_reg, op1_reg, op2_reg);
559 }
560}
561
562static void CreateFPFPToFPLocations(ArenaAllocator* arena, HInvoke* invoke) {
563 LocationSummary* locations = new (arena) LocationSummary(invoke,
564 LocationSummary::kNoCall,
565 kIntrinsified);
566 locations->SetInAt(0, Location::RequiresFpuRegister());
567 locations->SetInAt(1, Location::RequiresFpuRegister());
568 locations->SetOut(Location::RequiresFpuRegister(), Location::kNoOutputOverlap);
569}
570
571void IntrinsicLocationsBuilderARM64::VisitMathMinDoubleDouble(HInvoke* invoke) {
572 CreateFPFPToFPLocations(arena_, invoke);
573}
574
575void IntrinsicCodeGeneratorARM64::VisitMathMinDoubleDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000576 GenMinMaxFP(invoke->GetLocations(), /* is_min */ true, /* is_double */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800577}
578
579void IntrinsicLocationsBuilderARM64::VisitMathMinFloatFloat(HInvoke* invoke) {
580 CreateFPFPToFPLocations(arena_, invoke);
581}
582
583void IntrinsicCodeGeneratorARM64::VisitMathMinFloatFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000584 GenMinMaxFP(invoke->GetLocations(), /* is_min */ true, /* is_double */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800585}
586
587void IntrinsicLocationsBuilderARM64::VisitMathMaxDoubleDouble(HInvoke* invoke) {
588 CreateFPFPToFPLocations(arena_, invoke);
589}
590
591void IntrinsicCodeGeneratorARM64::VisitMathMaxDoubleDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000592 GenMinMaxFP(invoke->GetLocations(), /* is_min */ false, /* is_double */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800593}
594
595void IntrinsicLocationsBuilderARM64::VisitMathMaxFloatFloat(HInvoke* invoke) {
596 CreateFPFPToFPLocations(arena_, invoke);
597}
598
599void IntrinsicCodeGeneratorARM64::VisitMathMaxFloatFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000600 GenMinMaxFP(
601 invoke->GetLocations(), /* is_min */ false, /* is_double */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800602}
603
604static void GenMinMax(LocationSummary* locations,
605 bool is_min,
606 bool is_long,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100607 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800608 Location op1 = locations->InAt(0);
609 Location op2 = locations->InAt(1);
610 Location out = locations->Out();
611
612 Register op1_reg = is_long ? XRegisterFrom(op1) : WRegisterFrom(op1);
613 Register op2_reg = is_long ? XRegisterFrom(op2) : WRegisterFrom(op2);
614 Register out_reg = is_long ? XRegisterFrom(out) : WRegisterFrom(out);
615
616 __ Cmp(op1_reg, op2_reg);
617 __ Csel(out_reg, op1_reg, op2_reg, is_min ? lt : gt);
618}
619
Andreas Gampe878d58c2015-01-15 23:24:00 -0800620void IntrinsicLocationsBuilderARM64::VisitMathMinIntInt(HInvoke* invoke) {
621 CreateIntIntToIntLocations(arena_, invoke);
622}
623
624void IntrinsicCodeGeneratorARM64::VisitMathMinIntInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000625 GenMinMax(invoke->GetLocations(), /* is_min */ true, /* is_long */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800626}
627
628void IntrinsicLocationsBuilderARM64::VisitMathMinLongLong(HInvoke* invoke) {
629 CreateIntIntToIntLocations(arena_, invoke);
630}
631
632void IntrinsicCodeGeneratorARM64::VisitMathMinLongLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000633 GenMinMax(invoke->GetLocations(), /* is_min */ true, /* is_long */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800634}
635
636void IntrinsicLocationsBuilderARM64::VisitMathMaxIntInt(HInvoke* invoke) {
637 CreateIntIntToIntLocations(arena_, invoke);
638}
639
640void IntrinsicCodeGeneratorARM64::VisitMathMaxIntInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000641 GenMinMax(invoke->GetLocations(), /* is_min */ false, /* is_long */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800642}
643
644void IntrinsicLocationsBuilderARM64::VisitMathMaxLongLong(HInvoke* invoke) {
645 CreateIntIntToIntLocations(arena_, invoke);
646}
647
648void IntrinsicCodeGeneratorARM64::VisitMathMaxLongLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000649 GenMinMax(invoke->GetLocations(), /* is_min */ false, /* is_long */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800650}
651
652void IntrinsicLocationsBuilderARM64::VisitMathSqrt(HInvoke* invoke) {
653 CreateFPToFPLocations(arena_, invoke);
654}
655
656void IntrinsicCodeGeneratorARM64::VisitMathSqrt(HInvoke* invoke) {
657 LocationSummary* locations = invoke->GetLocations();
Scott Wakeling97c72b72016-06-24 16:19:36 +0100658 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800659 __ Fsqrt(DRegisterFrom(locations->Out()), DRegisterFrom(locations->InAt(0)));
660}
661
662void IntrinsicLocationsBuilderARM64::VisitMathCeil(HInvoke* invoke) {
663 CreateFPToFPLocations(arena_, invoke);
664}
665
666void IntrinsicCodeGeneratorARM64::VisitMathCeil(HInvoke* invoke) {
667 LocationSummary* locations = invoke->GetLocations();
Scott Wakeling97c72b72016-06-24 16:19:36 +0100668 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800669 __ Frintp(DRegisterFrom(locations->Out()), DRegisterFrom(locations->InAt(0)));
670}
671
672void IntrinsicLocationsBuilderARM64::VisitMathFloor(HInvoke* invoke) {
673 CreateFPToFPLocations(arena_, invoke);
674}
675
676void IntrinsicCodeGeneratorARM64::VisitMathFloor(HInvoke* invoke) {
677 LocationSummary* locations = invoke->GetLocations();
Scott Wakeling97c72b72016-06-24 16:19:36 +0100678 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800679 __ Frintm(DRegisterFrom(locations->Out()), DRegisterFrom(locations->InAt(0)));
680}
681
682void IntrinsicLocationsBuilderARM64::VisitMathRint(HInvoke* invoke) {
683 CreateFPToFPLocations(arena_, invoke);
684}
685
686void IntrinsicCodeGeneratorARM64::VisitMathRint(HInvoke* invoke) {
687 LocationSummary* locations = invoke->GetLocations();
Scott Wakeling97c72b72016-06-24 16:19:36 +0100688 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800689 __ Frintn(DRegisterFrom(locations->Out()), DRegisterFrom(locations->InAt(0)));
690}
691
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100692static void CreateFPToIntPlusFPTempLocations(ArenaAllocator* arena, HInvoke* invoke) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800693 LocationSummary* locations = new (arena) LocationSummary(invoke,
694 LocationSummary::kNoCall,
695 kIntrinsified);
696 locations->SetInAt(0, Location::RequiresFpuRegister());
697 locations->SetOut(Location::RequiresRegister());
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100698 locations->AddTemp(Location::RequiresFpuRegister());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800699}
700
Scott Wakeling97c72b72016-06-24 16:19:36 +0100701static void GenMathRound(HInvoke* invoke, bool is_double, vixl::aarch64::MacroAssembler* masm) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100702 // Java 8 API definition for Math.round():
703 // Return the closest long or int to the argument, with ties rounding to positive infinity.
704 //
705 // There is no single instruction in ARMv8 that can support the above definition.
706 // We choose to use FCVTAS here, because it has closest semantic.
707 // FCVTAS performs rounding to nearest integer, ties away from zero.
708 // For most inputs (positive values, zero or NaN), this instruction is enough.
709 // We only need a few handling code after FCVTAS if the input is negative half value.
710 //
711 // The reason why we didn't choose FCVTPS instruction here is that
712 // although it performs rounding toward positive infinity, it doesn't perform rounding to nearest.
713 // For example, FCVTPS(-1.9) = -1 and FCVTPS(1.1) = 2.
714 // If we were using this instruction, for most inputs, more handling code would be needed.
715 LocationSummary* l = invoke->GetLocations();
716 FPRegister in_reg = is_double ? DRegisterFrom(l->InAt(0)) : SRegisterFrom(l->InAt(0));
717 FPRegister tmp_fp = is_double ? DRegisterFrom(l->GetTemp(0)) : SRegisterFrom(l->GetTemp(0));
718 Register out_reg = is_double ? XRegisterFrom(l->Out()) : WRegisterFrom(l->Out());
Scott Wakeling97c72b72016-06-24 16:19:36 +0100719 vixl::aarch64::Label done;
Andreas Gampe878d58c2015-01-15 23:24:00 -0800720
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100721 // Round to nearest integer, ties away from zero.
722 __ Fcvtas(out_reg, in_reg);
723
724 // For positive values, zero or NaN inputs, rounding is done.
Scott Wakeling97c72b72016-06-24 16:19:36 +0100725 __ Tbz(out_reg, out_reg.GetSizeInBits() - 1, &done);
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100726
727 // Handle input < 0 cases.
728 // If input is negative but not a tie, previous result (round to nearest) is valid.
729 // If input is a negative tie, out_reg += 1.
730 __ Frinta(tmp_fp, in_reg);
731 __ Fsub(tmp_fp, in_reg, tmp_fp);
732 __ Fcmp(tmp_fp, 0.5);
733 __ Cinc(out_reg, out_reg, eq);
734
735 __ Bind(&done);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800736}
737
738void IntrinsicLocationsBuilderARM64::VisitMathRoundDouble(HInvoke* invoke) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100739 CreateFPToIntPlusFPTempLocations(arena_, invoke);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800740}
741
742void IntrinsicCodeGeneratorARM64::VisitMathRoundDouble(HInvoke* invoke) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100743 GenMathRound(invoke, /* is_double */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800744}
745
746void IntrinsicLocationsBuilderARM64::VisitMathRoundFloat(HInvoke* invoke) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100747 CreateFPToIntPlusFPTempLocations(arena_, invoke);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800748}
749
750void IntrinsicCodeGeneratorARM64::VisitMathRoundFloat(HInvoke* invoke) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100751 GenMathRound(invoke, /* is_double */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800752}
753
754void IntrinsicLocationsBuilderARM64::VisitMemoryPeekByte(HInvoke* invoke) {
755 CreateIntToIntLocations(arena_, invoke);
756}
757
758void IntrinsicCodeGeneratorARM64::VisitMemoryPeekByte(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100759 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800760 __ Ldrsb(WRegisterFrom(invoke->GetLocations()->Out()),
761 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
762}
763
764void IntrinsicLocationsBuilderARM64::VisitMemoryPeekIntNative(HInvoke* invoke) {
765 CreateIntToIntLocations(arena_, invoke);
766}
767
768void IntrinsicCodeGeneratorARM64::VisitMemoryPeekIntNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100769 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800770 __ Ldr(WRegisterFrom(invoke->GetLocations()->Out()),
771 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
772}
773
774void IntrinsicLocationsBuilderARM64::VisitMemoryPeekLongNative(HInvoke* invoke) {
775 CreateIntToIntLocations(arena_, invoke);
776}
777
778void IntrinsicCodeGeneratorARM64::VisitMemoryPeekLongNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100779 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800780 __ Ldr(XRegisterFrom(invoke->GetLocations()->Out()),
781 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
782}
783
784void IntrinsicLocationsBuilderARM64::VisitMemoryPeekShortNative(HInvoke* invoke) {
785 CreateIntToIntLocations(arena_, invoke);
786}
787
788void IntrinsicCodeGeneratorARM64::VisitMemoryPeekShortNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100789 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800790 __ Ldrsh(WRegisterFrom(invoke->GetLocations()->Out()),
791 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
792}
793
794static void CreateIntIntToVoidLocations(ArenaAllocator* arena, HInvoke* invoke) {
795 LocationSummary* locations = new (arena) LocationSummary(invoke,
796 LocationSummary::kNoCall,
797 kIntrinsified);
798 locations->SetInAt(0, Location::RequiresRegister());
799 locations->SetInAt(1, Location::RequiresRegister());
800}
801
802void IntrinsicLocationsBuilderARM64::VisitMemoryPokeByte(HInvoke* invoke) {
803 CreateIntIntToVoidLocations(arena_, invoke);
804}
805
806void IntrinsicCodeGeneratorARM64::VisitMemoryPokeByte(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100807 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800808 __ Strb(WRegisterFrom(invoke->GetLocations()->InAt(1)),
809 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
810}
811
812void IntrinsicLocationsBuilderARM64::VisitMemoryPokeIntNative(HInvoke* invoke) {
813 CreateIntIntToVoidLocations(arena_, invoke);
814}
815
816void IntrinsicCodeGeneratorARM64::VisitMemoryPokeIntNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100817 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800818 __ Str(WRegisterFrom(invoke->GetLocations()->InAt(1)),
819 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
820}
821
822void IntrinsicLocationsBuilderARM64::VisitMemoryPokeLongNative(HInvoke* invoke) {
823 CreateIntIntToVoidLocations(arena_, invoke);
824}
825
826void IntrinsicCodeGeneratorARM64::VisitMemoryPokeLongNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100827 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800828 __ Str(XRegisterFrom(invoke->GetLocations()->InAt(1)),
829 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
830}
831
832void IntrinsicLocationsBuilderARM64::VisitMemoryPokeShortNative(HInvoke* invoke) {
833 CreateIntIntToVoidLocations(arena_, invoke);
834}
835
836void IntrinsicCodeGeneratorARM64::VisitMemoryPokeShortNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100837 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800838 __ Strh(WRegisterFrom(invoke->GetLocations()->InAt(1)),
839 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
840}
841
842void IntrinsicLocationsBuilderARM64::VisitThreadCurrentThread(HInvoke* invoke) {
843 LocationSummary* locations = new (arena_) LocationSummary(invoke,
844 LocationSummary::kNoCall,
845 kIntrinsified);
846 locations->SetOut(Location::RequiresRegister());
847}
848
849void IntrinsicCodeGeneratorARM64::VisitThreadCurrentThread(HInvoke* invoke) {
850 codegen_->Load(Primitive::kPrimNot, WRegisterFrom(invoke->GetLocations()->Out()),
Andreas Gampe542451c2016-07-26 09:02:02 -0700851 MemOperand(tr, Thread::PeerOffset<kArm64PointerSize>().Int32Value()));
Andreas Gampe878d58c2015-01-15 23:24:00 -0800852}
853
854static void GenUnsafeGet(HInvoke* invoke,
855 Primitive::Type type,
856 bool is_volatile,
857 CodeGeneratorARM64* codegen) {
858 LocationSummary* locations = invoke->GetLocations();
859 DCHECK((type == Primitive::kPrimInt) ||
860 (type == Primitive::kPrimLong) ||
861 (type == Primitive::kPrimNot));
Roland Levillain22ccc3a2015-11-24 13:10:05 +0000862 Location base_loc = locations->InAt(1);
863 Register base = WRegisterFrom(base_loc); // Object pointer.
864 Location offset_loc = locations->InAt(2);
865 Register offset = XRegisterFrom(offset_loc); // Long offset.
866 Location trg_loc = locations->Out();
867 Register trg = RegisterFrom(trg_loc, type);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800868
Roland Levillain44015862016-01-22 11:47:17 +0000869 if (type == Primitive::kPrimNot && kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
870 // UnsafeGetObject/UnsafeGetObjectVolatile with Baker's read barrier case.
Roland Levillain54f869e2017-03-06 13:54:11 +0000871 Register temp = WRegisterFrom(locations->GetTemp(0));
Roland Levillainbfea3352016-06-23 13:48:47 +0100872 codegen->GenerateReferenceLoadWithBakerReadBarrier(invoke,
873 trg_loc,
874 base,
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100875 /* offset */ 0u,
Roland Levillainbfea3352016-06-23 13:48:47 +0100876 /* index */ offset_loc,
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100877 /* scale_factor */ 0u,
Roland Levillainbfea3352016-06-23 13:48:47 +0100878 temp,
879 /* needs_null_check */ false,
880 is_volatile);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800881 } else {
Roland Levillain44015862016-01-22 11:47:17 +0000882 // Other cases.
883 MemOperand mem_op(base.X(), offset);
884 if (is_volatile) {
Serban Constantinescu4a6a67c2016-01-27 09:19:56 +0000885 codegen->LoadAcquire(invoke, trg, mem_op, /* needs_null_check */ true);
Roland Levillain44015862016-01-22 11:47:17 +0000886 } else {
887 codegen->Load(type, trg, mem_op);
888 }
Roland Levillain4d027112015-07-01 15:41:14 +0100889
Roland Levillain44015862016-01-22 11:47:17 +0000890 if (type == Primitive::kPrimNot) {
891 DCHECK(trg.IsW());
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100892 codegen->MaybeGenerateReadBarrierSlow(invoke, trg_loc, trg_loc, base_loc, 0u, offset_loc);
Roland Levillain44015862016-01-22 11:47:17 +0000893 }
Roland Levillain4d027112015-07-01 15:41:14 +0100894 }
Andreas Gampe878d58c2015-01-15 23:24:00 -0800895}
896
897static void CreateIntIntIntToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
Roland Levillain22ccc3a2015-11-24 13:10:05 +0000898 bool can_call = kEmitCompilerReadBarrier &&
899 (invoke->GetIntrinsic() == Intrinsics::kUnsafeGetObject ||
900 invoke->GetIntrinsic() == Intrinsics::kUnsafeGetObjectVolatile);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800901 LocationSummary* locations = new (arena) LocationSummary(invoke,
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100902 (can_call
903 ? LocationSummary::kCallOnSlowPath
904 : LocationSummary::kNoCall),
Andreas Gampe878d58c2015-01-15 23:24:00 -0800905 kIntrinsified);
Vladimir Marko70e97462016-08-09 11:04:26 +0100906 if (can_call && kUseBakerReadBarrier) {
Vladimir Marko804b03f2016-09-14 16:26:36 +0100907 locations->SetCustomSlowPathCallerSaves(RegisterSet::Empty()); // No caller-save registers.
Roland Levillain54f869e2017-03-06 13:54:11 +0000908 // We need a temporary register for the read barrier marking slow
909 // path in CodeGeneratorARM64::GenerateReferenceLoadWithBakerReadBarrier.
910 locations->AddTemp(Location::RequiresRegister());
Vladimir Marko70e97462016-08-09 11:04:26 +0100911 }
Andreas Gampe878d58c2015-01-15 23:24:00 -0800912 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
913 locations->SetInAt(1, Location::RequiresRegister());
914 locations->SetInAt(2, Location::RequiresRegister());
Roland Levillainbfea3352016-06-23 13:48:47 +0100915 locations->SetOut(Location::RequiresRegister(),
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100916 (can_call ? Location::kOutputOverlap : Location::kNoOutputOverlap));
Andreas Gampe878d58c2015-01-15 23:24:00 -0800917}
918
919void IntrinsicLocationsBuilderARM64::VisitUnsafeGet(HInvoke* invoke) {
920 CreateIntIntIntToIntLocations(arena_, invoke);
921}
922void IntrinsicLocationsBuilderARM64::VisitUnsafeGetVolatile(HInvoke* invoke) {
923 CreateIntIntIntToIntLocations(arena_, invoke);
924}
925void IntrinsicLocationsBuilderARM64::VisitUnsafeGetLong(HInvoke* invoke) {
926 CreateIntIntIntToIntLocations(arena_, invoke);
927}
928void IntrinsicLocationsBuilderARM64::VisitUnsafeGetLongVolatile(HInvoke* invoke) {
929 CreateIntIntIntToIntLocations(arena_, invoke);
930}
931void IntrinsicLocationsBuilderARM64::VisitUnsafeGetObject(HInvoke* invoke) {
932 CreateIntIntIntToIntLocations(arena_, invoke);
933}
934void IntrinsicLocationsBuilderARM64::VisitUnsafeGetObjectVolatile(HInvoke* invoke) {
935 CreateIntIntIntToIntLocations(arena_, invoke);
936}
937
938void IntrinsicCodeGeneratorARM64::VisitUnsafeGet(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000939 GenUnsafeGet(invoke, Primitive::kPrimInt, /* is_volatile */ false, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800940}
941void IntrinsicCodeGeneratorARM64::VisitUnsafeGetVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000942 GenUnsafeGet(invoke, Primitive::kPrimInt, /* is_volatile */ true, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800943}
944void IntrinsicCodeGeneratorARM64::VisitUnsafeGetLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000945 GenUnsafeGet(invoke, Primitive::kPrimLong, /* is_volatile */ false, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800946}
947void IntrinsicCodeGeneratorARM64::VisitUnsafeGetLongVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000948 GenUnsafeGet(invoke, Primitive::kPrimLong, /* is_volatile */ true, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800949}
950void IntrinsicCodeGeneratorARM64::VisitUnsafeGetObject(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000951 GenUnsafeGet(invoke, Primitive::kPrimNot, /* is_volatile */ false, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800952}
953void IntrinsicCodeGeneratorARM64::VisitUnsafeGetObjectVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000954 GenUnsafeGet(invoke, Primitive::kPrimNot, /* is_volatile */ true, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800955}
956
957static void CreateIntIntIntIntToVoid(ArenaAllocator* arena, HInvoke* invoke) {
958 LocationSummary* locations = new (arena) LocationSummary(invoke,
959 LocationSummary::kNoCall,
960 kIntrinsified);
961 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
962 locations->SetInAt(1, Location::RequiresRegister());
963 locations->SetInAt(2, Location::RequiresRegister());
964 locations->SetInAt(3, Location::RequiresRegister());
965}
966
967void IntrinsicLocationsBuilderARM64::VisitUnsafePut(HInvoke* invoke) {
968 CreateIntIntIntIntToVoid(arena_, invoke);
969}
970void IntrinsicLocationsBuilderARM64::VisitUnsafePutOrdered(HInvoke* invoke) {
971 CreateIntIntIntIntToVoid(arena_, invoke);
972}
973void IntrinsicLocationsBuilderARM64::VisitUnsafePutVolatile(HInvoke* invoke) {
974 CreateIntIntIntIntToVoid(arena_, invoke);
975}
976void IntrinsicLocationsBuilderARM64::VisitUnsafePutObject(HInvoke* invoke) {
977 CreateIntIntIntIntToVoid(arena_, invoke);
978}
979void IntrinsicLocationsBuilderARM64::VisitUnsafePutObjectOrdered(HInvoke* invoke) {
980 CreateIntIntIntIntToVoid(arena_, invoke);
981}
982void IntrinsicLocationsBuilderARM64::VisitUnsafePutObjectVolatile(HInvoke* invoke) {
983 CreateIntIntIntIntToVoid(arena_, invoke);
984}
985void IntrinsicLocationsBuilderARM64::VisitUnsafePutLong(HInvoke* invoke) {
986 CreateIntIntIntIntToVoid(arena_, invoke);
987}
988void IntrinsicLocationsBuilderARM64::VisitUnsafePutLongOrdered(HInvoke* invoke) {
989 CreateIntIntIntIntToVoid(arena_, invoke);
990}
991void IntrinsicLocationsBuilderARM64::VisitUnsafePutLongVolatile(HInvoke* invoke) {
992 CreateIntIntIntIntToVoid(arena_, invoke);
993}
994
Artem Serov914d7a82017-02-07 14:33:49 +0000995static void GenUnsafePut(HInvoke* invoke,
Andreas Gampe878d58c2015-01-15 23:24:00 -0800996 Primitive::Type type,
997 bool is_volatile,
998 bool is_ordered,
999 CodeGeneratorARM64* codegen) {
Artem Serov914d7a82017-02-07 14:33:49 +00001000 LocationSummary* locations = invoke->GetLocations();
Alexandre Rames087930f2016-08-02 13:45:28 +01001001 MacroAssembler* masm = codegen->GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -08001002
1003 Register base = WRegisterFrom(locations->InAt(1)); // Object pointer.
1004 Register offset = XRegisterFrom(locations->InAt(2)); // Long offset.
1005 Register value = RegisterFrom(locations->InAt(3), type);
Roland Levillain4d027112015-07-01 15:41:14 +01001006 Register source = value;
Andreas Gampe878d58c2015-01-15 23:24:00 -08001007 MemOperand mem_op(base.X(), offset);
1008
Roland Levillain4d027112015-07-01 15:41:14 +01001009 {
1010 // We use a block to end the scratch scope before the write barrier, thus
1011 // freeing the temporary registers so they can be used in `MarkGCCard`.
1012 UseScratchRegisterScope temps(masm);
1013
1014 if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
1015 DCHECK(value.IsW());
1016 Register temp = temps.AcquireW();
1017 __ Mov(temp.W(), value.W());
1018 codegen->GetAssembler()->PoisonHeapReference(temp.W());
1019 source = temp;
Andreas Gampe878d58c2015-01-15 23:24:00 -08001020 }
Roland Levillain4d027112015-07-01 15:41:14 +01001021
1022 if (is_volatile || is_ordered) {
Artem Serov914d7a82017-02-07 14:33:49 +00001023 codegen->StoreRelease(invoke, type, source, mem_op, /* needs_null_check */ false);
Roland Levillain4d027112015-07-01 15:41:14 +01001024 } else {
1025 codegen->Store(type, source, mem_op);
1026 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001027 }
1028
1029 if (type == Primitive::kPrimNot) {
Nicolas Geoffray07276db2015-05-18 14:22:09 +01001030 bool value_can_be_null = true; // TODO: Worth finding out this information?
1031 codegen->MarkGCCard(base, value, value_can_be_null);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001032 }
1033}
1034
1035void IntrinsicCodeGeneratorARM64::VisitUnsafePut(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001036 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001037 Primitive::kPrimInt,
1038 /* is_volatile */ false,
1039 /* is_ordered */ false,
1040 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001041}
1042void IntrinsicCodeGeneratorARM64::VisitUnsafePutOrdered(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001043 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001044 Primitive::kPrimInt,
1045 /* is_volatile */ false,
1046 /* is_ordered */ true,
1047 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001048}
1049void IntrinsicCodeGeneratorARM64::VisitUnsafePutVolatile(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001050 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001051 Primitive::kPrimInt,
1052 /* is_volatile */ true,
1053 /* is_ordered */ false,
1054 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001055}
1056void IntrinsicCodeGeneratorARM64::VisitUnsafePutObject(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001057 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001058 Primitive::kPrimNot,
1059 /* is_volatile */ false,
1060 /* is_ordered */ false,
1061 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001062}
1063void IntrinsicCodeGeneratorARM64::VisitUnsafePutObjectOrdered(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001064 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001065 Primitive::kPrimNot,
1066 /* is_volatile */ false,
1067 /* is_ordered */ true,
1068 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001069}
1070void IntrinsicCodeGeneratorARM64::VisitUnsafePutObjectVolatile(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001071 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001072 Primitive::kPrimNot,
1073 /* is_volatile */ true,
1074 /* is_ordered */ false,
1075 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001076}
1077void IntrinsicCodeGeneratorARM64::VisitUnsafePutLong(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001078 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001079 Primitive::kPrimLong,
1080 /* is_volatile */ false,
1081 /* is_ordered */ false,
1082 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001083}
1084void IntrinsicCodeGeneratorARM64::VisitUnsafePutLongOrdered(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001085 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001086 Primitive::kPrimLong,
1087 /* is_volatile */ false,
1088 /* is_ordered */ true,
1089 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001090}
1091void IntrinsicCodeGeneratorARM64::VisitUnsafePutLongVolatile(HInvoke* invoke) {
Artem Serov914d7a82017-02-07 14:33:49 +00001092 GenUnsafePut(invoke,
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001093 Primitive::kPrimLong,
1094 /* is_volatile */ true,
1095 /* is_ordered */ false,
1096 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001097}
1098
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001099static void CreateIntIntIntIntIntToInt(ArenaAllocator* arena,
1100 HInvoke* invoke,
1101 Primitive::Type type) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001102 bool can_call = kEmitCompilerReadBarrier &&
1103 kUseBakerReadBarrier &&
1104 (invoke->GetIntrinsic() == Intrinsics::kUnsafeCASObject);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001105 LocationSummary* locations = new (arena) LocationSummary(invoke,
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001106 (can_call
1107 ? LocationSummary::kCallOnSlowPath
1108 : LocationSummary::kNoCall),
Andreas Gampe878d58c2015-01-15 23:24:00 -08001109 kIntrinsified);
1110 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
1111 locations->SetInAt(1, Location::RequiresRegister());
1112 locations->SetInAt(2, Location::RequiresRegister());
1113 locations->SetInAt(3, Location::RequiresRegister());
1114 locations->SetInAt(4, Location::RequiresRegister());
1115
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001116 // If heap poisoning is enabled, we don't want the unpoisoning
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001117 // operations to potentially clobber the output. Likewise when
1118 // emitting a (Baker) read barrier, which may call.
1119 Location::OutputOverlap overlaps =
1120 ((kPoisonHeapReferences && type == Primitive::kPrimNot) || can_call)
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001121 ? Location::kOutputOverlap
1122 : Location::kNoOutputOverlap;
1123 locations->SetOut(Location::RequiresRegister(), overlaps);
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001124 if (type == Primitive::kPrimNot && kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
1125 // Temporary register for (Baker) read barrier.
1126 locations->AddTemp(Location::RequiresRegister());
1127 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001128}
1129
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001130static void GenCas(HInvoke* invoke, Primitive::Type type, CodeGeneratorARM64* codegen) {
Alexandre Rames087930f2016-08-02 13:45:28 +01001131 MacroAssembler* masm = codegen->GetVIXLAssembler();
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001132 LocationSummary* locations = invoke->GetLocations();
Andreas Gampe878d58c2015-01-15 23:24:00 -08001133
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001134 Location out_loc = locations->Out();
1135 Register out = WRegisterFrom(out_loc); // Boolean result.
Andreas Gampe878d58c2015-01-15 23:24:00 -08001136
1137 Register base = WRegisterFrom(locations->InAt(1)); // Object pointer.
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001138 Location offset_loc = locations->InAt(2);
1139 Register offset = XRegisterFrom(offset_loc); // Long offset.
Andreas Gampe878d58c2015-01-15 23:24:00 -08001140 Register expected = RegisterFrom(locations->InAt(3), type); // Expected.
1141 Register value = RegisterFrom(locations->InAt(4), type); // Value.
1142
1143 // This needs to be before the temp registers, as MarkGCCard also uses VIXL temps.
1144 if (type == Primitive::kPrimNot) {
1145 // Mark card for object assuming new value is stored.
Nicolas Geoffray07276db2015-05-18 14:22:09 +01001146 bool value_can_be_null = true; // TODO: Worth finding out this information?
1147 codegen->MarkGCCard(base, value, value_can_be_null);
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001148
1149 // The only read barrier implementation supporting the
1150 // UnsafeCASObject intrinsic is the Baker-style read barriers.
1151 DCHECK(!kEmitCompilerReadBarrier || kUseBakerReadBarrier);
1152
1153 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
1154 Register temp = WRegisterFrom(locations->GetTemp(0));
1155 // Need to make sure the reference stored in the field is a to-space
1156 // one before attempting the CAS or the CAS could fail incorrectly.
1157 codegen->GenerateReferenceLoadWithBakerReadBarrier(
1158 invoke,
1159 out_loc, // Unused, used only as a "temporary" within the read barrier.
1160 base,
1161 /* offset */ 0u,
1162 /* index */ offset_loc,
1163 /* scale_factor */ 0u,
1164 temp,
1165 /* needs_null_check */ false,
1166 /* use_load_acquire */ false,
1167 /* always_update_field */ true);
1168 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001169 }
1170
1171 UseScratchRegisterScope temps(masm);
1172 Register tmp_ptr = temps.AcquireX(); // Pointer to actual memory.
1173 Register tmp_value = temps.AcquireSameSizeAs(value); // Value in memory.
1174
1175 Register tmp_32 = tmp_value.W();
1176
1177 __ Add(tmp_ptr, base.X(), Operand(offset));
1178
Roland Levillain4d027112015-07-01 15:41:14 +01001179 if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
1180 codegen->GetAssembler()->PoisonHeapReference(expected);
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001181 if (value.Is(expected)) {
1182 // Do not poison `value`, as it is the same register as
1183 // `expected`, which has just been poisoned.
1184 } else {
1185 codegen->GetAssembler()->PoisonHeapReference(value);
1186 }
Roland Levillain4d027112015-07-01 15:41:14 +01001187 }
1188
Andreas Gampe878d58c2015-01-15 23:24:00 -08001189 // do {
1190 // tmp_value = [tmp_ptr] - expected;
1191 // } while (tmp_value == 0 && failure([tmp_ptr] <- r_new_value));
1192 // result = tmp_value != 0;
1193
Scott Wakeling97c72b72016-06-24 16:19:36 +01001194 vixl::aarch64::Label loop_head, exit_loop;
Serban Constantinescu4a6a67c2016-01-27 09:19:56 +00001195 __ Bind(&loop_head);
Serban Constantinescu4a6a67c2016-01-27 09:19:56 +00001196 __ Ldaxr(tmp_value, MemOperand(tmp_ptr));
1197 __ Cmp(tmp_value, expected);
1198 __ B(&exit_loop, ne);
1199 __ Stlxr(tmp_32, value, MemOperand(tmp_ptr));
1200 __ Cbnz(tmp_32, &loop_head);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001201 __ Bind(&exit_loop);
1202 __ Cset(out, eq);
Roland Levillain4d027112015-07-01 15:41:14 +01001203
1204 if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
Roland Levillain4d027112015-07-01 15:41:14 +01001205 codegen->GetAssembler()->UnpoisonHeapReference(expected);
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001206 if (value.Is(expected)) {
1207 // Do not unpoison `value`, as it is the same register as
1208 // `expected`, which has just been unpoisoned.
1209 } else {
1210 codegen->GetAssembler()->UnpoisonHeapReference(value);
1211 }
Roland Levillain4d027112015-07-01 15:41:14 +01001212 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001213}
1214
1215void IntrinsicLocationsBuilderARM64::VisitUnsafeCASInt(HInvoke* invoke) {
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001216 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimInt);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001217}
1218void IntrinsicLocationsBuilderARM64::VisitUnsafeCASLong(HInvoke* invoke) {
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001219 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimLong);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001220}
1221void IntrinsicLocationsBuilderARM64::VisitUnsafeCASObject(HInvoke* invoke) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001222 // The only read barrier implementation supporting the
1223 // UnsafeCASObject intrinsic is the Baker-style read barriers.
1224 if (kEmitCompilerReadBarrier && !kUseBakerReadBarrier) {
Roland Levillain985ff702015-10-23 13:25:35 +01001225 return;
1226 }
1227
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001228 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimNot);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001229}
1230
1231void IntrinsicCodeGeneratorARM64::VisitUnsafeCASInt(HInvoke* invoke) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001232 GenCas(invoke, Primitive::kPrimInt, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001233}
1234void IntrinsicCodeGeneratorARM64::VisitUnsafeCASLong(HInvoke* invoke) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001235 GenCas(invoke, Primitive::kPrimLong, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001236}
1237void IntrinsicCodeGeneratorARM64::VisitUnsafeCASObject(HInvoke* invoke) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001238 // The only read barrier implementation supporting the
1239 // UnsafeCASObject intrinsic is the Baker-style read barriers.
1240 DCHECK(!kEmitCompilerReadBarrier || kUseBakerReadBarrier);
Roland Levillain3d312422016-06-23 13:53:42 +01001241
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001242 GenCas(invoke, Primitive::kPrimNot, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001243}
1244
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001245void IntrinsicLocationsBuilderARM64::VisitStringCompareTo(HInvoke* invoke) {
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001246 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Scott Wakeling1f36f412016-04-21 11:13:45 +01001247 invoke->InputAt(1)->CanBeNull()
1248 ? LocationSummary::kCallOnSlowPath
1249 : LocationSummary::kNoCall,
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001250 kIntrinsified);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001251 locations->SetInAt(0, Location::RequiresRegister());
1252 locations->SetInAt(1, Location::RequiresRegister());
1253 locations->AddTemp(Location::RequiresRegister());
1254 locations->AddTemp(Location::RequiresRegister());
1255 locations->AddTemp(Location::RequiresRegister());
jessicahandojo05765752016-09-09 19:01:32 -07001256 // Need temporary registers for String compression's feature.
1257 if (mirror::kUseStringCompression) {
1258 locations->AddTemp(Location::RequiresRegister());
jessicahandojo05765752016-09-09 19:01:32 -07001259 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001260 locations->SetOut(Location::RequiresRegister(), Location::kOutputOverlap);
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001261}
1262
1263void IntrinsicCodeGeneratorARM64::VisitStringCompareTo(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001264 MacroAssembler* masm = GetVIXLAssembler();
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001265 LocationSummary* locations = invoke->GetLocations();
1266
Alexandre Rames2ea91532016-08-11 17:04:14 +01001267 Register str = InputRegisterAt(invoke, 0);
1268 Register arg = InputRegisterAt(invoke, 1);
1269 DCHECK(str.IsW());
1270 DCHECK(arg.IsW());
Scott Wakeling1f36f412016-04-21 11:13:45 +01001271 Register out = OutputRegister(invoke);
1272
1273 Register temp0 = WRegisterFrom(locations->GetTemp(0));
1274 Register temp1 = WRegisterFrom(locations->GetTemp(1));
1275 Register temp2 = WRegisterFrom(locations->GetTemp(2));
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001276 Register temp3;
jessicahandojo05765752016-09-09 19:01:32 -07001277 if (mirror::kUseStringCompression) {
1278 temp3 = WRegisterFrom(locations->GetTemp(3));
jessicahandojo05765752016-09-09 19:01:32 -07001279 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001280
Scott Wakeling97c72b72016-06-24 16:19:36 +01001281 vixl::aarch64::Label loop;
1282 vixl::aarch64::Label find_char_diff;
1283 vixl::aarch64::Label end;
jessicahandojo05765752016-09-09 19:01:32 -07001284 vixl::aarch64::Label different_compression;
Scott Wakeling1f36f412016-04-21 11:13:45 +01001285
1286 // Get offsets of count and value fields within a string object.
1287 const int32_t count_offset = mirror::String::CountOffset().Int32Value();
1288 const int32_t value_offset = mirror::String::ValueOffset().Int32Value();
1289
Nicolas Geoffray512e04d2015-03-27 17:21:24 +00001290 // Note that the null check must have been done earlier.
Calin Juravle641547a2015-04-21 22:08:51 +01001291 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001292
Scott Wakeling1f36f412016-04-21 11:13:45 +01001293 // Take slow path and throw if input can be and is null.
1294 SlowPathCodeARM64* slow_path = nullptr;
1295 const bool can_slow_path = invoke->InputAt(1)->CanBeNull();
1296 if (can_slow_path) {
1297 slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1298 codegen_->AddSlowPath(slow_path);
1299 __ Cbz(arg, slow_path->GetEntryLabel());
1300 }
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001301
Scott Wakeling1f36f412016-04-21 11:13:45 +01001302 // Reference equality check, return 0 if same reference.
1303 __ Subs(out, str, arg);
1304 __ B(&end, eq);
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001305
jessicahandojo05765752016-09-09 19:01:32 -07001306 if (mirror::kUseStringCompression) {
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001307 // Load `count` fields of this and argument strings.
jessicahandojo05765752016-09-09 19:01:32 -07001308 __ Ldr(temp3, HeapOperand(str, count_offset));
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001309 __ Ldr(temp2, HeapOperand(arg, count_offset));
jessicahandojo05765752016-09-09 19:01:32 -07001310 // Clean out compression flag from lengths.
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001311 __ Lsr(temp0, temp3, 1u);
1312 __ Lsr(temp1, temp2, 1u);
jessicahandojo05765752016-09-09 19:01:32 -07001313 } else {
1314 // Load lengths of this and argument strings.
1315 __ Ldr(temp0, HeapOperand(str, count_offset));
1316 __ Ldr(temp1, HeapOperand(arg, count_offset));
1317 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001318 // out = length diff.
1319 __ Subs(out, temp0, temp1);
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001320 // temp0 = min(len(str), len(arg)).
1321 __ Csel(temp0, temp1, temp0, ge);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001322 // Shorter string is empty?
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001323 __ Cbz(temp0, &end);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001324
jessicahandojo05765752016-09-09 19:01:32 -07001325 if (mirror::kUseStringCompression) {
1326 // Check if both strings using same compression style to use this comparison loop.
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001327 __ Eor(temp2, temp2, Operand(temp3));
1328 // Interleave with compression flag extraction which is needed for both paths
1329 // and also set flags which is needed only for the different compressions path.
1330 __ Ands(temp3.W(), temp3.W(), Operand(1));
1331 __ Tbnz(temp2, 0, &different_compression); // Does not use flags.
jessicahandojo05765752016-09-09 19:01:32 -07001332 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001333 // Store offset of string value in preparation for comparison loop.
1334 __ Mov(temp1, value_offset);
jessicahandojo05765752016-09-09 19:01:32 -07001335 if (mirror::kUseStringCompression) {
1336 // For string compression, calculate the number of bytes to compare (not chars).
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001337 // This could in theory exceed INT32_MAX, so treat temp0 as unsigned.
1338 __ Lsl(temp0, temp0, temp3);
jessicahandojo05765752016-09-09 19:01:32 -07001339 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001340
1341 UseScratchRegisterScope scratch_scope(masm);
1342 Register temp4 = scratch_scope.AcquireX();
1343
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001344 // Assertions that must hold in order to compare strings 8 bytes at a time.
Scott Wakeling1f36f412016-04-21 11:13:45 +01001345 DCHECK_ALIGNED(value_offset, 8);
1346 static_assert(IsAligned<8>(kObjectAlignment), "String of odd length is not zero padded");
1347
1348 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1349 DCHECK_EQ(char_size, 2u);
1350
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001351 // Promote temp2 to an X reg, ready for LDR.
1352 temp2 = temp2.X();
Scott Wakeling1f36f412016-04-21 11:13:45 +01001353
1354 // Loop to compare 4x16-bit characters at a time (ok because of string data alignment).
1355 __ Bind(&loop);
Alexandre Rames2ea91532016-08-11 17:04:14 +01001356 __ Ldr(temp4, MemOperand(str.X(), temp1.X()));
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001357 __ Ldr(temp2, MemOperand(arg.X(), temp1.X()));
1358 __ Cmp(temp4, temp2);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001359 __ B(ne, &find_char_diff);
1360 __ Add(temp1, temp1, char_size * 4);
jessicahandojo05765752016-09-09 19:01:32 -07001361 // With string compression, we have compared 8 bytes, otherwise 4 chars.
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001362 __ Subs(temp0, temp0, (mirror::kUseStringCompression) ? 8 : 4);
1363 __ B(&loop, hi);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001364 __ B(&end);
1365
1366 // Promote temp1 to an X reg, ready for EOR.
1367 temp1 = temp1.X();
1368
jessicahandojo05765752016-09-09 19:01:32 -07001369 // Find the single character difference.
Scott Wakeling1f36f412016-04-21 11:13:45 +01001370 __ Bind(&find_char_diff);
1371 // Get the bit position of the first character that differs.
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001372 __ Eor(temp1, temp2, temp4);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001373 __ Rbit(temp1, temp1);
1374 __ Clz(temp1, temp1);
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001375
jessicahandojo05765752016-09-09 19:01:32 -07001376 // If the number of chars remaining <= the index where the difference occurs (0-3), then
Scott Wakeling1f36f412016-04-21 11:13:45 +01001377 // the difference occurs outside the remaining string data, so just return length diff (out).
jessicahandojo05765752016-09-09 19:01:32 -07001378 // Unlike ARM, we're doing the comparison in one go here, without the subtraction at the
1379 // find_char_diff_2nd_cmp path, so it doesn't matter whether the comparison is signed or
1380 // unsigned when string compression is disabled.
1381 // When it's enabled, the comparison must be unsigned.
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001382 __ Cmp(temp0, Operand(temp1.W(), LSR, (mirror::kUseStringCompression) ? 3 : 4));
jessicahandojo05765752016-09-09 19:01:32 -07001383 __ B(ls, &end);
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001384
Scott Wakeling1f36f412016-04-21 11:13:45 +01001385 // Extract the characters and calculate the difference.
jessicahandojo05765752016-09-09 19:01:32 -07001386 if (mirror:: kUseStringCompression) {
jessicahandojo05765752016-09-09 19:01:32 -07001387 __ Bic(temp1, temp1, 0x7);
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001388 __ Bic(temp1, temp1, Operand(temp3.X(), LSL, 3u));
1389 } else {
1390 __ Bic(temp1, temp1, 0xf);
jessicahandojo05765752016-09-09 19:01:32 -07001391 }
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001392 __ Lsr(temp2, temp2, temp1);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001393 __ Lsr(temp4, temp4, temp1);
jessicahandojo05765752016-09-09 19:01:32 -07001394 if (mirror::kUseStringCompression) {
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001395 // Prioritize the case of compressed strings and calculate such result first.
1396 __ Uxtb(temp1, temp4);
1397 __ Sub(out, temp1.W(), Operand(temp2.W(), UXTB));
1398 __ Tbz(temp3, 0u, &end); // If actually compressed, we're done.
jessicahandojo05765752016-09-09 19:01:32 -07001399 }
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001400 __ Uxth(temp4, temp4);
1401 __ Sub(out, temp4.W(), Operand(temp2.W(), UXTH));
jessicahandojo05765752016-09-09 19:01:32 -07001402
1403 if (mirror::kUseStringCompression) {
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001404 __ B(&end);
1405 __ Bind(&different_compression);
1406
1407 // Comparison for different compression style.
jessicahandojo05765752016-09-09 19:01:32 -07001408 const size_t c_char_size = Primitive::ComponentSize(Primitive::kPrimByte);
1409 DCHECK_EQ(c_char_size, 1u);
jessicahandojo05765752016-09-09 19:01:32 -07001410 temp1 = temp1.W();
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001411 temp2 = temp2.W();
1412 temp4 = temp4.W();
jessicahandojo05765752016-09-09 19:01:32 -07001413
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001414 // `temp1` will hold the compressed data pointer, `temp2` the uncompressed data pointer.
1415 // Note that flags have been set by the `str` compression flag extraction to `temp3`
1416 // before branching to the `different_compression` label.
1417 __ Csel(temp1, str, arg, eq); // Pointer to the compressed string.
1418 __ Csel(temp2, str, arg, ne); // Pointer to the uncompressed string.
jessicahandojo05765752016-09-09 19:01:32 -07001419
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001420 // We want to free up the temp3, currently holding `str` compression flag, for comparison.
1421 // So, we move it to the bottom bit of the iteration count `temp0` which we then need to treat
1422 // as unsigned. Start by freeing the bit with a LSL and continue further down by a SUB which
1423 // will allow `subs temp0, #2; bhi different_compression_loop` to serve as the loop condition.
1424 __ Lsl(temp0, temp0, 1u);
1425
1426 // Adjust temp1 and temp2 from string pointers to data pointers.
1427 __ Add(temp1, temp1, Operand(value_offset));
1428 __ Add(temp2, temp2, Operand(value_offset));
1429
1430 // Complete the move of the compression flag.
1431 __ Sub(temp0, temp0, Operand(temp3));
1432
1433 vixl::aarch64::Label different_compression_loop;
1434 vixl::aarch64::Label different_compression_diff;
1435
1436 __ Bind(&different_compression_loop);
1437 __ Ldrb(temp4, MemOperand(temp1.X(), c_char_size, PostIndex));
1438 __ Ldrh(temp3, MemOperand(temp2.X(), char_size, PostIndex));
1439 __ Subs(temp4, temp4, Operand(temp3));
1440 __ B(&different_compression_diff, ne);
1441 __ Subs(temp0, temp0, 2);
1442 __ B(&different_compression_loop, hi);
jessicahandojo05765752016-09-09 19:01:32 -07001443 __ B(&end);
1444
1445 // Calculate the difference.
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001446 __ Bind(&different_compression_diff);
1447 __ Tst(temp0, Operand(1));
1448 static_assert(static_cast<uint32_t>(mirror::StringCompressionFlag::kCompressed) == 0u,
1449 "Expecting 0=compressed, 1=uncompressed");
1450 __ Cneg(out, temp4, ne);
jessicahandojo05765752016-09-09 19:01:32 -07001451 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001452
1453 __ Bind(&end);
1454
1455 if (can_slow_path) {
1456 __ Bind(slow_path->GetExitLabel());
1457 }
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001458}
1459
Vladimir Markoe39f14f2017-02-10 15:44:25 +00001460// The cut off for unrolling the loop in String.equals() intrinsic for const strings.
1461// The normal loop plus the pre-header is 9 instructions without string compression and 12
1462// instructions with string compression. We can compare up to 8 bytes in 4 instructions
1463// (LDR+LDR+CMP+BNE) and up to 16 bytes in 5 instructions (LDP+LDP+CMP+CCMP+BNE). Allow up
1464// to 10 instructions for the unrolled loop.
1465constexpr size_t kShortConstStringEqualsCutoffInBytes = 32;
1466
1467static const char* GetConstString(HInstruction* candidate, uint32_t* utf16_length) {
1468 if (candidate->IsLoadString()) {
1469 HLoadString* load_string = candidate->AsLoadString();
1470 const DexFile& dex_file = load_string->GetDexFile();
1471 return dex_file.StringDataAndUtf16LengthByIdx(load_string->GetStringIndex(), utf16_length);
1472 }
1473 return nullptr;
1474}
1475
Agi Csakiea34b402015-08-13 17:51:19 -07001476void IntrinsicLocationsBuilderARM64::VisitStringEquals(HInvoke* invoke) {
1477 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1478 LocationSummary::kNoCall,
1479 kIntrinsified);
1480 locations->SetInAt(0, Location::RequiresRegister());
1481 locations->SetInAt(1, Location::RequiresRegister());
Agi Csakiea34b402015-08-13 17:51:19 -07001482
Vladimir Markoe39f14f2017-02-10 15:44:25 +00001483 // For the generic implementation and for long const strings we need a temporary.
1484 // We do not need it for short const strings, up to 8 bytes, see code generation below.
1485 uint32_t const_string_length = 0u;
1486 const char* const_string = GetConstString(invoke->InputAt(0), &const_string_length);
1487 if (const_string == nullptr) {
1488 const_string = GetConstString(invoke->InputAt(1), &const_string_length);
1489 }
1490 bool is_compressed =
1491 mirror::kUseStringCompression &&
1492 const_string != nullptr &&
1493 mirror::String::DexFileStringAllASCII(const_string, const_string_length);
1494 if (const_string == nullptr || const_string_length > (is_compressed ? 8u : 4u)) {
1495 locations->AddTemp(Location::RequiresRegister());
1496 }
1497
1498 // TODO: If the String.equals() is used only for an immediately following HIf, we can
1499 // mark it as emitted-at-use-site and emit branches directly to the appropriate blocks.
1500 // Then we shall need an extra temporary register instead of the output register.
Agi Csakiea34b402015-08-13 17:51:19 -07001501 locations->SetOut(Location::RequiresRegister(), Location::kOutputOverlap);
1502}
1503
1504void IntrinsicCodeGeneratorARM64::VisitStringEquals(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001505 MacroAssembler* masm = GetVIXLAssembler();
Agi Csakiea34b402015-08-13 17:51:19 -07001506 LocationSummary* locations = invoke->GetLocations();
1507
1508 Register str = WRegisterFrom(locations->InAt(0));
1509 Register arg = WRegisterFrom(locations->InAt(1));
1510 Register out = XRegisterFrom(locations->Out());
1511
1512 UseScratchRegisterScope scratch_scope(masm);
1513 Register temp = scratch_scope.AcquireW();
Vladimir Markoe39f14f2017-02-10 15:44:25 +00001514 Register temp1 = scratch_scope.AcquireW();
Agi Csakiea34b402015-08-13 17:51:19 -07001515
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001516 vixl::aarch64::Label loop;
Scott Wakeling97c72b72016-06-24 16:19:36 +01001517 vixl::aarch64::Label end;
1518 vixl::aarch64::Label return_true;
1519 vixl::aarch64::Label return_false;
Agi Csakiea34b402015-08-13 17:51:19 -07001520
1521 // Get offsets of count, value, and class fields within a string object.
1522 const int32_t count_offset = mirror::String::CountOffset().Int32Value();
1523 const int32_t value_offset = mirror::String::ValueOffset().Int32Value();
1524 const int32_t class_offset = mirror::Object::ClassOffset().Int32Value();
1525
1526 // Note that the null check must have been done earlier.
1527 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1528
Vladimir Marko53b52002016-05-24 19:30:45 +01001529 StringEqualsOptimizations optimizations(invoke);
1530 if (!optimizations.GetArgumentNotNull()) {
1531 // Check if input is null, return false if it is.
1532 __ Cbz(arg, &return_false);
1533 }
Agi Csakiea34b402015-08-13 17:51:19 -07001534
1535 // Reference equality check, return true if same reference.
1536 __ Cmp(str, arg);
1537 __ B(&return_true, eq);
1538
Vladimir Marko53b52002016-05-24 19:30:45 +01001539 if (!optimizations.GetArgumentIsString()) {
1540 // Instanceof check for the argument by comparing class fields.
1541 // All string objects must have the same type since String cannot be subclassed.
1542 // Receiver must be a string object, so its class field is equal to all strings' class fields.
1543 // If the argument is a string object, its class field must be equal to receiver's class field.
1544 __ Ldr(temp, MemOperand(str.X(), class_offset));
1545 __ Ldr(temp1, MemOperand(arg.X(), class_offset));
1546 __ Cmp(temp, temp1);
1547 __ B(&return_false, ne);
1548 }
Agi Csakiea34b402015-08-13 17:51:19 -07001549
Vladimir Markoe39f14f2017-02-10 15:44:25 +00001550 // Check if one of the inputs is a const string. Do not special-case both strings
1551 // being const, such cases should be handled by constant folding if needed.
1552 uint32_t const_string_length = 0u;
1553 const char* const_string = GetConstString(invoke->InputAt(0), &const_string_length);
1554 if (const_string == nullptr) {
1555 const_string = GetConstString(invoke->InputAt(1), &const_string_length);
1556 if (const_string != nullptr) {
1557 std::swap(str, arg); // Make sure the const string is in `str`.
1558 }
1559 }
1560 bool is_compressed =
1561 mirror::kUseStringCompression &&
1562 const_string != nullptr &&
1563 mirror::String::DexFileStringAllASCII(const_string, const_string_length);
1564
1565 if (const_string != nullptr) {
1566 // Load `count` field of the argument string and check if it matches the const string.
1567 // Also compares the compression style, if differs return false.
1568 __ Ldr(temp, MemOperand(arg.X(), count_offset));
Vladimir Marko26ec3ca2017-03-14 13:37:14 +00001569 // Temporarily release temp1 as we may not be able to embed the flagged count in CMP immediate.
1570 scratch_scope.Release(temp1);
Vladimir Markoe39f14f2017-02-10 15:44:25 +00001571 __ Cmp(temp, Operand(mirror::String::GetFlaggedCount(const_string_length, is_compressed)));
Vladimir Marko26ec3ca2017-03-14 13:37:14 +00001572 temp1 = scratch_scope.AcquireW();
Vladimir Markoe39f14f2017-02-10 15:44:25 +00001573 __ B(&return_false, ne);
1574 } else {
1575 // Load `count` fields of this and argument strings.
1576 __ Ldr(temp, MemOperand(str.X(), count_offset));
1577 __ Ldr(temp1, MemOperand(arg.X(), count_offset));
1578 // Check if `count` fields are equal, return false if they're not.
1579 // Also compares the compression style, if differs return false.
1580 __ Cmp(temp, temp1);
1581 __ B(&return_false, ne);
1582 }
Agi Csakiea34b402015-08-13 17:51:19 -07001583
Vladimir Markofdaf0f42016-10-13 19:29:53 +01001584 // Assertions that must hold in order to compare strings 8 bytes at a time.
Agi Csakiea34b402015-08-13 17:51:19 -07001585 DCHECK_ALIGNED(value_offset, 8);
1586 static_assert(IsAligned<8>(kObjectAlignment), "String of odd length is not zero padded");
1587
Vladimir Markoe39f14f2017-02-10 15:44:25 +00001588 if (const_string != nullptr &&
1589 const_string_length < (is_compressed ? kShortConstStringEqualsCutoffInBytes
1590 : kShortConstStringEqualsCutoffInBytes / 2u)) {
1591 // Load and compare the contents. Though we know the contents of the short const string
1592 // at compile time, materializing constants may be more code than loading from memory.
1593 int32_t offset = value_offset;
1594 size_t remaining_bytes =
1595 RoundUp(is_compressed ? const_string_length : const_string_length * 2u, 8u);
1596 temp = temp.X();
1597 temp1 = temp1.X();
1598 while (remaining_bytes > 8u) {
1599 Register temp2 = XRegisterFrom(locations->GetTemp(0));
1600 __ Ldp(temp, temp1, MemOperand(str.X(), offset));
1601 __ Ldp(temp2, out, MemOperand(arg.X(), offset));
1602 __ Cmp(temp, temp2);
1603 __ Ccmp(temp1, out, NoFlag, eq);
1604 __ B(&return_false, ne);
1605 offset += 2u * sizeof(uint64_t);
1606 remaining_bytes -= 2u * sizeof(uint64_t);
1607 }
1608 if (remaining_bytes != 0u) {
1609 __ Ldr(temp, MemOperand(str.X(), offset));
1610 __ Ldr(temp1, MemOperand(arg.X(), offset));
1611 __ Cmp(temp, temp1);
1612 __ B(&return_false, ne);
1613 }
1614 } else {
1615 // Return true if both strings are empty. Even with string compression `count == 0` means empty.
1616 static_assert(static_cast<uint32_t>(mirror::StringCompressionFlag::kCompressed) == 0u,
1617 "Expecting 0=compressed, 1=uncompressed");
1618 __ Cbz(temp, &return_true);
1619
1620 if (mirror::kUseStringCompression) {
1621 // For string compression, calculate the number of bytes to compare (not chars).
1622 // This could in theory exceed INT32_MAX, so treat temp as unsigned.
1623 __ And(temp1, temp, Operand(1)); // Extract compression flag.
1624 __ Lsr(temp, temp, 1u); // Extract length.
1625 __ Lsl(temp, temp, temp1); // Calculate number of bytes to compare.
1626 }
1627
1628 // Store offset of string value in preparation for comparison loop
1629 __ Mov(temp1, value_offset);
1630
1631 temp1 = temp1.X();
1632 Register temp2 = XRegisterFrom(locations->GetTemp(0));
1633 // Loop to compare strings 8 bytes at a time starting at the front of the string.
1634 // Ok to do this because strings are zero-padded to kObjectAlignment.
1635 __ Bind(&loop);
1636 __ Ldr(out, MemOperand(str.X(), temp1));
1637 __ Ldr(temp2, MemOperand(arg.X(), temp1));
1638 __ Add(temp1, temp1, Operand(sizeof(uint64_t)));
1639 __ Cmp(out, temp2);
1640 __ B(&return_false, ne);
1641 // With string compression, we have compared 8 bytes, otherwise 4 chars.
1642 __ Sub(temp, temp, Operand(mirror::kUseStringCompression ? 8 : 4), SetFlags);
1643 __ B(&loop, hi);
jessicahandojo05765752016-09-09 19:01:32 -07001644 }
1645
Agi Csakiea34b402015-08-13 17:51:19 -07001646 // Return true and exit the function.
1647 // If loop does not result in returning false, we return true.
1648 __ Bind(&return_true);
1649 __ Mov(out, 1);
1650 __ B(&end);
1651
1652 // Return false and exit the function.
1653 __ Bind(&return_false);
1654 __ Mov(out, 0);
1655 __ Bind(&end);
1656}
1657
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001658static void GenerateVisitStringIndexOf(HInvoke* invoke,
Scott Wakeling97c72b72016-06-24 16:19:36 +01001659 MacroAssembler* masm,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001660 CodeGeneratorARM64* codegen,
1661 ArenaAllocator* allocator,
1662 bool start_at_zero) {
1663 LocationSummary* locations = invoke->GetLocations();
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001664
1665 // Note that the null check must have been done earlier.
1666 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1667
1668 // Check for code points > 0xFFFF. Either a slow-path check when we don't know statically,
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001669 // or directly dispatch for a large constant, or omit slow-path for a small constant or a char.
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001670 SlowPathCodeARM64* slow_path = nullptr;
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001671 HInstruction* code_point = invoke->InputAt(1);
1672 if (code_point->IsIntConstant()) {
Vladimir Markoda051082016-05-17 16:10:20 +01001673 if (static_cast<uint32_t>(code_point->AsIntConstant()->GetValue()) > 0xFFFFU) {
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001674 // Always needs the slow-path. We could directly dispatch to it, but this case should be
1675 // rare, so for simplicity just put the full slow-path down and branch unconditionally.
1676 slow_path = new (allocator) IntrinsicSlowPathARM64(invoke);
1677 codegen->AddSlowPath(slow_path);
1678 __ B(slow_path->GetEntryLabel());
1679 __ Bind(slow_path->GetExitLabel());
1680 return;
1681 }
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001682 } else if (code_point->GetType() != Primitive::kPrimChar) {
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001683 Register char_reg = WRegisterFrom(locations->InAt(1));
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001684 __ Tst(char_reg, 0xFFFF0000);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001685 slow_path = new (allocator) IntrinsicSlowPathARM64(invoke);
1686 codegen->AddSlowPath(slow_path);
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001687 __ B(ne, slow_path->GetEntryLabel());
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001688 }
1689
1690 if (start_at_zero) {
1691 // Start-index = 0.
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001692 Register tmp_reg = WRegisterFrom(locations->GetTemp(0));
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001693 __ Mov(tmp_reg, 0);
1694 }
1695
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001696 codegen->InvokeRuntime(kQuickIndexOf, invoke, invoke->GetDexPc(), slow_path);
Roland Levillain42ad2882016-02-29 18:26:54 +00001697 CheckEntrypointTypes<kQuickIndexOf, int32_t, void*, uint32_t, uint32_t>();
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001698
1699 if (slow_path != nullptr) {
1700 __ Bind(slow_path->GetExitLabel());
1701 }
1702}
1703
1704void IntrinsicLocationsBuilderARM64::VisitStringIndexOf(HInvoke* invoke) {
1705 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001706 LocationSummary::kCallOnMainAndSlowPath,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001707 kIntrinsified);
1708 // We have a hand-crafted assembly stub that follows the runtime calling convention. So it's
1709 // best to align the inputs accordingly.
1710 InvokeRuntimeCallingConvention calling_convention;
1711 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1712 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1713 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimInt));
1714
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001715 // Need to send start_index=0.
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001716 locations->AddTemp(LocationFrom(calling_convention.GetRegisterAt(2)));
1717}
1718
1719void IntrinsicCodeGeneratorARM64::VisitStringIndexOf(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001720 GenerateVisitStringIndexOf(
1721 invoke, GetVIXLAssembler(), codegen_, GetAllocator(), /* start_at_zero */ true);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001722}
1723
1724void IntrinsicLocationsBuilderARM64::VisitStringIndexOfAfter(HInvoke* invoke) {
1725 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001726 LocationSummary::kCallOnMainAndSlowPath,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001727 kIntrinsified);
1728 // We have a hand-crafted assembly stub that follows the runtime calling convention. So it's
1729 // best to align the inputs accordingly.
1730 InvokeRuntimeCallingConvention calling_convention;
1731 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1732 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1733 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1734 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimInt));
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001735}
1736
1737void IntrinsicCodeGeneratorARM64::VisitStringIndexOfAfter(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001738 GenerateVisitStringIndexOf(
1739 invoke, GetVIXLAssembler(), codegen_, GetAllocator(), /* start_at_zero */ false);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001740}
1741
Jeff Hao848f70a2014-01-15 13:49:50 -08001742void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromBytes(HInvoke* invoke) {
1743 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001744 LocationSummary::kCallOnMainAndSlowPath,
Jeff Hao848f70a2014-01-15 13:49:50 -08001745 kIntrinsified);
1746 InvokeRuntimeCallingConvention calling_convention;
1747 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1748 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1749 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1750 locations->SetInAt(3, LocationFrom(calling_convention.GetRegisterAt(3)));
1751 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1752}
1753
1754void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromBytes(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001755 MacroAssembler* masm = GetVIXLAssembler();
Jeff Hao848f70a2014-01-15 13:49:50 -08001756 LocationSummary* locations = invoke->GetLocations();
1757
1758 Register byte_array = WRegisterFrom(locations->InAt(0));
1759 __ Cmp(byte_array, 0);
1760 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1761 codegen_->AddSlowPath(slow_path);
1762 __ B(eq, slow_path->GetEntryLabel());
1763
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001764 codegen_->InvokeRuntime(kQuickAllocStringFromBytes, invoke, invoke->GetDexPc(), slow_path);
Roland Levillainf969a202016-03-09 16:14:00 +00001765 CheckEntrypointTypes<kQuickAllocStringFromBytes, void*, void*, int32_t, int32_t, int32_t>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001766 __ Bind(slow_path->GetExitLabel());
1767}
1768
1769void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromChars(HInvoke* invoke) {
1770 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001771 LocationSummary::kCallOnMainOnly,
Jeff Hao848f70a2014-01-15 13:49:50 -08001772 kIntrinsified);
1773 InvokeRuntimeCallingConvention calling_convention;
1774 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1775 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1776 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1777 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1778}
1779
1780void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromChars(HInvoke* invoke) {
Roland Levillaincc3839c2016-02-29 16:23:48 +00001781 // No need to emit code checking whether `locations->InAt(2)` is a null
1782 // pointer, as callers of the native method
1783 //
1784 // java.lang.StringFactory.newStringFromChars(int offset, int charCount, char[] data)
1785 //
1786 // all include a null check on `data` before calling that method.
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001787 codegen_->InvokeRuntime(kQuickAllocStringFromChars, invoke, invoke->GetDexPc());
Roland Levillainf969a202016-03-09 16:14:00 +00001788 CheckEntrypointTypes<kQuickAllocStringFromChars, void*, int32_t, int32_t, void*>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001789}
1790
1791void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromString(HInvoke* invoke) {
Jeff Hao848f70a2014-01-15 13:49:50 -08001792 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001793 LocationSummary::kCallOnMainAndSlowPath,
Jeff Hao848f70a2014-01-15 13:49:50 -08001794 kIntrinsified);
1795 InvokeRuntimeCallingConvention calling_convention;
1796 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
Jeff Hao848f70a2014-01-15 13:49:50 -08001797 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1798}
1799
1800void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromString(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001801 MacroAssembler* masm = GetVIXLAssembler();
Jeff Hao848f70a2014-01-15 13:49:50 -08001802 LocationSummary* locations = invoke->GetLocations();
1803
1804 Register string_to_copy = WRegisterFrom(locations->InAt(0));
1805 __ Cmp(string_to_copy, 0);
1806 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1807 codegen_->AddSlowPath(slow_path);
1808 __ B(eq, slow_path->GetEntryLabel());
1809
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001810 codegen_->InvokeRuntime(kQuickAllocStringFromString, invoke, invoke->GetDexPc(), slow_path);
Roland Levillainf969a202016-03-09 16:14:00 +00001811 CheckEntrypointTypes<kQuickAllocStringFromString, void*, void*>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001812 __ Bind(slow_path->GetExitLabel());
1813}
1814
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001815static void CreateFPToFPCallLocations(ArenaAllocator* arena, HInvoke* invoke) {
1816 DCHECK_EQ(invoke->GetNumberOfArguments(), 1U);
1817 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(0)->GetType()));
1818 DCHECK(Primitive::IsFloatingPointType(invoke->GetType()));
1819
1820 LocationSummary* const locations = new (arena) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001821 LocationSummary::kCallOnMainOnly,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001822 kIntrinsified);
1823 InvokeRuntimeCallingConvention calling_convention;
1824
1825 locations->SetInAt(0, LocationFrom(calling_convention.GetFpuRegisterAt(0)));
1826 locations->SetOut(calling_convention.GetReturnLocation(invoke->GetType()));
1827}
1828
1829static void CreateFPFPToFPCallLocations(ArenaAllocator* arena, HInvoke* invoke) {
1830 DCHECK_EQ(invoke->GetNumberOfArguments(), 2U);
1831 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(0)->GetType()));
1832 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(1)->GetType()));
1833 DCHECK(Primitive::IsFloatingPointType(invoke->GetType()));
1834
1835 LocationSummary* const locations = new (arena) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001836 LocationSummary::kCallOnMainOnly,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001837 kIntrinsified);
1838 InvokeRuntimeCallingConvention calling_convention;
1839
1840 locations->SetInAt(0, LocationFrom(calling_convention.GetFpuRegisterAt(0)));
1841 locations->SetInAt(1, LocationFrom(calling_convention.GetFpuRegisterAt(1)));
1842 locations->SetOut(calling_convention.GetReturnLocation(invoke->GetType()));
1843}
1844
1845static void GenFPToFPCall(HInvoke* invoke,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001846 CodeGeneratorARM64* codegen,
1847 QuickEntrypointEnum entry) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001848 codegen->InvokeRuntime(entry, invoke, invoke->GetDexPc());
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001849}
1850
1851void IntrinsicLocationsBuilderARM64::VisitMathCos(HInvoke* invoke) {
1852 CreateFPToFPCallLocations(arena_, invoke);
1853}
1854
1855void IntrinsicCodeGeneratorARM64::VisitMathCos(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001856 GenFPToFPCall(invoke, codegen_, kQuickCos);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001857}
1858
1859void IntrinsicLocationsBuilderARM64::VisitMathSin(HInvoke* invoke) {
1860 CreateFPToFPCallLocations(arena_, invoke);
1861}
1862
1863void IntrinsicCodeGeneratorARM64::VisitMathSin(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001864 GenFPToFPCall(invoke, codegen_, kQuickSin);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001865}
1866
1867void IntrinsicLocationsBuilderARM64::VisitMathAcos(HInvoke* invoke) {
1868 CreateFPToFPCallLocations(arena_, invoke);
1869}
1870
1871void IntrinsicCodeGeneratorARM64::VisitMathAcos(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001872 GenFPToFPCall(invoke, codegen_, kQuickAcos);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001873}
1874
1875void IntrinsicLocationsBuilderARM64::VisitMathAsin(HInvoke* invoke) {
1876 CreateFPToFPCallLocations(arena_, invoke);
1877}
1878
1879void IntrinsicCodeGeneratorARM64::VisitMathAsin(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001880 GenFPToFPCall(invoke, codegen_, kQuickAsin);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001881}
1882
1883void IntrinsicLocationsBuilderARM64::VisitMathAtan(HInvoke* invoke) {
1884 CreateFPToFPCallLocations(arena_, invoke);
1885}
1886
1887void IntrinsicCodeGeneratorARM64::VisitMathAtan(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001888 GenFPToFPCall(invoke, codegen_, kQuickAtan);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001889}
1890
1891void IntrinsicLocationsBuilderARM64::VisitMathCbrt(HInvoke* invoke) {
1892 CreateFPToFPCallLocations(arena_, invoke);
1893}
1894
1895void IntrinsicCodeGeneratorARM64::VisitMathCbrt(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001896 GenFPToFPCall(invoke, codegen_, kQuickCbrt);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001897}
1898
1899void IntrinsicLocationsBuilderARM64::VisitMathCosh(HInvoke* invoke) {
1900 CreateFPToFPCallLocations(arena_, invoke);
1901}
1902
1903void IntrinsicCodeGeneratorARM64::VisitMathCosh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001904 GenFPToFPCall(invoke, codegen_, kQuickCosh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001905}
1906
1907void IntrinsicLocationsBuilderARM64::VisitMathExp(HInvoke* invoke) {
1908 CreateFPToFPCallLocations(arena_, invoke);
1909}
1910
1911void IntrinsicCodeGeneratorARM64::VisitMathExp(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001912 GenFPToFPCall(invoke, codegen_, kQuickExp);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001913}
1914
1915void IntrinsicLocationsBuilderARM64::VisitMathExpm1(HInvoke* invoke) {
1916 CreateFPToFPCallLocations(arena_, invoke);
1917}
1918
1919void IntrinsicCodeGeneratorARM64::VisitMathExpm1(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001920 GenFPToFPCall(invoke, codegen_, kQuickExpm1);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001921}
1922
1923void IntrinsicLocationsBuilderARM64::VisitMathLog(HInvoke* invoke) {
1924 CreateFPToFPCallLocations(arena_, invoke);
1925}
1926
1927void IntrinsicCodeGeneratorARM64::VisitMathLog(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001928 GenFPToFPCall(invoke, codegen_, kQuickLog);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001929}
1930
1931void IntrinsicLocationsBuilderARM64::VisitMathLog10(HInvoke* invoke) {
1932 CreateFPToFPCallLocations(arena_, invoke);
1933}
1934
1935void IntrinsicCodeGeneratorARM64::VisitMathLog10(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001936 GenFPToFPCall(invoke, codegen_, kQuickLog10);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001937}
1938
1939void IntrinsicLocationsBuilderARM64::VisitMathSinh(HInvoke* invoke) {
1940 CreateFPToFPCallLocations(arena_, invoke);
1941}
1942
1943void IntrinsicCodeGeneratorARM64::VisitMathSinh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001944 GenFPToFPCall(invoke, codegen_, kQuickSinh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001945}
1946
1947void IntrinsicLocationsBuilderARM64::VisitMathTan(HInvoke* invoke) {
1948 CreateFPToFPCallLocations(arena_, invoke);
1949}
1950
1951void IntrinsicCodeGeneratorARM64::VisitMathTan(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001952 GenFPToFPCall(invoke, codegen_, kQuickTan);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001953}
1954
1955void IntrinsicLocationsBuilderARM64::VisitMathTanh(HInvoke* invoke) {
1956 CreateFPToFPCallLocations(arena_, invoke);
1957}
1958
1959void IntrinsicCodeGeneratorARM64::VisitMathTanh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001960 GenFPToFPCall(invoke, codegen_, kQuickTanh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001961}
1962
1963void IntrinsicLocationsBuilderARM64::VisitMathAtan2(HInvoke* invoke) {
1964 CreateFPFPToFPCallLocations(arena_, invoke);
1965}
1966
1967void IntrinsicCodeGeneratorARM64::VisitMathAtan2(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001968 GenFPToFPCall(invoke, codegen_, kQuickAtan2);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001969}
1970
1971void IntrinsicLocationsBuilderARM64::VisitMathHypot(HInvoke* invoke) {
1972 CreateFPFPToFPCallLocations(arena_, invoke);
1973}
1974
1975void IntrinsicCodeGeneratorARM64::VisitMathHypot(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001976 GenFPToFPCall(invoke, codegen_, kQuickHypot);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001977}
1978
1979void IntrinsicLocationsBuilderARM64::VisitMathNextAfter(HInvoke* invoke) {
1980 CreateFPFPToFPCallLocations(arena_, invoke);
1981}
1982
1983void IntrinsicCodeGeneratorARM64::VisitMathNextAfter(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001984 GenFPToFPCall(invoke, codegen_, kQuickNextAfter);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001985}
1986
Tim Zhang25abd6c2016-01-19 23:39:24 +08001987void IntrinsicLocationsBuilderARM64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
1988 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1989 LocationSummary::kNoCall,
1990 kIntrinsified);
1991 locations->SetInAt(0, Location::RequiresRegister());
1992 locations->SetInAt(1, Location::RequiresRegister());
1993 locations->SetInAt(2, Location::RequiresRegister());
1994 locations->SetInAt(3, Location::RequiresRegister());
1995 locations->SetInAt(4, Location::RequiresRegister());
1996
1997 locations->AddTemp(Location::RequiresRegister());
1998 locations->AddTemp(Location::RequiresRegister());
Scott Wakelingdf109d92016-04-22 11:35:56 +01001999 locations->AddTemp(Location::RequiresRegister());
Tim Zhang25abd6c2016-01-19 23:39:24 +08002000}
2001
2002void IntrinsicCodeGeneratorARM64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01002003 MacroAssembler* masm = GetVIXLAssembler();
Tim Zhang25abd6c2016-01-19 23:39:24 +08002004 LocationSummary* locations = invoke->GetLocations();
2005
2006 // Check assumption that sizeof(Char) is 2 (used in scaling below).
2007 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
2008 DCHECK_EQ(char_size, 2u);
2009
2010 // Location of data in char array buffer.
2011 const uint32_t data_offset = mirror::Array::DataOffset(char_size).Uint32Value();
2012
2013 // Location of char array data in string.
2014 const uint32_t value_offset = mirror::String::ValueOffset().Uint32Value();
2015
2016 // void getCharsNoCheck(int srcBegin, int srcEnd, char[] dst, int dstBegin);
2017 // Since getChars() calls getCharsNoCheck() - we use registers rather than constants.
2018 Register srcObj = XRegisterFrom(locations->InAt(0));
2019 Register srcBegin = XRegisterFrom(locations->InAt(1));
2020 Register srcEnd = XRegisterFrom(locations->InAt(2));
2021 Register dstObj = XRegisterFrom(locations->InAt(3));
2022 Register dstBegin = XRegisterFrom(locations->InAt(4));
2023
2024 Register src_ptr = XRegisterFrom(locations->GetTemp(0));
Scott Wakelingdf109d92016-04-22 11:35:56 +01002025 Register num_chr = XRegisterFrom(locations->GetTemp(1));
2026 Register tmp1 = XRegisterFrom(locations->GetTemp(2));
Tim Zhang25abd6c2016-01-19 23:39:24 +08002027
2028 UseScratchRegisterScope temps(masm);
2029 Register dst_ptr = temps.AcquireX();
Scott Wakelingdf109d92016-04-22 11:35:56 +01002030 Register tmp2 = temps.AcquireX();
Tim Zhang25abd6c2016-01-19 23:39:24 +08002031
jessicahandojo05765752016-09-09 19:01:32 -07002032 vixl::aarch64::Label done;
2033 vixl::aarch64::Label compressed_string_loop;
2034 __ Sub(num_chr, srcEnd, srcBegin);
2035 // Early out for valid zero-length retrievals.
2036 __ Cbz(num_chr, &done);
Tim Zhang25abd6c2016-01-19 23:39:24 +08002037
Scott Wakelingdf109d92016-04-22 11:35:56 +01002038 // dst address start to copy to.
Tim Zhang25abd6c2016-01-19 23:39:24 +08002039 __ Add(dst_ptr, dstObj, Operand(data_offset));
2040 __ Add(dst_ptr, dst_ptr, Operand(dstBegin, LSL, 1));
2041
jessicahandojo05765752016-09-09 19:01:32 -07002042 // src address to copy from.
2043 __ Add(src_ptr, srcObj, Operand(value_offset));
2044 vixl::aarch64::Label compressed_string_preloop;
2045 if (mirror::kUseStringCompression) {
2046 // Location of count in string.
2047 const uint32_t count_offset = mirror::String::CountOffset().Uint32Value();
2048 // String's length.
Vladimir Markofdaf0f42016-10-13 19:29:53 +01002049 __ Ldr(tmp2, MemOperand(srcObj, count_offset));
2050 __ Tbz(tmp2, 0, &compressed_string_preloop);
jessicahandojo05765752016-09-09 19:01:32 -07002051 }
2052 __ Add(src_ptr, src_ptr, Operand(srcBegin, LSL, 1));
Scott Wakelingdf109d92016-04-22 11:35:56 +01002053
Tim Zhang25abd6c2016-01-19 23:39:24 +08002054 // Do the copy.
Scott Wakeling97c72b72016-06-24 16:19:36 +01002055 vixl::aarch64::Label loop;
Scott Wakeling97c72b72016-06-24 16:19:36 +01002056 vixl::aarch64::Label remainder;
Scott Wakelingdf109d92016-04-22 11:35:56 +01002057
Scott Wakelingdf109d92016-04-22 11:35:56 +01002058 // Save repairing the value of num_chr on the < 8 character path.
2059 __ Subs(tmp1, num_chr, 8);
2060 __ B(lt, &remainder);
2061
2062 // Keep the result of the earlier subs, we are going to fetch at least 8 characters.
2063 __ Mov(num_chr, tmp1);
2064
2065 // Main loop used for longer fetches loads and stores 8x16-bit characters at a time.
2066 // (Unaligned addresses are acceptable here and not worth inlining extra code to rectify.)
Tim Zhang25abd6c2016-01-19 23:39:24 +08002067 __ Bind(&loop);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002068 __ Ldp(tmp1, tmp2, MemOperand(src_ptr, char_size * 8, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01002069 __ Subs(num_chr, num_chr, 8);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002070 __ Stp(tmp1, tmp2, MemOperand(dst_ptr, char_size * 8, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01002071 __ B(ge, &loop);
2072
2073 __ Adds(num_chr, num_chr, 8);
2074 __ B(eq, &done);
2075
2076 // Main loop for < 8 character case and remainder handling. Loads and stores one
2077 // 16-bit Java character at a time.
2078 __ Bind(&remainder);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002079 __ Ldrh(tmp1, MemOperand(src_ptr, char_size, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01002080 __ Subs(num_chr, num_chr, 1);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002081 __ Strh(tmp1, MemOperand(dst_ptr, char_size, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01002082 __ B(gt, &remainder);
jessicahandojo05765752016-09-09 19:01:32 -07002083 __ B(&done);
2084
2085 if (mirror::kUseStringCompression) {
2086 const size_t c_char_size = Primitive::ComponentSize(Primitive::kPrimByte);
2087 DCHECK_EQ(c_char_size, 1u);
2088 __ Bind(&compressed_string_preloop);
2089 __ Add(src_ptr, src_ptr, Operand(srcBegin));
2090 // Copy loop for compressed src, copying 1 character (8-bit) to (16-bit) at a time.
2091 __ Bind(&compressed_string_loop);
2092 __ Ldrb(tmp1, MemOperand(src_ptr, c_char_size, PostIndex));
2093 __ Strh(tmp1, MemOperand(dst_ptr, char_size, PostIndex));
2094 __ Subs(num_chr, num_chr, Operand(1));
2095 __ B(gt, &compressed_string_loop);
2096 }
Scott Wakelingdf109d92016-04-22 11:35:56 +01002097
Tim Zhang25abd6c2016-01-19 23:39:24 +08002098 __ Bind(&done);
2099}
2100
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002101// Mirrors ARRAYCOPY_SHORT_CHAR_ARRAY_THRESHOLD in libcore, so we can choose to use the native
2102// implementation there for longer copy lengths.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002103static constexpr int32_t kSystemArrayCopyCharThreshold = 32;
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002104
2105static void SetSystemArrayCopyLocationRequires(LocationSummary* locations,
2106 uint32_t at,
2107 HInstruction* input) {
2108 HIntConstant* const_input = input->AsIntConstant();
Scott Wakeling97c72b72016-06-24 16:19:36 +01002109 if (const_input != nullptr && !vixl::aarch64::Assembler::IsImmAddSub(const_input->GetValue())) {
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002110 locations->SetInAt(at, Location::RequiresRegister());
2111 } else {
2112 locations->SetInAt(at, Location::RegisterOrConstant(input));
2113 }
2114}
2115
2116void IntrinsicLocationsBuilderARM64::VisitSystemArrayCopyChar(HInvoke* invoke) {
2117 // Check to see if we have known failures that will cause us to have to bail out
2118 // to the runtime, and just generate the runtime call directly.
2119 HIntConstant* src_pos = invoke->InputAt(1)->AsIntConstant();
2120 HIntConstant* dst_pos = invoke->InputAt(3)->AsIntConstant();
2121
2122 // The positions must be non-negative.
2123 if ((src_pos != nullptr && src_pos->GetValue() < 0) ||
2124 (dst_pos != nullptr && dst_pos->GetValue() < 0)) {
2125 // We will have to fail anyways.
2126 return;
2127 }
2128
2129 // The length must be >= 0 and not so long that we would (currently) prefer libcore's
2130 // native implementation.
2131 HIntConstant* length = invoke->InputAt(4)->AsIntConstant();
2132 if (length != nullptr) {
2133 int32_t len = length->GetValue();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002134 if (len < 0 || len > kSystemArrayCopyCharThreshold) {
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002135 // Just call as normal.
2136 return;
2137 }
2138 }
2139
2140 ArenaAllocator* allocator = invoke->GetBlock()->GetGraph()->GetArena();
2141 LocationSummary* locations = new (allocator) LocationSummary(invoke,
2142 LocationSummary::kCallOnSlowPath,
2143 kIntrinsified);
2144 // arraycopy(char[] src, int src_pos, char[] dst, int dst_pos, int length).
2145 locations->SetInAt(0, Location::RequiresRegister());
2146 SetSystemArrayCopyLocationRequires(locations, 1, invoke->InputAt(1));
2147 locations->SetInAt(2, Location::RequiresRegister());
2148 SetSystemArrayCopyLocationRequires(locations, 3, invoke->InputAt(3));
2149 SetSystemArrayCopyLocationRequires(locations, 4, invoke->InputAt(4));
2150
2151 locations->AddTemp(Location::RequiresRegister());
2152 locations->AddTemp(Location::RequiresRegister());
2153 locations->AddTemp(Location::RequiresRegister());
2154}
2155
Scott Wakeling97c72b72016-06-24 16:19:36 +01002156static void CheckSystemArrayCopyPosition(MacroAssembler* masm,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002157 const Location& pos,
2158 const Register& input,
2159 const Location& length,
2160 SlowPathCodeARM64* slow_path,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002161 const Register& temp,
2162 bool length_is_input_length = false) {
2163 const int32_t length_offset = mirror::Array::LengthOffset().Int32Value();
2164 if (pos.IsConstant()) {
2165 int32_t pos_const = pos.GetConstant()->AsIntConstant()->GetValue();
2166 if (pos_const == 0) {
2167 if (!length_is_input_length) {
2168 // Check that length(input) >= length.
2169 __ Ldr(temp, MemOperand(input, length_offset));
2170 __ Cmp(temp, OperandFrom(length, Primitive::kPrimInt));
2171 __ B(slow_path->GetEntryLabel(), lt);
2172 }
2173 } else {
2174 // Check that length(input) >= pos.
Nicolas Geoffrayfea1abd2016-07-06 12:09:12 +01002175 __ Ldr(temp, MemOperand(input, length_offset));
2176 __ Subs(temp, temp, pos_const);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002177 __ B(slow_path->GetEntryLabel(), lt);
2178
2179 // Check that (length(input) - pos) >= length.
2180 __ Cmp(temp, OperandFrom(length, Primitive::kPrimInt));
2181 __ B(slow_path->GetEntryLabel(), lt);
2182 }
2183 } else if (length_is_input_length) {
2184 // The only way the copy can succeed is if pos is zero.
2185 __ Cbnz(WRegisterFrom(pos), slow_path->GetEntryLabel());
2186 } else {
2187 // Check that pos >= 0.
2188 Register pos_reg = WRegisterFrom(pos);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002189 __ Tbnz(pos_reg, pos_reg.GetSizeInBits() - 1, slow_path->GetEntryLabel());
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002190
2191 // Check that pos <= length(input) && (length(input) - pos) >= length.
2192 __ Ldr(temp, MemOperand(input, length_offset));
2193 __ Subs(temp, temp, pos_reg);
2194 // Ccmp if length(input) >= pos, else definitely bail to slow path (N!=V == lt).
2195 __ Ccmp(temp, OperandFrom(length, Primitive::kPrimInt), NFlag, ge);
2196 __ B(slow_path->GetEntryLabel(), lt);
2197 }
2198}
2199
Roland Levillain9cc0ea82017-03-16 11:25:59 +00002200// Compute base source address, base destination address, and end
2201// source address for System.arraycopy* intrinsics in `src_base`,
2202// `dst_base` and `src_end` respectively.
Scott Wakeling97c72b72016-06-24 16:19:36 +01002203static void GenSystemArrayCopyAddresses(MacroAssembler* masm,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002204 Primitive::Type type,
2205 const Register& src,
2206 const Location& src_pos,
2207 const Register& dst,
2208 const Location& dst_pos,
2209 const Location& copy_length,
2210 const Register& src_base,
2211 const Register& dst_base,
2212 const Register& src_end) {
Roland Levillain9cc0ea82017-03-16 11:25:59 +00002213 // This routine is used by the SystemArrayCopy and the SystemArrayCopyChar intrinsics.
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002214 DCHECK(type == Primitive::kPrimNot || type == Primitive::kPrimChar)
Roland Levillainebea3d22016-04-12 15:42:57 +01002215 << "Unexpected element type: " << type;
2216 const int32_t element_size = Primitive::ComponentSize(type);
2217 const int32_t element_size_shift = Primitive::ComponentSizeShift(type);
Roland Levillain9cc0ea82017-03-16 11:25:59 +00002218 const uint32_t data_offset = mirror::Array::DataOffset(element_size).Uint32Value();
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002219
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002220 if (src_pos.IsConstant()) {
2221 int32_t constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002222 __ Add(src_base, src, element_size * constant + data_offset);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002223 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002224 __ Add(src_base, src, data_offset);
2225 __ Add(src_base, src_base, Operand(XRegisterFrom(src_pos), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002226 }
2227
2228 if (dst_pos.IsConstant()) {
2229 int32_t constant = dst_pos.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002230 __ Add(dst_base, dst, element_size * constant + data_offset);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002231 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002232 __ Add(dst_base, dst, data_offset);
2233 __ Add(dst_base, dst_base, Operand(XRegisterFrom(dst_pos), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002234 }
2235
2236 if (copy_length.IsConstant()) {
2237 int32_t constant = copy_length.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002238 __ Add(src_end, src_base, element_size * constant);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002239 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002240 __ Add(src_end, src_base, Operand(XRegisterFrom(copy_length), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002241 }
2242}
2243
2244void IntrinsicCodeGeneratorARM64::VisitSystemArrayCopyChar(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01002245 MacroAssembler* masm = GetVIXLAssembler();
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002246 LocationSummary* locations = invoke->GetLocations();
2247 Register src = XRegisterFrom(locations->InAt(0));
2248 Location src_pos = locations->InAt(1);
2249 Register dst = XRegisterFrom(locations->InAt(2));
2250 Location dst_pos = locations->InAt(3);
2251 Location length = locations->InAt(4);
2252
2253 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
2254 codegen_->AddSlowPath(slow_path);
2255
2256 // If source and destination are the same, take the slow path. Overlapping copy regions must be
2257 // copied in reverse and we can't know in all cases if it's needed.
2258 __ Cmp(src, dst);
2259 __ B(slow_path->GetEntryLabel(), eq);
2260
2261 // Bail out if the source is null.
2262 __ Cbz(src, slow_path->GetEntryLabel());
2263
2264 // Bail out if the destination is null.
2265 __ Cbz(dst, slow_path->GetEntryLabel());
2266
2267 if (!length.IsConstant()) {
Vladimir Markoc5646202016-11-28 16:03:15 +00002268 // Merge the following two comparisons into one:
2269 // If the length is negative, bail out (delegate to libcore's native implementation).
2270 // If the length > 32 then (currently) prefer libcore's native implementation.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002271 __ Cmp(WRegisterFrom(length), kSystemArrayCopyCharThreshold);
Vladimir Markoc5646202016-11-28 16:03:15 +00002272 __ B(slow_path->GetEntryLabel(), hi);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002273 } else {
2274 // We have already checked in the LocationsBuilder for the constant case.
2275 DCHECK_GE(length.GetConstant()->AsIntConstant()->GetValue(), 0);
2276 DCHECK_LE(length.GetConstant()->AsIntConstant()->GetValue(), 32);
2277 }
2278
2279 Register src_curr_addr = WRegisterFrom(locations->GetTemp(0));
2280 Register dst_curr_addr = WRegisterFrom(locations->GetTemp(1));
2281 Register src_stop_addr = WRegisterFrom(locations->GetTemp(2));
2282
2283 CheckSystemArrayCopyPosition(masm,
2284 src_pos,
2285 src,
2286 length,
2287 slow_path,
2288 src_curr_addr,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002289 false);
2290
2291 CheckSystemArrayCopyPosition(masm,
2292 dst_pos,
2293 dst,
2294 length,
2295 slow_path,
2296 src_curr_addr,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002297 false);
2298
2299 src_curr_addr = src_curr_addr.X();
2300 dst_curr_addr = dst_curr_addr.X();
2301 src_stop_addr = src_stop_addr.X();
2302
2303 GenSystemArrayCopyAddresses(masm,
2304 Primitive::kPrimChar,
2305 src,
2306 src_pos,
2307 dst,
2308 dst_pos,
2309 length,
2310 src_curr_addr,
2311 dst_curr_addr,
2312 src_stop_addr);
2313
2314 // Iterate over the arrays and do a raw copy of the chars.
2315 const int32_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
2316 UseScratchRegisterScope temps(masm);
2317 Register tmp = temps.AcquireW();
Scott Wakeling97c72b72016-06-24 16:19:36 +01002318 vixl::aarch64::Label loop, done;
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002319 __ Bind(&loop);
2320 __ Cmp(src_curr_addr, src_stop_addr);
2321 __ B(&done, eq);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002322 __ Ldrh(tmp, MemOperand(src_curr_addr, char_size, PostIndex));
2323 __ Strh(tmp, MemOperand(dst_curr_addr, char_size, PostIndex));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002324 __ B(&loop);
2325 __ Bind(&done);
2326
2327 __ Bind(slow_path->GetExitLabel());
2328}
2329
donghui.baic2ec9ad2016-03-10 14:02:55 +08002330// We can choose to use the native implementation there for longer copy lengths.
2331static constexpr int32_t kSystemArrayCopyThreshold = 128;
2332
2333// CodeGenerator::CreateSystemArrayCopyLocationSummary use three temporary registers.
2334// We want to use two temporary registers in order to reduce the register pressure in arm64.
2335// So we don't use the CodeGenerator::CreateSystemArrayCopyLocationSummary.
2336void IntrinsicLocationsBuilderARM64::VisitSystemArrayCopy(HInvoke* invoke) {
Roland Levillain0b671c02016-08-19 12:02:34 +01002337 // The only read barrier implementation supporting the
2338 // SystemArrayCopy intrinsic is the Baker-style read barriers.
2339 if (kEmitCompilerReadBarrier && !kUseBakerReadBarrier) {
Roland Levillain3d312422016-06-23 13:53:42 +01002340 return;
2341 }
2342
donghui.baic2ec9ad2016-03-10 14:02:55 +08002343 // Check to see if we have known failures that will cause us to have to bail out
2344 // to the runtime, and just generate the runtime call directly.
2345 HIntConstant* src_pos = invoke->InputAt(1)->AsIntConstant();
2346 HIntConstant* dest_pos = invoke->InputAt(3)->AsIntConstant();
2347
2348 // The positions must be non-negative.
2349 if ((src_pos != nullptr && src_pos->GetValue() < 0) ||
2350 (dest_pos != nullptr && dest_pos->GetValue() < 0)) {
2351 // We will have to fail anyways.
2352 return;
2353 }
2354
2355 // The length must be >= 0.
2356 HIntConstant* length = invoke->InputAt(4)->AsIntConstant();
2357 if (length != nullptr) {
2358 int32_t len = length->GetValue();
2359 if (len < 0 || len >= kSystemArrayCopyThreshold) {
2360 // Just call as normal.
2361 return;
2362 }
2363 }
2364
2365 SystemArrayCopyOptimizations optimizations(invoke);
2366
2367 if (optimizations.GetDestinationIsSource()) {
2368 if (src_pos != nullptr && dest_pos != nullptr && src_pos->GetValue() < dest_pos->GetValue()) {
2369 // We only support backward copying if source and destination are the same.
2370 return;
2371 }
2372 }
2373
2374 if (optimizations.GetDestinationIsPrimitiveArray() || optimizations.GetSourceIsPrimitiveArray()) {
2375 // We currently don't intrinsify primitive copying.
2376 return;
2377 }
2378
2379 ArenaAllocator* allocator = invoke->GetBlock()->GetGraph()->GetArena();
2380 LocationSummary* locations = new (allocator) LocationSummary(invoke,
2381 LocationSummary::kCallOnSlowPath,
2382 kIntrinsified);
2383 // arraycopy(Object src, int src_pos, Object dest, int dest_pos, int length).
2384 locations->SetInAt(0, Location::RequiresRegister());
2385 SetSystemArrayCopyLocationRequires(locations, 1, invoke->InputAt(1));
2386 locations->SetInAt(2, Location::RequiresRegister());
2387 SetSystemArrayCopyLocationRequires(locations, 3, invoke->InputAt(3));
2388 SetSystemArrayCopyLocationRequires(locations, 4, invoke->InputAt(4));
2389
2390 locations->AddTemp(Location::RequiresRegister());
2391 locations->AddTemp(Location::RequiresRegister());
Roland Levillain0b671c02016-08-19 12:02:34 +01002392 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2393 // Temporary register IP0, obtained from the VIXL scratch register
2394 // pool, cannot be used in ReadBarrierSystemArrayCopySlowPathARM64
2395 // (because that register is clobbered by ReadBarrierMarkRegX
Roland Levillain54f869e2017-03-06 13:54:11 +00002396 // entry points). It cannot be used in calls to
2397 // CodeGeneratorARM64::GenerateFieldLoadWithBakerReadBarrier
2398 // either. For these reasons, get a third extra temporary register
2399 // from the register allocator.
Roland Levillain0b671c02016-08-19 12:02:34 +01002400 locations->AddTemp(Location::RequiresRegister());
Roland Levillain54f869e2017-03-06 13:54:11 +00002401 } else {
2402 // Cases other than Baker read barriers: the third temporary will
2403 // be acquired from the VIXL scratch register pool.
Roland Levillain0b671c02016-08-19 12:02:34 +01002404 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002405}
2406
2407void IntrinsicCodeGeneratorARM64::VisitSystemArrayCopy(HInvoke* invoke) {
Roland Levillain0b671c02016-08-19 12:02:34 +01002408 // The only read barrier implementation supporting the
2409 // SystemArrayCopy intrinsic is the Baker-style read barriers.
2410 DCHECK(!kEmitCompilerReadBarrier || kUseBakerReadBarrier);
Roland Levillain3d312422016-06-23 13:53:42 +01002411
Scott Wakeling97c72b72016-06-24 16:19:36 +01002412 MacroAssembler* masm = GetVIXLAssembler();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002413 LocationSummary* locations = invoke->GetLocations();
2414
2415 uint32_t class_offset = mirror::Object::ClassOffset().Int32Value();
2416 uint32_t super_offset = mirror::Class::SuperClassOffset().Int32Value();
2417 uint32_t component_offset = mirror::Class::ComponentTypeOffset().Int32Value();
2418 uint32_t primitive_offset = mirror::Class::PrimitiveTypeOffset().Int32Value();
Roland Levillain0b671c02016-08-19 12:02:34 +01002419 uint32_t monitor_offset = mirror::Object::MonitorOffset().Int32Value();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002420
2421 Register src = XRegisterFrom(locations->InAt(0));
2422 Location src_pos = locations->InAt(1);
2423 Register dest = XRegisterFrom(locations->InAt(2));
2424 Location dest_pos = locations->InAt(3);
2425 Location length = locations->InAt(4);
2426 Register temp1 = WRegisterFrom(locations->GetTemp(0));
Roland Levillain0b671c02016-08-19 12:02:34 +01002427 Location temp1_loc = LocationFrom(temp1);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002428 Register temp2 = WRegisterFrom(locations->GetTemp(1));
Roland Levillain0b671c02016-08-19 12:02:34 +01002429 Location temp2_loc = LocationFrom(temp2);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002430
Roland Levillain0b671c02016-08-19 12:02:34 +01002431 SlowPathCodeARM64* intrinsic_slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
2432 codegen_->AddSlowPath(intrinsic_slow_path);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002433
Scott Wakeling97c72b72016-06-24 16:19:36 +01002434 vixl::aarch64::Label conditions_on_positions_validated;
donghui.baic2ec9ad2016-03-10 14:02:55 +08002435 SystemArrayCopyOptimizations optimizations(invoke);
2436
donghui.baic2ec9ad2016-03-10 14:02:55 +08002437 // If source and destination are the same, we go to slow path if we need to do
2438 // forward copying.
2439 if (src_pos.IsConstant()) {
2440 int32_t src_pos_constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
2441 if (dest_pos.IsConstant()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002442 int32_t dest_pos_constant = dest_pos.GetConstant()->AsIntConstant()->GetValue();
2443 if (optimizations.GetDestinationIsSource()) {
2444 // Checked when building locations.
2445 DCHECK_GE(src_pos_constant, dest_pos_constant);
2446 } else if (src_pos_constant < dest_pos_constant) {
2447 __ Cmp(src, dest);
Roland Levillain0b671c02016-08-19 12:02:34 +01002448 __ B(intrinsic_slow_path->GetEntryLabel(), eq);
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002449 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002450 // Checked when building locations.
2451 DCHECK(!optimizations.GetDestinationIsSource()
2452 || (src_pos_constant >= dest_pos.GetConstant()->AsIntConstant()->GetValue()));
2453 } else {
2454 if (!optimizations.GetDestinationIsSource()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002455 __ Cmp(src, dest);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002456 __ B(&conditions_on_positions_validated, ne);
2457 }
2458 __ Cmp(WRegisterFrom(dest_pos), src_pos_constant);
Roland Levillain0b671c02016-08-19 12:02:34 +01002459 __ B(intrinsic_slow_path->GetEntryLabel(), gt);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002460 }
2461 } else {
2462 if (!optimizations.GetDestinationIsSource()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002463 __ Cmp(src, dest);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002464 __ B(&conditions_on_positions_validated, ne);
2465 }
2466 __ Cmp(RegisterFrom(src_pos, invoke->InputAt(1)->GetType()),
2467 OperandFrom(dest_pos, invoke->InputAt(3)->GetType()));
Roland Levillain0b671c02016-08-19 12:02:34 +01002468 __ B(intrinsic_slow_path->GetEntryLabel(), lt);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002469 }
2470
2471 __ Bind(&conditions_on_positions_validated);
2472
2473 if (!optimizations.GetSourceIsNotNull()) {
2474 // Bail out if the source is null.
Roland Levillain0b671c02016-08-19 12:02:34 +01002475 __ Cbz(src, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002476 }
2477
2478 if (!optimizations.GetDestinationIsNotNull() && !optimizations.GetDestinationIsSource()) {
2479 // Bail out if the destination is null.
Roland Levillain0b671c02016-08-19 12:02:34 +01002480 __ Cbz(dest, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002481 }
2482
2483 // We have already checked in the LocationsBuilder for the constant case.
2484 if (!length.IsConstant() &&
2485 !optimizations.GetCountIsSourceLength() &&
2486 !optimizations.GetCountIsDestinationLength()) {
Vladimir Markoc5646202016-11-28 16:03:15 +00002487 // Merge the following two comparisons into one:
2488 // If the length is negative, bail out (delegate to libcore's native implementation).
2489 // If the length >= 128 then (currently) prefer native implementation.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002490 __ Cmp(WRegisterFrom(length), kSystemArrayCopyThreshold);
Vladimir Markoc5646202016-11-28 16:03:15 +00002491 __ B(intrinsic_slow_path->GetEntryLabel(), hs);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002492 }
2493 // Validity checks: source.
2494 CheckSystemArrayCopyPosition(masm,
2495 src_pos,
2496 src,
2497 length,
Roland Levillain0b671c02016-08-19 12:02:34 +01002498 intrinsic_slow_path,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002499 temp1,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002500 optimizations.GetCountIsSourceLength());
2501
2502 // Validity checks: dest.
2503 CheckSystemArrayCopyPosition(masm,
2504 dest_pos,
2505 dest,
2506 length,
Roland Levillain0b671c02016-08-19 12:02:34 +01002507 intrinsic_slow_path,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002508 temp1,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002509 optimizations.GetCountIsDestinationLength());
2510 {
2511 // We use a block to end the scratch scope before the write barrier, thus
2512 // freeing the temporary registers so they can be used in `MarkGCCard`.
2513 UseScratchRegisterScope temps(masm);
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002514 Location temp3_loc; // Used only for Baker read barrier.
Roland Levillain54f869e2017-03-06 13:54:11 +00002515 Register temp3;
2516 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002517 temp3_loc = locations->GetTemp(2);
2518 temp3 = WRegisterFrom(temp3_loc);
Roland Levillain54f869e2017-03-06 13:54:11 +00002519 } else {
2520 temp3 = temps.AcquireW();
2521 }
Roland Levillain0b671c02016-08-19 12:02:34 +01002522
donghui.baic2ec9ad2016-03-10 14:02:55 +08002523 if (!optimizations.GetDoesNotNeedTypeCheck()) {
2524 // Check whether all elements of the source array are assignable to the component
2525 // type of the destination array. We do two checks: the classes are the same,
2526 // or the destination is Object[]. If none of these checks succeed, we go to the
2527 // slow path.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002528
Roland Levillain0b671c02016-08-19 12:02:34 +01002529 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2530 if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2531 // /* HeapReference<Class> */ temp1 = src->klass_
2532 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2533 temp1_loc,
2534 src.W(),
2535 class_offset,
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002536 temp3_loc,
Roland Levillain0b671c02016-08-19 12:02:34 +01002537 /* needs_null_check */ false,
2538 /* use_load_acquire */ false);
2539 // Bail out if the source is not a non primitive array.
2540 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2541 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2542 temp1_loc,
2543 temp1,
2544 component_offset,
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002545 temp3_loc,
Roland Levillain0b671c02016-08-19 12:02:34 +01002546 /* needs_null_check */ false,
2547 /* use_load_acquire */ false);
2548 __ Cbz(temp1, intrinsic_slow_path->GetEntryLabel());
2549 // If heap poisoning is enabled, `temp1` has been unpoisoned
2550 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2551 // /* uint16_t */ temp1 = static_cast<uint16>(temp1->primitive_type_);
2552 __ Ldrh(temp1, HeapOperand(temp1, primitive_offset));
2553 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2554 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002555 }
Roland Levillain0b671c02016-08-19 12:02:34 +01002556
2557 // /* HeapReference<Class> */ temp1 = dest->klass_
2558 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2559 temp1_loc,
2560 dest.W(),
2561 class_offset,
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002562 temp3_loc,
Roland Levillain0b671c02016-08-19 12:02:34 +01002563 /* needs_null_check */ false,
2564 /* use_load_acquire */ false);
2565
2566 if (!optimizations.GetDestinationIsNonPrimitiveArray()) {
2567 // Bail out if the destination is not a non primitive array.
2568 //
2569 // Register `temp1` is not trashed by the read barrier emitted
2570 // by GenerateFieldLoadWithBakerReadBarrier below, as that
2571 // method produces a call to a ReadBarrierMarkRegX entry point,
2572 // which saves all potentially live registers, including
2573 // temporaries such a `temp1`.
2574 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2575 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2576 temp2_loc,
2577 temp1,
2578 component_offset,
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002579 temp3_loc,
Roland Levillain0b671c02016-08-19 12:02:34 +01002580 /* needs_null_check */ false,
2581 /* use_load_acquire */ false);
2582 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2583 // If heap poisoning is enabled, `temp2` has been unpoisoned
2584 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2585 // /* uint16_t */ temp2 = static_cast<uint16>(temp2->primitive_type_);
2586 __ Ldrh(temp2, HeapOperand(temp2, primitive_offset));
2587 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2588 __ Cbnz(temp2, intrinsic_slow_path->GetEntryLabel());
2589 }
2590
2591 // For the same reason given earlier, `temp1` is not trashed by the
2592 // read barrier emitted by GenerateFieldLoadWithBakerReadBarrier below.
2593 // /* HeapReference<Class> */ temp2 = src->klass_
2594 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2595 temp2_loc,
2596 src.W(),
2597 class_offset,
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002598 temp3_loc,
Roland Levillain0b671c02016-08-19 12:02:34 +01002599 /* needs_null_check */ false,
2600 /* use_load_acquire */ false);
2601 // Note: if heap poisoning is on, we are comparing two unpoisoned references here.
2602 __ Cmp(temp1, temp2);
2603
2604 if (optimizations.GetDestinationIsTypedObjectArray()) {
2605 vixl::aarch64::Label do_copy;
2606 __ B(&do_copy, eq);
2607 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2608 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2609 temp1_loc,
2610 temp1,
2611 component_offset,
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002612 temp3_loc,
Roland Levillain0b671c02016-08-19 12:02:34 +01002613 /* needs_null_check */ false,
2614 /* use_load_acquire */ false);
2615 // /* HeapReference<Class> */ temp1 = temp1->super_class_
2616 // We do not need to emit a read barrier for the following
2617 // heap reference load, as `temp1` is only used in a
2618 // comparison with null below, and this reference is not
2619 // kept afterwards.
2620 __ Ldr(temp1, HeapOperand(temp1, super_offset));
2621 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
2622 __ Bind(&do_copy);
2623 } else {
2624 __ B(intrinsic_slow_path->GetEntryLabel(), ne);
2625 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002626 } else {
Roland Levillain0b671c02016-08-19 12:02:34 +01002627 // Non read barrier code.
2628
2629 // /* HeapReference<Class> */ temp1 = dest->klass_
2630 __ Ldr(temp1, MemOperand(dest, class_offset));
2631 // /* HeapReference<Class> */ temp2 = src->klass_
2632 __ Ldr(temp2, MemOperand(src, class_offset));
2633 bool did_unpoison = false;
2634 if (!optimizations.GetDestinationIsNonPrimitiveArray() ||
2635 !optimizations.GetSourceIsNonPrimitiveArray()) {
2636 // One or two of the references need to be unpoisoned. Unpoison them
2637 // both to make the identity check valid.
2638 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2639 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp2);
2640 did_unpoison = true;
2641 }
2642
2643 if (!optimizations.GetDestinationIsNonPrimitiveArray()) {
2644 // Bail out if the destination is not a non primitive array.
2645 // /* HeapReference<Class> */ temp3 = temp1->component_type_
2646 __ Ldr(temp3, HeapOperand(temp1, component_offset));
2647 __ Cbz(temp3, intrinsic_slow_path->GetEntryLabel());
2648 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp3);
2649 // /* uint16_t */ temp3 = static_cast<uint16>(temp3->primitive_type_);
2650 __ Ldrh(temp3, HeapOperand(temp3, primitive_offset));
2651 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2652 __ Cbnz(temp3, intrinsic_slow_path->GetEntryLabel());
2653 }
2654
2655 if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2656 // Bail out if the source is not a non primitive array.
2657 // /* HeapReference<Class> */ temp3 = temp2->component_type_
2658 __ Ldr(temp3, HeapOperand(temp2, component_offset));
2659 __ Cbz(temp3, intrinsic_slow_path->GetEntryLabel());
2660 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp3);
2661 // /* uint16_t */ temp3 = static_cast<uint16>(temp3->primitive_type_);
2662 __ Ldrh(temp3, HeapOperand(temp3, primitive_offset));
2663 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2664 __ Cbnz(temp3, intrinsic_slow_path->GetEntryLabel());
2665 }
2666
2667 __ Cmp(temp1, temp2);
2668
2669 if (optimizations.GetDestinationIsTypedObjectArray()) {
2670 vixl::aarch64::Label do_copy;
2671 __ B(&do_copy, eq);
2672 if (!did_unpoison) {
2673 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2674 }
2675 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2676 __ Ldr(temp1, HeapOperand(temp1, component_offset));
2677 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2678 // /* HeapReference<Class> */ temp1 = temp1->super_class_
2679 __ Ldr(temp1, HeapOperand(temp1, super_offset));
2680 // No need to unpoison the result, we're comparing against null.
2681 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
2682 __ Bind(&do_copy);
2683 } else {
2684 __ B(intrinsic_slow_path->GetEntryLabel(), ne);
2685 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002686 }
2687 } else if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2688 DCHECK(optimizations.GetDestinationIsNonPrimitiveArray());
2689 // Bail out if the source is not a non primitive array.
Roland Levillain0b671c02016-08-19 12:02:34 +01002690 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2691 // /* HeapReference<Class> */ temp1 = src->klass_
2692 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2693 temp1_loc,
2694 src.W(),
2695 class_offset,
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002696 temp3_loc,
Roland Levillain0b671c02016-08-19 12:02:34 +01002697 /* needs_null_check */ false,
2698 /* use_load_acquire */ false);
2699 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2700 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2701 temp2_loc,
2702 temp1,
2703 component_offset,
Vladimir Markof4f2daa2017-03-20 18:26:59 +00002704 temp3_loc,
Roland Levillain0b671c02016-08-19 12:02:34 +01002705 /* needs_null_check */ false,
2706 /* use_load_acquire */ false);
2707 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2708 // If heap poisoning is enabled, `temp2` has been unpoisoned
2709 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2710 } else {
2711 // /* HeapReference<Class> */ temp1 = src->klass_
2712 __ Ldr(temp1, HeapOperand(src.W(), class_offset));
2713 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2714 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2715 __ Ldr(temp2, HeapOperand(temp1, component_offset));
2716 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2717 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp2);
2718 }
2719 // /* uint16_t */ temp2 = static_cast<uint16>(temp2->primitive_type_);
2720 __ Ldrh(temp2, HeapOperand(temp2, primitive_offset));
donghui.baic2ec9ad2016-03-10 14:02:55 +08002721 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
Roland Levillain0b671c02016-08-19 12:02:34 +01002722 __ Cbnz(temp2, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002723 }
2724
Roland Levillain1663d162017-03-17 15:15:21 +00002725 if (length.IsConstant() && length.GetConstant()->AsIntConstant()->GetValue() == 0) {
2726 // Null constant length: not need to emit the loop code at all.
Roland Levillain0b671c02016-08-19 12:02:34 +01002727 } else {
Roland Levillain1663d162017-03-17 15:15:21 +00002728 Register src_curr_addr = temp1.X();
2729 Register dst_curr_addr = temp2.X();
2730 Register src_stop_addr = temp3.X();
2731 vixl::aarch64::Label done;
2732 const Primitive::Type type = Primitive::kPrimNot;
2733 const int32_t element_size = Primitive::ComponentSize(type);
2734
2735 if (length.IsRegister()) {
2736 // Don't enter the copy loop if the length is null.
2737 __ Cbz(WRegisterFrom(length), &done);
2738 }
2739
2740 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2741 // TODO: Also convert this intrinsic to the IsGcMarking strategy?
2742
2743 // SystemArrayCopy implementation for Baker read barriers (see
2744 // also CodeGeneratorARM::GenerateReferenceLoadWithBakerReadBarrier):
2745 //
2746 // uint32_t rb_state = Lockword(src->monitor_).ReadBarrierState();
2747 // lfence; // Load fence or artificial data dependency to prevent load-load reordering
2748 // bool is_gray = (rb_state == ReadBarrier::GrayState());
2749 // if (is_gray) {
2750 // // Slow-path copy.
2751 // do {
2752 // *dest_ptr++ = MaybePoison(ReadBarrier::Mark(MaybeUnpoison(*src_ptr++)));
2753 // } while (src_ptr != end_ptr)
2754 // } else {
2755 // // Fast-path copy.
2756 // do {
2757 // *dest_ptr++ = *src_ptr++;
2758 // } while (src_ptr != end_ptr)
2759 // }
2760
2761 // Make sure `tmp` is not IP0, as it is clobbered by
2762 // ReadBarrierMarkRegX entry points in
2763 // ReadBarrierSystemArrayCopySlowPathARM64.
Roland Levillain1ca955d2017-04-13 19:34:30 +01002764 DCHECK(temps.IsAvailable(ip0));
Roland Levillain1663d162017-03-17 15:15:21 +00002765 temps.Exclude(ip0);
Roland Levillain0b671c02016-08-19 12:02:34 +01002766 Register tmp = temps.AcquireW();
Roland Levillain1663d162017-03-17 15:15:21 +00002767 DCHECK_NE(LocationFrom(tmp).reg(), IP0);
Roland Levillain1ca955d2017-04-13 19:34:30 +01002768 // Put IP0 back in the pool so that VIXL has at least one
2769 // scratch register available to emit macro-instructions (note
2770 // that IP1 is already used for `tmp`). Indeed some
2771 // macro-instructions used in GenSystemArrayCopyAddresses
2772 // (invoked hereunder) may require a scratch register (for
2773 // instance to emit a load with a large constant offset).
2774 temps.Include(ip0);
Roland Levillain1663d162017-03-17 15:15:21 +00002775
2776 // /* int32_t */ monitor = src->monitor_
2777 __ Ldr(tmp, HeapOperand(src.W(), monitor_offset));
2778 // /* LockWord */ lock_word = LockWord(monitor)
2779 static_assert(sizeof(LockWord) == sizeof(int32_t),
2780 "art::LockWord and int32_t have different sizes.");
2781
2782 // Introduce a dependency on the lock_word including rb_state,
2783 // to prevent load-load reordering, and without using
2784 // a memory barrier (which would be more expensive).
2785 // `src` is unchanged by this operation, but its value now depends
2786 // on `tmp`.
2787 __ Add(src.X(), src.X(), Operand(tmp.X(), LSR, 32));
2788
2789 // Compute base source address, base destination address, and end
2790 // source address for System.arraycopy* intrinsics in `src_base`,
2791 // `dst_base` and `src_end` respectively.
2792 // Note that `src_curr_addr` is computed from from `src` (and
2793 // `src_pos`) here, and thus honors the artificial dependency
2794 // of `src` on `tmp`.
2795 GenSystemArrayCopyAddresses(masm,
2796 type,
2797 src,
2798 src_pos,
2799 dest,
2800 dest_pos,
2801 length,
2802 src_curr_addr,
2803 dst_curr_addr,
2804 src_stop_addr);
2805
2806 // Slow path used to copy array when `src` is gray.
2807 SlowPathCodeARM64* read_barrier_slow_path =
2808 new (GetAllocator()) ReadBarrierSystemArrayCopySlowPathARM64(invoke, LocationFrom(tmp));
2809 codegen_->AddSlowPath(read_barrier_slow_path);
2810
2811 // Given the numeric representation, it's enough to check the low bit of the rb_state.
2812 static_assert(ReadBarrier::WhiteState() == 0, "Expecting white to have value 0");
2813 static_assert(ReadBarrier::GrayState() == 1, "Expecting gray to have value 1");
2814 __ Tbnz(tmp, LockWord::kReadBarrierStateShift, read_barrier_slow_path->GetEntryLabel());
2815
2816 // Fast-path copy.
2817 // Iterate over the arrays and do a raw copy of the objects. We don't need to
2818 // poison/unpoison.
2819 vixl::aarch64::Label loop;
2820 __ Bind(&loop);
Roland Levillain0b671c02016-08-19 12:02:34 +01002821 __ Ldr(tmp, MemOperand(src_curr_addr, element_size, PostIndex));
2822 __ Str(tmp, MemOperand(dst_curr_addr, element_size, PostIndex));
Roland Levillain1663d162017-03-17 15:15:21 +00002823 __ Cmp(src_curr_addr, src_stop_addr);
2824 __ B(&loop, ne);
2825
2826 __ Bind(read_barrier_slow_path->GetExitLabel());
2827 } else {
2828 // Non read barrier code.
2829 // Compute base source address, base destination address, and end
2830 // source address for System.arraycopy* intrinsics in `src_base`,
2831 // `dst_base` and `src_end` respectively.
2832 GenSystemArrayCopyAddresses(masm,
2833 type,
2834 src,
2835 src_pos,
2836 dest,
2837 dest_pos,
2838 length,
2839 src_curr_addr,
2840 dst_curr_addr,
2841 src_stop_addr);
2842 // Iterate over the arrays and do a raw copy of the objects. We don't need to
2843 // poison/unpoison.
2844 vixl::aarch64::Label loop;
2845 __ Bind(&loop);
2846 {
2847 Register tmp = temps.AcquireW();
2848 __ Ldr(tmp, MemOperand(src_curr_addr, element_size, PostIndex));
2849 __ Str(tmp, MemOperand(dst_curr_addr, element_size, PostIndex));
2850 }
2851 __ Cmp(src_curr_addr, src_stop_addr);
2852 __ B(&loop, ne);
Roland Levillain0b671c02016-08-19 12:02:34 +01002853 }
Roland Levillain0b671c02016-08-19 12:02:34 +01002854 __ Bind(&done);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002855 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002856 }
Roland Levillain9cc0ea82017-03-16 11:25:59 +00002857
donghui.baic2ec9ad2016-03-10 14:02:55 +08002858 // We only need one card marking on the destination array.
2859 codegen_->MarkGCCard(dest.W(), Register(), /* value_can_be_null */ false);
2860
Roland Levillain0b671c02016-08-19 12:02:34 +01002861 __ Bind(intrinsic_slow_path->GetExitLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002862}
2863
Anton Kirilova3ffea22016-04-07 17:02:37 +01002864static void GenIsInfinite(LocationSummary* locations,
2865 bool is64bit,
Scott Wakeling97c72b72016-06-24 16:19:36 +01002866 MacroAssembler* masm) {
Anton Kirilova3ffea22016-04-07 17:02:37 +01002867 Operand infinity;
2868 Register out;
2869
2870 if (is64bit) {
2871 infinity = kPositiveInfinityDouble;
2872 out = XRegisterFrom(locations->Out());
2873 } else {
2874 infinity = kPositiveInfinityFloat;
2875 out = WRegisterFrom(locations->Out());
2876 }
2877
Scott Wakeling97c72b72016-06-24 16:19:36 +01002878 const Register zero = vixl::aarch64::Assembler::AppropriateZeroRegFor(out);
Anton Kirilova3ffea22016-04-07 17:02:37 +01002879
2880 MoveFPToInt(locations, is64bit, masm);
2881 __ Eor(out, out, infinity);
2882 // We don't care about the sign bit, so shift left.
2883 __ Cmp(zero, Operand(out, LSL, 1));
2884 __ Cset(out, eq);
2885}
2886
2887void IntrinsicLocationsBuilderARM64::VisitFloatIsInfinite(HInvoke* invoke) {
2888 CreateFPToIntLocations(arena_, invoke);
2889}
2890
2891void IntrinsicCodeGeneratorARM64::VisitFloatIsInfinite(HInvoke* invoke) {
2892 GenIsInfinite(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
2893}
2894
2895void IntrinsicLocationsBuilderARM64::VisitDoubleIsInfinite(HInvoke* invoke) {
2896 CreateFPToIntLocations(arena_, invoke);
2897}
2898
2899void IntrinsicCodeGeneratorARM64::VisitDoubleIsInfinite(HInvoke* invoke) {
2900 GenIsInfinite(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
2901}
2902
TatWai Chongd8c052a2016-11-02 16:12:48 +08002903void IntrinsicLocationsBuilderARM64::VisitReferenceGetReferent(HInvoke* invoke) {
2904 if (kEmitCompilerReadBarrier) {
2905 // Do not intrinsify this call with the read barrier configuration.
2906 return;
2907 }
2908 LocationSummary* locations = new (arena_) LocationSummary(invoke,
2909 LocationSummary::kCallOnSlowPath,
2910 kIntrinsified);
2911 locations->SetInAt(0, Location::RequiresRegister());
2912 locations->SetOut(Location::SameAsFirstInput());
2913 locations->AddTemp(Location::RequiresRegister());
2914}
2915
2916void IntrinsicCodeGeneratorARM64::VisitReferenceGetReferent(HInvoke* invoke) {
2917 DCHECK(!kEmitCompilerReadBarrier);
2918 MacroAssembler* masm = GetVIXLAssembler();
2919 LocationSummary* locations = invoke->GetLocations();
2920
2921 Register obj = InputRegisterAt(invoke, 0);
2922 Register out = OutputRegister(invoke);
2923
2924 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
2925 codegen_->AddSlowPath(slow_path);
2926
2927 // Load ArtMethod first.
2928 HInvokeStaticOrDirect* invoke_direct = invoke->AsInvokeStaticOrDirect();
2929 DCHECK(invoke_direct != nullptr);
2930 Register temp0 = XRegisterFrom(codegen_->GenerateCalleeMethodStaticOrDirectCall(
2931 invoke_direct, locations->GetTemp(0)));
2932
2933 // Now get declaring class.
2934 __ Ldr(temp0.W(), MemOperand(temp0, ArtMethod::DeclaringClassOffset().Int32Value()));
2935
2936 uint32_t slow_path_flag_offset = codegen_->GetReferenceSlowFlagOffset();
2937 uint32_t disable_flag_offset = codegen_->GetReferenceDisableFlagOffset();
2938 DCHECK_NE(slow_path_flag_offset, 0u);
2939 DCHECK_NE(disable_flag_offset, 0u);
2940 DCHECK_NE(slow_path_flag_offset, disable_flag_offset);
2941
2942 // Check static flags that prevent using intrinsic.
2943 if (slow_path_flag_offset == disable_flag_offset + 1) {
2944 // Load two adjacent flags in one 64-bit load.
2945 __ Ldr(temp0, MemOperand(temp0, disable_flag_offset));
2946 } else {
2947 UseScratchRegisterScope temps(masm);
2948 Register temp1 = temps.AcquireW();
2949 __ Ldr(temp1.W(), MemOperand(temp0, disable_flag_offset));
2950 __ Ldr(temp0.W(), MemOperand(temp0, slow_path_flag_offset));
2951 __ Orr(temp0, temp1, temp0);
2952 }
2953 __ Cbnz(temp0, slow_path->GetEntryLabel());
2954
Artem Serov914d7a82017-02-07 14:33:49 +00002955 {
2956 // Ensure that between load and MaybeRecordImplicitNullCheck there are no pools emitted.
2957 vixl::EmissionCheckScope guard(codegen_->GetVIXLAssembler(), kMaxMacroInstructionSizeInBytes);
2958 // Fast path.
2959 __ Ldr(out, HeapOperand(obj, mirror::Reference::ReferentOffset().Int32Value()));
2960 codegen_->MaybeRecordImplicitNullCheck(invoke);
2961 }
TatWai Chongd8c052a2016-11-02 16:12:48 +08002962 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(out);
2963 __ Bind(slow_path->GetExitLabel());
2964}
2965
Nicolas Geoffray331605a2017-03-01 11:01:41 +00002966void IntrinsicLocationsBuilderARM64::VisitIntegerValueOf(HInvoke* invoke) {
2967 InvokeRuntimeCallingConvention calling_convention;
2968 IntrinsicVisitor::ComputeIntegerValueOfLocations(
2969 invoke,
2970 codegen_,
2971 calling_convention.GetReturnLocation(Primitive::kPrimNot),
2972 Location::RegisterLocation(calling_convention.GetRegisterAt(0).GetCode()));
2973}
2974
2975void IntrinsicCodeGeneratorARM64::VisitIntegerValueOf(HInvoke* invoke) {
2976 IntrinsicVisitor::IntegerValueOfInfo info = IntrinsicVisitor::ComputeIntegerValueOfInfo();
2977 LocationSummary* locations = invoke->GetLocations();
2978 MacroAssembler* masm = GetVIXLAssembler();
2979
2980 Register out = RegisterFrom(locations->Out(), Primitive::kPrimNot);
2981 UseScratchRegisterScope temps(masm);
2982 Register temp = temps.AcquireW();
2983 InvokeRuntimeCallingConvention calling_convention;
2984 Register argument = calling_convention.GetRegisterAt(0);
2985 if (invoke->InputAt(0)->IsConstant()) {
2986 int32_t value = invoke->InputAt(0)->AsIntConstant()->GetValue();
2987 if (value >= info.low && value <= info.high) {
2988 // Just embed the j.l.Integer in the code.
2989 ScopedObjectAccess soa(Thread::Current());
2990 mirror::Object* boxed = info.cache->Get(value + (-info.low));
2991 DCHECK(boxed != nullptr && Runtime::Current()->GetHeap()->ObjectIsInBootImageSpace(boxed));
2992 uint32_t address = dchecked_integral_cast<uint32_t>(reinterpret_cast<uintptr_t>(boxed));
2993 __ Ldr(out.W(), codegen_->DeduplicateBootImageAddressLiteral(address));
2994 } else {
2995 // Allocate and initialize a new j.l.Integer.
2996 // TODO: If we JIT, we could allocate the j.l.Integer now, and store it in the
2997 // JIT object table.
2998 uint32_t address =
2999 dchecked_integral_cast<uint32_t>(reinterpret_cast<uintptr_t>(info.integer));
3000 __ Ldr(argument.W(), codegen_->DeduplicateBootImageAddressLiteral(address));
3001 codegen_->InvokeRuntime(kQuickAllocObjectInitialized, invoke, invoke->GetDexPc());
3002 CheckEntrypointTypes<kQuickAllocObjectWithChecks, void*, mirror::Class*>();
3003 __ Mov(temp.W(), value);
3004 __ Str(temp.W(), HeapOperand(out.W(), info.value_offset));
3005 // `value` is a final field :-( Ideally, we'd merge this memory barrier with the allocation
3006 // one.
3007 codegen_->GenerateMemoryBarrier(MemBarrierKind::kStoreStore);
3008 }
3009 } else {
3010 Register in = RegisterFrom(locations->InAt(0), Primitive::kPrimInt);
3011 // Check bounds of our cache.
3012 __ Add(out.W(), in.W(), -info.low);
3013 __ Cmp(out.W(), info.high - info.low + 1);
3014 vixl::aarch64::Label allocate, done;
3015 __ B(&allocate, hs);
3016 // If the value is within the bounds, load the j.l.Integer directly from the array.
3017 uint32_t data_offset = mirror::Array::DataOffset(kHeapReferenceSize).Uint32Value();
3018 uint32_t address = dchecked_integral_cast<uint32_t>(reinterpret_cast<uintptr_t>(info.cache));
3019 __ Ldr(temp.W(), codegen_->DeduplicateBootImageAddressLiteral(data_offset + address));
3020 MemOperand source = HeapOperand(
3021 temp, out.X(), LSL, Primitive::ComponentSizeShift(Primitive::kPrimNot));
3022 codegen_->Load(Primitive::kPrimNot, out, source);
3023 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(out);
3024 __ B(&done);
3025 __ Bind(&allocate);
3026 // Otherwise allocate and initialize a new j.l.Integer.
3027 address = dchecked_integral_cast<uint32_t>(reinterpret_cast<uintptr_t>(info.integer));
3028 __ Ldr(argument.W(), codegen_->DeduplicateBootImageAddressLiteral(address));
3029 codegen_->InvokeRuntime(kQuickAllocObjectInitialized, invoke, invoke->GetDexPc());
3030 CheckEntrypointTypes<kQuickAllocObjectWithChecks, void*, mirror::Class*>();
3031 __ Str(in.W(), HeapOperand(out.W(), info.value_offset));
3032 // `value` is a final field :-( Ideally, we'd merge this memory barrier with the allocation
3033 // one.
3034 codegen_->GenerateMemoryBarrier(MemBarrierKind::kStoreStore);
3035 __ Bind(&done);
3036 }
3037}
3038
Aart Bik2f9fcc92016-03-01 15:16:54 -08003039UNIMPLEMENTED_INTRINSIC(ARM64, IntegerHighestOneBit)
3040UNIMPLEMENTED_INTRINSIC(ARM64, LongHighestOneBit)
3041UNIMPLEMENTED_INTRINSIC(ARM64, IntegerLowestOneBit)
3042UNIMPLEMENTED_INTRINSIC(ARM64, LongLowestOneBit)
Andreas Gampe878d58c2015-01-15 23:24:00 -08003043
Aart Bikff7d89c2016-11-07 08:49:28 -08003044UNIMPLEMENTED_INTRINSIC(ARM64, StringStringIndexOf);
3045UNIMPLEMENTED_INTRINSIC(ARM64, StringStringIndexOfAfter);
Aart Bik71bf7b42016-11-16 10:17:46 -08003046UNIMPLEMENTED_INTRINSIC(ARM64, StringBufferAppend);
3047UNIMPLEMENTED_INTRINSIC(ARM64, StringBufferLength);
3048UNIMPLEMENTED_INTRINSIC(ARM64, StringBufferToString);
3049UNIMPLEMENTED_INTRINSIC(ARM64, StringBuilderAppend);
3050UNIMPLEMENTED_INTRINSIC(ARM64, StringBuilderLength);
3051UNIMPLEMENTED_INTRINSIC(ARM64, StringBuilderToString);
Aart Bikff7d89c2016-11-07 08:49:28 -08003052
Aart Bik0e54c012016-03-04 12:08:31 -08003053// 1.8.
3054UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndAddInt)
3055UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndAddLong)
3056UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetInt)
3057UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetLong)
3058UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetObject)
Aart Bik0e54c012016-03-04 12:08:31 -08003059
Aart Bik2f9fcc92016-03-01 15:16:54 -08003060UNREACHABLE_INTRINSICS(ARM64)
Roland Levillain4d027112015-07-01 15:41:14 +01003061
3062#undef __
3063
Andreas Gampe878d58c2015-01-15 23:24:00 -08003064} // namespace arm64
3065} // namespace art