Merge pull request #219 from gryffus/patch-2

Fix check if module uses msfvenom, fixes CLI
Veil-Framework · Sep 20, 2015 · cc89c4a · cc89c4a
2 parents ac7bd61 + 63cf378
commit cc89c4a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/common/controller.py b/modules/common/controller.py
@@ -327,10 +327,11 @@ def SetPayload(self, payloadname, options):
                     print helpers.color("\n [!] Internal error #4.", warning=True)
             # options['msfvenom'] = ["windows/meterpreter/reverse_tcp", ["LHOST=192.168.1.1","LPORT=443"]
             if 'msfvenom' in options:
-                if hasattr(options, 'msfvenom'):
+                if hasattr(self.payload, 'shellcode'):
                     self.payload.shellcode.SetPayload(options['msfvenom'])
                 else:
                     print helpers.color("\n [!] Internal error #3: This module does not use msfvenom!", warning=True)
+                    sys.exit()
 
             if not self.ValidatePayload(self.payload):