Include Docker and non-docker paths in template search, fix #904
parent
5dbf668500
commit
d7c9d43d22
|
@ -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')
|
||||
|
||||
|
|
|
@ -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')
|
||||
|
||||
|
|
Loading…
Reference in New Issue