RePlugin字节码修改流程分析

关键代码分析：RePlugin 字节码修改实现

根据 BaseInjector.groovy 的代码结构和 RePlugin 框架特性，字节码修改的核心逻辑通常通过以下步骤实现：

---

1. 核心类职责

类/接口	说明
IClassInjector	定义字节码注入接口（如 injectClass() onHookCompleted() 等方法）
BaseInjector	注入器的抽象基类，提供公共属性（项目对象、变体目录等）
具体子类	如 ActivityInjector ServiceInjector，实现具体组件的字节码修改逻辑

---

2. 字节码操作流程

Transform BaseInjector Javassist 遍历所有 .class 文件 加载 CtClass 对象 判断是否需要修改（如目标类名匹配） 插入/修改方法体 添加字段或接口 返回修改后的字节码 alt [需要修改] loop [每个类文件] 回写修改后的 .class 文件 Transform BaseInjector Javassist

---

3. 关键代码实现（伪代码）

示例：Hook ActivityThread 的 handleBindApplication 方法

// 子类实现（如 ApplicationInjector.groovy）
class ApplicationInjector extends BaseInjector {

    @Override
    void injectClass(CtClass ctClass) {
        if (ctClass.name == "android.app.ActivityThread") {
            CtMethod method = ctClass.getDeclaredMethod("handleBindApplication")
            method.insertBefore('''
                // 插入插件化初始化逻辑
                if (RePluginInternal.init(appContext)) {
                    RePluginInternal.attachBaseContext(appContext, data.info);
                }
            ''')
        }
    }
}

BaseInjector 核心逻辑扩展

abstract class BaseInjector implements IClassInjector {

    // 公共工具方法：查找并返回目标方法
    protected CtMethod findMethod(CtClass ctClass, String methodName, String desc) {
        return ctClass.getDeclaredMethod(methodName, desc)
    }

    // 公共工具方法：在方法体开头插入代码
    protected void insertBefore(CtMethod method, String code) {
        method.insertBefore(code)
    }
}

---

4. 典型 Hook 点

RePlugin 通常会修改以下系统类：

目标类	修改点	目的
ActivityThread	handleBindApplication	在应用初始化时注入插件框架
Instrumentation	newActivity/callActivityOnCreate	接管 Activity 生命周期，实现插件组件加载
LoadedApk	makeApplication	替换 ClassLoader 为 PluginClassLoader
PackageManagerService	getActivityInfo	欺骗系统，使插件组件通过合法性校验

---

5. 技术要点

1. Javassist 使用模式：

   // 典型操作步骤
   CtClass ctClass = pool.getCtClass(className)
   CtMethod method = ctClass.getDeclaredMethod("targetMethod")
   method.insertBefore("{ System.out.println(\"Injected!\"); }")
   ctClass.writeFile(outputDir)

2. Transform 集成：

   // 在自定义 Transform 中调用注入器
   void transform(...) {
     inputs.each { input ->
       input.jarInput.file.each { jar ->
         // 解压 JAR 并处理 .class 文件
       }
       input.directoryInput.file.each { dir ->
         // 遍历目录中的 .class 文件
         new BaseInjector().injectClass(file)
       }
     }
   }

3. 变体处理：

   // 通过 variantDir 区分不同构建变体（如 debug/release）
   String fullPath = "$project.buildDir/intermediates/classes/$variantDir/"

---

6. 注意事项

• 兼容性：不同 Android 版本中系统类结构可能变化，需通过版本判断实现差异化注入

• 性能优化 ：

  • 使用 @Poolable 注解复用 CtClass 对象
  • 通过缓存机制避免重复解析类文件

• 调试支持 ：

  // 输出修改后的字节码到 build/replugin 目录
  ctClass.debugWriteFile("build/replugin")

字节码修改过程时序图

基于 RePlugin 框架的字节码修改逻辑，以下是其典型的工作流程时序图：

---

时序图描述

Gradle RePluginHostPlugin Transform BaseInjector Javassist FileSystem 应用插件 注册自定义 Transform 开始构建任务 遍历所有 .class 文件 调用注入器处理类 加载 CtClass 对象 判断是否需要修改（如目标类名匹配） 插入/修改方法体 添加字段或接口 返回修改后的字节码 alt [需要修改] 返回修改后的字节码 loop [每个类文件] 回写修改后的 .class 文件 完成 Transform 任务 继续后续构建流程（打包、签名等） Gradle RePluginHostPlugin Transform BaseInjector Javassist FileSystem

---

关键步骤解析

1. 插件注册：

   • Gradle 应用 RePluginHostPlugin 插件。
   • 插件注册自定义 Transform 到 AGP 构建流程中。

2. Transform 启动：

   • Gradle 开始构建任务，触发自定义 Transform。
   • Transform 遍历所有 .class 文件（包括项目代码和依赖库）。

3. 字节码修改：

   • 对于每个 .class 文件，调用 BaseInjector 进行处理。
   • 使用 Javassist 加载 CtClass 对象，判断是否需要修改。
   • 如果需要修改，插入或修改方法体（如 Hook 系统方法）。

4. 回写文件：

   • 将修改后的字节码回写到文件系统。
   • 完成 Transform 任务，Gradle 继续后续构建流程。

---

示例场景：Hook ActivityThread

以下是修改 ActivityThread 类的伪代码流程：

// 在 BaseInjector 子类中实现
class ActivityInjector extends BaseInjector {

    @Override
    void injectClass(CtClass ctClass) {
        if (ctClass.name == "android.app.ActivityThread") {
            // 查找目标方法
            CtMethod method = ctClass.getDeclaredMethod("handleBindApplication")
            // 在方法体开头插入插件化初始化逻辑
            method.insertBefore('''
                if (RePluginInternal.init(appContext)) {
                    RePluginInternal.attachBaseContext(appContext, data.info);
                }
            ''')
        }
    }
}

---

技术要点

1. Transform API：

   • 通过 Transform 干预构建流程，修改字节码。
   • 支持处理 JAR 文件和目录中的 .class 文件。

2. Javassist 操作：

   • 使用 CtClass 表示类文件，CtMethod 表示方法。
   • 支持插入代码、添加字段、实现接口等操作。

3. 性能优化：

   • 通过缓存机制避免重复加载类文件。
   • 使用增量构建减少不必要的字节码修改。

---