Merge pull request #754 from ccsplit/dev

Fix #749 - Add check for 'Agent' in self.module.options
websockets-multiuser
Chris Ross 2017-10-15 15:15:48 -04:00 committed by GitHub
commit 9ab30eb6ea
1 changed files with 6 additions and 1 deletions

View File

@ -3394,7 +3394,12 @@ class ModuleMenu(cmd.Cmd):
if module not in self.mainMenu.modules.modules: if module not in self.mainMenu.modules.modules:
print helpers.color("[!] Error: invalid module") print helpers.color("[!] Error: invalid module")
else: else:
module_menu = ModuleMenu(self.mainMenu, line, agent=self.module.options['Agent']['Value']) _agent = ''
if 'Agent' in self.module.options:
_agent = self.module.options['Agent']['Value']
line = line.strip("*")
module_menu = ModuleMenu(self.mainMenu, line, agent=_agent)
module_menu.cmdloop() module_menu.cmdloop()