Mp4Encrypt.cpp
上传用户:xjjlds
上传日期:2015-12-05
资源大小:22823k
文件大小:5k
源码类别:

多媒体编程

开发平台:

Visual C++

  1. /*****************************************************************
  2. |
  3. |    AP4 - MP4 Encrypter
  4. |
  5. |    Copyright 2005 Gilles Boccon-Gibod
  6. |
  7. |
  8. |    This file is part of Bento4/AP4 (MP4 Atom Processing Library).
  9. |
  10. |    Unless you have obtained Bento4 under a difference license,
  11. |    this version of Bento4 is Bento4|GPL.
  12. |    Bento4|GPL is free software; you can redistribute it and/or modify
  13. |    it under the terms of the GNU General Public License as published by
  14. |    the Free Software Foundation; either version 2, or (at your option)
  15. |    any later version.
  16. |
  17. |    Bento4|GPL is distributed in the hope that it will be useful,
  18. |    but WITHOUT ANY WARRANTY; without even the implied warranty of
  19. |    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  20. |    GNU General Public License for more details.
  21. |
  22. |    You should have received a copy of the GNU General Public License
  23. |    along with Bento4|GPL; see the file COPYING.  If not, write to the
  24. |    Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  25. |    02111-1307, USA.
  26. |
  27. ****************************************************************/
  28. /*----------------------------------------------------------------------
  29. |       includes
  30. +---------------------------------------------------------------------*/
  31. #include <stdio.h>
  32. #include <stdlib.h>
  33. #include "Ap4.h"
  34. #include "Ap4FileByteStream.h"
  35. #include "Ap4Atom.h"
  36. #include "Ap4File.h"
  37. #include "Ap4Sample.h"
  38. #include "Ap4SampleDescription.h"
  39. #include "Ap4IsmaCryp.h"
  40. #include "Ap4Utils.h"
  41. /*----------------------------------------------------------------------
  42. |       constants
  43. +---------------------------------------------------------------------*/
  44. #define BANNER "MP4 Encrypter - Version 0.1an"
  45.                "(c) 2002-2005 Gilles Boccon-Gibod & Julien Boeuf"
  46. /*----------------------------------------------------------------------
  47. |       PrintUsageAndExit
  48. +---------------------------------------------------------------------*/
  49. static void
  50. PrintUsageAndExit()
  51. {
  52.     fprintf(stderr, 
  53.         BANNER 
  54.         "nn"
  55.         "usage: mp4encrypt --kms-uri <uri> [--key <n>:<k>:<salt>] <input> <output>n"
  56.         "    where <n> is a track index, <k> a 128-bit key in hexn"
  57.         "    and <salt> a 128-bit salting keyn"
  58.         "    (several --key options can be used, one for each track)n");
  59.     exit(1);
  60. }
  61. /*----------------------------------------------------------------------
  62. |       main
  63. +---------------------------------------------------------------------*/
  64. int
  65. main(int argc, char** argv)
  66. {
  67.     if (argc < 5) {
  68.         PrintUsageAndExit();
  69.     }
  70.     // parse options
  71.     if (strcmp(*++argv, "--kms-uri")) {
  72.         fprintf(stderr, "ERROR: the first option must be --kms-urin");
  73.         return 1;
  74.     }
  75.     const char* kms_uri = *++argv;
  76.     // create an encrypting processor
  77.     AP4_IsmaEncryptingProcessor processor(kms_uri);
  78.     // setup default values
  79.     const char* input_filename = NULL;
  80.     const char* output_filename = NULL;
  81.     char* arg;
  82.     while ((arg = *++argv)) {
  83.         if (!strcmp(arg, "--key")) {
  84.             arg = *++argv;
  85.             if (arg == NULL) {
  86.                 fprintf(stderr, "ERROR: missing argument for --key optionn");
  87.                 return 1;
  88.             }
  89.             char* track_ascii = NULL;
  90.             char* key_ascii = NULL;
  91.             char* salt_ascii = NULL;
  92.             if (AP4_FAILED(AP4_SplitArgs(arg, track_ascii, key_ascii, salt_ascii))) {
  93.                 fprintf(stderr, "ERROR: invalid argument for --key optionn");
  94.                 return 1;
  95.             }
  96.             unsigned char key[16];
  97.             unsigned char salt[8];
  98.             unsigned int track = strtoul(track_ascii, NULL, 10);
  99.             if (AP4_ParseHex(key_ascii, key, 16)) {
  100.                 fprintf(stderr, "ERROR: invalid hex format for keyn");
  101.             }
  102.             if (AP4_ParseHex(salt_ascii, salt, 8)) {
  103.                 fprintf(stderr, "ERROR: invalid hex format for saltn");
  104.             }
  105.             // set the key in the map
  106.             processor.GetKeyMap().SetKey(track, key, salt);
  107.         } else if (input_filename == NULL) {
  108.             input_filename = arg;
  109.         } else if (output_filename == NULL) {
  110.             output_filename = arg;
  111.         } else {
  112.             fprintf(stderr, "ERROR: unexpected argument (%s)n", arg);
  113.             return 1;
  114.         }
  115.     }
  116.     // check the arguments
  117.     if (input_filename == NULL) {
  118.         fprintf(stderr, "ERROR: missing input filenamen");
  119.         return 1;
  120.     }
  121.     if (output_filename == NULL) {
  122.         fprintf(stderr, "ERROR: missing output filenamen");
  123.         return 1;
  124.     }
  125.     if (kms_uri == NULL) {
  126.         fprintf(stderr, "ERROR: missing kms urin");
  127.         return 1;
  128.     }
  129.     // create the input stream
  130.     AP4_ByteStream* input;
  131.     try{
  132.         input = new AP4_FileByteStream(input_filename,
  133.             AP4_FileByteStream::STREAM_MODE_READ);
  134.     } catch (AP4_Exception) {
  135.         fprintf(stderr, "ERROR: cannot open input file (%s)n", input_filename);
  136.         return 1;
  137.     }
  138.     // create the output stream
  139.     AP4_ByteStream* output;
  140.     try {
  141.         output = new AP4_FileByteStream(output_filename,
  142.             AP4_FileByteStream::STREAM_MODE_WRITE);
  143.     } catch (AP4_Exception) {
  144.         fprintf(stderr, "ERROR: cannot open output file (%s)n", output_filename);
  145.         return 1;
  146.     }
  147.     // process/decrypt the file
  148.     processor.Process(*input, *output);
  149.     // cleanup
  150.     input->Release();
  151.     output->Release();
  152.     return 0;
  153. }