Пример #1
0
        /// <summary>
        /// Parses a connection string.
        /// </summary>
        /// <param name="connectionString">The connection string to parse.</param>
        /// <returns>The parsed connection string.</returns>
        private string ParseConnectionString(string connectionString)
        {
            var  updatedConnectionString = string.Empty;
            bool portProvided            = false;
            bool isDnsSrv = false;
            var  connectionOptionsDictionary = connectionString.Split(CONNECTION_DATA_KEY_SEPARATOR)
                                               .Select(item => item.Split(new char[] { CONNECTION_DATA_VALUE_SEPARATOR }, 2))
                                               .Where(item => item.Length == 2)
                                               .ToDictionary(item => item[0], item => item[1]);
            var serverOption         = MySqlXConnectionStringBuilder.Options.Options.First(item => item.Keyword == SERVER_CONNECTION_OPTION_KEYWORD);
            var connecttimeoutOption = MySqlXConnectionStringBuilder.Options.Options.First(item => item.Keyword == CONNECT_TIMEOUT_CONNECTION_OPTION_KEYWORD);

            foreach (KeyValuePair <string, string> keyValuePair in connectionOptionsDictionary)
            {
                // Value is an equal or a semicolon
                if (keyValuePair.Value == "=" || keyValuePair.Value == "\"")
                {
                    throw new MySqlException(string.Format(Resources.InvalidConnectionStringValue, (keyValuePair.Value == "\"" ? ";" : "="), keyValuePair.Key));
                }

                // Key is not server or any of its synonyms.
                if (keyValuePair.Key != serverOption.Keyword && !serverOption.Synonyms.Contains(keyValuePair.Key))
                {
                    if ((connecttimeoutOption.Keyword == keyValuePair.Key || connecttimeoutOption.Synonyms.Contains(keyValuePair.Key)) &&
                        String.IsNullOrWhiteSpace(keyValuePair.Value))
                    {
                        throw new FormatException(ResourcesX.InvalidConnectionTimeoutValue);
                    }
                    if (keyValuePair.Key == PORT_CONNECTION_OPTION_KEYWORD)
                    {
                        portProvided = true;
                    }
                    if (keyValuePair.Key == DNS_SRV_CONNECTION_OPTION_KEYWORD)
                    {
                        isDnsSrv = Convert.ToBoolean(keyValuePair.Value);
                    }

                    updatedConnectionString += $"{keyValuePair.Key}{CONNECTION_DATA_VALUE_SEPARATOR}{keyValuePair.Value}{CONNECTION_DATA_KEY_SEPARATOR}";
                    continue;
                }

                // Key is server or one of its synonyms.
                var updatedValue = keyValuePair.Value;
                if (IsUnixSocket(keyValuePair.Value))
                {
                    updatedValue = NormalizeUnixSocket(keyValuePair.Value);
                }

                // The value for the server connection option doesn't have a server list format.
                if (FailoverManager.ParseHostList(updatedValue, true, false) == 1 && FailoverManager.FailoverGroup == null)
                {
                    updatedConnectionString = $"{SERVER_CONNECTION_OPTION_KEYWORD}{CONNECTION_DATA_VALUE_SEPARATOR}{updatedValue}{CONNECTION_DATA_KEY_SEPARATOR}{updatedConnectionString}";
                }
            }

            // DNS SRV Validation - Port cannot be provided by the user and multihost is not allowed if dns-srv is true
            if (isDnsSrv)
            {
                if (portProvided)
                {
                    throw new ArgumentException(Resources.DnsSrvInvalidConnOptionPort);
                }
                if (FailoverManager.FailoverGroup != null)
                {
                    throw new ArgumentException(Resources.DnsSrvInvalidConnOptionMultihost);
                }
            }

            // Default port must be added if not provided by the user.
            if (FailoverManager.FailoverGroup == null)
            {
                return(portProvided ? updatedConnectionString : $"{updatedConnectionString}{CONNECTION_DATA_KEY_SEPARATOR}{PORT_CONNECTION_OPTION_KEYWORD}{CONNECTION_DATA_VALUE_SEPARATOR}{X_PROTOCOL_DEFAULT_PORT}");
            }

            return($"{SERVER_CONNECTION_OPTION_KEYWORD}{CONNECTION_DATA_VALUE_SEPARATOR}{FailoverManager.FailoverGroup.ActiveHost.Host}{CONNECTION_DATA_KEY_SEPARATOR}" +
                   (!portProvided ? $"{PORT_CONNECTION_OPTION_KEYWORD}{CONNECTION_DATA_VALUE_SEPARATOR}{X_PROTOCOL_DEFAULT_PORT}{CONNECTION_DATA_KEY_SEPARATOR}" : string.Empty) +
                   updatedConnectionString);
        }
Пример #2
0
        /// <summary>
        /// Initializes a new instance of the BaseSession class based on the specified anonymous type object.
        /// </summary>
        /// <param name="connectionData">The connection data as an anonymous type used to create the session.</param>
        /// <param name="client">A <see cref="Client"/> object.</param>
        /// <exception cref="ArgumentNullException"><paramref name="connectionData"/> is null.</exception>
        /// <remarks>
        /// <para>Multiple hosts can be specified as part of the <paramref name="connectionData"/>, which enables client-side failover when trying to
        /// establish a connection.</para>
        /// <para>&#160;</para>
        /// <para>To assign multiple hosts, create a property similar to the connection string examples shown in
        /// <see cref="BaseSession(string)"/>. Note that the value of the property must be a string.
        /// </para>
        /// </remarks>
        internal BaseSession(object connectionData, Client client = null) : this()
        {
            if (connectionData == null)
            {
                throw new ArgumentNullException("connectionData");
            }

            _client = client;
            if (client == null)
            {
                FailoverManager.Reset();
            }

            var values = Tools.GetDictionaryFromAnonymous(connectionData);

            if (!values.Keys.Any(s => s.ToLowerInvariant() == PORT_CONNECTION_OPTION_KEYWORD))
            {
                values.Add(PORT_CONNECTION_OPTION_KEYWORD, X_PROTOCOL_DEFAULT_PORT);
            }

            bool hostsParsed = false;

            foreach (var value in values)
            {
                if (!Settings.ContainsKey(value.Key))
                {
                    throw new KeyNotFoundException(string.Format(ResourcesX.InvalidConnectionStringAttribute, value.Key));
                }

                Settings.SetValue(value.Key, value.Value);
                if (!hostsParsed && !string.IsNullOrEmpty(Settings[SERVER_CONNECTION_OPTION_KEYWORD].ToString()))
                {
                    var server = value.Value.ToString();
                    if (IsUnixSocket(server))
                    {
                        Settings.SetValue(value.Key, server = NormalizeUnixSocket(server));
                    }

                    FailoverManager.ParseHostList(server, true, false);
                    if (FailoverManager.FailoverGroup != null && FailoverManager.FailoverGroup.Hosts?.Count > 1)
                    {
                        Settings[SERVER_CONNECTION_OPTION_KEYWORD] = null;
                    }
                    else if (FailoverManager.FailoverGroup != null)
                    {
                        Settings[SERVER_CONNECTION_OPTION_KEYWORD] = FailoverManager.FailoverGroup.Hosts[0].Host;
                    }

                    hostsParsed = true;
                }
            }
            this._connectionString = Settings.ToString();

            Settings.AnalyzeConnectionString(this._connectionString, true, _isDefaultPort);
            if (FailoverManager.FailoverGroup != null && FailoverManager.FailoverGroup.Hosts?.Count > 1)
            {
                // Multiple hosts were specified.
                _internalSession          = FailoverManager.AttemptConnectionXProtocol(this._connectionString, out this._connectionString, _isDefaultPort, client);
                Settings.ConnectionString = _connectionString;
            }
            else
            {
                if (Settings.DnsSrv)
                {
                    var dnsSrvRecords = DnsResolver.GetDnsSrvRecords(Settings.Server);
                    FailoverManager.SetHostList(dnsSrvRecords.ConvertAll(r => new FailoverServer(r.Target, r.Port, null)),
                                                FailoverMethod.Sequential);
                    _internalSession          = FailoverManager.AttemptConnectionXProtocol(this._connectionString, out this._connectionString, _isDefaultPort, client);
                    Settings.ConnectionString = this._connectionString;
                }
                else
                {
                    _internalSession = InternalSession.GetSession(Settings);
                }
            }

            if (!string.IsNullOrWhiteSpace(Settings.Database))
            {
                DefaultSchema = GetSchema(Settings.Database);
            }
        }
Пример #3
0
        /// <summary>
        /// Parses a connection URI.
        /// </summary>
        /// <param name="connectionUri">The connection URI to parse.</param>
        /// <returns>The connection string representation of the provided <paramref name="connectionUri"/>.</returns>
        private string ParseConnectionUri(string connectionUri)
        {
            Uri    uri        = null;
            string updatedUri = null;
            bool   parseServerAsUnixSocket = false;
            string hierPart = null;

            try
            {
                uri = new Uri(connectionUri);
            }
            catch (UriFormatException ex)
            {
                if (ex.Message != "Invalid URI: The hostname could not be parsed.")
                {
                    throw ex;
                }

                // Identify if multiple hosts were specified.
                string[] splitUri = connectionUri.Split('@', '?');
                if (splitUri.Length == 1)
                {
                    throw ex;
                }

                hierPart = splitUri[1];
                var schema = string.Empty;
                parseServerAsUnixSocket = IsUnixSocket(hierPart);
                bool isArray = hierPart.StartsWith("[") && hierPart.Contains("]");

                // Remove schema.
                if ((!parseServerAsUnixSocket && hierPart.Contains("/")) && !isArray ||
                    (parseServerAsUnixSocket && hierPart.Contains(")/")) ||
                    (hierPart.StartsWith("[") && hierPart.Contains("]/") && isArray))
                {
                    schema   = hierPart.Substring(hierPart.LastIndexOf('/') + 1);
                    hierPart = hierPart.Substring(0, hierPart.Length - schema.Length - 1);
                }

                if (parseServerAsUnixSocket)
                {
                    updatedUri = splitUri[0] + "@localhost" +
                                 (schema != string.Empty ? "/" + schema : string.Empty) +
                                 (splitUri.Length > 2 ? "?" + splitUri[2] : string.Empty);
                }
                else if (isArray)
                {
                    hierPart = hierPart.Substring(1, hierPart.Length - 2);
                    int hostCount = FailoverManager.ParseHostList(hierPart, true, true);
                    if (FailoverManager.FailoverGroup != null)
                    {
                        hierPart = FailoverManager.FailoverGroup.ActiveHost.Host;
                        parseServerAsUnixSocket = IsUnixSocket(FailoverManager.FailoverGroup.ActiveHost.Host);
                        updatedUri = splitUri[0] + "@" +
                                     (parseServerAsUnixSocket ? "localhost" : hierPart) +
                                     (FailoverManager.FailoverGroup.ActiveHost.Port != -1 ? ":" + FailoverManager.FailoverGroup.ActiveHost.Port : string.Empty) +
                                     (schema != string.Empty ? "/" + schema : string.Empty) +
                                     (splitUri.Length == 3 ? "?" + splitUri[2] : string.Empty);
                    }
                    else if (hostCount == 1)
                    {
                        updatedUri = splitUri[0] + "@" + hierPart +
                                     (schema != string.Empty ? "/" + schema : string.Empty) +
                                     (splitUri.Length == 3 ? "?" + splitUri[2] : string.Empty);
                    }
                    else
                    {
                        throw ex;
                    }
                }
            }

            if (uri == null)
            {
                uri = updatedUri == null ? new Uri(connectionUri) : new Uri(updatedUri);
            }

            if (uri.Scheme == DNS_SRV_URI_SCHEME)
            {
                if (FailoverManager.FailoverGroup != null && FailoverManager.FailoverGroup.Hosts?.Count > 1)
                {
                    throw new ArgumentException(Resources.DnsSrvInvalidConnOptionMultihost);
                }
                if (!uri.IsDefaultPort)
                {
                    throw new ArgumentException(Resources.DnsSrvInvalidConnOptionPort);
                }
                if (parseServerAsUnixSocket)
                {
                    throw new ArgumentException(Resources.DnsSrvInvalidConnOptionUnixSocket);
                }
            }
            else if (uri.Scheme != MYSQLX_URI_SCHEME && uri.Scheme != SSH_URI_SCHEME)
            {
                throw new ArgumentException(string.Format(ResourcesX.DnsSrvInvalidScheme, uri.Scheme));
            }

            return(ConvertToConnectionString(uri, hierPart, parseServerAsUnixSocket, uri.Scheme == DNS_SRV_URI_SCHEME));
        }
Пример #4
0
        /// <include file='docs/MySqlConnection.xml' path='docs/Open/*'/>
        public override void Open()
        {
            if (State == ConnectionState.Open)
            {
                Throw(new InvalidOperationException(Resources.ConnectionAlreadyOpen));
            }

            // start up our interceptors
            _exceptionInterceptor = new ExceptionInterceptor(this);
            commandInterceptor    = new CommandInterceptor(this);

            SetState(ConnectionState.Connecting, true);

            AssertPermissions();

            //TODO: SUPPORT FOR 452 AND 46X
            // if we are auto enlisting in a current transaction, then we will be
            // treating the connection as pooled
            if (Settings.AutoEnlist && Transaction.Current != null)
            {
                driver = DriverTransactionManager.GetDriverInTransaction(Transaction.Current);
                if (driver != null &&
                    (driver.IsInActiveUse ||
                     !driver.Settings.EquivalentTo(this.Settings)))
                {
                    Throw(new NotSupportedException(Resources.MultipleConnectionsInTransactionNotSupported));
                }
            }

            MySqlConnectionStringBuilder currentSettings = Settings;

            try
            {
                if (Settings.ConnectionProtocol == MySqlConnectionProtocol.Tcp && Settings.IsSshEnabled())
                {
                    _sshHandler = new Ssh(
                        Settings.SshHostName,
                        Settings.SshUserName,
                        Settings.SshPassword,
                        Settings.SshKeyFile,
                        Settings.SshPassphrase,
                        Settings.SshPort,
                        Settings.Server,
                        Settings.Port,
                        false
                        );
                    _sshHandler.StartClient();
                }

                if (!Settings.Pooling || MySqlPoolManager.Hosts == null)
                {
                    FailoverManager.Reset();

                    if (Settings.DnsSrv)
                    {
                        var dnsSrvRecords = DnsResolver.GetDnsSrvRecords(Settings.Server);
                        FailoverManager.SetHostList(dnsSrvRecords.ConvertAll(r => new FailoverServer(r.Target, r.Port, null)),
                                                    FailoverMethod.Sequential);
                    }
                    else
                    {
                        FailoverManager.ParseHostList(Settings.Server, false);
                    }
                }

                // Load balancing && Failover
                if (ReplicationManager.IsReplicationGroup(Settings.Server))
                {
                    if (driver == null)
                    {
                        ReplicationManager.GetNewConnection(Settings.Server, false, this);
                    }
                    else
                    {
                        currentSettings = driver.Settings;
                    }
                }
                else if (FailoverManager.FailoverGroup != null && !Settings.Pooling)
                {
                    FailoverManager.AttemptConnection(this, Settings.ConnectionString, out string connectionString);
                    currentSettings.ConnectionString = connectionString;
                }

                if (Settings.Pooling)
                {
                    if (FailoverManager.FailoverGroup != null)
                    {
                        FailoverManager.AttemptConnection(this, Settings.ConnectionString, out string connectionString, true);
                        currentSettings.ConnectionString = connectionString;
                    }

                    MySqlPool pool = MySqlPoolManager.GetPool(currentSettings);
                    if (driver == null || !driver.IsOpen)
                    {
                        driver = pool.GetConnection();
                    }
                    ProcedureCache = pool.ProcedureCache;
                }
                else
                {
                    if (driver == null || !driver.IsOpen)
                    {
                        driver = Driver.Create(currentSettings);
                    }
                    ProcedureCache = new ProcedureCache((int)Settings.ProcedureCacheSize);
                }
            }
            catch (Exception)
            {
                SetState(ConnectionState.Closed, true);
                throw;
            }

            SetState(ConnectionState.Open, false);
            driver.Configure(this);

            if (driver.IsPasswordExpired && Settings.Pooling)
            {
                MySqlPoolManager.ClearPool(currentSettings);
            }

            if (!(driver.SupportsPasswordExpiration && driver.IsPasswordExpired))
            {
                if (!string.IsNullOrEmpty(Settings.Database))
                {
                    ChangeDatabase(Settings.Database);
                }
            }

            // setup our schema provider
            _schemaProvider = new ISSchemaProvider(this);
            PerfMonitor     = new PerformanceMonitor(this);

            // if we are opening up inside a current transaction, then autoenlist
            // TODO: control this with a connection string option
            if (Transaction.Current != null && Settings.AutoEnlist)
            {
                EnlistTransaction(Transaction.Current);
            }

            hasBeenOpen = true;
            SetState(ConnectionState.Open, true);
        }