| // Copyright (c) 2018, 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.optimize.checkcast; |
| |
| import static org.junit.Assert.assertTrue; |
| |
| import com.android.tools.r8.TestParameters; |
| import com.android.tools.r8.TestParametersCollection; |
| import com.android.tools.r8.jasmin.JasminBuilder; |
| import com.android.tools.r8.jasmin.JasminBuilder.ClassBuilder; |
| import com.android.tools.r8.jasmin.JasminTestBase; |
| import com.android.tools.r8.utils.codeinspector.InstructionSubject; |
| import com.android.tools.r8.utils.codeinspector.MethodSubject; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.Parameterized; |
| import org.junit.runners.Parameterized.Parameter; |
| import org.junit.runners.Parameterized.Parameters; |
| |
| @RunWith(Parameterized.class) |
| public class CheckCastRemovalTest extends JasminTestBase { |
| |
| private static final String CLASS_NAME = "Example"; |
| |
| @Parameter(0) |
| public TestParameters parameters; |
| |
| @Parameters(name = "{0}") |
| public static TestParametersCollection data() { |
| return getTestParameters().withAllRuntimesAndApiLevels().build(); |
| } |
| |
| @Test |
| public void exactMatch() throws Exception { |
| JasminBuilder builder = new JasminBuilder(); |
| ClassBuilder classBuilder = builder.addClass(CLASS_NAME); |
| classBuilder.addDefaultConstructor(); |
| classBuilder.addMainMethod( |
| ".limit stack 3", |
| ".limit locals 1", |
| "new Example", |
| "dup", |
| "invokespecial Example/<init>()V", |
| "checkcast Example", // Gone |
| "return"); |
| |
| testForR8(parameters.getBackend()) |
| .addProgramClassFileData(builder.buildClasses()) |
| .addKeepMainRule(CLASS_NAME) |
| .setMinApi(parameters) |
| .compile() |
| .inspect(inspector -> checkCheckCasts(inspector.clazz(CLASS_NAME).mainMethod())) |
| .run(parameters.getRuntime(), CLASS_NAME) |
| .assertSuccessWithEmptyOutput(); |
| } |
| |
| @Test |
| public void upCasts() throws Exception { |
| JasminBuilder builder = new JasminBuilder(); |
| // A < B < C |
| ClassBuilder c = builder.addClass("C"); |
| c.addDefaultConstructor(); |
| ClassBuilder b = builder.addClass("B", "C"); |
| b.addDefaultConstructor(); |
| ClassBuilder a = builder.addClass("A", "B"); |
| a.addDefaultConstructor(); |
| ClassBuilder classBuilder = builder.addClass(CLASS_NAME); |
| classBuilder.addMainMethod( |
| ".limit stack 3", |
| ".limit locals 1", |
| "new A", |
| "dup", |
| "invokespecial A/<init>()V", |
| "checkcast B", // Gone |
| "checkcast C", // Gone |
| "return"); |
| |
| testForR8(parameters.getBackend()) |
| .addProgramClassFileData(builder.buildClasses()) |
| .addKeepClassAndMembersRules(CLASS_NAME, "A", "B", "C") |
| .setMinApi(parameters) |
| .compile() |
| .inspect(inspector -> checkCheckCasts(inspector.clazz(CLASS_NAME).mainMethod())) |
| .run(parameters.getRuntime(), CLASS_NAME) |
| .assertSuccessWithEmptyOutput(); |
| } |
| |
| @Test |
| public void bothUpAndDowncast() throws Exception { |
| JasminBuilder builder = new JasminBuilder(); |
| ClassBuilder classBuilder = builder.addClass(CLASS_NAME); |
| classBuilder.addMainMethod( |
| ".limit stack 4", |
| ".limit locals 1", |
| "iconst_1", |
| "anewarray java/lang/String", // args parameter |
| "dup", |
| "iconst_0", |
| "ldc \"a string\"", |
| "aastore", |
| "checkcast [Ljava/lang/Object;", // This upcast can be removed. |
| "iconst_0", |
| "aaload", |
| "checkcast java/lang/String", // Then, this downcast can be removed, too. |
| "invokevirtual java/lang/String/length()I", |
| "return"); |
| |
| // That is, both checkcasts should be removed together or kept together. |
| testForR8(parameters.getBackend()) |
| .addProgramClassFileData(builder.buildClasses()) |
| .addKeepMainRule(CLASS_NAME) |
| .setMinApi(parameters) |
| .compile() |
| .inspect(inspector -> checkCheckCasts(inspector.clazz(CLASS_NAME).mainMethod())) |
| .run(parameters.getRuntime(), CLASS_NAME) |
| .assertSuccessWithEmptyOutput(); |
| } |
| |
| @Test |
| public void nullCast() throws Exception { |
| JasminBuilder builder = new JasminBuilder(); |
| ClassBuilder classBuilder = builder.addClass(CLASS_NAME); |
| classBuilder.addField("public", "fld", "Ljava/lang/String;", null); |
| classBuilder.addMainMethod( |
| ".limit stack 3", |
| ".limit locals 1", |
| "aconst_null", |
| "checkcast Example", // Should be kept |
| "getfield Example.fld Ljava/lang/String;", |
| "return"); |
| |
| testForR8(parameters.getBackend()) |
| .addProgramClassFileData(builder.buildClasses()) |
| .addKeepClassAndMembersRules(CLASS_NAME) |
| .setMinApi(parameters) |
| .compile() |
| .inspect(inspector -> checkCheckCasts(inspector.clazz(CLASS_NAME).mainMethod())) |
| .run(parameters.getRuntime(), CLASS_NAME) |
| .assertFailureWithErrorThatThrows(NullPointerException.class); |
| } |
| |
| private void checkCheckCasts(MethodSubject method) { |
| assertTrue(method.streamInstructions().noneMatch(InstructionSubject::isCheckCast)); |
| } |
| } |