2 Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
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.
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.
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.
22 #include <boost/algorithm/string.hpp>
23 #include <openssl/rsa.h>
24 #include <openssl/pem.h>
25 #include <openssl/err.h>
26 #include <libcxml/cxml.h>
31 #include "compose.hpp"
32 #include "exceptions.h"
35 #include "mxf_asset.h"
36 #include "xml/kdm_smpte.h"
40 using std::stringstream;
45 using boost::shared_ptr;
46 using namespace libdcp;
48 KDM::KDM (boost::filesystem::path kdm, boost::filesystem::path private_key)
49 : _xml_kdm (new xml::DCinemaSecurityMessage (kdm))
51 /* Read the private key */
53 FILE* private_key_file = fopen (private_key.string().c_str(), "r");
54 if (!private_key_file) {
55 throw FileError ("could not find RSA private key file", private_key);
58 RSA* rsa = PEM_read_RSAPrivateKey (private_key_file, 0, 0, 0);
59 fclose (private_key_file);
61 throw FileError ("could not read RSA private key file", private_key);
64 /* Use it to decrypt the keys */
66 list<string> encrypted_keys = _xml_kdm->authenticated_private.encrypted_keys;
68 for (list<string>::iterator i = encrypted_keys.begin(); i != encrypted_keys.end(); ++i) {
70 /* Decode the base-64-encoded cipher value from the KDM */
71 unsigned char cipher_value[256];
72 int const cipher_value_len = base64_decode (*i, cipher_value, sizeof (cipher_value));
75 unsigned char* decrypted = new unsigned char[RSA_size(rsa)];
76 int const decrypted_len = RSA_private_decrypt (cipher_value_len, cipher_value, decrypted, rsa, RSA_PKCS1_OAEP_PADDING);
77 if (decrypted_len == -1) {
79 throw MiscError (String::compose ("Could not decrypt KDM (%1)", ERR_error_string (ERR_get_error(), 0)));
82 _keys.push_back (KDMKey (decrypted, decrypted_len));
90 shared_ptr<const CPL> cpl, shared_ptr<const Signer> signer, shared_ptr<const Certificate> recipient_cert,
91 boost::posix_time::ptime not_valid_before, boost::posix_time::ptime not_valid_after,
92 string annotation_text, string issue_date
94 : _xml_kdm (new xml::DCinemaSecurityMessage)
96 xml::AuthenticatedPublic& apu = _xml_kdm->authenticated_public;
98 /* AuthenticatedPublic */
100 apu.message_id = "urn:uuid:" + make_uuid ();
101 apu.message_type = "http://www.smpte-ra.org/430-1/2006/KDM#kdm-key-type";
102 apu.annotation_text = annotation_text;
103 apu.issue_date = issue_date;
104 apu.signer.x509_issuer_name = signer->certificates().leaf()->issuer ();
105 apu.signer.x509_serial_number = signer->certificates().leaf()->serial ();
106 apu.recipient.x509_issuer_serial.x509_issuer_name = recipient_cert->issuer ();
107 apu.recipient.x509_issuer_serial.x509_serial_number = recipient_cert->serial ();
108 apu.recipient.x509_subject_name = recipient_cert->subject ();
109 apu.composition_playlist_id = "urn:uuid:" + cpl->id ();
110 // apu.content_authenticator = signer->certificates().leaf()->thumbprint ();
111 apu.content_title_text = cpl->name ();
112 apu.content_keys_not_valid_before = ptime_to_string (not_valid_before);
113 apu.content_keys_not_valid_after = ptime_to_string (not_valid_after);
114 apu.authorized_device_info.device_list_identifier = "urn:uuid:" + make_uuid ();
115 string n = recipient_cert->common_name ();
116 if (n.find (".") != string::npos) {
117 n = n.substr (n.find (".") + 1);
119 apu.authorized_device_info.device_list_description = n;
120 apu.authorized_device_info.device_list.push_back (recipient_cert->thumbprint ());
122 list<shared_ptr<const Asset> > assets = cpl->assets ();
123 for (list<shared_ptr<const Asset> >::iterator i = assets.begin(); i != assets.end(); ++i) {
124 /* XXX: non-MXF assets? */
125 shared_ptr<const MXFAsset> mxf = boost::dynamic_pointer_cast<const MXFAsset> (*i);
127 apu.key_id_list.push_back (xml::TypedKeyId (mxf->key_type(), "urn:uuid:" + mxf->key_id()));
131 apu.forensic_mark_flag_list.push_back ("http://www.smpte-ra.org/430-1/2006/KDM#mrkflg-picture-disable");
132 apu.forensic_mark_flag_list.push_back ("http://www.smpte-ra.org/430-1/2006/KDM#mrkflg-audio-disable");
134 /* AuthenticatedPrivate */
136 for (list<shared_ptr<const Asset> >::iterator i = assets.begin(); i != assets.end(); ++i) {
137 /* XXX: non-MXF assets? */
138 shared_ptr<const MXFAsset> mxf = boost::dynamic_pointer_cast<const MXFAsset> (*i);
141 signer, cpl->id (), mxf->key_type (), mxf->key_id (),
142 not_valid_before, not_valid_after, mxf->key().get()
145 _keys.push_back (kkey);
146 _xml_kdm->authenticated_private.encrypted_keys.push_back (kkey.encrypted_base64 (recipient_cert));
152 shared_ptr<xmlpp::Document> doc = _xml_kdm->as_xml ();
153 shared_ptr<cxml::Node> root (new cxml::Node (doc->get_root_node ()));
154 xmlpp::Node* signature = root->node_child("Signature")->node();
155 signer->add_signature_value (signature, "ds");
156 _xml_kdm->signature = xml::Signature (shared_ptr<cxml::Node> (new cxml::Node (signature)));
159 KDM::KDM (KDM const & other)
160 : _keys (other._keys)
161 , _xml_kdm (new xml::DCinemaSecurityMessage (*other._xml_kdm.get()))
167 KDM::operator= (KDM const & other)
169 if (this == &other) {
174 _xml_kdm.reset (new xml::DCinemaSecurityMessage (*other._xml_kdm.get ()));
180 KDM::as_xml (boost::filesystem::path path) const
182 shared_ptr<xmlpp::Document> doc = _xml_kdm->as_xml ();
183 /* This must *not* be the _formatted version, otherwise the signature
186 doc->write_to_file (path.string(), "UTF-8");
192 shared_ptr<xmlpp::Document> doc = _xml_kdm->as_xml ();
193 /* This must *not* be the _formatted version, otherwise the signature
196 return doc->write_to_string ("UTF-8");
200 shared_ptr<const Signer> signer, string cpl_id, string key_type, string key_id, boost::posix_time::ptime from, boost::posix_time::ptime until, Key key
203 , _key_type (key_type)
205 , _not_valid_before (ptime_to_string (from))
206 , _not_valid_after (ptime_to_string (until))
209 base64_decode (signer->certificates().leaf()->thumbprint (), _signer_thumbprint, 20);
212 KDMKey::KDMKey (uint8_t const * raw, int len)
217 /* [0-15] is structure id (fixed sequence specified by standard) */
219 get (_signer_thumbprint, &raw, 20);
220 _cpl_id = get_uuid (&raw);
221 _key_id = get_uuid (&raw);
222 _not_valid_before = get (&raw, 25);
223 _not_valid_after = get (&raw, 25);
228 /* [0-15] is structure id (fixed sequence specified by standard) */
230 get (_signer_thumbprint, &raw, 20);
231 _cpl_id = get_uuid (&raw);
232 _key_type = get (&raw, 4);
233 _key_id = get_uuid (&raw);
234 _not_valid_before = get (&raw, 25);
235 _not_valid_after = get (&raw, 25);
243 KDMKey::KDMKey (KDMKey const & other)
244 : _cpl_id (other._cpl_id)
245 , _key_type (other._key_type)
246 , _key_id (other._key_id)
247 , _not_valid_before (other._not_valid_before)
248 , _not_valid_after (other._not_valid_after)
251 memcpy (_signer_thumbprint, other._signer_thumbprint, 20);
255 KDMKey::operator= (KDMKey const & other)
257 if (&other == this) {
261 _cpl_id = other._cpl_id;
262 _key_type = other._key_type;
263 _key_id = other._key_id;
264 _not_valid_before = other._not_valid_before;
265 _not_valid_after = other._not_valid_after;
267 memcpy (_signer_thumbprint, other._signer_thumbprint, 20);
273 KDMKey::encrypted_base64 (shared_ptr<const Certificate> recipient_cert) const
275 assert (_key_type.length() == 4);
276 assert (_not_valid_before.length() == 25);
277 assert (_not_valid_after.length() == 25);
279 /* XXX: SMPTE only */
283 /* Magic value specified by SMPTE S430-1-2006 */
284 uint8_t structure_id[] = { 0xf1, 0xdc, 0x12, 0x44, 0x60, 0x16, 0x9a, 0x0e, 0x85, 0xbc, 0x30, 0x06, 0x42, 0xf8, 0x66, 0xab };
285 put (&p, structure_id, 16);
286 put (&p, _signer_thumbprint, 20);
287 put_uuid (&p, _cpl_id);
289 put_uuid (&p, _key_id);
290 put (&p, _not_valid_before);
291 put (&p, _not_valid_after);
292 put (&p, _key.value(), ASDCP::KeyLen);
294 /* Encrypt using the projector's public key */
295 RSA* rsa = recipient_cert->public_key ();
296 unsigned char encrypted[RSA_size(rsa)];
297 int const encrypted_len = RSA_public_encrypt (p - block, block, encrypted, rsa, RSA_PKCS1_OAEP_PADDING);
298 if (encrypted_len == -1) {
299 throw MiscError (String::compose ("Could not encrypt KDM (%1)", ERR_error_string (ERR_get_error(), 0)));
302 /* Lazy overallocation */
303 char out[encrypted_len * 2];
304 Kumu::base64encode (encrypted, encrypted_len, out, encrypted_len * 2);
305 int const N = strlen (out);
307 for (int i = 0; i < N; ++i) {
308 if (i > 0 && (i % 64) == 0) {
318 KDMKey::get (uint8_t const ** p, int N) const
321 for (int i = 0; i < N; ++i) {
330 KDMKey::get (uint8_t* o, uint8_t const ** p, int N) const
337 KDMKey::get_uuid (unsigned char const ** p) const
341 for (int i = 0; i < 16; ++i) {
342 g << setw(2) << setfill('0') << hex << static_cast<int> (**p);
344 if (i == 3 || i == 5 || i == 7 || i == 9) {
353 KDMKey::put (uint8_t ** d, uint8_t const * s, int N) const
360 KDMKey::put (uint8_t ** d, string s) const
362 memcpy (*d, s.c_str(), s.length());
367 KDMKey::put_uuid (uint8_t ** d, string id) const
369 id.erase (std::remove (id.begin(), id.end(), '-'));
370 for (int i = 0; i < 32; i += 2) {
372 s << id[i] << id[i + 1];
381 libdcp::operator== (libdcp::KDMKey const & a, libdcp::KDMKey const & b)
383 if (memcmp (a._signer_thumbprint, b._signer_thumbprint, 20) != 0) {
388 a._cpl_id == b._cpl_id &&
389 a._key_type == b._key_type &&
390 a._key_id == b._key_id &&
391 a._not_valid_before == b._not_valid_before &&
392 a._not_valid_after == b._not_valid_after &&