Various probably quite untidy progress on KDMs.

[libdcp.git] / src / mxf_asset.cc

diff --git a/src/mxf_asset.cc b/src/mxf_asset.cc
index 95412d0cdda42a4e01b8e188e09050ae0c27fe22..6ba42d754f5ed7a15edd7ebbddf4bd03edd53175 100644 (file)
--- a/src/mxf_asset.cc
+++ b/src/mxf_asset.cc
@@ -24,11 +24,14 @@
 #include <iostream>
 #include <fstream>
 #include <boost/filesystem.hpp>
+#include <libxml++/nodes/element.h>
 #include "AS_DCP.h"
+#include "KM_prng.h"
 #include "KM_util.h"
 #include "mxf_asset.h"
 #include "util.h"
 #include "metadata.h"
+#include "exceptions.h"
 
 using std::string;
 using std::list;
@@ -36,14 +39,40 @@ using boost::shared_ptr;
 using boost::dynamic_pointer_cast;
 using namespace libdcp;
 
-MXFAsset::MXFAsset (string directory, string file_name, boost::signals2::signal<void (float)>* progress, int fps, int entry_point, int length)
+MXFAsset::MXFAsset (string directory, string file_name, boost::signals2::signal<void (float)>* progress, int fps, int entry_point, int length, bool encrypted)
        : Asset (directory, file_name)
        , _progress (progress)
        , _fps (fps)
        , _entry_point (entry_point)
        , _length (length)
+       , _encrypted (encrypted)
+       , _encryption_context (0)
 {
-       
+       if (_encrypted) {
+               _key_id = make_uuid ();
+               uint8_t key_buffer[ASDCP::KeyLen];
+               Kumu::FortunaRNG rng;
+               rng.FillRandom (key_buffer, ASDCP::KeyLen);
+               char key_string[ASDCP::KeyLen * 4];
+               Kumu::bin2hex (key_buffer, ASDCP::KeyLen, key_string, ASDCP::KeyLen * 4);
+               _key_value = key_string;
+                       
+               _encryption_context = new ASDCP::AESEncContext;
+               if (ASDCP_FAILURE (_encryption_context->InitKey (key_buffer))) {
+                       throw MiscError ("could not set up encryption context");
+               }
+
+               uint8_t cbc_buffer[ASDCP::CBC_BLOCK_SIZE];
+               
+               if (ASDCP_FAILURE (_encryption_context->SetIVec (rng.FillRandom (cbc_buffer, ASDCP::CBC_BLOCK_SIZE)))) {
+                       throw MiscError ("could not set up CBC initialization vector");
+               }
+       }
+}
+
+MXFAsset::~MXFAsset ()
+{
+       delete _encryption_context;
 }
 
 void
@@ -57,6 +86,15 @@ MXFAsset::fill_writer_info (ASDCP::WriterInfo* writer_info) const
        unsigned int c;
        Kumu::hex2bin (_uuid.c_str(), writer_info->AssetUUID, Kumu::UUID_Length, &c);
        assert (c == Kumu::UUID_Length);
+
+       if (_encrypted) {
+               Kumu::GenRandomUUID (writer_info->ContextID);
+               writer_info->EncryptedEssence = true;
+
+               unsigned int c;
+               Kumu::hex2bin (_key_id.c_str(), writer_info->CryptographicKeyID, Kumu::UUID_Length, &c);
+               assert (c == Kumu::UUID_Length);
+       }
 }
 
 bool
@@ -91,3 +129,11 @@ MXFAsset::length () const
 {
        return _length;
 }
+
+void
+MXFAsset::add_typed_key_id (xmlpp::Element* parent) const
+{
+       xmlpp::Element* typed_key_id = parent->add_child("TypedKeyId");
+       typed_key_id->add_child("KeyType")->add_child_text(key_type ());
+       typed_key_id->add_child("KeyId")->add_child_text("urn:uuid:" + _key_id);
+}