| // Copyright (c) 2020, 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.kotlin.metadata; |
| |
| import static com.android.tools.r8.KotlinCompilerTool.KOTLINC; |
| import static com.android.tools.r8.utils.DescriptorUtils.getDescriptorFromKotlinClassifier; |
| import static com.android.tools.r8.utils.codeinspector.Matchers.isPresent; |
| import static com.android.tools.r8.utils.codeinspector.Matchers.isRenamed; |
| import static org.hamcrest.CoreMatchers.equalTo; |
| import static org.hamcrest.CoreMatchers.not; |
| import static org.hamcrest.MatcherAssert.assertThat; |
| import static org.junit.Assert.assertEquals; |
| |
| import com.android.tools.r8.TestParameters; |
| import com.android.tools.r8.ToolHelper; |
| import com.android.tools.r8.ToolHelper.KotlinTargetVersion; |
| import com.android.tools.r8.shaking.ProguardKeepAttributes; |
| import com.android.tools.r8.utils.codeinspector.AnnotationSubject; |
| import com.android.tools.r8.utils.codeinspector.ClassSubject; |
| import com.android.tools.r8.utils.codeinspector.CodeInspector; |
| import com.android.tools.r8.utils.codeinspector.KmClassSubject; |
| import java.nio.file.Path; |
| import java.util.Collection; |
| import java.util.HashMap; |
| import java.util.Map; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.Parameterized; |
| |
| @RunWith(Parameterized.class) |
| public class MetadataRewriteInRenamedTypeTest extends KotlinMetadataTestBase { |
| private static final String OBFUSCATE_RENAMED = "-keep,allowobfuscation class **.Renamed { *; }"; |
| private static final String KEEP_KEPT = "-keep class **.Kept { *; }"; |
| |
| private final TestParameters parameters; |
| |
| @Parameterized.Parameters(name = "{0} target: {1}") |
| public static Collection<Object[]> data() { |
| return buildParameters( |
| getTestParameters().withAllRuntimesAndApiLevels().build(), KotlinTargetVersion.values()); |
| } |
| |
| public MetadataRewriteInRenamedTypeTest( |
| TestParameters parameters, KotlinTargetVersion targetVersion) { |
| super(targetVersion); |
| this.parameters = parameters; |
| } |
| |
| private static final Map<KotlinTargetVersion, Path> annoJarMap = new HashMap<>(); |
| private static final Map<KotlinTargetVersion, Path> inputJarMap = new HashMap<>(); |
| |
| @BeforeClass |
| public static void createInputJar() throws Exception { |
| String inputFolder = PKG_PREFIX + "/anno"; |
| for (KotlinTargetVersion targetVersion : KotlinTargetVersion.values()) { |
| Path annoJar = |
| kotlinc(KOTLINC, targetVersion) |
| .addSourceFiles(getKotlinFileInTest(inputFolder, "Anno")) |
| .compile(); |
| Path inputJar = |
| kotlinc(KOTLINC, targetVersion) |
| .addClasspathFiles(annoJar) |
| .addSourceFiles(getKotlinFileInTest(inputFolder, "main")) |
| .compile(); |
| annoJarMap.put(targetVersion, annoJar); |
| inputJarMap.put(targetVersion, inputJar); |
| } |
| } |
| |
| @Test |
| public void testR8_kotlinStdlibAsLib() throws Exception { |
| testForR8(parameters.getBackend()) |
| .addLibraryFiles( |
| annoJarMap.get(targetVersion), |
| ToolHelper.getJava8RuntimeJar(), |
| ToolHelper.getKotlinStdlibJar()) |
| .addProgramFiles(inputJarMap.get(targetVersion)) |
| .addKeepRules(OBFUSCATE_RENAMED, KEEP_KEPT) |
| .addKeepAttributes(ProguardKeepAttributes.RUNTIME_VISIBLE_ANNOTATIONS) |
| .compile() |
| .inspect(this::inspect); |
| } |
| |
| @Test |
| public void testR8_kotlinStdlibAsClassPath() throws Exception { |
| testForR8(parameters.getBackend()) |
| .addClasspathFiles(annoJarMap.get(targetVersion), ToolHelper.getKotlinStdlibJar()) |
| .addProgramFiles(inputJarMap.get(targetVersion)) |
| .addKeepRules(OBFUSCATE_RENAMED, KEEP_KEPT) |
| .addKeepAttributes(ProguardKeepAttributes.RUNTIME_VISIBLE_ANNOTATIONS) |
| .compile() |
| .inspect(this::inspect); |
| } |
| |
| @Test |
| public void testR8_kotlinStdlibAsProgramFile() throws Exception { |
| testForR8(parameters.getBackend()) |
| .addProgramFiles(annoJarMap.get(targetVersion), ToolHelper.getKotlinStdlibJar()) |
| .addProgramFiles(inputJarMap.get(targetVersion)) |
| .addKeepRules(OBFUSCATE_RENAMED, KEEP_KEPT) |
| .addKeepRules("-keep class **.Anno") |
| .addKeepRules("-keep class kotlin.Metadata") |
| .addKeepAttributes(ProguardKeepAttributes.RUNTIME_VISIBLE_ANNOTATIONS) |
| .allowDiagnosticWarningMessages() |
| .compile() |
| .assertWarningMessageThatMatches( |
| equalTo("Resource 'META-INF/MANIFEST.MF' already exists.")) |
| .inspect(this::inspect); |
| } |
| |
| private void inspect(CodeInspector inspector) { |
| String pkg = getClass().getPackage().getName(); |
| ClassSubject kept = inspector.clazz(pkg + ".anno.Kept"); |
| assertThat(kept, isPresent()); |
| assertThat(kept, not(isRenamed())); |
| // API entry is kept, hence @Metadata exists. |
| KmClassSubject kmClass = kept.getKmClass(); |
| assertThat(kmClass, isPresent()); |
| assertEquals(kept.getFinalDescriptor(), getDescriptorFromKotlinClassifier(kmClass.getName())); |
| // @Anno is kept. |
| String annoName = pkg + ".anno.Anno"; |
| AnnotationSubject anno = kept.annotation(annoName); |
| assertThat(anno, isPresent()); |
| |
| ClassSubject renamed = inspector.clazz(pkg + ".anno.Renamed"); |
| assertThat(renamed, isRenamed()); |
| // @Anno is kept. |
| anno = renamed.annotation(annoName); |
| assertThat(anno, isPresent()); |
| // @Metadata is kept even though the class is renamed. |
| kmClass = renamed.getKmClass(); |
| assertThat(kmClass, isPresent()); |
| assertEquals( |
| renamed.getFinalDescriptor(), getDescriptorFromKotlinClassifier(kmClass.getName())); |
| } |
| } |