| /////////////////////////////////////////////////////////////////////////////// |
| // |
| /// \file stream_flags_decoder.c |
| /// \brief Decodes Stream Header and tail from .lzma files |
| // |
| // Copyright (C) 2007 Lasse Collin |
| // |
| // This library is free software; you can redistribute it and/or |
| // modify it under the terms of the GNU Lesser General Public |
| // License as published by the Free Software Foundation; either |
| // version 2.1 of the License, or (at your option) any later version. |
| // |
| // This library is distributed in the hope that it will be useful, |
| // but WITHOUT ANY WARRANTY; without even the implied warranty of |
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| // Lesser General Public License for more details. |
| // |
| /////////////////////////////////////////////////////////////////////////////// |
| |
| #include "stream_flags_decoder.h" |
| #include "stream_common.h" |
| |
| |
| //////////// |
| // Common // |
| //////////// |
| |
| struct lzma_coder_s { |
| enum { |
| SEQ_HEADER_MAGIC, |
| SEQ_HEADER_FLAGS, |
| SEQ_HEADER_CRC32, |
| |
| SEQ_FOOTER_FLAGS, |
| SEQ_FOOTER_MAGIC, |
| } sequence; |
| |
| size_t pos; |
| uint32_t crc32; |
| |
| lzma_stream_flags *options; |
| }; |
| |
| |
| static void |
| stream_header_decoder_end(lzma_coder *coder, lzma_allocator *allocator) |
| { |
| lzma_free(coder, allocator); |
| return; |
| } |
| |
| |
| static bool |
| stream_flags_decode(const uint8_t *in, lzma_stream_flags *options) |
| { |
| // Reserved bits must be unset. |
| if (*in & 0xE0) |
| return true; |
| |
| options->check = *in & 0x07; |
| options->has_crc32 = (*in & 0x08) != 0; |
| options->is_multi = (*in & 0x10) != 0; |
| |
| return false; |
| } |
| |
| |
| //////////// |
| // Header // |
| //////////// |
| |
| static lzma_ret |
| stream_header_decode(lzma_coder *coder, |
| lzma_allocator *allocator lzma_attribute((unused)), |
| const uint8_t *restrict in, size_t *restrict in_pos, |
| size_t in_size, uint8_t *restrict out lzma_attribute((unused)), |
| size_t *restrict out_pos lzma_attribute((unused)), |
| size_t out_size lzma_attribute((unused)), |
| lzma_action action lzma_attribute((unused))) |
| { |
| while (*in_pos < in_size) |
| switch (coder->sequence) { |
| case SEQ_HEADER_MAGIC: |
| if (in[*in_pos] != lzma_header_magic[coder->pos]) |
| return LZMA_DATA_ERROR; |
| |
| ++*in_pos; |
| |
| if (++coder->pos == sizeof(lzma_header_magic)) { |
| coder->pos = 0; |
| coder->sequence = SEQ_HEADER_FLAGS; |
| } |
| |
| break; |
| |
| case SEQ_HEADER_FLAGS: |
| if (stream_flags_decode(in + *in_pos, coder->options)) |
| return LZMA_HEADER_ERROR; |
| |
| coder->crc32 = lzma_crc32(in + *in_pos, 1, 0); |
| |
| ++*in_pos; |
| coder->sequence = SEQ_HEADER_CRC32; |
| break; |
| |
| case SEQ_HEADER_CRC32: |
| if (in[*in_pos] != ((coder->crc32 >> (coder->pos * 8)) & 0xFF)) |
| return LZMA_DATA_ERROR; |
| |
| ++*in_pos; |
| |
| if (++coder->pos == 4) |
| return LZMA_STREAM_END; |
| |
| break; |
| |
| default: |
| return LZMA_PROG_ERROR; |
| } |
| |
| return LZMA_OK; |
| } |
| |
| |
| static lzma_ret |
| stream_header_decoder_init(lzma_next_coder *next, |
| lzma_allocator *allocator, lzma_stream_flags *options) |
| { |
| if (options == NULL) |
| return LZMA_PROG_ERROR; |
| |
| if (next->coder == NULL) { |
| next->coder = lzma_alloc(sizeof(lzma_coder), allocator); |
| if (next->coder == NULL) |
| return LZMA_MEM_ERROR; |
| } |
| |
| // Set the function pointers unconditionally, because they may |
| // have been pointing to footer decoder too. |
| next->code = &stream_header_decode; |
| next->end = &stream_header_decoder_end; |
| |
| next->coder->sequence = SEQ_HEADER_MAGIC; |
| next->coder->pos = 0; |
| next->coder->crc32 = 0; |
| next->coder->options = options; |
| |
| return LZMA_OK; |
| } |
| |
| |
| extern lzma_ret |
| lzma_stream_header_decoder_init(lzma_next_coder *next, |
| lzma_allocator *allocator, lzma_stream_flags *options) |
| { |
| lzma_next_coder_init( |
| stream_header_decoder_init, next, allocator, options); |
| } |
| |
| |
| extern LZMA_API lzma_ret |
| lzma_stream_header_decoder(lzma_stream *strm, lzma_stream_flags *options) |
| { |
| lzma_next_strm_init(strm, stream_header_decoder_init, options); |
| |
| strm->internal->supported_actions[LZMA_RUN] = true; |
| |
| return LZMA_OK; |
| } |
| |
| |
| ////////// |
| // Tail // |
| ////////// |
| |
| static lzma_ret |
| stream_tail_decode(lzma_coder *coder, |
| lzma_allocator *allocator lzma_attribute((unused)), |
| const uint8_t *restrict in, size_t *restrict in_pos, |
| size_t in_size, uint8_t *restrict out lzma_attribute((unused)), |
| size_t *restrict out_pos lzma_attribute((unused)), |
| size_t out_size lzma_attribute((unused)), |
| lzma_action action lzma_attribute((unused))) |
| { |
| while (*in_pos < in_size) |
| switch (coder->sequence) { |
| case SEQ_FOOTER_FLAGS: |
| if (stream_flags_decode(in + *in_pos, coder->options)) |
| return LZMA_HEADER_ERROR; |
| |
| ++*in_pos; |
| coder->sequence = SEQ_FOOTER_MAGIC; |
| break; |
| |
| case SEQ_FOOTER_MAGIC: |
| if (in[*in_pos] != lzma_footer_magic[coder->pos]) |
| return LZMA_DATA_ERROR; |
| |
| ++*in_pos; |
| |
| if (++coder->pos == sizeof(lzma_footer_magic)) |
| return LZMA_STREAM_END; |
| |
| break; |
| |
| default: |
| return LZMA_PROG_ERROR; |
| } |
| |
| return LZMA_OK; |
| } |
| |
| |
| static lzma_ret |
| stream_tail_decoder_init(lzma_next_coder *next, |
| lzma_allocator *allocator, lzma_stream_flags *options) |
| { |
| if (options == NULL) |
| return LZMA_PROG_ERROR; |
| |
| if (next->coder == NULL) { |
| next->coder = lzma_alloc(sizeof(lzma_coder), allocator); |
| if (next->coder == NULL) |
| return LZMA_MEM_ERROR; |
| } |
| |
| // Set the function pointers unconditionally, because they may |
| // have been pointing to footer decoder too. |
| next->code = &stream_tail_decode; |
| next->end = &stream_header_decoder_end; |
| |
| next->coder->sequence = SEQ_FOOTER_FLAGS; |
| next->coder->pos = 0; |
| next->coder->options = options; |
| |
| return LZMA_OK; |
| } |
| |
| |
| extern lzma_ret |
| lzma_stream_tail_decoder_init(lzma_next_coder *next, |
| lzma_allocator *allocator, lzma_stream_flags *options) |
| { |
| lzma_next_coder_init2(next, allocator, stream_header_decoder_init, |
| stream_tail_decoder_init, allocator, options); |
| } |
| |
| |
| extern LZMA_API lzma_ret |
| lzma_stream_tail_decoder(lzma_stream *strm, lzma_stream_flags *options) |
| { |
| lzma_next_strm_init2(strm, stream_header_decoder_init, |
| stream_tail_decoder_init, strm->allocator, options); |
| |
| strm->internal->supported_actions[LZMA_RUN] = true; |
| |
| return LZMA_OK; |
| } |