deps/lzma-16.04/C/Xz.c

   1 /* Xz.c - Xz\r
   2 2015-05-01 : Igor Pavlov : Public domain */\r
   3 \r
   4 #include "Precomp.h"\r
   5 \r
   6 #include "7zCrc.h"\r
   7 #include "CpuArch.h"\r
   8 #include "Xz.h"\r
   9 #include "XzCrc64.h"\r
  10 \r
  11 const Byte XZ_SIG[XZ_SIG_SIZE] = { 0xFD, '7', 'z', 'X', 'Z', 0 };\r
  12 const Byte XZ_FOOTER_SIG[XZ_FOOTER_SIG_SIZE] = { 'Y', 'Z' };\r
  13 \r
  14 unsigned Xz_WriteVarInt(Byte *buf, UInt64 v)\r
  15 {\r
  16   unsigned i = 0;\r
  17   do\r
  18   {\r
  19     buf[i++] = (Byte)((v & 0x7F) | 0x80);\r
  20     v >>= 7;\r
  21   }\r
  22   while (v != 0);\r
  23   buf[i - 1] &= 0x7F;\r
  24   return i;\r
  25 }\r
  26 \r
  27 void Xz_Construct(CXzStream *p)\r
  28 {\r
  29   p->numBlocks = p->numBlocksAllocated = 0;\r
  30   p->blocks = 0;\r
  31   p->flags = 0;\r
  32 }\r
  33 \r
  34 void Xz_Free(CXzStream *p, ISzAlloc *alloc)\r
  35 {\r
  36   alloc->Free(alloc, p->blocks);\r
  37   p->numBlocks = p->numBlocksAllocated = 0;\r
  38   p->blocks = 0;\r
  39 }\r
  40 \r
  41 unsigned XzFlags_GetCheckSize(CXzStreamFlags f)\r
  42 {\r
  43   unsigned t = XzFlags_GetCheckType(f);\r
  44   return (t == 0) ? 0 : (4 << ((t - 1) / 3));\r
  45 }\r
  46 \r
  47 void XzCheck_Init(CXzCheck *p, unsigned mode)\r
  48 {\r
  49   p->mode = mode;\r
  50   switch (mode)\r
  51   {\r
  52     case XZ_CHECK_CRC32: p->crc = CRC_INIT_VAL; break;\r
  53     case XZ_CHECK_CRC64: p->crc64 = CRC64_INIT_VAL; break;\r
  54     case XZ_CHECK_SHA256: Sha256_Init(&p->sha); break;\r
  55   }\r
  56 }\r
  57 \r
  58 void XzCheck_Update(CXzCheck *p, const void *data, size_t size)\r
  59 {\r
  60   switch (p->mode)\r
  61   {\r
  62     case XZ_CHECK_CRC32: p->crc = CrcUpdate(p->crc, data, size); break;\r
  63     case XZ_CHECK_CRC64: p->crc64 = Crc64Update(p->crc64, data, size); break;\r
  64     case XZ_CHECK_SHA256: Sha256_Update(&p->sha, (const Byte *)data, size); break;\r
  65   }\r
  66 }\r
  67 \r
  68 int XzCheck_Final(CXzCheck *p, Byte *digest)\r
  69 {\r
  70   switch (p->mode)\r
  71   {\r
  72     case XZ_CHECK_CRC32:\r
  73       SetUi32(digest, CRC_GET_DIGEST(p->crc));\r
  74       break;\r
  75     case XZ_CHECK_CRC64:\r
  76     {\r
  77       int i;\r
  78       UInt64 v = CRC64_GET_DIGEST(p->crc64);\r
  79       for (i = 0; i < 8; i++, v >>= 8)\r
  80         digest[i] = (Byte)(v & 0xFF);\r
  81       break;\r
  82     }\r
  83     case XZ_CHECK_SHA256:\r
  84       Sha256_Final(&p->sha, digest);\r
  85       break;\r
  86     default:\r
  87       return 0;\r
  88   }\r
  89   return 1;\r
  90 }\r