src/liblzma/check/check.c

   1 ///////////////////////////////////////////////////////////////////////////////
   2 //
   3 /// \file       check.c
   4 /// \brief      Single API to access different integrity checks
   5 //
   6 //  Author:     Lasse Collin
   7 //
   8 //  This file has been put into the public domain.
   9 //  You can do whatever you want with this file.
  10 //
  11 ///////////////////////////////////////////////////////////////////////////////
  12
  13 #include "check.h"
  14
  15
  16 extern LZMA_API(lzma_bool)
  17 lzma_check_is_supported(lzma_check type)
  18 {
  19         if ((unsigned int)(type) > LZMA_CHECK_ID_MAX)
  20                 return false;
  21
  22         static const lzma_bool available_checks[LZMA_CHECK_ID_MAX + 1] = {
  23                 true,   // LZMA_CHECK_NONE
  24
  25 #ifdef HAVE_CHECK_CRC32
  26                 true,
  27 #else
  28                 false,
  29 #endif
  30
  31                 false,  // Reserved
  32                 false,  // Reserved
  33
  34 #ifdef HAVE_CHECK_CRC64
  35                 true,
  36 #else
  37                 false,
  38 #endif
  39
  40                 false,  // Reserved
  41                 false,  // Reserved
  42                 false,  // Reserved
  43                 false,  // Reserved
  44                 false,  // Reserved
  45
  46 #ifdef HAVE_CHECK_SHA256
  47                 true,
  48 #else
  49                 false,
  50 #endif
  51
  52                 false,  // Reserved
  53                 false,  // Reserved
  54                 false,  // Reserved
  55                 false,  // Reserved
  56                 false,  // Reserved
  57         };
  58
  59         return available_checks[(unsigned int)(type)];
  60 }
  61
  62
  63 extern LZMA_API(uint32_t)
  64 lzma_check_size(lzma_check type)
  65 {
  66         if ((unsigned int)(type) > LZMA_CHECK_ID_MAX)
  67                 return UINT32_MAX;
  68
  69         // See file-format.txt section 2.1.1.2.
  70         static const uint8_t check_sizes[LZMA_CHECK_ID_MAX + 1] = {
  71                 0,
  72                 4, 4, 4,
  73                 8, 8, 8,
  74                 16, 16, 16,
  75                 32, 32, 32,
  76                 64, 64, 64
  77         };
  78
  79         return check_sizes[(unsigned int)(type)];
  80 }
  81
  82
  83 extern void
  84 lzma_check_init(lzma_check_state *check, lzma_check type)
  85 {
  86         switch (type) {
  87         case LZMA_CHECK_NONE:
  88                 break;
  89
  90 #ifdef HAVE_CHECK_CRC32
  91         case LZMA_CHECK_CRC32:
  92                 check->state.crc32 = 0;
  93                 break;
  94 #endif
  95
  96 #ifdef HAVE_CHECK_CRC64
  97         case LZMA_CHECK_CRC64:
  98                 check->state.crc64 = 0;
  99                 break;
 100 #endif
 101
 102 #ifdef HAVE_CHECK_SHA256
 103         case LZMA_CHECK_SHA256:
 104                 lzma_sha256_init(check);
 105                 break;
 106 #endif
 107
 108         default:
 109                 break;
 110         }
 111
 112         return;
 113 }
 114
 115
 116 extern void
 117 lzma_check_update(lzma_check_state *check, lzma_check type,
 118                 const uint8_t *buf, size_t size)
 119 {
 120         switch (type) {
 121 #ifdef HAVE_CHECK_CRC32
 122         case LZMA_CHECK_CRC32:
 123                 check->state.crc32 = lzma_crc32(buf, size, check->state.crc32);
 124                 break;
 125 #endif
 126
 127 #ifdef HAVE_CHECK_CRC64
 128         case LZMA_CHECK_CRC64:
 129                 check->state.crc64 = lzma_crc64(buf, size, check->state.crc64);
 130                 break;
 131 #endif
 132
 133 #ifdef HAVE_CHECK_SHA256
 134         case LZMA_CHECK_SHA256:
 135                 lzma_sha256_update(buf, size, check);
 136                 break;
 137 #endif
 138
 139         default:
 140                 break;
 141         }
 142
 143         return;
 144 }
 145
 146
 147 extern void
 148 lzma_check_finish(lzma_check_state *check, lzma_check type)
 149 {
 150         switch (type) {
 151 #ifdef HAVE_CHECK_CRC32
 152         case LZMA_CHECK_CRC32:
 153                 check->buffer.u32[0] = conv32le(check->state.crc32);
 154                 break;
 155 #endif
 156
 157 #ifdef HAVE_CHECK_CRC64
 158         case LZMA_CHECK_CRC64:
 159                 check->buffer.u64[0] = conv64le(check->state.crc64);
 160                 break;
 161 #endif
 162
 163 #ifdef HAVE_CHECK_SHA256
 164         case LZMA_CHECK_SHA256:
 165                 lzma_sha256_finish(check);
 166                 break;
 167 #endif
 168
 169         default:
 170                 break;
 171         }
 172
 173         return;
 174 }