lib/libz/test/minigzip.c

   1 /* minigzip.c -- simulate gzip using the zlib compression library
   2  * Copyright (C) 1995-2006, 2010, 2011 Jean-loup Gailly.
   3  * For conditions of distribution and use, see copyright notice in zlib.h
   4  */
   5
   6 /*
   7  * minigzip is a minimal implementation of the gzip utility. This is
   8  * only an example of using zlib and isn't meant to replace the
   9  * full-featured gzip. No attempt is made to deal with file systems
  10  * limiting names to 14 or 8+3 characters, etc... Error checking is
  11  * very limited. So use minigzip only for testing; use gzip for the
  12  * real thing. On MSDOS, use only on file names without extension
  13  * or in pipe mode.
  14  */
  15
  16 /* @(#) $Id$ */
  17
  18 #include "zlib.h"
  19 #include <stdio.h>
  20
  21 #ifdef STDC
  22 #  include <string.h>
  23 #  include <stdlib.h>
  24 #endif
  25
  26 #ifdef USE_MMAP
  27 #  include <sys/types.h>
  28 #  include <sys/mman.h>
  29 #  include <sys/stat.h>
  30 #endif
  31
  32 #if defined(MSDOS) || defined(OS2) || defined(WIN32) || defined(__CYGWIN__)
  33 #  include <fcntl.h>
  34 #  include <io.h>
  35 #  ifdef UNDER_CE
  36 #    include <stdlib.h>
  37 #  endif
  38 #  define SET_BINARY_MODE(file) setmode(fileno(file), O_BINARY)
  39 #else
  40 #  define SET_BINARY_MODE(file)
  41 #endif
  42
  43 #ifdef _MSC_VER
  44 #  define snprintf _snprintf
  45 #endif
  46
  47 #ifdef VMS
  48 #  define unlink delete
  49 #  define GZ_SUFFIX "-gz"
  50 #endif
  51 #ifdef RISCOS
  52 #  define unlink remove
  53 #  define GZ_SUFFIX "-gz"
  54 #  define fileno(file) file->__file
  55 #endif
  56 #if defined(__MWERKS__) && __dest_os != __be_os && __dest_os != __win32_os
  57 #  include <unix.h> /* for fileno */
  58 #endif
  59
  60 #if !defined(Z_HAVE_UNISTD_H) && !defined(_LARGEFILE64_SOURCE)
  61 #ifndef WIN32 /* unlink already in stdio.h for WIN32 */
  62   extern int unlink OF((const char *));
  63 #endif
  64 #endif
  65
  66 #if defined(UNDER_CE)
  67 #  include <windows.h>
  68 #  define perror(s) pwinerror(s)
  69
  70 /* Map the Windows error number in ERROR to a locale-dependent error
  71    message string and return a pointer to it.  Typically, the values
  72    for ERROR come from GetLastError.
  73
  74    The string pointed to shall not be modified by the application,
  75    but may be overwritten by a subsequent call to strwinerror
  76
  77    The strwinerror function does not change the current setting
  78    of GetLastError.  */
  79
  80 static char *strwinerror (error)
  81      DWORD error;
  82 {
  83     static char buf[1024];
  84
  85     wchar_t *msgbuf;
  86     DWORD lasterr = GetLastError();
  87     DWORD chars = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM
  88         | FORMAT_MESSAGE_ALLOCATE_BUFFER,
  89         NULL,
  90         error,
  91         0, /* Default language */
  92         (LPVOID)&msgbuf,
  93         0,
  94         NULL);
  95     if (chars != 0) {
  96         /* If there is an \r\n appended, zap it.  */
  97         if (chars >= 2
  98             && msgbuf[chars - 2] == '\r' && msgbuf[chars - 1] == '\n') {
  99             chars -= 2;
 100             msgbuf[chars] = 0;
 101         }
 102
 103         if (chars > sizeof (buf) - 1) {
 104             chars = sizeof (buf) - 1;
 105             msgbuf[chars] = 0;
 106         }
 107
 108         wcstombs(buf, msgbuf, chars + 1);
 109         LocalFree(msgbuf);
 110     }
 111     else {
 112         sprintf(buf, "unknown win32 error (%ld)", error);
 113     }
 114
 115     SetLastError(lasterr);
 116     return buf;
 117 }
 118
 119 static void pwinerror (s)
 120     const char *s;
 121 {
 122     if (s && *s)
 123         fprintf(stderr, "%s: %s\n", s, strwinerror(GetLastError ()));
 124     else
 125         fprintf(stderr, "%s\n", strwinerror(GetLastError ()));
 126 }
 127
 128 #endif /* UNDER_CE */
 129
 130 #ifndef GZ_SUFFIX
 131 #  define GZ_SUFFIX ".gz"
 132 #endif
 133 #define SUFFIX_LEN (sizeof(GZ_SUFFIX)-1)
 134
 135 #define BUFLEN      16384
 136 #define MAX_NAME_LEN 1024
 137
 138 #ifdef MAXSEG_64K
 139 #  define local static
 140    /* Needed for systems with limitation on stack size. */
 141 #else
 142 #  define local
 143 #endif
 144
 145 #ifdef Z_SOLO
 146 /* for Z_SOLO, create simplified gz* functions using deflate and inflate */
 147
 148 #if defined(Z_HAVE_UNISTD_H) || defined(Z_LARGE)
 149 #  include <unistd.h>       /* for unlink() */
 150 #endif
 151
 152 void *myalloc OF((void *, unsigned, unsigned));
 153 void myfree OF((void *, void *));
 154
 155 void *myalloc(q, n, m)
 156     void *q;
 157     unsigned n, m;
 158 {
 159     q = Z_NULL;
 160     return calloc(n, m);
 161 }
 162
 163 void myfree(q, p)
 164     void *q, *p;
 165 {
 166     q = Z_NULL;
 167     free(p);
 168 }
 169
 170 typedef struct gzFile_s {
 171     FILE *file;
 172     int write;
 173     int err;
 174     char *msg;
 175     z_stream strm;
 176 } *gzFile;
 177
 178 gzFile gzopen OF((const char *, const char *));
 179 gzFile gzdopen OF((int, const char *));
 180 gzFile gz_open OF((const char *, int, const char *));
 181
 182 gzFile gzopen(path, mode)
 183 const char *path;
 184 const char *mode;
 185 {
 186     return gz_open(path, -1, mode);
 187 }
 188
 189 gzFile gzdopen(fd, mode)
 190 int fd;
 191 const char *mode;
 192 {
 193     return gz_open(NULL, fd, mode);
 194 }
 195
 196 gzFile gz_open(path, fd, mode)
 197     const char *path;
 198     int fd;
 199     const char *mode;
 200 {
 201     gzFile gz;
 202     int ret;
 203
 204     gz = malloc(sizeof(struct gzFile_s));
 205     if (gz == NULL)
 206         return NULL;
 207     gz->write = strchr(mode, 'w') != NULL;
 208     gz->strm.zalloc = myalloc;
 209     gz->strm.zfree = myfree;
 210     gz->strm.opaque = Z_NULL;
 211     if (gz->write)
 212         ret = deflateInit2(&(gz->strm), -1, 8, 15 + 16, 8, 0);
 213     else {
 214         gz->strm.next_in = 0;
 215         gz->strm.avail_in = Z_NULL;
 216         ret = inflateInit2(&(gz->strm), 15 + 16);
 217     }
 218     if (ret != Z_OK) {
 219         free(gz);
 220         return NULL;
 221     }
 222     gz->file = path == NULL ? fdopen(fd, gz->write ? "wb" : "rb") :
 223                               fopen(path, gz->write ? "wb" : "rb");
 224     if (gz->file == NULL) {
 225         gz->write ? deflateEnd(&(gz->strm)) : inflateEnd(&(gz->strm));
 226         free(gz);
 227         return NULL;
 228     }
 229     gz->err = 0;
 230     gz->msg = "";
 231     return gz;
 232 }
 233
 234 int gzwrite OF((gzFile, const void *, unsigned));
 235
 236 int gzwrite(gz, buf, len)
 237     gzFile gz;
 238     const void *buf;
 239     unsigned len;
 240 {
 241     z_stream *strm;
 242     unsigned char out[BUFLEN];
 243
 244     if (gz == NULL || !gz->write)
 245         return 0;
 246     strm = &(gz->strm);
 247     strm->next_in = (void *)buf;
 248     strm->avail_in = len;
 249     do {
 250         strm->next_out = out;
 251         strm->avail_out = BUFLEN;
 252         (void)deflate(strm, Z_NO_FLUSH);
 253         fwrite(out, 1, BUFLEN - strm->avail_out, gz->file);
 254     } while (strm->avail_out == 0);
 255     return len;
 256 }
 257
 258 int gzread OF((gzFile, void *, unsigned));
 259
 260 int gzread(gz, buf, len)
 261     gzFile gz;
 262     void *buf;
 263     unsigned len;
 264 {
 265     int ret;
 266     unsigned got;
 267     unsigned char in[1];
 268     z_stream *strm;
 269
 270     if (gz == NULL || gz->write)
 271         return 0;
 272     if (gz->err)
 273         return 0;
 274     strm = &(gz->strm);
 275     strm->next_out = (void *)buf;
 276     strm->avail_out = len;
 277     do {
 278         got = fread(in, 1, 1, gz->file);
 279         if (got == 0)
 280             break;
 281         strm->next_in = in;
 282         strm->avail_in = 1;
 283         ret = inflate(strm, Z_NO_FLUSH);
 284         if (ret == Z_DATA_ERROR) {
 285             gz->err = Z_DATA_ERROR;
 286             gz->msg = strm->msg;
 287             return 0;
 288         }
 289         if (ret == Z_STREAM_END)
 290             inflateReset(strm);
 291     } while (strm->avail_out);
 292     return len - strm->avail_out;
 293 }
 294
 295 int gzclose OF((gzFile));
 296
 297 int gzclose(gz)
 298     gzFile gz;
 299 {
 300     z_stream *strm;
 301     unsigned char out[BUFLEN];
 302
 303     if (gz == NULL)
 304         return Z_STREAM_ERROR;
 305     strm = &(gz->strm);
 306     if (gz->write) {
 307         strm->next_in = Z_NULL;
 308         strm->avail_in = 0;
 309         do {
 310             strm->next_out = out;
 311             strm->avail_out = BUFLEN;
 312             (void)deflate(strm, Z_FINISH);
 313             fwrite(out, 1, BUFLEN - strm->avail_out, gz->file);
 314         } while (strm->avail_out == 0);
 315         deflateEnd(strm);
 316     }
 317     else
 318         inflateEnd(strm);
 319     fclose(gz->file);
 320     free(gz);
 321     return Z_OK;
 322 }
 323
 324 const char *gzerror OF((gzFile, int *));
 325
 326 const char *gzerror(gz, err)
 327     gzFile gz;
 328     int *err;
 329 {
 330     *err = gz->err;
 331     return gz->msg;
 332 }
 333
 334 #endif
 335
 336 char *prog;
 337
 338 void error            OF((const char *msg));
 339 void gz_compress      OF((FILE   *in, gzFile out));
 340 #ifdef USE_MMAP
 341 int  gz_compress_mmap OF((FILE   *in, gzFile out));
 342 #endif
 343 void gz_uncompress    OF((gzFile in, FILE   *out));
 344 void file_compress    OF((char  *file, char *mode));
 345 void file_uncompress  OF((char  *file));
 346 int  main             OF((int argc, char *argv[]));
 347
 348 /* ===========================================================================
 349  * Display error message and exit
 350  */
 351 void error(msg)
 352     const char *msg;
 353 {
 354     fprintf(stderr, "%s: %s\n", prog, msg);
 355     exit(1);
 356 }
 357
 358 /* ===========================================================================
 359  * Compress input to output then close both files.
 360  */
 361
 362 void gz_compress(in, out)
 363     FILE   *in;
 364     gzFile out;
 365 {
 366     local char buf[BUFLEN];
 367     int len;
 368     int err;
 369
 370 #ifdef USE_MMAP
 371     /* Try first compressing with mmap. If mmap fails (minigzip used in a
 372      * pipe), use the normal fread loop.
 373      */
 374     if (gz_compress_mmap(in, out) == Z_OK) return;
 375 #endif
 376     for (;;) {
 377         len = (int)fread(buf, 1, sizeof(buf), in);
 378         if (ferror(in)) {
 379             perror("fread");
 380             exit(1);
 381         }
 382         if (len == 0) break;
 383
 384         if (gzwrite(out, buf, (unsigned)len) != len) error(gzerror(out, &err));
 385     }
 386     fclose(in);
 387     if (gzclose(out) != Z_OK) error("failed gzclose");
 388 }
 389
 390 #ifdef USE_MMAP /* MMAP version, Miguel Albrecht <malbrech@eso.org> */
 391
 392 /* Try compressing the input file at once using mmap. Return Z_OK if
 393  * if success, Z_ERRNO otherwise.
 394  */
 395 int gz_compress_mmap(in, out)
 396     FILE   *in;
 397     gzFile out;
 398 {
 399     int len;
 400     int err;
 401     int ifd = fileno(in);
 402     caddr_t buf;    /* mmap'ed buffer for the entire input file */
 403     off_t buf_len;  /* length of the input file */
 404     struct stat sb;
 405
 406     /* Determine the size of the file, needed for mmap: */
 407     if (fstat(ifd, &sb) < 0) return Z_ERRNO;
 408     buf_len = sb.st_size;
 409     if (buf_len <= 0) return Z_ERRNO;
 410
 411     /* Now do the actual mmap: */
 412     buf = mmap((caddr_t) 0, buf_len, PROT_READ, MAP_SHARED, ifd, (off_t)0);
 413     if (buf == (caddr_t)(-1)) return Z_ERRNO;
 414
 415     /* Compress the whole file at once: */
 416     len = gzwrite(out, (char *)buf, (unsigned)buf_len);
 417
 418     if (len != (int)buf_len) error(gzerror(out, &err));
 419
 420     munmap(buf, buf_len);
 421     fclose(in);
 422     if (gzclose(out) != Z_OK) error("failed gzclose");
 423     return Z_OK;
 424 }
 425 #endif /* USE_MMAP */
 426
 427 /* ===========================================================================
 428  * Uncompress input to output then close both files.
 429  */
 430 void gz_uncompress(in, out)
 431     gzFile in;
 432     FILE   *out;
 433 {
 434     local char buf[BUFLEN];
 435     int len;
 436     int err;
 437
 438     for (;;) {
 439         len = gzread(in, buf, sizeof(buf));
 440         if (len < 0) error (gzerror(in, &err));
 441         if (len == 0) break;
 442
 443         if ((int)fwrite(buf, 1, (unsigned)len, out) != len) {
 444             error("failed fwrite");
 445         }
 446     }
 447     if (fclose(out)) error("failed fclose");
 448
 449     if (gzclose(in) != Z_OK) error("failed gzclose");
 450 }
 451
 452
 453 /* ===========================================================================
 454  * Compress the given file: create a corresponding .gz file and remove the
 455  * original.
 456  */
 457 void file_compress(file, mode)
 458     char  *file;
 459     char  *mode;
 460 {
 461     local char outfile[MAX_NAME_LEN];
 462     FILE  *in;
 463     gzFile out;
 464
 465     if (strlen(file) + strlen(GZ_SUFFIX) >= sizeof(outfile)) {
 466         fprintf(stderr, "%s: filename too long\n", prog);
 467         exit(1);
 468     }
 469
 470 #if !defined(NO_snprintf) && !defined(NO_vsnprintf)
 471     snprintf(outfile, sizeof(outfile), "%s%s", file, GZ_SUFFIX);
 472 #else
 473     strcpy(outfile, file);
 474     strcat(outfile, GZ_SUFFIX);
 475 #endif
 476
 477     in = fopen(file, "rb");
 478     if (in == NULL) {
 479         perror(file);
 480         exit(1);
 481     }
 482     out = gzopen(outfile, mode);
 483     if (out == NULL) {
 484         fprintf(stderr, "%s: can't gzopen %s\n", prog, outfile);
 485         exit(1);
 486     }
 487     gz_compress(in, out);
 488
 489     unlink(file);
 490 }
 491
 492
 493 /* ===========================================================================
 494  * Uncompress the given file and remove the original.
 495  */
 496 void file_uncompress(file)
 497     char  *file;
 498 {
 499     local char buf[MAX_NAME_LEN];
 500     char *infile, *outfile;
 501     FILE  *out;
 502     gzFile in;
 503     size_t len = strlen(file);
 504
 505     if (len + strlen(GZ_SUFFIX) >= sizeof(buf)) {
 506         fprintf(stderr, "%s: filename too long\n", prog);
 507         exit(1);
 508     }
 509
 510 #if !defined(NO_snprintf) && !defined(NO_vsnprintf)
 511     snprintf(buf, sizeof(buf), "%s", file);
 512 #else
 513     strcpy(buf, file);
 514 #endif
 515
 516     if (len > SUFFIX_LEN && strcmp(file+len-SUFFIX_LEN, GZ_SUFFIX) == 0) {
 517         infile = file;
 518         outfile = buf;
 519         outfile[len-3] = '\0';
 520     } else {
 521         outfile = file;
 522         infile = buf;
 523 #if !defined(NO_snprintf) && !defined(NO_vsnprintf)
 524         snprintf(buf + len, sizeof(buf) - len, "%s", GZ_SUFFIX);
 525 #else
 526         strcat(infile, GZ_SUFFIX);
 527 #endif
 528     }
 529     in = gzopen(infile, "rb");
 530     if (in == NULL) {
 531         fprintf(stderr, "%s: can't gzopen %s\n", prog, infile);
 532         exit(1);
 533     }
 534     out = fopen(outfile, "wb");
 535     if (out == NULL) {
 536         perror(file);
 537         exit(1);
 538     }
 539
 540     gz_uncompress(in, out);
 541
 542     unlink(infile);
 543 }
 544
 545
 546 /* ===========================================================================
 547  * Usage:  minigzip [-c] [-d] [-f] [-h] [-r] [-1 to -9] [files...]
 548  *   -c : write to standard output
 549  *   -d : decompress
 550  *   -f : compress with Z_FILTERED
 551  *   -h : compress with Z_HUFFMAN_ONLY
 552  *   -r : compress with Z_RLE
 553  *   -1 to -9 : compression level
 554  */
 555
 556 int main(argc, argv)
 557     int argc;
 558     char *argv[];
 559 {
 560     int copyout = 0;
 561     int uncompr = 0;
 562     gzFile file;
 563     char *bname, outmode[20];
 564
 565 #if !defined(NO_snprintf) && !defined(NO_vsnprintf)
 566     snprintf(outmode, sizeof(outmode), "%s", "wb6 ");
 567 #else
 568     strcpy(outmode, "wb6 ");
 569 #endif
 570
 571     prog = argv[0];
 572     bname = strrchr(argv[0], '/');
 573     if (bname)
 574       bname++;
 575     else
 576       bname = argv[0];
 577     argc--, argv++;
 578
 579     if (!strcmp(bname, "gunzip"))
 580       uncompr = 1;
 581     else if (!strcmp(bname, "zcat"))
 582       copyout = uncompr = 1;
 583
 584     while (argc > 0) {
 585       if (strcmp(*argv, "-c") == 0)
 586         copyout = 1;
 587       else if (strcmp(*argv, "-d") == 0)
 588         uncompr = 1;
 589       else if (strcmp(*argv, "-f") == 0)
 590         outmode[3] = 'f';
 591       else if (strcmp(*argv, "-h") == 0)
 592         outmode[3] = 'h';
 593       else if (strcmp(*argv, "-r") == 0)
 594         outmode[3] = 'R';
 595       else if ((*argv)[0] == '-' && (*argv)[1] >= '1' && (*argv)[1] <= '9' &&
 596                (*argv)[2] == 0)
 597         outmode[2] = (*argv)[1];
 598       else
 599         break;
 600       argc--, argv++;
 601     }
 602     if (outmode[3] == ' ')
 603         outmode[3] = 0;
 604     if (argc == 0) {
 605         SET_BINARY_MODE(stdin);
 606         SET_BINARY_MODE(stdout);
 607         if (uncompr) {
 608             file = gzdopen(fileno(stdin), "rb");
 609             if (file == NULL) error("can't gzdopen stdin");
 610             gz_uncompress(file, stdout);
 611         } else {
 612             file = gzdopen(fileno(stdout), outmode);
 613             if (file == NULL) error("can't gzdopen stdout");
 614             gz_compress(stdin, file);
 615         }
 616     } else {
 617         if (copyout) {
 618             SET_BINARY_MODE(stdout);
 619         }
 620         do {
 621             if (uncompr) {
 622                 if (copyout) {
 623                     file = gzopen(*argv, "rb");
 624                     if (file == NULL)
 625                         fprintf(stderr, "%s: can't gzopen %s\n", prog, *argv);
 626                     else
 627                         gz_uncompress(file, stdout);
 628                 } else {
 629                     file_uncompress(*argv);
 630                 }
 631             } else {
 632                 if (copyout) {
 633                     FILE * in = fopen(*argv, "rb");
 634
 635                     if (in == NULL) {
 636                         perror(*argv);
 637                     } else {
 638                         file = gzdopen(fileno(stdout), outmode);
 639                         if (file == NULL) error("can't gzdopen stdout");
 640
 641                         gz_compress(in, file);
 642                     }
 643
 644                 } else {
 645                     file_compress(*argv, outmode);
 646                 }
 647             }
 648         } while (argv++, --argc);
 649     }
 650     return 0;
 651 }