swaroop: basics of encrypted MP4 playback.
[dcpomatic.git] / src / lib / encrypted_ecinema_kdm.cc
1 /*
2     Copyright (C) 2019 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 #ifdef DCPOMATIC_VARIANT_SWAROOP
22
23 #include "encrypted_ecinema_kdm.h"
24 #include <dcp/key.h>
25 #include <dcp/certificate.h>
26 #include <dcp/util.h>
27 #include <libcxml/cxml.h>
28 #include <libxml++/libxml++.h>
29 #include <openssl/rsa.h>
30 #include <iostream>
31
32 using std::cout;
33 using std::string;
34 using boost::shared_ptr;
35 using dcp::Certificate;
36
37 EncryptedECinemaKDM::EncryptedECinemaKDM (string id, dcp::Key content_key, Certificate recipient)
38         : _id (id)
39 {
40         RSA* rsa = recipient.public_key ();
41         _content_key = dcp::Data (RSA_size(rsa));
42         int const N = RSA_public_encrypt (content_key.length(), content_key.value(), _content_key.data().get(), rsa, RSA_PKCS1_OAEP_PADDING);
43 }
44
45 EncryptedECinemaKDM::EncryptedECinemaKDM (string xml)
46 {
47         cxml::Document doc ("ECinemaSecurityMessage");
48         doc.read_string (xml);
49         _id = doc.string_child ("Id");
50         _content_key = dcp::Data (256);
51         int const len = dcp::base64_decode (doc.string_child("Key"), _content_key.data().get(), _content_key.size());
52         _content_key.set_size (len);
53 }
54
55 string
56 EncryptedECinemaKDM::as_xml () const
57 {
58         string key;
59
60         /* Lazy overallocation */
61         char out[_content_key.size() * 2];
62         Kumu::base64encode (_content_key.data().get(), _content_key.size(), out, _content_key.size() * 2);
63         int const N = strlen (out);
64         string lines;
65         for (int i = 0; i < N; ++i) {
66                 if (i > 0 && (i % 64) == 0) {
67                         lines += "\n";
68                 }
69                 lines += out[i];
70         }
71
72         xmlpp::Document document;
73         xmlpp::Element* root = document.create_root_node ("ECinemaSecurityMessage");
74         root->add_child("Id")->add_child_text(_id);
75         root->add_child("Key")->add_child_text(lines);
76         return document.write_to_string ("UTF-8");
77 }
78
79 #endif