1 // Copyright (c) 2011-2015 The Bitcoin Core developers
2 // Distributed under the MIT software license, see the accompanying
3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5 #if defined(HAVE_CONFIG_H)
6 #include "config/bitcoin-config.h"
7 #endif
8
9 #include "askpassphrasedialog.h"
10 #include "ui_askpassphrasedialog.h"
11
12 #include "guiconstants.h"
13 #include "walletmodel.h"
14
15 #include "support/allocators/secure.h"
16
17 #include <QKeyEvent>
18 #include <QMessageBox>
19 #include <QPushButton>
20
AskPassphraseDialog(Mode mode,QWidget * parent)21 AskPassphraseDialog::AskPassphraseDialog(Mode mode, QWidget *parent) :
22 QDialog(parent),
23 ui(new Ui::AskPassphraseDialog),
24 mode(mode),
25 model(0),
26 fCapsLock(false)
27 {
28 ui->setupUi(this);
29
30 ui->passEdit1->setMinimumSize(ui->passEdit1->sizeHint());
31 ui->passEdit2->setMinimumSize(ui->passEdit2->sizeHint());
32 ui->passEdit3->setMinimumSize(ui->passEdit3->sizeHint());
33
34 ui->passEdit1->setMaxLength(MAX_PASSPHRASE_SIZE);
35 ui->passEdit2->setMaxLength(MAX_PASSPHRASE_SIZE);
36 ui->passEdit3->setMaxLength(MAX_PASSPHRASE_SIZE);
37
38 // Setup Caps Lock detection.
39 ui->passEdit1->installEventFilter(this);
40 ui->passEdit2->installEventFilter(this);
41 ui->passEdit3->installEventFilter(this);
42
43 switch(mode)
44 {
45 case Encrypt: // Ask passphrase x2
46 ui->warningLabel->setText(tr("Enter the new passphrase to the wallet.<br/>Please use a passphrase of <b>ten or more random characters</b>, or <b>eight or more words</b>."));
47 ui->passLabel1->hide();
48 ui->passEdit1->hide();
49 setWindowTitle(tr("Encrypt wallet"));
50 break;
51 case Unlock: // Ask passphrase
52 ui->warningLabel->setText(tr("This operation needs your wallet passphrase to unlock the wallet."));
53 ui->passLabel2->hide();
54 ui->passEdit2->hide();
55 ui->passLabel3->hide();
56 ui->passEdit3->hide();
57 setWindowTitle(tr("Unlock wallet"));
58 break;
59 case Decrypt: // Ask passphrase
60 ui->warningLabel->setText(tr("This operation needs your wallet passphrase to decrypt the wallet."));
61 ui->passLabel2->hide();
62 ui->passEdit2->hide();
63 ui->passLabel3->hide();
64 ui->passEdit3->hide();
65 setWindowTitle(tr("Decrypt wallet"));
66 break;
67 case ChangePass: // Ask old passphrase + new passphrase x2
68 setWindowTitle(tr("Change passphrase"));
69 ui->warningLabel->setText(tr("Enter the old passphrase and new passphrase to the wallet."));
70 break;
71 }
72 textChanged();
73 connect(ui->passEdit1, SIGNAL(textChanged(QString)), this, SLOT(textChanged()));
74 connect(ui->passEdit2, SIGNAL(textChanged(QString)), this, SLOT(textChanged()));
75 connect(ui->passEdit3, SIGNAL(textChanged(QString)), this, SLOT(textChanged()));
76 }
77
~AskPassphraseDialog()78 AskPassphraseDialog::~AskPassphraseDialog()
79 {
80 secureClearPassFields();
81 delete ui;
82 }
83
setModel(WalletModel * model)84 void AskPassphraseDialog::setModel(WalletModel *model)
85 {
86 this->model = model;
87 }
88
accept()89 void AskPassphraseDialog::accept()
90 {
91 SecureString oldpass, newpass1, newpass2;
92 if(!model)
93 return;
94 oldpass.reserve(MAX_PASSPHRASE_SIZE);
95 newpass1.reserve(MAX_PASSPHRASE_SIZE);
96 newpass2.reserve(MAX_PASSPHRASE_SIZE);
97 // TODO: get rid of this .c_str() by implementing SecureString::operator=(std::string)
98 // Alternately, find a way to make this input mlock()'d to begin with.
99 oldpass.assign(ui->passEdit1->text().toStdString().c_str());
100 newpass1.assign(ui->passEdit2->text().toStdString().c_str());
101 newpass2.assign(ui->passEdit3->text().toStdString().c_str());
102
103 secureClearPassFields();
104
105 switch(mode)
106 {
107 case Encrypt: {
108 if(newpass1.empty() || newpass2.empty())
109 {
110 // Cannot encrypt with empty passphrase
111 break;
112 }
113 QMessageBox::StandardButton retval = QMessageBox::question(this, tr("Confirm wallet encryption"),
114 tr("Warning: If you encrypt your wallet and lose your passphrase, you will <b>LOSE ALL OF YOUR ZETACOINS</b>!") + "<br><br>" + tr("Are you sure you wish to encrypt your wallet?"),
115 QMessageBox::Yes|QMessageBox::Cancel,
116 QMessageBox::Cancel);
117 if(retval == QMessageBox::Yes)
118 {
119 if(newpass1 == newpass2)
120 {
121 if(model->setWalletEncrypted(true, newpass1))
122 {
123 QMessageBox::warning(this, tr("Wallet encrypted"),
124 "<qt>" +
125 tr("%1 will close now to finish the encryption process. "
126 "Remember that encrypting your wallet cannot fully protect "
127 "your zetacoins from being stolen by malware infecting your computer.").arg(tr(PACKAGE_NAME)) +
128 "<br><br><b>" +
129 tr("IMPORTANT: Any previous backups you have made of your wallet file "
130 "should be replaced with the newly generated, encrypted wallet file. "
131 "For security reasons, previous backups of the unencrypted wallet file "
132 "will become useless as soon as you start using the new, encrypted wallet.") +
133 "</b></qt>");
134 QApplication::quit();
135 }
136 else
137 {
138 QMessageBox::critical(this, tr("Wallet encryption failed"),
139 tr("Wallet encryption failed due to an internal error. Your wallet was not encrypted."));
140 }
141 QDialog::accept(); // Success
142 }
143 else
144 {
145 QMessageBox::critical(this, tr("Wallet encryption failed"),
146 tr("The supplied passphrases do not match."));
147 }
148 }
149 else
150 {
151 QDialog::reject(); // Cancelled
152 }
153 } break;
154 case Unlock:
155 if(!model->setWalletLocked(false, oldpass))
156 {
157 QMessageBox::critical(this, tr("Wallet unlock failed"),
158 tr("The passphrase entered for the wallet decryption was incorrect."));
159 }
160 else
161 {
162 QDialog::accept(); // Success
163 }
164 break;
165 case Decrypt:
166 if(!model->setWalletEncrypted(false, oldpass))
167 {
168 QMessageBox::critical(this, tr("Wallet decryption failed"),
169 tr("The passphrase entered for the wallet decryption was incorrect."));
170 }
171 else
172 {
173 QDialog::accept(); // Success
174 }
175 break;
176 case ChangePass:
177 if(newpass1 == newpass2)
178 {
179 if(model->changePassphrase(oldpass, newpass1))
180 {
181 QMessageBox::information(this, tr("Wallet encrypted"),
182 tr("Wallet passphrase was successfully changed."));
183 QDialog::accept(); // Success
184 }
185 else
186 {
187 QMessageBox::critical(this, tr("Wallet encryption failed"),
188 tr("The passphrase entered for the wallet decryption was incorrect."));
189 }
190 }
191 else
192 {
193 QMessageBox::critical(this, tr("Wallet encryption failed"),
194 tr("The supplied passphrases do not match."));
195 }
196 break;
197 }
198 }
199
textChanged()200 void AskPassphraseDialog::textChanged()
201 {
202 // Validate input, set Ok button to enabled when acceptable
203 bool acceptable = false;
204 switch(mode)
205 {
206 case Encrypt: // New passphrase x2
207 acceptable = !ui->passEdit2->text().isEmpty() && !ui->passEdit3->text().isEmpty();
208 break;
209 case Unlock: // Old passphrase x1
210 case Decrypt:
211 acceptable = !ui->passEdit1->text().isEmpty();
212 break;
213 case ChangePass: // Old passphrase x1, new passphrase x2
214 acceptable = !ui->passEdit1->text().isEmpty() && !ui->passEdit2->text().isEmpty() && !ui->passEdit3->text().isEmpty();
215 break;
216 }
217 ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(acceptable);
218 }
219
event(QEvent * event)220 bool AskPassphraseDialog::event(QEvent *event)
221 {
222 // Detect Caps Lock key press.
223 if (event->type() == QEvent::KeyPress) {
224 QKeyEvent *ke = static_cast<QKeyEvent *>(event);
225 if (ke->key() == Qt::Key_CapsLock) {
226 fCapsLock = !fCapsLock;
227 }
228 if (fCapsLock) {
229 ui->capsLabel->setText(tr("Warning: The Caps Lock key is on!"));
230 } else {
231 ui->capsLabel->clear();
232 }
233 }
234 return QWidget::event(event);
235 }
236
eventFilter(QObject * object,QEvent * event)237 bool AskPassphraseDialog::eventFilter(QObject *object, QEvent *event)
238 {
239 /* Detect Caps Lock.
240 * There is no good OS-independent way to check a key state in Qt, but we
241 * can detect Caps Lock by checking for the following condition:
242 * Shift key is down and the result is a lower case character, or
243 * Shift key is not down and the result is an upper case character.
244 */
245 if (event->type() == QEvent::KeyPress) {
246 QKeyEvent *ke = static_cast<QKeyEvent *>(event);
247 QString str = ke->text();
248 if (str.length() != 0) {
249 const QChar *psz = str.unicode();
250 bool fShift = (ke->modifiers() & Qt::ShiftModifier) != 0;
251 if ((fShift && *psz >= 'a' && *psz <= 'z') || (!fShift && *psz >= 'A' && *psz <= 'Z')) {
252 fCapsLock = true;
253 ui->capsLabel->setText(tr("Warning: The Caps Lock key is on!"));
254 } else if (psz->isLetter()) {
255 fCapsLock = false;
256 ui->capsLabel->clear();
257 }
258 }
259 }
260 return QDialog::eventFilter(object, event);
261 }
262
SecureClearQLineEdit(QLineEdit * edit)263 static void SecureClearQLineEdit(QLineEdit* edit)
264 {
265 // Attempt to overwrite text so that they do not linger around in memory
266 edit->setText(QString(" ").repeated(edit->text().size()));
267 edit->clear();
268 }
269
secureClearPassFields()270 void AskPassphraseDialog::secureClearPassFields()
271 {
272 SecureClearQLineEdit(ui->passEdit1);
273 SecureClearQLineEdit(ui->passEdit2);
274 SecureClearQLineEdit(ui->passEdit3);
275 }
276