verify exploit name input

Author: jm33-m0

lib/cli/main.py

@@ -91,7 +91,7 @@ def execute(cmd):
             colors.END)
     elif cmd.startswith('target'):
         target = ''.join(cmd.split()[1:])
-        if not target.endswith('.txt'):
+        if not target in os.listdir(SessionParameters.INIT_DIR + '\\data'):
             return
         print(colors.BLUE + '[i] Target changed to {}'.format(target))
         SessionParameters.IP_LIST = SessionParameters.INIT_DIR + \
@@ -395,11 +395,16 @@ def main():
             '[?] Use ip_list.txt as target list? [y/n] ' +
             colors.END)).strip()
     if answ.lower() == 'n':
-        os.system("ls data")
+        for item in os.listdir('data'):
+            print(colors.BLUE,
+                  item,
+                  colors.END,
+                  end=' ')
+        print('\n')
         SessionParameters.IP_LIST = SessionParameters.INIT_DIR + '\\data\\' + \
             input_check(
                 '[=] Choose your target IP list, eg. ip_list.txt ',
-                allow_blank=False)
+                choices=os.listdir('data'))
     while True:
         try:
             cmd = input(