AntiSpam: Prevent heavy unicode & utf16/32 from lagging the game (lambda-client#425)

* lag prevent

* Change to safeListener

Co-authored-by: Constructor <[email protected]>

Author: 2 people

src/main/kotlin/com/lambda/client/module/modules/chat/AntiSpam.kt

@@ -1,12 +1,12 @@
 package com.lambda.client.module.modules.chat
 
 import com.lambda.client.LambdaMod
-import com.lambda.client.event.listener.listener
 import com.lambda.client.module.Category
 import com.lambda.client.module.Module
 import com.lambda.client.util.text.MessageDetection
 import com.lambda.client.util.text.MessageSendHelper
 import com.lambda.client.util.text.SpamFilters
+import com.lambda.client.util.threads.safeListener
 import net.minecraft.util.text.TextComponentString
 import net.minecraftforge.client.event.ClientChatReceivedEvent
 import java.util.concurrent.ConcurrentHashMap
@@ -32,6 +32,8 @@ object AntiSpam : Module(
     private val specialCharBegin = setting("Special Begin", true, { page == Page.TYPE })
     private val greenText = setting("Green Text", false, { page == Page.TYPE })
     private val fancyChat by setting("Fancy Chat", false, { page == Page.TYPE })
+    private val lagMessage by setting("Lag Message", true, { page == Page.TYPE })
+    private val thresholdLagMessage by setting("Threshold Lag", 256, 256..1024, 1, { page == Page.TYPE && lagMessage }) // Is 1024 the max?
 
     /* Page Two */
     private val aggressiveFiltering by setting("Aggressive Filtering", true, { page == Page.SETTINGS })
@@ -82,8 +84,10 @@ object AntiSpam : Module(
             messageHistory.clear()
         }
 
-        listener<ClientChatReceivedEvent> { event ->
-            if (mc.player == null) return@listener
+        safeListener<ClientChatReceivedEvent> { event ->
+            if (lagMessage && isLagMessage(event.message.unformattedText)) {
+                event.isCanceled = true
+            }
 
             messageHistory.values.removeIf { System.currentTimeMillis() - it > 600000 }
 
@@ -131,6 +135,16 @@ object AntiSpam : Module(
         }
     }
 
+    private fun isLagMessage(message: String): Boolean {
+        return if (!filterOwn && isOwn(message)
+            || !filterDMs && MessageDetection.Direct.ANY detect message
+            || !filterServer && MessageDetection.Server.ANY detect message) {
+            false
+        } else {
+            message.getBytes() > thresholdLagMessage
+        }
+    }
+
     private fun detectSpam(message: String): String? {
         for ((key, value) in settingMap) {
             findPatterns(value, message)?.let {
@@ -169,6 +183,9 @@ object AntiSpam : Module(
         return isDuplicate
     }
 
+    private fun String.getBytes(): Int {
+        return this.toByteArray().size
+    }
 
     private fun isOwn(message: String): Boolean {
         /* mc.player is null when the module is being registered, so this matcher isn't added alongside the other FilterPatterns */