github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-15 06:49:17 +00:00
Code Issues Projects Releases Wiki Activity

feat: 系统工具改为异步，增加tcpdump工具

Browse Source
This commit is contained in:
jiangweidong
2023-08-07 09:37:24 +08:00
committed by Bryan
parent 8accd296b8
commit 7636255533
13 changed files with 358 additions and 165 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
apps/settings/ws.py
View File

@@ -1,53 +1,73 @@
# -*- coding: utf-8 -*-
#
import json
from channels.generic.websocket import JsonWebsocketConsumer
from channels.generic.websocket import AsyncJsonWebsocketConsumer
from common.db.utils import close_old_connections
from common.utils import get_logger
from .utils import verbose_ping, verbose_telnet, verbose_nmap
from .tools import verbose_ping, verbose_telnet, verbose_nmap, verbose_tcpdump
logger = get_logger(__name__)
class ToolsWebsocket(JsonWebsocketConsumer):
class ToolsWebsocket(AsyncJsonWebsocketConsumer):
def connect(self):
async def connect(self):
user = self.scope["user"]
if user.is_authenticated:
self.accept()
await self.accept()
else:
self.close()
await self.close()
def send_msg(self, msg):
self.send_json({'msg': msg + '\r\n'})
async def send_msg(self, msg=''):
await self.send_json({'msg': msg + '\r\n'})
def imitate_ping(self, dest_ip, timeout=3, count=5, psize=64):
"""
Send `count' ping with `psize' size to `dest_ip' with
the given `timeout' and display the result.
"""
logger.info('receive request ping {}'.format(dest_ip))
verbose_ping(dest_ip, timeout, count, psize, display=self.send_msg)
async def imitate_ping(self, dest_ip, timeout=3, count=5, psize=64):
params = {
'dest_ip': dest_ip, 'timeout': timeout,
'count': count, 'psize': psize
}
logger.info(f'Receive request ping: {params}')
await verbose_ping(display=self.send_msg, **params)
def imitate_telnet(self, dest_ip, dest_port=23, timeout=10):
logger.info('receive request telnet {}'.format(dest_ip))
verbose_telnet(dest_ip, dest_port, timeout, display=self.send_msg)
async def imitate_telnet(self, dest_ip, dest_port=23, timeout=10):
params = {
'dest_ip': dest_ip, 'dest_port': dest_port, 'timeout': timeout,
}
logger.info(f'Receive request telnet: {params}')
await verbose_telnet(display=self.send_msg, **params)
def imitate_nmap(self, dest_ip, dest_port=None, timeout=None):
logger.info('receive request nmap {}'.format(dest_ip))
verbose_nmap(dest_ip, dest_port, timeout, display=self.send_msg)
async def imitate_nmap(self, dest_ips, dest_ports=None, timeout=None):
params = {
'dest_ips': dest_ips, 'dest_ports': dest_ports, 'timeout': timeout,
}
logger.info(f'Receive request nmap: {params}')
await verbose_nmap(display=self.send_msg, **params)
def receive(self, text_data=None, bytes_data=None, **kwargs):
async def imitate_tcpdump(
self, interfaces=None, src_ips='',
src_ports='', dest_ips='', dest_ports=''
):
params = {
'interfaces': interfaces, 'src_ips': src_ips, 'src_ports': src_ports,
'dest_ips': dest_ips, 'dest_ports': dest_ports
}
logger.info(f'Receive request tcpdump: {params}')
await verbose_tcpdump(display=self.send_msg, **params)
async def receive(self, text_data=None, bytes_data=None, **kwargs):
data = json.loads(text_data)
tool_type = data.pop('tool_type', 'Ping')
try:
tool_func = getattr(self, f'imitate_{tool_type.lower()}')
await tool_func(**data)
except Exception as error:
await self.send_msg('Exception: %s' % error)
await self.send_msg()
await self.close()
tool_func = getattr(self, f'imitate_{tool_type.lower()}')
tool_func(**data)
self.close()
def disconnect(self, code):
self.close()
async def disconnect(self, code):
await self.close()
close_old_connections()