aboutsummaryrefslogtreecommitdiffstats
path: root/packages/meshbay-node/src/meshbay_node/transport/webrtc_server.py
diff options
context:
space:
mode:
authorChristophe Besson <cbesson@gmail.com>2026-09-12 09:47:34 +0200
committerChristophe Besson <cbesson@gmail.com>2026-09-12 16:36:54 +0200
commitbf7ff9ec311660318c8562abe03ef4db62475c98 (patch)
tree2fd49c42c59e1207574bcf2842b726e4aec82c90 /packages/meshbay-node/src/meshbay_node/transport/webrtc_server.py
parent4cce50f09a73739387d5058a6f8183ebac65ae2c (diff)
downloadmeshbay-bf7ff9ec311660318c8562abe03ef4db62475c98.tar.gz
fix: bound what one member can cost the others
An availability review, prompted by the group claim above: a participant supplies input — who else bears the cost? Six answers where the cost fell on someone other than the sender, and none of them needs an attacker. AV3 `chat_notify` carried a `group_id` the hub believed, so any connected node could write a notification to every member of any group on the hub, carrying a display string of its choosing, with its account having no relation to that group. This is the group claim again, two hundred lines further down the same socket. Gated on what the node is registered for, and metered: the fan-out is one write per member. The budget expires by time rather than on disconnect, or reconnecting would refill it and a node token is good for an hour. AV4 A swarm source named its own `endpoint` as free text documented as "ip:port", so an account could publish a third party's address — H6's `peer_ip` defect, never applied here. Nothing dials a swarm source today, which is the only reason it was not already a reflection primitive. It is a transport and a port now, never a host, and the number of hashes one account may claim is bounded: rows were keyed (hash, account) with no cap at all. AV5 `handle_webrtc_answer` resolved any pending `peer_id` from any node's socket. The answer is the SDP a browser then connects to. That this had not happened rested on a uuid4 being unguessable. AV6 `relay_register` had no authentication of any kind: it compared `pk_relay` against the approved value, which is a *public* key, so anyone who could read it could rewrite where the hub tells nodes to send relayed traffic. The module docstring promised signed JWTs and `jwt` was imported and never used. AV7 The node held unlimited peer connections and kept one that never completed a handshake for the life of the daemon. H6 bounded what one unauthenticated peer costs; the hub's cap is three offers in flight per *account*, a limit on each caller and not on the machine, so an operator's exposure grew with the size of their groups. AV8 `invite-notify` put a request-supplied `group_name` into the subject of an email the hub sends under its own domain, to any account, with no rate limit. The name comes from the group row now. The tests are two accounts each, in one file that says why: a one-member test proves a one-member property, and every finding here needed a second person to exist at all. Each was checked against the unfixed code. Two did not survive that check and were rewritten — one re-enacted the disconnect path instead of running it (hence `forget_node`), the other called the reaper itself and would have passed with the call removed from `handle_offer`. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01T4YmK41VsEURWFdop4EEeT
Diffstat (limited to 'packages/meshbay-node/src/meshbay_node/transport/webrtc_server.py')
-rw-r--r--packages/meshbay-node/src/meshbay_node/transport/webrtc_server.py60
1 files changed, 60 insertions, 0 deletions
diff --git a/packages/meshbay-node/src/meshbay_node/transport/webrtc_server.py b/packages/meshbay-node/src/meshbay_node/transport/webrtc_server.py
index a0776d2..29d6e7f 100644
--- a/packages/meshbay-node/src/meshbay_node/transport/webrtc_server.py
+++ b/packages/meshbay-node/src/meshbay_node/transport/webrtc_server.py
@@ -192,6 +192,18 @@ MAX_UPLOAD_BYTES = 4 * 1024 * 1024 * 1024 # 4 GB per file
# Budget for an unauthenticated peer: enough for a handshake and a bundle fetch,
# nowhere near enough to be a memory-exhaustion primitive (H6).
PRE_HANDSHAKE_MAX_MSG = 64 * 1024
+
+# How many peer connections this node holds at once, and how long one may stay
+# without completing the MNP handshake. The budget above bounds what *one*
+# unauthenticated peer costs; these bound how many there may be and how long
+# each lasts, which is the other half and was missing. The hub caps three
+# offers in flight per account — a limit on each caller, not on this machine —
+# so the cost to an operator grew with the number of people in their groups.
+# Sized to be unreachable in ordinary use: a browser holds one connection per
+# open group, and a handshake unfinished after a minute is not going to finish.
+MAX_PEER_SESSIONS = 64
+UNAUTHENTICATED_SESSION_TIMEOUT = 60 # seconds
+
# ffmpeg is spawned per stream request; without a cap any member can fork-bomb
# the node by requesting many streams at once (H6).
#
@@ -5882,6 +5894,7 @@ class WebRTCTransport:
from meshbay_node.config import DEFAULT_STUN_SERVERS
self._stun = stun_servers or list(DEFAULT_STUN_SERVERS)
self._sessions: dict[str, WebRTCPeerSession] = {}
+ self._reapers: set[asyncio.Task] = set()
def set_capacity(self, *, max_concurrent_streams: int | None = None,
max_concurrent_downloads: int | None = None,
@@ -5991,9 +6004,21 @@ class WebRTCTransport:
config = RTCConfiguration(
iceServers=[RTCIceServer(urls=s) for s in self._stun] if self._stun else []
)
+ # Before anything is allocated. Every offer costs an RTCPeerConnection
+ # with its own DTLS and SCTP stacks, and nothing here used to bound how
+ # many a node would hold: the hub caps three in flight *per account*,
+ # which is a limit on each caller and not on this machine, so the cost
+ # grew with the number of members in the group. An operator's node must
+ # not be exhaustible by the people they invited.
+ if len(self._sessions) >= MAX_PEER_SESSIONS:
+ log.warning("Refusing WebRTC offer: %d peer sessions already open",
+ len(self._sessions))
+ raise RuntimeError("Node is at its peer-connection limit")
+
pc = RTCPeerConnection(configuration=config)
session = WebRTCPeerSession(pc, self._ctx, peer_id=peer_id)
self._sessions[peer_id] = session
+ self._reap_if_unauthenticated(peer_id)
@pc.on("datachannel")
def on_datachannel(channel: RTCDataChannel):
@@ -6048,12 +6073,47 @@ class WebRTCTransport:
", ".join(sorted(host_addrs)) or "none", srflx)
return answer_sdp, []
+ def _reap_if_unauthenticated(self, peer_id: str) -> None:
+ """Close a session that never completes the handshake.
+
+ A peer that connects and then says nothing is indistinguishable from a
+ working one until it is asked to prove something, and it was never
+ asked: `connectionstatechange` reaps a connection that *fails*, and one
+ that succeeds and stays silent was held for the node's lifetime. That
+ is the cheapest way to spend someone else's memory — no GEK, no token,
+ no group, just an open connection. `_user_id` is set by the GEK proof
+ (`_do_handshake_response`), so it is the one honest test of whether
+ this peer ever became anybody.
+ """
+ async def reap() -> None:
+ try:
+ await asyncio.sleep(UNAUTHENTICATED_SESSION_TIMEOUT)
+ session = self._sessions.get(peer_id)
+ if session is not None and not session._user_id:
+ log.warning("Closing peer %s: no handshake within %ds",
+ peer_id[:8], UNAUTHENTICATED_SESSION_TIMEOUT)
+ await self.close_peer(peer_id)
+ except asyncio.CancelledError:
+ raise
+ except Exception as e:
+ log.warning("Reaping peer %s failed: %s", peer_id[:8], e)
+
+ # Held in a set for the same reason every other task here is: asyncio
+ # keeps only a weak reference, and a reaper collected mid-sleep reaps
+ # nothing (see WebRTCPeerSession.__init__).
+ task = asyncio.ensure_future(reap())
+ self._reapers.add(task)
+ task.add_done_callback(self._reapers.discard)
+
async def close_peer(self, peer_id: str) -> None:
session = self._sessions.pop(peer_id, None)
if session:
await session.close()
async def close_all(self) -> None:
+ for task in list(self._reapers):
+ task.cancel()
+ self._reapers.clear()
for session in list(self._sessions.values()):
await session.close()
self._sessions.clear()