minigzip.c

   1 /* minigzip.c -- simulate gzip using the zlib compression library
   2  * Copyright (C) 1995 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: minigzip.c,v 1.4 1995/05/02 15:54:22 jloup Exp $ */
  17
  18 #include <stdio.h>
  19 #include "zlib.h"
  20
  21 #ifndef __GO32__
  22 extern void exit  __P((int));
  23 #endif
  24 extern int unlink __P((const char *));
  25
  26 #ifdef STDC
  27 #  include <string.h>
  28 #endif
  29
  30 #ifdef MSDOS
  31 #  include <fcntl.h>
  32 #  include <io.h>
  33 #  define SET_BINARY_MODE(file) setmode(fileno(file), O_BINARY)
  34 #else
  35 #  define SET_BINARY_MODE(file)
  36 #endif
  37
  38 #define BUFLEN 4096
  39 #define MAX_NAME_LEN 1024
  40
  41 #define local static
  42 /* For MSDOS and other systems with limitation on stack size. For Unix,
  43     #define local
  44    works also.
  45  */
  46
  47 char *prog;
  48
  49 void error           __P((char *msg));
  50 void gz_compress     __P((FILE   *in, gzFile out));
  51 void gz_uncompress   __P((gzFile in, FILE   *out));
  52 void file_compress   __P((char  *file));
  53 void file_uncompress __P((char  *file));
  54 void main            __P((int argc, char *argv[]));
  55
  56 /* ===========================================================================
  57  * Display error message and exit
  58  */
  59 void error(msg)
  60     char *msg;
  61 {
  62     fprintf(stderr, "%s: %s\n", prog, msg);
  63     exit(1);
  64 }
  65
  66 /* ===========================================================================
  67  * Compress input to output then close both files.
  68  */
  69 void gz_compress(in, out)
  70     FILE   *in;
  71     gzFile out;
  72 {
  73     local char buf[BUFLEN];
  74     int len;
  75     int err;
  76
  77     for (;;) {
  78         len = fread(buf, 1, sizeof(buf), in);
  79         if (ferror(in)) {
  80             perror("fread");
  81             exit(1);
  82         }
  83         if (len == 0) break;
  84
  85         if (gzwrite(out, buf, len) != len) error(gzerror(out, &err));
  86     }
  87     fclose(in);
  88     if (gzclose(out) != Z_OK) error("failed gzclose");
  89 }
  90
  91 /* ===========================================================================
  92  * Uncompress input to output then close both files.
  93  */
  94 void gz_uncompress(in, out)
  95     gzFile in;
  96     FILE   *out;
  97 {
  98     local char buf[BUFLEN];
  99     int len;
 100     int err;
 101
 102     for (;;) {
 103         len = gzread(in, buf, sizeof(buf));
 104         if (len < 0) error (gzerror(in, &err));
 105         if (len == 0) break;
 106
 107         if (fwrite(buf, 1, len, out) != (uInt)len) error("failed fwrite");
 108     }
 109     if (fclose(out)) error("failed fclose");
 110
 111     if (gzclose(in) != Z_OK) error("failed gzclose");
 112 }
 113
 114
 115 /* ===========================================================================
 116  * Compress the given file: create a corresponding .gz file and remove the
 117  * original.
 118  */
 119 void file_compress(file)
 120     char  *file;
 121 {
 122     local char outfile[MAX_NAME_LEN];
 123     FILE  *in;
 124     gzFile out;
 125
 126     strcpy(outfile, file);
 127     strcat(outfile, ".gz");
 128
 129     in = fopen(file, "rb");
 130     if (in == NULL) {
 131         perror(file);
 132         exit(1);
 133     }
 134     out = gzopen(outfile, "wb");
 135     if (out == NULL) {
 136         fprintf(stderr, "%s: can't gzopen %s\n", prog, outfile);
 137         exit(1);
 138     }
 139     gz_compress(in, out);
 140
 141     unlink(file);
 142 }
 143
 144
 145 /* ===========================================================================
 146  * Uncompress the given file and remove the original.
 147  */
 148 void file_uncompress(file)
 149     char  *file;
 150 {
 151     local char buf[MAX_NAME_LEN];
 152     char *infile, *outfile;
 153     FILE  *out;
 154     gzFile in;
 155     int len = strlen(file);
 156
 157     strcpy(buf, file);
 158
 159     if (len > 3 && strcmp(file+len-3, ".gz") == 0) {
 160         infile = file;
 161         outfile = buf;
 162         outfile[len-3] = '\0';
 163     } else {
 164         outfile = file;
 165         infile = buf;
 166         strcat(infile, ".gz");
 167     }
 168     in = gzopen(infile, "rb");
 169     if (in == NULL) {
 170         fprintf(stderr, "%s: can't gzopen %s\n", prog, infile);
 171         exit(1);
 172     }
 173     out = fopen(outfile, "wb");
 174     if (out == NULL) {
 175         perror(file);
 176         exit(1);
 177     }
 178
 179     gz_uncompress(in, out);
 180
 181     unlink(infile);
 182 }
 183
 184
 185 /* ===========================================================================
 186  * Usage:  minigzip [-d] [files...]
 187  */
 188
 189 void main(argc, argv)
 190     int argc;
 191     char *argv[];
 192 {
 193     int uncompr = 0;
 194     gzFile file;
 195
 196     prog = argv[0];
 197     argc--, argv++;
 198
 199     if (argc > 0) {
 200         uncompr = (strcmp(*argv, "-d") == 0);
 201         if (uncompr) {
 202             argc--, argv++;
 203         }
 204     }
 205     if (argc == 0) {
 206         SET_BINARY_MODE(stdin);
 207         SET_BINARY_MODE(stdout);
 208         if (uncompr) {
 209             file = gzdopen(fileno(stdin), "rb");
 210             if (file == NULL) error("can't gzdopen stdin");
 211             gz_uncompress(file, stdout);
 212         } else {
 213             file = gzdopen(fileno(stdout), "wb");
 214             if (file == NULL) error("can't gzdopen stdout");
 215             gz_compress(stdin, file);
 216         }
 217     } else {
 218         do {
 219             if (uncompr) {
 220                 file_uncompress(*argv);
 221             } else {
 222                 file_compress(*argv);
 223             }
 224         } while (argv++, --argc);
 225     }
 226     exit(0);
 227 }