Add Py2 alert, catch socket error, changer payload sender timeout

Author: Al-Azif

start.py

@@ -4,6 +4,17 @@
    Source: https://github.com/Al-Azif/ps4-exploit-host
 """
 
+from __future__ import print_function
+
+
+import sys
+if sys.version_info.major < 3:
+    print('This cannot be run under Python 2')
+    try:
+        input('Press [ENTER] to exit')
+    finally:
+        sys.exit()
+
 import argparse
 import hashlib
 from http.server import BaseHTTPRequestHandler
@@ -13,7 +24,6 @@
 import os
 import re
 import socket
-import sys
 import threading
 import time
 
@@ -307,25 +317,28 @@ def payload_menu(payloads):
 def send_payload(hostname, port, content):
     """Netcat implementation"""
     soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-    timeout = time.time() + 60
+    timeout = time.time() + 15
     while True:
         result = soc.connect_ex((hostname, port))
         if result == 0:
             print('>> Connected to PS4')
             timed_out = False
             break
         if time.time() >= timeout:
-            print('>> Netcat service timed out')
+            print('ERROR: Payload sender timed out')
             timed_out = True
             break
     if not timed_out:
-        soc.sendall(content)
-        soc.shutdown(socket.SHUT_WR)
-        while True:
-            data = soc.recv(1024)
-            if not data:
-                break
-        print('>> Payload Sent!')
+        try:
+            soc.sendall(content)
+            soc.shutdown(socket.SHUT_WR)
+            while True:
+                data = soc.recv(1024)
+                if not data:
+                    break
+            print('>> Payload Sent!')
+        except socket.error:
+            print('ERROR: Broken Pipe (Out of Memory Error?)')
     soc.close()
 
 
@@ -418,7 +431,8 @@ def main():
         if not args.e_type:
             args.e_type = exploit_menu()
 
-        if os.path.isdir(os.path.join(EXPLOIT_LOC, args.e_type)) and args.e_type:
+        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')