Add new fuzzer engine, siddhis exploit and general improvements /fix #8

Author: s4dhul4bs

core/vmnf_arg_parser.py

@@ -60,13 +60,14 @@ def parse_args(self):
         list_cmd = subparsers.add_parser('list', 
             help='List available resources'
         )
+        list_cmd.add_argument('--payloads', action='store_true',dest='list_payloads')
         list_cmd.add_argument('--modules', action='store_true',dest='module_list')
         list_cmd.add_argument('-t', '--type', action='store')
         list_cmd.add_argument('-c', '--category', action='store', dest='category')
         list_cmd.add_argument('-f', '--framework', action='store', dest='framework')
         list_cmd.add_argument('-x', '--exploits', action='store_true')
         list_cmd.add_argument('-p', '--payload',action='store',
-	    choices=('reverse-shell', 'bind-port', 'backdoor', 'exfiltration')
+	    choices=('reverse-shell', 'bind-port', 'backdoor', 'exfiltration-server')
         )
 
         # -----------------------------------------------------------------
@@ -112,7 +113,7 @@ def start_handler(self):
         mod_type_by_id = {
             0:'tracker',
             1:'fuzzer',
-            2:'brute',
+            2:'attack',
             3:'leaker',
             4:'exploit'
         }
@@ -163,6 +164,7 @@ def start_handler(self):
             module_info     = False,
             module_run      = False,
             module_list     = False,
+            list_payloads   = False,
             module_args     = False,
             framework       = False,
             url_conf        = False,

core/vmnf_check_target.py

@@ -46,7 +46,7 @@ def worker(self):
                 _status_ = colored('Open', 'green', attrs=['bold'])
             else:
                 self.closed_ports.append(port)
-                _status_ = colored('Closed', 'red', attrs=['bold'])
+                _status_ = colored('Closed', 'red', attrs=[])
 
             _target_ = colored(self.target.strip(), 'yellow')
             _port_   = colored(port, 'yellow')

core/vmnf_engine.py

@@ -4,28 +4,32 @@
 import sys
 sys.path.insert(0, '../../../')
 
+from termcolor import colored, cprint
 from pathlib import Path
 from time import sleep
 import pathlib
 import argparse
 import os
 
 # vimana core modules
+from siddhis.dmt.dmt import siddhi as dmt_siddhi
 from siddhis.djunch.djunch import siddhi as Djunch
+
 from core.vmnf_fuzz_scope import handle_fuzz_scope
 from core.vmnf_scope_parser import ScopeParser
 from core.vmnf_urls_parser import digest_scope
 from core.vmnf_arg_parser import VimanaParser
+from core.vmnf_payloads import  VMNFPayloads 
 from core.vmnf_manager import vmng
-from siddhis.dmt.dmt import siddhi as dmt_siddhi
+
 
 # vimana helpers 
 from helpers.vmnf_helpers import VimanaHelp
 
 # vimana resources
 from resources import vmnf_banners 
 from resources.vmnf_banners import s4dhu0nv1m4n4
-
+from resources.colors import *
 
 def abduct():
     # commands that require --module argument
@@ -107,12 +111,38 @@ def abduct():
     elif handler_ns.module_info:
         vmng(**vars(handler_ns))  
 
-    # start fuzzer
+    # start only fuzzer directly
     elif handler_ns.fuzzer:
         scope = handle_fuzz_scope(**vars(handler_ns))
+    
     # start discovery
     elif handler_ns.discovery:
         print('Wait future releases for this feature. [:')
 
-    
+    # list available payloads
+    elif handler_ns.list_payloads:
+        VMNFPayloads()._vmnfp_payload_types_(False,True)
 
+        
+        """
+        vmnf_payloads = {}
+        [vmnf_payloads.__setitem__(payload, getattr(_payloads_, payload).__doc__) \
+            for payload in [attr for attr in dir(_payloads_) \
+                if not attr.startswith('_')
+	    ]
+        ]
+        
+        print("\033c", end="") 
+        print('\n\t {}'.format(
+            colored('⣷⣒⠂    ', 'green')) + colored(' Vimana Payloads (v0.1)', 'magenta', attrs=[]) +\
+            colored('    ⣸⣼⡀', 'green') + '\n\n'
+        )
+
+        for k,v in vmnf_payloads.items():
+            print('{}{}:\t   \x1B[3m{}\x1B[23m'.format(
+                    (' ' * int(5-len(k) + 14)),
+                    colored(k,'cyan'),colored(v, 'green')
+                )
+            )
+        print('\n\n')
+        """