keysmith/autotests/secrets/key-derivation.cpp

55 lines
2.1 KiB
C++

/*
* SPDX-License-Identifier: GPL-3.0-or-later
* SPDX-FileCopyrightText: 2020 Johan Ouwerkerk <jm.ouwerkerk@gmail.com>
*/
#include "secrets/secrets.h"
#include "test-utils/random.h"
#include <QTest>
#include <QtDebug>
class KeyDerivationTest: public QObject
{
Q_OBJECT
private Q_SLOTS:
void testRecovery(void);
};
void KeyDerivationTest::testRecovery(void)
{
QScopedPointer<secrets::SecureMemory> passwd(secrets::SecureMemory::allocate(13ULL));
QVERIFY2(passwd, "password memory should be allocated");
memcpy(passwd->data(), "Hello, world!", passwd->size());
std::optional<secrets::KeyDerivationParameters> defaults = secrets::KeyDerivationParameters::create();
QVERIFY2(defaults, "defaults should yield a valid key parameters object");
QScopedPointer<secrets::SecureMasterKey> origMasterKey(secrets::SecureMasterKey::derive(passwd.data(), *defaults, &test::fakeRandom));
QVERIFY2(origMasterKey, "key derivation should succeed");
QByteArray expectedSalt(crypto_pwhash_SALTBYTES, 'A');
QCOMPARE(origMasterKey->salt(), expectedSalt);
QScopedPointer<secrets::SecureMasterKey> copyKey(secrets::SecureMasterKey::derive(passwd.data(), *defaults, expectedSalt, &test::fakeRandom));
QVERIFY2(copyKey, "recovering/re-deriving a copy of the master key should succeed");
QScopedPointer<secrets::SecureMemory> payload(secrets::SecureMemory::allocate(42ULL));
QVERIFY2(payload, "allocating the secure memory input buffer should succeed");
memset(payload->data(), 'B', 42ULL);
std::optional<secrets::EncryptedSecret> fromOrigKey = origMasterKey->encrypt(payload.data());
QVERIFY2(fromOrigKey, "encryption of the payload should succeed with the original master key");
std::optional<secrets::EncryptedSecret> fromCopyKey = copyKey->encrypt(payload.data());
QVERIFY2(fromCopyKey, "encryption of the payload should also succeed with the recovered copy of the master key");
QCOMPARE(fromOrigKey->cryptText(), fromCopyKey->cryptText());
QCOMPARE(fromOrigKey->nonce(), fromCopyKey->nonce());
}
QTEST_APPLESS_MAIN(KeyDerivationTest)
#include "key-derivation.moc"