blob: 2986dc884293d60855cab691e3a084981a4d6e7e [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) {
613 CreateSSE41FPToIntLocations(arena_, invoke, codegen_);
614}
615
616void IntrinsicCodeGeneratorX86_64::VisitMathRoundFloat(HInvoke* invoke) {
617 LocationSummary* locations = invoke->GetLocations();
618 if (locations->WillCall()) {
619 InvokeOutOfLineIntrinsic(codegen_, invoke);
620 return;
621 }
622
623 // Implement RoundFloat as t1 = floor(input + 0.5f); convert to int.
624 XmmRegister in = locations->InAt(0).AsFpuRegister<XmmRegister>();
625 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
Mark Mendell40741f32015-04-20 22:10:34 -0400626 XmmRegister inPlusPointFive = locations->GetTemp(0).AsFpuRegister<XmmRegister>();
Mark Mendell0c9497d2015-08-21 09:30:05 -0400627 NearLabel done, nan;
Mark Mendellfb8d2792015-03-31 22:16:59 -0400628 X86_64Assembler* assembler = GetAssembler();
629
Mark Mendell40741f32015-04-20 22:10:34 -0400630 // Load 0.5 into inPlusPointFive.
631 __ movss(inPlusPointFive, codegen_->LiteralFloatAddress(0.5f));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400632
633 // Add in the input.
634 __ addss(inPlusPointFive, in);
635
636 // And truncate to an integer.
637 __ roundss(inPlusPointFive, inPlusPointFive, Immediate(1));
638
Pavel Vyssotski9ca25712015-07-31 13:03:17 +0600639 // Load maxInt into out.
640 codegen_->Load64BitValue(out, kPrimIntMax);
641
Mark Mendellfb8d2792015-03-31 22:16:59 -0400642 // if inPlusPointFive >= maxInt goto done
Mark Mendell40741f32015-04-20 22:10:34 -0400643 __ comiss(inPlusPointFive, codegen_->LiteralFloatAddress(static_cast<float>(kPrimIntMax)));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400644 __ j(kAboveEqual, &done);
645
646 // if input == NaN goto nan
647 __ j(kUnordered, &nan);
648
649 // output = float-to-int-truncate(input)
650 __ cvttss2si(out, inPlusPointFive);
651 __ jmp(&done);
652 __ Bind(&nan);
653
654 // output = 0
655 __ xorl(out, out);
656 __ Bind(&done);
657}
658
659void IntrinsicLocationsBuilderX86_64::VisitMathRoundDouble(HInvoke* invoke) {
660 CreateSSE41FPToIntLocations(arena_, invoke, codegen_);
661}
662
663void IntrinsicCodeGeneratorX86_64::VisitMathRoundDouble(HInvoke* invoke) {
664 LocationSummary* locations = invoke->GetLocations();
665 if (locations->WillCall()) {
666 InvokeOutOfLineIntrinsic(codegen_, invoke);
667 return;
668 }
669
670 // Implement RoundDouble as t1 = floor(input + 0.5); convert to long.
671 XmmRegister in = locations->InAt(0).AsFpuRegister<XmmRegister>();
672 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
Mark Mendell40741f32015-04-20 22:10:34 -0400673 XmmRegister inPlusPointFive = locations->GetTemp(0).AsFpuRegister<XmmRegister>();
Mark Mendell0c9497d2015-08-21 09:30:05 -0400674 NearLabel done, nan;
Mark Mendellfb8d2792015-03-31 22:16:59 -0400675 X86_64Assembler* assembler = GetAssembler();
676
Mark Mendell40741f32015-04-20 22:10:34 -0400677 // Load 0.5 into inPlusPointFive.
678 __ movsd(inPlusPointFive, codegen_->LiteralDoubleAddress(0.5));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400679
680 // Add in the input.
681 __ addsd(inPlusPointFive, in);
682
683 // And truncate to an integer.
684 __ roundsd(inPlusPointFive, inPlusPointFive, Immediate(1));
685
Pavel Vyssotski9ca25712015-07-31 13:03:17 +0600686 // Load maxLong into out.
687 codegen_->Load64BitValue(out, kPrimLongMax);
688
Mark Mendellfb8d2792015-03-31 22:16:59 -0400689 // if inPlusPointFive >= maxLong goto done
Mark Mendell40741f32015-04-20 22:10:34 -0400690 __ comisd(inPlusPointFive, codegen_->LiteralDoubleAddress(static_cast<double>(kPrimLongMax)));
Mark Mendellfb8d2792015-03-31 22:16:59 -0400691 __ j(kAboveEqual, &done);
692
693 // if input == NaN goto nan
694 __ j(kUnordered, &nan);
695
696 // output = double-to-long-truncate(input)
Roland Levillainbf84a3d2015-12-04 14:33:02 +0000697 __ cvttsd2si(out, inPlusPointFive, /* is64bit */ true);
Mark Mendellfb8d2792015-03-31 22:16:59 -0400698 __ jmp(&done);
699 __ Bind(&nan);
700
701 // output = 0
Mark Mendell92e83bf2015-05-07 11:25:03 -0400702 __ xorl(out, out);
Mark Mendellfb8d2792015-03-31 22:16:59 -0400703 __ Bind(&done);
704}
705
Mark Mendella4f12202015-08-06 15:23:34 -0400706static void CreateFPToFPCallLocations(ArenaAllocator* arena,
707 HInvoke* invoke) {
708 LocationSummary* locations = new (arena) LocationSummary(invoke,
709 LocationSummary::kCall,
710 kIntrinsified);
711 InvokeRuntimeCallingConvention calling_convention;
712 locations->SetInAt(0, Location::FpuRegisterLocation(calling_convention.GetFpuRegisterAt(0)));
713 locations->SetOut(Location::FpuRegisterLocation(XMM0));
714
715 // We have to ensure that the native code doesn't clobber the XMM registers which are
716 // non-volatile for ART, but volatile for Native calls. This will ensure that they are
717 // saved in the prologue and properly restored.
718 for (auto fp_reg : non_volatile_xmm_regs) {
719 locations->AddTemp(Location::FpuRegisterLocation(fp_reg));
720 }
721}
722
723static void GenFPToFPCall(HInvoke* invoke, CodeGeneratorX86_64* codegen,
724 QuickEntrypointEnum entry) {
725 LocationSummary* locations = invoke->GetLocations();
726 DCHECK(locations->WillCall());
727 DCHECK(invoke->IsInvokeStaticOrDirect());
728 X86_64Assembler* assembler = codegen->GetAssembler();
729
730 __ gs()->call(Address::Absolute(GetThreadOffset<kX86_64WordSize>(entry), true));
731 codegen->RecordPcInfo(invoke, invoke->GetDexPc());
732}
733
734void IntrinsicLocationsBuilderX86_64::VisitMathCos(HInvoke* invoke) {
735 CreateFPToFPCallLocations(arena_, invoke);
736}
737
738void IntrinsicCodeGeneratorX86_64::VisitMathCos(HInvoke* invoke) {
739 GenFPToFPCall(invoke, codegen_, kQuickCos);
740}
741
742void IntrinsicLocationsBuilderX86_64::VisitMathSin(HInvoke* invoke) {
743 CreateFPToFPCallLocations(arena_, invoke);
744}
745
746void IntrinsicCodeGeneratorX86_64::VisitMathSin(HInvoke* invoke) {
747 GenFPToFPCall(invoke, codegen_, kQuickSin);
748}
749
750void IntrinsicLocationsBuilderX86_64::VisitMathAcos(HInvoke* invoke) {
751 CreateFPToFPCallLocations(arena_, invoke);
752}
753
754void IntrinsicCodeGeneratorX86_64::VisitMathAcos(HInvoke* invoke) {
755 GenFPToFPCall(invoke, codegen_, kQuickAcos);
756}
757
758void IntrinsicLocationsBuilderX86_64::VisitMathAsin(HInvoke* invoke) {
759 CreateFPToFPCallLocations(arena_, invoke);
760}
761
762void IntrinsicCodeGeneratorX86_64::VisitMathAsin(HInvoke* invoke) {
763 GenFPToFPCall(invoke, codegen_, kQuickAsin);
764}
765
766void IntrinsicLocationsBuilderX86_64::VisitMathAtan(HInvoke* invoke) {
767 CreateFPToFPCallLocations(arena_, invoke);
768}
769
770void IntrinsicCodeGeneratorX86_64::VisitMathAtan(HInvoke* invoke) {
771 GenFPToFPCall(invoke, codegen_, kQuickAtan);
772}
773
774void IntrinsicLocationsBuilderX86_64::VisitMathCbrt(HInvoke* invoke) {
775 CreateFPToFPCallLocations(arena_, invoke);
776}
777
778void IntrinsicCodeGeneratorX86_64::VisitMathCbrt(HInvoke* invoke) {
779 GenFPToFPCall(invoke, codegen_, kQuickCbrt);
780}
781
782void IntrinsicLocationsBuilderX86_64::VisitMathCosh(HInvoke* invoke) {
783 CreateFPToFPCallLocations(arena_, invoke);
784}
785
786void IntrinsicCodeGeneratorX86_64::VisitMathCosh(HInvoke* invoke) {
787 GenFPToFPCall(invoke, codegen_, kQuickCosh);
788}
789
790void IntrinsicLocationsBuilderX86_64::VisitMathExp(HInvoke* invoke) {
791 CreateFPToFPCallLocations(arena_, invoke);
792}
793
794void IntrinsicCodeGeneratorX86_64::VisitMathExp(HInvoke* invoke) {
795 GenFPToFPCall(invoke, codegen_, kQuickExp);
796}
797
798void IntrinsicLocationsBuilderX86_64::VisitMathExpm1(HInvoke* invoke) {
799 CreateFPToFPCallLocations(arena_, invoke);
800}
801
802void IntrinsicCodeGeneratorX86_64::VisitMathExpm1(HInvoke* invoke) {
803 GenFPToFPCall(invoke, codegen_, kQuickExpm1);
804}
805
806void IntrinsicLocationsBuilderX86_64::VisitMathLog(HInvoke* invoke) {
807 CreateFPToFPCallLocations(arena_, invoke);
808}
809
810void IntrinsicCodeGeneratorX86_64::VisitMathLog(HInvoke* invoke) {
811 GenFPToFPCall(invoke, codegen_, kQuickLog);
812}
813
814void IntrinsicLocationsBuilderX86_64::VisitMathLog10(HInvoke* invoke) {
815 CreateFPToFPCallLocations(arena_, invoke);
816}
817
818void IntrinsicCodeGeneratorX86_64::VisitMathLog10(HInvoke* invoke) {
819 GenFPToFPCall(invoke, codegen_, kQuickLog10);
820}
821
822void IntrinsicLocationsBuilderX86_64::VisitMathSinh(HInvoke* invoke) {
823 CreateFPToFPCallLocations(arena_, invoke);
824}
825
826void IntrinsicCodeGeneratorX86_64::VisitMathSinh(HInvoke* invoke) {
827 GenFPToFPCall(invoke, codegen_, kQuickSinh);
828}
829
830void IntrinsicLocationsBuilderX86_64::VisitMathTan(HInvoke* invoke) {
831 CreateFPToFPCallLocations(arena_, invoke);
832}
833
834void IntrinsicCodeGeneratorX86_64::VisitMathTan(HInvoke* invoke) {
835 GenFPToFPCall(invoke, codegen_, kQuickTan);
836}
837
838void IntrinsicLocationsBuilderX86_64::VisitMathTanh(HInvoke* invoke) {
839 CreateFPToFPCallLocations(arena_, invoke);
840}
841
842void IntrinsicCodeGeneratorX86_64::VisitMathTanh(HInvoke* invoke) {
843 GenFPToFPCall(invoke, codegen_, kQuickTanh);
844}
845
846static void CreateFPFPToFPCallLocations(ArenaAllocator* arena,
847 HInvoke* invoke) {
848 LocationSummary* locations = new (arena) LocationSummary(invoke,
849 LocationSummary::kCall,
850 kIntrinsified);
851 InvokeRuntimeCallingConvention calling_convention;
852 locations->SetInAt(0, Location::FpuRegisterLocation(calling_convention.GetFpuRegisterAt(0)));
853 locations->SetInAt(1, Location::FpuRegisterLocation(calling_convention.GetFpuRegisterAt(1)));
854 locations->SetOut(Location::FpuRegisterLocation(XMM0));
855
856 // We have to ensure that the native code doesn't clobber the XMM registers which are
857 // non-volatile for ART, but volatile for Native calls. This will ensure that they are
858 // saved in the prologue and properly restored.
859 for (auto fp_reg : non_volatile_xmm_regs) {
860 locations->AddTemp(Location::FpuRegisterLocation(fp_reg));
861 }
862}
863
864void IntrinsicLocationsBuilderX86_64::VisitMathAtan2(HInvoke* invoke) {
865 CreateFPFPToFPCallLocations(arena_, invoke);
866}
867
868void IntrinsicCodeGeneratorX86_64::VisitMathAtan2(HInvoke* invoke) {
869 GenFPToFPCall(invoke, codegen_, kQuickAtan2);
870}
871
872void IntrinsicLocationsBuilderX86_64::VisitMathHypot(HInvoke* invoke) {
873 CreateFPFPToFPCallLocations(arena_, invoke);
874}
875
876void IntrinsicCodeGeneratorX86_64::VisitMathHypot(HInvoke* invoke) {
877 GenFPToFPCall(invoke, codegen_, kQuickHypot);
878}
879
880void IntrinsicLocationsBuilderX86_64::VisitMathNextAfter(HInvoke* invoke) {
881 CreateFPFPToFPCallLocations(arena_, invoke);
882}
883
884void IntrinsicCodeGeneratorX86_64::VisitMathNextAfter(HInvoke* invoke) {
885 GenFPToFPCall(invoke, codegen_, kQuickNextAfter);
886}
887
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800888void IntrinsicLocationsBuilderX86_64::VisitStringCharAt(HInvoke* invoke) {
889 // The inputs plus one temp.
890 LocationSummary* locations = new (arena_) LocationSummary(invoke,
891 LocationSummary::kCallOnSlowPath,
892 kIntrinsified);
893 locations->SetInAt(0, Location::RequiresRegister());
894 locations->SetInAt(1, Location::RequiresRegister());
895 locations->SetOut(Location::SameAsFirstInput());
896 locations->AddTemp(Location::RequiresRegister());
897}
898
899void IntrinsicCodeGeneratorX86_64::VisitStringCharAt(HInvoke* invoke) {
900 LocationSummary* locations = invoke->GetLocations();
901
Mark Mendell6bc53a92015-07-01 14:26:52 -0400902 // Location of reference to data array.
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800903 const int32_t value_offset = mirror::String::ValueOffset().Int32Value();
Mark Mendell6bc53a92015-07-01 14:26:52 -0400904 // Location of count.
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800905 const int32_t count_offset = mirror::String::CountOffset().Int32Value();
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800906
907 CpuRegister obj = locations->InAt(0).AsRegister<CpuRegister>();
908 CpuRegister idx = locations->InAt(1).AsRegister<CpuRegister>();
909 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800910
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800911 // TODO: Maybe we can support range check elimination. Overall, though, I think it's not worth
912 // the cost.
913 // TODO: For simplicity, the index parameter is requested in a register, so different from Quick
914 // we will not optimize the code for constants (which would save a register).
915
Andreas Gampe85b62f22015-09-09 13:15:38 -0700916 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800917 codegen_->AddSlowPath(slow_path);
918
919 X86_64Assembler* assembler = GetAssembler();
920
921 __ cmpl(idx, Address(obj, count_offset));
Andreas Gampe878d58c2015-01-15 23:24:00 -0800922 codegen_->MaybeRecordImplicitNullCheck(invoke);
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800923 __ j(kAboveEqual, slow_path->GetEntryLabel());
924
Jeff Hao848f70a2014-01-15 13:49:50 -0800925 // out = out[2*idx].
926 __ movzxw(out, Address(out, idx, ScaleFactor::TIMES_2, value_offset));
Andreas Gampe71fb52f2014-12-29 17:43:08 -0800927
928 __ Bind(slow_path->GetExitLabel());
929}
930
Mark Mendell6bc53a92015-07-01 14:26:52 -0400931void IntrinsicLocationsBuilderX86_64::VisitSystemArrayCopyChar(HInvoke* invoke) {
932 // Check to see if we have known failures that will cause us to have to bail out
933 // to the runtime, and just generate the runtime call directly.
934 HIntConstant* src_pos = invoke->InputAt(1)->AsIntConstant();
935 HIntConstant* dest_pos = invoke->InputAt(3)->AsIntConstant();
936
937 // The positions must be non-negative.
938 if ((src_pos != nullptr && src_pos->GetValue() < 0) ||
939 (dest_pos != nullptr && dest_pos->GetValue() < 0)) {
940 // We will have to fail anyways.
941 return;
942 }
943
944 // The length must be > 0.
945 HIntConstant* length = invoke->InputAt(4)->AsIntConstant();
946 if (length != nullptr) {
947 int32_t len = length->GetValue();
948 if (len < 0) {
949 // Just call as normal.
950 return;
951 }
952 }
953
954 LocationSummary* locations = new (arena_) LocationSummary(invoke,
955 LocationSummary::kCallOnSlowPath,
956 kIntrinsified);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +0100957 // arraycopy(Object src, int src_pos, Object dest, int dest_pos, int length).
Mark Mendell6bc53a92015-07-01 14:26:52 -0400958 locations->SetInAt(0, Location::RequiresRegister());
959 locations->SetInAt(1, Location::RegisterOrConstant(invoke->InputAt(1)));
960 locations->SetInAt(2, Location::RequiresRegister());
961 locations->SetInAt(3, Location::RegisterOrConstant(invoke->InputAt(3)));
962 locations->SetInAt(4, Location::RegisterOrConstant(invoke->InputAt(4)));
963
964 // And we need some temporaries. We will use REP MOVSW, so we need fixed registers.
965 locations->AddTemp(Location::RegisterLocation(RSI));
966 locations->AddTemp(Location::RegisterLocation(RDI));
967 locations->AddTemp(Location::RegisterLocation(RCX));
968}
969
970static void CheckPosition(X86_64Assembler* assembler,
971 Location pos,
972 CpuRegister input,
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +0100973 Location length,
Andreas Gampe85b62f22015-09-09 13:15:38 -0700974 SlowPathCode* slow_path,
Mark Mendell6bc53a92015-07-01 14:26:52 -0400975 CpuRegister input_len,
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +0100976 CpuRegister temp,
977 bool length_is_input_length = false) {
978 // Where is the length in the Array?
Mark Mendell6bc53a92015-07-01 14:26:52 -0400979 const uint32_t length_offset = mirror::Array::LengthOffset().Uint32Value();
980
981 if (pos.IsConstant()) {
982 int32_t pos_const = pos.GetConstant()->AsIntConstant()->GetValue();
983 if (pos_const == 0) {
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +0100984 if (!length_is_input_length) {
985 // Check that length(input) >= length.
986 if (length.IsConstant()) {
987 __ cmpl(Address(input, length_offset),
988 Immediate(length.GetConstant()->AsIntConstant()->GetValue()));
989 } else {
990 __ cmpl(Address(input, length_offset), length.AsRegister<CpuRegister>());
991 }
992 __ j(kLess, slow_path->GetEntryLabel());
993 }
Mark Mendell6bc53a92015-07-01 14:26:52 -0400994 } else {
995 // Check that length(input) >= pos.
996 __ movl(input_len, Address(input, length_offset));
997 __ cmpl(input_len, Immediate(pos_const));
998 __ j(kLess, slow_path->GetEntryLabel());
999
1000 // Check that (length(input) - pos) >= length.
1001 __ leal(temp, Address(input_len, -pos_const));
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001002 if (length.IsConstant()) {
1003 __ cmpl(temp, Immediate(length.GetConstant()->AsIntConstant()->GetValue()));
1004 } else {
1005 __ cmpl(temp, length.AsRegister<CpuRegister>());
1006 }
Mark Mendell6bc53a92015-07-01 14:26:52 -04001007 __ j(kLess, slow_path->GetEntryLabel());
1008 }
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001009 } else if (length_is_input_length) {
1010 // The only way the copy can succeed is if pos is zero.
1011 CpuRegister pos_reg = pos.AsRegister<CpuRegister>();
1012 __ testl(pos_reg, pos_reg);
1013 __ j(kNotEqual, slow_path->GetEntryLabel());
Mark Mendell6bc53a92015-07-01 14:26:52 -04001014 } else {
1015 // Check that pos >= 0.
1016 CpuRegister pos_reg = pos.AsRegister<CpuRegister>();
1017 __ testl(pos_reg, pos_reg);
1018 __ j(kLess, slow_path->GetEntryLabel());
1019
1020 // Check that pos <= length(input).
1021 __ cmpl(Address(input, length_offset), pos_reg);
1022 __ j(kLess, slow_path->GetEntryLabel());
1023
1024 // Check that (length(input) - pos) >= length.
1025 __ movl(temp, Address(input, length_offset));
1026 __ subl(temp, pos_reg);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001027 if (length.IsConstant()) {
1028 __ cmpl(temp, Immediate(length.GetConstant()->AsIntConstant()->GetValue()));
1029 } else {
1030 __ cmpl(temp, length.AsRegister<CpuRegister>());
1031 }
Mark Mendell6bc53a92015-07-01 14:26:52 -04001032 __ j(kLess, slow_path->GetEntryLabel());
1033 }
1034}
1035
1036void IntrinsicCodeGeneratorX86_64::VisitSystemArrayCopyChar(HInvoke* invoke) {
1037 X86_64Assembler* assembler = GetAssembler();
1038 LocationSummary* locations = invoke->GetLocations();
1039
1040 CpuRegister src = locations->InAt(0).AsRegister<CpuRegister>();
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001041 Location src_pos = locations->InAt(1);
Mark Mendell6bc53a92015-07-01 14:26:52 -04001042 CpuRegister dest = locations->InAt(2).AsRegister<CpuRegister>();
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001043 Location dest_pos = locations->InAt(3);
Mark Mendell6bc53a92015-07-01 14:26:52 -04001044 Location length = locations->InAt(4);
1045
1046 // Temporaries that we need for MOVSW.
1047 CpuRegister src_base = locations->GetTemp(0).AsRegister<CpuRegister>();
1048 DCHECK_EQ(src_base.AsRegister(), RSI);
1049 CpuRegister dest_base = locations->GetTemp(1).AsRegister<CpuRegister>();
1050 DCHECK_EQ(dest_base.AsRegister(), RDI);
1051 CpuRegister count = locations->GetTemp(2).AsRegister<CpuRegister>();
1052 DCHECK_EQ(count.AsRegister(), RCX);
1053
Andreas Gampe85b62f22015-09-09 13:15:38 -07001054 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Mark Mendell6bc53a92015-07-01 14:26:52 -04001055 codegen_->AddSlowPath(slow_path);
1056
1057 // Bail out if the source and destination are the same.
1058 __ cmpl(src, dest);
1059 __ j(kEqual, slow_path->GetEntryLabel());
1060
1061 // Bail out if the source is null.
1062 __ testl(src, src);
1063 __ j(kEqual, slow_path->GetEntryLabel());
1064
1065 // Bail out if the destination is null.
1066 __ testl(dest, dest);
1067 __ j(kEqual, slow_path->GetEntryLabel());
1068
1069 // If the length is negative, bail out.
1070 // We have already checked in the LocationsBuilder for the constant case.
1071 if (!length.IsConstant()) {
1072 __ testl(length.AsRegister<CpuRegister>(), length.AsRegister<CpuRegister>());
1073 __ j(kLess, slow_path->GetEntryLabel());
1074 }
1075
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001076 // Validity checks: source.
1077 CheckPosition(assembler, src_pos, src, length, slow_path, src_base, dest_base);
1078
1079 // Validity checks: dest.
1080 CheckPosition(assembler, dest_pos, dest, length, slow_path, src_base, dest_base);
1081
Mark Mendell6bc53a92015-07-01 14:26:52 -04001082 // We need the count in RCX.
1083 if (length.IsConstant()) {
1084 __ movl(count, Immediate(length.GetConstant()->AsIntConstant()->GetValue()));
1085 } else {
1086 __ movl(count, length.AsRegister<CpuRegister>());
1087 }
1088
Mark Mendell6bc53a92015-07-01 14:26:52 -04001089 // Okay, everything checks out. Finally time to do the copy.
1090 // Check assumption that sizeof(Char) is 2 (used in scaling below).
1091 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1092 DCHECK_EQ(char_size, 2u);
1093
1094 const uint32_t data_offset = mirror::Array::DataOffset(char_size).Uint32Value();
1095
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001096 if (src_pos.IsConstant()) {
1097 int32_t src_pos_const = src_pos.GetConstant()->AsIntConstant()->GetValue();
1098 __ leal(src_base, Address(src, char_size * src_pos_const + data_offset));
Mark Mendell6bc53a92015-07-01 14:26:52 -04001099 } else {
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001100 __ leal(src_base, Address(src, src_pos.AsRegister<CpuRegister>(),
Mark Mendell6bc53a92015-07-01 14:26:52 -04001101 ScaleFactor::TIMES_2, data_offset));
1102 }
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001103 if (dest_pos.IsConstant()) {
1104 int32_t dest_pos_const = dest_pos.GetConstant()->AsIntConstant()->GetValue();
1105 __ leal(dest_base, Address(dest, char_size * dest_pos_const + data_offset));
Mark Mendell6bc53a92015-07-01 14:26:52 -04001106 } else {
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001107 __ leal(dest_base, Address(dest, dest_pos.AsRegister<CpuRegister>(),
Mark Mendell6bc53a92015-07-01 14:26:52 -04001108 ScaleFactor::TIMES_2, data_offset));
1109 }
1110
1111 // Do the move.
1112 __ rep_movsw();
1113
1114 __ Bind(slow_path->GetExitLabel());
1115}
1116
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001117
1118void IntrinsicLocationsBuilderX86_64::VisitSystemArrayCopy(HInvoke* invoke) {
Nicolas Geoffray5bd05a52015-10-13 09:48:30 +01001119 CodeGenerator::CreateSystemArrayCopyLocationSummary(invoke);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001120}
1121
Roland Levillain0d5a2812015-11-13 10:07:31 +00001122// TODO: Implement read barriers in the SystemArrayCopy intrinsic.
1123// Note that this code path is not used (yet) because we do not
1124// intrinsify methods that can go into the IntrinsicSlowPathX86_64
1125// slow path.
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001126void IntrinsicCodeGeneratorX86_64::VisitSystemArrayCopy(HInvoke* invoke) {
1127 X86_64Assembler* assembler = GetAssembler();
1128 LocationSummary* locations = invoke->GetLocations();
1129
1130 uint32_t class_offset = mirror::Object::ClassOffset().Int32Value();
1131 uint32_t super_offset = mirror::Class::SuperClassOffset().Int32Value();
1132 uint32_t component_offset = mirror::Class::ComponentTypeOffset().Int32Value();
1133 uint32_t primitive_offset = mirror::Class::PrimitiveTypeOffset().Int32Value();
1134
1135 CpuRegister src = locations->InAt(0).AsRegister<CpuRegister>();
1136 Location src_pos = locations->InAt(1);
1137 CpuRegister dest = locations->InAt(2).AsRegister<CpuRegister>();
1138 Location dest_pos = locations->InAt(3);
1139 Location length = locations->InAt(4);
1140 CpuRegister temp1 = locations->GetTemp(0).AsRegister<CpuRegister>();
1141 CpuRegister temp2 = locations->GetTemp(1).AsRegister<CpuRegister>();
1142 CpuRegister temp3 = locations->GetTemp(2).AsRegister<CpuRegister>();
1143
1144 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
1145 codegen_->AddSlowPath(slow_path);
1146
1147 NearLabel ok;
1148 SystemArrayCopyOptimizations optimizations(invoke);
1149
1150 if (!optimizations.GetDestinationIsSource()) {
Nicolas Geoffray5bd05a52015-10-13 09:48:30 +01001151 if (!src_pos.IsConstant() || !dest_pos.IsConstant()) {
1152 __ cmpl(src, dest);
1153 }
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001154 }
1155
1156 // If source and destination are the same, we go to slow path if we need to do
1157 // forward copying.
1158 if (src_pos.IsConstant()) {
1159 int32_t src_pos_constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
1160 if (dest_pos.IsConstant()) {
1161 // Checked when building locations.
1162 DCHECK(!optimizations.GetDestinationIsSource()
1163 || (src_pos_constant >= dest_pos.GetConstant()->AsIntConstant()->GetValue()));
1164 } else {
1165 if (!optimizations.GetDestinationIsSource()) {
1166 __ j(kNotEqual, &ok);
1167 }
1168 __ cmpl(dest_pos.AsRegister<CpuRegister>(), Immediate(src_pos_constant));
1169 __ j(kGreater, slow_path->GetEntryLabel());
1170 }
1171 } else {
1172 if (!optimizations.GetDestinationIsSource()) {
1173 __ j(kNotEqual, &ok);
1174 }
1175 if (dest_pos.IsConstant()) {
1176 int32_t dest_pos_constant = dest_pos.GetConstant()->AsIntConstant()->GetValue();
1177 __ cmpl(src_pos.AsRegister<CpuRegister>(), Immediate(dest_pos_constant));
1178 __ j(kLess, slow_path->GetEntryLabel());
1179 } else {
1180 __ cmpl(src_pos.AsRegister<CpuRegister>(), dest_pos.AsRegister<CpuRegister>());
1181 __ j(kLess, slow_path->GetEntryLabel());
1182 }
1183 }
1184
1185 __ Bind(&ok);
1186
1187 if (!optimizations.GetSourceIsNotNull()) {
1188 // Bail out if the source is null.
1189 __ testl(src, src);
1190 __ j(kEqual, slow_path->GetEntryLabel());
1191 }
1192
1193 if (!optimizations.GetDestinationIsNotNull() && !optimizations.GetDestinationIsSource()) {
1194 // Bail out if the destination is null.
1195 __ testl(dest, dest);
1196 __ j(kEqual, slow_path->GetEntryLabel());
1197 }
1198
1199 // If the length is negative, bail out.
1200 // We have already checked in the LocationsBuilder for the constant case.
1201 if (!length.IsConstant() &&
1202 !optimizations.GetCountIsSourceLength() &&
1203 !optimizations.GetCountIsDestinationLength()) {
1204 __ testl(length.AsRegister<CpuRegister>(), length.AsRegister<CpuRegister>());
1205 __ j(kLess, slow_path->GetEntryLabel());
1206 }
1207
1208 // Validity checks: source.
1209 CheckPosition(assembler,
1210 src_pos,
1211 src,
1212 length,
1213 slow_path,
1214 temp1,
1215 temp2,
1216 optimizations.GetCountIsSourceLength());
1217
1218 // Validity checks: dest.
1219 CheckPosition(assembler,
1220 dest_pos,
1221 dest,
1222 length,
1223 slow_path,
1224 temp1,
1225 temp2,
1226 optimizations.GetCountIsDestinationLength());
1227
1228 if (!optimizations.GetDoesNotNeedTypeCheck()) {
1229 // Check whether all elements of the source array are assignable to the component
1230 // type of the destination array. We do two checks: the classes are the same,
1231 // or the destination is Object[]. If none of these checks succeed, we go to the
1232 // slow path.
1233 __ movl(temp1, Address(dest, class_offset));
1234 __ movl(temp2, Address(src, class_offset));
1235 bool did_unpoison = false;
1236 if (!optimizations.GetDestinationIsNonPrimitiveArray() ||
1237 !optimizations.GetSourceIsNonPrimitiveArray()) {
1238 // One or two of the references need to be unpoisoned. Unpoisoned them
1239 // both to make the identity check valid.
1240 __ MaybeUnpoisonHeapReference(temp1);
1241 __ MaybeUnpoisonHeapReference(temp2);
1242 did_unpoison = true;
1243 }
1244
1245 if (!optimizations.GetDestinationIsNonPrimitiveArray()) {
1246 // Bail out if the destination is not a non primitive array.
1247 __ movl(CpuRegister(TMP), Address(temp1, component_offset));
1248 __ testl(CpuRegister(TMP), CpuRegister(TMP));
1249 __ j(kEqual, slow_path->GetEntryLabel());
1250 __ MaybeUnpoisonHeapReference(CpuRegister(TMP));
1251 __ cmpw(Address(CpuRegister(TMP), primitive_offset), Immediate(Primitive::kPrimNot));
1252 __ j(kNotEqual, slow_path->GetEntryLabel());
1253 }
1254
1255 if (!optimizations.GetSourceIsNonPrimitiveArray()) {
1256 // Bail out if the source is not a non primitive array.
1257 __ movl(CpuRegister(TMP), Address(temp2, component_offset));
1258 __ testl(CpuRegister(TMP), CpuRegister(TMP));
1259 __ j(kEqual, slow_path->GetEntryLabel());
1260 __ MaybeUnpoisonHeapReference(CpuRegister(TMP));
1261 __ cmpw(Address(CpuRegister(TMP), primitive_offset), Immediate(Primitive::kPrimNot));
1262 __ j(kNotEqual, slow_path->GetEntryLabel());
1263 }
1264
1265 __ cmpl(temp1, temp2);
1266
1267 if (optimizations.GetDestinationIsTypedObjectArray()) {
1268 NearLabel do_copy;
1269 __ j(kEqual, &do_copy);
1270 if (!did_unpoison) {
1271 __ MaybeUnpoisonHeapReference(temp1);
1272 }
1273 __ movl(temp1, Address(temp1, component_offset));
1274 __ MaybeUnpoisonHeapReference(temp1);
1275 __ movl(temp1, Address(temp1, super_offset));
1276 // No need to unpoison the result, we're comparing against null.
1277 __ testl(temp1, temp1);
1278 __ j(kNotEqual, slow_path->GetEntryLabel());
1279 __ Bind(&do_copy);
1280 } else {
1281 __ j(kNotEqual, slow_path->GetEntryLabel());
1282 }
1283 } else if (!optimizations.GetSourceIsNonPrimitiveArray()) {
1284 DCHECK(optimizations.GetDestinationIsNonPrimitiveArray());
1285 // Bail out if the source is not a non primitive array.
1286 __ movl(temp1, Address(src, class_offset));
1287 __ MaybeUnpoisonHeapReference(temp1);
1288 __ movl(CpuRegister(TMP), Address(temp1, component_offset));
1289 __ testl(CpuRegister(TMP), CpuRegister(TMP));
1290 __ j(kEqual, slow_path->GetEntryLabel());
1291 __ MaybeUnpoisonHeapReference(CpuRegister(TMP));
1292 __ cmpw(Address(CpuRegister(TMP), primitive_offset), Immediate(Primitive::kPrimNot));
1293 __ j(kNotEqual, slow_path->GetEntryLabel());
1294 }
1295
1296 // Compute base source address, base destination address, and end source address.
1297
1298 uint32_t element_size = sizeof(int32_t);
1299 uint32_t offset = mirror::Array::DataOffset(element_size).Uint32Value();
1300 if (src_pos.IsConstant()) {
1301 int32_t constant = src_pos.GetConstant()->AsIntConstant()->GetValue();
1302 __ leal(temp1, Address(src, element_size * constant + offset));
1303 } else {
1304 __ leal(temp1, Address(src, src_pos.AsRegister<CpuRegister>(), ScaleFactor::TIMES_4, offset));
1305 }
1306
1307 if (dest_pos.IsConstant()) {
1308 int32_t constant = dest_pos.GetConstant()->AsIntConstant()->GetValue();
1309 __ leal(temp2, Address(dest, element_size * constant + offset));
1310 } else {
1311 __ leal(temp2, Address(dest, dest_pos.AsRegister<CpuRegister>(), ScaleFactor::TIMES_4, offset));
1312 }
1313
1314 if (length.IsConstant()) {
1315 int32_t constant = length.GetConstant()->AsIntConstant()->GetValue();
1316 __ leal(temp3, Address(temp1, element_size * constant));
1317 } else {
1318 __ leal(temp3, Address(temp1, length.AsRegister<CpuRegister>(), ScaleFactor::TIMES_4, 0));
1319 }
1320
1321 // Iterate over the arrays and do a raw copy of the objects. We don't need to
1322 // poison/unpoison, nor do any read barrier as the next uses of the destination
1323 // array will do it.
1324 NearLabel loop, done;
1325 __ cmpl(temp1, temp3);
1326 __ j(kEqual, &done);
1327 __ Bind(&loop);
1328 __ movl(CpuRegister(TMP), Address(temp1, 0));
1329 __ movl(Address(temp2, 0), CpuRegister(TMP));
1330 __ addl(temp1, Immediate(element_size));
1331 __ addl(temp2, Immediate(element_size));
1332 __ cmpl(temp1, temp3);
1333 __ j(kNotEqual, &loop);
1334 __ Bind(&done);
1335
1336 // We only need one card marking on the destination array.
1337 codegen_->MarkGCCard(temp1,
1338 temp2,
1339 dest,
1340 CpuRegister(kNoRegister),
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001341 /* value_can_be_null */ false);
Nicolas Geoffrayee3cf072015-10-06 11:45:02 +01001342
1343 __ Bind(slow_path->GetExitLabel());
1344}
1345
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001346void IntrinsicLocationsBuilderX86_64::VisitStringCompareTo(HInvoke* invoke) {
1347 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1348 LocationSummary::kCall,
1349 kIntrinsified);
1350 InvokeRuntimeCallingConvention calling_convention;
1351 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetRegisterAt(0)));
1352 locations->SetInAt(1, Location::RegisterLocation(calling_convention.GetRegisterAt(1)));
1353 locations->SetOut(Location::RegisterLocation(RAX));
1354}
1355
1356void IntrinsicCodeGeneratorX86_64::VisitStringCompareTo(HInvoke* invoke) {
1357 X86_64Assembler* assembler = GetAssembler();
1358 LocationSummary* locations = invoke->GetLocations();
1359
Nicolas Geoffray512e04d2015-03-27 17:21:24 +00001360 // Note that the null check must have been done earlier.
Calin Juravle641547a2015-04-21 22:08:51 +01001361 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001362
1363 CpuRegister argument = locations->InAt(1).AsRegister<CpuRegister>();
1364 __ testl(argument, argument);
Andreas Gampe85b62f22015-09-09 13:15:38 -07001365 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001366 codegen_->AddSlowPath(slow_path);
1367 __ j(kEqual, slow_path->GetEntryLabel());
1368
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001369 __ gs()->call(Address::Absolute(QUICK_ENTRYPOINT_OFFSET(kX86_64WordSize, pStringCompareTo),
1370 /* no_rip */ true));
Nicolas Geoffrayd75948a2015-03-27 09:53:16 +00001371 __ Bind(slow_path->GetExitLabel());
1372}
1373
Agi Csakif8cfb202015-08-13 17:54:54 -07001374void IntrinsicLocationsBuilderX86_64::VisitStringEquals(HInvoke* invoke) {
1375 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1376 LocationSummary::kNoCall,
1377 kIntrinsified);
1378 locations->SetInAt(0, Location::RequiresRegister());
1379 locations->SetInAt(1, Location::RequiresRegister());
1380
1381 // Request temporary registers, RCX and RDI needed for repe_cmpsq instruction.
1382 locations->AddTemp(Location::RegisterLocation(RCX));
1383 locations->AddTemp(Location::RegisterLocation(RDI));
1384
1385 // Set output, RSI needed for repe_cmpsq instruction anyways.
1386 locations->SetOut(Location::RegisterLocation(RSI), Location::kOutputOverlap);
1387}
1388
1389void IntrinsicCodeGeneratorX86_64::VisitStringEquals(HInvoke* invoke) {
1390 X86_64Assembler* assembler = GetAssembler();
1391 LocationSummary* locations = invoke->GetLocations();
1392
1393 CpuRegister str = locations->InAt(0).AsRegister<CpuRegister>();
1394 CpuRegister arg = locations->InAt(1).AsRegister<CpuRegister>();
1395 CpuRegister rcx = locations->GetTemp(0).AsRegister<CpuRegister>();
1396 CpuRegister rdi = locations->GetTemp(1).AsRegister<CpuRegister>();
1397 CpuRegister rsi = locations->Out().AsRegister<CpuRegister>();
1398
Mark Mendell0c9497d2015-08-21 09:30:05 -04001399 NearLabel end, return_true, return_false;
Agi Csakif8cfb202015-08-13 17:54:54 -07001400
1401 // Get offsets of count, value, and class fields within a string object.
1402 const uint32_t count_offset = mirror::String::CountOffset().Uint32Value();
1403 const uint32_t value_offset = mirror::String::ValueOffset().Uint32Value();
1404 const uint32_t class_offset = mirror::Object::ClassOffset().Uint32Value();
1405
1406 // Note that the null check must have been done earlier.
1407 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1408
1409 // Check if input is null, return false if it is.
1410 __ testl(arg, arg);
1411 __ j(kEqual, &return_false);
1412
1413 // Instanceof check for the argument by comparing class fields.
1414 // All string objects must have the same type since String cannot be subclassed.
1415 // Receiver must be a string object, so its class field is equal to all strings' class fields.
1416 // If the argument is a string object, its class field must be equal to receiver's class field.
1417 __ movl(rcx, Address(str, class_offset));
1418 __ cmpl(rcx, Address(arg, class_offset));
1419 __ j(kNotEqual, &return_false);
1420
1421 // Reference equality check, return true if same reference.
1422 __ cmpl(str, arg);
1423 __ j(kEqual, &return_true);
1424
1425 // Load length of receiver string.
1426 __ movl(rcx, Address(str, count_offset));
1427 // Check if lengths are equal, return false if they're not.
1428 __ cmpl(rcx, Address(arg, count_offset));
1429 __ j(kNotEqual, &return_false);
1430 // Return true if both strings are empty.
Mark Mendell0c9497d2015-08-21 09:30:05 -04001431 __ jrcxz(&return_true);
Agi Csakif8cfb202015-08-13 17:54:54 -07001432
1433 // Load starting addresses of string values into RSI/RDI as required for repe_cmpsq instruction.
1434 __ leal(rsi, Address(str, value_offset));
1435 __ leal(rdi, Address(arg, value_offset));
1436
1437 // Divide string length by 4 and adjust for lengths not divisible by 4.
1438 __ addl(rcx, Immediate(3));
1439 __ shrl(rcx, Immediate(2));
1440
1441 // Assertions that must hold in order to compare strings 4 characters at a time.
1442 DCHECK_ALIGNED(value_offset, 8);
1443 static_assert(IsAligned<8>(kObjectAlignment), "String is not zero padded");
1444
1445 // Loop to compare strings four characters at a time starting at the beginning of the string.
1446 __ repe_cmpsq();
1447 // If strings are not equal, zero flag will be cleared.
1448 __ j(kNotEqual, &return_false);
1449
1450 // Return true and exit the function.
1451 // If loop does not result in returning false, we return true.
1452 __ Bind(&return_true);
1453 __ movl(rsi, Immediate(1));
1454 __ jmp(&end);
1455
1456 // Return false and exit the function.
1457 __ Bind(&return_false);
1458 __ xorl(rsi, rsi);
1459 __ Bind(&end);
1460}
1461
Andreas Gampe21030dd2015-05-07 14:46:15 -07001462static void CreateStringIndexOfLocations(HInvoke* invoke,
1463 ArenaAllocator* allocator,
1464 bool start_at_zero) {
1465 LocationSummary* locations = new (allocator) LocationSummary(invoke,
1466 LocationSummary::kCallOnSlowPath,
1467 kIntrinsified);
1468 // The data needs to be in RDI for scasw. So request that the string is there, anyways.
1469 locations->SetInAt(0, Location::RegisterLocation(RDI));
1470 // If we look for a constant char, we'll still have to copy it into RAX. So just request the
1471 // allocator to do that, anyways. We can still do the constant check by checking the parameter
1472 // of the instruction explicitly.
1473 // Note: This works as we don't clobber RAX anywhere.
1474 locations->SetInAt(1, Location::RegisterLocation(RAX));
1475 if (!start_at_zero) {
1476 locations->SetInAt(2, Location::RequiresRegister()); // The starting index.
1477 }
1478 // As we clobber RDI during execution anyways, also use it as the output.
1479 locations->SetOut(Location::SameAsFirstInput());
1480
1481 // repne scasw uses RCX as the counter.
1482 locations->AddTemp(Location::RegisterLocation(RCX));
1483 // Need another temporary to be able to compute the result.
1484 locations->AddTemp(Location::RequiresRegister());
1485}
1486
1487static void GenerateStringIndexOf(HInvoke* invoke,
1488 X86_64Assembler* assembler,
1489 CodeGeneratorX86_64* codegen,
1490 ArenaAllocator* allocator,
1491 bool start_at_zero) {
1492 LocationSummary* locations = invoke->GetLocations();
1493
1494 // Note that the null check must have been done earlier.
1495 DCHECK(!invoke->CanDoImplicitNullCheckOn(invoke->InputAt(0)));
1496
1497 CpuRegister string_obj = locations->InAt(0).AsRegister<CpuRegister>();
1498 CpuRegister search_value = locations->InAt(1).AsRegister<CpuRegister>();
1499 CpuRegister counter = locations->GetTemp(0).AsRegister<CpuRegister>();
1500 CpuRegister string_length = locations->GetTemp(1).AsRegister<CpuRegister>();
1501 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
1502
1503 // Check our assumptions for registers.
1504 DCHECK_EQ(string_obj.AsRegister(), RDI);
1505 DCHECK_EQ(search_value.AsRegister(), RAX);
1506 DCHECK_EQ(counter.AsRegister(), RCX);
1507 DCHECK_EQ(out.AsRegister(), RDI);
1508
1509 // Check for code points > 0xFFFF. Either a slow-path check when we don't know statically,
1510 // or directly dispatch if we have a constant.
Andreas Gampe85b62f22015-09-09 13:15:38 -07001511 SlowPathCode* slow_path = nullptr;
Andreas Gampe21030dd2015-05-07 14:46:15 -07001512 if (invoke->InputAt(1)->IsIntConstant()) {
1513 if (static_cast<uint32_t>(invoke->InputAt(1)->AsIntConstant()->GetValue()) >
1514 std::numeric_limits<uint16_t>::max()) {
1515 // Always needs the slow-path. We could directly dispatch to it, but this case should be
1516 // rare, so for simplicity just put the full slow-path down and branch unconditionally.
1517 slow_path = new (allocator) IntrinsicSlowPathX86_64(invoke);
1518 codegen->AddSlowPath(slow_path);
1519 __ jmp(slow_path->GetEntryLabel());
1520 __ Bind(slow_path->GetExitLabel());
1521 return;
1522 }
1523 } else {
1524 __ cmpl(search_value, Immediate(std::numeric_limits<uint16_t>::max()));
1525 slow_path = new (allocator) IntrinsicSlowPathX86_64(invoke);
1526 codegen->AddSlowPath(slow_path);
1527 __ j(kAbove, slow_path->GetEntryLabel());
1528 }
1529
1530 // From here down, we know that we are looking for a char that fits in 16 bits.
1531 // Location of reference to data array within the String object.
1532 int32_t value_offset = mirror::String::ValueOffset().Int32Value();
1533 // Location of count within the String object.
1534 int32_t count_offset = mirror::String::CountOffset().Int32Value();
1535
1536 // Load string length, i.e., the count field of the string.
1537 __ movl(string_length, Address(string_obj, count_offset));
1538
1539 // Do a length check.
1540 // TODO: Support jecxz.
Mark Mendell0c9497d2015-08-21 09:30:05 -04001541 NearLabel not_found_label;
Andreas Gampe21030dd2015-05-07 14:46:15 -07001542 __ testl(string_length, string_length);
1543 __ j(kEqual, &not_found_label);
1544
1545 if (start_at_zero) {
1546 // Number of chars to scan is the same as the string length.
1547 __ movl(counter, string_length);
1548
1549 // Move to the start of the string.
1550 __ addq(string_obj, Immediate(value_offset));
1551 } else {
1552 CpuRegister start_index = locations->InAt(2).AsRegister<CpuRegister>();
1553
1554 // Do a start_index check.
1555 __ cmpl(start_index, string_length);
1556 __ j(kGreaterEqual, &not_found_label);
1557
1558 // Ensure we have a start index >= 0;
1559 __ xorl(counter, counter);
1560 __ cmpl(start_index, Immediate(0));
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001561 __ cmov(kGreater, counter, start_index, /* is64bit */ false); // 32-bit copy is enough.
Andreas Gampe21030dd2015-05-07 14:46:15 -07001562
1563 // Move to the start of the string: string_obj + value_offset + 2 * start_index.
1564 __ leaq(string_obj, Address(string_obj, counter, ScaleFactor::TIMES_2, value_offset));
1565
1566 // Now update ecx, the work counter: it's gonna be string.length - start_index.
1567 __ negq(counter); // Needs to be 64-bit negation, as the address computation is 64-bit.
1568 __ leaq(counter, Address(string_length, counter, ScaleFactor::TIMES_1, 0));
1569 }
1570
1571 // Everything is set up for repne scasw:
1572 // * Comparison address in RDI.
1573 // * Counter in ECX.
1574 __ repne_scasw();
1575
1576 // Did we find a match?
1577 __ j(kNotEqual, &not_found_label);
1578
1579 // Yes, we matched. Compute the index of the result.
1580 __ subl(string_length, counter);
1581 __ leal(out, Address(string_length, -1));
1582
Mark Mendell0c9497d2015-08-21 09:30:05 -04001583 NearLabel done;
Andreas Gampe21030dd2015-05-07 14:46:15 -07001584 __ jmp(&done);
1585
1586 // Failed to match; return -1.
1587 __ Bind(&not_found_label);
1588 __ movl(out, Immediate(-1));
1589
1590 // And join up at the end.
1591 __ Bind(&done);
1592 if (slow_path != nullptr) {
1593 __ Bind(slow_path->GetExitLabel());
1594 }
1595}
1596
1597void IntrinsicLocationsBuilderX86_64::VisitStringIndexOf(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001598 CreateStringIndexOfLocations(invoke, arena_, /* start_at_zero */ true);
Andreas Gampe21030dd2015-05-07 14:46:15 -07001599}
1600
1601void IntrinsicCodeGeneratorX86_64::VisitStringIndexOf(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001602 GenerateStringIndexOf(invoke, GetAssembler(), codegen_, GetAllocator(), /* start_at_zero */ true);
Andreas Gampe21030dd2015-05-07 14:46:15 -07001603}
1604
1605void IntrinsicLocationsBuilderX86_64::VisitStringIndexOfAfter(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001606 CreateStringIndexOfLocations(invoke, arena_, /* start_at_zero */ false);
Andreas Gampe21030dd2015-05-07 14:46:15 -07001607}
1608
1609void IntrinsicCodeGeneratorX86_64::VisitStringIndexOfAfter(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001610 GenerateStringIndexOf(
1611 invoke, GetAssembler(), codegen_, GetAllocator(), /* start_at_zero */ false);
Andreas Gampe21030dd2015-05-07 14:46:15 -07001612}
1613
Jeff Hao848f70a2014-01-15 13:49:50 -08001614void IntrinsicLocationsBuilderX86_64::VisitStringNewStringFromBytes(HInvoke* invoke) {
1615 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1616 LocationSummary::kCall,
1617 kIntrinsified);
1618 InvokeRuntimeCallingConvention calling_convention;
1619 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetRegisterAt(0)));
1620 locations->SetInAt(1, Location::RegisterLocation(calling_convention.GetRegisterAt(1)));
1621 locations->SetInAt(2, Location::RegisterLocation(calling_convention.GetRegisterAt(2)));
1622 locations->SetInAt(3, Location::RegisterLocation(calling_convention.GetRegisterAt(3)));
1623 locations->SetOut(Location::RegisterLocation(RAX));
1624}
1625
1626void IntrinsicCodeGeneratorX86_64::VisitStringNewStringFromBytes(HInvoke* invoke) {
1627 X86_64Assembler* assembler = GetAssembler();
1628 LocationSummary* locations = invoke->GetLocations();
1629
1630 CpuRegister byte_array = locations->InAt(0).AsRegister<CpuRegister>();
1631 __ testl(byte_array, byte_array);
Andreas Gampe85b62f22015-09-09 13:15:38 -07001632 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Jeff Hao848f70a2014-01-15 13:49:50 -08001633 codegen_->AddSlowPath(slow_path);
1634 __ j(kEqual, slow_path->GetEntryLabel());
1635
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001636 __ gs()->call(Address::Absolute(QUICK_ENTRYPOINT_OFFSET(kX86_64WordSize, pAllocStringFromBytes),
1637 /* no_rip */ true));
Jeff Hao848f70a2014-01-15 13:49:50 -08001638 codegen_->RecordPcInfo(invoke, invoke->GetDexPc());
1639 __ Bind(slow_path->GetExitLabel());
1640}
1641
1642void IntrinsicLocationsBuilderX86_64::VisitStringNewStringFromChars(HInvoke* invoke) {
1643 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1644 LocationSummary::kCall,
1645 kIntrinsified);
1646 InvokeRuntimeCallingConvention calling_convention;
1647 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetRegisterAt(0)));
1648 locations->SetInAt(1, Location::RegisterLocation(calling_convention.GetRegisterAt(1)));
1649 locations->SetInAt(2, Location::RegisterLocation(calling_convention.GetRegisterAt(2)));
1650 locations->SetOut(Location::RegisterLocation(RAX));
1651}
1652
1653void IntrinsicCodeGeneratorX86_64::VisitStringNewStringFromChars(HInvoke* invoke) {
1654 X86_64Assembler* assembler = GetAssembler();
1655
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001656 __ gs()->call(Address::Absolute(QUICK_ENTRYPOINT_OFFSET(kX86_64WordSize, pAllocStringFromChars),
1657 /* no_rip */ true));
Jeff Hao848f70a2014-01-15 13:49:50 -08001658 codegen_->RecordPcInfo(invoke, invoke->GetDexPc());
1659}
1660
1661void IntrinsicLocationsBuilderX86_64::VisitStringNewStringFromString(HInvoke* invoke) {
1662 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1663 LocationSummary::kCall,
1664 kIntrinsified);
1665 InvokeRuntimeCallingConvention calling_convention;
1666 locations->SetInAt(0, Location::RegisterLocation(calling_convention.GetRegisterAt(0)));
1667 locations->SetOut(Location::RegisterLocation(RAX));
1668}
1669
1670void IntrinsicCodeGeneratorX86_64::VisitStringNewStringFromString(HInvoke* invoke) {
1671 X86_64Assembler* assembler = GetAssembler();
1672 LocationSummary* locations = invoke->GetLocations();
1673
1674 CpuRegister string_to_copy = locations->InAt(0).AsRegister<CpuRegister>();
1675 __ testl(string_to_copy, string_to_copy);
Andreas Gampe85b62f22015-09-09 13:15:38 -07001676 SlowPathCode* slow_path = new (GetAllocator()) IntrinsicSlowPathX86_64(invoke);
Jeff Hao848f70a2014-01-15 13:49:50 -08001677 codegen_->AddSlowPath(slow_path);
1678 __ j(kEqual, slow_path->GetEntryLabel());
1679
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001680 __ gs()->call(Address::Absolute(QUICK_ENTRYPOINT_OFFSET(kX86_64WordSize, pAllocStringFromString),
1681 /* no_rip */ true));
Jeff Hao848f70a2014-01-15 13:49:50 -08001682 codegen_->RecordPcInfo(invoke, invoke->GetDexPc());
1683 __ Bind(slow_path->GetExitLabel());
1684}
1685
Mark Mendell8f8926a2015-08-17 11:39:06 -04001686void IntrinsicLocationsBuilderX86_64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
1687 // public void getChars(int srcBegin, int srcEnd, char[] dst, int dstBegin);
1688 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1689 LocationSummary::kNoCall,
1690 kIntrinsified);
1691 locations->SetInAt(0, Location::RequiresRegister());
1692 locations->SetInAt(1, Location::RegisterOrConstant(invoke->InputAt(1)));
1693 locations->SetInAt(2, Location::RequiresRegister());
1694 locations->SetInAt(3, Location::RequiresRegister());
1695 locations->SetInAt(4, Location::RequiresRegister());
1696
1697 // And we need some temporaries. We will use REP MOVSW, so we need fixed registers.
1698 locations->AddTemp(Location::RegisterLocation(RSI));
1699 locations->AddTemp(Location::RegisterLocation(RDI));
1700 locations->AddTemp(Location::RegisterLocation(RCX));
1701}
1702
1703void IntrinsicCodeGeneratorX86_64::VisitStringGetCharsNoCheck(HInvoke* invoke) {
1704 X86_64Assembler* assembler = GetAssembler();
1705 LocationSummary* locations = invoke->GetLocations();
1706
1707 size_t char_component_size = Primitive::ComponentSize(Primitive::kPrimChar);
1708 // Location of data in char array buffer.
1709 const uint32_t data_offset = mirror::Array::DataOffset(char_component_size).Uint32Value();
1710 // Location of char array data in string.
1711 const uint32_t value_offset = mirror::String::ValueOffset().Uint32Value();
1712
1713 // public void getChars(int srcBegin, int srcEnd, char[] dst, int dstBegin);
1714 CpuRegister obj = locations->InAt(0).AsRegister<CpuRegister>();
1715 Location srcBegin = locations->InAt(1);
1716 int srcBegin_value =
1717 srcBegin.IsConstant() ? srcBegin.GetConstant()->AsIntConstant()->GetValue() : 0;
1718 CpuRegister srcEnd = locations->InAt(2).AsRegister<CpuRegister>();
1719 CpuRegister dst = locations->InAt(3).AsRegister<CpuRegister>();
1720 CpuRegister dstBegin = locations->InAt(4).AsRegister<CpuRegister>();
1721
1722 // Check assumption that sizeof(Char) is 2 (used in scaling below).
1723 const size_t char_size = Primitive::ComponentSize(Primitive::kPrimChar);
1724 DCHECK_EQ(char_size, 2u);
1725
1726 // Compute the address of the destination buffer.
1727 __ leaq(CpuRegister(RDI), Address(dst, dstBegin, ScaleFactor::TIMES_2, data_offset));
1728
1729 // Compute the address of the source string.
1730 if (srcBegin.IsConstant()) {
1731 // Compute the address of the source string by adding the number of chars from
1732 // the source beginning to the value offset of a string.
1733 __ leaq(CpuRegister(RSI), Address(obj, srcBegin_value * char_size + value_offset));
1734 } else {
1735 __ leaq(CpuRegister(RSI), Address(obj, srcBegin.AsRegister<CpuRegister>(),
1736 ScaleFactor::TIMES_2, value_offset));
1737 }
1738
1739 // Compute the number of chars (words) to move.
1740 __ movl(CpuRegister(RCX), srcEnd);
1741 if (srcBegin.IsConstant()) {
1742 if (srcBegin_value != 0) {
1743 __ subl(CpuRegister(RCX), Immediate(srcBegin_value));
1744 }
1745 } else {
1746 DCHECK(srcBegin.IsRegister());
1747 __ subl(CpuRegister(RCX), srcBegin.AsRegister<CpuRegister>());
1748 }
1749
1750 // Do the move.
1751 __ rep_movsw();
1752}
1753
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001754static void GenPeek(LocationSummary* locations, Primitive::Type size, X86_64Assembler* assembler) {
1755 CpuRegister address = locations->InAt(0).AsRegister<CpuRegister>();
1756 CpuRegister out = locations->Out().AsRegister<CpuRegister>(); // == address, here for clarity.
1757 // x86 allows unaligned access. We do not have to check the input or use specific instructions
1758 // to avoid a SIGBUS.
1759 switch (size) {
1760 case Primitive::kPrimByte:
1761 __ movsxb(out, Address(address, 0));
1762 break;
1763 case Primitive::kPrimShort:
1764 __ movsxw(out, Address(address, 0));
1765 break;
1766 case Primitive::kPrimInt:
1767 __ movl(out, Address(address, 0));
1768 break;
1769 case Primitive::kPrimLong:
1770 __ movq(out, Address(address, 0));
1771 break;
1772 default:
1773 LOG(FATAL) << "Type not recognized for peek: " << size;
1774 UNREACHABLE();
1775 }
1776}
1777
1778void IntrinsicLocationsBuilderX86_64::VisitMemoryPeekByte(HInvoke* invoke) {
1779 CreateIntToIntLocations(arena_, invoke);
1780}
1781
1782void IntrinsicCodeGeneratorX86_64::VisitMemoryPeekByte(HInvoke* invoke) {
1783 GenPeek(invoke->GetLocations(), Primitive::kPrimByte, GetAssembler());
1784}
1785
1786void IntrinsicLocationsBuilderX86_64::VisitMemoryPeekIntNative(HInvoke* invoke) {
1787 CreateIntToIntLocations(arena_, invoke);
1788}
1789
1790void IntrinsicCodeGeneratorX86_64::VisitMemoryPeekIntNative(HInvoke* invoke) {
1791 GenPeek(invoke->GetLocations(), Primitive::kPrimInt, GetAssembler());
1792}
1793
1794void IntrinsicLocationsBuilderX86_64::VisitMemoryPeekLongNative(HInvoke* invoke) {
1795 CreateIntToIntLocations(arena_, invoke);
1796}
1797
1798void IntrinsicCodeGeneratorX86_64::VisitMemoryPeekLongNative(HInvoke* invoke) {
1799 GenPeek(invoke->GetLocations(), Primitive::kPrimLong, GetAssembler());
1800}
1801
1802void IntrinsicLocationsBuilderX86_64::VisitMemoryPeekShortNative(HInvoke* invoke) {
1803 CreateIntToIntLocations(arena_, invoke);
1804}
1805
1806void IntrinsicCodeGeneratorX86_64::VisitMemoryPeekShortNative(HInvoke* invoke) {
1807 GenPeek(invoke->GetLocations(), Primitive::kPrimShort, GetAssembler());
1808}
1809
1810static void CreateIntIntToVoidLocations(ArenaAllocator* arena, HInvoke* invoke) {
1811 LocationSummary* locations = new (arena) LocationSummary(invoke,
1812 LocationSummary::kNoCall,
1813 kIntrinsified);
1814 locations->SetInAt(0, Location::RequiresRegister());
Mark Mendellea5af682015-10-22 17:35:49 -04001815 locations->SetInAt(1, Location::RegisterOrInt32Constant(invoke->InputAt(1)));
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001816}
1817
1818static void GenPoke(LocationSummary* locations, Primitive::Type size, X86_64Assembler* assembler) {
1819 CpuRegister address = locations->InAt(0).AsRegister<CpuRegister>();
Mark Mendell40741f32015-04-20 22:10:34 -04001820 Location value = locations->InAt(1);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001821 // x86 allows unaligned access. We do not have to check the input or use specific instructions
1822 // to avoid a SIGBUS.
1823 switch (size) {
1824 case Primitive::kPrimByte:
Mark Mendell40741f32015-04-20 22:10:34 -04001825 if (value.IsConstant()) {
1826 __ movb(Address(address, 0),
1827 Immediate(CodeGenerator::GetInt32ValueOf(value.GetConstant())));
1828 } else {
1829 __ movb(Address(address, 0), value.AsRegister<CpuRegister>());
1830 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001831 break;
1832 case Primitive::kPrimShort:
Mark Mendell40741f32015-04-20 22:10:34 -04001833 if (value.IsConstant()) {
1834 __ movw(Address(address, 0),
1835 Immediate(CodeGenerator::GetInt32ValueOf(value.GetConstant())));
1836 } else {
1837 __ movw(Address(address, 0), value.AsRegister<CpuRegister>());
1838 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001839 break;
1840 case Primitive::kPrimInt:
Mark Mendell40741f32015-04-20 22:10:34 -04001841 if (value.IsConstant()) {
1842 __ movl(Address(address, 0),
1843 Immediate(CodeGenerator::GetInt32ValueOf(value.GetConstant())));
1844 } else {
1845 __ movl(Address(address, 0), value.AsRegister<CpuRegister>());
1846 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001847 break;
1848 case Primitive::kPrimLong:
Mark Mendell40741f32015-04-20 22:10:34 -04001849 if (value.IsConstant()) {
1850 int64_t v = value.GetConstant()->AsLongConstant()->GetValue();
1851 DCHECK(IsInt<32>(v));
1852 int32_t v_32 = v;
1853 __ movq(Address(address, 0), Immediate(v_32));
1854 } else {
1855 __ movq(Address(address, 0), value.AsRegister<CpuRegister>());
1856 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001857 break;
1858 default:
1859 LOG(FATAL) << "Type not recognized for poke: " << size;
1860 UNREACHABLE();
1861 }
1862}
1863
1864void IntrinsicLocationsBuilderX86_64::VisitMemoryPokeByte(HInvoke* invoke) {
1865 CreateIntIntToVoidLocations(arena_, invoke);
1866}
1867
1868void IntrinsicCodeGeneratorX86_64::VisitMemoryPokeByte(HInvoke* invoke) {
1869 GenPoke(invoke->GetLocations(), Primitive::kPrimByte, GetAssembler());
1870}
1871
1872void IntrinsicLocationsBuilderX86_64::VisitMemoryPokeIntNative(HInvoke* invoke) {
1873 CreateIntIntToVoidLocations(arena_, invoke);
1874}
1875
1876void IntrinsicCodeGeneratorX86_64::VisitMemoryPokeIntNative(HInvoke* invoke) {
1877 GenPoke(invoke->GetLocations(), Primitive::kPrimInt, GetAssembler());
1878}
1879
1880void IntrinsicLocationsBuilderX86_64::VisitMemoryPokeLongNative(HInvoke* invoke) {
1881 CreateIntIntToVoidLocations(arena_, invoke);
1882}
1883
1884void IntrinsicCodeGeneratorX86_64::VisitMemoryPokeLongNative(HInvoke* invoke) {
1885 GenPoke(invoke->GetLocations(), Primitive::kPrimLong, GetAssembler());
1886}
1887
1888void IntrinsicLocationsBuilderX86_64::VisitMemoryPokeShortNative(HInvoke* invoke) {
1889 CreateIntIntToVoidLocations(arena_, invoke);
1890}
1891
1892void IntrinsicCodeGeneratorX86_64::VisitMemoryPokeShortNative(HInvoke* invoke) {
1893 GenPoke(invoke->GetLocations(), Primitive::kPrimShort, GetAssembler());
1894}
1895
1896void IntrinsicLocationsBuilderX86_64::VisitThreadCurrentThread(HInvoke* invoke) {
1897 LocationSummary* locations = new (arena_) LocationSummary(invoke,
1898 LocationSummary::kNoCall,
1899 kIntrinsified);
1900 locations->SetOut(Location::RequiresRegister());
1901}
1902
1903void IntrinsicCodeGeneratorX86_64::VisitThreadCurrentThread(HInvoke* invoke) {
1904 CpuRegister out = invoke->GetLocations()->Out().AsRegister<CpuRegister>();
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001905 GetAssembler()->gs()->movl(out, Address::Absolute(Thread::PeerOffset<kX86_64WordSize>(),
1906 /* no_rip */ true));
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001907}
1908
Roland Levillain0d5a2812015-11-13 10:07:31 +00001909static void GenUnsafeGet(HInvoke* invoke,
1910 Primitive::Type type,
1911 bool is_volatile ATTRIBUTE_UNUSED,
1912 CodeGeneratorX86_64* codegen) {
1913 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen->GetAssembler());
1914 LocationSummary* locations = invoke->GetLocations();
1915 Location base_loc = locations->InAt(1);
1916 CpuRegister base = base_loc.AsRegister<CpuRegister>();
1917 Location offset_loc = locations->InAt(2);
1918 CpuRegister offset = offset_loc.AsRegister<CpuRegister>();
1919 Location output_loc = locations->Out();
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001920 CpuRegister output = output_loc.AsRegister<CpuRegister>();
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001921
Andreas Gampe878d58c2015-01-15 23:24:00 -08001922 switch (type) {
1923 case Primitive::kPrimInt:
Roland Levillain0d5a2812015-11-13 10:07:31 +00001924 __ movl(output, Address(base, offset, ScaleFactor::TIMES_1, 0));
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001925 break;
1926
1927 case Primitive::kPrimNot: {
1928 if (kEmitCompilerReadBarrier) {
1929 if (kUseBakerReadBarrier) {
1930 Location temp = locations->GetTemp(0);
1931 codegen->GenerateArrayLoadWithBakerReadBarrier(
1932 invoke, output_loc, base, 0U, offset_loc, temp, /* needs_null_check */ false);
1933 } else {
1934 __ movl(output, Address(base, offset, ScaleFactor::TIMES_1, 0));
1935 codegen->GenerateReadBarrierSlow(
1936 invoke, output_loc, output_loc, base_loc, 0U, offset_loc);
1937 }
1938 } else {
1939 __ movl(output, Address(base, offset, ScaleFactor::TIMES_1, 0));
1940 __ MaybeUnpoisonHeapReference(output);
Roland Levillain4d027112015-07-01 15:41:14 +01001941 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001942 break;
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001943 }
Andreas Gampe878d58c2015-01-15 23:24:00 -08001944
1945 case Primitive::kPrimLong:
Roland Levillain0d5a2812015-11-13 10:07:31 +00001946 __ movq(output, Address(base, offset, ScaleFactor::TIMES_1, 0));
Andreas Gampe878d58c2015-01-15 23:24:00 -08001947 break;
1948
1949 default:
1950 LOG(FATAL) << "Unsupported op size " << type;
1951 UNREACHABLE();
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001952 }
1953}
1954
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001955static void CreateIntIntIntToIntLocations(ArenaAllocator* arena,
1956 HInvoke* invoke,
1957 Primitive::Type type) {
Roland Levillain0d5a2812015-11-13 10:07:31 +00001958 bool can_call = kEmitCompilerReadBarrier &&
1959 (invoke->GetIntrinsic() == Intrinsics::kUnsafeGetObject ||
1960 invoke->GetIntrinsic() == Intrinsics::kUnsafeGetObjectVolatile);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001961 LocationSummary* locations = new (arena) LocationSummary(invoke,
Roland Levillain0d5a2812015-11-13 10:07:31 +00001962 can_call ?
1963 LocationSummary::kCallOnSlowPath :
1964 LocationSummary::kNoCall,
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001965 kIntrinsified);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001966 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001967 locations->SetInAt(1, Location::RequiresRegister());
1968 locations->SetInAt(2, Location::RequiresRegister());
Andreas Gampe878d58c2015-01-15 23:24:00 -08001969 locations->SetOut(Location::RequiresRegister());
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001970 if (type == Primitive::kPrimNot && kEmitCompilerReadBarrier && kUseBakerReadBarrier) {
1971 // We need a temporary register for the read barrier marking slow
1972 // path in InstructionCodeGeneratorX86_64::GenerateArrayLoadWithBakerReadBarrier.
1973 locations->AddTemp(Location::RequiresRegister());
1974 }
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001975}
1976
1977void IntrinsicLocationsBuilderX86_64::VisitUnsafeGet(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001978 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimInt);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001979}
1980void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetVolatile(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001981 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimInt);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001982}
1983void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetLong(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001984 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimLong);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001985}
1986void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetLongVolatile(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001987 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimLong);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001988}
Andreas Gampe878d58c2015-01-15 23:24:00 -08001989void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetObject(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001990 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimNot);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001991}
1992void IntrinsicLocationsBuilderX86_64::VisitUnsafeGetObjectVolatile(HInvoke* invoke) {
Roland Levillain1e7f8db2015-12-15 10:54:19 +00001993 CreateIntIntIntToIntLocations(arena_, invoke, Primitive::kPrimNot);
Andreas Gampe878d58c2015-01-15 23:24:00 -08001994}
1995
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001996
1997void IntrinsicCodeGeneratorX86_64::VisitUnsafeGet(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00001998 GenUnsafeGet(invoke, Primitive::kPrimInt, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08001999}
2000void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002001 GenUnsafeGet(invoke, Primitive::kPrimInt, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002002}
2003void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002004 GenUnsafeGet(invoke, Primitive::kPrimLong, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002005}
2006void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetLongVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002007 GenUnsafeGet(invoke, Primitive::kPrimLong, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002008}
Andreas Gampe878d58c2015-01-15 23:24:00 -08002009void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetObject(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002010 GenUnsafeGet(invoke, Primitive::kPrimNot, /* is_volatile */ false, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08002011}
2012void IntrinsicCodeGeneratorX86_64::VisitUnsafeGetObjectVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002013 GenUnsafeGet(invoke, Primitive::kPrimNot, /* is_volatile */ true, codegen_);
Andreas Gampe878d58c2015-01-15 23:24:00 -08002014}
2015
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002016
2017static void CreateIntIntIntIntToVoidPlusTempsLocations(ArenaAllocator* arena,
2018 Primitive::Type type,
2019 HInvoke* invoke) {
2020 LocationSummary* locations = new (arena) LocationSummary(invoke,
2021 LocationSummary::kNoCall,
2022 kIntrinsified);
Andreas Gampe878d58c2015-01-15 23:24:00 -08002023 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002024 locations->SetInAt(1, Location::RequiresRegister());
2025 locations->SetInAt(2, Location::RequiresRegister());
2026 locations->SetInAt(3, Location::RequiresRegister());
2027 if (type == Primitive::kPrimNot) {
2028 // Need temp registers for card-marking.
Roland Levillain4d027112015-07-01 15:41:14 +01002029 locations->AddTemp(Location::RequiresRegister()); // Possibly used for reference poisoning too.
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002030 locations->AddTemp(Location::RequiresRegister());
2031 }
2032}
2033
2034void IntrinsicLocationsBuilderX86_64::VisitUnsafePut(HInvoke* invoke) {
2035 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimInt, invoke);
2036}
2037void IntrinsicLocationsBuilderX86_64::VisitUnsafePutOrdered(HInvoke* invoke) {
2038 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimInt, invoke);
2039}
2040void IntrinsicLocationsBuilderX86_64::VisitUnsafePutVolatile(HInvoke* invoke) {
2041 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimInt, invoke);
2042}
2043void IntrinsicLocationsBuilderX86_64::VisitUnsafePutObject(HInvoke* invoke) {
2044 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimNot, invoke);
2045}
2046void IntrinsicLocationsBuilderX86_64::VisitUnsafePutObjectOrdered(HInvoke* invoke) {
2047 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimNot, invoke);
2048}
2049void IntrinsicLocationsBuilderX86_64::VisitUnsafePutObjectVolatile(HInvoke* invoke) {
2050 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimNot, invoke);
2051}
2052void IntrinsicLocationsBuilderX86_64::VisitUnsafePutLong(HInvoke* invoke) {
2053 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimLong, invoke);
2054}
2055void IntrinsicLocationsBuilderX86_64::VisitUnsafePutLongOrdered(HInvoke* invoke) {
2056 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimLong, invoke);
2057}
2058void IntrinsicLocationsBuilderX86_64::VisitUnsafePutLongVolatile(HInvoke* invoke) {
2059 CreateIntIntIntIntToVoidPlusTempsLocations(arena_, Primitive::kPrimLong, invoke);
2060}
2061
2062// We don't care for ordered: it requires an AnyStore barrier, which is already given by the x86
2063// memory model.
2064static void GenUnsafePut(LocationSummary* locations, Primitive::Type type, bool is_volatile,
2065 CodeGeneratorX86_64* codegen) {
Roland Levillainb488b782015-10-22 11:38:49 +01002066 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen->GetAssembler());
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002067 CpuRegister base = locations->InAt(1).AsRegister<CpuRegister>();
2068 CpuRegister offset = locations->InAt(2).AsRegister<CpuRegister>();
2069 CpuRegister value = locations->InAt(3).AsRegister<CpuRegister>();
2070
2071 if (type == Primitive::kPrimLong) {
2072 __ movq(Address(base, offset, ScaleFactor::TIMES_1, 0), value);
Roland Levillain4d027112015-07-01 15:41:14 +01002073 } else if (kPoisonHeapReferences && type == Primitive::kPrimNot) {
2074 CpuRegister temp = locations->GetTemp(0).AsRegister<CpuRegister>();
2075 __ movl(temp, value);
2076 __ PoisonHeapReference(temp);
2077 __ movl(Address(base, offset, ScaleFactor::TIMES_1, 0), temp);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002078 } else {
2079 __ movl(Address(base, offset, ScaleFactor::TIMES_1, 0), value);
2080 }
2081
2082 if (is_volatile) {
Aart Bik0da3b912015-12-16 19:06:17 +00002083 __ mfence();
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002084 }
2085
2086 if (type == Primitive::kPrimNot) {
Nicolas Geoffray07276db2015-05-18 14:22:09 +01002087 bool value_can_be_null = true; // TODO: Worth finding out this information?
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002088 codegen->MarkGCCard(locations->GetTemp(0).AsRegister<CpuRegister>(),
2089 locations->GetTemp(1).AsRegister<CpuRegister>(),
2090 base,
Nicolas Geoffray07276db2015-05-18 14:22:09 +01002091 value,
2092 value_can_be_null);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002093 }
2094}
2095
2096void IntrinsicCodeGeneratorX86_64::VisitUnsafePut(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002097 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimInt, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002098}
2099void IntrinsicCodeGeneratorX86_64::VisitUnsafePutOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002100 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimInt, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002101}
2102void IntrinsicCodeGeneratorX86_64::VisitUnsafePutVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002103 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimInt, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002104}
2105void IntrinsicCodeGeneratorX86_64::VisitUnsafePutObject(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002106 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimNot, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002107}
2108void IntrinsicCodeGeneratorX86_64::VisitUnsafePutObjectOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002109 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimNot, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002110}
2111void IntrinsicCodeGeneratorX86_64::VisitUnsafePutObjectVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002112 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimNot, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002113}
2114void IntrinsicCodeGeneratorX86_64::VisitUnsafePutLong(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002115 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimLong, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002116}
2117void IntrinsicCodeGeneratorX86_64::VisitUnsafePutLongOrdered(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002118 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimLong, /* is_volatile */ false, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002119}
2120void IntrinsicCodeGeneratorX86_64::VisitUnsafePutLongVolatile(HInvoke* invoke) {
Roland Levillainbf84a3d2015-12-04 14:33:02 +00002121 GenUnsafePut(invoke->GetLocations(), Primitive::kPrimLong, /* is_volatile */ true, codegen_);
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002122}
2123
Mark Mendell58d25fd2015-04-03 14:52:31 -04002124static void CreateIntIntIntIntIntToInt(ArenaAllocator* arena, Primitive::Type type,
2125 HInvoke* invoke) {
2126 LocationSummary* locations = new (arena) LocationSummary(invoke,
2127 LocationSummary::kNoCall,
2128 kIntrinsified);
2129 locations->SetInAt(0, Location::NoLocation()); // Unused receiver.
2130 locations->SetInAt(1, Location::RequiresRegister());
2131 locations->SetInAt(2, Location::RequiresRegister());
2132 // expected value must be in EAX/RAX.
2133 locations->SetInAt(3, Location::RegisterLocation(RAX));
2134 locations->SetInAt(4, Location::RequiresRegister());
2135
2136 locations->SetOut(Location::RequiresRegister());
2137 if (type == Primitive::kPrimNot) {
2138 // Need temp registers for card-marking.
Roland Levillainb488b782015-10-22 11:38:49 +01002139 locations->AddTemp(Location::RequiresRegister()); // Possibly used for reference poisoning too.
Mark Mendell58d25fd2015-04-03 14:52:31 -04002140 locations->AddTemp(Location::RequiresRegister());
2141 }
2142}
2143
2144void IntrinsicLocationsBuilderX86_64::VisitUnsafeCASInt(HInvoke* invoke) {
2145 CreateIntIntIntIntIntToInt(arena_, Primitive::kPrimInt, invoke);
2146}
2147
2148void IntrinsicLocationsBuilderX86_64::VisitUnsafeCASLong(HInvoke* invoke) {
2149 CreateIntIntIntIntIntToInt(arena_, Primitive::kPrimLong, invoke);
2150}
2151
2152void IntrinsicLocationsBuilderX86_64::VisitUnsafeCASObject(HInvoke* invoke) {
Roland Levillain391b8662015-12-18 11:43:38 +00002153 // The UnsafeCASObject intrinsic is missing a read barrier, and
2154 // therefore sometimes does not work as expected (b/25883050).
2155 // Turn it off temporarily as a quick fix, until the read barrier is
2156 // implemented.
2157 //
2158 // TODO(rpl): Implement a read barrier in GenCAS below and re-enable
2159 // this intrinsic.
2160 if (kEmitCompilerReadBarrier) {
2161 return;
2162 }
2163
Mark Mendell58d25fd2015-04-03 14:52:31 -04002164 CreateIntIntIntIntIntToInt(arena_, Primitive::kPrimNot, invoke);
2165}
2166
2167static void GenCAS(Primitive::Type type, HInvoke* invoke, CodeGeneratorX86_64* codegen) {
Roland Levillainb488b782015-10-22 11:38:49 +01002168 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen->GetAssembler());
Mark Mendell58d25fd2015-04-03 14:52:31 -04002169 LocationSummary* locations = invoke->GetLocations();
2170
2171 CpuRegister base = locations->InAt(1).AsRegister<CpuRegister>();
2172 CpuRegister offset = locations->InAt(2).AsRegister<CpuRegister>();
2173 CpuRegister expected = locations->InAt(3).AsRegister<CpuRegister>();
Roland Levillainb488b782015-10-22 11:38:49 +01002174 // Ensure `expected` is in RAX (required by the CMPXCHG instruction).
Mark Mendell58d25fd2015-04-03 14:52:31 -04002175 DCHECK_EQ(expected.AsRegister(), RAX);
2176 CpuRegister value = locations->InAt(4).AsRegister<CpuRegister>();
2177 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2178
Roland Levillainb488b782015-10-22 11:38:49 +01002179 if (type == Primitive::kPrimNot) {
2180 // Mark card for object assuming new value is stored.
2181 bool value_can_be_null = true; // TODO: Worth finding out this information?
2182 codegen->MarkGCCard(locations->GetTemp(0).AsRegister<CpuRegister>(),
2183 locations->GetTemp(1).AsRegister<CpuRegister>(),
2184 base,
2185 value,
2186 value_can_be_null);
Roland Levillain4d027112015-07-01 15:41:14 +01002187
Roland Levillainb488b782015-10-22 11:38:49 +01002188 bool base_equals_value = (base.AsRegister() == value.AsRegister());
2189 Register value_reg = value.AsRegister();
2190 if (kPoisonHeapReferences) {
2191 if (base_equals_value) {
2192 // If `base` and `value` are the same register location, move
2193 // `value_reg` to a temporary register. This way, poisoning
2194 // `value_reg` won't invalidate `base`.
2195 value_reg = locations->GetTemp(0).AsRegister<CpuRegister>().AsRegister();
2196 __ movl(CpuRegister(value_reg), base);
Roland Levillain4d027112015-07-01 15:41:14 +01002197 }
Roland Levillainb488b782015-10-22 11:38:49 +01002198
2199 // Check that the register allocator did not assign the location
2200 // of `expected` (RAX) to `value` nor to `base`, so that heap
2201 // poisoning (when enabled) works as intended below.
2202 // - If `value` were equal to `expected`, both references would
2203 // be poisoned twice, meaning they would not be poisoned at
2204 // all, as heap poisoning uses address negation.
2205 // - If `base` were equal to `expected`, poisoning `expected`
2206 // would invalidate `base`.
2207 DCHECK_NE(value_reg, expected.AsRegister());
2208 DCHECK_NE(base.AsRegister(), expected.AsRegister());
2209
2210 __ PoisonHeapReference(expected);
2211 __ PoisonHeapReference(CpuRegister(value_reg));
Mark Mendell58d25fd2015-04-03 14:52:31 -04002212 }
2213
Roland Levillain391b8662015-12-18 11:43:38 +00002214 // TODO: Add a read barrier for the reference stored in the object
2215 // before attempting the CAS, similar to the one in the
2216 // art::Unsafe_compareAndSwapObject JNI implementation.
2217 //
2218 // Note that this code is not (yet) used when read barriers are
2219 // enabled (see IntrinsicLocationsBuilderX86_64::VisitUnsafeCASObject).
2220 DCHECK(!kEmitCompilerReadBarrier);
Roland Levillainb488b782015-10-22 11:38:49 +01002221 __ LockCmpxchgl(Address(base, offset, TIMES_1, 0), CpuRegister(value_reg));
Mark Mendell58d25fd2015-04-03 14:52:31 -04002222
Roland Levillain0d5a2812015-11-13 10:07:31 +00002223 // LOCK CMPXCHG has full barrier semantics, and we don't need
Roland Levillainb488b782015-10-22 11:38:49 +01002224 // scheduling barriers at this time.
Mark Mendell58d25fd2015-04-03 14:52:31 -04002225
Roland Levillainb488b782015-10-22 11:38:49 +01002226 // Convert ZF into the boolean result.
2227 __ setcc(kZero, out);
2228 __ movzxb(out, out);
Roland Levillain4d027112015-07-01 15:41:14 +01002229
Roland Levillain391b8662015-12-18 11:43:38 +00002230 // If heap poisoning is enabled, we need to unpoison the values
2231 // that were poisoned earlier.
Roland Levillainb488b782015-10-22 11:38:49 +01002232 if (kPoisonHeapReferences) {
2233 if (base_equals_value) {
2234 // `value_reg` has been moved to a temporary register, no need
2235 // to unpoison it.
2236 } else {
2237 // Ensure `value` is different from `out`, so that unpoisoning
2238 // the former does not invalidate the latter.
2239 DCHECK_NE(value_reg, out.AsRegister());
2240 __ UnpoisonHeapReference(CpuRegister(value_reg));
2241 }
2242 // Ensure `expected` is different from `out`, so that unpoisoning
2243 // the former does not invalidate the latter.
2244 DCHECK_NE(expected.AsRegister(), out.AsRegister());
2245 __ UnpoisonHeapReference(expected);
2246 }
2247 } else {
2248 if (type == Primitive::kPrimInt) {
2249 __ LockCmpxchgl(Address(base, offset, TIMES_1, 0), value);
2250 } else if (type == Primitive::kPrimLong) {
2251 __ LockCmpxchgq(Address(base, offset, TIMES_1, 0), value);
2252 } else {
2253 LOG(FATAL) << "Unexpected CAS type " << type;
2254 }
2255
Roland Levillain0d5a2812015-11-13 10:07:31 +00002256 // LOCK CMPXCHG has full barrier semantics, and we don't need
Roland Levillainb488b782015-10-22 11:38:49 +01002257 // scheduling barriers at this time.
2258
2259 // Convert ZF into the boolean result.
2260 __ setcc(kZero, out);
2261 __ movzxb(out, out);
Roland Levillain4d027112015-07-01 15:41:14 +01002262 }
Mark Mendell58d25fd2015-04-03 14:52:31 -04002263}
2264
2265void IntrinsicCodeGeneratorX86_64::VisitUnsafeCASInt(HInvoke* invoke) {
2266 GenCAS(Primitive::kPrimInt, invoke, codegen_);
2267}
2268
2269void IntrinsicCodeGeneratorX86_64::VisitUnsafeCASLong(HInvoke* invoke) {
2270 GenCAS(Primitive::kPrimLong, invoke, codegen_);
2271}
2272
2273void IntrinsicCodeGeneratorX86_64::VisitUnsafeCASObject(HInvoke* invoke) {
2274 GenCAS(Primitive::kPrimNot, invoke, codegen_);
2275}
2276
2277void IntrinsicLocationsBuilderX86_64::VisitIntegerReverse(HInvoke* invoke) {
2278 LocationSummary* locations = new (arena_) LocationSummary(invoke,
2279 LocationSummary::kNoCall,
2280 kIntrinsified);
2281 locations->SetInAt(0, Location::RequiresRegister());
2282 locations->SetOut(Location::SameAsFirstInput());
2283 locations->AddTemp(Location::RequiresRegister());
2284}
2285
2286static void SwapBits(CpuRegister reg, CpuRegister temp, int32_t shift, int32_t mask,
2287 X86_64Assembler* assembler) {
2288 Immediate imm_shift(shift);
2289 Immediate imm_mask(mask);
2290 __ movl(temp, reg);
2291 __ shrl(reg, imm_shift);
2292 __ andl(temp, imm_mask);
2293 __ andl(reg, imm_mask);
2294 __ shll(temp, imm_shift);
2295 __ orl(reg, temp);
2296}
2297
2298void IntrinsicCodeGeneratorX86_64::VisitIntegerReverse(HInvoke* invoke) {
Roland Levillainb488b782015-10-22 11:38:49 +01002299 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen_->GetAssembler());
Mark Mendell58d25fd2015-04-03 14:52:31 -04002300 LocationSummary* locations = invoke->GetLocations();
2301
2302 CpuRegister reg = locations->InAt(0).AsRegister<CpuRegister>();
2303 CpuRegister temp = locations->GetTemp(0).AsRegister<CpuRegister>();
2304
2305 /*
2306 * Use one bswap instruction to reverse byte order first and then use 3 rounds of
2307 * swapping bits to reverse bits in a number x. Using bswap to save instructions
2308 * compared to generic luni implementation which has 5 rounds of swapping bits.
2309 * x = bswap x
2310 * x = (x & 0x55555555) << 1 | (x >> 1) & 0x55555555;
2311 * x = (x & 0x33333333) << 2 | (x >> 2) & 0x33333333;
2312 * x = (x & 0x0F0F0F0F) << 4 | (x >> 4) & 0x0F0F0F0F;
2313 */
2314 __ bswapl(reg);
2315 SwapBits(reg, temp, 1, 0x55555555, assembler);
2316 SwapBits(reg, temp, 2, 0x33333333, assembler);
2317 SwapBits(reg, temp, 4, 0x0f0f0f0f, assembler);
2318}
2319
2320void IntrinsicLocationsBuilderX86_64::VisitLongReverse(HInvoke* invoke) {
2321 LocationSummary* locations = new (arena_) LocationSummary(invoke,
2322 LocationSummary::kNoCall,
2323 kIntrinsified);
2324 locations->SetInAt(0, Location::RequiresRegister());
2325 locations->SetOut(Location::SameAsFirstInput());
2326 locations->AddTemp(Location::RequiresRegister());
2327 locations->AddTemp(Location::RequiresRegister());
2328}
2329
2330static void SwapBits64(CpuRegister reg, CpuRegister temp, CpuRegister temp_mask,
2331 int32_t shift, int64_t mask, X86_64Assembler* assembler) {
2332 Immediate imm_shift(shift);
2333 __ movq(temp_mask, Immediate(mask));
2334 __ movq(temp, reg);
2335 __ shrq(reg, imm_shift);
2336 __ andq(temp, temp_mask);
2337 __ andq(reg, temp_mask);
2338 __ shlq(temp, imm_shift);
2339 __ orq(reg, temp);
2340}
2341
2342void IntrinsicCodeGeneratorX86_64::VisitLongReverse(HInvoke* invoke) {
Roland Levillainb488b782015-10-22 11:38:49 +01002343 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen_->GetAssembler());
Mark Mendell58d25fd2015-04-03 14:52:31 -04002344 LocationSummary* locations = invoke->GetLocations();
2345
2346 CpuRegister reg = locations->InAt(0).AsRegister<CpuRegister>();
2347 CpuRegister temp1 = locations->GetTemp(0).AsRegister<CpuRegister>();
2348 CpuRegister temp2 = locations->GetTemp(1).AsRegister<CpuRegister>();
2349
2350 /*
2351 * Use one bswap instruction to reverse byte order first and then use 3 rounds of
2352 * swapping bits to reverse bits in a long number x. Using bswap to save instructions
2353 * compared to generic luni implementation which has 5 rounds of swapping bits.
2354 * x = bswap x
2355 * x = (x & 0x5555555555555555) << 1 | (x >> 1) & 0x5555555555555555;
2356 * x = (x & 0x3333333333333333) << 2 | (x >> 2) & 0x3333333333333333;
2357 * x = (x & 0x0F0F0F0F0F0F0F0F) << 4 | (x >> 4) & 0x0F0F0F0F0F0F0F0F;
2358 */
2359 __ bswapq(reg);
2360 SwapBits64(reg, temp1, temp2, 1, INT64_C(0x5555555555555555), assembler);
2361 SwapBits64(reg, temp1, temp2, 2, INT64_C(0x3333333333333333), assembler);
2362 SwapBits64(reg, temp1, temp2, 4, INT64_C(0x0f0f0f0f0f0f0f0f), assembler);
2363}
2364
Mark Mendelld5897672015-08-12 21:16:41 -04002365static void CreateLeadingZeroLocations(ArenaAllocator* arena, HInvoke* invoke) {
2366 LocationSummary* locations = new (arena) LocationSummary(invoke,
2367 LocationSummary::kNoCall,
2368 kIntrinsified);
2369 locations->SetInAt(0, Location::Any());
2370 locations->SetOut(Location::RequiresRegister());
2371}
2372
2373static void GenLeadingZeros(X86_64Assembler* assembler, HInvoke* invoke, bool is_long) {
2374 LocationSummary* locations = invoke->GetLocations();
2375 Location src = locations->InAt(0);
2376 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2377
2378 int zero_value_result = is_long ? 64 : 32;
2379 if (invoke->InputAt(0)->IsConstant()) {
2380 // Evaluate this at compile time.
2381 int64_t value = Int64FromConstant(invoke->InputAt(0)->AsConstant());
2382 if (value == 0) {
2383 value = zero_value_result;
2384 } else {
2385 value = is_long ? CLZ(static_cast<uint64_t>(value)) : CLZ(static_cast<uint32_t>(value));
2386 }
2387 if (value == 0) {
2388 __ xorl(out, out);
2389 } else {
2390 __ movl(out, Immediate(value));
2391 }
2392 return;
2393 }
2394
2395 // Handle the non-constant cases.
2396 if (src.IsRegister()) {
2397 if (is_long) {
2398 __ bsrq(out, src.AsRegister<CpuRegister>());
2399 } else {
2400 __ bsrl(out, src.AsRegister<CpuRegister>());
2401 }
2402 } else if (is_long) {
2403 DCHECK(src.IsDoubleStackSlot());
2404 __ bsrq(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2405 } else {
2406 DCHECK(src.IsStackSlot());
2407 __ bsrl(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2408 }
2409
2410 // BSR sets ZF if the input was zero, and the output is undefined.
Mark Mendell0c9497d2015-08-21 09:30:05 -04002411 NearLabel is_zero, done;
Mark Mendelld5897672015-08-12 21:16:41 -04002412 __ j(kEqual, &is_zero);
2413
2414 // Correct the result from BSR to get the CLZ result.
2415 __ xorl(out, Immediate(zero_value_result - 1));
2416 __ jmp(&done);
2417
2418 // Fix the zero case with the expected result.
2419 __ Bind(&is_zero);
2420 __ movl(out, Immediate(zero_value_result));
2421
2422 __ Bind(&done);
2423}
2424
2425void IntrinsicLocationsBuilderX86_64::VisitIntegerNumberOfLeadingZeros(HInvoke* invoke) {
2426 CreateLeadingZeroLocations(arena_, invoke);
2427}
2428
2429void IntrinsicCodeGeneratorX86_64::VisitIntegerNumberOfLeadingZeros(HInvoke* invoke) {
2430 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen_->GetAssembler());
2431 GenLeadingZeros(assembler, invoke, /* is_long */ false);
2432}
2433
2434void IntrinsicLocationsBuilderX86_64::VisitLongNumberOfLeadingZeros(HInvoke* invoke) {
2435 CreateLeadingZeroLocations(arena_, invoke);
2436}
2437
2438void IntrinsicCodeGeneratorX86_64::VisitLongNumberOfLeadingZeros(HInvoke* invoke) {
2439 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen_->GetAssembler());
2440 GenLeadingZeros(assembler, invoke, /* is_long */ true);
2441}
2442
Mark Mendell2d554792015-09-15 21:45:18 -04002443static void CreateTrailingZeroLocations(ArenaAllocator* arena, HInvoke* invoke) {
2444 LocationSummary* locations = new (arena) LocationSummary(invoke,
2445 LocationSummary::kNoCall,
2446 kIntrinsified);
2447 locations->SetInAt(0, Location::Any());
2448 locations->SetOut(Location::RequiresRegister());
2449}
2450
2451static void GenTrailingZeros(X86_64Assembler* assembler, HInvoke* invoke, bool is_long) {
2452 LocationSummary* locations = invoke->GetLocations();
2453 Location src = locations->InAt(0);
2454 CpuRegister out = locations->Out().AsRegister<CpuRegister>();
2455
2456 int zero_value_result = is_long ? 64 : 32;
2457 if (invoke->InputAt(0)->IsConstant()) {
2458 // Evaluate this at compile time.
2459 int64_t value = Int64FromConstant(invoke->InputAt(0)->AsConstant());
2460 if (value == 0) {
2461 value = zero_value_result;
2462 } else {
2463 value = is_long ? CTZ(static_cast<uint64_t>(value)) : CTZ(static_cast<uint32_t>(value));
2464 }
2465 if (value == 0) {
2466 __ xorl(out, out);
2467 } else {
2468 __ movl(out, Immediate(value));
2469 }
2470 return;
2471 }
2472
2473 // Handle the non-constant cases.
2474 if (src.IsRegister()) {
2475 if (is_long) {
2476 __ bsfq(out, src.AsRegister<CpuRegister>());
2477 } else {
2478 __ bsfl(out, src.AsRegister<CpuRegister>());
2479 }
2480 } else if (is_long) {
2481 DCHECK(src.IsDoubleStackSlot());
2482 __ bsfq(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2483 } else {
2484 DCHECK(src.IsStackSlot());
2485 __ bsfl(out, Address(CpuRegister(RSP), src.GetStackIndex()));
2486 }
2487
2488 // BSF sets ZF if the input was zero, and the output is undefined.
2489 NearLabel done;
2490 __ j(kNotEqual, &done);
2491
2492 // Fix the zero case with the expected result.
2493 __ movl(out, Immediate(zero_value_result));
2494
2495 __ Bind(&done);
2496}
2497
2498void IntrinsicLocationsBuilderX86_64::VisitIntegerNumberOfTrailingZeros(HInvoke* invoke) {
2499 CreateTrailingZeroLocations(arena_, invoke);
2500}
2501
2502void IntrinsicCodeGeneratorX86_64::VisitIntegerNumberOfTrailingZeros(HInvoke* invoke) {
2503 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen_->GetAssembler());
2504 GenTrailingZeros(assembler, invoke, /* is_long */ false);
2505}
2506
2507void IntrinsicLocationsBuilderX86_64::VisitLongNumberOfTrailingZeros(HInvoke* invoke) {
2508 CreateTrailingZeroLocations(arena_, invoke);
2509}
2510
2511void IntrinsicCodeGeneratorX86_64::VisitLongNumberOfTrailingZeros(HInvoke* invoke) {
2512 X86_64Assembler* assembler = down_cast<X86_64Assembler*>(codegen_->GetAssembler());
2513 GenTrailingZeros(assembler, invoke, /* is_long */ true);
2514}
2515
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002516// Unimplemented intrinsics.
2517
2518#define UNIMPLEMENTED_INTRINSIC(Name) \
2519void IntrinsicLocationsBuilderX86_64::Visit ## Name(HInvoke* invoke ATTRIBUTE_UNUSED) { \
2520} \
2521void IntrinsicCodeGeneratorX86_64::Visit ## Name(HInvoke* invoke ATTRIBUTE_UNUSED) { \
2522}
2523
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002524UNIMPLEMENTED_INTRINSIC(ReferenceGetReferent)
Scott Wakeling40a04bf2015-12-11 09:50:36 +00002525UNIMPLEMENTED_INTRINSIC(IntegerRotateLeft)
2526UNIMPLEMENTED_INTRINSIC(IntegerRotateRight)
2527UNIMPLEMENTED_INTRINSIC(LongRotateLeft)
2528UNIMPLEMENTED_INTRINSIC(LongRotateRight)
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002529
Roland Levillain4d027112015-07-01 15:41:14 +01002530#undef UNIMPLEMENTED_INTRINSIC
2531
2532#undef __
2533
Andreas Gampe71fb52f2014-12-29 17:43:08 -08002534} // namespace x86_64
2535} // namespace art