| 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 | }
|
|---|
