X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=src%2Flib%2Fencrypted_ecinema_kdm.cc;h=faea034249a5d8c9294737aad8f408182376db4f;hp=e277eb997c46602b34e116f6d7ada4b544d21b55;hb=5eb8b5c3a1566aef638e9d9df03b88d320735092;hpb=cb0a41355327f0308e5b56c21bff1d1ac004a9cb

diff --git a/src/lib/encrypted_ecinema_kdm.cc b/src/lib/encrypted_ecinema_kdm.cc
index e277eb997..faea03424 100644
--- a/src/lib/encrypted_ecinema_kdm.cc
+++ b/src/lib/encrypted_ecinema_kdm.cc
@@ -21,22 +21,59 @@
 #ifdef DCPOMATIC_VARIANT_SWAROOP
 
 #include "encrypted_ecinema_kdm.h"
+#include "ecinema_kdm_data.h"
+#include "exceptions.h"
+#include "cross.h"
 #include <dcp/key.h>
 #include <dcp/certificate.h>
+#include <dcp/util.h>
+#include <libcxml/cxml.h>
 #include <libxml++/libxml++.h>
 #include <openssl/rsa.h>
+#include <openssl/err.h>
 #include <iostream>
 
 using std::cout;
 using std::string;
 using boost::shared_ptr;
+using boost::optional;
 using dcp::Certificate;
 
-EncryptedECinemaKDM::EncryptedECinemaKDM (dcp::Key content_key, Certificate recipient)
+EncryptedECinemaKDM::EncryptedECinemaKDM (string id, string name, dcp::Key content_key, optional<dcp::LocalTime> not_valid_before, optional<dcp::LocalTime> not_valid_after, Certificate recipient)
+	: _id (id)
+	, _name (name)
 {
 	RSA* rsa = recipient.public_key ();
-	_content_key = dcp::Data (RSA_size(rsa));
-	int const N = RSA_public_encrypt (content_key.length(), content_key.value(), _content_key.data().get(), rsa, RSA_PKCS1_OAEP_PADDING);
+	_data = dcp::Data (RSA_size(rsa));
+
+	int input_size = ECINEMA_KDM_KEY_LENGTH;
+	if (not_valid_before && not_valid_after) {
+		input_size += ECINEMA_KDM_NOT_VALID_BEFORE_LENGTH + ECINEMA_KDM_NOT_VALID_AFTER_LENGTH;
+	}
+
+	dcp::Data input (input_size);
+	memcpy (input.data().get(), content_key.value(), ECINEMA_KDM_KEY_LENGTH);
+	if (not_valid_before && not_valid_after) {
+		memcpy (input.data().get() + ECINEMA_KDM_NOT_VALID_BEFORE, not_valid_before->as_string().c_str(), ECINEMA_KDM_NOT_VALID_BEFORE_LENGTH);
+		memcpy (input.data().get() + ECINEMA_KDM_NOT_VALID_AFTER, not_valid_after->as_string().c_str(), ECINEMA_KDM_NOT_VALID_AFTER_LENGTH);
+	}
+
+	int const N = RSA_public_encrypt (input_size, input.data().get(), _data.data().get(), rsa, RSA_PKCS1_OAEP_PADDING);
+	if (N == -1) {
+		throw KDMError ("Could not encrypt ECinema KDM", ERR_error_string(ERR_get_error(), 0));
+	}
+
+}
+
+EncryptedECinemaKDM::EncryptedECinemaKDM (string xml)
+{
+	cxml::Document doc ("ECinemaSecurityMessage");
+	doc.read_string (xml);
+	_id = doc.string_child ("Id");
+	_name = doc.string_child ("Name");
+	_data = dcp::Data (256);
+	int const len = dcp::base64_decode (doc.string_child("Data"), _data.data().get(), _data.size());
+	_data.set_size (len);
 }
 
 string
@@ -45,8 +82,8 @@ EncryptedECinemaKDM::as_xml () const
 	string key;
 
 	/* Lazy overallocation */
-	char out[_content_key.size() * 2];
-	Kumu::base64encode (_content_key.data().get(), _content_key.size(), out, _content_key.size() * 2);
+	char out[_data.size() * 2];
+	Kumu::base64encode (_data.data().get(), _data.size(), out, _data.size() * 2);
 	int const N = strlen (out);
 	string lines;
 	for (int i = 0; i < N; ++i) {
@@ -57,9 +94,20 @@ EncryptedECinemaKDM::as_xml () const
 	}
 
 	xmlpp::Document document;
-	xmlpp::Element* root = document.create_root_node ("ECinemaSecurityMesage");
-	root->add_child("Key")->add_child_text(lines);
+	xmlpp::Element* root = document.create_root_node ("ECinemaSecurityMessage");
+	root->add_child("Id")->add_child_text(_id);
+	root->add_child("Name")->add_child_text(_name);
+	root->add_child("Data")->add_child_text(lines);
 	return document.write_to_string ("UTF-8");
 }
 
+void
+EncryptedECinemaKDM::as_xml (boost::filesystem::path path) const
+{
+	FILE* f = fopen_boost (path, "w");
+	string const x = as_xml ();
+	fwrite (x.c_str(), 1, x.length(), f);
+	fclose (f);
+}
+
 #endif