Merged PR 4036: [5.1.2] Fix | LocalDb and managed SNI (#2129)

Ports [#2129](#2129)

Author: DavoudEshtehari

src/Microsoft.Data.SqlClient/netcore/src/Microsoft/Data/SqlClient/SNI/LocalDB.Windows.cs

@@ -50,8 +50,14 @@ private string GetConnectionString(string localDbInstance)
         {
             StringBuilder localDBConnectionString = new StringBuilder(MAX_LOCAL_DB_CONNECTION_STRING_SIZE + 1);
             int sizeOfbuffer = localDBConnectionString.Capacity;
-            localDBStartInstanceFunc(localDbInstance, 0, localDBConnectionString, ref sizeOfbuffer);
-            return localDBConnectionString.ToString();
+            int result = localDBStartInstanceFunc(localDbInstance, 0, localDBConnectionString, ref sizeOfbuffer);
+            if (result != TdsEnums.SNI_SUCCESS)
+            {
+                SNILoadHandle.SingletonInstance.LastError = new SNIError(SNIProviders.INVALID_PROV, 0, SNICommon.LocalDBErrorCode, Strings.SNI_ERROR_50);
+                SqlClientEventSource.Log.TrySNITraceEvent(nameof(LocalDB), EventType.ERR, "Unsuccessful 'LocalDBStartInstance' method call with {0} result to start '{1}' localDb instance", args0: result, args1: localDbInstance);
+                localDBConnectionString = null;
+            }
+            return localDBConnectionString?.ToString();
         }
 
         internal enum LocalDBErrorState

src/Microsoft.Data.SqlClient/netcore/src/Microsoft/Data/SqlClient/SNI/SNICommon.cs

@@ -21,16 +21,18 @@ namespace Microsoft.Data.SqlClient.SNI
     /// </summary>
     internal enum SNIProviders
     {
-        HTTP_PROV, // HTTP Provider
-        NP_PROV, // Named Pipes Provider
-        SESSION_PROV, // Session Provider
-        SIGN_PROV, // Sign Provider
-        SM_PROV, // Shared Memory Provider
-        SMUX_PROV, // SMUX Provider
-        SSL_PROV, // SSL Provider
-        TCP_PROV, // TCP Provider
-        MAX_PROVS, // Number of providers
-        INVALID_PROV // SQL Network Interfaces
+        HTTP_PROV = 0, // HTTP Provider
+        NP_PROV = 1, // Named Pipes Provider
+        SESSION_PROV = 2, // Session Provider
+        SIGN_PROV = 3, // Sign Provider
+        SM_PROV = 4, // Shared Memory Provider
+        SMUX_PROV = 5, // SMUX Provider
+        SSL_PROV = 6, // SSL Provider
+        TCP_PROV = 7, // TCP Provider
+        VIA_PROV = 8, // Virtual Interface Architecture Provider
+        CTAIP_PROV = 9,
+        MAX_PROVS = 10, // Number of providers
+        INVALID_PROV = 11 // SQL Network Interfaces
     }
 
     /// <summary>

src/Microsoft.Data.SqlClient/netcore/src/Microsoft/Data/SqlClient/SNI/SNIProxy.cs

@@ -190,7 +190,7 @@ internal static SNIHandle CreateConnectionHandle(
                 case DataSource.Protocol.TCP:
                     sniHandle = CreateTcpHandle(details, timerExpire, parallel, ipPreference, cachedFQDN, ref pendingDNSInfo,
                         tlsFirst, hostNameInCertificate, serverCertificateFilename);
-                    break;
+                     break;
                 case DataSource.Protocol.NP:
                     sniHandle = CreateNpHandle(details, timerExpire, parallel, tlsFirst);
                     break;
@@ -392,7 +392,7 @@ private static string GetLocalDBDataSource(string fullServerName, out bool error
                 Debug.Assert(!string.IsNullOrWhiteSpace(localDBInstance), "Local DB Instance name cannot be empty.");
                 localDBConnectionString = LocalDB.GetLocalDBConnectionString(localDBInstance);
 
-                if (fullServerName == null)
+                if (fullServerName == null || string.IsNullOrEmpty(localDBConnectionString))
                 {
                     // The Last error is set in LocalDB.GetLocalDBConnectionString. We don't need to set Last here.
                     error = true;
@@ -522,7 +522,18 @@ internal static string GetLocalDBInstance(string dataSource, out bool error)
             ReadOnlySpan<char> input = dataSource.AsSpan().TrimStart();
             error = false;
             // NetStandard 2.0 does not support passing a string to ReadOnlySpan<char>
-            if (input.StartsWith(LocalDbHost.AsSpan().Trim(), StringComparison.InvariantCultureIgnoreCase))
+            int index = input.IndexOf(LocalDbHost.AsSpan().Trim(), StringComparison.InvariantCultureIgnoreCase);
+            if (input.StartsWith(LocalDbHost_NP.AsSpan().Trim(), StringComparison.InvariantCultureIgnoreCase))
+            {
+                instanceName = input.Trim().ToString();
+            }
+            else if (index > 0)
+            {
+                SNILoadHandle.SingletonInstance.LastError = new SNIError(SNIProviders.INVALID_PROV, 0, SNICommon.ErrorLocatingServerInstance, Strings.SNI_ERROR_26);
+                SqlClientEventSource.Log.TrySNITraceEvent(nameof(SNIProxy), EventType.ERR, "Incompatible use of prefix with LocalDb: '{0}'", dataSource);
+                error = true;
+            }
+            else if (index == 0)
             {
                 // When netcoreapp support for netcoreapp2.1 is dropped these slice calls could be converted to System.Range\System.Index
                 // Such ad input = input[1..];
@@ -541,10 +552,6 @@ internal static string GetLocalDBInstance(string dataSource, out bool error)
                     error = true;
                 }
             }
-            else if (input.StartsWith(LocalDbHost_NP.AsSpan().Trim(), StringComparison.InvariantCultureIgnoreCase))
-            {
-                instanceName = input.Trim().ToString();
-            }
 
             return instanceName;
         }

src/Microsoft.Data.SqlClient/netcore/src/Microsoft/Data/SqlClient/TdsParser.cs

@@ -1516,20 +1516,17 @@ internal SqlError ProcessSNIError(TdsParserStateObject stateObj)
                 }
                 else
                 {
-
                     if (TdsParserStateObjectFactory.UseManagedSNI)
                     {
-                        // SNI error. Append additional error message info if available.
-                        //
+                        // SNI error. Append additional error message info if available and hasn't been included.
                         string sniLookupMessage = SQL.GetSNIErrorMessage((int)details.sniErrorNumber);
-                        errorMessage = (errorMessage != string.Empty) ?
-                                        (sniLookupMessage + ": " + errorMessage) :
-                                        sniLookupMessage;
+                        errorMessage = (string.IsNullOrEmpty(errorMessage) || errorMessage.Contains(sniLookupMessage))
+                                        ? sniLookupMessage
+                                        : (sniLookupMessage + ": " + errorMessage);
                     }
                     else
                     {
                         // SNI error. Replace the entire message.
-                        //
                         errorMessage = SQL.GetSNIErrorMessage((int)details.sniErrorNumber);
 
                         // If its a LocalDB error, then nativeError actually contains a LocalDB-specific error code, not a win32 error code
@@ -1538,6 +1535,7 @@ internal SqlError ProcessSNIError(TdsParserStateObject stateObj)
                             errorMessage += LocalDBAPI.GetLocalDBMessage((int)details.nativeError);
                             win32ErrorCode = 0;
                         }
+                        SqlClientEventSource.Log.TryAdvancedTraceEvent("<sc.TdsParser.ProcessSNIError |ERR|ADV > Extracting the latest exception from native SNI. errorMessage: {0}", errorMessage);
                     }
                 }
                 errorMessage = string.Format("{0} (provider: {1}, error: {2} - {3})",
@@ -12545,8 +12543,7 @@ internal bool TryReadPlpUnicodeChars(ref char[] buff, int offst, int len, TdsPar
                 return true;       // No data
             }
 
-            Debug.Assert(((ulong)stateObj._longlen != TdsEnums.SQL_PLP_NULL),
-                    "Out of sync plp read request");
+            Debug.Assert(((ulong)stateObj._longlen != TdsEnums.SQL_PLP_NULL), "Out of sync plp read request");
 
             Debug.Assert((buff == null && offst == 0) || (buff.Length >= offst + len), "Invalid length sent to ReadPlpUnicodeChars()!");
             charsLeft = len;

src/Microsoft.Data.SqlClient/netcore/src/Resources/Strings.Designer.cs

@@ -1195,10 +1195,10 @@
     <value>TCP Provider</value>
   </data>
   <data name="SNI_PN8" xml:space="preserve">
-    <value />
+    <value>VIA Provider</value>
   </data>
   <data name="SNI_PN9" xml:space="preserve">
-    <value>SQL Network Interfaces</value>
+    <value>CTAIP Provider</value>
   </data>
   <data name="AZURESQL_GenericEndpoint" xml:space="preserve">
     <value>.database.windows.net</value>
@@ -1941,4 +1941,10 @@
   <data name="SQL_TDS8_NotSupported_Netstandard2.0" xml:space="preserve">
     <value>Encrypt=Strict is not supported when targeting .NET Standard 2.0. Use .NET Standard 2.1, .NET Framework, or .NET.</value>
   </data>
-</root>
+  <data name="SNI_PN10" xml:space="preserve">
+    <value />
+  </data>
+  <data name="SNI_PN11" xml:space="preserve">
+    <value>SQL Network Interfaces</value>
+  </data>
+</root>

src/Microsoft.Data.SqlClient/netcore/src/Resources/Strings.resx

@@ -345,8 +345,12 @@ public static bool IsAKVSetupAvailable()
 
         public static bool IsNotUsingManagedSNIOnWindows() => !UseManagedSNIOnWindows;
 
-        public static bool IsUsingNativeSNI() => !IsUsingManagedSNI();
-
+        public static bool IsUsingNativeSNI() =>
+#if !NETFRAMEWORK
+        DataTestUtility.IsNotUsingManagedSNIOnWindows();
+#else 
+            true;
+#endif
         // Synapse: UTF8 collations are not supported with Azure Synapse.
         //          Ref: https://feedback.azure.com/forums/307516-azure-synapse-analytics/suggestions/40103791-utf-8-collations-should-be-supported-in-azure-syna
         public static bool IsUTF8Supported()

src/Microsoft.Data.SqlClient/tests/ManualTests/DataCommon/DataTestUtility.cs

@@ -16,6 +16,7 @@ private enum InfoType
             state
         }
         private static bool IsLocalDBEnvironmentSet() => DataTestUtility.IsLocalDBInstalled();
+        private static bool IsNativeSNI() => DataTestUtility.IsUsingNativeSNI();
         private static bool IsLocalDbSharedInstanceSet() => DataTestUtility.IsLocalDbSharedInstanceSetup();
         private static readonly string s_localDbConnectionString = @$"server=(localdb)\{DataTestUtility.LocalDbAppName}";
         private static readonly string[] s_sharedLocalDbInstances = new string[] { @$"server=(localdb)\.\{DataTestUtility.LocalDbSharedInstanceName}", @$"server=(localdb)\." };
@@ -123,6 +124,47 @@ public static void LocalDBNamepipeMarsTest()
 
         #endregion
 
+        #region Failures
+        // ToDo: After adding shared memory support on managed SNI, the IsNativeSNI could be taken out
+        [SkipOnTargetFramework(TargetFrameworkMonikers.Uap)] // No Registry support on UAP
+        [ConditionalTheory(nameof(IsLocalDBEnvironmentSet), nameof(IsNativeSNI))]
+        [InlineData("lpc:")]
+        public static void SharedMemoryAndSqlLocalDbConnectionTest(string prefix)
+        {
+            SqlConnectionStringBuilder stringBuilder = new(s_localDbConnectionString);
+            stringBuilder.DataSource = prefix + stringBuilder.DataSource;
+            SqlException ex = Assert.Throws<SqlException>(() => ConnectionTest(stringBuilder.ConnectionString));
+            Assert.Contains("A network-related or instance-specific error occurred while establishing a connection to SQL Server. The server was not found or was not accessible. Verify that the instance name is correct and that SQL Server is configured to allow remote connections. (provider: SQL Network Interfaces, error: 41 - Cannot open a Shared Memory connection to a remote SQL server)", ex.Message);
+        }
+
+        [SkipOnTargetFramework(TargetFrameworkMonikers.Uap)] // No Registry support on UAP
+        [InlineData("tcp:")]
+        [InlineData("np:")]
+        [InlineData("undefinded:")]
+        [ConditionalTheory(nameof(IsLocalDBEnvironmentSet))]
+        public static void PrefixAndSqlLocalDbConnectionTest(string prefix)
+        {
+            SqlConnectionStringBuilder stringBuilder = new(s_localDbConnectionString);
+            stringBuilder.DataSource = prefix + stringBuilder.DataSource;
+            SqlException ex = Assert.Throws<SqlException>(() => ConnectionTest(stringBuilder.ConnectionString));
+            Assert.Contains("A network-related or instance-specific error occurred while establishing a connection to SQL Server. The server was not found or was not accessible. Verify that the instance name is correct and that SQL Server is configured to allow remote connections. (provider: SQL Network Interfaces, error: 26 - Error Locating Server/Instance Specified)", ex.Message);
+        }
+
+        [SkipOnTargetFramework(TargetFrameworkMonikers.Uap)] // No Registry support on UAP
+        [ConditionalFact(nameof(IsLocalDBEnvironmentSet))]
+        public static void InvalidSqlLocalDbConnectionTest()
+        {
+            SqlConnectionStringBuilder stringBuilder = new(s_localDbConnectionString);
+            stringBuilder.DataSource = stringBuilder.DataSource + "Invalid123";
+            SqlException ex = Assert.Throws<SqlException>(() => ConnectionTest(stringBuilder.ConnectionString));
+            Assert.Contains("A network-related or instance-specific error occurred while establishing a connection to SQL Server. The server was not found or was not accessible. Verify that the instance name is correct and that SQL Server is configured to allow remote connections. (provider: SQL Network Interfaces, error: 50 - Local Database Runtime error occurred.", ex.Message);
+            if (IsNativeSNI())
+            {
+                Assert.Contains("The specified LocalDB instance does not exist.", ex.Message);
+            }
+        }
+        #endregion
+
         private static void ConnectionWithMarsTest(string connectionString)
         {
             SqlConnectionStringBuilder builder = new(connectionString)
@@ -178,13 +220,13 @@ private static void RestartLocalDB()
         {
             string state = ExecuteLocalDBCommandProcess(s_commandPrompt, s_sqlLocalDbInfo, InfoType.state);
             int count = 5;
-            while (state.Equals("stopped", StringComparison.InvariantCultureIgnoreCase) && count>0)
+            while (state.Equals("stopped", StringComparison.InvariantCultureIgnoreCase) && count > 0)
             {
                 count--;
                 state = ExecuteLocalDBCommandProcess(s_commandPrompt, s_startLocalDbCommand, InfoType.state);
                 Thread.Sleep(2000);
             }
-            if(state == null || state != "Running")
+            if (state == null || state != "Running")
             {
                 throw new LocalDBNotStartedException();
             }