gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-03-12 00:45:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update server.py

Browse Source
This commit is contained in:
Namhyeon Go 2024-02-23 15:02:14 +09:00 committed by GitHub
parent 766b068641
commit 40b459a3c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server.py
View File

@ -18,7 +18,7 @@ import hashlib
import resource import resource
import traceback import traceback
import io import io
#import textwrap import textwrap
from subprocess import Popen, PIPE from subprocess import Popen, PIPE
from datetime import datetime from datetime import datetime
from platform import python_version from platform import python_version
@ -300,9 +300,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
# override data # override data
if is_ssl: if is_ssl:
_, _, _, method, _url = parse_first_data(data) _, _, _, method, url = parse_first_data(data)
else:
_url = b''
# https://stackoverflow.com/questions/44343739/python-sockets-ssl-eof-occurred-in-violation-of-protocol # https://stackoverflow.com/questions/44343739/python-sockets-ssl-eof-occurred-in-violation-of-protocol
def sock_close(sock, is_ssl = False): def sock_close(sock, is_ssl = False):
@ -387,22 +385,20 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
return return
# https://stackoverflow.com/questions/20658572/python-requests-print-entire-http-request-raw # https://stackoverflow.com/questions/20658572/python-requests-print-entire-http-request-raw
'''
format_headers = lambda d: '\r\n'.join(f'{k}: {v}' for k, v in d.items()) format_headers = lambda d: '\r\n'.join(f'{k}: {v}' for k, v in d.items())
first_data = textwrap.dedent('{res.status_code} {res.reason} {res.url}\r\n{reshdrs}\r\n\r\n').format( first_data = textwrap.dedent('HTTP/1.1 {res.status_code} {res.reason}\r\n{reshdrs}\r\n\r\n').format(
res=response, res=response,
reshdrs=format_headers(response.headers), reshdrs=format_headers(response.headers),
).encode(client_encoding) ).encode(client_encoding)
conn.send(first_data) conn.send(first_data)
'''
for chunk in response.iter_content(chunk_size=buffer_size): for chunk in response.iter_content(chunk_size=buffer_size):
conn.send(chunk) conn.send(chunk)
if is_ssl and method == b'GET': if is_ssl and method == b'GET':
print ("[*] Trying to bypass blocked request...") print ("[*] Trying to bypass blocked request...")
remote_url = "%s://%s%s" % (scheme.decode(client_encoding), webserver.decode(client_encoding), _url.decode(client_encoding)) remote_url = "%s://%s%s" % (scheme.decode(client_encoding), webserver.decode(client_encoding), url.decode(client_encoding))
requests.get(remote_url, stream=True, hooks={'response': bypass_callback}) requests.get(remote_url, stream=True, hooks={'response': bypass_callback})
else: else:
conn.sendall(b"HTTP/1.1 403 Forbidden\r\n\r\n{\"status\":403}") conn.sendall(b"HTTP/1.1 403 Forbidden\r\n\r\n{\"status\":403}")