src/Hash_Blake2.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Simple x264 Launcher
   3 // Copyright (C) 2004-2016 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 //MUtils
  36 #include <MUtils/Hash_Blake2.h>
  37 #include <MUtils/Exception.h>
  38
  39 //Internal
  40 #include "3rd_party/blake2/include/blake2.h"
  41
  42 #include <malloc.h>
  43 #include <string.h>
  44 #include <stdexcept>
  45
  46 static const size_t HASH_SIZE = 64;
  47
  48 class MUtils::Hash::Blake2_Context
  49 {
  50         friend Blake2;
  51
  52         Blake2_Context(void)
  53         {
  54                 if(!(state = (MUtils::Hash::Internal::Blake2Impl::blake2b_state*) _aligned_malloc(sizeof(MUtils::Hash::Internal::Blake2Impl::blake2b_state), 64)))
  55                 {
  56                         MUTILS_THROW("Aligend malloc has failed!");
  57                 }
  58                 memset(state, 0, sizeof(MUtils::Hash::Internal::Blake2Impl::blake2b_state));
  59         }
  60
  61         ~Blake2_Context(void)
  62         {
  63                 memset(state, 0, sizeof(MUtils::Hash::Internal::Blake2Impl::blake2b_state));
  64                 _aligned_free(state);
  65         }
  66
  67 private:
  68         MUtils::Hash::Internal::Blake2Impl::blake2b_state *state;
  69 };
  70
  71 MUtils::Hash::Blake2::Blake2(const char* key)
  72 :
  73         m_hash(HASH_SIZE, '\0'),
  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 void MUtils::Hash::Blake2::update(const QByteArray &data)
  93 {
  94         if(m_finalized)
  95         {
  96                 MUTILS_THROW("BLAKE2 was already finalized!");
  97         }
  98
  99         if(data.size() > 0)
 100         {
 101                 if(blake2b_update(m_context->state, (const uint8_t*) data.constData(), data.size()) != 0)
 102                 {
 103                         MUTILS_THROW("BLAKE2 internal error!");
 104                 }
 105         }
 106 }
 107
 108 void MUtils::Hash::Blake2::update(QFile &file)
 109 {
 110         bool okay = false;
 111
 112         for(;;)
 113         {
 114                 QByteArray data = file.read(16384);
 115                 if(data.size() > 0)
 116                 {
 117                         okay = true;
 118                         update(data);
 119                         continue;
 120                 }
 121                 break;
 122         }
 123
 124         if(!okay)
 125         {
 126                 qWarning("[QBlake2Checksum] Could not ready any data from file!");
 127         }
 128 }
 129
 130 QByteArray MUtils::Hash::Blake2::finalize(const bool bAsHex)
 131 {
 132         if(!m_finalized)
 133         {
 134                 if(blake2b_final(m_context->state, (uint8_t*) m_hash.data(), m_hash.size()) != 0)
 135                 {
 136                         MUTILS_THROW("BLAKE2 internal error!");
 137                 }
 138
 139                 m_finalized = true;
 140         }
 141
 142         return bAsHex ? m_hash.toHex() : m_hash;
 143 }