xref: /OK3568_Linux_fs/u-boot/lib/bzip2/bzlib.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
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