fix

RTAkland · Sep 12, 2023 · f12a539 · f12a539
1 parent 09f979d
commit f12a539
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/kotlin/cn/rtast/kwsify/Kwsify.kt b/src/main/kotlin/cn/rtast/kwsify/Kwsify.kt
@@ -32,7 +32,13 @@ class Kwsify(address: InetSocketAddress) : WebSocketServer(address) {
     }
 
     override fun onOpen(conn: WebSocket, handshake: ClientHandshake) {
-        println("New connection connected. ${conn.remoteSocketAddress}")
+        val validEndpoints = listOf("subscribe", "publish")
+        val endpoint = conn.resourceDescriptor.replace("/", "")
+        if (!validEndpoints.contains(endpoint)) {
+            conn.close(1000, "This endpoint is invalid.")
+        } else {
+            println("New connection connected. ${conn.remoteSocketAddress}")
+        }
     }
 
     override fun onClose(conn: WebSocket, code: Int, reason: String, remote: Boolean) {