mirror of
				https://github.com/keepassxreboot/keepassxc.git
				synced 2025-11-03 23:24:06 -05:00 
			
		
		
		
	
		
			
				
	
	
		
			90 lines
		
	
	
	
		
			2.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			90 lines
		
	
	
	
		
			2.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
 *  Copyright (C) 2019 KeePassXC Team <team@keepassxc.org>
 | 
						|
 *
 | 
						|
 *  This program is free software: you can redistribute it and/or modify
 | 
						|
 *  it under the terms of the GNU General Public License as published by
 | 
						|
 *  the Free Software Foundation, either version 2 or (at your option)
 | 
						|
 *  version 3 of the License.
 | 
						|
 *
 | 
						|
 *  This program is distributed in the hope that it will be useful,
 | 
						|
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
 *  GNU General Public License for more details.
 | 
						|
 *
 | 
						|
 *  You should have received a copy of the GNU General Public License
 | 
						|
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 */
 | 
						|
 | 
						|
#include "PasswordKey.h"
 | 
						|
 | 
						|
#include "crypto/CryptoHash.h"
 | 
						|
 | 
						|
#include <QDataStream>
 | 
						|
#include <QSharedPointer>
 | 
						|
 | 
						|
QUuid PasswordKey::UUID("77e90411-303a-43f2-b773-853b05635ead");
 | 
						|
 | 
						|
constexpr int PasswordKey::SHA256_SIZE;
 | 
						|
 | 
						|
PasswordKey::PasswordKey()
 | 
						|
    : Key(UUID)
 | 
						|
    , m_key(SHA256_SIZE)
 | 
						|
{
 | 
						|
}
 | 
						|
 | 
						|
PasswordKey::PasswordKey(const QString& password)
 | 
						|
    : Key(UUID)
 | 
						|
    , m_key(SHA256_SIZE)
 | 
						|
{
 | 
						|
    setPassword(password);
 | 
						|
}
 | 
						|
 | 
						|
QByteArray PasswordKey::rawKey() const
 | 
						|
{
 | 
						|
    if (!m_isInitialized) {
 | 
						|
        return {};
 | 
						|
    }
 | 
						|
    return QByteArray(m_key.data(), m_key.size());
 | 
						|
}
 | 
						|
 | 
						|
void PasswordKey::setRawKey(const QByteArray& data)
 | 
						|
{
 | 
						|
    if (data.isEmpty()) {
 | 
						|
        m_key.clear();
 | 
						|
        m_isInitialized = false;
 | 
						|
    } else {
 | 
						|
        Q_ASSERT(data.size() == SHA256_SIZE);
 | 
						|
        m_key.assign(data.begin(), data.end());
 | 
						|
        m_isInitialized = true;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void PasswordKey::setPassword(const QString& password)
 | 
						|
{
 | 
						|
    setRawKey(CryptoHash::hash(password.toUtf8(), CryptoHash::Sha256));
 | 
						|
}
 | 
						|
 | 
						|
QSharedPointer<PasswordKey> PasswordKey::fromRawKey(const QByteArray& rawKey)
 | 
						|
{
 | 
						|
    auto result = QSharedPointer<PasswordKey>::create();
 | 
						|
    result->setRawKey(rawKey);
 | 
						|
    return result;
 | 
						|
}
 | 
						|
 | 
						|
QByteArray PasswordKey::serialize() const
 | 
						|
{
 | 
						|
    QByteArray data;
 | 
						|
    QDataStream stream(&data, QIODevice::WriteOnly);
 | 
						|
    stream << uuid().toRfc4122() << rawKey();
 | 
						|
    return data;
 | 
						|
}
 | 
						|
 | 
						|
void PasswordKey::deserialize(const QByteArray& data)
 | 
						|
{
 | 
						|
    QByteArray uuidData, key;
 | 
						|
    QDataStream stream(data);
 | 
						|
    stream >> uuidData >> key;
 | 
						|
    if (uuid().toRfc4122() == uuidData) {
 | 
						|
        setRawKey(key);
 | 
						|
    }
 | 
						|
}
 |