infosecn1nja
/
Empire
mirror of https://github.com/infosecn1nja/Empire.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Include Docker and non-docker paths in template search, fix #904

php_fix
Dakota Nelson 2018-01-05 14:12:51 -07:00
parent 5dbf668500
commit d7c9d43d22
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/listeners/dbx.py
View File

@ -431,7 +431,9 @@ class Listener:
elif language.lower() == 'python':
template_path = os.path.join(self.mainMenu.installPath, '/data/agent/stagers')
template_path = [
os.path.join(self.mainMenu.installPath, '/data/agent/stagers'),
os.path.join(self.mainMenu.installPath, './data/agent/stagers')]
eng = templating.TemplateEngine(template_path)
template = eng.get_template('dropbox.py')

8
lib/listeners/http.py
View File

@ -591,7 +591,9 @@ class Listener:
return randomizedStager
elif language.lower() == 'python':
template_path = os.path.join(self.mainMenu.installPath, 'data/agent/stagers')
template_path = [
os.path.join(self.mainMenu.installPath, '/data/agent/stagers'),
os.path.join(self.mainMenu.installPath, './data/agent/stagers')]
eng = templating.TemplateEngine(template_path)
template = eng.get_template('http.py')
@ -898,7 +900,7 @@ def send_message(packets=None):
return launcher
else:
return make_response(self.default_response(), 404)
@app.before_request
def check_ip():
"""
@ -930,7 +932,7 @@ def send_message(packets=None):
"""
Return default server web page if user navigates to index.
"""
static_dir = self.mainMenu.installPath + "data/misc/"
return make_response(self.index_page(), 200)