Cookies_Github_mirror/AquaDX
4
0
Fork 1
mirror of https://github.com/MewoLab/AquaDX.git synced 2026-02-04 18:07:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

[F] Fix udp sendclass and recv

Browse Source
This commit is contained in:
Azalea
2025-01-12 22:27:01 -05:00
parent 0f87ed82e3
commit 18d95a1ccd
1 changed files with 16 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -18,18 +18,15 @@ const val PROTO_VERSION = 1
private object Command { private object Command {
// Control plane // Control plane
const val CTL_START = 1 const val CTL_START = 1
// const val CTL_BIND = 2 // Actually maybe we don't need BIND const val CTL_BIND = 2
const val CTL_HEARTBEAT = 3 const val CTL_HEARTBEAT = 3
const val CTL_ACCEPT = 4 // Accept a new multiplexed TCP stream const val CTL_TCP_ACCEPT = 4 // Accept a new multiplexed TCP stream
const val CTL_TCP_CONNECT = 5
const val CTL_TCP_CLOSE = 6
// Data plane // Data plane
const val DATA_SEND = 21 const val DATA_SEND = 21
const val DATA_BROADCAST = 22 const val DATA_BROADCAST = 22
const val DATA_POLL = 23
// Server-to-client message
const val INCOMING_DATA = 41
const val INCOMING_CONN = 42
} }
private object Proto { private object Proto {
@@ -43,9 +40,9 @@ data class Message(
val sid: Int? = null, // Stream ID, only applicable in TCP streams, 0 in UDP val sid: Int? = null, // Stream ID, only applicable in TCP streams, 0 in UDP
// Src and dst should be simulated ipv4 addresses and 0-65535 ports // Src and dst should be simulated ipv4 addresses and 0-65535 ports
val src: Int? = null, val src: UInt? = null,
val sPort: Int? = null, val sPort: Int? = null,
val dst: Int? = null, val dst: UInt? = null,
val dPort: Int? = null, val dPort: Int? = null,
val data: Any? = null val data: Any? = null
@@ -59,7 +56,7 @@ data class ActiveClient(
val writer: BufferedWriter val writer: BufferedWriter
) { ) {
val log = logger() val log = logger()
val stubIp by lazy { keychipToStubIp(clientKey) } val stubIp = keychipToStubIp(clientKey)
val mutex = ReentrantLock() val mutex = ReentrantLock()
var lastHeartbeat = millis() var lastHeartbeat = millis()
@@ -78,26 +75,24 @@ fun ActiveClient.handle(message: Message) {
Command.CTL_HEARTBEAT -> lastHeartbeat = millis() Command.CTL_HEARTBEAT -> lastHeartbeat = millis()
Command.DATA_SEND -> { Command.DATA_SEND -> {
// Find target by dst IP address // Find target by dst IP address
val tId = stubIpToKeychip(message.dst ?: return log.warn("No destination IP")) val target = message.dst?.let { clients[it] } ?: return log.warn("Target not found: ${message.dst}")
val target = clients[tId] ?: return log.warn("Target not found: $tId")
// Send to target TODO: SID // Send to target TODO: SID
target.send(message.copy(cmd = Command.INCOMING_DATA, sid = 0)) target.send(message.copy(sid = 0))
} }
Command.DATA_BROADCAST -> { Command.DATA_BROADCAST -> {
// Broadcast to all clients. This is only used in UDP so SID is always 0 // Broadcast to all clients. This is only used in UDP so SID is always 0
assert(message.proto == Proto.UDP) assert(message.proto == Proto.UDP)
clients.values.filter { it.clientKey != clientKey } clients.values.filter { it.clientKey != clientKey }
.forEach { it.send(message.copy(cmd = Command.INCOMING_DATA, sid = 0, src = stubIp)) } .forEach { it.send(message.copy(sid = 0, src = stubIp)) }
} }
} }
} }
fun keychipToStubIp(keychip: String) = "1${keychip.substring(1)}".toInt() fun keychipToStubIp(keychip: String) = "1${keychip.substring(2)}".toUInt()
fun stubIpToKeychip(ip: Int) = "A${ip.toString().substring(1)}"
// Keychip ID to Socket // Keychip ID to Socket
val clients = ConcurrentHashMap<String, ActiveClient>() val clients = ConcurrentHashMap<UInt, ActiveClient>()
/** /**
* Service for the party linker for AquaMai * Service for the party linker for AquaMai
@@ -130,8 +125,9 @@ class MaimaiFutari(private val port: Int = 20101) {
// Start: Register the client. Payload is the keychip // Start: Register the client. Payload is the keychip
Command.CTL_START -> { Command.CTL_START -> {
val id = message.data as String val id = message.data as String
clients[id] = ActiveClient(id, socket, reader, writer) val client = ActiveClient(id, socket, reader, writer)
handler = clients[id] clients[client.stubIp] = client
handler = clients[client.stubIp]
log.info("[+] Client registered: $id") log.info("[+] Client registered: $id")
// Send back the version // Send back the version
@@ -150,7 +146,7 @@ class MaimaiFutari(private val port: Int = 20101) {
} else log.error("Error in client handler", e) } else log.error("Error in client handler", e)
} finally { } finally {
// Remove client // Remove client
handler?.clientKey?.let { clients.remove(it) } handler?.stubIp?.let { clients.remove(it) }
socket.close() socket.close()
} }
} }