Refactored MUtils::Hash functions.

This commit is contained in:
LoRd_MuldeR 2016-12-26 00:55:45 +01:00
parent 1fab453c11
commit 9fb7fcdac4
8 changed files with 207 additions and 90 deletions

View File

@ -26,6 +26,7 @@
<ClCompile Include="src\Global.cpp" />
<ClCompile Include="src\GUI.cpp" />
<ClCompile Include="src\GUI_Win32.cpp" />
<ClCompile Include="src\Hash.cpp" />
<ClCompile Include="src\Hash_Blake2.cpp" />
<ClCompile Include="src\IPCChannel.cpp" />
<ClCompile Include="src\JobObject_Win32.cpp" />
@ -47,10 +48,9 @@
<ClInclude Include="include\MUtils\Exception.h" />
<ClInclude Include="include\MUtils\Global.h" />
<ClInclude Include="include\MUtils\GUI.h" />
<ClInclude Include="include\MUtils\Hash_Blake2.h" />
<ClInclude Include="include\MUtils\Hash.h" />
<ClInclude Include="include\MUtils\IPCChannel.h" />
<ClInclude Include="include\MUtils\JobObject.h" />
<ClInclude Include="include\MUtils\Hash_Keccak.h" />
<ClInclude Include="include\MUtils\OSSupport.h" />
<ClInclude Include="include\MUtils\Registry.h" />
<ClInclude Include="include\MUtils\Sound.h" />

View File

@ -102,6 +102,9 @@
<ClCompile Include="src\Utils_Win32.cpp">
<Filter>Source Files</Filter>
</ClCompile>
<ClCompile Include="src\Hash.cpp">
<Filter>Source Files</Filter>
</ClCompile>
</ItemGroup>
<ItemGroup>
<ClInclude Include="src\CriticalSection_Win32.h">
@ -161,12 +164,6 @@
<ClInclude Include="include\MUtils\Translation.h">
<Filter>Public Headers</Filter>
</ClInclude>
<ClInclude Include="include\MUtils\Hash_Blake2.h">
<Filter>Public Headers</Filter>
</ClInclude>
<ClInclude Include="include\MUtils\Hash_Keccak.h">
<Filter>Public Headers</Filter>
</ClInclude>
<ClInclude Include="src\3rd_party\blake2\include\blake2.h">
<Filter>Header Files\3rd Party</Filter>
</ClInclude>
@ -176,6 +173,9 @@
<ClInclude Include="src\Utils_Win32.h">
<Filter>Header Files</Filter>
</ClInclude>
<ClInclude Include="include\MUtils\Hash.h">
<Filter>Public Headers</Filter>
</ClInclude>
</ItemGroup>
<ItemGroup>
<CustomBuild Include="include\Mutils\UpdateChecker.h">

60
include/MUtils/Hash.h Normal file
View File

@ -0,0 +1,60 @@
///////////////////////////////////////////////////////////////////////////////
// MuldeR's Utilities for Qt
// Copyright (C) 2004-2016 LoRd_MuldeR <MuldeR2@GMX.de>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
//
// http://www.gnu.org/licenses/lgpl-2.1.txt
//////////////////////////////////////////////////////////////////////////////////
#pragma once
//MUtils
#include <MUtils/Global.h>
//Qt
#include <QByteArray>
#include <QFile>
namespace MUtils
{
namespace Hash
{
static const quint16 HASH_BLAKE2_512 = 0x0000U;
static const quint16 HASH_KECCAK_224 = 0x0100U;
static const quint16 HASH_KECCAK_256 = 0x0101U;
static const quint16 HASH_KECCAK_384 = 0x0102U;
static const quint16 HASH_KECCAK_512 = 0x0103U;
class MUTILS_API Hash
{
public:
Hash(const char* key = NULL) {};
virtual ~Hash(void) {};
bool update(const quint8 *const data, const quint32 len) { return process(data, len); }
bool update(const QByteArray &data) { return process(((const quint8*)data.constData()), ((quint32)data.length())); }
bool update(QFile &file);
QByteArray digest(const bool bAsHex = true) { return bAsHex ? finalize().toHex() : finalize(); }
protected:
virtual bool process(const quint8 *const data, const quint32 len) = 0;
virtual QByteArray finalize(void) = 0;
};
Hash *create(const quint16 &hashId, const char *const key = NULL);
}
}

68
src/Hash.cpp Normal file
View File

@ -0,0 +1,68 @@
///////////////////////////////////////////////////////////////////////////////
// Simple x264 Launcher
// Copyright (C) 2004-2016 LoRd_MuldeR <MuldeR2@GMX.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 of the License, or
// (at your option) any later version.
//
// 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, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
// http://www.gnu.org/licenses/gpl-2.0.txt
///////////////////////////////////////////////////////////////////////////////
//MUtils
#include <MUtils/Hash.h>
#include <MUtils/Exception.h>
//Hash Functions
#include "Hash_Keccak.h"
#include "Hash_Blake2.h"
// ==========================================================================
// Abstract Hash Class
// ==========================================================================
bool MUtils::Hash::Hash::update(QFile &file)
{
while (!file.atEnd())
{
const QByteArray data = file.read(4194304i64);
if (data.isEmpty() || (!update(data)))
{
return false;
}
}
return true;
}
// ==========================================================================
// Hash Factory
// ==========================================================================
MUtils::Hash::Hash *MUtils::Hash::create(const quint16 &hashId, const char *const key)
{
switch (hashId)
{
case HASH_KECCAK_224:
return Keccak::create(Keccak::hb224, key);
case HASH_KECCAK_256:
return Keccak::create(Keccak::hb256, key);
case HASH_KECCAK_384:
return Keccak::create(Keccak::hb384, key);
case HASH_KECCAK_512:
return Keccak::create(Keccak::hb512, key);
case HASH_BLAKE2_512:
return new Blake2(key);
default:
MUTILS_THROW_FMT("Hash algorithm 0x%02X is unknown!", quint32(hashId));
}
}

View File

@ -32,8 +32,9 @@
this software. If not, see <http://creativecommons.org/publicdomain/zero/1.0/>.
*/
#include "Hash_Blake2.h"
//MUtils
#include <MUtils/Hash_Blake2.h>
#include <MUtils/Exception.h>
//Internal
@ -51,7 +52,7 @@ class MUtils::Hash::Blake2_Context
Blake2_Context(void)
{
if(!(state = (MUtils::Hash::Internal::Blake2Impl::blake2b_state*) _aligned_malloc(sizeof(MUtils::Hash::Internal::Blake2Impl::blake2b_state), 64)))
if(!(state = (MUtils::Hash::Internal::Blake2Impl::blake2b_state*) _aligned_malloc(sizeof(MUtils::Hash::Internal::Blake2Impl::blake2b_state), HASH_SIZE)))
{
MUTILS_THROW("Aligend malloc has failed!");
}
@ -68,9 +69,8 @@ private:
MUtils::Hash::Internal::Blake2Impl::blake2b_state *state;
};
MUtils::Hash::Blake2::Blake2(const char* key)
MUtils::Hash::Blake2::Blake2(const char *const key)
:
m_hash(HASH_SIZE, '\0'),
m_context(new Blake2_Context()),
m_finalized(false)
{
@ -89,55 +89,31 @@ MUtils::Hash::Blake2::~Blake2(void)
delete m_context;
}
void MUtils::Hash::Blake2::update(const QByteArray &data)
bool MUtils::Hash::Blake2::process(const quint8 *const data, const quint32 len)
{
if(m_finalized)
{
MUTILS_THROW("BLAKE2 was already finalized!");
}
if(data.size() > 0)
if(data && (len > 0))
{
if(blake2b_update(m_context->state, (const uint8_t*) data.constData(), data.size()) != 0)
if(blake2b_update(m_context->state, data, len) != 0)
{
MUTILS_THROW("BLAKE2 internal error!");
}
}
return true;
}
void MUtils::Hash::Blake2::update(QFile &file)
QByteArray MUtils::Hash::Blake2::finalize(void)
{
bool okay = false;
for(;;)
QByteArray result(HASH_SIZE, '\0');
if(blake2b_final(m_context->state, (uint8_t*) result.data(), result.size()) != 0)
{
QByteArray data = file.read(16384);
if(data.size() > 0)
{
okay = true;
update(data);
continue;
}
break;
}
if(!okay)
{
qWarning("[QBlake2Checksum] Could not ready any data from file!");
MUTILS_THROW("BLAKE2 internal error!");
}
}
QByteArray MUtils::Hash::Blake2::finalize(const bool bAsHex)
{
if(!m_finalized)
{
if(blake2b_final(m_context->state, (uint8_t*) m_hash.data(), m_hash.size()) != 0)
{
MUTILS_THROW("BLAKE2 internal error!");
}
m_finalized = true;
}
return bAsHex ? m_hash.toHex() : m_hash;
m_finalized = true;
return result;
}

View File

@ -36,6 +36,7 @@
//MUtils
#include <MUtils/Global.h>
#include <MUtils/Hash.h>
//Qt
#include <QByteArray>
@ -47,20 +48,18 @@ namespace MUtils
{
class MUTILS_API Blake2_Context;
class MUTILS_API Blake2
class MUTILS_API Blake2 : public Hash
{
public:
Blake2(const char* key = NULL);
~Blake2(void);
void update(const QByteArray &data);
void update(QFile &file);
QByteArray finalize(const bool bAsHex = true);
Blake2(const char *const key = NULL);
virtual ~Blake2(void);
private:
QByteArray m_hash;
Blake2_Context *const m_context;
bool m_finalized;
virtual bool process(const quint8 *const data, const quint32 len);
virtual QByteArray finalize(void);
};
}

View File

@ -44,9 +44,15 @@
** Date: 12.01.12 **
****************************************************************************/
#include <MUtils/Hash_Keccak.h>
#include "Hash_Keccak.h"
//MUtils
#include <MUtils/Exception.h>
//Qt
#include <QDebug>
//Implementation
#include "3rd_party/keccak/include/keccak_impl.h"
MUtils::Hash::Keccak::Keccak()
@ -55,16 +61,13 @@ MUtils::Hash::Keccak::Keccak()
m_state = (MUtils::Hash::Internal::KeccakImpl::hashState*) _aligned_malloc(sizeof(MUtils::Hash::Internal::KeccakImpl::hashState), 32);
if(!m_state)
{
throw "[MUtils::KeccakHash] Error: _aligned_malloc() has failed, probably out of heap space!";
MUTILS_THROW("_aligned_malloc() has failed, probably out of heap space!");
}
memset(m_state, 0, sizeof(MUtils::Hash::Internal::KeccakImpl::hashState));
m_hashResult.clear();
}
MUtils::Hash::Keccak::~Keccak()
{
m_hashResult.clear();
if(m_state)
{
_aligned_free(m_state);
@ -72,7 +75,7 @@ MUtils::Hash::Keccak::~Keccak()
}
}
bool MUtils::Hash::Keccak::init(HashBits hashBits)
bool MUtils::Hash::Keccak::init(const HashBits hashBits)
{
if(m_initialized)
{
@ -80,7 +83,6 @@ bool MUtils::Hash::Keccak::init(HashBits hashBits)
return false;
}
m_hashResult.clear();
memset(m_state, 0, sizeof(MUtils::Hash::Internal::KeccakImpl::hashState));
int hashBitLength = 0;
@ -99,18 +101,12 @@ bool MUtils::Hash::Keccak::init(HashBits hashBits)
return false;
}
m_hashResult.fill(char(0), hashBitLength/8);
m_initialized = true;
return true;
}
bool MUtils::Hash::Keccak::addData(const QByteArray &data)
{
return addData(data.constData(), data.size());
}
bool MUtils::Hash::Keccak::addData(const char *data, int size)
bool MUtils::Hash::Keccak::process(const quint8 *const data, const quint32 len)
{
if(!m_initialized)
{
@ -118,10 +114,9 @@ bool MUtils::Hash::Keccak::addData(const char *data, int size)
return false;
}
if(MUtils::Hash::Internal::KeccakImpl::Update(m_state, (MUtils::Hash::Internal::KeccakImpl::BitSequence*)data, size*8) != MUtils::Hash::Internal::KeccakImpl::SUCCESS)
if(MUtils::Hash::Internal::KeccakImpl::Update(m_state, (MUtils::Hash::Internal::KeccakImpl::BitSequence*)data, len*8U) != MUtils::Hash::Internal::KeccakImpl::SUCCESS)
{
qWarning("KeccakImpl::Update() has failed unexpectedly!");
m_hashResult.clear();
m_initialized = false;
return false;
}
@ -129,34 +124,52 @@ bool MUtils::Hash::Keccak::addData(const char *data, int size)
return true;
}
const QByteArray &MUtils::Hash::Keccak::finalize()
QByteArray MUtils::Hash::Keccak::finalize(void)
{
if(!m_initialized)
{
qWarning("MUtils::KeccakHash has not been initialized yet!");
m_hashResult.clear();
return m_hashResult;
return QByteArray();
}
if(MUtils::Hash::Internal::KeccakImpl::Final(m_state, (MUtils::Hash::Internal::KeccakImpl::BitSequence*)m_hashResult.data()) != MUtils::Hash::Internal::KeccakImpl::SUCCESS)
Q_ASSERT(m_state->fixedOutputLength > 0);
Q_ASSERT((m_state->fixedOutputLength % 8) == 0);
QByteArray hashResult(m_state->fixedOutputLength / 8, '\0');
if(MUtils::Hash::Internal::KeccakImpl::Final(m_state, (MUtils::Hash::Internal::KeccakImpl::BitSequence*)hashResult.data()) != MUtils::Hash::Internal::KeccakImpl::SUCCESS)
{
qWarning("KeccakImpl::Final() has failed unexpectedly!");
m_hashResult.clear();
hashResult.clear();
}
m_initialized = false;
return m_hashResult;
return hashResult;
}
MUtils::Hash::Keccak *MUtils::Hash::Keccak::create(const HashBits hashBit, const char *const key)
{
Keccak *const keccak = new Keccak();
if (!keccak->init(hashBit))
{
MUTILS_THROW("Keccak initialization has failed!");
}
if (key)
{
keccak->update(((const uint8_t*)key), strlen(key));
}
return keccak;
}
bool MUtils::Hash::Keccak::selfTest(void)
{
MUtils::Hash::Keccak hash;
Keccak hash;
const QByteArray input("The quick brown fox jumps over the lazy dog");
bool passed[4] = {false, false, false, false};
if(hash.init(MUtils::Hash::Keccak::hb224))
{
if(hash.addData(input))
if(hash.update(input))
{
QByteArray result = hash.finalize();
if(!result.isEmpty())
@ -169,7 +182,7 @@ bool MUtils::Hash::Keccak::selfTest(void)
if(hash.init(MUtils::Hash::Keccak::hb256))
{
if(hash.addData(input))
if(hash.update(input))
{
QByteArray result = hash.finalize();
if(!result.isEmpty())
@ -182,7 +195,7 @@ bool MUtils::Hash::Keccak::selfTest(void)
if(hash.init(MUtils::Hash::Keccak::hb384))
{
if(hash.addData(input))
if(hash.update(input))
{
QByteArray result = hash.finalize();
if(!result.isEmpty())
@ -195,7 +208,7 @@ bool MUtils::Hash::Keccak::selfTest(void)
if(hash.init(MUtils::Hash::Keccak::hb512))
{
if(hash.addData(input))
if(hash.update(input))
{
QByteArray result = hash.finalize();
if(!result.isEmpty())

View File

@ -48,6 +48,7 @@
//MUtils
#include <MUtils/Global.h>
#include <MUtils/Hash.h>
//Qt
#include <QString>
@ -95,25 +96,25 @@ namespace MUtils
// End Section from KeccakSponge.h
}
class MUTILS_API Keccak
class MUTILS_API Keccak : public Hash
{
public:
enum HashBits {hb224, hb256, hb384, hb512};
Keccak();
~Keccak();
virtual ~Keccak();
bool init(const HashBits hashBits=hb256);
static Keccak *create(const HashBits hashBits = hb256, const char *const key = NULL);
static bool selfTest(void);
bool init(HashBits hashBits=hb256);
bool addData(const QByteArray &data);
bool addData(const char *data, int size);
const QByteArray &finalize();
protected:
bool m_initialized;
Internal::KeccakImpl::hashState *m_state;
QByteArray m_hashResult;
virtual bool process(const quint8 *const data, const quint32 len);
virtual QByteArray finalize(void);
};
}
};