Merge pull request #754 from ccsplit/dev

Fix #749 - Add check for 'Agent' in self.module.options
2017-10-15 15:15:48 -04:00 · 2017-10-15 15:15:48 -04:00 · 9ab30eb6ea
parent 9cc02b211b 1474fa9975
commit 9ab30eb6ea
1 changed files with 6 additions and 1 deletions
--- a/lib/common/empire.py
+++ b/lib/common/empire.py
@ -3394,7 +3394,12 @@ class ModuleMenu(cmd.Cmd):
        if module not in self.mainMenu.modules.modules:
            print helpers.color("[!] Error: invalid module")
        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()