lzc256
/
AquaDX
mirror of https://github.com/hykilpikonna/AquaDX
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

[O] Futari serailization

pull/109/head
Azalea 2025-01-13 21:14:44 -05:00
parent 661af76ed6
commit 5290597b2b
1 changed files with 56 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

99
src/main/java/icu/samnyan/aqua/sega/maimai2/MaimaiFutari.kt
View File

@ -20,38 +20,54 @@ const val SO_TIMEOUT = 10000
private object Command { private object Command {
// Control plane // Control plane
const val CTL_START = 1 const val CTL_START = 1u
const val CTL_BIND = 2 const val CTL_BIND = 2u
const val CTL_HEARTBEAT = 3 const val CTL_HEARTBEAT = 3u
const val CTL_TCP_CONNECT = 4 // Accept a new multiplexed TCP stream const val CTL_TCP_CONNECT = 4u // Accept a new multiplexed TCP stream
const val CTL_TCP_ACCEPT = 5 const val CTL_TCP_ACCEPT = 5u
const val CTL_TCP_ACCEPT_ACK = 6 const val CTL_TCP_ACCEPT_ACK = 6u
const val CTL_TCP_CLOSE = 7 const val CTL_TCP_CLOSE = 7u
// Data plane // Data plane
const val DATA_SEND = 21 const val DATA_SEND = 21u
const val DATA_BROADCAST = 22 const val DATA_BROADCAST = 22u
} }
private object Proto { private object Proto {
const val TCP = 6 const val TCP = 6u
const val UDP = 17 const val UDP = 17u
} }
data class Message( data class Msg(
val cmd: Int, var cmd: UInt,
val proto: Int? = null, // Protocol, TCP or UDP var proto: UInt? = null,
val sid: Int? = null, // Stream ID, only applicable in TCP streams, 0 in UDP var sid: UInt? = null,
var src: UInt? = null,
var sPort: UInt? = null,
var dst: UInt? = null,
var dPort: UInt? = null,
var data: String? = null
) {
override fun toString() = ls(
1, cmd, proto, sid, src, sPort, dst, dPort,
null, null, null, null, null, null, null, null, // reserved for future use
data
).joinToString(",") { it?.str ?: "" }.trimEnd(',')
// Src and dst should be simulated ipv4 addresses and 0-65535 ports companion object {
val src: UInt? = null, val fields = arr(Msg::proto, Msg::sid, Msg::src, Msg::sPort, Msg::dst, Msg::dPort)
val sPort: Int? = null,
val dst: UInt? = null,
val dPort: Int? = null,
val data: Any? = null fun fromString(str: String): Msg {
) val sp = str.split(',')
fun ctlMsg(cmd: Int, data: Any? = null) = Message(cmd, data = data) return Msg(0u).apply {
cmd = sp[1].toUInt()
fields.forEachIndexed { i, f -> f.set(this, sp.getOrNull(i + 2)?.some?.toUIntOrNull()) }
data = sp.drop(16).joinToString(",")
}
}
}
}
fun ctlMsg(cmd: UInt, data: String? = null) = Msg(cmd, data = data)
data class ActiveClient( data class ActiveClient(
val clientKey: String, val clientKey: String,
@ -59,8 +75,8 @@ data class ActiveClient(
val reader: BufferedReader, val reader: BufferedReader,
val writer: BufferedWriter, val writer: BufferedWriter,
// <Stream ID, Destination client stub IP> // <Stream ID, Destination client stub IP>
val tcpStreams: MutableMap<Int, UInt> = mutableMapOf(), val tcpStreams: MutableMap<UInt, UInt> = mutableMapOf(),
val pendingStreams: MutableSet<Int> = mutableSetOf(), val pendingStreams: MutableSet<UInt> = mutableSetOf(),
) { ) {
val log = logger() val log = logger()
val stubIp = keychipToStubIp(clientKey) val stubIp = keychipToStubIp(clientKey)
@ -68,16 +84,16 @@ data class ActiveClient(
var lastHeartbeat = millis() var lastHeartbeat = millis()
fun send(msg: Message) { fun send(msg: Msg) {
writeMutex.withLock { writeMutex.withLock {
writer.write(msg.toJson()) writer.write(msg.toString())
writer.newLine() writer.newLine()
writer.flush() writer.flush()
} }
} }
} }
fun ActiveClient.handle(msg: Message) { fun ActiveClient.handle(msg: Msg) {
// Find target by dst IP address or TCP stream ID // Find target by dst IP address or TCP stream ID
val target = (msg.dst ?: msg.sid?.let { tcpStreams[it] } )?.let { clients[it] } val target = (msg.dst ?: msg.sid?.let { tcpStreams[it] } )?.let { clients[it] }
@ -95,19 +111,16 @@ fun ActiveClient.handle(msg: Message) {
return log.warn("Stream ID not found: ${msg.sid}") return log.warn("Stream ID not found: ${msg.sid}")
target.send(msg.copy(src = stubIp, dst = target.stubIp)) target.send(msg.copy(src = stubIp, dst = target.stubIp))
// 1020844165
// 2245580860
} }
Command.CTL_TCP_CONNECT -> { Command.CTL_TCP_CONNECT -> {
target ?: return log.warn("Connect: Target not found: ${msg.dst}") target ?: return log.warn("Connect: Target not found: ${msg.dst}")
msg.sid ?: return log.warn("Connect: Stream ID not found") val sid = msg.sid ?: return log.warn("Connect: Stream ID not found")
if (msg.sid in tcpStreams || msg.sid in pendingStreams) if (sid in tcpStreams || sid in pendingStreams)
return log.warn("Stream ID already in use: ${msg.sid}") return log.warn("Stream ID already in use: $sid")
// Add the stream to the pending list // Add the stream to the pending list
pendingStreams.add(msg.sid) pendingStreams.add(sid)
if (pendingStreams.size > MAX_STREAMS) { if (pendingStreams.size > MAX_STREAMS) {
log.warn("Too many pending streams, closing connection") log.warn("Too many pending streams, closing connection")
return socket.close() return socket.close()
@ -117,15 +130,15 @@ fun ActiveClient.handle(msg: Message) {
} }
Command.CTL_TCP_ACCEPT -> { Command.CTL_TCP_ACCEPT -> {
target ?: return log.warn("Accept: Target not found: ${msg.dst}") target ?: return log.warn("Accept: Target not found: ${msg.dst}")
msg.sid ?: return log.warn("Accept: Stream ID not found") val sid = msg.sid ?: return log.warn("Accept: Stream ID not found")
if (msg.sid !in target.pendingStreams) if (sid !in target.pendingStreams)
return log.warn("Stream ID not found in pending list: ${msg.sid}") return log.warn("Stream ID not found in pending list: ${sid}")
// Add the stream to the active list // Add the stream to the active list
target.pendingStreams.remove(msg.sid) target.pendingStreams.remove(sid)
target.tcpStreams[msg.sid] = stubIp target.tcpStreams[sid] = stubIp
tcpStreams[msg.sid] = target.stubIp tcpStreams[sid] = target.stubIp
target.send(msg.copy(src = stubIp, dst = target.stubIp)) target.send(msg.copy(src = stubIp, dst = target.stubIp))
} }
@ -165,7 +178,7 @@ class MaimaiFutari(private val port: Int = 20101) {
while (true) { while (true) {
val input = (reader.readLine() ?: continue).trim('\uFEFF') val input = (reader.readLine() ?: continue).trim('\uFEFF')
log.debug("Received: $input") log.debug("Received: $input")
val message = input.json<Message>() val message = Msg.fromString(input)
when (message.cmd) { when (message.cmd) {
// Start: Register the client. Payload is the keychip // Start: Register the client. Payload is the keychip
@ -178,7 +191,7 @@ class MaimaiFutari(private val port: Int = 20101) {
log.info("[+] Client registered: ${socket.remoteSocketAddress} -> $id") log.info("[+] Client registered: ${socket.remoteSocketAddress} -> $id")
// Send back the version // Send back the version
handler?.send(ctlMsg(Command.CTL_START, mapOf("version" to PROTO_VERSION))) handler?.send(ctlMsg(Command.CTL_START, "version=$PROTO_VERSION"))
} }
// Handle any other command using the handler // Handle any other command using the handler