blob: 504e746876d8114f1bc4306dcf434368d5023447 [file] [log] [blame]
Andreas Gampe71fb52f2014-12-29 17:43:08 -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_x86_64.h"
18
Andreas Gampe21030dd2015-05-07 14:46:15 -070019#include <limits>
20
Mark Mendellfb8d2792015-03-31 22:16:59 -040021#include "arch/x86_64/instruction_set_features_x86_64.h"
Mathieu Chartiere401d142015-04-22 13:56:20 -070022#include "art_method-inl.h"
Mark Mendelld5897672015-08-12 21:16:41 -040023#include "base/bit_utils.h"
Andreas Gampe71fb52f2014-12-29 17:43:08 -080024#include "code_generator_x86_64.h"
25#include "entrypoints/quick/quick_entrypoints.h"
26#include "intrinsics.h"
Andreas Gampe85b62f22015-09-09 13:15:38 -070027#include "intrinsics_utils.h"
Andreas Gampe71fb52f2014-12-29 17:43:08 -080028#include "mirror/array-inl.h"
Andreas Gampe71fb52f2014-12-29 17:43:08 -080029#include "mirror/string.h"
30#include "thread.h"
31#include "utils/x86_64/assembler_x86_64.h"
32#include "utils/x86_64/constants_x86_64.h"
33
34namespace art {
35
36namespace x86_64 {
37
Mark Mendellfb8d2792015-03-31 22:16:59 -040038IntrinsicLocationsBuilderX86_64::IntrinsicLocationsBuilderX86_64(CodeGeneratorX86_64* codegen)
39 : arena_(codegen->GetGraph()->GetArena()), codegen_(codegen) {
40}
41
42
Andreas Gampe71fb52f2014-12-29 17:43:08 -080043X86_64Assembler* IntrinsicCodeGeneratorX86_64::GetAssembler() {
Roland Levillainb488b782015-10-22 11:38:49 +010044 return down_cast<X86_64Assembler*>(codegen_->GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -080045}
46
Andreas Gampe878d58c2015-01-15 23:24:00 -080047ArenaAllocator* IntrinsicCodeGeneratorX86_64::GetAllocator() {
Andreas Gampe71fb52f2014-12-29 17:43:08 -080048 return codegen_->GetGraph()->GetArena();
49}
50
51bool IntrinsicLocationsBuilderX86_64::TryDispatch(HInvoke* invoke) {
52 Dispatch(invoke);
Roland Levillain0d5a2812015-11-13 10:07:31 +000053 LocationSummary* res = invoke->GetLocations();
54 if (res == nullptr) {
55 return false;
56 }
57 if (kEmitCompilerReadBarrier && res->CanCall()) {
58 // Generating an intrinsic for this HInvoke may produce an
59 // IntrinsicSlowPathX86_64 slow path. Currently this approach
60 // does not work when using read barriers, as the emitted
61 // calling sequence will make use of another slow path
62 // (ReadBarrierForRootSlowPathX86_64 for HInvokeStaticOrDirect,
63 // ReadBarrierSlowPathX86_64 for HInvokeVirtual). So we bail
64 // out in this case.
65 //
66 // TODO: Find a way to have intrinsics work with read barriers.
67 invoke->SetLocations(nullptr);
68 return false;
69 }
70 return res->Intrinsified();
Andreas Gampe71fb52f2014-12-29 17:43:08 -080071}
72
Roland Levillainec525fc2015-04-28 15:50:20 +010073static void MoveArguments(HInvoke* invoke, CodeGeneratorX86_64* codegen) {
Roland Levillain2d27c8e2015-04-28 15:48:45 +010074 InvokeDexCallingConventionVisitorX86_64 calling_convention_visitor;
Roland Levillainec525fc2015-04-28 15:50:20 +010075 IntrinsicVisitor::MoveArguments(invoke, codegen, &calling_convention_visitor);
Andreas Gampe71fb52f2014-12-29 17:43:08 -080076}
77
Andreas Gampe85b62f22015-09-09 13:15:38 -070078using IntrinsicSlowPathX86_64 = IntrinsicSlowPath<InvokeDexCallingConventionVisitorX86_64>;
Andreas Gampe71fb52f2014-12-29 17:43:08 -080079
Andreas Gampe71fb52f2014-12-29 17:43:08 -080080#define __ assembler->
81
82static void CreateFPToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
83 LocationSummary* locations = new (arena) LocationSummary(invoke,
84 LocationSummary::kNoCall,
85 kIntrinsified);
86 locations->SetInAt(0, Location::RequiresFpuRegister());
87 locations->SetOut(Location::RequiresRegister());
88}
89
90static void CreateIntToFPLocations(ArenaAllocator* arena, HInvoke* invoke) {
91 LocationSummary* locations = new (arena) LocationSummary(invoke,
92 LocationSummary::kNoCall,
93 kIntrinsified);
94 locations->SetInAt(0, Location::RequiresRegister());
95 locations->SetOut(Location::RequiresFpuRegister());
96}
97
98static void MoveFPToInt(LocationSummary* locations, bool is64bit, X86_64Assembler* assembler) {
99 Location input = locations->InAt(0);
100 Location output = locations->Out();
101 __ movd(output.AsRegister<CpuRegister>(), input.AsFpuRegister<XmmRegister>(), is64bit);
102}
103
104static void MoveIntToFP(LocationSummary* locations, bool is64bit, X86_64Assembler* assembler) {
105 Location input = locations->InAt(0);
106 Location output = locations->Out();
107 __ movd(output.AsFpuRegister<XmmRegister>(), input.AsRegister<CpuRegister>(), is64bit);
108}
109
110void IntrinsicLocationsBuilderX86_64::VisitDoubleDoubleToRawLongBits(HInvoke* invoke) {
111 CreateFPToIntLocations(arena_, invoke);
112}
113void IntrinsicLocationsBuilderX86_64::VisitDoubleLongBitsToDouble(HInvoke* invoke) {
114 CreateIntToFPLocations(arena_, invoke);
115}
116
117void IntrinsicCodeGeneratorX86_64::VisitDoubleDoubleToRawLongBits(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000118 MoveFPToInt(invoke->GetLocations(), /* is64bit */ true, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800119}
120void IntrinsicCodeGeneratorX86_64::VisitDoubleLongBitsToDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000121 MoveIntToFP(invoke->GetLocations(), /* is64bit */ true, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800122}
123
124void IntrinsicLocationsBuilderX86_64::VisitFloatFloatToRawIntBits(HInvoke* invoke) {
125 CreateFPToIntLocations(arena_, invoke);
126}
127void IntrinsicLocationsBuilderX86_64::VisitFloatIntBitsToFloat(HInvoke* invoke) {
128 CreateIntToFPLocations(arena_, invoke);
129}
130
131void IntrinsicCodeGeneratorX86_64::VisitFloatFloatToRawIntBits(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000132 MoveFPToInt(invoke->GetLocations(), /* is64bit */ false, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800133}
134void IntrinsicCodeGeneratorX86_64::VisitFloatIntBitsToFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000135 MoveIntToFP(invoke->GetLocations(), /* is64bit */ false, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800136}
137
138static void CreateIntToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
139 LocationSummary* locations = new (arena) LocationSummary(invoke,
140 LocationSummary::kNoCall,
141 kIntrinsified);
142 locations->SetInAt(0, Location::RequiresRegister());
143 locations->SetOut(Location::SameAsFirstInput());
144}
145
146static void GenReverseBytes(LocationSummary* locations,
147 Primitive::Type size,
148 X86_64Assembler* assembler) {
149 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
150
151 switch (size) {
152 case Primitive::kPrimShort:
153 // TODO: Can be done with an xchg of 8b registers. This is straight from Quick.
154 __ bswapl(out);
155 __ sarl(out, Immediate(16));
156 break;
157 case Primitive::kPrimInt:
158 __ bswapl(out);
159 break;
160 case Primitive::kPrimLong:
161 __ bswapq(out);
162 break;
163 default:
164 LOG(FATAL) << "Unexpected size for reverse-bytes: " << size;
165 UNREACHABLE();
166 }
167}
168
169void IntrinsicLocationsBuilderX86_64::VisitIntegerReverseBytes(HInvoke* invoke) {
170 CreateIntToIntLocations(arena_, invoke);
171}
172
173void IntrinsicCodeGeneratorX86_64::VisitIntegerReverseBytes(HInvoke* invoke) {
174 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimInt, GetAssembler());
175}
176
177void IntrinsicLocationsBuilderX86_64::VisitLongReverseBytes(HInvoke* invoke) {
178 CreateIntToIntLocations(arena_, invoke);
179}
180
181void IntrinsicCodeGeneratorX86_64::VisitLongReverseBytes(HInvoke* invoke) {
182 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimLong, GetAssembler());
183}
184
185void IntrinsicLocationsBuilderX86_64::VisitShortReverseBytes(HInvoke* invoke) {
186 CreateIntToIntLocations(arena_, invoke);
187}
188
189void IntrinsicCodeGeneratorX86_64::VisitShortReverseBytes(HInvoke* invoke) {
190 GenReverseBytes(invoke->GetLocations(), Primitive::kPrimShort, GetAssembler());
191}
192
193
194// TODO: Consider Quick's way of doing Double abs through integer operations, as the immediate we
195// need is 64b.
196
197static void CreateFloatToFloatPlusTemps(ArenaAllocator* arena, HInvoke* invoke) {
198 // TODO: Enable memory operations when the assembler supports them.
199 LocationSummary* locations = new (arena) LocationSummary(invoke,
200 LocationSummary::kNoCall,
201 kIntrinsified);
202 locations->SetInAt(0, Location::RequiresFpuRegister());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800203 locations->SetOut(Location::SameAsFirstInput());
Mark Mendellf55c3e02015-03-26 21:07:46 -0400204 locations->AddTemp(Location::RequiresFpuRegister()); // FP reg to hold mask.
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800205}
206
Mark Mendell39dcf552015-04-09 20:42:42 -0400207static void MathAbsFP(LocationSummary* locations,
208 bool is64bit,
209 X86_64Assembler* assembler,
210 CodeGeneratorX86_64* codegen) {
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800211 Location output = locations->Out();
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800212
Mark Mendellcfa410b2015-05-25 16:02:44 -0400213 DCHECK(output.IsFpuRegister());
214 XmmRegister xmm_temp = locations->GetTemp(0).AsFpuRegister<XmmRegister>();
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800215
Mark Mendellcfa410b2015-05-25 16:02:44 -0400216 // TODO: Can mask directly with constant area using pand if we can guarantee
217 // that the literal is aligned on a 16 byte boundary. This will avoid a
218 // temporary.
219 if (is64bit) {
220 __ movsd(xmm_temp, codegen->LiteralInt64Address(INT64_C(0x7FFFFFFFFFFFFFFF)));
221 __ andpd(output.AsFpuRegister<XmmRegister>(), xmm_temp);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800222 } else {
Mark Mendellcfa410b2015-05-25 16:02:44 -0400223 __ movss(xmm_temp, codegen->LiteralInt32Address(INT32_C(0x7FFFFFFF)));
224 __ andps(output.AsFpuRegister<XmmRegister>(), xmm_temp);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800225 }
226}
227
228void IntrinsicLocationsBuilderX86_64::VisitMathAbsDouble(HInvoke* invoke) {
229 CreateFloatToFloatPlusTemps(arena_, invoke);
230}
231
232void IntrinsicCodeGeneratorX86_64::VisitMathAbsDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000233 MathAbsFP(invoke->GetLocations(), /* is64bit */ true, GetAssembler(), codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800234}
235
236void IntrinsicLocationsBuilderX86_64::VisitMathAbsFloat(HInvoke* invoke) {
237 CreateFloatToFloatPlusTemps(arena_, invoke);
238}
239
240void IntrinsicCodeGeneratorX86_64::VisitMathAbsFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000241 MathAbsFP(invoke->GetLocations(), /* is64bit */ false, GetAssembler(), codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800242}
243
244static void CreateIntToIntPlusTemp(ArenaAllocator* arena, HInvoke* invoke) {
245 LocationSummary* locations = new (arena) LocationSummary(invoke,
246 LocationSummary::kNoCall,
247 kIntrinsified);
248 locations->SetInAt(0, Location::RequiresRegister());
249 locations->SetOut(Location::SameAsFirstInput());
250 locations->AddTemp(Location::RequiresRegister());
251}
252
253static void GenAbsInteger(LocationSummary* locations, bool is64bit, X86_64Assembler* assembler) {
254 Location output = locations->Out();
255 CpuRegister out = output.AsRegister<CpuRegister>();
256 CpuRegister mask = locations->GetTemp(0).AsRegister<CpuRegister>();
257
258 if (is64bit) {
259 // Create mask.
260 __ movq(mask, out);
261 __ sarq(mask, Immediate(63));
262 // Add mask.
263 __ addq(out, mask);
264 __ xorq(out, mask);
265 } else {
266 // Create mask.
267 __ movl(mask, out);
268 __ sarl(mask, Immediate(31));
269 // Add mask.
270 __ addl(out, mask);
271 __ xorl(out, mask);
272 }
273}
274
275void IntrinsicLocationsBuilderX86_64::VisitMathAbsInt(HInvoke* invoke) {
276 CreateIntToIntPlusTemp(arena_, invoke);
277}
278
279void IntrinsicCodeGeneratorX86_64::VisitMathAbsInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000280 GenAbsInteger(invoke->GetLocations(), /* is64bit */ false, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800281}
282
283void IntrinsicLocationsBuilderX86_64::VisitMathAbsLong(HInvoke* invoke) {
284 CreateIntToIntPlusTemp(arena_, invoke);
285}
286
287void IntrinsicCodeGeneratorX86_64::VisitMathAbsLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000288 GenAbsInteger(invoke->GetLocations(), /* is64bit */ true, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800289}
290
Mark Mendell39dcf552015-04-09 20:42:42 -0400291static void GenMinMaxFP(LocationSummary* locations,
292 bool is_min,
293 bool is_double,
294 X86_64Assembler* assembler,
295 CodeGeneratorX86_64* codegen) {
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800296 Location op1_loc = locations->InAt(0);
297 Location op2_loc = locations->InAt(1);
298 Location out_loc = locations->Out();
299 XmmRegister out = out_loc.AsFpuRegister<XmmRegister>();
300
301 // Shortcut for same input locations.
302 if (op1_loc.Equals(op2_loc)) {
303 DCHECK(out_loc.Equals(op1_loc));
304 return;
305 }
306
307 // (out := op1)
308 // out <=? op2
309 // if Nan jmp Nan_label
310 // if out is min jmp done
311 // if op2 is min jmp op2_label
312 // handle -0/+0
313 // jmp done
314 // Nan_label:
315 // out := NaN
316 // op2_label:
317 // out := op2
318 // done:
319 //
320 // This removes one jmp, but needs to copy one input (op1) to out.
321 //
Mark Mendellf55c3e02015-03-26 21:07:46 -0400322 // TODO: This is straight from Quick. Make NaN an out-of-line slowpath?
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800323
324 XmmRegister op2 = op2_loc.AsFpuRegister<XmmRegister>();
325
Mark Mendell0c9497d2015-08-21 09:30:05 -0400326 NearLabel nan, done, op2_label;
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800327 if (is_double) {
328 __ ucomisd(out, op2);
329 } else {
330 __ ucomiss(out, op2);
331 }
332
333 __ j(Condition::kParityEven, &nan);
334
335 __ j(is_min ? Condition::kAbove : Condition::kBelow, &op2_label);
336 __ j(is_min ? Condition::kBelow : Condition::kAbove, &done);
337
338 // Handle 0.0/-0.0.
339 if (is_min) {
340 if (is_double) {
341 __ orpd(out, op2);
342 } else {
343 __ orps(out, op2);
344 }
345 } else {
346 if (is_double) {
347 __ andpd(out, op2);
348 } else {
349 __ andps(out, op2);
350 }
351 }
352 __ jmp(&done);
353
354 // NaN handling.
355 __ Bind(&nan);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800356 if (is_double) {
Mark Mendellf55c3e02015-03-26 21:07:46 -0400357 __ movsd(out, codegen->LiteralInt64Address(INT64_C(0x7FF8000000000000)));
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800358 } else {
Mark Mendellf55c3e02015-03-26 21:07:46 -0400359 __ movss(out, codegen->LiteralInt32Address(INT32_C(0x7FC00000)));
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800360 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800361 __ jmp(&done);
362
363 // out := op2;
364 __ Bind(&op2_label);
365 if (is_double) {
366 __ movsd(out, op2);
367 } else {
368 __ movss(out, op2);
369 }
370
371 // Done.
372 __ Bind(&done);
373}
374
Mark Mendellf55c3e02015-03-26 21:07:46 -0400375static void CreateFPFPToFP(ArenaAllocator* arena, HInvoke* invoke) {
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800376 LocationSummary* locations = new (arena) LocationSummary(invoke,
377 LocationSummary::kNoCall,
378 kIntrinsified);
379 locations->SetInAt(0, Location::RequiresFpuRegister());
380 locations->SetInAt(1, Location::RequiresFpuRegister());
381 // The following is sub-optimal, but all we can do for now. It would be fine to also accept
382 // the second input to be the output (we can simply swap inputs).
383 locations->SetOut(Location::SameAsFirstInput());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800384}
385
386void IntrinsicLocationsBuilderX86_64::VisitMathMinDoubleDouble(HInvoke* invoke) {
Mark Mendellf55c3e02015-03-26 21:07:46 -0400387 CreateFPFPToFP(arena_, invoke);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800388}
389
390void IntrinsicCodeGeneratorX86_64::VisitMathMinDoubleDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000391 GenMinMaxFP(
392 invoke->GetLocations(), /* is_min */ true, /* is_double */ true, GetAssembler(), codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800393}
394
395void IntrinsicLocationsBuilderX86_64::VisitMathMinFloatFloat(HInvoke* invoke) {
Mark Mendellf55c3e02015-03-26 21:07:46 -0400396 CreateFPFPToFP(arena_, invoke);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800397}
398
399void IntrinsicCodeGeneratorX86_64::VisitMathMinFloatFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000400 GenMinMaxFP(
401 invoke->GetLocations(), /* is_min */ true, /* is_double */ false, GetAssembler(), codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800402}
403
404void IntrinsicLocationsBuilderX86_64::VisitMathMaxDoubleDouble(HInvoke* invoke) {
Mark Mendellf55c3e02015-03-26 21:07:46 -0400405 CreateFPFPToFP(arena_, invoke);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800406}
407
408void IntrinsicCodeGeneratorX86_64::VisitMathMaxDoubleDouble(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000409 GenMinMaxFP(
410 invoke->GetLocations(), /* is_min */ false, /* is_double */ true, GetAssembler(), codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800411}
412
413void IntrinsicLocationsBuilderX86_64::VisitMathMaxFloatFloat(HInvoke* invoke) {
Mark Mendellf55c3e02015-03-26 21:07:46 -0400414 CreateFPFPToFP(arena_, invoke);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800415}
416
417void IntrinsicCodeGeneratorX86_64::VisitMathMaxFloatFloat(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000418 GenMinMaxFP(
419 invoke->GetLocations(), /* is_min */ false, /* is_double */ false, GetAssembler(), codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800420}
421
422static void GenMinMax(LocationSummary* locations, bool is_min, bool is_long,
423 X86_64Assembler* assembler) {
424 Location op1_loc = locations->InAt(0);
425 Location op2_loc = locations->InAt(1);
426
427 // Shortcut for same input locations.
428 if (op1_loc.Equals(op2_loc)) {
429 // Can return immediately, as op1_loc == out_loc.
430 // Note: if we ever support separate registers, e.g., output into memory, we need to check for
431 // a copy here.
432 DCHECK(locations->Out().Equals(op1_loc));
433 return;
434 }
435
436 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
437 CpuRegister op2 = op2_loc.AsRegister<CpuRegister>();
438
439 // (out := op1)
440 // out <=? op2
441 // if out is min jmp done
442 // out := op2
443 // done:
444
445 if (is_long) {
446 __ cmpq(out, op2);
447 } else {
448 __ cmpl(out, op2);
449 }
450
451 __ cmov(is_min ? Condition::kGreater : Condition::kLess, out, op2, is_long);
452}
453
454static void CreateIntIntToIntLocations(ArenaAllocator* arena, HInvoke* invoke) {
455 LocationSummary* locations = new (arena) LocationSummary(invoke,
456 LocationSummary::kNoCall,
457 kIntrinsified);
458 locations->SetInAt(0, Location::RequiresRegister());
459 locations->SetInAt(1, Location::RequiresRegister());
460 locations->SetOut(Location::SameAsFirstInput());
461}
462
463void IntrinsicLocationsBuilderX86_64::VisitMathMinIntInt(HInvoke* invoke) {
464 CreateIntIntToIntLocations(arena_, invoke);
465}
466
467void IntrinsicCodeGeneratorX86_64::VisitMathMinIntInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000468 GenMinMax(invoke->GetLocations(), /* is_min */ true, /* is_long */ false, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800469}
470
471void IntrinsicLocationsBuilderX86_64::VisitMathMinLongLong(HInvoke* invoke) {
472 CreateIntIntToIntLocations(arena_, invoke);
473}
474
475void IntrinsicCodeGeneratorX86_64::VisitMathMinLongLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000476 GenMinMax(invoke->GetLocations(), /* is_min */ true, /* is_long */ true, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800477}
478
479void IntrinsicLocationsBuilderX86_64::VisitMathMaxIntInt(HInvoke* invoke) {
480 CreateIntIntToIntLocations(arena_, invoke);
481}
482
483void IntrinsicCodeGeneratorX86_64::VisitMathMaxIntInt(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000484 GenMinMax(invoke->GetLocations(), /* is_min */ false, /* is_long */ false, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800485}
486
487void IntrinsicLocationsBuilderX86_64::VisitMathMaxLongLong(HInvoke* invoke) {
488 CreateIntIntToIntLocations(arena_, invoke);
489}
490
491void IntrinsicCodeGeneratorX86_64::VisitMathMaxLongLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000492 GenMinMax(invoke->GetLocations(), /* is_min */ false, /* is_long */ true, GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800493}
494
495static void CreateFPToFPLocations(ArenaAllocator* arena, HInvoke* invoke) {
496 LocationSummary* locations = new (arena) LocationSummary(invoke,
497 LocationSummary::kNoCall,
498 kIntrinsified);
499 locations->SetInAt(0, Location::RequiresFpuRegister());
500 locations->SetOut(Location::RequiresFpuRegister());
501}
502
503void IntrinsicLocationsBuilderX86_64::VisitMathSqrt(HInvoke* invoke) {
504 CreateFPToFPLocations(arena_, invoke);
505}
506
507void IntrinsicCodeGeneratorX86_64::VisitMathSqrt(HInvoke* invoke) {
508 LocationSummary* locations = invoke->GetLocations();
509 XmmRegister in = locations->InAt(0).AsFpuRegister<XmmRegister>();
510 XmmRegister out = locations->Out().AsFpuRegister<XmmRegister>();
511
512 GetAssembler()->sqrtsd(out, in);
513}
514
Mark Mendellfb8d2792015-03-31 22:16:59 -0400515static void InvokeOutOfLineIntrinsic(CodeGeneratorX86_64* codegen, HInvoke* invoke) {
Roland Levillainec525fc2015-04-28 15:50:20 +0100516 MoveArguments(invoke, codegen);
Mark Mendellfb8d2792015-03-31 22:16:59 -0400517
518 DCHECK(invoke->IsInvokeStaticOrDirect());
Nicolas Geoffray94015b92015-06-04 18:21:04 +0100519 codegen->GenerateStaticOrDirectCall(
520 invoke->AsInvokeStaticOrDirect(), Location::RegisterLocation(RDI));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400521 codegen->RecordPcInfo(invoke, invoke->GetDexPc());
522
523 // Copy the result back to the expected output.
524 Location out = invoke->GetLocations()->Out();
525 if (out.IsValid()) {
526 DCHECK(out.IsRegister());
Andreas Gampe85b62f22015-09-09 13:15:38 -0700527 codegen->MoveFromReturnRegister(out, invoke->GetType());
Mark Mendellfb8d2792015-03-31 22:16:59 -0400528 }
529}
530
531static void CreateSSE41FPToFPLocations(ArenaAllocator* arena,
532 HInvoke* invoke,
533 CodeGeneratorX86_64* codegen) {
534 // Do we have instruction support?
535 if (codegen->GetInstructionSetFeatures().HasSSE4_1()) {
536 CreateFPToFPLocations(arena, invoke);
537 return;
538 }
539
540 // We have to fall back to a call to the intrinsic.
541 LocationSummary* locations = new (arena) LocationSummary(invoke,
542 LocationSummary::kCall);
543 InvokeRuntimeCallingConvention calling_convention;
544 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetFpuRegisterAt(0)));
545 locations->SetOut(Location::FpuRegisterLocation(XMM0));
546 // Needs to be RDI for the invoke.
547 locations->AddTemp(Location::RegisterLocation(RDI));
548}
549
550static void GenSSE41FPToFPIntrinsic(CodeGeneratorX86_64* codegen,
551 HInvoke* invoke,
552 X86_64Assembler* assembler,
553 int round_mode) {
554 LocationSummary* locations = invoke->GetLocations();
555 if (locations->WillCall()) {
556 InvokeOutOfLineIntrinsic(codegen, invoke);
557 } else {
558 XmmRegister in = locations->InAt(0).AsFpuRegister<XmmRegister>();
559 XmmRegister out = locations->Out().AsFpuRegister<XmmRegister>();
560 __ roundsd(out, in, Immediate(round_mode));
561 }
562}
563
564void IntrinsicLocationsBuilderX86_64::VisitMathCeil(HInvoke* invoke) {
565 CreateSSE41FPToFPLocations(arena_, invoke, codegen_);
566}
567
568void IntrinsicCodeGeneratorX86_64::VisitMathCeil(HInvoke* invoke) {
569 GenSSE41FPToFPIntrinsic(codegen_, invoke, GetAssembler(), 2);
570}
571
572void IntrinsicLocationsBuilderX86_64::VisitMathFloor(HInvoke* invoke) {
573 CreateSSE41FPToFPLocations(arena_, invoke, codegen_);
574}
575
576void IntrinsicCodeGeneratorX86_64::VisitMathFloor(HInvoke* invoke) {
577 GenSSE41FPToFPIntrinsic(codegen_, invoke, GetAssembler(), 1);
578}
579
580void IntrinsicLocationsBuilderX86_64::VisitMathRint(HInvoke* invoke) {
581 CreateSSE41FPToFPLocations(arena_, invoke, codegen_);
582}
583
584void IntrinsicCodeGeneratorX86_64::VisitMathRint(HInvoke* invoke) {
585 GenSSE41FPToFPIntrinsic(codegen_, invoke, GetAssembler(), 0);
586}
587
588static void CreateSSE41FPToIntLocations(ArenaAllocator* arena,
589 HInvoke* invoke,
590 CodeGeneratorX86_64* codegen) {
591 // Do we have instruction support?
592 if (codegen->GetInstructionSetFeatures().HasSSE4_1()) {
593 LocationSummary* locations = new (arena) LocationSummary(invoke,
594 LocationSummary::kNoCall,
595 kIntrinsified);
596 locations->SetInAt(0, Location::RequiresFpuRegister());
Pavel Vyssotski9ca25712015-07-31 13:03:17 +0600597 locations->SetOut(Location::RequiresRegister());
Mark Mendellfb8d2792015-03-31 22:16:59 -0400598 locations->AddTemp(Location::RequiresFpuRegister());
Mark Mendellfb8d2792015-03-31 22:16:59 -0400599 return;
600 }
601
602 // We have to fall back to a call to the intrinsic.
603 LocationSummary* locations = new (arena) LocationSummary(invoke,
604 LocationSummary::kCall);
605 InvokeRuntimeCallingConvention calling_convention;
606 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetFpuRegisterAt(0)));
607 locations->SetOut(Location::RegisterLocation(RAX));
608 // Needs to be RDI for the invoke.
609 locations->AddTemp(Location::RegisterLocation(RDI));
610}
611
612void IntrinsicLocationsBuilderX86_64::VisitMathRoundFloat(HInvoke* invoke) {
Andreas Gampee6d0d8d2015-12-28 09:54:29 -0800613 // See intrinsics.h.
614 if (kRoundIsPlusPointFive) {
615 CreateSSE41FPToIntLocations(arena_, invoke, codegen_);
616 }
Mark Mendellfb8d2792015-03-31 22:16:59 -0400617}
618
619void IntrinsicCodeGeneratorX86_64::VisitMathRoundFloat(HInvoke* invoke) {
620 LocationSummary* locations = invoke->GetLocations();
621 if (locations->WillCall()) {
622 InvokeOutOfLineIntrinsic(codegen_, invoke);
623 return;
624 }
625
626 // Implement RoundFloat as t1 = floor(input + 0.5f); convert to int.
627 XmmRegister in = locations->InAt(0).AsFpuRegister<XmmRegister>();
628 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
Mark Mendell40741f32015-04-20 22:10:34 -0400629 XmmRegister inPlusPointFive = locations->GetTemp(0).AsFpuRegister<XmmRegister>();
Mark Mendell0c9497d2015-08-21 09:30:05 -0400630 NearLabel done, nan;
Mark Mendellfb8d2792015-03-31 22:16:59 -0400631 X86_64Assembler* assembler = GetAssembler();
632
Mark Mendell40741f32015-04-20 22:10:34 -0400633 // Load 0.5 into inPlusPointFive.
634 __ movss(inPlusPointFive, codegen_->LiteralFloatAddress(0.5f));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400635
636 // Add in the input.
637 __ addss(inPlusPointFive, in);
638
639 // And truncate to an integer.
640 __ roundss(inPlusPointFive, inPlusPointFive, Immediate(1));
641
Pavel Vyssotski9ca25712015-07-31 13:03:17 +0600642 // Load maxInt into out.
643 codegen_->Load64BitValue(out, kPrimIntMax);
644
Mark Mendellfb8d2792015-03-31 22:16:59 -0400645 // if inPlusPointFive >= maxInt goto done
Mark Mendell40741f32015-04-20 22:10:34 -0400646 __ comiss(inPlusPointFive, codegen_->LiteralFloatAddress(static_cast<float>(kPrimIntMax)));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400647 __ j(kAboveEqual, &done);
648
649 // if input == NaN goto nan
650 __ j(kUnordered, &nan);
651
652 // output = float-to-int-truncate(input)
653 __ cvttss2si(out, inPlusPointFive);
654 __ jmp(&done);
655 __ Bind(&nan);
656
657 // output = 0
658 __ xorl(out, out);
659 __ Bind(&done);
660}
661
662void IntrinsicLocationsBuilderX86_64::VisitMathRoundDouble(HInvoke* invoke) {
Andreas Gampee6d0d8d2015-12-28 09:54:29 -0800663 // See intrinsics.h.
664 if (kRoundIsPlusPointFive) {
665 CreateSSE41FPToIntLocations(arena_, invoke, codegen_);
666 }
Mark Mendellfb8d2792015-03-31 22:16:59 -0400667}
668
669void IntrinsicCodeGeneratorX86_64::VisitMathRoundDouble(HInvoke* invoke) {
670 LocationSummary* locations = invoke->GetLocations();
671 if (locations->WillCall()) {
672 InvokeOutOfLineIntrinsic(codegen_, invoke);
673 return;
674 }
675
676 // Implement RoundDouble as t1 = floor(input + 0.5); convert to long.
677 XmmRegister in = locations->InAt(0).AsFpuRegister<XmmRegister>();
678 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
Mark Mendell40741f32015-04-20 22:10:34 -0400679 XmmRegister inPlusPointFive = locations->GetTemp(0).AsFpuRegister<XmmRegister>();
Mark Mendell0c9497d2015-08-21 09:30:05 -0400680 NearLabel done, nan;
Mark Mendellfb8d2792015-03-31 22:16:59 -0400681 X86_64Assembler* assembler = GetAssembler();
682
Mark Mendell40741f32015-04-20 22:10:34 -0400683 // Load 0.5 into inPlusPointFive.
684 __ movsd(inPlusPointFive, codegen_->LiteralDoubleAddress(0.5));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400685
686 // Add in the input.
687 __ addsd(inPlusPointFive, in);
688
689 // And truncate to an integer.
690 __ roundsd(inPlusPointFive, inPlusPointFive, Immediate(1));
691
Pavel Vyssotski9ca25712015-07-31 13:03:17 +0600692 // Load maxLong into out.
693 codegen_->Load64BitValue(out, kPrimLongMax);
694
Mark Mendellfb8d2792015-03-31 22:16:59 -0400695 // if inPlusPointFive >= maxLong goto done
Mark Mendell40741f32015-04-20 22:10:34 -0400696 __ comisd(inPlusPointFive, codegen_->LiteralDoubleAddress(static_cast<double>(kPrimLongMax)));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400697 __ j(kAboveEqual, &done);
698
699 // if input == NaN goto nan
700 __ j(kUnordered, &nan);
701
702 // output = double-to-long-truncate(input)
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000703 __ cvttsd2si(out, inPlusPointFive, /* is64bit */ true);
Mark Mendellfb8d2792015-03-31 22:16:59 -0400704 __ jmp(&done);
705 __ Bind(&nan);
706
707 // output = 0
Mark Mendell92e83bf2015-05-07 11:25:03 -0400708 __ xorl(out, out);
Mark Mendellfb8d2792015-03-31 22:16:59 -0400709 __ Bind(&done);
710}
711
Mark Mendella4f12202015-08-06 15:23:34 -0400712static void CreateFPToFPCallLocations(ArenaAllocator* arena,
713 HInvoke* invoke) {
714 LocationSummary* locations = new (arena) LocationSummary(invoke,
715 LocationSummary::kCall,
716 kIntrinsified);
717 InvokeRuntimeCallingConvention calling_convention;
718 locations->SetInAt(0, Location::FpuRegisterLocation(calling_convention.GetFpuRegisterAt(0)));
719 locations->SetOut(Location::FpuRegisterLocation(XMM0));
720
721 // We have to ensure that the native code doesn't clobber the XMM registers which are
722 // non-volatile for ART, but volatile for Native calls. This will ensure that they are
723 // saved in the prologue and properly restored.
724 for (auto fp_reg : non_volatile_xmm_regs) {
725 locations->AddTemp(Location::FpuRegisterLocation(fp_reg));
726 }
727}
728
729static void GenFPToFPCall(HInvoke* invoke, CodeGeneratorX86_64* codegen,
730 QuickEntrypointEnum entry) {
731 LocationSummary* locations = invoke->GetLocations();
732 DCHECK(locations->WillCall());
733 DCHECK(invoke->IsInvokeStaticOrDirect());
734 X86_64Assembler* assembler = codegen->GetAssembler();
735
736 __ gs()->call(Address::Absolute(GetThreadOffset<kX86_64WordSize>(entry), true));
737 codegen->RecordPcInfo(invoke, invoke->GetDexPc());
738}
739
740void IntrinsicLocationsBuilderX86_64::VisitMathCos(HInvoke* invoke) {
741 CreateFPToFPCallLocations(arena_, invoke);
742}
743
744void IntrinsicCodeGeneratorX86_64::VisitMathCos(HInvoke* invoke) {
745 GenFPToFPCall(invoke, codegen_, kQuickCos);
746}
747
748void IntrinsicLocationsBuilderX86_64::VisitMathSin(HInvoke* invoke) {
749 CreateFPToFPCallLocations(arena_, invoke);
750}
751
752void IntrinsicCodeGeneratorX86_64::VisitMathSin(HInvoke* invoke) {
753 GenFPToFPCall(invoke, codegen_, kQuickSin);
754}
755
756void IntrinsicLocationsBuilderX86_64::VisitMathAcos(HInvoke* invoke) {
757 CreateFPToFPCallLocations(arena_, invoke);
758}
759
760void IntrinsicCodeGeneratorX86_64::VisitMathAcos(HInvoke* invoke) {
761 GenFPToFPCall(invoke, codegen_, kQuickAcos);
762}
763
764void IntrinsicLocationsBuilderX86_64::VisitMathAsin(HInvoke* invoke) {
765 CreateFPToFPCallLocations(arena_, invoke);
766}
767
768void IntrinsicCodeGeneratorX86_64::VisitMathAsin(HInvoke* invoke) {
769 GenFPToFPCall(invoke, codegen_, kQuickAsin);
770}
771
772void IntrinsicLocationsBuilderX86_64::VisitMathAtan(HInvoke* invoke) {
773 CreateFPToFPCallLocations(arena_, invoke);
774}
775
776void IntrinsicCodeGeneratorX86_64::VisitMathAtan(HInvoke* invoke) {
777 GenFPToFPCall(invoke, codegen_, kQuickAtan);
778}
779
780void IntrinsicLocationsBuilderX86_64::VisitMathCbrt(HInvoke* invoke) {
781 CreateFPToFPCallLocations(arena_, invoke);
782}
783
784void IntrinsicCodeGeneratorX86_64::VisitMathCbrt(HInvoke* invoke) {
785 GenFPToFPCall(invoke, codegen_, kQuickCbrt);
786}
787
788void IntrinsicLocationsBuilderX86_64::VisitMathCosh(HInvoke* invoke) {
789 CreateFPToFPCallLocations(arena_, invoke);
790}
791
792void IntrinsicCodeGeneratorX86_64::VisitMathCosh(HInvoke* invoke) {
793 GenFPToFPCall(invoke, codegen_, kQuickCosh);
794}
795
796void IntrinsicLocationsBuilderX86_64::VisitMathExp(HInvoke* invoke) {
797 CreateFPToFPCallLocations(arena_, invoke);
798}
799
800void IntrinsicCodeGeneratorX86_64::VisitMathExp(HInvoke* invoke) {
801 GenFPToFPCall(invoke, codegen_, kQuickExp);
802}
803
804void IntrinsicLocationsBuilderX86_64::VisitMathExpm1(HInvoke* invoke) {
805 CreateFPToFPCallLocations(arena_, invoke);
806}
807
808void IntrinsicCodeGeneratorX86_64::VisitMathExpm1(HInvoke* invoke) {
809 GenFPToFPCall(invoke, codegen_, kQuickExpm1);
810}
811
812void IntrinsicLocationsBuilderX86_64::VisitMathLog(HInvoke* invoke) {
813 CreateFPToFPCallLocations(arena_, invoke);
814}
815
816void IntrinsicCodeGeneratorX86_64::VisitMathLog(HInvoke* invoke) {
817 GenFPToFPCall(invoke, codegen_, kQuickLog);
818}
819
820void IntrinsicLocationsBuilderX86_64::VisitMathLog10(HInvoke* invoke) {
821 CreateFPToFPCallLocations(arena_, invoke);
822}
823
824void IntrinsicCodeGeneratorX86_64::VisitMathLog10(HInvoke* invoke) {
825 GenFPToFPCall(invoke, codegen_, kQuickLog10);
826}
827
828void IntrinsicLocationsBuilderX86_64::VisitMathSinh(HInvoke* invoke) {
829 CreateFPToFPCallLocations(arena_, invoke);
830}
831
832void IntrinsicCodeGeneratorX86_64::VisitMathSinh(HInvoke* invoke) {
833 GenFPToFPCall(invoke, codegen_, kQuickSinh);
834}
835
836void IntrinsicLocationsBuilderX86_64::VisitMathTan(HInvoke* invoke) {
837 CreateFPToFPCallLocations(arena_, invoke);
838}
839
840void IntrinsicCodeGeneratorX86_64::VisitMathTan(HInvoke* invoke) {
841 GenFPToFPCall(invoke, codegen_, kQuickTan);
842}
843
844void IntrinsicLocationsBuilderX86_64::VisitMathTanh(HInvoke* invoke) {
845 CreateFPToFPCallLocations(arena_, invoke);
846}
847
848void IntrinsicCodeGeneratorX86_64::VisitMathTanh(HInvoke* invoke) {
849 GenFPToFPCall(invoke, codegen_, kQuickTanh);
850}
851
852static void CreateFPFPToFPCallLocations(ArenaAllocator* arena,
853 HInvoke* invoke) {
854 LocationSummary* locations = new (arena) LocationSummary(invoke,
855 LocationSummary::kCall,
856 kIntrinsified);
857 InvokeRuntimeCallingConvention calling_convention;
858 locations->SetInAt(0, Location::FpuRegisterLocation(calling_convention.GetFpuRegisterAt(0)));
859 locations->SetInAt(1, Location::FpuRegisterLocation(calling_convention.GetFpuRegisterAt(1)));
860 locations->SetOut(Location::FpuRegisterLocation(XMM0));
861
862 // We have to ensure that the native code doesn't clobber the XMM registers which are
863 // non-volatile for ART, but volatile for Native calls. This will ensure that they are
864 // saved in the prologue and properly restored.
865 for (auto fp_reg : non_volatile_xmm_regs) {
866 locations->AddTemp(Location::FpuRegisterLocation(fp_reg));
867 }
868}
869
870void IntrinsicLocationsBuilderX86_64::VisitMathAtan2(HInvoke* invoke) {
871 CreateFPFPToFPCallLocations(arena_, invoke);
872}
873
874void IntrinsicCodeGeneratorX86_64::VisitMathAtan2(HInvoke* invoke) {
875 GenFPToFPCall(invoke, codegen_, kQuickAtan2);
876}
877
878void IntrinsicLocationsBuilderX86_64::VisitMathHypot(HInvoke* invoke) {
879 CreateFPFPToFPCallLocations(arena_, invoke);
880}
881
882void IntrinsicCodeGeneratorX86_64::VisitMathHypot(HInvoke* invoke) {
883 GenFPToFPCall(invoke, codegen_, kQuickHypot);
884}
885
886void IntrinsicLocationsBuilderX86_64::VisitMathNextAfter(HInvoke* invoke) {
887 CreateFPFPToFPCallLocations(arena_, invoke);
888}
889
890void IntrinsicCodeGeneratorX86_64::VisitMathNextAfter(HInvoke* invoke) {
891 GenFPToFPCall(invoke, codegen_, kQuickNextAfter);
892}
893
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800894void IntrinsicLocationsBuilderX86_64::VisitStringCharAt(HInvoke* invoke) {
895 // The inputs plus one temp.
896 LocationSummary* locations = new (arena_) LocationSummary(invoke,
897 LocationSummary::kCallOnSlowPath,
898 kIntrinsified);
899 locations->SetInAt(0, Location::RequiresRegister());
900 locations->SetInAt(1, Location::RequiresRegister());
901 locations->SetOut(Location::SameAsFirstInput());
902 locations->AddTemp(Location::RequiresRegister());
903}
904
905void IntrinsicCodeGeneratorX86_64::VisitStringCharAt(HInvoke* invoke) {
906 LocationSummary* locations = invoke->GetLocations();
907
Mark Mendell6bc53a92015-07-01 14:26:52 -0400908 // Location of reference to data array.
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800909 const int32_t value_offset = mirror::String::ValueOffset().Int32Value();
Mark Mendell6bc53a92015-07-01 14:26:52 -0400910 // Location of count.
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800911 const int32_t count_offset = mirror::String::CountOffset().Int32Value();
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800912
913 CpuRegister obj = locations->InAt(0).AsRegister<CpuRegister>();
914 CpuRegister idx = locations->InAt(1).AsRegister<CpuRegister>();
915 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800916
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800917 // TODO: Maybe we can support range check elimination. Overall, though, I think it's not worth
918 // the cost.
919 // TODO: For simplicity, the index parameter is requested in a register, so different from Quick
920 // we will not optimize the code for constants (which would save a register).
921
Andreas Gampe85b62f22015-09-09 13:15:38 -0700922 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800923 codegen_->AddSlowPath(slow_path);
924
925 X86_64Assembler* assembler = GetAssembler();
926
927 __ cmpl(idx, Address(obj, count_offset));
Andreas Gampe878d58c2015-01-15 23:24:00 -0800928 codegen_->MaybeRecordImplicitNullCheck(invoke);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800929 __ j(kAboveEqual, slow_path->GetEntryLabel());
930
Jeff Hao848f70a2014-01-15 13:49:50 -0800931 // out = out[2*idx].
932 __ movzxw(out, Address(out, idx, ScaleFactor::TIMES_2, value_offset));
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800933
934 __ Bind(slow_path->GetExitLabel());
935}
936
Mark Mendell6bc53a92015-07-01 14:26:52 -0400937void IntrinsicLocationsBuilderX86_64::VisitSystemArrayCopyChar(HInvoke* invoke) {
938 // Check to see if we have known failures that will cause us to have to bail out
939 // to the runtime, and just generate the runtime call directly.
940 HIntConstant* src_pos = invoke->InputAt(1)->AsIntConstant();
941 HIntConstant* dest_pos = invoke->InputAt(3)->AsIntConstant();
942
943 // The positions must be non-negative.
944 if ((src_pos != nullptr && src_pos->GetValue() < 0) ||
945 (dest_pos != nullptr && dest_pos->GetValue() < 0)) {
946 // We will have to fail anyways.
947 return;
948 }
949
950 // The length must be > 0.
951 HIntConstant* length = invoke->InputAt(4)->AsIntConstant();
952 if (length != nullptr) {
953 int32_t len = length->GetValue();
954 if (len < 0) {
955 // Just call as normal.
956 return;
957 }
958 }
959
960 LocationSummary* locations = new (arena_) LocationSummary(invoke,
961 LocationSummary::kCallOnSlowPath,
962 kIntrinsified);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +0100963 // arraycopy(Object src, int src_pos, Object dest, int dest_pos, int length).
Mark Mendell6bc53a92015-07-01 14:26:52 -0400964 locations->SetInAt(0, Location::RequiresRegister());
965 locations->SetInAt(1, Location::RegisterOrConstant(invoke->InputAt(1)));
966 locations->SetInAt(2, Location::RequiresRegister());
967 locations->SetInAt(3, Location::RegisterOrConstant(invoke->InputAt(3)));
968 locations->SetInAt(4, Location::RegisterOrConstant(invoke->InputAt(4)));
969
970 // And we need some temporaries. We will use REP MOVSW, so we need fixed registers.
971 locations->AddTemp(Location::RegisterLocation(RSI));
972 locations->AddTemp(Location::RegisterLocation(RDI));
973 locations->AddTemp(Location::RegisterLocation(RCX));
974}
975
976static void CheckPosition(X86_64Assembler* assembler,
977 Location pos,
978 CpuRegister input,
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +0100979 Location length,
Andreas Gampe85b62f22015-09-09 13:15:38 -0700980 SlowPathCode* slow_path,
Mark Mendell6bc53a92015-07-01 14:26:52 -0400981 CpuRegister input_len,
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +0100982 CpuRegister temp,
983 bool length_is_input_length = false) {
984 // Where is the length in the Array?
Mark Mendell6bc53a92015-07-01 14:26:52 -0400985 const uint32_t length_offset = mirror::Array::LengthOffset().Uint32Value();
986
987 if (pos.IsConstant()) {
988 int32_t pos_const = pos.GetConstant()->AsIntConstant()->GetValue();
989 if (pos_const == 0) {
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +0100990 if (!length_is_input_length) {
991 // Check that length(input) >= length.
992 if (length.IsConstant()) {
993 __ cmpl(Address(input, length_offset),
994 Immediate(length.GetConstant()->AsIntConstant()->GetValue()));
995 } else {
996 __ cmpl(Address(input, length_offset), length.AsRegister<CpuRegister>());
997 }
998 __ j(kLess, slow_path->GetEntryLabel());
999 }
Mark Mendell6bc53a92015-07-01 14:26:52 -04001000 } else {
1001 // Check that length(input) >= pos.
1002 __ movl(input_len, Address(input, length_offset));
1003 __ cmpl(input_len, Immediate(pos_const));
1004 __ j(kLess, slow_path->GetEntryLabel());
1005
1006 // Check that (length(input) - pos) >= length.
1007 __ leal(temp, Address(input_len, -pos_const));
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001008 if (length.IsConstant()) {
1009 __ cmpl(temp, Immediate(length.GetConstant()->AsIntConstant()->GetValue()));
1010 } else {
1011 __ cmpl(temp, length.AsRegister<CpuRegister>());
1012 }
Mark Mendell6bc53a92015-07-01 14:26:52 -04001013 __ j(kLess, slow_path->GetEntryLabel());
1014 }
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001015 } else if (length_is_input_length) {
1016 // The only way the copy can succeed is if pos is zero.
1017 CpuRegister pos_reg = pos.AsRegister<CpuRegister>();
1018 __ testl(pos_reg, pos_reg);
1019 __ j(kNotEqual, slow_path->GetEntryLabel());
Mark Mendell6bc53a92015-07-01 14:26:52 -04001020 } else {
1021 // Check that pos >= 0.
1022 CpuRegister pos_reg = pos.AsRegister<CpuRegister>();
1023 __ testl(pos_reg, pos_reg);
1024 __ j(kLess, slow_path->GetEntryLabel());
1025
1026 // Check that pos <= length(input).
1027 __ cmpl(Address(input, length_offset), pos_reg);
1028 __ j(kLess, slow_path->GetEntryLabel());
1029
1030 // Check that (length(input) - pos) >= length.
1031 __ movl(temp, Address(input, length_offset));
1032 __ subl(temp, pos_reg);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001033 if (length.IsConstant()) {
1034 __ cmpl(temp, Immediate(length.GetConstant()->AsIntConstant()->GetValue()));
1035 } else {
1036 __ cmpl(temp, length.AsRegister<CpuRegister>());
1037 }
Mark Mendell6bc53a92015-07-01 14:26:52 -04001038 __ j(kLess, slow_path->GetEntryLabel());
1039 }
1040}
1041
1042void IntrinsicCodeGeneratorX86_64::VisitSystemArrayCopyChar(HInvoke* invoke) {
1043 X86_64Assembler* assembler = GetAssembler();
1044 LocationSummary* locations = invoke->GetLocations();
1045
1046 CpuRegister src = locations->InAt(0).AsRegister<CpuRegister>();
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001047 Location src_pos = locations->InAt(1);
Mark Mendell6bc53a92015-07-01 14:26:52 -04001048 CpuRegister dest = locations->InAt(2).AsRegister<CpuRegister>();
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001049 Location dest_pos = locations->InAt(3);
Mark Mendell6bc53a92015-07-01 14:26:52 -04001050 Location length = locations->InAt(4);
1051
1052 // Temporaries that we need for MOVSW.
1053 CpuRegister src_base = locations->GetTemp(0).AsRegister<CpuRegister>();
1054 DCHECK_EQ(src_base.AsRegister(), RSI);
1055 CpuRegister dest_base = locations->GetTemp(1).AsRegister<CpuRegister>();
1056 DCHECK_EQ(dest_base.AsRegister(), RDI);
1057 CpuRegister count = locations->GetTemp(2).AsRegister<CpuRegister>();
1058 DCHECK_EQ(count.AsRegister(), RCX);
1059
Andreas Gampe85b62f22015-09-09 13:15:38 -07001060 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Mark Mendell6bc53a92015-07-01 14:26:52 -04001061 codegen_->AddSlowPath(slow_path);
1062
1063 // Bail out if the source and destination are the same.
1064 __ cmpl(src, dest);
1065 __ j(kEqual, slow_path->GetEntryLabel());
1066
1067 // Bail out if the source is null.
1068 __ testl(src, src);
1069 __ j(kEqual, slow_path->GetEntryLabel());
1070
1071 // Bail out if the destination is null.
1072 __ testl(dest, dest);
1073 __ j(kEqual, slow_path->GetEntryLabel());
1074
1075 // If the length is negative, bail out.
1076 // We have already checked in the LocationsBuilder for the constant case.
1077 if (!length.IsConstant()) {
1078 __ testl(length.AsRegister<CpuRegister>(), length.AsRegister<CpuRegister>());
1079 __ j(kLess, slow_path->GetEntryLabel());
1080 }
1081
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001082 // Validity checks: source.
1083 CheckPosition(assembler, src_pos, src, length, slow_path, src_base, dest_base);
1084
1085 // Validity checks: dest.
1086 CheckPosition(assembler, dest_pos, dest, length, slow_path, src_base, dest_base);
1087
Mark Mendell6bc53a92015-07-01 14:26:52 -04001088 // We need the count in RCX.
1089 if (length.IsConstant()) {
1090 __ movl(count, Immediate(length.GetConstant()->AsIntConstant()->GetValue()));
1091 } else {
1092 __ movl(count, length.AsRegister<CpuRegister>());
1093 }
1094
Mark Mendell6bc53a92015-07-01 14:26:52 -04001095 // Okay, everything checks out. Finally time to do the copy.
1096 // Check assumption that sizeof(Char) is 2 (used in scaling below).
1097 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1098 DCHECK_EQ(char_size, 2u);
1099
1100 const uint32_t data_offset = mirror::Array::DataOffset(char_size).Uint32Value();
1101
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001102 if (src_pos.IsConstant()) {
1103 int32_t src_pos_const = src_pos.GetConstant()->AsIntConstant()->GetValue();
1104 __ leal(src_base, Address(src, char_size * src_pos_const + data_offset));
Mark Mendell6bc53a92015-07-01 14:26:52 -04001105 } else {
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001106 __ leal(src_base, Address(src, src_pos.AsRegister<CpuRegister>(),
Mark Mendell6bc53a92015-07-01 14:26:52 -04001107 ScaleFactor::TIMES_2, data_offset));
1108 }
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001109 if (dest_pos.IsConstant()) {
1110 int32_t dest_pos_const = dest_pos.GetConstant()->AsIntConstant()->GetValue();
1111 __ leal(dest_base, Address(dest, char_size * dest_pos_const + data_offset));
Mark Mendell6bc53a92015-07-01 14:26:52 -04001112 } else {
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001113 __ leal(dest_base, Address(dest, dest_pos.AsRegister<CpuRegister>(),
Mark Mendell6bc53a92015-07-01 14:26:52 -04001114 ScaleFactor::TIMES_2, data_offset));
1115 }
1116
1117 // Do the move.
1118 __ rep_movsw();
1119
1120 __ Bind(slow_path->GetExitLabel());
1121}
1122
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001123
1124void IntrinsicLocationsBuilderX86_64::VisitSystemArrayCopy(HInvoke* invoke) {
Nicolas Geoffray5bd05a52015-10-13 09:48:30 +01001125 CodeGenerator::CreateSystemArrayCopyLocationSummary(invoke);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001126}
1127
Roland Levillain0d5a2812015-11-13 10:07:31 +00001128// TODO: Implement read barriers in the SystemArrayCopy intrinsic.
1129// Note that this code path is not used (yet) because we do not
1130// intrinsify methods that can go into the IntrinsicSlowPathX86_64
1131// slow path.
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001132void IntrinsicCodeGeneratorX86_64::VisitSystemArrayCopy(HInvoke* invoke) {
1133 X86_64Assembler* assembler = GetAssembler();
1134 LocationSummary* locations = invoke->GetLocations();
1135
1136 uint32_t class_offset = mirror::Object::ClassOffset().Int32Value();
1137 uint32_t super_offset = mirror::Class::SuperClassOffset().Int32Value();
1138 uint32_t component_offset = mirror::Class::ComponentTypeOffset().Int32Value();
1139 uint32_t primitive_offset = mirror::Class::PrimitiveTypeOffset().Int32Value();
1140
1141 CpuRegister src = locations->InAt(0).AsRegister<CpuRegister>();
1142 Location src_pos = locations->InAt(1);
1143 CpuRegister dest = locations->InAt(2).AsRegister<CpuRegister>();
1144 Location dest_pos = locations->InAt(3);
1145 Location length = locations->InAt(4);
1146 CpuRegister temp1 = locations->GetTemp(0).AsRegister<CpuRegister>();
1147 CpuRegister temp2 = locations->GetTemp(1).AsRegister<CpuRegister>();
1148 CpuRegister temp3 = locations->GetTemp(2).AsRegister<CpuRegister>();
1149
1150 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
1151 codegen_->AddSlowPath(slow_path);
1152
Roland Levillainebea3d22016-04-12 15:42:57 +01001153 NearLabel conditions_on_positions_validated;
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001154 SystemArrayCopyOptimizations optimizations(invoke);
1155
Roland Levillainebea3d22016-04-12 15:42:57 +01001156 if (!optimizations.GetDestinationIsSource() &&
1157 (!src_pos.IsConstant() || !dest_pos.IsConstant())) {
1158 __ cmpl(src, dest);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001159 }
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001160 // If source and destination are the same, we go to slow path if we need to do
1161 // forward copying.
1162 if (src_pos.IsConstant()) {
1163 int32_t src_pos_constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
1164 if (dest_pos.IsConstant()) {
1165 // Checked when building locations.
1166 DCHECK(!optimizations.GetDestinationIsSource()
1167 || (src_pos_constant >= dest_pos.GetConstant()->AsIntConstant()->GetValue()));
1168 } else {
1169 if (!optimizations.GetDestinationIsSource()) {
Roland Levillainebea3d22016-04-12 15:42:57 +01001170 __ j(kNotEqual, &conditions_on_positions_validated);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001171 }
1172 __ cmpl(dest_pos.AsRegister<CpuRegister>(), Immediate(src_pos_constant));
1173 __ j(kGreater, slow_path->GetEntryLabel());
1174 }
1175 } else {
1176 if (!optimizations.GetDestinationIsSource()) {
Roland Levillainebea3d22016-04-12 15:42:57 +01001177 __ j(kNotEqual, &conditions_on_positions_validated);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001178 }
1179 if (dest_pos.IsConstant()) {
1180 int32_t dest_pos_constant = dest_pos.GetConstant()->AsIntConstant()->GetValue();
1181 __ cmpl(src_pos.AsRegister<CpuRegister>(), Immediate(dest_pos_constant));
1182 __ j(kLess, slow_path->GetEntryLabel());
1183 } else {
1184 __ cmpl(src_pos.AsRegister<CpuRegister>(), dest_pos.AsRegister<CpuRegister>());
1185 __ j(kLess, slow_path->GetEntryLabel());
1186 }
1187 }
1188
Roland Levillainebea3d22016-04-12 15:42:57 +01001189 __ Bind(&conditions_on_positions_validated);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001190
1191 if (!optimizations.GetSourceIsNotNull()) {
1192 // Bail out if the source is null.
1193 __ testl(src, src);
1194 __ j(kEqual, slow_path->GetEntryLabel());
1195 }
1196
1197 if (!optimizations.GetDestinationIsNotNull() && !optimizations.GetDestinationIsSource()) {
1198 // Bail out if the destination is null.
1199 __ testl(dest, dest);
1200 __ j(kEqual, slow_path->GetEntryLabel());
1201 }
1202
1203 // If the length is negative, bail out.
1204 // We have already checked in the LocationsBuilder for the constant case.
1205 if (!length.IsConstant() &&
1206 !optimizations.GetCountIsSourceLength() &&
1207 !optimizations.GetCountIsDestinationLength()) {
1208 __ testl(length.AsRegister<CpuRegister>(), length.AsRegister<CpuRegister>());
1209 __ j(kLess, slow_path->GetEntryLabel());
1210 }
1211
1212 // Validity checks: source.
1213 CheckPosition(assembler,
1214 src_pos,
1215 src,
1216 length,
1217 slow_path,
1218 temp1,
1219 temp2,
1220 optimizations.GetCountIsSourceLength());
1221
1222 // Validity checks: dest.
1223 CheckPosition(assembler,
1224 dest_pos,
1225 dest,
1226 length,
1227 slow_path,
1228 temp1,
1229 temp2,
1230 optimizations.GetCountIsDestinationLength());
1231
1232 if (!optimizations.GetDoesNotNeedTypeCheck()) {
1233 // Check whether all elements of the source array are assignable to the component
1234 // type of the destination array. We do two checks: the classes are the same,
1235 // or the destination is Object[]. If none of these checks succeed, we go to the
1236 // slow path.
1237 __ movl(temp1, Address(dest, class_offset));
1238 __ movl(temp2, Address(src, class_offset));
1239 bool did_unpoison = false;
1240 if (!optimizations.GetDestinationIsNonPrimitiveArray() ||
1241 !optimizations.GetSourceIsNonPrimitiveArray()) {
Roland Levillainebea3d22016-04-12 15:42:57 +01001242 // One or two of the references need to be unpoisoned. Unpoison them
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001243 // both to make the identity check valid.
1244 __ MaybeUnpoisonHeapReference(temp1);
1245 __ MaybeUnpoisonHeapReference(temp2);
1246 did_unpoison = true;
1247 }
1248
1249 if (!optimizations.GetDestinationIsNonPrimitiveArray()) {
1250 // Bail out if the destination is not a non primitive array.
Roland Levillainebea3d22016-04-12 15:42:57 +01001251 // /* HeapReference<Class> */ TMP = temp1->component_type_
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001252 __ movl(CpuRegister(TMP), Address(temp1, component_offset));
1253 __ testl(CpuRegister(TMP), CpuRegister(TMP));
1254 __ j(kEqual, slow_path->GetEntryLabel());
1255 __ MaybeUnpoisonHeapReference(CpuRegister(TMP));
1256 __ cmpw(Address(CpuRegister(TMP), primitive_offset), Immediate(Primitive::kPrimNot));
1257 __ j(kNotEqual, slow_path->GetEntryLabel());
1258 }
1259
1260 if (!optimizations.GetSourceIsNonPrimitiveArray()) {
1261 // Bail out if the source is not a non primitive array.
Roland Levillainebea3d22016-04-12 15:42:57 +01001262 // /* HeapReference<Class> */ TMP = temp2->component_type_
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001263 __ movl(CpuRegister(TMP), Address(temp2, component_offset));
1264 __ testl(CpuRegister(TMP), CpuRegister(TMP));
1265 __ j(kEqual, slow_path->GetEntryLabel());
1266 __ MaybeUnpoisonHeapReference(CpuRegister(TMP));
1267 __ cmpw(Address(CpuRegister(TMP), primitive_offset), Immediate(Primitive::kPrimNot));
1268 __ j(kNotEqual, slow_path->GetEntryLabel());
1269 }
1270
1271 __ cmpl(temp1, temp2);
1272
1273 if (optimizations.GetDestinationIsTypedObjectArray()) {
1274 NearLabel do_copy;
1275 __ j(kEqual, &do_copy);
1276 if (!did_unpoison) {
1277 __ MaybeUnpoisonHeapReference(temp1);
1278 }
Roland Levillainebea3d22016-04-12 15:42:57 +01001279 // /* HeapReference<Class> */ temp1 = temp1->component_type_
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001280 __ movl(temp1, Address(temp1, component_offset));
1281 __ MaybeUnpoisonHeapReference(temp1);
Roland Levillainebea3d22016-04-12 15:42:57 +01001282 // /* HeapReference<Class> */ temp1 = temp1->super_class_
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001283 __ movl(temp1, Address(temp1, super_offset));
1284 // No need to unpoison the result, we're comparing against null.
1285 __ testl(temp1, temp1);
1286 __ j(kNotEqual, slow_path->GetEntryLabel());
1287 __ Bind(&do_copy);
1288 } else {
1289 __ j(kNotEqual, slow_path->GetEntryLabel());
1290 }
1291 } else if (!optimizations.GetSourceIsNonPrimitiveArray()) {
1292 DCHECK(optimizations.GetDestinationIsNonPrimitiveArray());
1293 // Bail out if the source is not a non primitive array.
Roland Levillainebea3d22016-04-12 15:42:57 +01001294 // /* HeapReference<Class> */ temp1 = src->klass_
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001295 __ movl(temp1, Address(src, class_offset));
1296 __ MaybeUnpoisonHeapReference(temp1);
Roland Levillainebea3d22016-04-12 15:42:57 +01001297 // /* HeapReference<Class> */ TMP = temp1->component_type_
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001298 __ movl(CpuRegister(TMP), Address(temp1, component_offset));
1299 __ testl(CpuRegister(TMP), CpuRegister(TMP));
1300 __ j(kEqual, slow_path->GetEntryLabel());
1301 __ MaybeUnpoisonHeapReference(CpuRegister(TMP));
1302 __ cmpw(Address(CpuRegister(TMP), primitive_offset), Immediate(Primitive::kPrimNot));
1303 __ j(kNotEqual, slow_path->GetEntryLabel());
1304 }
1305
1306 // Compute base source address, base destination address, and end source address.
1307
1308 uint32_t element_size = sizeof(int32_t);
1309 uint32_t offset = mirror::Array::DataOffset(element_size).Uint32Value();
1310 if (src_pos.IsConstant()) {
1311 int32_t constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
1312 __ leal(temp1, Address(src, element_size * constant + offset));
1313 } else {
1314 __ leal(temp1, Address(src, src_pos.AsRegister<CpuRegister>(), ScaleFactor::TIMES_4, offset));
1315 }
1316
1317 if (dest_pos.IsConstant()) {
1318 int32_t constant = dest_pos.GetConstant()->AsIntConstant()->GetValue();
1319 __ leal(temp2, Address(dest, element_size * constant + offset));
1320 } else {
1321 __ leal(temp2, Address(dest, dest_pos.AsRegister<CpuRegister>(), ScaleFactor::TIMES_4, offset));
1322 }
1323
1324 if (length.IsConstant()) {
1325 int32_t constant = length.GetConstant()->AsIntConstant()->GetValue();
1326 __ leal(temp3, Address(temp1, element_size * constant));
1327 } else {
1328 __ leal(temp3, Address(temp1, length.AsRegister<CpuRegister>(), ScaleFactor::TIMES_4, 0));
1329 }
1330
1331 // Iterate over the arrays and do a raw copy of the objects. We don't need to
1332 // poison/unpoison, nor do any read barrier as the next uses of the destination
1333 // array will do it.
1334 NearLabel loop, done;
1335 __ cmpl(temp1, temp3);
1336 __ j(kEqual, &done);
1337 __ Bind(&loop);
1338 __ movl(CpuRegister(TMP), Address(temp1, 0));
1339 __ movl(Address(temp2, 0), CpuRegister(TMP));
1340 __ addl(temp1, Immediate(element_size));
1341 __ addl(temp2, Immediate(element_size));
1342 __ cmpl(temp1, temp3);
1343 __ j(kNotEqual, &loop);
1344 __ Bind(&done);
1345
1346 // We only need one card marking on the destination array.
1347 codegen_->MarkGCCard(temp1,
1348 temp2,
1349 dest,
1350 CpuRegister(kNoRegister),
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001351 /* value_can_be_null */ false);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001352
1353 __ Bind(slow_path->GetExitLabel());
1354}
1355
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001356void IntrinsicLocationsBuilderX86_64::VisitStringCompareTo(HInvoke* invoke) {
1357 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1358 LocationSummary::kCall,
1359 kIntrinsified);
1360 InvokeRuntimeCallingConvention calling_convention;
1361 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetRegisterAt(0)));
1362 locations->SetInAt(1, Location::RegisterLocation(calling_convention.GetRegisterAt(1)));
1363 locations->SetOut(Location::RegisterLocation(RAX));
1364}
1365
1366void IntrinsicCodeGeneratorX86_64::VisitStringCompareTo(HInvoke* invoke) {
1367 X86_64Assembler* assembler = GetAssembler();
1368 LocationSummary* locations = invoke->GetLocations();
1369
Nicolas Geoffray512e04d2015-03-27 17:21:24 +00001370 // Note that the null check must have been done earlier.
Calin Juravle641547a2015-04-21 22:08:51 +01001371 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001372
1373 CpuRegister argument = locations->InAt(1).AsRegister<CpuRegister>();
1374 __ testl(argument, argument);
Andreas Gampe85b62f22015-09-09 13:15:38 -07001375 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001376 codegen_->AddSlowPath(slow_path);
1377 __ j(kEqual, slow_path->GetEntryLabel());
1378
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001379 __ gs()->call(Address::Absolute(QUICK_ENTRYPOINT_OFFSET(kX86_64WordSize, pStringCompareTo),
1380 /* no_rip */ true));
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001381 __ Bind(slow_path->GetExitLabel());
1382}
1383
Agi Csakif8cfb202015-08-13 17:54:54 -07001384void IntrinsicLocationsBuilderX86_64::VisitStringEquals(HInvoke* invoke) {
1385 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1386 LocationSummary::kNoCall,
1387 kIntrinsified);
1388 locations->SetInAt(0, Location::RequiresRegister());
1389 locations->SetInAt(1, Location::RequiresRegister());
1390
1391 // Request temporary registers, RCX and RDI needed for repe_cmpsq instruction.
1392 locations->AddTemp(Location::RegisterLocation(RCX));
1393 locations->AddTemp(Location::RegisterLocation(RDI));
1394
1395 // Set output, RSI needed for repe_cmpsq instruction anyways.
1396 locations->SetOut(Location::RegisterLocation(RSI), Location::kOutputOverlap);
1397}
1398
1399void IntrinsicCodeGeneratorX86_64::VisitStringEquals(HInvoke* invoke) {
1400 X86_64Assembler* assembler = GetAssembler();
1401 LocationSummary* locations = invoke->GetLocations();
1402
1403 CpuRegister str = locations->InAt(0).AsRegister<CpuRegister>();
1404 CpuRegister arg = locations->InAt(1).AsRegister<CpuRegister>();
1405 CpuRegister rcx = locations->GetTemp(0).AsRegister<CpuRegister>();
1406 CpuRegister rdi = locations->GetTemp(1).AsRegister<CpuRegister>();
1407 CpuRegister rsi = locations->Out().AsRegister<CpuRegister>();
1408
Mark Mendell0c9497d2015-08-21 09:30:05 -04001409 NearLabel end, return_true, return_false;
Agi Csakif8cfb202015-08-13 17:54:54 -07001410
1411 // Get offsets of count, value, and class fields within a string object.
1412 const uint32_t count_offset = mirror::String::CountOffset().Uint32Value();
1413 const uint32_t value_offset = mirror::String::ValueOffset().Uint32Value();
1414 const uint32_t class_offset = mirror::Object::ClassOffset().Uint32Value();
1415
1416 // Note that the null check must have been done earlier.
1417 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1418
Vladimir Marko53b52002016-05-24 19:30:45 +01001419 StringEqualsOptimizations optimizations(invoke);
1420 if (!optimizations.GetArgumentNotNull()) {
1421 // Check if input is null, return false if it is.
1422 __ testl(arg, arg);
1423 __ j(kEqual, &return_false);
1424 }
Agi Csakif8cfb202015-08-13 17:54:54 -07001425
Vladimir Marko53b52002016-05-24 19:30:45 +01001426 if (!optimizations.GetArgumentIsString()) {
1427 // Instanceof check for the argument by comparing class fields.
1428 // All string objects must have the same type since String cannot be subclassed.
1429 // Receiver must be a string object, so its class field is equal to all strings' class fields.
1430 // If the argument is a string object, its class field must be equal to receiver's class field.
1431 __ movl(rcx, Address(str, class_offset));
1432 __ cmpl(rcx, Address(arg, class_offset));
1433 __ j(kNotEqual, &return_false);
1434 }
Agi Csakif8cfb202015-08-13 17:54:54 -07001435
1436 // Reference equality check, return true if same reference.
1437 __ cmpl(str, arg);
1438 __ j(kEqual, &return_true);
1439
1440 // Load length of receiver string.
1441 __ movl(rcx, Address(str, count_offset));
1442 // Check if lengths are equal, return false if they're not.
1443 __ cmpl(rcx, Address(arg, count_offset));
1444 __ j(kNotEqual, &return_false);
1445 // Return true if both strings are empty.
Mark Mendell0c9497d2015-08-21 09:30:05 -04001446 __ jrcxz(&return_true);
Agi Csakif8cfb202015-08-13 17:54:54 -07001447
1448 // Load starting addresses of string values into RSI/RDI as required for repe_cmpsq instruction.
1449 __ leal(rsi, Address(str, value_offset));
1450 __ leal(rdi, Address(arg, value_offset));
1451
1452 // Divide string length by 4 and adjust for lengths not divisible by 4.
1453 __ addl(rcx, Immediate(3));
1454 __ shrl(rcx, Immediate(2));
1455
1456 // Assertions that must hold in order to compare strings 4 characters at a time.
1457 DCHECK_ALIGNED(value_offset, 8);
1458 static_assert(IsAligned<8>(kObjectAlignment), "String is not zero padded");
1459
1460 // Loop to compare strings four characters at a time starting at the beginning of the string.
1461 __ repe_cmpsq();
1462 // If strings are not equal, zero flag will be cleared.
1463 __ j(kNotEqual, &return_false);
1464
1465 // Return true and exit the function.
1466 // If loop does not result in returning false, we return true.
1467 __ Bind(&return_true);
1468 __ movl(rsi, Immediate(1));
1469 __ jmp(&end);
1470
1471 // Return false and exit the function.
1472 __ Bind(&return_false);
1473 __ xorl(rsi, rsi);
1474 __ Bind(&end);
1475}
1476
Andreas Gampe21030dd2015-05-07 14:46:15 -07001477static void CreateStringIndexOfLocations(HInvoke* invoke,
1478 ArenaAllocator* allocator,
1479 bool start_at_zero) {
1480 LocationSummary* locations = new (allocator) LocationSummary(invoke,
1481 LocationSummary::kCallOnSlowPath,
1482 kIntrinsified);
1483 // The data needs to be in RDI for scasw. So request that the string is there, anyways.
1484 locations->SetInAt(0, Location::RegisterLocation(RDI));
1485 // If we look for a constant char, we'll still have to copy it into RAX. So just request the
1486 // allocator to do that, anyways. We can still do the constant check by checking the parameter
1487 // of the instruction explicitly.
1488 // Note: This works as we don't clobber RAX anywhere.
1489 locations->SetInAt(1, Location::RegisterLocation(RAX));
1490 if (!start_at_zero) {
1491 locations->SetInAt(2, Location::RequiresRegister()); // The starting index.
1492 }
1493 // As we clobber RDI during execution anyways, also use it as the output.
1494 locations->SetOut(Location::SameAsFirstInput());
1495
1496 // repne scasw uses RCX as the counter.
1497 locations->AddTemp(Location::RegisterLocation(RCX));
1498 // Need another temporary to be able to compute the result.
1499 locations->AddTemp(Location::RequiresRegister());
1500}
1501
1502static void GenerateStringIndexOf(HInvoke* invoke,
1503 X86_64Assembler* assembler,
1504 CodeGeneratorX86_64* codegen,
1505 ArenaAllocator* allocator,
1506 bool start_at_zero) {
1507 LocationSummary* locations = invoke->GetLocations();
1508
1509 // Note that the null check must have been done earlier.
1510 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1511
1512 CpuRegister string_obj = locations->InAt(0).AsRegister<CpuRegister>();
1513 CpuRegister search_value = locations->InAt(1).AsRegister<CpuRegister>();
1514 CpuRegister counter = locations->GetTemp(0).AsRegister<CpuRegister>();
1515 CpuRegister string_length = locations->GetTemp(1).AsRegister<CpuRegister>();
1516 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
1517
1518 // Check our assumptions for registers.
1519 DCHECK_EQ(string_obj.AsRegister(), RDI);
1520 DCHECK_EQ(search_value.AsRegister(), RAX);
1521 DCHECK_EQ(counter.AsRegister(), RCX);
1522 DCHECK_EQ(out.AsRegister(), RDI);
1523
1524 // Check for code points > 0xFFFF. Either a slow-path check when we don't know statically,
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001525 // or directly dispatch for a large constant, or omit slow-path for a small constant or a char.
Andreas Gampe85b62f22015-09-09 13:15:38 -07001526 SlowPathCode* slow_path = nullptr;
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001527 HInstruction* code_point = invoke->InputAt(1);
1528 if (code_point->IsIntConstant()) {
Vladimir Markoda051082016-05-17 16:10:20 +01001529 if (static_cast<uint32_t>(code_point->AsIntConstant()->GetValue()) >
Andreas Gampe21030dd2015-05-07 14:46:15 -07001530 std::numeric_limits<uint16_t>::max()) {
1531 // Always needs the slow-path. We could directly dispatch to it, but this case should be
1532 // rare, so for simplicity just put the full slow-path down and branch unconditionally.
1533 slow_path = new (allocator) IntrinsicSlowPathX86_64(invoke);
1534 codegen->AddSlowPath(slow_path);
1535 __ jmp(slow_path->GetEntryLabel());
1536 __ Bind(slow_path->GetExitLabel());
1537 return;
1538 }
Vladimir Markofb6c90a2016-05-06 15:52:12 +01001539 } else if (code_point->GetType() != Primitive::kPrimChar) {
Andreas Gampe21030dd2015-05-07 14:46:15 -07001540 __ cmpl(search_value, Immediate(std::numeric_limits<uint16_t>::max()));
1541 slow_path = new (allocator) IntrinsicSlowPathX86_64(invoke);
1542 codegen->AddSlowPath(slow_path);
1543 __ j(kAbove, slow_path->GetEntryLabel());
1544 }
1545
1546 // From here down, we know that we are looking for a char that fits in 16 bits.
1547 // Location of reference to data array within the String object.
1548 int32_t value_offset = mirror::String::ValueOffset().Int32Value();
1549 // Location of count within the String object.
1550 int32_t count_offset = mirror::String::CountOffset().Int32Value();
1551
1552 // Load string length, i.e., the count field of the string.
1553 __ movl(string_length, Address(string_obj, count_offset));
1554
1555 // Do a length check.
1556 // TODO: Support jecxz.
Mark Mendell0c9497d2015-08-21 09:30:05 -04001557 NearLabel not_found_label;
Andreas Gampe21030dd2015-05-07 14:46:15 -07001558 __ testl(string_length, string_length);
1559 __ j(kEqual, &not_found_label);
1560
1561 if (start_at_zero) {
1562 // Number of chars to scan is the same as the string length.
1563 __ movl(counter, string_length);
1564
1565 // Move to the start of the string.
1566 __ addq(string_obj, Immediate(value_offset));
1567 } else {
1568 CpuRegister start_index = locations->InAt(2).AsRegister<CpuRegister>();
1569
1570 // Do a start_index check.
1571 __ cmpl(start_index, string_length);
1572 __ j(kGreaterEqual, &not_found_label);
1573
1574 // Ensure we have a start index >= 0;
1575 __ xorl(counter, counter);
1576 __ cmpl(start_index, Immediate(0));
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001577 __ cmov(kGreater, counter, start_index, /* is64bit */ false); // 32-bit copy is enough.
Andreas Gampe21030dd2015-05-07 14:46:15 -07001578
1579 // Move to the start of the string: string_obj + value_offset + 2 * start_index.
1580 __ leaq(string_obj, Address(string_obj, counter, ScaleFactor::TIMES_2, value_offset));
1581
1582 // Now update ecx, the work counter: it's gonna be string.length - start_index.
1583 __ negq(counter); // Needs to be 64-bit negation, as the address computation is 64-bit.
1584 __ leaq(counter, Address(string_length, counter, ScaleFactor::TIMES_1, 0));
1585 }
1586
1587 // Everything is set up for repne scasw:
1588 // * Comparison address in RDI.
1589 // * Counter in ECX.
1590 __ repne_scasw();
1591
1592 // Did we find a match?
1593 __ j(kNotEqual, &not_found_label);
1594
1595 // Yes, we matched. Compute the index of the result.
1596 __ subl(string_length, counter);
1597 __ leal(out, Address(string_length, -1));
1598
Mark Mendell0c9497d2015-08-21 09:30:05 -04001599 NearLabel done;
Andreas Gampe21030dd2015-05-07 14:46:15 -07001600 __ jmp(&done);
1601
1602 // Failed to match; return -1.
1603 __ Bind(&not_found_label);
1604 __ movl(out, Immediate(-1));
1605
1606 // And join up at the end.
1607 __ Bind(&done);
1608 if (slow_path != nullptr) {
1609 __ Bind(slow_path->GetExitLabel());
1610 }
1611}
1612
1613void IntrinsicLocationsBuilderX86_64::VisitStringIndexOf(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001614 CreateStringIndexOfLocations(invoke, arena_, /* start_at_zero */ true);
Andreas Gampe21030dd2015-05-07 14:46:15 -07001615}
1616
1617void IntrinsicCodeGeneratorX86_64::VisitStringIndexOf(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001618 GenerateStringIndexOf(invoke, GetAssembler(), codegen_, GetAllocator(), /* start_at_zero */ true);
Andreas Gampe21030dd2015-05-07 14:46:15 -07001619}
1620
1621void IntrinsicLocationsBuilderX86_64::VisitStringIndexOfAfter(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001622 CreateStringIndexOfLocations(invoke, arena_, /* start_at_zero */ false);
Andreas Gampe21030dd2015-05-07 14:46:15 -07001623}
1624
1625void IntrinsicCodeGeneratorX86_64::VisitStringIndexOfAfter(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001626 GenerateStringIndexOf(
1627 invoke, GetAssembler(), codegen_, GetAllocator(), /* start_at_zero */ false);
Andreas Gampe21030dd2015-05-07 14:46:15 -07001628}
1629
Jeff Hao848f70a2014-01-15 13:49:50 -08001630void IntrinsicLocationsBuilderX86_64::VisitStringNewStringFromBytes(HInvoke* invoke) {
1631 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1632 LocationSummary::kCall,
1633 kIntrinsified);
1634 InvokeRuntimeCallingConvention calling_convention;
1635 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetRegisterAt(0)));
1636 locations->SetInAt(1, Location::RegisterLocation(calling_convention.GetRegisterAt(1)));
1637 locations->SetInAt(2, Location::RegisterLocation(calling_convention.GetRegisterAt(2)));
1638 locations->SetInAt(3, Location::RegisterLocation(calling_convention.GetRegisterAt(3)));
1639 locations->SetOut(Location::RegisterLocation(RAX));
1640}
1641
1642void IntrinsicCodeGeneratorX86_64::VisitStringNewStringFromBytes(HInvoke* invoke) {
1643 X86_64Assembler* assembler = GetAssembler();
1644 LocationSummary* locations = invoke->GetLocations();
1645
1646 CpuRegister byte_array = locations->InAt(0).AsRegister<CpuRegister>();
1647 __ testl(byte_array, byte_array);
Andreas Gampe85b62f22015-09-09 13:15:38 -07001648 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Jeff Hao848f70a2014-01-15 13:49:50 -08001649 codegen_->AddSlowPath(slow_path);
1650 __ j(kEqual, slow_path->GetEntryLabel());
1651
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001652 __ gs()->call(Address::Absolute(QUICK_ENTRYPOINT_OFFSET(kX86_64WordSize, pAllocStringFromBytes),
1653 /* no_rip */ true));
Roland Levillainf969a202016-03-09 16:14:00 +00001654 CheckEntrypointTypes<kQuickAllocStringFromBytes, void*, void*, int32_t, int32_t, int32_t>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001655 codegen_->RecordPcInfo(invoke, invoke->GetDexPc());
1656 __ Bind(slow_path->GetExitLabel());
1657}
1658
1659void IntrinsicLocationsBuilderX86_64::VisitStringNewStringFromChars(HInvoke* invoke) {
1660 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1661 LocationSummary::kCall,
1662 kIntrinsified);
1663 InvokeRuntimeCallingConvention calling_convention;
1664 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetRegisterAt(0)));
1665 locations->SetInAt(1, Location::RegisterLocation(calling_convention.GetRegisterAt(1)));
1666 locations->SetInAt(2, Location::RegisterLocation(calling_convention.GetRegisterAt(2)));
1667 locations->SetOut(Location::RegisterLocation(RAX));
1668}
1669
1670void IntrinsicCodeGeneratorX86_64::VisitStringNewStringFromChars(HInvoke* invoke) {
1671 X86_64Assembler* assembler = GetAssembler();
1672
Roland Levillaincc3839c2016-02-29 16:23:48 +00001673 // No need to emit code checking whether `locations->InAt(2)` is a null
1674 // pointer, as callers of the native method
1675 //
1676 // java.lang.StringFactory.newStringFromChars(int offset, int charCount, char[] data)
1677 //
1678 // all include a null check on `data` before calling that method.
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001679 __ gs()->call(Address::Absolute(QUICK_ENTRYPOINT_OFFSET(kX86_64WordSize, pAllocStringFromChars),
1680 /* no_rip */ true));
Roland Levillainf969a202016-03-09 16:14:00 +00001681 CheckEntrypointTypes<kQuickAllocStringFromChars, void*, int32_t, int32_t, void*>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001682 codegen_->RecordPcInfo(invoke, invoke->GetDexPc());
1683}
1684
1685void IntrinsicLocationsBuilderX86_64::VisitStringNewStringFromString(HInvoke* invoke) {
1686 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1687 LocationSummary::kCall,
1688 kIntrinsified);
1689 InvokeRuntimeCallingConvention calling_convention;
1690 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetRegisterAt(0)));
1691 locations->SetOut(Location::RegisterLocation(RAX));
1692}
1693
1694void IntrinsicCodeGeneratorX86_64::VisitStringNewStringFromString(HInvoke* invoke) {
1695 X86_64Assembler* assembler = GetAssembler();
1696 LocationSummary* locations = invoke->GetLocations();
1697
1698 CpuRegister string_to_copy = locations->InAt(0).AsRegister<CpuRegister>();
1699 __ testl(string_to_copy, string_to_copy);
Andreas Gampe85b62f22015-09-09 13:15:38 -07001700 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Jeff Hao848f70a2014-01-15 13:49:50 -08001701 codegen_->AddSlowPath(slow_path);
1702 __ j(kEqual, slow_path->GetEntryLabel());
1703
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001704 __ gs()->call(Address::Absolute(QUICK_ENTRYPOINT_OFFSET(kX86_64WordSize, pAllocStringFromString),
1705 /* no_rip */ true));
Roland Levillainf969a202016-03-09 16:14:00 +00001706 CheckEntrypointTypes<kQuickAllocStringFromString, void*, void*>();
Jeff Hao848f70a2014-01-15 13:49:50 -08001707 codegen_->RecordPcInfo(invoke, invoke->GetDexPc());
1708 __ Bind(slow_path->GetExitLabel());
1709}
1710
Mark Mendell8f8926a2015-08-17 11:39:06 -04001711void IntrinsicLocationsBuilderX86_64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
1712 // public void getChars(int srcBegin, int srcEnd, char[] dst, int dstBegin);
1713 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1714 LocationSummary::kNoCall,
1715 kIntrinsified);
1716 locations->SetInAt(0, Location::RequiresRegister());
1717 locations->SetInAt(1, Location::RegisterOrConstant(invoke->InputAt(1)));
1718 locations->SetInAt(2, Location::RequiresRegister());
1719 locations->SetInAt(3, Location::RequiresRegister());
1720 locations->SetInAt(4, Location::RequiresRegister());
1721
1722 // And we need some temporaries. We will use REP MOVSW, so we need fixed registers.
1723 locations->AddTemp(Location::RegisterLocation(RSI));
1724 locations->AddTemp(Location::RegisterLocation(RDI));
1725 locations->AddTemp(Location::RegisterLocation(RCX));
1726}
1727
1728void IntrinsicCodeGeneratorX86_64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
1729 X86_64Assembler* assembler = GetAssembler();
1730 LocationSummary* locations = invoke->GetLocations();
1731
1732 size_t char_component_size = Primitive::ComponentSize(Primitive::kPrimChar);
1733 // Location of data in char array buffer.
1734 const uint32_t data_offset = mirror::Array::DataOffset(char_component_size).Uint32Value();
1735 // Location of char array data in string.
1736 const uint32_t value_offset = mirror::String::ValueOffset().Uint32Value();
1737
1738 // public void getChars(int srcBegin, int srcEnd, char[] dst, int dstBegin);
1739 CpuRegister obj = locations->InAt(0).AsRegister<CpuRegister>();
1740 Location srcBegin = locations->InAt(1);
1741 int srcBegin_value =
1742 srcBegin.IsConstant() ? srcBegin.GetConstant()->AsIntConstant()->GetValue() : 0;
1743 CpuRegister srcEnd = locations->InAt(2).AsRegister<CpuRegister>();
1744 CpuRegister dst = locations->InAt(3).AsRegister<CpuRegister>();
1745 CpuRegister dstBegin = locations->InAt(4).AsRegister<CpuRegister>();
1746
1747 // Check assumption that sizeof(Char) is 2 (used in scaling below).
1748 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1749 DCHECK_EQ(char_size, 2u);
1750
1751 // Compute the address of the destination buffer.
1752 __ leaq(CpuRegister(RDI), Address(dst, dstBegin, ScaleFactor::TIMES_2, data_offset));
1753
1754 // Compute the address of the source string.
1755 if (srcBegin.IsConstant()) {
1756 // Compute the address of the source string by adding the number of chars from
1757 // the source beginning to the value offset of a string.
1758 __ leaq(CpuRegister(RSI), Address(obj, srcBegin_value * char_size + value_offset));
1759 } else {
1760 __ leaq(CpuRegister(RSI), Address(obj, srcBegin.AsRegister<CpuRegister>(),
1761 ScaleFactor::TIMES_2, value_offset));
1762 }
1763
1764 // Compute the number of chars (words) to move.
1765 __ movl(CpuRegister(RCX), srcEnd);
1766 if (srcBegin.IsConstant()) {
1767 if (srcBegin_value != 0) {
1768 __ subl(CpuRegister(RCX), Immediate(srcBegin_value));
1769 }
1770 } else {
1771 DCHECK(srcBegin.IsRegister());
1772 __ subl(CpuRegister(RCX), srcBegin.AsRegister<CpuRegister>());
1773 }
1774
1775 // Do the move.
1776 __ rep_movsw();
1777}
1778
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001779static void GenPeek(LocationSummary* locations, Primitive::Type size, X86_64Assembler* assembler) {
1780 CpuRegister address = locations->InAt(0).AsRegister<CpuRegister>();
1781 CpuRegister out = locations->Out().AsRegister<CpuRegister>(); // == address, here for clarity.
1782 // x86 allows unaligned access. We do not have to check the input or use specific instructions
1783 // to avoid a SIGBUS.
1784 switch (size) {
1785 case Primitive::kPrimByte:
1786 __ movsxb(out, Address(address, 0));
1787 break;
1788 case Primitive::kPrimShort:
1789 __ movsxw(out, Address(address, 0));
1790 break;
1791 case Primitive::kPrimInt:
1792 __ movl(out, Address(address, 0));
1793 break;
1794 case Primitive::kPrimLong:
1795 __ movq(out, Address(address, 0));
1796 break;
1797 default:
1798 LOG(FATAL) << "Type not recognized for peek: " << size;
1799 UNREACHABLE();
1800 }
1801}
1802
1803void IntrinsicLocationsBuilderX86_64::VisitMemoryPeekByte(HInvoke* invoke) {
1804 CreateIntToIntLocations(arena_, invoke);
1805}
1806
1807void IntrinsicCodeGeneratorX86_64::VisitMemoryPeekByte(HInvoke* invoke) {
1808 GenPeek(invoke->GetLocations(), Primitive::kPrimByte, GetAssembler());
1809}
1810
1811void IntrinsicLocationsBuilderX86_64::VisitMemoryPeekIntNative(HInvoke* invoke) {
1812 CreateIntToIntLocations(arena_, invoke);
1813}
1814
1815void IntrinsicCodeGeneratorX86_64::VisitMemoryPeekIntNative(HInvoke* invoke) {
1816 GenPeek(invoke->GetLocations(), Primitive::kPrimInt, GetAssembler());
1817}
1818
1819void IntrinsicLocationsBuilderX86_64::VisitMemoryPeekLongNative(HInvoke* invoke) {
1820 CreateIntToIntLocations(arena_, invoke);
1821}
1822
1823void IntrinsicCodeGeneratorX86_64::VisitMemoryPeekLongNative(HInvoke* invoke) {
1824 GenPeek(invoke->GetLocations(), Primitive::kPrimLong, GetAssembler());
1825}
1826
1827void IntrinsicLocationsBuilderX86_64::VisitMemoryPeekShortNative(HInvoke* invoke) {
1828 CreateIntToIntLocations(arena_, invoke);
1829}
1830
1831void IntrinsicCodeGeneratorX86_64::VisitMemoryPeekShortNative(HInvoke* invoke) {
1832 GenPeek(invoke->GetLocations(), Primitive::kPrimShort, GetAssembler());
1833}
1834
1835static void CreateIntIntToVoidLocations(ArenaAllocator* arena, HInvoke* invoke) {
1836 LocationSummary* locations = new (arena) LocationSummary(invoke,
1837 LocationSummary::kNoCall,
1838 kIntrinsified);
1839 locations->SetInAt(0, Location::RequiresRegister());
Mark Mendellea5af682015-10-22 17:35:49 -04001840 locations->SetInAt(1, Location::RegisterOrInt32Constant(invoke->InputAt(1)));
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001841}
1842
1843static void GenPoke(LocationSummary* locations, Primitive::Type size, X86_64Assembler* assembler) {
1844 CpuRegister address = locations->InAt(0).AsRegister<CpuRegister>();
Mark Mendell40741f32015-04-20 22:10:34 -04001845 Location value = locations->InAt(1);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001846 // x86 allows unaligned access. We do not have to check the input or use specific instructions
1847 // to avoid a SIGBUS.
1848 switch (size) {
1849 case Primitive::kPrimByte:
Mark Mendell40741f32015-04-20 22:10:34 -04001850 if (value.IsConstant()) {
1851 __ movb(Address(address, 0),
1852 Immediate(CodeGenerator::GetInt32ValueOf(value.GetConstant())));
1853 } else {
1854 __ movb(Address(address, 0), value.AsRegister<CpuRegister>());
1855 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001856 break;
1857 case Primitive::kPrimShort:
Mark Mendell40741f32015-04-20 22:10:34 -04001858 if (value.IsConstant()) {
1859 __ movw(Address(address, 0),
1860 Immediate(CodeGenerator::GetInt32ValueOf(value.GetConstant())));
1861 } else {
1862 __ movw(Address(address, 0), value.AsRegister<CpuRegister>());
1863 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001864 break;
1865 case Primitive::kPrimInt:
Mark Mendell40741f32015-04-20 22:10:34 -04001866 if (value.IsConstant()) {
1867 __ movl(Address(address, 0),
1868 Immediate(CodeGenerator::GetInt32ValueOf(value.GetConstant())));
1869 } else {
1870 __ movl(Address(address, 0), value.AsRegister<CpuRegister>());
1871 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001872 break;
1873 case Primitive::kPrimLong:
Mark Mendell40741f32015-04-20 22:10:34 -04001874 if (value.IsConstant()) {
1875 int64_t v = value.GetConstant()->AsLongConstant()->GetValue();
1876 DCHECK(IsInt<32>(v));
1877 int32_t v_32 = v;
1878 __ movq(Address(address, 0), Immediate(v_32));
1879 } else {
1880 __ movq(Address(address, 0), value.AsRegister<CpuRegister>());
1881 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001882 break;
1883 default:
1884 LOG(FATAL) << "Type not recognized for poke: " << size;
1885 UNREACHABLE();
1886 }
1887}
1888
1889void IntrinsicLocationsBuilderX86_64::VisitMemoryPokeByte(HInvoke* invoke) {
1890 CreateIntIntToVoidLocations(arena_, invoke);
1891}
1892
1893void IntrinsicCodeGeneratorX86_64::VisitMemoryPokeByte(HInvoke* invoke) {
1894 GenPoke(invoke->GetLocations(), Primitive::kPrimByte, GetAssembler());
1895}
1896
1897void IntrinsicLocationsBuilderX86_64::VisitMemoryPokeIntNative(HInvoke* invoke) {
1898 CreateIntIntToVoidLocations(arena_, invoke);
1899}
1900
1901void IntrinsicCodeGeneratorX86_64::VisitMemoryPokeIntNative(HInvoke* invoke) {
1902 GenPoke(invoke->GetLocations(), Primitive::kPrimInt, GetAssembler());
1903}
1904
1905void IntrinsicLocationsBuilderX86_64::VisitMemoryPokeLongNative(HInvoke* invoke) {
1906 CreateIntIntToVoidLocations(arena_, invoke);
1907}
1908
1909void IntrinsicCodeGeneratorX86_64::VisitMemoryPokeLongNative(HInvoke* invoke) {
1910 GenPoke(invoke->GetLocations(), Primitive::kPrimLong, GetAssembler());
1911}
1912
1913void IntrinsicLocationsBuilderX86_64::VisitMemoryPokeShortNative(HInvoke* invoke) {
1914 CreateIntIntToVoidLocations(arena_, invoke);
1915}
1916
1917void IntrinsicCodeGeneratorX86_64::VisitMemoryPokeShortNative(HInvoke* invoke) {
1918 GenPoke(invoke->GetLocations(), Primitive::kPrimShort, GetAssembler());
1919}
1920
1921void IntrinsicLocationsBuilderX86_64::VisitThreadCurrentThread(HInvoke* invoke) {
1922 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1923 LocationSummary::kNoCall,
1924 kIntrinsified);
1925 locations->SetOut(Location::RequiresRegister());
1926}
1927
1928void IntrinsicCodeGeneratorX86_64::VisitThreadCurrentThread(HInvoke* invoke) {
1929 CpuRegister out = invoke->GetLocations()->Out().AsRegister<CpuRegister>();
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001930 GetAssembler()->gs()->movl(out, Address::Absolute(Thread::PeerOffset<kX86_64WordSize>(),
1931 /* no_rip */ true));
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001932}
1933
Roland Levillain0d5a2812015-11-13 10:07:31 +00001934static void GenUnsafeGet(HInvoke* invoke,
1935 Primitive::Type type,
1936 bool is_volatile ATTRIBUTE_UNUSED,
1937 CodeGeneratorX86_64* codegen) {
1938 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen->GetAssembler());
1939 LocationSummary* locations = invoke->GetLocations();
1940 Location base_loc = locations->InAt(1);
1941 CpuRegister base = base_loc.AsRegister<CpuRegister>();
1942 Location offset_loc = locations->InAt(2);
1943 CpuRegister offset = offset_loc.AsRegister<CpuRegister>();
1944 Location output_loc = locations->Out();
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001945 CpuRegister output = output_loc.AsRegister<CpuRegister>();
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001946
Andreas Gampe878d58c2015-01-15 23:24:00 -08001947 switch (type) {
1948 case Primitive::kPrimInt:
Roland Levillain0d5a2812015-11-13 10:07:31 +00001949 __ movl(output, Address(base, offset, ScaleFactor::TIMES_1, 0));
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001950 break;
1951
1952 case Primitive::kPrimNot: {
1953 if (kEmitCompilerReadBarrier) {
1954 if (kUseBakerReadBarrier) {
1955 Location temp = locations->GetTemp(0);
Sang, Chunlei0fcd2b82016-04-05 17:12:59 +08001956 Address src(base, offset, ScaleFactor::TIMES_1, 0);
1957 codegen->GenerateReferenceLoadWithBakerReadBarrier(
1958 invoke, output_loc, base, src, temp, /* needs_null_check */ false);
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001959 } else {
1960 __ movl(output, Address(base, offset, ScaleFactor::TIMES_1, 0));
1961 codegen->GenerateReadBarrierSlow(
1962 invoke, output_loc, output_loc, base_loc, 0U, offset_loc);
1963 }
1964 } else {
1965 __ movl(output, Address(base, offset, ScaleFactor::TIMES_1, 0));
1966 __ MaybeUnpoisonHeapReference(output);
Roland Levillain4d027112015-07-01 15:41:14 +01001967 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001968 break;
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001969 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001970
1971 case Primitive::kPrimLong:
Roland Levillain0d5a2812015-11-13 10:07:31 +00001972 __ movq(output, Address(base, offset, ScaleFactor::TIMES_1, 0));
Andreas Gampe878d58c2015-01-15 23:24:00 -08001973 break;
1974
1975 default:
1976 LOG(FATAL) << "Unsupported op size " << type;
1977 UNREACHABLE();
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001978 }
1979}
1980
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001981static void CreateIntIntIntToIntLocations(ArenaAllocator* arena,
1982 HInvoke* invoke,
1983 Primitive::Type type) {
Roland Levillain0d5a2812015-11-13 10:07:31 +00001984 bool can_call = kEmitCompilerReadBarrier &&
1985 (invoke->GetIntrinsic() == Intrinsics::kUnsafeGetObject ||
1986 invoke->GetIntrinsic() == Intrinsics::kUnsafeGetObjectVolatile);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001987 LocationSummary* locations = new (arena) LocationSummary(invoke,
Roland Levillain0d5a2812015-11-13 10:07:31 +00001988 can_call ?
1989 LocationSummary::kCallOnSlowPath :
1990 LocationSummary::kNoCall,
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001991 kIntrinsified);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001992 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001993 locations->SetInAt(1, Location::RequiresRegister());
1994 locations->SetInAt(2, Location::RequiresRegister());
Andreas Gampe878d58c2015-01-15 23:24:00 -08001995 locations->SetOut(Location::RequiresRegister());
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001996 if (type == Primitive::kPrimNot && kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
1997 // We need a temporary register for the read barrier marking slow
Sang, Chunlei0fcd2b82016-04-05 17:12:59 +08001998 // path in InstructionCodeGeneratorX86_64::GenerateReferenceLoadWithBakerReadBarrier.
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001999 locations->AddTemp(Location::RequiresRegister());
2000 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002001}
2002
2003void IntrinsicLocationsBuilderX86_64::VisitUnsafeGet(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00002004 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimInt);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002005}
2006void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetVolatile(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00002007 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimInt);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002008}
2009void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetLong(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00002010 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimLong);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002011}
2012void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetLongVolatile(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00002013 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimLong);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002014}
Andreas Gampe878d58c2015-01-15 23:24:00 -08002015void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetObject(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00002016 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimNot);
Andreas Gampe878d58c2015-01-15 23:24:00 -08002017}
2018void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetObjectVolatile(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00002019 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimNot);
Andreas Gampe878d58c2015-01-15 23:24:00 -08002020}
2021
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002022
2023void IntrinsicCodeGeneratorX86_64::VisitUnsafeGet(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002024 GenUnsafeGet(invoke, Primitive::kPrimInt, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002025}
2026void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002027 GenUnsafeGet(invoke, Primitive::kPrimInt, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002028}
2029void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002030 GenUnsafeGet(invoke, Primitive::kPrimLong, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002031}
2032void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetLongVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002033 GenUnsafeGet(invoke, Primitive::kPrimLong, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002034}
Andreas Gampe878d58c2015-01-15 23:24:00 -08002035void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetObject(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002036 GenUnsafeGet(invoke, Primitive::kPrimNot, /* is_volatile */ false, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08002037}
2038void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetObjectVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002039 GenUnsafeGet(invoke, Primitive::kPrimNot, /* is_volatile */ true, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08002040}
2041
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002042
2043static void CreateIntIntIntIntToVoidPlusTempsLocations(ArenaAllocator* arena,
2044 Primitive::Type type,
2045 HInvoke* invoke) {
2046 LocationSummary* locations = new (arena) LocationSummary(invoke,
2047 LocationSummary::kNoCall,
2048 kIntrinsified);
Andreas Gampe878d58c2015-01-15 23:24:00 -08002049 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002050 locations->SetInAt(1, Location::RequiresRegister());
2051 locations->SetInAt(2, Location::RequiresRegister());
2052 locations->SetInAt(3, Location::RequiresRegister());
2053 if (type == Primitive::kPrimNot) {
2054 // Need temp registers for card-marking.
Roland Levillain4d027112015-07-01 15:41:14 +01002055 locations->AddTemp(Location::RequiresRegister()); // Possibly used for reference poisoning too.
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002056 locations->AddTemp(Location::RequiresRegister());
2057 }
2058}
2059
2060void IntrinsicLocationsBuilderX86_64::VisitUnsafePut(HInvoke* invoke) {
2061 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimInt, invoke);
2062}
2063void IntrinsicLocationsBuilderX86_64::VisitUnsafePutOrdered(HInvoke* invoke) {
2064 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimInt, invoke);
2065}
2066void IntrinsicLocationsBuilderX86_64::VisitUnsafePutVolatile(HInvoke* invoke) {
2067 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimInt, invoke);
2068}
2069void IntrinsicLocationsBuilderX86_64::VisitUnsafePutObject(HInvoke* invoke) {
2070 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimNot, invoke);
2071}
2072void IntrinsicLocationsBuilderX86_64::VisitUnsafePutObjectOrdered(HInvoke* invoke) {
2073 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimNot, invoke);
2074}
2075void IntrinsicLocationsBuilderX86_64::VisitUnsafePutObjectVolatile(HInvoke* invoke) {
2076 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimNot, invoke);
2077}
2078void IntrinsicLocationsBuilderX86_64::VisitUnsafePutLong(HInvoke* invoke) {
2079 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimLong, invoke);
2080}
2081void IntrinsicLocationsBuilderX86_64::VisitUnsafePutLongOrdered(HInvoke* invoke) {
2082 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimLong, invoke);
2083}
2084void IntrinsicLocationsBuilderX86_64::VisitUnsafePutLongVolatile(HInvoke* invoke) {
2085 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimLong, invoke);
2086}
2087
2088// We don't care for ordered: it requires an AnyStore barrier, which is already given by the x86
2089// memory model.
2090static void GenUnsafePut(LocationSummary* locations, Primitive::Type type, bool is_volatile,
2091 CodeGeneratorX86_64* codegen) {
Roland Levillainb488b782015-10-22 11:38:49 +01002092 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen->GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002093 CpuRegister base = locations->InAt(1).AsRegister<CpuRegister>();
2094 CpuRegister offset = locations->InAt(2).AsRegister<CpuRegister>();
2095 CpuRegister value = locations->InAt(3).AsRegister<CpuRegister>();
2096
2097 if (type == Primitive::kPrimLong) {
2098 __ movq(Address(base, offset, ScaleFactor::TIMES_1, 0), value);
Roland Levillain4d027112015-07-01 15:41:14 +01002099 } else if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
2100 CpuRegister temp = locations->GetTemp(0).AsRegister<CpuRegister>();
2101 __ movl(temp, value);
2102 __ PoisonHeapReference(temp);
2103 __ movl(Address(base, offset, ScaleFactor::TIMES_1, 0), temp);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002104 } else {
2105 __ movl(Address(base, offset, ScaleFactor::TIMES_1, 0), value);
2106 }
2107
2108 if (is_volatile) {
Mark P Mendell17077d82015-12-16 19:15:59 +00002109 codegen->MemoryFence();
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002110 }
2111
2112 if (type == Primitive::kPrimNot) {
Nicolas Geoffray07276db2015-05-18 14:22:09 +01002113 bool value_can_be_null = true; // TODO: Worth finding out this information?
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002114 codegen->MarkGCCard(locations->GetTemp(0).AsRegister<CpuRegister>(),
2115 locations->GetTemp(1).AsRegister<CpuRegister>(),
2116 base,
Nicolas Geoffray07276db2015-05-18 14:22:09 +01002117 value,
2118 value_can_be_null);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002119 }
2120}
2121
2122void IntrinsicCodeGeneratorX86_64::VisitUnsafePut(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002123 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimInt, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002124}
2125void IntrinsicCodeGeneratorX86_64::VisitUnsafePutOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002126 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimInt, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002127}
2128void IntrinsicCodeGeneratorX86_64::VisitUnsafePutVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002129 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimInt, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002130}
2131void IntrinsicCodeGeneratorX86_64::VisitUnsafePutObject(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002132 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimNot, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002133}
2134void IntrinsicCodeGeneratorX86_64::VisitUnsafePutObjectOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002135 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimNot, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002136}
2137void IntrinsicCodeGeneratorX86_64::VisitUnsafePutObjectVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002138 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimNot, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002139}
2140void IntrinsicCodeGeneratorX86_64::VisitUnsafePutLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002141 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimLong, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002142}
2143void IntrinsicCodeGeneratorX86_64::VisitUnsafePutLongOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002144 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimLong, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002145}
2146void IntrinsicCodeGeneratorX86_64::VisitUnsafePutLongVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002147 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimLong, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002148}
2149
Mark Mendell58d25fd2015-04-03 14:52:31 -04002150static void CreateIntIntIntIntIntToInt(ArenaAllocator* arena, Primitive::Type type,
2151 HInvoke* invoke) {
2152 LocationSummary* locations = new (arena) LocationSummary(invoke,
2153 LocationSummary::kNoCall,
2154 kIntrinsified);
2155 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
2156 locations->SetInAt(1, Location::RequiresRegister());
2157 locations->SetInAt(2, Location::RequiresRegister());
2158 // expected value must be in EAX/RAX.
2159 locations->SetInAt(3, Location::RegisterLocation(RAX));
2160 locations->SetInAt(4, Location::RequiresRegister());
2161
2162 locations->SetOut(Location::RequiresRegister());
2163 if (type == Primitive::kPrimNot) {
2164 // Need temp registers for card-marking.
Roland Levillainb488b782015-10-22 11:38:49 +01002165 locations->AddTemp(Location::RequiresRegister()); // Possibly used for reference poisoning too.
Mark Mendell58d25fd2015-04-03 14:52:31 -04002166 locations->AddTemp(Location::RequiresRegister());
2167 }
2168}
2169
2170void IntrinsicLocationsBuilderX86_64::VisitUnsafeCASInt(HInvoke* invoke) {
2171 CreateIntIntIntIntIntToInt(arena_, Primitive::kPrimInt, invoke);
2172}
2173
2174void IntrinsicLocationsBuilderX86_64::VisitUnsafeCASLong(HInvoke* invoke) {
2175 CreateIntIntIntIntIntToInt(arena_, Primitive::kPrimLong, invoke);
2176}
2177
2178void IntrinsicLocationsBuilderX86_64::VisitUnsafeCASObject(HInvoke* invoke) {
Roland Levillain391b8662015-12-18 11:43:38 +00002179 // The UnsafeCASObject intrinsic is missing a read barrier, and
2180 // therefore sometimes does not work as expected (b/25883050).
2181 // Turn it off temporarily as a quick fix, until the read barrier is
2182 // implemented.
2183 //
2184 // TODO(rpl): Implement a read barrier in GenCAS below and re-enable
2185 // this intrinsic.
2186 if (kEmitCompilerReadBarrier) {
2187 return;
2188 }
2189
Mark Mendell58d25fd2015-04-03 14:52:31 -04002190 CreateIntIntIntIntIntToInt(arena_, Primitive::kPrimNot, invoke);
2191}
2192
2193static void GenCAS(Primitive::Type type, HInvoke* invoke, CodeGeneratorX86_64* codegen) {
Roland Levillainb488b782015-10-22 11:38:49 +01002194 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen->GetAssembler());
Mark Mendell58d25fd2015-04-03 14:52:31 -04002195 LocationSummary* locations = invoke->GetLocations();
2196
2197 CpuRegister base = locations->InAt(1).AsRegister<CpuRegister>();
2198 CpuRegister offset = locations->InAt(2).AsRegister<CpuRegister>();
2199 CpuRegister expected = locations->InAt(3).AsRegister<CpuRegister>();
Roland Levillainb488b782015-10-22 11:38:49 +01002200 // Ensure `expected` is in RAX (required by the CMPXCHG instruction).
Mark Mendell58d25fd2015-04-03 14:52:31 -04002201 DCHECK_EQ(expected.AsRegister(), RAX);
2202 CpuRegister value = locations->InAt(4).AsRegister<CpuRegister>();
2203 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2204
Roland Levillainb488b782015-10-22 11:38:49 +01002205 if (type == Primitive::kPrimNot) {
2206 // Mark card for object assuming new value is stored.
2207 bool value_can_be_null = true; // TODO: Worth finding out this information?
2208 codegen->MarkGCCard(locations->GetTemp(0).AsRegister<CpuRegister>(),
2209 locations->GetTemp(1).AsRegister<CpuRegister>(),
2210 base,
2211 value,
2212 value_can_be_null);
Roland Levillain4d027112015-07-01 15:41:14 +01002213
Roland Levillainb488b782015-10-22 11:38:49 +01002214 bool base_equals_value = (base.AsRegister() == value.AsRegister());
2215 Register value_reg = value.AsRegister();
2216 if (kPoisonHeapReferences) {
2217 if (base_equals_value) {
2218 // If `base` and `value` are the same register location, move
2219 // `value_reg` to a temporary register. This way, poisoning
2220 // `value_reg` won't invalidate `base`.
2221 value_reg = locations->GetTemp(0).AsRegister<CpuRegister>().AsRegister();
2222 __ movl(CpuRegister(value_reg), base);
Roland Levillain4d027112015-07-01 15:41:14 +01002223 }
Roland Levillainb488b782015-10-22 11:38:49 +01002224
2225 // Check that the register allocator did not assign the location
2226 // of `expected` (RAX) to `value` nor to `base`, so that heap
2227 // poisoning (when enabled) works as intended below.
2228 // - If `value` were equal to `expected`, both references would
2229 // be poisoned twice, meaning they would not be poisoned at
2230 // all, as heap poisoning uses address negation.
2231 // - If `base` were equal to `expected`, poisoning `expected`
2232 // would invalidate `base`.
2233 DCHECK_NE(value_reg, expected.AsRegister());
2234 DCHECK_NE(base.AsRegister(), expected.AsRegister());
2235
2236 __ PoisonHeapReference(expected);
2237 __ PoisonHeapReference(CpuRegister(value_reg));
Mark Mendell58d25fd2015-04-03 14:52:31 -04002238 }
2239
Roland Levillain391b8662015-12-18 11:43:38 +00002240 // TODO: Add a read barrier for the reference stored in the object
2241 // before attempting the CAS, similar to the one in the
2242 // art::Unsafe_compareAndSwapObject JNI implementation.
2243 //
2244 // Note that this code is not (yet) used when read barriers are
2245 // enabled (see IntrinsicLocationsBuilderX86_64::VisitUnsafeCASObject).
2246 DCHECK(!kEmitCompilerReadBarrier);
Roland Levillainb488b782015-10-22 11:38:49 +01002247 __ LockCmpxchgl(Address(base, offset, TIMES_1, 0), CpuRegister(value_reg));
Mark Mendell58d25fd2015-04-03 14:52:31 -04002248
Roland Levillain0d5a2812015-11-13 10:07:31 +00002249 // LOCK CMPXCHG has full barrier semantics, and we don't need
Roland Levillainb488b782015-10-22 11:38:49 +01002250 // scheduling barriers at this time.
Mark Mendell58d25fd2015-04-03 14:52:31 -04002251
Roland Levillainb488b782015-10-22 11:38:49 +01002252 // Convert ZF into the boolean result.
2253 __ setcc(kZero, out);
2254 __ movzxb(out, out);
Roland Levillain4d027112015-07-01 15:41:14 +01002255
Roland Levillain391b8662015-12-18 11:43:38 +00002256 // If heap poisoning is enabled, we need to unpoison the values
2257 // that were poisoned earlier.
Roland Levillainb488b782015-10-22 11:38:49 +01002258 if (kPoisonHeapReferences) {
2259 if (base_equals_value) {
2260 // `value_reg` has been moved to a temporary register, no need
2261 // to unpoison it.
2262 } else {
2263 // Ensure `value` is different from `out`, so that unpoisoning
2264 // the former does not invalidate the latter.
2265 DCHECK_NE(value_reg, out.AsRegister());
2266 __ UnpoisonHeapReference(CpuRegister(value_reg));
2267 }
2268 // Ensure `expected` is different from `out`, so that unpoisoning
2269 // the former does not invalidate the latter.
2270 DCHECK_NE(expected.AsRegister(), out.AsRegister());
2271 __ UnpoisonHeapReference(expected);
2272 }
2273 } else {
2274 if (type == Primitive::kPrimInt) {
2275 __ LockCmpxchgl(Address(base, offset, TIMES_1, 0), value);
2276 } else if (type == Primitive::kPrimLong) {
2277 __ LockCmpxchgq(Address(base, offset, TIMES_1, 0), value);
2278 } else {
2279 LOG(FATAL) << "Unexpected CAS type " << type;
2280 }
2281
Roland Levillain0d5a2812015-11-13 10:07:31 +00002282 // LOCK CMPXCHG has full barrier semantics, and we don't need
Roland Levillainb488b782015-10-22 11:38:49 +01002283 // scheduling barriers at this time.
2284
2285 // Convert ZF into the boolean result.
2286 __ setcc(kZero, out);
2287 __ movzxb(out, out);
Roland Levillain4d027112015-07-01 15:41:14 +01002288 }
Mark Mendell58d25fd2015-04-03 14:52:31 -04002289}
2290
2291void IntrinsicCodeGeneratorX86_64::VisitUnsafeCASInt(HInvoke* invoke) {
2292 GenCAS(Primitive::kPrimInt, invoke, codegen_);
2293}
2294
2295void IntrinsicCodeGeneratorX86_64::VisitUnsafeCASLong(HInvoke* invoke) {
2296 GenCAS(Primitive::kPrimLong, invoke, codegen_);
2297}
2298
2299void IntrinsicCodeGeneratorX86_64::VisitUnsafeCASObject(HInvoke* invoke) {
2300 GenCAS(Primitive::kPrimNot, invoke, codegen_);
2301}
2302
2303void IntrinsicLocationsBuilderX86_64::VisitIntegerReverse(HInvoke* invoke) {
2304 LocationSummary* locations = new (arena_) LocationSummary(invoke,
2305 LocationSummary::kNoCall,
2306 kIntrinsified);
2307 locations->SetInAt(0, Location::RequiresRegister());
2308 locations->SetOut(Location::SameAsFirstInput());
2309 locations->AddTemp(Location::RequiresRegister());
2310}
2311
2312static void SwapBits(CpuRegister reg, CpuRegister temp, int32_t shift, int32_t mask,
2313 X86_64Assembler* assembler) {
2314 Immediate imm_shift(shift);
2315 Immediate imm_mask(mask);
2316 __ movl(temp, reg);
2317 __ shrl(reg, imm_shift);
2318 __ andl(temp, imm_mask);
2319 __ andl(reg, imm_mask);
2320 __ shll(temp, imm_shift);
2321 __ orl(reg, temp);
2322}
2323
2324void IntrinsicCodeGeneratorX86_64::VisitIntegerReverse(HInvoke* invoke) {
Aart Bikc5d47542016-01-27 17:00:35 -08002325 X86_64Assembler* assembler = GetAssembler();
Mark Mendell58d25fd2015-04-03 14:52:31 -04002326 LocationSummary* locations = invoke->GetLocations();
2327
2328 CpuRegister reg = locations->InAt(0).AsRegister<CpuRegister>();
2329 CpuRegister temp = locations->GetTemp(0).AsRegister<CpuRegister>();
2330
2331 /*
2332 * Use one bswap instruction to reverse byte order first and then use 3 rounds of
2333 * swapping bits to reverse bits in a number x. Using bswap to save instructions
2334 * compared to generic luni implementation which has 5 rounds of swapping bits.
2335 * x = bswap x
2336 * x = (x & 0x55555555) << 1 | (x >> 1) & 0x55555555;
2337 * x = (x & 0x33333333) << 2 | (x >> 2) & 0x33333333;
2338 * x = (x & 0x0F0F0F0F) << 4 | (x >> 4) & 0x0F0F0F0F;
2339 */
2340 __ bswapl(reg);
2341 SwapBits(reg, temp, 1, 0x55555555, assembler);
2342 SwapBits(reg, temp, 2, 0x33333333, assembler);
2343 SwapBits(reg, temp, 4, 0x0f0f0f0f, assembler);
2344}
2345
2346void IntrinsicLocationsBuilderX86_64::VisitLongReverse(HInvoke* invoke) {
2347 LocationSummary* locations = new (arena_) LocationSummary(invoke,
2348 LocationSummary::kNoCall,
2349 kIntrinsified);
2350 locations->SetInAt(0, Location::RequiresRegister());
2351 locations->SetOut(Location::SameAsFirstInput());
2352 locations->AddTemp(Location::RequiresRegister());
2353 locations->AddTemp(Location::RequiresRegister());
2354}
2355
2356static void SwapBits64(CpuRegister reg, CpuRegister temp, CpuRegister temp_mask,
2357 int32_t shift, int64_t mask, X86_64Assembler* assembler) {
2358 Immediate imm_shift(shift);
2359 __ movq(temp_mask, Immediate(mask));
2360 __ movq(temp, reg);
2361 __ shrq(reg, imm_shift);
2362 __ andq(temp, temp_mask);
2363 __ andq(reg, temp_mask);
2364 __ shlq(temp, imm_shift);
2365 __ orq(reg, temp);
2366}
2367
2368void IntrinsicCodeGeneratorX86_64::VisitLongReverse(HInvoke* invoke) {
Aart Bikc5d47542016-01-27 17:00:35 -08002369 X86_64Assembler* assembler = GetAssembler();
Mark Mendell58d25fd2015-04-03 14:52:31 -04002370 LocationSummary* locations = invoke->GetLocations();
2371
2372 CpuRegister reg = locations->InAt(0).AsRegister<CpuRegister>();
2373 CpuRegister temp1 = locations->GetTemp(0).AsRegister<CpuRegister>();
2374 CpuRegister temp2 = locations->GetTemp(1).AsRegister<CpuRegister>();
2375
2376 /*
2377 * Use one bswap instruction to reverse byte order first and then use 3 rounds of
2378 * swapping bits to reverse bits in a long number x. Using bswap to save instructions
2379 * compared to generic luni implementation which has 5 rounds of swapping bits.
2380 * x = bswap x
2381 * x = (x & 0x5555555555555555) << 1 | (x >> 1) & 0x5555555555555555;
2382 * x = (x & 0x3333333333333333) << 2 | (x >> 2) & 0x3333333333333333;
2383 * x = (x & 0x0F0F0F0F0F0F0F0F) << 4 | (x >> 4) & 0x0F0F0F0F0F0F0F0F;
2384 */
2385 __ bswapq(reg);
2386 SwapBits64(reg, temp1, temp2, 1, INT64_C(0x5555555555555555), assembler);
2387 SwapBits64(reg, temp1, temp2, 2, INT64_C(0x3333333333333333), assembler);
2388 SwapBits64(reg, temp1, temp2, 4, INT64_C(0x0f0f0f0f0f0f0f0f), assembler);
2389}
2390
Aart Bik3f67e692016-01-15 14:35:12 -08002391static void CreateBitCountLocations(
2392 ArenaAllocator* arena, CodeGeneratorX86_64* codegen, HInvoke* invoke) {
2393 if (!codegen->GetInstructionSetFeatures().HasPopCnt()) {
2394 // Do nothing if there is no popcnt support. This results in generating
2395 // a call for the intrinsic rather than direct code.
2396 return;
2397 }
2398 LocationSummary* locations = new (arena) LocationSummary(invoke,
2399 LocationSummary::kNoCall,
2400 kIntrinsified);
2401 locations->SetInAt(0, Location::Any());
2402 locations->SetOut(Location::RequiresRegister());
2403}
2404
Aart Bikc5d47542016-01-27 17:00:35 -08002405static void GenBitCount(X86_64Assembler* assembler,
2406 CodeGeneratorX86_64* codegen,
2407 HInvoke* invoke,
2408 bool is_long) {
Aart Bik3f67e692016-01-15 14:35:12 -08002409 LocationSummary* locations = invoke->GetLocations();
2410 Location src = locations->InAt(0);
2411 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2412
2413 if (invoke->InputAt(0)->IsConstant()) {
2414 // Evaluate this at compile time.
2415 int64_t value = Int64FromConstant(invoke->InputAt(0)->AsConstant());
Roland Levillainfa3912e2016-04-01 18:21:55 +01002416 int32_t result = is_long
Aart Bik3f67e692016-01-15 14:35:12 -08002417 ? POPCOUNT(static_cast<uint64_t>(value))
2418 : POPCOUNT(static_cast<uint32_t>(value));
Roland Levillainfa3912e2016-04-01 18:21:55 +01002419 codegen->Load32BitValue(out, result);
Aart Bik3f67e692016-01-15 14:35:12 -08002420 return;
2421 }
2422
2423 if (src.IsRegister()) {
2424 if (is_long) {
2425 __ popcntq(out, src.AsRegister<CpuRegister>());
2426 } else {
2427 __ popcntl(out, src.AsRegister<CpuRegister>());
2428 }
2429 } else if (is_long) {
2430 DCHECK(src.IsDoubleStackSlot());
2431 __ popcntq(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2432 } else {
2433 DCHECK(src.IsStackSlot());
2434 __ popcntl(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2435 }
2436}
2437
2438void IntrinsicLocationsBuilderX86_64::VisitIntegerBitCount(HInvoke* invoke) {
2439 CreateBitCountLocations(arena_, codegen_, invoke);
2440}
2441
2442void IntrinsicCodeGeneratorX86_64::VisitIntegerBitCount(HInvoke* invoke) {
Aart Bikc5d47542016-01-27 17:00:35 -08002443 GenBitCount(GetAssembler(), codegen_, invoke, /* is_long */ false);
Aart Bik3f67e692016-01-15 14:35:12 -08002444}
2445
2446void IntrinsicLocationsBuilderX86_64::VisitLongBitCount(HInvoke* invoke) {
2447 CreateBitCountLocations(arena_, codegen_, invoke);
2448}
2449
2450void IntrinsicCodeGeneratorX86_64::VisitLongBitCount(HInvoke* invoke) {
Aart Bikc5d47542016-01-27 17:00:35 -08002451 GenBitCount(GetAssembler(), codegen_, invoke, /* is_long */ true);
2452}
2453
Aart Bikc5d47542016-01-27 17:00:35 -08002454static void CreateOneBitLocations(ArenaAllocator* arena, HInvoke* invoke, bool is_high) {
2455 LocationSummary* locations = new (arena) LocationSummary(invoke,
2456 LocationSummary::kNoCall,
2457 kIntrinsified);
2458 locations->SetInAt(0, Location::Any());
2459 locations->SetOut(Location::RequiresRegister());
2460 locations->AddTemp(is_high ? Location::RegisterLocation(RCX) // needs CL
2461 : Location::RequiresRegister()); // any will do
2462}
2463
2464static void GenOneBit(X86_64Assembler* assembler,
2465 CodeGeneratorX86_64* codegen,
2466 HInvoke* invoke,
2467 bool is_high, bool is_long) {
2468 LocationSummary* locations = invoke->GetLocations();
2469 Location src = locations->InAt(0);
2470 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2471
2472 if (invoke->InputAt(0)->IsConstant()) {
2473 // Evaluate this at compile time.
2474 int64_t value = Int64FromConstant(invoke->InputAt(0)->AsConstant());
2475 if (value == 0) {
2476 __ xorl(out, out); // Clears upper bits too.
2477 return;
2478 }
2479 // Nonzero value.
2480 if (is_high) {
2481 value = is_long ? 63 - CLZ(static_cast<uint64_t>(value))
2482 : 31 - CLZ(static_cast<uint32_t>(value));
2483 } else {
2484 value = is_long ? CTZ(static_cast<uint64_t>(value))
2485 : CTZ(static_cast<uint32_t>(value));
2486 }
2487 if (is_long) {
2488 codegen->Load64BitValue(out, 1L << value);
2489 } else {
2490 codegen->Load32BitValue(out, 1 << value);
2491 }
2492 return;
2493 }
2494
2495 // Handle the non-constant cases.
2496 CpuRegister tmp = locations->GetTemp(0).AsRegister<CpuRegister>();
2497 if (is_high) {
2498 // Use architectural support: basically 1 << bsr.
2499 if (src.IsRegister()) {
2500 if (is_long) {
2501 __ bsrq(tmp, src.AsRegister<CpuRegister>());
2502 } else {
2503 __ bsrl(tmp, src.AsRegister<CpuRegister>());
2504 }
2505 } else if (is_long) {
2506 DCHECK(src.IsDoubleStackSlot());
2507 __ bsrq(tmp, Address(CpuRegister(RSP), src.GetStackIndex()));
2508 } else {
2509 DCHECK(src.IsStackSlot());
2510 __ bsrl(tmp, Address(CpuRegister(RSP), src.GetStackIndex()));
2511 }
2512 // BSR sets ZF if the input was zero.
2513 NearLabel is_zero, done;
2514 __ j(kEqual, &is_zero);
2515 __ movl(out, Immediate(1)); // Clears upper bits too.
2516 if (is_long) {
2517 __ shlq(out, tmp);
2518 } else {
2519 __ shll(out, tmp);
2520 }
2521 __ jmp(&done);
2522 __ Bind(&is_zero);
2523 __ xorl(out, out); // Clears upper bits too.
2524 __ Bind(&done);
2525 } else {
2526 // Copy input into temporary.
2527 if (src.IsRegister()) {
2528 if (is_long) {
2529 __ movq(tmp, src.AsRegister<CpuRegister>());
2530 } else {
2531 __ movl(tmp, src.AsRegister<CpuRegister>());
2532 }
2533 } else if (is_long) {
2534 DCHECK(src.IsDoubleStackSlot());
2535 __ movq(tmp, Address(CpuRegister(RSP), src.GetStackIndex()));
2536 } else {
2537 DCHECK(src.IsStackSlot());
2538 __ movl(tmp, Address(CpuRegister(RSP), src.GetStackIndex()));
2539 }
2540 // Do the bit twiddling: basically tmp & -tmp;
2541 if (is_long) {
2542 __ movq(out, tmp);
2543 __ negq(tmp);
2544 __ andq(out, tmp);
2545 } else {
2546 __ movl(out, tmp);
2547 __ negl(tmp);
2548 __ andl(out, tmp);
2549 }
2550 }
2551}
2552
2553void IntrinsicLocationsBuilderX86_64::VisitIntegerHighestOneBit(HInvoke* invoke) {
2554 CreateOneBitLocations(arena_, invoke, /* is_high */ true);
2555}
2556
2557void IntrinsicCodeGeneratorX86_64::VisitIntegerHighestOneBit(HInvoke* invoke) {
2558 GenOneBit(GetAssembler(), codegen_, invoke, /* is_high */ true, /* is_long */ false);
2559}
2560
2561void IntrinsicLocationsBuilderX86_64::VisitLongHighestOneBit(HInvoke* invoke) {
2562 CreateOneBitLocations(arena_, invoke, /* is_high */ true);
2563}
2564
2565void IntrinsicCodeGeneratorX86_64::VisitLongHighestOneBit(HInvoke* invoke) {
2566 GenOneBit(GetAssembler(), codegen_, invoke, /* is_high */ true, /* is_long */ true);
2567}
2568
2569void IntrinsicLocationsBuilderX86_64::VisitIntegerLowestOneBit(HInvoke* invoke) {
2570 CreateOneBitLocations(arena_, invoke, /* is_high */ false);
2571}
2572
2573void IntrinsicCodeGeneratorX86_64::VisitIntegerLowestOneBit(HInvoke* invoke) {
2574 GenOneBit(GetAssembler(), codegen_, invoke, /* is_high */ false, /* is_long */ false);
2575}
2576
2577void IntrinsicLocationsBuilderX86_64::VisitLongLowestOneBit(HInvoke* invoke) {
2578 CreateOneBitLocations(arena_, invoke, /* is_high */ false);
2579}
2580
2581void IntrinsicCodeGeneratorX86_64::VisitLongLowestOneBit(HInvoke* invoke) {
2582 GenOneBit(GetAssembler(), codegen_, invoke, /* is_high */ false, /* is_long */ true);
Aart Bik3f67e692016-01-15 14:35:12 -08002583}
2584
Mark Mendelld5897672015-08-12 21:16:41 -04002585static void CreateLeadingZeroLocations(ArenaAllocator* arena, HInvoke* invoke) {
2586 LocationSummary* locations = new (arena) LocationSummary(invoke,
2587 LocationSummary::kNoCall,
2588 kIntrinsified);
2589 locations->SetInAt(0, Location::Any());
2590 locations->SetOut(Location::RequiresRegister());
2591}
2592
Aart Bikc5d47542016-01-27 17:00:35 -08002593static void GenLeadingZeros(X86_64Assembler* assembler,
2594 CodeGeneratorX86_64* codegen,
2595 HInvoke* invoke, bool is_long) {
Mark Mendelld5897672015-08-12 21:16:41 -04002596 LocationSummary* locations = invoke->GetLocations();
2597 Location src = locations->InAt(0);
2598 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2599
2600 int zero_value_result = is_long ? 64 : 32;
2601 if (invoke->InputAt(0)->IsConstant()) {
2602 // Evaluate this at compile time.
2603 int64_t value = Int64FromConstant(invoke->InputAt(0)->AsConstant());
2604 if (value == 0) {
2605 value = zero_value_result;
2606 } else {
2607 value = is_long ? CLZ(static_cast<uint64_t>(value)) : CLZ(static_cast<uint32_t>(value));
2608 }
Aart Bikc5d47542016-01-27 17:00:35 -08002609 codegen->Load32BitValue(out, value);
Mark Mendelld5897672015-08-12 21:16:41 -04002610 return;
2611 }
2612
2613 // Handle the non-constant cases.
2614 if (src.IsRegister()) {
2615 if (is_long) {
2616 __ bsrq(out, src.AsRegister<CpuRegister>());
2617 } else {
2618 __ bsrl(out, src.AsRegister<CpuRegister>());
2619 }
2620 } else if (is_long) {
2621 DCHECK(src.IsDoubleStackSlot());
2622 __ bsrq(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2623 } else {
2624 DCHECK(src.IsStackSlot());
2625 __ bsrl(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2626 }
2627
2628 // BSR sets ZF if the input was zero, and the output is undefined.
Mark Mendell0c9497d2015-08-21 09:30:05 -04002629 NearLabel is_zero, done;
Mark Mendelld5897672015-08-12 21:16:41 -04002630 __ j(kEqual, &is_zero);
2631
2632 // Correct the result from BSR to get the CLZ result.
2633 __ xorl(out, Immediate(zero_value_result - 1));
2634 __ jmp(&done);
2635
2636 // Fix the zero case with the expected result.
2637 __ Bind(&is_zero);
2638 __ movl(out, Immediate(zero_value_result));
2639
2640 __ Bind(&done);
2641}
2642
2643void IntrinsicLocationsBuilderX86_64::VisitIntegerNumberOfLeadingZeros(HInvoke* invoke) {
2644 CreateLeadingZeroLocations(arena_, invoke);
2645}
2646
2647void IntrinsicCodeGeneratorX86_64::VisitIntegerNumberOfLeadingZeros(HInvoke* invoke) {
Aart Bikc5d47542016-01-27 17:00:35 -08002648 GenLeadingZeros(GetAssembler(), codegen_, invoke, /* is_long */ false);
Mark Mendelld5897672015-08-12 21:16:41 -04002649}
2650
2651void IntrinsicLocationsBuilderX86_64::VisitLongNumberOfLeadingZeros(HInvoke* invoke) {
2652 CreateLeadingZeroLocations(arena_, invoke);
2653}
2654
2655void IntrinsicCodeGeneratorX86_64::VisitLongNumberOfLeadingZeros(HInvoke* invoke) {
Aart Bikc5d47542016-01-27 17:00:35 -08002656 GenLeadingZeros(GetAssembler(), codegen_, invoke, /* is_long */ true);
Mark Mendelld5897672015-08-12 21:16:41 -04002657}
2658
Mark Mendell2d554792015-09-15 21:45:18 -04002659static void CreateTrailingZeroLocations(ArenaAllocator* arena, HInvoke* invoke) {
2660 LocationSummary* locations = new (arena) LocationSummary(invoke,
2661 LocationSummary::kNoCall,
2662 kIntrinsified);
2663 locations->SetInAt(0, Location::Any());
2664 locations->SetOut(Location::RequiresRegister());
2665}
2666
Aart Bikc5d47542016-01-27 17:00:35 -08002667static void GenTrailingZeros(X86_64Assembler* assembler,
2668 CodeGeneratorX86_64* codegen,
2669 HInvoke* invoke, bool is_long) {
Mark Mendell2d554792015-09-15 21:45:18 -04002670 LocationSummary* locations = invoke->GetLocations();
2671 Location src = locations->InAt(0);
2672 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2673
2674 int zero_value_result = is_long ? 64 : 32;
2675 if (invoke->InputAt(0)->IsConstant()) {
2676 // Evaluate this at compile time.
2677 int64_t value = Int64FromConstant(invoke->InputAt(0)->AsConstant());
2678 if (value == 0) {
2679 value = zero_value_result;
2680 } else {
2681 value = is_long ? CTZ(static_cast<uint64_t>(value)) : CTZ(static_cast<uint32_t>(value));
2682 }
Aart Bikc5d47542016-01-27 17:00:35 -08002683 codegen->Load32BitValue(out, value);
Mark Mendell2d554792015-09-15 21:45:18 -04002684 return;
2685 }
2686
2687 // Handle the non-constant cases.
2688 if (src.IsRegister()) {
2689 if (is_long) {
2690 __ bsfq(out, src.AsRegister<CpuRegister>());
2691 } else {
2692 __ bsfl(out, src.AsRegister<CpuRegister>());
2693 }
2694 } else if (is_long) {
2695 DCHECK(src.IsDoubleStackSlot());
2696 __ bsfq(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2697 } else {
2698 DCHECK(src.IsStackSlot());
2699 __ bsfl(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2700 }
2701
2702 // BSF sets ZF if the input was zero, and the output is undefined.
2703 NearLabel done;
2704 __ j(kNotEqual, &done);
2705
2706 // Fix the zero case with the expected result.
2707 __ movl(out, Immediate(zero_value_result));
2708
2709 __ Bind(&done);
2710}
2711
2712void IntrinsicLocationsBuilderX86_64::VisitIntegerNumberOfTrailingZeros(HInvoke* invoke) {
2713 CreateTrailingZeroLocations(arena_, invoke);
2714}
2715
2716void IntrinsicCodeGeneratorX86_64::VisitIntegerNumberOfTrailingZeros(HInvoke* invoke) {
Aart Bikc5d47542016-01-27 17:00:35 -08002717 GenTrailingZeros(GetAssembler(), codegen_, invoke, /* is_long */ false);
Mark Mendell2d554792015-09-15 21:45:18 -04002718}
2719
2720void IntrinsicLocationsBuilderX86_64::VisitLongNumberOfTrailingZeros(HInvoke* invoke) {
2721 CreateTrailingZeroLocations(arena_, invoke);
2722}
2723
2724void IntrinsicCodeGeneratorX86_64::VisitLongNumberOfTrailingZeros(HInvoke* invoke) {
Aart Bikc5d47542016-01-27 17:00:35 -08002725 GenTrailingZeros(GetAssembler(), codegen_, invoke, /* is_long */ true);
2726}
2727
Serguei Katkov288c7a82016-05-16 11:53:15 +06002728void IntrinsicLocationsBuilderX86_64::VisitReferenceGetReferent(HInvoke* invoke) {
2729 if (kEmitCompilerReadBarrier) {
2730 // Do not intrinsify this call with the read barrier configuration.
2731 return;
2732 }
2733 LocationSummary* locations = new (arena_) LocationSummary(invoke,
2734 LocationSummary::kCallOnSlowPath,
2735 kIntrinsified);
2736 locations->SetInAt(0, Location::RequiresRegister());
2737 locations->SetOut(Location::SameAsFirstInput());
2738 locations->AddTemp(Location::RequiresRegister());
2739}
2740
2741void IntrinsicCodeGeneratorX86_64::VisitReferenceGetReferent(HInvoke* invoke) {
2742 DCHECK(!kEmitCompilerReadBarrier);
2743 LocationSummary* locations = invoke->GetLocations();
2744 X86_64Assembler* assembler = GetAssembler();
2745
2746 CpuRegister obj = locations->InAt(0).AsRegister<CpuRegister>();
2747 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2748
2749 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
2750 codegen_->AddSlowPath(slow_path);
2751
2752 // Load ArtMethod first.
2753 HInvokeStaticOrDirect* invoke_direct = invoke->AsInvokeStaticOrDirect();
2754 DCHECK(invoke_direct != nullptr);
2755 Location temp_loc = codegen_->GenerateCalleeMethodStaticOrDirectCall(
2756 invoke_direct, locations->GetTemp(0));
2757 DCHECK(temp_loc.Equals(locations->GetTemp(0)));
2758 CpuRegister temp = temp_loc.AsRegister<CpuRegister>();
2759
2760 // Now get declaring class.
2761 __ movl(temp, Address(temp, ArtMethod::DeclaringClassOffset().Int32Value()));
2762
2763 uint32_t slow_path_flag_offset = codegen_->GetReferenceSlowFlagOffset();
2764 uint32_t disable_flag_offset = codegen_->GetReferenceDisableFlagOffset();
2765 DCHECK_NE(slow_path_flag_offset, 0u);
2766 DCHECK_NE(disable_flag_offset, 0u);
2767 DCHECK_NE(slow_path_flag_offset, disable_flag_offset);
2768
2769 // Check static flags preventing us for using intrinsic.
2770 if (slow_path_flag_offset == disable_flag_offset + 1) {
2771 __ cmpw(Address(temp, disable_flag_offset), Immediate(0));
2772 __ j(kNotEqual, slow_path->GetEntryLabel());
2773 } else {
2774 __ cmpb(Address(temp, disable_flag_offset), Immediate(0));
2775 __ j(kNotEqual, slow_path->GetEntryLabel());
2776 __ cmpb(Address(temp, slow_path_flag_offset), Immediate(0));
2777 __ j(kNotEqual, slow_path->GetEntryLabel());
2778 }
2779
2780 // Fast path.
2781 __ movl(out, Address(obj, mirror::Reference::ReferentOffset().Int32Value()));
2782 codegen_->MaybeRecordImplicitNullCheck(invoke);
2783 __ MaybeUnpoisonHeapReference(out);
2784 __ Bind(slow_path->GetExitLabel());
2785}
2786
Aart Bik2f9fcc92016-03-01 15:16:54 -08002787UNIMPLEMENTED_INTRINSIC(X86_64, FloatIsInfinite)
2788UNIMPLEMENTED_INTRINSIC(X86_64, DoubleIsInfinite)
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002789
Aart Bik0e54c012016-03-04 12:08:31 -08002790// 1.8.
2791UNIMPLEMENTED_INTRINSIC(X86_64, UnsafeGetAndAddInt)
2792UNIMPLEMENTED_INTRINSIC(X86_64, UnsafeGetAndAddLong)
2793UNIMPLEMENTED_INTRINSIC(X86_64, UnsafeGetAndSetInt)
2794UNIMPLEMENTED_INTRINSIC(X86_64, UnsafeGetAndSetLong)
2795UNIMPLEMENTED_INTRINSIC(X86_64, UnsafeGetAndSetObject)
Aart Bik0e54c012016-03-04 12:08:31 -08002796
Aart Bik2f9fcc92016-03-01 15:16:54 -08002797UNREACHABLE_INTRINSICS(X86_64)
Roland Levillain4d027112015-07-01 15:41:14 +01002798
2799#undef __
2800
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002801} // namespace x86_64
2802} // namespace art