1 /*-
2 * Copyright (c) 2004 Max Khon
3 * Copyright (c) 2014 Juniper Networks, Inc.
4 * Copyright (c) 2006-2016 Maxim Sobolev <sobomax@FreeBSD.org>
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include <sys/cdefs.h>
30 #include <sys/types.h>
31 #include <sys/systm.h>
32 #include <sys/malloc.h>
33
34 #include <contrib/zlib/zlib.h>
35 #include <dev/zlib/zcalloc.h>
36
37 #include <geom/uzip/g_uzip.h>
38 #include <geom/uzip/g_uzip_dapi.h>
39 #include <geom/uzip/g_uzip_zlib.h>
40
41 struct g_uzip_zlib {
42 uint32_t blksz;
43 struct g_uzip_dapi pub;
44 /* Zlib decoder structs */
45 z_stream zs;
46 };
47
48 static int g_uzip_zlib_rewind(struct g_uzip_dapi *, const char *);
49
50 static void
g_uzip_zlib_free(struct g_uzip_dapi * zpp)51 g_uzip_zlib_free(struct g_uzip_dapi *zpp)
52 {
53 struct g_uzip_zlib *zp;
54
55 zp = (struct g_uzip_zlib *)zpp->pvt;
56 inflateEnd(&zp->zs);
57 free(zp, M_GEOM_UZIP);
58 }
59
60 static int
g_uzip_zlib_decompress(struct g_uzip_dapi * zpp,const char * gp_name,void * ibp,size_t ilen,void * obp)61 g_uzip_zlib_decompress(struct g_uzip_dapi *zpp, const char *gp_name, void *ibp,
62 size_t ilen, void *obp)
63 {
64 int err;
65 struct g_uzip_zlib *zp;
66
67 zp = (struct g_uzip_zlib *)zpp->pvt;
68
69 zp->zs.next_in = ibp;
70 zp->zs.avail_in = ilen;
71 zp->zs.next_out = obp;
72 zp->zs.avail_out = zp->blksz;
73
74 err = (inflate(&zp->zs, Z_FINISH) != Z_STREAM_END) ? 1 : 0;
75 if (err != 0) {
76 printf("%s: UZIP(zlib) inflate() failed\n", gp_name);
77 }
78 return (err);
79 }
80
81 static int
g_uzip_zlib_rewind(struct g_uzip_dapi * zpp,const char * gp_name)82 g_uzip_zlib_rewind(struct g_uzip_dapi *zpp, const char *gp_name)
83 {
84 int err;
85 struct g_uzip_zlib *zp;
86
87 zp = (struct g_uzip_zlib *)zpp->pvt;
88
89 err = 0;
90 if (inflateReset(&zp->zs) != Z_OK) {
91 printf("%s: UZIP(zlib) decoder reset failed\n", gp_name);
92 err = 1;
93 }
94 return (err);
95 }
96
97 struct g_uzip_dapi *
g_uzip_zlib_ctor(uint32_t blksz)98 g_uzip_zlib_ctor(uint32_t blksz)
99 {
100 struct g_uzip_zlib *zp;
101
102 zp = malloc(sizeof(struct g_uzip_zlib), M_GEOM_UZIP, M_WAITOK | M_ZERO);
103 if (inflateInit(&zp->zs) != Z_OK) {
104 goto e1;
105 }
106 zp->blksz = blksz;
107 zp->pub.max_blen = compressBound(blksz);
108 zp->pub.decompress = &g_uzip_zlib_decompress;
109 zp->pub.free = &g_uzip_zlib_free;
110 zp->pub.rewind = &g_uzip_zlib_rewind;
111 zp->pub.pvt = (void *)zp;
112 return (&zp->pub);
113 e1:
114 free(zp, M_GEOM_UZIP);
115 return (NULL);
116 }
117