1/*
2 * Copyright (c) 2004-2016 Maxim Sobolev <sobomax@FreeBSD.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD$");
29
30#include <sys/param.h>
31#include <err.h>
32#include <stdint.h>
33
34#include <zlib.h>
35
36#include "mkuzip.h"
37#include "mkuz_zlib.h"
38#include "mkuz_blk.h"
39
40struct mkuz_zlib {
41	uLongf oblen;
42	uint32_t blksz;
43};
44
45void *
46mkuz_zlib_init(uint32_t blksz)
47{
48	struct mkuz_zlib *zp;
49
50	if (blksz % DEV_BSIZE != 0) {
51		errx(1, "cluster size should be multiple of %d",
52		    DEV_BSIZE);
53		/* Not reached */
54	}
55	if (compressBound(blksz) > MAXPHYS) {
56		errx(1, "cluster size is too large");
57		/* Not reached */
58	}
59	zp = mkuz_safe_zmalloc(sizeof(struct mkuz_zlib));
60	zp->oblen = compressBound(blksz);
61	zp->blksz = blksz;
62
63	return (void *)zp;
64}
65
66struct mkuz_blk *
67mkuz_zlib_compress(void *p, const struct mkuz_blk *iblk)
68{
69	uLongf destlen_z;
70	struct mkuz_blk *rval;
71	struct mkuz_zlib *zp;
72
73	zp = (struct mkuz_zlib *)p;
74
75	rval = mkuz_blk_ctor(zp->oblen);
76
77	destlen_z = rval->alen;
78	if (compress2(rval->data, &destlen_z, iblk->data, zp->blksz,
79	    Z_BEST_COMPRESSION) != Z_OK) {
80		errx(1, "can't compress data: compress2() "
81		    "failed");
82		/* Not reached */
83	}
84
85	rval->info.len = (uint32_t)destlen_z;
86	return (rval);
87}
88