63 lines
3.3 KiB
Kotlin
63 lines
3.3 KiB
Kotlin
/* app_process + APK: JNI и настоящий worker без Activity, профиля приложения и внешних соединений.
|
|
* Аргумент — новый каталог внутри /data/local/tmp; libutun.so/зависимости должны быть в java.library.path.
|
|
* JNI restart вызывается напрямую: frontend-владельцы камеры/аудио в этом процессе не создаются. */
|
|
import com.utun.chat.data.ChatEventCallback
|
|
import com.utun.chat.data.NativeLib
|
|
import java.io.File
|
|
import java.util.concurrent.LinkedBlockingQueue
|
|
import java.util.concurrent.TimeUnit
|
|
import kotlin.system.exitProcess
|
|
|
|
private fun nativeCommand(name: String, config: String? = null): Any? {
|
|
val method = if (config == null) NativeLib::class.java.getDeclaredMethod(name)
|
|
else NativeLib::class.java.getDeclaredMethod(name, String::class.java)
|
|
method.isAccessible = true
|
|
return if (config == null) method.invoke(NativeLib) else method.invoke(NativeLib, config)
|
|
}
|
|
|
|
fun main(args: Array<String>) {
|
|
check(args.size == 1)
|
|
val directory = File(args.single()).canonicalFile
|
|
check(directory.path.startsWith("/data/local/tmp/utun-identity-audit-") && !directory.exists())
|
|
check(directory.mkdirs())
|
|
val events = LinkedBlockingQueue<Int>()
|
|
NativeLib.setEventCallback(object : ChatEventCallback {
|
|
override fun onEvent(type: Int, data: ByteArray?) {
|
|
if (type == 12 || type == 13) events.offer(type)
|
|
}
|
|
})
|
|
fun awaitStarted(expectStop: Boolean) {
|
|
if (expectStop) check(events.poll(10, TimeUnit.SECONDS) == 13) { "old core did not stop" }
|
|
check(events.poll(10, TimeUnit.SECONDS) == 12 && NativeLib.isReady()) { "new core did not start" }
|
|
}
|
|
fun identity(): Pair<String, String> {
|
|
val lines = File(directory, "keys.conf").readLines()
|
|
check(lines.size == 2 && lines[0].startsWith("public_key=") && lines[1].startsWith("private_key="))
|
|
val keys = lines[0].removePrefix("public_key=") to lines[1].removePrefix("private_key=")
|
|
check(keys.first.length == 64 && keys.second.length == 64)
|
|
val public = keys.first.chunked(2).map { it.toInt(16).toByte() }.toByteArray()
|
|
check(NativeLib.deriveNodeId(public) == NativeLib.getMyNodeId()) { "file/core identity mismatch" }
|
|
return keys
|
|
}
|
|
val config = "[global]\ndb_path=${directory.path}\nauto_sockets=no\n[ntp]\nenabled=no\n" +
|
|
"[stun]\nenabled=no\n[chatserver]\nstorage_autoload=0\n"
|
|
try {
|
|
check(NativeLib.start(config))
|
|
awaitStarted(expectStop = false)
|
|
val original = identity()
|
|
val staleConfig = "[global]\nmy_public_key=${original.first}\nmy_private_key=${original.second}\n$config"
|
|
check(nativeCommand("nativeRegenerateKeys", staleConfig) == true)
|
|
awaitStarted(expectStop = true)
|
|
val generated = identity()
|
|
check(generated != original) { "key regeneration retained old pair" }
|
|
check(nativeCommand("nativeRestart", staleConfig) == true)
|
|
awaitStarted(expectStop = true)
|
|
check(identity() == generated) { "stale config reverted persisted identity" }
|
|
} finally {
|
|
nativeCommand("nativeStop")
|
|
}
|
|
check(!NativeLib.isReady())
|
|
println("PASS: Android JNI start, full identity regeneration, persisted keys, stale-config restart and stop")
|
|
exitProcess(0)
|
|
}
|