开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 多媒体编程 > 查看源码
Ap4StreamCipher.cpp
资源名称：fullMPCcode.rar　[点击查看]
上传用户：xjjlds
上传日期：2015-12-05
资源大小：22823k
文件大小：5k
源码类别：

多媒体编程

开发平台：

Visual C++

Ap4StreamCipher.cpp：源码内容
  1. /*
  2.  * Copyright (c) 2005 Gilles Boccon-Gibod 
  3.  */
  5. /*----------------------------------------------------------------------
  6. |       includes
  7. +---------------------------------------------------------------------*/
  8. #include "Ap4StreamCipher.h"
  10. /*----------------------------------------------------------------------
  11. |       AP4_StreamCipher::AP4_StreamCipher
  12. +---------------------------------------------------------------------*/
  13. AP4_StreamCipher::AP4_StreamCipher(const AP4_UI08* key,
  14.                                    const AP4_UI08* salt,
  15.                                    AP4_Size        iv_size) :
  16.     m_StreamOffset(0),
  17.     m_IvSize(iv_size),
  18.     m_BlockCipher(NULL)
  19. {
  20.     // clamp the IV size to the max supported size
  21.     if (iv_size > 4) {
  22.         m_IvSize = 4;
  23.     }
  25.     // set the initial state
  26.     Reset(key, salt);
  27. }
  29. /*----------------------------------------------------------------------
  30. |       AP4_StreamCipher::~AP4_StreamCipher
  31. +---------------------------------------------------------------------*/
  32. AP4_StreamCipher::~AP4_StreamCipher()
  33. {
  34.     // delete the block cipher
  35.     if (m_BlockCipher) {
  36.         delete m_BlockCipher;
  37.     }
  38. }
  40. /*----------------------------------------------------------------------
  41. |       AP4_StreamCipher::Reset
  42. +---------------------------------------------------------------------*/
  43. AP4_Result
  44. AP4_StreamCipher::Reset(const AP4_UI08* key, const AP4_UI08* salt) 
  45. {
  46.     if (salt) {
  47.         // initialize the counter with a salting key
  48.         for (AP4_UI32 i=0; i<AP4_AES_BLOCK_SIZE; i++) {
  49.             m_CBlock[i] = salt[i];
  50.         }
  51.     } else {
  52.         // initialize the counter with no salting key
  53.         for (AP4_UI32 i = 0; i < AP4_AES_BLOCK_SIZE; i++) {
  54.             m_CBlock[i] = 0;
  55.         }
  56.     }
  58.     // (re)create the block cipher
  59.     if (key != NULL) {
  60.         // delete the block cipher if needed
  61.         if (m_BlockCipher) {
  62.             delete m_BlockCipher;
  63.         }
  64.         
  65.         // (re)create one
  66.         m_BlockCipher = new AP4_AesBlockCipher(key);
  67.     }
  69.     // reset the stream offset
  70.     SetStreamOffset(0);
  71.   
  72.     return AP4_SUCCESS;
  73. }
  75. /*----------------------------------------------------------------------
  76. |       AP4_StreamCipher::SetCounter
  77. +---------------------------------------------------------------------*/
  78. void
  79. AP4_StreamCipher::SetCounter(AP4_Offset block_offset)
  80. {
  81.     // set the counter bytes
  82.     for (AP4_UI32 i = 0; i < m_IvSize; i++) {
  83.         m_CBlock[AP4_AES_BLOCK_SIZE-1-i] = 
  84. (AP4_UI08)((block_offset>>(8*i)) & 0xFF);
  85.     }
  86. }
  88. /*----------------------------------------------------------------------
  89. |       AP4_StreamCipher::SetStreamOffset
  90. +---------------------------------------------------------------------*/
  91. AP4_Result
  92. AP4_StreamCipher::SetStreamOffset(AP4_Offset offset)
  93. {
  94.     // do nothing if we're already at that offset
  95.     if (offset == m_StreamOffset) return AP4_SUCCESS;
  97.     // update the offset
  98.     m_StreamOffset = offset;
  100.     // update the key stream if necessary
  101.     if (m_StreamOffset & 0xF) {
  102.         return UpdateKeyStream(m_StreamOffset/AP4_AES_BLOCK_SIZE);
  103.     }
  104.     
  105.     return AP4_SUCCESS;
  106. }
  108. /*----------------------------------------------------------------------
  109. |       AP4_StreamCipher::UpdateKeyStream
  110. +---------------------------------------------------------------------*/
  111. AP4_Result
  112. AP4_StreamCipher::UpdateKeyStream(AP4_Offset block_offset)
  113. {
  114.     // compute the new counter
  115.     SetCounter(block_offset);
  117.     // compute the key block (x) from the counter block (c)
  118.     return m_BlockCipher->EncryptBlock(m_CBlock, m_XBlock);
  119. }
  121. /*----------------------------------------------------------------------
  122. |       AP4_StreamCipher::ProcessBuffer
  123. +---------------------------------------------------------------------*/
  124. AP4_Result
  125. AP4_StreamCipher::ProcessBuffer(const AP4_UI08* in, 
  126.                                 AP4_UI08*       out, 
  127.                                 AP4_Size        size)
  128. {
  129.     if (m_BlockCipher == NULL) return AP4_ERROR_INVALID_STATE;
  131.     while (size) {
  132.         // compute the number of bytes available in this chunk
  133.         AP4_UI32 index = m_StreamOffset & (AP4_AES_BLOCK_SIZE-1);
  134.         AP4_UI32 chunk;
  136.         // update the key stream if we are on a boundary
  137.         if (index == 0) {
  138.             UpdateKeyStream(m_StreamOffset/AP4_AES_BLOCK_SIZE);
  139.             chunk = AP4_AES_BLOCK_SIZE;
  140.         }
  142.         // compute the number of bytes remaining in the chunk
  143.         chunk = AP4_AES_BLOCK_SIZE - index;
  144.         if (chunk > size) chunk = size;
  146.         // encrypt/decrypt the chunk
  147.         AP4_UI08* x = &m_XBlock[index];
  148.         for (AP4_UI32 i = 0; i < chunk; i++) {
  149.             *out++ = *in++ ^ *x++;
  150.         }
  152.         // update offset and size
  153.         m_StreamOffset += chunk;
  154.         size -= chunk;
  155.     }
  157.     return AP4_SUCCESS;
  158. }