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/cdefs.h> 29 #include <sys/param.h> 30 #include <err.h> 31 #include <stdint.h> 32 33 #include <lzma.h> 34 35 #include "mkuzip.h" 36 #include "mkuz_blk.h" 37 #include "mkuz_lzma.h" 38 39 struct mkuz_lzma { 40 lzma_filter filters[2]; 41 lzma_options_lzma opt_lzma; 42 lzma_stream strm; 43 }; 44 45 size_t 46 mkuz_lzma_cbound(size_t blksz) 47 { 48 return (lzma_stream_buffer_bound(blksz)); 49 } 50 51 void * 52 mkuz_lzma_init(int *comp_level) 53 { 54 struct mkuz_lzma *ulp; 55 56 if (*comp_level == USE_DEFAULT_LEVEL) 57 *comp_level = LZMA_PRESET_DEFAULT; 58 if (*comp_level < 0 || *comp_level > 9) 59 errx(1, "provided compression level %d is invalid", 60 *comp_level); 61 /* Not reached */ 62 63 ulp = mkuz_safe_zmalloc(sizeof(struct mkuz_lzma)); 64 65 /* Init lzma encoder */ 66 ulp->strm = (lzma_stream)LZMA_STREAM_INIT; 67 if (lzma_lzma_preset(&ulp->opt_lzma, *comp_level)) 68 errx(1, "Error loading LZMA preset"); 69 70 ulp->filters[0].id = LZMA_FILTER_LZMA2; 71 ulp->filters[0].options = &ulp->opt_lzma; 72 ulp->filters[1].id = LZMA_VLI_UNKNOWN; 73 74 return (void *)ulp; 75 } 76 77 void 78 mkuz_lzma_compress(void *p, const struct mkuz_blk *iblk, struct mkuz_blk *oblk) 79 { 80 lzma_ret ret; 81 struct mkuz_lzma *ulp; 82 83 ulp = (struct mkuz_lzma *)p; 84 85 ret = lzma_stream_encoder(&ulp->strm, ulp->filters, LZMA_CHECK_CRC32); 86 if (ret != LZMA_OK) { 87 if (ret == LZMA_MEMLIMIT_ERROR) 88 errx(1, "can't compress data: LZMA_MEMLIMIT_ERROR"); 89 90 errx(1, "can't compress data: LZMA compressor ERROR"); 91 } 92 93 ulp->strm.next_in = iblk->data; 94 ulp->strm.avail_in = iblk->info.len; 95 ulp->strm.next_out = oblk->data; 96 ulp->strm.avail_out = oblk->alen; 97 98 ret = lzma_code(&ulp->strm, LZMA_FINISH); 99 100 if (ret != LZMA_STREAM_END) 101 errx(1, "lzma_code FINISH failed, code=%d, pos(in=%zd, " 102 "out=%zd)", ret, (iblk->info.len - ulp->strm.avail_in), 103 (oblk->alen - ulp->strm.avail_out)); 104 105 #if 0 106 lzma_end(&ulp->strm); 107 #endif 108 109 oblk->info.len = oblk->alen - ulp->strm.avail_out; 110 } 111