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':
|
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)
|
eng = templating.TemplateEngine(template_path)
|
||||||
template = eng.get_template('dropbox.py')
|
template = eng.get_template('dropbox.py')
|
||||||
|
|
||||||
|
|
|
@ -591,7 +591,9 @@ class Listener:
|
||||||
return randomizedStager
|
return randomizedStager
|
||||||
|
|
||||||
elif language.lower() == 'python':
|
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)
|
eng = templating.TemplateEngine(template_path)
|
||||||
template = eng.get_template('http.py')
|
template = eng.get_template('http.py')
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue