HighCapable/YukiHookAPI
5
0
Fork 0
mirror of https://github.com/HighCapable/YukiHookAPI.git synced 2025-09-06 02:35:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Modify make hooker safety

Browse Source
This commit is contained in:
fankesyooni
2022-07-29 22:34:38 +08:00
parent 2954b70dc0
commit 34d7d00b87
2 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
yukihookapi/src/api/kotlin/com/highcapable/yukihookapi/hook/core/YukiMemberHookCreater.kt
View File

@@ -524,11 +524,11 @@ class YukiMemberHookCreater(@PublishedApi internal val packageParam: PackagePara
(if (isReplaceHookMode) YukiHookHelper.hookMethod(member, replaceMent)
else YukiHookHelper.hookMethod(member, beforeAfterHook)).also {
when {
it.first.member == null -> error("Hook Member [$member] failed")
it.second -> onAlreadyHookedCallback?.invoke(it.first.member!!)
it.first?.member == null -> error("Hook Member [$member] failed")
it.second -> onAlreadyHookedCallback?.invoke(it.first?.member!!)
else -> {
memberUnhook = it.first
onHookedCallback?.invoke(it.first.member!!)
onHookedCallback?.invoke(it.first?.member!!)
}
}
}

20
yukihookapi/src/api/kotlin/com/highcapable/yukihookapi/hook/xposed/bridge/factory/YukiHookFactory.kt
View File

@@ -28,6 +28,7 @@
package com.highcapable.yukihookapi.hook.xposed.bridge.factory
import com.highcapable.yukihookapi.hook.param.wrapper.HookParamWrapper
import com.highcapable.yukihookapi.hook.xposed.bridge.YukiHookBridge
import de.robv.android.xposed.XC_MethodHook
import de.robv.android.xposed.XposedBridge
import java.lang.reflect.Constructor
@@ -108,18 +109,20 @@ internal object YukiHookHelper {
* 对接 [XposedBridge.hookMethod]
* @param hookMethod 需要 Hook 的方法、构造方法
* @param callback 回调
* @return [Pair] - ([YukiMemberHook.Unhook],[Boolean] 是否已经 Hook)
* @return [Pair] - ([YukiMemberHook.Unhook] or null,[Boolean] 是否已经 Hook)
*/
internal fun hookMethod(hookMethod: Member?, callback: YukiHookCallback): Pair<YukiMemberHook.Unhook, Boolean> {
internal fun hookMethod(hookMethod: Member?, callback: YukiHookCallback): Pair<YukiMemberHook.Unhook?, Boolean> {
runCatching {
YukiHookedMembers.hookedMembers.takeIf { it.isNotEmpty() }?.forEach {
if (it.member.toString() == hookMethod.toString()) return@runCatching it
}
}
return YukiMemberHook.Unhook.wrapper(XposedBridge.hookMethod(hookMethod, callback.compat())).let {
YukiHookedMembers.hookedMembers.add(it)
Pair(it, false)
}
return if (YukiHookBridge.hasXposedBridge)
YukiMemberHook.Unhook.wrapper(XposedBridge.hookMethod(hookMethod, callback.compat())).let {
YukiHookedMembers.hookedMembers.add(it)
Pair(it, false)
}
else Pair(null, false)
}
/**
@@ -142,6 +145,7 @@ internal object YukiHookHelper {
YukiHookedMembers.hookedQueueMethods[allMethodsName]?.forEach { e -> it.add(e) }
return@also
}
if (YukiHookBridge.hasXposedBridge.not()) return@also
XposedBridge.hookAllMethods(hookClass, methodName, callback.compat()).takeIf { e -> e.isNotEmpty() }
?.forEach { e -> it.add(YukiMemberHook.Unhook.wrapper(e, allMethodsName)) }
YukiHookedMembers.hookedQueueMethods[allMethodsName] = it
@@ -166,6 +170,7 @@ internal object YukiHookHelper {
YukiHookedMembers.hookedQueueConstructors[allConstructorsName]?.forEach { e -> it.add(e) }
return@also
}
if (YukiHookBridge.hasXposedBridge.not()) return@also
XposedBridge.hookAllConstructors(hookClass, callback.compat()).takeIf { e -> e.isNotEmpty() }
?.forEach { e -> it.add(YukiMemberHook.Unhook.wrapper(e, allConstructorsName)) }
YukiHookedMembers.hookedQueueConstructors[allConstructorsName] = it
@@ -185,7 +190,7 @@ internal object YukiHookHelper {
val isHookedMember = YukiHookedMembers.hookedMembers.any { it.member.toString() == member.toString() }
val isQueueMethod = YukiHookedMembers.hookedQueueMethods.any { it.value.any { e -> e.member.toString() == member.toString() } }
val isQueueConstructor = YukiHookedMembers.hookedQueueConstructors.any { it.value.any { e -> e.member.toString() == member.toString() } }
return if (isHookedMember || isQueueMethod || isQueueConstructor)
return if (YukiHookBridge.hasXposedBridge && (isHookedMember || isQueueMethod || isQueueConstructor))
XposedBridge.invokeOriginalMethod(member, instance, args)
else null
}
@@ -308,6 +313,7 @@ internal abstract class YukiMemberHook(override val priority: Int = YukiHookPrio
* @param isNeedRemove 是否需要从 [YukiHookedMembers.hookedMembers] 中移除
*/
private fun unhook(isNeedRemove: Boolean = true) {
if (YukiHookBridge.hasXposedBridge.not()) return
instance.unhook()
if (isNeedRemove) runCatching { YukiHookedMembers.hookedMembers.remove(this) }
}