12940b44dSPeter Avalos ///////////////////////////////////////////////////////////////////////////////
22940b44dSPeter Avalos //
32940b44dSPeter Avalos /// \file filter_buffer_encoder.c
42940b44dSPeter Avalos /// \brief Single-call raw encoding
52940b44dSPeter Avalos //
62940b44dSPeter Avalos // Author: Lasse Collin
72940b44dSPeter Avalos //
82940b44dSPeter Avalos // This file has been put into the public domain.
92940b44dSPeter Avalos // You can do whatever you want with this file.
102940b44dSPeter Avalos //
112940b44dSPeter Avalos ///////////////////////////////////////////////////////////////////////////////
122940b44dSPeter Avalos
132940b44dSPeter Avalos #include "filter_encoder.h"
142940b44dSPeter Avalos
152940b44dSPeter Avalos
162940b44dSPeter Avalos extern LZMA_API(lzma_ret)
lzma_raw_buffer_encode(const lzma_filter * filters,const lzma_allocator * allocator,const uint8_t * in,size_t in_size,uint8_t * out,size_t * out_pos,size_t out_size)17*15ab8c86SJohn Marino lzma_raw_buffer_encode(
18*15ab8c86SJohn Marino const lzma_filter *filters, const lzma_allocator *allocator,
19*15ab8c86SJohn Marino const uint8_t *in, size_t in_size,
20*15ab8c86SJohn Marino uint8_t *out, size_t *out_pos, size_t out_size)
212940b44dSPeter Avalos {
222940b44dSPeter Avalos // Validate what isn't validated later in filter_common.c.
232940b44dSPeter Avalos if ((in == NULL && in_size != 0) || out == NULL
242940b44dSPeter Avalos || out_pos == NULL || *out_pos > out_size)
252940b44dSPeter Avalos return LZMA_PROG_ERROR;
262940b44dSPeter Avalos
272940b44dSPeter Avalos // Initialize the encoder
282940b44dSPeter Avalos lzma_next_coder next = LZMA_NEXT_CODER_INIT;
292940b44dSPeter Avalos return_if_error(lzma_raw_encoder_init(&next, allocator, filters));
302940b44dSPeter Avalos
312940b44dSPeter Avalos // Store the output position so that we can restore it if
322940b44dSPeter Avalos // something goes wrong.
332940b44dSPeter Avalos const size_t out_start = *out_pos;
342940b44dSPeter Avalos
352940b44dSPeter Avalos // Do the actual encoding and free coder's memory.
362940b44dSPeter Avalos size_t in_pos = 0;
372940b44dSPeter Avalos lzma_ret ret = next.code(next.coder, allocator, in, &in_pos, in_size,
382940b44dSPeter Avalos out, out_pos, out_size, LZMA_FINISH);
392940b44dSPeter Avalos lzma_next_end(&next, allocator);
402940b44dSPeter Avalos
412940b44dSPeter Avalos if (ret == LZMA_STREAM_END) {
422940b44dSPeter Avalos ret = LZMA_OK;
432940b44dSPeter Avalos } else {
442940b44dSPeter Avalos if (ret == LZMA_OK) {
452940b44dSPeter Avalos // Output buffer was too small.
462940b44dSPeter Avalos assert(*out_pos == out_size);
472940b44dSPeter Avalos ret = LZMA_BUF_ERROR;
482940b44dSPeter Avalos }
492940b44dSPeter Avalos
502940b44dSPeter Avalos // Restore the output position.
512940b44dSPeter Avalos *out_pos = out_start;
522940b44dSPeter Avalos }
532940b44dSPeter Avalos
542940b44dSPeter Avalos return ret;
552940b44dSPeter Avalos }
56