From 13c9be92fec399d56de01049537bb5f1721ebd7d Mon Sep 17 00:00:00 2001
From: swee <meow@swee.codes>
Date: Sat, 11 Jan 2025 00:44:12 -0800
Subject: [PATCH] Update modules/botnet_protect.py

---
 modules/botnet_protect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/botnet_protect.py b/modules/botnet_protect.py
index 77b521a..3416637 100644
--- a/modules/botnet_protect.py
+++ b/modules/botnet_protect.py
@@ -21,7 +21,7 @@ class IRCatModule:
         raise Exception("Botnet detected!")
     def onSocket(self, ip, socket, value, cachedNick=None, validated=False):
         if validated:
-            if self.memory[cachedNick + "|" + ip] != True:
+            if cachedNick + "|" + ip in self.memory and self.memory[cachedNick + "|" + ip] != True:
                 if "JOIN" in value:
                     target = value.split(" ")[1]
                     self.memory[cachedNick + "|" + ip][target] = 1 # 1: Just joined the channel, continue observing.