1*4882a593Smuzhiyun #include <config.h>
2*4882a593Smuzhiyun #include <common.h>
3*4882a593Smuzhiyun #include <watchdog.h>
4*4882a593Smuzhiyun
5*4882a593Smuzhiyun /*
6*4882a593Smuzhiyun * This file is a modified version of bzlib.c from the bzip2-1.0.2
7*4882a593Smuzhiyun * distribution which can be found at http://sources.redhat.com/bzip2/
8*4882a593Smuzhiyun */
9*4882a593Smuzhiyun
10*4882a593Smuzhiyun /*-------------------------------------------------------------*/
11*4882a593Smuzhiyun /*--- Library top-level functions. ---*/
12*4882a593Smuzhiyun /*--- bzlib.c ---*/
13*4882a593Smuzhiyun /*-------------------------------------------------------------*/
14*4882a593Smuzhiyun
15*4882a593Smuzhiyun /*--
16*4882a593Smuzhiyun This file is a part of bzip2 and/or libbzip2, a program and
17*4882a593Smuzhiyun library for lossless, block-sorting data compression.
18*4882a593Smuzhiyun
19*4882a593Smuzhiyun Copyright (C) 1996-2002 Julian R Seward. All rights reserved.
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun Redistribution and use in source and binary forms, with or without
22*4882a593Smuzhiyun modification, are permitted provided that the following conditions
23*4882a593Smuzhiyun are met:
24*4882a593Smuzhiyun
25*4882a593Smuzhiyun 1. Redistributions of source code must retain the above copyright
26*4882a593Smuzhiyun notice, this list of conditions and the following disclaimer.
27*4882a593Smuzhiyun
28*4882a593Smuzhiyun 2. The origin of this software must not be misrepresented; you must
29*4882a593Smuzhiyun not claim that you wrote the original software. If you use this
30*4882a593Smuzhiyun software in a product, an acknowledgment in the product
31*4882a593Smuzhiyun documentation would be appreciated but is not required.
32*4882a593Smuzhiyun
33*4882a593Smuzhiyun 3. Altered source versions must be plainly marked as such, and must
34*4882a593Smuzhiyun not be misrepresented as being the original software.
35*4882a593Smuzhiyun
36*4882a593Smuzhiyun 4. The name of the author may not be used to endorse or promote
37*4882a593Smuzhiyun products derived from this software without specific prior written
38*4882a593Smuzhiyun permission.
39*4882a593Smuzhiyun
40*4882a593Smuzhiyun THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
41*4882a593Smuzhiyun OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
42*4882a593Smuzhiyun WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
43*4882a593Smuzhiyun ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
44*4882a593Smuzhiyun DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
45*4882a593Smuzhiyun DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
46*4882a593Smuzhiyun GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
47*4882a593Smuzhiyun INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
48*4882a593Smuzhiyun WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
49*4882a593Smuzhiyun NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
50*4882a593Smuzhiyun SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
51*4882a593Smuzhiyun
52*4882a593Smuzhiyun Julian Seward, Cambridge, UK.
53*4882a593Smuzhiyun jseward@acm.org
54*4882a593Smuzhiyun bzip2/libbzip2 version 1.0 of 21 March 2000
55*4882a593Smuzhiyun
56*4882a593Smuzhiyun This program is based on (at least) the work of:
57*4882a593Smuzhiyun Mike Burrows
58*4882a593Smuzhiyun David Wheeler
59*4882a593Smuzhiyun Peter Fenwick
60*4882a593Smuzhiyun Alistair Moffat
61*4882a593Smuzhiyun Radford Neal
62*4882a593Smuzhiyun Ian H. Witten
63*4882a593Smuzhiyun Robert Sedgewick
64*4882a593Smuzhiyun Jon L. Bentley
65*4882a593Smuzhiyun
66*4882a593Smuzhiyun For more information on these sources, see the manual.
67*4882a593Smuzhiyun --*/
68*4882a593Smuzhiyun
69*4882a593Smuzhiyun /*--
70*4882a593Smuzhiyun CHANGES
71*4882a593Smuzhiyun ~~~~~~~
72*4882a593Smuzhiyun 0.9.0 -- original version.
73*4882a593Smuzhiyun
74*4882a593Smuzhiyun 0.9.0a/b -- no changes in this file.
75*4882a593Smuzhiyun
76*4882a593Smuzhiyun 0.9.0c
77*4882a593Smuzhiyun * made zero-length BZ_FLUSH work correctly in bzCompress().
78*4882a593Smuzhiyun * fixed bzWrite/bzRead to ignore zero-length requests.
79*4882a593Smuzhiyun * fixed bzread to correctly handle read requests after EOF.
80*4882a593Smuzhiyun * wrong parameter order in call to bzDecompressInit in
81*4882a593Smuzhiyun bzBuffToBuffDecompress. Fixed.
82*4882a593Smuzhiyun --*/
83*4882a593Smuzhiyun
84*4882a593Smuzhiyun #include "bzlib_private.h"
85*4882a593Smuzhiyun
86*4882a593Smuzhiyun /*---------------------------------------------------*/
87*4882a593Smuzhiyun /*--- Compression stuff ---*/
88*4882a593Smuzhiyun /*---------------------------------------------------*/
89*4882a593Smuzhiyun
90*4882a593Smuzhiyun
91*4882a593Smuzhiyun /*---------------------------------------------------*/
92*4882a593Smuzhiyun #ifndef BZ_NO_STDIO
BZ2_bz__AssertH__fail(int errcode)93*4882a593Smuzhiyun void BZ2_bz__AssertH__fail ( int errcode )
94*4882a593Smuzhiyun {
95*4882a593Smuzhiyun fprintf(stderr,
96*4882a593Smuzhiyun "\n\nbzip2/libbzip2: internal error number %d.\n"
97*4882a593Smuzhiyun "This is a bug in bzip2/libbzip2, %s.\n"
98*4882a593Smuzhiyun "Please report it to me at: jseward@acm.org. If this happened\n"
99*4882a593Smuzhiyun "when you were using some program which uses libbzip2 as a\n"
100*4882a593Smuzhiyun "component, you should also report this bug to the author(s)\n"
101*4882a593Smuzhiyun "of that program. Please make an effort to report this bug;\n"
102*4882a593Smuzhiyun "timely and accurate bug reports eventually lead to higher\n"
103*4882a593Smuzhiyun "quality software. Thanks. Julian Seward, 30 December 2001.\n\n",
104*4882a593Smuzhiyun errcode,
105*4882a593Smuzhiyun BZ2_bzlibVersion()
106*4882a593Smuzhiyun );
107*4882a593Smuzhiyun
108*4882a593Smuzhiyun if (errcode == 1007) {
109*4882a593Smuzhiyun fprintf(stderr,
110*4882a593Smuzhiyun "\n*** A special note about internal error number 1007 ***\n"
111*4882a593Smuzhiyun "\n"
112*4882a593Smuzhiyun "Experience suggests that a common cause of i.e. 1007\n"
113*4882a593Smuzhiyun "is unreliable memory or other hardware. The 1007 assertion\n"
114*4882a593Smuzhiyun "just happens to cross-check the results of huge numbers of\n"
115*4882a593Smuzhiyun "memory reads/writes, and so acts (unintendedly) as a stress\n"
116*4882a593Smuzhiyun "test of your memory system.\n"
117*4882a593Smuzhiyun "\n"
118*4882a593Smuzhiyun "I suggest the following: try compressing the file again,\n"
119*4882a593Smuzhiyun "possibly monitoring progress in detail with the -vv flag.\n"
120*4882a593Smuzhiyun "\n"
121*4882a593Smuzhiyun "* If the error cannot be reproduced, and/or happens at different\n"
122*4882a593Smuzhiyun " points in compression, you may have a flaky memory system.\n"
123*4882a593Smuzhiyun " Try a memory-test program. I have used Memtest86\n"
124*4882a593Smuzhiyun " (www.memtest86.com). At the time of writing it is free (GPLd).\n"
125*4882a593Smuzhiyun " Memtest86 tests memory much more thorougly than your BIOSs\n"
126*4882a593Smuzhiyun " power-on test, and may find failures that the BIOS doesn't.\n"
127*4882a593Smuzhiyun "\n"
128*4882a593Smuzhiyun "* If the error can be repeatably reproduced, this is a bug in\n"
129*4882a593Smuzhiyun " bzip2, and I would very much like to hear about it. Please\n"
130*4882a593Smuzhiyun " let me know, and, ideally, save a copy of the file causing the\n"
131*4882a593Smuzhiyun " problem -- without which I will be unable to investigate it.\n"
132*4882a593Smuzhiyun "\n"
133*4882a593Smuzhiyun );
134*4882a593Smuzhiyun }
135*4882a593Smuzhiyun
136*4882a593Smuzhiyun exit(3);
137*4882a593Smuzhiyun }
138*4882a593Smuzhiyun #endif
139*4882a593Smuzhiyun
140*4882a593Smuzhiyun
141*4882a593Smuzhiyun /*---------------------------------------------------*/
142*4882a593Smuzhiyun static
bz_config_ok(void)143*4882a593Smuzhiyun int bz_config_ok ( void )
144*4882a593Smuzhiyun {
145*4882a593Smuzhiyun if (sizeof(int) != 4) return 0;
146*4882a593Smuzhiyun if (sizeof(short) != 2) return 0;
147*4882a593Smuzhiyun if (sizeof(char) != 1) return 0;
148*4882a593Smuzhiyun return 1;
149*4882a593Smuzhiyun }
150*4882a593Smuzhiyun
151*4882a593Smuzhiyun
152*4882a593Smuzhiyun /*---------------------------------------------------*/
153*4882a593Smuzhiyun static
default_bzalloc(void * opaque,Int32 items,Int32 size)154*4882a593Smuzhiyun void* default_bzalloc ( void* opaque, Int32 items, Int32 size )
155*4882a593Smuzhiyun {
156*4882a593Smuzhiyun void* v = malloc ( items * size );
157*4882a593Smuzhiyun return v;
158*4882a593Smuzhiyun }
159*4882a593Smuzhiyun
160*4882a593Smuzhiyun static
default_bzfree(void * opaque,void * addr)161*4882a593Smuzhiyun void default_bzfree ( void* opaque, void* addr )
162*4882a593Smuzhiyun {
163*4882a593Smuzhiyun if (addr != NULL) free ( addr );
164*4882a593Smuzhiyun }
165*4882a593Smuzhiyun
166*4882a593Smuzhiyun #ifndef BZ_NO_COMPRESS
167*4882a593Smuzhiyun /*---------------------------------------------------*/
168*4882a593Smuzhiyun static
prepare_new_block(EState * s)169*4882a593Smuzhiyun void prepare_new_block ( EState* s )
170*4882a593Smuzhiyun {
171*4882a593Smuzhiyun Int32 i;
172*4882a593Smuzhiyun s->nblock = 0;
173*4882a593Smuzhiyun s->numZ = 0;
174*4882a593Smuzhiyun s->state_out_pos = 0;
175*4882a593Smuzhiyun BZ_INITIALISE_CRC ( s->blockCRC );
176*4882a593Smuzhiyun for (i = 0; i < 256; i++) s->inUse[i] = False;
177*4882a593Smuzhiyun s->blockNo++;
178*4882a593Smuzhiyun }
179*4882a593Smuzhiyun
180*4882a593Smuzhiyun
181*4882a593Smuzhiyun /*---------------------------------------------------*/
182*4882a593Smuzhiyun static
init_RL(EState * s)183*4882a593Smuzhiyun void init_RL ( EState* s )
184*4882a593Smuzhiyun {
185*4882a593Smuzhiyun s->state_in_ch = 256;
186*4882a593Smuzhiyun s->state_in_len = 0;
187*4882a593Smuzhiyun }
188*4882a593Smuzhiyun
189*4882a593Smuzhiyun
190*4882a593Smuzhiyun static
isempty_RL(EState * s)191*4882a593Smuzhiyun Bool isempty_RL ( EState* s )
192*4882a593Smuzhiyun {
193*4882a593Smuzhiyun if (s->state_in_ch < 256 && s->state_in_len > 0)
194*4882a593Smuzhiyun return False; else
195*4882a593Smuzhiyun return True;
196*4882a593Smuzhiyun }
197*4882a593Smuzhiyun
198*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzCompressInit)199*4882a593Smuzhiyun int BZ_API(BZ2_bzCompressInit)
200*4882a593Smuzhiyun ( bz_stream* strm,
201*4882a593Smuzhiyun int blockSize100k,
202*4882a593Smuzhiyun int verbosity,
203*4882a593Smuzhiyun int workFactor )
204*4882a593Smuzhiyun {
205*4882a593Smuzhiyun Int32 n;
206*4882a593Smuzhiyun EState* s;
207*4882a593Smuzhiyun
208*4882a593Smuzhiyun if (!bz_config_ok()) return BZ_CONFIG_ERROR;
209*4882a593Smuzhiyun
210*4882a593Smuzhiyun if (strm == NULL ||
211*4882a593Smuzhiyun blockSize100k < 1 || blockSize100k > 9 ||
212*4882a593Smuzhiyun workFactor < 0 || workFactor > 250)
213*4882a593Smuzhiyun return BZ_PARAM_ERROR;
214*4882a593Smuzhiyun
215*4882a593Smuzhiyun if (workFactor == 0) workFactor = 30;
216*4882a593Smuzhiyun if (strm->bzalloc == NULL) strm->bzalloc = default_bzalloc;
217*4882a593Smuzhiyun if (strm->bzfree == NULL) strm->bzfree = default_bzfree;
218*4882a593Smuzhiyun
219*4882a593Smuzhiyun s = BZALLOC( sizeof(EState) );
220*4882a593Smuzhiyun if (s == NULL) return BZ_MEM_ERROR;
221*4882a593Smuzhiyun s->strm = strm;
222*4882a593Smuzhiyun
223*4882a593Smuzhiyun s->arr1 = NULL;
224*4882a593Smuzhiyun s->arr2 = NULL;
225*4882a593Smuzhiyun s->ftab = NULL;
226*4882a593Smuzhiyun
227*4882a593Smuzhiyun n = 100000 * blockSize100k;
228*4882a593Smuzhiyun s->arr1 = BZALLOC( n * sizeof(UInt32) );
229*4882a593Smuzhiyun s->arr2 = BZALLOC( (n+BZ_N_OVERSHOOT) * sizeof(UInt32) );
230*4882a593Smuzhiyun s->ftab = BZALLOC( 65537 * sizeof(UInt32) );
231*4882a593Smuzhiyun
232*4882a593Smuzhiyun if (s->arr1 == NULL || s->arr2 == NULL || s->ftab == NULL) {
233*4882a593Smuzhiyun if (s->arr1 != NULL) BZFREE(s->arr1);
234*4882a593Smuzhiyun if (s->arr2 != NULL) BZFREE(s->arr2);
235*4882a593Smuzhiyun if (s->ftab != NULL) BZFREE(s->ftab);
236*4882a593Smuzhiyun if (s != NULL) BZFREE(s);
237*4882a593Smuzhiyun return BZ_MEM_ERROR;
238*4882a593Smuzhiyun }
239*4882a593Smuzhiyun
240*4882a593Smuzhiyun s->blockNo = 0;
241*4882a593Smuzhiyun s->state = BZ_S_INPUT;
242*4882a593Smuzhiyun s->mode = BZ_M_RUNNING;
243*4882a593Smuzhiyun s->combinedCRC = 0;
244*4882a593Smuzhiyun s->blockSize100k = blockSize100k;
245*4882a593Smuzhiyun s->nblockMAX = 100000 * blockSize100k - 19;
246*4882a593Smuzhiyun s->verbosity = verbosity;
247*4882a593Smuzhiyun s->workFactor = workFactor;
248*4882a593Smuzhiyun
249*4882a593Smuzhiyun s->block = (UChar*)s->arr2;
250*4882a593Smuzhiyun s->mtfv = (UInt16*)s->arr1;
251*4882a593Smuzhiyun s->zbits = NULL;
252*4882a593Smuzhiyun s->ptr = (UInt32*)s->arr1;
253*4882a593Smuzhiyun
254*4882a593Smuzhiyun strm->state = s;
255*4882a593Smuzhiyun strm->total_in_lo32 = 0;
256*4882a593Smuzhiyun strm->total_in_hi32 = 0;
257*4882a593Smuzhiyun strm->total_out_lo32 = 0;
258*4882a593Smuzhiyun strm->total_out_hi32 = 0;
259*4882a593Smuzhiyun init_RL ( s );
260*4882a593Smuzhiyun prepare_new_block ( s );
261*4882a593Smuzhiyun return BZ_OK;
262*4882a593Smuzhiyun }
263*4882a593Smuzhiyun
264*4882a593Smuzhiyun
265*4882a593Smuzhiyun /*---------------------------------------------------*/
266*4882a593Smuzhiyun static
add_pair_to_block(EState * s)267*4882a593Smuzhiyun void add_pair_to_block ( EState* s )
268*4882a593Smuzhiyun {
269*4882a593Smuzhiyun Int32 i;
270*4882a593Smuzhiyun UChar ch = (UChar)(s->state_in_ch);
271*4882a593Smuzhiyun for (i = 0; i < s->state_in_len; i++) {
272*4882a593Smuzhiyun BZ_UPDATE_CRC( s->blockCRC, ch );
273*4882a593Smuzhiyun }
274*4882a593Smuzhiyun s->inUse[s->state_in_ch] = True;
275*4882a593Smuzhiyun switch (s->state_in_len) {
276*4882a593Smuzhiyun case 1:
277*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
278*4882a593Smuzhiyun break;
279*4882a593Smuzhiyun case 2:
280*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
281*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
282*4882a593Smuzhiyun break;
283*4882a593Smuzhiyun case 3:
284*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
285*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
286*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
287*4882a593Smuzhiyun break;
288*4882a593Smuzhiyun default:
289*4882a593Smuzhiyun s->inUse[s->state_in_len-4] = True;
290*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
291*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
292*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
293*4882a593Smuzhiyun s->block[s->nblock] = (UChar)ch; s->nblock++;
294*4882a593Smuzhiyun s->block[s->nblock] = ((UChar)(s->state_in_len-4));
295*4882a593Smuzhiyun s->nblock++;
296*4882a593Smuzhiyun break;
297*4882a593Smuzhiyun }
298*4882a593Smuzhiyun }
299*4882a593Smuzhiyun
300*4882a593Smuzhiyun
301*4882a593Smuzhiyun /*---------------------------------------------------*/
302*4882a593Smuzhiyun static
flush_RL(EState * s)303*4882a593Smuzhiyun void flush_RL ( EState* s )
304*4882a593Smuzhiyun {
305*4882a593Smuzhiyun if (s->state_in_ch < 256) add_pair_to_block ( s );
306*4882a593Smuzhiyun init_RL ( s );
307*4882a593Smuzhiyun }
308*4882a593Smuzhiyun
309*4882a593Smuzhiyun
310*4882a593Smuzhiyun /*---------------------------------------------------*/
311*4882a593Smuzhiyun #define ADD_CHAR_TO_BLOCK(zs,zchh0) \
312*4882a593Smuzhiyun { \
313*4882a593Smuzhiyun UInt32 zchh = (UInt32)(zchh0); \
314*4882a593Smuzhiyun /*-- fast track the common case --*/ \
315*4882a593Smuzhiyun if (zchh != zs->state_in_ch && \
316*4882a593Smuzhiyun zs->state_in_len == 1) { \
317*4882a593Smuzhiyun UChar ch = (UChar)(zs->state_in_ch); \
318*4882a593Smuzhiyun BZ_UPDATE_CRC( zs->blockCRC, ch ); \
319*4882a593Smuzhiyun zs->inUse[zs->state_in_ch] = True; \
320*4882a593Smuzhiyun zs->block[zs->nblock] = (UChar)ch; \
321*4882a593Smuzhiyun zs->nblock++; \
322*4882a593Smuzhiyun zs->state_in_ch = zchh; \
323*4882a593Smuzhiyun } \
324*4882a593Smuzhiyun else \
325*4882a593Smuzhiyun /*-- general, uncommon cases --*/ \
326*4882a593Smuzhiyun if (zchh != zs->state_in_ch || \
327*4882a593Smuzhiyun zs->state_in_len == 255) { \
328*4882a593Smuzhiyun if (zs->state_in_ch < 256) \
329*4882a593Smuzhiyun add_pair_to_block ( zs ); \
330*4882a593Smuzhiyun zs->state_in_ch = zchh; \
331*4882a593Smuzhiyun zs->state_in_len = 1; \
332*4882a593Smuzhiyun } else { \
333*4882a593Smuzhiyun zs->state_in_len++; \
334*4882a593Smuzhiyun } \
335*4882a593Smuzhiyun }
336*4882a593Smuzhiyun
337*4882a593Smuzhiyun
338*4882a593Smuzhiyun /*---------------------------------------------------*/
339*4882a593Smuzhiyun static
copy_input_until_stop(EState * s)340*4882a593Smuzhiyun Bool copy_input_until_stop ( EState* s )
341*4882a593Smuzhiyun {
342*4882a593Smuzhiyun Bool progress_in = False;
343*4882a593Smuzhiyun
344*4882a593Smuzhiyun if (s->mode == BZ_M_RUNNING) {
345*4882a593Smuzhiyun
346*4882a593Smuzhiyun /*-- fast track the common case --*/
347*4882a593Smuzhiyun while (True) {
348*4882a593Smuzhiyun /*-- block full? --*/
349*4882a593Smuzhiyun if (s->nblock >= s->nblockMAX) break;
350*4882a593Smuzhiyun /*-- no input? --*/
351*4882a593Smuzhiyun if (s->strm->avail_in == 0) break;
352*4882a593Smuzhiyun progress_in = True;
353*4882a593Smuzhiyun ADD_CHAR_TO_BLOCK ( s, (UInt32)(*((UChar*)(s->strm->next_in))) );
354*4882a593Smuzhiyun s->strm->next_in++;
355*4882a593Smuzhiyun s->strm->avail_in--;
356*4882a593Smuzhiyun s->strm->total_in_lo32++;
357*4882a593Smuzhiyun if (s->strm->total_in_lo32 == 0) s->strm->total_in_hi32++;
358*4882a593Smuzhiyun }
359*4882a593Smuzhiyun
360*4882a593Smuzhiyun } else {
361*4882a593Smuzhiyun
362*4882a593Smuzhiyun /*-- general, uncommon case --*/
363*4882a593Smuzhiyun while (True) {
364*4882a593Smuzhiyun /*-- block full? --*/
365*4882a593Smuzhiyun if (s->nblock >= s->nblockMAX) break;
366*4882a593Smuzhiyun /*-- no input? --*/
367*4882a593Smuzhiyun if (s->strm->avail_in == 0) break;
368*4882a593Smuzhiyun /*-- flush/finish end? --*/
369*4882a593Smuzhiyun if (s->avail_in_expect == 0) break;
370*4882a593Smuzhiyun progress_in = True;
371*4882a593Smuzhiyun ADD_CHAR_TO_BLOCK ( s, (UInt32)(*((UChar*)(s->strm->next_in))) );
372*4882a593Smuzhiyun s->strm->next_in++;
373*4882a593Smuzhiyun s->strm->avail_in--;
374*4882a593Smuzhiyun s->strm->total_in_lo32++;
375*4882a593Smuzhiyun if (s->strm->total_in_lo32 == 0) s->strm->total_in_hi32++;
376*4882a593Smuzhiyun s->avail_in_expect--;
377*4882a593Smuzhiyun }
378*4882a593Smuzhiyun }
379*4882a593Smuzhiyun return progress_in;
380*4882a593Smuzhiyun }
381*4882a593Smuzhiyun
382*4882a593Smuzhiyun
383*4882a593Smuzhiyun /*---------------------------------------------------*/
384*4882a593Smuzhiyun static
copy_output_until_stop(EState * s)385*4882a593Smuzhiyun Bool copy_output_until_stop ( EState* s )
386*4882a593Smuzhiyun {
387*4882a593Smuzhiyun Bool progress_out = False;
388*4882a593Smuzhiyun
389*4882a593Smuzhiyun while (True) {
390*4882a593Smuzhiyun
391*4882a593Smuzhiyun /*-- no output space? --*/
392*4882a593Smuzhiyun if (s->strm->avail_out == 0) break;
393*4882a593Smuzhiyun
394*4882a593Smuzhiyun /*-- block done? --*/
395*4882a593Smuzhiyun if (s->state_out_pos >= s->numZ) break;
396*4882a593Smuzhiyun
397*4882a593Smuzhiyun progress_out = True;
398*4882a593Smuzhiyun *(s->strm->next_out) = s->zbits[s->state_out_pos];
399*4882a593Smuzhiyun s->state_out_pos++;
400*4882a593Smuzhiyun s->strm->avail_out--;
401*4882a593Smuzhiyun s->strm->next_out++;
402*4882a593Smuzhiyun s->strm->total_out_lo32++;
403*4882a593Smuzhiyun if (s->strm->total_out_lo32 == 0) s->strm->total_out_hi32++;
404*4882a593Smuzhiyun }
405*4882a593Smuzhiyun
406*4882a593Smuzhiyun return progress_out;
407*4882a593Smuzhiyun }
408*4882a593Smuzhiyun
409*4882a593Smuzhiyun
410*4882a593Smuzhiyun /*---------------------------------------------------*/
411*4882a593Smuzhiyun static
handle_compress(bz_stream * strm)412*4882a593Smuzhiyun Bool handle_compress ( bz_stream* strm )
413*4882a593Smuzhiyun {
414*4882a593Smuzhiyun Bool progress_in = False;
415*4882a593Smuzhiyun Bool progress_out = False;
416*4882a593Smuzhiyun EState* s = strm->state;
417*4882a593Smuzhiyun
418*4882a593Smuzhiyun while (True) {
419*4882a593Smuzhiyun
420*4882a593Smuzhiyun if (s->state == BZ_S_OUTPUT) {
421*4882a593Smuzhiyun progress_out |= copy_output_until_stop ( s );
422*4882a593Smuzhiyun if (s->state_out_pos < s->numZ) break;
423*4882a593Smuzhiyun if (s->mode == BZ_M_FINISHING &&
424*4882a593Smuzhiyun s->avail_in_expect == 0 &&
425*4882a593Smuzhiyun isempty_RL(s)) break;
426*4882a593Smuzhiyun prepare_new_block ( s );
427*4882a593Smuzhiyun s->state = BZ_S_INPUT;
428*4882a593Smuzhiyun if (s->mode == BZ_M_FLUSHING &&
429*4882a593Smuzhiyun s->avail_in_expect == 0 &&
430*4882a593Smuzhiyun isempty_RL(s)) break;
431*4882a593Smuzhiyun }
432*4882a593Smuzhiyun
433*4882a593Smuzhiyun if (s->state == BZ_S_INPUT) {
434*4882a593Smuzhiyun progress_in |= copy_input_until_stop ( s );
435*4882a593Smuzhiyun if (s->mode != BZ_M_RUNNING && s->avail_in_expect == 0) {
436*4882a593Smuzhiyun flush_RL ( s );
437*4882a593Smuzhiyun BZ2_compressBlock ( s, (Bool)(s->mode == BZ_M_FINISHING) );
438*4882a593Smuzhiyun s->state = BZ_S_OUTPUT;
439*4882a593Smuzhiyun }
440*4882a593Smuzhiyun else
441*4882a593Smuzhiyun if (s->nblock >= s->nblockMAX) {
442*4882a593Smuzhiyun BZ2_compressBlock ( s, False );
443*4882a593Smuzhiyun s->state = BZ_S_OUTPUT;
444*4882a593Smuzhiyun }
445*4882a593Smuzhiyun else
446*4882a593Smuzhiyun if (s->strm->avail_in == 0) {
447*4882a593Smuzhiyun break;
448*4882a593Smuzhiyun }
449*4882a593Smuzhiyun }
450*4882a593Smuzhiyun
451*4882a593Smuzhiyun }
452*4882a593Smuzhiyun
453*4882a593Smuzhiyun return progress_in || progress_out;
454*4882a593Smuzhiyun }
455*4882a593Smuzhiyun
456*4882a593Smuzhiyun
457*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzCompress)458*4882a593Smuzhiyun int BZ_API(BZ2_bzCompress) ( bz_stream *strm, int action )
459*4882a593Smuzhiyun {
460*4882a593Smuzhiyun Bool progress;
461*4882a593Smuzhiyun EState* s;
462*4882a593Smuzhiyun if (strm == NULL) return BZ_PARAM_ERROR;
463*4882a593Smuzhiyun s = strm->state;
464*4882a593Smuzhiyun if (s == NULL) return BZ_PARAM_ERROR;
465*4882a593Smuzhiyun if (s->strm != strm) return BZ_PARAM_ERROR;
466*4882a593Smuzhiyun
467*4882a593Smuzhiyun preswitch:
468*4882a593Smuzhiyun switch (s->mode) {
469*4882a593Smuzhiyun
470*4882a593Smuzhiyun case BZ_M_IDLE:
471*4882a593Smuzhiyun return BZ_SEQUENCE_ERROR;
472*4882a593Smuzhiyun
473*4882a593Smuzhiyun case BZ_M_RUNNING:
474*4882a593Smuzhiyun if (action == BZ_RUN) {
475*4882a593Smuzhiyun progress = handle_compress ( strm );
476*4882a593Smuzhiyun return progress ? BZ_RUN_OK : BZ_PARAM_ERROR;
477*4882a593Smuzhiyun }
478*4882a593Smuzhiyun else
479*4882a593Smuzhiyun if (action == BZ_FLUSH) {
480*4882a593Smuzhiyun s->avail_in_expect = strm->avail_in;
481*4882a593Smuzhiyun s->mode = BZ_M_FLUSHING;
482*4882a593Smuzhiyun goto preswitch;
483*4882a593Smuzhiyun }
484*4882a593Smuzhiyun else
485*4882a593Smuzhiyun if (action == BZ_FINISH) {
486*4882a593Smuzhiyun s->avail_in_expect = strm->avail_in;
487*4882a593Smuzhiyun s->mode = BZ_M_FINISHING;
488*4882a593Smuzhiyun goto preswitch;
489*4882a593Smuzhiyun }
490*4882a593Smuzhiyun else
491*4882a593Smuzhiyun return BZ_PARAM_ERROR;
492*4882a593Smuzhiyun
493*4882a593Smuzhiyun case BZ_M_FLUSHING:
494*4882a593Smuzhiyun if (action != BZ_FLUSH) return BZ_SEQUENCE_ERROR;
495*4882a593Smuzhiyun if (s->avail_in_expect != s->strm->avail_in)
496*4882a593Smuzhiyun return BZ_SEQUENCE_ERROR;
497*4882a593Smuzhiyun progress = handle_compress ( strm );
498*4882a593Smuzhiyun if (s->avail_in_expect > 0 || !isempty_RL(s) ||
499*4882a593Smuzhiyun s->state_out_pos < s->numZ) return BZ_FLUSH_OK;
500*4882a593Smuzhiyun s->mode = BZ_M_RUNNING;
501*4882a593Smuzhiyun return BZ_RUN_OK;
502*4882a593Smuzhiyun
503*4882a593Smuzhiyun case BZ_M_FINISHING:
504*4882a593Smuzhiyun if (action != BZ_FINISH) return BZ_SEQUENCE_ERROR;
505*4882a593Smuzhiyun if (s->avail_in_expect != s->strm->avail_in)
506*4882a593Smuzhiyun return BZ_SEQUENCE_ERROR;
507*4882a593Smuzhiyun progress = handle_compress ( strm );
508*4882a593Smuzhiyun if (!progress) return BZ_SEQUENCE_ERROR;
509*4882a593Smuzhiyun if (s->avail_in_expect > 0 || !isempty_RL(s) ||
510*4882a593Smuzhiyun s->state_out_pos < s->numZ) return BZ_FINISH_OK;
511*4882a593Smuzhiyun s->mode = BZ_M_IDLE;
512*4882a593Smuzhiyun return BZ_STREAM_END;
513*4882a593Smuzhiyun }
514*4882a593Smuzhiyun return BZ_OK; /*--not reached--*/
515*4882a593Smuzhiyun }
516*4882a593Smuzhiyun
517*4882a593Smuzhiyun
518*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzCompressEnd)519*4882a593Smuzhiyun int BZ_API(BZ2_bzCompressEnd) ( bz_stream *strm )
520*4882a593Smuzhiyun {
521*4882a593Smuzhiyun EState* s;
522*4882a593Smuzhiyun if (strm == NULL) return BZ_PARAM_ERROR;
523*4882a593Smuzhiyun s = strm->state;
524*4882a593Smuzhiyun if (s == NULL) return BZ_PARAM_ERROR;
525*4882a593Smuzhiyun if (s->strm != strm) return BZ_PARAM_ERROR;
526*4882a593Smuzhiyun
527*4882a593Smuzhiyun if (s->arr1 != NULL) BZFREE(s->arr1);
528*4882a593Smuzhiyun if (s->arr2 != NULL) BZFREE(s->arr2);
529*4882a593Smuzhiyun if (s->ftab != NULL) BZFREE(s->ftab);
530*4882a593Smuzhiyun BZFREE(strm->state);
531*4882a593Smuzhiyun
532*4882a593Smuzhiyun strm->state = NULL;
533*4882a593Smuzhiyun
534*4882a593Smuzhiyun return BZ_OK;
535*4882a593Smuzhiyun }
536*4882a593Smuzhiyun #endif /* BZ_NO_COMPRESS */
537*4882a593Smuzhiyun
538*4882a593Smuzhiyun /*---------------------------------------------------*/
539*4882a593Smuzhiyun /*--- Decompression stuff ---*/
540*4882a593Smuzhiyun /*---------------------------------------------------*/
541*4882a593Smuzhiyun
542*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzDecompressInit)543*4882a593Smuzhiyun int BZ_API(BZ2_bzDecompressInit)
544*4882a593Smuzhiyun ( bz_stream* strm,
545*4882a593Smuzhiyun int verbosity,
546*4882a593Smuzhiyun int small )
547*4882a593Smuzhiyun {
548*4882a593Smuzhiyun DState* s;
549*4882a593Smuzhiyun
550*4882a593Smuzhiyun if (!bz_config_ok()) return BZ_CONFIG_ERROR;
551*4882a593Smuzhiyun
552*4882a593Smuzhiyun if (strm == NULL) return BZ_PARAM_ERROR;
553*4882a593Smuzhiyun if (small != 0 && small != 1) return BZ_PARAM_ERROR;
554*4882a593Smuzhiyun if (verbosity < 0 || verbosity > 4) return BZ_PARAM_ERROR;
555*4882a593Smuzhiyun
556*4882a593Smuzhiyun if (strm->bzalloc == NULL) strm->bzalloc = default_bzalloc;
557*4882a593Smuzhiyun if (strm->bzfree == NULL) strm->bzfree = default_bzfree;
558*4882a593Smuzhiyun
559*4882a593Smuzhiyun s = BZALLOC( sizeof(DState) );
560*4882a593Smuzhiyun if (s == NULL) return BZ_MEM_ERROR;
561*4882a593Smuzhiyun s->strm = strm;
562*4882a593Smuzhiyun strm->state = s;
563*4882a593Smuzhiyun s->state = BZ_X_MAGIC_1;
564*4882a593Smuzhiyun s->bsLive = 0;
565*4882a593Smuzhiyun s->bsBuff = 0;
566*4882a593Smuzhiyun s->calculatedCombinedCRC = 0;
567*4882a593Smuzhiyun strm->total_in_lo32 = 0;
568*4882a593Smuzhiyun strm->total_in_hi32 = 0;
569*4882a593Smuzhiyun strm->total_out_lo32 = 0;
570*4882a593Smuzhiyun strm->total_out_hi32 = 0;
571*4882a593Smuzhiyun s->smallDecompress = (Bool)small;
572*4882a593Smuzhiyun s->ll4 = NULL;
573*4882a593Smuzhiyun s->ll16 = NULL;
574*4882a593Smuzhiyun s->tt = NULL;
575*4882a593Smuzhiyun s->currBlockNo = 0;
576*4882a593Smuzhiyun s->verbosity = verbosity;
577*4882a593Smuzhiyun
578*4882a593Smuzhiyun return BZ_OK;
579*4882a593Smuzhiyun }
580*4882a593Smuzhiyun
581*4882a593Smuzhiyun
582*4882a593Smuzhiyun /*---------------------------------------------------*/
583*4882a593Smuzhiyun static
unRLE_obuf_to_output_FAST(DState * s)584*4882a593Smuzhiyun void unRLE_obuf_to_output_FAST ( DState* s )
585*4882a593Smuzhiyun {
586*4882a593Smuzhiyun UChar k1;
587*4882a593Smuzhiyun
588*4882a593Smuzhiyun if (s->blockRandomised) {
589*4882a593Smuzhiyun
590*4882a593Smuzhiyun while (True) {
591*4882a593Smuzhiyun /* try to finish existing run */
592*4882a593Smuzhiyun while (True) {
593*4882a593Smuzhiyun if (s->strm->avail_out == 0) return;
594*4882a593Smuzhiyun if (s->state_out_len == 0) break;
595*4882a593Smuzhiyun *( (UChar*)(s->strm->next_out) ) = s->state_out_ch;
596*4882a593Smuzhiyun BZ_UPDATE_CRC ( s->calculatedBlockCRC, s->state_out_ch );
597*4882a593Smuzhiyun s->state_out_len--;
598*4882a593Smuzhiyun s->strm->next_out++;
599*4882a593Smuzhiyun s->strm->avail_out--;
600*4882a593Smuzhiyun s->strm->total_out_lo32++;
601*4882a593Smuzhiyun if (s->strm->total_out_lo32 == 0) s->strm->total_out_hi32++;
602*4882a593Smuzhiyun }
603*4882a593Smuzhiyun
604*4882a593Smuzhiyun /* can a new run be started? */
605*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) return;
606*4882a593Smuzhiyun
607*4882a593Smuzhiyun
608*4882a593Smuzhiyun s->state_out_len = 1;
609*4882a593Smuzhiyun s->state_out_ch = s->k0;
610*4882a593Smuzhiyun BZ_GET_FAST(k1); BZ_RAND_UPD_MASK;
611*4882a593Smuzhiyun k1 ^= BZ_RAND_MASK; s->nblock_used++;
612*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
613*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
614*4882a593Smuzhiyun
615*4882a593Smuzhiyun s->state_out_len = 2;
616*4882a593Smuzhiyun BZ_GET_FAST(k1); BZ_RAND_UPD_MASK;
617*4882a593Smuzhiyun k1 ^= BZ_RAND_MASK; s->nblock_used++;
618*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
619*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
620*4882a593Smuzhiyun
621*4882a593Smuzhiyun s->state_out_len = 3;
622*4882a593Smuzhiyun BZ_GET_FAST(k1); BZ_RAND_UPD_MASK;
623*4882a593Smuzhiyun k1 ^= BZ_RAND_MASK; s->nblock_used++;
624*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
625*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
626*4882a593Smuzhiyun
627*4882a593Smuzhiyun BZ_GET_FAST(k1); BZ_RAND_UPD_MASK;
628*4882a593Smuzhiyun k1 ^= BZ_RAND_MASK; s->nblock_used++;
629*4882a593Smuzhiyun s->state_out_len = ((Int32)k1) + 4;
630*4882a593Smuzhiyun BZ_GET_FAST(s->k0); BZ_RAND_UPD_MASK;
631*4882a593Smuzhiyun s->k0 ^= BZ_RAND_MASK; s->nblock_used++;
632*4882a593Smuzhiyun }
633*4882a593Smuzhiyun
634*4882a593Smuzhiyun } else {
635*4882a593Smuzhiyun
636*4882a593Smuzhiyun /* restore */
637*4882a593Smuzhiyun UInt32 c_calculatedBlockCRC = s->calculatedBlockCRC;
638*4882a593Smuzhiyun UChar c_state_out_ch = s->state_out_ch;
639*4882a593Smuzhiyun Int32 c_state_out_len = s->state_out_len;
640*4882a593Smuzhiyun Int32 c_nblock_used = s->nblock_used;
641*4882a593Smuzhiyun Int32 c_k0 = s->k0;
642*4882a593Smuzhiyun UInt32* c_tt = s->tt;
643*4882a593Smuzhiyun UInt32 c_tPos = s->tPos;
644*4882a593Smuzhiyun char* cs_next_out = s->strm->next_out;
645*4882a593Smuzhiyun unsigned int cs_avail_out = s->strm->avail_out;
646*4882a593Smuzhiyun /* end restore */
647*4882a593Smuzhiyun
648*4882a593Smuzhiyun UInt32 avail_out_INIT = cs_avail_out;
649*4882a593Smuzhiyun Int32 s_save_nblockPP = s->save_nblock+1;
650*4882a593Smuzhiyun unsigned int total_out_lo32_old;
651*4882a593Smuzhiyun
652*4882a593Smuzhiyun while (True) {
653*4882a593Smuzhiyun
654*4882a593Smuzhiyun /* try to finish existing run */
655*4882a593Smuzhiyun if (c_state_out_len > 0) {
656*4882a593Smuzhiyun while (True) {
657*4882a593Smuzhiyun if (cs_avail_out == 0) goto return_notr;
658*4882a593Smuzhiyun if (c_state_out_len == 1) break;
659*4882a593Smuzhiyun *( (UChar*)(cs_next_out) ) = c_state_out_ch;
660*4882a593Smuzhiyun BZ_UPDATE_CRC ( c_calculatedBlockCRC, c_state_out_ch );
661*4882a593Smuzhiyun c_state_out_len--;
662*4882a593Smuzhiyun cs_next_out++;
663*4882a593Smuzhiyun cs_avail_out--;
664*4882a593Smuzhiyun }
665*4882a593Smuzhiyun s_state_out_len_eq_one:
666*4882a593Smuzhiyun {
667*4882a593Smuzhiyun if (cs_avail_out == 0) {
668*4882a593Smuzhiyun c_state_out_len = 1; goto return_notr;
669*4882a593Smuzhiyun };
670*4882a593Smuzhiyun *( (UChar*)(cs_next_out) ) = c_state_out_ch;
671*4882a593Smuzhiyun BZ_UPDATE_CRC ( c_calculatedBlockCRC, c_state_out_ch );
672*4882a593Smuzhiyun cs_next_out++;
673*4882a593Smuzhiyun cs_avail_out--;
674*4882a593Smuzhiyun }
675*4882a593Smuzhiyun }
676*4882a593Smuzhiyun /* can a new run be started? */
677*4882a593Smuzhiyun if (c_nblock_used == s_save_nblockPP) {
678*4882a593Smuzhiyun c_state_out_len = 0; goto return_notr;
679*4882a593Smuzhiyun };
680*4882a593Smuzhiyun c_state_out_ch = c_k0;
681*4882a593Smuzhiyun BZ_GET_FAST_C(k1); c_nblock_used++;
682*4882a593Smuzhiyun if (k1 != c_k0) {
683*4882a593Smuzhiyun c_k0 = k1; goto s_state_out_len_eq_one;
684*4882a593Smuzhiyun };
685*4882a593Smuzhiyun if (c_nblock_used == s_save_nblockPP)
686*4882a593Smuzhiyun goto s_state_out_len_eq_one;
687*4882a593Smuzhiyun
688*4882a593Smuzhiyun c_state_out_len = 2;
689*4882a593Smuzhiyun BZ_GET_FAST_C(k1); c_nblock_used++;
690*4882a593Smuzhiyun if (c_nblock_used == s_save_nblockPP) continue;
691*4882a593Smuzhiyun if (k1 != c_k0) { c_k0 = k1; continue; };
692*4882a593Smuzhiyun
693*4882a593Smuzhiyun c_state_out_len = 3;
694*4882a593Smuzhiyun BZ_GET_FAST_C(k1); c_nblock_used++;
695*4882a593Smuzhiyun if (c_nblock_used == s_save_nblockPP) continue;
696*4882a593Smuzhiyun if (k1 != c_k0) { c_k0 = k1; continue; };
697*4882a593Smuzhiyun
698*4882a593Smuzhiyun BZ_GET_FAST_C(k1); c_nblock_used++;
699*4882a593Smuzhiyun c_state_out_len = ((Int32)k1) + 4;
700*4882a593Smuzhiyun BZ_GET_FAST_C(c_k0); c_nblock_used++;
701*4882a593Smuzhiyun }
702*4882a593Smuzhiyun
703*4882a593Smuzhiyun return_notr:
704*4882a593Smuzhiyun total_out_lo32_old = s->strm->total_out_lo32;
705*4882a593Smuzhiyun s->strm->total_out_lo32 += (avail_out_INIT - cs_avail_out);
706*4882a593Smuzhiyun if (s->strm->total_out_lo32 < total_out_lo32_old)
707*4882a593Smuzhiyun s->strm->total_out_hi32++;
708*4882a593Smuzhiyun
709*4882a593Smuzhiyun /* save */
710*4882a593Smuzhiyun s->calculatedBlockCRC = c_calculatedBlockCRC;
711*4882a593Smuzhiyun s->state_out_ch = c_state_out_ch;
712*4882a593Smuzhiyun s->state_out_len = c_state_out_len;
713*4882a593Smuzhiyun s->nblock_used = c_nblock_used;
714*4882a593Smuzhiyun s->k0 = c_k0;
715*4882a593Smuzhiyun s->tt = c_tt;
716*4882a593Smuzhiyun s->tPos = c_tPos;
717*4882a593Smuzhiyun s->strm->next_out = cs_next_out;
718*4882a593Smuzhiyun s->strm->avail_out = cs_avail_out;
719*4882a593Smuzhiyun /* end save */
720*4882a593Smuzhiyun }
721*4882a593Smuzhiyun }
722*4882a593Smuzhiyun
723*4882a593Smuzhiyun
724*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ2_indexIntoF(Int32 indx,Int32 * cftab)725*4882a593Smuzhiyun __inline__ Int32 BZ2_indexIntoF ( Int32 indx, Int32 *cftab )
726*4882a593Smuzhiyun {
727*4882a593Smuzhiyun Int32 nb, na, mid;
728*4882a593Smuzhiyun nb = 0;
729*4882a593Smuzhiyun na = 256;
730*4882a593Smuzhiyun do {
731*4882a593Smuzhiyun mid = (nb + na) >> 1;
732*4882a593Smuzhiyun if (indx >= cftab[mid]) nb = mid; else na = mid;
733*4882a593Smuzhiyun }
734*4882a593Smuzhiyun while (na - nb != 1);
735*4882a593Smuzhiyun return nb;
736*4882a593Smuzhiyun }
737*4882a593Smuzhiyun
738*4882a593Smuzhiyun
739*4882a593Smuzhiyun /*---------------------------------------------------*/
740*4882a593Smuzhiyun static
unRLE_obuf_to_output_SMALL(DState * s)741*4882a593Smuzhiyun void unRLE_obuf_to_output_SMALL ( DState* s )
742*4882a593Smuzhiyun {
743*4882a593Smuzhiyun UChar k1;
744*4882a593Smuzhiyun
745*4882a593Smuzhiyun if (s->blockRandomised) {
746*4882a593Smuzhiyun
747*4882a593Smuzhiyun while (True) {
748*4882a593Smuzhiyun /* try to finish existing run */
749*4882a593Smuzhiyun while (True) {
750*4882a593Smuzhiyun if (s->strm->avail_out == 0) return;
751*4882a593Smuzhiyun if (s->state_out_len == 0) break;
752*4882a593Smuzhiyun *( (UChar*)(s->strm->next_out) ) = s->state_out_ch;
753*4882a593Smuzhiyun BZ_UPDATE_CRC ( s->calculatedBlockCRC, s->state_out_ch );
754*4882a593Smuzhiyun s->state_out_len--;
755*4882a593Smuzhiyun s->strm->next_out++;
756*4882a593Smuzhiyun s->strm->avail_out--;
757*4882a593Smuzhiyun s->strm->total_out_lo32++;
758*4882a593Smuzhiyun if (s->strm->total_out_lo32 == 0) s->strm->total_out_hi32++;
759*4882a593Smuzhiyun }
760*4882a593Smuzhiyun
761*4882a593Smuzhiyun /* can a new run be started? */
762*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) return;
763*4882a593Smuzhiyun
764*4882a593Smuzhiyun
765*4882a593Smuzhiyun s->state_out_len = 1;
766*4882a593Smuzhiyun s->state_out_ch = s->k0;
767*4882a593Smuzhiyun BZ_GET_SMALL(k1); BZ_RAND_UPD_MASK;
768*4882a593Smuzhiyun k1 ^= BZ_RAND_MASK; s->nblock_used++;
769*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
770*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
771*4882a593Smuzhiyun
772*4882a593Smuzhiyun s->state_out_len = 2;
773*4882a593Smuzhiyun BZ_GET_SMALL(k1); BZ_RAND_UPD_MASK;
774*4882a593Smuzhiyun k1 ^= BZ_RAND_MASK; s->nblock_used++;
775*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
776*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
777*4882a593Smuzhiyun
778*4882a593Smuzhiyun s->state_out_len = 3;
779*4882a593Smuzhiyun BZ_GET_SMALL(k1); BZ_RAND_UPD_MASK;
780*4882a593Smuzhiyun k1 ^= BZ_RAND_MASK; s->nblock_used++;
781*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
782*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
783*4882a593Smuzhiyun
784*4882a593Smuzhiyun BZ_GET_SMALL(k1); BZ_RAND_UPD_MASK;
785*4882a593Smuzhiyun k1 ^= BZ_RAND_MASK; s->nblock_used++;
786*4882a593Smuzhiyun s->state_out_len = ((Int32)k1) + 4;
787*4882a593Smuzhiyun BZ_GET_SMALL(s->k0); BZ_RAND_UPD_MASK;
788*4882a593Smuzhiyun s->k0 ^= BZ_RAND_MASK; s->nblock_used++;
789*4882a593Smuzhiyun }
790*4882a593Smuzhiyun
791*4882a593Smuzhiyun } else {
792*4882a593Smuzhiyun
793*4882a593Smuzhiyun while (True) {
794*4882a593Smuzhiyun /* try to finish existing run */
795*4882a593Smuzhiyun while (True) {
796*4882a593Smuzhiyun if (s->strm->avail_out == 0) return;
797*4882a593Smuzhiyun if (s->state_out_len == 0) break;
798*4882a593Smuzhiyun *( (UChar*)(s->strm->next_out) ) = s->state_out_ch;
799*4882a593Smuzhiyun BZ_UPDATE_CRC ( s->calculatedBlockCRC, s->state_out_ch );
800*4882a593Smuzhiyun s->state_out_len--;
801*4882a593Smuzhiyun s->strm->next_out++;
802*4882a593Smuzhiyun s->strm->avail_out--;
803*4882a593Smuzhiyun s->strm->total_out_lo32++;
804*4882a593Smuzhiyun if (s->strm->total_out_lo32 == 0) s->strm->total_out_hi32++;
805*4882a593Smuzhiyun }
806*4882a593Smuzhiyun
807*4882a593Smuzhiyun /* can a new run be started? */
808*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) return;
809*4882a593Smuzhiyun
810*4882a593Smuzhiyun s->state_out_len = 1;
811*4882a593Smuzhiyun s->state_out_ch = s->k0;
812*4882a593Smuzhiyun BZ_GET_SMALL(k1); s->nblock_used++;
813*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
814*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
815*4882a593Smuzhiyun
816*4882a593Smuzhiyun s->state_out_len = 2;
817*4882a593Smuzhiyun BZ_GET_SMALL(k1); s->nblock_used++;
818*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
819*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
820*4882a593Smuzhiyun
821*4882a593Smuzhiyun s->state_out_len = 3;
822*4882a593Smuzhiyun BZ_GET_SMALL(k1); s->nblock_used++;
823*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1) continue;
824*4882a593Smuzhiyun if (k1 != s->k0) { s->k0 = k1; continue; };
825*4882a593Smuzhiyun
826*4882a593Smuzhiyun BZ_GET_SMALL(k1); s->nblock_used++;
827*4882a593Smuzhiyun s->state_out_len = ((Int32)k1) + 4;
828*4882a593Smuzhiyun BZ_GET_SMALL(s->k0); s->nblock_used++;
829*4882a593Smuzhiyun }
830*4882a593Smuzhiyun
831*4882a593Smuzhiyun }
832*4882a593Smuzhiyun }
833*4882a593Smuzhiyun
834*4882a593Smuzhiyun
835*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzDecompress)836*4882a593Smuzhiyun int BZ_API(BZ2_bzDecompress) ( bz_stream *strm )
837*4882a593Smuzhiyun {
838*4882a593Smuzhiyun DState* s;
839*4882a593Smuzhiyun if (strm == NULL) return BZ_PARAM_ERROR;
840*4882a593Smuzhiyun s = strm->state;
841*4882a593Smuzhiyun if (s == NULL) return BZ_PARAM_ERROR;
842*4882a593Smuzhiyun if (s->strm != strm) return BZ_PARAM_ERROR;
843*4882a593Smuzhiyun
844*4882a593Smuzhiyun while (True) {
845*4882a593Smuzhiyun #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
846*4882a593Smuzhiyun WATCHDOG_RESET();
847*4882a593Smuzhiyun #endif
848*4882a593Smuzhiyun if (s->state == BZ_X_IDLE) return BZ_SEQUENCE_ERROR;
849*4882a593Smuzhiyun if (s->state == BZ_X_OUTPUT) {
850*4882a593Smuzhiyun if (s->smallDecompress)
851*4882a593Smuzhiyun unRLE_obuf_to_output_SMALL ( s ); else
852*4882a593Smuzhiyun unRLE_obuf_to_output_FAST ( s );
853*4882a593Smuzhiyun if (s->nblock_used == s->save_nblock+1 && s->state_out_len == 0) {
854*4882a593Smuzhiyun BZ_FINALISE_CRC ( s->calculatedBlockCRC );
855*4882a593Smuzhiyun if (s->verbosity >= 3)
856*4882a593Smuzhiyun VPrintf2 ( " {0x%x, 0x%x}", s->storedBlockCRC,
857*4882a593Smuzhiyun s->calculatedBlockCRC );
858*4882a593Smuzhiyun if (s->verbosity >= 2) VPrintf0 ( "]" );
859*4882a593Smuzhiyun if (s->calculatedBlockCRC != s->storedBlockCRC)
860*4882a593Smuzhiyun return BZ_DATA_ERROR;
861*4882a593Smuzhiyun s->calculatedCombinedCRC
862*4882a593Smuzhiyun = (s->calculatedCombinedCRC << 1) |
863*4882a593Smuzhiyun (s->calculatedCombinedCRC >> 31);
864*4882a593Smuzhiyun s->calculatedCombinedCRC ^= s->calculatedBlockCRC;
865*4882a593Smuzhiyun s->state = BZ_X_BLKHDR_1;
866*4882a593Smuzhiyun } else {
867*4882a593Smuzhiyun return BZ_OK;
868*4882a593Smuzhiyun }
869*4882a593Smuzhiyun }
870*4882a593Smuzhiyun if (s->state >= BZ_X_MAGIC_1) {
871*4882a593Smuzhiyun Int32 r = BZ2_decompress ( s );
872*4882a593Smuzhiyun if (r == BZ_STREAM_END) {
873*4882a593Smuzhiyun if (s->verbosity >= 3)
874*4882a593Smuzhiyun VPrintf2 ( "\n combined CRCs: stored = 0x%x, computed = 0x%x",
875*4882a593Smuzhiyun s->storedCombinedCRC, s->calculatedCombinedCRC );
876*4882a593Smuzhiyun if (s->calculatedCombinedCRC != s->storedCombinedCRC)
877*4882a593Smuzhiyun return BZ_DATA_ERROR;
878*4882a593Smuzhiyun return r;
879*4882a593Smuzhiyun }
880*4882a593Smuzhiyun if (s->state != BZ_X_OUTPUT) return r;
881*4882a593Smuzhiyun }
882*4882a593Smuzhiyun }
883*4882a593Smuzhiyun
884*4882a593Smuzhiyun AssertH ( 0, 6001 );
885*4882a593Smuzhiyun
886*4882a593Smuzhiyun return 0; /*NOTREACHED*/
887*4882a593Smuzhiyun }
888*4882a593Smuzhiyun
889*4882a593Smuzhiyun
890*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzDecompressEnd)891*4882a593Smuzhiyun int BZ_API(BZ2_bzDecompressEnd) ( bz_stream *strm )
892*4882a593Smuzhiyun {
893*4882a593Smuzhiyun DState* s;
894*4882a593Smuzhiyun if (strm == NULL) return BZ_PARAM_ERROR;
895*4882a593Smuzhiyun s = strm->state;
896*4882a593Smuzhiyun if (s == NULL) return BZ_PARAM_ERROR;
897*4882a593Smuzhiyun if (s->strm != strm) return BZ_PARAM_ERROR;
898*4882a593Smuzhiyun
899*4882a593Smuzhiyun if (s->tt != NULL) BZFREE(s->tt);
900*4882a593Smuzhiyun if (s->ll16 != NULL) BZFREE(s->ll16);
901*4882a593Smuzhiyun if (s->ll4 != NULL) BZFREE(s->ll4);
902*4882a593Smuzhiyun
903*4882a593Smuzhiyun BZFREE(strm->state);
904*4882a593Smuzhiyun strm->state = NULL;
905*4882a593Smuzhiyun
906*4882a593Smuzhiyun return BZ_OK;
907*4882a593Smuzhiyun }
908*4882a593Smuzhiyun
909*4882a593Smuzhiyun
910*4882a593Smuzhiyun #ifndef BZ_NO_STDIO
911*4882a593Smuzhiyun /*---------------------------------------------------*/
912*4882a593Smuzhiyun /*--- File I/O stuff ---*/
913*4882a593Smuzhiyun /*---------------------------------------------------*/
914*4882a593Smuzhiyun
915*4882a593Smuzhiyun #define BZ_SETERR(eee) \
916*4882a593Smuzhiyun { \
917*4882a593Smuzhiyun if (bzerror != NULL) *bzerror = eee; \
918*4882a593Smuzhiyun if (bzf != NULL) bzf->lastErr = eee; \
919*4882a593Smuzhiyun }
920*4882a593Smuzhiyun
921*4882a593Smuzhiyun typedef
922*4882a593Smuzhiyun struct {
923*4882a593Smuzhiyun FILE* handle;
924*4882a593Smuzhiyun Char buf[BZ_MAX_UNUSED];
925*4882a593Smuzhiyun Int32 bufN;
926*4882a593Smuzhiyun Bool writing;
927*4882a593Smuzhiyun bz_stream strm;
928*4882a593Smuzhiyun Int32 lastErr;
929*4882a593Smuzhiyun Bool initialisedOk;
930*4882a593Smuzhiyun }
931*4882a593Smuzhiyun bzFile;
932*4882a593Smuzhiyun
933*4882a593Smuzhiyun
934*4882a593Smuzhiyun /*---------------------------------------------*/
myfeof(FILE * f)935*4882a593Smuzhiyun static Bool myfeof ( FILE* f )
936*4882a593Smuzhiyun {
937*4882a593Smuzhiyun Int32 c = fgetc ( f );
938*4882a593Smuzhiyun if (c == EOF) return True;
939*4882a593Smuzhiyun ungetc ( c, f );
940*4882a593Smuzhiyun return False;
941*4882a593Smuzhiyun }
942*4882a593Smuzhiyun
943*4882a593Smuzhiyun
944*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzWriteOpen)945*4882a593Smuzhiyun BZFILE* BZ_API(BZ2_bzWriteOpen)
946*4882a593Smuzhiyun ( int* bzerror,
947*4882a593Smuzhiyun FILE* f,
948*4882a593Smuzhiyun int blockSize100k,
949*4882a593Smuzhiyun int verbosity,
950*4882a593Smuzhiyun int workFactor )
951*4882a593Smuzhiyun {
952*4882a593Smuzhiyun Int32 ret;
953*4882a593Smuzhiyun bzFile* bzf = NULL;
954*4882a593Smuzhiyun
955*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
956*4882a593Smuzhiyun
957*4882a593Smuzhiyun if (f == NULL ||
958*4882a593Smuzhiyun (blockSize100k < 1 || blockSize100k > 9) ||
959*4882a593Smuzhiyun (workFactor < 0 || workFactor > 250) ||
960*4882a593Smuzhiyun (verbosity < 0 || verbosity > 4))
961*4882a593Smuzhiyun { BZ_SETERR(BZ_PARAM_ERROR); return NULL; };
962*4882a593Smuzhiyun
963*4882a593Smuzhiyun if (ferror(f))
964*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return NULL; };
965*4882a593Smuzhiyun
966*4882a593Smuzhiyun bzf = malloc ( sizeof(bzFile) );
967*4882a593Smuzhiyun if (bzf == NULL)
968*4882a593Smuzhiyun { BZ_SETERR(BZ_MEM_ERROR); return NULL; };
969*4882a593Smuzhiyun
970*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
971*4882a593Smuzhiyun bzf->initialisedOk = False;
972*4882a593Smuzhiyun bzf->bufN = 0;
973*4882a593Smuzhiyun bzf->handle = f;
974*4882a593Smuzhiyun bzf->writing = True;
975*4882a593Smuzhiyun bzf->strm.bzalloc = NULL;
976*4882a593Smuzhiyun bzf->strm.bzfree = NULL;
977*4882a593Smuzhiyun bzf->strm.opaque = NULL;
978*4882a593Smuzhiyun
979*4882a593Smuzhiyun if (workFactor == 0) workFactor = 30;
980*4882a593Smuzhiyun ret = BZ2_bzCompressInit ( &(bzf->strm), blockSize100k,
981*4882a593Smuzhiyun verbosity, workFactor );
982*4882a593Smuzhiyun if (ret != BZ_OK)
983*4882a593Smuzhiyun { BZ_SETERR(ret); free(bzf); return NULL; };
984*4882a593Smuzhiyun
985*4882a593Smuzhiyun bzf->strm.avail_in = 0;
986*4882a593Smuzhiyun bzf->initialisedOk = True;
987*4882a593Smuzhiyun return bzf;
988*4882a593Smuzhiyun }
989*4882a593Smuzhiyun
990*4882a593Smuzhiyun
991*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzWrite)992*4882a593Smuzhiyun void BZ_API(BZ2_bzWrite)
993*4882a593Smuzhiyun ( int* bzerror,
994*4882a593Smuzhiyun BZFILE* b,
995*4882a593Smuzhiyun void* buf,
996*4882a593Smuzhiyun int len )
997*4882a593Smuzhiyun {
998*4882a593Smuzhiyun Int32 n, n2, ret;
999*4882a593Smuzhiyun bzFile* bzf = (bzFile*)b;
1000*4882a593Smuzhiyun
1001*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
1002*4882a593Smuzhiyun if (bzf == NULL || buf == NULL || len < 0)
1003*4882a593Smuzhiyun { BZ_SETERR(BZ_PARAM_ERROR); return; };
1004*4882a593Smuzhiyun if (!(bzf->writing))
1005*4882a593Smuzhiyun { BZ_SETERR(BZ_SEQUENCE_ERROR); return; };
1006*4882a593Smuzhiyun if (ferror(bzf->handle))
1007*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return; };
1008*4882a593Smuzhiyun
1009*4882a593Smuzhiyun if (len == 0)
1010*4882a593Smuzhiyun { BZ_SETERR(BZ_OK); return; };
1011*4882a593Smuzhiyun
1012*4882a593Smuzhiyun bzf->strm.avail_in = len;
1013*4882a593Smuzhiyun bzf->strm.next_in = buf;
1014*4882a593Smuzhiyun
1015*4882a593Smuzhiyun while (True) {
1016*4882a593Smuzhiyun bzf->strm.avail_out = BZ_MAX_UNUSED;
1017*4882a593Smuzhiyun bzf->strm.next_out = bzf->buf;
1018*4882a593Smuzhiyun ret = BZ2_bzCompress ( &(bzf->strm), BZ_RUN );
1019*4882a593Smuzhiyun if (ret != BZ_RUN_OK)
1020*4882a593Smuzhiyun { BZ_SETERR(ret); return; };
1021*4882a593Smuzhiyun
1022*4882a593Smuzhiyun if (bzf->strm.avail_out < BZ_MAX_UNUSED) {
1023*4882a593Smuzhiyun n = BZ_MAX_UNUSED - bzf->strm.avail_out;
1024*4882a593Smuzhiyun n2 = fwrite ( (void*)(bzf->buf), sizeof(UChar),
1025*4882a593Smuzhiyun n, bzf->handle );
1026*4882a593Smuzhiyun if (n != n2 || ferror(bzf->handle))
1027*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return; };
1028*4882a593Smuzhiyun }
1029*4882a593Smuzhiyun
1030*4882a593Smuzhiyun if (bzf->strm.avail_in == 0)
1031*4882a593Smuzhiyun { BZ_SETERR(BZ_OK); return; };
1032*4882a593Smuzhiyun }
1033*4882a593Smuzhiyun }
1034*4882a593Smuzhiyun
1035*4882a593Smuzhiyun
1036*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzWriteClose)1037*4882a593Smuzhiyun void BZ_API(BZ2_bzWriteClose)
1038*4882a593Smuzhiyun ( int* bzerror,
1039*4882a593Smuzhiyun BZFILE* b,
1040*4882a593Smuzhiyun int abandon,
1041*4882a593Smuzhiyun unsigned int* nbytes_in,
1042*4882a593Smuzhiyun unsigned int* nbytes_out )
1043*4882a593Smuzhiyun {
1044*4882a593Smuzhiyun BZ2_bzWriteClose64 ( bzerror, b, abandon,
1045*4882a593Smuzhiyun nbytes_in, NULL, nbytes_out, NULL );
1046*4882a593Smuzhiyun }
1047*4882a593Smuzhiyun
1048*4882a593Smuzhiyun
BZ_API(BZ2_bzWriteClose64)1049*4882a593Smuzhiyun void BZ_API(BZ2_bzWriteClose64)
1050*4882a593Smuzhiyun ( int* bzerror,
1051*4882a593Smuzhiyun BZFILE* b,
1052*4882a593Smuzhiyun int abandon,
1053*4882a593Smuzhiyun unsigned int* nbytes_in_lo32,
1054*4882a593Smuzhiyun unsigned int* nbytes_in_hi32,
1055*4882a593Smuzhiyun unsigned int* nbytes_out_lo32,
1056*4882a593Smuzhiyun unsigned int* nbytes_out_hi32 )
1057*4882a593Smuzhiyun {
1058*4882a593Smuzhiyun Int32 n, n2, ret;
1059*4882a593Smuzhiyun bzFile* bzf = (bzFile*)b;
1060*4882a593Smuzhiyun
1061*4882a593Smuzhiyun if (bzf == NULL)
1062*4882a593Smuzhiyun { BZ_SETERR(BZ_OK); return; };
1063*4882a593Smuzhiyun if (!(bzf->writing))
1064*4882a593Smuzhiyun { BZ_SETERR(BZ_SEQUENCE_ERROR); return; };
1065*4882a593Smuzhiyun if (ferror(bzf->handle))
1066*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return; };
1067*4882a593Smuzhiyun
1068*4882a593Smuzhiyun if (nbytes_in_lo32 != NULL) *nbytes_in_lo32 = 0;
1069*4882a593Smuzhiyun if (nbytes_in_hi32 != NULL) *nbytes_in_hi32 = 0;
1070*4882a593Smuzhiyun if (nbytes_out_lo32 != NULL) *nbytes_out_lo32 = 0;
1071*4882a593Smuzhiyun if (nbytes_out_hi32 != NULL) *nbytes_out_hi32 = 0;
1072*4882a593Smuzhiyun
1073*4882a593Smuzhiyun if ((!abandon) && bzf->lastErr == BZ_OK) {
1074*4882a593Smuzhiyun while (True) {
1075*4882a593Smuzhiyun bzf->strm.avail_out = BZ_MAX_UNUSED;
1076*4882a593Smuzhiyun bzf->strm.next_out = bzf->buf;
1077*4882a593Smuzhiyun ret = BZ2_bzCompress ( &(bzf->strm), BZ_FINISH );
1078*4882a593Smuzhiyun if (ret != BZ_FINISH_OK && ret != BZ_STREAM_END)
1079*4882a593Smuzhiyun { BZ_SETERR(ret); return; };
1080*4882a593Smuzhiyun
1081*4882a593Smuzhiyun if (bzf->strm.avail_out < BZ_MAX_UNUSED) {
1082*4882a593Smuzhiyun n = BZ_MAX_UNUSED - bzf->strm.avail_out;
1083*4882a593Smuzhiyun n2 = fwrite ( (void*)(bzf->buf), sizeof(UChar),
1084*4882a593Smuzhiyun n, bzf->handle );
1085*4882a593Smuzhiyun if (n != n2 || ferror(bzf->handle))
1086*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return; };
1087*4882a593Smuzhiyun }
1088*4882a593Smuzhiyun
1089*4882a593Smuzhiyun if (ret == BZ_STREAM_END) break;
1090*4882a593Smuzhiyun }
1091*4882a593Smuzhiyun }
1092*4882a593Smuzhiyun
1093*4882a593Smuzhiyun if ( !abandon && !ferror ( bzf->handle ) ) {
1094*4882a593Smuzhiyun fflush ( bzf->handle );
1095*4882a593Smuzhiyun if (ferror(bzf->handle))
1096*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return; };
1097*4882a593Smuzhiyun }
1098*4882a593Smuzhiyun
1099*4882a593Smuzhiyun if (nbytes_in_lo32 != NULL)
1100*4882a593Smuzhiyun *nbytes_in_lo32 = bzf->strm.total_in_lo32;
1101*4882a593Smuzhiyun if (nbytes_in_hi32 != NULL)
1102*4882a593Smuzhiyun *nbytes_in_hi32 = bzf->strm.total_in_hi32;
1103*4882a593Smuzhiyun if (nbytes_out_lo32 != NULL)
1104*4882a593Smuzhiyun *nbytes_out_lo32 = bzf->strm.total_out_lo32;
1105*4882a593Smuzhiyun if (nbytes_out_hi32 != NULL)
1106*4882a593Smuzhiyun *nbytes_out_hi32 = bzf->strm.total_out_hi32;
1107*4882a593Smuzhiyun
1108*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
1109*4882a593Smuzhiyun BZ2_bzCompressEnd ( &(bzf->strm) );
1110*4882a593Smuzhiyun free ( bzf );
1111*4882a593Smuzhiyun }
1112*4882a593Smuzhiyun
1113*4882a593Smuzhiyun
1114*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzReadOpen)1115*4882a593Smuzhiyun BZFILE* BZ_API(BZ2_bzReadOpen)
1116*4882a593Smuzhiyun ( int* bzerror,
1117*4882a593Smuzhiyun FILE* f,
1118*4882a593Smuzhiyun int verbosity,
1119*4882a593Smuzhiyun int small,
1120*4882a593Smuzhiyun void* unused,
1121*4882a593Smuzhiyun int nUnused )
1122*4882a593Smuzhiyun {
1123*4882a593Smuzhiyun bzFile* bzf = NULL;
1124*4882a593Smuzhiyun int ret;
1125*4882a593Smuzhiyun
1126*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
1127*4882a593Smuzhiyun
1128*4882a593Smuzhiyun if (f == NULL ||
1129*4882a593Smuzhiyun (small != 0 && small != 1) ||
1130*4882a593Smuzhiyun (verbosity < 0 || verbosity > 4) ||
1131*4882a593Smuzhiyun (unused == NULL && nUnused != 0) ||
1132*4882a593Smuzhiyun (unused != NULL && (nUnused < 0 || nUnused > BZ_MAX_UNUSED)))
1133*4882a593Smuzhiyun { BZ_SETERR(BZ_PARAM_ERROR); return NULL; };
1134*4882a593Smuzhiyun
1135*4882a593Smuzhiyun if (ferror(f))
1136*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return NULL; };
1137*4882a593Smuzhiyun
1138*4882a593Smuzhiyun bzf = malloc ( sizeof(bzFile) );
1139*4882a593Smuzhiyun if (bzf == NULL)
1140*4882a593Smuzhiyun { BZ_SETERR(BZ_MEM_ERROR); return NULL; };
1141*4882a593Smuzhiyun
1142*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
1143*4882a593Smuzhiyun
1144*4882a593Smuzhiyun bzf->initialisedOk = False;
1145*4882a593Smuzhiyun bzf->handle = f;
1146*4882a593Smuzhiyun bzf->bufN = 0;
1147*4882a593Smuzhiyun bzf->writing = False;
1148*4882a593Smuzhiyun bzf->strm.bzalloc = NULL;
1149*4882a593Smuzhiyun bzf->strm.bzfree = NULL;
1150*4882a593Smuzhiyun bzf->strm.opaque = NULL;
1151*4882a593Smuzhiyun
1152*4882a593Smuzhiyun while (nUnused > 0) {
1153*4882a593Smuzhiyun bzf->buf[bzf->bufN] = *((UChar*)(unused)); bzf->bufN++;
1154*4882a593Smuzhiyun unused = ((void*)( 1 + ((UChar*)(unused)) ));
1155*4882a593Smuzhiyun nUnused--;
1156*4882a593Smuzhiyun }
1157*4882a593Smuzhiyun
1158*4882a593Smuzhiyun ret = BZ2_bzDecompressInit ( &(bzf->strm), verbosity, small );
1159*4882a593Smuzhiyun if (ret != BZ_OK)
1160*4882a593Smuzhiyun { BZ_SETERR(ret); free(bzf); return NULL; };
1161*4882a593Smuzhiyun
1162*4882a593Smuzhiyun bzf->strm.avail_in = bzf->bufN;
1163*4882a593Smuzhiyun bzf->strm.next_in = bzf->buf;
1164*4882a593Smuzhiyun
1165*4882a593Smuzhiyun bzf->initialisedOk = True;
1166*4882a593Smuzhiyun return bzf;
1167*4882a593Smuzhiyun }
1168*4882a593Smuzhiyun
1169*4882a593Smuzhiyun
1170*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzReadClose)1171*4882a593Smuzhiyun void BZ_API(BZ2_bzReadClose) ( int *bzerror, BZFILE *b )
1172*4882a593Smuzhiyun {
1173*4882a593Smuzhiyun bzFile* bzf = (bzFile*)b;
1174*4882a593Smuzhiyun
1175*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
1176*4882a593Smuzhiyun if (bzf == NULL)
1177*4882a593Smuzhiyun { BZ_SETERR(BZ_OK); return; };
1178*4882a593Smuzhiyun
1179*4882a593Smuzhiyun if (bzf->writing)
1180*4882a593Smuzhiyun { BZ_SETERR(BZ_SEQUENCE_ERROR); return; };
1181*4882a593Smuzhiyun
1182*4882a593Smuzhiyun if (bzf->initialisedOk)
1183*4882a593Smuzhiyun (void)BZ2_bzDecompressEnd ( &(bzf->strm) );
1184*4882a593Smuzhiyun free ( bzf );
1185*4882a593Smuzhiyun }
1186*4882a593Smuzhiyun
1187*4882a593Smuzhiyun
1188*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzRead)1189*4882a593Smuzhiyun int BZ_API(BZ2_bzRead)
1190*4882a593Smuzhiyun ( int* bzerror,
1191*4882a593Smuzhiyun BZFILE* b,
1192*4882a593Smuzhiyun void* buf,
1193*4882a593Smuzhiyun int len )
1194*4882a593Smuzhiyun {
1195*4882a593Smuzhiyun Int32 n, ret;
1196*4882a593Smuzhiyun bzFile* bzf = (bzFile*)b;
1197*4882a593Smuzhiyun
1198*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
1199*4882a593Smuzhiyun
1200*4882a593Smuzhiyun if (bzf == NULL || buf == NULL || len < 0)
1201*4882a593Smuzhiyun { BZ_SETERR(BZ_PARAM_ERROR); return 0; };
1202*4882a593Smuzhiyun
1203*4882a593Smuzhiyun if (bzf->writing)
1204*4882a593Smuzhiyun { BZ_SETERR(BZ_SEQUENCE_ERROR); return 0; };
1205*4882a593Smuzhiyun
1206*4882a593Smuzhiyun if (len == 0)
1207*4882a593Smuzhiyun { BZ_SETERR(BZ_OK); return 0; };
1208*4882a593Smuzhiyun
1209*4882a593Smuzhiyun bzf->strm.avail_out = len;
1210*4882a593Smuzhiyun bzf->strm.next_out = buf;
1211*4882a593Smuzhiyun
1212*4882a593Smuzhiyun while (True) {
1213*4882a593Smuzhiyun
1214*4882a593Smuzhiyun if (ferror(bzf->handle))
1215*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return 0; };
1216*4882a593Smuzhiyun
1217*4882a593Smuzhiyun if (bzf->strm.avail_in == 0 && !myfeof(bzf->handle)) {
1218*4882a593Smuzhiyun n = fread ( bzf->buf, sizeof(UChar),
1219*4882a593Smuzhiyun BZ_MAX_UNUSED, bzf->handle );
1220*4882a593Smuzhiyun if (ferror(bzf->handle))
1221*4882a593Smuzhiyun { BZ_SETERR(BZ_IO_ERROR); return 0; };
1222*4882a593Smuzhiyun bzf->bufN = n;
1223*4882a593Smuzhiyun bzf->strm.avail_in = bzf->bufN;
1224*4882a593Smuzhiyun bzf->strm.next_in = bzf->buf;
1225*4882a593Smuzhiyun }
1226*4882a593Smuzhiyun
1227*4882a593Smuzhiyun ret = BZ2_bzDecompress ( &(bzf->strm) );
1228*4882a593Smuzhiyun
1229*4882a593Smuzhiyun if (ret != BZ_OK && ret != BZ_STREAM_END)
1230*4882a593Smuzhiyun { BZ_SETERR(ret); return 0; };
1231*4882a593Smuzhiyun
1232*4882a593Smuzhiyun if (ret == BZ_OK && myfeof(bzf->handle) &&
1233*4882a593Smuzhiyun bzf->strm.avail_in == 0 && bzf->strm.avail_out > 0)
1234*4882a593Smuzhiyun { BZ_SETERR(BZ_UNEXPECTED_EOF); return 0; };
1235*4882a593Smuzhiyun
1236*4882a593Smuzhiyun if (ret == BZ_STREAM_END)
1237*4882a593Smuzhiyun { BZ_SETERR(BZ_STREAM_END);
1238*4882a593Smuzhiyun return len - bzf->strm.avail_out; };
1239*4882a593Smuzhiyun if (bzf->strm.avail_out == 0)
1240*4882a593Smuzhiyun { BZ_SETERR(BZ_OK); return len; };
1241*4882a593Smuzhiyun
1242*4882a593Smuzhiyun }
1243*4882a593Smuzhiyun
1244*4882a593Smuzhiyun return 0; /*not reached*/
1245*4882a593Smuzhiyun }
1246*4882a593Smuzhiyun
1247*4882a593Smuzhiyun
1248*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzReadGetUnused)1249*4882a593Smuzhiyun void BZ_API(BZ2_bzReadGetUnused)
1250*4882a593Smuzhiyun ( int* bzerror,
1251*4882a593Smuzhiyun BZFILE* b,
1252*4882a593Smuzhiyun void** unused,
1253*4882a593Smuzhiyun int* nUnused )
1254*4882a593Smuzhiyun {
1255*4882a593Smuzhiyun bzFile* bzf = (bzFile*)b;
1256*4882a593Smuzhiyun if (bzf == NULL)
1257*4882a593Smuzhiyun { BZ_SETERR(BZ_PARAM_ERROR); return; };
1258*4882a593Smuzhiyun if (bzf->lastErr != BZ_STREAM_END)
1259*4882a593Smuzhiyun { BZ_SETERR(BZ_SEQUENCE_ERROR); return; };
1260*4882a593Smuzhiyun if (unused == NULL || nUnused == NULL)
1261*4882a593Smuzhiyun { BZ_SETERR(BZ_PARAM_ERROR); return; };
1262*4882a593Smuzhiyun
1263*4882a593Smuzhiyun BZ_SETERR(BZ_OK);
1264*4882a593Smuzhiyun *nUnused = bzf->strm.avail_in;
1265*4882a593Smuzhiyun *unused = bzf->strm.next_in;
1266*4882a593Smuzhiyun }
1267*4882a593Smuzhiyun #endif
1268*4882a593Smuzhiyun
1269*4882a593Smuzhiyun
1270*4882a593Smuzhiyun /*---------------------------------------------------*/
1271*4882a593Smuzhiyun /*--- Misc convenience stuff ---*/
1272*4882a593Smuzhiyun /*---------------------------------------------------*/
1273*4882a593Smuzhiyun #ifndef BZ_NO_COMPRESS
1274*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzBuffToBuffCompress)1275*4882a593Smuzhiyun int BZ_API(BZ2_bzBuffToBuffCompress)
1276*4882a593Smuzhiyun ( char* dest,
1277*4882a593Smuzhiyun unsigned int* destLen,
1278*4882a593Smuzhiyun char* source,
1279*4882a593Smuzhiyun unsigned int sourceLen,
1280*4882a593Smuzhiyun int blockSize100k,
1281*4882a593Smuzhiyun int verbosity,
1282*4882a593Smuzhiyun int workFactor )
1283*4882a593Smuzhiyun {
1284*4882a593Smuzhiyun bz_stream strm;
1285*4882a593Smuzhiyun int ret;
1286*4882a593Smuzhiyun
1287*4882a593Smuzhiyun if (dest == NULL || destLen == NULL ||
1288*4882a593Smuzhiyun source == NULL ||
1289*4882a593Smuzhiyun blockSize100k < 1 || blockSize100k > 9 ||
1290*4882a593Smuzhiyun verbosity < 0 || verbosity > 4 ||
1291*4882a593Smuzhiyun workFactor < 0 || workFactor > 250)
1292*4882a593Smuzhiyun return BZ_PARAM_ERROR;
1293*4882a593Smuzhiyun
1294*4882a593Smuzhiyun if (workFactor == 0) workFactor = 30;
1295*4882a593Smuzhiyun strm.bzalloc = NULL;
1296*4882a593Smuzhiyun strm.bzfree = NULL;
1297*4882a593Smuzhiyun strm.opaque = NULL;
1298*4882a593Smuzhiyun ret = BZ2_bzCompressInit ( &strm, blockSize100k,
1299*4882a593Smuzhiyun verbosity, workFactor );
1300*4882a593Smuzhiyun if (ret != BZ_OK) return ret;
1301*4882a593Smuzhiyun
1302*4882a593Smuzhiyun strm.next_in = source;
1303*4882a593Smuzhiyun strm.next_out = dest;
1304*4882a593Smuzhiyun strm.avail_in = sourceLen;
1305*4882a593Smuzhiyun strm.avail_out = *destLen;
1306*4882a593Smuzhiyun
1307*4882a593Smuzhiyun ret = BZ2_bzCompress ( &strm, BZ_FINISH );
1308*4882a593Smuzhiyun if (ret == BZ_FINISH_OK) goto output_overflow;
1309*4882a593Smuzhiyun if (ret != BZ_STREAM_END) goto errhandler;
1310*4882a593Smuzhiyun
1311*4882a593Smuzhiyun /* normal termination */
1312*4882a593Smuzhiyun *destLen -= strm.avail_out;
1313*4882a593Smuzhiyun BZ2_bzCompressEnd ( &strm );
1314*4882a593Smuzhiyun return BZ_OK;
1315*4882a593Smuzhiyun
1316*4882a593Smuzhiyun output_overflow:
1317*4882a593Smuzhiyun BZ2_bzCompressEnd ( &strm );
1318*4882a593Smuzhiyun return BZ_OUTBUFF_FULL;
1319*4882a593Smuzhiyun
1320*4882a593Smuzhiyun errhandler:
1321*4882a593Smuzhiyun BZ2_bzCompressEnd ( &strm );
1322*4882a593Smuzhiyun return ret;
1323*4882a593Smuzhiyun }
1324*4882a593Smuzhiyun #endif /* BZ_NO_COMPRESS */
1325*4882a593Smuzhiyun
1326*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzBuffToBuffDecompress)1327*4882a593Smuzhiyun int BZ_API(BZ2_bzBuffToBuffDecompress)
1328*4882a593Smuzhiyun ( char* dest,
1329*4882a593Smuzhiyun unsigned int* destLen,
1330*4882a593Smuzhiyun char* source,
1331*4882a593Smuzhiyun unsigned int sourceLen,
1332*4882a593Smuzhiyun int small,
1333*4882a593Smuzhiyun int verbosity )
1334*4882a593Smuzhiyun {
1335*4882a593Smuzhiyun bz_stream strm;
1336*4882a593Smuzhiyun int ret;
1337*4882a593Smuzhiyun
1338*4882a593Smuzhiyun if (destLen == NULL || source == NULL)
1339*4882a593Smuzhiyun return BZ_PARAM_ERROR;
1340*4882a593Smuzhiyun
1341*4882a593Smuzhiyun strm.bzalloc = NULL;
1342*4882a593Smuzhiyun strm.bzfree = NULL;
1343*4882a593Smuzhiyun strm.opaque = NULL;
1344*4882a593Smuzhiyun ret = BZ2_bzDecompressInit ( &strm, verbosity, small );
1345*4882a593Smuzhiyun if (ret != BZ_OK) return ret;
1346*4882a593Smuzhiyun
1347*4882a593Smuzhiyun strm.next_in = source;
1348*4882a593Smuzhiyun strm.next_out = dest;
1349*4882a593Smuzhiyun strm.avail_in = sourceLen;
1350*4882a593Smuzhiyun strm.avail_out = *destLen;
1351*4882a593Smuzhiyun
1352*4882a593Smuzhiyun ret = BZ2_bzDecompress ( &strm );
1353*4882a593Smuzhiyun *destLen -= strm.avail_out;
1354*4882a593Smuzhiyun if (ret == BZ_OK) goto output_overflow_or_eof;
1355*4882a593Smuzhiyun if (ret != BZ_STREAM_END) goto errhandler;
1356*4882a593Smuzhiyun
1357*4882a593Smuzhiyun /* normal termination */
1358*4882a593Smuzhiyun BZ2_bzDecompressEnd ( &strm );
1359*4882a593Smuzhiyun return BZ_OK;
1360*4882a593Smuzhiyun
1361*4882a593Smuzhiyun output_overflow_or_eof:
1362*4882a593Smuzhiyun if (strm.avail_out > 0) {
1363*4882a593Smuzhiyun BZ2_bzDecompressEnd ( &strm );
1364*4882a593Smuzhiyun return BZ_UNEXPECTED_EOF;
1365*4882a593Smuzhiyun } else {
1366*4882a593Smuzhiyun BZ2_bzDecompressEnd ( &strm );
1367*4882a593Smuzhiyun return BZ_OUTBUFF_FULL;
1368*4882a593Smuzhiyun };
1369*4882a593Smuzhiyun
1370*4882a593Smuzhiyun errhandler:
1371*4882a593Smuzhiyun BZ2_bzDecompressEnd ( &strm );
1372*4882a593Smuzhiyun return ret;
1373*4882a593Smuzhiyun }
1374*4882a593Smuzhiyun
1375*4882a593Smuzhiyun
1376*4882a593Smuzhiyun /*---------------------------------------------------*/
1377*4882a593Smuzhiyun /*--
1378*4882a593Smuzhiyun Code contributed by Yoshioka Tsuneo
1379*4882a593Smuzhiyun (QWF00133@niftyserve.or.jp/tsuneo-y@is.aist-nara.ac.jp),
1380*4882a593Smuzhiyun to support better zlib compatibility.
1381*4882a593Smuzhiyun This code is not _officially_ part of libbzip2 (yet);
1382*4882a593Smuzhiyun I haven't tested it, documented it, or considered the
1383*4882a593Smuzhiyun threading-safeness of it.
1384*4882a593Smuzhiyun If this code breaks, please contact both Yoshioka and me.
1385*4882a593Smuzhiyun --*/
1386*4882a593Smuzhiyun /*---------------------------------------------------*/
1387*4882a593Smuzhiyun
1388*4882a593Smuzhiyun /*---------------------------------------------------*/
1389*4882a593Smuzhiyun /*--
1390*4882a593Smuzhiyun return version like "0.9.0c".
1391*4882a593Smuzhiyun --*/
BZ_API(BZ2_bzlibVersion)1392*4882a593Smuzhiyun const char * BZ_API(BZ2_bzlibVersion)(void)
1393*4882a593Smuzhiyun {
1394*4882a593Smuzhiyun return BZ_VERSION;
1395*4882a593Smuzhiyun }
1396*4882a593Smuzhiyun
1397*4882a593Smuzhiyun
1398*4882a593Smuzhiyun #ifndef BZ_NO_STDIO
1399*4882a593Smuzhiyun /*---------------------------------------------------*/
1400*4882a593Smuzhiyun
1401*4882a593Smuzhiyun #if defined(_WIN32) || defined(OS2) || defined(MSDOS)
1402*4882a593Smuzhiyun # include <fcntl.h>
1403*4882a593Smuzhiyun # include <io.h>
1404*4882a593Smuzhiyun # define SET_BINARY_MODE(file) setmode(fileno(file),O_BINARY)
1405*4882a593Smuzhiyun #else
1406*4882a593Smuzhiyun # define SET_BINARY_MODE(file)
1407*4882a593Smuzhiyun #endif
1408*4882a593Smuzhiyun static
bzopen_or_bzdopen(const char * path,int fd,const char * mode,int open_mode)1409*4882a593Smuzhiyun BZFILE * bzopen_or_bzdopen
1410*4882a593Smuzhiyun ( const char *path, /* no use when bzdopen */
1411*4882a593Smuzhiyun int fd, /* no use when bzdopen */
1412*4882a593Smuzhiyun const char *mode,
1413*4882a593Smuzhiyun int open_mode) /* bzopen: 0, bzdopen:1 */
1414*4882a593Smuzhiyun {
1415*4882a593Smuzhiyun int bzerr;
1416*4882a593Smuzhiyun char unused[BZ_MAX_UNUSED];
1417*4882a593Smuzhiyun int blockSize100k = 9;
1418*4882a593Smuzhiyun int writing = 0;
1419*4882a593Smuzhiyun char mode2[10] = "";
1420*4882a593Smuzhiyun FILE *fp = NULL;
1421*4882a593Smuzhiyun BZFILE *bzfp = NULL;
1422*4882a593Smuzhiyun int verbosity = 0;
1423*4882a593Smuzhiyun int workFactor = 30;
1424*4882a593Smuzhiyun int smallMode = 0;
1425*4882a593Smuzhiyun int nUnused = 0;
1426*4882a593Smuzhiyun
1427*4882a593Smuzhiyun if (mode == NULL) return NULL;
1428*4882a593Smuzhiyun while (*mode) {
1429*4882a593Smuzhiyun switch (*mode) {
1430*4882a593Smuzhiyun case 'r':
1431*4882a593Smuzhiyun writing = 0; break;
1432*4882a593Smuzhiyun case 'w':
1433*4882a593Smuzhiyun writing = 1; break;
1434*4882a593Smuzhiyun case 's':
1435*4882a593Smuzhiyun smallMode = 1; break;
1436*4882a593Smuzhiyun default:
1437*4882a593Smuzhiyun if (isdigit((int)(*mode))) {
1438*4882a593Smuzhiyun blockSize100k = *mode-BZ_HDR_0;
1439*4882a593Smuzhiyun }
1440*4882a593Smuzhiyun }
1441*4882a593Smuzhiyun mode++;
1442*4882a593Smuzhiyun }
1443*4882a593Smuzhiyun strcat(mode2, writing ? "w" : "r" );
1444*4882a593Smuzhiyun strcat(mode2,"b"); /* binary mode */
1445*4882a593Smuzhiyun
1446*4882a593Smuzhiyun if (open_mode==0) {
1447*4882a593Smuzhiyun if (path==NULL || strcmp(path,"")==0) {
1448*4882a593Smuzhiyun fp = (writing ? stdout : stdin);
1449*4882a593Smuzhiyun SET_BINARY_MODE(fp);
1450*4882a593Smuzhiyun } else {
1451*4882a593Smuzhiyun fp = fopen(path,mode2);
1452*4882a593Smuzhiyun }
1453*4882a593Smuzhiyun } else {
1454*4882a593Smuzhiyun #ifdef BZ_STRICT_ANSI
1455*4882a593Smuzhiyun fp = NULL;
1456*4882a593Smuzhiyun #else
1457*4882a593Smuzhiyun fp = fdopen(fd,mode2);
1458*4882a593Smuzhiyun #endif
1459*4882a593Smuzhiyun }
1460*4882a593Smuzhiyun if (fp == NULL) return NULL;
1461*4882a593Smuzhiyun
1462*4882a593Smuzhiyun if (writing) {
1463*4882a593Smuzhiyun /* Guard against total chaos and anarchy -- JRS */
1464*4882a593Smuzhiyun if (blockSize100k < 1) blockSize100k = 1;
1465*4882a593Smuzhiyun if (blockSize100k > 9) blockSize100k = 9;
1466*4882a593Smuzhiyun bzfp = BZ2_bzWriteOpen(&bzerr,fp,blockSize100k,
1467*4882a593Smuzhiyun verbosity,workFactor);
1468*4882a593Smuzhiyun } else {
1469*4882a593Smuzhiyun bzfp = BZ2_bzReadOpen(&bzerr,fp,verbosity,smallMode,
1470*4882a593Smuzhiyun unused,nUnused);
1471*4882a593Smuzhiyun }
1472*4882a593Smuzhiyun if (bzfp == NULL) {
1473*4882a593Smuzhiyun if (fp != stdin && fp != stdout) fclose(fp);
1474*4882a593Smuzhiyun return NULL;
1475*4882a593Smuzhiyun }
1476*4882a593Smuzhiyun return bzfp;
1477*4882a593Smuzhiyun }
1478*4882a593Smuzhiyun
1479*4882a593Smuzhiyun
1480*4882a593Smuzhiyun /*---------------------------------------------------*/
1481*4882a593Smuzhiyun /*--
1482*4882a593Smuzhiyun open file for read or write.
1483*4882a593Smuzhiyun ex) bzopen("file","w9")
1484*4882a593Smuzhiyun case path="" or NULL => use stdin or stdout.
1485*4882a593Smuzhiyun --*/
BZ_API(BZ2_bzopen)1486*4882a593Smuzhiyun BZFILE * BZ_API(BZ2_bzopen)
1487*4882a593Smuzhiyun ( const char *path,
1488*4882a593Smuzhiyun const char *mode )
1489*4882a593Smuzhiyun {
1490*4882a593Smuzhiyun return bzopen_or_bzdopen(path,-1,mode,/*bzopen*/0);
1491*4882a593Smuzhiyun }
1492*4882a593Smuzhiyun
1493*4882a593Smuzhiyun
1494*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzdopen)1495*4882a593Smuzhiyun BZFILE * BZ_API(BZ2_bzdopen)
1496*4882a593Smuzhiyun ( int fd,
1497*4882a593Smuzhiyun const char *mode )
1498*4882a593Smuzhiyun {
1499*4882a593Smuzhiyun return bzopen_or_bzdopen(NULL,fd,mode,/*bzdopen*/1);
1500*4882a593Smuzhiyun }
1501*4882a593Smuzhiyun
1502*4882a593Smuzhiyun
1503*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzread)1504*4882a593Smuzhiyun int BZ_API(BZ2_bzread) (BZFILE* b, void* buf, int len )
1505*4882a593Smuzhiyun {
1506*4882a593Smuzhiyun int bzerr, nread;
1507*4882a593Smuzhiyun if (((bzFile*)b)->lastErr == BZ_STREAM_END) return 0;
1508*4882a593Smuzhiyun nread = BZ2_bzRead(&bzerr,b,buf,len);
1509*4882a593Smuzhiyun if (bzerr == BZ_OK || bzerr == BZ_STREAM_END) {
1510*4882a593Smuzhiyun return nread;
1511*4882a593Smuzhiyun } else {
1512*4882a593Smuzhiyun return -1;
1513*4882a593Smuzhiyun }
1514*4882a593Smuzhiyun }
1515*4882a593Smuzhiyun
1516*4882a593Smuzhiyun
1517*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzwrite)1518*4882a593Smuzhiyun int BZ_API(BZ2_bzwrite) (BZFILE* b, void* buf, int len )
1519*4882a593Smuzhiyun {
1520*4882a593Smuzhiyun int bzerr;
1521*4882a593Smuzhiyun
1522*4882a593Smuzhiyun BZ2_bzWrite(&bzerr,b,buf,len);
1523*4882a593Smuzhiyun if(bzerr == BZ_OK){
1524*4882a593Smuzhiyun return len;
1525*4882a593Smuzhiyun }else{
1526*4882a593Smuzhiyun return -1;
1527*4882a593Smuzhiyun }
1528*4882a593Smuzhiyun }
1529*4882a593Smuzhiyun
1530*4882a593Smuzhiyun
1531*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzflush)1532*4882a593Smuzhiyun int BZ_API(BZ2_bzflush) (BZFILE *b)
1533*4882a593Smuzhiyun {
1534*4882a593Smuzhiyun /* do nothing now... */
1535*4882a593Smuzhiyun return 0;
1536*4882a593Smuzhiyun }
1537*4882a593Smuzhiyun
1538*4882a593Smuzhiyun
1539*4882a593Smuzhiyun /*---------------------------------------------------*/
BZ_API(BZ2_bzclose)1540*4882a593Smuzhiyun void BZ_API(BZ2_bzclose) (BZFILE* b)
1541*4882a593Smuzhiyun {
1542*4882a593Smuzhiyun int bzerr;
1543*4882a593Smuzhiyun FILE *fp = ((bzFile *)b)->handle;
1544*4882a593Smuzhiyun
1545*4882a593Smuzhiyun if (b==NULL) {return;}
1546*4882a593Smuzhiyun if(((bzFile*)b)->writing){
1547*4882a593Smuzhiyun BZ2_bzWriteClose(&bzerr,b,0,NULL,NULL);
1548*4882a593Smuzhiyun if(bzerr != BZ_OK){
1549*4882a593Smuzhiyun BZ2_bzWriteClose(NULL,b,1,NULL,NULL);
1550*4882a593Smuzhiyun }
1551*4882a593Smuzhiyun }else{
1552*4882a593Smuzhiyun BZ2_bzReadClose(&bzerr,b);
1553*4882a593Smuzhiyun }
1554*4882a593Smuzhiyun if(fp!=stdin && fp!=stdout){
1555*4882a593Smuzhiyun fclose(fp);
1556*4882a593Smuzhiyun }
1557*4882a593Smuzhiyun }
1558*4882a593Smuzhiyun
1559*4882a593Smuzhiyun
1560*4882a593Smuzhiyun /*---------------------------------------------------*/
1561*4882a593Smuzhiyun /*--
1562*4882a593Smuzhiyun return last error code
1563*4882a593Smuzhiyun --*/
1564*4882a593Smuzhiyun static char *bzerrorstrings[] = {
1565*4882a593Smuzhiyun "OK"
1566*4882a593Smuzhiyun ,"SEQUENCE_ERROR"
1567*4882a593Smuzhiyun ,"PARAM_ERROR"
1568*4882a593Smuzhiyun ,"MEM_ERROR"
1569*4882a593Smuzhiyun ,"DATA_ERROR"
1570*4882a593Smuzhiyun ,"DATA_ERROR_MAGIC"
1571*4882a593Smuzhiyun ,"IO_ERROR"
1572*4882a593Smuzhiyun ,"UNEXPECTED_EOF"
1573*4882a593Smuzhiyun ,"OUTBUFF_FULL"
1574*4882a593Smuzhiyun ,"CONFIG_ERROR"
1575*4882a593Smuzhiyun ,"???" /* for future */
1576*4882a593Smuzhiyun ,"???" /* for future */
1577*4882a593Smuzhiyun ,"???" /* for future */
1578*4882a593Smuzhiyun ,"???" /* for future */
1579*4882a593Smuzhiyun ,"???" /* for future */
1580*4882a593Smuzhiyun ,"???" /* for future */
1581*4882a593Smuzhiyun };
1582*4882a593Smuzhiyun
1583*4882a593Smuzhiyun
BZ_API(BZ2_bzerror)1584*4882a593Smuzhiyun const char * BZ_API(BZ2_bzerror) (BZFILE *b, int *errnum)
1585*4882a593Smuzhiyun {
1586*4882a593Smuzhiyun int err = ((bzFile *)b)->lastErr;
1587*4882a593Smuzhiyun
1588*4882a593Smuzhiyun if(err>0) err = 0;
1589*4882a593Smuzhiyun *errnum = err;
1590*4882a593Smuzhiyun return bzerrorstrings[err*-1];
1591*4882a593Smuzhiyun }
1592*4882a593Smuzhiyun #endif
1593*4882a593Smuzhiyun
bz_internal_error(int errcode)1594*4882a593Smuzhiyun void bz_internal_error(int errcode)
1595*4882a593Smuzhiyun {
1596*4882a593Smuzhiyun printf ("BZIP2 internal error %d\n", errcode);
1597*4882a593Smuzhiyun }
1598*4882a593Smuzhiyun
1599*4882a593Smuzhiyun /*-------------------------------------------------------------*/
1600*4882a593Smuzhiyun /*--- end bzlib.c ---*/
1601*4882a593Smuzhiyun /*-------------------------------------------------------------*/
1602