2 Copyright (C) 2013-2019 Carl Hetherington <cth@carlh.net>
4 This file is part of libdcp.
6 libdcp 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.
11 libdcp 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.
16 You should have received a copy of the GNU General Public License
17 along with libdcp. If not, see <http://www.gnu.org/licenses/>.
19 In addition, as a special exception, the copyright holders give
20 permission to link the code of portions of this program with the
21 OpenSSL library under certain conditions as described in each
22 individual source file, and distribute linked combinations
25 You must obey the GNU General Public License in all respects
26 for all of the code used other than OpenSSL. If you modify
27 file(s) with this exception, you may extend this exception to your
28 version of the file(s), but you are not obligated to do so. If you
29 do not wish to do so, delete this exception statement from your
30 version. If you delete this exception statement from all source
31 files in the program, then also delete it here.
34 #include "colour_conversion.h"
37 #include "decrypted_kdm.h"
38 #include "encrypted_kdm.h"
39 #include "mono_picture_asset.h"
40 #include "mono_picture_asset_reader.h"
41 #include "mono_picture_frame.h"
42 #include "openjpeg_image.h"
43 #include "picture_asset_writer.h"
45 #include "reel_file_asset.h"
46 #include "reel_mono_picture_asset.h"
47 #include "reel_picture_asset.h"
48 #include "reel_sound_asset.h"
49 #include "reel_smpte_subtitle_asset.h"
51 #include "smpte_subtitle_asset.h"
52 #include "sound_asset.h"
53 #include "sound_asset_writer.h"
54 #include "stream_operators.h"
56 #include <boost/test/unit_test.hpp>
57 #include <boost/scoped_array.hpp>
61 #define M_PI (3.14159265358979323846)
65 using std::dynamic_pointer_cast;
69 using std::shared_ptr;
72 using boost::optional;
73 using boost::scoped_array;
76 pair<vector<uint8_t>, dcp::Size>
77 get_frame (dcp::DCP const & dcp)
79 auto reel = dcp.cpls().front()->reels()[0];
80 auto picture = reel->main_picture()->asset();
81 BOOST_CHECK (picture);
83 auto mono_picture = dynamic_pointer_cast<const dcp::MonoPictureAsset>(picture);
84 auto reader = mono_picture->start_read();
85 auto j2k_frame = reader->get_frame(0);
86 auto xyz = j2k_frame->xyz_image();
88 std::vector<uint8_t> argb(xyz->size().width * xyz->size().height * 4);
89 dcp::xyz_to_rgba(j2k_frame->xyz_image(), dcp::ColourConversion::srgb_to_xyz(), argb.data(), xyz->size().width * 4);
90 return make_pair (argb, xyz->size ());
93 /** Decrypt an encrypted test DCP and check that its first frame is the same as the unencrypted version */
94 BOOST_AUTO_TEST_CASE (decryption_test1)
96 auto plaintext_path = private_test;
97 plaintext_path /= "TONEPLATES-SMPTE-PLAINTEXT_TST_F_XX-XX_ITL-TD_51-XX_2K_WOE_20111001_WOE_OV";
98 dcp::DCP plaintext (plaintext_path.string ());
100 BOOST_CHECK_EQUAL (plaintext.any_encrypted(), false);
102 auto encrypted_path = private_test;
103 encrypted_path /= "TONEPLATES-SMPTE-ENCRYPTED_TST_F_XX-XX_ITL-TD_51-XX_2K_WOE_20111001_WOE_OV";
104 dcp::DCP encrypted (encrypted_path.string ());
106 BOOST_CHECK_EQUAL (encrypted.any_encrypted(), true);
108 dcp::DecryptedKDM kdm (
110 dcp::file_to_string ("test/data/kdm_TONEPLATES-SMPTE-ENC_.smpte-430-2.ROOT.NOT_FOR_PRODUCTION_20130706_20230702_CAR_OV_t1_8971c838.xml")
112 dcp::file_to_string ("test/data/private.key")
117 auto plaintext_frame = get_frame(plaintext);
118 auto encrypted_frame = get_frame(encrypted);
120 /* Check that plaintext and encrypted are the same */
122 BOOST_CHECK_EQUAL (plaintext_frame.second, encrypted_frame.second);
126 plaintext_frame.first.data(),
127 encrypted_frame.first.data(),
128 plaintext_frame.second.width * plaintext_frame.second.height * 4
134 /** Load in a KDM that didn't work at first */
135 BOOST_AUTO_TEST_CASE (failing_kdm_test)
137 dcp::DecryptedKDM kdm (
138 dcp::EncryptedKDM (dcp::file_to_string ("test/data/target.pem.crt.de5d4eba-e683-41ca-bdda-aa4ad96af3f4.kdm.xml")),
139 dcp::file_to_string("test/data/failing-private.key")
144 /** Make an encrypted SMPTE DCP with picture, sound and subs and check that the keys get distributed to the assets
145 * when we read it back in.
147 BOOST_AUTO_TEST_CASE (decryption_test2)
149 boost::filesystem::path dir = "build/test/decryption_test2";
150 boost::filesystem::create_directory(dir);
152 auto context_id = dcp::make_uuid();
155 auto picture_asset = std::make_shared<dcp::MonoPictureAsset>(dcp::Fraction(24, 1), dcp::Standard::SMPTE);
156 picture_asset->set_key (key);
157 picture_asset->set_context_id (context_id);
158 auto picture_writer = picture_asset->start_write(dir / "picture.mxf", dcp::PictureAsset::Behaviour::MAKE_NEW);
159 dcp::ArrayData picture("test/data/flat_red.j2c");
160 for (int i = 0; i < 24; ++i) {
161 picture_writer->write(picture);
163 picture_writer->finalize();
165 auto sound_asset = std::make_shared<dcp::SoundAsset>(dcp::Fraction(24, 1), 48000, 2, dcp::LanguageTag("en-GB"), dcp::Standard::SMPTE);
166 sound_asset->set_key (key);
167 sound_asset->set_context_id (context_id);
168 auto sound_writer = sound_asset->start_write(dir / "sound.mxf", {}, dcp::SoundAsset::AtmosSync::DISABLED, dcp::SoundAsset::MCASubDescriptors::ENABLED);
169 std::array<float, 48000> left;
170 std::array<float, 48000> right;
171 for (int i = 0; i < 48000; ++i) {
172 left[i] = sin (2 * M_PI * i * 440 / 48000) * 0.25;
173 right[i] = sin (2 * M_PI * i * 880 / 48000) * 0.25;
175 std::array<float*, 2> audio;
176 audio[0] = left.data();
177 audio[1] = right.data();
178 sound_writer->write(audio.data(), 2, 48000);
179 sound_writer->finalize ();
181 auto subs_asset = std::make_shared<dcp::SMPTESubtitleAsset>();
182 subs_asset->set_key (key);
183 subs_asset->set_context_id (context_id);
184 subs_asset->add(std::make_shared<dcp::SubtitleString>(
187 dcp::Colour(255, 255, 255),
191 dcp::Time(0, 0, 5, 0, 24),
192 0.5, dcp::HAlign::CENTER,
193 0.5, dcp::VAlign::CENTER,
198 dcp::Colour(0, 0, 0),
199 dcp::Time(), dcp::Time(), 0, std::vector<dcp::Ruby>()
201 subs_asset->write (dir / "subs.mxf");
203 auto reel = std::make_shared<dcp::Reel>();
204 auto reel_picture_asset = std::make_shared<dcp::ReelMonoPictureAsset>(picture_asset, 0);
205 auto reel_sound_asset = std::make_shared<dcp::ReelSoundAsset>(sound_asset, 0);
206 auto reel_subs_asset = std::make_shared<dcp::ReelSMPTESubtitleAsset>(subs_asset, dcp::Fraction(24, 1), 120, 0);
207 reel->add(reel_picture_asset);
208 reel->add(reel_sound_asset);
209 reel->add(reel_subs_asset);
211 auto cpl = std::make_shared<dcp::CPL>("My film", dcp::ContentKind::FEATURE, dcp::Standard::SMPTE);
218 map<shared_ptr<const dcp::ReelFileAsset>, dcp::Key> keys;
219 keys[reel_picture_asset] = key;
220 keys[reel_sound_asset] = key;
221 keys[reel_subs_asset] = key;
223 dcp::DecryptedKDM kdm (cpl->id(), keys, dcp::LocalTime(), dcp::LocalTime(), "foo", "bar", "baz");
225 dcp::DCP dcp_read (dir);
229 BOOST_REQUIRE_EQUAL (dcp_read.cpls().size(), 1U);
230 auto cpl_read = dcp_read.cpls()[0];
231 BOOST_REQUIRE_EQUAL (cpl_read->reels().size(), 1U);
232 auto reel_read = cpl_read->reels()[0];
234 BOOST_REQUIRE (reel_read->main_picture());
235 BOOST_CHECK (reel_read->main_picture()->asset()->key());
236 BOOST_REQUIRE (reel_read->main_sound());
237 BOOST_CHECK (reel_read->main_sound()->asset()->key());
238 BOOST_REQUIRE (reel_read->main_subtitle());
239 auto smpte_sub = dynamic_pointer_cast<dcp::SMPTESubtitleAsset>(reel_read->main_subtitle()->asset());
240 BOOST_REQUIRE (smpte_sub);
241 BOOST_CHECK (smpte_sub->key());