diff options
| author | jhurst <jhurst@cinecert.com> | 2005-12-20 01:55:40 +0000 |
|---|---|---|
| committer | jhurst <> | 2005-12-20 01:55:40 +0000 |
| commit | 8095eaa320551b6795d0368c0ad0c227a3167caa (patch) | |
| tree | e522d5137671fffbc8fcc084831b5d8806ef44f2 /src/asdcp-test.cpp | |
wheee!
Diffstat (limited to 'src/asdcp-test.cpp')
| -rwxr-xr-x | src/asdcp-test.cpp | 1366 |
1 files changed, 1366 insertions, 0 deletions
diff --git a/src/asdcp-test.cpp b/src/asdcp-test.cpp new file mode 100755 index 0000000..c034626 --- /dev/null +++ b/src/asdcp-test.cpp @@ -0,0 +1,1366 @@ +/* +Copyright (c) 2003-2005, John Hurst +All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions +are met: +1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. +2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. +3. The name of the author may not be used to endorse or promote products + derived from this software without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR +IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES +OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. +IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, +INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT +NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF +THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +*/ +/*! \file asdcp-test.cpp + \version $Id$ + \brief AS-DCP file manipulation utility + + This program provides command line access to the major features of the asdcplib + library, and serves as a library unit test which provides the functionality of + the supported use cases. + + For more information about asdcplib, please refer to the header file AS_DCP.h + + WARNING: While the asdcplib library attempts to provide a complete and secure + implementation of the cryptographic features of the AS-DCP file formats, this + unit test program is NOT secure and is therefore NOT SUITABLE FOR USE in a + production environment without some modification. + + In particular, this program uses weak IV generation and externally generated + plaintext keys. These shortcomings exist because cryptographic-quality + random number generation and key management are outside the scope of the + asdcplib library. Developers using asdcplib for commercial implementations + claiming SMPTE conformance are expected to provide proper implementations of + these features. +*/ + +#include <iostream> +#include <assert.h> + +#include <FileIO.h> +#include <PCMParserList.h> +#include <WavFileWriter.h> +#include <hex_utils.h> +#include <AS_DCP_UUID.h> +#include <MXF.h> +#include <Metadata.h> + +using namespace ASDCP; + +const ui32_t FRAME_BUFFER_SIZE = 4*1024*1024; + +//------------------------------------------------------------------------------------------ +// +// command line option parser class + +static const char* PACKAGE = "asdcp-test"; // program name for messages +const ui32_t MAX_IN_FILES = 16; // maximum number of input files handled by + // the command option parser + +// local program identification info written to file headers +class MyInfo : public WriterInfo +{ +public: + MyInfo() + { + static byte_t default_ProductUUID_Data[UUIDlen] = + { 0x7d, 0x83, 0x6e, 0x16, 0x37, 0xc7, 0x4c, 0x22, + 0xb2, 0xe0, 0x46, 0xa7, 0x17, 0xe8, 0x4f, 0x42 }; + + memcpy(ProductUUID, default_ProductUUID_Data, UUIDlen); + CompanyName = "WidgetCo"; + ProductName = "asdcp-test"; + + char s_buf[128]; + sprintf(s_buf, "%lu.%lu.%lu", VERSION_MAJOR, VERSION_APIMINOR, VERSION_IMPMINOR); + ProductVersion = s_buf; + } +} s_MyInfo; + + +// Macros used to test command option data state. + +// True if a major mode has already been selected. +#define TEST_MAJOR_MODE() ( info_flag||create_flag||extract_flag||genkey_flag||genid_flag||gop_start_flag ) + +// Causes the caller to return if a major mode has already been selected, +// otherwise sets the given flag. +#define TEST_SET_MAJOR_MODE(f) if ( TEST_MAJOR_MODE() ) \ + { \ + fputs("Conflicting major mode, choose one of -(gcixG)).\n", stderr); \ + return; \ + } \ + (f) = true; + +// Increment the iterator, test for an additional non-option command line argument. +// Causes the caller to return if there are no remaining arguments or if the next +// argument begins with '-'. +#define TEST_EXTRA_ARG(i,c) if ( ++i >= argc || argv[(i)][0] == '-' ) \ + { \ + fprintf(stderr, "Argument not found for option %c.\n", (c)); \ + return; \ + } +// +void +banner(FILE* stream = stderr) +{ + fprintf(stream, "\n\ +%s (asdcplib %s)\n\n\ +Copyright (c) 2003-2005 John Hurst\n\n\ +asdcplib may be copied only under the terms of the license found at\n\ +the top of every file in the asdcplib distribution kit.\n\n\ +Specify the -h (help) option for further information about %s\n\n", + PACKAGE, ASDCP::Version(), PACKAGE); +} + +// +void +usage(FILE* stream = stderr) +{ + fprintf(stream, "\ +USAGE: %s [-i [-H, -n]|-c <filename> [-p <rate>, -e, -M, -R]|-x <root-name> [-m][-S]|-g|-u|-G|-V|-h]\n\ + [-k <key-string>] [-j <key-id-string>] [-f <start-frame-num>] [-d <duration>]\n\ + [-b <buf-size>] [-W] [-v [-s]] [<filename>, ...]\n\ +\n", PACKAGE); + + fprintf(stream, "\ +Major modes:\n\ + -i - show file info\n\ + -c <filename> - create AS-DCP file from input(s)\n\ + -x <root-name> - extract essence from AS-DCP file to named file(s)\n\ + -g - generate a random 16 byte value to stdout\n\ + -u - generate a random UUID value to stdout\n\ + -G - Perform GOP start lookup test on MPEG file\n\ + -V - show version\n\ + -h - show help\n\ +\n"); + + fprintf(stream, "\ +Security Options:\n\ + -j <key-id-str> - write key ID instead of creating a random value\n\ + -k <key-string> - use key for ciphertext operations\n\ + -e - encrypt MPEG or JP2K headers (default)\n\ + -E - do not encrypt MPEG or JP2K headers\n\ + -M - do not create HMAC values when writing\n\ + -m - verify HMAC values when reading\n\ +\n"); + + fprintf(stream, "\ +Read/Write Options:\n\ + -b <buf-size> - Size (in bytes) of the picture frame buffer, default: 2097152 (2MB)\n\ + -f <frame-num> - starting frame number, default 0\n\ + -d <duration> - number of frames to process, default all\n\ + -p <rate> - fps of picture when wrapping PCM or JP2K:, use one of [23|24|48], 24 is default\n\ + -R - Repeat the first frame over the entire file (picture essence only, requires -c, -d)\n\ + -S - Split Wave essence to stereo WAV files during extract (default = multichannel WAV)\n\ + -W - read input file only, do not write source file\n\ +\n"); + + fprintf(stream, "\ +Info Options:\n\ + -H - show MXF header metadata, used with option -i\n\ + -n - show index, used with option -i\n\ +\n\ +Other Options:\n\ + -s <number> - number of bytes of frame buffer to be dumped as hex to stderr (use with -v)\n\ + -v - verbose, show extra detail during run\n\ +\n\ + NOTES: o There is no option grouping, all options must be distinct arguments.\n\ + o All option arguments must be separated from the option by whitespace.\n\ + o An argument of \"23\" to the -p option will be interpreted as 23000/1001 fps.\n\ +\n"); +} + +// +// +class CommandOptions +{ + CommandOptions(); + +public: + bool error_flag; // true if the given options are in error or not complete + bool info_flag; // true if the file info mode was selected + bool create_flag; // true if the file create mode was selected + bool extract_flag; // true if the file extract mode was selected + bool genkey_flag; // true if we are to generate a new key value + bool genid_flag; // true if we are to generate a new UUID value + bool gop_start_flag; // true if we are to perform a GOP start lookup test + bool key_flag; // true if an encryption key was given + bool key_id_flag; // true if a key ID was given + bool encrypt_header_flag; // true if mpeg headers are to be encrypted + bool write_hmac; // true if HMAC values are to be generated and written + bool read_hmac; // true if HMAC values are to be validated + bool split_wav; // true if PCM is to be extracted to stereo WAV files + bool verbose_flag; // true if the verbose option was selected + ui32_t fb_dump_size; // number of bytes of frame buffer to dump + bool showindex_flag; // true if index is to be displayed + bool showheader_flag; // true if MXF file header is to be displayed + bool no_write_flag; // true if no output files are to be written + bool version_flag; // true if the version display option was selected + bool help_flag; // true if the help display option was selected + ui32_t start_frame; // frame number to begin processing + ui32_t duration; // number of frames to be processed + bool duration_flag; // true if duration argument given + bool do_repeat; // if true and -c -d, repeat first input frame + ui32_t picture_rate; // fps of picture when wrapping PCM + ui32_t fb_size; // size of picture frame buffer + ui32_t file_count; // number of elements in filenames[] + const char* file_root; // filename pre for files written by the extract mode + const char* out_file; // name of mxf file created by create mode + byte_t key_value[KeyLen]; // value of given encryption key (when key_flag is true) + byte_t key_id_value[KeyIDlen];// value of given key ID (when key_id_flag is true) + const char* filenames[MAX_IN_FILES]; // list of filenames to be processed + + // + Rational PictureRate() + { + if ( picture_rate == 23 ) return EditRate_23_98; + if ( picture_rate == 48 ) return EditRate_48; + return EditRate_24; + } + + // + const char* szPictureRate() + { + if ( picture_rate == 23 ) return "23.976"; + if ( picture_rate == 48 ) return "48"; + return "24"; + } + + // + CommandOptions(int argc, const char** argv) : + error_flag(true), info_flag(false), create_flag(false), + extract_flag(false), genkey_flag(false), genid_flag(false), gop_start_flag(false), + key_flag(false), encrypt_header_flag(true), write_hmac(true), read_hmac(false), split_wav(false), + verbose_flag(false), fb_dump_size(0), showindex_flag(false), showheader_flag(false), + no_write_flag(false), version_flag(false), help_flag(false), start_frame(0), + duration(0xffffffff), duration_flag(false), do_repeat(false), picture_rate(24), + fb_size(FRAME_BUFFER_SIZE), file_count(0), file_root(0), out_file(0) + { + memset(key_value, 0, KeyLen); + memset(key_id_value, 0, KeyIDlen); + + for ( int i = 1; i < argc; i++ ) + { + if ( argv[i][0] == '-' && isalpha(argv[i][1]) && argv[i][2] == 0 ) + { + switch ( argv[i][1] ) + { + case 'i': TEST_SET_MAJOR_MODE(info_flag); break; + case 'G': TEST_SET_MAJOR_MODE(gop_start_flag); break; + case 'W': no_write_flag = true; break; + case 'n': showindex_flag = true; break; + case 'H': showheader_flag = true; break; + case 'R': do_repeat = true; break; + case 'S': split_wav = true; break; + case 'V': version_flag = true; break; + case 'h': help_flag = true; break; + case 'v': verbose_flag = true; break; + case 'g': +#ifdef ASDCP_WITHOUT_OPENSSL + fputs("Program compiled without encryption support.\n", stderr); + return; +#else + genkey_flag = true; +#endif + break; + + case 'u': +#ifdef ASDCP_WITHOUT_OPENSSL + fputs("Program compiled without encryption support.\n", stderr); + return; +#else + genid_flag = true; +#endif + break; + + case 'e': encrypt_header_flag = true; break; + case 'E': encrypt_header_flag = false; break; + case 'M': write_hmac = false; break; + case 'm': read_hmac = true; break; + + case 'c': + TEST_SET_MAJOR_MODE(create_flag); + TEST_EXTRA_ARG(i, 'c'); + out_file = argv[i]; + break; + + case 'x': + TEST_SET_MAJOR_MODE(extract_flag); + TEST_EXTRA_ARG(i, 'x'); + file_root = argv[i]; + break; + + case 'k': key_flag = true; +#ifdef ASDCP_WITHOUT_OPENSSL + fputs("Program compiled without encryption support.\n", stderr); + return; +#else + TEST_EXTRA_ARG(i, 'k'); + { + ui32_t length; + hex2bin(argv[i], key_value, KeyLen, &length); + + if ( length != KeyLen ) + { + fprintf(stderr, "Unexpected key length: %lu, expecting %lu characters.\n", KeyLen, length); + return; + } + } +#endif + break; + + case 'j': key_id_flag = true; +#ifdef ASDCP_WITHOUT_OPENSSL + fputs("Program compiled without encryption support.\n", stderr); + return; +#else + TEST_EXTRA_ARG(i, 'j'); + { + ui32_t length; + hex2bin(argv[i], key_id_value, KeyIDlen, &length); + + if ( length != KeyIDlen ) + { + fprintf(stderr, "Unexpected key ID length: %lu, expecting %lu characters.\n", KeyIDlen, length); + return; + } + } +#endif + break; + + case 'f': + TEST_EXTRA_ARG(i, 'f'); + start_frame = atoi(argv[i]); // TODO: test for negative value, should use strtol() + break; + + case 'd': + TEST_EXTRA_ARG(i, 'd'); + duration_flag = true; + duration = atoi(argv[i]); // TODO: test for negative value, should use strtol() + break; + + case 'p': + TEST_EXTRA_ARG(i, 'p'); + picture_rate = atoi(argv[i]); + break; + + case 's': + TEST_EXTRA_ARG(i, 's'); + fb_dump_size = atoi(argv[i]); + break; + + case 'b': + TEST_EXTRA_ARG(i, 'b'); + fb_size = atoi(argv[i]); + + if ( verbose_flag ) + fprintf(stderr, "Frame Buffer size: %lu bytes.\n", fb_size); + + break; + + default: + fprintf(stderr, "Unrecognized option: %c\n", argv[i][1]); + return; + } + } + else + { + filenames[file_count++] = argv[i]; + + if ( file_count >= MAX_IN_FILES ) + { + fprintf(stderr, "Filename lists exceeds maximum list size: %lu\n", MAX_IN_FILES); + return; + } + } + } + + if ( TEST_MAJOR_MODE() ) + { + if ( ! genkey_flag && ! genid_flag && file_count == 0 ) + { + fputs("Option requires at least one filename argument.\n", stderr); + return; + } + } + + if ( ! TEST_MAJOR_MODE() && ! help_flag && ! version_flag ) + { + fputs("No operation selected (use one of -(gcixG) or -h for help).\n", stderr); + return; + } + + error_flag = false; + } +}; + +//------------------------------------------------------------------------------------------ +// MPEG2 essence +#if 0 +// Write a plaintext MPEG2 Video Elementary Stream to a plaintext ASDCP file +// Write a plaintext MPEG2 Video Elementary Stream to a ciphertext ASDCP file +// +Result_t +write_MPEG2_file(CommandOptions& Options) +{ + AESEncContext* Context = 0; + HMACContext* HMAC = 0; + MPEG2::FrameBuffer FrameBuffer(Options.fb_size); + MPEG2::Parser Parser; + MPEG2::MXFWriter Writer; + MPEG2::VideoDescriptor VDesc; + +#ifndef ASDCP_WITHOUT_OPENSSL + byte_t IV_buf[CBC_BLOCK_SIZE]; + FortunaRNG RNG; +#endif + + // set up essence parser + Result_t result = Parser.OpenRead(Options.filenames[0]); + + // set up MXF writer + if ( ASDCP_SUCCESS(result) ) + { + Parser.FillVideoDescriptor(VDesc); + + if ( Options.verbose_flag ) + { + fputs("MPEG-2 Pictures\n", stderr); + fputs("VideoDescriptor:\n", stderr); + fprintf(stderr, "Frame Buffer size: %lu\n", Options.fb_size); + MPEG2::VideoDescriptorDump(VDesc); + } + } + + if ( ASDCP_SUCCESS(result) && ! Options.no_write_flag ) + { + WriterInfo Info = s_MyInfo; // fill in your favorite identifiers here +#ifndef ASDCP_WITHOUT_OPENSSL + GenRandomUUID(RNG, Info.AssetUUID); + + // configure encryption + if( Options.key_flag ) + { + GenRandomUUID(RNG, Info.ContextID); + Info.EncryptedEssence = true; + + if ( Options.key_id_flag ) + memcpy(Info.CryptographicKeyID, Options.key_id_value, KeyIDlen); + else + RNG.FillRandom(Info.CryptographicKeyID, KeyIDlen); + + Context = new AESEncContext; + result = Context->InitKey(Options.key_value); + + if ( ASDCP_SUCCESS(result) ) + result = Context->SetIVec(RNG.FillRandom(IV_buf, CBC_BLOCK_SIZE)); + + if ( ASDCP_SUCCESS(result) && Options.write_hmac ) + { + Info.UsesHMAC = true; + HMAC = new HMACContext; + result = HMAC->InitKey(Options.key_value); + } + } +#endif // ASDCP_WITHOUT_OPENSSL + + if ( ASDCP_SUCCESS(result) ) + result = Writer.OpenWrite(Options.out_file, Info, VDesc); + } + + if ( ASDCP_SUCCESS(result) ) + // loop through the frames + { + result = Parser.Reset(); + ui32_t duration = 0; + + while ( ASDCP_SUCCESS(result) && duration++ < Options.duration ) + { + if ( ! Options.do_repeat || duration == 1 ) + { + result = Parser.ReadFrame(FrameBuffer); + + if ( ASDCP_SUCCESS(result) ) + { + if ( Options.verbose_flag ) + FrameBuffer.Dump(stderr, Options.fb_dump_size); + + if ( Options.encrypt_header_flag ) + FrameBuffer.PlaintextOffset(0); + } + } + + if ( ASDCP_SUCCESS(result) && ! Options.no_write_flag ) + { + result = Writer.WriteFrame(FrameBuffer, Context, HMAC); + +#ifndef ASDCP_WITHOUT_OPENSSL + // The Writer class will forward the last block of ciphertext + // to the encryption context for use as the IV for the next + // frame. If you want to use non-sequitur IV values, un-comment + // the following line of code. + // if ( ASDCP_SUCCESS(result) && Options.key_flag ) + // Context->SetIVec(RNG.FillRandom(IV_buf, CBC_BLOCK_SIZE)); +#endif + } + } + + if ( result == RESULT_ENDOFFILE ) + result = RESULT_OK; + } + + if ( ASDCP_SUCCESS(result) && ! Options.no_write_flag ) + result = Writer.Finalize(); + + return result; +} +#endif + +// Read a plaintext MPEG2 Video Elementary Stream from a plaintext ASDCP file +// Read a plaintext MPEG2 Video Elementary Stream from a ciphertext ASDCP file +// Read a ciphertext MPEG2 Video Elementary Stream from a ciphertext ASDCP file +// +Result_t +read_MPEG2_file(CommandOptions& Options) +{ + AESDecContext* Context = 0; + HMACContext* HMAC = 0; + MPEG2::MXFReader Reader; + MPEG2::FrameBuffer FrameBuffer(Options.fb_size); + FileWriter OutFile; + ui32_t frame_count = 0; + + Result_t result = Reader.OpenRead(Options.filenames[0]); + + if ( ASDCP_SUCCESS(result) ) + { + MPEG2::VideoDescriptor VDesc; + Reader.FillVideoDescriptor(VDesc); + frame_count = VDesc.ContainerDuration; + + if ( Options.verbose_flag ) + { + fprintf(stderr, "Frame Buffer size: %lu\n", Options.fb_size); + MPEG2::VideoDescriptorDump(VDesc); + } + } + + if ( ASDCP_SUCCESS(result) ) + { + char filename[256]; + sprintf(filename, "%s.ves", Options.file_root); + result = OutFile.OpenWrite(filename); + } + + if ( ASDCP_SUCCESS(result) && Options.key_flag ) + { + Context = new AESDecContext; + result = Context->InitKey(Options.key_value); + + if ( ASDCP_SUCCESS(result) && Options.read_hmac ) + { + WriterInfo Info; + Reader.FillWriterInfo(Info); + + if ( Info.UsesHMAC ) + { + HMAC = new HMACContext; + result = HMAC->InitKey(Options.key_value); + } + else + { + fputs("File does not contain HMAC values, ignoring -m option.\n", stderr); + } + } + } + + ui32_t last_frame = Options.start_frame + ( Options.duration ? Options.duration : frame_count); + if ( last_frame > frame_count ) + last_frame = frame_count; + + for ( ui32_t i = Options.start_frame; ASDCP_SUCCESS(result) && i < last_frame; i++ ) + { + result = Reader.ReadFrame(i, FrameBuffer, Context, HMAC); + + if ( ASDCP_SUCCESS(result) ) + { + if ( Options.verbose_flag ) + FrameBuffer.Dump(stderr, Options.fb_dump_size); + + ui32_t write_count = 0; + result = OutFile.Write(FrameBuffer.Data(), FrameBuffer.Size(), &write_count); + } + } + + return result; +} + + +// +Result_t +gop_start_test(CommandOptions& Options) +{ + using namespace ASDCP::MPEG2; + + MXFReader Reader; + MPEG2::FrameBuffer FrameBuffer(Options.fb_size); + ui32_t frame_count = 0; + + Result_t result = Reader.OpenRead(Options.filenames[0]); + + if ( ASDCP_SUCCESS(result) ) + { + MPEG2::VideoDescriptor VDesc; + Reader.FillVideoDescriptor(VDesc); + frame_count = VDesc.ContainerDuration; + + if ( Options.verbose_flag ) + { + fprintf(stderr, "Frame Buffer size: %lu\n", Options.fb_size); + MPEG2::VideoDescriptorDump(VDesc); + } + } + + ui32_t last_frame = Options.start_frame + ( Options.duration ? Options.duration : frame_count); + if ( last_frame > frame_count ) + last_frame = frame_count; + + for ( ui32_t i = Options.start_frame; ASDCP_SUCCESS(result) && i < last_frame; i++ ) + { + result = Reader.ReadFrameGOPStart(i, FrameBuffer); + + if ( ASDCP_SUCCESS(result) ) + { + if ( Options.verbose_flag ) + FrameBuffer.Dump(stderr, Options.fb_dump_size); + + if ( FrameBuffer.FrameType() != FRAME_I ) + fprintf(stderr, "Expecting an I frame, got %c\n", FrameTypeChar(FrameBuffer.FrameType())); + + fprintf(stderr, "Requested frame %lu, got %lu\n", i, FrameBuffer.FrameNumber()); + } + } + + return result; +} + +#if 0 +//------------------------------------------------------------------------------------------ +// JPEG 2000 essence + +// Write one or more plaintext JPEG 2000 codestreams to a plaintext ASDCP file +// Write one or more plaintext JPEG 2000 codestreams to a ciphertext ASDCP file +// +Result_t +write_JP2K_file(CommandOptions& Options) +{ + AESEncContext* Context = 0; + HMACContext* HMAC = 0; + JP2K::MXFWriter Writer; + JP2K::FrameBuffer FrameBuffer(Options.fb_size); + JP2K::PictureDescriptor PDesc; + JP2K::SequenceParser Parser; + +#ifndef ASDCP_WITHOUT_OPENSSL + byte_t IV_buf[CBC_BLOCK_SIZE]; + FortunaRNG RNG; +#endif + + // set up essence parser + Result_t result = Parser.OpenRead(Options.filenames[0]); + + // set up MXF writer + if ( ASDCP_SUCCESS(result) ) + { + Parser.FillPictureDescriptor(PDesc); + PDesc.EditRate = Options.PictureRate(); + + if ( Options.verbose_flag ) + { + fprintf(stderr, "JPEG 2000 pictures\n"); + fputs("PictureDescriptor:\n", stderr); + fprintf(stderr, "Frame Buffer size: %lu\n", Options.fb_size); + JP2K::PictureDescriptorDump(PDesc); + } + } + + if ( ASDCP_SUCCESS(result) && ! Options.no_write_flag ) + { + WriterInfo Info = s_MyInfo; // fill in your favorite identifiers here +#ifndef ASDCP_WITHOUT_OPENSSL + GenRandomUUID(RNG, Info.AssetUUID); + + // configure encryption + if( Options.key_flag ) + { + GenRandomUUID(RNG, Info.ContextID); + Info.EncryptedEssence = true; + + if ( Options.key_id_flag ) + memcpy(Info.CryptographicKeyID, Options.key_id_value, KeyIDlen); + else + RNG.FillRandom(Info.CryptographicKeyID, KeyIDlen); + + Context = new AESEncContext; + result = Context->InitKey(Options.key_value); + + if ( ASDCP_SUCCESS(result) ) + result = Context->SetIVec(RNG.FillRandom(IV_buf, CBC_BLOCK_SIZE)); + + if ( ASDCP_SUCCESS(result) && Options.write_hmac ) + { + Info.UsesHMAC = true; + HMAC = new HMACContext; + result = HMAC->InitKey(Options.key_value); + } + } +#endif // ASDCP_WITHOUT_OPENSSL + + if ( ASDCP_SUCCESS(result) ) + result = Writer.OpenWrite(Options.out_file, Info, PDesc); + } + + if ( ASDCP_SUCCESS(result) ) + { + ui32_t duration = 0; + result = Parser.Reset(); + + while ( ASDCP_SUCCESS(result) && duration++ < Options.duration ) + { + if ( ! Options.do_repeat || duration == 1 ) + { + result = Parser.ReadFrame(FrameBuffer); + + if ( ASDCP_SUCCESS(result) ) + { + if ( Options.verbose_flag ) + FrameBuffer.Dump(stderr, Options.fb_dump_size); + + if ( Options.encrypt_header_flag ) + FrameBuffer.PlaintextOffset(0); + } + } + + if ( ASDCP_SUCCESS(result) && ! Options.no_write_flag ) + { + result = Writer.WriteFrame(FrameBuffer, Context, HMAC); +#ifndef ASDCP_WITHOUT_OPENSSL + // The Writer class will forward the last block of ciphertext + // to the encryption context for use as the IV for the next + // frame. If you want to use non-sequitur IV values, un-comment + // the following line of code. + // if ( ASDCP_SUCCESS(result) && Options.key_flag ) + // Context->SetIVec(RNG.FillRandom(IV_buf, CBC_BLOCK_SIZE)); +#endif + } + } + + if ( result == RESULT_ENDOFFILE ) + result = RESULT_OK; + } + + if ( ASDCP_SUCCESS(result) && ! Options.no_write_flag ) + result = Writer.Finalize(); + + return result; +} +#endif + +// Read one or more plaintext JPEG 2000 codestreams from a plaintext ASDCP file +// Read one or more plaintext JPEG 2000 codestreams from a ciphertext ASDCP file +// Read one or more ciphertext JPEG 2000 codestreams from a ciphertext ASDCP file +// +Result_t +read_JP2K_file(CommandOptions& Options) +{ + AESDecContext* Context = 0; + HMACContext* HMAC = 0; + JP2K::MXFReader Reader; + JP2K::FrameBuffer FrameBuffer(Options.fb_size); + ui32_t frame_count = 0; + + Result_t result = Reader.OpenRead(Options.filenames[0]); + + if ( ASDCP_SUCCESS(result) ) + { + JP2K::PictureDescriptor PDesc; + Reader.FillPictureDescriptor(PDesc); + + frame_count = PDesc.ContainerDuration; + + if ( Options.verbose_flag ) + { + fprintf(stderr, "Frame Buffer size: %lu\n", Options.fb_size); + JP2K::PictureDescriptorDump(PDesc); + } + } + + if ( ASDCP_SUCCESS(result) && Options.key_flag ) + { + Context = new AESDecContext; + result = Context->InitKey(Options.key_value); + + if ( ASDCP_SUCCESS(result) && Options.read_hmac ) + { + WriterInfo Info; + Reader.FillWriterInfo(Info); + + if ( Info.UsesHMAC ) + { + HMAC = new HMACContext; + result = HMAC->InitKey(Options.key_value); + } + else + { + fputs("File does not contain HMAC values, ignoring -m option.\n", stderr); + } + } + } + + ui32_t last_frame = Options.start_frame + ( Options.duration ? Options.duration : frame_count); + if ( last_frame > frame_count ) + last_frame = frame_count; + + for ( ui32_t i = Options.start_frame; ASDCP_SUCCESS(result) && i < last_frame; i++ ) + { + result = Reader.ReadFrame(i, FrameBuffer, Context, HMAC); + + if ( ASDCP_SUCCESS(result) ) + { + FileWriter OutFile; + char filename[256]; + ui32_t write_count; + sprintf(filename, "%s%06lu.j2c", Options.file_root, i); + result = OutFile.OpenWrite(filename); + + if ( ASDCP_SUCCESS(result) ) + result = OutFile.Write(FrameBuffer.Data(), FrameBuffer.Size(), &write_count); + + if ( Options.verbose_flag ) + FrameBuffer.Dump(stderr, Options.fb_dump_size); + } + } + + return result; +} + +#if 0 +//------------------------------------------------------------------------------------------ +// PCM essence + + +// Write one or more plaintext PCM audio streams to a plaintext ASDCP file +// Write one or more plaintext PCM audio streams to a ciphertext ASDCP file +// +Result_t +write_PCM_file(CommandOptions& Options) +{ + AESEncContext* Context = 0; + HMACContext* HMAC = 0; + PCMParserList Parser; + PCM::MXFWriter Writer; + PCM::FrameBuffer FrameBuffer; + PCM::AudioDescriptor ADesc; + Rational PictureRate = Options.PictureRate(); + +#ifndef ASDCP_WITHOUT_OPENSSL + byte_t IV_buf[CBC_BLOCK_SIZE]; + FortunaRNG RNG; +#endif + + // set up essence parser + Result_t result = Parser.OpenRead(Options.file_count, Options.filenames, PictureRate); + + // set up MXF writer + if ( ASDCP_SUCCESS(result) ) + { + Parser.FillAudioDescriptor(ADesc); + + ADesc.SampleRate = PictureRate; + FrameBuffer.Capacity(PCM::CalcFrameBufferSize(ADesc)); + + if ( Options.verbose_flag ) + { + fprintf(stderr, "48Khz PCM Audio, %s fps (%lu spf)\n", + Options.szPictureRate(), + PCM::CalcSamplesPerFrame(ADesc)); + fputs("AudioDescriptor:\n", stderr); + PCM::AudioDescriptorDump(ADesc); + } + } + + if ( ASDCP_SUCCESS(result) && ! Options.no_write_flag ) + { + WriterInfo Info = s_MyInfo; // fill in your favorite identifiers here +#ifndef ASDCP_WITHOUT_OPENSSL + GenRandomUUID(RNG, Info.AssetUUID); + + // configure encryption + if( Options.key_flag ) + { + GenRandomUUID(RNG, Info.ContextID); + Info.EncryptedEssence = true; + + if ( Options.key_id_flag ) + memcpy(Info.CryptographicKeyID, Options.key_id_value, KeyIDlen); + else + RNG.FillRandom(Info.CryptographicKeyID, KeyIDlen); + + Context = new AESEncContext; + result = Context->InitKey(Options.key_value); + + if ( ASDCP_SUCCESS(result) ) + result = Context->SetIVec(RNG.FillRandom(IV_buf, CBC_BLOCK_SIZE)); + + if ( ASDCP_SUCCESS(result) && Options.write_hmac ) + { + Info.UsesHMAC = true; + HMAC = new HMACContext; + result = HMAC->InitKey(Options.key_value); + } + } +#endif // ASDCP_WITHOUT_OPENSSL + + if ( ASDCP_SUCCESS(result) ) + result = Writer.OpenWrite(Options.out_file, Info, ADesc); + } + + if ( ASDCP_SUCCESS(result) ) + { + result = Parser.Reset(); + ui32_t duration = 0; + + while ( ASDCP_SUCCESS(result) && duration++ < Options.duration ) + { + result = Parser.ReadFrame(FrameBuffer); + + if ( ASDCP_SUCCESS(result) ) + { + if ( FrameBuffer.Size() != FrameBuffer.Capacity() ) + { + fprintf(stderr, "WARNING: Last frame read was short, PCM input is possibly not frame aligned.\n"); + fprintf(stderr, "Expecting %lu bytes, got %lu.\n", FrameBuffer.Capacity(), FrameBuffer.Size()); + result = RESULT_ENDOFFILE; + continue; + } + + if ( Options.verbose_flag ) + FrameBuffer.Dump(stderr, Options.fb_dump_size); + + if ( ! Options.no_write_flag ) + { + result = Writer.WriteFrame(FrameBuffer, Context, HMAC); + +#ifndef ASDCP_WITHOUT_OPENSSL + // The Writer class will forward the last block of ciphertext + // to the encryption context for use as the IV for the next + // frame. If you want to use non-sequitur IV values, un-comment + // the following line of code. + // if ( ASDCP_SUCCESS(result) && Options.key_flag ) + // Context->SetIVec(RNG.FillRandom(IV_buf, CBC_BLOCK_SIZE)); +#endif + } + } + } + + if ( result == RESULT_ENDOFFILE ) + result = RESULT_OK; + } + + if ( ASDCP_SUCCESS(result) && ! Options.no_write_flag ) + result = Writer.Finalize(); + + return result; +} +#endif + +// Read one or more plaintext PCM audio streams from a plaintext ASDCP file +// Read one or more plaintext PCM audio streams from a ciphertext ASDCP file +// Read one or more ciphertext PCM audio streams from a ciphertext ASDCP file +// +Result_t +read_PCM_file(CommandOptions& Options) +{ + AESDecContext* Context = 0; + HMACContext* HMAC = 0; + PCM::MXFReader Reader; + PCM::FrameBuffer FrameBuffer; + WavFileWriter OutWave; + PCM::AudioDescriptor ADesc; + ui32_t last_frame = 0; + + Result_t result = Reader.OpenRead(Options.filenames[0]); + + if ( ASDCP_SUCCESS(result) ) + { + Reader.FillAudioDescriptor(ADesc); + + if ( ADesc.SampleRate != EditRate_23_98 + && ADesc.SampleRate != EditRate_24 + && ADesc.SampleRate != EditRate_48 ) + ADesc.SampleRate = Options.PictureRate(); + + FrameBuffer.Capacity(PCM::CalcFrameBufferSize(ADesc)); + + if ( Options.verbose_flag ) + PCM::AudioDescriptorDump(ADesc); + } + + if ( ASDCP_SUCCESS(result) ) + { + last_frame = ADesc.ContainerDuration; + + if ( Options.duration > 0 && Options.duration < last_frame ) + last_frame = Options.duration; + + if ( Options.start_frame > 0 ) + { + if ( Options.start_frame > ADesc.ContainerDuration ) + { + fprintf(stderr, "Start value greater than file duration.\n"); + return RESULT_FAIL; + } + + last_frame = xmin(Options.start_frame + last_frame, ADesc.ContainerDuration); + } + + ADesc.ContainerDuration = last_frame - Options.start_frame; + OutWave.OpenWrite(ADesc, Options.file_root, Options.split_wav); + } + + if ( ASDCP_SUCCESS(result) && Options.key_flag ) + { + Context = new AESDecContext; + result = Context->InitKey(Options.key_value); + + if ( ASDCP_SUCCESS(result) && Options.read_hmac ) + { + WriterInfo Info; + Reader.FillWriterInfo(Info); + + if ( Info.UsesHMAC ) + { + HMAC = new HMACContext; + result = HMAC->InitKey(Options.key_value); + } + else + { + fputs("File does not contain HMAC values, ignoring -m option.\n", stderr); + } + } + } + + for ( ui32_t i = Options.start_frame; ASDCP_SUCCESS(result) && i < last_frame; i++ ) + { + result = Reader.ReadFrame(i, FrameBuffer, Context, HMAC); + + if ( ASDCP_SUCCESS(result) ) + { + if ( Options.verbose_flag ) + FrameBuffer.Dump(stderr, Options.fb_dump_size); + + result = OutWave.WriteFrame(FrameBuffer); + } + } + + return result; +} + + +//------------------------------------------------------------------------------------------ +// + +// +// These classes wrap the irregular names in the asdcplib API +// so that I can use a template to simplify the implementation +// of show_file_info() + +class MyVideoDescriptor : public MPEG2::VideoDescriptor +{ + public: + void FillDescriptor(MPEG2::MXFReader& Reader) { + Reader.FillVideoDescriptor(*this); + } + + void Dump(FILE* stream) { + MPEG2::VideoDescriptorDump(*this, stream); + } +}; +#if 0 +class MyPictureDescriptor : public JP2K::PictureDescriptor +{ + public: + void FillDescriptor(JP2K::MXFReader& Reader) { + Reader.FillPictureDescriptor(*this); + } + + void Dump(FILE* stream) { + JP2K::PictureDescriptorDump(*this, stream); + } +}; + +class MyAudioDescriptor : public PCM::AudioDescriptor +{ + public: + void FillDescriptor(PCM::MXFReader& Reader) { + Reader.FillAudioDescriptor(*this); + } + + void Dump(FILE* stream) { + PCM::AudioDescriptorDump(*this, stream); + } +}; +#endif + +// MSVC didn't like the function template, so now it's a static class method +template<class ReaderT, class DescriptorT> +class FileInfoWrapper +{ +public: + static void file_info(CommandOptions& Options, FILE* stream = 0) + { + if ( stream == 0 ) + stream = stdout; + + if ( Options.verbose_flag || Options.showheader_flag ) + { + ReaderT Reader; + Result_t result = Reader.OpenRead(Options.filenames[0]); + + if ( ASDCP_SUCCESS(result) ) + { + if ( Options.showheader_flag ) + Reader.DumpHeaderMetadata(stream); + + WriterInfo WI; + Reader.FillWriterInfo(WI); + WriterInfoDump(WI, stream); + + DescriptorT Desc; + Desc.FillDescriptor(Reader); + Desc.Dump(stream); + + if ( Options.showindex_flag ) + Reader.DumpIndex(stream); + } + else if ( result == RESULT_FORMAT && Options.showheader_flag ) + { + Reader.DumpHeaderMetadata(stream); + } + } + } +}; + +// Read header metadata from an ASDCP file +// +Result_t +show_file_info(CommandOptions& Options) +{ + EssenceType_t EssenceType; + Result_t result = ASDCP::EssenceType(Options.filenames[0], EssenceType); + + if ( ASDCP_FAILURE(result) ) + return result; + + if ( EssenceType == ESS_MPEG2_VES ) + { + fputs("File essence type is MPEG2 video.\n", stdout); + FileInfoWrapper<ASDCP::MPEG2::MXFReader, MyVideoDescriptor>::file_info(Options); + } +#if 0 + else if ( EssenceType == ESS_PCM_24b_48k ) + { + fputs("File essence type is PCM audio.\n", stdout); + FileInfoWrapper<ASDCP::PCM::MXFReader, MyAudioDescriptor>::file_info(Options); + } + else if ( EssenceType == ESS_JPEG_2000 ) + { + fputs("File essence type is JPEG 2000 pictures.\n", stdout); + FileInfoWrapper<ASDCP::JP2K::MXFReader, MyPictureDescriptor>::file_info(Options); + } +#endif + else + { + fprintf(stderr, "File is not AS-DCP: %s\n", Options.filenames[0]); + FileReader Reader; + MXF::OPAtomHeader TestHeader; + + result = Reader.OpenRead(Options.filenames[0]); + + if ( ASDCP_SUCCESS(result) ) + result = TestHeader.InitFromFile(Reader); // test UL and OP + + if ( ASDCP_SUCCESS(result) ) + { + if ( MXF::Identification* ID = TestHeader.GetIdentification() ) + { + TestHeader.Dump(); + ID->Dump(); + // show OP + // show + } + else + { + fputs("File contains no Identification object.\n", stdout); + } + } + else + { + fputs("File is not MXF.\n", stdout); + } + } + + return result; +} + + +// +int +main(int argc, const char** argv) +{ + Result_t result = RESULT_OK; + CommandOptions Options(argc, argv); + + if ( Options.help_flag ) + { + usage(); + return 0; + } + + if ( Options.error_flag ) + return 3; + + if ( Options.version_flag ) + banner(); + + if ( Options.info_flag ) + { + result = show_file_info(Options); + } + else if ( Options.gop_start_flag ) + { + result = gop_start_test(Options); + } +#ifndef ASDCP_WITHOUT_OPENSSL + else if ( Options.genkey_flag ) + { + FortunaRNG RNG; + byte_t bin_buf[KeyLen]; + char str_buf[40]; + + RNG.FillRandom(bin_buf, KeyLen); + printf("%s\n", bin2hex(bin_buf, KeyLen, str_buf, 40)); + } + else if ( Options.genid_flag ) + { + FortunaRNG RNG; + byte_t bin_buf[KeyLen]; + char str_buf[40]; + + GenRandomUUID(RNG, bin_buf); + bin2hex(bin_buf, KeyLen, str_buf, 40); + printf("%s\n", hyphenate_UUID(str_buf, 40)); + } +#endif // ASDCP_WITHOUT_OPENSSL + else if ( Options.extract_flag ) + { + EssenceType_t EssenceType; + result = ASDCP::EssenceType(Options.filenames[0], EssenceType); +#ifdef SMPTE_LABELS + fprintf(stderr, "ATTENTION! Expecting SMPTE Universal Labels\n"); +#endif + if ( ASDCP_SUCCESS(result) ) + { + switch ( EssenceType ) + { + case ESS_MPEG2_VES: + result = read_MPEG2_file(Options); + break; + + case ESS_JPEG_2000: + result = read_JP2K_file(Options); + break; + + case ESS_PCM_24b_48k: + result = read_PCM_file(Options); + break; + + default: + fprintf(stderr, "%s: Unknown file type, not ASDCP essence.\n", Options.filenames[0]); + return 5; + } + } + } + else if ( Options.create_flag ) + { + if ( Options.do_repeat && ! Options.duration_flag ) + { + fputs("Option -R requires -d <duration>\n", stderr); + return RESULT_FAIL; + } + + EssenceType_t EssenceType; + result = ASDCP::RawEssenceType(Options.filenames[0], EssenceType); +#ifdef SMPTE_LABELS + fprintf(stderr, "ATTENTION! Writing SMPTE Universal Labels\n"); +#endif +#if 0 + if ( ASDCP_SUCCESS(result) ) + { + switch ( EssenceType ) + { + case ESS_MPEG2_VES: + result = write_MPEG2_file(Options); + break; + + case ESS_JPEG_2000: + result = write_JP2K_file(Options); + break; + + case ESS_PCM_24b_48k: + result = write_PCM_file(Options); + break; + + default: + fprintf(stderr, "%s: Unknown file type, not ASDCP-compatible essence.\n", + Options.filenames[0]); + return 5; + } + } +#endif + } + + if ( result != RESULT_OK ) + { + fputs("Program stopped on error.\n", stderr); + + if ( result != RESULT_FAIL ) + { + fputs(GetResultString(result), stderr); + fputc('\n', stderr); + } + + return 1; + } + + return 0; +} + + +// +// end asdcp-test.cpp +// |
