src/checksum.cpp

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