| // Copyright (c) 2019, 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.membervaluepropagation; |
| |
| import static com.android.tools.r8.utils.codeinspector.Matchers.isPresent; |
| import static org.hamcrest.MatcherAssert.assertThat; |
| import static org.junit.Assert.assertTrue; |
| |
| import com.android.tools.r8.TestBase; |
| import com.android.tools.r8.TestParameters; |
| import com.android.tools.r8.jasmin.JasminBuilder; |
| import com.android.tools.r8.utils.BooleanUtils; |
| import com.android.tools.r8.utils.StringUtils; |
| import com.android.tools.r8.utils.codeinspector.ClassSubject; |
| import com.android.tools.r8.utils.codeinspector.CodeInspector; |
| import com.android.tools.r8.utils.codeinspector.InstructionSubject; |
| import com.android.tools.r8.utils.codeinspector.MethodSubject; |
| import com.google.common.collect.ImmutableList; |
| import java.util.List; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.Parameterized; |
| |
| @RunWith(Parameterized.class) |
| public class LibraryFieldPropagationTest extends TestBase { |
| |
| private static final String MAIN = "TestClass"; |
| private static List<byte[]> programClassFileData; |
| |
| private final TestParameters parameters; |
| private final boolean withAssumeValuesRule; |
| |
| @Parameterized.Parameters(name = "{0}, with assume values rule: {1}") |
| public static List<Object[]> data() { |
| return buildParameters(getTestParameters().withAllRuntimes().build(), BooleanUtils.values()); |
| } |
| |
| public LibraryFieldPropagationTest(TestParameters parameters, boolean withAssumeValuesRule) { |
| this.parameters = parameters; |
| this.withAssumeValuesRule = withAssumeValuesRule; |
| } |
| |
| @BeforeClass |
| public static void generateProgramInput() throws Exception { |
| // Generate a program that prints Thread.MIN_PRIORITY, which is a final field that has the |
| // constant value 1. The program is being generated via Jasmin, because javac will inline the |
| // field. |
| JasminBuilder jasminBuilder = new JasminBuilder(); |
| jasminBuilder |
| .addClass(MAIN) |
| .staticMethodBuilder("main", ImmutableList.of("[Ljava/lang/String;"), "V") |
| .setCode( |
| "getstatic java/lang/System/out Ljava/io/PrintStream;", |
| "getstatic java/lang/Thread/MIN_PRIORITY I", |
| "invokevirtual java/io/PrintStream/println(I)V", |
| "return") |
| .build(); |
| programClassFileData = jasminBuilder.buildClasses(); |
| } |
| |
| @Test |
| public void test() throws Exception { |
| String expectedOutput = StringUtils.lines("1"); |
| testForR8(parameters.getBackend()) |
| .addProgramClassFileData(programClassFileData) |
| .addKeepMainRule(MAIN) |
| .addKeepRules( |
| withAssumeValuesRule |
| ? "-assumevalues class java.lang.Thread { public int MIN_PRIORITY return 1; }" |
| : "") |
| .setMinApi(parameters.getRuntime()) |
| .compile() |
| .inspect(this::verifyFieldValueNotPropagated) |
| .run(parameters.getRuntime(), MAIN) |
| .assertSuccessWithOutput(expectedOutput); |
| } |
| |
| private void verifyFieldValueNotPropagated(CodeInspector inspector) { |
| ClassSubject classSubject = inspector.clazz(MAIN); |
| assertThat(classSubject, isPresent()); |
| |
| MethodSubject methodSubject = classSubject.mainMethod(); |
| assertThat(methodSubject, isPresent()); |
| |
| if (withAssumeValuesRule) { |
| // A const-number instruction has been introduced. |
| assertTrue(methodSubject.streamInstructions().anyMatch(InstructionSubject::isConstNumber)); |
| |
| // The static-get instruction is still there, since it could in principle lead to class |
| // initialization. |
| assertTrue( |
| methodSubject |
| .streamInstructions() |
| .anyMatch( |
| instruction -> |
| instruction.isStaticGet() |
| && instruction.getField().name.toSourceString().equals("MIN_PRIORITY"))); |
| } else { |
| // Verify that the static-get instruction is still there, and that no const-number instruction |
| // has been introduced. |
| assertTrue( |
| methodSubject |
| .streamInstructions() |
| .anyMatch( |
| instruction -> |
| instruction.isStaticGet() |
| && instruction.getField().name.toSourceString().equals("MIN_PRIORITY"))); |
| assertTrue(methodSubject.streamInstructions().noneMatch(InstructionSubject::isConstNumber)); |
| } |
| } |
| } |