src/lib/digester.cc

   1 /*
   2     Copyright (C) 2014-2021 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of DCP-o-matic.
   5
   6     DCP-o-matic is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     DCP-o-matic is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21
  22 #include "digester.h"
  23 #include "dcpomatic_assert.h"
  24 #include <nettle/md5.h>
  25 #include <iomanip>
  26 #include <cstdio>
  27
  28
  29 using std::string;
  30 using std::hex;
  31 using std::setfill;
  32 using std::setw;
  33
  34
  35 Digester::Digester ()
  36 {
  37         md5_init (&_context);
  38 }
  39
  40
  41 Digester::~Digester ()
  42 {
  43         get ();
  44 }
  45
  46
  47 void
  48 Digester::add (void const * data, size_t size)
  49 {
  50         md5_update (&_context, size, reinterpret_cast<uint8_t const *>(data));
  51 }
  52
  53
  54 void
  55 Digester::add (string const & s)
  56 {
  57         add (s.c_str(), s.length());
  58 }
  59
  60
  61 string
  62 Digester::get () const
  63 {
  64         if (!_digest) {
  65                 unsigned char digest[MD5_DIGEST_SIZE];
  66                 md5_digest (&_context, MD5_DIGEST_SIZE, digest);
  67
  68                 char hex[MD5_DIGEST_SIZE * 2 + 1];
  69                 for (int i = 0; i < MD5_DIGEST_SIZE; ++i) {
  70                         sprintf(hex + i * 2, "%02x", digest[i]);
  71                 }
  72
  73                 _digest = hex;
  74         }
  75
  76         return _digest.get ();
  77 }
  78
  79
  80 void
  81 Digester::get (uint8_t* buffer) const
  82 {
  83         md5_digest (&_context, MD5_DIGEST_SIZE, buffer);
  84 }
  85
  86
  87 int
  88 Digester::size () const
  89 {
  90         return MD5_DIGEST_SIZE;
  91 }