This commit is contained in:
2022-02-08 04:19:11 +08:00
parent 589f7ae95f
commit 12a5593d25
4 changed files with 108 additions and 85 deletions

View File

@@ -38,7 +38,7 @@ import com.highcapable.yukihookapi.annotation.xposed.InjectYukiHookWithXposed
import java.io.File import java.io.File
/** /**
* 这是 YukiHook API 的自动生成处理类 - 核心基于 KSP * 这是 YukiHookAPI 的自动生成处理类 - 核心基于 KSP
* *
* 可以帮你快速生成 Xposed 入口类和包名 * 可以帮你快速生成 Xposed 入口类和包名
* *
@@ -65,9 +65,22 @@ class YukiHookXposedProcessor : SymbolProcessorProvider {
/** /**
* 创建一个环境方法体方便调用 * 创建一个环境方法体方便调用
* @param ignoredError 是否忽略错误 - 默认否
* @param env 方法体 * @param env 方法体
*/ */
private fun environment(env: SymbolProcessorEnvironment.() -> Unit) = runCatching { environment.apply(env) } private fun environment(ignoredError: Boolean = false, env: SymbolProcessorEnvironment.() -> Unit) {
if (ignoredError) runCatching { environment.apply(env) }
else environment.apply(env)
}
/**
* 终止并报错
* @param msg 错误消息
*/
private fun SymbolProcessorEnvironment.error(msg: String) {
logger.error(msg)
throw RuntimeException(msg)
}
override fun process(resolver: Resolver) = emptyList<KSAnnotated>().let { override fun process(resolver: Resolver) = emptyList<KSAnnotated>().let {
injectProcess(resolver) injectProcess(resolver)
@@ -80,80 +93,102 @@ class YukiHookXposedProcessor : SymbolProcessorProvider {
*/ */
private fun injectProcess(resolver: Resolver) = environment { private fun injectProcess(resolver: Resolver) = environment {
var injectOnce = true var injectOnce = true
resolver.getSymbolsWithAnnotation(InjectYukiHookWithXposed::class.java.name) resolver.getSymbolsWithAnnotation(InjectYukiHookWithXposed::class.java.name).apply {
.asSequence() /**
.filterIsInstance<KSClassDeclaration>().forEach { * 检索需要注入的类
if (injectOnce) * @param sourcePath 指定的 source 路径
if (it.superName == "YukiHookXposedInitProxy") { * @param modulePackageName 模块包名
injectAssets( */
codePath = (it.location as? FileLocation?)?.filePath ?: "", fun fetchKSClassDeclaration(sourcePath: String, modulePackageName: String) {
packageName = it.packageName.asString(), asSequence().filterIsInstance<KSClassDeclaration>().forEach {
className = it.simpleName.asString() if (injectOnce)
) if (it.superName == "YukiHookXposedInitProxy") {
injectClass(it.packageName.asString(), it.simpleName.asString()) injectAssets(
} else logger.error(message = "HookEntryClass \"${it.simpleName.asString()}\" must be implements YukiHookXposedInitProxy") codePath = (it.location as? FileLocation?)?.filePath ?: "",
else logger.error(message = "@InjectYukiHookWithXposed only can be use in once times") sourcePath = sourcePath,
/** 仅处理第一个标记的类 - 再次处理将拦截并报错 */ packageName = it.packageName.asString(),
injectOnce = false className = it.simpleName.asString(),
)
injectClass(it.packageName.asString(), it.simpleName.asString(), modulePackageName)
} else error(msg = "HookEntryClass \"${it.simpleName.asString()}\" must be implements YukiHookXposedInitProxy")
else error(msg = "@InjectYukiHookWithXposed only can be use in once times")
/** 仅处理第一个标记的类 - 再次处理将拦截并报错 */
injectOnce = false
}
} }
forEach {
it.annotations.forEach { e ->
val sourcePath = e.arguments[0].value.toString()
val modulePackageName = e.arguments[1].value.toString()
if ((modulePackageName.startsWith(".") ||
modulePackageName.endsWith(".") ||
!modulePackageName.contains(".") ||
modulePackageName.contains("..")) &&
modulePackageName.isNotEmpty()
) error(msg = "Invalid Module Package name \"$modulePackageName\"")
else fetchKSClassDeclaration(sourcePath, modulePackageName)
}
}
}
} }
/** /**
* 自动生成 Xposed assets 入口文件 * 自动生成 Xposed assets 入口文件
* @param codePath 注释类的完整代码文件路径 * @param codePath 注释类的完整代码文件路径
* @param sourcePath 指定的 source 路径
* @param packageName 包名 * @param packageName 包名
* @param className 类名 * @param className 类名
*/ */
private fun injectAssets(codePath: String, packageName: String, className: String) = environment { private fun injectAssets(codePath: String, sourcePath: String, packageName: String, className: String) =
runCatching { environment {
if (codePath.isBlank()) error("Project CodePath is empty") if (codePath.isBlank()) error("Project CodePath is empty")
val projectPath = when (File.separator) { val projectPath = when (File.separator) {
"\\" -> { "\\" -> sourcePath.replace("/", "\\")
if (codePath.contains("\\src\\main\\")) "/" -> sourcePath.replace("\\", "/")
codePath.split("\\src\\main\\")[0] + "\\src\\main\\" else -> kotlin.error("Unix File Separator unknown")
else error("Project source path must be ..\\src\\main\\..") }.let {
} if (codePath.contains(it))
"/" -> { codePath.split(it)[0] + it
if (codePath.contains("/src/main/")) else error(msg = "Project Source Path \"$it\" not matched")
codePath.split("/src/main/")[0] + "/src/main/"
else error("Project source path must be ../src/main/..")
}
else -> error("Unix File Separator unknown")
} }
File("$projectPath${File.separator}assets").also { assFile -> File("$projectPath${File.separator}assets").also { assFile ->
if (!assFile.exists() || !assFile.isDirectory) { if (File("$projectPath${File.separator}AndroidManifest.xml").exists()) {
assFile.delete() if (!assFile.exists() || !assFile.isDirectory) {
assFile.mkdirs() assFile.delete()
} assFile.mkdirs()
File("${assFile.absolutePath}${File.separator}xposed_init") }
.writeText(text = "$packageName.$className$xposedClassShortName") File("${assFile.absolutePath}${File.separator}xposed_init")
.writeText(text = "$packageName.$className$xposedClassShortName")
} else error(msg = "Project Source Path \"$sourcePath\" verify failed! Is this an Android Project?")
} }
}.onFailure {
logger.error(message = "Inject XposedAssets Failed! $it")
} }
}
/** /**
* 注入并生成指定类 * 注入并生成指定类
* @param packageName 包名 * @param packageName 包名
* @param className 类名 * @param className 类名
* @param modulePackageName 模块包名
*/ */
private fun injectClass(packageName: String, className: String) = environment { private fun injectClass(packageName: String, className: String, modulePackageName: String) =
var realPackageName = "unknown" environment(ignoredError = true) {
if (packageName.contains(".hook.")) if (modulePackageName.isNotBlank()) logger.warn(message = "You set the customize module package name to \"$modulePackageName\",please check for yourself if it is correct")
realPackageName = packageName.split(".hook.")[0] val realPackageName =
else logger.warn(message = "YukiHook cannot identify your App's package name,please refer to the wiki https://github.com/fankes/YukiHookAPI/wiki to fix the package name or manually configure the package name") modulePackageName.ifBlank {
codeGenerator.createNewFile(Dependencies.ALL_FILES, packageName, fileName = "$className$xposedClassShortName") if (packageName.contains(".hook."))
.apply { packageName.split(".hook.")[0]
/** 🤡 由于插入的代码量不大就不想用工具生成了 */ else error(msg = "YukiHook cannot identify your App's package name,please refer to the wiki https://github.com/fankes/YukiHookAPI/wiki to fix the package name or manually configure the package name")
}
codeGenerator.createNewFile(
Dependencies.ALL_FILES,
packageName,
fileName = "$className$xposedClassShortName"
).apply {
/** 插入 xposed_init 代码 */
write( write(
("@file:Suppress(\"EXPERIMENTAL_API_USAGE\")\n" + ("package $packageName\n" +
"\n" +
"package $packageName\n" +
"\n" + "\n" +
"import androidx.annotation.Keep\n" + "import androidx.annotation.Keep\n" +
"import com.highcapable.yukihookapi.YukiHookAPI\n" + "import com.highcapable.yukihookapi.YukiHookAPI\n" +
"import com.highcapable.yukihookapi.YukiHookAPI.Configs\n" +
"import com.highcapable.yukihookapi.hook.xposed.YukiHookModuleStatus\n" + "import com.highcapable.yukihookapi.hook.xposed.YukiHookModuleStatus\n" +
"import com.highcapable.yukihookapi.hook.log.loggerE\n" + "import com.highcapable.yukihookapi.hook.log.loggerE\n" +
"import de.robv.android.xposed.IXposedHookLoadPackage\n" + "import de.robv.android.xposed.IXposedHookLoadPackage\n" +
@@ -181,19 +216,14 @@ class YukiHookXposedProcessor : SymbolProcessorProvider {
" }.onFailure {\n" + " }.onFailure {\n" +
" loggerE(tag = \"YukiHookAPI\", msg = \"YukiHookAPI try to load HookEntryClass failed\", e = it)\n" + " loggerE(tag = \"YukiHookAPI\", msg = \"YukiHookAPI try to load HookEntryClass failed\", e = it)\n" +
" }\n" + " }\n" +
" YukiHookAPI.Configs.modulePackageName.ifEmpty {\n" + " if (lpparam.packageName == \"$realPackageName\")\n" +
" YukiHookAPI.Configs.modulePackageName = \"$realPackageName\"\n" + " XposedHelpers.findAndHookMethod(\n" +
" \"$realPackageName\"\n" + " YukiHookModuleStatus::class.java.name,\n" +
" }.also {\n" + " lpparam.classLoader,\n" +
" if (lpparam.packageName == it)\n" + " \"isActive\",\n" +
" XposedHelpers.findAndHookMethod(\n" + " object : XC_MethodReplacement() {\n" +
" YukiHookModuleStatus::class.java.name,\n" + " override fun replaceHookedMethod(param: MethodHookParam?) = true\n" +
" lpparam.classLoader,\n" + " })\n" +
" \"isActive\",\n" +
" object : XC_MethodReplacement() {\n" +
" override fun replaceHookedMethod(param: MethodHookParam?) = true\n" +
" })\n" +
" }\n" +
" YukiHookAPI.onXposedLoaded(lpparam)\n" + " YukiHookAPI.onXposedLoaded(lpparam)\n" +
" }\n" + " }\n" +
"}").toByteArray() "}").toByteArray()
@@ -201,6 +231,6 @@ class YukiHookXposedProcessor : SymbolProcessorProvider {
flush() flush()
close() close()
} }
} }
} }
} }

View File

@@ -40,7 +40,6 @@ import com.highcapable.yukihookapi.hook.factory.processName
import com.highcapable.yukihookapi.hook.log.loggerE import com.highcapable.yukihookapi.hook.log.loggerE
import com.highcapable.yukihookapi.hook.param.PackageParam import com.highcapable.yukihookapi.hook.param.PackageParam
import com.highcapable.yukihookapi.hook.param.wrapper.PackageParamWrapper import com.highcapable.yukihookapi.hook.param.wrapper.PackageParamWrapper
import com.highcapable.yukihookapi.hook.xposed.YukiHookModuleStatus
import de.robv.android.xposed.XposedBridge import de.robv.android.xposed.XposedBridge
import de.robv.android.xposed.callbacks.XC_LoadPackage import de.robv.android.xposed.callbacks.XC_LoadPackage
@@ -82,15 +81,6 @@ object YukiHookAPI {
* 请过滤 [debugTag] 即可找到每条日志 * 请过滤 [debugTag] 即可找到每条日志
*/ */
var isDebug = true var isDebug = true
/**
* Xposed Hook API 绑定的模块包名
*
* - 用于 [YukiHookModuleStatus] 判断模块激活状态
*
* 未写将自动生成
*/
var modulePackageName = ""
} }
/** /**

View File

@@ -29,15 +29,16 @@
package com.highcapable.yukihookapi.annotation.xposed package com.highcapable.yukihookapi.annotation.xposed
import com.highcapable.yukihookapi.YukiHookAPI
import com.highcapable.yukihookapi.hook.xposed.proxy.YukiHookXposedInitProxy import com.highcapable.yukihookapi.hook.xposed.proxy.YukiHookXposedInitProxy
/** /**
* 标识 YukiHook 注入 Xposed 入口的类注释 * 标识 YukiHook 注入 Xposed 入口的类注释
* *
* - 你的项目 source 目录必须为 ../src/main/ 后方内容根据你自行喜好定义即可 - 否则编译会报错 * - 你的项目 source 目录默认为 "src/main/" 可在 [sourcePath] 中进行自定义 - 自动处理程序将只检查 ..app/[sourcePath]/java.. 中间部分
* *
* - 你的 Hook 入口类(HookEntryClass) 需要按照此格式创建 --> 你的模块APP包名/hook/...可允许子包名存在.../你的入口类 * - 自动处理程序将自动在 ../[sourcePath]/assets/ 下建立 xposed_init 文件
*
* - 你的 Hook 入口类(HookEntryClass) 需要按照此格式创建 --> 你的模块 APP 包名/hook/...可允许子包名存在.../你的入口类
* *
* 例子com.example.module.hook.MainHook、com.example.module.hook.inject.MainInject、com.example.module.hook.custom.CustomClass * 例子com.example.module.hook.MainHook、com.example.module.hook.inject.MainInject、com.example.module.hook.custom.CustomClass
* *
@@ -47,15 +48,19 @@ import com.highcapable.yukihookapi.hook.xposed.proxy.YukiHookXposedInitProxy
* *
* - 你的模块包名将被这样识别:|com.example.module|.hook... * - 你的模块包名将被这样识别:|com.example.module|.hook...
* *
* - 若你不喜欢这样创建类 - 没问题 - 请在 [YukiHookAPI.Configs.modulePackageName] 填写模块包名即可 - 但不按照规则定义包名你将会收到编译警告 * - 若你不喜欢这样创建类 - 没问题 - 请在 [modulePackageName] 填写你的模块包名即可 - 默认会按照标准识别 - 失败编译会报错
* *
* - ⚡最后这一点很重要:请不要随意修改项目 ../src/main/assets/xposed_init 中的内容 - 否则可能会导致模块装载发生错误 * - 为了防止模块包名无法正常被识别 - 自定义 [modulePackageName] 会在编译时产生警告
* *
* 你必须将被注释的类继承于 [YukiHookXposedInitProxy] 接口实现 [YukiHookXposedInitProxy.onHook] 方法 - 否则编译会报错 * - ⚡最后这一点很重要:请不要随意修改项目 ../[sourcePath]/assets/xposed_init 中的内容 - 否则可能会导致模块无法装载
* *
* 只能拥有一个 Hook 入口 - 若存在多个注释编译会报错 * - ⚡你必须将被注释的类继承于 [YukiHookXposedInitProxy] 接口实现 [YukiHookXposedInitProxy.onHook] 方法 - 否则编译会报错
*
* - ⚡只能拥有一个 Hook 入口 - 若存在多个注释编译会报错
* *
* 详情请参考 [YukiHookAPI Wiki](https://github.com/fankes/YukiHookAPI/wiki) * 详情请参考 [YukiHookAPI Wiki](https://github.com/fankes/YukiHookAPI/wiki)
* @param sourcePath 你的项目 source 相对路径 - 默认为 ..src/main..
* @param modulePackageName 模块包名 - 使用标准路径可不填会自动生成
*/ */
@Target(AnnotationTarget.CLASS) @Target(AnnotationTarget.CLASS)
annotation class InjectYukiHookWithXposed annotation class InjectYukiHookWithXposed(val sourcePath: String = "src/main", val modulePackageName: String = "")

View File

@@ -50,8 +50,6 @@ import com.highcapable.yukihookapi.hook.factory.encase
* *
* ....debugTag = "自定义 TAG" * ....debugTag = "自定义 TAG"
* *
* ....modulePackageName = "模块包名/可选"
*
* ....isDebug = true * ....isDebug = true
* *
* } * }