diff --git a/TcpDotNet/ClientNode.cs b/TcpDotNet/ClientNode.cs index bb29390..46aea66 100644 --- a/TcpDotNet/ClientNode.cs +++ b/TcpDotNet/ClientNode.cs @@ -16,6 +16,7 @@ public abstract class ClientNode : Node { private readonly ConcurrentDictionary>> _packetCompletionSources = new(); private readonly ConcurrentDictionary> _callbackCompletionSources = new(); + private EndPoint? _remoteEP; /// @@ -160,6 +161,9 @@ public abstract class ClientNode : Node } } + if (packet is ResponsePacket response && _callbackCompletionSources.TryGetValue(response.CallbackId, out TaskCompletionSource? callback)) + callback.SetResult(response); + return packet; }