You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
232 lines
6.4 KiB
232 lines
6.4 KiB
#if MYSQL_6_9
|
|
|
|
// Copyright © 2012, Oracle and/or its affiliates. All rights reserved.
|
|
//
|
|
// MySQL Connector/NET is licensed under the terms of the GPLv2
|
|
// <http://www.gnu.org/licenses/old-licenses/gpl-2.0.html>, like most
|
|
// MySQL Connectors. There are special exceptions to the terms and
|
|
// conditions of the GPLv2 as it is applied to this software, see the
|
|
// FLOSS License Exception
|
|
// <http://www.mysql.com/about/legal/licensing/foss-exception.html>.
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published
|
|
// by the Free Software Foundation; version 2 of the License.
|
|
//
|
|
// This program is distributed in the hope that it will be useful, but
|
|
// WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
// or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
// for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License along
|
|
// with this program; if not, write to the Free Software Foundation, Inc.,
|
|
// 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
using System.IO;
|
|
using System;
|
|
using Externals.MySql.Data.MySqlClient.Properties;
|
|
using Externals.MySql.Data.Common;
|
|
using System.Text;
|
|
using System.Diagnostics;
|
|
|
|
namespace Externals.MySql.Data.MySqlClient.Authentication
|
|
{
|
|
internal abstract class MySqlAuthenticationPlugin
|
|
{
|
|
private NativeDriver driver;
|
|
protected byte[] AuthenticationData;
|
|
|
|
/// <summary>
|
|
/// This is a factory method that is used only internally. It creates an auth plugin based on the method type
|
|
/// </summary>
|
|
internal static MySqlAuthenticationPlugin GetPlugin(string method, NativeDriver driver, byte[] authData)
|
|
{
|
|
if (method == "mysql_old_password")
|
|
{
|
|
driver.Close(true);
|
|
throw new MySqlException(Resources.OldPasswordsNotSupported);
|
|
}
|
|
MySqlAuthenticationPlugin plugin = AuthenticationPluginManager.GetPlugin(method);
|
|
if (plugin == null)
|
|
throw new MySqlException(String.Format(Resources.UnknownAuthenticationMethod, method));
|
|
|
|
plugin.driver = driver;
|
|
plugin.SetAuthData(authData);
|
|
return plugin;
|
|
}
|
|
|
|
protected MySqlConnectionStringBuilder Settings
|
|
{
|
|
get { return driver.Settings; }
|
|
}
|
|
|
|
protected Version ServerVersion
|
|
{
|
|
get { return new Version(driver.Version.Major, driver.Version.Minor, driver.Version.Build); }
|
|
}
|
|
|
|
internal ClientFlags Flags
|
|
{
|
|
get { return driver.Flags; }
|
|
}
|
|
|
|
protected Encoding Encoding
|
|
{
|
|
get { return driver.Encoding; }
|
|
}
|
|
|
|
protected virtual void SetAuthData(byte[] data)
|
|
{
|
|
AuthenticationData = data;
|
|
}
|
|
|
|
protected virtual void CheckConstraints()
|
|
{
|
|
}
|
|
|
|
protected virtual void AuthenticationFailed(Exception ex)
|
|
{
|
|
string msg = String.Format(Resources.AuthenticationFailed, Settings.Server, GetUsername(), PluginName, ex.Message);
|
|
throw new MySqlException(msg, ex);
|
|
}
|
|
|
|
protected virtual void AuthenticationSuccessful()
|
|
{
|
|
}
|
|
|
|
protected virtual byte[] MoreData(byte[] data)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
internal void Authenticate(bool reset)
|
|
{
|
|
CheckConstraints();
|
|
|
|
MySqlPacket packet = driver.Packet;
|
|
|
|
// send auth response
|
|
packet.WriteString(GetUsername());
|
|
|
|
// now write the password
|
|
WritePassword(packet);
|
|
|
|
if ((Flags & ClientFlags.CONNECT_WITH_DB) != 0 || reset)
|
|
{
|
|
if (!String.IsNullOrEmpty(Settings.Database))
|
|
packet.WriteString(Settings.Database);
|
|
}
|
|
|
|
if (reset)
|
|
packet.WriteInteger(8, 2);
|
|
|
|
if ((Flags & ClientFlags.PLUGIN_AUTH) != 0)
|
|
packet.WriteString(PluginName);
|
|
|
|
driver.SetConnectAttrs();
|
|
driver.SendPacket(packet);
|
|
//read server response
|
|
packet = ReadPacket();
|
|
byte[] b = packet.Buffer;
|
|
if (b[0] == 0xfe)
|
|
{
|
|
if (packet.IsLastPacket)
|
|
{
|
|
driver.Close(true);
|
|
throw new MySqlException( Resources.OldPasswordsNotSupported );
|
|
}
|
|
else
|
|
{
|
|
HandleAuthChange(packet);
|
|
}
|
|
}
|
|
driver.ReadOk(false);
|
|
AuthenticationSuccessful();
|
|
}
|
|
|
|
private void WritePassword(MySqlPacket packet)
|
|
{
|
|
bool secure = (Flags & ClientFlags.SECURE_CONNECTION) != 0;
|
|
object password = GetPassword();
|
|
if (password is string)
|
|
{
|
|
if (secure)
|
|
packet.WriteLenString((string)password);
|
|
else
|
|
packet.WriteString((string)password);
|
|
}
|
|
else if (password == null)
|
|
packet.WriteByte(0);
|
|
else if (password is byte[])
|
|
packet.Write(password as byte[]);
|
|
else throw new MySqlException("Unexpected password format: " + password.GetType());
|
|
}
|
|
|
|
private MySqlPacket ReadPacket()
|
|
{
|
|
try
|
|
{
|
|
MySqlPacket p = driver.ReadPacket();
|
|
return p;
|
|
}
|
|
catch (MySqlException ex)
|
|
{
|
|
// make sure this is an auth failed ex
|
|
AuthenticationFailed(ex);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
private void HandleAuthChange(MySqlPacket packet)
|
|
{
|
|
byte b = packet.ReadByte();
|
|
Debug.Assert(b == 0xfe);
|
|
|
|
string method = packet.ReadString();
|
|
byte[] authData = new byte[packet.Length - packet.Position];
|
|
Array.Copy(packet.Buffer, packet.Position, authData, 0, authData.Length);
|
|
|
|
MySqlAuthenticationPlugin plugin = MySqlAuthenticationPlugin.GetPlugin(method, driver, authData);
|
|
plugin.AuthenticationChange();
|
|
}
|
|
|
|
private void AuthenticationChange()
|
|
{
|
|
MySqlPacket packet = driver.Packet;
|
|
packet.Clear();
|
|
byte[] moreData = MoreData(null);
|
|
while (moreData != null && moreData.Length > 0)
|
|
{
|
|
packet.Clear();
|
|
packet.Write(moreData);
|
|
driver.SendPacket(packet);
|
|
|
|
packet = ReadPacket();
|
|
byte prefixByte = packet.Buffer[0];
|
|
if (prefixByte != 1) return;
|
|
|
|
// a prefix of 0x01 means need more auth data
|
|
byte[] responseData = new byte[packet.Length - 1];
|
|
Array.Copy(packet.Buffer, 1, responseData, 0, responseData.Length);
|
|
moreData = MoreData(responseData);
|
|
}
|
|
// we get here if MoreData returned null but the last packet read was a more data packet
|
|
ReadPacket();
|
|
}
|
|
|
|
public abstract string PluginName { get; }
|
|
|
|
public virtual string GetUsername()
|
|
{
|
|
return Settings.UserID;
|
|
}
|
|
|
|
public virtual object GetPassword()
|
|
{
|
|
return null;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#endif
|
|
|