tigrcorn-protocols 0.3.18.dev2__tar.gz → 0.3.18.dev3__tar.gz

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tigrcorn-protocols
-Version: 0.3.18.dev2
+Version: 0.3.18.dev3
 Summary: Protocol handlers for Tigrcorn HTTP/1.1, HTTP/2, HTTP/3, QUIC, WebSocket, WebTransport, lifespan, and ASGI3 traffic.
 Author-email: Jacob Stewart <jacob@swarmauri.com>
 License: Apache License
@@ -187,11 +187,11 @@ Classifier: Typing :: Typed
 Requires-Python: <3.15,>=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: tigrcorn-core==0.3.18.dev2
-Requires-Dist: tigrcorn-config==0.3.18.dev2
-Requires-Dist: tigrcorn-asgi==0.3.18.dev2
-Requires-Dist: tigrcorn-http==0.3.18.dev2
-Requires-Dist: tigrcorn-transports==0.3.18.dev2
+Requires-Dist: tigrcorn-core==0.3.18.dev3
+Requires-Dist: tigrcorn-config==0.3.18.dev3
+Requires-Dist: tigrcorn-asgi==0.3.18.dev3
+Requires-Dist: tigrcorn-http==0.3.18.dev3
+Requires-Dist: tigrcorn-transports==0.3.18.dev3
 Dynamic: license-file
 
 <div align="center">
@@ -284,7 +284,7 @@ The package exposes its supported public surface through the <code>tigrcorn_prot
 
 ## Package Graph
 
-[tigrcorn-core](https://pypi.org/project/tigrcorn-core/) | [tigrcorn-config](https://pypi.org/project/tigrcorn-config/) | [tigrcorn-http](https://pypi.org/project/tigrcorn-http/) | [tigrcorn-asgi](https://pypi.org/project/tigrcorn-asgi/) | [tigrcorn-contract](https://pypi.org/project/tigrcorn-contract/) | [tigrcorn-transports](https://pypi.org/project/tigrcorn-transports/) | [tigrcorn-security](https://pypi.org/project/tigrcorn-security/) | [tigrcorn-protocols](https://pypi.org/project/tigrcorn-protocols/) | [tigrcorn-static](https://pypi.org/project/tigrcorn-static/) | [tigrcorn-observability](https://pypi.org/project/tigrcorn-observability/) | [tigrcorn-runtime](https://pypi.org/project/tigrcorn-runtime/) | [tigrcorn-compat](https://pypi.org/project/tigrcorn-compat/) | [tigrcorn-certification](https://pypi.org/project/tigrcorn-certification/)
+[tigrcorn-core](https://pypi.org/project/tigrcorn-core/) | [tigrcorn-quic-cc](https://pypi.org/project/tigrcorn-quic-cc/) | [tigrcorn-quic-cc-reno](https://pypi.org/project/tigrcorn-quic-cc-reno/) | [tigrcorn-config](https://pypi.org/project/tigrcorn-config/) | [tigrcorn-http](https://pypi.org/project/tigrcorn-http/) | [tigrcorn-asgi](https://pypi.org/project/tigrcorn-asgi/) | [tigrcorn-contract](https://pypi.org/project/tigrcorn-contract/) | [tigrcorn-transports](https://pypi.org/project/tigrcorn-transports/) | [tigrcorn-security](https://pypi.org/project/tigrcorn-security/) | [tigrcorn-protocols](https://pypi.org/project/tigrcorn-protocols/) | [tigrcorn-static](https://pypi.org/project/tigrcorn-static/) | [tigrcorn-observability](https://pypi.org/project/tigrcorn-observability/) | [tigrcorn-runtime](https://pypi.org/project/tigrcorn-runtime/) | [tigrcorn-compat](https://pypi.org/project/tigrcorn-compat/) | [tigrcorn-certification](https://pypi.org/project/tigrcorn-certification/)
 
 ## Best Practices
 

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/README.md

@@ -88,7 +88,7 @@ The package exposes its supported public surface through the <code>tigrcorn_prot
 
 ## Package Graph
 
-[tigrcorn-core](https://pypi.org/project/tigrcorn-core/) | [tigrcorn-config](https://pypi.org/project/tigrcorn-config/) | [tigrcorn-http](https://pypi.org/project/tigrcorn-http/) | [tigrcorn-asgi](https://pypi.org/project/tigrcorn-asgi/) | [tigrcorn-contract](https://pypi.org/project/tigrcorn-contract/) | [tigrcorn-transports](https://pypi.org/project/tigrcorn-transports/) | [tigrcorn-security](https://pypi.org/project/tigrcorn-security/) | [tigrcorn-protocols](https://pypi.org/project/tigrcorn-protocols/) | [tigrcorn-static](https://pypi.org/project/tigrcorn-static/) | [tigrcorn-observability](https://pypi.org/project/tigrcorn-observability/) | [tigrcorn-runtime](https://pypi.org/project/tigrcorn-runtime/) | [tigrcorn-compat](https://pypi.org/project/tigrcorn-compat/) | [tigrcorn-certification](https://pypi.org/project/tigrcorn-certification/)
+[tigrcorn-core](https://pypi.org/project/tigrcorn-core/) | [tigrcorn-quic-cc](https://pypi.org/project/tigrcorn-quic-cc/) | [tigrcorn-quic-cc-reno](https://pypi.org/project/tigrcorn-quic-cc-reno/) | [tigrcorn-config](https://pypi.org/project/tigrcorn-config/) | [tigrcorn-http](https://pypi.org/project/tigrcorn-http/) | [tigrcorn-asgi](https://pypi.org/project/tigrcorn-asgi/) | [tigrcorn-contract](https://pypi.org/project/tigrcorn-contract/) | [tigrcorn-transports](https://pypi.org/project/tigrcorn-transports/) | [tigrcorn-security](https://pypi.org/project/tigrcorn-security/) | [tigrcorn-protocols](https://pypi.org/project/tigrcorn-protocols/) | [tigrcorn-static](https://pypi.org/project/tigrcorn-static/) | [tigrcorn-observability](https://pypi.org/project/tigrcorn-observability/) | [tigrcorn-runtime](https://pypi.org/project/tigrcorn-runtime/) | [tigrcorn-compat](https://pypi.org/project/tigrcorn-compat/) | [tigrcorn-certification](https://pypi.org/project/tigrcorn-certification/)
 
 ## Best Practices
 

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tigrcorn-protocols"
-version = "0.3.18.dev2"
+version = "0.3.18.dev3"
 description = "Protocol handlers for Tigrcorn HTTP/1.1, HTTP/2, HTTP/3, QUIC, WebSocket, WebTransport, lifespan, and ASGI3 traffic."
 readme = "README.md"
 requires-python = ">=3.10,<3.15"
@@ -30,11 +30,11 @@ classifiers = [
   "Typing :: Typed",
 ]
 dependencies = [
-  "tigrcorn-core==0.3.18.dev2",
-  "tigrcorn-config==0.3.18.dev2",
-  "tigrcorn-asgi==0.3.18.dev2",
-  "tigrcorn-http==0.3.18.dev2",
-  "tigrcorn-transports==0.3.18.dev2",
+  "tigrcorn-core==0.3.18.dev3",
+  "tigrcorn-config==0.3.18.dev3",
+  "tigrcorn-asgi==0.3.18.dev3",
+  "tigrcorn-http==0.3.18.dev3",
+  "tigrcorn-transports==0.3.18.dev3",
 ]
 
 [tool.setuptools]

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http2/handler/connect.py

@@ -67,6 +67,7 @@ class _HTTP2ConnectTunnel:
         if self.work_lease is not None:
             self.work_lease.release()
         await close_tcp_writer(self.upstream_writer)
+        self.handler._close_inventory_session(self.stream_id, reason="connect-complete")
         self.handler._finalize_stream_if_complete(self.stream_id)
 
     async def _relay_upstream_to_client(self) -> None:
@@ -108,6 +109,7 @@ class HTTP2ConnectMixin:
                 await self._write_raw(serialize_rst_stream(stream_id, H2_CONNECT_ERROR))
             state.mark_reset_sent()
         self._cancel_stream(stream_id)
+        self._close_inventory_session(stream_id, reason="connect-reset")
         self.streams.close(stream_id)
         self._maybe_finish_after_goaway()
 
@@ -163,11 +165,17 @@ class HTTP2ConnectMixin:
             work_lease=self.stream_work_leases.get(stream_id),
         )
         state.connect_tunnel = tunnel
+        self._open_inventory_session(
+            stream_id,
+            kind="connect-tunnel",
+            metadata={"authority": request.target},
+        )
         self.state.last_stream_id = max(self.state.last_stream_id, stream_id)
         try:
             await tunnel.start()
         except Exception:
             state.connect_tunnel = None
+            self._close_inventory_session(stream_id, reason="connect-start-failed")
             await close_tcp_writer(upstream_writer)
             raise
         if state.end_stream_received:

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http2/handler/core.py

@@ -5,6 +5,7 @@ from .connect import HTTP2ConnectMixin
 from .flow_control import HTTP2FlowControlMixin
 from .frames import HTTP2FrameDispatchMixin
 from .headers import HTTP2HeadersMixin
+from .inventory import HTTP2InventoryMixin
 from .io import HTTP2IOMixin
 from .requests import HTTP2RequestsMixin
 from .responses import HTTP2ResponsesMixin
@@ -22,6 +23,7 @@ class HTTP2ConnectionHandler(
     HTTP2ResponsesMixin,
     HTTP2ConnectMixin,
     HTTP2WebSocketSupportMixin,
+    HTTP2InventoryMixin,
 ):
     def __init__(
         self,
@@ -38,6 +40,8 @@ class HTTP2ConnectionHandler(
         scheme: str,
         prebuffer: bytes = b"",
         scope_extensions: dict | None = None,
+        connection_id: str | None = None,
+        connection_inventory: RuntimeConnectionInventory | None = None,
     ) -> None:
         self.app = app
         self.config = config
@@ -51,6 +55,8 @@ class HTTP2ConnectionHandler(
         self.scheme = scheme
         self.prebuffer = prebuffer
         self.scope_extensions = dict(scope_extensions or {})
+        self.connection_id = connection_id
+        self.connection_inventory = connection_inventory
         self.state = H2ConnectionState()
         self.state.local_settings[SETTING_MAX_CONCURRENT_STREAMS] = self.config.http.http2_max_concurrent_streams
         self.state.local_settings[SETTING_MAX_HEADER_LIST_SIZE] = self.config.http.http2_max_headers_size

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http2/handler/frames.py

@@ -116,6 +116,7 @@ class HTTP2FrameDispatchMixin:
             await state.connect_tunnel.abort()
         self._cancel_stream(frame.stream_id)
         state.mark_reset_received()
+        self._close_inventory_session(frame.stream_id, reason="rst-stream")
         self.streams.close(frame.stream_id)
         self._notify_waiter(frame.stream_id)
         self._maybe_finish_after_goaway()

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http2/handler/imports.py

@@ -63,6 +63,7 @@ from tigrcorn_protocols.scheduler.runtime import ProductionScheduler, WorkLease
 from tigrcorn_protocols.http2.streams import H2StreamRegistry
 from tigrcorn_protocols.connect import close_tcp_writer, half_close_tcp_writer, is_connect_allowed, parse_connect_authority
 from tigrcorn_protocols.http2.websocket import H2WebSocketSession
+from tigrcorn_protocols.sessions import RuntimeConnectionInventory
 from tigrcorn_core.types import ASGIApp
 from tigrcorn_core.utils.authority import authority_allowed
 from tigrcorn_core.utils.headers import apply_response_header_policy, sanitize_early_hints_headers, strip_connection_specific_headers

tigrcorn_protocols-0.3.18.dev3/src/tigrcorn_protocols/http2/handler/inventory.py

@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+from .imports import *
+
+
+class HTTP2InventoryMixin:
+    def _inventory_session_id(self, stream_id: int) -> str | None:
+        if self.connection_inventory is None or self.connection_id is None:
+            return None
+        return f"h2:{self.connection_id}:{stream_id}"
+
+    def _open_inventory_session(
+        self,
+        stream_id: int,
+        *,
+        kind: str,
+        metadata: dict | None = None,
+    ) -> str | None:
+        session_id = self._inventory_session_id(stream_id)
+        if session_id is None:
+            return None
+        payload = {"protocol": "http2", "stream_id": stream_id}
+        payload.update(metadata or {})
+        self.connection_inventory.open_session(
+            session_id,
+            connection_id=self.connection_id,
+            kind=kind,
+            stream_ids=(str(stream_id),),
+            metadata=payload,
+        )
+        self.connection_inventory.increment_connection_counter(self.connection_id, "streams")
+        if kind == "http-request":
+            self.connection_inventory.increment_connection_counter(self.connection_id, "requests")
+        elif kind == "websocket":
+            self.connection_inventory.increment_connection_counter(self.connection_id, "websockets")
+        elif kind == "connect-tunnel":
+            self.connection_inventory.increment_connection_counter(self.connection_id, "connect_tunnels")
+        return session_id
+
+    def _update_inventory_session(
+        self,
+        stream_id: int,
+        *,
+        state: str | None = None,
+        metadata: dict | None = None,
+    ) -> None:
+        session_id = self._inventory_session_id(stream_id)
+        if session_id is None:
+            return
+        self.connection_inventory.update_session(session_id, state=state, metadata=metadata)
+
+    def _close_inventory_session(self, stream_id: int, *, reason: str | None = None) -> None:
+        session_id = self._inventory_session_id(stream_id)
+        if session_id is None:
+            return
+        self.connection_inventory.close_session(session_id, reason=reason)

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http2/handler/streams.py

@@ -51,6 +51,12 @@ class HTTP2StreamsMixin:
             await self._start_connect_tunnel(stream_id)
             return
         self.state.last_stream_id = max(self.state.last_stream_id, stream_id)
+        request = self._build_request(state)
+        self._open_inventory_session(
+            stream_id,
+            kind="http-request",
+            metadata={"method": request.method, "path": request.path},
+        )
         task = asyncio.create_task(self._run_stream(stream_id), name=f"tigrcorn-h2-stream-{stream_id}")
         self.stream_tasks[stream_id] = task
 
@@ -66,6 +72,7 @@ class HTTP2StreamsMixin:
         if state is not None:
             state.websocket_session = None
         self._release_stream_work_lease(stream_id)
+        self._close_inventory_session(stream_id, reason="websocket-complete")
         self._finalize_stream_if_complete(stream_id)
 
 
@@ -87,6 +94,7 @@ class HTTP2StreamsMixin:
         if state is None or state.websocket_session is not None or state.connect_tunnel is not None:
             return
         if state.local_closed and state.end_stream_received:
+            self._close_inventory_session(stream_id, reason="stream-complete")
             self._release_stream_work_lease(stream_id)
             self._cancel_stream(stream_id)
             self.streams.close(stream_id)
@@ -125,6 +133,7 @@ class HTTP2StreamsMixin:
                 self._finalize_stream_if_complete(stream_id)
             self._maybe_finish_after_goaway()
         finally:
+            self._close_inventory_session(stream_id, reason="request-complete")
             self._release_stream_work_lease(stream_id)
 
 
@@ -141,14 +150,17 @@ class HTTP2StreamsMixin:
             if state.websocket_session is not None:
                 with suppress(Exception):
                     await state.websocket_session.abort()
+                self._close_inventory_session(state.stream_id, reason="shutdown")
             if state.connect_tunnel is not None:
                 with suppress(Exception):
                     await state.connect_tunnel.abort()
+                self._close_inventory_session(state.stream_id, reason="shutdown")
         for stream_id, task in list(self.stream_tasks.items()):
             task.cancel()
             with suppress(asyncio.CancelledError):
                 await task
             self.stream_tasks.pop(stream_id, None)
+            self._close_inventory_session(stream_id, reason="shutdown")
         if not self.state.local_goaway_sent:
             self.state.local_goaway_sent = True
             self.state.local_goaway_last_stream_id = self.state.last_stream_id

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http2/handler/websocket_support.py

@@ -41,6 +41,15 @@ class HTTP2WebSocketSupportMixin:
             on_close=lambda stream_id=stream_id: self._on_websocket_stream_closed(stream_id),
         )
         state.websocket_session = session
+        self._open_inventory_session(
+            stream_id,
+            kind="websocket",
+            metadata={"path": request.path},
+        )
         self.state.last_stream_id = max(self.state.last_stream_id, stream_id)
-        await session.start()
+        try:
+            await session.start()
+        except Exception:
+            self._close_inventory_session(stream_id, reason="websocket-start-failed")
+            raise
 

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http3/codec.py

@@ -25,6 +25,7 @@ SETTING_ENABLE_CONNECT_PROTOCOL = 0x08
 SETTING_H3_DATAGRAM = 0x33
 SETTING_ENABLE_WEBTRANSPORT = 0x2B603742
 SETTING_WT_MAX_SESSIONS = 0x14E9CD29
+SETTING_WT_ENABLED = 0x2C7CF000
 
 H3_DATAGRAM_ERROR = 0x33
 H3_NO_ERROR = 0x0100
@@ -69,6 +70,7 @@ HTTP3_IANA_SETTINGS = {
     "SETTINGS_ENABLE_CONNECT_PROTOCOL": SETTING_ENABLE_CONNECT_PROTOCOL,
     "SETTINGS_H3_DATAGRAM": SETTING_H3_DATAGRAM,
     "SETTINGS_WT_MAX_SESSIONS": SETTING_WT_MAX_SESSIONS,
+    "SETTINGS_WT_ENABLED": SETTING_WT_ENABLED,
 }
 
 HTTP3_IANA_ERROR_CODES = {

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http3/handler/connect.py

@@ -70,6 +70,11 @@ class _HTTP3ConnectTunnel:
             with suppress(asyncio.CancelledError):
                 await self.relay_task
         self.session.connect_tunnels.pop(self.stream_id, None)
+        if self.handler.connection_inventory is not None:
+            self.handler.connection_inventory.close_session(
+                f"{self.handler._connection_id_for_session(self.session)}:connect:{self.stream_id}",
+                reason='connect-abort',
+            )
         lease = self.session.stream_work_leases.pop(self.stream_id, None)
         if lease is not None:
             lease.release()
@@ -128,7 +133,7 @@ class HTTP3ConnectMixin:
         already_locked: bool = False,
     ) -> None:
         if not already_locked:
-            async with self._lock:
+            async with session.lock:
                 await self._send_http3_tunnel_data(
                     session,
                     stream_id,
@@ -154,7 +159,7 @@ class HTTP3ConnectMixin:
         already_locked: bool = False,
     ) -> None:
         if not already_locked:
-            async with self._lock:
+            async with session.lock:
                 await self._reset_http3_tunnel_stream(
                     session,
                     stream_id,
@@ -252,6 +257,14 @@ class HTTP3ConnectMixin:
             work_lease=session.stream_work_leases.get(stream_id),
         )
         session.connect_tunnels[stream_id] = tunnel
+        if self.connection_inventory is not None:
+            self.connection_inventory.open_session(
+                f"{self._connection_id_for_session(session)}:connect:{stream_id}",
+                connection_id=self._connection_id_for_session(session),
+                kind='connect-tunnel',
+                stream_ids=(str(stream_id),),
+                metadata={'authority': authority, 'protocol': 'http3'},
+            )
         tunnel.start()
         self.access_logger.log_http(session.addr, 'CONNECT', authority, 200, 'HTTP/3')
         return self._build_http3_response_datagrams_locked(session, stream_id, 200, [], b'', end_stream=False)

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http3/handler/core.py

@@ -6,16 +6,19 @@ from .custom import HTTP3CustomQuicMixin
 from .lifecycle import HTTP3LifecycleMixin
 from .outbound import HTTP3OutboundMixin
 from .packet import HTTP3PacketMixin
+from .priority_lock import PriorityLock
 from .requests import HTTP3RequestsMixin
 from .responses import HTTP3ResponsesMixin
 from .websocket_runtime import HTTP3WebSocketRuntimeMixin
 from .webtransport_datagrams import HTTP3WebTransportDatagramsMixin
+from .webtransport_stream_flow import HTTP3WebTransportStreamFlowMixin
 from .webtransport_streams import HTTP3WebTransportStreamsMixin
 from .webtransport_support import HTTP3WebTransportSupportMixin
 
 
 class HTTP3DatagramHandler(
     HTTP3PacketMixin,
+    HTTP3WebTransportStreamFlowMixin,
     HTTP3WebTransportStreamsMixin,
     HTTP3WebTransportDatagramsMixin,
     HTTP3WebTransportSupportMixin,
@@ -41,6 +44,9 @@ class HTTP3DatagramHandler(
         scheduler: ProductionScheduler | None = None,
         metrics: Metrics | None = None,
         webtransport_governance: Any | None = None,
+        connection_inventory: RuntimeConnectionInventory | None = None,
+        congestion_controller_factory: Any | None = None,
+        congestion_controller_options: Any | None = None,
     ) -> None:
         self.app = app
         self.config = config
@@ -49,10 +55,13 @@ class HTTP3DatagramHandler(
         self.scheduler = scheduler
         self.metrics = metrics
         self.webtransport_governance = webtransport_governance
+        self.connection_inventory = connection_inventory
+        self.congestion_controller_factory = congestion_controller_factory
+        self.congestion_controller_options = dict(congestion_controller_options or {})
         self.sessions: dict[tuple[str, int], HTTP3Session] = {}
         self.sessions_by_local_cid: dict[bytes, HTTP3Session] = {}
         self._session_sequence = 0
-        self._lock = asyncio.Lock()
+        self._lock = PriorityLock()
         self.webtransport_trace: list[dict[str, object]] = []
 
 

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http3/handler/custom.py

@@ -47,7 +47,7 @@ class _CustomQuicSend:
             raise RuntimeError(f'unexpected custom quic send event: {typ!r}')
         data = bytes(message.get('data', b''))
         fin = not bool(message.get('more_data', False))
-        self.messages.append(self.session.quic.send_stream_data(self.stream_id, data, fin=fin))
+        self.messages.extend(self.session.quic.send_stream_data_packets(self.stream_id, data, fin=fin))
 
     def flush(self) -> list[bytes]:
         return list(self.messages)

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http3/handler/imports.py

@@ -1,6 +1,8 @@
 from __future__ import annotations
 
 import asyncio
+import logging
+import time
 from contextlib import suppress
 from dataclasses import dataclass, field
 from typing import Any
@@ -31,6 +33,7 @@ from tigrcorn_protocols.http3.codec import (
     SETTING_MAX_FIELD_SECTION_SIZE,
     SETTING_QPACK_MAX_TABLE_CAPACITY,
     SETTING_WT_MAX_SESSIONS,
+    SETTING_WT_ENABLED,
     HTTP3ConnectionError,
     HTTP3StreamError,
     encode_frame,
@@ -42,7 +45,10 @@ from tigrcorn_protocols.http3.streams import (
     HTTP3ConnectionCore,
 )
 from tigrcorn_protocols.http3.websocket import H3WebSocketSession
+from tigrcorn_protocols.sessions import RuntimeConnectionInventory, peer_id_from_address
 from tigrcorn_protocols.webtransport.governance import WebTransportGovernanceError
+from tigrcorn_protocols.webtransport.negotiation import negotiate_profiles, settings_for_profiles
+from tigrcorn_protocols.webtransport.profiles import conflicting_request_profile, missing_peer_requirement, missing_request_requirement, profile_spec
 from tigrcorn_transports.quic.connection import QuicConnection
 from tigrcorn_transports.quic.handshake import QuicTlsHandshakeDriver, TransportParameters
 from tigrcorn_transports.quic.packets import QuicLongHeaderPacket, QuicLongHeaderType, QuicRetryPacket, QuicShortHeaderPacket, QuicVersionNegotiationPacket, decode_packet

{tigrcorn_protocols-0.3.18.dev2 → tigrcorn_protocols-0.3.18.dev3}/src/tigrcorn_protocols/http3/handler/lifecycle.py

@@ -25,6 +25,62 @@ class HTTP3LifecycleMixin:
     def _webtransport_address(addr: tuple[str, int]) -> str:
         return f"{addr[0]}:{addr[1]}"
 
+    def _connection_id_for_session(self, session: HTTP3Session) -> str:
+        return f"conn:h3:{session.runtime_id or session.quic.local_cid.hex() or 'session'}"
+
+    def _listener_id(self) -> str:
+        return self.listener.label or f"{self.listener.kind}:{self.listener.host}:{self.listener.port}"
+
+    def _register_h3_connection(self, session: HTTP3Session) -> None:
+        inventory = self.connection_inventory
+        if inventory is None:
+            return
+        address = self._webtransport_address(session.addr)
+        handshake = session.quic.handshake_driver
+        inventory.open_connection(
+            self._connection_id_for_session(session),
+            transport='quic',
+            protocols=tuple(sorted(self.listener.enabled_protocols)),
+            listener_id=self._listener_id(),
+            peer_id=peer_id_from_address(address),
+            remote_address=address,
+            local_address=self.listener.label,
+            security={
+                'alpn': getattr(handshake, 'selected_alpn', None) if handshake is not None else None,
+                'local_cid': session.quic.local_cid.hex(),
+                'remote_cid': session.quic.remote_cid.hex() if isinstance(session.quic.remote_cid, bytes) else None,
+                'tls': bool(handshake is not None),
+            },
+        )
+
+    def _update_h3_connection(self, session: HTTP3Session) -> None:
+        inventory = self.connection_inventory
+        if inventory is None:
+            return
+        inventory.update_connection(
+            self._connection_id_for_session(session),
+            remote_address=self._webtransport_address(session.addr),
+            counters={
+                'bytes_received': session.bytes_received,
+                'bytes_sent': session.bytes_sent,
+                'requests': len(session.responded_streams),
+                'streams': len(session.h3.requests) + len(session.webtransport_streams),
+                'webtransport_sessions': len(session.webtransport_sessions),
+                'websocket_sessions': len(session.websocket_sessions),
+            },
+            security={
+                'local_cid': session.quic.local_cid.hex(),
+                'remote_cid': session.quic.remote_cid.hex() if isinstance(session.quic.remote_cid, bytes) else None,
+            },
+        )
+
+    def _close_h3_connection(self, session: HTTP3Session, *, reason: str) -> None:
+        inventory = self.connection_inventory
+        if inventory is None:
+            return
+        self._update_h3_connection(session)
+        inventory.close_connection(self._connection_id_for_session(session), reason=reason)
+
     def _webtransport_budget_snapshot(self) -> dict[str, Any] | None:
         manager = self.webtransport_governance
         if manager is None:
@@ -37,16 +93,29 @@ class HTTP3LifecycleMixin:
             return None
         return manager.open_session(
             webtransport.session_id,
-            peer_id=self._webtransport_address(session.addr),
+            peer_id=peer_id_from_address(self._webtransport_address(session.addr)),
             address=self._webtransport_address(session.addr),
         )
 
     def _webtransport_register_stream(self, webtransport: _HTTP3WebTransportSession, stream_id: int | str) -> dict[str, Any] | None:
+        if self.connection_inventory is not None:
+            self.connection_inventory.update_session(
+                webtransport.session_id,
+                stream_ids=(str(stream_id),),
+            )
         manager = self.webtransport_governance
         if manager is None:
             return None
         return manager.open_stream(webtransport.session_id, str(stream_id))
 
+    def _webtransport_release_stream(
+        self, webtransport: _HTTP3WebTransportSession, stream_id: int | str
+    ) -> dict[str, Any] | None:
+        manager = self.webtransport_governance
+        if manager is None:
+            return None
+        return manager.close_stream(webtransport.session_id, str(stream_id))
+
     def _webtransport_register_datagram(
         self,
         webtransport: _HTTP3WebTransportSession,
@@ -107,6 +176,13 @@ class HTTP3LifecycleMixin:
             max_streams_uni=self.config.scheduler.max_streams or 128,
             idle_timeout=int(self.config.quic.idle_timeout * 1000),
             max_datagram_frame_size=self.listener.max_datagram_size,
+            reset_stream_at=(
+                'webtransport' in self.listener.enabled_protocols
+                and any(
+                    profile_spec(name, max_sessions=int(self.config.webtransport.max_sessions or 1)).requires_reset_stream_at
+                    for name in self.config.webtransport.profiles
+                )
+            ),
         )
         session.quic.configure_handshake(
             QuicTlsHandshakeDriver(