2042 lines
		
	
	
		
			57 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			2042 lines
		
	
	
		
			57 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*	$NetBSD: bzip2.c,v 1.8 2009/04/11 11:10:43 lukem Exp $	*/
 | 
						|
 | 
						|
 | 
						|
/*-----------------------------------------------------------*/
 | 
						|
/*--- A block-sorting, lossless compressor        bzip2.c ---*/
 | 
						|
/*-----------------------------------------------------------*/
 | 
						|
 | 
						|
/* ------------------------------------------------------------------
 | 
						|
   This file is part of bzip2/libbzip2, a program and library for
 | 
						|
   lossless, block-sorting data compression.
 | 
						|
 | 
						|
   bzip2/libbzip2 version 1.0.5 of 10 December 2007
 | 
						|
   Copyright (C) 1996-2007 Julian Seward <jseward@bzip.org>
 | 
						|
 | 
						|
   Please read the WARNING, DISCLAIMER and PATENTS sections in the 
 | 
						|
   README file.
 | 
						|
 | 
						|
   This program is released under the terms of the license contained
 | 
						|
   in the file LICENSE.
 | 
						|
   ------------------------------------------------------------------ */
 | 
						|
 | 
						|
 | 
						|
/* 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                 ( const 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*, const Char* );
 | 
						|
static void*   myMalloc     ( Int32 );
 | 
						|
static void    applySavedFileAttrToOutputFile ( IntNative fd );
 | 
						|
 | 
						|
 | 
						|
static FILE* fopen_output_safely ( Char*, const char* );
 | 
						|
 | 
						|
/*---------------------------------------------------*/
 | 
						|
/*--- 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) {
 | 
						|
      Int32 fd = fileno ( zStream );
 | 
						|
      if (fd < 0) goto errhandler_io;
 | 
						|
      applySavedFileAttrToOutputFile ( fd );
 | 
						|
      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 = NULL;
 | 
						|
   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, (void*)(&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;
 | 
						|
   if (stream != stdout) {
 | 
						|
      Int32 fd = fileno ( stream );
 | 
						|
      if (fd < 0) goto errhandler_io;
 | 
						|
      applySavedFileAttrToOutputFile ( fd );
 | 
						|
   }
 | 
						|
   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 = NULL;
 | 
						|
   UChar*  unusedTmp = NULL;
 | 
						|
 | 
						|
   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*)(&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 ( const 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);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/*---------------------------------------------*/
 | 
						|
#ifndef SMALL
 | 
						|
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 ); }
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
/*---------------------------------------------*/
 | 
						|
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, const 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.
 | 
						|
*/
 | 
						|
static
 | 
						|
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 applySavedTimeInfoToOutputFile ( Char *dstName )
 | 
						|
{
 | 
						|
#  if BZ_UNIX
 | 
						|
   IntNative      retVal;
 | 
						|
   struct utimbuf uTimBuf;
 | 
						|
 | 
						|
   uTimBuf.actime = fileMetaInfo.st_atime;
 | 
						|
   uTimBuf.modtime = fileMetaInfo.st_mtime;
 | 
						|
 | 
						|
   retVal = utime ( dstName, &uTimBuf );
 | 
						|
   ERROR_IF_NOT_ZERO ( retVal );
 | 
						|
#  endif
 | 
						|
}
 | 
						|
 | 
						|
static 
 | 
						|
void applySavedFileAttrToOutputFile ( IntNative fd )
 | 
						|
{
 | 
						|
#  if BZ_UNIX
 | 
						|
   IntNative retVal;
 | 
						|
 | 
						|
   retVal = fchmod ( fd, fileMetaInfo.st_mode );
 | 
						|
   ERROR_IF_NOT_ZERO ( retVal );
 | 
						|
 | 
						|
   (void) fchown ( fd, 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
 | 
						|
 | 
						|
const Char* zSuffix[BZ_N_SUFFIX_PAIRS] 
 | 
						|
   = { ".bz2", ".bz", ".tbz2", ".tbz" };
 | 
						|
const Char* unzSuffix[BZ_N_SUFFIX_PAIRS] 
 | 
						|
   = { "", "", ".tar", ".tar" };
 | 
						|
 | 
						|
static 
 | 
						|
Bool hasSuffix ( Char* s, const 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, 
 | 
						|
                 const Char* oldSuffix, 
 | 
						|
                 const 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 ) {
 | 
						|
      applySavedTimeInfoToOutputFile ( 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 ) {
 | 
						|
         applySavedTimeInfoToOutputFile ( 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-2007 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.5 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, const 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 */
 | 
						|
 | 
						|
#ifndef SMALL
 | 
						|
   /*-- Set up signal handlers for mem access errors --*/
 | 
						|
   signal (SIGSEGV, mySIGSEGVorSIGBUScatcher);
 | 
						|
#  if BZ_UNIX
 | 
						|
#  ifndef __DJGPP__
 | 
						|
   signal (SIGBUS,  mySIGSEGVorSIGBUScatcher);
 | 
						|
#  endif
 | 
						|
#  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("--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 ---*/
 | 
						|
/*-----------------------------------------------------------*/
 |