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 __FBSDID("$FreeBSD$");
31
32 #include <sys/types.h>
33 #include <sys/systm.h>
34 #include <sys/malloc.h>
35
36 #include <contrib/zlib/zlib.h>
37 #include <dev/zlib/zcalloc.h>
38
39 #include <geom/uzip/g_uzip.h>
40 #include <geom/uzip/g_uzip_dapi.h>
41 #include <geom/uzip/g_uzip_zlib.h>
42
43 struct g_uzip_zlib {
44 uint32_t blksz;
45 struct g_uzip_dapi pub;
46 /* Zlib decoder structs */
47 z_stream zs;
48 };
49
50 static int g_uzip_zlib_rewind(struct g_uzip_dapi *, const char *);
51
52 static void
53 g_uzip_zlib_free(struct g_uzip_dapi *zpp)
54 {
55 struct g_uzip_zlib *zp;
56
57 zp = (struct g_uzip_zlib *)zpp->pvt;
58 inflateEnd(&zp->zs);
59 free(zp, M_GEOM_UZIP);
60 }
61
62 static int
63 g_uzip_zlib_decompress(struct g_uzip_dapi *zpp, const char *gp_name, void *ibp,
64 size_t ilen, void *obp)
65 {
66 int err;
67 struct g_uzip_zlib *zp;
68
69 zp = (struct g_uzip_zlib *)zpp->pvt;
70
71 zp->zs.next_in = ibp;
72 zp->zs.avail_in = ilen;
73 zp->zs.next_out = obp;
74 zp->zs.avail_out = zp->blksz;
75
76 err = (inflate(&zp->zs, Z_FINISH) != Z_STREAM_END) ? 1 : 0;
77 if (err != 0) {
78 printf("%s: UZIP(zlib) inflate() failed\n", gp_name);
79 }
80 return (err);
81 }
82
83 static int
84 g_uzip_zlib_rewind(struct g_uzip_dapi *zpp, const char *gp_name)
85 {
86 int err;
87 struct g_uzip_zlib *zp;
88
89 zp = (struct g_uzip_zlib *)zpp->pvt;
90
91 err = 0;
92 if (inflateReset(&zp->zs) != Z_OK) {
93 printf("%s: UZIP(zlib) decoder reset failed\n", gp_name);
94 err = 1;
95 }
96 return (err);
97 }
98
99 struct g_uzip_dapi *
100 g_uzip_zlib_ctor(uint32_t blksz)
101 {
102 struct g_uzip_zlib *zp;
103
104 zp = malloc(sizeof(struct g_uzip_zlib), M_GEOM_UZIP, M_WAITOK | M_ZERO);
105 if (inflateInit(&zp->zs) != Z_OK) {
106 goto e1;
107 }
108 zp->blksz = blksz;
109 zp->pub.max_blen = compressBound(blksz);
110 zp->pub.decompress = &g_uzip_zlib_decompress;
111 zp->pub.free = &g_uzip_zlib_free;
112 zp->pub.rewind = &g_uzip_zlib_rewind;
113 zp->pub.pvt = (void *)zp;
114 return (&zp->pub);
115 e1:
116 free(zp, M_GEOM_UZIP);
117 return (NULL);
118 }
Cache object: e458fc81a1fbdb19941bdec71ba1fb17
|