Sancaros/phantasmal-world
1
0
Fork 0
mirror of https://github.com/DaanVandenBosch/phantasmal-world.git synced 2025-04-04 06:28:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Simplified proxy server code.

Browse Source
This commit is contained in:
Daan Vanden Bosch 2021-07-31 22:16:01 +02:00
parent 6daddcfd65
commit 089832c2fe
4 changed files with 39 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
psoserv/src/main/kotlin/world/phantasmal/psoserv/messages/BbMessages.kt
View File

@ -62,13 +62,15 @@ class GuildCard(
val entries: List<GuildCardEntry>
)
sealed class BbMessage(override val buffer: Buffer) : Message(BB_HEADER_SIZE) {
sealed class BbMessage(override val buffer: Buffer) : AbstractMessage(BB_HEADER_SIZE) {
override val code: Int get() = buffer.getUShort(BB_MSG_CODE_POS).toInt()
override val size: Int get() = buffer.getUShort(BB_MSG_SIZE_POS).toInt()
class InitEncryption(buffer: Buffer) : BbMessage(buffer) {
val serverKey: ByteArray get() = byteArray(INIT_MSG_SIZE, size = KEY_SIZE)
val clientKey: ByteArray get() = byteArray(INIT_MSG_SIZE + KEY_SIZE, size = KEY_SIZE)
class InitEncryption(buffer: Buffer) : BbMessage(buffer), InitEncryptionMessage {
override val serverKey: ByteArray
get() = byteArray(INIT_MSG_SIZE, size = KEY_SIZE)
override val clientKey: ByteArray
get() = byteArray(INIT_MSG_SIZE + KEY_SIZE, size = KEY_SIZE)
constructor(message: String, serverKey: ByteArray, clientKey: ByteArray) : this(
buf(0x0003, INIT_MSG_SIZE + 2 * KEY_SIZE) {
@ -87,14 +89,14 @@ sealed class BbMessage(override val buffer: Buffer) : Message(BB_HEADER_SIZE) {
constructor() : this(buf(0x0005))
}
class Redirect(buffer: Buffer) : BbMessage(buffer) {
var ipAddress: ByteArray
class Redirect(buffer: Buffer) : BbMessage(buffer), RedirectMessage {
override var ipAddress: ByteArray
get() = byteArray(0, size = 4)
set(value) {
require(value.size == 4)
setByteArray(0, value)
}
var port: Int
override var port: Int
get() = uShort(4).toInt()
set(value) {
require(value in 0..65535)

22
psoserv/src/main/kotlin/world/phantasmal/psoserv/messages/Messages.kt
View File

@ -24,12 +24,25 @@ fun messageString(
data class Header(val code: Int, val size: Int)
abstract class Message(val headerSize: Int) {
abstract val buffer: Buffer
abstract val code: Int
abstract val size: Int
interface Message {
val buffer: Buffer
val code: Int
val size: Int
val headerSize: Int
val bodySize: Int get() = size - headerSize
}
interface InitEncryptionMessage : Message {
val serverKey: ByteArray
val clientKey: ByteArray
}
interface RedirectMessage : Message {
var ipAddress: ByteArray
var port: Int
}
abstract class AbstractMessage(override val headerSize: Int) : Message {
override fun toString(): String = messageString()
protected fun uByte(offset: Int) = buffer.getUByte(headerSize + offset)
@ -58,5 +71,4 @@ abstract class Message(val headerSize: Int) {
protected fun messageString(vararg props: Pair<String, Any>): String =
messageString(code, size, this::class.simpleName, *props)
}

16
psoserv/src/main/kotlin/world/phantasmal/psoserv/messages/PcMessages.kt
View File

@ -13,13 +13,15 @@ const val PC_HEADER_SIZE: Int = 4
const val PC_MSG_SIZE_POS: Int = 0
const val PC_MSG_CODE_POS: Int = 2
sealed class PcMessage(override val buffer: Buffer) : Message(PC_HEADER_SIZE) {
sealed class PcMessage(override val buffer: Buffer) : AbstractMessage(PC_HEADER_SIZE) {
override val code: Int get() = buffer.getUByte(PC_MSG_CODE_POS).toInt()
override val size: Int get() = buffer.getUShort(PC_MSG_SIZE_POS).toInt()
class InitEncryption(buffer: Buffer) : PcMessage(buffer) {
val serverKey: ByteArray get() = byteArray(INIT_MSG_SIZE, size = KEY_SIZE)
val clientKey: ByteArray get() = byteArray(INIT_MSG_SIZE + KEY_SIZE, size = KEY_SIZE)
class InitEncryption(buffer: Buffer) : PcMessage(buffer), InitEncryptionMessage {
override val serverKey: ByteArray
get() = byteArray(INIT_MSG_SIZE, size = KEY_SIZE)
override val clientKey: ByteArray
get() = byteArray(INIT_MSG_SIZE + KEY_SIZE, size = KEY_SIZE)
constructor(message: String, serverKey: ByteArray, clientKey: ByteArray) : this(
buf(0x02, INIT_MSG_SIZE + 2 * KEY_SIZE) {
@ -60,14 +62,14 @@ sealed class PcMessage(override val buffer: Buffer) : Message(PC_HEADER_SIZE) {
)
}
class Redirect(buffer: Buffer) : PcMessage(buffer) {
var ipAddress: ByteArray
class Redirect(buffer: Buffer) : PcMessage(buffer), RedirectMessage {
override var ipAddress: ByteArray
get() = byteArray(0, size = 4)
set(value) {
require(value.size == 4)
setByteArray(0, value)
}
var port: Int
override var port: Int
get() {
buffer.endianness = Endianness.Big
val p = uShort(4).toInt()

50
psoserv/src/main/kotlin/world/phantasmal/psoserv/servers/ProxyServer.kt
View File

@ -4,10 +4,7 @@ import mu.KotlinLogging
import world.phantasmal.core.disposable.TrackedDisposable
import world.phantasmal.psolib.buffer.Buffer
import world.phantasmal.psoserv.encryption.Cipher
import world.phantasmal.psoserv.messages.BbMessage
import world.phantasmal.psoserv.messages.Header
import world.phantasmal.psoserv.messages.Message
import world.phantasmal.psoserv.messages.PcMessage
import world.phantasmal.psoserv.messages.*
import java.net.*
class ProxyServer(
@ -119,7 +116,7 @@ class ProxyServer(
override fun processMessage(message: Message): ProcessResult {
when (message) {
is PcMessage.InitEncryption -> if (decryptCipher == null) {
is InitEncryptionMessage -> if (decryptCipher == null) {
decryptCipher = createCipher(message.serverKey)
encryptCipher = createCipher(message.serverKey)
@ -135,7 +132,7 @@ class ProxyServer(
clientSocket,
this,
clientDecryptCipher,
clientEncryptCipher
clientEncryptCipher,
)
this.clientHandler = clientListener
val thread = Thread(clientListener::listen)
@ -143,46 +140,7 @@ class ProxyServer(
thread.start()
}
is BbMessage.InitEncryption -> if (decryptCipher == null) {
decryptCipher = createCipher(message.serverKey)
encryptCipher = createCipher(message.serverKey)
val clientDecryptCipher = createCipher(message.clientKey)
val clientEncryptCipher = createCipher(message.clientKey)
logger.info {
"Encryption initialized, start listening to client."
}
// Start listening to client on another thread.
val clientListener = ClientHandler(
clientSocket,
this,
clientDecryptCipher,
clientEncryptCipher
)
this.clientHandler = clientListener
val thread = Thread(clientListener::listen)
thread.name = "${ProxyServer::class.simpleName} client"
thread.start()
}
is PcMessage.Redirect -> {
val oldAddress = InetAddress.getByAddress(message.ipAddress)
redirectMap[Pair(oldAddress, message.port)]?.let { (newAddress, newPort) ->
logger.debug {
"Rewriting redirect from $oldAddress:${message.port} to $newAddress:$newPort."
}
message.ipAddress = newAddress.address
message.port = newPort
return ProcessResult.Changed
}
}
is BbMessage.Redirect -> {
is RedirectMessage -> {
val oldAddress = InetAddress.getByAddress(message.ipAddress)
redirectMap[Pair(oldAddress, message.port)]?.let { (newAddress, newPort) ->