use wisp server instead of wsproxy

This commit is contained in:
ading2210 2024-01-16 00:25:29 -05:00
parent 95121068bc
commit badeaa8ac3
6 changed files with 9 additions and 60 deletions

5
.gitmodules vendored
View file

@ -1,3 +1,6 @@
[submodule "client/wisp_client"] [submodule "client/wisp_client"]
path = client/wisp_client path = client/wisp_client
url = https://github.com/mercuryWorkshop/wisp-client-js url = https://github.com/MercuryWorkshop/wisp-client-js
[submodule "server/wisp_server"]
path = server/wisp_server
url = https://github.com/MercuryWorkshop/wisp-server-python

View file

@ -10,7 +10,7 @@ This is an experimental port of [libcurl](https://curl.se/libcurl/) to WebAssemb
## Building: ## Building:
You can build this project by running the following commands: You can build this project by running the following commands:
``` ```
git clone https://github.com/ading2210/libcurl.js git clone https://github.com/ading2210/libcurl.js --recursive
cd libcurl.js/client cd libcurl.js/client
./build.sh ./build.sh
``` ```
@ -48,11 +48,11 @@ libcurl.set_websocket("ws://localhost:6001/");
``` ```
## Proxy Server: ## Proxy Server:
The proxy server consists of a [SOCKS5 proxy server](https://github.com/Amaindex/asyncio-socks-server) behind a [websocket TCP proxy](https://github.com/novnc/websockify). The proxy server consists of a standard [Wisp](https://github.com/MercuryWorkshop/wisp-protocol) server, allowing multiple TCP connections to share the same websocket.
To host the proxy server, run the following commands: To host the proxy server, run the following commands:
``` ```
git clone https://github.com/ading2210/libcurl.js git clone https://github.com/ading2210/libcurl.js --recursive
cd libcurl.js/server cd libcurl.js/server
./run.sh ./run.sh
``` ```

View file

@ -1,55 +0,0 @@
import asyncio
from websockets.server import serve
from websockets.exceptions import ConnectionClosed
buffer_size = 1024*1024
class Connection:
def __init__(self, ws, path):
self.ws = ws
self.path = path
async def setup_connection(self):
addr_str = self.path.split("/")[-1]
self.tcp_host, self.tcp_port = addr_str.split(":")
self.tcp_port = int(self.tcp_port)
self.tcp_reader, self.tcp_writer = await asyncio.open_connection(host=self.tcp_host, port=self.tcp_port, limit=buffer_size)
async def handle_ws(self):
while True:
try:
data = await self.ws.recv()
except ConnectionClosed:
break
self.tcp_writer.write(data)
await self.tcp_writer.drain()
print("sent data to tcp")
self.tcp_writer.close()
async def handle_tcp(self):
while True:
data = await self.tcp_reader.read(buffer_size)
if len(data) == 0:
break #socket closed
await self.ws.send(data)
print("sent data to ws")
await self.ws.close()
async def connection_handler(websocket, path):
print("incoming connection from "+path)
connection = Connection(websocket, path)
await connection.setup_connection()
ws_handler = asyncio.create_task(connection.handle_ws())
tcp_handler = asyncio.create_task(connection.handle_tcp())
await asyncio.gather(ws_handler, tcp_handler)
async def main():
async with serve(connection_handler, "127.0.0.1", 6001, subprotocols=["binary"]):
await asyncio.Future()
if __name__ == "__main__":
asyncio.run(main())

View file

@ -1 +0,0 @@
websockets

View file

@ -4,6 +4,7 @@
set -e set -e
cd wisp_server
if [ ! -d ".venv" ]; then if [ ! -d ".venv" ]; then
python3 -m venv .venv python3 -m venv .venv
fi fi

1
server/wisp_server Submodule

@ -0,0 +1 @@
Subproject commit 874734623e4dfc4652b34a1bc61e1e35ca86dee8