test/encryption_test.cc

   1 /*
   2     Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of libdcp.
   5
   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.
  10
  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.
  15
  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/>.
  18 */
  19
  20 #include "metadata.h"
  21 #include "certificate.h"
  22 #include "dcp.h"
  23 #include "certificate_chain.h"
  24 #include "cpl.h"
  25 #include "mono_picture_asset.h"
  26 #include "picture_asset_writer.h"
  27 #include "sound_asset_writer.h"
  28 #include "sound_asset.h"
  29 #include "reel.h"
  30 #include "test.h"
  31 #include "file.h"
  32 #include "subtitle_asset.h"
  33 #include "reel_mono_picture_asset.h"
  34 #include "reel_sound_asset.h"
  35 #include "encrypted_kdm.h"
  36 #include "decrypted_kdm.h"
  37 #include <asdcp/KM_util.h>
  38 #include <sndfile.h>
  39 #include <boost/test/unit_test.hpp>
  40 #include <boost/shared_ptr.hpp>
  41
  42 using std::vector;
  43 using boost::shared_ptr;
  44
  45 /** Load a certificate chain from build/test/data/ *.pem and then build
  46  *  an encrypted DCP and a KDM using it.
  47  */
  48 BOOST_AUTO_TEST_CASE (encryption_test)
  49 {
  50         boost::filesystem::remove_all ("build/test/signer");
  51         boost::filesystem::create_directory ("build/test/signer");
  52
  53         Kumu::cth_test = true;
  54
  55         dcp::MXFMetadata mxf_metadata;
  56         mxf_metadata.company_name = "OpenDCP";
  57         mxf_metadata.product_name = "OpenDCP";
  58         mxf_metadata.product_version = "0.0.25";
  59
  60         dcp::XMLMetadata xml_metadata;
  61         xml_metadata.annotation_text = "A Test DCP";
  62         xml_metadata.issuer = "OpenDCP 0.0.25";
  63         xml_metadata.creator = "OpenDCP 0.0.25";
  64         xml_metadata.issue_date = "2012-07-17T04:45:18+00:00";
  65
  66         boost::filesystem::remove_all ("build/test/DCP/encryption_test");
  67         boost::filesystem::create_directories ("build/test/DCP/encryption_test");
  68         dcp::DCP d ("build/test/DCP/encryption_test");
  69
  70         /* Use test/ref/crypt so this test is repeatable */
  71         shared_ptr<dcp::CertificateChain> signer (new dcp::CertificateChain ());
  72         signer->add (dcp::Certificate (dcp::file_to_string ("test/ref/crypt/ca.self-signed.pem")));
  73         signer->add (dcp::Certificate (dcp::file_to_string ("test/ref/crypt/intermediate.signed.pem")));
  74         signer->add (dcp::Certificate (dcp::file_to_string ("test/ref/crypt/leaf.signed.pem")));
  75         signer->set_key (dcp::file_to_string ("test/ref/crypt/leaf.key"));
  76
  77         shared_ptr<dcp::CPL> cpl (new dcp::CPL ("A Test DCP", dcp::FEATURE));
  78
  79         dcp::Key key;
  80
  81         shared_ptr<dcp::MonoPictureAsset> mp (new dcp::MonoPictureAsset (dcp::Fraction (24, 1), dcp::SMPTE));
  82         mp->set_metadata (mxf_metadata);
  83         mp->set_key (key);
  84
  85         shared_ptr<dcp::PictureAssetWriter> writer = mp->start_write ("build/test/DCP/encryption_test/video.mxf", false);
  86         dcp::File j2c ("test/data/32x32_red_square.j2c");
  87         for (int i = 0; i < 24; ++i) {
  88                 writer->write (j2c.data (), j2c.size ());
  89         }
  90         writer->finalize ();
  91
  92         shared_ptr<dcp::SoundAsset> ms (new dcp::SoundAsset (dcp::Fraction (24, 1), 48000, 1, dcp::SMPTE));
  93         ms->set_metadata (mxf_metadata);
  94         ms->set_key (key);
  95         shared_ptr<dcp::SoundAssetWriter> sound_writer = ms->start_write ("build/test/DCP/encryption_test/audio.mxf");
  96
  97         SF_INFO info;
  98         info.format = 0;
  99         SNDFILE* sndfile = sf_open ("test/data/1s_24-bit_48k_silence.wav", SFM_READ, &info);
 100         BOOST_CHECK (sndfile);
 101         float buffer[4096*6];
 102         float* channels[1];
 103         channels[0] = buffer;
 104         while (1) {
 105                 sf_count_t N = sf_readf_float (sndfile, buffer, 4096);
 106                 sound_writer->write (channels, N);
 107                 if (N < 4096) {
 108                         break;
 109                 }
 110         }
 111
 112         sound_writer->finalize ();
 113
 114         cpl->add (shared_ptr<dcp::Reel> (new dcp::Reel (
 115                                                  shared_ptr<dcp::ReelMonoPictureAsset> (new dcp::ReelMonoPictureAsset (mp, 0)),
 116                                                  shared_ptr<dcp::ReelSoundAsset> (new dcp::ReelSoundAsset (ms, 0)),
 117                                                  shared_ptr<dcp::ReelSubtitleAsset> ()
 118                                                  )));
 119         cpl->set_content_version_id ("urn:uri:81fb54df-e1bf-4647-8788-ea7ba154375b_2012-07-17T04:45:18+00:00");
 120         cpl->set_content_version_label_text ("81fb54df-e1bf-4647-8788-ea7ba154375b_2012-07-17T04:45:18+00:00");
 121         cpl->set_metadata (xml_metadata);
 122
 123         d.add (cpl);
 124
 125         xml_metadata.annotation_text = "Created by libdcp";
 126         d.write_xml (dcp::SMPTE, xml_metadata, signer);
 127
 128         dcp::DecryptedKDM kdm (
 129                 cpl,
 130                 key,
 131                 dcp::LocalTime ("2013-01-01T00:00:00+00:00"),
 132                 dcp::LocalTime ("2017-01-08T00:00:00+00:00"),
 133                 "libdcp",
 134                 "test",
 135                 "2012-07-17T04:45:18+00:00"
 136                 );
 137
 138         kdm.encrypt (signer, signer->leaf(), vector<dcp::Certificate>(), dcp::MODIFIED_TRANSITIONAL_1, true, 0).as_xml ("build/test/encryption_test.kdm.xml");
 139
 140         int r = system (
 141                 "xmllint --path schema --nonet --noout --schema schema/SMPTE-430-1-2006-Amd-1-2009-KDM.xsd build/test/encryption_test.kdm.xml "
 142                 "> build/test/xmllint.log 2>&1 < /dev/null"
 143                 );
 144
 145 #ifdef LIBDCP_POSIX
 146         BOOST_CHECK_EQUAL (WEXITSTATUS (r), 0);
 147 #else
 148         BOOST_CHECK_EQUAL (r, 0);
 149 #endif
 150
 151         r = system ("xmlsec1 verify "
 152                 "--pubkey-cert-pem test/ref/crypt/leaf.signed.pem "
 153                 "--trusted-pem test/ref/crypt/intermediate.signed.pem "
 154                 "--trusted-pem test/ref/crypt/ca.self-signed.pem "
 155                 "--id-attr:Id http://www.smpte-ra.org/schemas/430-3/2006/ETM:AuthenticatedPublic "
 156                 "--id-attr:Id http://www.smpte-ra.org/schemas/430-3/2006/ETM:AuthenticatedPrivate "
 157                     "build/test/encryption_test.kdm.xml > build/test/xmlsec1.log 2>&1 < /dev/null");
 158
 159 #ifdef LIBDCP_POSIX
 160         BOOST_CHECK_EQUAL (WEXITSTATUS (r), 0);
 161 #else
 162         BOOST_CHECK_EQUAL (r, 0);
 163 #endif
 164 }