Merge bed1dec33b7cbac9dc1e2a169288425c8e24b1bb on remote branch

Change-Id: I82d4fb6229331befc565982eff3a612856e980b7
diff --git a/Android.bp b/Android.bp
index c1340d9..ea48443 100644
--- a/Android.bp
+++ b/Android.bp
@@ -16,7 +16,10 @@
     name: "jarjar",
     manifest: "manifest.txt",
 
-    srcs: ["src/main/**/*.java"],
+    srcs: [
+        "src/android/**/*.java",
+        "src/main/**/*.java",
+    ],
     java_resource_dirs: ["res"],
 
     static_libs: [
diff --git a/src/android/com/android/jarjar/RemoveAndroidCompatAnnotationsJarTransformer.java b/src/android/com/android/jarjar/RemoveAndroidCompatAnnotationsJarTransformer.java
new file mode 100644
index 0000000..990e048
--- /dev/null
+++ b/src/android/com/android/jarjar/RemoveAndroidCompatAnnotationsJarTransformer.java
@@ -0,0 +1,120 @@
+/*
+ * Copyright (C) 2020 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.jarjar;
+
+import com.tonicsystems.jarjar.util.JarTransformer;
+
+import org.objectweb.asm.AnnotationVisitor;
+import org.objectweb.asm.ClassVisitor;
+import org.objectweb.asm.FieldVisitor;
+import org.objectweb.asm.MethodVisitor;
+import org.objectweb.asm.Opcodes;
+import org.objectweb.asm.commons.Remapper;
+
+import java.util.Set;
+import java.util.function.Supplier;
+
+/**
+ * A transformer that removes annotations from repackaged classes.
+ */
+public final class RemoveAndroidCompatAnnotationsJarTransformer extends JarTransformer {
+
+    private static final Set<String> REMOVE_ANNOTATIONS = Set.of(
+            "Landroid/compat/annotation/UnsupportedAppUsage;");
+
+    private final Remapper remapper;
+
+    public RemoveAndroidCompatAnnotationsJarTransformer(Remapper remapper) {
+        this.remapper = remapper;
+    }
+
+    protected ClassVisitor transform(ClassVisitor classVisitor) {
+        return new AnnotationRemover(classVisitor);
+    }
+
+    private class AnnotationRemover extends ClassVisitor {
+
+        private boolean isClassRemapped;
+
+        AnnotationRemover(ClassVisitor cv) {
+            super(Opcodes.ASM6, cv);
+        }
+
+        @Override
+        public void visit(int version, int access, String name, String signature, String superName,
+                String[] interfaces) {
+            String newName = remapper.map(name);
+            // On every new class header visit, remember whether the class is repackaged.
+            isClassRemapped = newName != null && !newName.equals(name);
+            super.visit(version, access, name, signature, superName, interfaces);
+        }
+
+        @Override
+        public AnnotationVisitor visitAnnotation(String descriptor, boolean visible) {
+            return visitAnnotationCommon(descriptor,
+                    () -> super.visitAnnotation(descriptor, visible));
+        }
+
+        @Override
+        public FieldVisitor visitField(int access, String name, String descriptor, String signature,
+                Object value) {
+            FieldVisitor superVisitor =
+                    super.visitField(access, name, descriptor, signature, value);
+            return new FieldVisitor(Opcodes.ASM6, superVisitor) {
+                @Override
+                public AnnotationVisitor visitAnnotation(String descriptor, boolean visible) {
+                    return visitAnnotationCommon(descriptor,
+                            () -> super.visitAnnotation(descriptor, visible));
+
+                }
+            };
+        }
+
+        @Override
+        public MethodVisitor visitMethod(int access, String name, String descriptor,
+                String signature, String[] exceptions) {
+            MethodVisitor superVisitor =
+                    super.visitMethod(access, name, descriptor, signature, exceptions);
+            return new MethodVisitor(Opcodes.ASM6, superVisitor) {
+                @Override
+                public AnnotationVisitor visitAnnotation(String descriptor, boolean visible) {
+                    return visitAnnotationCommon(descriptor,
+                            () -> super.visitAnnotation(descriptor, visible));
+                }
+            };
+        }
+
+        /**
+         * Create an {@link AnnotationVisitor} that removes any annotations from {@link
+         * #REMOVE_ANNOTATIONS} if the class is being repackaged.
+         *
+         * <p>For the annotations to be dropped correctly, do not visit the annotation beforehand,
+         * provide a supplier instead.
+         */
+        private AnnotationVisitor visitAnnotationCommon(String annotation,
+                Supplier<AnnotationVisitor> defaultVisitorSupplier) {
+            if (isClassRemapped && REMOVE_ANNOTATIONS.contains(annotation)) {
+                return null;
+            }
+            // Only get() the default AnnotationVisitor if the annotation is to be included.
+            // Invoking super.visitAnnotation(descriptor, visible) causes the annotation to be
+            // included in the output even if the resulting AnnotationVisitor is not returned or
+            // used.
+            return defaultVisitorSupplier.get();
+        }
+    }
+}
diff --git a/src/main/com/tonicsystems/jarjar/Main.java b/src/main/com/tonicsystems/jarjar/Main.java
index 117bdbb..1eaa3e3 100644
--- a/src/main/com/tonicsystems/jarjar/Main.java
+++ b/src/main/com/tonicsystems/jarjar/Main.java
@@ -90,7 +90,11 @@
     List<PatternElement> rules = RulesFileParser.parse(rulesFile);
     boolean verbose = Boolean.getBoolean("verbose");
     boolean skipManifest = Boolean.getBoolean("skipManifest");
-    MainProcessor proc = new MainProcessor(rules, verbose, skipManifest);
+    // ANDROID-BEGIN: b/146418363 Add an Android-specific transformer to strip compat annotation
+    boolean removeAndroidCompatAnnotations = Boolean.getBoolean("removeAndroidCompatAnnotations");
+    MainProcessor proc = new MainProcessor(rules, verbose, skipManifest,
+            removeAndroidCompatAnnotations);
+    // ANDROID-END: b/146418363 Add an Android-specific transformer to strip compat annotation
     StandaloneJarProcessor.run(inJar, outJar, proc);
     proc.strip(outJar);
   }
diff --git a/src/main/com/tonicsystems/jarjar/MainProcessor.java b/src/main/com/tonicsystems/jarjar/MainProcessor.java
index de15924..2778cc5 100644
--- a/src/main/com/tonicsystems/jarjar/MainProcessor.java
+++ b/src/main/com/tonicsystems/jarjar/MainProcessor.java
@@ -21,14 +21,23 @@
 import java.io.IOException;
 import java.util.*;
 
+import com.android.jarjar.RemoveAndroidCompatAnnotationsJarTransformer;
+
 class MainProcessor implements JarProcessor
 {
     private final boolean verbose;
     private final JarProcessorChain chain;
     private final KeepProcessor kp;
     private final Map<String, String> renames = new HashMap<String, String>();
-    
+
+    // ANDROID-BEGIN: b/146418363 Add an Android-specific transformer to strip compat annotation
     public MainProcessor(List<PatternElement> patterns, boolean verbose, boolean skipManifest) {
+        this(patterns, verbose, skipManifest, false /* removeAndroidCompatAnnotations */);
+    }
+
+    public MainProcessor(List<PatternElement> patterns, boolean verbose, boolean skipManifest,
+            boolean removeAndroidCompatAnnotations) {
+        // ANDROID-END: b/146418363 Add an Android-specific transformer to strip compat annotation
         this.verbose = verbose;
         List<Zap> zapList = new ArrayList<Zap>();
         List<Rule> ruleList = new ArrayList<Rule>();
@@ -52,6 +61,10 @@
         if (kp != null)
             processors.add(kp);
         processors.add(new ZapProcessor(zapList));
+        // ANDROID-BEGIN: b/146418363 Add an Android-specific transformer to strip compat annotation
+        if (removeAndroidCompatAnnotations)
+            processors.add(new RemoveAndroidCompatAnnotationsJarTransformer(pr));
+        // ANDROID-END: b/146418363 Add an Android-specific transformer to strip compat annotation
         processors.add(new JarTransformerChain(new RemappingClassTransformer[]{ new RemappingClassTransformer(pr) }));
         processors.add(new ResourceProcessor(pr));
         chain = new JarProcessorChain(processors.toArray(new JarProcessor[processors.size()]));