1 /* -*- mode: C; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2 // vim: expandtab:ts=8:sw=4:softtabstop=4:
3 ///////////////////////////////////////////////////////////////////////////////
4 //
5 /// \file       easy_encoder.c
6 /// \brief      Easy .xz Stream encoder initialization
7 //
8 //  Author:     Lasse Collin
9 //
10 //  This file has been put into the public domain.
11 //  You can do whatever you want with this file.
12 //
13 ///////////////////////////////////////////////////////////////////////////////
14 
15 #include "easy_preset.h"
16 #include "stream_encoder.h"
17 
18 
19 struct lzma_coder_s {
20 	lzma_next_coder stream_encoder;
21 	lzma_options_easy opt_easy;
22 };
23 
24 
25 static lzma_ret
easy_encode(lzma_coder * coder,lzma_allocator * allocator,const uint8_t * restrict in,size_t * restrict in_pos,size_t in_size,uint8_t * restrict out,size_t * restrict out_pos,size_t out_size,lzma_action action)26 easy_encode(lzma_coder *coder, lzma_allocator *allocator,
27 		const uint8_t *restrict in, size_t *restrict in_pos,
28 		size_t in_size, uint8_t *restrict out,
29 		size_t *restrict out_pos, size_t out_size, lzma_action action)
30 {
31 	return coder->stream_encoder.code(
32 			coder->stream_encoder.coder, allocator,
33 			in, in_pos, in_size, out, out_pos, out_size, action);
34 }
35 
36 
37 static void
easy_encoder_end(lzma_coder * coder,lzma_allocator * allocator)38 easy_encoder_end(lzma_coder *coder, lzma_allocator *allocator)
39 {
40 	lzma_next_end(&coder->stream_encoder, allocator);
41 	lzma_free(coder, allocator);
42 	return;
43 }
44 
45 
46 static lzma_ret
easy_encoder_init(lzma_next_coder * next,lzma_allocator * allocator,uint32_t preset,lzma_check check)47 easy_encoder_init(lzma_next_coder *next, lzma_allocator *allocator,
48 		uint32_t preset, lzma_check check)
49 {
50 	lzma_next_coder_init(&easy_encoder_init, next, allocator);
51 
52 	if (next->coder == NULL) {
53 		next->coder = lzma_alloc(sizeof(lzma_coder), allocator);
54 		if (next->coder == NULL)
55 			return LZMA_MEM_ERROR;
56 
57 		next->code = &easy_encode;
58 		next->end = &easy_encoder_end;
59 
60 		next->coder->stream_encoder = LZMA_NEXT_CODER_INIT;
61 	}
62 
63 	if (lzma_easy_preset(&next->coder->opt_easy, preset))
64 		return LZMA_OPTIONS_ERROR;
65 
66 	return lzma_stream_encoder_init(&next->coder->stream_encoder,
67 			allocator, next->coder->opt_easy.filters, check);
68 }
69 
70 
71 extern LZMA_API(lzma_ret)
lzma_easy_encoder(lzma_stream * strm,uint32_t preset,lzma_check check)72 lzma_easy_encoder(lzma_stream *strm, uint32_t preset, lzma_check check)
73 {
74 	lzma_next_strm_init(easy_encoder_init, strm, preset, check);
75 
76 	strm->internal->supported_actions[LZMA_RUN] = true;
77 	strm->internal->supported_actions[LZMA_SYNC_FLUSH] = true;
78 	strm->internal->supported_actions[LZMA_FULL_FLUSH] = true;
79 	strm->internal->supported_actions[LZMA_FINISH] = true;
80 
81 	return LZMA_OK;
82 }
83