Event based communication over sockets.

scripts/test.sh

@@ -113,7 +113,7 @@ function invoke_test()
     local dotnet=$(_get_dotnet_path)
     local vstest=$TP_OUT_DIR/$TPB_Configuration/$TPB_TargetFrameworkCore/vstest.console.dll
 
-    find ./test -path $PROJECT_NAME_PATTERNS | xargs $dotnet $vstest --parallel
+    find ./test -path $PROJECT_NAME_PATTERNS | xargs --verbose $dotnet $vstest --parallel
 }
 
 #

src/Microsoft.TestPlatform.Common/Interfaces/Communication/ConnectedEventArgs.cs

@@ -24,11 +24,28 @@ public ConnectedEventArgs()
         public ConnectedEventArgs(ICommunicationChannel channel)
         {
             this.Channel = channel;
+            this.Connected = true;
+        }
+
+        public ConnectedEventArgs(Exception faultException)
+        {
+            this.Connected = false;
+            this.Fault = faultException;
         }
 
         /// <summary>
         /// Gets the communication channel based on this connection.
         /// </summary>
         public ICommunicationChannel Channel { get; private set; }
+
+        /// <summary>
+        /// Gets true if it's connected.
+        /// </summary>
+        public bool Connected { get; private set; }
+
+        /// <summary>
+        /// Gets the exception if it's not connected.
+        /// </summary>
+        public Exception Fault { get; private set; }
     }
 }

src/Microsoft.TestPlatform.CommunicationUtilities/DataCollectionRequestHandler.cs

@@ -121,11 +121,10 @@ public static DataCollectionRequestHandler Create(
             ICommunicationManager communicationManager,
             IMessageSink messageSink)
         {
+            ValidateArg.NotNull(communicationManager, nameof(communicationManager));
+            ValidateArg.NotNull(messageSink, nameof(messageSink));
             if (Instance == null)
             {
-                ValidateArg.NotNull(communicationManager, nameof(communicationManager));
-                ValidateArg.NotNull(messageSink, nameof(messageSink));
-
                 lock (SyncObject)
                 {
                     if (Instance == null)

src/Microsoft.TestPlatform.CommunicationUtilities/Interfaces/ICommunicationEndpoint.cs

@@ -0,0 +1,32 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+
+namespace Microsoft.VisualStudio.TestPlatform.CommunicationUtilities.Interfaces
+{
+    using System;
+
+    public interface ICommunicationEndPoint
+    {
+        /// <summary>
+        /// Event raised when an endpoint is connected.
+        /// </summary>
+        event EventHandler<ConnectedEventArgs> Connected;
+
+        /// <summary>
+        /// Event raised when an endpoint is disconnected.
+        /// </summary>
+        event EventHandler<DisconnectedEventArgs> Disconnected;
+
+        /// <summary>
+        /// Starts the endpoint and channel.
+        /// </summary>
+        /// <param name="endpoint">Address to connect</param>
+        /// <returns>Address of the connected endpoint</returns>
+        string Start(string endpoint);
+
+        /// <summary>
+        /// Stops the endpoint and closes the underlying communication channel.
+        /// </summary>
+        void Stop();
+    }
+}

src/Microsoft.TestPlatform.CommunicationUtilities/LengthPrefixCommunicationChannel.cs

@@ -57,10 +57,7 @@ public Task NotifyDataAvailable()
             // connection is closed.
             var data = this.reader.ReadString();
 
-            if (this.MessageReceived != null)
-            {
-                this.MessageReceived.SafeInvoke(this, new MessageReceivedEventArgs { Data = data }, "LengthPrefixCommunicationChannel: MessageReceived");
-            }
+            this.MessageReceived?.SafeInvoke(this, new MessageReceivedEventArgs { Data = data }, "LengthPrefixCommunicationChannel: MessageReceived");
 
             return Task.FromResult(0);
         }

src/Microsoft.TestPlatform.CommunicationUtilities/SocketClient.cs

@@ -17,7 +17,7 @@ namespace Microsoft.VisualStudio.TestPlatform.CommunicationUtilities
     /// <summary>
     /// Communication client implementation over sockets.
     /// </summary>
-    public class SocketClient : ICommunicationClient
+    public class SocketClient : ICommunicationEndPoint
     {
         private readonly CancellationTokenSource cancellation;
         private readonly TcpClient tcpClient;
@@ -41,16 +41,19 @@ protected SocketClient(Func<Stream, ICommunicationChannel> channelFactory)
         }
 
         /// <inheritdoc />
-        public event EventHandler<ConnectedEventArgs> ServerConnected;
+        public event EventHandler<ConnectedEventArgs> Connected;
 
         /// <inheritdoc />
-        public event EventHandler<DisconnectedEventArgs> ServerDisconnected;
+        public event EventHandler<DisconnectedEventArgs> Disconnected;
 
         /// <inheritdoc />
-        public void Start(string connectionInfo)
+        public string Start(string endpoint)
         {
-            this.tcpClient.ConnectAsync(IPAddress.Loopback, int.Parse(connectionInfo))
-                .ContinueWith(this.OnServerConnected);
+            var ipEndPoint = endpoint.GetIPEndPoint();
+
+            // Don't start if the endpoint port is zero
+            this.tcpClient.ConnectAsync(ipEndPoint.Address, ipEndPoint.Port).ContinueWith(this.OnServerConnected);
+            return endpoint;
         }
 
         /// <inheritdoc />
@@ -65,22 +68,24 @@ public void Stop()
 
         private void OnServerConnected(Task connectAsyncTask)
         {
-            if (connectAsyncTask.IsFaulted)
-            {
-                throw connectAsyncTask.Exception;
-            }
-
-            this.channel = this.channelFactory(this.tcpClient.GetStream());
-            if (this.ServerConnected != null)
+            if (this.Connected != null)
             {
-                this.ServerConnected.SafeInvoke(this, new ConnectedEventArgs(this.channel), "SocketClient: ServerConnected");
+                if (connectAsyncTask.IsFaulted)
+                {
+                    this.Connected.SafeInvoke(this, new ConnectedEventArgs(connectAsyncTask.Exception), "SocketClient: ServerConnected");
+                }
+                else
+                {
+                    this.channel = this.channelFactory(this.tcpClient.GetStream());
+                    this.Connected.SafeInvoke(this, new ConnectedEventArgs(this.channel), "SocketClient: ServerConnected");
 
-                // Start the message loop
-                Task.Run(() => this.tcpClient.MessageLoopAsync(
-                        this.channel,
-                        this.Stop,
-                        this.cancellation.Token))
-                    .ConfigureAwait(false);
+                    // Start the message loop
+                    Task.Run(() => this.tcpClient.MessageLoopAsync(
+                            this.channel,
+                            this.Stop,
+                            this.cancellation.Token))
+                        .ConfigureAwait(false);
+                }
             }
         }
 
@@ -102,7 +107,7 @@ private void Stop(Exception error)
                 this.channel.Dispose();
                 this.cancellation.Dispose();
 
-                this.ServerDisconnected?.SafeInvoke(this, new DisconnectedEventArgs(), "SocketClient: ServerDisconnected");
+                this.Disconnected?.SafeInvoke(this, new DisconnectedEventArgs(), "SocketClient: ServerDisconnected");
             }
         }
     }

src/Microsoft.TestPlatform.CommunicationUtilities/SocketServer.cs

@@ -17,7 +17,7 @@ namespace Microsoft.VisualStudio.TestPlatform.CommunicationUtilities
     /// <summary>
     /// Communication server implementation over sockets.
     /// </summary>
-    public class SocketServer : ICommunicationServer
+    public class SocketServer : ICommunicationEndPoint
     {
         private readonly CancellationTokenSource cancellation;
 
@@ -35,7 +35,7 @@ public class SocketServer : ICommunicationServer
         /// Initializes a new instance of the <see cref="SocketServer"/> class.
         /// </summary>
         public SocketServer()
-            : this((stream) => new LengthPrefixCommunicationChannel(stream))
+            : this(stream => new LengthPrefixCommunicationChannel(stream))
         {
         }
 
@@ -53,21 +53,19 @@ protected SocketServer(Func<Stream, ICommunicationChannel> channelFactory)
         }
 
         /// <inheritdoc />
-        public event EventHandler<ConnectedEventArgs> ClientConnected;
+        public event EventHandler<ConnectedEventArgs> Connected;
 
         /// <inheritdoc />
-        public event EventHandler<DisconnectedEventArgs> ClientDisconnected;
+        public event EventHandler<DisconnectedEventArgs> Disconnected;
 
-        /// <inheritdoc />
-        public string Start()
+        public string Start(string endpoint)
         {
-            var endpoint = new IPEndPoint(IPAddress.Loopback, 0);
-            this.tcpListener = new TcpListener(endpoint);
+            this.tcpListener = new TcpListener(endpoint.GetIPEndPoint());
 
             this.tcpListener.Start();
 
-            var connectionInfo = ((IPEndPoint)this.tcpListener.LocalEndpoint).Port.ToString();
-            EqtTrace.Info("SocketServer: Listening on port : {0}", connectionInfo);
+            var connectionInfo = ((IPEndPoint)this.tcpListener.LocalEndpoint).ToString();
+            EqtTrace.Info("SocketServer: Listening on end point : {0}", connectionInfo);
 
             // Serves a single client at the moment. An error in connection, or message loop just
             // terminates the entire server.
@@ -90,10 +88,10 @@ private void OnClientConnected(TcpClient client)
             this.tcpClient = client;
             this.tcpClient.Client.NoDelay = true;
 
-            if (this.ClientConnected != null)
+            if (this.Connected != null)
             {
                 this.channel = this.channelFactory(this.tcpClient.GetStream());
-                this.ClientConnected.SafeInvoke(this, new ConnectedEventArgs(this.channel), "SocketServer: ClientConnected");
+                this.Connected.SafeInvoke(this, new ConnectedEventArgs(this.channel), "SocketServer: ClientConnected");
 
                 // Start the message loop
                 Task.Run(() => this.tcpClient.MessageLoopAsync(this.channel, error => this.Stop(error), this.cancellation.Token)).ConfigureAwait(false);
@@ -121,7 +119,7 @@ private void Stop(Exception error)
                 this.channel.Dispose();
                 this.cancellation.Dispose();
 
-                this.ClientDisconnected?.SafeInvoke(this, new DisconnectedEventArgs { Error = error }, "SocketServer: ClientDisconnected");
+                this.Disconnected?.SafeInvoke(this, new DisconnectedEventArgs { Error = error }, "SocketServer: ClientDisconnected");
             }
         }
     }

src/Microsoft.TestPlatform.CommunicationUtilities/TcpClientExtensions.cs

@@ -5,6 +5,7 @@ namespace Microsoft.VisualStudio.TestPlatform.CommunicationUtilities
 {
     using System;
     using System.IO;
+    using System.Net;
     using System.Net.Sockets;
     using System.Threading;
     using System.Threading.Tasks;
@@ -69,5 +70,20 @@ internal static Task MessageLoopAsync(
 
             return Task.FromResult(0);
         }
+
+        /// <summary>
+        /// Converts a given string endpoint address to valid Ipv4, Ipv6 IPEndpoint
+        /// </summary>
+        /// <param name="value">Input endpoint address</param>
+        /// <returns>IPEndpoint from give string, if its not a valid string. It will create endpoint with loop back address with port 0</returns>
+        internal static IPEndPoint GetIPEndPoint(this string value)
+        {
+            if (Uri.TryCreate(string.Concat("tcp://", value), UriKind.Absolute, out Uri uri))
+            {
+                return new IPEndPoint(IPAddress.Parse(uri.Host), uri.Port < 0 ? 0 : uri.Port);
+            }
+
+            return new IPEndPoint(IPAddress.Loopback, 0);
+        }
     }
 }