HighCapable/YukiHookAPI
5
0
Fork 0
mirror of https://github.com/HighCapable/YukiHookAPI.git synced 2025-09-04 17:55:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
Fankesyooni
2022-02-08 04:19:11 +08:00
parent 589f7ae95f
commit 12a5593d25
4 changed files with 108 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

160
yukihookapi-ksp-xposed/src/api/kotlin/com/highcapable/yukihookapi_ksp_xposed/YukiHookXposedProcessor.kt
View File

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