1 /* Copyright (C) 2010-2020 The RetroArch team
3 * ---------------------------------------------------------------------------------------
4 * The following license statement only applies to this file (trans_stream_zlib.c).
5 * ---------------------------------------------------------------------------------------
7 * Permission is hereby granted, free of charge,
8 * to any person obtaining a copy of this software and associated documentation files (the "Software"),
9 * to deal in the Software without restriction, including without limitation the rights to
10 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software,
11 * and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
13 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
16 * INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
18 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
19 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
27 #include <string/stdstring.h>
28 #include <streams/trans_stream.h>
30 struct zlib_trans_stream
33 int ex; /* window_bits or level */
37 static void *zlib_deflate_stream_new(void)
39 struct zlib_trans_stream *ret = (struct zlib_trans_stream*)
46 ret->z.next_in = NULL;
49 ret->z.next_out = NULL;
66 static void *zlib_inflate_stream_new(void)
68 struct zlib_trans_stream *ret = (struct zlib_trans_stream*)
75 ret->z.next_in = NULL;
78 ret->z.next_out = NULL;
95 static void zlib_deflate_stream_free(void *data)
97 struct zlib_trans_stream *z = (struct zlib_trans_stream *) data;
105 static void zlib_inflate_stream_free(void *data)
107 struct zlib_trans_stream *z = (struct zlib_trans_stream *) data;
116 static bool zlib_deflate_define(void *data, const char *prop, uint32_t val)
118 struct zlib_trans_stream *z = (struct zlib_trans_stream *) data;
119 if (string_is_equal(prop, "level"))
128 static bool zlib_inflate_define(void *data, const char *prop, uint32_t val)
130 struct zlib_trans_stream *z = (struct zlib_trans_stream *) data;
131 if (string_is_equal(prop, "window_bits"))
140 static void zlib_deflate_set_in(void *data, const uint8_t *in, uint32_t in_size)
142 struct zlib_trans_stream *z = (struct zlib_trans_stream *) data;
147 z->z.next_in = (uint8_t *) in;
148 z->z.avail_in = in_size;
152 deflateInit(&z->z, z->ex);
157 static void zlib_inflate_set_in(void *data, const uint8_t *in, uint32_t in_size)
159 struct zlib_trans_stream *z = (struct zlib_trans_stream *) data;
164 z->z.next_in = (uint8_t *) in;
165 z->z.avail_in = in_size;
168 inflateInit2(&z->z, z->ex);
173 static void zlib_set_out(void *data, uint8_t *out, uint32_t out_size)
175 struct zlib_trans_stream *z = (struct zlib_trans_stream *) data;
181 z->z.avail_out = out_size;
184 static bool zlib_deflate_trans(
185 void *data, bool flush,
186 uint32_t *rd, uint32_t *wn,
187 enum trans_stream_error *error)
191 uint32_t pre_avail_in = 0;
192 uint32_t pre_avail_out = 0;
193 struct zlib_trans_stream *zt = (struct zlib_trans_stream *) data;
194 z_stream *z = &zt->z;
198 deflateInit(z, zt->ex);
202 pre_avail_in = z->avail_in;
203 pre_avail_out = z->avail_out;
204 zret = deflate(z, flush ? Z_FINISH : Z_NO_FLUSH);
209 *error = TRANS_STREAM_ERROR_AGAIN;
211 else if (zret == Z_STREAM_END)
214 *error = TRANS_STREAM_ERROR_NONE;
219 *error = TRANS_STREAM_ERROR_OTHER;
224 if (z->avail_out == 0)
226 /* Filled buffer, maybe an error */
227 if (z->avail_in != 0)
231 *error = TRANS_STREAM_ERROR_BUFFER_FULL;
235 *rd = pre_avail_in - z->avail_in;
236 *wn = pre_avail_out - z->avail_out;
238 if (flush && zret == Z_STREAM_END)
247 static bool zlib_inflate_trans(
248 void *data, bool flush,
249 uint32_t *rd, uint32_t *wn,
250 enum trans_stream_error *error)
254 uint32_t pre_avail_in = 0;
255 uint32_t pre_avail_out = 0;
256 struct zlib_trans_stream *zt = (struct zlib_trans_stream *) data;
257 z_stream *z = &zt->z;
261 inflateInit2(z, zt->ex);
265 pre_avail_in = z->avail_in;
266 pre_avail_out = z->avail_out;
267 zret = inflate(z, flush ? Z_FINISH : Z_NO_FLUSH);
272 *error = TRANS_STREAM_ERROR_AGAIN;
274 else if (zret == Z_STREAM_END)
277 *error = TRANS_STREAM_ERROR_NONE;
282 *error = TRANS_STREAM_ERROR_OTHER;
287 if (z->avail_out == 0)
289 /* Filled buffer, maybe an error */
290 if (z->avail_in != 0)
294 *error = TRANS_STREAM_ERROR_BUFFER_FULL;
298 *rd = pre_avail_in - z->avail_in;
299 *wn = pre_avail_out - z->avail_out;
301 if (flush && zret == Z_STREAM_END)
310 const struct trans_stream_backend zlib_deflate_backend = {
312 &zlib_inflate_backend,
313 zlib_deflate_stream_new,
314 zlib_deflate_stream_free,
321 const struct trans_stream_backend zlib_inflate_backend = {
323 &zlib_deflate_backend,
324 zlib_inflate_stream_new,
325 zlib_inflate_stream_free,