Fixed a threading issue, shuffled main()

Author: Al-Azif

start.py

@@ -92,11 +92,18 @@ def do_GET(self):
                 self.send_error(404)
 
             if path == 'rop.js':
-                print('>> Exploit Sent...')
-                thread = threading.Thread(target=payload_brain,
-                                          args=(self.client_address[0],))
-                thread.daemon = True
-                thread.start()
+                payload_menu = True
+                for thread in threading.enumerate():
+                    if thread.name == 'payload_brain':
+                        payload_menu = False
+
+                if payload_menu:
+                    print('>> Exploit Sent...')
+                    thread = threading.Thread(name='payload_brain',
+                                              target=payload_brain,
+                                              args=(self.client_address[0],),
+                                              daemon=True)
+                    thread.start()
 
 
 def check_root():
@@ -249,8 +256,9 @@ def start_servers():
 
     print('>> Starting HTTP server thread...')
     server = ThreadedHTTPServer(('', 80), MyHandler)
-    thread = threading.Thread(target=server.serve_forever, args=())
-    thread.daemon = True
+    thread = threading.Thread(target=server.serve_forever,
+                              args=(),
+                              daemon=True)
     thread.start()
     clear_line()
     print('>> HTTP server thread is running...')
@@ -396,32 +404,32 @@ def main():
                         required=False, help='Print debug statements')
     args = parser.parse_args()
 
-    if not args.e_type:
-        args.e_type = exploit_menu()
+    try:
+        if args.debug:
+            DEBUG = True
+        else:
+            DEBUG = False
+            MyHandler.log_message = silence_http
 
-    if os.path.isdir(os.path.join(EXPLOIT_LOC, args.e_type)) and args.e_type:
-        EXPLOIT_LOC = os.path.join(EXPLOIT_LOC, args.e_type)
-    else:
-        closer('ERROR: Could not find exploit specified')
+        check_update_pups()
 
-    if args.debug:
-        DEBUG = True
-    else:
-        DEBUG = False
-        MyHandler.log_message = silence_http
+        if not args.e_type:
+            args.e_type = exploit_menu()
 
-    check_update_pups()
+        if os.path.isdir(os.path.join(EXPLOIT_LOC, args.e_type)) and args.e_type:
+            EXPLOIT_LOC = os.path.join(EXPLOIT_LOC, args.e_type)
+        else:
+            closer('ERROR: Could not find exploit specified')
 
-    lan = get_lan()
+        lan = get_lan()
 
-    if write_conf(lan):
-        print('>> Your DNS IP is {}'.format(lan))
-    else:
-        closer('ERROR: Unable to write {}'.format(DNS_LOC))
+        if write_conf(lan):
+            print('>> Your DNS IP is {}'.format(lan))
+        else:
+            closer('ERROR: Unable to write {}'.format(DNS_LOC))
 
-    start_servers()
+        start_servers()
 
-    try:
         while True:
             pass
     except KeyboardInterrupt: