Use server redirection information.

Signed-off-by: Bradley Grainger <[email protected]>

Author: bgrainger

src/MySqlConnector/Core/ConnectionSettings.cs

@@ -125,6 +125,8 @@ public ConnectionSettings(MySqlConnectionStringBuilder csb)
 			UseXaTransactions = csb.UseXaTransactions;
 		}
 
+		public ConnectionSettings CloneWith(string host, int port, string userId, bool isRedirected) => new ConnectionSettings(this, host, port, userId, isRedirected);
+
 		private static MySqlGuidFormat GetEffectiveGuidFormat(MySqlGuidFormat guidFormat, bool oldGuids)
 		{
 			switch (guidFormat)
@@ -210,6 +212,7 @@ private static MySqlGuidFormat GetEffectiveGuidFormat(MySqlGuidFormat guidFormat
 		public bool UseXaTransactions { get; }
 
 		public byte[]? ConnectionAttributes { get; set; }
+		public bool IsRedirected { get; }
 
 		// Helper Functions
 		int? m_connectionTimeoutMilliseconds;
@@ -235,6 +238,66 @@ public int ConnectionTimeoutMilliseconds
 			}
 		}
 
+		private ConnectionSettings(ConnectionSettings other, string host, int port, string userId, bool isRedirected)
+		{
+			ConnectionStringBuilder = other.ConnectionStringBuilder;
+			ConnectionString = other.ConnectionString;
+
+			ConnectionProtocol = MySqlConnectionProtocol.Sockets;
+			HostNames = new[] { host };
+			LoadBalance = other.LoadBalance;
+			Port = port;
+			PipeName = other.PipeName;
+
+			UserID = userId;
+			Password = other.Password;
+			Database = other.Database;
+
+			SslMode = other.SslMode;
+			CertificateFile = other.CertificateFile;
+			CertificatePassword = other.CertificatePassword;
+			SslCertificateFile = other.SslCertificateFile;
+			SslKeyFile = other.SslKeyFile;
+			CACertificateFile = other.CACertificateFile;
+			CertificateStoreLocation = other.CertificateStoreLocation;
+			CertificateThumbprint = other.CertificateThumbprint;
+
+			Pooling = other.Pooling;
+			ConnectionLifeTime = other.ConnectionLifeTime;
+			ConnectionReset = other.ConnectionReset;
+			ConnectionIdlePingTime = other.ConnectionIdlePingTime;
+			ConnectionIdleTimeout = other.ConnectionIdleTimeout;
+			MinimumPoolSize = other.MinimumPoolSize;
+			MaximumPoolSize = other.MaximumPoolSize;
+
+			AllowLoadLocalInfile = other.AllowLoadLocalInfile;
+			AllowPublicKeyRetrieval = other.AllowPublicKeyRetrieval;
+			AllowUserVariables = other.AllowUserVariables;
+			AllowZeroDateTime = other.AllowZeroDateTime;
+			ApplicationName = other.ApplicationName;
+			AutoEnlist = other.AutoEnlist;
+			ConnectionTimeout = other.ConnectionTimeout;
+			ConvertZeroDateTime = other.ConvertZeroDateTime;
+			DateTimeKind = other.DateTimeKind;
+			DefaultCommandTimeout = other.DefaultCommandTimeout;
+			ForceSynchronous = other.ForceSynchronous;
+			IgnoreCommandTransaction = other.IgnoreCommandTransaction;
+			IgnorePrepare = other.IgnorePrepare;
+			InteractiveSession = other.InteractiveSession;
+			GuidFormat = other.GuidFormat;
+			Keepalive = other.Keepalive;
+			NoBackslashEscapes = other.NoBackslashEscapes;
+			PersistSecurityInfo = other.PersistSecurityInfo;
+			ServerRsaPublicKeyFile = other.ServerRsaPublicKeyFile;
+			ServerSPN = other.ServerSPN;
+			TreatTinyAsBoolean = other.TreatTinyAsBoolean;
+			UseAffectedRows = other.UseAffectedRows;
+			UseCompression = other.UseCompression;
+			UseXaTransactions = other.UseXaTransactions;
+
+			IsRedirected = isRedirected;
+		}
+
 		static readonly string[] s_localhostPipeServer = { "." };
 	}
 }

src/MySqlConnector/Core/ServerSession.cs

@@ -348,6 +348,7 @@ public async Task ConnectAsync(ConnectionSettings cs, int startTickCount, ILoadB
 					m_state = State.Connecting;
 				}
 
+serverRedirection:
 				// TLS negotiation should automatically fall back to the best version supported by client and server. However,
 				// Windows Schannel clients will fail to connect to a yaSSL-based MySQL Server if TLS 1.2 is requested and
 				// have to use only TLS 1.1: https://github.com/mysql-net/MySqlConnector/pull/101
@@ -462,7 +463,51 @@ public async Task ConnectAsync(ConnectionSettings cs, int startTickCount, ILoadB
 					payload = await SwitchAuthenticationAsync(cs, payload, ioBehavior, cancellationToken).ConfigureAwait(false);
 				}
 
-				OkPayload.Create(payload.Span, SupportsDeprecateEof, SupportsSessionTrack);
+				var ok = OkPayload.Create(payload.Span, SupportsDeprecateEof, SupportsSessionTrack);
+				if (ok.StatusInfo is not null && ok.StatusInfo.StartsWith("Location: mysql://", StringComparison.Ordinal))
+				{
+					// server redirection string has the format "Location: mysql://{host}:{port}/user={userId}[&ttl={ttl}]"
+					m_logArguments[1] = ok.StatusInfo;
+					Log.Info("Session{0} has server redirection header {1}", m_logArguments);
+
+					if (cs.IsRedirected)
+					{
+						Log.Info("Session{0} is already redirected; ignoring it.", m_logArguments);
+					}
+					else
+					{
+						var hostIndex = 18;
+						var colonIndex = ok.StatusInfo.IndexOf(':', hostIndex);
+						if (colonIndex != -1)
+						{
+							var host = ok.StatusInfo.Substring(hostIndex, colonIndex - hostIndex);
+							var portIndex = colonIndex + 1;
+							var userIndex = ok.StatusInfo.IndexOf("/user=", StringComparison.Ordinal);
+							if (userIndex != -1)
+							{
+								if (int.TryParse(ok.StatusInfo.Substring(portIndex, userIndex - portIndex), out var port))
+								{
+									var ampersandIndex = ok.StatusInfo.IndexOf('&', userIndex);
+									var userId = ampersandIndex == -1 ? ok.StatusInfo.Substring(userIndex + 6) : ok.StatusInfo.Substring(userIndex + 6, ampersandIndex - userIndex - 6);
+									Log.Info("Session{0} found server redirection Host={1}; Port={2}; User={3}", m_logArguments[0], host, port, userId);
+
+									if (host != cs.HostNames![0] || port != cs.Port || userId != cs.UserID)
+									{
+										Log.Info("Session{0} closing existing connection", m_logArguments);
+										await SendAsync(QuitPayload.Instance, ioBehavior, cancellationToken).ConfigureAwait(false);
+										Log.Info("Session{0} opening new connection to Host={1}; Port={2}; User={3}", m_logArguments[0], host, port, userId);
+										cs = cs.CloneWith(host, port, userId, isRedirected: true);
+										goto serverRedirection;
+									}
+									else
+									{
+										Log.Info("Session{0} is already connected to this server; ignoring redirection", m_logArguments);
+									}
+								}
+							}
+						}
+					}
+				}
 
 				if (m_useCompression)
 					m_payloadHandler = new CompressedPayloadHandler(m_payloadHandler.ByteHandler);
@@ -1364,6 +1409,7 @@ private bool ShouldGetRealServerDetails(ConnectionSettings cs)
 			if (cs.ConnectionProtocol == MySqlConnectionProtocol.Sockets && cs.HostNames!.Count == 1)
 			{
 				return cs.HostNames[0].EndsWith(".mysql.database.azure.com", StringComparison.OrdinalIgnoreCase) ||
+					cs.HostNames[0].EndsWith(".database.windows.net", StringComparison.OrdinalIgnoreCase) ||
 					cs.HostNames[0].EndsWith(".mysql.database.chinacloudapi.cn", StringComparison.OrdinalIgnoreCase);
 			}