signmessagewidgets.cpp

Go to the documentation of this file.

// Copyright (c) 2019-2022 The PIVX Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#include "qt/settings/signmessagewidgets.h"
#include "qt/settings/forms/ui_signmessagewidgets.h"
 
#include "key_io.h"
#include "messagesigner.h"
#include "qt/askpassphrasedialog.h"
#include "qt/walletmodel.h"
#include "qtutils.h"
#include "util/validation.h"
 
#include <string>
#include <vector>
 
#include <QClipboard>
 
SettingsSignMessageWidgets::SettingsSignMessageWidgets(PIVXGUI* _window, QWidget *parent) :
    PWidget(_window, parent),
    ui(new Ui::SettingsSignMessageWidgets)
{
    ui->setupUi(this);
 
    this->setStyleSheet(parent->styleSheet());
 
    // Containers
    ui->left->setProperty("cssClass", "container");
    ui->left->setContentsMargins(10,10,10,10);
 
    // Title
    ui->labelTitle->setProperty("cssClass", "text-title-screen");
    ui->labelSubtitle1->setProperty("cssClass", "text-subtitle");
 
    // Address
    ui->labelSubtitleAddress->setProperty("cssClass", "text-title");
    ui->addressIn_SM->setProperty("cssClass", "edit-primary-multi-book");
    ui->addressIn_SM->setAttribute(Qt::WA_MacShowFocusRect, 0);
    setShadow(ui->addressIn_SM);
 
    /* Button Group */
    setCssProperty(ui->pushSign, "btn-check-right");
    setCssProperty(ui->pushVerify, "btn-check-right");
    ui->labelSubtitleSwitch->setText(tr("Select mode"));
    setCssProperty(ui->labelSubtitleSwitch, "text-subtitle");
    ui->pushSign->setChecked(true);
    updateMode();
 
    // Message
    ui->labelSubtitleMessage->setProperty("cssClass", "text-title");
    ui->messageIn_SM->setProperty("cssClass","edit-primary");
    setShadow(ui->messageIn_SM);
    ui->messageIn_SM->setAttribute(Qt::WA_MacShowFocusRect, 0);
 
    // Signature
    ui->labelSubtitleSignature->setProperty("cssClass", "text-title");
    ui->signatureOut_SM->setAttribute(Qt::WA_MacShowFocusRect, 0);
 
    initCssEditLine(ui->signatureOut_SM);
    setShadow(ui->signatureOut_SM);
 
    // Buttons
    btnContact = ui->addressIn_SM->addAction(QIcon("://ic-contact-arrow-down"), QLineEdit::TrailingPosition);
 
    setCssBtnPrimary(ui->pushButtonSave);
    setCssBtnSecondary(ui->pushButtonClear);
 
    ui->statusLabel_SM->setStyleSheet("QLabel { color: transparent; }");
 
    connect(ui->pushButtonSave, &QPushButton::clicked, this, &SettingsSignMessageWidgets::onGoClicked);
    connect(btnContact, &QAction::triggered, this, &SettingsSignMessageWidgets::onAddressesClicked);
    connect(ui->pushButtonClear, &QPushButton::clicked, this, &SettingsSignMessageWidgets::onClearAll);
    connect(ui->pushSign, &QPushButton::clicked, [this](){onModeSelected(true);});
    connect(ui->pushVerify,  &QPushButton::clicked, [this](){onModeSelected(false);});
}
 
SettingsSignMessageWidgets::~SettingsSignMessageWidgets()
{
    delete ui;
}
 
void SettingsSignMessageWidgets::showEvent(QShowEvent *event)
{
    if (ui->addressIn_SM) ui->addressIn_SM->setFocus();
}
 
void SettingsSignMessageWidgets::onModeSelected(bool isSign)
{
    this->isSign = isSign;
    updateMode();
}
 
void SettingsSignMessageWidgets::onGoClicked()
{
    if (isSign) {
        onSignMessageButtonSMClicked();
    } else {
        onVerifyMessage();
    }
}
 
void SettingsSignMessageWidgets::updateMode()
{
    QString subtitle;
    QString go;
    if (isSign) {
        subtitle = tr("You can sign messages with your addresses to prove you own them. Be careful not to sign anything vague, as phishing attacks may try to trick you into signing your identity over to them. Only sign fully-detailed statements you agree to.");
        go = tr("SIGN");
        ui->signatureOut_SM->setReadOnly(true);
        ui->signatureOut_SM->clear();
    } else {
        subtitle = tr("Enter the signing address, message (ensure you copy line breaks, spaces, tabs, etc. exactly) and signature below to verify the message. Be careful not to read more into the signature than what is in the signed message itself, to avoid being tricked by a man-in-the-middle attack.");
        go = tr("VERIFY");
        ui->signatureOut_SM->setReadOnly(false);
    }
    ui->labelSubtitle1->setText(subtitle);
    ui->pushButtonSave->setText(go);
}
 
void SettingsSignMessageWidgets::setAddress_SM(const QString& address)
{
    ui->addressIn_SM->setText(address);
    ui->messageIn_SM->setFocus();
}
 
void SettingsSignMessageWidgets::onClearAll()
{
    ui->addressIn_SM->clear();
    ui->signatureOut_SM->clear();
    ui->messageIn_SM->clear();
    ui->statusLabel_SM->setStyleSheet("QLabel { color: transparent; }");
}
 
void SettingsSignMessageWidgets::onSignMessageButtonSMClicked()
{
    if (!walletModel)
        return;
 
    /* Clear old signature to ensure users don't get confused on error with an old signature displayed */
    ui->signatureOut_SM->clear();
 
    CTxDestination addr = DecodeDestination(ui->addressIn_SM->text().toStdString());
    if (!IsValidDestination(addr)) {
        ui->statusLabel_SM->setStyleSheet("QLabel { color: red; }");
        ui->statusLabel_SM->setText(tr("The entered address is invalid.") + QString(" ") + tr("Please check the address and try again."));
        return;
    }
    const CKeyID* keyID = boost::get<CKeyID>(&addr);
    if (!keyID) {
        // TODO: change css..
        //ui->addressIn_SM->setValid(false);
        ui->statusLabel_SM->setStyleSheet("QLabel { color: red; }");
        ui->statusLabel_SM->setText(tr("The entered address does not refer to a key.") + QString(" ") + tr("Please check the address and try again."));
        return;
    }
 
    WalletModel::UnlockContext ctx(walletModel->requestUnlock());
    if (!ctx.isValid()) {
        ui->statusLabel_SM->setStyleSheet("QLabel { color: red; }");
        ui->statusLabel_SM->setText(tr("Wallet unlock was cancelled."));
        return;
    }
 
    CKey key;
    if (!walletModel->getKey(*keyID, key)) {
        ui->statusLabel_SM->setStyleSheet("QLabel { color: red; }");
        ui->statusLabel_SM->setText(tr("Private key for the entered address is not available."));
        return;
    }
 
    const std::string& message = ui->messageIn_SM->document()->toPlainText().toStdString();
 
    std::vector<unsigned char> vchSig;
    if (!CMessageSigner::SignMessage(message, vchSig, key)) {
        ui->statusLabel_SM->setStyleSheet("QLabel { color: red; }");
        ui->statusLabel_SM->setText(QString("<nobr>") + tr("Message signing failed.") + QString("</nobr>"));
        return;
    }
 
    ui->statusLabel_SM->setStyleSheet("QLabel { color: green; }");
    ui->statusLabel_SM->setText(QString("<nobr>") + tr("Message signed.") + QString("</nobr>"));
    ui->signatureOut_SM->setText(QString::fromStdString(EncodeBase64(vchSig)));
}
 
void SettingsSignMessageWidgets::onVerifyMessage()
{
    CTxDestination addr = DecodeDestination(ui->addressIn_SM->text().toStdString());
    if (!IsValidDestination(addr)) {
        ui->statusLabel_SM->setStyleSheet("QLabel { color: red; }");
        ui->statusLabel_SM->setText(tr("The entered address is invalid.") + QString(" ") + tr("Please check the address and try again."));
        return;
    }
    const CKeyID* keyID = boost::get<CKeyID>(&addr);
    if (!keyID) {
        //ui->addressIn_SM->setValid(false);
        ui->statusLabel_SM->setStyleSheet("QLabel { color: red; }");
        ui->statusLabel_SM->setText(tr("The entered address does not refer to a key.") + QString(" ") + tr("Please check the address and try again."));
        return;
    }
 
    bool fInvalid = false;
    std::vector<unsigned char> vchSig = DecodeBase64(ui->signatureOut_SM->text().toStdString().c_str(), &fInvalid);
 
    if (fInvalid) {
        //ui->signatureOut_SM->setValid(false);
        ui->signatureOut_SM->setStyleSheet("QLabel { color: red; }");
        ui->signatureOut_SM->setText(tr("The signature could not be decoded.") + QString(" ") + tr("Please check the signature and try again."));
        return;
    }
 
    const std::string& message = ui->messageIn_SM->document()->toPlainText().toStdString();
 
    std::string err_log;
    if (!CMessageSigner::VerifyMessage(*keyID, vchSig, message, err_log)) {
        ui->statusLabel_SM->setStyleSheet("QLabel { color: red; }");
        ui->statusLabel_SM->setText(QString("<nobr>") + tr("Message verification failed.") + QString("</nobr>"));
        return;
    }
 
    ui->statusLabel_SM->setStyleSheet("QLabel { color: green; }");
    ui->statusLabel_SM->setText(QString("<nobr>") + tr("Message verified.") + QString("</nobr>"));
}
 
void SettingsSignMessageWidgets::onAddressesClicked()
{
    int addressSize = walletModel->getAddressTableModel()->sizeRecv();
    if (addressSize == 0) {
        inform(tr("No addresses available, you can go to the receive screen and add some there!"));
        return;
    }
 
    int height = (addressSize <= 2) ? ui->addressIn_SM->height() * ( 2 * (addressSize + 1 )) : ui->addressIn_SM->height() * 4;
    int width = ui->containerAddress->width();
 
    if (!menuContacts) {
        menuContacts = new ContactsDropdown(
                width,
                height,
                this
        );
        menuContacts->setWalletModel(walletModel, {AddressTableModel::Receive});
        connect(menuContacts, &ContactsDropdown::contactSelected, [this](QString address, QString label){
            setAddress_SM(address);
        });
 
    }
 
    if (menuContacts->isVisible()) {
        menuContacts->hide();
        return;
    }
 
    menuContacts->resizeList(width, height);
    menuContacts->setStyleSheet(this->styleSheet());
    menuContacts->adjustSize();
 
    QPoint pos = ui->container_sign->mapToParent(ui->containerAddress->rect().bottomLeft());
    pos.setY(pos.y() + (ui->containerAddress->height() * 1.4) - 10);
    menuContacts->move(pos);
    menuContacts->show();
}
 
void SettingsSignMessageWidgets::resizeMenu()
{
    if (menuContacts && menuContacts->isVisible()) {
        int width = ui->containerAddress->width();
        menuContacts->resizeList(width, menuContacts->height());
        menuContacts->resize(width, menuContacts->height());
        QPoint pos = ui->container_sign->mapToParent(ui->containerAddress->rect().bottomLeft());
        pos.setY(pos.y() + (ui->containerAddress->height() * 1.4) - 10);
        menuContacts->move(pos);
    }
}
 
void SettingsSignMessageWidgets::resizeEvent(QResizeEvent *event)
{
    resizeMenu();
    QWidget::resizeEvent(event);
}