aboutsummaryrefslogtreecommitdiffstats
path: root/packages/meshbay-node/tests/test_quic_transport.py
diff options
context:
space:
mode:
authorChristophe Besson <cbesson@gmail.com>2026-08-10 03:07:56 +0200
committerChristophe Besson <cbesson@gmail.com>2026-08-10 03:07:56 +0200
commit4b3e8c3b8b9d10c8ac333dd8db614a7569052472 (patch)
tree8828d7d549adc8bb21d8a7f6533c8e9db9c298b9 /packages/meshbay-node/tests/test_quic_transport.py
parent8ccbe262ecf4a8f7545cbf1e9f1cc5a485acae67 (diff)
downloadmeshbay-4b3e8c3b8b9d10c8ac333dd8db614a7569052472.tar.gz
feat: Phase 7 — Node v2 (multi-group, Sender Keys, 0-RTT, chat, denylist)
Implements all 8 milestones (7.0-7.7): - 7.0: JWT carries `groups` claim; node verifies group membership at MNP handshake (QUIC + TCP+TLS). Resolves security review C2. - 7.1: QUIC 0-RTT session resumption via stored session tickets (17-21ms reconnect vs 47ms cold). - 7.2: Hub→node WebSocket signaling for NAT punch coordination (`client_incoming`/`punch_ready`) + jti denylist push. Denylist class blocks revoked users/jtis at handshake. - 7.3: Multi-group daemon — one QUIC port serves N groups with per-group GEK, shared_root, and index routing. - 7.4: HLS streaming via QUIC (STREAM_SEGMENT message type, ffmpeg segment extraction). - 7.5: Sender Keys protocol for group chat (Signal Groups approach). Each member has own sending chain key, HKDF chain ratchet, AES-256-GCM encryption, Ed25519 signing. Resolves security review C1. - 7.6: Chat store (SQLite via aiosqlite), CHAT_MESSAGE MNP wire type with peer broadcast, web UI with WebSocket push. - 7.7: Argon2id calibration CLI. First security review included (first-review.md). 109 tests, demo-v3 validated against meshbay.org production hub. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Diffstat (limited to 'packages/meshbay-node/tests/test_quic_transport.py')
-rw-r--r--packages/meshbay-node/tests/test_quic_transport.py134
1 files changed, 132 insertions, 2 deletions
diff --git a/packages/meshbay-node/tests/test_quic_transport.py b/packages/meshbay-node/tests/test_quic_transport.py
index 2abd465..0c1a1cd 100644
--- a/packages/meshbay-node/tests/test_quic_transport.py
+++ b/packages/meshbay-node/tests/test_quic_transport.py
@@ -14,7 +14,7 @@ from cryptography.hazmat.primitives import serialization
from meshbay_common.crypto import generate_gek, pk_to_b64
from meshbay_node.indexer import DirectoryIndexer, GroupIndex
-from meshbay_node.transport.quic_server import QuicChunkServer
+from meshbay_node.transport.quic_server import QuicChunkServer, Denylist
from meshbay_node.transport.quic_client import QuicChunkClient
@@ -38,7 +38,7 @@ def shared_dir(tmp_path):
(d / "small.txt").write_bytes(b"hello quic " * 100)
return d
-def make_jwt(sk_hub, pk_node_b64, ttl=3600):
+def make_jwt(sk_hub, pk_node_b64, ttl=3600, groups=None):
sk_pem = sk_hub.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.PKCS8,
@@ -49,6 +49,7 @@ def make_jwt(sk_hub, pk_node_b64, ttl=3600):
"iss": "test-hub", "sub": "user-001",
"pk_user": pk_node_b64, "hub_id": "test-hub",
"jti": "test-jti", "iat": now, "exp": now + ttl,
+ "groups": groups or [],
}, sk_pem, algorithm="EdDSA")
@@ -155,3 +156,132 @@ async def test_quic_invalid_jwt_rejected(sk_node, sk_hub, gek, shared_dir, tmp_p
await client.fetch_index()
await server.stop()
+
+
+@pytest.mark.asyncio
+async def test_quic_wrong_group_rejected(sk_node, sk_hub, gek, shared_dir, tmp_path):
+ """QUIC server rejects a client whose JWT groups don't include the requested group_id."""
+ hub_pk_pem = sk_hub.public_key().public_bytes(
+ serialization.Encoding.PEM, serialization.PublicFormat.SubjectPublicKeyInfo)
+
+ indexer = DirectoryIndexer(root=shared_dir, group_id="g", sk_node=sk_node, gek=gek)
+ await indexer.initial_scan()
+
+ cert_path = tmp_path / "node.crt"
+ key_path = tmp_path / "node.key"
+
+ server = QuicChunkServer(
+ sk_node=sk_node, hub_pk_pem=hub_pk_pem, gek=gek,
+ shared_root=shared_dir, index=indexer.index,
+ host="127.0.0.1", port=19103,
+ cert_path=cert_path, key_path=key_path,
+ )
+ await server.start()
+
+ token = make_jwt(sk_hub, pk_to_b64(sk_node.public_key()), groups=["group-a"])
+
+ with pytest.raises(ConnectionError, match="rejected"):
+ async with QuicChunkClient(
+ host="127.0.0.1", port=19103,
+ jwt_token=token, gek=gek,
+ pk_node_b64=pk_to_b64(sk_node.public_key()),
+ group_id="group-b",
+ ) as client:
+ await client.fetch_index()
+
+ await server.stop()
+
+
+@pytest.mark.asyncio
+async def test_quic_session_resumption(sk_node, sk_hub, gek, shared_dir, tmp_path):
+ """QUIC 0-RTT: connect, save session ticket, reconnect with ticket."""
+ hub_pk_pem = sk_hub.public_key().public_bytes(
+ serialization.Encoding.PEM, serialization.PublicFormat.SubjectPublicKeyInfo)
+
+ indexer = DirectoryIndexer(root=shared_dir, group_id="g", sk_node=sk_node, gek=gek)
+ await indexer.initial_scan()
+
+ cert_path = tmp_path / "node.crt"
+ key_path = tmp_path / "node.key"
+
+ server = QuicChunkServer(
+ sk_node=sk_node, hub_pk_pem=hub_pk_pem, gek=gek,
+ shared_root=shared_dir, index=indexer.index,
+ host="127.0.0.1", port=19104,
+ cert_path=cert_path, key_path=key_path,
+ )
+ await server.start()
+
+ token = make_jwt(sk_hub, pk_to_b64(sk_node.public_key()))
+ pk_b64 = pk_to_b64(sk_node.public_key())
+
+ # First connection — captures session ticket
+ saved_ticket = None
+ async with QuicChunkClient(
+ host="127.0.0.1", port=19104,
+ jwt_token=token, gek=gek, pk_node_b64=pk_b64,
+ ) as client:
+ wire = await client.fetch_index()
+ assert GroupIndex.deserialize(wire, sk_node=sk_node, gek=gek).count == 2
+ saved_ticket = client.session_ticket
+
+ # Allow server to process the close
+ await asyncio.sleep(0.1)
+
+ # Second connection — reuses session ticket (0-RTT)
+ async with QuicChunkClient(
+ host="127.0.0.1", port=19104,
+ jwt_token=token, gek=gek, pk_node_b64=pk_b64,
+ session_ticket=saved_ticket,
+ ) as client:
+ wire = await client.fetch_index()
+ assert GroupIndex.deserialize(wire, sk_node=sk_node, gek=gek).count == 2
+
+ await server.stop()
+
+
+@pytest.mark.asyncio
+async def test_quic_denylist_blocks_user(sk_node, sk_hub, gek, shared_dir, tmp_path):
+ """QUIC server rejects a connection when the user is on the denylist."""
+ hub_pk_pem = sk_hub.public_key().public_bytes(
+ serialization.Encoding.PEM, serialization.PublicFormat.SubjectPublicKeyInfo)
+
+ indexer = DirectoryIndexer(root=shared_dir, group_id="g", sk_node=sk_node, gek=gek)
+ await indexer.initial_scan()
+
+ cert_path = tmp_path / "node.crt"
+ key_path = tmp_path / "node.key"
+
+ denylist = Denylist()
+ server = QuicChunkServer(
+ sk_node=sk_node, hub_pk_pem=hub_pk_pem, gek=gek,
+ shared_root=shared_dir, index=indexer.index,
+ host="127.0.0.1", port=19105,
+ cert_path=cert_path, key_path=key_path,
+ denylist=denylist,
+ )
+ await server.start()
+
+ token = make_jwt(sk_hub, pk_to_b64(sk_node.public_key()))
+ pk_b64 = pk_to_b64(sk_node.public_key())
+
+ # Connection works before denylisting
+ async with QuicChunkClient(
+ host="127.0.0.1", port=19105,
+ jwt_token=token, gek=gek, pk_node_b64=pk_b64,
+ ) as client:
+ wire = await client.fetch_index()
+ assert GroupIndex.deserialize(wire, sk_node=sk_node, gek=gek).count == 2
+
+ # Add user to denylist
+ denylist.deny_user("user-001")
+
+ # Connection now rejected
+ with pytest.raises(Exception):
+ async with QuicChunkClient(
+ host="127.0.0.1", port=19105,
+ jwt_token=token, gek=gek, pk_node_b64=pk_b64,
+ ) as client:
+ await client.fetch_index()
+
+ await server.stop()