Avalonia/src/Avalonia.Remote.Protocol/TcpTransportBase.cs

using System;
using System.IO;
using System.Net;
using System.Net.Sockets;
using System.Threading.Tasks;

namespace Avalonia.Remote.Protocol
{
    public abstract class TcpTransportBase
    {
        private readonly IMessageTypeResolver _resolver;

        public TcpTransportBase(IMessageTypeResolver resolver)
        {
            _resolver = resolver;
        }
        
        protected abstract IAvaloniaRemoteTransportConnection CreateTransport(IMessageTypeResolver resolver,
            Stream stream, Action disposeCallback);

        class DisposableServer : IDisposable
        {
            private readonly TcpListener _l;

            public DisposableServer(TcpListener l)
            {
                _l = l;
            }
            public void Dispose()
            {
                try
                {
                    _l.Stop();
                }
                catch
                {
                    //Ignore
                }
            }
        }
        
        public IDisposable Listen(IPAddress address, int port, Action<IAvaloniaRemoteTransportConnection> cb)
        {
            var server = new TcpListener(address, port);
            async void AcceptNew()
            {
                try
                {
                    var cl = await server.AcceptTcpClientAsync();
                    AcceptNew();
                    await Task.Run(async () =>
                    {
                        var tcs = new TaskCompletionSource<int>();
                        var t = CreateTransport(_resolver, cl.GetStream(), () => tcs.TrySetResult(0));
                        cb(t);
                        await tcs.Task;
                    });
                }
                catch
                {
                    //Ignore and stop
                }
            }
            server.Start();
            AcceptNew();
            return new DisposableServer(server);
        }

        public async Task<IAvaloniaRemoteTransportConnection> Connect(IPAddress address, int port)
        {
            var c = new TcpClient();
            await c.ConnectAsync(address, port);
            return CreateTransport(_resolver, c.GetStream(), ((IDisposable)c).Dispose);
        }
    }
}