wisp-python 0.6.1__py3-none-any.whl → 0.7.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
wisp/server/__main__.py CHANGED
@@ -1,26 +1,4 @@
1
- import argparse
2
- import asyncio
3
- import sys
4
-
5
- try:
6
- import uvloop
7
- use_uvloop = True
8
- except ImportError:
9
- use_uvloop = False
10
-
11
- from wisp.server import http
12
- from wisp.server import common
1
+ from wisp.server import cli
13
2
 
14
3
  if __name__ == "__main__":
15
- args = common.setup_main()
16
-
17
- try:
18
- if use_uvloop:
19
- uvloop.run(http.main(args))
20
- else:
21
- #uvloop doesn't support windows at all so we don't need to print the error
22
- if not sys.platform in ("win32", "cygwin"):
23
- logging.error("Importing uvloop failed. Falling back to asyncio, which is slower.")
24
- asyncio.run(http.main(args))
25
- except KeyboardInterrupt:
26
- pass
4
+ cli.main()
wisp/server/cli.py ADDED
@@ -0,0 +1,86 @@
1
+ import asyncio
2
+ import argparse
3
+ import pathlib
4
+ import sys
5
+ import logging
6
+ import multiprocessing
7
+
8
+ try:
9
+ import uvloop
10
+ use_uvloop = True
11
+ except ImportError:
12
+ use_uvloop = False
13
+
14
+ import wisp
15
+ from wisp.server import http
16
+ from wisp.server import net
17
+
18
+ def run_async(func, *args, **kwargs):
19
+ try:
20
+ if use_uvloop:
21
+ uvloop.run(func(*args, **kwargs))
22
+ else:
23
+ #uvloop doesn't support windows at all so we don't need to print the error
24
+ if not sys.platform in ("win32", "cygwin"):
25
+ logging.error("Importing uvloop failed. Falling back to asyncio, which is slower.")
26
+ asyncio.run(func(*args, **kwargs))
27
+ except KeyboardInterrupt:
28
+ pass
29
+
30
+ def run_http(args):
31
+ run_async(http.main, args)
32
+
33
+ def main():
34
+ parser = argparse.ArgumentParser(
35
+ prog="wisp-server-python",
36
+ description=f"A Wisp server implementation, written in Python (v{wisp.version})"
37
+ )
38
+
39
+ parser.add_argument("--host", default="127.0.0.1", help="The hostname the server will listen on.")
40
+ parser.add_argument("--port", default=6001, help="The TCP port the server will listen on.")
41
+ parser.add_argument("--static", help="Where static files are served from.")
42
+ parser.add_argument("--limits", action="store_true", help="Enable rate limits.")
43
+ parser.add_argument("--bandwidth", default=1000, help="Bandwidth limit per IP, in kilobytes per second.")
44
+ parser.add_argument("--connections", default=30, help="New connections limit per IP.")
45
+ parser.add_argument("--window", default=60, help="Fixed window length for rate limits, in seconds.")
46
+ parser.add_argument("--allow-loopback", action="store_true", help="Allow connections to loopback IP addresses.")
47
+ parser.add_argument("--allow-private", action="store_true", help="Allow connections to private IP addresses.")
48
+ parser.add_argument("--log-level", default="info", help="The log level (either debug, info, warning, error, or critical).")
49
+ parser.add_argument("--threads", default=0, help="The number of threads to run the server on. By default it uses all CPU cores.")
50
+ args = parser.parse_args()
51
+
52
+ logging.basicConfig(
53
+ format="[%(asctime)s] %(levelname)-8s %(message)s",
54
+ level=getattr(logging, args.log_level.upper()),
55
+ datefmt="%Y/%m/%d - %H:%M:%S"
56
+ )
57
+
58
+ logging.info(f"running wisp-server-python v{wisp.version} (async)")
59
+ if args.static:
60
+ static_path = pathlib.Path(args.static).resolve()
61
+ logging.info(f"serving static files from {static_path}")
62
+ if args.limits:
63
+ logging.info("enabled rate limits")
64
+ logging.info(f"listening on {args.host}:{args.port}")
65
+
66
+ threads = int(args.threads)
67
+ if net.reuse_port_supported():
68
+ if threads == 0:
69
+ threads = multiprocessing.cpu_count()
70
+ logging.info(f"running using {threads} threads")
71
+
72
+ processes = []
73
+ for i in range(0, int(threads)):
74
+ process = multiprocessing.Process(target=run_http, args=(args,), daemon=True)
75
+ processes.append(process)
76
+ process.start()
77
+ try:
78
+ for process in processes:
79
+ process.join()
80
+ except KeyboardInterrupt:
81
+ pass
82
+
83
+ else:
84
+ if threads != 0:
85
+ logging.warn("the --threads option is not supported on this platform")
86
+ run_http(args)
wisp/server/http.py CHANGED
@@ -61,18 +61,15 @@ async def static_handler(path, request_headers):
61
61
 
62
62
  async def main(args):
63
63
  global static_path
64
- logging.info(f"running wisp-server-python v{wisp.version} (async)")
65
64
 
66
65
  if args.static:
67
66
  static_path = pathlib.Path(args.static).resolve()
68
67
  request_handler = static_handler
69
68
  mimetypes.init()
70
- logging.info(f"serving static files from {static_path}")
71
69
  else:
72
70
  request_handler = None
73
71
 
74
72
  if args.limits:
75
- logging.info("enabled rate limits")
76
73
  ratelimit.enabled = True
77
74
  ratelimit.connections_limit = int(args.connections)
78
75
  ratelimit.bandwidth_limit = float(args.bandwidth)
@@ -82,10 +79,10 @@ async def main(args):
82
79
  net.block_private = not args.allow_private
83
80
 
84
81
  limit_task = asyncio.create_task(ratelimit.reset_limits_timer())
85
- logging.info(f"listening on {args.host}:{args.port}")
86
-
87
82
  ws_logger = logging.getLogger("websockets")
88
83
  ws_logger.setLevel(logging.WARN)
89
84
 
90
- async with serve(connection_handler, args.host, int(args.port), process_request=request_handler, compression=None):
85
+ reuse_port = net.reuse_port_supported()
86
+
87
+ async with serve(connection_handler, args.host, int(args.port), reuse_port=reuse_port, process_request=request_handler, compression=None):
91
88
  await asyncio.Future()
wisp/server/net.py CHANGED
@@ -9,6 +9,9 @@ tcp_size = 64*1024
9
9
  block_loopback = False
10
10
  block_private = False
11
11
 
12
+ def reuse_port_supported():
13
+ return hasattr(socket, "SO_REUSEPORT")
14
+
12
15
  def get_ip(host, port, stream_type):
13
16
  if stream_type == 0x01:
14
17
  proto = socket.IPPROTO_TCP
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: wisp-python
3
- Version: 0.6.1
3
+ Version: 0.7.0
4
4
  Summary: An implementation of a Wisp proxy server in Python
5
5
  Author: ading2210
6
6
  Project-URL: Homepage, https://github.com/MercuryWorkshop/wisp-server-python
@@ -36,18 +36,16 @@ pip3 install wisp-python
36
36
  ```
37
37
 
38
38
  ## Running the Server:
39
- There are two different versions of the server:
40
- - Async (`wisp.server`) - The default, based on asyncio. It is more stable, but slower.
41
- - Threading (`wisp.server.threading`) - An faster alternative based on threading. It is about 40-80% faster, but possibly unstable. If you use Python's experimental [free threading](https://docs.python.org/3/howto/free-threading-python.html) builds, performance will scale linearly with your available CPU cores.
42
-
43
39
  For the best performance use [CPython](https://github.com/python/cpython) 3.11 or newer. [PyPy](https://github.com/pypy/pypy) is not recommended as it is a lot slower than CPython here.
44
40
 
45
- To start the server, run `python3 -m wisp.server` (or `python3 -m wisp.server.threading`). The program accepts the following arguments:
41
+ There used to be an experimental "threading" variant of the server. This has been removed due to its instability and difficulty of maintenance.
42
+
43
+ To start the server, run `python3 -m wisp.server`. The program accepts the following arguments:
46
44
  ```
47
- usage: wisp-server-python [-h] [--host HOST] [--port PORT] [--static STATIC] [--limits] [--bandwidth BANDWIDTH] [--connections CONNECTIONS] [--window WINDOW] [--allow-loopback] [--allow-private]
48
- [--log-level LOG_LEVEL]
45
+ usage: wisp-server-python [-h] [--host HOST] [--port PORT] [--static STATIC] [--limits] [--bandwidth BANDWIDTH] [--connections CONNECTIONS] [--window WINDOW] [--allow-loopback]
46
+ [--allow-private] [--log-level LOG_LEVEL] [--threads THREADS]
49
47
 
50
- A Wisp server implementation, written in Python (v0.5.0)
48
+ A Wisp server implementation, written in Python (v0.7.0)
51
49
 
52
50
  options:
53
51
  -h, --help show this help message and exit
@@ -64,6 +62,7 @@ options:
64
62
  --allow-private Allow connections to private IP addresses.
65
63
  --log-level LOG_LEVEL
66
64
  The log level (either debug, info, warning, error, or critical).
65
+ --threads THREADS The number of threads to run the server on. By default it uses all CPU cores.
67
66
  ```
68
67
 
69
68
  ## Roadmap:
@@ -0,0 +1,14 @@
1
+ wisp/__init__.py,sha256=02Fv0s_jJG8U4y5B6AzKPIOzxAuqJyY_So8Guz5bdTk,75
2
+ wisp/util.py,sha256=AU7si8VwJqS8Dwc1PT2pu_f_9Y32GQxivRYwJ3-U0YQ,144
3
+ wisp/server/__init__.py,sha256=LmgQ1nTlnqcTPqRJVc1NGapV9qVR-BK8iPVUMSJ74Qs,102
4
+ wisp/server/__main__.py,sha256=zhDqmhOY9HgHx-BOq-kU5IoxZCMJwt_PEEEISaZr9Mk,68
5
+ wisp/server/cli.py,sha256=6lO0HqDsaMISoOyN6r7DnTnsqqITrlosylxydLd5T2A,3153
6
+ wisp/server/connection.py,sha256=V_Skmmsvn30lQKY5w9x2rJQFR_FsA3lngNDpaYOfX6U,7669
7
+ wisp/server/http.py,sha256=Es8_xR4xAFsYY-tezSzA8XPFvpetS1VUFjlyNXO4CrE,2983
8
+ wisp/server/net.py,sha256=XXZolHei7VoyQgRtrFF02COaENVG9q8FOXfJqHLiApc,2335
9
+ wisp/server/ratelimit.py,sha256=cic_rlBaM-tp6kJaSz4ITGvUbsX4JLSGSUgF5cPO0FY,1926
10
+ wisp_python-0.7.0.dist-info/licenses/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
11
+ wisp_python-0.7.0.dist-info/METADATA,sha256=O76KfKGQa3lWdwu_0Uw8i36mNtvCS4zpOMycZtqaXPU,3860
12
+ wisp_python-0.7.0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
13
+ wisp_python-0.7.0.dist-info/top_level.txt,sha256=sv0LDAu8cHUc3_VTsCKfXU-6B9HdV4dW22ubeW8KPVs,5
14
+ wisp_python-0.7.0.dist-info/RECORD,,
wisp/server/common.py DELETED
@@ -1,30 +0,0 @@
1
- import argparse
2
- import logging
3
-
4
- import wisp
5
-
6
- def setup_main():
7
- parser = argparse.ArgumentParser(
8
- prog="wisp-server-python",
9
- description=f"A Wisp server implementation, written in Python (v{wisp.version})"
10
- )
11
-
12
- parser.add_argument("--host", default="127.0.0.1", help="The hostname the server will listen on.")
13
- parser.add_argument("--port", default=6001, help="The TCP port the server will listen on.")
14
- parser.add_argument("--static", help="Where static files are served from.")
15
- parser.add_argument("--limits", action="store_true", help="Enable rate limits.")
16
- parser.add_argument("--bandwidth", default=1000, help="Bandwidth limit per IP, in kilobytes per second.")
17
- parser.add_argument("--connections", default=30, help="New connections limit per IP.")
18
- parser.add_argument("--window", default=60, help="Fixed window length for rate limits, in seconds.")
19
- parser.add_argument("--allow-loopback", action="store_true",help="Allow connections to loopback IP addresses.")
20
- parser.add_argument("--allow-private", action="store_true", help="Allow connections to private IP addresses.")
21
- parser.add_argument("--log-level", default="info", help="The log level (either debug, info, warning, error, or critical).")
22
- args = parser.parse_args()
23
-
24
- logging.basicConfig(
25
- format="[%(asctime)s] %(levelname)-8s %(message)s",
26
- level=getattr(logging, args.log_level.upper()),
27
- datefmt="%Y/%m/%d - %H:%M:%S"
28
- )
29
-
30
- return args
@@ -1,2 +0,0 @@
1
- from wisp.server.threading.connection import WispConnection
2
- from wisp.server.threading.connection import WSProxyConnection
@@ -1,6 +0,0 @@
1
- from wisp.server.threading import http
2
- from wisp.server import common
3
-
4
- if __name__ == "__main__":
5
- args = common.setup_main()
6
- http.main(args)
@@ -1,219 +0,0 @@
1
- import struct
2
- import os
3
- import logging
4
- import queue
5
- import threading
6
-
7
- from websockets.exceptions import ConnectionClosed, ConnectionClosedError
8
-
9
- from wisp.server import ratelimit
10
- from wisp.server.threading import netsync
11
-
12
- queue_size = 128
13
- static_path = None
14
-
15
- #wisp packet format definitions
16
- #see https://docs.python.org/3/library/struct.html for what these characters mean
17
- packet_format = "<BI"
18
- connect_format = "<BH"
19
- continue_format = "<I"
20
- close_format = "<B"
21
-
22
- class WSProxyConnection:
23
- def __init__(self, ws, path, client_ip):
24
- self.ws = ws
25
- self.path = path
26
- self.client_ip = client_ip
27
-
28
- def setup_connection(self):
29
- addr_str = self.path.split("/")[-1]
30
- self.tcp_host, self.tcp_port = addr_str.split(":")
31
- self.tcp_port = int(self.tcp_port)
32
-
33
- try:
34
- self.conn = netsync.TCPConnection(self.tcp_host, self.tcp_port)
35
- self.conn.connect()
36
- except Exception as e:
37
- logging.info(f"Creating a WSProxy stream to {self.tcp_host}:{self.tcp_port} failed: {e}")
38
- self.ws.close()
39
-
40
- def handle_ws(self):
41
- while True:
42
- try:
43
- data = self.ws.recv()
44
- except (ConnectionClosed, ConnectionClosedError) as e:
45
- break
46
-
47
- ratelimit.limit_client_bandwidth_sync(self.client_ip, len(data), "ws")
48
- self.conn.send(data)
49
-
50
- self.conn.close()
51
-
52
- def handle_tcp(self):
53
- while True:
54
- data = self.conn.recv()
55
- if len(data) == 0:
56
- break #socket closed
57
-
58
- ratelimit.limit_client_bandwidth_sync(self.client_ip, len(data), "tcp")
59
- self.ws.send(data)
60
-
61
- self.ws.close()
62
-
63
- class WispConnection:
64
- def __init__(self, ws, path, client_ip, id=None):
65
- self.ws = ws
66
- self.path = path
67
- self.active_streams = {}
68
- self.client_ip = client_ip
69
- self.id = id
70
-
71
- #send the initial CONTINUE packet
72
- def setup(self):
73
- continue_payload = struct.pack(continue_format, queue_size)
74
- continue_packet = struct.pack(packet_format, 0x03, 0) + continue_payload
75
- self.ws.send(continue_packet)
76
-
77
- def close(self):
78
- #close all active streams when the websocket disconnects
79
- for stream_id in list(self.active_streams.keys()):
80
- self.close_stream(stream_id)
81
- self.ws.close()
82
-
83
- def new_stream(self, stream_id, payload):
84
- stream_type, destination_port = struct.unpack(connect_format, payload[:3])
85
- hostname = bytes(payload[3:]).decode()
86
- logging.debug(f"({self.id}) Creating a new stream to {hostname}:{destination_port}")
87
-
88
- #rate limited
89
- stream_count = ratelimit.get_client_attr(self.client_ip, "streams")
90
- if ratelimit.enabled and stream_count > ratelimit.connections_limit:
91
- self.send_close_packet(stream_id, 0x49)
92
- self.close_stream(stream_id)
93
- return
94
-
95
- #info looks valid - try to open the connection now
96
- try:
97
- if stream_type == 0x01:
98
- connection = netsync.TCPConnection(hostname, destination_port)
99
- elif stream_type == 0x02:
100
- connection = netsync.UDPConnection(hostname, destination_port)
101
- else:
102
- raise Exception("Invalid stream type.")
103
- self.active_streams[stream_id]["conn"] = connection
104
- connection.connect()
105
-
106
- except Exception as e:
107
- logging.warn(f"({self.id}) Creating a new stream to {hostname}:{destination_port} failed: {e}")
108
- self.send_close_packet(stream_id, 0x42)
109
- self.close_stream(stream_id)
110
- return
111
-
112
- self.active_streams[stream_id]["type"] = stream_type
113
- threading.Thread(target=self.stream_ws_to_tcp, args=(stream_id,), daemon=True).start()
114
- threading.Thread(target=self.stream_tcp_to_ws, args=(stream_id,), daemon=True).start()
115
- ratelimit.inc_client_attr(self.client_ip, "streams")
116
-
117
- def stream_ws_to_tcp(self, stream_id):
118
- while True:
119
- stream = self.active_streams[stream_id]
120
- data = stream["queue"].get()
121
- if data is None:
122
- break
123
- try:
124
- stream["conn"].send(data)
125
- except:
126
- break
127
-
128
- #send a CONTINUE packet periodically
129
- stream["packets_sent"] += 1
130
- if stream["packets_sent"] % (queue_size // 4) == 0:
131
- buffer_remaining = stream["queue"].maxsize - stream["queue"].qsize()
132
- continue_payload = struct.pack(continue_format, buffer_remaining)
133
- continue_packet = struct.pack(packet_format, 0x03, stream_id) + continue_payload
134
- self.ws.send(continue_packet)
135
-
136
- def stream_tcp_to_ws(self, stream_id):
137
- while True:
138
- stream = self.active_streams[stream_id]
139
- try:
140
- data = stream["conn"].recv()
141
- except Exception as e:
142
- logging.warn(f"({self.id}) Receiving data from stream failed: {e}")
143
- self.send_close_packet(stream_id, 0x03)
144
- self.close_stream(stream_id)
145
- return
146
-
147
- if len(data) == 0: #connection closed
148
- break
149
- data_header = struct.pack(packet_format, 0x02, stream_id)
150
-
151
- ratelimit.limit_client_bandwidth_sync(self.client_ip, len(data_header)+len(data), "tcp")
152
- try:
153
- self.ws.send(data_header + data)
154
- except (ConnectionClosed, ConnectionClosedError) as e:
155
- return
156
-
157
- self.send_close_packet(stream_id, 0x02)
158
- self.close_stream(stream_id)
159
-
160
- def send_close_packet(self, stream_id, reason):
161
- if not stream_id in self.active_streams:
162
- return
163
- close_payload = struct.pack(close_format, reason)
164
- close_packet = struct.pack(packet_format, 0x04, stream_id) + close_payload
165
- self.ws.send(close_packet)
166
-
167
- def close_stream(self, stream_id):
168
- if not stream_id in self.active_streams:
169
- return #stream already closed
170
- stream = self.active_streams[stream_id]
171
- if stream["conn"]:
172
- stream["conn"].close()
173
-
174
- #empty the queue and kill it
175
- while not stream["queue"].empty():
176
- stream["queue"].get()
177
- stream["queue"].put(None)
178
- del self.active_streams[stream_id]
179
-
180
- def handle_ws(self):
181
- while True:
182
- try:
183
- data = self.ws.recv()
184
- except (ConnectionClosed, ConnectionClosedError):
185
- break
186
- except Exception as e:
187
- logging.warn(f"({self.id}) Receiving data from websocket failed: {e}")
188
- break
189
-
190
- if not isinstance(data, bytes):
191
- continue #ignore non binary frames
192
-
193
- #implement bandwidth limits
194
- ratelimit.limit_client_bandwidth_sync(self.client_ip, len(data), "ws")
195
-
196
- #get basic packet info
197
- payload = memoryview(data)[5:]
198
- packet_type, stream_id = struct.unpack(packet_format, data[:5])
199
-
200
- if packet_type == 0x01: #CONNECT packet
201
- self.active_streams[stream_id] = {
202
- "conn": None,
203
- "type": None,
204
- "queue": queue.Queue(queue_size),
205
- "packets_sent": 0
206
- }
207
- threading.Thread(target=self.new_stream, args=(stream_id, payload), daemon=True).start()
208
-
209
- elif packet_type == 0x02: #DATA packet
210
- stream = self.active_streams.get(stream_id)
211
- if not stream:
212
- continue
213
- stream["queue"].put(payload)
214
-
215
- elif packet_type == 0x04: #CLOSE packet
216
- reason = struct.unpack(close_format, payload)[0]
217
- self.close_stream(stream_id)
218
-
219
- self.close()
@@ -1,104 +0,0 @@
1
- import threading
2
- import multiprocessing
3
- import pathlib
4
- import mimetypes
5
- import logging
6
- import random
7
- import sys
8
-
9
- from websockets.sync.server import serve
10
- from websockets.http11 import Response, Headers
11
-
12
- import wisp
13
- from wisp.server import ratelimit
14
- from wisp.server import net
15
- from wisp.server.threading import connection
16
-
17
- static_path = None
18
- connections = {}
19
-
20
- def connection_handler(websocket):
21
- path = websocket.request.path
22
- client_ip = websocket.remote_address[0]
23
- if client_ip == "127.0.0.1" and "X-Real-IP" in websocket.request.headers:
24
- client_ip = websocket.request.headers["X-Real-IP"]
25
- origin = websocket.request.headers.get("Origin")
26
-
27
- conn_id = "".join(random.choices("1234567890abcdef", k=8))
28
- logging.info(f"({conn_id}) incoming connection on {path} from {client_ip} (origin: {origin})")
29
- ratelimit.inc_client_attr(client_ip, "streams")
30
-
31
- if path.endswith("/"):
32
- wisp_conn = connection.WispConnection(websocket, path, client_ip, conn_id)
33
- connections[websocket] = wisp_conn
34
- wisp_conn.setup()
35
- wisp_conn.handle_ws()
36
- del connections[websocket]
37
-
38
- else:
39
- stream_count = ratelimit.get_client_attr(client_ip, "streams")
40
- if ratelimit.enabled and stream_count > ratelimit.connections_limit:
41
- return
42
- wsproxy_conn = connection.WSProxyConnection(websocket, path, client_ip)
43
- wsproxy_conn.setup_connection()
44
- t1 = threading.Thread(target=wsproxy_conn.handle_ws, daemon=True)
45
- t2 = threading.Thread(target=wsproxy_conn.handle_tcp, daemon=True)
46
- t1.start(); t2.start()
47
- t1.join(); t2.join()
48
-
49
- def static_handler(connection, request):
50
- if "Upgrade" in request.headers:
51
- return
52
-
53
- response_headers = Headers()
54
- response_headers["Server"] = f"wisp-server-python v{wisp.version}"
55
- if static_path is None:
56
- return Response(204, None, response_headers, body=b"")
57
-
58
- target_path = static_path / request.path[1:]
59
-
60
- if target_path.is_dir():
61
- target_path = target_path / "index.html"
62
- if not target_path.is_relative_to(static_path):
63
- return Response(403, None, response_headers, body="403 forbidden".encode())
64
- if not target_path.exists():
65
- return Response(404, None, response_headers, body="404 not found".encode())
66
-
67
- mimetype = mimetypes.guess_type(target_path.name)[0]
68
- response_headers["Content-Type"] = mimetype
69
-
70
- static_data = target_path.read_bytes()
71
- return Response(200, None, response_headers, body=static_data)
72
-
73
- def main(args):
74
- global static_path
75
- logging.info(f"running wisp-server-python v{wisp.version} (threading)")
76
-
77
- if args.static:
78
- static_path = pathlib.Path(args.static).resolve()
79
- mimetypes.init()
80
- logging.info(f"serving static files from {static_path}")
81
-
82
- if args.limits:
83
- logging.info("enabled rate limits")
84
- ratelimit.enabled = True
85
- ratelimit.connections_limit = int(args.connections)
86
- ratelimit.bandwidth_limit = float(args.bandwidth)
87
- ratelimit.window_size = float(args.window)
88
-
89
- net.block_loopback = not args.allow_loopback
90
- net.block_private = not args.allow_private
91
-
92
- threading.Thread(target=ratelimit.reset_limits_timer_sync, daemon=True).start()
93
- logging.info(f"listening on {args.host}:{args.port}")
94
-
95
- ws_logger = logging.getLogger("websockets")
96
- ws_logger.setLevel(logging.WARN)
97
-
98
- with serve(connection_handler, args.host, int(args.port), process_request=static_handler, compression=None) as server:
99
- try:
100
- server.serve_forever()
101
- except KeyboardInterrupt:
102
- for connection in connections.values():
103
- connection.close()
104
- server.shutdown()
@@ -1,58 +0,0 @@
1
- import socket
2
- import ipaddress
3
-
4
- #various network utilities and wrappers - synchronous version
5
-
6
- from wisp.server import net
7
-
8
- def validate_hostname(host, port, stream_type):
9
- addr_str = net.get_ip(host, port, stream_type)
10
- net.validate_ip(addr_str)
11
- return addr_str
12
-
13
- class TCPConnection:
14
- def __init__(self, hostname, port):
15
- self.hostname = hostname
16
- self.port = port
17
- self.socket = None
18
-
19
- def connect(self):
20
- addr_str = validate_hostname(self.hostname, self.port, 0x01)
21
- self.socket = socket.socket()
22
- self.socket.connect((self.hostname, self.port))
23
-
24
- def recv(self):
25
- return self.socket.recv(net.tcp_size)
26
-
27
- def send(self, data):
28
- self.socket.send(data)
29
-
30
- def close(self):
31
- if self.socket is None:
32
- return
33
- self.socket.shutdown(socket.SHUT_RDWR)
34
- self.socket.close()
35
-
36
- class UDPConnection:
37
- def __init__(self, hostname, port):
38
- self.hostname = hostname
39
- self.port = port
40
- self.socket = None
41
-
42
- def connect(self):
43
- addr_str = validate_hostname(self.hostname, self.port, 0x02)
44
- self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
45
- self.socket.bind((self.hostname, self.port))
46
-
47
- def recv(self):
48
- data, addr = self.socket.recvfrom(net.tcp_size)
49
- return data
50
-
51
- def send(self, data):
52
- self.socket.sendto(data, (self.hostname, self.port))
53
-
54
- def close(self):
55
- if self.socket is None:
56
- return
57
- self.socket.shutdown(socket.SHUT_RDWR)
58
- self.socket.close()
@@ -1,19 +0,0 @@
1
- wisp/__init__.py,sha256=02Fv0s_jJG8U4y5B6AzKPIOzxAuqJyY_So8Guz5bdTk,75
2
- wisp/util.py,sha256=AU7si8VwJqS8Dwc1PT2pu_f_9Y32GQxivRYwJ3-U0YQ,144
3
- wisp/server/__init__.py,sha256=LmgQ1nTlnqcTPqRJVc1NGapV9qVR-BK8iPVUMSJ74Qs,102
4
- wisp/server/__main__.py,sha256=oEDuHOpC7DaHTFIX-9HViwOmWJqdIyktA8whRTmGesk,610
5
- wisp/server/common.py,sha256=0NheE8nJeUmAMPt-DomesnkqCvB8gITi-dsT7VBRJCw,1442
6
- wisp/server/connection.py,sha256=V_Skmmsvn30lQKY5w9x2rJQFR_FsA3lngNDpaYOfX6U,7669
7
- wisp/server/http.py,sha256=YL-N3ajptgcAq5P50V6b8KlSJlVo6fRPnz65El9ZUQY,3145
8
- wisp/server/net.py,sha256=3EL5FzM9BAF2VW6ysTFX1vyJG4xMvrcu_YMlxTYqhI8,2265
9
- wisp/server/ratelimit.py,sha256=cic_rlBaM-tp6kJaSz4ITGvUbsX4JLSGSUgF5cPO0FY,1926
10
- wisp/server/threading/__init__.py,sha256=_34ZRPKommA9uazP9UBw59EByf6m9-VGNqc7hAjaJkc,122
11
- wisp/server/threading/__main__.py,sha256=xM65KttaBDFg3FS5H5msEkx3p-mPfP7grpGslqKG4uY,146
12
- wisp/server/threading/connection.py,sha256=2bMEPBxMl28OAVehNS3sk4C33dECZCHKBZSHTkEzZoM,7044
13
- wisp/server/threading/http.py,sha256=0pzsHVrJoP4QZ9yNPPZYGStah9g3pJZvNMGbzVBRk5w,3521
14
- wisp/server/threading/netsync.py,sha256=7RBlLhEtrF1fsWUIt6aGeoD_lUui2Q83FJdEhi5gnTY,1462
15
- wisp_python-0.6.1.dist-info/licenses/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
16
- wisp_python-0.6.1.dist-info/METADATA,sha256=YE9Ent8HbMHBW95GSgF8irmLlOxRWEDpMcyCCS7r3Fg,4083
17
- wisp_python-0.6.1.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
18
- wisp_python-0.6.1.dist-info/top_level.txt,sha256=sv0LDAu8cHUc3_VTsCKfXU-6B9HdV4dW22ubeW8KPVs,5
19
- wisp_python-0.6.1.dist-info/RECORD,,