2112 lines
		
	
	
		
			60 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			2112 lines
		
	
	
		
			60 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| 
 | |
| /*-----------------------------------------------------------*/
 | |
| /*--- 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-2005 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@bzip.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.
 | |
|   This is the default.
 | |
| --*/
 | |
| #define BZ_UNIX      1
 | |
| 
 | |
| /*--
 | |
|   Win32, as seen by Jacob Navia's excellent
 | |
|   port of (Chris Fraser & David Hanson)'s excellent
 | |
|   lcc compiler.  Or with MS Visual C.
 | |
|   This is selected automatically if compiled by a compiler which
 | |
|   defines _WIN32, not including the Cygwin GCC.
 | |
| --*/
 | |
| #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 <fcntl.h>
 | |
| #   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_STAT     stat
 | |
| #   define MY_S_ISREG  S_ISREG
 | |
| #   define MY_S_ISDIR  S_ISDIR
 | |
| 
 | |
| #   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
 | |
| #     undef MY_STAT
 | |
| #     define MY_LSTAT stat
 | |
| #     define MY_STAT 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 /* BZ_UNIX */
 | |
| 
 | |
| 
 | |
| 
 | |
| #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_ISREG(x)  ((x) & _S_IFREG)
 | |
| #   define MY_S_ISDIR(x)  ((x) & _S_IFDIR)
 | |
| 
 | |
| #   define APPEND_FLAG(root, name) \
 | |
|       root=snocString((root), (name))
 | |
| 
 | |
| #   define APPEND_FILESPEC(root, name)                \
 | |
|       root = snocString ((root), (name))
 | |
| 
 | |
| #   define SET_BINARY_MODE(fd)                        \
 | |
|       do {                                            \
 | |
|          int retVal = setmode ( fileno ( fd ),        \
 | |
|                                 O_BINARY );           \
 | |
|          ERROR_IF_MINUS_ONE ( retVal );               \
 | |
|       } while ( 0 )
 | |
| 
 | |
| #endif /* BZ_LCCWIN32 */
 | |
| 
 | |
| 
 | |
| /*---------------------------------------------*/
 | |
| /*--
 | |
|   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
 | |
| Bool uInt64_isZero ( UInt64* n )
 | |
| {
 | |
|    Int32 i;
 | |
|    for (i = 0; i < 8; i++)
 | |
|       if (n->b[i] != 0) return 0;
 | |
|    return 1;
 | |
| }
 | |
| 
 | |
| 
 | |
| /* Divide *n by 10, and return the remainder.  */
 | |
| static 
 | |
| Int32 uInt64_qrm10 ( UInt64* n )
 | |
| {
 | |
|    UInt32 rem, tmp;
 | |
|    Int32  i;
 | |
|    rem = 0;
 | |
|    for (i = 7; i >= 0; i--) {
 | |
|       tmp = rem * 256 + n->b[i];
 | |
|       n->b[i] = tmp / 10;
 | |
|       rem = tmp % 10;
 | |
|    }
 | |
|    return rem;
 | |
| }
 | |
| 
 | |
| 
 | |
| /* ... 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 );
 | |
|       outputHandleJustInCase = NULL;
 | |
|       if (ret == EOF) goto errhandler_io;
 | |
|    }
 | |
|    outputHandleJustInCase = NULL;
 | |
|    if (ferror(stream)) goto errhandler_io;
 | |
|    ret = fclose ( stream );
 | |
|    if (ret == EOF) goto errhandler_io;
 | |
| 
 | |
|    if (verbosity >= 1) {
 | |
|       if (nbytes_in_lo32 == 0 && nbytes_in_hi32 == 0) {
 | |
| 	 fprintf ( stderr, " no data compressed.\n");
 | |
|       } else {
 | |
| 	 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;
 | |
|    void*   unusedTmpV;
 | |
|    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 trycat;
 | |
|          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, &unusedTmpV, &nUnused );
 | |
|       if (bzerr != BZ_OK) panic ( "decompress:bzReadGetUnused" );
 | |
| 
 | |
|       unusedTmp = (UChar*)unusedTmpV;
 | |
|       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;
 | |
|    }
 | |
| 
 | |
|    closeok:
 | |
|    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 );
 | |
|       outputHandleJustInCase = NULL;
 | |
|       if (ret == EOF) goto errhandler_io;
 | |
|    }
 | |
|    outputHandleJustInCase = NULL;
 | |
|    if (verbosity >= 2) fprintf ( stderr, "\n    " );
 | |
|    return True;
 | |
| 
 | |
|    trycat: 
 | |
|    if (forceOverwrite) {
 | |
|       rewind(zStream);
 | |
|       while (True) {
 | |
|       	 if (myfeof(zStream)) break;
 | |
|       	 nread = fread ( obuf, sizeof(UChar), 5000, zStream );
 | |
|       	 if (ferror(zStream)) goto errhandler_io;
 | |
|       	 if (nread > 0) fwrite ( obuf, sizeof(UChar), nread, stream );
 | |
|       	 if (ferror(stream)) goto errhandler_io;
 | |
|       }
 | |
|       goto closeok;
 | |
|    }
 | |
|   
 | |
|    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;
 | |
|    void*   unusedTmpV;
 | |
|    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, &unusedTmpV, &nUnused );
 | |
|       if (bzerr != BZ_OK) panic ( "test:bzReadGetUnused" );
 | |
| 
 | |
|       unusedTmp = (UChar*)unusedTmpV;
 | |
|       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;
 | |
|    struct MY_STAT statBuf;
 | |
| 
 | |
|    if ( srcMode == SM_F2F 
 | |
|         && opMode != OM_TEST
 | |
|         && deleteOutputOnInterrupt ) {
 | |
| 
 | |
|       /* Check whether input file still exists.  Delete output file
 | |
|          only if input exists to avoid loss of data.  Joerg Prante, 5
 | |
|          January 2002.  (JRS 06-Jan-2002: other changes in 1.0.2 mean
 | |
|          this is less likely to happen.  But to be ultra-paranoid, we
 | |
|          do the check anyway.)  */
 | |
|       retVal = MY_STAT ( inName, &statBuf );
 | |
|       if (retVal == 0) {
 | |
|          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 );
 | |
|       } else {
 | |
|          fprintf ( stderr,
 | |
|                    "%s: WARNING: deletion of output file suppressed\n",
 | |
|                     progName );
 | |
|          fprintf ( stderr,
 | |
|                    "%s:    since input file no longer exists.  Output file\n",
 | |
|                    progName );
 | |
|          fprintf ( stderr,
 | |
|                    "%s:    `%s' may be incomplete.\n",
 | |
|                    progName, outName );
 | |
|          fprintf ( stderr, 
 | |
|                    "%s:    I suggest doing an integrity test (bzip2 -tv)"
 | |
|                    " of it.\n",
 | |
|                    progName );
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    if (noisy && numFileNames > 0 && numFilesProcessed < numFileNames) {
 | |
|       fprintf ( stderr, 
 | |
|                 "%s: WARNING: some files have not been processed:\n"
 | |
|                 "%s:    %d specified on command line, %d not processed yet.\n\n",
 | |
|                 progName, 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@bzip.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 )
 | |
| {
 | |
|   if (noisy) {
 | |
|     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@bzip.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@bzip.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                   ---*/
 | |
| /*---------------------------------------------------*/
 | |
| 
 | |
| /* All rather crufty.  The main problem is that input files
 | |
|    are stat()d multiple times before use.  This should be
 | |
|    cleaned up. 
 | |
| */
 | |
| 
 | |
| /*---------------------------------------------*/
 | |
| 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;
 | |
| }
 | |
| 
 | |
| 
 | |
| /*---------------------------------------------*/
 | |
| /* Open an output file safely with O_EXCL and good permissions.
 | |
|    This avoids a race condition in versions < 1.0.2, in which
 | |
|    the file was first opened and then had its interim permissions
 | |
|    set safely.  We instead use open() to create the file with
 | |
|    the interim permissions required. (--- --- rw-).
 | |
| 
 | |
|    For non-Unix platforms, if we are not worrying about
 | |
|    security issues, simple this simply behaves like fopen.
 | |
| */
 | |
| FILE* fopen_output_safely ( Char* name, const char* mode )
 | |
| {
 | |
| #  if BZ_UNIX
 | |
|    FILE*     fp;
 | |
|    IntNative fh;
 | |
|    fh = open(name, O_WRONLY|O_CREAT|O_EXCL, S_IWUSR|S_IRUSR);
 | |
|    if (fh == -1) return NULL;
 | |
|    fp = fdopen(fh, mode);
 | |
|    if (fp == NULL) close(fh);
 | |
|    return fp;
 | |
| #  else
 | |
|    return fopen(name, mode);
 | |
| #  endif
 | |
| }
 | |
| 
 | |
| 
 | |
| /*---------------------------------------------*/
 | |
| /*--
 | |
|   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_ISREG(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);
 | |
| }
 | |
| 
 | |
| 
 | |
| /*---------------------------------------------*/
 | |
| /* Copy modification date, access date, permissions and owner from the
 | |
|    source to destination file.  We have to copy this meta-info off
 | |
|    into fileMetaInfo before starting to compress / decompress it,
 | |
|    because doing it afterwards means we get the wrong access time.
 | |
| 
 | |
|    To complicate matters, in compress() and decompress() below, the
 | |
|    sequence of tests preceding the call to saveInputFileMetaInfo()
 | |
|    involves calling fileExists(), which in turn establishes its result
 | |
|    by attempting to fopen() the file, and if successful, immediately
 | |
|    fclose()ing it again.  So we have to assume that the fopen() call
 | |
|    does not cause the access time field to be updated.
 | |
| 
 | |
|    Reading of the man page for stat() (man 2 stat) on RedHat 7.2 seems
 | |
|    to imply that merely doing open() will not affect the access time.
 | |
|    Therefore we merely need to hope that the C library only does
 | |
|    open() as a result of fopen(), and not any kind of read()-ahead
 | |
|    cleverness.
 | |
| 
 | |
|    It sounds pretty fragile to me.  Whether this carries across
 | |
|    robustly to arbitrary Unix-like platforms (or even works robustly
 | |
|    on this one, RedHat 7.2) is unknown to me.  Nevertheless ...  
 | |
| */
 | |
| #if BZ_UNIX
 | |
| static 
 | |
| struct MY_STAT fileMetaInfo;
 | |
| #endif
 | |
| 
 | |
| static 
 | |
| void saveInputFileMetaInfo ( Char *srcName )
 | |
| {
 | |
| #  if BZ_UNIX
 | |
|    IntNative retVal;
 | |
|    /* Note use of stat here, not lstat. */
 | |
|    retVal = MY_STAT( srcName, &fileMetaInfo );
 | |
|    ERROR_IF_NOT_ZERO ( retVal );
 | |
| #  endif
 | |
| }
 | |
| 
 | |
| 
 | |
| static 
 | |
| void applySavedMetaInfoToOutputFile ( Char *dstName )
 | |
| {
 | |
| #  if BZ_UNIX
 | |
|    IntNative      retVal;
 | |
|    struct utimbuf uTimBuf;
 | |
| 
 | |
|    uTimBuf.actime = fileMetaInfo.st_atime;
 | |
|    uTimBuf.modtime = fileMetaInfo.st_mtime;
 | |
| 
 | |
|    retVal = chmod ( dstName, fileMetaInfo.st_mode );
 | |
|    ERROR_IF_NOT_ZERO ( retVal );
 | |
| 
 | |
|    retVal = utime ( dstName, &uTimBuf );
 | |
|    ERROR_IF_NOT_ZERO ( retVal );
 | |
| 
 | |
|    retVal = chown ( dstName, fileMetaInfo.st_uid, fileMetaInfo.st_gid );
 | |
|    /* chown() will in many cases return with EPERM, which can
 | |
|       be safely ignored.
 | |
|    */
 | |
| #  endif
 | |
| }
 | |
| 
 | |
| 
 | |
| /*---------------------------------------------*/
 | |
| static 
 | |
| Bool containsDubiousChars ( Char* name )
 | |
| {
 | |
| #  if BZ_UNIX
 | |
|    /* On unix, files can contain any characters and the file expansion
 | |
|     * is performed by the shell.
 | |
|     */
 | |
|    return False;
 | |
| #  else /* ! BZ_UNIX */
 | |
|    /* On non-unix (Win* platforms), wildcard characters are not allowed in 
 | |
|     * filenames.
 | |
|     */
 | |
|    for (; *name != '\0'; name++)
 | |
|       if (*name == '?' || *name == '*') return True;
 | |
|    return False;
 | |
| #  endif /* BZ_UNIX */
 | |
| }
 | |
| 
 | |
| 
 | |
| /*---------------------------------------------*/
 | |
| #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;
 | |
|    struct MY_STAT statBuf;
 | |
| 
 | |
|    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 || srcMode == SM_F2O ) {
 | |
|       MY_STAT(inName, &statBuf);
 | |
|       if ( MY_S_ISDIR(statBuf.st_mode) ) {
 | |
|          fprintf( stderr,
 | |
|                   "%s: Input file %s is a directory.\n",
 | |
|                   progName,inName);
 | |
|          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 && fileExists ( outName ) ) {
 | |
|       if (forceOverwrite) {
 | |
| 	 remove(outName);
 | |
|       } else {
 | |
| 	 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;
 | |
|    }
 | |
| 
 | |
|    if ( srcMode == SM_F2F ) {
 | |
|       /* Save the file's meta-info before we open it.  Doing it later
 | |
|          means we mess up the access times. */
 | |
|       saveInputFileMetaInfo ( inName );
 | |
|    }
 | |
| 
 | |
|    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_output_safely ( 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;
 | |
|          };
 | |
|          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 ) {
 | |
|       applySavedMetaInfoToOutputFile ( 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;
 | |
|    struct MY_STAT statBuf;
 | |
| 
 | |
|    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 || srcMode == SM_F2O ) {
 | |
|       MY_STAT(inName, &statBuf);
 | |
|       if ( MY_S_ISDIR(statBuf.st_mode) ) {
 | |
|          fprintf( stderr,
 | |
|                   "%s: Input file %s is a directory.\n",
 | |
|                   progName,inName);
 | |
|          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 && fileExists ( outName ) ) {
 | |
|       if (forceOverwrite) {
 | |
| 	remove(outName);
 | |
|       } else {
 | |
|         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;
 | |
|    }
 | |
| 
 | |
|    if ( srcMode == SM_F2F ) {
 | |
|       /* Save the file's meta-info before we open it.  Doing it later
 | |
|          means we mess up the access times. */
 | |
|       saveInputFileMetaInfo ( inName );
 | |
|    }
 | |
| 
 | |
|    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_output_safely ( 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;
 | |
|          };
 | |
|          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 ) {
 | |
|          applySavedMetaInfoToOutputFile ( 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;
 | |
|    struct MY_STAT statBuf;
 | |
| 
 | |
|    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;
 | |
|    }
 | |
|    if ( srcMode != SM_I2O ) {
 | |
|       MY_STAT(inName, &statBuf);
 | |
|       if ( MY_S_ISDIR(statBuf.st_mode) ) {
 | |
|          fprintf( stderr,
 | |
|                   "%s: Input file %s is a directory.\n",
 | |
|                   progName,inName);
 | |
|          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. ---*/
 | |
|    outputHandleJustInCase = NULL;
 | |
|    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-2005 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"
 | |
|       "   --fast              alias for -1\n"
 | |
|       "   --best              alias for -9\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;
 | |
|    }
 | |
| 
 | |
|    if(strstr ( progName, "small" ) != 0) {
 | |
|    	smallMode = True;
 | |
|    }
 | |
| 
 | |
| 
 | |
|    /*-- 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("--fast"))              blockSize100k = 1;          else
 | |
|       if (ISFLAG("--best"))              blockSize100k = 9;          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 ---*/
 | |
| /*-----------------------------------------------------------*/
 | 
