Git-Mirrors/keepassxc
1
0
Fork 0
mirror of https://github.com/keepassxreboot/keepassxc.git synced 2025-06-29 00:47:32 -04:00
Code Issues Projects Releases Packages Wiki Activity

Add KeePass2RandomStream class.

Browse source 
It's responsible for processing protected strings in kdbx files.
This commit is contained in:
Felix Geyer 2011-07-07 00:08:37 +02:00
parent 1d6e106aee
commit 58e048be96
6 changed files with 228 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

79
src/format/KeePass2RandomStream.cpp Normal file
View file

@ -0,0 +1,79 @@
/*
* Copyright (C) 2010 Felix Geyer <debfx@fobos.de>
*
* 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 "KeePass2RandomStream.h"
#include "format/KeePass2.h"
#include "crypto/CryptoHash.h"
KeePass2RandomStream::KeePass2RandomStream(QByteArray key)
: m_cipher(SymmetricCipher::Salsa20, SymmetricCipher::Stream, SymmetricCipher::Encrypt,
CryptoHash::hash(key, CryptoHash::Sha256), KeePass2::INNER_STREAM_SALSA20_IV)
, m_offset(0)
{
}
QByteArray KeePass2RandomStream::randomBytes(int size)
{
QByteArray result;
int bytesRemaining = size;
while (bytesRemaining > 0) {
if (m_buffer.size() == m_offset) {
loadBlock();
}
int bytesToCopy = qMin(bytesRemaining, m_buffer.size() - m_offset);
result.append(m_buffer.mid(m_offset, bytesToCopy));
m_offset += bytesToCopy;
bytesRemaining -= bytesToCopy;
}
return result;
}
QByteArray KeePass2RandomStream::process(const QByteArray& data)
{
QByteArray randomData = randomBytes(data.size());
QByteArray result;
result.resize(data.size());
for (int i=0; i<data.size(); i++) {
result[i] = data[i] ^ randomData[i];
}
return result;
}
void KeePass2RandomStream::processInPlace(QByteArray& data)
{
QByteArray randomData = randomBytes(data.size());
for (int i=0; i<data.size(); i++) {
data[i] = data[i] ^ randomData[i];
}
}
void KeePass2RandomStream::loadBlock()
{
Q_ASSERT(m_offset == m_buffer.size());
m_buffer.fill('\0', m_cipher.blockSize());
m_cipher.processInPlace(m_buffer);
m_offset = 0;
}