Import.

1 files changed, 95 insertions, 0 deletions

diff --git a/src/util.cc b/src/util.cc
new file mode 100644
index 00000000..aab2e184
--- /dev/null
+++ b/src/util.cc
@@ -0,0 +1,95 @@
+/*
+    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, write to the Free Software
+    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+*/
+
+#include <stdexcept>
+#include <sstream>
+#include <iomanip>
+#include <openssl/sha.h>
+#include "KM_util.h"
+#include "KM_fileio.h"
+#include "AS_DCP.h"
+#include "util.h"
+
+using namespace std;
+
+bool libdcp::libdcp_test = false;
+
+/** Create a UUID.
+ *  @return UUID.
+ */
+string
+libdcp::make_uuid ()
+{
+	char buffer[64];
+	Kumu::UUID id;
+
+	if (libdcp_test) {
+		static int N = 0;
+		byte_t t[16];
+		for (int i = 0; i < 16; ++i) {
+			t[i] = N;
+		}
+		++N;
+		
+		id = Kumu::UUID (t);
+	} else {
+		Kumu::GenRandomValue (id);
+	}
+	
+	id.EncodeHex (buffer, 64);
+	return string (buffer);
+}
+
+/** Create a digest for a file.
+ *  @param filename File name.
+ *  @return Digest.
+ */
+string
+libdcp::make_digest (string filename)
+{
+	Kumu::FileReader reader;
+	if (ASDCP_FAILURE (reader.OpenRead (filename.c_str ()))) {
+		throw runtime_error ("could not open file to compute digest");
+	}
+	
+	SHA_CTX sha;
+	SHA1_Init (&sha);
+	
+	Kumu::ByteString read_buffer (65536);
+	while (1) {
+		ui32_t read = 0;
+		Kumu::Result_t r = reader.Read (read_buffer.Data(), read_buffer.Capacity(), &read);
+		
+		if (r == Kumu::RESULT_ENDOFFILE) {
+			break;
+		} else if (ASDCP_FAILURE (r)) {
+			throw runtime_error ("could not read file to compute digest");
+		}
+		
+		SHA1_Update (&sha, read_buffer.Data(), read);
+	}
+
+	byte_t byte_buffer[20];
+	SHA1_Final (byte_buffer, &sha);
+
+	stringstream s;
+	char digest[64];
+	s << setfill('0') << setw(36) << Kumu::base64encode (byte_buffer, 20, digest, 64);
+	return s.str ();
+}