diff --git a/bumble/transport/common.py b/bumble/transport/common.py index 5d5bdf1..c030308 100644 --- a/bumble/transport/common.py +++ b/bumble/transport/common.py @@ -63,6 +63,8 @@ class TransportSink(Protocol): class TransportSource(Protocol): + terminated: asyncio.Future[None] + def set_packet_sink(self, sink: TransportSink) -> None: ... @@ -430,6 +432,7 @@ class SnoopingTransport(Transport): def __init__(self, source: TransportSource, snooper: Snooper): self.source = source self.snooper = snooper + self.terminated = source.terminated def set_packet_sink(self, sink: TransportSink) -> None: self.sink = sink