mirror of https://github.com/Justuser3310/jetwork/
parent
f89269a1f6
commit
27eba1fbf4
@ -0,0 +1,71 @@
|
||||
import http.server
|
||||
import socketserver
|
||||
import urllib.request
|
||||
import logging
|
||||
|
||||
from db import *
|
||||
domain = read()['domain']
|
||||
|
||||
# Логирование
|
||||
#logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s')
|
||||
from icecream import ic
|
||||
|
||||
class Proxy(http.server.SimpleHTTPRequestHandler):
|
||||
def do_GET(self):
|
||||
# Log the requested domain
|
||||
logging.info(f"Request for: {self.path}")
|
||||
|
||||
# 'js-check.jet/favicon.ico' -> '127.0.0.1:8000/favicon.ico'
|
||||
target = self.path
|
||||
# Если статичный
|
||||
if 'jet' in target:
|
||||
target = f'{domain}/{target[7:]}' # http://127.0.0.1:8000 / js-check.jet
|
||||
logging.info(f"Modded request: {target}")
|
||||
elif 'dyn' in target:
|
||||
addr = target[target.find('://')+3:]
|
||||
addr = addr[:addr.find('/')]
|
||||
port = read(f'cached/{addr}/config.json')['port']
|
||||
target = f'http://bore.pub:{port}/{target[target.find("dyn")+4:]}'
|
||||
logging.info(f"Modded request: {target}")
|
||||
|
||||
# Forward the request to the actual server
|
||||
try:
|
||||
with urllib.request.urlopen(target) as response:
|
||||
self.send_response(response.getcode())
|
||||
self.send_header("Content-type", response.headers.get_content_type())
|
||||
self.end_headers()
|
||||
self.wfile.write(response.read())
|
||||
except Exception as e:
|
||||
self.send_error(500, f"Error: {str(e)}")
|
||||
|
||||
def do_POST(self):
|
||||
# Log the requested domain
|
||||
logging.info(f"Request for: {self.path}")
|
||||
|
||||
if 'jet' in target:
|
||||
target = f'{domain}/{target[7:]}' # http://127.0.0.1:8000 / js-check.jet
|
||||
logging.info(f"Modded request: {target}")
|
||||
else:
|
||||
pass
|
||||
|
||||
# Forward the request to the actual server
|
||||
try:
|
||||
content_length = int(self.headers['Content-Length'])
|
||||
post_data = self.rfile.read(content_length)
|
||||
req = urllib.request.Request(self.path, data=post_data, method='POST')
|
||||
with urllib.request.urlopen(req) as response:
|
||||
self.send_response(response.getcode())
|
||||
self.send_header("Content-type", response.headers.get_content_type())
|
||||
self.end_headers()
|
||||
self.wfile.write(response.read())
|
||||
except Exception as e:
|
||||
self.send_error(500, f"Error: {str(e)}")
|
||||
|
||||
#if __name__ == "__main__":
|
||||
def web_proxy():
|
||||
PORT = 8080
|
||||
with socketserver.TCPServer(("", PORT), Proxy) as httpd:
|
||||
logging.info(f"Serving on port {PORT}")
|
||||
httpd.serve_forever()
|
||||
|
||||
web_proxy()
|
Loading…
Reference in new issue