Delete session functionality (#590)

mushorg · Dec 24, 2024 · ee28118 · ee28118
1 parent e9a8415
commit ee28118
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/conpot/core/__init__.py b/conpot/core/__init__.py
@@ -41,6 +41,9 @@ def get_databus():
 def get_session(*args, **kwargs):
     return sessionManager.get_session(*args, **kwargs)
 
+def delete_session(*args, **kwargs):
+    return sessionManager.delete_session(*args, **kwargs)
+
 
 # file-system related  --
 

diff --git a/conpot/core/session_manager.py b/conpot/core/session_manager.py
@@ -55,6 +55,12 @@ def get_session(
             self._sessions.append(attack_session)
         return attack_session
 
+    def delete_session(self, id):
+        for i, session in enumerate(self._sessions):
+            if session.id == id:
+                del self._sessions[i]
+                break
+
     def purge_sessions(self):
         # there is no native purge/clear mechanism for gevent queues, so...
         self.log_queue = Queue()