ParameterOperate.cpp

// Author: Kang Lin <kl222@126.com>
 
#include <QLoggingCategory>
#include <QCryptographicHash>
#include <QInputDialog>
 
#include "RabbitCommonEncrypt.h"
#include "RabbitCommonTools.h"
#include "DlgInputPassword.h"
#include "ParameterOperate.h"
 
static Q_LOGGING_CATEGORY(log, "Parameter.Operate")
 
CParameterOperate::CParameterOperate(QObject *parent, const QString &szPrefix)
    : CParameter(parent, szPrefix)
    , m_Parent(nullptr)
    , m_pParameterPlugin(nullptr)
{
    bool check = false;
    check = connect(this, SIGNAL(sigSetGlobalParameters()),
                    this, SLOT(slotSetGlobalParameters()));
    Q_ASSERT(check);
    CParameterOperate* p = qobject_cast<CParameterOperate*>(parent);
    if(p) {
        m_Parent = p;
        check = connect(m_Parent, SIGNAL(sigSetGlobalParameters()),
                        this, SIGNAL(sigSetGlobalParameters()));
        Q_ASSERT(check);
    }
}
 
CParameterPlugin* CParameterOperate::GetGlobalParameters()
{
    if(m_Parent) return m_Parent->GetGlobalParameters();
    return m_pParameterPlugin;
}
 
int CParameterOperate::SetGlobalParameters(CParameterPlugin *p)
{
    if(m_pParameterPlugin == p) return 0;
    m_pParameterPlugin = p;
    emit sigSetGlobalParameters();
    return 0;
}
 
void CParameterOperate::slotSetGlobalParameters()
{
    return;
}
 
QByteArray CParameterOperate::PasswordSum(const std::string &password,
                                          const std::string &key)
{
    std::string pw = "RabbitRemoteControl";
    QCryptographicHash sum(QCryptographicHash::Md5);
 
#if QT_VERSION < QT_VERSION_CHECK(6, 4, 0)
    if(!password.empty())
        sum.addData(password.c_str(), password.length());
    sum.addData(pw.c_str(), pw.length());
    if(!key.empty())
        sum.addData(key.c_str(), key.length());
#else
    if(!password.empty())
        sum.addData(QByteArrayView(password.c_str(), password.length()));
    sum.addData(QByteArrayView(pw.c_str(), pw.length()));
    if(!key.empty())
        sum.addData(QByteArrayView(key.c_str(), key.length()));
#endif
    return sum.result();
}
 
int CParameterOperate::LoadPassword(const QString &szTitle,
                                    const QString &szKey,
                                    QString &password,
                                    QSettings &set)
{
    QByteArray sum = set.value(szKey + "_sum").toByteArray();
    QByteArray pwByte = set.value(szKey).toByteArray();
    if(pwByte.isEmpty())
        return 0;
    RabbitCommon::CEncrypt e;
 
    std::string key;
    if(GetGlobalParameters())
        key = GetGlobalParameters()->GetEncryptKey().toStdString().c_str();
    if(!key.empty())
        e.SetPassword(key.c_str());
 
    if(!e.Dencode(pwByte, password)
            && PasswordSum(password.toStdString(), key) == sum)
            return 0;
 
    qDebug(log) << "Password don't dencode or sum is error";
    CDlgInputPassword d(GetGlobalParameters()->GetViewPassowrd(), szTitle);
    if(QDialog::Accepted != d.exec())
    {
        return -1;
    }
 
    CDlgInputPassword::InputType t;
    int nRet = d.GetValue(t, password);
    if(nRet) return nRet;
    if(CDlgInputPassword::InputType::Password == t)
        return 0;
    GetGlobalParameters()->SetEncryptKey(password);
    return LoadPassword(szTitle, szKey, password, set);
}
 
int CParameterOperate::SavePassword(const QString &szKey,
                                    const QString &password,
                                    QSettings &set, bool bSave)
{
    if(!bSave)
    {
        set.remove(szKey);
        set.remove(szKey + "_sum");
        return 0;
    }
 
    QByteArray encryptPassword;
    RabbitCommon::CEncrypt e;
    std::string key = GetGlobalParameters()->GetEncryptKey().toStdString();
    if(key.empty())
    {
        switch (GetGlobalParameters()->GetPromptType()) {
        case CParameterPlugin::PromptType::First:
        {
            int nCount = GetGlobalParameters()->GetPromptCount();
            if(nCount >= 1)
                break;
            GetGlobalParameters()->SetPromptCount(nCount + 1);
        }
        case CParameterPlugin::PromptType::Always:
        {
            QString szKey;
            CDlgInputPassword::InputType t = CDlgInputPassword::InputType::Encrypt;
            CDlgInputPassword dlg(GetGlobalParameters()->GetViewPassowrd());
            if(QDialog::Accepted == dlg.exec())
                dlg.GetValue(t, szKey);
            if(CDlgInputPassword::InputType::Encrypt == t)
                GetGlobalParameters()->SetEncryptKey(szKey);
            break;
        }
        case CParameterPlugin::PromptType::No:
            break;
        }
    } else
        e.SetPassword(key.c_str());
    if(password.isEmpty())
        return 0;
    e.Encode(password, encryptPassword);
    set.setValue(szKey, encryptPassword);
    set.setValue(szKey + "_sum", PasswordSum(password.toStdString(), key));
    return 0;
}