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 29 #include <sys/cdefs.h> 30 __FBSDID("$FreeBSD$"); 31 32 #include <sys/param.h> 33 #include <err.h> 34 #include <stdint.h> 35 36 #include <lzma.h> 37 38 #include "mkuzip.h" 39 #include "mkuz_lzma.h" 40 41 #define USED_BLOCKSIZE DEV_BSIZE 42 43 struct mkuz_lzma { 44 lzma_filter filters[2]; 45 lzma_options_lzma opt_lzma; 46 lzma_stream strm; 47 char *obuf; 48 uint32_t blksz; 49 }; 50 51 static struct mkuz_lzma ulzma = {.strm = LZMA_STREAM_INIT}; 52 53 void * 54 mkuz_lzma_init(uint32_t blksz) 55 { 56 if (blksz % USED_BLOCKSIZE != 0) { 57 errx(1, "cluster size should be multiple of %d", 58 USED_BLOCKSIZE); 59 /* Not reached */ 60 } 61 if (blksz > MAXPHYS) { 62 errx(1, "cluster size is too large"); 63 /* Not reached */ 64 } 65 ulzma.obuf = mkuz_safe_malloc(blksz * 2); 66 67 /* Init lzma encoder */ 68 if (lzma_lzma_preset(&ulzma.opt_lzma, LZMA_PRESET_DEFAULT)) 69 errx(1, "Error loading LZMA preset"); 70 71 ulzma.filters[0].id = LZMA_FILTER_LZMA2; 72 ulzma.filters[0].options = &ulzma.opt_lzma; 73 ulzma.filters[1].id = LZMA_VLI_UNKNOWN; 74 75 ulzma.blksz = blksz; 76 77 return (ulzma.obuf); 78 } 79 80 void 81 mkuz_lzma_compress(const char *ibuf, uint32_t *destlen) 82 { 83 lzma_ret ret; 84 85 ret = lzma_stream_encoder(&ulzma.strm, ulzma.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 ulzma.strm.next_in = ibuf; 94 ulzma.strm.avail_in = ulzma.blksz; 95 ulzma.strm.next_out = ulzma.obuf; 96 ulzma.strm.avail_out = ulzma.blksz * 2; 97 98 ret = lzma_code(&ulzma.strm, LZMA_FINISH); 99 100 if (ret != LZMA_STREAM_END) { 101 /* Error */ 102 errx(1, "lzma_code FINISH failed, code=%d, pos(in=%zd, " 103 "out=%zd)", ret, (ulzma.blksz - ulzma.strm.avail_in), 104 (ulzma.blksz * 2 - ulzma.strm.avail_out)); 105 } 106 107 lzma_end(&ulzma.strm); 108 109 *destlen = (ulzma.blksz * 2) - ulzma.strm.avail_out; 110 } 111