Large number of cookies exceeding the defined buffer size can crash the script

jsh.py

@@ -1,9 +1,10 @@
 #!/usr/bin/env python3
 import socket
 import sys
-from requests import get
 import argparse
 
+from requests import get
+
 
 red = '\033[1;31m'
 white = '\033[1;m'
@@ -37,7 +38,6 @@
 parser.add_argument('-w', help='timeout for shell connection', dest='secs', type=float, default=0)
 parser.add_argument('-q', help='quiet mode', dest='quiet', action='store_true')
 
-
 args = parser.parse_args()
 
 host = args.host
@@ -174,14 +174,20 @@ def main():
 
     try:
         c, addr = s.accept()
-        resp = c.recv(1024).decode()
+        resp = ""
+        while True:
+            b = c.recv(1024).decode()
+            if not b:
+                break
+            resp += b
     except KeyboardInterrupt:
         if sys.platform == 'win32':
                 print('\nControl-C')
         exit()
-    except:
+    except(e):
+        print(e)
         s.close()
-        main()
+
 
     if 'Accept' in resp and 'HTTP' in resp:
         print('Got JS shell from [%s] port %s to %s %s' % (addr[0], addr[1], socket.gethostname(), port))