1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/src/win32/7zip/7z/CPP/7zip/Crypto/Rar20Crypto.cpp	Sat Mar 03 10:31:27 2012 -0600
     1.3 @@ -0,0 +1,136 @@
     1.4 +// Crypto/Rar20Crypto.cpp
     1.5 +
     1.6 +#include "StdAfx.h"
     1.7 +
     1.8 +extern "C"
     1.9 +{
    1.10 +  #include "../../../C/7zCrc.h"
    1.11 +  #include "../../../C/CpuArch.h"
    1.12 +  #include "../../../C/RotateDefs.h"
    1.13 +}
    1.14 +
    1.15 +#include "Rar20Crypto.h"
    1.16 +
    1.17 +namespace NCrypto {
    1.18 +namespace NRar20 {
    1.19 +
    1.20 +static const int kNumRounds = 32;
    1.21 +
    1.22 +static const Byte InitSubstTable[256] = {
    1.23 +  215, 19,149, 35, 73,197,192,205,249, 28, 16,119, 48,221,  2, 42,
    1.24 +  232,  1,177,233, 14, 88,219, 25,223,195,244, 90, 87,239,153,137,
    1.25 +  255,199,147, 70, 92, 66,246, 13,216, 40, 62, 29,217,230, 86,  6,
    1.26 +   71, 24,171,196,101,113,218,123, 93, 91,163,178,202, 67, 44,235,
    1.27 +  107,250, 75,234, 49,167,125,211, 83,114,157,144, 32,193,143, 36,
    1.28 +  158,124,247,187, 89,214,141, 47,121,228, 61,130,213,194,174,251,
    1.29 +   97,110, 54,229,115, 57,152, 94,105,243,212, 55,209,245, 63, 11,
    1.30 +  164,200, 31,156, 81,176,227, 21, 76, 99,139,188,127, 17,248, 51,
    1.31 +  207,120,189,210,  8,226, 41, 72,183,203,135,165,166, 60, 98,  7,
    1.32 +  122, 38,155,170, 69,172,252,238, 39,134, 59,128,236, 27,240, 80,
    1.33 +  131,  3, 85,206,145, 79,154,142,159,220,201,133, 74, 64, 20,129,
    1.34 +  224,185,138,103,173,182, 43, 34,254, 82,198,151,231,180, 58, 10,
    1.35 +  118, 26,102, 12, 50,132, 22,191,136,111,162,179, 45,  4,148,108,
    1.36 +  161, 56, 78,126,242,222, 15,175,146, 23, 33,241,181,190, 77,225,
    1.37 +    0, 46,169,186, 68, 95,237, 65, 53,208,253,168,  9, 18,100, 52,
    1.38 +  116,184,160, 96,109, 37, 30,106,140,104,150,  5,204,117,112, 84
    1.39 +};
    1.40 +
    1.41 +void CData::UpdateKeys(const Byte *data)
    1.42 +{
    1.43 +  for (int i = 0; i < 16; i += 4)
    1.44 +    for (int j = 0; j < 4; j++)
    1.45 +      Keys[j] ^= g_CrcTable[data[i + j]];
    1.46 +}
    1.47 +
    1.48 +static void Swap(Byte *b1, Byte *b2)
    1.49 +{
    1.50 +  Byte b = *b1;
    1.51 +  *b1 = *b2;
    1.52 +  *b2 = b;
    1.53 +}
    1.54 +
    1.55 +void CData::SetPassword(const Byte *password, UInt32 passwordLen)
    1.56 +{
    1.57 +  Keys[0] = 0xD3A3B879L;
    1.58 +  Keys[1] = 0x3F6D12F7L;
    1.59 +  Keys[2] = 0x7515A235L;
    1.60 +  Keys[3] = 0xA4E7F123L;
    1.61 +  
    1.62 +  Byte psw[256];
    1.63 +  memset(psw, 0, sizeof(psw));
    1.64 +  memcpy(psw, password, passwordLen);
    1.65 +  memcpy(SubstTable, InitSubstTable, sizeof(SubstTable));
    1.66 +
    1.67 +  for (UInt32 j = 0; j < 256; j++)
    1.68 +    for (UInt32 i = 0; i < passwordLen; i += 2)
    1.69 +    {
    1.70 +      UInt32 n2 = (Byte)g_CrcTable[(psw[i + 1] + j) & 0xFF];
    1.71 +      UInt32 n1 = (Byte)g_CrcTable[(psw[i] - j) & 0xFF];
    1.72 +      for (UInt32 k = 1; (n1 & 0xFF) != n2; n1++, k++)
    1.73 +        Swap(&SubstTable[n1 & 0xFF], &SubstTable[(n1 + i + k) & 0xFF]);
    1.74 +    }
    1.75 +  for (UInt32 i = 0; i < passwordLen; i+= 16)
    1.76 +    EncryptBlock(&psw[i]);
    1.77 +}
    1.78 +
    1.79 +void CData::CryptBlock(Byte *buf, bool encrypt)
    1.80 +{
    1.81 +  Byte inBuf[16];
    1.82 +  UInt32 A, B, C, D, T, TA, TB;
    1.83 +
    1.84 +  A = GetUi32(buf +  0) ^ Keys[0];
    1.85 +  B = GetUi32(buf +  4) ^ Keys[1];
    1.86 +  C = GetUi32(buf +  8) ^ Keys[2];
    1.87 +  D = GetUi32(buf + 12) ^ Keys[3];
    1.88 +
    1.89 +  if (!encrypt)
    1.90 +    memcpy(inBuf, buf, sizeof(inBuf));
    1.91 +  
    1.92 +  for (int i = 0; i < kNumRounds; i++)
    1.93 +  {
    1.94 +    UInt32 key = Keys[(encrypt ? i : (kNumRounds - 1 - i)) & 3];
    1.95 +    T = ((C + rotlFixed(D, 11)) ^ key);
    1.96 +    TA = A ^ SubstLong(T);
    1.97 +    T = ((D ^ rotlFixed(C, 17)) + key);
    1.98 +    TB = B ^ SubstLong(T);
    1.99 +    A = C;
   1.100 +    B = D;
   1.101 +    C = TA;
   1.102 +    D = TB;
   1.103 +  }
   1.104 +
   1.105 +  SetUi32(buf +  0, C ^ Keys[0]);
   1.106 +  SetUi32(buf +  4, D ^ Keys[1]);
   1.107 +  SetUi32(buf +  8, A ^ Keys[2]);
   1.108 +  SetUi32(buf + 12, B ^ Keys[3]);
   1.109 +
   1.110 +  UpdateKeys(encrypt ? buf : inBuf);
   1.111 +}
   1.112 +
   1.113 +STDMETHODIMP CDecoder::CryptoSetPassword(const Byte *data, UInt32 size)
   1.114 +{
   1.115 +  _cipher.SetPassword(data, size);
   1.116 +  return S_OK;
   1.117 +}
   1.118 +
   1.119 +STDMETHODIMP CDecoder::Init()
   1.120 +{
   1.121 +  return S_OK;
   1.122 +}
   1.123 +
   1.124 +static const UInt32 kBlockSize = 16;
   1.125 +
   1.126 +STDMETHODIMP_(UInt32) CDecoder::Filter(Byte *data, UInt32 size)
   1.127 +{
   1.128 +  if (size == 0)
   1.129 +    return 0;
   1.130 +  if (size < kBlockSize)
   1.131 +    return kBlockSize;
   1.132 +  UInt32 i;
   1.133 +  size -= kBlockSize;
   1.134 +  for (i = 0; i <= size; i += kBlockSize)
   1.135 +    _cipher.DecryptBlock(data + i);
   1.136 +  return i;
   1.137 +}
   1.138 +
   1.139 +}}