fix: memory leak

Author: Ikaleio

src/index.ts

@@ -6,7 +6,7 @@ import { MinecraftProxy } from './proxy'
 import { fromError } from 'zod-validation-error'
 import { PluginLoader } from './plugins'
 import { getGitCommitHash } from './macros/getGitCommitHash' with { type: 'macro' }
-import { generateHeapSnapshot } from 'bun'
+import { heapStats } from 'bun:jsc'
 
 export type TransitLogger = Logger<'packet', boolean>
 
@@ -31,6 +31,10 @@ async function main() {
 		}),
 	)
 
+	// setInterval(() => {
+	// 	logger.info(heapStats())
+	// }, 5000)
+
 	const minecraftProxy = new MinecraftProxy()
 
 	logger.info(`Transit Proxy by Ikaleio (build: ${await getGitCommitHash()})`)

src/proxy.ts

@@ -47,12 +47,14 @@ export type C2RSocketData = {
 	originIP: IP | null // 客户端 IP 地址（若开启 ProxyProtocol 入站，从其中解析）
 	proxyProtocol: boolean | null // 是否启用 Proxy Protocol v2 出站
 	FML: 0 | 1 | 2 | null // 是否为 Forge Mod Loader (2) 客户端，0 代表非 FML
+	timeouts: NodeJS.Timer[] // 存储所有的 timeout
 }
 
 // Relay to server socket data
 type R2SSocketData = {
 	client: Bun.Socket<C2RSocketData> | null // 客户端连接
 	sendBuffer: ArrayBufferSink // 发送缓冲区
+	timeouts: NodeJS.Timer[] // 存储所有的 timeout
 }
 
 const writeToBuffer = (
@@ -77,6 +79,10 @@ const sendBuffer = (
 	}
 }
 
+const clearAllTimeouts = (timeouts: NodeJS.Timer[]) => {
+	timeouts.forEach(timeout => clearTimeout(timeout))
+}
+
 export const InboundSchema = z
 	.object({
 		bind: z.string().default('0.0.0.0:25565'),
@@ -142,6 +148,7 @@ export class MinecraftProxy {
 						remoteSocket.data = {
 							client: clientSocket,
 							sendBuffer: new ArrayBufferSink(),
+							timeouts: [],
 						}
 						remoteSocket.data.sendBuffer.start({
 							asUint8Array: true,
@@ -162,6 +169,7 @@ export class MinecraftProxy {
 						writeToBuffer(clientSocket.data.remote, initPacket)
 					},
 					close: remoteSocket => {
+						remoteSocket.data.sendBuffer.end()
 						clientSocket.end()
 					},
 					end: remoteSocket => {
@@ -229,6 +237,7 @@ export class MinecraftProxy {
 						originIP: null,
 						proxyProtocol: null,
 						FML: null,
+						timeouts: [],
 					}
 
 					logger.debug(
@@ -246,16 +255,19 @@ export class MinecraftProxy {
 						clientSocket.data.originIP = IP.parse(clientSocket.remoteAddress)
 
 					// 若 3 秒内未成功读取握手包，则断开连接
-					setTimeout(() => {
+					const handshakeTimeout = setTimeout(() => {
 						if (clientSocket.data.state === null) {
 							logger.warn(
 								`${colorHash(clientSocket.data.connId)} Handshake timeout`,
 							)
 							clientSocket.end()
 						}
 					}, 3000)
+					clientSocket.data.timeouts.push(handshakeTimeout)
 				},
 				close: async clientSocket => {
+					clientSocket.data.sendBuffer.end()
+					clearAllTimeouts(clientSocket.data.timeouts)
 					if (clientSocket.data.username) {
 						this.onlinePlayers.delete(clientSocket.data.username)
 					}
@@ -417,14 +429,15 @@ export class MinecraftProxy {
 
 							if (nextState === State.Login) {
 								// 若 3 秒内未成功读取登录包，则断开连接
-								setTimeout(() => {
+								const loginTimeout = setTimeout(() => {
 									if (clientSocket.data.state !== State.Play) {
 										logger.warn(
 											`${colorHash(clientSocket.data.connId)} Login timeout`,
 										)
 										clientSocket.end()
 									}
 								}, 3000)
+								clientSocket.data.timeouts.push(loginTimeout)
 							}
 						}
 					} // 考虑一次发送两个数据包，应当直接在后面处理登录