Socket extension

Source.cs

    public static class SocketExtensions
    {
        public static Task ConnectExAsync(this Socket socket, string host, int port, CancellationToken cancellationToken = default)
        {
            var xCompletion = new TaskCompletionSource<bool>();
            socket.BeginConnect(host,
                                port,
                                r =>
                                {
                                    try
                                    {
                                        socket.EndConnect(r);
                                        xCompletion.TrySetResult(true);
                                    }
                                    catch (Exception E)
                                    {
                                        xCompletion.TrySetException(E);
                                    }
                                },
                                null);

            cancellationToken.Register(() =>
                                       {
                                           xCompletion.SetException(new TimeoutException());
                                           socket.Close();
                                       });

            return xCompletion.Task;
        }
    }