src/util.cc

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include <stdexcept>
  21 #include <sstream>
  22 #include <iostream>
  23 #include <iomanip>
  24 #include <boost/filesystem.hpp>
  25 #include <openssl/sha.h>
  26 #include "KM_util.h"
  27 #include "KM_fileio.h"
  28 #include "AS_DCP.h"
  29 #include "util.h"
  30
  31 using namespace std;
  32 using namespace boost;
  33
  34 /** Create a UUID.
  35  *  @return UUID.
  36  */
  37 string
  38 libdcp::make_uuid ()
  39 {
  40         char buffer[64];
  41         Kumu::UUID id;
  42         Kumu::GenRandomValue (id);
  43         id.EncodeHex (buffer, 64);
  44         return string (buffer);
  45 }
  46
  47 /** Create a digest for a file.
  48  *  @param filename File name.
  49  *  @return Digest.
  50  */
  51 string
  52 libdcp::make_digest (string filename, sigc::signal1<void, float>* progress)
  53 {
  54         int const file_size = filesystem::file_size (filename);
  55
  56         Kumu::FileReader reader;
  57         if (ASDCP_FAILURE (reader.OpenRead (filename.c_str ()))) {
  58                 throw runtime_error ("could not open file to compute digest");
  59         }
  60
  61         SHA_CTX sha;
  62         SHA1_Init (&sha);
  63
  64         Kumu::ByteString read_buffer (65536);
  65         int done = 0;
  66         while (1) {
  67                 ui32_t read = 0;
  68                 Kumu::Result_t r = reader.Read (read_buffer.Data(), read_buffer.Capacity(), &read);
  69
  70                 if (r == Kumu::RESULT_ENDOFFILE) {
  71                         break;
  72                 } else if (ASDCP_FAILURE (r)) {
  73                         throw runtime_error ("could not read file to compute digest");
  74                 }
  75
  76                 SHA1_Update (&sha, read_buffer.Data(), read);
  77                 done += read;
  78
  79                 if (progress) {
  80                         (*progress) (0.5 + (0.5 * done / file_size));
  81                 }
  82         }
  83
  84         byte_t byte_buffer[20];
  85         SHA1_Final (byte_buffer, &sha);
  86
  87         stringstream s;
  88         char digest[64];
  89         return Kumu::base64encode (byte_buffer, 20, digest, 64);
  90 }