majjo/XSnap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(core/security_features): better explanation of login response

Browse Source
This commit is contained in:
rhunk
2024-08-16 12:57:45 +02:00
parent ef1566add2
commit 47efdf165d
5 changed files with 70 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
core/src/main/kotlin/me/rhunk/snapenhance/core/features/impl/SecurityFeatures.kt
View File

@@ -5,11 +5,7 @@ import androidx.compose.foundation.background
import androidx.compose.foundation.layout.padding
import androidx.compose.foundation.shape.RoundedCornerShape
import androidx.compose.material3.Text
import androidx.compose.runtime.LaunchedEffect
import androidx.compose.runtime.getValue
import androidx.compose.runtime.mutableStateOf
import androidx.compose.runtime.remember
import androidx.compose.runtime.setValue
import androidx.compose.runtime.*
import androidx.compose.ui.Alignment
import androidx.compose.ui.Modifier
import androidx.compose.ui.graphics.Color
@@ -18,22 +14,66 @@ import androidx.compose.ui.unit.sp
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.delay
import kotlinx.coroutines.withContext
import me.rhunk.snapenhance.common.util.protobuf.ProtoEditor
import me.rhunk.snapenhance.common.util.protobuf.ProtoReader
import me.rhunk.snapenhance.core.event.events.impl.UnaryCallEvent
import me.rhunk.snapenhance.core.features.Feature
import me.rhunk.snapenhance.core.util.ktx.setObjectField
import java.io.FileDescriptor
class SecurityFeatures : Feature("Security Features") {
private fun transact(option: Int) = Os.prctl(option, 0, 0, 0, 0)
private fun transact(option: Int, option2: Long) = kotlin.runCatching { Os.prctl(option, option2, 0, 0, 0) }.getOrNull()
private val token by lazy {
runCatching { transact(0) }.getOrNull()
transact(0, 0)
}
private fun getStatus() = token?.run {
transact(this).toString(2).padStart(32, '0').count { it == '1' }
transact(this, 0)?.toString(2)?.padStart(32, '0')?.count { it == '1' }
}
override fun init() {
token // pre init token
context.event.subscribe(UnaryCallEvent::class) { event ->
if (!event.uri.contains("/Login")) return@subscribe
// intercept login response
event.addResponseCallback {
val response = ProtoReader(buffer)
val isBlocked = when {
event.uri.contains("TLv") -> response.getVarInt(1) == 14L
else -> response.getVarInt(1) == 16L
}
val errorDataIndex = when {
response.contains(11) -> 11
response.contains(10) -> 10
response.contains(8) -> 8
else -> return@addResponseCallback
}
if (isBlocked) {
val status = transact(token ?: return@addResponseCallback, 1)?.let {
val buffer = ByteArray(8192)
val fd = FileDescriptor().apply {
setObjectField("descriptor", it)
}
val read = Os.read(fd, buffer, 0, buffer.size)
Os.close(fd)
buffer.copyOfRange(0, read).decodeToString()
}!!
buffer = ProtoEditor(buffer).apply {
edit(errorDataIndex) {
remove(1)
addString(1, status)
}
}.toByteArray()
}
}
}
context.inAppOverlay.addCustomComposable {
var statusText by remember {
mutableStateOf("")
@@ -49,10 +89,10 @@ class SecurityFeatures : Feature("Security Features") {
withContext(Dispatchers.Main) {
if (status == null || status == 0) {
textColor = Color.Red
statusText = "sif not loaded. Can't get status"
statusText = "SIF not loaded!"
} else {
textColor = Color.Green
statusText = "sif = $status"
statusText = "SIF = $status"
}
}
delay(1000)