diff --git a/jsh.py b/jsh.py index e0b7162..9ef78e6 100644 --- a/jsh.py +++ b/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))