| // Copyright (c) 2017, the R8 project authors. Please see the AUTHORS file |
| // for details. All rights reserved. Use of this source code is governed by a |
| // BSD-style license that can be found in the LICENSE file. |
| package com.android.tools.r8.ir.code; |
| |
| import com.android.tools.r8.cf.LoadStoreHelper; |
| import com.android.tools.r8.cf.TypeVerificationHelper; |
| import com.android.tools.r8.cf.code.CfInvokeDynamic; |
| import com.android.tools.r8.code.InvokeCustomRange; |
| import com.android.tools.r8.graph.AppView; |
| import com.android.tools.r8.graph.DexCallSite; |
| import com.android.tools.r8.graph.DexType; |
| import com.android.tools.r8.ir.analysis.type.ClassTypeLatticeElement; |
| import com.android.tools.r8.ir.analysis.type.Nullability; |
| import com.android.tools.r8.ir.analysis.type.TypeLatticeElement; |
| import com.android.tools.r8.ir.conversion.CfBuilder; |
| import com.android.tools.r8.ir.conversion.DexBuilder; |
| import com.android.tools.r8.ir.desugar.LambdaDescriptor; |
| import com.android.tools.r8.ir.optimize.Inliner.ConstraintWithTarget; |
| import com.android.tools.r8.ir.optimize.InliningConstraints; |
| import com.google.common.collect.ImmutableSet; |
| import java.util.List; |
| import java.util.Set; |
| |
| public final class InvokeCustom extends Invoke { |
| |
| private final DexCallSite callSite; |
| |
| public InvokeCustom(DexCallSite callSite, Value result, List<Value> arguments) { |
| super(result, arguments); |
| assert callSite != null; |
| this.callSite = callSite; |
| } |
| |
| @Override |
| public int opcode() { |
| return Opcodes.INVOKE_CUSTOM; |
| } |
| |
| @Override |
| public <T> T accept(InstructionVisitor<T> visitor) { |
| return visitor.visit(this); |
| } |
| |
| private static boolean verifyLambdaInterfaces( |
| TypeLatticeElement returnTypeLattice, Set<DexType> lambdaInterfaceSet, DexType objectType) { |
| Set<DexType> primaryInterfaces = returnTypeLattice.asClassTypeLatticeElement().getInterfaces(); |
| if (returnTypeLattice.asClassTypeLatticeElement().getClassType() == objectType) { |
| assert primaryInterfaces.size() == 1; |
| // The interfaces returned by the LambdaDescriptor assumed to already contain the primary |
| // interface. If they're both singleton lists they must be identical and we can return the |
| // primary return type. |
| assert lambdaInterfaceSet.contains(primaryInterfaces.iterator().next()); |
| } else { |
| // We arrive here if the primary interface is a missing class. In that case the |
| // returnTypeLattice will be the missing type as the class type. |
| assert primaryInterfaces.isEmpty(); |
| assert lambdaInterfaceSet.contains( |
| returnTypeLattice.asClassTypeLatticeElement().getClassType()); |
| } |
| return true; |
| } |
| |
| @Override |
| public TypeLatticeElement evaluate(AppView<?> appView) { |
| TypeLatticeElement returnTypeLattice = super.evaluate(appView); |
| if (!appView.appInfo().hasSubtyping()) { |
| return returnTypeLattice; |
| } |
| |
| List<DexType> lambdaInterfaces = LambdaDescriptor.getInterfaces(callSite, appView.appInfo()); |
| if (lambdaInterfaces == null || lambdaInterfaces.isEmpty()) { |
| return returnTypeLattice; |
| } |
| |
| // The primary return type is either an interface or a missing type. |
| assert returnTypeLattice instanceof ClassTypeLatticeElement; |
| |
| Set<DexType> primaryInterfaces = returnTypeLattice.asClassTypeLatticeElement().getInterfaces(); |
| DexType objectType = appView.dexItemFactory().objectType; |
| |
| if (returnTypeLattice.asClassTypeLatticeElement().getClassType() == objectType) { |
| assert primaryInterfaces.size() == 1; |
| // Shortcut for the common case: single interface. Save creating a new lattice type. |
| if (lambdaInterfaces.size() == 1) { |
| assert lambdaInterfaces.get(0) == primaryInterfaces.iterator().next(); |
| return returnTypeLattice; |
| } |
| } |
| |
| Set<DexType> lambdaInterfaceSet = |
| ImmutableSet.<DexType>builder().addAll(lambdaInterfaces).build(); |
| |
| assert verifyLambdaInterfaces(returnTypeLattice, lambdaInterfaceSet, objectType); |
| |
| return ClassTypeLatticeElement.create(objectType, Nullability.maybeNull(), lambdaInterfaceSet); |
| } |
| |
| @Override |
| public DexType getReturnType() { |
| return callSite.methodProto.returnType; |
| } |
| |
| public DexCallSite getCallSite() { |
| return callSite; |
| } |
| |
| @Override |
| public Type getType() { |
| return Type.CUSTOM; |
| } |
| |
| @Override |
| protected String getTypeString() { |
| return "Custom"; |
| } |
| |
| @Override |
| public String toString() { |
| return super.toString() + "; call site: " + callSite.toSourceString(); |
| } |
| |
| @Override |
| public void buildDex(DexBuilder builder) { |
| com.android.tools.r8.code.Instruction instruction; |
| int argumentRegisters = requiredArgumentRegisters(); |
| builder.requestOutgoingRegisters(argumentRegisters); |
| if (needsRangedInvoke(builder)) { |
| assert argumentsConsecutive(builder); |
| int firstRegister = argumentRegisterValue(0, builder); |
| instruction = new InvokeCustomRange(firstRegister, argumentRegisters, getCallSite()); |
| } else { |
| int[] individualArgumentRegisters = new int[5]; |
| int argumentRegistersCount = fillArgumentRegisters(builder, individualArgumentRegisters); |
| instruction = new com.android.tools.r8.code.InvokeCustom( |
| argumentRegistersCount, |
| getCallSite(), |
| individualArgumentRegisters[0], // C |
| individualArgumentRegisters[1], // D |
| individualArgumentRegisters[2], // E |
| individualArgumentRegisters[3], // F |
| individualArgumentRegisters[4]); // G |
| } |
| addInvokeAndMoveResult(instruction, builder); |
| } |
| |
| @Override |
| public void buildCf(CfBuilder builder) { |
| builder.add(new CfInvokeDynamic(getCallSite())); |
| } |
| |
| @Override |
| public boolean identicalNonValueNonPositionParts(Instruction other) { |
| return other.isInvokeCustom() && callSite == other.asInvokeCustom().callSite; |
| } |
| |
| @Override |
| public boolean isInvokeCustom() { |
| return true; |
| } |
| |
| @Override |
| public InvokeCustom asInvokeCustom() { |
| return this; |
| } |
| |
| @Override |
| public ConstraintWithTarget inliningConstraint( |
| InliningConstraints inliningConstraints, DexType invocationContext) { |
| return inliningConstraints.forInvokeCustom(); |
| } |
| |
| @Override |
| public void insertLoadAndStores(InstructionListIterator it, LoadStoreHelper helper) { |
| // Essentially the same as InvokeMethod but with call site's method proto |
| // instead of a static called method. |
| helper.loadInValues(this, it); |
| if (getCallSite().methodProto.returnType.isVoidType()) { |
| return; |
| } |
| if (outValue == null) { |
| helper.popOutType(getCallSite().methodProto.returnType, this, it); |
| } else { |
| assert outValue.isUsed(); |
| helper.storeOutValue(this, it); |
| } |
| } |
| |
| @Override |
| public boolean hasInvariantOutType() { |
| return true; |
| } |
| |
| @Override |
| public DexType computeVerificationType(AppView<?> appView, TypeVerificationHelper helper) { |
| return getCallSite().methodProto.returnType; |
| } |
| |
| @Override |
| public boolean instructionMayTriggerMethodInvocation(AppView<?> appView, DexType context) { |
| return true; |
| } |
| } |