src/qt/askpassphrasedialog.cpp

   1 #include "askpassphrasedialog.h"
   2 #include "ui_askpassphrasedialog.h"
   3
   4 #include "guiconstants.h"
   5 #include "walletmodel.h"
   6
   7 #include <QMessageBox>
   8 #include <QPushButton>
   9
  10 AskPassphraseDialog::AskPassphraseDialog(Mode mode, QWidget *parent) :
  11     QDialog(parent),
  12     ui(new Ui::AskPassphraseDialog),
  13     mode(mode),
  14     model(0)
  15 {
  16     ui->setupUi(this);
  17     ui->passEdit1->setMaxLength(MAX_PASSPHRASE_SIZE);
  18     ui->passEdit2->setMaxLength(MAX_PASSPHRASE_SIZE);
  19     ui->passEdit3->setMaxLength(MAX_PASSPHRASE_SIZE);
  20
  21     switch(mode)
  22     {
  23         case Encrypt: // Ask passphrase x2
  24             ui->passLabel1->hide();
  25             ui->passEdit1->hide();
  26             ui->warningLabel->setText(tr("Enter the new passphrase to the wallet.<br/>Please use a passphrase of <b>10 or more random characters</b>, or <b>eight or more words</b>."));
  27             setWindowTitle(tr("Encrypt wallet"));
  28             break;
  29         case Unlock: // Ask passphrase
  30             ui->warningLabel->setText(tr("This operation needs your wallet passphrase to unlock the wallet."));
  31             ui->passLabel2->hide();
  32             ui->passEdit2->hide();
  33             ui->passLabel3->hide();
  34             ui->passEdit3->hide();
  35             setWindowTitle(tr("Unlock wallet"));
  36             break;
  37         case Decrypt:   // Ask passphrase
  38             ui->warningLabel->setText(tr("This operation needs your wallet passphrase to decrypt the wallet."));
  39             ui->passLabel2->hide();
  40             ui->passEdit2->hide();
  41             ui->passLabel3->hide();
  42             ui->passEdit3->hide();
  43             setWindowTitle(tr("Decrypt wallet"));
  44             break;
  45         case ChangePass: // Ask old passphrase + new passphrase x2
  46             setWindowTitle(tr("Change passphrase"));
  47             ui->warningLabel->setText(tr("Enter the old and new passphrase to the wallet."));
  48             break;
  49     }
  50     resize(minimumSize()); // Get rid of extra space in dialog
  51
  52     textChanged();
  53     connect(ui->passEdit1, SIGNAL(textChanged(QString)), this, SLOT(textChanged()));
  54     connect(ui->passEdit2, SIGNAL(textChanged(QString)), this, SLOT(textChanged()));
  55     connect(ui->passEdit3, SIGNAL(textChanged(QString)), this, SLOT(textChanged()));
  56 }
  57
  58 AskPassphraseDialog::~AskPassphraseDialog()
  59 {
  60     // Attempt to overwrite text so that they do not linger around in memory
  61     ui->passEdit1->setText(QString(" ").repeated(ui->passEdit1->text().size()));
  62     ui->passEdit2->setText(QString(" ").repeated(ui->passEdit2->text().size()));
  63     ui->passEdit3->setText(QString(" ").repeated(ui->passEdit3->text().size()));
  64     delete ui;
  65 }
  66
  67 void AskPassphraseDialog::setModel(WalletModel *model)
  68 {
  69     this->model = model;
  70 }
  71
  72 void AskPassphraseDialog::accept()
  73 {
  74     std::string oldpass, newpass1, newpass2;
  75     if(!model)
  76         return;
  77     // TODO: mlock memory / munlock on return so they will not be swapped out, really need "mlockedstring" wrapper class to do this safely
  78     oldpass.reserve(MAX_PASSPHRASE_SIZE);
  79     newpass1.reserve(MAX_PASSPHRASE_SIZE);
  80     newpass2.reserve(MAX_PASSPHRASE_SIZE);
  81     oldpass.assign(ui->passEdit1->text().toStdString());
  82     newpass1.assign(ui->passEdit2->text().toStdString());
  83     newpass2.assign(ui->passEdit3->text().toStdString());
  84
  85     switch(mode)
  86     {
  87     case Encrypt: {
  88         if(newpass1.empty() || newpass2.empty())
  89         {
  90             // Cannot encrypt with empty passphrase
  91             break;
  92         }
  93         QMessageBox::StandardButton retval = QMessageBox::question(this, tr("Confirm wallet encryption"),
  94                  tr("WARNING: If you encrypt your wallet and lose your passphrase, you will <b>LOSE ALL OF YOUR BITCOINS</b>!\nAre you sure you wish to encrypt your wallet?"),
  95                  QMessageBox::Yes|QMessageBox::Cancel,
  96                  QMessageBox::Cancel);
  97         if(retval == QMessageBox::Yes)
  98         {
  99             if(newpass1 == newpass2)
 100             {
 101                 if(model->setWalletEncrypted(true, newpass1))
 102                 {
 103                     QMessageBox::warning(this, tr("Wallet encrypted"),
 104                                          tr("Bitcoin will close now to finish the encryption process. Remember that encrypting your wallet cannot fully protect your bitcoins from being stolen by malware infecting your computer."));
 105                     QApplication::quit();
 106                 }
 107                 else
 108                 {
 109                     QMessageBox::critical(this, tr("Wallet encryption failed"),
 110                                          tr("Wallet encryption failed due to an internal error. Your wallet was not encrypted."));
 111                 }
 112                 QDialog::accept(); // Success
 113             }
 114             else
 115             {
 116                 QMessageBox::critical(this, tr("Wallet encryption failed"),
 117                                      tr("The supplied passphrases do not match."));
 118             }
 119         }
 120         else
 121         {
 122             QDialog::reject(); // Cancelled
 123         }
 124         } break;
 125     case Unlock:
 126         if(!model->setWalletLocked(false, oldpass))
 127         {
 128             QMessageBox::critical(this, tr("Wallet unlock failed"),
 129                                   tr("The passphrase entered for the wallet decryption was incorrect."));
 130         }
 131         else
 132         {
 133             QDialog::accept(); // Success
 134         }
 135         break;
 136     case Decrypt:
 137         if(!model->setWalletEncrypted(false, oldpass))
 138         {
 139             QMessageBox::critical(this, tr("Wallet decryption failed"),
 140                                   tr("The passphrase entered for the wallet decryption was incorrect."));
 141         }
 142         else
 143         {
 144             QDialog::accept(); // Success
 145         }
 146         break;
 147     case ChangePass:
 148         if(newpass1 == newpass2)
 149         {
 150             if(model->changePassphrase(oldpass, newpass1))
 151             {
 152                 QMessageBox::information(this, tr("Wallet encrypted"),
 153                                      tr("Wallet passphrase was succesfully changed."));
 154                 QDialog::accept(); // Success
 155             }
 156             else
 157             {
 158                 QMessageBox::critical(this, tr("Wallet encryption failed"),
 159                                      tr("The passphrase entered for the wallet decryption was incorrect."));
 160             }
 161         }
 162         else
 163         {
 164             QMessageBox::critical(this, tr("Wallet encryption failed"),
 165                                  tr("The supplied passphrases do not match."));
 166         }
 167         break;
 168     }
 169 }
 170
 171 void AskPassphraseDialog::textChanged()
 172 {
 173     // Validate input, set Ok button to enabled when accepable
 174     bool acceptable = false;
 175     switch(mode)
 176     {
 177     case Encrypt: // New passphrase x2
 178         acceptable = !ui->passEdit2->text().isEmpty() && !ui->passEdit3->text().isEmpty();
 179         break;
 180     case Unlock: // Old passphrase x1
 181     case Decrypt:
 182         acceptable = !ui->passEdit1->text().isEmpty();
 183         break;
 184     case ChangePass: // Old passphrase x1, new passphrase x2
 185         acceptable = !ui->passEdit1->text().isEmpty() && !ui->passEdit2->text().isEmpty() && !ui->passEdit3->text().isEmpty();
 186         break;
 187     }
 188     ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(acceptable);
 189 }