blob: db1c022868065b3dd07da87d637bbee05d589f09 [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"
25#include "mirror/array-inl.h"
Andreas Gampe878d58c2015-01-15 23:24:00 -080026#include "mirror/string.h"
27#include "thread.h"
28#include "utils/arm64/assembler_arm64.h"
Andreas Gampe878d58c2015-01-15 23:24:00 -080029
Scott Wakeling97c72b72016-06-24 16:19:36 +010030using namespace vixl::aarch64; // NOLINT(build/namespaces)
Andreas Gampe878d58c2015-01-15 23:24:00 -080031
Artem Serovaf4e42a2016-08-08 15:11:24 +010032// TODO(VIXL): Make VIXL compile with -Wshadow.
Scott Wakeling97c72b72016-06-24 16:19:36 +010033#pragma GCC diagnostic push
34#pragma GCC diagnostic ignored "-Wshadow"
Artem Serovaf4e42a2016-08-08 15:11:24 +010035#include "aarch64/disasm-aarch64.h"
36#include "aarch64/macro-assembler-aarch64.h"
Scott Wakeling97c72b72016-06-24 16:19:36 +010037#pragma GCC diagnostic pop
Andreas Gampe878d58c2015-01-15 23:24:00 -080038
39namespace art {
40
41namespace arm64 {
42
43using helpers::DRegisterFrom;
44using helpers::FPRegisterFrom;
45using helpers::HeapOperand;
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +000046using helpers::LocationFrom;
Scott Wakeling9ee23f42015-07-23 10:44:35 +010047using helpers::OperandFrom;
Andreas Gampe878d58c2015-01-15 23:24:00 -080048using helpers::RegisterFrom;
49using helpers::SRegisterFrom;
50using helpers::WRegisterFrom;
51using helpers::XRegisterFrom;
xueliang.zhong49924c92016-03-03 10:52:51 +000052using helpers::InputRegisterAt;
Scott Wakeling1f36f412016-04-21 11:13:45 +010053using helpers::OutputRegister;
Andreas Gampe878d58c2015-01-15 23:24:00 -080054
Andreas Gampe878d58c2015-01-15 23:24:00 -080055namespace {
56
57ALWAYS_INLINE inline MemOperand AbsoluteHeapOperandFrom(Location location, size_t offset = 0) {
58 return MemOperand(XRegisterFrom(location), offset);
59}
60
61} // namespace
62
Scott Wakeling97c72b72016-06-24 16:19:36 +010063MacroAssembler* IntrinsicCodeGeneratorARM64::GetVIXLAssembler() {
Alexandre Rames087930f2016-08-02 13:45:28 +010064 return codegen_->GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -080065}
66
67ArenaAllocator* IntrinsicCodeGeneratorARM64::GetAllocator() {
68 return codegen_->GetGraph()->GetArena();
69}
70
Alexandre Rames087930f2016-08-02 13:45:28 +010071#define __ codegen->GetVIXLAssembler()->
Andreas Gampe878d58c2015-01-15 23:24:00 -080072
73static void MoveFromReturnRegister(Location trg,
74 Primitive::Type type,
75 CodeGeneratorARM64* codegen) {
76 if (!trg.IsValid()) {
77 DCHECK(type == Primitive::kPrimVoid);
78 return;
79 }
80
81 DCHECK_NE(type, Primitive::kPrimVoid);
82
Jeff Hao848f70a2014-01-15 13:49:50 -080083 if (Primitive::IsIntegralType(type) || type == Primitive::kPrimNot) {
Andreas Gampe878d58c2015-01-15 23:24:00 -080084 Register trg_reg = RegisterFrom(trg, type);
85 Register res_reg = RegisterFrom(ARM64ReturnLocation(type), type);
86 __ Mov(trg_reg, res_reg, kDiscardForSameWReg);
87 } else {
88 FPRegister trg_reg = FPRegisterFrom(trg, type);
89 FPRegister res_reg = FPRegisterFrom(ARM64ReturnLocation(type), type);
90 __ Fmov(trg_reg, res_reg);
91 }
92}
93
Roland Levillainec525fc2015-04-28 15:50:20 +010094static void MoveArguments(HInvoke* invoke, CodeGeneratorARM64* codegen) {
Roland Levillain2d27c8e2015-04-28 15:48:45 +010095 InvokeDexCallingConventionVisitorARM64 calling_convention_visitor;
Roland Levillainec525fc2015-04-28 15:50:20 +010096 IntrinsicVisitor::MoveArguments(invoke, codegen, &calling_convention_visitor);
Andreas Gampe878d58c2015-01-15 23:24:00 -080097}
98
99// Slow-path for fallback (calling the managed code to handle the intrinsic) in an intrinsified
100// call. This will copy the arguments into the positions for a regular call.
101//
102// Note: The actual parameters are required to be in the locations given by the invoke's location
103// summary. If an intrinsic modifies those locations before a slowpath call, they must be
104// restored!
105class IntrinsicSlowPathARM64 : public SlowPathCodeARM64 {
106 public:
David Srbecky9cd6d372016-02-09 15:24:47 +0000107 explicit IntrinsicSlowPathARM64(HInvoke* invoke)
108 : SlowPathCodeARM64(invoke), invoke_(invoke) { }
Andreas Gampe878d58c2015-01-15 23:24:00 -0800109
110 void EmitNativeCode(CodeGenerator* codegen_in) OVERRIDE {
111 CodeGeneratorARM64* codegen = down_cast<CodeGeneratorARM64*>(codegen_in);
112 __ Bind(GetEntryLabel());
113
Nicolas Geoffraya8ac9132015-03-13 16:36:36 +0000114 SaveLiveRegisters(codegen, invoke_->GetLocations());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800115
Roland Levillainec525fc2015-04-28 15:50:20 +0100116 MoveArguments(invoke_, codegen);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800117
118 if (invoke_->IsInvokeStaticOrDirect()) {
Nicolas Geoffray94015b92015-06-04 18:21:04 +0100119 codegen->GenerateStaticOrDirectCall(invoke_->AsInvokeStaticOrDirect(),
120 LocationFrom(kArtMethodRegister));
Andreas Gampe878d58c2015-01-15 23:24:00 -0800121 } else {
Andreas Gampebfb5ba92015-09-01 15:45:02 +0000122 codegen->GenerateVirtualCall(invoke_->AsInvokeVirtual(), LocationFrom(kArtMethodRegister));
Andreas Gampe878d58c2015-01-15 23:24:00 -0800123 }
Andreas Gampebfb5ba92015-09-01 15:45:02 +0000124 codegen->RecordPcInfo(invoke_, invoke_->GetDexPc(), this);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800125
126 // Copy the result back to the expected output.
127 Location out = invoke_->GetLocations()->Out();
128 if (out.IsValid()) {
129 DCHECK(out.IsRegister()); // TODO: Replace this when we support output in memory.
130 DCHECK(!invoke_->GetLocations()->GetLiveRegisters()->ContainsCoreRegister(out.reg()));
131 MoveFromReturnRegister(out, invoke_->GetType(), codegen);
132 }
133
Nicolas Geoffraya8ac9132015-03-13 16:36:36 +0000134 RestoreLiveRegisters(codegen, invoke_->GetLocations());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800135 __ B(GetExitLabel());
136 }
137
Alexandre Rames9931f312015-06-19 14:47:01 +0100138 const char* GetDescription() const OVERRIDE { return "IntrinsicSlowPathARM64"; }
139
Andreas Gampe878d58c2015-01-15 23:24:00 -0800140 private:
141 // The instruction where this slow path is happening.
142 HInvoke* const invoke_;
143
144 DISALLOW_COPY_AND_ASSIGN(IntrinsicSlowPathARM64);
145};
146
Roland Levillain0b671c02016-08-19 12:02:34 +0100147// Slow path implementing the SystemArrayCopy intrinsic copy loop with read barriers.
148class ReadBarrierSystemArrayCopySlowPathARM64 : public SlowPathCodeARM64 {
149 public:
150 ReadBarrierSystemArrayCopySlowPathARM64(HInstruction* instruction, Location tmp)
151 : SlowPathCodeARM64(instruction), tmp_(tmp) {
152 DCHECK(kEmitCompilerReadBarrier);
153 DCHECK(kUseBakerReadBarrier);
154 }
155
156 void EmitNativeCode(CodeGenerator* codegen_in) OVERRIDE {
157 CodeGeneratorARM64* codegen = down_cast<CodeGeneratorARM64*>(codegen_in);
158 LocationSummary* locations = instruction_->GetLocations();
159 DCHECK(locations->CanCall());
160 DCHECK(instruction_->IsInvokeStaticOrDirect())
161 << "Unexpected instruction in read barrier arraycopy slow path: "
162 << instruction_->DebugName();
163 DCHECK(instruction_->GetLocations()->Intrinsified());
164 DCHECK_EQ(instruction_->AsInvoke()->GetIntrinsic(), Intrinsics::kSystemArrayCopy);
165
166 const int32_t element_size = Primitive::ComponentSize(Primitive::kPrimNot);
167
168 Register src_curr_addr = XRegisterFrom(locations->GetTemp(0));
169 Register dst_curr_addr = XRegisterFrom(locations->GetTemp(1));
170 Register src_stop_addr = XRegisterFrom(locations->GetTemp(2));
171 Register tmp_reg = WRegisterFrom(tmp_);
172
173 __ Bind(GetEntryLabel());
174 vixl::aarch64::Label slow_copy_loop;
175 __ Bind(&slow_copy_loop);
176 __ Ldr(tmp_reg, MemOperand(src_curr_addr, element_size, PostIndex));
177 codegen->GetAssembler()->MaybeUnpoisonHeapReference(tmp_reg);
178 // TODO: Inline the mark bit check before calling the runtime?
179 // tmp_reg = ReadBarrier::Mark(tmp_reg);
180 // No need to save live registers; it's taken care of by the
181 // entrypoint. Also, there is no need to update the stack mask,
182 // as this runtime call will not trigger a garbage collection.
183 // (See ReadBarrierMarkSlowPathARM64::EmitNativeCode for more
184 // explanations.)
185 DCHECK_NE(tmp_.reg(), LR);
186 DCHECK_NE(tmp_.reg(), WSP);
187 DCHECK_NE(tmp_.reg(), WZR);
188 // IP0 is used internally by the ReadBarrierMarkRegX entry point
189 // as a temporary (and not preserved). It thus cannot be used by
190 // any live register in this slow path.
191 DCHECK_NE(LocationFrom(src_curr_addr).reg(), IP0);
192 DCHECK_NE(LocationFrom(dst_curr_addr).reg(), IP0);
193 DCHECK_NE(LocationFrom(src_stop_addr).reg(), IP0);
194 DCHECK_NE(tmp_.reg(), IP0);
195 DCHECK(0 <= tmp_.reg() && tmp_.reg() < kNumberOfWRegisters) << tmp_.reg();
196 int32_t entry_point_offset =
197 CodeGenerator::GetReadBarrierMarkEntryPointsOffset<kArm64PointerSize>(tmp_.reg());
198 // This runtime call does not require a stack map.
199 codegen->InvokeRuntimeWithoutRecordingPcInfo(entry_point_offset, instruction_, this);
200 codegen->GetAssembler()->MaybePoisonHeapReference(tmp_reg);
201 __ Str(tmp_reg, MemOperand(dst_curr_addr, element_size, PostIndex));
202 __ Cmp(src_curr_addr, src_stop_addr);
203 __ B(&slow_copy_loop, ne);
204 __ B(GetExitLabel());
205 }
206
207 const char* GetDescription() const OVERRIDE { return "ReadBarrierSystemArrayCopySlowPathARM64"; }
208
209 private:
210 Location tmp_;
211
212 DISALLOW_COPY_AND_ASSIGN(ReadBarrierSystemArrayCopySlowPathARM64);
213};
Andreas Gampe878d58c2015-01-15 23:24:00 -0800214#undef __
215
216bool IntrinsicLocationsBuilderARM64::TryDispatch(HInvoke* invoke) {
217 Dispatch(invoke);
218 LocationSummary* res = invoke->GetLocations();
Roland Levillain22ccc3a2015-11-24 13:10:05 +0000219 if (res == nullptr) {
220 return false;
221 }
Roland Levillain22ccc3a2015-11-24 13:10:05 +0000222 return res->Intrinsified();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800223}
224
225#define __ masm->
226
227static void CreateFPToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
228 LocationSummary* locations = new (arena) LocationSummary(invoke,
229 LocationSummary::kNoCall,
230 kIntrinsified);
231 locations->SetInAt(0, Location::RequiresFpuRegister());
232 locations->SetOut(Location::RequiresRegister());
233}
234
235static void CreateIntToFPLocations(ArenaAllocator* arena, HInvoke* invoke) {
236 LocationSummary* locations = new (arena) LocationSummary(invoke,
237 LocationSummary::kNoCall,
238 kIntrinsified);
239 locations->SetInAt(0, Location::RequiresRegister());
240 locations->SetOut(Location::RequiresFpuRegister());
241}
242
Scott Wakeling97c72b72016-06-24 16:19:36 +0100243static void MoveFPToInt(LocationSummary* locations, bool is64bit, MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800244 Location input = locations->InAt(0);
245 Location output = locations->Out();
246 __ Fmov(is64bit ? XRegisterFrom(output) : WRegisterFrom(output),
247 is64bit ? DRegisterFrom(input) : SRegisterFrom(input));
248}
249
Scott Wakeling97c72b72016-06-24 16:19:36 +0100250static void MoveIntToFP(LocationSummary* locations, bool is64bit, MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800251 Location input = locations->InAt(0);
252 Location output = locations->Out();
253 __ Fmov(is64bit ? DRegisterFrom(output) : SRegisterFrom(output),
254 is64bit ? XRegisterFrom(input) : WRegisterFrom(input));
255}
256
257void IntrinsicLocationsBuilderARM64::VisitDoubleDoubleToRawLongBits(HInvoke* invoke) {
258 CreateFPToIntLocations(arena_, invoke);
259}
260void IntrinsicLocationsBuilderARM64::VisitDoubleLongBitsToDouble(HInvoke* invoke) {
261 CreateIntToFPLocations(arena_, invoke);
262}
263
264void IntrinsicCodeGeneratorARM64::VisitDoubleDoubleToRawLongBits(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000265 MoveFPToInt(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800266}
267void IntrinsicCodeGeneratorARM64::VisitDoubleLongBitsToDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000268 MoveIntToFP(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800269}
270
271void IntrinsicLocationsBuilderARM64::VisitFloatFloatToRawIntBits(HInvoke* invoke) {
272 CreateFPToIntLocations(arena_, invoke);
273}
274void IntrinsicLocationsBuilderARM64::VisitFloatIntBitsToFloat(HInvoke* invoke) {
275 CreateIntToFPLocations(arena_, invoke);
276}
277
278void IntrinsicCodeGeneratorARM64::VisitFloatFloatToRawIntBits(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000279 MoveFPToInt(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800280}
281void IntrinsicCodeGeneratorARM64::VisitFloatIntBitsToFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000282 MoveIntToFP(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800283}
284
285static void CreateIntToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
286 LocationSummary* locations = new (arena) LocationSummary(invoke,
287 LocationSummary::kNoCall,
288 kIntrinsified);
289 locations->SetInAt(0, Location::RequiresRegister());
290 locations->SetOut(Location::RequiresRegister(), Location::kNoOutputOverlap);
291}
292
293static void GenReverseBytes(LocationSummary* locations,
294 Primitive::Type type,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100295 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800296 Location in = locations->InAt(0);
297 Location out = locations->Out();
298
299 switch (type) {
300 case Primitive::kPrimShort:
301 __ Rev16(WRegisterFrom(out), WRegisterFrom(in));
302 __ Sxth(WRegisterFrom(out), WRegisterFrom(out));
303 break;
304 case Primitive::kPrimInt:
305 case Primitive::kPrimLong:
306 __ Rev(RegisterFrom(out, type), RegisterFrom(in, type));
307 break;
308 default:
309 LOG(FATAL) << "Unexpected size for reverse-bytes: " << type;
310 UNREACHABLE();
311 }
312}
313
314void IntrinsicLocationsBuilderARM64::VisitIntegerReverseBytes(HInvoke* invoke) {
315 CreateIntToIntLocations(arena_, invoke);
316}
317
318void IntrinsicCodeGeneratorARM64::VisitIntegerReverseBytes(HInvoke* invoke) {
319 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimInt, GetVIXLAssembler());
320}
321
322void IntrinsicLocationsBuilderARM64::VisitLongReverseBytes(HInvoke* invoke) {
323 CreateIntToIntLocations(arena_, invoke);
324}
325
326void IntrinsicCodeGeneratorARM64::VisitLongReverseBytes(HInvoke* invoke) {
327 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimLong, GetVIXLAssembler());
328}
329
330void IntrinsicLocationsBuilderARM64::VisitShortReverseBytes(HInvoke* invoke) {
331 CreateIntToIntLocations(arena_, invoke);
332}
333
334void IntrinsicCodeGeneratorARM64::VisitShortReverseBytes(HInvoke* invoke) {
335 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimShort, GetVIXLAssembler());
336}
337
Aart Bik7b565022016-01-28 14:36:22 -0800338static void CreateIntIntToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
339 LocationSummary* locations = new (arena) LocationSummary(invoke,
340 LocationSummary::kNoCall,
341 kIntrinsified);
342 locations->SetInAt(0, Location::RequiresRegister());
343 locations->SetInAt(1, Location::RequiresRegister());
344 locations->SetOut(Location::RequiresRegister(), Location::kNoOutputOverlap);
345}
346
Scott Wakeling611d3392015-07-10 11:42:06 +0100347static void GenNumberOfLeadingZeros(LocationSummary* locations,
348 Primitive::Type type,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100349 MacroAssembler* masm) {
Scott Wakeling611d3392015-07-10 11:42:06 +0100350 DCHECK(type == Primitive::kPrimInt || type == Primitive::kPrimLong);
351
352 Location in = locations->InAt(0);
353 Location out = locations->Out();
354
355 __ Clz(RegisterFrom(out, type), RegisterFrom(in, type));
356}
357
358void IntrinsicLocationsBuilderARM64::VisitIntegerNumberOfLeadingZeros(HInvoke* invoke) {
359 CreateIntToIntLocations(arena_, invoke);
360}
361
362void IntrinsicCodeGeneratorARM64::VisitIntegerNumberOfLeadingZeros(HInvoke* invoke) {
363 GenNumberOfLeadingZeros(invoke->GetLocations(), Primitive::kPrimInt, GetVIXLAssembler());
364}
365
366void IntrinsicLocationsBuilderARM64::VisitLongNumberOfLeadingZeros(HInvoke* invoke) {
367 CreateIntToIntLocations(arena_, invoke);
368}
369
370void IntrinsicCodeGeneratorARM64::VisitLongNumberOfLeadingZeros(HInvoke* invoke) {
371 GenNumberOfLeadingZeros(invoke->GetLocations(), Primitive::kPrimLong, GetVIXLAssembler());
372}
373
Scott Wakeling9ee23f42015-07-23 10:44:35 +0100374static void GenNumberOfTrailingZeros(LocationSummary* locations,
375 Primitive::Type type,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100376 MacroAssembler* masm) {
Scott Wakeling9ee23f42015-07-23 10:44:35 +0100377 DCHECK(type == Primitive::kPrimInt || type == Primitive::kPrimLong);
378
379 Location in = locations->InAt(0);
380 Location out = locations->Out();
381
382 __ Rbit(RegisterFrom(out, type), RegisterFrom(in, type));
383 __ Clz(RegisterFrom(out, type), RegisterFrom(out, type));
384}
385
386void IntrinsicLocationsBuilderARM64::VisitIntegerNumberOfTrailingZeros(HInvoke* invoke) {
387 CreateIntToIntLocations(arena_, invoke);
388}
389
390void IntrinsicCodeGeneratorARM64::VisitIntegerNumberOfTrailingZeros(HInvoke* invoke) {
391 GenNumberOfTrailingZeros(invoke->GetLocations(), Primitive::kPrimInt, GetVIXLAssembler());
392}
393
394void IntrinsicLocationsBuilderARM64::VisitLongNumberOfTrailingZeros(HInvoke* invoke) {
395 CreateIntToIntLocations(arena_, invoke);
396}
397
398void IntrinsicCodeGeneratorARM64::VisitLongNumberOfTrailingZeros(HInvoke* invoke) {
399 GenNumberOfTrailingZeros(invoke->GetLocations(), Primitive::kPrimLong, GetVIXLAssembler());
400}
401
Andreas Gampe878d58c2015-01-15 23:24:00 -0800402static void GenReverse(LocationSummary* locations,
403 Primitive::Type type,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100404 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800405 DCHECK(type == Primitive::kPrimInt || type == Primitive::kPrimLong);
406
407 Location in = locations->InAt(0);
408 Location out = locations->Out();
409
410 __ Rbit(RegisterFrom(out, type), RegisterFrom(in, type));
411}
412
413void IntrinsicLocationsBuilderARM64::VisitIntegerReverse(HInvoke* invoke) {
414 CreateIntToIntLocations(arena_, invoke);
415}
416
417void IntrinsicCodeGeneratorARM64::VisitIntegerReverse(HInvoke* invoke) {
418 GenReverse(invoke->GetLocations(), Primitive::kPrimInt, GetVIXLAssembler());
419}
420
421void IntrinsicLocationsBuilderARM64::VisitLongReverse(HInvoke* invoke) {
422 CreateIntToIntLocations(arena_, invoke);
423}
424
425void IntrinsicCodeGeneratorARM64::VisitLongReverse(HInvoke* invoke) {
426 GenReverse(invoke->GetLocations(), Primitive::kPrimLong, GetVIXLAssembler());
427}
428
Scott Wakeling97c72b72016-06-24 16:19:36 +0100429static void GenBitCount(HInvoke* instr, Primitive::Type type, MacroAssembler* masm) {
Roland Levillainfa3912e2016-04-01 18:21:55 +0100430 DCHECK(Primitive::IsIntOrLongType(type)) << type;
431 DCHECK_EQ(instr->GetType(), Primitive::kPrimInt);
432 DCHECK_EQ(Primitive::PrimitiveKind(instr->InputAt(0)->GetType()), type);
xueliang.zhong49924c92016-03-03 10:52:51 +0000433
xueliang.zhong49924c92016-03-03 10:52:51 +0000434 UseScratchRegisterScope temps(masm);
435
Nicolas Geoffray457413a2016-03-04 11:10:17 +0000436 Register src = InputRegisterAt(instr, 0);
Roland Levillainfa3912e2016-04-01 18:21:55 +0100437 Register dst = RegisterFrom(instr->GetLocations()->Out(), type);
438 FPRegister fpr = (type == Primitive::kPrimLong) ? temps.AcquireD() : temps.AcquireS();
xueliang.zhong49924c92016-03-03 10:52:51 +0000439
440 __ Fmov(fpr, src);
Nicolas Geoffray457413a2016-03-04 11:10:17 +0000441 __ Cnt(fpr.V8B(), fpr.V8B());
442 __ Addv(fpr.B(), fpr.V8B());
xueliang.zhong49924c92016-03-03 10:52:51 +0000443 __ Fmov(dst, fpr);
444}
445
446void IntrinsicLocationsBuilderARM64::VisitLongBitCount(HInvoke* invoke) {
447 CreateIntToIntLocations(arena_, invoke);
448}
449
450void IntrinsicCodeGeneratorARM64::VisitLongBitCount(HInvoke* invoke) {
Roland Levillainfa3912e2016-04-01 18:21:55 +0100451 GenBitCount(invoke, Primitive::kPrimLong, GetVIXLAssembler());
xueliang.zhong49924c92016-03-03 10:52:51 +0000452}
453
454void IntrinsicLocationsBuilderARM64::VisitIntegerBitCount(HInvoke* invoke) {
455 CreateIntToIntLocations(arena_, invoke);
456}
457
458void IntrinsicCodeGeneratorARM64::VisitIntegerBitCount(HInvoke* invoke) {
Roland Levillainfa3912e2016-04-01 18:21:55 +0100459 GenBitCount(invoke, Primitive::kPrimInt, GetVIXLAssembler());
xueliang.zhong49924c92016-03-03 10:52:51 +0000460}
461
Andreas Gampe878d58c2015-01-15 23:24:00 -0800462static void CreateFPToFPLocations(ArenaAllocator* arena, HInvoke* invoke) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800463 LocationSummary* locations = new (arena) LocationSummary(invoke,
464 LocationSummary::kNoCall,
465 kIntrinsified);
466 locations->SetInAt(0, Location::RequiresFpuRegister());
467 locations->SetOut(Location::RequiresFpuRegister(), Location::kNoOutputOverlap);
468}
469
Scott Wakeling97c72b72016-06-24 16:19:36 +0100470static void MathAbsFP(LocationSummary* locations, bool is64bit, MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800471 Location in = locations->InAt(0);
472 Location out = locations->Out();
473
474 FPRegister in_reg = is64bit ? DRegisterFrom(in) : SRegisterFrom(in);
475 FPRegister out_reg = is64bit ? DRegisterFrom(out) : SRegisterFrom(out);
476
477 __ Fabs(out_reg, in_reg);
478}
479
480void IntrinsicLocationsBuilderARM64::VisitMathAbsDouble(HInvoke* invoke) {
481 CreateFPToFPLocations(arena_, invoke);
482}
483
484void IntrinsicCodeGeneratorARM64::VisitMathAbsDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000485 MathAbsFP(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800486}
487
488void IntrinsicLocationsBuilderARM64::VisitMathAbsFloat(HInvoke* invoke) {
489 CreateFPToFPLocations(arena_, invoke);
490}
491
492void IntrinsicCodeGeneratorARM64::VisitMathAbsFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000493 MathAbsFP(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800494}
495
496static void CreateIntToInt(ArenaAllocator* arena, HInvoke* invoke) {
497 LocationSummary* locations = new (arena) LocationSummary(invoke,
498 LocationSummary::kNoCall,
499 kIntrinsified);
500 locations->SetInAt(0, Location::RequiresRegister());
501 locations->SetOut(Location::RequiresRegister(), Location::kNoOutputOverlap);
502}
503
504static void GenAbsInteger(LocationSummary* locations,
505 bool is64bit,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100506 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800507 Location in = locations->InAt(0);
508 Location output = locations->Out();
509
510 Register in_reg = is64bit ? XRegisterFrom(in) : WRegisterFrom(in);
511 Register out_reg = is64bit ? XRegisterFrom(output) : WRegisterFrom(output);
512
513 __ Cmp(in_reg, Operand(0));
514 __ Cneg(out_reg, in_reg, lt);
515}
516
517void IntrinsicLocationsBuilderARM64::VisitMathAbsInt(HInvoke* invoke) {
518 CreateIntToInt(arena_, invoke);
519}
520
521void IntrinsicCodeGeneratorARM64::VisitMathAbsInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000522 GenAbsInteger(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800523}
524
525void IntrinsicLocationsBuilderARM64::VisitMathAbsLong(HInvoke* invoke) {
526 CreateIntToInt(arena_, invoke);
527}
528
529void IntrinsicCodeGeneratorARM64::VisitMathAbsLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000530 GenAbsInteger(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800531}
532
533static void GenMinMaxFP(LocationSummary* locations,
534 bool is_min,
535 bool is_double,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100536 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800537 Location op1 = locations->InAt(0);
538 Location op2 = locations->InAt(1);
539 Location out = locations->Out();
540
541 FPRegister op1_reg = is_double ? DRegisterFrom(op1) : SRegisterFrom(op1);
542 FPRegister op2_reg = is_double ? DRegisterFrom(op2) : SRegisterFrom(op2);
543 FPRegister out_reg = is_double ? DRegisterFrom(out) : SRegisterFrom(out);
544 if (is_min) {
545 __ Fmin(out_reg, op1_reg, op2_reg);
546 } else {
547 __ Fmax(out_reg, op1_reg, op2_reg);
548 }
549}
550
551static void CreateFPFPToFPLocations(ArenaAllocator* arena, HInvoke* invoke) {
552 LocationSummary* locations = new (arena) LocationSummary(invoke,
553 LocationSummary::kNoCall,
554 kIntrinsified);
555 locations->SetInAt(0, Location::RequiresFpuRegister());
556 locations->SetInAt(1, Location::RequiresFpuRegister());
557 locations->SetOut(Location::RequiresFpuRegister(), Location::kNoOutputOverlap);
558}
559
560void IntrinsicLocationsBuilderARM64::VisitMathMinDoubleDouble(HInvoke* invoke) {
561 CreateFPFPToFPLocations(arena_, invoke);
562}
563
564void IntrinsicCodeGeneratorARM64::VisitMathMinDoubleDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000565 GenMinMaxFP(invoke->GetLocations(), /* is_min */ true, /* is_double */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800566}
567
568void IntrinsicLocationsBuilderARM64::VisitMathMinFloatFloat(HInvoke* invoke) {
569 CreateFPFPToFPLocations(arena_, invoke);
570}
571
572void IntrinsicCodeGeneratorARM64::VisitMathMinFloatFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000573 GenMinMaxFP(invoke->GetLocations(), /* is_min */ true, /* is_double */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800574}
575
576void IntrinsicLocationsBuilderARM64::VisitMathMaxDoubleDouble(HInvoke* invoke) {
577 CreateFPFPToFPLocations(arena_, invoke);
578}
579
580void IntrinsicCodeGeneratorARM64::VisitMathMaxDoubleDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000581 GenMinMaxFP(invoke->GetLocations(), /* is_min */ false, /* is_double */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800582}
583
584void IntrinsicLocationsBuilderARM64::VisitMathMaxFloatFloat(HInvoke* invoke) {
585 CreateFPFPToFPLocations(arena_, invoke);
586}
587
588void IntrinsicCodeGeneratorARM64::VisitMathMaxFloatFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000589 GenMinMaxFP(
590 invoke->GetLocations(), /* is_min */ false, /* is_double */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800591}
592
593static void GenMinMax(LocationSummary* locations,
594 bool is_min,
595 bool is_long,
Scott Wakeling97c72b72016-06-24 16:19:36 +0100596 MacroAssembler* masm) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800597 Location op1 = locations->InAt(0);
598 Location op2 = locations->InAt(1);
599 Location out = locations->Out();
600
601 Register op1_reg = is_long ? XRegisterFrom(op1) : WRegisterFrom(op1);
602 Register op2_reg = is_long ? XRegisterFrom(op2) : WRegisterFrom(op2);
603 Register out_reg = is_long ? XRegisterFrom(out) : WRegisterFrom(out);
604
605 __ Cmp(op1_reg, op2_reg);
606 __ Csel(out_reg, op1_reg, op2_reg, is_min ? lt : gt);
607}
608
Andreas Gampe878d58c2015-01-15 23:24:00 -0800609void IntrinsicLocationsBuilderARM64::VisitMathMinIntInt(HInvoke* invoke) {
610 CreateIntIntToIntLocations(arena_, invoke);
611}
612
613void IntrinsicCodeGeneratorARM64::VisitMathMinIntInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000614 GenMinMax(invoke->GetLocations(), /* is_min */ true, /* is_long */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800615}
616
617void IntrinsicLocationsBuilderARM64::VisitMathMinLongLong(HInvoke* invoke) {
618 CreateIntIntToIntLocations(arena_, invoke);
619}
620
621void IntrinsicCodeGeneratorARM64::VisitMathMinLongLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000622 GenMinMax(invoke->GetLocations(), /* is_min */ true, /* is_long */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800623}
624
625void IntrinsicLocationsBuilderARM64::VisitMathMaxIntInt(HInvoke* invoke) {
626 CreateIntIntToIntLocations(arena_, invoke);
627}
628
629void IntrinsicCodeGeneratorARM64::VisitMathMaxIntInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000630 GenMinMax(invoke->GetLocations(), /* is_min */ false, /* is_long */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800631}
632
633void IntrinsicLocationsBuilderARM64::VisitMathMaxLongLong(HInvoke* invoke) {
634 CreateIntIntToIntLocations(arena_, invoke);
635}
636
637void IntrinsicCodeGeneratorARM64::VisitMathMaxLongLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000638 GenMinMax(invoke->GetLocations(), /* is_min */ false, /* is_long */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800639}
640
641void IntrinsicLocationsBuilderARM64::VisitMathSqrt(HInvoke* invoke) {
642 CreateFPToFPLocations(arena_, invoke);
643}
644
645void IntrinsicCodeGeneratorARM64::VisitMathSqrt(HInvoke* invoke) {
646 LocationSummary* locations = invoke->GetLocations();
Scott Wakeling97c72b72016-06-24 16:19:36 +0100647 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800648 __ Fsqrt(DRegisterFrom(locations->Out()), DRegisterFrom(locations->InAt(0)));
649}
650
651void IntrinsicLocationsBuilderARM64::VisitMathCeil(HInvoke* invoke) {
652 CreateFPToFPLocations(arena_, invoke);
653}
654
655void IntrinsicCodeGeneratorARM64::VisitMathCeil(HInvoke* invoke) {
656 LocationSummary* locations = invoke->GetLocations();
Scott Wakeling97c72b72016-06-24 16:19:36 +0100657 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800658 __ Frintp(DRegisterFrom(locations->Out()), DRegisterFrom(locations->InAt(0)));
659}
660
661void IntrinsicLocationsBuilderARM64::VisitMathFloor(HInvoke* invoke) {
662 CreateFPToFPLocations(arena_, invoke);
663}
664
665void IntrinsicCodeGeneratorARM64::VisitMathFloor(HInvoke* invoke) {
666 LocationSummary* locations = invoke->GetLocations();
Scott Wakeling97c72b72016-06-24 16:19:36 +0100667 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800668 __ Frintm(DRegisterFrom(locations->Out()), DRegisterFrom(locations->InAt(0)));
669}
670
671void IntrinsicLocationsBuilderARM64::VisitMathRint(HInvoke* invoke) {
672 CreateFPToFPLocations(arena_, invoke);
673}
674
675void IntrinsicCodeGeneratorARM64::VisitMathRint(HInvoke* invoke) {
676 LocationSummary* locations = invoke->GetLocations();
Scott Wakeling97c72b72016-06-24 16:19:36 +0100677 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800678 __ Frintn(DRegisterFrom(locations->Out()), DRegisterFrom(locations->InAt(0)));
679}
680
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100681static void CreateFPToIntPlusFPTempLocations(ArenaAllocator* arena, HInvoke* invoke) {
Andreas Gampe878d58c2015-01-15 23:24:00 -0800682 LocationSummary* locations = new (arena) LocationSummary(invoke,
683 LocationSummary::kNoCall,
684 kIntrinsified);
685 locations->SetInAt(0, Location::RequiresFpuRegister());
686 locations->SetOut(Location::RequiresRegister());
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100687 locations->AddTemp(Location::RequiresFpuRegister());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800688}
689
Scott Wakeling97c72b72016-06-24 16:19:36 +0100690static void GenMathRound(HInvoke* invoke, bool is_double, vixl::aarch64::MacroAssembler* masm) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100691 // Java 8 API definition for Math.round():
692 // Return the closest long or int to the argument, with ties rounding to positive infinity.
693 //
694 // There is no single instruction in ARMv8 that can support the above definition.
695 // We choose to use FCVTAS here, because it has closest semantic.
696 // FCVTAS performs rounding to nearest integer, ties away from zero.
697 // For most inputs (positive values, zero or NaN), this instruction is enough.
698 // We only need a few handling code after FCVTAS if the input is negative half value.
699 //
700 // The reason why we didn't choose FCVTPS instruction here is that
701 // although it performs rounding toward positive infinity, it doesn't perform rounding to nearest.
702 // For example, FCVTPS(-1.9) = -1 and FCVTPS(1.1) = 2.
703 // If we were using this instruction, for most inputs, more handling code would be needed.
704 LocationSummary* l = invoke->GetLocations();
705 FPRegister in_reg = is_double ? DRegisterFrom(l->InAt(0)) : SRegisterFrom(l->InAt(0));
706 FPRegister tmp_fp = is_double ? DRegisterFrom(l->GetTemp(0)) : SRegisterFrom(l->GetTemp(0));
707 Register out_reg = is_double ? XRegisterFrom(l->Out()) : WRegisterFrom(l->Out());
Scott Wakeling97c72b72016-06-24 16:19:36 +0100708 vixl::aarch64::Label done;
Andreas Gampe878d58c2015-01-15 23:24:00 -0800709
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100710 // Round to nearest integer, ties away from zero.
711 __ Fcvtas(out_reg, in_reg);
712
713 // For positive values, zero or NaN inputs, rounding is done.
Scott Wakeling97c72b72016-06-24 16:19:36 +0100714 __ Tbz(out_reg, out_reg.GetSizeInBits() - 1, &done);
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100715
716 // Handle input < 0 cases.
717 // If input is negative but not a tie, previous result (round to nearest) is valid.
718 // If input is a negative tie, out_reg += 1.
719 __ Frinta(tmp_fp, in_reg);
720 __ Fsub(tmp_fp, in_reg, tmp_fp);
721 __ Fcmp(tmp_fp, 0.5);
722 __ Cinc(out_reg, out_reg, eq);
723
724 __ Bind(&done);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800725}
726
727void IntrinsicLocationsBuilderARM64::VisitMathRoundDouble(HInvoke* invoke) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100728 CreateFPToIntPlusFPTempLocations(arena_, invoke);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800729}
730
731void IntrinsicCodeGeneratorARM64::VisitMathRoundDouble(HInvoke* invoke) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100732 GenMathRound(invoke, /* is_double */ true, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800733}
734
735void IntrinsicLocationsBuilderARM64::VisitMathRoundFloat(HInvoke* invoke) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100736 CreateFPToIntPlusFPTempLocations(arena_, invoke);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800737}
738
739void IntrinsicCodeGeneratorARM64::VisitMathRoundFloat(HInvoke* invoke) {
xueliang.zhongd1e153c2016-05-27 18:56:13 +0100740 GenMathRound(invoke, /* is_double */ false, GetVIXLAssembler());
Andreas Gampe878d58c2015-01-15 23:24:00 -0800741}
742
743void IntrinsicLocationsBuilderARM64::VisitMemoryPeekByte(HInvoke* invoke) {
744 CreateIntToIntLocations(arena_, invoke);
745}
746
747void IntrinsicCodeGeneratorARM64::VisitMemoryPeekByte(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100748 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800749 __ Ldrsb(WRegisterFrom(invoke->GetLocations()->Out()),
750 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
751}
752
753void IntrinsicLocationsBuilderARM64::VisitMemoryPeekIntNative(HInvoke* invoke) {
754 CreateIntToIntLocations(arena_, invoke);
755}
756
757void IntrinsicCodeGeneratorARM64::VisitMemoryPeekIntNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100758 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800759 __ Ldr(WRegisterFrom(invoke->GetLocations()->Out()),
760 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
761}
762
763void IntrinsicLocationsBuilderARM64::VisitMemoryPeekLongNative(HInvoke* invoke) {
764 CreateIntToIntLocations(arena_, invoke);
765}
766
767void IntrinsicCodeGeneratorARM64::VisitMemoryPeekLongNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100768 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800769 __ Ldr(XRegisterFrom(invoke->GetLocations()->Out()),
770 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
771}
772
773void IntrinsicLocationsBuilderARM64::VisitMemoryPeekShortNative(HInvoke* invoke) {
774 CreateIntToIntLocations(arena_, invoke);
775}
776
777void IntrinsicCodeGeneratorARM64::VisitMemoryPeekShortNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100778 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800779 __ Ldrsh(WRegisterFrom(invoke->GetLocations()->Out()),
780 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
781}
782
783static void CreateIntIntToVoidLocations(ArenaAllocator* arena, HInvoke* invoke) {
784 LocationSummary* locations = new (arena) LocationSummary(invoke,
785 LocationSummary::kNoCall,
786 kIntrinsified);
787 locations->SetInAt(0, Location::RequiresRegister());
788 locations->SetInAt(1, Location::RequiresRegister());
789}
790
791void IntrinsicLocationsBuilderARM64::VisitMemoryPokeByte(HInvoke* invoke) {
792 CreateIntIntToVoidLocations(arena_, invoke);
793}
794
795void IntrinsicCodeGeneratorARM64::VisitMemoryPokeByte(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100796 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800797 __ Strb(WRegisterFrom(invoke->GetLocations()->InAt(1)),
798 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
799}
800
801void IntrinsicLocationsBuilderARM64::VisitMemoryPokeIntNative(HInvoke* invoke) {
802 CreateIntIntToVoidLocations(arena_, invoke);
803}
804
805void IntrinsicCodeGeneratorARM64::VisitMemoryPokeIntNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100806 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800807 __ Str(WRegisterFrom(invoke->GetLocations()->InAt(1)),
808 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
809}
810
811void IntrinsicLocationsBuilderARM64::VisitMemoryPokeLongNative(HInvoke* invoke) {
812 CreateIntIntToVoidLocations(arena_, invoke);
813}
814
815void IntrinsicCodeGeneratorARM64::VisitMemoryPokeLongNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100816 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800817 __ Str(XRegisterFrom(invoke->GetLocations()->InAt(1)),
818 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
819}
820
821void IntrinsicLocationsBuilderARM64::VisitMemoryPokeShortNative(HInvoke* invoke) {
822 CreateIntIntToVoidLocations(arena_, invoke);
823}
824
825void IntrinsicCodeGeneratorARM64::VisitMemoryPokeShortNative(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +0100826 MacroAssembler* masm = GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800827 __ Strh(WRegisterFrom(invoke->GetLocations()->InAt(1)),
828 AbsoluteHeapOperandFrom(invoke->GetLocations()->InAt(0), 0));
829}
830
831void IntrinsicLocationsBuilderARM64::VisitThreadCurrentThread(HInvoke* invoke) {
832 LocationSummary* locations = new (arena_) LocationSummary(invoke,
833 LocationSummary::kNoCall,
834 kIntrinsified);
835 locations->SetOut(Location::RequiresRegister());
836}
837
838void IntrinsicCodeGeneratorARM64::VisitThreadCurrentThread(HInvoke* invoke) {
839 codegen_->Load(Primitive::kPrimNot, WRegisterFrom(invoke->GetLocations()->Out()),
Andreas Gampe542451c2016-07-26 09:02:02 -0700840 MemOperand(tr, Thread::PeerOffset<kArm64PointerSize>().Int32Value()));
Andreas Gampe878d58c2015-01-15 23:24:00 -0800841}
842
843static void GenUnsafeGet(HInvoke* invoke,
844 Primitive::Type type,
845 bool is_volatile,
846 CodeGeneratorARM64* codegen) {
847 LocationSummary* locations = invoke->GetLocations();
848 DCHECK((type == Primitive::kPrimInt) ||
849 (type == Primitive::kPrimLong) ||
850 (type == Primitive::kPrimNot));
Alexandre Rames087930f2016-08-02 13:45:28 +0100851 MacroAssembler* masm = codegen->GetVIXLAssembler();
Roland Levillain22ccc3a2015-11-24 13:10:05 +0000852 Location base_loc = locations->InAt(1);
853 Register base = WRegisterFrom(base_loc); // Object pointer.
854 Location offset_loc = locations->InAt(2);
855 Register offset = XRegisterFrom(offset_loc); // Long offset.
856 Location trg_loc = locations->Out();
857 Register trg = RegisterFrom(trg_loc, type);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800858
Roland Levillain44015862016-01-22 11:47:17 +0000859 if (type == Primitive::kPrimNot && kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
860 // UnsafeGetObject/UnsafeGetObjectVolatile with Baker's read barrier case.
861 UseScratchRegisterScope temps(masm);
862 Register temp = temps.AcquireW();
Roland Levillainbfea3352016-06-23 13:48:47 +0100863 codegen->GenerateReferenceLoadWithBakerReadBarrier(invoke,
864 trg_loc,
865 base,
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100866 /* offset */ 0u,
Roland Levillainbfea3352016-06-23 13:48:47 +0100867 /* index */ offset_loc,
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100868 /* scale_factor */ 0u,
Roland Levillainbfea3352016-06-23 13:48:47 +0100869 temp,
870 /* needs_null_check */ false,
871 is_volatile);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800872 } else {
Roland Levillain44015862016-01-22 11:47:17 +0000873 // Other cases.
874 MemOperand mem_op(base.X(), offset);
875 if (is_volatile) {
Serban Constantinescu4a6a67c2016-01-27 09:19:56 +0000876 codegen->LoadAcquire(invoke, trg, mem_op, /* needs_null_check */ true);
Roland Levillain44015862016-01-22 11:47:17 +0000877 } else {
878 codegen->Load(type, trg, mem_op);
879 }
Roland Levillain4d027112015-07-01 15:41:14 +0100880
Roland Levillain44015862016-01-22 11:47:17 +0000881 if (type == Primitive::kPrimNot) {
882 DCHECK(trg.IsW());
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100883 codegen->MaybeGenerateReadBarrierSlow(invoke, trg_loc, trg_loc, base_loc, 0u, offset_loc);
Roland Levillain44015862016-01-22 11:47:17 +0000884 }
Roland Levillain4d027112015-07-01 15:41:14 +0100885 }
Andreas Gampe878d58c2015-01-15 23:24:00 -0800886}
887
888static void CreateIntIntIntToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
Roland Levillain22ccc3a2015-11-24 13:10:05 +0000889 bool can_call = kEmitCompilerReadBarrier &&
890 (invoke->GetIntrinsic() == Intrinsics::kUnsafeGetObject ||
891 invoke->GetIntrinsic() == Intrinsics::kUnsafeGetObjectVolatile);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800892 LocationSummary* locations = new (arena) LocationSummary(invoke,
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100893 (can_call
894 ? LocationSummary::kCallOnSlowPath
895 : LocationSummary::kNoCall),
Andreas Gampe878d58c2015-01-15 23:24:00 -0800896 kIntrinsified);
Vladimir Marko70e97462016-08-09 11:04:26 +0100897 if (can_call && kUseBakerReadBarrier) {
Vladimir Marko804b03f2016-09-14 16:26:36 +0100898 locations->SetCustomSlowPathCallerSaves(RegisterSet::Empty()); // No caller-save registers.
Vladimir Marko70e97462016-08-09 11:04:26 +0100899 }
Andreas Gampe878d58c2015-01-15 23:24:00 -0800900 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
901 locations->SetInAt(1, Location::RequiresRegister());
902 locations->SetInAt(2, Location::RequiresRegister());
Roland Levillainbfea3352016-06-23 13:48:47 +0100903 locations->SetOut(Location::RequiresRegister(),
Roland Levillaina1aa3b12016-10-26 13:03:38 +0100904 (can_call ? Location::kOutputOverlap : Location::kNoOutputOverlap));
Andreas Gampe878d58c2015-01-15 23:24:00 -0800905}
906
907void IntrinsicLocationsBuilderARM64::VisitUnsafeGet(HInvoke* invoke) {
908 CreateIntIntIntToIntLocations(arena_, invoke);
909}
910void IntrinsicLocationsBuilderARM64::VisitUnsafeGetVolatile(HInvoke* invoke) {
911 CreateIntIntIntToIntLocations(arena_, invoke);
912}
913void IntrinsicLocationsBuilderARM64::VisitUnsafeGetLong(HInvoke* invoke) {
914 CreateIntIntIntToIntLocations(arena_, invoke);
915}
916void IntrinsicLocationsBuilderARM64::VisitUnsafeGetLongVolatile(HInvoke* invoke) {
917 CreateIntIntIntToIntLocations(arena_, invoke);
918}
919void IntrinsicLocationsBuilderARM64::VisitUnsafeGetObject(HInvoke* invoke) {
920 CreateIntIntIntToIntLocations(arena_, invoke);
921}
922void IntrinsicLocationsBuilderARM64::VisitUnsafeGetObjectVolatile(HInvoke* invoke) {
923 CreateIntIntIntToIntLocations(arena_, invoke);
924}
925
926void IntrinsicCodeGeneratorARM64::VisitUnsafeGet(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000927 GenUnsafeGet(invoke, Primitive::kPrimInt, /* is_volatile */ false, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800928}
929void IntrinsicCodeGeneratorARM64::VisitUnsafeGetVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000930 GenUnsafeGet(invoke, Primitive::kPrimInt, /* is_volatile */ true, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800931}
932void IntrinsicCodeGeneratorARM64::VisitUnsafeGetLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000933 GenUnsafeGet(invoke, Primitive::kPrimLong, /* is_volatile */ false, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800934}
935void IntrinsicCodeGeneratorARM64::VisitUnsafeGetLongVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000936 GenUnsafeGet(invoke, Primitive::kPrimLong, /* is_volatile */ true, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800937}
938void IntrinsicCodeGeneratorARM64::VisitUnsafeGetObject(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000939 GenUnsafeGet(invoke, Primitive::kPrimNot, /* is_volatile */ false, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800940}
941void IntrinsicCodeGeneratorARM64::VisitUnsafeGetObjectVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000942 GenUnsafeGet(invoke, Primitive::kPrimNot, /* is_volatile */ true, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -0800943}
944
945static void CreateIntIntIntIntToVoid(ArenaAllocator* arena, HInvoke* invoke) {
946 LocationSummary* locations = new (arena) LocationSummary(invoke,
947 LocationSummary::kNoCall,
948 kIntrinsified);
949 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
950 locations->SetInAt(1, Location::RequiresRegister());
951 locations->SetInAt(2, Location::RequiresRegister());
952 locations->SetInAt(3, Location::RequiresRegister());
953}
954
955void IntrinsicLocationsBuilderARM64::VisitUnsafePut(HInvoke* invoke) {
956 CreateIntIntIntIntToVoid(arena_, invoke);
957}
958void IntrinsicLocationsBuilderARM64::VisitUnsafePutOrdered(HInvoke* invoke) {
959 CreateIntIntIntIntToVoid(arena_, invoke);
960}
961void IntrinsicLocationsBuilderARM64::VisitUnsafePutVolatile(HInvoke* invoke) {
962 CreateIntIntIntIntToVoid(arena_, invoke);
963}
964void IntrinsicLocationsBuilderARM64::VisitUnsafePutObject(HInvoke* invoke) {
965 CreateIntIntIntIntToVoid(arena_, invoke);
966}
967void IntrinsicLocationsBuilderARM64::VisitUnsafePutObjectOrdered(HInvoke* invoke) {
968 CreateIntIntIntIntToVoid(arena_, invoke);
969}
970void IntrinsicLocationsBuilderARM64::VisitUnsafePutObjectVolatile(HInvoke* invoke) {
971 CreateIntIntIntIntToVoid(arena_, invoke);
972}
973void IntrinsicLocationsBuilderARM64::VisitUnsafePutLong(HInvoke* invoke) {
974 CreateIntIntIntIntToVoid(arena_, invoke);
975}
976void IntrinsicLocationsBuilderARM64::VisitUnsafePutLongOrdered(HInvoke* invoke) {
977 CreateIntIntIntIntToVoid(arena_, invoke);
978}
979void IntrinsicLocationsBuilderARM64::VisitUnsafePutLongVolatile(HInvoke* invoke) {
980 CreateIntIntIntIntToVoid(arena_, invoke);
981}
982
983static void GenUnsafePut(LocationSummary* locations,
984 Primitive::Type type,
985 bool is_volatile,
986 bool is_ordered,
987 CodeGeneratorARM64* codegen) {
Alexandre Rames087930f2016-08-02 13:45:28 +0100988 MacroAssembler* masm = codegen->GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -0800989
990 Register base = WRegisterFrom(locations->InAt(1)); // Object pointer.
991 Register offset = XRegisterFrom(locations->InAt(2)); // Long offset.
992 Register value = RegisterFrom(locations->InAt(3), type);
Roland Levillain4d027112015-07-01 15:41:14 +0100993 Register source = value;
Andreas Gampe878d58c2015-01-15 23:24:00 -0800994 MemOperand mem_op(base.X(), offset);
995
Roland Levillain4d027112015-07-01 15:41:14 +0100996 {
997 // We use a block to end the scratch scope before the write barrier, thus
998 // freeing the temporary registers so they can be used in `MarkGCCard`.
999 UseScratchRegisterScope temps(masm);
1000
1001 if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
1002 DCHECK(value.IsW());
1003 Register temp = temps.AcquireW();
1004 __ Mov(temp.W(), value.W());
1005 codegen->GetAssembler()->PoisonHeapReference(temp.W());
1006 source = temp;
Andreas Gampe878d58c2015-01-15 23:24:00 -08001007 }
Roland Levillain4d027112015-07-01 15:41:14 +01001008
1009 if (is_volatile || is_ordered) {
Serban Constantinescu4a6a67c2016-01-27 09:19:56 +00001010 codegen->StoreRelease(type, source, mem_op);
Roland Levillain4d027112015-07-01 15:41:14 +01001011 } else {
1012 codegen->Store(type, source, mem_op);
1013 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001014 }
1015
1016 if (type == Primitive::kPrimNot) {
Nicolas Geoffray07276db2015-05-18 14:22:09 +01001017 bool value_can_be_null = true; // TODO: Worth finding out this information?
1018 codegen->MarkGCCard(base, value, value_can_be_null);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001019 }
1020}
1021
1022void IntrinsicCodeGeneratorARM64::VisitUnsafePut(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001023 GenUnsafePut(invoke->GetLocations(),
1024 Primitive::kPrimInt,
1025 /* is_volatile */ false,
1026 /* is_ordered */ false,
1027 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001028}
1029void IntrinsicCodeGeneratorARM64::VisitUnsafePutOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001030 GenUnsafePut(invoke->GetLocations(),
1031 Primitive::kPrimInt,
1032 /* is_volatile */ false,
1033 /* is_ordered */ true,
1034 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001035}
1036void IntrinsicCodeGeneratorARM64::VisitUnsafePutVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001037 GenUnsafePut(invoke->GetLocations(),
1038 Primitive::kPrimInt,
1039 /* is_volatile */ true,
1040 /* is_ordered */ false,
1041 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001042}
1043void IntrinsicCodeGeneratorARM64::VisitUnsafePutObject(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001044 GenUnsafePut(invoke->GetLocations(),
1045 Primitive::kPrimNot,
1046 /* is_volatile */ false,
1047 /* is_ordered */ false,
1048 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001049}
1050void IntrinsicCodeGeneratorARM64::VisitUnsafePutObjectOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001051 GenUnsafePut(invoke->GetLocations(),
1052 Primitive::kPrimNot,
1053 /* is_volatile */ false,
1054 /* is_ordered */ true,
1055 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001056}
1057void IntrinsicCodeGeneratorARM64::VisitUnsafePutObjectVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001058 GenUnsafePut(invoke->GetLocations(),
1059 Primitive::kPrimNot,
1060 /* is_volatile */ true,
1061 /* is_ordered */ false,
1062 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001063}
1064void IntrinsicCodeGeneratorARM64::VisitUnsafePutLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001065 GenUnsafePut(invoke->GetLocations(),
1066 Primitive::kPrimLong,
1067 /* is_volatile */ false,
1068 /* is_ordered */ false,
1069 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001070}
1071void IntrinsicCodeGeneratorARM64::VisitUnsafePutLongOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001072 GenUnsafePut(invoke->GetLocations(),
1073 Primitive::kPrimLong,
1074 /* is_volatile */ false,
1075 /* is_ordered */ true,
1076 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001077}
1078void IntrinsicCodeGeneratorARM64::VisitUnsafePutLongVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001079 GenUnsafePut(invoke->GetLocations(),
1080 Primitive::kPrimLong,
1081 /* is_volatile */ true,
1082 /* is_ordered */ false,
1083 codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001084}
1085
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001086static void CreateIntIntIntIntIntToInt(ArenaAllocator* arena,
1087 HInvoke* invoke,
1088 Primitive::Type type) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001089 bool can_call = kEmitCompilerReadBarrier &&
1090 kUseBakerReadBarrier &&
1091 (invoke->GetIntrinsic() == Intrinsics::kUnsafeCASObject);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001092 LocationSummary* locations = new (arena) LocationSummary(invoke,
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001093 (can_call
1094 ? LocationSummary::kCallOnSlowPath
1095 : LocationSummary::kNoCall),
Andreas Gampe878d58c2015-01-15 23:24:00 -08001096 kIntrinsified);
1097 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
1098 locations->SetInAt(1, Location::RequiresRegister());
1099 locations->SetInAt(2, Location::RequiresRegister());
1100 locations->SetInAt(3, Location::RequiresRegister());
1101 locations->SetInAt(4, Location::RequiresRegister());
1102
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001103 // If heap poisoning is enabled, we don't want the unpoisoning
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001104 // operations to potentially clobber the output. Likewise when
1105 // emitting a (Baker) read barrier, which may call.
1106 Location::OutputOverlap overlaps =
1107 ((kPoisonHeapReferences && type == Primitive::kPrimNot) || can_call)
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001108 ? Location::kOutputOverlap
1109 : Location::kNoOutputOverlap;
1110 locations->SetOut(Location::RequiresRegister(), overlaps);
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001111 if (type == Primitive::kPrimNot && kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
1112 // Temporary register for (Baker) read barrier.
1113 locations->AddTemp(Location::RequiresRegister());
1114 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001115}
1116
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001117static void GenCas(HInvoke* invoke, Primitive::Type type, CodeGeneratorARM64* codegen) {
Alexandre Rames087930f2016-08-02 13:45:28 +01001118 MacroAssembler* masm = codegen->GetVIXLAssembler();
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001119 LocationSummary* locations = invoke->GetLocations();
Andreas Gampe878d58c2015-01-15 23:24:00 -08001120
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001121 Location out_loc = locations->Out();
1122 Register out = WRegisterFrom(out_loc); // Boolean result.
Andreas Gampe878d58c2015-01-15 23:24:00 -08001123
1124 Register base = WRegisterFrom(locations->InAt(1)); // Object pointer.
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001125 Location offset_loc = locations->InAt(2);
1126 Register offset = XRegisterFrom(offset_loc); // Long offset.
Andreas Gampe878d58c2015-01-15 23:24:00 -08001127 Register expected = RegisterFrom(locations->InAt(3), type); // Expected.
1128 Register value = RegisterFrom(locations->InAt(4), type); // Value.
1129
1130 // This needs to be before the temp registers, as MarkGCCard also uses VIXL temps.
1131 if (type == Primitive::kPrimNot) {
1132 // Mark card for object assuming new value is stored.
Nicolas Geoffray07276db2015-05-18 14:22:09 +01001133 bool value_can_be_null = true; // TODO: Worth finding out this information?
1134 codegen->MarkGCCard(base, value, value_can_be_null);
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001135
1136 // The only read barrier implementation supporting the
1137 // UnsafeCASObject intrinsic is the Baker-style read barriers.
1138 DCHECK(!kEmitCompilerReadBarrier || kUseBakerReadBarrier);
1139
1140 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
1141 Register temp = WRegisterFrom(locations->GetTemp(0));
1142 // Need to make sure the reference stored in the field is a to-space
1143 // one before attempting the CAS or the CAS could fail incorrectly.
1144 codegen->GenerateReferenceLoadWithBakerReadBarrier(
1145 invoke,
1146 out_loc, // Unused, used only as a "temporary" within the read barrier.
1147 base,
1148 /* offset */ 0u,
1149 /* index */ offset_loc,
1150 /* scale_factor */ 0u,
1151 temp,
1152 /* needs_null_check */ false,
1153 /* use_load_acquire */ false,
1154 /* always_update_field */ true);
1155 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001156 }
1157
1158 UseScratchRegisterScope temps(masm);
1159 Register tmp_ptr = temps.AcquireX(); // Pointer to actual memory.
1160 Register tmp_value = temps.AcquireSameSizeAs(value); // Value in memory.
1161
1162 Register tmp_32 = tmp_value.W();
1163
1164 __ Add(tmp_ptr, base.X(), Operand(offset));
1165
Roland Levillain4d027112015-07-01 15:41:14 +01001166 if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
1167 codegen->GetAssembler()->PoisonHeapReference(expected);
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001168 if (value.Is(expected)) {
1169 // Do not poison `value`, as it is the same register as
1170 // `expected`, which has just been poisoned.
1171 } else {
1172 codegen->GetAssembler()->PoisonHeapReference(value);
1173 }
Roland Levillain4d027112015-07-01 15:41:14 +01001174 }
1175
Andreas Gampe878d58c2015-01-15 23:24:00 -08001176 // do {
1177 // tmp_value = [tmp_ptr] - expected;
1178 // } while (tmp_value == 0 && failure([tmp_ptr] <- r_new_value));
1179 // result = tmp_value != 0;
1180
Scott Wakeling97c72b72016-06-24 16:19:36 +01001181 vixl::aarch64::Label loop_head, exit_loop;
Serban Constantinescu4a6a67c2016-01-27 09:19:56 +00001182 __ Bind(&loop_head);
Serban Constantinescu4a6a67c2016-01-27 09:19:56 +00001183 __ Ldaxr(tmp_value, MemOperand(tmp_ptr));
1184 __ Cmp(tmp_value, expected);
1185 __ B(&exit_loop, ne);
1186 __ Stlxr(tmp_32, value, MemOperand(tmp_ptr));
1187 __ Cbnz(tmp_32, &loop_head);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001188 __ Bind(&exit_loop);
1189 __ Cset(out, eq);
Roland Levillain4d027112015-07-01 15:41:14 +01001190
1191 if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
Roland Levillain4d027112015-07-01 15:41:14 +01001192 codegen->GetAssembler()->UnpoisonHeapReference(expected);
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001193 if (value.Is(expected)) {
1194 // Do not unpoison `value`, as it is the same register as
1195 // `expected`, which has just been unpoisoned.
1196 } else {
1197 codegen->GetAssembler()->UnpoisonHeapReference(value);
1198 }
Roland Levillain4d027112015-07-01 15:41:14 +01001199 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001200}
1201
1202void IntrinsicLocationsBuilderARM64::VisitUnsafeCASInt(HInvoke* invoke) {
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001203 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimInt);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001204}
1205void IntrinsicLocationsBuilderARM64::VisitUnsafeCASLong(HInvoke* invoke) {
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001206 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimLong);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001207}
1208void IntrinsicLocationsBuilderARM64::VisitUnsafeCASObject(HInvoke* invoke) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001209 // The only read barrier implementation supporting the
1210 // UnsafeCASObject intrinsic is the Baker-style read barriers.
1211 if (kEmitCompilerReadBarrier && !kUseBakerReadBarrier) {
Roland Levillain985ff702015-10-23 13:25:35 +01001212 return;
1213 }
1214
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001215 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimNot);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001216}
1217
1218void IntrinsicCodeGeneratorARM64::VisitUnsafeCASInt(HInvoke* invoke) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001219 GenCas(invoke, Primitive::kPrimInt, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001220}
1221void IntrinsicCodeGeneratorARM64::VisitUnsafeCASLong(HInvoke* invoke) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001222 GenCas(invoke, Primitive::kPrimLong, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001223}
1224void IntrinsicCodeGeneratorARM64::VisitUnsafeCASObject(HInvoke* invoke) {
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001225 // The only read barrier implementation supporting the
1226 // UnsafeCASObject intrinsic is the Baker-style read barriers.
1227 DCHECK(!kEmitCompilerReadBarrier || kUseBakerReadBarrier);
Roland Levillain3d312422016-06-23 13:53:42 +01001228
Roland Levillaina1aa3b12016-10-26 13:03:38 +01001229 GenCas(invoke, Primitive::kPrimNot, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001230}
1231
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001232void IntrinsicLocationsBuilderARM64::VisitStringCompareTo(HInvoke* invoke) {
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001233 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Scott Wakeling1f36f412016-04-21 11:13:45 +01001234 invoke->InputAt(1)->CanBeNull()
1235 ? LocationSummary::kCallOnSlowPath
1236 : LocationSummary::kNoCall,
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001237 kIntrinsified);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001238 locations->SetInAt(0, Location::RequiresRegister());
1239 locations->SetInAt(1, Location::RequiresRegister());
1240 locations->AddTemp(Location::RequiresRegister());
1241 locations->AddTemp(Location::RequiresRegister());
1242 locations->AddTemp(Location::RequiresRegister());
jessicahandojo05765752016-09-09 19:01:32 -07001243 // Need temporary registers for String compression's feature.
1244 if (mirror::kUseStringCompression) {
1245 locations->AddTemp(Location::RequiresRegister());
1246 locations->AddTemp(Location::RequiresRegister());
1247 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001248 locations->SetOut(Location::RequiresRegister(), Location::kOutputOverlap);
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001249}
1250
1251void IntrinsicCodeGeneratorARM64::VisitStringCompareTo(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001252 MacroAssembler* masm = GetVIXLAssembler();
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001253 LocationSummary* locations = invoke->GetLocations();
1254
Alexandre Rames2ea91532016-08-11 17:04:14 +01001255 Register str = InputRegisterAt(invoke, 0);
1256 Register arg = InputRegisterAt(invoke, 1);
1257 DCHECK(str.IsW());
1258 DCHECK(arg.IsW());
Scott Wakeling1f36f412016-04-21 11:13:45 +01001259 Register out = OutputRegister(invoke);
1260
1261 Register temp0 = WRegisterFrom(locations->GetTemp(0));
1262 Register temp1 = WRegisterFrom(locations->GetTemp(1));
1263 Register temp2 = WRegisterFrom(locations->GetTemp(2));
jessicahandojo05765752016-09-09 19:01:32 -07001264 Register temp3, temp5;
1265 if (mirror::kUseStringCompression) {
1266 temp3 = WRegisterFrom(locations->GetTemp(3));
1267 temp5 = WRegisterFrom(locations->GetTemp(4));
1268 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001269
Scott Wakeling97c72b72016-06-24 16:19:36 +01001270 vixl::aarch64::Label loop;
1271 vixl::aarch64::Label find_char_diff;
1272 vixl::aarch64::Label end;
jessicahandojo05765752016-09-09 19:01:32 -07001273 vixl::aarch64::Label different_compression;
Scott Wakeling1f36f412016-04-21 11:13:45 +01001274
1275 // Get offsets of count and value fields within a string object.
1276 const int32_t count_offset = mirror::String::CountOffset().Int32Value();
1277 const int32_t value_offset = mirror::String::ValueOffset().Int32Value();
1278
Nicolas Geoffray512e04d2015-03-27 17:21:24 +00001279 // Note that the null check must have been done earlier.
Calin Juravle641547a2015-04-21 22:08:51 +01001280 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001281
Scott Wakeling1f36f412016-04-21 11:13:45 +01001282 // Take slow path and throw if input can be and is null.
1283 SlowPathCodeARM64* slow_path = nullptr;
1284 const bool can_slow_path = invoke->InputAt(1)->CanBeNull();
1285 if (can_slow_path) {
1286 slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1287 codegen_->AddSlowPath(slow_path);
1288 __ Cbz(arg, slow_path->GetEntryLabel());
1289 }
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001290
Scott Wakeling1f36f412016-04-21 11:13:45 +01001291 // Reference equality check, return 0 if same reference.
1292 __ Subs(out, str, arg);
1293 __ B(&end, eq);
jessicahandojo05765752016-09-09 19:01:32 -07001294 if (mirror::kUseStringCompression) {
1295 // Load lengths of this and argument strings.
1296 __ Ldr(temp3, HeapOperand(str, count_offset));
1297 __ Ldr(temp5, HeapOperand(arg, count_offset));
1298 // Clean out compression flag from lengths.
1299 __ Bic(temp0, temp3, Operand(static_cast<int32_t>(0x80000000)));
1300 __ Bic(temp1, temp5, Operand(static_cast<int32_t>(0x80000000)));
1301 } else {
1302 // Load lengths of this and argument strings.
1303 __ Ldr(temp0, HeapOperand(str, count_offset));
1304 __ Ldr(temp1, HeapOperand(arg, count_offset));
1305 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001306 // Return zero if both strings are empty.
1307 __ Orr(out, temp0, temp1);
1308 __ Cbz(out, &end);
1309 // out = length diff.
1310 __ Subs(out, temp0, temp1);
1311 // temp2 = min(len(str), len(arg)).
1312 __ Csel(temp2, temp1, temp0, ge);
1313 // Shorter string is empty?
1314 __ Cbz(temp2, &end);
1315
jessicahandojo05765752016-09-09 19:01:32 -07001316 if (mirror::kUseStringCompression) {
1317 // Check if both strings using same compression style to use this comparison loop.
1318 __ Eor(temp3.W(), temp3, Operand(temp5));
1319 __ Tbnz(temp3.W(), kWRegSize - 1, &different_compression);
1320 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001321 // Store offset of string value in preparation for comparison loop.
1322 __ Mov(temp1, value_offset);
jessicahandojo05765752016-09-09 19:01:32 -07001323 if (mirror::kUseStringCompression) {
1324 // For string compression, calculate the number of bytes to compare (not chars).
1325 // This could be in theory exceed INT32_MAX, so treat temp2 as unsigned.
1326 vixl::aarch64::Label let_it_signed;
1327 __ Cmp(temp5, Operand(0));
1328 __ B(lt, &let_it_signed);
1329 __ Add(temp2, temp2, Operand(temp2));
1330 __ Bind(&let_it_signed);
1331 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001332
1333 UseScratchRegisterScope scratch_scope(masm);
1334 Register temp4 = scratch_scope.AcquireX();
1335
1336 // Assertions that must hold in order to compare strings 4 characters at a time.
1337 DCHECK_ALIGNED(value_offset, 8);
1338 static_assert(IsAligned<8>(kObjectAlignment), "String of odd length is not zero padded");
1339
1340 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1341 DCHECK_EQ(char_size, 2u);
1342
1343 // Promote temp0 to an X reg, ready for LDR.
1344 temp0 = temp0.X();
1345
1346 // Loop to compare 4x16-bit characters at a time (ok because of string data alignment).
1347 __ Bind(&loop);
Alexandre Rames2ea91532016-08-11 17:04:14 +01001348 __ Ldr(temp4, MemOperand(str.X(), temp1.X()));
1349 __ Ldr(temp0, MemOperand(arg.X(), temp1.X()));
Scott Wakeling1f36f412016-04-21 11:13:45 +01001350 __ Cmp(temp4, temp0);
1351 __ B(ne, &find_char_diff);
1352 __ Add(temp1, temp1, char_size * 4);
jessicahandojo05765752016-09-09 19:01:32 -07001353 // With string compression, we have compared 8 bytes, otherwise 4 chars.
1354 __ Subs(temp2, temp2, (mirror::kUseStringCompression) ? 8 : 4);
1355 __ B(hi, &loop);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001356 __ B(&end);
1357
1358 // Promote temp1 to an X reg, ready for EOR.
1359 temp1 = temp1.X();
1360
jessicahandojo05765752016-09-09 19:01:32 -07001361 // Find the single character difference.
Scott Wakeling1f36f412016-04-21 11:13:45 +01001362 __ Bind(&find_char_diff);
1363 // Get the bit position of the first character that differs.
1364 __ Eor(temp1, temp0, temp4);
1365 __ Rbit(temp1, temp1);
1366 __ Clz(temp1, temp1);
jessicahandojo05765752016-09-09 19:01:32 -07001367 // If the number of chars remaining <= the index where the difference occurs (0-3), then
Scott Wakeling1f36f412016-04-21 11:13:45 +01001368 // the difference occurs outside the remaining string data, so just return length diff (out).
jessicahandojo05765752016-09-09 19:01:32 -07001369 // Unlike ARM, we're doing the comparison in one go here, without the subtraction at the
1370 // find_char_diff_2nd_cmp path, so it doesn't matter whether the comparison is signed or
1371 // unsigned when string compression is disabled.
1372 // When it's enabled, the comparison must be unsigned.
1373 __ Cmp(temp2, Operand(temp1.W(), LSR, (mirror::kUseStringCompression) ? 3 : 4));
1374 __ B(ls, &end);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001375 // Extract the characters and calculate the difference.
jessicahandojo05765752016-09-09 19:01:32 -07001376 vixl::aarch64::Label uncompressed_string, continue_process;
1377 if (mirror:: kUseStringCompression) {
1378 __ Tbz(temp5, kWRegSize - 1, &uncompressed_string);
1379 __ Bic(temp1, temp1, 0x7);
1380 __ B(&continue_process);
1381 }
1382 __ Bind(&uncompressed_string);
Scott Wakelinge5ed20b2016-05-20 10:41:38 +01001383 __ Bic(temp1, temp1, 0xf);
jessicahandojo05765752016-09-09 19:01:32 -07001384 __ Bind(&continue_process);
1385
Scott Wakeling1f36f412016-04-21 11:13:45 +01001386 __ Lsr(temp0, temp0, temp1);
1387 __ Lsr(temp4, temp4, temp1);
jessicahandojo05765752016-09-09 19:01:32 -07001388 vixl::aarch64::Label uncompressed_string_extract_chars;
1389 if (mirror::kUseStringCompression) {
1390 __ Tbz(temp5, kWRegSize - 1, &uncompressed_string_extract_chars);
1391 __ And(temp4, temp4, 0xff);
1392 __ Sub(out, temp4.W(), Operand(temp0.W(), UXTB));
1393 __ B(&end);
1394 }
1395 __ Bind(&uncompressed_string_extract_chars);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001396 __ And(temp4, temp4, 0xffff);
Alexandre Rames2ea91532016-08-11 17:04:14 +01001397 __ Sub(out, temp4.W(), Operand(temp0.W(), UXTH));
jessicahandojo05765752016-09-09 19:01:32 -07001398 __ B(&end);
1399
1400 if (mirror::kUseStringCompression) {
1401 vixl::aarch64::Label loop_this_compressed, loop_arg_compressed, find_diff;
1402 const size_t c_char_size = Primitive::ComponentSize(Primitive::kPrimByte);
1403 DCHECK_EQ(c_char_size, 1u);
1404 temp0 = temp0.W();
1405 temp1 = temp1.W();
1406 // Comparison for different compression style.
1407 // This part is when THIS is compressed and ARG is not.
1408 __ Bind(&different_compression);
1409 __ Add(temp0, str, Operand(value_offset));
1410 __ Add(temp1, arg, Operand(value_offset));
1411 __ Cmp(temp5, Operand(0));
1412 __ B(lt, &loop_arg_compressed);
1413
1414 __ Bind(&loop_this_compressed);
1415 __ Ldrb(temp3, MemOperand(temp0.X(), c_char_size, PostIndex));
1416 __ Ldrh(temp5, MemOperand(temp1.X(), char_size, PostIndex));
1417 __ Cmp(temp3, Operand(temp5));
1418 __ B(ne, &find_diff);
1419 __ Subs(temp2, temp2, 1);
1420 __ B(gt, &loop_this_compressed);
1421 __ B(&end);
1422
1423 // This part is when THIS is not compressed and ARG is.
1424 __ Bind(&loop_arg_compressed);
1425 __ Ldrh(temp3, MemOperand(temp0.X(), char_size, PostIndex));
1426 __ Ldrb(temp5, MemOperand(temp1.X(), c_char_size, PostIndex));
1427 __ Cmp(temp3, Operand(temp5));
1428 __ B(ne, &find_diff);
1429 __ Subs(temp2, temp2, 1);
1430 __ B(gt, &loop_arg_compressed);
1431 __ B(&end);
1432
1433 // Calculate the difference.
1434 __ Bind(&find_diff);
1435 __ Sub(out, temp3.W(), Operand(temp5.W(), UXTH));
1436 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001437
1438 __ Bind(&end);
1439
1440 if (can_slow_path) {
1441 __ Bind(slow_path->GetExitLabel());
1442 }
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001443}
1444
Agi Csakiea34b402015-08-13 17:51:19 -07001445void IntrinsicLocationsBuilderARM64::VisitStringEquals(HInvoke* invoke) {
1446 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1447 LocationSummary::kNoCall,
1448 kIntrinsified);
1449 locations->SetInAt(0, Location::RequiresRegister());
1450 locations->SetInAt(1, Location::RequiresRegister());
1451 // Temporary registers to store lengths of strings and for calculations.
1452 locations->AddTemp(Location::RequiresRegister());
1453 locations->AddTemp(Location::RequiresRegister());
1454
1455 locations->SetOut(Location::RequiresRegister(), Location::kOutputOverlap);
1456}
1457
1458void IntrinsicCodeGeneratorARM64::VisitStringEquals(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001459 MacroAssembler* masm = GetVIXLAssembler();
Agi Csakiea34b402015-08-13 17:51:19 -07001460 LocationSummary* locations = invoke->GetLocations();
1461
1462 Register str = WRegisterFrom(locations->InAt(0));
1463 Register arg = WRegisterFrom(locations->InAt(1));
1464 Register out = XRegisterFrom(locations->Out());
1465
1466 UseScratchRegisterScope scratch_scope(masm);
1467 Register temp = scratch_scope.AcquireW();
1468 Register temp1 = WRegisterFrom(locations->GetTemp(0));
1469 Register temp2 = WRegisterFrom(locations->GetTemp(1));
1470
jessicahandojo05765752016-09-09 19:01:32 -07001471 vixl::aarch64::Label loop, preloop;
Scott Wakeling97c72b72016-06-24 16:19:36 +01001472 vixl::aarch64::Label end;
1473 vixl::aarch64::Label return_true;
1474 vixl::aarch64::Label return_false;
Agi Csakiea34b402015-08-13 17:51:19 -07001475
1476 // Get offsets of count, value, and class fields within a string object.
1477 const int32_t count_offset = mirror::String::CountOffset().Int32Value();
1478 const int32_t value_offset = mirror::String::ValueOffset().Int32Value();
1479 const int32_t class_offset = mirror::Object::ClassOffset().Int32Value();
1480
1481 // Note that the null check must have been done earlier.
1482 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1483
Vladimir Marko53b52002016-05-24 19:30:45 +01001484 StringEqualsOptimizations optimizations(invoke);
1485 if (!optimizations.GetArgumentNotNull()) {
1486 // Check if input is null, return false if it is.
1487 __ Cbz(arg, &return_false);
1488 }
Agi Csakiea34b402015-08-13 17:51:19 -07001489
1490 // Reference equality check, return true if same reference.
1491 __ Cmp(str, arg);
1492 __ B(&return_true, eq);
1493
Vladimir Marko53b52002016-05-24 19:30:45 +01001494 if (!optimizations.GetArgumentIsString()) {
1495 // Instanceof check for the argument by comparing class fields.
1496 // All string objects must have the same type since String cannot be subclassed.
1497 // Receiver must be a string object, so its class field is equal to all strings' class fields.
1498 // If the argument is a string object, its class field must be equal to receiver's class field.
1499 __ Ldr(temp, MemOperand(str.X(), class_offset));
1500 __ Ldr(temp1, MemOperand(arg.X(), class_offset));
1501 __ Cmp(temp, temp1);
1502 __ B(&return_false, ne);
1503 }
Agi Csakiea34b402015-08-13 17:51:19 -07001504
1505 // Load lengths of this and argument strings.
1506 __ Ldr(temp, MemOperand(str.X(), count_offset));
1507 __ Ldr(temp1, MemOperand(arg.X(), count_offset));
1508 // Check if lengths are equal, return false if they're not.
jessicahandojo05765752016-09-09 19:01:32 -07001509 // Also compares the compression style, if differs return false.
Agi Csakiea34b402015-08-13 17:51:19 -07001510 __ Cmp(temp, temp1);
1511 __ B(&return_false, ne);
Agi Csakiea34b402015-08-13 17:51:19 -07001512 // Return true if both strings are empty.
jessicahandojo05765752016-09-09 19:01:32 -07001513 if (mirror::kUseStringCompression) {
1514 // Length needs to be masked out first because 0 is treated as compressed.
1515 __ Bic(temp, temp, Operand(static_cast<int32_t>(0x80000000)));
1516 }
Agi Csakiea34b402015-08-13 17:51:19 -07001517 __ Cbz(temp, &return_true);
1518
1519 // Assertions that must hold in order to compare strings 4 characters at a time.
1520 DCHECK_ALIGNED(value_offset, 8);
1521 static_assert(IsAligned<8>(kObjectAlignment), "String of odd length is not zero padded");
1522
jessicahandojo05765752016-09-09 19:01:32 -07001523 if (mirror::kUseStringCompression) {
1524 // If not compressed, directly to fast compare. Else do preprocess on length.
1525 __ Cmp(temp1, Operand(0));
1526 __ B(&preloop, gt);
1527 // Mask out compression flag and adjust length for compressed string (8-bit)
1528 // as if it is a 16-bit data, new_length = (length + 1) / 2
1529 __ Add(temp, temp, 1);
1530 __ Lsr(temp, temp, 1);
1531 }
1532
Agi Csakiea34b402015-08-13 17:51:19 -07001533 temp1 = temp1.X();
1534 temp2 = temp2.X();
Agi Csakiea34b402015-08-13 17:51:19 -07001535 // Loop to compare strings 4 characters at a time starting at the beginning of the string.
1536 // Ok to do this because strings are zero-padded to be 8-byte aligned.
jessicahandojo05765752016-09-09 19:01:32 -07001537 // Store offset of string value in preparation for comparison loop
1538 __ Bind(&preloop);
1539 __ Mov(temp1, value_offset);
Agi Csakiea34b402015-08-13 17:51:19 -07001540 __ Bind(&loop);
1541 __ Ldr(out, MemOperand(str.X(), temp1));
1542 __ Ldr(temp2, MemOperand(arg.X(), temp1));
1543 __ Add(temp1, temp1, Operand(sizeof(uint64_t)));
1544 __ Cmp(out, temp2);
1545 __ B(&return_false, ne);
1546 __ Sub(temp, temp, Operand(4), SetFlags);
1547 __ B(&loop, gt);
1548
1549 // Return true and exit the function.
1550 // If loop does not result in returning false, we return true.
1551 __ Bind(&return_true);
1552 __ Mov(out, 1);
1553 __ B(&end);
1554
1555 // Return false and exit the function.
1556 __ Bind(&return_false);
1557 __ Mov(out, 0);
1558 __ Bind(&end);
1559}
1560
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001561static void GenerateVisitStringIndexOf(HInvoke* invoke,
Scott Wakeling97c72b72016-06-24 16:19:36 +01001562 MacroAssembler* masm,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001563 CodeGeneratorARM64* codegen,
1564 ArenaAllocator* allocator,
1565 bool start_at_zero) {
1566 LocationSummary* locations = invoke->GetLocations();
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001567
1568 // Note that the null check must have been done earlier.
1569 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1570
1571 // Check for code points > 0xFFFF. Either a slow-path check when we don't know statically,
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001572 // 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 -07001573 SlowPathCodeARM64* slow_path = nullptr;
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001574 HInstruction* code_point = invoke->InputAt(1);
1575 if (code_point->IsIntConstant()) {
Vladimir Markoda051082016-05-17 16:10:20 +01001576 if (static_cast<uint32_t>(code_point->AsIntConstant()->GetValue()) > 0xFFFFU) {
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001577 // Always needs the slow-path. We could directly dispatch to it, but this case should be
1578 // rare, so for simplicity just put the full slow-path down and branch unconditionally.
1579 slow_path = new (allocator) IntrinsicSlowPathARM64(invoke);
1580 codegen->AddSlowPath(slow_path);
1581 __ B(slow_path->GetEntryLabel());
1582 __ Bind(slow_path->GetExitLabel());
1583 return;
1584 }
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001585 } else if (code_point->GetType() != Primitive::kPrimChar) {
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001586 Register char_reg = WRegisterFrom(locations->InAt(1));
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001587 __ Tst(char_reg, 0xFFFF0000);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001588 slow_path = new (allocator) IntrinsicSlowPathARM64(invoke);
1589 codegen->AddSlowPath(slow_path);
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001590 __ B(ne, slow_path->GetEntryLabel());
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001591 }
1592
1593 if (start_at_zero) {
1594 // Start-index = 0.
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001595 Register tmp_reg = WRegisterFrom(locations->GetTemp(0));
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001596 __ Mov(tmp_reg, 0);
1597 }
1598
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001599 codegen->InvokeRuntime(kQuickIndexOf, invoke, invoke->GetDexPc(), slow_path);
Roland Levillain42ad2882016-02-29 18:26:54 +00001600 CheckEntrypointTypes<kQuickIndexOf, int32_t, void*, uint32_t, uint32_t>();
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001601
1602 if (slow_path != nullptr) {
1603 __ Bind(slow_path->GetExitLabel());
1604 }
1605}
1606
1607void IntrinsicLocationsBuilderARM64::VisitStringIndexOf(HInvoke* invoke) {
1608 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001609 LocationSummary::kCallOnMainAndSlowPath,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001610 kIntrinsified);
1611 // We have a hand-crafted assembly stub that follows the runtime calling convention. So it's
1612 // best to align the inputs accordingly.
1613 InvokeRuntimeCallingConvention calling_convention;
1614 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1615 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1616 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimInt));
1617
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001618 // Need to send start_index=0.
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001619 locations->AddTemp(LocationFrom(calling_convention.GetRegisterAt(2)));
1620}
1621
1622void IntrinsicCodeGeneratorARM64::VisitStringIndexOf(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001623 GenerateVisitStringIndexOf(
1624 invoke, GetVIXLAssembler(), codegen_, GetAllocator(), /* start_at_zero */ true);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001625}
1626
1627void IntrinsicLocationsBuilderARM64::VisitStringIndexOfAfter(HInvoke* invoke) {
1628 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001629 LocationSummary::kCallOnMainAndSlowPath,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001630 kIntrinsified);
1631 // We have a hand-crafted assembly stub that follows the runtime calling convention. So it's
1632 // best to align the inputs accordingly.
1633 InvokeRuntimeCallingConvention calling_convention;
1634 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1635 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1636 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1637 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimInt));
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001638}
1639
1640void IntrinsicCodeGeneratorARM64::VisitStringIndexOfAfter(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001641 GenerateVisitStringIndexOf(
1642 invoke, GetVIXLAssembler(), codegen_, GetAllocator(), /* start_at_zero */ false);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001643}
1644
Jeff Hao848f70a2014-01-15 13:49:50 -08001645void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromBytes(HInvoke* invoke) {
1646 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001647 LocationSummary::kCallOnMainAndSlowPath,
Jeff Hao848f70a2014-01-15 13:49:50 -08001648 kIntrinsified);
1649 InvokeRuntimeCallingConvention calling_convention;
1650 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1651 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1652 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1653 locations->SetInAt(3, LocationFrom(calling_convention.GetRegisterAt(3)));
1654 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1655}
1656
1657void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromBytes(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001658 MacroAssembler* masm = GetVIXLAssembler();
Jeff Hao848f70a2014-01-15 13:49:50 -08001659 LocationSummary* locations = invoke->GetLocations();
1660
1661 Register byte_array = WRegisterFrom(locations->InAt(0));
1662 __ Cmp(byte_array, 0);
1663 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1664 codegen_->AddSlowPath(slow_path);
1665 __ B(eq, slow_path->GetEntryLabel());
1666
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001667 codegen_->InvokeRuntime(kQuickAllocStringFromBytes, invoke, invoke->GetDexPc(), slow_path);
Roland Levillainf969a202016-03-09 16:14:00 +00001668 CheckEntrypointTypes<kQuickAllocStringFromBytes, void*, void*, int32_t, int32_t, int32_t>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001669 __ Bind(slow_path->GetExitLabel());
1670}
1671
1672void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromChars(HInvoke* invoke) {
1673 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001674 LocationSummary::kCallOnMainOnly,
Jeff Hao848f70a2014-01-15 13:49:50 -08001675 kIntrinsified);
1676 InvokeRuntimeCallingConvention calling_convention;
1677 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1678 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1679 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1680 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1681}
1682
1683void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromChars(HInvoke* invoke) {
Roland Levillaincc3839c2016-02-29 16:23:48 +00001684 // No need to emit code checking whether `locations->InAt(2)` is a null
1685 // pointer, as callers of the native method
1686 //
1687 // java.lang.StringFactory.newStringFromChars(int offset, int charCount, char[] data)
1688 //
1689 // all include a null check on `data` before calling that method.
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001690 codegen_->InvokeRuntime(kQuickAllocStringFromChars, invoke, invoke->GetDexPc());
Roland Levillainf969a202016-03-09 16:14:00 +00001691 CheckEntrypointTypes<kQuickAllocStringFromChars, void*, int32_t, int32_t, void*>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001692}
1693
1694void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromString(HInvoke* invoke) {
Jeff Hao848f70a2014-01-15 13:49:50 -08001695 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001696 LocationSummary::kCallOnMainAndSlowPath,
Jeff Hao848f70a2014-01-15 13:49:50 -08001697 kIntrinsified);
1698 InvokeRuntimeCallingConvention calling_convention;
1699 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
Jeff Hao848f70a2014-01-15 13:49:50 -08001700 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1701}
1702
1703void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromString(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001704 MacroAssembler* masm = GetVIXLAssembler();
Jeff Hao848f70a2014-01-15 13:49:50 -08001705 LocationSummary* locations = invoke->GetLocations();
1706
1707 Register string_to_copy = WRegisterFrom(locations->InAt(0));
1708 __ Cmp(string_to_copy, 0);
1709 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1710 codegen_->AddSlowPath(slow_path);
1711 __ B(eq, slow_path->GetEntryLabel());
1712
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001713 codegen_->InvokeRuntime(kQuickAllocStringFromString, invoke, invoke->GetDexPc(), slow_path);
Roland Levillainf969a202016-03-09 16:14:00 +00001714 CheckEntrypointTypes<kQuickAllocStringFromString, void*, void*>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001715 __ Bind(slow_path->GetExitLabel());
1716}
1717
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001718static void CreateFPToFPCallLocations(ArenaAllocator* arena, HInvoke* invoke) {
1719 DCHECK_EQ(invoke->GetNumberOfArguments(), 1U);
1720 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(0)->GetType()));
1721 DCHECK(Primitive::IsFloatingPointType(invoke->GetType()));
1722
1723 LocationSummary* const locations = new (arena) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001724 LocationSummary::kCallOnMainOnly,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001725 kIntrinsified);
1726 InvokeRuntimeCallingConvention calling_convention;
1727
1728 locations->SetInAt(0, LocationFrom(calling_convention.GetFpuRegisterAt(0)));
1729 locations->SetOut(calling_convention.GetReturnLocation(invoke->GetType()));
1730}
1731
1732static void CreateFPFPToFPCallLocations(ArenaAllocator* arena, HInvoke* invoke) {
1733 DCHECK_EQ(invoke->GetNumberOfArguments(), 2U);
1734 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(0)->GetType()));
1735 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(1)->GetType()));
1736 DCHECK(Primitive::IsFloatingPointType(invoke->GetType()));
1737
1738 LocationSummary* const locations = new (arena) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001739 LocationSummary::kCallOnMainOnly,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001740 kIntrinsified);
1741 InvokeRuntimeCallingConvention calling_convention;
1742
1743 locations->SetInAt(0, LocationFrom(calling_convention.GetFpuRegisterAt(0)));
1744 locations->SetInAt(1, LocationFrom(calling_convention.GetFpuRegisterAt(1)));
1745 locations->SetOut(calling_convention.GetReturnLocation(invoke->GetType()));
1746}
1747
1748static void GenFPToFPCall(HInvoke* invoke,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001749 CodeGeneratorARM64* codegen,
1750 QuickEntrypointEnum entry) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001751 codegen->InvokeRuntime(entry, invoke, invoke->GetDexPc());
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001752}
1753
1754void IntrinsicLocationsBuilderARM64::VisitMathCos(HInvoke* invoke) {
1755 CreateFPToFPCallLocations(arena_, invoke);
1756}
1757
1758void IntrinsicCodeGeneratorARM64::VisitMathCos(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001759 GenFPToFPCall(invoke, codegen_, kQuickCos);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001760}
1761
1762void IntrinsicLocationsBuilderARM64::VisitMathSin(HInvoke* invoke) {
1763 CreateFPToFPCallLocations(arena_, invoke);
1764}
1765
1766void IntrinsicCodeGeneratorARM64::VisitMathSin(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001767 GenFPToFPCall(invoke, codegen_, kQuickSin);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001768}
1769
1770void IntrinsicLocationsBuilderARM64::VisitMathAcos(HInvoke* invoke) {
1771 CreateFPToFPCallLocations(arena_, invoke);
1772}
1773
1774void IntrinsicCodeGeneratorARM64::VisitMathAcos(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001775 GenFPToFPCall(invoke, codegen_, kQuickAcos);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001776}
1777
1778void IntrinsicLocationsBuilderARM64::VisitMathAsin(HInvoke* invoke) {
1779 CreateFPToFPCallLocations(arena_, invoke);
1780}
1781
1782void IntrinsicCodeGeneratorARM64::VisitMathAsin(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001783 GenFPToFPCall(invoke, codegen_, kQuickAsin);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001784}
1785
1786void IntrinsicLocationsBuilderARM64::VisitMathAtan(HInvoke* invoke) {
1787 CreateFPToFPCallLocations(arena_, invoke);
1788}
1789
1790void IntrinsicCodeGeneratorARM64::VisitMathAtan(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001791 GenFPToFPCall(invoke, codegen_, kQuickAtan);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001792}
1793
1794void IntrinsicLocationsBuilderARM64::VisitMathCbrt(HInvoke* invoke) {
1795 CreateFPToFPCallLocations(arena_, invoke);
1796}
1797
1798void IntrinsicCodeGeneratorARM64::VisitMathCbrt(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001799 GenFPToFPCall(invoke, codegen_, kQuickCbrt);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001800}
1801
1802void IntrinsicLocationsBuilderARM64::VisitMathCosh(HInvoke* invoke) {
1803 CreateFPToFPCallLocations(arena_, invoke);
1804}
1805
1806void IntrinsicCodeGeneratorARM64::VisitMathCosh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001807 GenFPToFPCall(invoke, codegen_, kQuickCosh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001808}
1809
1810void IntrinsicLocationsBuilderARM64::VisitMathExp(HInvoke* invoke) {
1811 CreateFPToFPCallLocations(arena_, invoke);
1812}
1813
1814void IntrinsicCodeGeneratorARM64::VisitMathExp(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001815 GenFPToFPCall(invoke, codegen_, kQuickExp);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001816}
1817
1818void IntrinsicLocationsBuilderARM64::VisitMathExpm1(HInvoke* invoke) {
1819 CreateFPToFPCallLocations(arena_, invoke);
1820}
1821
1822void IntrinsicCodeGeneratorARM64::VisitMathExpm1(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001823 GenFPToFPCall(invoke, codegen_, kQuickExpm1);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001824}
1825
1826void IntrinsicLocationsBuilderARM64::VisitMathLog(HInvoke* invoke) {
1827 CreateFPToFPCallLocations(arena_, invoke);
1828}
1829
1830void IntrinsicCodeGeneratorARM64::VisitMathLog(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001831 GenFPToFPCall(invoke, codegen_, kQuickLog);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001832}
1833
1834void IntrinsicLocationsBuilderARM64::VisitMathLog10(HInvoke* invoke) {
1835 CreateFPToFPCallLocations(arena_, invoke);
1836}
1837
1838void IntrinsicCodeGeneratorARM64::VisitMathLog10(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001839 GenFPToFPCall(invoke, codegen_, kQuickLog10);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001840}
1841
1842void IntrinsicLocationsBuilderARM64::VisitMathSinh(HInvoke* invoke) {
1843 CreateFPToFPCallLocations(arena_, invoke);
1844}
1845
1846void IntrinsicCodeGeneratorARM64::VisitMathSinh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001847 GenFPToFPCall(invoke, codegen_, kQuickSinh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001848}
1849
1850void IntrinsicLocationsBuilderARM64::VisitMathTan(HInvoke* invoke) {
1851 CreateFPToFPCallLocations(arena_, invoke);
1852}
1853
1854void IntrinsicCodeGeneratorARM64::VisitMathTan(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001855 GenFPToFPCall(invoke, codegen_, kQuickTan);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001856}
1857
1858void IntrinsicLocationsBuilderARM64::VisitMathTanh(HInvoke* invoke) {
1859 CreateFPToFPCallLocations(arena_, invoke);
1860}
1861
1862void IntrinsicCodeGeneratorARM64::VisitMathTanh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001863 GenFPToFPCall(invoke, codegen_, kQuickTanh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001864}
1865
1866void IntrinsicLocationsBuilderARM64::VisitMathAtan2(HInvoke* invoke) {
1867 CreateFPFPToFPCallLocations(arena_, invoke);
1868}
1869
1870void IntrinsicCodeGeneratorARM64::VisitMathAtan2(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001871 GenFPToFPCall(invoke, codegen_, kQuickAtan2);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001872}
1873
1874void IntrinsicLocationsBuilderARM64::VisitMathHypot(HInvoke* invoke) {
1875 CreateFPFPToFPCallLocations(arena_, invoke);
1876}
1877
1878void IntrinsicCodeGeneratorARM64::VisitMathHypot(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001879 GenFPToFPCall(invoke, codegen_, kQuickHypot);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001880}
1881
1882void IntrinsicLocationsBuilderARM64::VisitMathNextAfter(HInvoke* invoke) {
1883 CreateFPFPToFPCallLocations(arena_, invoke);
1884}
1885
1886void IntrinsicCodeGeneratorARM64::VisitMathNextAfter(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001887 GenFPToFPCall(invoke, codegen_, kQuickNextAfter);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001888}
1889
Tim Zhang25abd6c2016-01-19 23:39:24 +08001890void IntrinsicLocationsBuilderARM64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
1891 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1892 LocationSummary::kNoCall,
1893 kIntrinsified);
1894 locations->SetInAt(0, Location::RequiresRegister());
1895 locations->SetInAt(1, Location::RequiresRegister());
1896 locations->SetInAt(2, Location::RequiresRegister());
1897 locations->SetInAt(3, Location::RequiresRegister());
1898 locations->SetInAt(4, Location::RequiresRegister());
1899
1900 locations->AddTemp(Location::RequiresRegister());
1901 locations->AddTemp(Location::RequiresRegister());
Scott Wakelingdf109d92016-04-22 11:35:56 +01001902 locations->AddTemp(Location::RequiresRegister());
jessicahandojo05765752016-09-09 19:01:32 -07001903 // Need temporary register for String compression feature.
1904 if (mirror::kUseStringCompression) {
1905 locations->AddTemp(Location::RequiresRegister());
1906 }
Tim Zhang25abd6c2016-01-19 23:39:24 +08001907}
1908
1909void IntrinsicCodeGeneratorARM64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001910 MacroAssembler* masm = GetVIXLAssembler();
Tim Zhang25abd6c2016-01-19 23:39:24 +08001911 LocationSummary* locations = invoke->GetLocations();
1912
1913 // Check assumption that sizeof(Char) is 2 (used in scaling below).
1914 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1915 DCHECK_EQ(char_size, 2u);
1916
1917 // Location of data in char array buffer.
1918 const uint32_t data_offset = mirror::Array::DataOffset(char_size).Uint32Value();
1919
1920 // Location of char array data in string.
1921 const uint32_t value_offset = mirror::String::ValueOffset().Uint32Value();
1922
1923 // void getCharsNoCheck(int srcBegin, int srcEnd, char[] dst, int dstBegin);
1924 // Since getChars() calls getCharsNoCheck() - we use registers rather than constants.
1925 Register srcObj = XRegisterFrom(locations->InAt(0));
1926 Register srcBegin = XRegisterFrom(locations->InAt(1));
1927 Register srcEnd = XRegisterFrom(locations->InAt(2));
1928 Register dstObj = XRegisterFrom(locations->InAt(3));
1929 Register dstBegin = XRegisterFrom(locations->InAt(4));
1930
1931 Register src_ptr = XRegisterFrom(locations->GetTemp(0));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001932 Register num_chr = XRegisterFrom(locations->GetTemp(1));
1933 Register tmp1 = XRegisterFrom(locations->GetTemp(2));
jessicahandojo05765752016-09-09 19:01:32 -07001934 Register tmp3;
1935 if (mirror::kUseStringCompression) {
1936 tmp3 = WRegisterFrom(locations->GetTemp(3));
1937 }
Tim Zhang25abd6c2016-01-19 23:39:24 +08001938
1939 UseScratchRegisterScope temps(masm);
1940 Register dst_ptr = temps.AcquireX();
Scott Wakelingdf109d92016-04-22 11:35:56 +01001941 Register tmp2 = temps.AcquireX();
Tim Zhang25abd6c2016-01-19 23:39:24 +08001942
jessicahandojo05765752016-09-09 19:01:32 -07001943 vixl::aarch64::Label done;
1944 vixl::aarch64::Label compressed_string_loop;
1945 __ Sub(num_chr, srcEnd, srcBegin);
1946 // Early out for valid zero-length retrievals.
1947 __ Cbz(num_chr, &done);
Tim Zhang25abd6c2016-01-19 23:39:24 +08001948
Scott Wakelingdf109d92016-04-22 11:35:56 +01001949 // dst address start to copy to.
Tim Zhang25abd6c2016-01-19 23:39:24 +08001950 __ Add(dst_ptr, dstObj, Operand(data_offset));
1951 __ Add(dst_ptr, dst_ptr, Operand(dstBegin, LSL, 1));
1952
jessicahandojo05765752016-09-09 19:01:32 -07001953 // src address to copy from.
1954 __ Add(src_ptr, srcObj, Operand(value_offset));
1955 vixl::aarch64::Label compressed_string_preloop;
1956 if (mirror::kUseStringCompression) {
1957 // Location of count in string.
1958 const uint32_t count_offset = mirror::String::CountOffset().Uint32Value();
1959 // String's length.
1960 __ Ldr(tmp3, MemOperand(srcObj, count_offset));
1961 __ Tbnz(tmp3, kWRegSize - 1, &compressed_string_preloop);
1962 }
1963 __ Add(src_ptr, src_ptr, Operand(srcBegin, LSL, 1));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001964
Tim Zhang25abd6c2016-01-19 23:39:24 +08001965 // Do the copy.
Scott Wakeling97c72b72016-06-24 16:19:36 +01001966 vixl::aarch64::Label loop;
Scott Wakeling97c72b72016-06-24 16:19:36 +01001967 vixl::aarch64::Label remainder;
Scott Wakelingdf109d92016-04-22 11:35:56 +01001968
Scott Wakelingdf109d92016-04-22 11:35:56 +01001969 // Save repairing the value of num_chr on the < 8 character path.
1970 __ Subs(tmp1, num_chr, 8);
1971 __ B(lt, &remainder);
1972
1973 // Keep the result of the earlier subs, we are going to fetch at least 8 characters.
1974 __ Mov(num_chr, tmp1);
1975
1976 // Main loop used for longer fetches loads and stores 8x16-bit characters at a time.
1977 // (Unaligned addresses are acceptable here and not worth inlining extra code to rectify.)
Tim Zhang25abd6c2016-01-19 23:39:24 +08001978 __ Bind(&loop);
Scott Wakeling97c72b72016-06-24 16:19:36 +01001979 __ Ldp(tmp1, tmp2, MemOperand(src_ptr, char_size * 8, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001980 __ Subs(num_chr, num_chr, 8);
Scott Wakeling97c72b72016-06-24 16:19:36 +01001981 __ Stp(tmp1, tmp2, MemOperand(dst_ptr, char_size * 8, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001982 __ B(ge, &loop);
1983
1984 __ Adds(num_chr, num_chr, 8);
1985 __ B(eq, &done);
1986
1987 // Main loop for < 8 character case and remainder handling. Loads and stores one
1988 // 16-bit Java character at a time.
1989 __ Bind(&remainder);
Scott Wakeling97c72b72016-06-24 16:19:36 +01001990 __ Ldrh(tmp1, MemOperand(src_ptr, char_size, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001991 __ Subs(num_chr, num_chr, 1);
Scott Wakeling97c72b72016-06-24 16:19:36 +01001992 __ Strh(tmp1, MemOperand(dst_ptr, char_size, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001993 __ B(gt, &remainder);
jessicahandojo05765752016-09-09 19:01:32 -07001994 __ B(&done);
1995
1996 if (mirror::kUseStringCompression) {
1997 const size_t c_char_size = Primitive::ComponentSize(Primitive::kPrimByte);
1998 DCHECK_EQ(c_char_size, 1u);
1999 __ Bind(&compressed_string_preloop);
2000 __ Add(src_ptr, src_ptr, Operand(srcBegin));
2001 // Copy loop for compressed src, copying 1 character (8-bit) to (16-bit) at a time.
2002 __ Bind(&compressed_string_loop);
2003 __ Ldrb(tmp1, MemOperand(src_ptr, c_char_size, PostIndex));
2004 __ Strh(tmp1, MemOperand(dst_ptr, char_size, PostIndex));
2005 __ Subs(num_chr, num_chr, Operand(1));
2006 __ B(gt, &compressed_string_loop);
2007 }
Scott Wakelingdf109d92016-04-22 11:35:56 +01002008
Tim Zhang25abd6c2016-01-19 23:39:24 +08002009 __ Bind(&done);
2010}
2011
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002012// Mirrors ARRAYCOPY_SHORT_CHAR_ARRAY_THRESHOLD in libcore, so we can choose to use the native
2013// implementation there for longer copy lengths.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002014static constexpr int32_t kSystemArrayCopyCharThreshold = 32;
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002015
2016static void SetSystemArrayCopyLocationRequires(LocationSummary* locations,
2017 uint32_t at,
2018 HInstruction* input) {
2019 HIntConstant* const_input = input->AsIntConstant();
Scott Wakeling97c72b72016-06-24 16:19:36 +01002020 if (const_input != nullptr && !vixl::aarch64::Assembler::IsImmAddSub(const_input->GetValue())) {
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002021 locations->SetInAt(at, Location::RequiresRegister());
2022 } else {
2023 locations->SetInAt(at, Location::RegisterOrConstant(input));
2024 }
2025}
2026
2027void IntrinsicLocationsBuilderARM64::VisitSystemArrayCopyChar(HInvoke* invoke) {
2028 // Check to see if we have known failures that will cause us to have to bail out
2029 // to the runtime, and just generate the runtime call directly.
2030 HIntConstant* src_pos = invoke->InputAt(1)->AsIntConstant();
2031 HIntConstant* dst_pos = invoke->InputAt(3)->AsIntConstant();
2032
2033 // The positions must be non-negative.
2034 if ((src_pos != nullptr && src_pos->GetValue() < 0) ||
2035 (dst_pos != nullptr && dst_pos->GetValue() < 0)) {
2036 // We will have to fail anyways.
2037 return;
2038 }
2039
2040 // The length must be >= 0 and not so long that we would (currently) prefer libcore's
2041 // native implementation.
2042 HIntConstant* length = invoke->InputAt(4)->AsIntConstant();
2043 if (length != nullptr) {
2044 int32_t len = length->GetValue();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002045 if (len < 0 || len > kSystemArrayCopyCharThreshold) {
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002046 // Just call as normal.
2047 return;
2048 }
2049 }
2050
2051 ArenaAllocator* allocator = invoke->GetBlock()->GetGraph()->GetArena();
2052 LocationSummary* locations = new (allocator) LocationSummary(invoke,
2053 LocationSummary::kCallOnSlowPath,
2054 kIntrinsified);
2055 // arraycopy(char[] src, int src_pos, char[] dst, int dst_pos, int length).
2056 locations->SetInAt(0, Location::RequiresRegister());
2057 SetSystemArrayCopyLocationRequires(locations, 1, invoke->InputAt(1));
2058 locations->SetInAt(2, Location::RequiresRegister());
2059 SetSystemArrayCopyLocationRequires(locations, 3, invoke->InputAt(3));
2060 SetSystemArrayCopyLocationRequires(locations, 4, invoke->InputAt(4));
2061
2062 locations->AddTemp(Location::RequiresRegister());
2063 locations->AddTemp(Location::RequiresRegister());
2064 locations->AddTemp(Location::RequiresRegister());
2065}
2066
Scott Wakeling97c72b72016-06-24 16:19:36 +01002067static void CheckSystemArrayCopyPosition(MacroAssembler* masm,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002068 const Location& pos,
2069 const Register& input,
2070 const Location& length,
2071 SlowPathCodeARM64* slow_path,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002072 const Register& temp,
2073 bool length_is_input_length = false) {
2074 const int32_t length_offset = mirror::Array::LengthOffset().Int32Value();
2075 if (pos.IsConstant()) {
2076 int32_t pos_const = pos.GetConstant()->AsIntConstant()->GetValue();
2077 if (pos_const == 0) {
2078 if (!length_is_input_length) {
2079 // Check that length(input) >= length.
2080 __ Ldr(temp, MemOperand(input, length_offset));
2081 __ Cmp(temp, OperandFrom(length, Primitive::kPrimInt));
2082 __ B(slow_path->GetEntryLabel(), lt);
2083 }
2084 } else {
2085 // Check that length(input) >= pos.
Nicolas Geoffrayfea1abd2016-07-06 12:09:12 +01002086 __ Ldr(temp, MemOperand(input, length_offset));
2087 __ Subs(temp, temp, pos_const);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002088 __ B(slow_path->GetEntryLabel(), lt);
2089
2090 // Check that (length(input) - pos) >= length.
2091 __ Cmp(temp, OperandFrom(length, Primitive::kPrimInt));
2092 __ B(slow_path->GetEntryLabel(), lt);
2093 }
2094 } else if (length_is_input_length) {
2095 // The only way the copy can succeed is if pos is zero.
2096 __ Cbnz(WRegisterFrom(pos), slow_path->GetEntryLabel());
2097 } else {
2098 // Check that pos >= 0.
2099 Register pos_reg = WRegisterFrom(pos);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002100 __ Tbnz(pos_reg, pos_reg.GetSizeInBits() - 1, slow_path->GetEntryLabel());
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002101
2102 // Check that pos <= length(input) && (length(input) - pos) >= length.
2103 __ Ldr(temp, MemOperand(input, length_offset));
2104 __ Subs(temp, temp, pos_reg);
2105 // Ccmp if length(input) >= pos, else definitely bail to slow path (N!=V == lt).
2106 __ Ccmp(temp, OperandFrom(length, Primitive::kPrimInt), NFlag, ge);
2107 __ B(slow_path->GetEntryLabel(), lt);
2108 }
2109}
2110
2111// Compute base source address, base destination address, and end source address
2112// for System.arraycopy* intrinsics.
Scott Wakeling97c72b72016-06-24 16:19:36 +01002113static void GenSystemArrayCopyAddresses(MacroAssembler* masm,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002114 Primitive::Type type,
2115 const Register& src,
2116 const Location& src_pos,
2117 const Register& dst,
2118 const Location& dst_pos,
2119 const Location& copy_length,
2120 const Register& src_base,
2121 const Register& dst_base,
2122 const Register& src_end) {
2123 DCHECK(type == Primitive::kPrimNot || type == Primitive::kPrimChar)
Roland Levillainebea3d22016-04-12 15:42:57 +01002124 << "Unexpected element type: " << type;
2125 const int32_t element_size = Primitive::ComponentSize(type);
2126 const int32_t element_size_shift = Primitive::ComponentSizeShift(type);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002127
Roland Levillainebea3d22016-04-12 15:42:57 +01002128 uint32_t data_offset = mirror::Array::DataOffset(element_size).Uint32Value();
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002129 if (src_pos.IsConstant()) {
2130 int32_t constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002131 __ Add(src_base, src, element_size * constant + data_offset);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002132 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002133 __ Add(src_base, src, data_offset);
2134 __ Add(src_base, src_base, Operand(XRegisterFrom(src_pos), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002135 }
2136
2137 if (dst_pos.IsConstant()) {
2138 int32_t constant = dst_pos.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002139 __ Add(dst_base, dst, element_size * constant + data_offset);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002140 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002141 __ Add(dst_base, dst, data_offset);
2142 __ Add(dst_base, dst_base, Operand(XRegisterFrom(dst_pos), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002143 }
2144
2145 if (copy_length.IsConstant()) {
2146 int32_t constant = copy_length.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002147 __ Add(src_end, src_base, element_size * constant);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002148 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002149 __ Add(src_end, src_base, Operand(XRegisterFrom(copy_length), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002150 }
2151}
2152
2153void IntrinsicCodeGeneratorARM64::VisitSystemArrayCopyChar(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01002154 MacroAssembler* masm = GetVIXLAssembler();
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002155 LocationSummary* locations = invoke->GetLocations();
2156 Register src = XRegisterFrom(locations->InAt(0));
2157 Location src_pos = locations->InAt(1);
2158 Register dst = XRegisterFrom(locations->InAt(2));
2159 Location dst_pos = locations->InAt(3);
2160 Location length = locations->InAt(4);
2161
2162 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
2163 codegen_->AddSlowPath(slow_path);
2164
2165 // If source and destination are the same, take the slow path. Overlapping copy regions must be
2166 // copied in reverse and we can't know in all cases if it's needed.
2167 __ Cmp(src, dst);
2168 __ B(slow_path->GetEntryLabel(), eq);
2169
2170 // Bail out if the source is null.
2171 __ Cbz(src, slow_path->GetEntryLabel());
2172
2173 // Bail out if the destination is null.
2174 __ Cbz(dst, slow_path->GetEntryLabel());
2175
2176 if (!length.IsConstant()) {
2177 // If the length is negative, bail out.
2178 __ Tbnz(WRegisterFrom(length), kWRegSize - 1, slow_path->GetEntryLabel());
2179 // If the length > 32 then (currently) prefer libcore's native implementation.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002180 __ Cmp(WRegisterFrom(length), kSystemArrayCopyCharThreshold);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002181 __ B(slow_path->GetEntryLabel(), gt);
2182 } else {
2183 // We have already checked in the LocationsBuilder for the constant case.
2184 DCHECK_GE(length.GetConstant()->AsIntConstant()->GetValue(), 0);
2185 DCHECK_LE(length.GetConstant()->AsIntConstant()->GetValue(), 32);
2186 }
2187
2188 Register src_curr_addr = WRegisterFrom(locations->GetTemp(0));
2189 Register dst_curr_addr = WRegisterFrom(locations->GetTemp(1));
2190 Register src_stop_addr = WRegisterFrom(locations->GetTemp(2));
2191
2192 CheckSystemArrayCopyPosition(masm,
2193 src_pos,
2194 src,
2195 length,
2196 slow_path,
2197 src_curr_addr,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002198 false);
2199
2200 CheckSystemArrayCopyPosition(masm,
2201 dst_pos,
2202 dst,
2203 length,
2204 slow_path,
2205 src_curr_addr,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002206 false);
2207
2208 src_curr_addr = src_curr_addr.X();
2209 dst_curr_addr = dst_curr_addr.X();
2210 src_stop_addr = src_stop_addr.X();
2211
2212 GenSystemArrayCopyAddresses(masm,
2213 Primitive::kPrimChar,
2214 src,
2215 src_pos,
2216 dst,
2217 dst_pos,
2218 length,
2219 src_curr_addr,
2220 dst_curr_addr,
2221 src_stop_addr);
2222
2223 // Iterate over the arrays and do a raw copy of the chars.
2224 const int32_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
2225 UseScratchRegisterScope temps(masm);
2226 Register tmp = temps.AcquireW();
Scott Wakeling97c72b72016-06-24 16:19:36 +01002227 vixl::aarch64::Label loop, done;
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002228 __ Bind(&loop);
2229 __ Cmp(src_curr_addr, src_stop_addr);
2230 __ B(&done, eq);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002231 __ Ldrh(tmp, MemOperand(src_curr_addr, char_size, PostIndex));
2232 __ Strh(tmp, MemOperand(dst_curr_addr, char_size, PostIndex));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002233 __ B(&loop);
2234 __ Bind(&done);
2235
2236 __ Bind(slow_path->GetExitLabel());
2237}
2238
donghui.baic2ec9ad2016-03-10 14:02:55 +08002239// We can choose to use the native implementation there for longer copy lengths.
2240static constexpr int32_t kSystemArrayCopyThreshold = 128;
2241
2242// CodeGenerator::CreateSystemArrayCopyLocationSummary use three temporary registers.
2243// We want to use two temporary registers in order to reduce the register pressure in arm64.
2244// So we don't use the CodeGenerator::CreateSystemArrayCopyLocationSummary.
2245void IntrinsicLocationsBuilderARM64::VisitSystemArrayCopy(HInvoke* invoke) {
Roland Levillain0b671c02016-08-19 12:02:34 +01002246 // The only read barrier implementation supporting the
2247 // SystemArrayCopy intrinsic is the Baker-style read barriers.
2248 if (kEmitCompilerReadBarrier && !kUseBakerReadBarrier) {
Roland Levillain3d312422016-06-23 13:53:42 +01002249 return;
2250 }
2251
donghui.baic2ec9ad2016-03-10 14:02:55 +08002252 // Check to see if we have known failures that will cause us to have to bail out
2253 // to the runtime, and just generate the runtime call directly.
2254 HIntConstant* src_pos = invoke->InputAt(1)->AsIntConstant();
2255 HIntConstant* dest_pos = invoke->InputAt(3)->AsIntConstant();
2256
2257 // The positions must be non-negative.
2258 if ((src_pos != nullptr && src_pos->GetValue() < 0) ||
2259 (dest_pos != nullptr && dest_pos->GetValue() < 0)) {
2260 // We will have to fail anyways.
2261 return;
2262 }
2263
2264 // The length must be >= 0.
2265 HIntConstant* length = invoke->InputAt(4)->AsIntConstant();
2266 if (length != nullptr) {
2267 int32_t len = length->GetValue();
2268 if (len < 0 || len >= kSystemArrayCopyThreshold) {
2269 // Just call as normal.
2270 return;
2271 }
2272 }
2273
2274 SystemArrayCopyOptimizations optimizations(invoke);
2275
2276 if (optimizations.GetDestinationIsSource()) {
2277 if (src_pos != nullptr && dest_pos != nullptr && src_pos->GetValue() < dest_pos->GetValue()) {
2278 // We only support backward copying if source and destination are the same.
2279 return;
2280 }
2281 }
2282
2283 if (optimizations.GetDestinationIsPrimitiveArray() || optimizations.GetSourceIsPrimitiveArray()) {
2284 // We currently don't intrinsify primitive copying.
2285 return;
2286 }
2287
2288 ArenaAllocator* allocator = invoke->GetBlock()->GetGraph()->GetArena();
2289 LocationSummary* locations = new (allocator) LocationSummary(invoke,
2290 LocationSummary::kCallOnSlowPath,
2291 kIntrinsified);
2292 // arraycopy(Object src, int src_pos, Object dest, int dest_pos, int length).
2293 locations->SetInAt(0, Location::RequiresRegister());
2294 SetSystemArrayCopyLocationRequires(locations, 1, invoke->InputAt(1));
2295 locations->SetInAt(2, Location::RequiresRegister());
2296 SetSystemArrayCopyLocationRequires(locations, 3, invoke->InputAt(3));
2297 SetSystemArrayCopyLocationRequires(locations, 4, invoke->InputAt(4));
2298
2299 locations->AddTemp(Location::RequiresRegister());
2300 locations->AddTemp(Location::RequiresRegister());
Roland Levillain0b671c02016-08-19 12:02:34 +01002301 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2302 // Temporary register IP0, obtained from the VIXL scratch register
2303 // pool, cannot be used in ReadBarrierSystemArrayCopySlowPathARM64
2304 // (because that register is clobbered by ReadBarrierMarkRegX
2305 // entry points). Get an extra temporary register from the
2306 // register allocator.
2307 locations->AddTemp(Location::RequiresRegister());
2308 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002309}
2310
2311void IntrinsicCodeGeneratorARM64::VisitSystemArrayCopy(HInvoke* invoke) {
Roland Levillain0b671c02016-08-19 12:02:34 +01002312 // The only read barrier implementation supporting the
2313 // SystemArrayCopy intrinsic is the Baker-style read barriers.
2314 DCHECK(!kEmitCompilerReadBarrier || kUseBakerReadBarrier);
Roland Levillain3d312422016-06-23 13:53:42 +01002315
Scott Wakeling97c72b72016-06-24 16:19:36 +01002316 MacroAssembler* masm = GetVIXLAssembler();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002317 LocationSummary* locations = invoke->GetLocations();
2318
2319 uint32_t class_offset = mirror::Object::ClassOffset().Int32Value();
2320 uint32_t super_offset = mirror::Class::SuperClassOffset().Int32Value();
2321 uint32_t component_offset = mirror::Class::ComponentTypeOffset().Int32Value();
2322 uint32_t primitive_offset = mirror::Class::PrimitiveTypeOffset().Int32Value();
Roland Levillain0b671c02016-08-19 12:02:34 +01002323 uint32_t monitor_offset = mirror::Object::MonitorOffset().Int32Value();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002324
2325 Register src = XRegisterFrom(locations->InAt(0));
2326 Location src_pos = locations->InAt(1);
2327 Register dest = XRegisterFrom(locations->InAt(2));
2328 Location dest_pos = locations->InAt(3);
2329 Location length = locations->InAt(4);
2330 Register temp1 = WRegisterFrom(locations->GetTemp(0));
Roland Levillain0b671c02016-08-19 12:02:34 +01002331 Location temp1_loc = LocationFrom(temp1);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002332 Register temp2 = WRegisterFrom(locations->GetTemp(1));
Roland Levillain0b671c02016-08-19 12:02:34 +01002333 Location temp2_loc = LocationFrom(temp2);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002334
Roland Levillain0b671c02016-08-19 12:02:34 +01002335 SlowPathCodeARM64* intrinsic_slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
2336 codegen_->AddSlowPath(intrinsic_slow_path);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002337
Scott Wakeling97c72b72016-06-24 16:19:36 +01002338 vixl::aarch64::Label conditions_on_positions_validated;
donghui.baic2ec9ad2016-03-10 14:02:55 +08002339 SystemArrayCopyOptimizations optimizations(invoke);
2340
donghui.baic2ec9ad2016-03-10 14:02:55 +08002341 // If source and destination are the same, we go to slow path if we need to do
2342 // forward copying.
2343 if (src_pos.IsConstant()) {
2344 int32_t src_pos_constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
2345 if (dest_pos.IsConstant()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002346 int32_t dest_pos_constant = dest_pos.GetConstant()->AsIntConstant()->GetValue();
2347 if (optimizations.GetDestinationIsSource()) {
2348 // Checked when building locations.
2349 DCHECK_GE(src_pos_constant, dest_pos_constant);
2350 } else if (src_pos_constant < dest_pos_constant) {
2351 __ Cmp(src, dest);
Roland Levillain0b671c02016-08-19 12:02:34 +01002352 __ B(intrinsic_slow_path->GetEntryLabel(), eq);
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002353 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002354 // Checked when building locations.
2355 DCHECK(!optimizations.GetDestinationIsSource()
2356 || (src_pos_constant >= dest_pos.GetConstant()->AsIntConstant()->GetValue()));
2357 } else {
2358 if (!optimizations.GetDestinationIsSource()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002359 __ Cmp(src, dest);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002360 __ B(&conditions_on_positions_validated, ne);
2361 }
2362 __ Cmp(WRegisterFrom(dest_pos), src_pos_constant);
Roland Levillain0b671c02016-08-19 12:02:34 +01002363 __ B(intrinsic_slow_path->GetEntryLabel(), gt);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002364 }
2365 } else {
2366 if (!optimizations.GetDestinationIsSource()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002367 __ Cmp(src, dest);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002368 __ B(&conditions_on_positions_validated, ne);
2369 }
2370 __ Cmp(RegisterFrom(src_pos, invoke->InputAt(1)->GetType()),
2371 OperandFrom(dest_pos, invoke->InputAt(3)->GetType()));
Roland Levillain0b671c02016-08-19 12:02:34 +01002372 __ B(intrinsic_slow_path->GetEntryLabel(), lt);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002373 }
2374
2375 __ Bind(&conditions_on_positions_validated);
2376
2377 if (!optimizations.GetSourceIsNotNull()) {
2378 // Bail out if the source is null.
Roland Levillain0b671c02016-08-19 12:02:34 +01002379 __ Cbz(src, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002380 }
2381
2382 if (!optimizations.GetDestinationIsNotNull() && !optimizations.GetDestinationIsSource()) {
2383 // Bail out if the destination is null.
Roland Levillain0b671c02016-08-19 12:02:34 +01002384 __ Cbz(dest, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002385 }
2386
2387 // We have already checked in the LocationsBuilder for the constant case.
2388 if (!length.IsConstant() &&
2389 !optimizations.GetCountIsSourceLength() &&
2390 !optimizations.GetCountIsDestinationLength()) {
2391 // If the length is negative, bail out.
Roland Levillain0b671c02016-08-19 12:02:34 +01002392 __ Tbnz(WRegisterFrom(length), kWRegSize - 1, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002393 // If the length >= 128 then (currently) prefer native implementation.
2394 __ Cmp(WRegisterFrom(length), kSystemArrayCopyThreshold);
Roland Levillain0b671c02016-08-19 12:02:34 +01002395 __ B(intrinsic_slow_path->GetEntryLabel(), ge);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002396 }
2397 // Validity checks: source.
2398 CheckSystemArrayCopyPosition(masm,
2399 src_pos,
2400 src,
2401 length,
Roland Levillain0b671c02016-08-19 12:02:34 +01002402 intrinsic_slow_path,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002403 temp1,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002404 optimizations.GetCountIsSourceLength());
2405
2406 // Validity checks: dest.
2407 CheckSystemArrayCopyPosition(masm,
2408 dest_pos,
2409 dest,
2410 length,
Roland Levillain0b671c02016-08-19 12:02:34 +01002411 intrinsic_slow_path,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002412 temp1,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002413 optimizations.GetCountIsDestinationLength());
2414 {
2415 // We use a block to end the scratch scope before the write barrier, thus
2416 // freeing the temporary registers so they can be used in `MarkGCCard`.
2417 UseScratchRegisterScope temps(masm);
Roland Levillain0b671c02016-08-19 12:02:34 +01002418 // Note: Because it is acquired from VIXL's scratch register pool,
2419 // `temp3` might be IP0, and thus cannot be used as `ref` argument
2420 // of CodeGeneratorARM64::GenerateFieldLoadWithBakerReadBarrier
2421 // calls below (see ReadBarrierMarkSlowPathARM64 for more details).
donghui.baic2ec9ad2016-03-10 14:02:55 +08002422 Register temp3 = temps.AcquireW();
Roland Levillain0b671c02016-08-19 12:02:34 +01002423
donghui.baic2ec9ad2016-03-10 14:02:55 +08002424 if (!optimizations.GetDoesNotNeedTypeCheck()) {
2425 // Check whether all elements of the source array are assignable to the component
2426 // type of the destination array. We do two checks: the classes are the same,
2427 // or the destination is Object[]. If none of these checks succeed, we go to the
2428 // slow path.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002429
Roland Levillain0b671c02016-08-19 12:02:34 +01002430 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2431 if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2432 // /* HeapReference<Class> */ temp1 = src->klass_
2433 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2434 temp1_loc,
2435 src.W(),
2436 class_offset,
2437 temp2,
2438 /* needs_null_check */ false,
2439 /* use_load_acquire */ false);
2440 // Bail out if the source is not a non primitive array.
2441 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2442 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2443 temp1_loc,
2444 temp1,
2445 component_offset,
2446 temp2,
2447 /* needs_null_check */ false,
2448 /* use_load_acquire */ false);
2449 __ Cbz(temp1, intrinsic_slow_path->GetEntryLabel());
2450 // If heap poisoning is enabled, `temp1` has been unpoisoned
2451 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2452 // /* uint16_t */ temp1 = static_cast<uint16>(temp1->primitive_type_);
2453 __ Ldrh(temp1, HeapOperand(temp1, primitive_offset));
2454 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2455 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002456 }
Roland Levillain0b671c02016-08-19 12:02:34 +01002457
2458 // /* HeapReference<Class> */ temp1 = dest->klass_
2459 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2460 temp1_loc,
2461 dest.W(),
2462 class_offset,
2463 temp2,
2464 /* needs_null_check */ false,
2465 /* use_load_acquire */ false);
2466
2467 if (!optimizations.GetDestinationIsNonPrimitiveArray()) {
2468 // Bail out if the destination is not a non primitive array.
2469 //
2470 // Register `temp1` is not trashed by the read barrier emitted
2471 // by GenerateFieldLoadWithBakerReadBarrier below, as that
2472 // method produces a call to a ReadBarrierMarkRegX entry point,
2473 // which saves all potentially live registers, including
2474 // temporaries such a `temp1`.
2475 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2476 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2477 temp2_loc,
2478 temp1,
2479 component_offset,
2480 temp3,
2481 /* needs_null_check */ false,
2482 /* use_load_acquire */ false);
2483 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2484 // If heap poisoning is enabled, `temp2` has been unpoisoned
2485 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2486 // /* uint16_t */ temp2 = static_cast<uint16>(temp2->primitive_type_);
2487 __ Ldrh(temp2, HeapOperand(temp2, primitive_offset));
2488 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2489 __ Cbnz(temp2, intrinsic_slow_path->GetEntryLabel());
2490 }
2491
2492 // For the same reason given earlier, `temp1` is not trashed by the
2493 // read barrier emitted by GenerateFieldLoadWithBakerReadBarrier below.
2494 // /* HeapReference<Class> */ temp2 = src->klass_
2495 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2496 temp2_loc,
2497 src.W(),
2498 class_offset,
2499 temp3,
2500 /* needs_null_check */ false,
2501 /* use_load_acquire */ false);
2502 // Note: if heap poisoning is on, we are comparing two unpoisoned references here.
2503 __ Cmp(temp1, temp2);
2504
2505 if (optimizations.GetDestinationIsTypedObjectArray()) {
2506 vixl::aarch64::Label do_copy;
2507 __ B(&do_copy, eq);
2508 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2509 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2510 temp1_loc,
2511 temp1,
2512 component_offset,
2513 temp2,
2514 /* needs_null_check */ false,
2515 /* use_load_acquire */ false);
2516 // /* HeapReference<Class> */ temp1 = temp1->super_class_
2517 // We do not need to emit a read barrier for the following
2518 // heap reference load, as `temp1` is only used in a
2519 // comparison with null below, and this reference is not
2520 // kept afterwards.
2521 __ Ldr(temp1, HeapOperand(temp1, super_offset));
2522 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
2523 __ Bind(&do_copy);
2524 } else {
2525 __ B(intrinsic_slow_path->GetEntryLabel(), ne);
2526 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002527 } else {
Roland Levillain0b671c02016-08-19 12:02:34 +01002528 // Non read barrier code.
2529
2530 // /* HeapReference<Class> */ temp1 = dest->klass_
2531 __ Ldr(temp1, MemOperand(dest, class_offset));
2532 // /* HeapReference<Class> */ temp2 = src->klass_
2533 __ Ldr(temp2, MemOperand(src, class_offset));
2534 bool did_unpoison = false;
2535 if (!optimizations.GetDestinationIsNonPrimitiveArray() ||
2536 !optimizations.GetSourceIsNonPrimitiveArray()) {
2537 // One or two of the references need to be unpoisoned. Unpoison them
2538 // both to make the identity check valid.
2539 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2540 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp2);
2541 did_unpoison = true;
2542 }
2543
2544 if (!optimizations.GetDestinationIsNonPrimitiveArray()) {
2545 // Bail out if the destination is not a non primitive array.
2546 // /* HeapReference<Class> */ temp3 = temp1->component_type_
2547 __ Ldr(temp3, HeapOperand(temp1, component_offset));
2548 __ Cbz(temp3, intrinsic_slow_path->GetEntryLabel());
2549 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp3);
2550 // /* uint16_t */ temp3 = static_cast<uint16>(temp3->primitive_type_);
2551 __ Ldrh(temp3, HeapOperand(temp3, primitive_offset));
2552 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2553 __ Cbnz(temp3, intrinsic_slow_path->GetEntryLabel());
2554 }
2555
2556 if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2557 // Bail out if the source is not a non primitive array.
2558 // /* HeapReference<Class> */ temp3 = temp2->component_type_
2559 __ Ldr(temp3, HeapOperand(temp2, component_offset));
2560 __ Cbz(temp3, intrinsic_slow_path->GetEntryLabel());
2561 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp3);
2562 // /* uint16_t */ temp3 = static_cast<uint16>(temp3->primitive_type_);
2563 __ Ldrh(temp3, HeapOperand(temp3, primitive_offset));
2564 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2565 __ Cbnz(temp3, intrinsic_slow_path->GetEntryLabel());
2566 }
2567
2568 __ Cmp(temp1, temp2);
2569
2570 if (optimizations.GetDestinationIsTypedObjectArray()) {
2571 vixl::aarch64::Label do_copy;
2572 __ B(&do_copy, eq);
2573 if (!did_unpoison) {
2574 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2575 }
2576 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2577 __ Ldr(temp1, HeapOperand(temp1, component_offset));
2578 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2579 // /* HeapReference<Class> */ temp1 = temp1->super_class_
2580 __ Ldr(temp1, HeapOperand(temp1, super_offset));
2581 // No need to unpoison the result, we're comparing against null.
2582 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
2583 __ Bind(&do_copy);
2584 } else {
2585 __ B(intrinsic_slow_path->GetEntryLabel(), ne);
2586 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002587 }
2588 } else if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2589 DCHECK(optimizations.GetDestinationIsNonPrimitiveArray());
2590 // Bail out if the source is not a non primitive array.
Roland Levillain0b671c02016-08-19 12:02:34 +01002591 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2592 // /* HeapReference<Class> */ temp1 = src->klass_
2593 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2594 temp1_loc,
2595 src.W(),
2596 class_offset,
2597 temp2,
2598 /* needs_null_check */ false,
2599 /* use_load_acquire */ false);
2600 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2601 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2602 temp2_loc,
2603 temp1,
2604 component_offset,
2605 temp3,
2606 /* needs_null_check */ false,
2607 /* use_load_acquire */ false);
2608 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2609 // If heap poisoning is enabled, `temp2` has been unpoisoned
2610 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2611 } else {
2612 // /* HeapReference<Class> */ temp1 = src->klass_
2613 __ Ldr(temp1, HeapOperand(src.W(), class_offset));
2614 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2615 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2616 __ Ldr(temp2, HeapOperand(temp1, component_offset));
2617 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2618 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp2);
2619 }
2620 // /* uint16_t */ temp2 = static_cast<uint16>(temp2->primitive_type_);
2621 __ Ldrh(temp2, HeapOperand(temp2, primitive_offset));
donghui.baic2ec9ad2016-03-10 14:02:55 +08002622 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
Roland Levillain0b671c02016-08-19 12:02:34 +01002623 __ Cbnz(temp2, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002624 }
2625
2626 Register src_curr_addr = temp1.X();
2627 Register dst_curr_addr = temp2.X();
Roland Levillain0b671c02016-08-19 12:02:34 +01002628 Register src_stop_addr;
2629 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2630 // Temporary register IP0, obtained from the VIXL scratch
2631 // register pool as `temp3`, cannot be used in
2632 // ReadBarrierSystemArrayCopySlowPathARM64 (because that
2633 // register is clobbered by ReadBarrierMarkRegX entry points).
2634 // So another temporary register allocated by the register
2635 // allocator instead.
2636 DCHECK_EQ(LocationFrom(temp3).reg(), IP0);
2637 src_stop_addr = XRegisterFrom(locations->GetTemp(2));
2638 } else {
2639 src_stop_addr = temp3.X();
2640 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002641
2642 GenSystemArrayCopyAddresses(masm,
2643 Primitive::kPrimNot,
2644 src,
2645 src_pos,
2646 dest,
2647 dest_pos,
2648 length,
2649 src_curr_addr,
2650 dst_curr_addr,
2651 src_stop_addr);
2652
donghui.baic2ec9ad2016-03-10 14:02:55 +08002653 const int32_t element_size = Primitive::ComponentSize(Primitive::kPrimNot);
Roland Levillain0b671c02016-08-19 12:02:34 +01002654
2655 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2656 // SystemArrayCopy implementation for Baker read barriers (see
2657 // also CodeGeneratorARM::GenerateReferenceLoadWithBakerReadBarrier):
2658 //
2659 // if (src_ptr != end_ptr) {
2660 // uint32_t rb_state = Lockword(src->monitor_).ReadBarrierState();
2661 // lfence; // Load fence or artificial data dependency to prevent load-load reordering
2662 // bool is_gray = (rb_state == ReadBarrier::gray_ptr_);
2663 // if (is_gray) {
2664 // // Slow-path copy.
2665 // do {
2666 // *dest_ptr++ = MaybePoison(ReadBarrier::Mark(MaybeUnpoison(*src_ptr++)));
2667 // } while (src_ptr != end_ptr)
2668 // } else {
2669 // // Fast-path copy.
2670 // do {
2671 // *dest_ptr++ = *src_ptr++;
2672 // } while (src_ptr != end_ptr)
2673 // }
2674 // }
2675
2676 vixl::aarch64::Label loop, done;
2677
2678 // Don't enter copy loop if `length == 0`.
2679 __ Cmp(src_curr_addr, src_stop_addr);
2680 __ B(&done, eq);
2681
donghui.baic2ec9ad2016-03-10 14:02:55 +08002682 Register tmp = temps.AcquireW();
Roland Levillain0b671c02016-08-19 12:02:34 +01002683 // Make sure `tmp` is not IP0, as it is clobbered by
2684 // ReadBarrierMarkRegX entry points in
2685 // ReadBarrierSystemArrayCopySlowPathARM64.
2686 DCHECK_NE(LocationFrom(tmp).reg(), IP0);
2687
2688 // /* int32_t */ monitor = src->monitor_
2689 __ Ldr(tmp, HeapOperand(src.W(), monitor_offset));
2690 // /* LockWord */ lock_word = LockWord(monitor)
2691 static_assert(sizeof(LockWord) == sizeof(int32_t),
2692 "art::LockWord and int32_t have different sizes.");
2693
2694 // Introduce a dependency on the lock_word including rb_state,
2695 // to prevent load-load reordering, and without using
2696 // a memory barrier (which would be more expensive).
2697 // `src` is unchanged by this operation, but its value now depends
2698 // on `tmp`.
2699 __ Add(src.X(), src.X(), Operand(tmp.X(), LSR, 32));
2700
2701 // Slow path used to copy array when `src` is gray.
2702 SlowPathCodeARM64* read_barrier_slow_path =
2703 new (GetAllocator()) ReadBarrierSystemArrayCopySlowPathARM64(invoke, LocationFrom(tmp));
2704 codegen_->AddSlowPath(read_barrier_slow_path);
2705
2706 // Given the numeric representation, it's enough to check the low bit of the rb_state.
2707 static_assert(ReadBarrier::white_ptr_ == 0, "Expecting white to have value 0");
2708 static_assert(ReadBarrier::gray_ptr_ == 1, "Expecting gray to have value 1");
2709 static_assert(ReadBarrier::black_ptr_ == 2, "Expecting black to have value 2");
2710 __ Tbnz(tmp, LockWord::kReadBarrierStateShift, read_barrier_slow_path->GetEntryLabel());
2711
2712 // Fast-path copy.
2713 // Iterate over the arrays and do a raw copy of the objects. We don't need to
2714 // poison/unpoison.
2715 __ Bind(&loop);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002716 __ Ldr(tmp, MemOperand(src_curr_addr, element_size, PostIndex));
2717 __ Str(tmp, MemOperand(dst_curr_addr, element_size, PostIndex));
Roland Levillain0b671c02016-08-19 12:02:34 +01002718 __ Cmp(src_curr_addr, src_stop_addr);
2719 __ B(&loop, ne);
2720
2721 __ Bind(read_barrier_slow_path->GetExitLabel());
2722 __ Bind(&done);
2723 } else {
2724 // Non read barrier code.
2725
2726 // Iterate over the arrays and do a raw copy of the objects. We don't need to
2727 // poison/unpoison.
2728 vixl::aarch64::Label loop, done;
2729 __ Bind(&loop);
2730 __ Cmp(src_curr_addr, src_stop_addr);
2731 __ B(&done, eq);
2732 {
2733 Register tmp = temps.AcquireW();
2734 __ Ldr(tmp, MemOperand(src_curr_addr, element_size, PostIndex));
2735 __ Str(tmp, MemOperand(dst_curr_addr, element_size, PostIndex));
2736 }
2737 __ B(&loop);
2738 __ Bind(&done);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002739 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002740 }
2741 // We only need one card marking on the destination array.
2742 codegen_->MarkGCCard(dest.W(), Register(), /* value_can_be_null */ false);
2743
Roland Levillain0b671c02016-08-19 12:02:34 +01002744 __ Bind(intrinsic_slow_path->GetExitLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002745}
2746
Anton Kirilova3ffea22016-04-07 17:02:37 +01002747static void GenIsInfinite(LocationSummary* locations,
2748 bool is64bit,
Scott Wakeling97c72b72016-06-24 16:19:36 +01002749 MacroAssembler* masm) {
Anton Kirilova3ffea22016-04-07 17:02:37 +01002750 Operand infinity;
2751 Register out;
2752
2753 if (is64bit) {
2754 infinity = kPositiveInfinityDouble;
2755 out = XRegisterFrom(locations->Out());
2756 } else {
2757 infinity = kPositiveInfinityFloat;
2758 out = WRegisterFrom(locations->Out());
2759 }
2760
Scott Wakeling97c72b72016-06-24 16:19:36 +01002761 const Register zero = vixl::aarch64::Assembler::AppropriateZeroRegFor(out);
Anton Kirilova3ffea22016-04-07 17:02:37 +01002762
2763 MoveFPToInt(locations, is64bit, masm);
2764 __ Eor(out, out, infinity);
2765 // We don't care about the sign bit, so shift left.
2766 __ Cmp(zero, Operand(out, LSL, 1));
2767 __ Cset(out, eq);
2768}
2769
2770void IntrinsicLocationsBuilderARM64::VisitFloatIsInfinite(HInvoke* invoke) {
2771 CreateFPToIntLocations(arena_, invoke);
2772}
2773
2774void IntrinsicCodeGeneratorARM64::VisitFloatIsInfinite(HInvoke* invoke) {
2775 GenIsInfinite(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
2776}
2777
2778void IntrinsicLocationsBuilderARM64::VisitDoubleIsInfinite(HInvoke* invoke) {
2779 CreateFPToIntLocations(arena_, invoke);
2780}
2781
2782void IntrinsicCodeGeneratorARM64::VisitDoubleIsInfinite(HInvoke* invoke) {
2783 GenIsInfinite(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
2784}
2785
Aart Bik2f9fcc92016-03-01 15:16:54 -08002786UNIMPLEMENTED_INTRINSIC(ARM64, ReferenceGetReferent)
Aart Bik2f9fcc92016-03-01 15:16:54 -08002787UNIMPLEMENTED_INTRINSIC(ARM64, IntegerHighestOneBit)
2788UNIMPLEMENTED_INTRINSIC(ARM64, LongHighestOneBit)
2789UNIMPLEMENTED_INTRINSIC(ARM64, IntegerLowestOneBit)
2790UNIMPLEMENTED_INTRINSIC(ARM64, LongLowestOneBit)
Andreas Gampe878d58c2015-01-15 23:24:00 -08002791
Aart Bik0e54c012016-03-04 12:08:31 -08002792// 1.8.
2793UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndAddInt)
2794UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndAddLong)
2795UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetInt)
2796UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetLong)
2797UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetObject)
Aart Bik0e54c012016-03-04 12:08:31 -08002798
Aart Bik2f9fcc92016-03-01 15:16:54 -08002799UNREACHABLE_INTRINSICS(ARM64)
Roland Levillain4d027112015-07-01 15:41:14 +01002800
2801#undef __
2802
Andreas Gampe878d58c2015-01-15 23:24:00 -08002803} // namespace arm64
2804} // namespace art