Fix: Show/Set Payloads issue (#501)

routersploit/core/exploit/shell.py

@@ -35,7 +35,7 @@ def shell(exploit, architecture="", method="", payloads=None, **params):
 
         payload_path = path.replace("/", ".")
         for p in all_payloads:
-            module = getattr(importlib.import_module("{}{}".format(payload_path, p)), 'Exploit')
+            module = getattr(importlib.import_module("{}{}".format(payload_path, p)), 'Payload')
 
             # if method/arch is cmd then filter out payloads
             if method is "cmd":
@@ -56,7 +56,7 @@ def shell(exploit, architecture="", method="", payloads=None, **params):
         if payload is None:
             cmd_str = "\001\033[4m\002cmd\001\033[0m\002 > "
         else:
-            cmd_str = "\001\033[4m\002cmd\001\033[0m\002 (\033[94m{}\033[0m) > ".format(payload._Exploit__info__["name"])
+            cmd_str = "\001\033[4m\002cmd\001\033[0m\002 (\033[94m{}\033[0m) > ".format(payload._Payload__info__["name"])
 
         cmd = input(cmd_str)
 
@@ -72,7 +72,7 @@ def shell(exploit, architecture="", method="", payloads=None, **params):
             headers = ("Payload", "Name", "Description")
             data = []
             for p in available_payloads.keys():
-                data.append((p, available_payloads[p]._Exploit__info__["name"], available_payloads[p]._Exploit__info__["description"]))
+                data.append((p, available_payloads[p]._Payload__info__["name"], available_payloads[p]._Payload__info__["description"]))
 
             print_table(headers, *data)