自定义Gradle插件 + ASM 实现无埋点监控

Author: 1170762202@qq.com

.idea/compiler.xml

@@ -1,6 +1,7 @@
 apply plugin: 'com.android.application'
 apply plugin: 'android-aspectjx'
 apply plugin: 'com.jakewharton.butterknife'
+apply plugin: 'com.zlx.plugin_lifecycle'
 
 
 def keystorePropertiesFile = rootProject.file("keystore.properties")

.idea/gradle.xml

@@ -3,6 +3,10 @@
 import com.zlx.module_base.BaseApplication;
 import com.zlx.module_base.config.ModuleLifecycleConfig;
 
+import org.greenrobot.eventbus.EventBus;
+import org.greenrobot.eventbus.Subscribe;
+import org.greenrobot.eventbus.ThreadMode;
+
 
 /**
  * @date: 2019\3\8 0008
@@ -17,9 +21,15 @@ public class MyApp extends BaseApplication {
     public void onCreate() {
         super.onCreate();
         ModuleLifecycleConfig.getInstance().initModuleAhead(this);
+        EventBus.getDefault().register(this);
     }
 
 
+    @Subscribe(threadMode = ThreadMode.MAIN)
+    public void onMessageEvent(Integer event) {
+        System.out.println("收到消息" + event.toString());
+    }
 
+    ;
 
 }

.idea/jarRepositories.xml

@@ -5,12 +5,15 @@ buildscript {
     repositories {
         google()
         jcenter()
-
+        maven {
+            url uri('./plugin-lifecycle/plugin-lifecycle')
+        }
     }
     dependencies {
         classpath "com.android.tools.build:gradle:4.0.2"
         classpath 'com.jakewharton:butterknife-gradle-plugin:10.2.2'
         classpath 'com.hujiang.aspectjx:gradle-android-plugin-aspectjx:2.0.10'
+        classpath 'com.zlx.plugin_lifecycle:plugin-lifecycle:0.0.1'
     }
 }
 

.idea/misc.xml

@@ -0,0 +1,8 @@
+## This file must *NOT* be checked into Version Control Systems,
+# as it contains information specific to your local configuration.
+#
+# Location of the SDK. This is only used by Gradle.
+# For customization when using a Version Control System, please read the
+# header note.
+#Thu Sep 10 14:41:54 CST 2020
+sdk.dir=D\:\\sdk\\Android

app/build.gradle

@@ -0,0 +1,33 @@
+apply plugin: 'groovy'
+apply plugin: 'maven'
+dependencies {
+    //gradle sdk
+    implementation gradleApi()
+    //groovy sdk
+    implementation localGroovy()
+    implementation 'com.android.tools.build:gradle:4.0.2'
+
+}
+
+
+repositories {
+    mavenCentral()
+    jcenter()
+}
+//group和version在后面使用自定义插件的时候会用到
+group='com.zlx.plugin_lifecycle'
+version='0.0.1'
+
+
+uploadArchives {
+    repositories {
+        mavenDeployer {
+            //提交到远程服务器：
+            // repository(url: "http://www.xxx.com/repos") {
+            //    authentication(userName: "admin", password: "admin")
+            // }
+            //本地的Maven地址:当前工程下
+            repository(url: uri('./plugin-lifecycle'))
+        }
+    }
+}

app/src/main/java/com/zlx/sharelive/MyApp.java

@@ -0,0 +1 @@
+3311e9e9a64a4bd57ea08f7f5bb2751a

build.gradle

@@ -0,0 +1 @@
+674fcd05682dd733cbf06626b6f4983dfed910e0

local.properties

@@ -0,0 +1,16 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd" xmlns="http://maven.apache.org/POM/4.0.0"
+    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
+  <modelVersion>4.0.0</modelVersion>
+  <groupId>com.zlx.plugin_lifecycle</groupId>
+  <artifactId>plugin-lifecycle</artifactId>
+  <version>0.0.1</version>
+  <dependencies>
+    <dependency>
+      <groupId>com.android.tools.build</groupId>
+      <artifactId>gradle</artifactId>
+      <version>4.0.2</version>
+      <scope>runtime</scope>
+    </dependency>
+  </dependencies>
+</project>

plugin-lifecycle/build.gradle

@@ -0,0 +1 @@
+51381e0efaf3e0daa4acb0fe9a8ff218

plugin-lifecycle/plugin-lifecycle/com/zlx/plugin_lifecycle/plugin-lifecycle/0.0.1/plugin-lifecycle-0.0.1.jar

@@ -0,0 +1 @@
+f3eea6d9164361fef3395c6d3e116019c78a6b38

plugin-lifecycle/plugin-lifecycle/com/zlx/plugin_lifecycle/plugin-lifecycle/0.0.1/plugin-lifecycle-0.0.1.jar.md5

@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<metadata>
+  <groupId>com.zlx.plugin_lifecycle</groupId>
+  <artifactId>plugin-lifecycle</artifactId>
+  <versioning>
+    <release>0.0.1</release>
+    <versions>
+      <version>0.0.1</version>
+    </versions>
+    <lastUpdated>20210329090202</lastUpdated>
+  </versioning>
+</metadata>

plugin-lifecycle/plugin-lifecycle/com/zlx/plugin_lifecycle/plugin-lifecycle/0.0.1/plugin-lifecycle-0.0.1.jar.sha1

@@ -0,0 +1 @@
+e60846202f41c8a105ba9b8c088d4167

plugin-lifecycle/plugin-lifecycle/com/zlx/plugin_lifecycle/plugin-lifecycle/0.0.1/plugin-lifecycle-0.0.1.pom

@@ -0,0 +1 @@
+ffbf6b927ad697cba2562508336ce27505708975

plugin-lifecycle/plugin-lifecycle/com/zlx/plugin_lifecycle/plugin-lifecycle/0.0.1/plugin-lifecycle-0.0.1.pom.md5

@@ -0,0 +1,169 @@
+package com.zlx.plugin_lifecycle
+
+import com.android.annotations.NonNull
+import com.android.build.api.transform.*
+import com.android.build.gradle.AppExtension
+import com.android.build.gradle.internal.pipeline.TransformManager
+import com.zlx.plugin_lifecycle.LifecycleClassVisitor
+import org.apache.commons.codec.digest.DigestUtils
+import org.apache.commons.io.FileUtils
+import org.apache.commons.io.IOUtils
+import org.gradle.api.Plugin
+import org.gradle.api.Project
+import org.objectweb.asm.ClassReader
+import org.objectweb.asm.ClassVisitor
+import org.objectweb.asm.ClassWriter
+
+import java.util.jar.JarEntry
+import java.util.jar.JarFile
+import java.util.jar.JarOutputStream
+import java.util.zip.ZipEntry
+
+import static org.objectweb.asm.ClassReader.EXPAND_FRAMES
+
+class LifecyclePlugin extends Transform implements Plugin<Project> {
+
+    @Override
+    void apply(Project project) {
+        //registerTransform
+        def android = project.extensions.getByType(AppExtension)
+        android.registerTransform(this)
+    }
+
+    @Override
+    String getName() {
+        return "LifecyclePlugin"
+    }
+
+    @Override
+    Set<QualifiedContent.ContentType> getInputTypes() {
+        return TransformManager.CONTENT_CLASS
+    }
+
+    @Override
+    Set<? super QualifiedContent.Scope> getScopes() {
+        return TransformManager.SCOPE_FULL_PROJECT
+    }
+
+    @Override
+    boolean isIncremental() {
+        return false
+    }
+
+    @Override
+    void transform(@NonNull TransformInvocation transformInvocation) {
+        println '--------------- LifecyclePlugin visit start --------------- '
+        def startTime = System.currentTimeMillis()
+        Collection<TransformInput> inputs = transformInvocation.inputs
+        TransformOutputProvider outputProvider = transformInvocation.outputProvider
+        //删除之前的输出
+        if (outputProvider != null)
+            outputProvider.deleteAll()
+        //遍历inputs
+        inputs.each { TransformInput input ->
+            //遍历directoryInputs
+            input.directoryInputs.each { DirectoryInput directoryInput ->
+                //处理directoryInputs
+                handleDirectoryInput(directoryInput, outputProvider)
+            }
+
+            //遍历jarInputs
+            input.jarInputs.each { JarInput jarInput ->
+                //处理jarInputs
+                handleJarInputs(jarInput, outputProvider)
+            }
+        }
+        def cost = (System.currentTimeMillis() - startTime) / 1000
+        println '--------------- LifecyclePlugin visit end --------------- '
+        println "LifecyclePlugin cost ： $cost s"
+    }
+
+    /**
+     * 处理文件目录下的class文件
+     */
+    static void handleDirectoryInput(DirectoryInput directoryInput, TransformOutputProvider outputProvider) {
+        //是否是目录
+        if (directoryInput.file.isDirectory()) {
+            //列出目录所有文件（包含子文件夹，子文件夹内文件）
+            directoryInput.file.eachFileRecurse { File file ->
+                def name = file.name
+                if (name.endsWith(".class") && !name.startsWith("R\$")
+                        && !"R.class".equals(name) && !"BuildConfig.class".equals(name)
+                        && "androidx/fragment/app/FragmentActivity.class".equals(name)) {
+                    println '----------- deal with "class" file <' + name + '> -----------'
+                    ClassReader classReader = new ClassReader(file.bytes)
+                    ClassWriter classWriter = new ClassWriter(classReader, ClassWriter.COMPUTE_MAXS)
+                    ClassVisitor cv = new LifecycleClassVisitor(classWriter)
+                    classReader.accept(cv, EXPAND_FRAMES)
+                    byte[] code = classWriter.toByteArray()
+                    FileOutputStream fos = new FileOutputStream(
+                            file.parentFile.absolutePath + File.separator + name)
+                    fos.write(code)
+                    fos.close()
+                }
+            }
+        }
+        //处理完输入文件之后，要把输出给下一个任务
+        def dest = outputProvider.getContentLocation(directoryInput.name,
+                directoryInput.contentTypes, directoryInput.scopes,
+                Format.DIRECTORY)
+        FileUtils.copyDirectory(directoryInput.file, dest)
+    }
+
+    /**
+     * 处理Jar中的class文件
+     */
+    static void handleJarInputs(JarInput jarInput, TransformOutputProvider outputProvider) {
+        if (jarInput.file.getAbsolutePath().endsWith(".jar")) {
+            //重名名输出文件,因为可能同名,会覆盖
+            def jarName = jarInput.name
+            def md5Name = DigestUtils.md5Hex(jarInput.file.getAbsolutePath())
+            if (jarName.endsWith(".jar")) {
+                jarName = jarName.substring(0, jarName.length() - 4)
+            }
+            JarFile jarFile = new JarFile(jarInput.file)
+            Enumeration enumeration = jarFile.entries()
+            File tmpFile = new File(jarInput.file.getParent() + File.separator + "classes_temp.jar")
+            //避免上次的缓存被重复插入
+            if (tmpFile.exists()) {
+                tmpFile.delete()
+            }
+            JarOutputStream jarOutputStream = new JarOutputStream(new FileOutputStream(tmpFile))
+            //用于保存
+            while (enumeration.hasMoreElements()) {
+                JarEntry jarEntry = (JarEntry) enumeration.nextElement()
+                String entryName = jarEntry.getName()
+                ZipEntry zipEntry = new ZipEntry(entryName)
+                InputStream inputStream = jarFile.getInputStream(jarEntry)
+
+                //插桩class
+                if (entryName.endsWith(".class") && !entryName.startsWith("R\$")
+                        && !"R.class".equals(entryName) && !"BuildConfig.class".equals(entryName)
+                        && ("androidx/fragment/app/FragmentActivity.class".equals(entryName) ||
+                        "androidx/fragment/app/Fragment.class".equals(entryName) ||
+                        "com/zlx/module_home/fragment/HomeFg.class".equals(entryName))) {
+                    //class文件处理
+                    println '----------- deal with "jar" class file <' + entryName + '> -----------'
+                    jarOutputStream.putNextEntry(zipEntry)
+                    ClassReader classReader = new ClassReader(IOUtils.toByteArray(inputStream))
+                    ClassWriter classWriter = new ClassWriter(classReader, ClassWriter.COMPUTE_MAXS)
+                    ClassVisitor cv = new LifecycleClassVisitor(classWriter)
+                    classReader.accept(cv, EXPAND_FRAMES)
+                    byte[] code = classWriter.toByteArray()
+                    jarOutputStream.write(code)
+                } else {
+                    jarOutputStream.putNextEntry(zipEntry)
+                    jarOutputStream.write(IOUtils.toByteArray(inputStream))
+                }
+                jarOutputStream.closeEntry()
+            }
+            //结束
+            jarOutputStream.close()
+            jarFile.close()
+            def dest = outputProvider.getContentLocation(jarName + md5Name,
+                    jarInput.contentTypes, jarInput.scopes, Format.JAR)
+            FileUtils.copyFile(tmpFile, dest)
+            tmpFile.delete()
+        }
+    }
+}