1 | /*
|
---|
2 | * Copyright 1998-2016 The OpenSSL Project Authors. All Rights Reserved.
|
---|
3 | *
|
---|
4 | * Licensed under the OpenSSL license (the "License"). You may not use
|
---|
5 | * this file except in compliance with the License. You can obtain a copy
|
---|
6 | * in the file LICENSE in the source distribution or at
|
---|
7 | * https://www.openssl.org/source/license.html
|
---|
8 | */
|
---|
9 |
|
---|
10 | #include <stdio.h>
|
---|
11 | #include <stdlib.h>
|
---|
12 | #include <string.h>
|
---|
13 | #include <openssl/objects.h>
|
---|
14 | #include "internal/comp.h"
|
---|
15 | #include <openssl/err.h>
|
---|
16 | #include "internal/cryptlib_int.h"
|
---|
17 | #include "internal/bio.h"
|
---|
18 | #include "comp_lcl.h"
|
---|
19 |
|
---|
20 | COMP_METHOD *COMP_zlib(void);
|
---|
21 |
|
---|
22 | static COMP_METHOD zlib_method_nozlib = {
|
---|
23 | NID_undef,
|
---|
24 | "(undef)",
|
---|
25 | NULL,
|
---|
26 | NULL,
|
---|
27 | NULL,
|
---|
28 | NULL,
|
---|
29 | };
|
---|
30 |
|
---|
31 | #ifndef ZLIB
|
---|
32 | # undef ZLIB_SHARED
|
---|
33 | #else
|
---|
34 |
|
---|
35 | # include <zlib.h>
|
---|
36 |
|
---|
37 | static int zlib_stateful_init(COMP_CTX *ctx);
|
---|
38 | static void zlib_stateful_finish(COMP_CTX *ctx);
|
---|
39 | static int zlib_stateful_compress_block(COMP_CTX *ctx, unsigned char *out,
|
---|
40 | unsigned int olen, unsigned char *in,
|
---|
41 | unsigned int ilen);
|
---|
42 | static int zlib_stateful_expand_block(COMP_CTX *ctx, unsigned char *out,
|
---|
43 | unsigned int olen, unsigned char *in,
|
---|
44 | unsigned int ilen);
|
---|
45 |
|
---|
46 | /* memory allocations functions for zlib initialisation */
|
---|
47 | static void *zlib_zalloc(void *opaque, unsigned int no, unsigned int size)
|
---|
48 | {
|
---|
49 | void *p;
|
---|
50 |
|
---|
51 | p = OPENSSL_zalloc(no * size);
|
---|
52 | return p;
|
---|
53 | }
|
---|
54 |
|
---|
55 | static void zlib_zfree(void *opaque, void *address)
|
---|
56 | {
|
---|
57 | OPENSSL_free(address);
|
---|
58 | }
|
---|
59 |
|
---|
60 |
|
---|
61 | static COMP_METHOD zlib_stateful_method = {
|
---|
62 | NID_zlib_compression,
|
---|
63 | LN_zlib_compression,
|
---|
64 | zlib_stateful_init,
|
---|
65 | zlib_stateful_finish,
|
---|
66 | zlib_stateful_compress_block,
|
---|
67 | zlib_stateful_expand_block
|
---|
68 | };
|
---|
69 |
|
---|
70 | /*
|
---|
71 | * When OpenSSL is built on Windows, we do not want to require that
|
---|
72 | * the ZLIB.DLL be available in order for the OpenSSL DLLs to
|
---|
73 | * work. Therefore, all ZLIB routines are loaded at run time
|
---|
74 | * and we do not link to a .LIB file when ZLIB_SHARED is set.
|
---|
75 | */
|
---|
76 | # if defined(OPENSSL_SYS_WINDOWS) || defined(OPENSSL_SYS_WIN32)
|
---|
77 | # include <windows.h>
|
---|
78 | # endif /* !(OPENSSL_SYS_WINDOWS ||
|
---|
79 | * OPENSSL_SYS_WIN32) */
|
---|
80 |
|
---|
81 | # ifdef ZLIB_SHARED
|
---|
82 | # include "internal/dso.h"
|
---|
83 |
|
---|
84 | /* Function pointers */
|
---|
85 | typedef int (*compress_ft) (Bytef *dest, uLongf * destLen,
|
---|
86 | const Bytef *source, uLong sourceLen);
|
---|
87 | typedef int (*inflateEnd_ft) (z_streamp strm);
|
---|
88 | typedef int (*inflate_ft) (z_streamp strm, int flush);
|
---|
89 | typedef int (*inflateInit__ft) (z_streamp strm,
|
---|
90 | const char *version, int stream_size);
|
---|
91 | typedef int (*deflateEnd_ft) (z_streamp strm);
|
---|
92 | typedef int (*deflate_ft) (z_streamp strm, int flush);
|
---|
93 | typedef int (*deflateInit__ft) (z_streamp strm, int level,
|
---|
94 | const char *version, int stream_size);
|
---|
95 | typedef const char *(*zError__ft) (int err);
|
---|
96 | static compress_ft p_compress = NULL;
|
---|
97 | static inflateEnd_ft p_inflateEnd = NULL;
|
---|
98 | static inflate_ft p_inflate = NULL;
|
---|
99 | static inflateInit__ft p_inflateInit_ = NULL;
|
---|
100 | static deflateEnd_ft p_deflateEnd = NULL;
|
---|
101 | static deflate_ft p_deflate = NULL;
|
---|
102 | static deflateInit__ft p_deflateInit_ = NULL;
|
---|
103 | static zError__ft p_zError = NULL;
|
---|
104 |
|
---|
105 | static int zlib_loaded = 0; /* only attempt to init func pts once */
|
---|
106 | static DSO *zlib_dso = NULL;
|
---|
107 |
|
---|
108 | # define compress p_compress
|
---|
109 | # define inflateEnd p_inflateEnd
|
---|
110 | # define inflate p_inflate
|
---|
111 | # define inflateInit_ p_inflateInit_
|
---|
112 | # define deflateEnd p_deflateEnd
|
---|
113 | # define deflate p_deflate
|
---|
114 | # define deflateInit_ p_deflateInit_
|
---|
115 | # define zError p_zError
|
---|
116 | # endif /* ZLIB_SHARED */
|
---|
117 |
|
---|
118 | struct zlib_state {
|
---|
119 | z_stream istream;
|
---|
120 | z_stream ostream;
|
---|
121 | };
|
---|
122 |
|
---|
123 | static int zlib_stateful_init(COMP_CTX *ctx)
|
---|
124 | {
|
---|
125 | int err;
|
---|
126 | struct zlib_state *state = OPENSSL_zalloc(sizeof(*state));
|
---|
127 |
|
---|
128 | if (state == NULL)
|
---|
129 | goto err;
|
---|
130 |
|
---|
131 | state->istream.zalloc = zlib_zalloc;
|
---|
132 | state->istream.zfree = zlib_zfree;
|
---|
133 | state->istream.opaque = Z_NULL;
|
---|
134 | state->istream.next_in = Z_NULL;
|
---|
135 | state->istream.next_out = Z_NULL;
|
---|
136 | err = inflateInit_(&state->istream, ZLIB_VERSION, sizeof(z_stream));
|
---|
137 | if (err != Z_OK)
|
---|
138 | goto err;
|
---|
139 |
|
---|
140 | state->ostream.zalloc = zlib_zalloc;
|
---|
141 | state->ostream.zfree = zlib_zfree;
|
---|
142 | state->ostream.opaque = Z_NULL;
|
---|
143 | state->ostream.next_in = Z_NULL;
|
---|
144 | state->ostream.next_out = Z_NULL;
|
---|
145 | err = deflateInit_(&state->ostream, Z_DEFAULT_COMPRESSION,
|
---|
146 | ZLIB_VERSION, sizeof(z_stream));
|
---|
147 | if (err != Z_OK)
|
---|
148 | goto err;
|
---|
149 |
|
---|
150 | ctx->data = state;
|
---|
151 | return 1;
|
---|
152 | err:
|
---|
153 | OPENSSL_free(state);
|
---|
154 | return 0;
|
---|
155 | }
|
---|
156 |
|
---|
157 | static void zlib_stateful_finish(COMP_CTX *ctx)
|
---|
158 | {
|
---|
159 | struct zlib_state *state = ctx->data;
|
---|
160 | inflateEnd(&state->istream);
|
---|
161 | deflateEnd(&state->ostream);
|
---|
162 | OPENSSL_free(state);
|
---|
163 | }
|
---|
164 |
|
---|
165 | static int zlib_stateful_compress_block(COMP_CTX *ctx, unsigned char *out,
|
---|
166 | unsigned int olen, unsigned char *in,
|
---|
167 | unsigned int ilen)
|
---|
168 | {
|
---|
169 | int err = Z_OK;
|
---|
170 | struct zlib_state *state = ctx->data;
|
---|
171 |
|
---|
172 | if (state == NULL)
|
---|
173 | return -1;
|
---|
174 |
|
---|
175 | state->ostream.next_in = in;
|
---|
176 | state->ostream.avail_in = ilen;
|
---|
177 | state->ostream.next_out = out;
|
---|
178 | state->ostream.avail_out = olen;
|
---|
179 | if (ilen > 0)
|
---|
180 | err = deflate(&state->ostream, Z_SYNC_FLUSH);
|
---|
181 | if (err != Z_OK)
|
---|
182 | return -1;
|
---|
183 | return olen - state->ostream.avail_out;
|
---|
184 | }
|
---|
185 |
|
---|
186 | static int zlib_stateful_expand_block(COMP_CTX *ctx, unsigned char *out,
|
---|
187 | unsigned int olen, unsigned char *in,
|
---|
188 | unsigned int ilen)
|
---|
189 | {
|
---|
190 | int err = Z_OK;
|
---|
191 | struct zlib_state *state = ctx->data;
|
---|
192 |
|
---|
193 | if (state == NULL)
|
---|
194 | return 0;
|
---|
195 |
|
---|
196 | state->istream.next_in = in;
|
---|
197 | state->istream.avail_in = ilen;
|
---|
198 | state->istream.next_out = out;
|
---|
199 | state->istream.avail_out = olen;
|
---|
200 | if (ilen > 0)
|
---|
201 | err = inflate(&state->istream, Z_SYNC_FLUSH);
|
---|
202 | if (err != Z_OK)
|
---|
203 | return -1;
|
---|
204 | return olen - state->istream.avail_out;
|
---|
205 | }
|
---|
206 |
|
---|
207 | #endif
|
---|
208 |
|
---|
209 | COMP_METHOD *COMP_zlib(void)
|
---|
210 | {
|
---|
211 | COMP_METHOD *meth = &zlib_method_nozlib;
|
---|
212 |
|
---|
213 | #ifdef ZLIB_SHARED
|
---|
214 | /* LIBZ may be externally defined, and we should respect that value */
|
---|
215 | # ifndef LIBZ
|
---|
216 | # if defined(OPENSSL_SYS_WINDOWS) || defined(OPENSSL_SYS_WIN32)
|
---|
217 | # define LIBZ "ZLIB1"
|
---|
218 | # elif defined(OPENSSL_SYS_VMS)
|
---|
219 | # define LIBZ "LIBZ"
|
---|
220 | # else
|
---|
221 | # define LIBZ "z"
|
---|
222 | # endif
|
---|
223 | # endif
|
---|
224 |
|
---|
225 | if (!zlib_loaded) {
|
---|
226 | zlib_dso = DSO_load(NULL, LIBZ, NULL, 0);
|
---|
227 | if (zlib_dso != NULL) {
|
---|
228 | p_compress = (compress_ft) DSO_bind_func(zlib_dso, "compress");
|
---|
229 | p_inflateEnd
|
---|
230 | = (inflateEnd_ft) DSO_bind_func(zlib_dso, "inflateEnd");
|
---|
231 | p_inflate = (inflate_ft) DSO_bind_func(zlib_dso, "inflate");
|
---|
232 | p_inflateInit_
|
---|
233 | = (inflateInit__ft) DSO_bind_func(zlib_dso, "inflateInit_");
|
---|
234 | p_deflateEnd
|
---|
235 | = (deflateEnd_ft) DSO_bind_func(zlib_dso, "deflateEnd");
|
---|
236 | p_deflate = (deflate_ft) DSO_bind_func(zlib_dso, "deflate");
|
---|
237 | p_deflateInit_
|
---|
238 | = (deflateInit__ft) DSO_bind_func(zlib_dso, "deflateInit_");
|
---|
239 | p_zError = (zError__ft) DSO_bind_func(zlib_dso, "zError");
|
---|
240 |
|
---|
241 | if (p_compress && p_inflateEnd && p_inflate
|
---|
242 | && p_inflateInit_ && p_deflateEnd
|
---|
243 | && p_deflate && p_deflateInit_ && p_zError)
|
---|
244 | zlib_loaded++;
|
---|
245 |
|
---|
246 | if (!OPENSSL_init_crypto(OPENSSL_INIT_ZLIB, NULL)) {
|
---|
247 | comp_zlib_cleanup_int();
|
---|
248 | return meth;
|
---|
249 | }
|
---|
250 | if (zlib_loaded)
|
---|
251 | meth = &zlib_stateful_method;
|
---|
252 | }
|
---|
253 | }
|
---|
254 | #endif
|
---|
255 | #if defined(ZLIB)
|
---|
256 | meth = &zlib_stateful_method;
|
---|
257 | #endif
|
---|
258 |
|
---|
259 | return (meth);
|
---|
260 | }
|
---|
261 |
|
---|
262 | void comp_zlib_cleanup_int(void)
|
---|
263 | {
|
---|
264 | #ifdef ZLIB_SHARED
|
---|
265 | if (zlib_dso != NULL)
|
---|
266 | DSO_free(zlib_dso);
|
---|
267 | zlib_dso = NULL;
|
---|
268 | #endif
|
---|
269 | }
|
---|
270 |
|
---|
271 | #ifdef ZLIB
|
---|
272 |
|
---|
273 | /* Zlib based compression/decompression filter BIO */
|
---|
274 |
|
---|
275 | typedef struct {
|
---|
276 | unsigned char *ibuf; /* Input buffer */
|
---|
277 | int ibufsize; /* Buffer size */
|
---|
278 | z_stream zin; /* Input decompress context */
|
---|
279 | unsigned char *obuf; /* Output buffer */
|
---|
280 | int obufsize; /* Output buffer size */
|
---|
281 | unsigned char *optr; /* Position in output buffer */
|
---|
282 | int ocount; /* Amount of data in output buffer */
|
---|
283 | int odone; /* deflate EOF */
|
---|
284 | int comp_level; /* Compression level to use */
|
---|
285 | z_stream zout; /* Output compression context */
|
---|
286 | } BIO_ZLIB_CTX;
|
---|
287 |
|
---|
288 | # define ZLIB_DEFAULT_BUFSIZE 1024
|
---|
289 |
|
---|
290 | static int bio_zlib_new(BIO *bi);
|
---|
291 | static int bio_zlib_free(BIO *bi);
|
---|
292 | static int bio_zlib_read(BIO *b, char *out, int outl);
|
---|
293 | static int bio_zlib_write(BIO *b, const char *in, int inl);
|
---|
294 | static long bio_zlib_ctrl(BIO *b, int cmd, long num, void *ptr);
|
---|
295 | static long bio_zlib_callback_ctrl(BIO *b, int cmd, bio_info_cb *fp);
|
---|
296 |
|
---|
297 | static const BIO_METHOD bio_meth_zlib = {
|
---|
298 | BIO_TYPE_COMP,
|
---|
299 | "zlib",
|
---|
300 | bio_zlib_write,
|
---|
301 | bio_zlib_read,
|
---|
302 | NULL,
|
---|
303 | NULL,
|
---|
304 | bio_zlib_ctrl,
|
---|
305 | bio_zlib_new,
|
---|
306 | bio_zlib_free,
|
---|
307 | bio_zlib_callback_ctrl
|
---|
308 | };
|
---|
309 |
|
---|
310 | const BIO_METHOD *BIO_f_zlib(void)
|
---|
311 | {
|
---|
312 | return &bio_meth_zlib;
|
---|
313 | }
|
---|
314 |
|
---|
315 | static int bio_zlib_new(BIO *bi)
|
---|
316 | {
|
---|
317 | BIO_ZLIB_CTX *ctx;
|
---|
318 | # ifdef ZLIB_SHARED
|
---|
319 | (void)COMP_zlib();
|
---|
320 | if (!zlib_loaded) {
|
---|
321 | COMPerr(COMP_F_BIO_ZLIB_NEW, COMP_R_ZLIB_NOT_SUPPORTED);
|
---|
322 | return 0;
|
---|
323 | }
|
---|
324 | # endif
|
---|
325 | ctx = OPENSSL_zalloc(sizeof(*ctx));
|
---|
326 | if (ctx == NULL) {
|
---|
327 | COMPerr(COMP_F_BIO_ZLIB_NEW, ERR_R_MALLOC_FAILURE);
|
---|
328 | return 0;
|
---|
329 | }
|
---|
330 | ctx->ibufsize = ZLIB_DEFAULT_BUFSIZE;
|
---|
331 | ctx->obufsize = ZLIB_DEFAULT_BUFSIZE;
|
---|
332 | ctx->zin.zalloc = Z_NULL;
|
---|
333 | ctx->zin.zfree = Z_NULL;
|
---|
334 | ctx->zout.zalloc = Z_NULL;
|
---|
335 | ctx->zout.zfree = Z_NULL;
|
---|
336 | ctx->comp_level = Z_DEFAULT_COMPRESSION;
|
---|
337 | BIO_set_init(bi, 1);
|
---|
338 | BIO_set_data(bi, ctx);
|
---|
339 |
|
---|
340 | return 1;
|
---|
341 | }
|
---|
342 |
|
---|
343 | static int bio_zlib_free(BIO *bi)
|
---|
344 | {
|
---|
345 | BIO_ZLIB_CTX *ctx;
|
---|
346 | if (!bi)
|
---|
347 | return 0;
|
---|
348 | ctx = BIO_get_data(bi);
|
---|
349 | if (ctx->ibuf) {
|
---|
350 | /* Destroy decompress context */
|
---|
351 | inflateEnd(&ctx->zin);
|
---|
352 | OPENSSL_free(ctx->ibuf);
|
---|
353 | }
|
---|
354 | if (ctx->obuf) {
|
---|
355 | /* Destroy compress context */
|
---|
356 | deflateEnd(&ctx->zout);
|
---|
357 | OPENSSL_free(ctx->obuf);
|
---|
358 | }
|
---|
359 | OPENSSL_free(ctx);
|
---|
360 | BIO_set_data(bi, NULL);
|
---|
361 | BIO_set_init(bi, 0);
|
---|
362 |
|
---|
363 | return 1;
|
---|
364 | }
|
---|
365 |
|
---|
366 | static int bio_zlib_read(BIO *b, char *out, int outl)
|
---|
367 | {
|
---|
368 | BIO_ZLIB_CTX *ctx;
|
---|
369 | int ret;
|
---|
370 | z_stream *zin;
|
---|
371 | BIO *next = BIO_next(b);
|
---|
372 |
|
---|
373 | if (!out || !outl)
|
---|
374 | return 0;
|
---|
375 | ctx = BIO_get_data(b);
|
---|
376 | zin = &ctx->zin;
|
---|
377 | BIO_clear_retry_flags(b);
|
---|
378 | if (!ctx->ibuf) {
|
---|
379 | ctx->ibuf = OPENSSL_malloc(ctx->ibufsize);
|
---|
380 | if (ctx->ibuf == NULL) {
|
---|
381 | COMPerr(COMP_F_BIO_ZLIB_READ, ERR_R_MALLOC_FAILURE);
|
---|
382 | return 0;
|
---|
383 | }
|
---|
384 | inflateInit(zin);
|
---|
385 | zin->next_in = ctx->ibuf;
|
---|
386 | zin->avail_in = 0;
|
---|
387 | }
|
---|
388 |
|
---|
389 | /* Copy output data directly to supplied buffer */
|
---|
390 | zin->next_out = (unsigned char *)out;
|
---|
391 | zin->avail_out = (unsigned int)outl;
|
---|
392 | for (;;) {
|
---|
393 | /* Decompress while data available */
|
---|
394 | while (zin->avail_in) {
|
---|
395 | ret = inflate(zin, 0);
|
---|
396 | if ((ret != Z_OK) && (ret != Z_STREAM_END)) {
|
---|
397 | COMPerr(COMP_F_BIO_ZLIB_READ, COMP_R_ZLIB_INFLATE_ERROR);
|
---|
398 | ERR_add_error_data(2, "zlib error:", zError(ret));
|
---|
399 | return 0;
|
---|
400 | }
|
---|
401 | /* If EOF or we've read everything then return */
|
---|
402 | if ((ret == Z_STREAM_END) || !zin->avail_out)
|
---|
403 | return outl - zin->avail_out;
|
---|
404 | }
|
---|
405 |
|
---|
406 | /*
|
---|
407 | * No data in input buffer try to read some in, if an error then
|
---|
408 | * return the total data read.
|
---|
409 | */
|
---|
410 | ret = BIO_read(next, ctx->ibuf, ctx->ibufsize);
|
---|
411 | if (ret <= 0) {
|
---|
412 | /* Total data read */
|
---|
413 | int tot = outl - zin->avail_out;
|
---|
414 | BIO_copy_next_retry(b);
|
---|
415 | if (ret < 0)
|
---|
416 | return (tot > 0) ? tot : ret;
|
---|
417 | return tot;
|
---|
418 | }
|
---|
419 | zin->avail_in = ret;
|
---|
420 | zin->next_in = ctx->ibuf;
|
---|
421 | }
|
---|
422 | }
|
---|
423 |
|
---|
424 | static int bio_zlib_write(BIO *b, const char *in, int inl)
|
---|
425 | {
|
---|
426 | BIO_ZLIB_CTX *ctx;
|
---|
427 | int ret;
|
---|
428 | z_stream *zout;
|
---|
429 | BIO *next = BIO_next(b);
|
---|
430 |
|
---|
431 | if (!in || !inl)
|
---|
432 | return 0;
|
---|
433 | ctx = BIO_get_data(b);
|
---|
434 | if (ctx->odone)
|
---|
435 | return 0;
|
---|
436 | zout = &ctx->zout;
|
---|
437 | BIO_clear_retry_flags(b);
|
---|
438 | if (!ctx->obuf) {
|
---|
439 | ctx->obuf = OPENSSL_malloc(ctx->obufsize);
|
---|
440 | /* Need error here */
|
---|
441 | if (ctx->obuf == NULL) {
|
---|
442 | COMPerr(COMP_F_BIO_ZLIB_WRITE, ERR_R_MALLOC_FAILURE);
|
---|
443 | return 0;
|
---|
444 | }
|
---|
445 | ctx->optr = ctx->obuf;
|
---|
446 | ctx->ocount = 0;
|
---|
447 | deflateInit(zout, ctx->comp_level);
|
---|
448 | zout->next_out = ctx->obuf;
|
---|
449 | zout->avail_out = ctx->obufsize;
|
---|
450 | }
|
---|
451 | /* Obtain input data directly from supplied buffer */
|
---|
452 | zout->next_in = (void *)in;
|
---|
453 | zout->avail_in = inl;
|
---|
454 | for (;;) {
|
---|
455 | /* If data in output buffer write it first */
|
---|
456 | while (ctx->ocount) {
|
---|
457 | ret = BIO_write(next, ctx->optr, ctx->ocount);
|
---|
458 | if (ret <= 0) {
|
---|
459 | /* Total data written */
|
---|
460 | int tot = inl - zout->avail_in;
|
---|
461 | BIO_copy_next_retry(b);
|
---|
462 | if (ret < 0)
|
---|
463 | return (tot > 0) ? tot : ret;
|
---|
464 | return tot;
|
---|
465 | }
|
---|
466 | ctx->optr += ret;
|
---|
467 | ctx->ocount -= ret;
|
---|
468 | }
|
---|
469 |
|
---|
470 | /* Have we consumed all supplied data? */
|
---|
471 | if (!zout->avail_in)
|
---|
472 | return inl;
|
---|
473 |
|
---|
474 | /* Compress some more */
|
---|
475 |
|
---|
476 | /* Reset buffer */
|
---|
477 | ctx->optr = ctx->obuf;
|
---|
478 | zout->next_out = ctx->obuf;
|
---|
479 | zout->avail_out = ctx->obufsize;
|
---|
480 | /* Compress some more */
|
---|
481 | ret = deflate(zout, 0);
|
---|
482 | if (ret != Z_OK) {
|
---|
483 | COMPerr(COMP_F_BIO_ZLIB_WRITE, COMP_R_ZLIB_DEFLATE_ERROR);
|
---|
484 | ERR_add_error_data(2, "zlib error:", zError(ret));
|
---|
485 | return 0;
|
---|
486 | }
|
---|
487 | ctx->ocount = ctx->obufsize - zout->avail_out;
|
---|
488 | }
|
---|
489 | }
|
---|
490 |
|
---|
491 | static int bio_zlib_flush(BIO *b)
|
---|
492 | {
|
---|
493 | BIO_ZLIB_CTX *ctx;
|
---|
494 | int ret;
|
---|
495 | z_stream *zout;
|
---|
496 | BIO *next = BIO_next(b);
|
---|
497 |
|
---|
498 | ctx = BIO_get_data(b);
|
---|
499 | /* If no data written or already flush show success */
|
---|
500 | if (!ctx->obuf || (ctx->odone && !ctx->ocount))
|
---|
501 | return 1;
|
---|
502 | zout = &ctx->zout;
|
---|
503 | BIO_clear_retry_flags(b);
|
---|
504 | /* No more input data */
|
---|
505 | zout->next_in = NULL;
|
---|
506 | zout->avail_in = 0;
|
---|
507 | for (;;) {
|
---|
508 | /* If data in output buffer write it first */
|
---|
509 | while (ctx->ocount) {
|
---|
510 | ret = BIO_write(next, ctx->optr, ctx->ocount);
|
---|
511 | if (ret <= 0) {
|
---|
512 | BIO_copy_next_retry(b);
|
---|
513 | return ret;
|
---|
514 | }
|
---|
515 | ctx->optr += ret;
|
---|
516 | ctx->ocount -= ret;
|
---|
517 | }
|
---|
518 | if (ctx->odone)
|
---|
519 | return 1;
|
---|
520 |
|
---|
521 | /* Compress some more */
|
---|
522 |
|
---|
523 | /* Reset buffer */
|
---|
524 | ctx->optr = ctx->obuf;
|
---|
525 | zout->next_out = ctx->obuf;
|
---|
526 | zout->avail_out = ctx->obufsize;
|
---|
527 | /* Compress some more */
|
---|
528 | ret = deflate(zout, Z_FINISH);
|
---|
529 | if (ret == Z_STREAM_END)
|
---|
530 | ctx->odone = 1;
|
---|
531 | else if (ret != Z_OK) {
|
---|
532 | COMPerr(COMP_F_BIO_ZLIB_FLUSH, COMP_R_ZLIB_DEFLATE_ERROR);
|
---|
533 | ERR_add_error_data(2, "zlib error:", zError(ret));
|
---|
534 | return 0;
|
---|
535 | }
|
---|
536 | ctx->ocount = ctx->obufsize - zout->avail_out;
|
---|
537 | }
|
---|
538 | }
|
---|
539 |
|
---|
540 | static long bio_zlib_ctrl(BIO *b, int cmd, long num, void *ptr)
|
---|
541 | {
|
---|
542 | BIO_ZLIB_CTX *ctx;
|
---|
543 | int ret, *ip;
|
---|
544 | int ibs, obs;
|
---|
545 | BIO *next = BIO_next(b);
|
---|
546 |
|
---|
547 | if (next == NULL)
|
---|
548 | return 0;
|
---|
549 | ctx = BIO_get_data(b);
|
---|
550 | switch (cmd) {
|
---|
551 |
|
---|
552 | case BIO_CTRL_RESET:
|
---|
553 | ctx->ocount = 0;
|
---|
554 | ctx->odone = 0;
|
---|
555 | ret = 1;
|
---|
556 | break;
|
---|
557 |
|
---|
558 | case BIO_CTRL_FLUSH:
|
---|
559 | ret = bio_zlib_flush(b);
|
---|
560 | if (ret > 0)
|
---|
561 | ret = BIO_flush(next);
|
---|
562 | break;
|
---|
563 |
|
---|
564 | case BIO_C_SET_BUFF_SIZE:
|
---|
565 | ibs = -1;
|
---|
566 | obs = -1;
|
---|
567 | if (ptr != NULL) {
|
---|
568 | ip = ptr;
|
---|
569 | if (*ip == 0)
|
---|
570 | ibs = (int)num;
|
---|
571 | else
|
---|
572 | obs = (int)num;
|
---|
573 | } else {
|
---|
574 | ibs = (int)num;
|
---|
575 | obs = ibs;
|
---|
576 | }
|
---|
577 |
|
---|
578 | if (ibs != -1) {
|
---|
579 | OPENSSL_free(ctx->ibuf);
|
---|
580 | ctx->ibuf = NULL;
|
---|
581 | ctx->ibufsize = ibs;
|
---|
582 | }
|
---|
583 |
|
---|
584 | if (obs != -1) {
|
---|
585 | OPENSSL_free(ctx->obuf);
|
---|
586 | ctx->obuf = NULL;
|
---|
587 | ctx->obufsize = obs;
|
---|
588 | }
|
---|
589 | ret = 1;
|
---|
590 | break;
|
---|
591 |
|
---|
592 | case BIO_C_DO_STATE_MACHINE:
|
---|
593 | BIO_clear_retry_flags(b);
|
---|
594 | ret = BIO_ctrl(next, cmd, num, ptr);
|
---|
595 | BIO_copy_next_retry(b);
|
---|
596 | break;
|
---|
597 |
|
---|
598 | default:
|
---|
599 | ret = BIO_ctrl(next, cmd, num, ptr);
|
---|
600 | break;
|
---|
601 |
|
---|
602 | }
|
---|
603 |
|
---|
604 | return ret;
|
---|
605 | }
|
---|
606 |
|
---|
607 | static long bio_zlib_callback_ctrl(BIO *b, int cmd, bio_info_cb *fp)
|
---|
608 | {
|
---|
609 | BIO *next = BIO_next(b);
|
---|
610 | if (next == NULL)
|
---|
611 | return 0;
|
---|
612 | return BIO_callback_ctrl(next, cmd, fp);
|
---|
613 | }
|
---|
614 |
|
---|
615 | #endif
|
---|