blob: e2c1802fdc3f52421b3e943e80bcfa2560064475 [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,
866 /* offset */ 0U,
867 /* index */ offset_loc,
868 /* scale_factor */ 0U,
869 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());
883 codegen->MaybeGenerateReadBarrierSlow(invoke, trg_loc, trg_loc, base_loc, 0U, offset_loc);
884 }
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 Levillain22ccc3a2015-11-24 13:10:05 +0000893 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(),
904 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) {
Andreas Gampe878d58c2015-01-15 23:24:00 -08001089 LocationSummary* locations = new (arena) LocationSummary(invoke,
1090 LocationSummary::kNoCall,
1091 kIntrinsified);
1092 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
1093 locations->SetInAt(1, Location::RequiresRegister());
1094 locations->SetInAt(2, Location::RequiresRegister());
1095 locations->SetInAt(3, Location::RequiresRegister());
1096 locations->SetInAt(4, Location::RequiresRegister());
1097
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001098 // If heap poisoning is enabled, we don't want the unpoisoning
1099 // operations to potentially clobber the output.
1100 Location::OutputOverlap overlaps = (kPoisonHeapReferences && type == Primitive::kPrimNot)
1101 ? Location::kOutputOverlap
1102 : Location::kNoOutputOverlap;
1103 locations->SetOut(Location::RequiresRegister(), overlaps);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001104}
1105
1106static void GenCas(LocationSummary* locations, Primitive::Type type, CodeGeneratorARM64* codegen) {
Alexandre Rames087930f2016-08-02 13:45:28 +01001107 MacroAssembler* masm = codegen->GetVIXLAssembler();
Andreas Gampe878d58c2015-01-15 23:24:00 -08001108
1109 Register out = WRegisterFrom(locations->Out()); // Boolean result.
1110
1111 Register base = WRegisterFrom(locations->InAt(1)); // Object pointer.
1112 Register offset = XRegisterFrom(locations->InAt(2)); // Long offset.
1113 Register expected = RegisterFrom(locations->InAt(3), type); // Expected.
1114 Register value = RegisterFrom(locations->InAt(4), type); // Value.
1115
1116 // This needs to be before the temp registers, as MarkGCCard also uses VIXL temps.
1117 if (type == Primitive::kPrimNot) {
1118 // Mark card for object assuming new value is stored.
Nicolas Geoffray07276db2015-05-18 14:22:09 +01001119 bool value_can_be_null = true; // TODO: Worth finding out this information?
1120 codegen->MarkGCCard(base, value, value_can_be_null);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001121 }
1122
1123 UseScratchRegisterScope temps(masm);
1124 Register tmp_ptr = temps.AcquireX(); // Pointer to actual memory.
1125 Register tmp_value = temps.AcquireSameSizeAs(value); // Value in memory.
1126
1127 Register tmp_32 = tmp_value.W();
1128
1129 __ Add(tmp_ptr, base.X(), Operand(offset));
1130
Roland Levillain4d027112015-07-01 15:41:14 +01001131 if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
1132 codegen->GetAssembler()->PoisonHeapReference(expected);
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001133 if (value.Is(expected)) {
1134 // Do not poison `value`, as it is the same register as
1135 // `expected`, which has just been poisoned.
1136 } else {
1137 codegen->GetAssembler()->PoisonHeapReference(value);
1138 }
Roland Levillain4d027112015-07-01 15:41:14 +01001139 }
1140
Andreas Gampe878d58c2015-01-15 23:24:00 -08001141 // do {
1142 // tmp_value = [tmp_ptr] - expected;
1143 // } while (tmp_value == 0 && failure([tmp_ptr] <- r_new_value));
1144 // result = tmp_value != 0;
1145
Scott Wakeling97c72b72016-06-24 16:19:36 +01001146 vixl::aarch64::Label loop_head, exit_loop;
Serban Constantinescu4a6a67c2016-01-27 09:19:56 +00001147 __ Bind(&loop_head);
1148 // TODO: When `type == Primitive::kPrimNot`, add a read barrier for
1149 // the reference stored in the object before attempting the CAS,
1150 // similar to the one in the art::Unsafe_compareAndSwapObject JNI
1151 // implementation.
1152 //
1153 // Note that this code is not (yet) used when read barriers are
1154 // enabled (see IntrinsicLocationsBuilderARM64::VisitUnsafeCASObject).
1155 DCHECK(!(type == Primitive::kPrimNot && kEmitCompilerReadBarrier));
1156 __ Ldaxr(tmp_value, MemOperand(tmp_ptr));
1157 __ Cmp(tmp_value, expected);
1158 __ B(&exit_loop, ne);
1159 __ Stlxr(tmp_32, value, MemOperand(tmp_ptr));
1160 __ Cbnz(tmp_32, &loop_head);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001161 __ Bind(&exit_loop);
1162 __ Cset(out, eq);
Roland Levillain4d027112015-07-01 15:41:14 +01001163
1164 if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
Roland Levillain4d027112015-07-01 15:41:14 +01001165 codegen->GetAssembler()->UnpoisonHeapReference(expected);
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001166 if (value.Is(expected)) {
1167 // Do not unpoison `value`, as it is the same register as
1168 // `expected`, which has just been unpoisoned.
1169 } else {
1170 codegen->GetAssembler()->UnpoisonHeapReference(value);
1171 }
Roland Levillain4d027112015-07-01 15:41:14 +01001172 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001173}
1174
1175void IntrinsicLocationsBuilderARM64::VisitUnsafeCASInt(HInvoke* invoke) {
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001176 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimInt);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001177}
1178void IntrinsicLocationsBuilderARM64::VisitUnsafeCASLong(HInvoke* invoke) {
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001179 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimLong);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001180}
1181void IntrinsicLocationsBuilderARM64::VisitUnsafeCASObject(HInvoke* invoke) {
Roland Levillain391b8662015-12-18 11:43:38 +00001182 // The UnsafeCASObject intrinsic is missing a read barrier, and
1183 // therefore sometimes does not work as expected (b/25883050).
1184 // Turn it off temporarily as a quick fix, until the read barrier is
Roland Levillain3d312422016-06-23 13:53:42 +01001185 // implemented (see TODO in GenCAS).
Roland Levillain391b8662015-12-18 11:43:38 +00001186 //
Roland Levillain3d312422016-06-23 13:53:42 +01001187 // TODO(rpl): Implement read barrier support in GenCAS and re-enable
1188 // this intrinsic.
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001189 if (kEmitCompilerReadBarrier) {
Roland Levillain985ff702015-10-23 13:25:35 +01001190 return;
1191 }
1192
Roland Levillain2e50ecb2016-01-27 14:08:33 +00001193 CreateIntIntIntIntIntToInt(arena_, invoke, Primitive::kPrimNot);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001194}
1195
1196void IntrinsicCodeGeneratorARM64::VisitUnsafeCASInt(HInvoke* invoke) {
1197 GenCas(invoke->GetLocations(), Primitive::kPrimInt, codegen_);
1198}
1199void IntrinsicCodeGeneratorARM64::VisitUnsafeCASLong(HInvoke* invoke) {
1200 GenCas(invoke->GetLocations(), Primitive::kPrimLong, codegen_);
1201}
1202void IntrinsicCodeGeneratorARM64::VisitUnsafeCASObject(HInvoke* invoke) {
Roland Levillain3d312422016-06-23 13:53:42 +01001203 // The UnsafeCASObject intrinsic is missing a read barrier, and
1204 // therefore sometimes does not work as expected (b/25883050).
1205 // Turn it off temporarily as a quick fix, until the read barrier is
1206 // implemented (see TODO in GenCAS).
1207 //
1208 // TODO(rpl): Implement read barrier support in GenCAS and re-enable
1209 // this intrinsic.
1210 DCHECK(!kEmitCompilerReadBarrier);
1211
Andreas Gampe878d58c2015-01-15 23:24:00 -08001212 GenCas(invoke->GetLocations(), Primitive::kPrimNot, codegen_);
1213}
1214
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001215void IntrinsicLocationsBuilderARM64::VisitStringCompareTo(HInvoke* invoke) {
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001216 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Scott Wakeling1f36f412016-04-21 11:13:45 +01001217 invoke->InputAt(1)->CanBeNull()
1218 ? LocationSummary::kCallOnSlowPath
1219 : LocationSummary::kNoCall,
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001220 kIntrinsified);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001221 locations->SetInAt(0, Location::RequiresRegister());
1222 locations->SetInAt(1, Location::RequiresRegister());
1223 locations->AddTemp(Location::RequiresRegister());
1224 locations->AddTemp(Location::RequiresRegister());
1225 locations->AddTemp(Location::RequiresRegister());
jessicahandojo05765752016-09-09 19:01:32 -07001226 // Need temporary registers for String compression's feature.
1227 if (mirror::kUseStringCompression) {
1228 locations->AddTemp(Location::RequiresRegister());
1229 locations->AddTemp(Location::RequiresRegister());
1230 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001231 locations->SetOut(Location::RequiresRegister(), Location::kOutputOverlap);
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001232}
1233
1234void IntrinsicCodeGeneratorARM64::VisitStringCompareTo(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001235 MacroAssembler* masm = GetVIXLAssembler();
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001236 LocationSummary* locations = invoke->GetLocations();
1237
Alexandre Rames2ea91532016-08-11 17:04:14 +01001238 Register str = InputRegisterAt(invoke, 0);
1239 Register arg = InputRegisterAt(invoke, 1);
1240 DCHECK(str.IsW());
1241 DCHECK(arg.IsW());
Scott Wakeling1f36f412016-04-21 11:13:45 +01001242 Register out = OutputRegister(invoke);
1243
1244 Register temp0 = WRegisterFrom(locations->GetTemp(0));
1245 Register temp1 = WRegisterFrom(locations->GetTemp(1));
1246 Register temp2 = WRegisterFrom(locations->GetTemp(2));
jessicahandojo05765752016-09-09 19:01:32 -07001247 Register temp3, temp5;
1248 if (mirror::kUseStringCompression) {
1249 temp3 = WRegisterFrom(locations->GetTemp(3));
1250 temp5 = WRegisterFrom(locations->GetTemp(4));
1251 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001252
Scott Wakeling97c72b72016-06-24 16:19:36 +01001253 vixl::aarch64::Label loop;
1254 vixl::aarch64::Label find_char_diff;
1255 vixl::aarch64::Label end;
jessicahandojo05765752016-09-09 19:01:32 -07001256 vixl::aarch64::Label different_compression;
Scott Wakeling1f36f412016-04-21 11:13:45 +01001257
1258 // Get offsets of count and value fields within a string object.
1259 const int32_t count_offset = mirror::String::CountOffset().Int32Value();
1260 const int32_t value_offset = mirror::String::ValueOffset().Int32Value();
1261
Nicolas Geoffray512e04d2015-03-27 17:21:24 +00001262 // Note that the null check must have been done earlier.
Calin Juravle641547a2015-04-21 22:08:51 +01001263 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001264
Scott Wakeling1f36f412016-04-21 11:13:45 +01001265 // Take slow path and throw if input can be and is null.
1266 SlowPathCodeARM64* slow_path = nullptr;
1267 const bool can_slow_path = invoke->InputAt(1)->CanBeNull();
1268 if (can_slow_path) {
1269 slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1270 codegen_->AddSlowPath(slow_path);
1271 __ Cbz(arg, slow_path->GetEntryLabel());
1272 }
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001273
Scott Wakeling1f36f412016-04-21 11:13:45 +01001274 // Reference equality check, return 0 if same reference.
1275 __ Subs(out, str, arg);
1276 __ B(&end, eq);
jessicahandojo05765752016-09-09 19:01:32 -07001277 if (mirror::kUseStringCompression) {
1278 // Load lengths of this and argument strings.
1279 __ Ldr(temp3, HeapOperand(str, count_offset));
1280 __ Ldr(temp5, HeapOperand(arg, count_offset));
1281 // Clean out compression flag from lengths.
1282 __ Bic(temp0, temp3, Operand(static_cast<int32_t>(0x80000000)));
1283 __ Bic(temp1, temp5, Operand(static_cast<int32_t>(0x80000000)));
1284 } else {
1285 // Load lengths of this and argument strings.
1286 __ Ldr(temp0, HeapOperand(str, count_offset));
1287 __ Ldr(temp1, HeapOperand(arg, count_offset));
1288 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001289 // Return zero if both strings are empty.
1290 __ Orr(out, temp0, temp1);
1291 __ Cbz(out, &end);
1292 // out = length diff.
1293 __ Subs(out, temp0, temp1);
1294 // temp2 = min(len(str), len(arg)).
1295 __ Csel(temp2, temp1, temp0, ge);
1296 // Shorter string is empty?
1297 __ Cbz(temp2, &end);
1298
jessicahandojo05765752016-09-09 19:01:32 -07001299 if (mirror::kUseStringCompression) {
1300 // Check if both strings using same compression style to use this comparison loop.
1301 __ Eor(temp3.W(), temp3, Operand(temp5));
1302 __ Tbnz(temp3.W(), kWRegSize - 1, &different_compression);
1303 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001304 // Store offset of string value in preparation for comparison loop.
1305 __ Mov(temp1, value_offset);
jessicahandojo05765752016-09-09 19:01:32 -07001306 if (mirror::kUseStringCompression) {
1307 // For string compression, calculate the number of bytes to compare (not chars).
1308 // This could be in theory exceed INT32_MAX, so treat temp2 as unsigned.
1309 vixl::aarch64::Label let_it_signed;
1310 __ Cmp(temp5, Operand(0));
1311 __ B(lt, &let_it_signed);
1312 __ Add(temp2, temp2, Operand(temp2));
1313 __ Bind(&let_it_signed);
1314 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001315
1316 UseScratchRegisterScope scratch_scope(masm);
1317 Register temp4 = scratch_scope.AcquireX();
1318
1319 // Assertions that must hold in order to compare strings 4 characters at a time.
1320 DCHECK_ALIGNED(value_offset, 8);
1321 static_assert(IsAligned<8>(kObjectAlignment), "String of odd length is not zero padded");
1322
1323 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1324 DCHECK_EQ(char_size, 2u);
1325
1326 // Promote temp0 to an X reg, ready for LDR.
1327 temp0 = temp0.X();
1328
1329 // Loop to compare 4x16-bit characters at a time (ok because of string data alignment).
1330 __ Bind(&loop);
Alexandre Rames2ea91532016-08-11 17:04:14 +01001331 __ Ldr(temp4, MemOperand(str.X(), temp1.X()));
1332 __ Ldr(temp0, MemOperand(arg.X(), temp1.X()));
Scott Wakeling1f36f412016-04-21 11:13:45 +01001333 __ Cmp(temp4, temp0);
1334 __ B(ne, &find_char_diff);
1335 __ Add(temp1, temp1, char_size * 4);
jessicahandojo05765752016-09-09 19:01:32 -07001336 // With string compression, we have compared 8 bytes, otherwise 4 chars.
1337 __ Subs(temp2, temp2, (mirror::kUseStringCompression) ? 8 : 4);
1338 __ B(hi, &loop);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001339 __ B(&end);
1340
1341 // Promote temp1 to an X reg, ready for EOR.
1342 temp1 = temp1.X();
1343
jessicahandojo05765752016-09-09 19:01:32 -07001344 // Find the single character difference.
Scott Wakeling1f36f412016-04-21 11:13:45 +01001345 __ Bind(&find_char_diff);
1346 // Get the bit position of the first character that differs.
1347 __ Eor(temp1, temp0, temp4);
1348 __ Rbit(temp1, temp1);
1349 __ Clz(temp1, temp1);
jessicahandojo05765752016-09-09 19:01:32 -07001350 // If the number of chars remaining <= the index where the difference occurs (0-3), then
Scott Wakeling1f36f412016-04-21 11:13:45 +01001351 // the difference occurs outside the remaining string data, so just return length diff (out).
jessicahandojo05765752016-09-09 19:01:32 -07001352 // Unlike ARM, we're doing the comparison in one go here, without the subtraction at the
1353 // find_char_diff_2nd_cmp path, so it doesn't matter whether the comparison is signed or
1354 // unsigned when string compression is disabled.
1355 // When it's enabled, the comparison must be unsigned.
1356 __ Cmp(temp2, Operand(temp1.W(), LSR, (mirror::kUseStringCompression) ? 3 : 4));
1357 __ B(ls, &end);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001358 // Extract the characters and calculate the difference.
jessicahandojo05765752016-09-09 19:01:32 -07001359 vixl::aarch64::Label uncompressed_string, continue_process;
1360 if (mirror:: kUseStringCompression) {
1361 __ Tbz(temp5, kWRegSize - 1, &uncompressed_string);
1362 __ Bic(temp1, temp1, 0x7);
1363 __ B(&continue_process);
1364 }
1365 __ Bind(&uncompressed_string);
Scott Wakelinge5ed20b2016-05-20 10:41:38 +01001366 __ Bic(temp1, temp1, 0xf);
jessicahandojo05765752016-09-09 19:01:32 -07001367 __ Bind(&continue_process);
1368
Scott Wakeling1f36f412016-04-21 11:13:45 +01001369 __ Lsr(temp0, temp0, temp1);
1370 __ Lsr(temp4, temp4, temp1);
jessicahandojo05765752016-09-09 19:01:32 -07001371 vixl::aarch64::Label uncompressed_string_extract_chars;
1372 if (mirror::kUseStringCompression) {
1373 __ Tbz(temp5, kWRegSize - 1, &uncompressed_string_extract_chars);
1374 __ And(temp4, temp4, 0xff);
1375 __ Sub(out, temp4.W(), Operand(temp0.W(), UXTB));
1376 __ B(&end);
1377 }
1378 __ Bind(&uncompressed_string_extract_chars);
Scott Wakeling1f36f412016-04-21 11:13:45 +01001379 __ And(temp4, temp4, 0xffff);
Alexandre Rames2ea91532016-08-11 17:04:14 +01001380 __ Sub(out, temp4.W(), Operand(temp0.W(), UXTH));
jessicahandojo05765752016-09-09 19:01:32 -07001381 __ B(&end);
1382
1383 if (mirror::kUseStringCompression) {
1384 vixl::aarch64::Label loop_this_compressed, loop_arg_compressed, find_diff;
1385 const size_t c_char_size = Primitive::ComponentSize(Primitive::kPrimByte);
1386 DCHECK_EQ(c_char_size, 1u);
1387 temp0 = temp0.W();
1388 temp1 = temp1.W();
1389 // Comparison for different compression style.
1390 // This part is when THIS is compressed and ARG is not.
1391 __ Bind(&different_compression);
1392 __ Add(temp0, str, Operand(value_offset));
1393 __ Add(temp1, arg, Operand(value_offset));
1394 __ Cmp(temp5, Operand(0));
1395 __ B(lt, &loop_arg_compressed);
1396
1397 __ Bind(&loop_this_compressed);
1398 __ Ldrb(temp3, MemOperand(temp0.X(), c_char_size, PostIndex));
1399 __ Ldrh(temp5, MemOperand(temp1.X(), char_size, PostIndex));
1400 __ Cmp(temp3, Operand(temp5));
1401 __ B(ne, &find_diff);
1402 __ Subs(temp2, temp2, 1);
1403 __ B(gt, &loop_this_compressed);
1404 __ B(&end);
1405
1406 // This part is when THIS is not compressed and ARG is.
1407 __ Bind(&loop_arg_compressed);
1408 __ Ldrh(temp3, MemOperand(temp0.X(), char_size, PostIndex));
1409 __ Ldrb(temp5, MemOperand(temp1.X(), c_char_size, PostIndex));
1410 __ Cmp(temp3, Operand(temp5));
1411 __ B(ne, &find_diff);
1412 __ Subs(temp2, temp2, 1);
1413 __ B(gt, &loop_arg_compressed);
1414 __ B(&end);
1415
1416 // Calculate the difference.
1417 __ Bind(&find_diff);
1418 __ Sub(out, temp3.W(), Operand(temp5.W(), UXTH));
1419 }
Scott Wakeling1f36f412016-04-21 11:13:45 +01001420
1421 __ Bind(&end);
1422
1423 if (can_slow_path) {
1424 __ Bind(slow_path->GetExitLabel());
1425 }
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001426}
1427
Agi Csakiea34b402015-08-13 17:51:19 -07001428void IntrinsicLocationsBuilderARM64::VisitStringEquals(HInvoke* invoke) {
1429 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1430 LocationSummary::kNoCall,
1431 kIntrinsified);
1432 locations->SetInAt(0, Location::RequiresRegister());
1433 locations->SetInAt(1, Location::RequiresRegister());
1434 // Temporary registers to store lengths of strings and for calculations.
1435 locations->AddTemp(Location::RequiresRegister());
1436 locations->AddTemp(Location::RequiresRegister());
1437
1438 locations->SetOut(Location::RequiresRegister(), Location::kOutputOverlap);
1439}
1440
1441void IntrinsicCodeGeneratorARM64::VisitStringEquals(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001442 MacroAssembler* masm = GetVIXLAssembler();
Agi Csakiea34b402015-08-13 17:51:19 -07001443 LocationSummary* locations = invoke->GetLocations();
1444
1445 Register str = WRegisterFrom(locations->InAt(0));
1446 Register arg = WRegisterFrom(locations->InAt(1));
1447 Register out = XRegisterFrom(locations->Out());
1448
1449 UseScratchRegisterScope scratch_scope(masm);
1450 Register temp = scratch_scope.AcquireW();
1451 Register temp1 = WRegisterFrom(locations->GetTemp(0));
1452 Register temp2 = WRegisterFrom(locations->GetTemp(1));
1453
jessicahandojo05765752016-09-09 19:01:32 -07001454 vixl::aarch64::Label loop, preloop;
Scott Wakeling97c72b72016-06-24 16:19:36 +01001455 vixl::aarch64::Label end;
1456 vixl::aarch64::Label return_true;
1457 vixl::aarch64::Label return_false;
Agi Csakiea34b402015-08-13 17:51:19 -07001458
1459 // Get offsets of count, value, and class fields within a string object.
1460 const int32_t count_offset = mirror::String::CountOffset().Int32Value();
1461 const int32_t value_offset = mirror::String::ValueOffset().Int32Value();
1462 const int32_t class_offset = mirror::Object::ClassOffset().Int32Value();
1463
1464 // Note that the null check must have been done earlier.
1465 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1466
Vladimir Marko53b52002016-05-24 19:30:45 +01001467 StringEqualsOptimizations optimizations(invoke);
1468 if (!optimizations.GetArgumentNotNull()) {
1469 // Check if input is null, return false if it is.
1470 __ Cbz(arg, &return_false);
1471 }
Agi Csakiea34b402015-08-13 17:51:19 -07001472
1473 // Reference equality check, return true if same reference.
1474 __ Cmp(str, arg);
1475 __ B(&return_true, eq);
1476
Vladimir Marko53b52002016-05-24 19:30:45 +01001477 if (!optimizations.GetArgumentIsString()) {
1478 // Instanceof check for the argument by comparing class fields.
1479 // All string objects must have the same type since String cannot be subclassed.
1480 // Receiver must be a string object, so its class field is equal to all strings' class fields.
1481 // If the argument is a string object, its class field must be equal to receiver's class field.
1482 __ Ldr(temp, MemOperand(str.X(), class_offset));
1483 __ Ldr(temp1, MemOperand(arg.X(), class_offset));
1484 __ Cmp(temp, temp1);
1485 __ B(&return_false, ne);
1486 }
Agi Csakiea34b402015-08-13 17:51:19 -07001487
1488 // Load lengths of this and argument strings.
1489 __ Ldr(temp, MemOperand(str.X(), count_offset));
1490 __ Ldr(temp1, MemOperand(arg.X(), count_offset));
1491 // Check if lengths are equal, return false if they're not.
jessicahandojo05765752016-09-09 19:01:32 -07001492 // Also compares the compression style, if differs return false.
Agi Csakiea34b402015-08-13 17:51:19 -07001493 __ Cmp(temp, temp1);
1494 __ B(&return_false, ne);
Agi Csakiea34b402015-08-13 17:51:19 -07001495 // Return true if both strings are empty.
jessicahandojo05765752016-09-09 19:01:32 -07001496 if (mirror::kUseStringCompression) {
1497 // Length needs to be masked out first because 0 is treated as compressed.
1498 __ Bic(temp, temp, Operand(static_cast<int32_t>(0x80000000)));
1499 }
Agi Csakiea34b402015-08-13 17:51:19 -07001500 __ Cbz(temp, &return_true);
1501
1502 // Assertions that must hold in order to compare strings 4 characters at a time.
1503 DCHECK_ALIGNED(value_offset, 8);
1504 static_assert(IsAligned<8>(kObjectAlignment), "String of odd length is not zero padded");
1505
jessicahandojo05765752016-09-09 19:01:32 -07001506 if (mirror::kUseStringCompression) {
1507 // If not compressed, directly to fast compare. Else do preprocess on length.
1508 __ Cmp(temp1, Operand(0));
1509 __ B(&preloop, gt);
1510 // Mask out compression flag and adjust length for compressed string (8-bit)
1511 // as if it is a 16-bit data, new_length = (length + 1) / 2
1512 __ Add(temp, temp, 1);
1513 __ Lsr(temp, temp, 1);
1514 }
1515
Agi Csakiea34b402015-08-13 17:51:19 -07001516 temp1 = temp1.X();
1517 temp2 = temp2.X();
Agi Csakiea34b402015-08-13 17:51:19 -07001518 // Loop to compare strings 4 characters at a time starting at the beginning of the string.
1519 // Ok to do this because strings are zero-padded to be 8-byte aligned.
jessicahandojo05765752016-09-09 19:01:32 -07001520 // Store offset of string value in preparation for comparison loop
1521 __ Bind(&preloop);
1522 __ Mov(temp1, value_offset);
Agi Csakiea34b402015-08-13 17:51:19 -07001523 __ Bind(&loop);
1524 __ Ldr(out, MemOperand(str.X(), temp1));
1525 __ Ldr(temp2, MemOperand(arg.X(), temp1));
1526 __ Add(temp1, temp1, Operand(sizeof(uint64_t)));
1527 __ Cmp(out, temp2);
1528 __ B(&return_false, ne);
1529 __ Sub(temp, temp, Operand(4), SetFlags);
1530 __ B(&loop, gt);
1531
1532 // Return true and exit the function.
1533 // If loop does not result in returning false, we return true.
1534 __ Bind(&return_true);
1535 __ Mov(out, 1);
1536 __ B(&end);
1537
1538 // Return false and exit the function.
1539 __ Bind(&return_false);
1540 __ Mov(out, 0);
1541 __ Bind(&end);
1542}
1543
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001544static void GenerateVisitStringIndexOf(HInvoke* invoke,
Scott Wakeling97c72b72016-06-24 16:19:36 +01001545 MacroAssembler* masm,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001546 CodeGeneratorARM64* codegen,
1547 ArenaAllocator* allocator,
1548 bool start_at_zero) {
1549 LocationSummary* locations = invoke->GetLocations();
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001550
1551 // Note that the null check must have been done earlier.
1552 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1553
1554 // Check for code points > 0xFFFF. Either a slow-path check when we don't know statically,
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001555 // 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 -07001556 SlowPathCodeARM64* slow_path = nullptr;
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001557 HInstruction* code_point = invoke->InputAt(1);
1558 if (code_point->IsIntConstant()) {
Vladimir Markoda051082016-05-17 16:10:20 +01001559 if (static_cast<uint32_t>(code_point->AsIntConstant()->GetValue()) > 0xFFFFU) {
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001560 // Always needs the slow-path. We could directly dispatch to it, but this case should be
1561 // rare, so for simplicity just put the full slow-path down and branch unconditionally.
1562 slow_path = new (allocator) IntrinsicSlowPathARM64(invoke);
1563 codegen->AddSlowPath(slow_path);
1564 __ B(slow_path->GetEntryLabel());
1565 __ Bind(slow_path->GetExitLabel());
1566 return;
1567 }
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001568 } else if (code_point->GetType() != Primitive::kPrimChar) {
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001569 Register char_reg = WRegisterFrom(locations->InAt(1));
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001570 __ Tst(char_reg, 0xFFFF0000);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001571 slow_path = new (allocator) IntrinsicSlowPathARM64(invoke);
1572 codegen->AddSlowPath(slow_path);
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001573 __ B(ne, slow_path->GetEntryLabel());
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001574 }
1575
1576 if (start_at_zero) {
1577 // Start-index = 0.
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001578 Register tmp_reg = WRegisterFrom(locations->GetTemp(0));
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001579 __ Mov(tmp_reg, 0);
1580 }
1581
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001582 codegen->InvokeRuntime(kQuickIndexOf, invoke, invoke->GetDexPc(), slow_path);
Roland Levillain42ad2882016-02-29 18:26:54 +00001583 CheckEntrypointTypes<kQuickIndexOf, int32_t, void*, uint32_t, uint32_t>();
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001584
1585 if (slow_path != nullptr) {
1586 __ Bind(slow_path->GetExitLabel());
1587 }
1588}
1589
1590void IntrinsicLocationsBuilderARM64::VisitStringIndexOf(HInvoke* invoke) {
1591 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001592 LocationSummary::kCallOnMainAndSlowPath,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001593 kIntrinsified);
1594 // We have a hand-crafted assembly stub that follows the runtime calling convention. So it's
1595 // best to align the inputs accordingly.
1596 InvokeRuntimeCallingConvention calling_convention;
1597 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1598 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1599 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimInt));
1600
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001601 // Need to send start_index=0.
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001602 locations->AddTemp(LocationFrom(calling_convention.GetRegisterAt(2)));
1603}
1604
1605void IntrinsicCodeGeneratorARM64::VisitStringIndexOf(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001606 GenerateVisitStringIndexOf(
1607 invoke, GetVIXLAssembler(), codegen_, GetAllocator(), /* start_at_zero */ true);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001608}
1609
1610void IntrinsicLocationsBuilderARM64::VisitStringIndexOfAfter(HInvoke* invoke) {
1611 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001612 LocationSummary::kCallOnMainAndSlowPath,
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001613 kIntrinsified);
1614 // We have a hand-crafted assembly stub that follows the runtime calling convention. So it's
1615 // best to align the inputs accordingly.
1616 InvokeRuntimeCallingConvention calling_convention;
1617 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1618 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1619 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1620 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimInt));
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001621}
1622
1623void IntrinsicCodeGeneratorARM64::VisitStringIndexOfAfter(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001624 GenerateVisitStringIndexOf(
1625 invoke, GetVIXLAssembler(), codegen_, GetAllocator(), /* start_at_zero */ false);
Andreas Gampeba6fdbc2015-05-07 22:31:55 -07001626}
1627
Jeff Hao848f70a2014-01-15 13:49:50 -08001628void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromBytes(HInvoke* invoke) {
1629 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001630 LocationSummary::kCallOnMainAndSlowPath,
Jeff Hao848f70a2014-01-15 13:49:50 -08001631 kIntrinsified);
1632 InvokeRuntimeCallingConvention calling_convention;
1633 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1634 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1635 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1636 locations->SetInAt(3, LocationFrom(calling_convention.GetRegisterAt(3)));
1637 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1638}
1639
1640void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromBytes(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001641 MacroAssembler* masm = GetVIXLAssembler();
Jeff Hao848f70a2014-01-15 13:49:50 -08001642 LocationSummary* locations = invoke->GetLocations();
1643
1644 Register byte_array = WRegisterFrom(locations->InAt(0));
1645 __ Cmp(byte_array, 0);
1646 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1647 codegen_->AddSlowPath(slow_path);
1648 __ B(eq, slow_path->GetEntryLabel());
1649
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001650 codegen_->InvokeRuntime(kQuickAllocStringFromBytes, invoke, invoke->GetDexPc(), slow_path);
Roland Levillainf969a202016-03-09 16:14:00 +00001651 CheckEntrypointTypes<kQuickAllocStringFromBytes, void*, void*, int32_t, int32_t, int32_t>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001652 __ Bind(slow_path->GetExitLabel());
1653}
1654
1655void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromChars(HInvoke* invoke) {
1656 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001657 LocationSummary::kCallOnMainOnly,
Jeff Hao848f70a2014-01-15 13:49:50 -08001658 kIntrinsified);
1659 InvokeRuntimeCallingConvention calling_convention;
1660 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
1661 locations->SetInAt(1, LocationFrom(calling_convention.GetRegisterAt(1)));
1662 locations->SetInAt(2, LocationFrom(calling_convention.GetRegisterAt(2)));
1663 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1664}
1665
1666void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromChars(HInvoke* invoke) {
Roland Levillaincc3839c2016-02-29 16:23:48 +00001667 // No need to emit code checking whether `locations->InAt(2)` is a null
1668 // pointer, as callers of the native method
1669 //
1670 // java.lang.StringFactory.newStringFromChars(int offset, int charCount, char[] data)
1671 //
1672 // all include a null check on `data` before calling that method.
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001673 codegen_->InvokeRuntime(kQuickAllocStringFromChars, invoke, invoke->GetDexPc());
Roland Levillainf969a202016-03-09 16:14:00 +00001674 CheckEntrypointTypes<kQuickAllocStringFromChars, void*, int32_t, int32_t, void*>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001675}
1676
1677void IntrinsicLocationsBuilderARM64::VisitStringNewStringFromString(HInvoke* invoke) {
Jeff Hao848f70a2014-01-15 13:49:50 -08001678 LocationSummary* locations = new (arena_) LocationSummary(invoke,
Serban Constantinescu806f0122016-03-09 11:10:16 +00001679 LocationSummary::kCallOnMainAndSlowPath,
Jeff Hao848f70a2014-01-15 13:49:50 -08001680 kIntrinsified);
1681 InvokeRuntimeCallingConvention calling_convention;
1682 locations->SetInAt(0, LocationFrom(calling_convention.GetRegisterAt(0)));
Jeff Hao848f70a2014-01-15 13:49:50 -08001683 locations->SetOut(calling_convention.GetReturnLocation(Primitive::kPrimNot));
1684}
1685
1686void IntrinsicCodeGeneratorARM64::VisitStringNewStringFromString(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001687 MacroAssembler* masm = GetVIXLAssembler();
Jeff Hao848f70a2014-01-15 13:49:50 -08001688 LocationSummary* locations = invoke->GetLocations();
1689
1690 Register string_to_copy = WRegisterFrom(locations->InAt(0));
1691 __ Cmp(string_to_copy, 0);
1692 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
1693 codegen_->AddSlowPath(slow_path);
1694 __ B(eq, slow_path->GetEntryLabel());
1695
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001696 codegen_->InvokeRuntime(kQuickAllocStringFromString, invoke, invoke->GetDexPc(), slow_path);
Roland Levillainf969a202016-03-09 16:14:00 +00001697 CheckEntrypointTypes<kQuickAllocStringFromString, void*, void*>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001698 __ Bind(slow_path->GetExitLabel());
1699}
1700
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001701static void CreateFPToFPCallLocations(ArenaAllocator* arena, HInvoke* invoke) {
1702 DCHECK_EQ(invoke->GetNumberOfArguments(), 1U);
1703 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(0)->GetType()));
1704 DCHECK(Primitive::IsFloatingPointType(invoke->GetType()));
1705
1706 LocationSummary* const locations = new (arena) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001707 LocationSummary::kCallOnMainOnly,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001708 kIntrinsified);
1709 InvokeRuntimeCallingConvention calling_convention;
1710
1711 locations->SetInAt(0, LocationFrom(calling_convention.GetFpuRegisterAt(0)));
1712 locations->SetOut(calling_convention.GetReturnLocation(invoke->GetType()));
1713}
1714
1715static void CreateFPFPToFPCallLocations(ArenaAllocator* arena, HInvoke* invoke) {
1716 DCHECK_EQ(invoke->GetNumberOfArguments(), 2U);
1717 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(0)->GetType()));
1718 DCHECK(Primitive::IsFloatingPointType(invoke->InputAt(1)->GetType()));
1719 DCHECK(Primitive::IsFloatingPointType(invoke->GetType()));
1720
1721 LocationSummary* const locations = new (arena) LocationSummary(invoke,
Serban Constantinescu54ff4822016-07-07 18:03:19 +01001722 LocationSummary::kCallOnMainOnly,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001723 kIntrinsified);
1724 InvokeRuntimeCallingConvention calling_convention;
1725
1726 locations->SetInAt(0, LocationFrom(calling_convention.GetFpuRegisterAt(0)));
1727 locations->SetInAt(1, LocationFrom(calling_convention.GetFpuRegisterAt(1)));
1728 locations->SetOut(calling_convention.GetReturnLocation(invoke->GetType()));
1729}
1730
1731static void GenFPToFPCall(HInvoke* invoke,
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001732 CodeGeneratorARM64* codegen,
1733 QuickEntrypointEnum entry) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001734 codegen->InvokeRuntime(entry, invoke, invoke->GetDexPc());
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001735}
1736
1737void IntrinsicLocationsBuilderARM64::VisitMathCos(HInvoke* invoke) {
1738 CreateFPToFPCallLocations(arena_, invoke);
1739}
1740
1741void IntrinsicCodeGeneratorARM64::VisitMathCos(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001742 GenFPToFPCall(invoke, codegen_, kQuickCos);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001743}
1744
1745void IntrinsicLocationsBuilderARM64::VisitMathSin(HInvoke* invoke) {
1746 CreateFPToFPCallLocations(arena_, invoke);
1747}
1748
1749void IntrinsicCodeGeneratorARM64::VisitMathSin(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001750 GenFPToFPCall(invoke, codegen_, kQuickSin);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001751}
1752
1753void IntrinsicLocationsBuilderARM64::VisitMathAcos(HInvoke* invoke) {
1754 CreateFPToFPCallLocations(arena_, invoke);
1755}
1756
1757void IntrinsicCodeGeneratorARM64::VisitMathAcos(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001758 GenFPToFPCall(invoke, codegen_, kQuickAcos);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001759}
1760
1761void IntrinsicLocationsBuilderARM64::VisitMathAsin(HInvoke* invoke) {
1762 CreateFPToFPCallLocations(arena_, invoke);
1763}
1764
1765void IntrinsicCodeGeneratorARM64::VisitMathAsin(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001766 GenFPToFPCall(invoke, codegen_, kQuickAsin);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001767}
1768
1769void IntrinsicLocationsBuilderARM64::VisitMathAtan(HInvoke* invoke) {
1770 CreateFPToFPCallLocations(arena_, invoke);
1771}
1772
1773void IntrinsicCodeGeneratorARM64::VisitMathAtan(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001774 GenFPToFPCall(invoke, codegen_, kQuickAtan);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001775}
1776
1777void IntrinsicLocationsBuilderARM64::VisitMathCbrt(HInvoke* invoke) {
1778 CreateFPToFPCallLocations(arena_, invoke);
1779}
1780
1781void IntrinsicCodeGeneratorARM64::VisitMathCbrt(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001782 GenFPToFPCall(invoke, codegen_, kQuickCbrt);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001783}
1784
1785void IntrinsicLocationsBuilderARM64::VisitMathCosh(HInvoke* invoke) {
1786 CreateFPToFPCallLocations(arena_, invoke);
1787}
1788
1789void IntrinsicCodeGeneratorARM64::VisitMathCosh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001790 GenFPToFPCall(invoke, codegen_, kQuickCosh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001791}
1792
1793void IntrinsicLocationsBuilderARM64::VisitMathExp(HInvoke* invoke) {
1794 CreateFPToFPCallLocations(arena_, invoke);
1795}
1796
1797void IntrinsicCodeGeneratorARM64::VisitMathExp(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001798 GenFPToFPCall(invoke, codegen_, kQuickExp);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001799}
1800
1801void IntrinsicLocationsBuilderARM64::VisitMathExpm1(HInvoke* invoke) {
1802 CreateFPToFPCallLocations(arena_, invoke);
1803}
1804
1805void IntrinsicCodeGeneratorARM64::VisitMathExpm1(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001806 GenFPToFPCall(invoke, codegen_, kQuickExpm1);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001807}
1808
1809void IntrinsicLocationsBuilderARM64::VisitMathLog(HInvoke* invoke) {
1810 CreateFPToFPCallLocations(arena_, invoke);
1811}
1812
1813void IntrinsicCodeGeneratorARM64::VisitMathLog(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001814 GenFPToFPCall(invoke, codegen_, kQuickLog);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001815}
1816
1817void IntrinsicLocationsBuilderARM64::VisitMathLog10(HInvoke* invoke) {
1818 CreateFPToFPCallLocations(arena_, invoke);
1819}
1820
1821void IntrinsicCodeGeneratorARM64::VisitMathLog10(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001822 GenFPToFPCall(invoke, codegen_, kQuickLog10);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001823}
1824
1825void IntrinsicLocationsBuilderARM64::VisitMathSinh(HInvoke* invoke) {
1826 CreateFPToFPCallLocations(arena_, invoke);
1827}
1828
1829void IntrinsicCodeGeneratorARM64::VisitMathSinh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001830 GenFPToFPCall(invoke, codegen_, kQuickSinh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001831}
1832
1833void IntrinsicLocationsBuilderARM64::VisitMathTan(HInvoke* invoke) {
1834 CreateFPToFPCallLocations(arena_, invoke);
1835}
1836
1837void IntrinsicCodeGeneratorARM64::VisitMathTan(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001838 GenFPToFPCall(invoke, codegen_, kQuickTan);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001839}
1840
1841void IntrinsicLocationsBuilderARM64::VisitMathTanh(HInvoke* invoke) {
1842 CreateFPToFPCallLocations(arena_, invoke);
1843}
1844
1845void IntrinsicCodeGeneratorARM64::VisitMathTanh(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001846 GenFPToFPCall(invoke, codegen_, kQuickTanh);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001847}
1848
1849void IntrinsicLocationsBuilderARM64::VisitMathAtan2(HInvoke* invoke) {
1850 CreateFPFPToFPCallLocations(arena_, invoke);
1851}
1852
1853void IntrinsicCodeGeneratorARM64::VisitMathAtan2(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001854 GenFPToFPCall(invoke, codegen_, kQuickAtan2);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001855}
1856
1857void IntrinsicLocationsBuilderARM64::VisitMathHypot(HInvoke* invoke) {
1858 CreateFPFPToFPCallLocations(arena_, invoke);
1859}
1860
1861void IntrinsicCodeGeneratorARM64::VisitMathHypot(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001862 GenFPToFPCall(invoke, codegen_, kQuickHypot);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001863}
1864
1865void IntrinsicLocationsBuilderARM64::VisitMathNextAfter(HInvoke* invoke) {
1866 CreateFPFPToFPCallLocations(arena_, invoke);
1867}
1868
1869void IntrinsicCodeGeneratorARM64::VisitMathNextAfter(HInvoke* invoke) {
Serban Constantinescu22f81d32016-02-18 16:06:31 +00001870 GenFPToFPCall(invoke, codegen_, kQuickNextAfter);
Anton Kirilov02fc24e2016-01-20 16:48:19 +00001871}
1872
Tim Zhang25abd6c2016-01-19 23:39:24 +08001873void IntrinsicLocationsBuilderARM64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
1874 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1875 LocationSummary::kNoCall,
1876 kIntrinsified);
1877 locations->SetInAt(0, Location::RequiresRegister());
1878 locations->SetInAt(1, Location::RequiresRegister());
1879 locations->SetInAt(2, Location::RequiresRegister());
1880 locations->SetInAt(3, Location::RequiresRegister());
1881 locations->SetInAt(4, Location::RequiresRegister());
1882
1883 locations->AddTemp(Location::RequiresRegister());
1884 locations->AddTemp(Location::RequiresRegister());
Scott Wakelingdf109d92016-04-22 11:35:56 +01001885 locations->AddTemp(Location::RequiresRegister());
jessicahandojo05765752016-09-09 19:01:32 -07001886 // Need temporary register for String compression feature.
1887 if (mirror::kUseStringCompression) {
1888 locations->AddTemp(Location::RequiresRegister());
1889 }
Tim Zhang25abd6c2016-01-19 23:39:24 +08001890}
1891
1892void IntrinsicCodeGeneratorARM64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01001893 MacroAssembler* masm = GetVIXLAssembler();
Tim Zhang25abd6c2016-01-19 23:39:24 +08001894 LocationSummary* locations = invoke->GetLocations();
1895
1896 // Check assumption that sizeof(Char) is 2 (used in scaling below).
1897 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1898 DCHECK_EQ(char_size, 2u);
1899
1900 // Location of data in char array buffer.
1901 const uint32_t data_offset = mirror::Array::DataOffset(char_size).Uint32Value();
1902
1903 // Location of char array data in string.
1904 const uint32_t value_offset = mirror::String::ValueOffset().Uint32Value();
1905
1906 // void getCharsNoCheck(int srcBegin, int srcEnd, char[] dst, int dstBegin);
1907 // Since getChars() calls getCharsNoCheck() - we use registers rather than constants.
1908 Register srcObj = XRegisterFrom(locations->InAt(0));
1909 Register srcBegin = XRegisterFrom(locations->InAt(1));
1910 Register srcEnd = XRegisterFrom(locations->InAt(2));
1911 Register dstObj = XRegisterFrom(locations->InAt(3));
1912 Register dstBegin = XRegisterFrom(locations->InAt(4));
1913
1914 Register src_ptr = XRegisterFrom(locations->GetTemp(0));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001915 Register num_chr = XRegisterFrom(locations->GetTemp(1));
1916 Register tmp1 = XRegisterFrom(locations->GetTemp(2));
jessicahandojo05765752016-09-09 19:01:32 -07001917 Register tmp3;
1918 if (mirror::kUseStringCompression) {
1919 tmp3 = WRegisterFrom(locations->GetTemp(3));
1920 }
Tim Zhang25abd6c2016-01-19 23:39:24 +08001921
1922 UseScratchRegisterScope temps(masm);
1923 Register dst_ptr = temps.AcquireX();
Scott Wakelingdf109d92016-04-22 11:35:56 +01001924 Register tmp2 = temps.AcquireX();
Tim Zhang25abd6c2016-01-19 23:39:24 +08001925
jessicahandojo05765752016-09-09 19:01:32 -07001926 vixl::aarch64::Label done;
1927 vixl::aarch64::Label compressed_string_loop;
1928 __ Sub(num_chr, srcEnd, srcBegin);
1929 // Early out for valid zero-length retrievals.
1930 __ Cbz(num_chr, &done);
Tim Zhang25abd6c2016-01-19 23:39:24 +08001931
Scott Wakelingdf109d92016-04-22 11:35:56 +01001932 // dst address start to copy to.
Tim Zhang25abd6c2016-01-19 23:39:24 +08001933 __ Add(dst_ptr, dstObj, Operand(data_offset));
1934 __ Add(dst_ptr, dst_ptr, Operand(dstBegin, LSL, 1));
1935
jessicahandojo05765752016-09-09 19:01:32 -07001936 // src address to copy from.
1937 __ Add(src_ptr, srcObj, Operand(value_offset));
1938 vixl::aarch64::Label compressed_string_preloop;
1939 if (mirror::kUseStringCompression) {
1940 // Location of count in string.
1941 const uint32_t count_offset = mirror::String::CountOffset().Uint32Value();
1942 // String's length.
1943 __ Ldr(tmp3, MemOperand(srcObj, count_offset));
1944 __ Tbnz(tmp3, kWRegSize - 1, &compressed_string_preloop);
1945 }
1946 __ Add(src_ptr, src_ptr, Operand(srcBegin, LSL, 1));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001947
Tim Zhang25abd6c2016-01-19 23:39:24 +08001948 // Do the copy.
Scott Wakeling97c72b72016-06-24 16:19:36 +01001949 vixl::aarch64::Label loop;
Scott Wakeling97c72b72016-06-24 16:19:36 +01001950 vixl::aarch64::Label remainder;
Scott Wakelingdf109d92016-04-22 11:35:56 +01001951
Scott Wakelingdf109d92016-04-22 11:35:56 +01001952 // Save repairing the value of num_chr on the < 8 character path.
1953 __ Subs(tmp1, num_chr, 8);
1954 __ B(lt, &remainder);
1955
1956 // Keep the result of the earlier subs, we are going to fetch at least 8 characters.
1957 __ Mov(num_chr, tmp1);
1958
1959 // Main loop used for longer fetches loads and stores 8x16-bit characters at a time.
1960 // (Unaligned addresses are acceptable here and not worth inlining extra code to rectify.)
Tim Zhang25abd6c2016-01-19 23:39:24 +08001961 __ Bind(&loop);
Scott Wakeling97c72b72016-06-24 16:19:36 +01001962 __ Ldp(tmp1, tmp2, MemOperand(src_ptr, char_size * 8, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001963 __ Subs(num_chr, num_chr, 8);
Scott Wakeling97c72b72016-06-24 16:19:36 +01001964 __ Stp(tmp1, tmp2, MemOperand(dst_ptr, char_size * 8, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001965 __ B(ge, &loop);
1966
1967 __ Adds(num_chr, num_chr, 8);
1968 __ B(eq, &done);
1969
1970 // Main loop for < 8 character case and remainder handling. Loads and stores one
1971 // 16-bit Java character at a time.
1972 __ Bind(&remainder);
Scott Wakeling97c72b72016-06-24 16:19:36 +01001973 __ Ldrh(tmp1, MemOperand(src_ptr, char_size, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001974 __ Subs(num_chr, num_chr, 1);
Scott Wakeling97c72b72016-06-24 16:19:36 +01001975 __ Strh(tmp1, MemOperand(dst_ptr, char_size, PostIndex));
Scott Wakelingdf109d92016-04-22 11:35:56 +01001976 __ B(gt, &remainder);
jessicahandojo05765752016-09-09 19:01:32 -07001977 __ B(&done);
1978
1979 if (mirror::kUseStringCompression) {
1980 const size_t c_char_size = Primitive::ComponentSize(Primitive::kPrimByte);
1981 DCHECK_EQ(c_char_size, 1u);
1982 __ Bind(&compressed_string_preloop);
1983 __ Add(src_ptr, src_ptr, Operand(srcBegin));
1984 // Copy loop for compressed src, copying 1 character (8-bit) to (16-bit) at a time.
1985 __ Bind(&compressed_string_loop);
1986 __ Ldrb(tmp1, MemOperand(src_ptr, c_char_size, PostIndex));
1987 __ Strh(tmp1, MemOperand(dst_ptr, char_size, PostIndex));
1988 __ Subs(num_chr, num_chr, Operand(1));
1989 __ B(gt, &compressed_string_loop);
1990 }
Scott Wakelingdf109d92016-04-22 11:35:56 +01001991
Tim Zhang25abd6c2016-01-19 23:39:24 +08001992 __ Bind(&done);
1993}
1994
Scott Wakelingd3d0da52016-02-29 15:17:20 +00001995// Mirrors ARRAYCOPY_SHORT_CHAR_ARRAY_THRESHOLD in libcore, so we can choose to use the native
1996// implementation there for longer copy lengths.
donghui.baic2ec9ad2016-03-10 14:02:55 +08001997static constexpr int32_t kSystemArrayCopyCharThreshold = 32;
Scott Wakelingd3d0da52016-02-29 15:17:20 +00001998
1999static void SetSystemArrayCopyLocationRequires(LocationSummary* locations,
2000 uint32_t at,
2001 HInstruction* input) {
2002 HIntConstant* const_input = input->AsIntConstant();
Scott Wakeling97c72b72016-06-24 16:19:36 +01002003 if (const_input != nullptr && !vixl::aarch64::Assembler::IsImmAddSub(const_input->GetValue())) {
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002004 locations->SetInAt(at, Location::RequiresRegister());
2005 } else {
2006 locations->SetInAt(at, Location::RegisterOrConstant(input));
2007 }
2008}
2009
2010void IntrinsicLocationsBuilderARM64::VisitSystemArrayCopyChar(HInvoke* invoke) {
2011 // Check to see if we have known failures that will cause us to have to bail out
2012 // to the runtime, and just generate the runtime call directly.
2013 HIntConstant* src_pos = invoke->InputAt(1)->AsIntConstant();
2014 HIntConstant* dst_pos = invoke->InputAt(3)->AsIntConstant();
2015
2016 // The positions must be non-negative.
2017 if ((src_pos != nullptr && src_pos->GetValue() < 0) ||
2018 (dst_pos != nullptr && dst_pos->GetValue() < 0)) {
2019 // We will have to fail anyways.
2020 return;
2021 }
2022
2023 // The length must be >= 0 and not so long that we would (currently) prefer libcore's
2024 // native implementation.
2025 HIntConstant* length = invoke->InputAt(4)->AsIntConstant();
2026 if (length != nullptr) {
2027 int32_t len = length->GetValue();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002028 if (len < 0 || len > kSystemArrayCopyCharThreshold) {
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002029 // Just call as normal.
2030 return;
2031 }
2032 }
2033
2034 ArenaAllocator* allocator = invoke->GetBlock()->GetGraph()->GetArena();
2035 LocationSummary* locations = new (allocator) LocationSummary(invoke,
2036 LocationSummary::kCallOnSlowPath,
2037 kIntrinsified);
2038 // arraycopy(char[] src, int src_pos, char[] dst, int dst_pos, int length).
2039 locations->SetInAt(0, Location::RequiresRegister());
2040 SetSystemArrayCopyLocationRequires(locations, 1, invoke->InputAt(1));
2041 locations->SetInAt(2, Location::RequiresRegister());
2042 SetSystemArrayCopyLocationRequires(locations, 3, invoke->InputAt(3));
2043 SetSystemArrayCopyLocationRequires(locations, 4, invoke->InputAt(4));
2044
2045 locations->AddTemp(Location::RequiresRegister());
2046 locations->AddTemp(Location::RequiresRegister());
2047 locations->AddTemp(Location::RequiresRegister());
2048}
2049
Scott Wakeling97c72b72016-06-24 16:19:36 +01002050static void CheckSystemArrayCopyPosition(MacroAssembler* masm,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002051 const Location& pos,
2052 const Register& input,
2053 const Location& length,
2054 SlowPathCodeARM64* slow_path,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002055 const Register& temp,
2056 bool length_is_input_length = false) {
2057 const int32_t length_offset = mirror::Array::LengthOffset().Int32Value();
2058 if (pos.IsConstant()) {
2059 int32_t pos_const = pos.GetConstant()->AsIntConstant()->GetValue();
2060 if (pos_const == 0) {
2061 if (!length_is_input_length) {
2062 // Check that length(input) >= length.
2063 __ Ldr(temp, MemOperand(input, length_offset));
2064 __ Cmp(temp, OperandFrom(length, Primitive::kPrimInt));
2065 __ B(slow_path->GetEntryLabel(), lt);
2066 }
2067 } else {
2068 // Check that length(input) >= pos.
Nicolas Geoffrayfea1abd2016-07-06 12:09:12 +01002069 __ Ldr(temp, MemOperand(input, length_offset));
2070 __ Subs(temp, temp, pos_const);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002071 __ B(slow_path->GetEntryLabel(), lt);
2072
2073 // Check that (length(input) - pos) >= length.
2074 __ Cmp(temp, OperandFrom(length, Primitive::kPrimInt));
2075 __ B(slow_path->GetEntryLabel(), lt);
2076 }
2077 } else if (length_is_input_length) {
2078 // The only way the copy can succeed is if pos is zero.
2079 __ Cbnz(WRegisterFrom(pos), slow_path->GetEntryLabel());
2080 } else {
2081 // Check that pos >= 0.
2082 Register pos_reg = WRegisterFrom(pos);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002083 __ Tbnz(pos_reg, pos_reg.GetSizeInBits() - 1, slow_path->GetEntryLabel());
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002084
2085 // Check that pos <= length(input) && (length(input) - pos) >= length.
2086 __ Ldr(temp, MemOperand(input, length_offset));
2087 __ Subs(temp, temp, pos_reg);
2088 // Ccmp if length(input) >= pos, else definitely bail to slow path (N!=V == lt).
2089 __ Ccmp(temp, OperandFrom(length, Primitive::kPrimInt), NFlag, ge);
2090 __ B(slow_path->GetEntryLabel(), lt);
2091 }
2092}
2093
2094// Compute base source address, base destination address, and end source address
2095// for System.arraycopy* intrinsics.
Scott Wakeling97c72b72016-06-24 16:19:36 +01002096static void GenSystemArrayCopyAddresses(MacroAssembler* masm,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002097 Primitive::Type type,
2098 const Register& src,
2099 const Location& src_pos,
2100 const Register& dst,
2101 const Location& dst_pos,
2102 const Location& copy_length,
2103 const Register& src_base,
2104 const Register& dst_base,
2105 const Register& src_end) {
2106 DCHECK(type == Primitive::kPrimNot || type == Primitive::kPrimChar)
Roland Levillainebea3d22016-04-12 15:42:57 +01002107 << "Unexpected element type: " << type;
2108 const int32_t element_size = Primitive::ComponentSize(type);
2109 const int32_t element_size_shift = Primitive::ComponentSizeShift(type);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002110
Roland Levillainebea3d22016-04-12 15:42:57 +01002111 uint32_t data_offset = mirror::Array::DataOffset(element_size).Uint32Value();
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002112 if (src_pos.IsConstant()) {
2113 int32_t constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002114 __ Add(src_base, src, element_size * constant + data_offset);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002115 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002116 __ Add(src_base, src, data_offset);
2117 __ Add(src_base, src_base, Operand(XRegisterFrom(src_pos), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002118 }
2119
2120 if (dst_pos.IsConstant()) {
2121 int32_t constant = dst_pos.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002122 __ Add(dst_base, dst, element_size * constant + data_offset);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002123 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002124 __ Add(dst_base, dst, data_offset);
2125 __ Add(dst_base, dst_base, Operand(XRegisterFrom(dst_pos), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002126 }
2127
2128 if (copy_length.IsConstant()) {
2129 int32_t constant = copy_length.GetConstant()->AsIntConstant()->GetValue();
Roland Levillainebea3d22016-04-12 15:42:57 +01002130 __ Add(src_end, src_base, element_size * constant);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002131 } else {
Roland Levillainebea3d22016-04-12 15:42:57 +01002132 __ Add(src_end, src_base, Operand(XRegisterFrom(copy_length), LSL, element_size_shift));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002133 }
2134}
2135
2136void IntrinsicCodeGeneratorARM64::VisitSystemArrayCopyChar(HInvoke* invoke) {
Scott Wakeling97c72b72016-06-24 16:19:36 +01002137 MacroAssembler* masm = GetVIXLAssembler();
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002138 LocationSummary* locations = invoke->GetLocations();
2139 Register src = XRegisterFrom(locations->InAt(0));
2140 Location src_pos = locations->InAt(1);
2141 Register dst = XRegisterFrom(locations->InAt(2));
2142 Location dst_pos = locations->InAt(3);
2143 Location length = locations->InAt(4);
2144
2145 SlowPathCodeARM64* slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
2146 codegen_->AddSlowPath(slow_path);
2147
2148 // If source and destination are the same, take the slow path. Overlapping copy regions must be
2149 // copied in reverse and we can't know in all cases if it's needed.
2150 __ Cmp(src, dst);
2151 __ B(slow_path->GetEntryLabel(), eq);
2152
2153 // Bail out if the source is null.
2154 __ Cbz(src, slow_path->GetEntryLabel());
2155
2156 // Bail out if the destination is null.
2157 __ Cbz(dst, slow_path->GetEntryLabel());
2158
2159 if (!length.IsConstant()) {
2160 // If the length is negative, bail out.
2161 __ Tbnz(WRegisterFrom(length), kWRegSize - 1, slow_path->GetEntryLabel());
2162 // If the length > 32 then (currently) prefer libcore's native implementation.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002163 __ Cmp(WRegisterFrom(length), kSystemArrayCopyCharThreshold);
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002164 __ B(slow_path->GetEntryLabel(), gt);
2165 } else {
2166 // We have already checked in the LocationsBuilder for the constant case.
2167 DCHECK_GE(length.GetConstant()->AsIntConstant()->GetValue(), 0);
2168 DCHECK_LE(length.GetConstant()->AsIntConstant()->GetValue(), 32);
2169 }
2170
2171 Register src_curr_addr = WRegisterFrom(locations->GetTemp(0));
2172 Register dst_curr_addr = WRegisterFrom(locations->GetTemp(1));
2173 Register src_stop_addr = WRegisterFrom(locations->GetTemp(2));
2174
2175 CheckSystemArrayCopyPosition(masm,
2176 src_pos,
2177 src,
2178 length,
2179 slow_path,
2180 src_curr_addr,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002181 false);
2182
2183 CheckSystemArrayCopyPosition(masm,
2184 dst_pos,
2185 dst,
2186 length,
2187 slow_path,
2188 src_curr_addr,
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002189 false);
2190
2191 src_curr_addr = src_curr_addr.X();
2192 dst_curr_addr = dst_curr_addr.X();
2193 src_stop_addr = src_stop_addr.X();
2194
2195 GenSystemArrayCopyAddresses(masm,
2196 Primitive::kPrimChar,
2197 src,
2198 src_pos,
2199 dst,
2200 dst_pos,
2201 length,
2202 src_curr_addr,
2203 dst_curr_addr,
2204 src_stop_addr);
2205
2206 // Iterate over the arrays and do a raw copy of the chars.
2207 const int32_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
2208 UseScratchRegisterScope temps(masm);
2209 Register tmp = temps.AcquireW();
Scott Wakeling97c72b72016-06-24 16:19:36 +01002210 vixl::aarch64::Label loop, done;
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002211 __ Bind(&loop);
2212 __ Cmp(src_curr_addr, src_stop_addr);
2213 __ B(&done, eq);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002214 __ Ldrh(tmp, MemOperand(src_curr_addr, char_size, PostIndex));
2215 __ Strh(tmp, MemOperand(dst_curr_addr, char_size, PostIndex));
Scott Wakelingd3d0da52016-02-29 15:17:20 +00002216 __ B(&loop);
2217 __ Bind(&done);
2218
2219 __ Bind(slow_path->GetExitLabel());
2220}
2221
donghui.baic2ec9ad2016-03-10 14:02:55 +08002222// We can choose to use the native implementation there for longer copy lengths.
2223static constexpr int32_t kSystemArrayCopyThreshold = 128;
2224
2225// CodeGenerator::CreateSystemArrayCopyLocationSummary use three temporary registers.
2226// We want to use two temporary registers in order to reduce the register pressure in arm64.
2227// So we don't use the CodeGenerator::CreateSystemArrayCopyLocationSummary.
2228void IntrinsicLocationsBuilderARM64::VisitSystemArrayCopy(HInvoke* invoke) {
Roland Levillain0b671c02016-08-19 12:02:34 +01002229 // The only read barrier implementation supporting the
2230 // SystemArrayCopy intrinsic is the Baker-style read barriers.
2231 if (kEmitCompilerReadBarrier && !kUseBakerReadBarrier) {
Roland Levillain3d312422016-06-23 13:53:42 +01002232 return;
2233 }
2234
donghui.baic2ec9ad2016-03-10 14:02:55 +08002235 // Check to see if we have known failures that will cause us to have to bail out
2236 // to the runtime, and just generate the runtime call directly.
2237 HIntConstant* src_pos = invoke->InputAt(1)->AsIntConstant();
2238 HIntConstant* dest_pos = invoke->InputAt(3)->AsIntConstant();
2239
2240 // The positions must be non-negative.
2241 if ((src_pos != nullptr && src_pos->GetValue() < 0) ||
2242 (dest_pos != nullptr && dest_pos->GetValue() < 0)) {
2243 // We will have to fail anyways.
2244 return;
2245 }
2246
2247 // The length must be >= 0.
2248 HIntConstant* length = invoke->InputAt(4)->AsIntConstant();
2249 if (length != nullptr) {
2250 int32_t len = length->GetValue();
2251 if (len < 0 || len >= kSystemArrayCopyThreshold) {
2252 // Just call as normal.
2253 return;
2254 }
2255 }
2256
2257 SystemArrayCopyOptimizations optimizations(invoke);
2258
2259 if (optimizations.GetDestinationIsSource()) {
2260 if (src_pos != nullptr && dest_pos != nullptr && src_pos->GetValue() < dest_pos->GetValue()) {
2261 // We only support backward copying if source and destination are the same.
2262 return;
2263 }
2264 }
2265
2266 if (optimizations.GetDestinationIsPrimitiveArray() || optimizations.GetSourceIsPrimitiveArray()) {
2267 // We currently don't intrinsify primitive copying.
2268 return;
2269 }
2270
2271 ArenaAllocator* allocator = invoke->GetBlock()->GetGraph()->GetArena();
2272 LocationSummary* locations = new (allocator) LocationSummary(invoke,
2273 LocationSummary::kCallOnSlowPath,
2274 kIntrinsified);
2275 // arraycopy(Object src, int src_pos, Object dest, int dest_pos, int length).
2276 locations->SetInAt(0, Location::RequiresRegister());
2277 SetSystemArrayCopyLocationRequires(locations, 1, invoke->InputAt(1));
2278 locations->SetInAt(2, Location::RequiresRegister());
2279 SetSystemArrayCopyLocationRequires(locations, 3, invoke->InputAt(3));
2280 SetSystemArrayCopyLocationRequires(locations, 4, invoke->InputAt(4));
2281
2282 locations->AddTemp(Location::RequiresRegister());
2283 locations->AddTemp(Location::RequiresRegister());
Roland Levillain0b671c02016-08-19 12:02:34 +01002284 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2285 // Temporary register IP0, obtained from the VIXL scratch register
2286 // pool, cannot be used in ReadBarrierSystemArrayCopySlowPathARM64
2287 // (because that register is clobbered by ReadBarrierMarkRegX
2288 // entry points). Get an extra temporary register from the
2289 // register allocator.
2290 locations->AddTemp(Location::RequiresRegister());
2291 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002292}
2293
2294void IntrinsicCodeGeneratorARM64::VisitSystemArrayCopy(HInvoke* invoke) {
Roland Levillain0b671c02016-08-19 12:02:34 +01002295 // The only read barrier implementation supporting the
2296 // SystemArrayCopy intrinsic is the Baker-style read barriers.
2297 DCHECK(!kEmitCompilerReadBarrier || kUseBakerReadBarrier);
Roland Levillain3d312422016-06-23 13:53:42 +01002298
Scott Wakeling97c72b72016-06-24 16:19:36 +01002299 MacroAssembler* masm = GetVIXLAssembler();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002300 LocationSummary* locations = invoke->GetLocations();
2301
2302 uint32_t class_offset = mirror::Object::ClassOffset().Int32Value();
2303 uint32_t super_offset = mirror::Class::SuperClassOffset().Int32Value();
2304 uint32_t component_offset = mirror::Class::ComponentTypeOffset().Int32Value();
2305 uint32_t primitive_offset = mirror::Class::PrimitiveTypeOffset().Int32Value();
Roland Levillain0b671c02016-08-19 12:02:34 +01002306 uint32_t monitor_offset = mirror::Object::MonitorOffset().Int32Value();
donghui.baic2ec9ad2016-03-10 14:02:55 +08002307
2308 Register src = XRegisterFrom(locations->InAt(0));
2309 Location src_pos = locations->InAt(1);
2310 Register dest = XRegisterFrom(locations->InAt(2));
2311 Location dest_pos = locations->InAt(3);
2312 Location length = locations->InAt(4);
2313 Register temp1 = WRegisterFrom(locations->GetTemp(0));
Roland Levillain0b671c02016-08-19 12:02:34 +01002314 Location temp1_loc = LocationFrom(temp1);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002315 Register temp2 = WRegisterFrom(locations->GetTemp(1));
Roland Levillain0b671c02016-08-19 12:02:34 +01002316 Location temp2_loc = LocationFrom(temp2);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002317
Roland Levillain0b671c02016-08-19 12:02:34 +01002318 SlowPathCodeARM64* intrinsic_slow_path = new (GetAllocator()) IntrinsicSlowPathARM64(invoke);
2319 codegen_->AddSlowPath(intrinsic_slow_path);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002320
Scott Wakeling97c72b72016-06-24 16:19:36 +01002321 vixl::aarch64::Label conditions_on_positions_validated;
donghui.baic2ec9ad2016-03-10 14:02:55 +08002322 SystemArrayCopyOptimizations optimizations(invoke);
2323
donghui.baic2ec9ad2016-03-10 14:02:55 +08002324 // If source and destination are the same, we go to slow path if we need to do
2325 // forward copying.
2326 if (src_pos.IsConstant()) {
2327 int32_t src_pos_constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
2328 if (dest_pos.IsConstant()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002329 int32_t dest_pos_constant = dest_pos.GetConstant()->AsIntConstant()->GetValue();
2330 if (optimizations.GetDestinationIsSource()) {
2331 // Checked when building locations.
2332 DCHECK_GE(src_pos_constant, dest_pos_constant);
2333 } else if (src_pos_constant < dest_pos_constant) {
2334 __ Cmp(src, dest);
Roland Levillain0b671c02016-08-19 12:02:34 +01002335 __ B(intrinsic_slow_path->GetEntryLabel(), eq);
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002336 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002337 // Checked when building locations.
2338 DCHECK(!optimizations.GetDestinationIsSource()
2339 || (src_pos_constant >= dest_pos.GetConstant()->AsIntConstant()->GetValue()));
2340 } else {
2341 if (!optimizations.GetDestinationIsSource()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002342 __ Cmp(src, dest);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002343 __ B(&conditions_on_positions_validated, ne);
2344 }
2345 __ Cmp(WRegisterFrom(dest_pos), src_pos_constant);
Roland Levillain0b671c02016-08-19 12:02:34 +01002346 __ B(intrinsic_slow_path->GetEntryLabel(), gt);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002347 }
2348 } else {
2349 if (!optimizations.GetDestinationIsSource()) {
Nicolas Geoffray9f65db82016-07-07 12:07:42 +01002350 __ Cmp(src, dest);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002351 __ B(&conditions_on_positions_validated, ne);
2352 }
2353 __ Cmp(RegisterFrom(src_pos, invoke->InputAt(1)->GetType()),
2354 OperandFrom(dest_pos, invoke->InputAt(3)->GetType()));
Roland Levillain0b671c02016-08-19 12:02:34 +01002355 __ B(intrinsic_slow_path->GetEntryLabel(), lt);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002356 }
2357
2358 __ Bind(&conditions_on_positions_validated);
2359
2360 if (!optimizations.GetSourceIsNotNull()) {
2361 // Bail out if the source is null.
Roland Levillain0b671c02016-08-19 12:02:34 +01002362 __ Cbz(src, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002363 }
2364
2365 if (!optimizations.GetDestinationIsNotNull() && !optimizations.GetDestinationIsSource()) {
2366 // Bail out if the destination is null.
Roland Levillain0b671c02016-08-19 12:02:34 +01002367 __ Cbz(dest, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002368 }
2369
2370 // We have already checked in the LocationsBuilder for the constant case.
2371 if (!length.IsConstant() &&
2372 !optimizations.GetCountIsSourceLength() &&
2373 !optimizations.GetCountIsDestinationLength()) {
2374 // If the length is negative, bail out.
Roland Levillain0b671c02016-08-19 12:02:34 +01002375 __ Tbnz(WRegisterFrom(length), kWRegSize - 1, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002376 // If the length >= 128 then (currently) prefer native implementation.
2377 __ Cmp(WRegisterFrom(length), kSystemArrayCopyThreshold);
Roland Levillain0b671c02016-08-19 12:02:34 +01002378 __ B(intrinsic_slow_path->GetEntryLabel(), ge);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002379 }
2380 // Validity checks: source.
2381 CheckSystemArrayCopyPosition(masm,
2382 src_pos,
2383 src,
2384 length,
Roland Levillain0b671c02016-08-19 12:02:34 +01002385 intrinsic_slow_path,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002386 temp1,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002387 optimizations.GetCountIsSourceLength());
2388
2389 // Validity checks: dest.
2390 CheckSystemArrayCopyPosition(masm,
2391 dest_pos,
2392 dest,
2393 length,
Roland Levillain0b671c02016-08-19 12:02:34 +01002394 intrinsic_slow_path,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002395 temp1,
donghui.baic2ec9ad2016-03-10 14:02:55 +08002396 optimizations.GetCountIsDestinationLength());
2397 {
2398 // We use a block to end the scratch scope before the write barrier, thus
2399 // freeing the temporary registers so they can be used in `MarkGCCard`.
2400 UseScratchRegisterScope temps(masm);
Roland Levillain0b671c02016-08-19 12:02:34 +01002401 // Note: Because it is acquired from VIXL's scratch register pool,
2402 // `temp3` might be IP0, and thus cannot be used as `ref` argument
2403 // of CodeGeneratorARM64::GenerateFieldLoadWithBakerReadBarrier
2404 // calls below (see ReadBarrierMarkSlowPathARM64 for more details).
donghui.baic2ec9ad2016-03-10 14:02:55 +08002405 Register temp3 = temps.AcquireW();
Roland Levillain0b671c02016-08-19 12:02:34 +01002406
donghui.baic2ec9ad2016-03-10 14:02:55 +08002407 if (!optimizations.GetDoesNotNeedTypeCheck()) {
2408 // Check whether all elements of the source array are assignable to the component
2409 // type of the destination array. We do two checks: the classes are the same,
2410 // or the destination is Object[]. If none of these checks succeed, we go to the
2411 // slow path.
donghui.baic2ec9ad2016-03-10 14:02:55 +08002412
Roland Levillain0b671c02016-08-19 12:02:34 +01002413 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2414 if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2415 // /* HeapReference<Class> */ temp1 = src->klass_
2416 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2417 temp1_loc,
2418 src.W(),
2419 class_offset,
2420 temp2,
2421 /* needs_null_check */ false,
2422 /* use_load_acquire */ false);
2423 // Bail out if the source is not a non primitive array.
2424 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2425 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2426 temp1_loc,
2427 temp1,
2428 component_offset,
2429 temp2,
2430 /* needs_null_check */ false,
2431 /* use_load_acquire */ false);
2432 __ Cbz(temp1, intrinsic_slow_path->GetEntryLabel());
2433 // If heap poisoning is enabled, `temp1` has been unpoisoned
2434 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2435 // /* uint16_t */ temp1 = static_cast<uint16>(temp1->primitive_type_);
2436 __ Ldrh(temp1, HeapOperand(temp1, primitive_offset));
2437 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2438 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002439 }
Roland Levillain0b671c02016-08-19 12:02:34 +01002440
2441 // /* HeapReference<Class> */ temp1 = dest->klass_
2442 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2443 temp1_loc,
2444 dest.W(),
2445 class_offset,
2446 temp2,
2447 /* needs_null_check */ false,
2448 /* use_load_acquire */ false);
2449
2450 if (!optimizations.GetDestinationIsNonPrimitiveArray()) {
2451 // Bail out if the destination is not a non primitive array.
2452 //
2453 // Register `temp1` is not trashed by the read barrier emitted
2454 // by GenerateFieldLoadWithBakerReadBarrier below, as that
2455 // method produces a call to a ReadBarrierMarkRegX entry point,
2456 // which saves all potentially live registers, including
2457 // temporaries such a `temp1`.
2458 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2459 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2460 temp2_loc,
2461 temp1,
2462 component_offset,
2463 temp3,
2464 /* needs_null_check */ false,
2465 /* use_load_acquire */ false);
2466 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2467 // If heap poisoning is enabled, `temp2` has been unpoisoned
2468 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2469 // /* uint16_t */ temp2 = static_cast<uint16>(temp2->primitive_type_);
2470 __ Ldrh(temp2, HeapOperand(temp2, primitive_offset));
2471 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2472 __ Cbnz(temp2, intrinsic_slow_path->GetEntryLabel());
2473 }
2474
2475 // For the same reason given earlier, `temp1` is not trashed by the
2476 // read barrier emitted by GenerateFieldLoadWithBakerReadBarrier below.
2477 // /* HeapReference<Class> */ temp2 = src->klass_
2478 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2479 temp2_loc,
2480 src.W(),
2481 class_offset,
2482 temp3,
2483 /* needs_null_check */ false,
2484 /* use_load_acquire */ false);
2485 // Note: if heap poisoning is on, we are comparing two unpoisoned references here.
2486 __ Cmp(temp1, temp2);
2487
2488 if (optimizations.GetDestinationIsTypedObjectArray()) {
2489 vixl::aarch64::Label do_copy;
2490 __ B(&do_copy, eq);
2491 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2492 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2493 temp1_loc,
2494 temp1,
2495 component_offset,
2496 temp2,
2497 /* needs_null_check */ false,
2498 /* use_load_acquire */ false);
2499 // /* HeapReference<Class> */ temp1 = temp1->super_class_
2500 // We do not need to emit a read barrier for the following
2501 // heap reference load, as `temp1` is only used in a
2502 // comparison with null below, and this reference is not
2503 // kept afterwards.
2504 __ Ldr(temp1, HeapOperand(temp1, super_offset));
2505 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
2506 __ Bind(&do_copy);
2507 } else {
2508 __ B(intrinsic_slow_path->GetEntryLabel(), ne);
2509 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002510 } else {
Roland Levillain0b671c02016-08-19 12:02:34 +01002511 // Non read barrier code.
2512
2513 // /* HeapReference<Class> */ temp1 = dest->klass_
2514 __ Ldr(temp1, MemOperand(dest, class_offset));
2515 // /* HeapReference<Class> */ temp2 = src->klass_
2516 __ Ldr(temp2, MemOperand(src, class_offset));
2517 bool did_unpoison = false;
2518 if (!optimizations.GetDestinationIsNonPrimitiveArray() ||
2519 !optimizations.GetSourceIsNonPrimitiveArray()) {
2520 // One or two of the references need to be unpoisoned. Unpoison them
2521 // both to make the identity check valid.
2522 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2523 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp2);
2524 did_unpoison = true;
2525 }
2526
2527 if (!optimizations.GetDestinationIsNonPrimitiveArray()) {
2528 // Bail out if the destination is not a non primitive array.
2529 // /* HeapReference<Class> */ temp3 = temp1->component_type_
2530 __ Ldr(temp3, HeapOperand(temp1, component_offset));
2531 __ Cbz(temp3, intrinsic_slow_path->GetEntryLabel());
2532 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp3);
2533 // /* uint16_t */ temp3 = static_cast<uint16>(temp3->primitive_type_);
2534 __ Ldrh(temp3, HeapOperand(temp3, primitive_offset));
2535 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2536 __ Cbnz(temp3, intrinsic_slow_path->GetEntryLabel());
2537 }
2538
2539 if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2540 // Bail out if the source is not a non primitive array.
2541 // /* HeapReference<Class> */ temp3 = temp2->component_type_
2542 __ Ldr(temp3, HeapOperand(temp2, component_offset));
2543 __ Cbz(temp3, intrinsic_slow_path->GetEntryLabel());
2544 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp3);
2545 // /* uint16_t */ temp3 = static_cast<uint16>(temp3->primitive_type_);
2546 __ Ldrh(temp3, HeapOperand(temp3, primitive_offset));
2547 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
2548 __ Cbnz(temp3, intrinsic_slow_path->GetEntryLabel());
2549 }
2550
2551 __ Cmp(temp1, temp2);
2552
2553 if (optimizations.GetDestinationIsTypedObjectArray()) {
2554 vixl::aarch64::Label do_copy;
2555 __ B(&do_copy, eq);
2556 if (!did_unpoison) {
2557 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2558 }
2559 // /* HeapReference<Class> */ temp1 = temp1->component_type_
2560 __ Ldr(temp1, HeapOperand(temp1, component_offset));
2561 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2562 // /* HeapReference<Class> */ temp1 = temp1->super_class_
2563 __ Ldr(temp1, HeapOperand(temp1, super_offset));
2564 // No need to unpoison the result, we're comparing against null.
2565 __ Cbnz(temp1, intrinsic_slow_path->GetEntryLabel());
2566 __ Bind(&do_copy);
2567 } else {
2568 __ B(intrinsic_slow_path->GetEntryLabel(), ne);
2569 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002570 }
2571 } else if (!optimizations.GetSourceIsNonPrimitiveArray()) {
2572 DCHECK(optimizations.GetDestinationIsNonPrimitiveArray());
2573 // Bail out if the source is not a non primitive array.
Roland Levillain0b671c02016-08-19 12:02:34 +01002574 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2575 // /* HeapReference<Class> */ temp1 = src->klass_
2576 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2577 temp1_loc,
2578 src.W(),
2579 class_offset,
2580 temp2,
2581 /* needs_null_check */ false,
2582 /* use_load_acquire */ false);
2583 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2584 codegen_->GenerateFieldLoadWithBakerReadBarrier(invoke,
2585 temp2_loc,
2586 temp1,
2587 component_offset,
2588 temp3,
2589 /* needs_null_check */ false,
2590 /* use_load_acquire */ false);
2591 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2592 // If heap poisoning is enabled, `temp2` has been unpoisoned
2593 // by the the previous call to GenerateFieldLoadWithBakerReadBarrier.
2594 } else {
2595 // /* HeapReference<Class> */ temp1 = src->klass_
2596 __ Ldr(temp1, HeapOperand(src.W(), class_offset));
2597 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp1);
2598 // /* HeapReference<Class> */ temp2 = temp1->component_type_
2599 __ Ldr(temp2, HeapOperand(temp1, component_offset));
2600 __ Cbz(temp2, intrinsic_slow_path->GetEntryLabel());
2601 codegen_->GetAssembler()->MaybeUnpoisonHeapReference(temp2);
2602 }
2603 // /* uint16_t */ temp2 = static_cast<uint16>(temp2->primitive_type_);
2604 __ Ldrh(temp2, HeapOperand(temp2, primitive_offset));
donghui.baic2ec9ad2016-03-10 14:02:55 +08002605 static_assert(Primitive::kPrimNot == 0, "Expected 0 for kPrimNot");
Roland Levillain0b671c02016-08-19 12:02:34 +01002606 __ Cbnz(temp2, intrinsic_slow_path->GetEntryLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002607 }
2608
2609 Register src_curr_addr = temp1.X();
2610 Register dst_curr_addr = temp2.X();
Roland Levillain0b671c02016-08-19 12:02:34 +01002611 Register src_stop_addr;
2612 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2613 // Temporary register IP0, obtained from the VIXL scratch
2614 // register pool as `temp3`, cannot be used in
2615 // ReadBarrierSystemArrayCopySlowPathARM64 (because that
2616 // register is clobbered by ReadBarrierMarkRegX entry points).
2617 // So another temporary register allocated by the register
2618 // allocator instead.
2619 DCHECK_EQ(LocationFrom(temp3).reg(), IP0);
2620 src_stop_addr = XRegisterFrom(locations->GetTemp(2));
2621 } else {
2622 src_stop_addr = temp3.X();
2623 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002624
2625 GenSystemArrayCopyAddresses(masm,
2626 Primitive::kPrimNot,
2627 src,
2628 src_pos,
2629 dest,
2630 dest_pos,
2631 length,
2632 src_curr_addr,
2633 dst_curr_addr,
2634 src_stop_addr);
2635
donghui.baic2ec9ad2016-03-10 14:02:55 +08002636 const int32_t element_size = Primitive::ComponentSize(Primitive::kPrimNot);
Roland Levillain0b671c02016-08-19 12:02:34 +01002637
2638 if (kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
2639 // SystemArrayCopy implementation for Baker read barriers (see
2640 // also CodeGeneratorARM::GenerateReferenceLoadWithBakerReadBarrier):
2641 //
2642 // if (src_ptr != end_ptr) {
2643 // uint32_t rb_state = Lockword(src->monitor_).ReadBarrierState();
2644 // lfence; // Load fence or artificial data dependency to prevent load-load reordering
2645 // bool is_gray = (rb_state == ReadBarrier::gray_ptr_);
2646 // if (is_gray) {
2647 // // Slow-path copy.
2648 // do {
2649 // *dest_ptr++ = MaybePoison(ReadBarrier::Mark(MaybeUnpoison(*src_ptr++)));
2650 // } while (src_ptr != end_ptr)
2651 // } else {
2652 // // Fast-path copy.
2653 // do {
2654 // *dest_ptr++ = *src_ptr++;
2655 // } while (src_ptr != end_ptr)
2656 // }
2657 // }
2658
2659 vixl::aarch64::Label loop, done;
2660
2661 // Don't enter copy loop if `length == 0`.
2662 __ Cmp(src_curr_addr, src_stop_addr);
2663 __ B(&done, eq);
2664
donghui.baic2ec9ad2016-03-10 14:02:55 +08002665 Register tmp = temps.AcquireW();
Roland Levillain0b671c02016-08-19 12:02:34 +01002666 // Make sure `tmp` is not IP0, as it is clobbered by
2667 // ReadBarrierMarkRegX entry points in
2668 // ReadBarrierSystemArrayCopySlowPathARM64.
2669 DCHECK_NE(LocationFrom(tmp).reg(), IP0);
2670
2671 // /* int32_t */ monitor = src->monitor_
2672 __ Ldr(tmp, HeapOperand(src.W(), monitor_offset));
2673 // /* LockWord */ lock_word = LockWord(monitor)
2674 static_assert(sizeof(LockWord) == sizeof(int32_t),
2675 "art::LockWord and int32_t have different sizes.");
2676
2677 // Introduce a dependency on the lock_word including rb_state,
2678 // to prevent load-load reordering, and without using
2679 // a memory barrier (which would be more expensive).
2680 // `src` is unchanged by this operation, but its value now depends
2681 // on `tmp`.
2682 __ Add(src.X(), src.X(), Operand(tmp.X(), LSR, 32));
2683
2684 // Slow path used to copy array when `src` is gray.
2685 SlowPathCodeARM64* read_barrier_slow_path =
2686 new (GetAllocator()) ReadBarrierSystemArrayCopySlowPathARM64(invoke, LocationFrom(tmp));
2687 codegen_->AddSlowPath(read_barrier_slow_path);
2688
2689 // Given the numeric representation, it's enough to check the low bit of the rb_state.
2690 static_assert(ReadBarrier::white_ptr_ == 0, "Expecting white to have value 0");
2691 static_assert(ReadBarrier::gray_ptr_ == 1, "Expecting gray to have value 1");
2692 static_assert(ReadBarrier::black_ptr_ == 2, "Expecting black to have value 2");
2693 __ Tbnz(tmp, LockWord::kReadBarrierStateShift, read_barrier_slow_path->GetEntryLabel());
2694
2695 // Fast-path copy.
2696 // Iterate over the arrays and do a raw copy of the objects. We don't need to
2697 // poison/unpoison.
2698 __ Bind(&loop);
Scott Wakeling97c72b72016-06-24 16:19:36 +01002699 __ Ldr(tmp, MemOperand(src_curr_addr, element_size, PostIndex));
2700 __ Str(tmp, MemOperand(dst_curr_addr, element_size, PostIndex));
Roland Levillain0b671c02016-08-19 12:02:34 +01002701 __ Cmp(src_curr_addr, src_stop_addr);
2702 __ B(&loop, ne);
2703
2704 __ Bind(read_barrier_slow_path->GetExitLabel());
2705 __ Bind(&done);
2706 } else {
2707 // Non read barrier code.
2708
2709 // Iterate over the arrays and do a raw copy of the objects. We don't need to
2710 // poison/unpoison.
2711 vixl::aarch64::Label loop, done;
2712 __ Bind(&loop);
2713 __ Cmp(src_curr_addr, src_stop_addr);
2714 __ B(&done, eq);
2715 {
2716 Register tmp = temps.AcquireW();
2717 __ Ldr(tmp, MemOperand(src_curr_addr, element_size, PostIndex));
2718 __ Str(tmp, MemOperand(dst_curr_addr, element_size, PostIndex));
2719 }
2720 __ B(&loop);
2721 __ Bind(&done);
donghui.baic2ec9ad2016-03-10 14:02:55 +08002722 }
donghui.baic2ec9ad2016-03-10 14:02:55 +08002723 }
2724 // We only need one card marking on the destination array.
2725 codegen_->MarkGCCard(dest.W(), Register(), /* value_can_be_null */ false);
2726
Roland Levillain0b671c02016-08-19 12:02:34 +01002727 __ Bind(intrinsic_slow_path->GetExitLabel());
donghui.baic2ec9ad2016-03-10 14:02:55 +08002728}
2729
Anton Kirilova3ffea22016-04-07 17:02:37 +01002730static void GenIsInfinite(LocationSummary* locations,
2731 bool is64bit,
Scott Wakeling97c72b72016-06-24 16:19:36 +01002732 MacroAssembler* masm) {
Anton Kirilova3ffea22016-04-07 17:02:37 +01002733 Operand infinity;
2734 Register out;
2735
2736 if (is64bit) {
2737 infinity = kPositiveInfinityDouble;
2738 out = XRegisterFrom(locations->Out());
2739 } else {
2740 infinity = kPositiveInfinityFloat;
2741 out = WRegisterFrom(locations->Out());
2742 }
2743
Scott Wakeling97c72b72016-06-24 16:19:36 +01002744 const Register zero = vixl::aarch64::Assembler::AppropriateZeroRegFor(out);
Anton Kirilova3ffea22016-04-07 17:02:37 +01002745
2746 MoveFPToInt(locations, is64bit, masm);
2747 __ Eor(out, out, infinity);
2748 // We don't care about the sign bit, so shift left.
2749 __ Cmp(zero, Operand(out, LSL, 1));
2750 __ Cset(out, eq);
2751}
2752
2753void IntrinsicLocationsBuilderARM64::VisitFloatIsInfinite(HInvoke* invoke) {
2754 CreateFPToIntLocations(arena_, invoke);
2755}
2756
2757void IntrinsicCodeGeneratorARM64::VisitFloatIsInfinite(HInvoke* invoke) {
2758 GenIsInfinite(invoke->GetLocations(), /* is64bit */ false, GetVIXLAssembler());
2759}
2760
2761void IntrinsicLocationsBuilderARM64::VisitDoubleIsInfinite(HInvoke* invoke) {
2762 CreateFPToIntLocations(arena_, invoke);
2763}
2764
2765void IntrinsicCodeGeneratorARM64::VisitDoubleIsInfinite(HInvoke* invoke) {
2766 GenIsInfinite(invoke->GetLocations(), /* is64bit */ true, GetVIXLAssembler());
2767}
2768
Aart Bik2f9fcc92016-03-01 15:16:54 -08002769UNIMPLEMENTED_INTRINSIC(ARM64, ReferenceGetReferent)
Aart Bik2f9fcc92016-03-01 15:16:54 -08002770UNIMPLEMENTED_INTRINSIC(ARM64, IntegerHighestOneBit)
2771UNIMPLEMENTED_INTRINSIC(ARM64, LongHighestOneBit)
2772UNIMPLEMENTED_INTRINSIC(ARM64, IntegerLowestOneBit)
2773UNIMPLEMENTED_INTRINSIC(ARM64, LongLowestOneBit)
Andreas Gampe878d58c2015-01-15 23:24:00 -08002774
Aart Bik0e54c012016-03-04 12:08:31 -08002775// 1.8.
2776UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndAddInt)
2777UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndAddLong)
2778UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetInt)
2779UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetLong)
2780UNIMPLEMENTED_INTRINSIC(ARM64, UnsafeGetAndSetObject)
Aart Bik0e54c012016-03-04 12:08:31 -08002781
Aart Bik2f9fcc92016-03-01 15:16:54 -08002782UNREACHABLE_INTRINSICS(ARM64)
Roland Levillain4d027112015-07-01 15:41:14 +01002783
2784#undef __
2785
Andreas Gampe878d58c2015-01-15 23:24:00 -08002786} // namespace arm64
2787} // namespace art