From 527eb75d67bc406bc38e3cff7242e66277ede87d Mon Sep 17 00:00:00 2001
From: acidvegas <acid.vegas@acid.vegas>
Date: Tue, 28 May 2024 20:40:00 -0400
Subject: [PATCH] Added memory constraints to things to keep everything memory
 safe from anomalies & attacks

---
 src/main.ino | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/main.ino b/src/main.ino
index 80783e1..2271bdb 100644
--- a/src/main.ino
+++ b/src/main.ino
@@ -274,7 +274,8 @@ void scanWiFiNetworks() {
         net.channel = WiFi.channel(i);
         net.rssi = WiFi.RSSI(i);
         net.encryption = (WiFi.encryptionType(i) == WIFI_AUTH_OPEN) ? "Open" : "Secured";
-        net.ssid = WiFi.SSID(i);
+        String ssid = WiFi.SSID(i).substring(0, 32); // WiFi SSIDs are limited to 32 characters
+        net.ssid = ssid;
         wifiNetworks.push_back(net);
     }
 
@@ -292,7 +293,7 @@ void handlePasswordInput(char key) {
             inputBuffer.remove(inputBuffer.length() - 1);
             displayPasswordInputLine();
         }
-    } else {
+    } else if (inputBuffer.length() < 63) { // WiFi passwords are limited to 63 characters
         inputBuffer += key;
         displayPasswordInputLine();
     }
@@ -415,6 +416,11 @@ bool connectToIRC() {
 
 
 void sendIRC(String command) {
+    if (command.length() > 510) {
+        Serial.println("Failed to send: Command too long");
+        return;
+    }
+
     if (client.connected()) {
         if (client.println(command))
             Serial.println("IRC: >>> " + command);
@@ -429,6 +435,12 @@ void sendIRC(String command) {
 void handleIRC() {
     while (client.available()) {
         String line = client.readStringUntil('\n');
+
+        // This is an anomaly, but it can happen and I wanted debug output for if it does
+        if (line.length() > 512)
+            Serial.println("WARNING: IRC line length exceeds 512 characters!");
+            line = line.substring(0, 512); // Truncate the line to 512 characters anyways
+
         Serial.println("IRC: " + line);
 
         int firstSpace = line.indexOf(' ');