src/Hash_Blake2.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Simple x264 Launcher
   3 // Copyright (C) 2004-2019 LoRd_MuldeR <MuldeR2@GMX.de>
   4 //
   5 // This program is free software; you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation; either version 2 of the License, or
   8 // (at your option) any later version.
   9 //
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14 //
  15 // You should have received a copy of the GNU General Public License along
  16 // with this program; if not, write to the Free Software Foundation, Inc.,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 // http://www.gnu.org/licenses/gpl-2.0.txt
  20 ///////////////////////////////////////////////////////////////////////////////
  21
  22 /*
  23    BLAKE2 reference source code package - reference C implementations
  24
  25    Written in 2012 by Samuel Neves <sneves@dei.uc.pt>
  26
  27    To the extent possible under law, the author(s) have dedicated all copyright
  28    and related and neighboring rights to this software to the public domain
  29    worldwide. This software is distributed without any warranty.
  30
  31    You should have received a copy of the CC0 Public Domain Dedication along with
  32    this software. If not, see <http://creativecommons.org/publicdomain/zero/1.0/>.
  33 */
  34
  35 #include "Hash_Blake2.h"
  36
  37 //MUtils
  38 #include <MUtils/Exception.h>
  39
  40 //Internal
  41 #include "3rd_party/blake2/include/blake2.h"
  42
  43 #include <malloc.h>
  44 #include <string.h>
  45 #include <stdexcept>
  46
  47 static const size_t HASH_SIZE = 64;
  48
  49 class MUtils::Hash::Blake2_Context
  50 {
  51         friend Blake2;
  52
  53         Blake2_Context(void)
  54         {
  55                 if(!(state = (MUtils::Hash::Internal::Blake2Impl::blake2b_state*) _aligned_malloc(sizeof(MUtils::Hash::Internal::Blake2Impl::blake2b_state), HASH_SIZE)))
  56                 {
  57                         MUTILS_THROW("Aligend malloc has failed!");
  58                 }
  59                 memset(state, 0, sizeof(MUtils::Hash::Internal::Blake2Impl::blake2b_state));
  60         }
  61
  62         ~Blake2_Context(void)
  63         {
  64                 memset(state, 0, sizeof(MUtils::Hash::Internal::Blake2Impl::blake2b_state));
  65                 _aligned_free(state);
  66         }
  67
  68 private:
  69         MUtils::Hash::Internal::Blake2Impl::blake2b_state *state;
  70 };
  71
  72 MUtils::Hash::Blake2::Blake2(const char *const key)
  73 :
  74         m_context(new Blake2_Context()),
  75         m_finalized(false)
  76 {
  77         if(key && key[0])
  78         {
  79                 blake2b_init_key(m_context->state, HASH_SIZE, key, (uint8_t)strlen(key));
  80         }
  81         else
  82         {
  83                 blake2b_init(m_context->state, HASH_SIZE);
  84         }
  85 }
  86
  87 MUtils::Hash::Blake2::~Blake2(void)
  88 {
  89         delete m_context;
  90 }
  91
  92 bool MUtils::Hash::Blake2::process(const quint8 *const data, const quint32 len)
  93 {
  94         if(m_finalized)
  95         {
  96                 MUTILS_THROW("BLAKE2 was already finalized!");
  97         }
  98
  99         if(data && (len > 0))
 100         {
 101                 if(blake2b_update(m_context->state, data, len) != 0)
 102                 {
 103                         MUTILS_THROW("BLAKE2 internal error!");
 104                 }
 105         }
 106
 107         return true;
 108 }
 109
 110 QByteArray MUtils::Hash::Blake2::finalize(void)
 111 {
 112         QByteArray result(HASH_SIZE, '\0');
 113         if(blake2b_final(m_context->state, (uint8_t*) result.data(), result.size()) != 0)
 114         {
 115                 MUTILS_THROW("BLAKE2 internal error!");
 116         }
 117         m_finalized = true;
 118         return result;
 119 }