| |
| /*-----------------------------------------------------------*/ |
| /*--- A block-sorting, lossless compressor bzip2.c ---*/ |
| /*-----------------------------------------------------------*/ |
| |
| /*-- |
| This file is a part of bzip2 and/or libbzip2, a program and |
| library for lossless, block-sorting data compression. |
| |
| Copyright (C) 1996-2000 Julian R Seward. All rights reserved. |
| |
| Redistribution and use in source and binary forms, with or without |
| modification, are permitted provided that the following conditions |
| are met: |
| |
| 1. Redistributions of source code must retain the above copyright |
| notice, this list of conditions and the following disclaimer. |
| |
| 2. The origin of this software must not be misrepresented; you must |
| not claim that you wrote the original software. If you use this |
| software in a product, an acknowledgment in the product |
| documentation would be appreciated but is not required. |
| |
| 3. Altered source versions must be plainly marked as such, and must |
| not be misrepresented as being the original software. |
| |
| 4. The name of the author may not be used to endorse or promote |
| products derived from this software without specific prior written |
| permission. |
| |
| THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS |
| OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY |
| DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE |
| GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, |
| WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
| NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
| SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| Julian Seward, Cambridge, UK. |
| jseward@acm.org |
| bzip2/libbzip2 version 1.0 of 21 March 2000 |
| |
| This program is based on (at least) the work of: |
| Mike Burrows |
| David Wheeler |
| Peter Fenwick |
| Alistair Moffat |
| Radford Neal |
| Ian H. Witten |
| Robert Sedgewick |
| Jon L. Bentley |
| |
| For more information on these sources, see the manual. |
| --*/ |
| |
| |
| /*----------------------------------------------------*/ |
| /*--- IMPORTANT ---*/ |
| /*----------------------------------------------------*/ |
| |
| /*-- |
| WARNING: |
| This program and library (attempts to) compress data by |
| performing several non-trivial transformations on it. |
| Unless you are 100% familiar with *all* the algorithms |
| contained herein, and with the consequences of modifying them, |
| you should NOT meddle with the compression or decompression |
| machinery. Incorrect changes can and very likely *will* |
| lead to disasterous loss of data. |
| |
| DISCLAIMER: |
| I TAKE NO RESPONSIBILITY FOR ANY LOSS OF DATA ARISING FROM THE |
| USE OF THIS PROGRAM, HOWSOEVER CAUSED. |
| |
| Every compression of a file implies an assumption that the |
| compressed file can be decompressed to reproduce the original. |
| Great efforts in design, coding and testing have been made to |
| ensure that this program works correctly. However, the |
| complexity of the algorithms, and, in particular, the presence |
| of various special cases in the code which occur with very low |
| but non-zero probability make it impossible to rule out the |
| possibility of bugs remaining in the program. DO NOT COMPRESS |
| ANY DATA WITH THIS PROGRAM AND/OR LIBRARY UNLESS YOU ARE PREPARED |
| TO ACCEPT THE POSSIBILITY, HOWEVER SMALL, THAT THE DATA WILL |
| NOT BE RECOVERABLE. |
| |
| That is not to say this program is inherently unreliable. |
| Indeed, I very much hope the opposite is true. bzip2/libbzip2 |
| has been carefully constructed and extensively tested. |
| |
| PATENTS: |
| To the best of my knowledge, bzip2/libbzip2 does not use any |
| patented algorithms. However, I do not have the resources |
| available to carry out a full patent search. Therefore I cannot |
| give any guarantee of the above statement. |
| --*/ |
| |
| |
| |
| /*----------------------------------------------------*/ |
| /*--- and now for something much more pleasant :-) ---*/ |
| /*----------------------------------------------------*/ |
| |
| /*---------------------------------------------*/ |
| /*-- |
| Place a 1 beside your platform, and 0 elsewhere. |
| --*/ |
| |
| /*-- |
| Generic 32-bit Unix. |
| Also works on 64-bit Unix boxes. |
| --*/ |
| #define BZ_UNIX 1 |
| |
| /*-- |
| Win32, as seen by Jacob Navia's excellent |
| port of (Chris Fraser & David Hanson)'s excellent |
| lcc compiler. |
| --*/ |
| #define BZ_LCCWIN32 0 |
| |
| #if defined(_WIN32) && !defined(__CYGWIN__) |
| #undef BZ_LCCWIN32 |
| #define BZ_LCCWIN32 1 |
| #undef BZ_UNIX |
| #define BZ_UNIX 0 |
| #endif |
| |
| |
| /*---------------------------------------------*/ |
| /*-- |
| Some stuff for all platforms. |
| --*/ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <signal.h> |
| #include <math.h> |
| #include <errno.h> |
| #include <ctype.h> |
| #include "bzlib.h" |
| |
| #define ERROR_IF_EOF(i) { if ((i) == EOF) ioError(); } |
| #define ERROR_IF_NOT_ZERO(i) { if ((i) != 0) ioError(); } |
| #define ERROR_IF_MINUS_ONE(i) { if ((i) == (-1)) ioError(); } |
| |
| |
| /*---------------------------------------------*/ |
| /*-- |
| Platform-specific stuff. |
| --*/ |
| |
| #if BZ_UNIX |
| # include <sys/types.h> |
| # include <utime.h> |
| # include <unistd.h> |
| # include <sys/stat.h> |
| # include <sys/times.h> |
| |
| # define PATH_SEP '/' |
| # define MY_LSTAT lstat |
| # define MY_S_IFREG S_ISREG |
| # define MY_STAT stat |
| |
| # define APPEND_FILESPEC(root, name) \ |
| root=snocString((root), (name)) |
| |
| # define APPEND_FLAG(root, name) \ |
| root=snocString((root), (name)) |
| |
| # define SET_BINARY_MODE(fd) /**/ |
| |
| # ifdef __GNUC__ |
| # define NORETURN __attribute__ ((noreturn)) |
| # else |
| # define NORETURN /**/ |
| # endif |
| # ifdef __DJGPP__ |
| # include <io.h> |
| # include <fcntl.h> |
| # undef MY_LSTAT |
| # define MY_LSTAT stat |
| # undef SET_BINARY_MODE |
| # define SET_BINARY_MODE(fd) \ |
| do { \ |
| int retVal = setmode ( fileno ( fd ), \ |
| O_BINARY ); \ |
| ERROR_IF_MINUS_ONE ( retVal ); \ |
| } while ( 0 ) |
| # endif |
| # ifdef __CYGWIN__ |
| # include <io.h> |
| # include <fcntl.h> |
| # undef SET_BINARY_MODE |
| # define SET_BINARY_MODE(fd) \ |
| do { \ |
| int retVal = setmode ( fileno ( fd ), \ |
| O_BINARY ); \ |
| ERROR_IF_MINUS_ONE ( retVal ); \ |
| } while ( 0 ) |
| # endif |
| #endif |
| |
| |
| |
| #if BZ_LCCWIN32 |
| # include <io.h> |
| # include <fcntl.h> |
| # include <sys\stat.h> |
| |
| # define NORETURN /**/ |
| # define PATH_SEP '\\' |
| # define MY_LSTAT _stat |
| # define MY_STAT _stat |
| # define MY_S_IFREG(x) ((x) & _S_IFREG) |
| |
| # define APPEND_FLAG(root, name) \ |
| root=snocString((root), (name)) |
| |
| # if 0 |
| /*-- lcc-win32 seems to expand wildcards itself --*/ |
| # define APPEND_FILESPEC(root, spec) \ |
| do { \ |
| if ((spec)[0] == '-') { \ |
| root = snocString((root), (spec)); \ |
| } else { \ |
| struct _finddata_t c_file; \ |
| long hFile; \ |
| hFile = _findfirst((spec), &c_file); \ |
| if ( hFile == -1L ) { \ |
| root = snocString ((root), (spec)); \ |
| } else { \ |
| int anInt = 0; \ |
| while ( anInt == 0 ) { \ |
| root = snocString((root), \ |
| &c_file.name[0]); \ |
| anInt = _findnext(hFile, &c_file); \ |
| } \ |
| } \ |
| } \ |
| } while ( 0 ) |
| # else |
| # define APPEND_FILESPEC(root, name) \ |
| root = snocString ((root), (name)) |
| # endif |
| |
| # define SET_BINARY_MODE(fd) \ |
| do { \ |
| int retVal = setmode ( fileno ( fd ), \ |
| O_BINARY ); \ |
| ERROR_IF_MINUS_ONE ( retVal ); \ |
| } while ( 0 ) |
| |
| #endif |
| |
| |
| /*---------------------------------------------*/ |
| /*-- |
| Some more stuff for all platforms :-) |
| --*/ |
| |
| typedef char Char; |
| typedef unsigned char Bool; |
| typedef unsigned char UChar; |
| typedef int Int32; |
| typedef unsigned int UInt32; |
| typedef short Int16; |
| typedef unsigned short UInt16; |
| |
| #define True ((Bool)1) |
| #define False ((Bool)0) |
| |
| /*-- |
| IntNative is your platform's `native' int size. |
| Only here to avoid probs with 64-bit platforms. |
| --*/ |
| typedef int IntNative; |
| |
| |
| /*---------------------------------------------------*/ |
| /*--- Misc (file handling) data decls ---*/ |
| /*---------------------------------------------------*/ |
| |
| Int32 verbosity; |
| Bool keepInputFiles, smallMode, deleteOutputOnInterrupt; |
| Bool forceOverwrite, testFailsExist, unzFailsExist, noisy; |
| Int32 numFileNames, numFilesProcessed, blockSize100k; |
| Int32 exitValue; |
| |
| /*-- source modes; F==file, I==stdin, O==stdout --*/ |
| #define SM_I2O 1 |
| #define SM_F2O 2 |
| #define SM_F2F 3 |
| |
| /*-- operation modes --*/ |
| #define OM_Z 1 |
| #define OM_UNZ 2 |
| #define OM_TEST 3 |
| |
| Int32 opMode; |
| Int32 srcMode; |
| |
| #define FILE_NAME_LEN 1034 |
| |
| Int32 longestFileName; |
| Char inName [FILE_NAME_LEN]; |
| Char outName[FILE_NAME_LEN]; |
| Char tmpName[FILE_NAME_LEN]; |
| Char *progName; |
| Char progNameReally[FILE_NAME_LEN]; |
| FILE *outputHandleJustInCase; |
| Int32 workFactor; |
| |
| static void panic ( Char* ) NORETURN; |
| static void ioError ( void ) NORETURN; |
| static void outOfMemory ( void ) NORETURN; |
| static void configError ( void ) NORETURN; |
| static void crcError ( void ) NORETURN; |
| static void cleanUpAndFail ( Int32 ) NORETURN; |
| static void compressedStreamEOF ( void ) NORETURN; |
| |
| static void copyFileName ( Char*, Char* ); |
| static void* myMalloc ( Int32 ); |
| |
| |
| |
| /*---------------------------------------------------*/ |
| /*--- An implementation of 64-bit ints. Sigh. ---*/ |
| /*--- Roll on widespread deployment of ANSI C9X ! ---*/ |
| /*---------------------------------------------------*/ |
| |
| typedef |
| struct { UChar b[8]; } |
| UInt64; |
| |
| static |
| void uInt64_from_UInt32s ( UInt64* n, UInt32 lo32, UInt32 hi32 ) |
| { |
| n->b[7] = (UChar)((hi32 >> 24) & 0xFF); |
| n->b[6] = (UChar)((hi32 >> 16) & 0xFF); |
| n->b[5] = (UChar)((hi32 >> 8) & 0xFF); |
| n->b[4] = (UChar) (hi32 & 0xFF); |
| n->b[3] = (UChar)((lo32 >> 24) & 0xFF); |
| n->b[2] = (UChar)((lo32 >> 16) & 0xFF); |
| n->b[1] = (UChar)((lo32 >> 8) & 0xFF); |
| n->b[0] = (UChar) (lo32 & 0xFF); |
| } |
| |
| static |
| double uInt64_to_double ( UInt64* n ) |
| { |
| Int32 i; |
| double base = 1.0; |
| double sum = 0.0; |
| for (i = 0; i < 8; i++) { |
| sum += base * (double)(n->b[i]); |
| base *= 256.0; |
| } |
| return sum; |
| } |
| |
| static |
| void uInt64_add ( UInt64* src, UInt64* dst ) |
| { |
| Int32 i; |
| Int32 carry = 0; |
| for (i = 0; i < 8; i++) { |
| carry += ( ((Int32)src->b[i]) + ((Int32)dst->b[i]) ); |
| dst->b[i] = (UChar)(carry & 0xFF); |
| carry >>= 8; |
| } |
| } |
| |
| static |
| void uInt64_sub ( UInt64* src, UInt64* dst ) |
| { |
| Int32 t, i; |
| Int32 borrow = 0; |
| for (i = 0; i < 8; i++) { |
| t = ((Int32)dst->b[i]) - ((Int32)src->b[i]) - borrow; |
| if (t < 0) { |
| dst->b[i] = (UChar)(t + 256); |
| borrow = 1; |
| } else { |
| dst->b[i] = (UChar)t; |
| borrow = 0; |
| } |
| } |
| } |
| |
| static |
| void uInt64_mul ( UInt64* a, UInt64* b, UInt64* r_hi, UInt64* r_lo ) |
| { |
| UChar sum[16]; |
| Int32 ia, ib, carry; |
| for (ia = 0; ia < 16; ia++) sum[ia] = 0; |
| for (ia = 0; ia < 8; ia++) { |
| carry = 0; |
| for (ib = 0; ib < 8; ib++) { |
| carry += ( ((Int32)sum[ia+ib]) |
| + ((Int32)a->b[ia]) * ((Int32)b->b[ib]) ); |
| sum[ia+ib] = (UChar)(carry & 0xFF); |
| carry >>= 8; |
| } |
| sum[ia+8] = (UChar)(carry & 0xFF); |
| if ((carry >>= 8) != 0) panic ( "uInt64_mul" ); |
| } |
| |
| for (ia = 0; ia < 8; ia++) r_hi->b[ia] = sum[ia+8]; |
| for (ia = 0; ia < 8; ia++) r_lo->b[ia] = sum[ia]; |
| } |
| |
| |
| static |
| void uInt64_shr1 ( UInt64* n ) |
| { |
| Int32 i; |
| for (i = 0; i < 8; i++) { |
| n->b[i] >>= 1; |
| if (i < 7 && (n->b[i+1] & 1)) n->b[i] |= 0x80; |
| } |
| } |
| |
| static |
| void uInt64_shl1 ( UInt64* n ) |
| { |
| Int32 i; |
| for (i = 7; i >= 0; i--) { |
| n->b[i] <<= 1; |
| if (i > 0 && (n->b[i-1] & 0x80)) n->b[i]++; |
| } |
| } |
| |
| static |
| Bool uInt64_isZero ( UInt64* n ) |
| { |
| Int32 i; |
| for (i = 0; i < 8; i++) |
| if (n->b[i] != 0) return 0; |
| return 1; |
| } |
| |
| static |
| Int32 uInt64_qrm10 ( UInt64* n ) |
| { |
| /* Divide *n by 10, and return the remainder. Long division |
| is difficult, so we cheat and instead multiply by |
| 0xCCCC CCCC CCCC CCCD, which is 0.8 (viz, 0.1 << 3). |
| */ |
| Int32 i; |
| UInt64 tmp1, tmp2, n_orig, zero_point_eight; |
| |
| zero_point_eight.b[1] = zero_point_eight.b[2] = |
| zero_point_eight.b[3] = zero_point_eight.b[4] = |
| zero_point_eight.b[5] = zero_point_eight.b[6] = |
| zero_point_eight.b[7] = 0xCC; |
| zero_point_eight.b[0] = 0xCD; |
| |
| n_orig = *n; |
| |
| /* divide n by 10, |
| by multiplying by 0.8 and then shifting right 3 times */ |
| uInt64_mul ( n, &zero_point_eight, &tmp1, &tmp2 ); |
| uInt64_shr1(&tmp1); uInt64_shr1(&tmp1); uInt64_shr1(&tmp1); |
| *n = tmp1; |
| |
| /* tmp1 = 8*n, tmp2 = 2*n */ |
| uInt64_shl1(&tmp1); uInt64_shl1(&tmp1); uInt64_shl1(&tmp1); |
| tmp2 = *n; uInt64_shl1(&tmp2); |
| |
| /* tmp1 = 10*n */ |
| uInt64_add ( &tmp2, &tmp1 ); |
| |
| /* n_orig = n_orig - 10*n */ |
| uInt64_sub ( &tmp1, &n_orig ); |
| |
| /* n_orig should now hold quotient, in range 0 .. 9 */ |
| for (i = 7; i >= 1; i--) |
| if (n_orig.b[i] != 0) panic ( "uInt64_qrm10(1)" ); |
| if (n_orig.b[0] > 9) |
| panic ( "uInt64_qrm10(2)" ); |
| |
| return (int)n_orig.b[0]; |
| } |
| |
| /* ... and the Whole Entire Point of all this UInt64 stuff is |
| so that we can supply the following function. |
| */ |
| static |
| void uInt64_toAscii ( char* outbuf, UInt64* n ) |
| { |
| Int32 i, q; |
| UChar buf[32]; |
| Int32 nBuf = 0; |
| UInt64 n_copy = *n; |
| do { |
| q = uInt64_qrm10 ( &n_copy ); |
| buf[nBuf] = q + '0'; |
| nBuf++; |
| } while (!uInt64_isZero(&n_copy)); |
| outbuf[nBuf] = 0; |
| for (i = 0; i < nBuf; i++) outbuf[i] = buf[nBuf-i-1]; |
| } |
| |
| |
| /*---------------------------------------------------*/ |
| /*--- Processing of complete files and streams ---*/ |
| /*---------------------------------------------------*/ |
| |
| /*---------------------------------------------*/ |
| static |
| Bool myfeof ( FILE* f ) |
| { |
| Int32 c = fgetc ( f ); |
| if (c == EOF) return True; |
| ungetc ( c, f ); |
| return False; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void compressStream ( FILE *stream, FILE *zStream ) |
| { |
| BZFILE* bzf = NULL; |
| UChar ibuf[5000]; |
| Int32 nIbuf; |
| UInt32 nbytes_in_lo32, nbytes_in_hi32; |
| UInt32 nbytes_out_lo32, nbytes_out_hi32; |
| Int32 bzerr, bzerr_dummy, ret; |
| |
| SET_BINARY_MODE(stream); |
| SET_BINARY_MODE(zStream); |
| |
| if (ferror(stream)) goto errhandler_io; |
| if (ferror(zStream)) goto errhandler_io; |
| |
| bzf = BZ2_bzWriteOpen ( &bzerr, zStream, |
| blockSize100k, verbosity, workFactor ); |
| if (bzerr != BZ_OK) goto errhandler; |
| |
| if (verbosity >= 2) fprintf ( stderr, "\n" ); |
| |
| while (True) { |
| |
| if (myfeof(stream)) break; |
| nIbuf = fread ( ibuf, sizeof(UChar), 5000, stream ); |
| if (ferror(stream)) goto errhandler_io; |
| if (nIbuf > 0) BZ2_bzWrite ( &bzerr, bzf, (void*)ibuf, nIbuf ); |
| if (bzerr != BZ_OK) goto errhandler; |
| |
| } |
| |
| BZ2_bzWriteClose64 ( &bzerr, bzf, 0, |
| &nbytes_in_lo32, &nbytes_in_hi32, |
| &nbytes_out_lo32, &nbytes_out_hi32 ); |
| if (bzerr != BZ_OK) goto errhandler; |
| |
| if (ferror(zStream)) goto errhandler_io; |
| ret = fflush ( zStream ); |
| if (ret == EOF) goto errhandler_io; |
| if (zStream != stdout) { |
| ret = fclose ( zStream ); |
| if (ret == EOF) goto errhandler_io; |
| } |
| if (ferror(stream)) goto errhandler_io; |
| ret = fclose ( stream ); |
| if (ret == EOF) goto errhandler_io; |
| |
| if (nbytes_in_lo32 == 0 && nbytes_in_hi32 == 0) |
| nbytes_in_lo32 = 1; |
| |
| if (verbosity >= 1) { |
| Char buf_nin[32], buf_nout[32]; |
| UInt64 nbytes_in, nbytes_out; |
| double nbytes_in_d, nbytes_out_d; |
| uInt64_from_UInt32s ( &nbytes_in, |
| nbytes_in_lo32, nbytes_in_hi32 ); |
| uInt64_from_UInt32s ( &nbytes_out, |
| nbytes_out_lo32, nbytes_out_hi32 ); |
| nbytes_in_d = uInt64_to_double ( &nbytes_in ); |
| nbytes_out_d = uInt64_to_double ( &nbytes_out ); |
| uInt64_toAscii ( buf_nin, &nbytes_in ); |
| uInt64_toAscii ( buf_nout, &nbytes_out ); |
| fprintf ( stderr, "%6.3f:1, %6.3f bits/byte, " |
| "%5.2f%% saved, %s in, %s out.\n", |
| nbytes_in_d / nbytes_out_d, |
| (8.0 * nbytes_out_d) / nbytes_in_d, |
| 100.0 * (1.0 - nbytes_out_d / nbytes_in_d), |
| buf_nin, |
| buf_nout |
| ); |
| } |
| |
| return; |
| |
| errhandler: |
| BZ2_bzWriteClose64 ( &bzerr_dummy, bzf, 1, |
| &nbytes_in_lo32, &nbytes_in_hi32, |
| &nbytes_out_lo32, &nbytes_out_hi32 ); |
| switch (bzerr) { |
| case BZ_CONFIG_ERROR: |
| configError(); break; |
| case BZ_MEM_ERROR: |
| outOfMemory (); break; |
| case BZ_IO_ERROR: |
| errhandler_io: |
| ioError(); break; |
| default: |
| panic ( "compress:unexpected error" ); |
| } |
| |
| panic ( "compress:end" ); |
| /*notreached*/ |
| } |
| |
| |
| |
| /*---------------------------------------------*/ |
| static |
| Bool uncompressStream ( FILE *zStream, FILE *stream ) |
| { |
| BZFILE* bzf = NULL; |
| Int32 bzerr, bzerr_dummy, ret, nread, streamNo, i; |
| UChar obuf[5000]; |
| UChar unused[BZ_MAX_UNUSED]; |
| Int32 nUnused; |
| UChar* unusedTmp; |
| |
| nUnused = 0; |
| streamNo = 0; |
| |
| SET_BINARY_MODE(stream); |
| SET_BINARY_MODE(zStream); |
| |
| if (ferror(stream)) goto errhandler_io; |
| if (ferror(zStream)) goto errhandler_io; |
| |
| while (True) { |
| |
| bzf = BZ2_bzReadOpen ( |
| &bzerr, zStream, verbosity, |
| (int)smallMode, unused, nUnused |
| ); |
| if (bzf == NULL || bzerr != BZ_OK) goto errhandler; |
| streamNo++; |
| |
| while (bzerr == BZ_OK) { |
| nread = BZ2_bzRead ( &bzerr, bzf, obuf, 5000 ); |
| if (bzerr == BZ_DATA_ERROR_MAGIC) goto errhandler; |
| if ((bzerr == BZ_OK || bzerr == BZ_STREAM_END) && nread > 0) |
| fwrite ( obuf, sizeof(UChar), nread, stream ); |
| if (ferror(stream)) goto errhandler_io; |
| } |
| if (bzerr != BZ_STREAM_END) goto errhandler; |
| |
| BZ2_bzReadGetUnused ( &bzerr, bzf, (void**)(&unusedTmp), &nUnused ); |
| if (bzerr != BZ_OK) panic ( "decompress:bzReadGetUnused" ); |
| |
| for (i = 0; i < nUnused; i++) unused[i] = unusedTmp[i]; |
| |
| BZ2_bzReadClose ( &bzerr, bzf ); |
| if (bzerr != BZ_OK) panic ( "decompress:bzReadGetUnused" ); |
| |
| if (nUnused == 0 && myfeof(zStream)) break; |
| |
| } |
| |
| if (ferror(zStream)) goto errhandler_io; |
| ret = fclose ( zStream ); |
| if (ret == EOF) goto errhandler_io; |
| |
| if (ferror(stream)) goto errhandler_io; |
| ret = fflush ( stream ); |
| if (ret != 0) goto errhandler_io; |
| if (stream != stdout) { |
| ret = fclose ( stream ); |
| if (ret == EOF) goto errhandler_io; |
| } |
| if (verbosity >= 2) fprintf ( stderr, "\n " ); |
| return True; |
| |
| errhandler: |
| BZ2_bzReadClose ( &bzerr_dummy, bzf ); |
| switch (bzerr) { |
| case BZ_CONFIG_ERROR: |
| configError(); break; |
| case BZ_IO_ERROR: |
| errhandler_io: |
| ioError(); break; |
| case BZ_DATA_ERROR: |
| crcError(); |
| case BZ_MEM_ERROR: |
| outOfMemory(); |
| case BZ_UNEXPECTED_EOF: |
| compressedStreamEOF(); |
| case BZ_DATA_ERROR_MAGIC: |
| if (zStream != stdin) fclose(zStream); |
| if (stream != stdout) fclose(stream); |
| if (streamNo == 1) { |
| return False; |
| } else { |
| if (noisy) |
| fprintf ( stderr, |
| "\n%s: %s: trailing garbage after EOF ignored\n", |
| progName, inName ); |
| return True; |
| } |
| default: |
| panic ( "decompress:unexpected error" ); |
| } |
| |
| panic ( "decompress:end" ); |
| return True; /*notreached*/ |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| Bool testStream ( FILE *zStream ) |
| { |
| BZFILE* bzf = NULL; |
| Int32 bzerr, bzerr_dummy, ret, nread, streamNo, i; |
| UChar obuf[5000]; |
| UChar unused[BZ_MAX_UNUSED]; |
| Int32 nUnused; |
| UChar* unusedTmp; |
| |
| nUnused = 0; |
| streamNo = 0; |
| |
| SET_BINARY_MODE(zStream); |
| if (ferror(zStream)) goto errhandler_io; |
| |
| while (True) { |
| |
| bzf = BZ2_bzReadOpen ( |
| &bzerr, zStream, verbosity, |
| (int)smallMode, unused, nUnused |
| ); |
| if (bzf == NULL || bzerr != BZ_OK) goto errhandler; |
| streamNo++; |
| |
| while (bzerr == BZ_OK) { |
| nread = BZ2_bzRead ( &bzerr, bzf, obuf, 5000 ); |
| if (bzerr == BZ_DATA_ERROR_MAGIC) goto errhandler; |
| } |
| if (bzerr != BZ_STREAM_END) goto errhandler; |
| |
| BZ2_bzReadGetUnused ( &bzerr, bzf, (void**)(&unusedTmp), &nUnused ); |
| if (bzerr != BZ_OK) panic ( "test:bzReadGetUnused" ); |
| |
| for (i = 0; i < nUnused; i++) unused[i] = unusedTmp[i]; |
| |
| BZ2_bzReadClose ( &bzerr, bzf ); |
| if (bzerr != BZ_OK) panic ( "test:bzReadGetUnused" ); |
| if (nUnused == 0 && myfeof(zStream)) break; |
| |
| } |
| |
| if (ferror(zStream)) goto errhandler_io; |
| ret = fclose ( zStream ); |
| if (ret == EOF) goto errhandler_io; |
| |
| if (verbosity >= 2) fprintf ( stderr, "\n " ); |
| return True; |
| |
| errhandler: |
| BZ2_bzReadClose ( &bzerr_dummy, bzf ); |
| if (verbosity == 0) |
| fprintf ( stderr, "%s: %s: ", progName, inName ); |
| switch (bzerr) { |
| case BZ_CONFIG_ERROR: |
| configError(); break; |
| case BZ_IO_ERROR: |
| errhandler_io: |
| ioError(); break; |
| case BZ_DATA_ERROR: |
| fprintf ( stderr, |
| "data integrity (CRC) error in data\n" ); |
| return False; |
| case BZ_MEM_ERROR: |
| outOfMemory(); |
| case BZ_UNEXPECTED_EOF: |
| fprintf ( stderr, |
| "file ends unexpectedly\n" ); |
| return False; |
| case BZ_DATA_ERROR_MAGIC: |
| if (zStream != stdin) fclose(zStream); |
| if (streamNo == 1) { |
| fprintf ( stderr, |
| "bad magic number (file not created by bzip2)\n" ); |
| return False; |
| } else { |
| if (noisy) |
| fprintf ( stderr, |
| "trailing garbage after EOF ignored\n" ); |
| return True; |
| } |
| default: |
| panic ( "test:unexpected error" ); |
| } |
| |
| panic ( "test:end" ); |
| return True; /*notreached*/ |
| } |
| |
| |
| /*---------------------------------------------------*/ |
| /*--- Error [non-] handling grunge ---*/ |
| /*---------------------------------------------------*/ |
| |
| /*---------------------------------------------*/ |
| static |
| void setExit ( Int32 v ) |
| { |
| if (v > exitValue) exitValue = v; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void cadvise ( void ) |
| { |
| if (noisy) |
| fprintf ( |
| stderr, |
| "\nIt is possible that the compressed file(s) have become corrupted.\n" |
| "You can use the -tvv option to test integrity of such files.\n\n" |
| "You can use the `bzip2recover' program to *attempt* to recover\n" |
| "data from undamaged sections of corrupted files.\n\n" |
| ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void showFileNames ( void ) |
| { |
| if (noisy) |
| fprintf ( |
| stderr, |
| "\tInput file = %s, output file = %s\n", |
| inName, outName |
| ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void cleanUpAndFail ( Int32 ec ) |
| { |
| IntNative retVal; |
| |
| if ( srcMode == SM_F2F |
| && opMode != OM_TEST |
| && deleteOutputOnInterrupt ) { |
| if (noisy) |
| fprintf ( stderr, "%s: Deleting output file %s, if it exists.\n", |
| progName, outName ); |
| if (outputHandleJustInCase != NULL) |
| fclose ( outputHandleJustInCase ); |
| retVal = remove ( outName ); |
| if (retVal != 0) |
| fprintf ( stderr, |
| "%s: WARNING: deletion of output file (apparently) failed.\n", |
| progName ); |
| } |
| if (noisy && numFileNames > 0 && numFilesProcessed < numFileNames) { |
| fprintf ( stderr, |
| "%s: WARNING: some files have not been processed:\n" |
| "\t%d specified on command line, %d not processed yet.\n\n", |
| progName, numFileNames, |
| numFileNames - numFilesProcessed ); |
| } |
| setExit(ec); |
| exit(exitValue); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void panic ( Char* s ) |
| { |
| fprintf ( stderr, |
| "\n%s: PANIC -- internal consistency error:\n" |
| "\t%s\n" |
| "\tThis is a BUG. Please report it to me at:\n" |
| "\tjseward@acm.org\n", |
| progName, s ); |
| showFileNames(); |
| cleanUpAndFail( 3 ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void crcError ( void ) |
| { |
| fprintf ( stderr, |
| "\n%s: Data integrity error when decompressing.\n", |
| progName ); |
| showFileNames(); |
| cadvise(); |
| cleanUpAndFail( 2 ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void compressedStreamEOF ( void ) |
| { |
| fprintf ( stderr, |
| "\n%s: Compressed file ends unexpectedly;\n\t" |
| "perhaps it is corrupted? *Possible* reason follows.\n", |
| progName ); |
| perror ( progName ); |
| showFileNames(); |
| cadvise(); |
| cleanUpAndFail( 2 ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void ioError ( void ) |
| { |
| fprintf ( stderr, |
| "\n%s: I/O or other error, bailing out. " |
| "Possible reason follows.\n", |
| progName ); |
| perror ( progName ); |
| showFileNames(); |
| cleanUpAndFail( 1 ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void mySignalCatcher ( IntNative n ) |
| { |
| fprintf ( stderr, |
| "\n%s: Control-C or similar caught, quitting.\n", |
| progName ); |
| cleanUpAndFail(1); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void mySIGSEGVorSIGBUScatcher ( IntNative n ) |
| { |
| if (opMode == OM_Z) |
| fprintf ( |
| stderr, |
| "\n%s: Caught a SIGSEGV or SIGBUS whilst compressing.\n" |
| "\n" |
| " Possible causes are (most likely first):\n" |
| " (1) This computer has unreliable memory or cache hardware\n" |
| " (a surprisingly common problem; try a different machine.)\n" |
| " (2) A bug in the compiler used to create this executable\n" |
| " (unlikely, if you didn't compile bzip2 yourself.)\n" |
| " (3) A real bug in bzip2 -- I hope this should never be the case.\n" |
| " The user's manual, Section 4.3, has more info on (1) and (2).\n" |
| " \n" |
| " If you suspect this is a bug in bzip2, or are unsure about (1)\n" |
| " or (2), feel free to report it to me at: jseward@acm.org.\n" |
| " Section 4.3 of the user's manual describes the info a useful\n" |
| " bug report should have. If the manual is available on your\n" |
| " system, please try and read it before mailing me. If you don't\n" |
| " have the manual or can't be bothered to read it, mail me anyway.\n" |
| "\n", |
| progName ); |
| else |
| fprintf ( |
| stderr, |
| "\n%s: Caught a SIGSEGV or SIGBUS whilst decompressing.\n" |
| "\n" |
| " Possible causes are (most likely first):\n" |
| " (1) The compressed data is corrupted, and bzip2's usual checks\n" |
| " failed to detect this. Try bzip2 -tvv my_file.bz2.\n" |
| " (2) This computer has unreliable memory or cache hardware\n" |
| " (a surprisingly common problem; try a different machine.)\n" |
| " (3) A bug in the compiler used to create this executable\n" |
| " (unlikely, if you didn't compile bzip2 yourself.)\n" |
| " (4) A real bug in bzip2 -- I hope this should never be the case.\n" |
| " The user's manual, Section 4.3, has more info on (2) and (3).\n" |
| " \n" |
| " If you suspect this is a bug in bzip2, or are unsure about (2)\n" |
| " or (3), feel free to report it to me at: jseward@acm.org.\n" |
| " Section 4.3 of the user's manual describes the info a useful\n" |
| " bug report should have. If the manual is available on your\n" |
| " system, please try and read it before mailing me. If you don't\n" |
| " have the manual or can't be bothered to read it, mail me anyway.\n" |
| "\n", |
| progName ); |
| |
| showFileNames(); |
| if (opMode == OM_Z) |
| cleanUpAndFail( 3 ); else |
| { cadvise(); cleanUpAndFail( 2 ); } |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void outOfMemory ( void ) |
| { |
| fprintf ( stderr, |
| "\n%s: couldn't allocate enough memory\n", |
| progName ); |
| showFileNames(); |
| cleanUpAndFail(1); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void configError ( void ) |
| { |
| fprintf ( stderr, |
| "bzip2: I'm not configured correctly for this platform!\n" |
| "\tI require Int32, Int16 and Char to have sizes\n" |
| "\tof 4, 2 and 1 bytes to run properly, and they don't.\n" |
| "\tProbably you can fix this by defining them correctly,\n" |
| "\tand recompiling. Bye!\n" ); |
| setExit(3); |
| exit(exitValue); |
| } |
| |
| |
| /*---------------------------------------------------*/ |
| /*--- The main driver machinery ---*/ |
| /*---------------------------------------------------*/ |
| |
| /*---------------------------------------------*/ |
| static |
| void pad ( Char *s ) |
| { |
| Int32 i; |
| if ( (Int32)strlen(s) >= longestFileName ) return; |
| for (i = 1; i <= longestFileName - (Int32)strlen(s); i++) |
| fprintf ( stderr, " " ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void copyFileName ( Char* to, Char* from ) |
| { |
| if ( strlen(from) > FILE_NAME_LEN-10 ) { |
| fprintf ( |
| stderr, |
| "bzip2: file name\n`%s'\n" |
| "is suspiciously (more than %d chars) long.\n" |
| "Try using a reasonable file name instead. Sorry! :-)\n", |
| from, FILE_NAME_LEN-10 |
| ); |
| setExit(1); |
| exit(exitValue); |
| } |
| |
| strncpy(to,from,FILE_NAME_LEN-10); |
| to[FILE_NAME_LEN-10]='\0'; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| Bool fileExists ( Char* name ) |
| { |
| FILE *tmp = fopen ( name, "rb" ); |
| Bool exists = (tmp != NULL); |
| if (tmp != NULL) fclose ( tmp ); |
| return exists; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| /*-- |
| if in doubt, return True |
| --*/ |
| static |
| Bool notAStandardFile ( Char* name ) |
| { |
| IntNative i; |
| struct MY_STAT statBuf; |
| |
| i = MY_LSTAT ( name, &statBuf ); |
| if (i != 0) return True; |
| if (MY_S_IFREG(statBuf.st_mode)) return False; |
| return True; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| /*-- |
| rac 11/21/98 see if file has hard links to it |
| --*/ |
| static |
| Int32 countHardLinks ( Char* name ) |
| { |
| IntNative i; |
| struct MY_STAT statBuf; |
| |
| i = MY_LSTAT ( name, &statBuf ); |
| if (i != 0) return 0; |
| return (statBuf.st_nlink - 1); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void copyDatePermissionsAndOwner ( Char *srcName, Char *dstName ) |
| { |
| #if BZ_UNIX |
| IntNative retVal; |
| struct MY_STAT statBuf; |
| struct utimbuf uTimBuf; |
| |
| retVal = MY_LSTAT ( srcName, &statBuf ); |
| ERROR_IF_NOT_ZERO ( retVal ); |
| uTimBuf.actime = statBuf.st_atime; |
| uTimBuf.modtime = statBuf.st_mtime; |
| |
| retVal = chmod ( dstName, statBuf.st_mode ); |
| ERROR_IF_NOT_ZERO ( retVal ); |
| |
| retVal = utime ( dstName, &uTimBuf ); |
| ERROR_IF_NOT_ZERO ( retVal ); |
| |
| retVal = chown ( dstName, statBuf.st_uid, statBuf.st_gid ); |
| /* chown() will in many cases return with EPERM, which can |
| be safely ignored. |
| */ |
| #endif |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void setInterimPermissions ( Char *dstName ) |
| { |
| #if BZ_UNIX |
| IntNative retVal; |
| retVal = chmod ( dstName, S_IRUSR | S_IWUSR ); |
| ERROR_IF_NOT_ZERO ( retVal ); |
| #endif |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| Bool containsDubiousChars ( Char* name ) |
| { |
| Bool cdc = False; |
| for (; *name != '\0'; name++) |
| if (*name == '?' || *name == '*') cdc = True; |
| return cdc; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| #define BZ_N_SUFFIX_PAIRS 4 |
| |
| Char* zSuffix[BZ_N_SUFFIX_PAIRS] |
| = { ".bz2", ".bz", ".tbz2", ".tbz" }; |
| Char* unzSuffix[BZ_N_SUFFIX_PAIRS] |
| = { "", "", ".tar", ".tar" }; |
| |
| static |
| Bool hasSuffix ( Char* s, Char* suffix ) |
| { |
| Int32 ns = strlen(s); |
| Int32 nx = strlen(suffix); |
| if (ns < nx) return False; |
| if (strcmp(s + ns - nx, suffix) == 0) return True; |
| return False; |
| } |
| |
| static |
| Bool mapSuffix ( Char* name, |
| Char* oldSuffix, Char* newSuffix ) |
| { |
| if (!hasSuffix(name,oldSuffix)) return False; |
| name[strlen(name)-strlen(oldSuffix)] = 0; |
| strcat ( name, newSuffix ); |
| return True; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void compress ( Char *name ) |
| { |
| FILE *inStr; |
| FILE *outStr; |
| Int32 n, i; |
| |
| deleteOutputOnInterrupt = False; |
| |
| if (name == NULL && srcMode != SM_I2O) |
| panic ( "compress: bad modes\n" ); |
| |
| switch (srcMode) { |
| case SM_I2O: |
| copyFileName ( inName, "(stdin)" ); |
| copyFileName ( outName, "(stdout)" ); |
| break; |
| case SM_F2F: |
| copyFileName ( inName, name ); |
| copyFileName ( outName, name ); |
| strcat ( outName, ".bz2" ); |
| break; |
| case SM_F2O: |
| copyFileName ( inName, name ); |
| copyFileName ( outName, "(stdout)" ); |
| break; |
| } |
| |
| if ( srcMode != SM_I2O && containsDubiousChars ( inName ) ) { |
| if (noisy) |
| fprintf ( stderr, "%s: There are no files matching `%s'.\n", |
| progName, inName ); |
| setExit(1); |
| return; |
| } |
| if ( srcMode != SM_I2O && !fileExists ( inName ) ) { |
| fprintf ( stderr, "%s: Can't open input file %s: %s.\n", |
| progName, inName, strerror(errno) ); |
| setExit(1); |
| return; |
| } |
| for (i = 0; i < BZ_N_SUFFIX_PAIRS; i++) { |
| if (hasSuffix(inName, zSuffix[i])) { |
| if (noisy) |
| fprintf ( stderr, |
| "%s: Input file %s already has %s suffix.\n", |
| progName, inName, zSuffix[i] ); |
| setExit(1); |
| return; |
| } |
| } |
| if ( srcMode == SM_F2F && !forceOverwrite && notAStandardFile ( inName )) { |
| if (noisy) |
| fprintf ( stderr, "%s: Input file %s is not a normal file.\n", |
| progName, inName ); |
| setExit(1); |
| return; |
| } |
| if ( srcMode == SM_F2F && !forceOverwrite && fileExists ( outName ) ) { |
| fprintf ( stderr, "%s: Output file %s already exists.\n", |
| progName, outName ); |
| setExit(1); |
| return; |
| } |
| if ( srcMode == SM_F2F && !forceOverwrite && |
| (n=countHardLinks ( inName )) > 0) { |
| fprintf ( stderr, "%s: Input file %s has %d other link%s.\n", |
| progName, inName, n, n > 1 ? "s" : "" ); |
| setExit(1); |
| return; |
| } |
| |
| switch ( srcMode ) { |
| |
| case SM_I2O: |
| inStr = stdin; |
| outStr = stdout; |
| if ( isatty ( fileno ( stdout ) ) ) { |
| fprintf ( stderr, |
| "%s: I won't write compressed data to a terminal.\n", |
| progName ); |
| fprintf ( stderr, "%s: For help, type: `%s --help'.\n", |
| progName, progName ); |
| setExit(1); |
| return; |
| }; |
| break; |
| |
| case SM_F2O: |
| inStr = fopen ( inName, "rb" ); |
| outStr = stdout; |
| if ( isatty ( fileno ( stdout ) ) ) { |
| fprintf ( stderr, |
| "%s: I won't write compressed data to a terminal.\n", |
| progName ); |
| fprintf ( stderr, "%s: For help, type: `%s --help'.\n", |
| progName, progName ); |
| if ( inStr != NULL ) fclose ( inStr ); |
| setExit(1); |
| return; |
| }; |
| if ( inStr == NULL ) { |
| fprintf ( stderr, "%s: Can't open input file %s: %s.\n", |
| progName, inName, strerror(errno) ); |
| setExit(1); |
| return; |
| }; |
| break; |
| |
| case SM_F2F: |
| inStr = fopen ( inName, "rb" ); |
| outStr = fopen ( outName, "wb" ); |
| if ( outStr == NULL) { |
| fprintf ( stderr, "%s: Can't create output file %s: %s.\n", |
| progName, outName, strerror(errno) ); |
| if ( inStr != NULL ) fclose ( inStr ); |
| setExit(1); |
| return; |
| } |
| if ( inStr == NULL ) { |
| fprintf ( stderr, "%s: Can't open input file %s: %s.\n", |
| progName, inName, strerror(errno) ); |
| if ( outStr != NULL ) fclose ( outStr ); |
| setExit(1); |
| return; |
| }; |
| setInterimPermissions ( outName ); |
| break; |
| |
| default: |
| panic ( "compress: bad srcMode" ); |
| break; |
| } |
| |
| if (verbosity >= 1) { |
| fprintf ( stderr, " %s: ", inName ); |
| pad ( inName ); |
| fflush ( stderr ); |
| } |
| |
| /*--- Now the input and output handles are sane. Do the Biz. ---*/ |
| outputHandleJustInCase = outStr; |
| deleteOutputOnInterrupt = True; |
| compressStream ( inStr, outStr ); |
| outputHandleJustInCase = NULL; |
| |
| /*--- If there was an I/O error, we won't get here. ---*/ |
| if ( srcMode == SM_F2F ) { |
| copyDatePermissionsAndOwner ( inName, outName ); |
| deleteOutputOnInterrupt = False; |
| if ( !keepInputFiles ) { |
| IntNative retVal = remove ( inName ); |
| ERROR_IF_NOT_ZERO ( retVal ); |
| } |
| } |
| |
| deleteOutputOnInterrupt = False; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void uncompress ( Char *name ) |
| { |
| FILE *inStr; |
| FILE *outStr; |
| Int32 n, i; |
| Bool magicNumberOK; |
| Bool cantGuess; |
| |
| deleteOutputOnInterrupt = False; |
| |
| if (name == NULL && srcMode != SM_I2O) |
| panic ( "uncompress: bad modes\n" ); |
| |
| cantGuess = False; |
| switch (srcMode) { |
| case SM_I2O: |
| copyFileName ( inName, "(stdin)" ); |
| copyFileName ( outName, "(stdout)" ); |
| break; |
| case SM_F2F: |
| copyFileName ( inName, name ); |
| copyFileName ( outName, name ); |
| for (i = 0; i < BZ_N_SUFFIX_PAIRS; i++) |
| if (mapSuffix(outName,zSuffix[i],unzSuffix[i])) |
| goto zzz; |
| cantGuess = True; |
| strcat ( outName, ".out" ); |
| break; |
| case SM_F2O: |
| copyFileName ( inName, name ); |
| copyFileName ( outName, "(stdout)" ); |
| break; |
| } |
| |
| zzz: |
| if ( srcMode != SM_I2O && containsDubiousChars ( inName ) ) { |
| if (noisy) |
| fprintf ( stderr, "%s: There are no files matching `%s'.\n", |
| progName, inName ); |
| setExit(1); |
| return; |
| } |
| if ( srcMode != SM_I2O && !fileExists ( inName ) ) { |
| fprintf ( stderr, "%s: Can't open input file %s: %s.\n", |
| progName, inName, strerror(errno) ); |
| setExit(1); |
| return; |
| } |
| if ( srcMode == SM_F2F && !forceOverwrite && notAStandardFile ( inName )) { |
| if (noisy) |
| fprintf ( stderr, "%s: Input file %s is not a normal file.\n", |
| progName, inName ); |
| setExit(1); |
| return; |
| } |
| if ( /* srcMode == SM_F2F implied && */ cantGuess ) { |
| if (noisy) |
| fprintf ( stderr, |
| "%s: Can't guess original name for %s -- using %s\n", |
| progName, inName, outName ); |
| /* just a warning, no return */ |
| } |
| if ( srcMode == SM_F2F && !forceOverwrite && fileExists ( outName ) ) { |
| fprintf ( stderr, "%s: Output file %s already exists.\n", |
| progName, outName ); |
| setExit(1); |
| return; |
| } |
| if ( srcMode == SM_F2F && !forceOverwrite && |
| (n=countHardLinks ( inName ) ) > 0) { |
| fprintf ( stderr, "%s: Input file %s has %d other link%s.\n", |
| progName, inName, n, n > 1 ? "s" : "" ); |
| setExit(1); |
| return; |
| } |
| |
| switch ( srcMode ) { |
| |
| case SM_I2O: |
| inStr = stdin; |
| outStr = stdout; |
| if ( isatty ( fileno ( stdin ) ) ) { |
| fprintf ( stderr, |
| "%s: I won't read compressed data from a terminal.\n", |
| progName ); |
| fprintf ( stderr, "%s: For help, type: `%s --help'.\n", |
| progName, progName ); |
| setExit(1); |
| return; |
| }; |
| break; |
| |
| case SM_F2O: |
| inStr = fopen ( inName, "rb" ); |
| outStr = stdout; |
| if ( inStr == NULL ) { |
| fprintf ( stderr, "%s: Can't open input file %s:%s.\n", |
| progName, inName, strerror(errno) ); |
| if ( inStr != NULL ) fclose ( inStr ); |
| setExit(1); |
| return; |
| }; |
| break; |
| |
| case SM_F2F: |
| inStr = fopen ( inName, "rb" ); |
| outStr = fopen ( outName, "wb" ); |
| if ( outStr == NULL) { |
| fprintf ( stderr, "%s: Can't create output file %s: %s.\n", |
| progName, outName, strerror(errno) ); |
| if ( inStr != NULL ) fclose ( inStr ); |
| setExit(1); |
| return; |
| } |
| if ( inStr == NULL ) { |
| fprintf ( stderr, "%s: Can't open input file %s: %s.\n", |
| progName, inName, strerror(errno) ); |
| if ( outStr != NULL ) fclose ( outStr ); |
| setExit(1); |
| return; |
| }; |
| setInterimPermissions ( outName ); |
| break; |
| |
| default: |
| panic ( "uncompress: bad srcMode" ); |
| break; |
| } |
| |
| if (verbosity >= 1) { |
| fprintf ( stderr, " %s: ", inName ); |
| pad ( inName ); |
| fflush ( stderr ); |
| } |
| |
| /*--- Now the input and output handles are sane. Do the Biz. ---*/ |
| outputHandleJustInCase = outStr; |
| deleteOutputOnInterrupt = True; |
| magicNumberOK = uncompressStream ( inStr, outStr ); |
| outputHandleJustInCase = NULL; |
| |
| /*--- If there was an I/O error, we won't get here. ---*/ |
| if ( magicNumberOK ) { |
| if ( srcMode == SM_F2F ) { |
| copyDatePermissionsAndOwner ( inName, outName ); |
| deleteOutputOnInterrupt = False; |
| if ( !keepInputFiles ) { |
| IntNative retVal = remove ( inName ); |
| ERROR_IF_NOT_ZERO ( retVal ); |
| } |
| } |
| } else { |
| unzFailsExist = True; |
| deleteOutputOnInterrupt = False; |
| if ( srcMode == SM_F2F ) { |
| IntNative retVal = remove ( outName ); |
| ERROR_IF_NOT_ZERO ( retVal ); |
| } |
| } |
| deleteOutputOnInterrupt = False; |
| |
| if ( magicNumberOK ) { |
| if (verbosity >= 1) |
| fprintf ( stderr, "done\n" ); |
| } else { |
| setExit(2); |
| if (verbosity >= 1) |
| fprintf ( stderr, "not a bzip2 file.\n" ); else |
| fprintf ( stderr, |
| "%s: %s is not a bzip2 file.\n", |
| progName, inName ); |
| } |
| |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void testf ( Char *name ) |
| { |
| FILE *inStr; |
| Bool allOK; |
| |
| deleteOutputOnInterrupt = False; |
| |
| if (name == NULL && srcMode != SM_I2O) |
| panic ( "testf: bad modes\n" ); |
| |
| copyFileName ( outName, "(none)" ); |
| switch (srcMode) { |
| case SM_I2O: copyFileName ( inName, "(stdin)" ); break; |
| case SM_F2F: copyFileName ( inName, name ); break; |
| case SM_F2O: copyFileName ( inName, name ); break; |
| } |
| |
| if ( srcMode != SM_I2O && containsDubiousChars ( inName ) ) { |
| if (noisy) |
| fprintf ( stderr, "%s: There are no files matching `%s'.\n", |
| progName, inName ); |
| setExit(1); |
| return; |
| } |
| if ( srcMode != SM_I2O && !fileExists ( inName ) ) { |
| fprintf ( stderr, "%s: Can't open input %s: %s.\n", |
| progName, inName, strerror(errno) ); |
| setExit(1); |
| return; |
| } |
| |
| switch ( srcMode ) { |
| |
| case SM_I2O: |
| if ( isatty ( fileno ( stdin ) ) ) { |
| fprintf ( stderr, |
| "%s: I won't read compressed data from a terminal.\n", |
| progName ); |
| fprintf ( stderr, "%s: For help, type: `%s --help'.\n", |
| progName, progName ); |
| setExit(1); |
| return; |
| }; |
| inStr = stdin; |
| break; |
| |
| case SM_F2O: case SM_F2F: |
| inStr = fopen ( inName, "rb" ); |
| if ( inStr == NULL ) { |
| fprintf ( stderr, "%s: Can't open input file %s:%s.\n", |
| progName, inName, strerror(errno) ); |
| setExit(1); |
| return; |
| }; |
| break; |
| |
| default: |
| panic ( "testf: bad srcMode" ); |
| break; |
| } |
| |
| if (verbosity >= 1) { |
| fprintf ( stderr, " %s: ", inName ); |
| pad ( inName ); |
| fflush ( stderr ); |
| } |
| |
| /*--- Now the input handle is sane. Do the Biz. ---*/ |
| allOK = testStream ( inStr ); |
| |
| if (allOK && verbosity >= 1) fprintf ( stderr, "ok\n" ); |
| if (!allOK) testFailsExist = True; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void license ( void ) |
| { |
| fprintf ( stderr, |
| |
| "bzip2, a block-sorting file compressor. " |
| "Version %s.\n" |
| " \n" |
| " Copyright (C) 1996-2000 by Julian Seward.\n" |
| " \n" |
| " This program is free software; you can redistribute it and/or modify\n" |
| " it under the terms set out in the LICENSE file, which is included\n" |
| " in the bzip2-1.0 source distribution.\n" |
| " \n" |
| " This program is distributed in the hope that it will be useful,\n" |
| " but WITHOUT ANY WARRANTY; without even the implied warranty of\n" |
| " MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n" |
| " LICENSE file for more details.\n" |
| " \n", |
| BZ2_bzlibVersion() |
| ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void usage ( Char *fullProgName ) |
| { |
| fprintf ( |
| stderr, |
| "bzip2, a block-sorting file compressor. " |
| "Version %s.\n" |
| "\n usage: %s [flags and input files in any order]\n" |
| "\n" |
| " -h --help print this message\n" |
| " -d --decompress force decompression\n" |
| " -z --compress force compression\n" |
| " -k --keep keep (don't delete) input files\n" |
| " -f --force overwrite existing output files\n" |
| " -t --test test compressed file integrity\n" |
| " -c --stdout output to standard out\n" |
| " -q --quiet suppress noncritical error messages\n" |
| " -v --verbose be verbose (a 2nd -v gives more)\n" |
| " -L --license display software version & license\n" |
| " -V --version display software version & license\n" |
| " -s --small use less memory (at most 2500k)\n" |
| " -1 .. -9 set block size to 100k .. 900k\n" |
| "\n" |
| " If invoked as `bzip2', default action is to compress.\n" |
| " as `bunzip2', default action is to decompress.\n" |
| " as `bzcat', default action is to decompress to stdout.\n" |
| "\n" |
| " If no file names are given, bzip2 compresses or decompresses\n" |
| " from standard input to standard output. You can combine\n" |
| " short flags, so `-v -4' means the same as -v4 or -4v, &c.\n" |
| #if BZ_UNIX |
| "\n" |
| #endif |
| , |
| |
| BZ2_bzlibVersion(), |
| fullProgName |
| ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void redundant ( Char* flag ) |
| { |
| fprintf ( |
| stderr, |
| "%s: %s is redundant in versions 0.9.5 and above\n", |
| progName, flag ); |
| } |
| |
| |
| /*---------------------------------------------*/ |
| /*-- |
| All the garbage from here to main() is purely to |
| implement a linked list of command-line arguments, |
| into which main() copies argv[1 .. argc-1]. |
| |
| The purpose of this exercise is to facilitate |
| the expansion of wildcard characters * and ? in |
| filenames for OSs which don't know how to do it |
| themselves, like MSDOS, Windows 95 and NT. |
| |
| The actual Dirty Work is done by the platform- |
| specific macro APPEND_FILESPEC. |
| --*/ |
| |
| typedef |
| struct zzzz { |
| Char *name; |
| struct zzzz *link; |
| } |
| Cell; |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void *myMalloc ( Int32 n ) |
| { |
| void* p; |
| |
| p = malloc ( (size_t)n ); |
| if (p == NULL) outOfMemory (); |
| return p; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| Cell *mkCell ( void ) |
| { |
| Cell *c; |
| |
| c = (Cell*) myMalloc ( sizeof ( Cell ) ); |
| c->name = NULL; |
| c->link = NULL; |
| return c; |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| Cell *snocString ( Cell *root, Char *name ) |
| { |
| if (root == NULL) { |
| Cell *tmp = mkCell(); |
| tmp->name = (Char*) myMalloc ( 5 + strlen(name) ); |
| strcpy ( tmp->name, name ); |
| return tmp; |
| } else { |
| Cell *tmp = root; |
| while (tmp->link != NULL) tmp = tmp->link; |
| tmp->link = snocString ( tmp->link, name ); |
| return root; |
| } |
| } |
| |
| |
| /*---------------------------------------------*/ |
| static |
| void addFlagsFromEnvVar ( Cell** argList, Char* varName ) |
| { |
| Int32 i, j, k; |
| Char *envbase, *p; |
| |
| envbase = getenv(varName); |
| if (envbase != NULL) { |
| p = envbase; |
| i = 0; |
| while (True) { |
| if (p[i] == 0) break; |
| p += i; |
| i = 0; |
| while (isspace((Int32)(p[0]))) p++; |
| while (p[i] != 0 && !isspace((Int32)(p[i]))) i++; |
| if (i > 0) { |
| k = i; if (k > FILE_NAME_LEN-10) k = FILE_NAME_LEN-10; |
| for (j = 0; j < k; j++) tmpName[j] = p[j]; |
| tmpName[k] = 0; |
| APPEND_FLAG(*argList, tmpName); |
| } |
| } |
| } |
| } |
| |
| |
| /*---------------------------------------------*/ |
| #define ISFLAG(s) (strcmp(aa->name, (s))==0) |
| |
| IntNative main ( IntNative argc, Char *argv[] ) |
| { |
| Int32 i, j; |
| Char *tmp; |
| Cell *argList; |
| Cell *aa; |
| Bool decode; |
| |
| /*-- Be really really really paranoid :-) --*/ |
| if (sizeof(Int32) != 4 || sizeof(UInt32) != 4 || |
| sizeof(Int16) != 2 || sizeof(UInt16) != 2 || |
| sizeof(Char) != 1 || sizeof(UChar) != 1) |
| configError(); |
| |
| /*-- Initialise --*/ |
| outputHandleJustInCase = NULL; |
| smallMode = False; |
| keepInputFiles = False; |
| forceOverwrite = False; |
| noisy = True; |
| verbosity = 0; |
| blockSize100k = 9; |
| testFailsExist = False; |
| unzFailsExist = False; |
| numFileNames = 0; |
| numFilesProcessed = 0; |
| workFactor = 30; |
| deleteOutputOnInterrupt = False; |
| exitValue = 0; |
| i = j = 0; /* avoid bogus warning from egcs-1.1.X */ |
| |
| /*-- Set up signal handlers for mem access errors --*/ |
| signal (SIGSEGV, mySIGSEGVorSIGBUScatcher); |
| #if BZ_UNIX |
| #ifndef __DJGPP__ |
| signal (SIGBUS, mySIGSEGVorSIGBUScatcher); |
| #endif |
| #endif |
| |
| copyFileName ( inName, "(none)" ); |
| copyFileName ( outName, "(none)" ); |
| |
| copyFileName ( progNameReally, argv[0] ); |
| progName = &progNameReally[0]; |
| for (tmp = &progNameReally[0]; *tmp != '\0'; tmp++) |
| if (*tmp == PATH_SEP) progName = tmp + 1; |
| |
| |
| /*-- Copy flags from env var BZIP2, and |
| expand filename wildcards in arg list. |
| --*/ |
| argList = NULL; |
| addFlagsFromEnvVar ( &argList, "BZIP2" ); |
| addFlagsFromEnvVar ( &argList, "BZIP" ); |
| for (i = 1; i <= argc-1; i++) |
| APPEND_FILESPEC(argList, argv[i]); |
| |
| |
| /*-- Find the length of the longest filename --*/ |
| longestFileName = 7; |
| numFileNames = 0; |
| decode = True; |
| for (aa = argList; aa != NULL; aa = aa->link) { |
| if (ISFLAG("--")) { decode = False; continue; } |
| if (aa->name[0] == '-' && decode) continue; |
| numFileNames++; |
| if (longestFileName < (Int32)strlen(aa->name) ) |
| longestFileName = (Int32)strlen(aa->name); |
| } |
| |
| |
| /*-- Determine source modes; flag handling may change this too. --*/ |
| if (numFileNames == 0) |
| srcMode = SM_I2O; else srcMode = SM_F2F; |
| |
| |
| /*-- Determine what to do (compress/uncompress/test/cat). --*/ |
| /*-- Note that subsequent flag handling may change this. --*/ |
| opMode = OM_Z; |
| |
| if ( (strstr ( progName, "unzip" ) != 0) || |
| (strstr ( progName, "UNZIP" ) != 0) ) |
| opMode = OM_UNZ; |
| |
| if ( (strstr ( progName, "z2cat" ) != 0) || |
| (strstr ( progName, "Z2CAT" ) != 0) || |
| (strstr ( progName, "zcat" ) != 0) || |
| (strstr ( progName, "ZCAT" ) != 0) ) { |
| opMode = OM_UNZ; |
| srcMode = (numFileNames == 0) ? SM_I2O : SM_F2O; |
| } |
| |
| |
| /*-- Look at the flags. --*/ |
| for (aa = argList; aa != NULL; aa = aa->link) { |
| if (ISFLAG("--")) break; |
| if (aa->name[0] == '-' && aa->name[1] != '-') { |
| for (j = 1; aa->name[j] != '\0'; j++) { |
| switch (aa->name[j]) { |
| case 'c': srcMode = SM_F2O; break; |
| case 'd': opMode = OM_UNZ; break; |
| case 'z': opMode = OM_Z; break; |
| case 'f': forceOverwrite = True; break; |
| case 't': opMode = OM_TEST; break; |
| case 'k': keepInputFiles = True; break; |
| case 's': smallMode = True; break; |
| case 'q': noisy = False; break; |
| case '1': blockSize100k = 1; break; |
| case '2': blockSize100k = 2; break; |
| case '3': blockSize100k = 3; break; |
| case '4': blockSize100k = 4; break; |
| case '5': blockSize100k = 5; break; |
| case '6': blockSize100k = 6; break; |
| case '7': blockSize100k = 7; break; |
| case '8': blockSize100k = 8; break; |
| case '9': blockSize100k = 9; break; |
| case 'V': |
| case 'L': license(); break; |
| case 'v': verbosity++; break; |
| case 'h': usage ( progName ); |
| exit ( 0 ); |
| break; |
| default: fprintf ( stderr, "%s: Bad flag `%s'\n", |
| progName, aa->name ); |
| usage ( progName ); |
| exit ( 1 ); |
| break; |
| } |
| } |
| } |
| } |
| |
| /*-- And again ... --*/ |
| for (aa = argList; aa != NULL; aa = aa->link) { |
| if (ISFLAG("--")) break; |
| if (ISFLAG("--stdout")) srcMode = SM_F2O; else |
| if (ISFLAG("--decompress")) opMode = OM_UNZ; else |
| if (ISFLAG("--compress")) opMode = OM_Z; else |
| if (ISFLAG("--force")) forceOverwrite = True; else |
| if (ISFLAG("--test")) opMode = OM_TEST; else |
| if (ISFLAG("--keep")) keepInputFiles = True; else |
| if (ISFLAG("--small")) smallMode = True; else |
| if (ISFLAG("--quiet")) noisy = False; else |
| if (ISFLAG("--version")) license(); else |
| if (ISFLAG("--license")) license(); else |
| if (ISFLAG("--exponential")) workFactor = 1; else |
| if (ISFLAG("--repetitive-best")) redundant(aa->name); else |
| if (ISFLAG("--repetitive-fast")) redundant(aa->name); else |
| if (ISFLAG("--verbose")) verbosity++; else |
| if (ISFLAG("--help")) { usage ( progName ); exit ( 0 ); } |
| else |
| if (strncmp ( aa->name, "--", 2) == 0) { |
| fprintf ( stderr, "%s: Bad flag `%s'\n", progName, aa->name ); |
| usage ( progName ); |
| exit ( 1 ); |
| } |
| } |
| |
| if (verbosity > 4) verbosity = 4; |
| if (opMode == OM_Z && smallMode && blockSize100k > 2) |
| blockSize100k = 2; |
| |
| if (opMode == OM_TEST && srcMode == SM_F2O) { |
| fprintf ( stderr, "%s: -c and -t cannot be used together.\n", |
| progName ); |
| exit ( 1 ); |
| } |
| |
| if (srcMode == SM_F2O && numFileNames == 0) |
| srcMode = SM_I2O; |
| |
| if (opMode != OM_Z) blockSize100k = 0; |
| |
| if (srcMode == SM_F2F) { |
| signal (SIGINT, mySignalCatcher); |
| signal (SIGTERM, mySignalCatcher); |
| # if BZ_UNIX |
| signal (SIGHUP, mySignalCatcher); |
| # endif |
| } |
| |
| if (opMode == OM_Z) { |
| if (srcMode == SM_I2O) { |
| compress ( NULL ); |
| } else { |
| decode = True; |
| for (aa = argList; aa != NULL; aa = aa->link) { |
| if (ISFLAG("--")) { decode = False; continue; } |
| if (aa->name[0] == '-' && decode) continue; |
| numFilesProcessed++; |
| compress ( aa->name ); |
| } |
| } |
| } |
| else |
| |
| if (opMode == OM_UNZ) { |
| unzFailsExist = False; |
| if (srcMode == SM_I2O) { |
| uncompress ( NULL ); |
| } else { |
| decode = True; |
| for (aa = argList; aa != NULL; aa = aa->link) { |
| if (ISFLAG("--")) { decode = False; continue; } |
| if (aa->name[0] == '-' && decode) continue; |
| numFilesProcessed++; |
| uncompress ( aa->name ); |
| } |
| } |
| if (unzFailsExist) { |
| setExit(2); |
| exit(exitValue); |
| } |
| } |
| |
| else { |
| testFailsExist = False; |
| if (srcMode == SM_I2O) { |
| testf ( NULL ); |
| } else { |
| decode = True; |
| for (aa = argList; aa != NULL; aa = aa->link) { |
| if (ISFLAG("--")) { decode = False; continue; } |
| if (aa->name[0] == '-' && decode) continue; |
| numFilesProcessed++; |
| testf ( aa->name ); |
| } |
| } |
| if (testFailsExist && noisy) { |
| fprintf ( stderr, |
| "\n" |
| "You can use the `bzip2recover' program to attempt to recover\n" |
| "data from undamaged sections of corrupted files.\n\n" |
| ); |
| setExit(2); |
| exit(exitValue); |
| } |
| } |
| |
| /* Free the argument list memory to mollify leak detectors |
| (eg) Purify, Checker. Serves no other useful purpose. |
| */ |
| aa = argList; |
| while (aa != NULL) { |
| Cell* aa2 = aa->link; |
| if (aa->name != NULL) free(aa->name); |
| free(aa); |
| aa = aa2; |
| } |
| |
| return exitValue; |
| } |
| |
| |
| /*-----------------------------------------------------------*/ |
| /*--- end bzip2.c ---*/ |
| /*-----------------------------------------------------------*/ |
