Coverage Report

Created: 2025-07-23 08:18

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