use latest websockify, allow user to change port

This commit is contained in:
ading2210 2024-01-08 01:38:41 -05:00
parent 953b6cb191
commit 7472c899db
4 changed files with 18 additions and 10 deletions

View file

@ -1,5 +1,4 @@
import logging
import threading
import os
from asyncio_socks_server.app import SocksServer
@ -16,28 +15,31 @@ def setup_logging(prefix):
root.addHandler(stderr_handler)
root.setLevel(logging.INFO)
def start_websockify():
def start_websockify(listen_port, proxy_port):
options = {
"listen_host": "127.0.0.1",
"listen_port": 6001,
"listen_port": int(listen_port),
"target_host": "127.0.0.1",
"target_port": 1080
"target_port": int(proxy_port)
}
server = WebSocketProxy(**options)
server.start_server()
def start_socks():
def start_socks(proxy_port):
socks_app = SocksServer(
LISTEN_HOST="127.0.0.1",
LISTEN_PORT=1080
LISTEN_PORT=int(proxy_port)
)
socks_app.run()
if __name__ == "__main__":
listen_port = os.environ.get("PORT") or 6001
proxy_port = os.environ.get("SOCKS5_PORT") or 6002
pid = os.fork()
if pid == 0:
setup_logging("[websockify] ")
start_websockify()
start_websockify(listen_port, proxy_port)
else:
start_socks()
start_socks(proxy_port)