Allow connections on non-default port (#462)

* Allow custom port

* Update unit tests for port property

src/Microsoft.SqlTools.ServiceLayer/Admin/AdminService.cs

@@ -82,6 +82,8 @@ namespace Microsoft.SqlTools.ServiceLayer.Admin
         public static async Task HandleDefaultDatabaseInfoRequest(
             DefaultDatabaseInfoParams optionsParams,
             RequestContext<DefaultDatabaseInfoResponse> requestContext)
+        {
+            try
             {
                 var response = new DefaultDatabaseInfoResponse();
                 ConnectionInfo connInfo;
@@ -97,6 +99,11 @@ namespace Microsoft.SqlTools.ServiceLayer.Admin
                 response.DefaultDatabaseInfo = DatabaseTaskHelper.DatabasePrototypeToDatabaseInfo(taskHelper.Prototype);
                 await requestContext.SendResult(response);
             }
+            catch (Exception ex)
+            {
+                await requestContext.SendError(ex.ToString());
+            }                
+        }
 
         /// <summary>
         /// Handles a create database request
@@ -104,6 +111,8 @@ namespace Microsoft.SqlTools.ServiceLayer.Admin
         internal static async Task HandleCreateDatabaseRequest(
             CreateDatabaseParams databaseParams,
             RequestContext<CreateDatabaseResponse> requestContext)
+        {
+            try
             {        
                 var response = new DefaultDatabaseInfoResponse();
                 ConnectionInfo connInfo;
@@ -131,6 +140,11 @@ namespace Microsoft.SqlTools.ServiceLayer.Admin
                     TaskId = 0
                 }); 
             }
+            catch (Exception ex)
+            {
+                await requestContext.SendError(ex.ToString());
+            }         
+        }
 
         /// <summary>
         /// Handle get database info request
@@ -138,6 +152,8 @@ namespace Microsoft.SqlTools.ServiceLayer.Admin
         internal static async Task HandleGetDatabaseInfoRequest(
             GetDatabaseInfoParams databaseParams,
             RequestContext<GetDatabaseInfoResponse> requestContext)
+        {
+            try
             {            
                 ConnectionInfo connInfo;
                 AdminService.ConnectionServiceInstance.TryFindConnection(
@@ -154,6 +170,11 @@ namespace Microsoft.SqlTools.ServiceLayer.Admin
                     DatabaseInfo = info
                 });
             }
+            catch (Exception ex)
+            {
+                await requestContext.SendError(ex.ToString());
+            }
+        }
         
         /// <summary>
         /// Return database info for a specific database
@@ -177,15 +198,21 @@ namespace Microsoft.SqlTools.ServiceLayer.Admin
             XmlDocument xmlDoc = CreateDataContainerDocument(connInfo, databaseExists);
             CDataContainer dataContainer;
 
+             // add alternate port to server name property if provided
+            var connectionDetails = connInfo.ConnectionDetails;
+            string serverName = !connectionDetails.Port.HasValue
+                ? connectionDetails.ServerName
+                : string.Format("{0},{1}", connectionDetails.ServerName, connectionDetails.Port.Value);
+
             // check if the connection is using SQL Auth or Integrated Auth
-            if (string.Equals(connInfo.ConnectionDetails.AuthenticationType, "SqlLogin", StringComparison.OrdinalIgnoreCase))
+            if (string.Equals(connectionDetails.AuthenticationType, "SqlLogin", StringComparison.OrdinalIgnoreCase))
             {
-                var passwordSecureString = BuildSecureStringFromPassword(connInfo.ConnectionDetails.Password);
+                var passwordSecureString = BuildSecureStringFromPassword(connectionDetails.Password);
                 dataContainer = new CDataContainer(
                     CDataContainer.ServerType.SQL,
-                    connInfo.ConnectionDetails.ServerName,
+                    serverName,
                     false,
-                    connInfo.ConnectionDetails.UserName,
+                    connectionDetails.UserName,
                     passwordSecureString,
                     xmlDoc.InnerXml);
             }
@@ -193,7 +220,7 @@ namespace Microsoft.SqlTools.ServiceLayer.Admin
             {
                 dataContainer = new CDataContainer(
                     CDataContainer.ServerType.SQL,
-                    connInfo.ConnectionDetails.ServerName,
+                    serverName,
                     true,
                     null,
                     null,

src/Microsoft.SqlTools.ServiceLayer/Connection/ConnectionService.cs

@@ -886,10 +886,16 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection
             {
                 connectionBuilder = new SqlConnectionStringBuilder(connectionDetails.ConnectionString);
             }
-            else {
+            else 
+            {
+                // add alternate port to data source property if provided
+                string dataSource = !connectionDetails.Port.HasValue
+                    ? connectionDetails.ServerName
+                    : string.Format("{0},{1}", connectionDetails.ServerName, connectionDetails.Port.Value);
+
                 connectionBuilder = new SqlConnectionStringBuilder
                 {
-                    ["Data Source"] = connectionDetails.ServerName,
+                    ["Data Source"] = dataSource,
                     ["User Id"] = connectionDetails.UserName,
                     ["Password"] = connectionDetails.Password
                 };

src/Microsoft.SqlTools.ServiceLayer/Connection/Contracts/ConnectionDetails.cs

@@ -418,6 +418,22 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection.Contracts
             }
         }
 
+        /// <summary>
+        /// Gets or sets the port to use for the TCP/IP connection
+        /// </summary>
+        public int? Port
+        {
+            get
+            {
+                return GetOptionValue<int?>("port");
+            }
+
+            set
+            {
+                SetOptionValue("port", value);
+            }
+        }        
+
         /// <summary>
         /// Gets or sets a string value that indicates the type system the application expects.
         /// </summary>

src/Microsoft.SqlTools.ServiceLayer/Connection/Contracts/ConnectionDetailsExtensions.cs

@@ -43,7 +43,8 @@ namespace Microsoft.SqlTools.ServiceLayer.Connection.Contracts
                 MultipleActiveResultSets = details.MultipleActiveResultSets,
                 PacketSize = details.PacketSize,
                 TypeSystemVersion = details.TypeSystemVersion,
-                ConnectionString = details.ConnectionString
+                ConnectionString = details.ConnectionString,
+                Port = details.Port
             };
         }
     }

test/Microsoft.SqlTools.ServiceLayer.UnitTests/Connection/ConnectionDetailsTests.cs

@@ -51,6 +51,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Connection
             Assert.Equal(details.Pooling, expectedForBoolean);
             Assert.Equal(details.Replication, expectedForBoolean);
             Assert.Equal(details.TrustServerCertificate, expectedForBoolean);
+            Assert.Equal(details.Port, expectedForInt);
         }
 
         [Fact]
@@ -87,6 +88,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Connection
             details.Pooling = (index++ % 2 == 0);
             details.Replication = (index++ % 2 == 0);
             details.TrustServerCertificate = (index++ % 2 == 0);
+            details.Port = expectedForInt + index++;
 
             index = 0;
             Assert.Equal(details.ApplicationIntent, expectedForStrings + index++);
@@ -115,6 +117,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Connection
             Assert.Equal(details.Pooling, (index++ % 2 == 0));
             Assert.Equal(details.Replication, (index++ % 2 == 0));
             Assert.Equal(details.TrustServerCertificate, (index++ % 2 == 0));
+            Assert.Equal(details.Port, (expectedForInt + index++));
         }
 
         [Fact]
@@ -152,6 +155,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Connection
             details.Pooling = (index++ % 2 == 0);
             details.Replication = (index++ % 2 == 0);
             details.TrustServerCertificate = (index++ % 2 == 0);
+            details.Port = expectedForInt + index++;
 
             if(optionMetadata.Options.Count() != details.Options.Count)
             {