fix(mapper): default media item mapper

This commit is contained in:
rhunk 2024-08-27 14:00:57 +02:00
parent c7745e206a
commit 7908fe25f1
2 changed files with 13 additions and 7 deletions

View File

@ -5,6 +5,7 @@ import android.os.FileObserver
import com.google.gson.JsonParser
import me.rhunk.snapenhance.core.event.events.impl.SendMessageWithContentEvent
import me.rhunk.snapenhance.core.features.Feature
import me.rhunk.snapenhance.core.util.dataBuilder
import me.rhunk.snapenhance.core.util.hook.HookStage
import me.rhunk.snapenhance.core.util.hook.hookConstructor
import me.rhunk.snapenhance.core.util.ktx.setObjectField
@ -47,9 +48,11 @@ class BypassVideoLengthRestriction :
}
context.mappings.useMapper(DefaultMediaItemMapper::class) {
defaultMediaItem.getAsClass()?.hookConstructor(HookStage.BEFORE) { param ->
defaultMediaItemClass.getAsClass()?.hookConstructor(HookStage.AFTER) { param ->
//set the video length argument
param.setArg(5, -1L)
param.thisObject<Any>().dataBuilder {
set(defaultMediaItemDurationMsField.getAsString()!!, -1L)
}
}
}
}

View File

@ -4,11 +4,13 @@ import me.rhunk.snapenhance.mapper.AbstractClassMapper
import me.rhunk.snapenhance.mapper.ext.findConstString
import me.rhunk.snapenhance.mapper.ext.getClassName
import me.rhunk.snapenhance.mapper.ext.isAbstract
import me.rhunk.snapenhance.mapper.ext.searchNextFieldReference
class DefaultMediaItemMapper : AbstractClassMapper("DefaultMediaItem") {
val cameraRollMediaId = classReference("cameraRollMediaIdClass")
val durationMsField = string("durationMsField")
val defaultMediaItem = classReference("defaultMediaItemClass")
val defaultMediaItemClass = classReference("defaultMediaItemClass")
val defaultMediaItemDurationMsField = string("defaultMediaItemDurationMsField")
init {
mapper {
@ -31,10 +33,11 @@ class DefaultMediaItemMapper : AbstractClassMapper("DefaultMediaItem") {
if (!superClass.isAbstract() || superClass.interfaces.isEmpty() || superClass.interfaces[0] != "Ljava/lang/Comparable;") continue
if (clazz.methods.none { it.returnType == "Landroid/net/Uri;" }) continue
val constructorParameters = clazz.directMethods.firstOrNull { it.name == "<init>" }?.parameterTypes ?: continue
if (constructorParameters.size < 6 || constructorParameters[5] != "J") continue
defaultMediaItem.set(clazz.getClassName())
val durationInMillisDexField = clazz.methods.firstOrNull { it.name == "toString" }?.implementation?.takeIf {
it.findConstString("metadata", contains = true)
}?.searchNextFieldReference("durationInMillis", contains = true) ?: continue
defaultMediaItemClass.set(clazz.getClassName())
defaultMediaItemDurationMsField.set(durationInMillisDexField.name)
return@mapper
}
}