1 /* 2 * Copyright (c) 2004-2016 Maxim Sobolev <sobomax@FreeBSD.org> 3 * Copyright (c) 2011 Aleksandr Rybalko <ray@ddteam.net> 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 */ 27 28 #include <sys/param.h> 29 #include <err.h> 30 #include <stdint.h> 31 32 #include <lzma.h> 33 34 #include "mkuzip.h" 35 #include "mkuz_blk.h" 36 #include "mkuz_lzma.h" 37 38 struct mkuz_lzma { 39 lzma_filter filters[2]; 40 lzma_options_lzma opt_lzma; 41 lzma_stream strm; 42 }; 43 44 size_t 45 mkuz_lzma_cbound(size_t blksz) 46 { 47 return (lzma_stream_buffer_bound(blksz)); 48 } 49 50 void * 51 mkuz_lzma_init(int *comp_level) 52 { 53 struct mkuz_lzma *ulp; 54 55 if (*comp_level == USE_DEFAULT_LEVEL) 56 *comp_level = LZMA_PRESET_DEFAULT; 57 if (*comp_level < 0 || *comp_level > 9) 58 errx(1, "provided compression level %d is invalid", 59 *comp_level); 60 /* Not reached */ 61 62 ulp = mkuz_safe_zmalloc(sizeof(struct mkuz_lzma)); 63 64 /* Init lzma encoder */ 65 ulp->strm = (lzma_stream)LZMA_STREAM_INIT; 66 if (lzma_lzma_preset(&ulp->opt_lzma, *comp_level)) 67 errx(1, "Error loading LZMA preset"); 68 69 ulp->filters[0].id = LZMA_FILTER_LZMA2; 70 ulp->filters[0].options = &ulp->opt_lzma; 71 ulp->filters[1].id = LZMA_VLI_UNKNOWN; 72 73 return (void *)ulp; 74 } 75 76 void 77 mkuz_lzma_compress(void *p, const struct mkuz_blk *iblk, struct mkuz_blk *oblk) 78 { 79 lzma_ret ret; 80 struct mkuz_lzma *ulp; 81 82 ulp = (struct mkuz_lzma *)p; 83 84 ret = lzma_stream_encoder(&ulp->strm, ulp->filters, LZMA_CHECK_CRC32); 85 if (ret != LZMA_OK) { 86 if (ret == LZMA_MEMLIMIT_ERROR) 87 errx(1, "can't compress data: LZMA_MEMLIMIT_ERROR"); 88 89 errx(1, "can't compress data: LZMA compressor ERROR"); 90 } 91 92 ulp->strm.next_in = iblk->data; 93 ulp->strm.avail_in = iblk->info.len; 94 ulp->strm.next_out = oblk->data; 95 ulp->strm.avail_out = oblk->alen; 96 97 ret = lzma_code(&ulp->strm, LZMA_FINISH); 98 99 if (ret != LZMA_STREAM_END) 100 errx(1, "lzma_code FINISH failed, code=%d, pos(in=%zd, " 101 "out=%zd)", ret, (iblk->info.len - ulp->strm.avail_in), 102 (oblk->alen - ulp->strm.avail_out)); 103 104 #if 0 105 lzma_end(&ulp->strm); 106 #endif 107 108 oblk->info.len = oblk->alen - ulp->strm.avail_out; 109 } 110