Add length limits to messages and names

ReconnectedCC · Oct 9, 2024 · 21986bb · 21986bb
1 parent 74d0d65
commit 21986bb
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/main/java/cc/reconnected/chatbox/ws/WsServer.java b/src/main/java/cc/reconnected/chatbox/ws/WsServer.java
@@ -29,6 +29,8 @@ public class WsServer extends WebSocketServer {
     public static final Pattern PATH_LICENSE = Pattern.compile("^/([0-9a-z-]+)$", Pattern.CASE_INSENSITIVE);
     private final HashMap<WebSocket, ChatboxClient> clients = new HashMap<>();
     private final InetAddress guestAddress;
+    public static int messageMaxLength = 1024;
+    public static int nameMaxLength = 64;
 
     public HashMap<WebSocket, ChatboxClient> clients() {
         return clients;
@@ -146,6 +148,12 @@ public void onMessage(WebSocket conn, String message) {
                     return;
                 }
 
+                sayPacket.text = sayPacket.text.substring(0, Math.min(sayPacket.text.length(), messageMaxLength));
+                if(sayPacket.name != null) {
+                    sayPacket.name = sayPacket.name.trim().replace("\n","");
+                    sayPacket.name = sayPacket.name.substring(0, Math.min(sayPacket.name.length(), nameMaxLength));
+                }
+
                 ChatboxSay.EVENT.invoker().say(client, sayPacket);
 
                 break;
@@ -179,6 +187,12 @@ public void onMessage(WebSocket conn, String message) {
                     return;
                 }
 
+                tellPacket.text = tellPacket.text.substring(0, Math.min(tellPacket.text.length(), messageMaxLength));
+                if(tellPacket.name != null) {
+                    tellPacket.name = tellPacket.name.trim().replace("\n","");
+                    tellPacket.name = tellPacket.name.substring(0, Math.min(tellPacket.name.length(), nameMaxLength));
+                }
+
                 ChatboxTell.EVENT.invoker().tell(client, tellPacket);
 
                 break;