UNIVPLMDataIntegration/Admin.NET/Admin.NET.Core/Mqtt/DefaultMqttEventInterceptor.cs

// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
//
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
//
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！

using MQTTnet.Protocol;
using MQTTnet.Server;

namespace Admin.NET.Core;

/// <summary>
/// 默认 MQTT 事件拦截器
/// </summary>
public class DefaultMqttEventInterceptor : MqttEventInterceptor
{
    public new int Order = int.MinValue;

    public override Task ValidatingConnectionAsync(ValidatingConnectionEventArgs arg)
    {
        var _db = App.GetRequiredService<ISqlSugarClient>();
        // 验证账号
        var user = _db.Queryable<SysUser>().First(u => u.Account == arg.UserName);
        if (user == null)
        {
            arg.ReasonCode = MqttConnectReasonCode.BadUserNameOrPassword;
            Logging($"客户端验证：客户端ID=【{arg.ClientId}】用户名不存在 {DateTime.Now} ");
            return Task.CompletedTask;
        }

        // 验证密码
        var password = arg.Password;
        if (CryptogramHelper.CryptoType == CryptogramEnum.MD5.ToString())
        {
            if (user.Password.Equals(MD5Encryption.Encrypt(password)))
                return Task.CompletedTask;
        }
        else
        {
            if (CryptogramHelper.Decrypt(user.Password).Equals(password))
                return Task.CompletedTask;
        }
        arg.ReasonCode = MqttConnectReasonCode.BadUserNameOrPassword;
        Logging($"客户端验证：客户端ID=【{arg.ClientId}】用户名或密码验证错误 {DateTime.Now} ");
        return Task.CompletedTask;
    }
}