1 | /* compress.c -- compress a memory buffer |
2 | * Copyright (C) 1995-2005, 2014, 2016 Jean-loup Gailly, Mark Adler |
3 | * For conditions of distribution and use, see copyright notice in zlib.h |
4 | */ |
5 | |
6 | /* @(#) $Id$ */ |
7 | |
8 | #define ZLIB_INTERNAL |
9 | #include "zlib.h" |
10 | |
11 | /* =========================================================================== |
12 | Compresses the source buffer into the destination buffer. The level |
13 | parameter has the same meaning as in deflateInit. sourceLen is the byte |
14 | length of the source buffer. Upon entry, destLen is the total size of the |
15 | destination buffer, which must be at least 0.1% larger than sourceLen plus |
16 | 12 bytes. Upon exit, destLen is the actual size of the compressed buffer. |
17 | |
18 | compress2 returns Z_OK if success, Z_MEM_ERROR if there was not enough |
19 | memory, Z_BUF_ERROR if there was not enough room in the output buffer, |
20 | Z_STREAM_ERROR if the level parameter is invalid. |
21 | */ |
22 | int ZEXPORT compress2 (dest, destLen, source, sourceLen, level) |
23 | Bytef *dest; |
24 | uLongf *destLen; |
25 | const Bytef *source; |
26 | uLong sourceLen; |
27 | int level; |
28 | { |
29 | z_stream stream; |
30 | int err; |
31 | const uInt max = (uInt)-1; |
32 | uLong left; |
33 | |
34 | left = *destLen; |
35 | *destLen = 0; |
36 | |
37 | stream.zalloc = (alloc_func)0; |
38 | stream.zfree = (free_func)0; |
39 | stream.opaque = (voidpf)0; |
40 | |
41 | err = deflateInit(&stream, level); |
42 | if (err != Z_OK) return err; |
43 | |
44 | stream.next_out = dest; |
45 | stream.avail_out = 0; |
46 | stream.next_in = (z_const Bytef *)source; |
47 | stream.avail_in = 0; |
48 | |
49 | do { |
50 | if (stream.avail_out == 0) { |
51 | stream.avail_out = left > (uLong)max ? max : (uInt)left; |
52 | left -= stream.avail_out; |
53 | } |
54 | if (stream.avail_in == 0) { |
55 | stream.avail_in = sourceLen > (uLong)max ? max : (uInt)sourceLen; |
56 | sourceLen -= stream.avail_in; |
57 | } |
58 | err = deflate(&stream, sourceLen ? Z_NO_FLUSH : Z_FINISH); |
59 | } while (err == Z_OK); |
60 | |
61 | *destLen = stream.total_out; |
62 | deflateEnd(&stream); |
63 | return err == Z_STREAM_END ? Z_OK : err; |
64 | } |
65 | |
66 | /* =========================================================================== |
67 | */ |
68 | int ZEXPORT compress (dest, destLen, source, sourceLen) |
69 | Bytef *dest; |
70 | uLongf *destLen; |
71 | const Bytef *source; |
72 | uLong sourceLen; |
73 | { |
74 | return compress2(dest, destLen, source, sourceLen, Z_DEFAULT_COMPRESSION); |
75 | } |
76 | |
77 | /* =========================================================================== |
78 | If the default memLevel or windowBits for deflateInit() is changed, then |
79 | this function needs to be updated. |
80 | */ |
81 | uLong ZEXPORT compressBound (sourceLen) |
82 | uLong sourceLen; |
83 | { |
84 | sourceLen = sourceLen + (sourceLen >> 12) + (sourceLen >> 14) + |
85 | (sourceLen >> 25) + 13; |
86 | /* FIXME(cavalcantii): usage of CRC32 Castagnoli as a hash function |
87 | * for the hash table of symbols used for compression has a side effect |
88 | * where for compression level [4, 5] it will increase the output buffer size |
89 | * by 0.1% (i.e. less than 1%) for a high entropy input (i.e. random data). |
90 | * To avoid a scenario where client code would fail, for safety we increase |
91 | * the expected output size by 0.8% (i.e. 8x more than the worst scenario). |
92 | * See: http://crbug.com/990489 |
93 | */ |
94 | sourceLen += sourceLen >> 7; // Equivalent to 1.0078125 |
95 | return sourceLen; |
96 | } |
97 | |