strips/foundation
2
0
Fork 0
mirror of https://github.com/GayPizzaSpecifications/foundation.git synced 2025-08-03 13:31:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement usage of the new Tailscale channel library.

Browse Source
This commit is contained in:
Alex Zenla
2023-03-26 21:21:04 -07:00
parent 02a5d02dad
commit 2262ceb1d1
5 changed files with 13 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
foundation-tailscale/build.gradle.kts
View File

@ -13,6 +13,7 @@ dependencies {
implementation(project(":common-plugin"))
compileOnly(project(":foundation-shared"))
implementation(libs.tailscale)
implementation(libs.tailscale.channel)
}
concreteItem {

44
foundation-tailscale/src/main/kotlin/gay/pizza/foundation/tailscale/TailscaleProxyServer.kt
View File

@ -1,16 +1,13 @@
package gay.pizza.foundation.tailscale
import gay.pizza.tailscale.channel.ChannelCopier
import gay.pizza.tailscale.core.Tailscale
import gay.pizza.tailscale.core.TailscaleConn
import gay.pizza.tailscale.core.TailscaleListener
import org.bukkit.Server
import java.net.InetSocketAddress
import java.net.StandardSocketOptions
import java.nio.ByteBuffer
import java.nio.channels.ClosedChannelException
import java.nio.channels.ReadableByteChannel
import java.nio.channels.SocketChannel
import java.nio.channels.WritableByteChannel
class TailscaleProxyServer(val server: Server, val tailscale: Tailscale) {
private var minecraftServerListener: TailscaleListener? = null
@ -34,38 +31,13 @@ class TailscaleProxyServer(val server: Server, val tailscale: Tailscale) {
val readChannel = conn.openReadChannel()
val writeChannel = conn.openWriteChannel()
fun closeAll() {
socketChannel.close()
readChannel.close()
writeChannel.close()
}
fun startCopyThread(name: String, from: ReadableByteChannel, to: WritableByteChannel) {
val thread = Thread {
try {
while (true) {
val buffer = ByteBuffer.allocate(2048)
val size = from.read(buffer)
if (size < 0) {
break
} else {
buffer.flip()
to.write(buffer)
}
buffer.clear()
}
} catch (_: ClosedChannelException) {
} finally {
closeAll()
}
}
thread.name = name
thread.start()
}
startCopyThread("Tailscale to Socket Pipe", readChannel, socketChannel)
startCopyThread("Socket to Tailscale Pipe", socketChannel, writeChannel)
val closeHandler = { socketChannel.close() }
val tailscaleSocketCopier = ChannelCopier(readChannel, socketChannel)
tailscaleSocketCopier.spawnCopyThread(
"Tailscale to Socket Copier", onClose = closeHandler)
val socketTailscaleCopier = ChannelCopier(socketChannel, writeChannel)
socketTailscaleCopier.spawnCopyThread(
"Socket to Tailscale Copier", onClose = closeHandler)
}
fun close() {