CPP/7zip/Crypto/Rar20Crypto.cpp

   1 // Crypto/Rar20Crypto.cpp\r
   2 \r
   3 #include "StdAfx.h"\r
   4 \r
   5 #include "../../../C/7zCrc.h"\r
   6 #include "../../../C/CpuArch.h"\r
   7 #include "../../../C/RotateDefs.h"\r
   8 \r
   9 #include "Rar20Crypto.h"\r
  10 \r
  11 namespace NCrypto {\r
  12 namespace NRar20 {\r
  13 \r
  14 static const int kNumRounds = 32;\r
  15 \r
  16 static const Byte InitSubstTable[256] = {\r
  17   215, 19,149, 35, 73,197,192,205,249, 28, 16,119, 48,221,  2, 42,\r
  18   232,  1,177,233, 14, 88,219, 25,223,195,244, 90, 87,239,153,137,\r
  19   255,199,147, 70, 92, 66,246, 13,216, 40, 62, 29,217,230, 86,  6,\r
  20    71, 24,171,196,101,113,218,123, 93, 91,163,178,202, 67, 44,235,\r
  21   107,250, 75,234, 49,167,125,211, 83,114,157,144, 32,193,143, 36,\r
  22   158,124,247,187, 89,214,141, 47,121,228, 61,130,213,194,174,251,\r
  23    97,110, 54,229,115, 57,152, 94,105,243,212, 55,209,245, 63, 11,\r
  24   164,200, 31,156, 81,176,227, 21, 76, 99,139,188,127, 17,248, 51,\r
  25   207,120,189,210,  8,226, 41, 72,183,203,135,165,166, 60, 98,  7,\r
  26   122, 38,155,170, 69,172,252,238, 39,134, 59,128,236, 27,240, 80,\r
  27   131,  3, 85,206,145, 79,154,142,159,220,201,133, 74, 64, 20,129,\r
  28   224,185,138,103,173,182, 43, 34,254, 82,198,151,231,180, 58, 10,\r
  29   118, 26,102, 12, 50,132, 22,191,136,111,162,179, 45,  4,148,108,\r
  30   161, 56, 78,126,242,222, 15,175,146, 23, 33,241,181,190, 77,225,\r
  31     0, 46,169,186, 68, 95,237, 65, 53,208,253,168,  9, 18,100, 52,\r
  32   116,184,160, 96,109, 37, 30,106,140,104,150,  5,204,117,112, 84\r
  33 };\r
  34 \r
  35 void CData::UpdateKeys(const Byte *data)\r
  36 {\r
  37   for (int i = 0; i < 16; i += 4)\r
  38     for (int j = 0; j < 4; j++)\r
  39       Keys[j] ^= g_CrcTable[data[i + j]];\r
  40 }\r
  41 \r
  42 static void Swap(Byte *b1, Byte *b2)\r
  43 {\r
  44   Byte b = *b1;\r
  45   *b1 = *b2;\r
  46   *b2 = b;\r
  47 }\r
  48 \r
  49 void CData::SetPassword(const Byte *password, UInt32 passwordLen)\r
  50 {\r
  51   Keys[0] = 0xD3A3B879L;\r
  52   Keys[1] = 0x3F6D12F7L;\r
  53   Keys[2] = 0x7515A235L;\r
  54   Keys[3] = 0xA4E7F123L;\r
  55   \r
  56   Byte psw[256];\r
  57   memset(psw, 0, sizeof(psw));\r
  58   memcpy(psw, password, passwordLen);\r
  59   memcpy(SubstTable, InitSubstTable, sizeof(SubstTable));\r
  60 \r
  61   for (UInt32 j = 0; j < 256; j++)\r
  62     for (UInt32 i = 0; i < passwordLen; i += 2)\r
  63     {\r
  64       UInt32 n2 = (Byte)g_CrcTable[(psw[i + 1] + j) & 0xFF];\r
  65       UInt32 n1 = (Byte)g_CrcTable[(psw[i] - j) & 0xFF];\r
  66       for (UInt32 k = 1; (n1 & 0xFF) != n2; n1++, k++)\r
  67         Swap(&SubstTable[n1 & 0xFF], &SubstTable[(n1 + i + k) & 0xFF]);\r
  68     }\r
  69   for (UInt32 i = 0; i < passwordLen; i+= 16)\r
  70     EncryptBlock(&psw[i]);\r
  71 }\r
  72 \r
  73 void CData::CryptBlock(Byte *buf, bool encrypt)\r
  74 {\r
  75   Byte inBuf[16];\r
  76   UInt32 A, B, C, D, T, TA, TB;\r
  77 \r
  78   A = GetUi32(buf +  0) ^ Keys[0];\r
  79   B = GetUi32(buf +  4) ^ Keys[1];\r
  80   C = GetUi32(buf +  8) ^ Keys[2];\r
  81   D = GetUi32(buf + 12) ^ Keys[3];\r
  82 \r
  83   if (!encrypt)\r
  84     memcpy(inBuf, buf, sizeof(inBuf));\r
  85   \r
  86   for (int i = 0; i < kNumRounds; i++)\r
  87   {\r
  88     UInt32 key = Keys[(encrypt ? i : (kNumRounds - 1 - i)) & 3];\r
  89     T = ((C + rotlFixed(D, 11)) ^ key);\r
  90     TA = A ^ SubstLong(T);\r
  91     T = ((D ^ rotlFixed(C, 17)) + key);\r
  92     TB = B ^ SubstLong(T);\r
  93     A = C;\r
  94     B = D;\r
  95     C = TA;\r
  96     D = TB;\r
  97   }\r
  98 \r
  99   SetUi32(buf +  0, C ^ Keys[0]);\r
 100   SetUi32(buf +  4, D ^ Keys[1]);\r
 101   SetUi32(buf +  8, A ^ Keys[2]);\r
 102   SetUi32(buf + 12, B ^ Keys[3]);\r
 103 \r
 104   UpdateKeys(encrypt ? buf : inBuf);\r
 105 }\r
 106 \r
 107 STDMETHODIMP CDecoder::CryptoSetPassword(const Byte *data, UInt32 size)\r
 108 {\r
 109   _cipher.SetPassword(data, size);\r
 110   return S_OK;\r
 111 }\r
 112 \r
 113 STDMETHODIMP CDecoder::Init()\r
 114 {\r
 115   return S_OK;\r
 116 }\r
 117 \r
 118 static const UInt32 kBlockSize = 16;\r
 119 \r
 120 STDMETHODIMP_(UInt32) CDecoder::Filter(Byte *data, UInt32 size)\r
 121 {\r
 122   if (size == 0)\r
 123     return 0;\r
 124   if (size < kBlockSize)\r
 125     return kBlockSize;\r
 126   UInt32 i;\r
 127   size -= kBlockSize;\r
 128   for (i = 0; i <= size; i += kBlockSize)\r
 129     _cipher.DecryptBlock(data + i);\r
 130   return i;\r
 131 }\r
 132 \r
 133 }}\r