keepassxc

1
/*
2
 *  Copyright (C) 2010 Felix Geyer <debfx@fobos.de>
3
 *
4
 *  This program is free software: you can redistribute it and/or modify
5
 *  it under the terms of the GNU General Public License as published by
6
 *  the Free Software Foundation, either version 2 or (at your option)
7
 *  version 3 of the License.
8
 *
9
 *  This program is distributed in the hope that it will be useful,
10
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 *  GNU General Public License for more details.
13
 *
14
 *  You should have received a copy of the GNU General Public License
15
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
16
 */
17

18
#include "TestKeePass1Reader.h"
19

20
#include <QBuffer>
21
#include <QTest>
22

23
#include "config-keepassx-tests.h"
24
#include "core/Group.h"
25
#include "core/Metadata.h"
26
#include "crypto/Crypto.h"
27
#include "format/KeePass1Reader.h"
28
#include "format/KeePass2Reader.h"
29
#include "format/KeePass2Writer.h"
30
#include "keys/FileKey.h"
31
#include "keys/PasswordKey.h"
32

33
QTEST_GUILESS_MAIN(TestKeePass1Reader)
34

35
void TestKeePass1Reader::initTestCase()
36
{
37
    QVERIFY(Crypto::init());
38

39
    QString filename = QString(KEEPASSX_TEST_DATA_DIR).append("/basic.kdb");
40

41
    KeePass1Reader reader;
42
    m_db = reader.readDatabase(filename, "masterpw", nullptr);
43
    QVERIFY(m_db);
44
    QVERIFY(!reader.hasError());
45
}
46

47
void TestKeePass1Reader::testBasic()
48
{
49
    QVERIFY(m_db->deletedObjects().isEmpty());
50

51
    QCOMPARE(m_db->rootGroup()->children().size(), 2);
52

53
    Group* group1 = m_db->rootGroup()->children().at(0);
54
    QVERIFY(!group1->uuid().isNull());
55
    QCOMPARE(group1->name(), QString("Internet"));
56
    QCOMPARE(group1->children().size(), 2);
57
    QCOMPARE(group1->entries().size(), 2);
58
    QCOMPARE(group1->iconNumber(), 1);
59

60
    Entry* entry11 = group1->entries().at(0);
61
    QVERIFY(!entry11->uuid().isNull());
62
    QCOMPARE(entry11->title(), QString("Test entry"));
63
    QCOMPARE(entry11->iconNumber(), 1);
64
    QCOMPARE(entry11->username(), QString("I"));
65
    QCOMPARE(entry11->url(), QString("http://example.com/"));
66
    QCOMPARE(entry11->password(), QString("secretpassword"));
67
    QCOMPARE(entry11->notes(), QString("Lorem ipsum\ndolor sit amet"));
68
    QVERIFY(entry11->timeInfo().expires());
69
    QCOMPARE(entry11->timeInfo().expiryTime(), genDT(2012, 5, 9, 10, 32));
70
    QCOMPARE(entry11->attachments()->keys().size(), 1);
71
    QCOMPARE(entry11->attachments()->keys().at(0), QString("attachment.txt"));
72
    QCOMPARE(entry11->attachments()->value("attachment.txt"), QByteArray("hello world\n"));
73

74
    Entry* entry12 = group1->entries().at(1);
75
    QCOMPARE(entry12->title(), QString(""));
76
    QCOMPARE(entry12->iconNumber(), 0);
77
    QCOMPARE(entry12->username(), QString(""));
78
    QCOMPARE(entry12->url(), QString(""));
79
    QCOMPARE(entry12->password(), QString(""));
80
    QCOMPARE(entry12->notes(), QString(""));
81
    QVERIFY(!entry12->timeInfo().expires());
82
    QCOMPARE(entry12->attachments()->keys().size(), 0);
83

84
    Group* group11 = group1->children().at(0);
85
    QCOMPARE(group11->name(), QString("Subgroup 1"));
86
    QCOMPARE(group11->children().size(), 1);
87

88
    Group* group111 = group11->children().at(0);
89
    QCOMPARE(group111->name(), QString("Unexpanded"));
90
    QCOMPARE(group111->children().size(), 1);
91

92
    Group* group1111 = group111->children().at(0);
93
    QCOMPARE(group1111->name(), QString("abc"));
94
    QCOMPARE(group1111->children().size(), 0);
95

96
    Group* group12 = group1->children().at(1);
97
    QCOMPARE(group12->name(), QString("Subgroup 2"));
98
    QCOMPARE(group12->children().size(), 0);
99

100
    Group* group2 = m_db->rootGroup()->children().at(1);
101
    QCOMPARE(group2->name(), QString("eMail"));
102
    QCOMPARE(group2->entries().size(), 1);
103
    QCOMPARE(group2->iconNumber(), 19);
104

105
    reopenDatabase(m_db, "masterpw", QString());
106
}
107

108
void TestKeePass1Reader::testMasterKey()
109
{
110
    QVERIFY(m_db->isInitialized());
111
    QCOMPARE(m_db->kdf()->rounds(), 713);
112
}
113

114
void TestKeePass1Reader::testCustomIcons()
115
{
116
    QCOMPARE(m_db->metadata()->customIconsOrder().size(), 1);
117
    QUuid uuid = m_db->metadata()->customIconsOrder().at(0);
118
    QVERIFY(m_db->metadata()->hasCustomIcon(uuid));
119
    QByteArray icon = m_db->metadata()->customIcon(uuid).data;
120

121
    QVERIFY(icon.startsWith(
122
        "\x89PNG\r\n\x1A\n\x00\x00\x00\rIHDR\x00\x00\x00\x10\x00\x00\x00\x10\b\x06\x00\x00\x00\x1F\xF3\xFF"));
123
}
124

125
void TestKeePass1Reader::testGroupExpanded()
126
{
127
    QCOMPARE(m_db->rootGroup()->children().at(0)->isExpanded(), true);
128
    QCOMPARE(m_db->rootGroup()->children().at(0)->children().at(0)->isExpanded(), true);
129
    QCOMPARE(m_db->rootGroup()->children().at(0)->children().at(0)->children().at(0)->isExpanded(), false);
130
}
131

132
void TestKeePass1Reader::testAutoType()
133
{
134
    Group* group = m_db->rootGroup()->children().at(0)->children().at(0);
135
    QCOMPARE(group->entries().size(), 2);
136

137
    Entry* entry1 = group->entries().at(0);
138
    QCOMPARE(entry1->notes(), QString("last line"));
139
    QCOMPARE(entry1->defaultAutoTypeSequence(), QString("{USERNAME}{ENTER}"));
140
    QCOMPARE(entry1->autoTypeAssociations()->size(), 5);
141
    QCOMPARE(entry1->autoTypeAssociations()->get(0).sequence, QString(""));
142
    QCOMPARE(entry1->autoTypeAssociations()->get(0).window, QString("a window"));
143
    QCOMPARE(entry1->autoTypeAssociations()->get(1).sequence, QString(""));
144
    QCOMPARE(entry1->autoTypeAssociations()->get(1).window, QString("a second window"));
145
    QCOMPARE(entry1->autoTypeAssociations()->get(2).sequence, QString("{PASSWORD}{ENTER}"));
146
    QCOMPARE(entry1->autoTypeAssociations()->get(2).window, QString("Window Nr 1a"));
147
    QCOMPARE(entry1->autoTypeAssociations()->get(3).sequence, QString("{PASSWORD}{ENTER}"));
148
    QCOMPARE(entry1->autoTypeAssociations()->get(3).window, QString("Window Nr 1b"));
149
    QCOMPARE(entry1->autoTypeAssociations()->get(4).sequence, QString(""));
150
    QCOMPARE(entry1->autoTypeAssociations()->get(4).window, QString("Window 2"));
151

152
    Entry* entry2 = group->entries().at(1);
153
    QCOMPARE(entry2->notes(), QString("start line\nend line"));
154
    QCOMPARE(entry2->defaultAutoTypeSequence(), QString(""));
155
    QCOMPARE(entry2->autoTypeAssociations()->size(), 2);
156
    QCOMPARE(entry2->autoTypeAssociations()->get(0).sequence, QString(""));
157
    QCOMPARE(entry2->autoTypeAssociations()->get(0).window, QString("Main Window"));
158
    QCOMPARE(entry2->autoTypeAssociations()->get(1).sequence, QString(""));
159
    QCOMPARE(entry2->autoTypeAssociations()->get(1).window, QString("Test Window"));
160
}
161

162
void TestKeePass1Reader::testFileKey()
163
{
164
    QFETCH(QString, type);
165

166
    QString name = QString("FileKey").append(type);
167

168
    KeePass1Reader reader;
169

170
    QString dbFilename = QString("%1/%2.kdb").arg(QString(KEEPASSX_TEST_DATA_DIR), name);
171
    QString keyFilename = QString("%1/%2.key").arg(QString(KEEPASSX_TEST_DATA_DIR), name);
172

173
    auto db = reader.readDatabase(dbFilename, QString(), keyFilename);
174
    QVERIFY(db);
175
    QVERIFY(!reader.hasError());
176
    QCOMPARE(db->rootGroup()->children().size(), 1);
177
    QCOMPARE(db->rootGroup()->children().at(0)->name(), name);
178

179
    reopenDatabase(db, QString(), keyFilename);
180
}
181

182
void TestKeePass1Reader::testFileKey_data()
183
{
184
    QTest::addColumn<QString>("type");
185
    QTest::newRow("Binary") << QString("Binary");
186
    QTest::newRow("Hex") << QString("Hex");
187
    QTest::newRow("Hashed") << QString("Hashed");
188
}
189

190
void TestKeePass1Reader::testCompositeKey()
191
{
192
    QString name = "CompositeKey";
193

194
    KeePass1Reader reader;
195

196
    QString dbFilename = QString("%1/%2.kdb").arg(QString(KEEPASSX_TEST_DATA_DIR), name);
197
    QString keyFilename = QString("%1/FileKeyHex.key").arg(QString(KEEPASSX_TEST_DATA_DIR));
198

199
    auto db = reader.readDatabase(dbFilename, "mypassword", keyFilename);
200
    QVERIFY(db);
201
    QVERIFY(!reader.hasError());
202
    QCOMPARE(db->rootGroup()->children().size(), 1);
203
    QCOMPARE(db->rootGroup()->children().at(0)->name(), name);
204

205
    reopenDatabase(db, "mypassword", keyFilename);
206
}
207

208
void TestKeePass1Reader::testTwofish()
209
{
210
    QString name = "Twofish";
211

212
    KeePass1Reader reader;
213

214
    QString dbFilename = QString("%1/%2.kdb").arg(QString(KEEPASSX_TEST_DATA_DIR), name);
215

216
    auto db = reader.readDatabase(dbFilename, "masterpw", nullptr);
217
    QVERIFY(db);
218
    QVERIFY(!reader.hasError());
219
    QCOMPARE(db->rootGroup()->children().size(), 1);
220
    QCOMPARE(db->rootGroup()->children().at(0)->name(), name);
221
}
222

223
void TestKeePass1Reader::testCP1252Password()
224
{
225
    QString name = "CP-1252";
226

227
    KeePass1Reader reader;
228

229
    QString dbFilename = QString("%1/%2.kdb").arg(QString(KEEPASSX_TEST_DATA_DIR), name);
230
    QString password = QString::fromUtf8("\xe2\x80\x9e\x70\x61\x73\x73\x77\x6f\x72\x64\xe2\x80\x9d");
231

232
    auto db = reader.readDatabase(dbFilename, password, nullptr);
233
    QVERIFY(db);
234
    QVERIFY(!reader.hasError());
235
    QCOMPARE(db->rootGroup()->children().size(), 1);
236
    QCOMPARE(db->rootGroup()->children().at(0)->name(), name);
237
}
238

239
void TestKeePass1Reader::cleanupTestCase()
240
{
241
}
242

243
QDateTime TestKeePass1Reader::genDT(int year, int month, int day, int hour, int min)
244
{
245
    QDate date(year, month, day);
246
    QTime time(hour, min, 0);
247
    return {date, time, Qt::UTC};
248
}
249

250
void TestKeePass1Reader::reopenDatabase(QSharedPointer<Database> db,
251
                                        const QString& password,
252
                                        const QString& keyfileName)
253
{
254
    QBuffer buffer;
255
    buffer.open(QIODevice::ReadWrite);
256

257
    KeePass2Writer writer;
258
    writer.writeDatabase(&buffer, db.data());
259
    QVERIFY(!writer.hasError());
260
    QVERIFY(buffer.seek(0));
261

262
    auto key = QSharedPointer<CompositeKey>::create();
263
    if (!password.isEmpty()) {
264
        key->addKey(QSharedPointer<PasswordKey>::create(password));
265
    }
266
    if (!keyfileName.isEmpty()) {
267
        auto fileKey = QSharedPointer<FileKey>::create();
268
        QVERIFY(fileKey->load(keyfileName));
269
        key->addKey(fileKey);
270
    }
271

272
    KeePass2Reader reader;
273
    auto newDb = QSharedPointer<Database>::create();
274
    QVERIFY(reader.readDatabase(&buffer, key, newDb.data()));
275
    QVERIFY(!reader.hasError());
276
}
277