home *** CD-ROM | disk | FTP | other *** search
/ Amiga Elysian Archive / AmigaElysianArchive.iso / compress / comprs16.lha / compress.c < prev    next >
C/C++ Source or Header  |  1989-10-18  |  30KB  |  1,150 lines
  1. static char sccsid[] = "@(#)compress.c  @(#)compress.c  5.9 (Berkeley) 5/11/86";
  2.  
  3. /*
  4.  * Compress - data compression program
  5.  */
  6. #define min(a,b)        ((a>b) ? b : a)
  7.  
  8. #define HSIZE  69001           /* 95% occupancy */
  9.  
  10. /*
  11.  * a code_int must be able to hold 2**BITS values of type int, and also -1
  12.  */
  13. typedef long int    code_int;
  14. typedef long int    count_int;
  15.  
  16. typedef unsigned char    char_type;
  17. char_type magic_header[] = { "\037\235" };      /* 1F 9D */
  18.  
  19. /* Defines for third byte of header */
  20. #define BIT_MASK    0x1f
  21. #define BLOCK_MASK    0x80
  22. /* Masks 0x40 and 0x20 are free.  I think 0x20 should mean that there is
  23.    a fourth header byte (for expansion).
  24. */
  25. #define INIT_BITS 9            /* initial number of bits/code */
  26.  
  27. static char rcs_ident[] = "$Header: compress.c,v 4.0 85/07/30 12:50:00 joe Release $";
  28.  
  29. #include <stdio.h>
  30. #include <ctype.h>
  31. #include <stat.h>
  32. #include <time.h>
  33.  
  34. #include <exec/types.h>
  35. #include <exec/memory.h>
  36. #include <exec/ports.h>
  37. #include <exec/io.h>
  38. #include <libraries/dos.h>
  39. #include <libraries/dosextens.h>
  40. #include <functions.h>
  41.  
  42.  
  43. #define ARGVAL() (*++(*argv) || (--argc && *++argv))
  44.  
  45. int n_bits;                /* number of bits/code */
  46. int maxbits = BITS;            /* user settable max # bits/code */
  47. code_int maxcode;            /* maximum code, given n_bits */
  48. code_int maxmaxcode = 1 << BITS;    /* should NEVER generate this code */
  49. #define MAXCODE(n_bits) ((1 << (n_bits)) - 1)
  50.  
  51. /* extern int errno; */
  52.  
  53. count_int *htab;
  54. unsigned short *codetab;
  55.  
  56. #define htabof(i)       htab[i]
  57. #define codetabof(i)    codetab[i]
  58.  
  59. code_int hsize = HSIZE;         /* for dynamic table sizing */
  60. count_int fsize;
  61.  
  62. /*
  63.  * To save much memory, we overlay the table used by compress() with those
  64.  * used by decompress().  The tab_prefix table is the same size and type
  65.  * as the codetab.  The tab_suffix table needs 2**BITS characters.  We
  66.  * get this from the beginning of htab.  The output stack uses the rest
  67.  * of htab, and contains characters.  There is plenty of room for any
  68.  * possible stack (stack used to be 8000 characters).
  69.  */
  70.  
  71. #define tab_prefixof(i) codetabof(i)
  72. #define tab_suffixof(i)        ((char_type *)(htab))[i]
  73. #define de_stack           ((char_type *)&tab_suffixof(1<<BITS))
  74.  
  75. code_int free_ent = 0;            /* first unused entry */
  76. int exit_stat = 0;            /* per-file status */
  77. int perm_stat = 0;            /* permanent status */
  78.  
  79. code_int getcode();
  80.  
  81. Usage() {
  82. fprintf(stderr,"Usage: compress [-fvc] [-b maxbits] [file ...]\n");
  83. }
  84. int nomagic = 0;    /* Use a 3-byte magic number header, unless old file */
  85. int zcat_flg = 0;    /* Write output on stdout, suppress messages */
  86. int precious = 1;    /* Don't unlink output file on interrupt */
  87. int quiet = 1;        /* don't tell me about compression */
  88.  
  89. /*
  90.  * block compression parameters -- after all codes are used up,
  91.  * and compression rate changes, start over.
  92.  */
  93. int block_compress = BLOCK_MASK;
  94. int clear_flg = 0;
  95. long int ratio = 0;
  96. #define CHECK_GAP 10000 /* ratio check interval */
  97. count_int checkpoint = CHECK_GAP;
  98. /*
  99.  * the next two codes should not be changed lightly, as they must not
  100.  * lie within the contiguous general code space.
  101.  */
  102. #define FIRST    257    /* first free entry */
  103. #define CLEAR    256    /* table clear output code */
  104.  
  105. int force = 0;
  106. char ofname [100];
  107. int (*oldint)();
  108.  
  109. int do_decomp = 0;
  110.  
  111. /*****************************************************************
  112.  * TAG( main )
  113.  *
  114.  * Algorithm from "A Technique for High Performance Data Compression",
  115.  * Terry A. Welch, IEEE Computer Vol 17, No 6 (June 1984), pp 8-19.
  116.  *
  117.  * Usage: compress [-dfvc] [-b bits] [file ...]
  118.  * Inputs:
  119.  *    -d:        If given, decompression is done instead.
  120.  *
  121.  *    -c:        Write output on stdout, don't remove original.
  122.  *
  123.  *    -b:        Parameter limits the max number of bits/code.
  124.  *
  125.  *    -f:        Forces output file to be generated, even if one already
  126.  *            exists, and even if no space is saved by compressing.
  127.  *            If -f is not used, the user will be prompted if stdin is
  128.  *            a tty, otherwise, the output file will not be overwritten.
  129.  *
  130.  *    -v:        Write compression statistics
  131.  *
  132.  *    file ...:   Files to be compressed.  If none specified, stdin
  133.  *            is used.
  134.  * Outputs:
  135.  *    file.Z:     Compressed form of file with same mode, owner, and utimes
  136.  *    or stdout   (if stdin used as input)
  137.  *
  138.  * Assumptions:
  139.  *    When filenames are given, replaces with the compressed version
  140.  *    (.Z suffix) only if the file decreases in size.
  141.  * Algorithm:
  142.  *    Modified Lempel-Ziv method (LZW).  Basically finds common
  143.  * substrings and replaces them with a variable size code.  This is
  144.  * deterministic, and can be done on the fly.  Thus, the decompression
  145.  * procedure needs no input table, but tracks the way the table was built.
  146.  */
  147.  
  148. main( argc, argv )
  149. register int argc; char **argv;
  150. {
  151.     int overwrite = 0;    /* Do not overwrite unless given -f flag */
  152.     char tempname[100];
  153.     char **filelist, **fileptr;
  154.     char *cp, *rindex(), *strcpy(), *malloc();
  155.     struct stat statbuf;
  156.     extern onintr(), oops();
  157.  
  158.     freopen("*","r+", stderr);
  159.  
  160.     htab = (count_int *)malloc(HSIZE * sizeof(count_int));
  161.     codetab = (unsigned short *)malloc(HSIZE * sizeof(unsigned short));
  162.     if (htab == NULL || codetab == NULL) {
  163.     fprintf(stderr,"compress: out of memory\n");
  164.     exit(1);
  165.     }
  166.     filelist = fileptr = (char **)(malloc(argc * sizeof(*argv)));
  167.     *filelist = NULL;
  168.  
  169.     if((cp = rindex(argv[0], '/')) != 0) {
  170.     cp++;
  171.     } else {
  172.     cp = argv[0];
  173.     }
  174.     if(strcmp(cp, "uncompress") == 0) {
  175.     do_decomp = 1;
  176.     } else if(strcmp(cp, "zcat") == 0) {
  177.     do_decomp = 1;
  178.     zcat_flg = 1;
  179.     }
  180.  
  181.     /* Argument Processing
  182.      * All flags are optional.
  183.      * -D => debug
  184.      * -V => print Version; debug verbose
  185.      * -d => do_decomp
  186.      * -v => unquiet
  187.      * -f => force overwrite of output file
  188.      * -n => no header: useful to uncompress old files
  189.      * -b maxbits => maxbits.  If -b is specified, then maxbits MUST be
  190.      *        given also.
  191.      * -c => cat all output to stdout
  192.      * -C => generate output compatible with compress 2.0.
  193.      * if a string is left, must be an input filename.
  194.      */
  195.     for (argc--, argv++; argc > 0; argc--, argv++) {
  196.     if (**argv == '-') {    /* A flag argument */
  197.         while (*++(*argv)) {        /* Process all flags in this arg */
  198.         switch (**argv) {
  199.             case 'V':
  200.             version();
  201.             break;
  202.             case 'v':
  203.             quiet = 0;
  204.             break;
  205.             case 'd':
  206.             do_decomp = 1;
  207.             break;
  208.             case 'f':
  209.             case 'F':
  210.             overwrite = 1;
  211.             force = 1;
  212.             break;
  213.             case 'n':
  214.             nomagic = 1;
  215.             break;
  216.             case 'C':
  217.             block_compress = 0;
  218.             break;
  219.             case 'b':
  220.             if (!ARGVAL()) {
  221.                 fprintf(stderr, "Missing maxbits\n");
  222.                 Usage();
  223.                 exit(1);
  224.             }
  225.             maxbits = atoi(*argv);
  226.             goto nextarg;
  227.             case 'c':
  228.             zcat_flg = 1;
  229.             break;
  230.             case 'q':
  231.             quiet = 1;
  232.             break;
  233.             default:
  234.             fprintf(stderr, "Unknown flag: '%c'; ", **argv);
  235.             Usage();
  236.             exit(1);
  237.         }
  238.         }
  239.     } else {      /* Input file name */
  240.         *fileptr++ = *argv; /* Build input file list */
  241.         *fileptr = NULL;
  242.         /* process nextarg; */
  243.     }
  244.     nextarg: continue;
  245.     }
  246.  
  247.     if (maxbits < INIT_BITS) maxbits = INIT_BITS;
  248.     if (maxbits > BITS) maxbits = BITS;
  249.     maxmaxcode = 1 << maxbits;
  250.  
  251.     if (*filelist != NULL) {
  252.     for (fileptr = filelist; *fileptr; fileptr++) {
  253.         exit_stat = 0;
  254.         if (do_decomp) {                    /* DECOMPRESSION */
  255.         /* Check for .Z suffix */
  256.         if (strcmp(*fileptr + strlen(*fileptr) - 2, ".Z") != 0) {
  257.             /* No .Z: tack one on */
  258.             strcpy(tempname, *fileptr);
  259.             strcat(tempname, ".Z");
  260.             *fileptr = tempname;
  261.         }
  262.         /* Open input file */
  263.         if ((freopen(*fileptr, "r", stdin)) == NULL) {
  264.             perror(*fileptr);
  265.             perm_stat = 1;
  266.             continue;
  267.         }
  268.         /* Check the magic number */
  269.         if (nomagic == 0) {
  270.             if ((getc(stdin) != (magic_header[0] & 0xFF))
  271.              || (getc(stdin) != (magic_header[1] & 0xFF))) {
  272.             fprintf(stderr, "%s: not in compressed format\n",
  273.                 *fileptr);
  274.             continue;
  275.             }
  276.             maxbits = getc(stdin);        /* set -b from file */
  277.             block_compress = maxbits & BLOCK_MASK;
  278.             maxbits &= BIT_MASK;
  279.             maxmaxcode = 1 << maxbits;
  280.             if(maxbits > BITS) {
  281.             fprintf(stderr,
  282.             "%s: compressed with %d bits, can only handle %d bits\n",
  283.             *fileptr, maxbits, BITS);
  284.             continue;
  285.             }
  286.         }
  287.         /* Generate output filename */
  288.         strcpy(ofname, *fileptr);
  289.         ofname[strlen(*fileptr) - 2] = '\0';  /* Strip off .Z */
  290.         } else {                    /* COMPRESSION */
  291.         if (strcmp(*fileptr + strlen(*fileptr) - 2, ".Z") == 0) {
  292.             fprintf(stderr, "%s: already has .Z suffix -- no change\n",
  293.                 *fileptr);
  294.             continue;
  295.         }
  296.         /* Open input file */
  297.         if ((freopen(*fileptr, "r", stdin)) == NULL) {
  298.             perror(*fileptr);
  299.             perm_stat = 1;
  300.             continue;
  301.         }
  302.         stat ( *fileptr, &statbuf );
  303.         fsize = (long) statbuf.st_size;
  304.         /*
  305.          * tune hash table size for small files -- ad hoc,
  306.          * but the sizes match earlier #defines, which
  307.          * serve as upper bounds on the number of output codes.
  308.          */
  309.         hsize = HSIZE;
  310.         if ( fsize < (1 << 12) )
  311.             hsize = min ( 5003, HSIZE );
  312.         else if ( fsize < (1 << 13) )
  313.             hsize = min ( 9001, HSIZE );
  314.         else if ( fsize < (1 << 14) )
  315.             hsize = min ( 18013, HSIZE );
  316.         else if ( fsize < (1 << 15) )
  317.             hsize = min ( 35023, HSIZE );
  318.         else if ( fsize < 47000 )
  319.             hsize = min ( 50021, HSIZE );
  320.  
  321.         /* Generate output filename */
  322.         strcpy(ofname, *fileptr);
  323.         strcat(ofname, ".Z");
  324.         }
  325.         /* Check for overwrite of existing file */
  326.         if (overwrite == 0 && zcat_flg == 0) {
  327.         if (stat(ofname, &statbuf) == 0) {
  328.             char response[2];
  329.             response[0] = 'n';
  330.             fprintf(stderr, "%s already exists;", ofname);
  331.             if (isatty(2)) {
  332.             fprintf(stderr, " do you wish to overwrite %s (y or n)? ",
  333.                     ofname);
  334.             fflush(stderr);
  335.             read(2, response, 2);
  336.             while (response[1] != '\n') {
  337.                 if (read(2, response+1, 1) < 0) {   /* Ack! */
  338.                 perror("stderr"); break;
  339.             }
  340.             }
  341.         }
  342.         if (response[0] != 'y') {
  343.             fprintf(stderr, "\tnot overwritten\n");
  344.             continue;
  345.         }
  346.         }
  347.     }
  348.     if(zcat_flg == 0) {         /* Open output file */
  349.         if (freopen(ofname, "w", stdout) == NULL) {
  350.         perror(ofname);
  351.         perm_stat = 1;
  352.         continue;
  353.         }
  354.         precious = 0;
  355.         if(!quiet)
  356.         fprintf(stderr, "%s: ", *fileptr);
  357.     }
  358.  
  359.         /* Actually do the compression/decompression */
  360.     if (do_decomp == 0)
  361.         compress();
  362.     else
  363.         decompress();
  364.     if(zcat_flg == 0) {
  365.         copystat(*fileptr, ofname);     /* Copy stats */
  366.         precious = 1;
  367.         if((exit_stat == 1) || (!quiet))
  368.         putc('\n', stderr);
  369.         }
  370.     }
  371.     } else {        /* Standard input */
  372.     if (do_decomp == 0) {
  373.         compress();
  374.         if(!quiet)
  375.         putc('\n', stderr);
  376.     } else {
  377.         /* Check the magic number */
  378.         if (nomagic == 0) {
  379.         if ((getc(stdin)!=(magic_header[0] & 0xFF))
  380.          || (getc(stdin)!=(magic_header[1] & 0xFF))) {
  381.             fprintf(stderr, "stdin: not in compressed format\n");
  382.             exit(1);
  383.         }
  384.         maxbits = getc(stdin);    /* set -b from file */
  385.         block_compress = maxbits & BLOCK_MASK;
  386.         maxbits &= BIT_MASK;
  387.         maxmaxcode = 1 << maxbits;
  388.         fsize = 100000;     /* assume stdin large for USERMEM */
  389.         if(maxbits > BITS) {
  390.             fprintf(stderr,
  391.              "stdin: compressed with %d bits, can only handle %d bits\n",
  392.              maxbits, BITS);
  393.             exit(1);
  394.         }
  395.         }
  396.         decompress();
  397.     }
  398.     }
  399.     exit(perm_stat ? perm_stat : exit_stat);
  400. }
  401.  
  402. static int offset;
  403. long int in_count = 1;            /* length of input */
  404. long int bytes_out;            /* length of compressed output */
  405. long int out_count = 0;         /* # of codes output (for debugging) */
  406.  
  407. /*
  408.  * compress stdin to stdout
  409.  *
  410.  * Algorithm:  use open addressing double hashing (no chaining) on the
  411.  * prefix code / next character combination.  We do a variant of Knuth's
  412.  * algorithm D (vol. 3, sec. 6.4) along with G. Knott's relatively-prime
  413.  * secondary probe.  Here, the modular division first probe is gives way
  414.  * to a faster exclusive-or manipulation.  Also do block compression with
  415.  * an adaptive reset, whereby the code table is cleared when the compression
  416.  * ratio decreases, but after the table fills.    The variable-length output
  417.  * codes are re-sized at this point, and a special CLEAR code is generated
  418.  * for the decompressor.  Late addition:  construct the table according to
  419.  * file size for noticeable speed improvement on small files.  Please direct
  420.  * questions about this implementation to ames!jaw.
  421.  */
  422.  
  423. compress() {
  424.     register long fcode;
  425.     register code_int i = 0;
  426.     register int c;
  427.     register code_int ent;
  428.     register int disp;
  429.     register code_int hsize_reg;
  430.     register int hshift;
  431.  
  432.     if (nomagic == 0) {
  433.     putc(magic_header[0], stdout); putc(magic_header[1], stdout);
  434.     putc((char)(maxbits | block_compress), stdout);
  435.     if(ferror(stdout))
  436.         writeerr();
  437.     }
  438.  
  439.     offset = 0;
  440.     bytes_out = 3;        /* includes 3-byte header mojo */
  441.     out_count = 0;
  442.     clear_flg = 0;
  443.     ratio = 0;
  444.     in_count = 1;
  445.     checkpoint = CHECK_GAP;
  446.     maxcode = MAXCODE(n_bits = INIT_BITS);
  447.     free_ent = ((block_compress) ? FIRST : 256 );
  448.  
  449.     ent = getc(stdin);
  450.  
  451.     hshift = 0;
  452.     for ( fcode = (long) hsize;  fcode < 65536L; fcode *= 2L )
  453.     hshift++;
  454.     hshift = 8 - hshift;        /* set hash code range bound */
  455.  
  456.     hsize_reg = hsize;
  457.     cl_hash( (count_int) hsize_reg);            /* clear hash table */
  458.  
  459.     while ((c = getc(stdin)) != EOF) {
  460.     in_count++;
  461.     fcode = (long) (((long) c << maxbits) + ent);
  462.     i = ((c << hshift) ^ ent);      /* xor hashing */
  463.  
  464.     if ( htabof (i) == fcode ) {
  465.         ent = codetabof (i);
  466.         continue;
  467.     } else if ((long)htabof (i) < 0)      /* empty slot */
  468.         goto nomatch;
  469.     disp = hsize_reg - i;        /* secondary hash (after G. Knott) */
  470.     if ( i == 0 )
  471.         disp = 1;
  472. probe:
  473.     if ((i -= disp) < 0)
  474.         i += hsize_reg;
  475.  
  476.     if (htabof (i) == fcode) {
  477.         ent = codetabof (i);
  478.         continue;
  479.     }
  480.     if ((long)htabof (i) > 0)
  481.         goto probe;
  482. nomatch:
  483.     output ((code_int) ent);
  484.     out_count++;
  485.     ent = c;
  486.     if ( free_ent < maxmaxcode ) {
  487.         codetabof (i) = free_ent++; /* code -> hashtable */
  488.         htabof (i) = fcode;
  489.     }
  490.     else if ( (count_int)in_count >= checkpoint && block_compress )
  491.         cl_block ();
  492.     }
  493.     /*
  494.      * Put out the final code.
  495.      */
  496.     output((code_int)ent);
  497.     out_count++;
  498.     output((code_int)-1);
  499.  
  500.     /*
  501.      * Print out stats on stderr
  502.      */
  503.     if(zcat_flg == 0 && !quiet) {
  504.     fprintf( stderr, "Compression: " );
  505.     prratio( stderr, in_count-bytes_out, in_count );
  506.     }
  507.     if(bytes_out > in_count)    /* exit(2) if no savings */
  508.     exit_stat = 2;
  509.     return;
  510. }
  511.  
  512. /*****************************************************************
  513.  * TAG(output)
  514.  *
  515.  * Output the given code.
  516.  * Inputs:
  517.  *    code:    A n_bits-bit integer.  If == -1, then EOF.  This assumes
  518.  *        that n_bits =< (long)wordsize - 1.
  519.  * Outputs:
  520.  *    Outputs code to the file.
  521.  * Assumptions:
  522.  *    Chars are 8 bits long.
  523.  * Algorithm:
  524.  *    Maintain a BITS character long buffer (so that 8 codes will
  525.  * fit in it exactly).    Use the VAX insv instruction to insert each
  526.  * code in turn.  When the buffer fills up empty it and start over.
  527.  */
  528.  
  529. static char buf[BITS];
  530.  
  531. char_type lmask[9] = {0xff, 0xfe, 0xfc, 0xf8, 0xf0, 0xe0, 0xc0, 0x80, 0x00};
  532. char_type rmask[9] = {0x00, 0x01, 0x03, 0x07, 0x0f, 0x1f, 0x3f, 0x7f, 0xff};
  533.  
  534. output( code )
  535. code_int  code;
  536. {
  537.  
  538.     /*
  539.      * On the VAX, it is important to have the register declarations
  540.      * in exactly the order given, or the asm will break.
  541.      */
  542.     register int r_off = offset, bits= n_bits;
  543.     register char * bp = buf;
  544.  
  545.     if (code >= 0) {
  546. /*
  547.  * byte/bit numbering on the VAX is simulated by the following code
  548.  */
  549.     /*
  550.      * Get to the first byte.
  551.      */
  552.     bp += (r_off >> 3);
  553.     r_off &= 7;
  554.     /*
  555.      * Since code is always >= 8 bits, only need to mask the first
  556.      * hunk on the left.
  557.      */
  558.     *bp = (*bp & rmask[r_off]) | (code << r_off) & lmask[r_off];
  559.     bp++;
  560.     bits -= (8 - r_off);
  561.     code >>= 8 - r_off;
  562.     /* Get any 8 bit parts in the middle (<=1 for up to 16 bits). */
  563.     if ( bits >= 8 ) {
  564.         *bp++ = code;
  565.         code >>= 8;
  566.         bits -= 8;
  567.     }
  568.     /* Last bits. */
  569.     if(bits)
  570.         *bp = code;
  571.     offset += n_bits;
  572.     if (offset == (n_bits << 3)) {
  573.         bp = buf;
  574.         bits = n_bits;
  575.         bytes_out += bits;
  576.         do
  577.         putc(*bp++, stdout);
  578.         while(--bits);
  579.         offset = 0;
  580.     }
  581.  
  582.     /*
  583.      * If the next entry is going to be too big for the code size,
  584.      * then increase it, if possible.
  585.      */
  586.     if (free_ent > maxcode || (clear_flg > 0))
  587.     {
  588.         /*
  589.          * Write the whole buffer, because the input side won't
  590.          * discover the size increase until after it has read it.
  591.          */
  592.         if (offset > 0) {
  593.         if( fwrite( buf, 1, n_bits, stdout ) != n_bits)
  594.             writeerr();
  595.         bytes_out += n_bits;
  596.         }
  597.         offset = 0;
  598.  
  599.         if ( clear_flg ) {
  600.         maxcode = MAXCODE (n_bits = INIT_BITS);
  601.         clear_flg = 0;
  602.         }
  603.         else {
  604.         n_bits++;
  605.         if ( n_bits == maxbits )
  606.             maxcode = maxmaxcode;
  607.         else
  608.             maxcode = MAXCODE(n_bits);
  609.         }
  610.     }
  611.     } else {
  612.     /*
  613.      * At EOF, write the rest of the buffer.
  614.      */
  615.     if ( offset > 0 )
  616.         fwrite( buf, 1, (offset + 7) / 8, stdout );
  617.     bytes_out += (offset + 7) / 8;
  618.     offset = 0;
  619.     fflush( stdout );
  620.     if(ferror(stdout))
  621.         writeerr();
  622.     }
  623. }
  624.  
  625. /*
  626.  * Decompress stdin to stdout.    This routine adapts to the codes in the
  627.  * file building the "string" table on-the-fly; requiring no table to
  628.  * be stored in the compressed file.  The tables used herein are shared
  629.  * with those of the compress() routine.  See the definitions above.
  630.  */
  631.  
  632. decompress() {
  633.     register char_type *stackp;
  634.     register int finchar;
  635.     register code_int code, oldcode, incode;
  636.  
  637.     /*
  638.      * As above, initialize the first 256 entries in the table.
  639.      */
  640.     maxcode = MAXCODE(n_bits = INIT_BITS);
  641.     for ( code = 255; code >= 0; code-- ) {
  642.     tab_prefixof(code) = 0;
  643.     tab_suffixof(code) = (char_type)code;
  644.     }
  645.     free_ent = ((block_compress) ? FIRST : 256 );
  646.  
  647.     finchar = oldcode = getcode();
  648.     if(oldcode == -1)   /* EOF already? */
  649.     return;         /* Get out of here */
  650.     putc((char)finchar, stdout);  /* first code must be 8 bits = char */
  651.     if(ferror(stdout))          /* Crash if can't write */
  652.     writeerr();
  653.     stackp = de_stack;
  654.  
  655.     while ((code = getcode()) > -1) {
  656.  
  657.     if ((code == CLEAR) && block_compress) {
  658.         for ( code = 255; code >= 0; code-- )
  659.         tab_prefixof(code) = 0;
  660.         clear_flg = 1;
  661.         free_ent = FIRST - 1;
  662.         if ( (code = getcode ()) == -1 )    /* O, untimely death! */
  663.         break;
  664.     }
  665.     incode = code;
  666.     /*
  667.      * Special case for KwKwK string.
  668.      */
  669.     if ( code >= free_ent ) {
  670.         *stackp++ = finchar;
  671.         code = oldcode;
  672.     }
  673.  
  674.     /*
  675.      * Generate output characters in reverse order
  676.      */
  677.     while ( code >= 256 ) {
  678.         *stackp++ = tab_suffixof(code);
  679.         code = tab_prefixof(code);
  680.     }
  681.     *stackp++ = finchar = tab_suffixof(code);
  682.  
  683.     /*
  684.      * And put them out in forward order
  685.      */
  686.     do
  687.         putc(*--stackp, stdout);
  688.     while ( stackp > de_stack );
  689.  
  690.     /*
  691.      * Generate the new entry.
  692.      */
  693.     if ( (code=free_ent) < maxmaxcode ) {
  694.         tab_prefixof(code) = (unsigned short)oldcode;
  695.         tab_suffixof(code) = finchar;
  696.         free_ent = code+1;
  697.     }
  698.     /*
  699.      * Remember previous code.
  700.      */
  701.     oldcode = incode;
  702.     }
  703.     fflush( stdout );
  704.     if(ferror(stdout))
  705.     writeerr();
  706. }
  707.  
  708. /*****************************************************************
  709.  * TAG( getcode )
  710.  *
  711.  * Read one code from the standard input.  If EOF, return -1.
  712.  * Inputs:
  713.  *    stdin
  714.  * Outputs:
  715.  *    code or -1 is returned.
  716.  */
  717.  
  718. code_int
  719. getcode() {
  720.     /*
  721.      * On the VAX, it is important to have the register declarations
  722.      * in exactly the order given, or the asm will break.
  723.      */
  724.     register code_int code;
  725.     static int offset = 0, size = 0;
  726.     static char_type buf[BITS];
  727.     register int r_off, bits;
  728.     register char_type *bp = buf;
  729.  
  730.     if ( clear_flg > 0 || offset >= size || free_ent > maxcode ) {
  731.     /*
  732.      * If the next entry will be too big for the current code
  733.      * size, then we must increase the size.  This implies reading
  734.      * a new buffer full, too.
  735.      */
  736.     if ( free_ent > maxcode ) {
  737.         n_bits++;
  738.         if ( n_bits == maxbits )
  739.         maxcode = maxmaxcode;    /* won't get any bigger now */
  740.         else
  741.         maxcode = MAXCODE(n_bits);
  742.     }
  743.     if ( clear_flg > 0) {
  744.         maxcode = MAXCODE (n_bits = INIT_BITS);
  745.         clear_flg = 0;
  746.     }
  747.     size = fread( buf, 1, n_bits, stdin );
  748.     if ( size <= 0 )
  749.         return -1;            /* end of file */
  750.     offset = 0;
  751.     /* Round size down to integral number of codes */
  752.     size = (size << 3) - (n_bits - 1);
  753.     }
  754.     r_off = offset;
  755.     bits = n_bits;
  756.     /*
  757.      * Get to the first byte.
  758.      */
  759.     bp += (r_off >> 3);
  760.     r_off &= 7;
  761.     /* Get first part (low order bits) */
  762.     code = (*bp++ >> r_off);
  763.     bits -= (8 - r_off);
  764.     r_off = 8 - r_off;            /* now, offset into code word */
  765.     /* Get any 8 bit parts in the middle (<=1 for up to 16 bits). */
  766.     if ( bits >= 8 ) {
  767.     code |= *bp++ << r_off;
  768.     r_off += 8;
  769.     bits -= 8;
  770.     }
  771.     /* high order bits. */
  772.     code |= (*bp & rmask[bits]) << r_off;
  773.     offset += n_bits;
  774.  
  775.     return code;
  776. }
  777.  
  778.  
  779. writeerr()
  780. {
  781.     perror ( ofname );
  782.     unlink ( ofname );
  783.     exit ( 1 );
  784. }
  785.  
  786. copystat(ifname, ofname)
  787. char *ifname, *ofname;
  788. {
  789.     BOOL CopyFileDate();
  790.  
  791.     fclose(stdout);
  792.     fclose(stdin);
  793.     if (exit_stat == 2 && (!force)) { /* No compression: remove file.Z */
  794.     if(!quiet)
  795.         fprintf(stderr, " -- file unchanged"), fflush(stderr);
  796.     } else {            /* ***** Successful Compression ***** */
  797.     exit_stat = 0;
  798.     if (CopyFileAttr(ifname, ofname) || CopyFileDate(ifname, ofname))
  799.         fprintf(stderr, " -- couldn't copy file attributes"), fflush(stderr);
  800.     if (unlink(ifname))     /* Remove input file */
  801.         perror(ifname), fflush(stderr);
  802.     else if(!quiet)
  803.         fprintf(stderr, " -- replaced with %s", ofname), fflush(stderr);
  804.     return;     /* Successful return */
  805.     }
  806.  
  807.     /* Unsuccessful return -- one of the tests failed */
  808.     if (unlink(ofname))
  809.     perror(ofname), fflush(stderr);
  810. }
  811.  
  812. onintr ( )
  813. {
  814.     if (!precious)
  815.     unlink ( ofname );
  816.     exit ( 1 );
  817. }
  818.  
  819. oops ( )        /* wild pointer -- assume bad input */
  820. {
  821.     if ( do_decomp )
  822.     fprintf ( stderr, "uncompress: corrupt input\n" );
  823.     unlink ( ofname );
  824.     exit ( 1 );
  825. }
  826.  
  827. cl_block ()             /* table clear for block compress */
  828. {
  829.     register long int rat;
  830.  
  831.     checkpoint = in_count + CHECK_GAP;
  832.  
  833.     if(in_count > 0x007fffff) { /* shift will overflow */
  834.     rat = bytes_out >> 8;
  835.     if(rat == 0) {          /* Don't divide by zero */
  836.         rat = 0x7fffffff;
  837.     } else {
  838.         rat = in_count / rat;
  839.     }
  840.     } else {
  841.     rat = (in_count << 8) / bytes_out;      /* 8 fractional bits */
  842.     }
  843.     if ( rat > ratio ) {
  844.     ratio = rat;
  845.     } else {
  846.     ratio = 0;
  847.     cl_hash ( (count_int) hsize );
  848.     free_ent = FIRST;
  849.     clear_flg = 1;
  850.     output ( (code_int) CLEAR );
  851.     }
  852. }
  853.  
  854. cl_hash(hsize)          /* reset code table */
  855.     register count_int hsize;
  856. {
  857.     register count_int *htab_p = &htab[hsize];
  858.     register long i;
  859.     register long m1 = -1;
  860.  
  861.     i = hsize - 16;
  862.     do {                /* might use Sys V memset(3) here */
  863.     *(htab_p-16) = m1;
  864.     *(htab_p-15) = m1;
  865.     *(htab_p-14) = m1;
  866.     *(htab_p-13) = m1;
  867.     *(htab_p-12) = m1;
  868.     *(htab_p-11) = m1;
  869.     *(htab_p-10) = m1;
  870.     *(htab_p-9) = m1;
  871.     *(htab_p-8) = m1;
  872.     *(htab_p-7) = m1;
  873.     *(htab_p-6) = m1;
  874.     *(htab_p-5) = m1;
  875.     *(htab_p-4) = m1;
  876.     *(htab_p-3) = m1;
  877.     *(htab_p-2) = m1;
  878.     *(htab_p-1) = m1;
  879.     htab_p -= 16;
  880.     } while ((i -= 16) >= 0);
  881.     for ( i += 16; i > 0; i-- )
  882.     *--htab_p = m1;
  883. }
  884.  
  885. prratio(stream, num, den)
  886. FILE *stream;
  887. long int num, den;
  888. {
  889.     register int q;            /* Doesn't need to be long */
  890.  
  891.     if(num > 214748L) {             /* 2147483647/10000 */
  892.     q = num / (den / 10000L);
  893.     } else {
  894.     q = 10000L * num / den;     /* Long calculations, though */
  895.     }
  896.     if (q < 0) {
  897.     putc('-', stream);
  898.     q = -q;
  899.     }
  900.     fprintf(stream, "%d.%02d%%", q / 100, q % 100);
  901. }
  902.  
  903. version()
  904. {
  905.     fprintf(stderr, "%s, Berkeley 5.9 5/11/86\n", rcs_ident);
  906.     fprintf(stderr, "Options: ");
  907.     fprintf(stderr, "AMIGA, ");
  908.     fprintf(stderr, "BITS = %d\n", BITS);
  909. }
  910.  
  911.  
  912.  
  913.  
  914. /* Function:
  915.  *        GetFileDate
  916.  *
  917.  * Called with:
  918.  *        name:    file name
  919.  *        date:    pointer to DateStamp structure
  920.  *
  921.  * Returns:
  922.  *        result: 1 => got a date, 0 => didn't
  923.  *
  924.  * Description:
  925.  *        GetFileDate attempts to get the creation/modification date
  926.  *        of a file (unfortunately, they're one and the same) and stores
  927.  *        it into the location pointed to by <date>.  If the file doesn't
  928.  *        exist or for some reason the date can't be obtained, <date>
  929.  *    is set to zeros and a zero is returned.  Otherwise, <date> is set
  930.  *        to the file date and a 1 is returned.
  931.  */
  932.  
  933. BOOL
  934. GetFileDate(name, date)
  935.     char   *name; struct DateStamp *date;
  936. {
  937.     struct FileInfoBlock *Fib;
  938.     ULONG    FLock;
  939.     int    result = FALSE;
  940.     register struct DateStamp  *d;
  941.  
  942.     if ((FLock = (ULONG) Lock(name,(long)(ACCESS_READ)))== NULL)
  943.         goto exit1;
  944.  
  945.     Fib = (struct FileInfoBlock *)
  946.         AllocMem((long)sizeof(struct FileInfoBlock),
  947.                     (long)(MEMF_CHIP|MEMF_PUBLIC));
  948.  
  949.     if (Fib == NULL )
  950.         result = FALSE;
  951.     else{
  952.         if (!Examine(FLock, Fib )) {
  953.             result = FALSE;
  954.         }
  955.         else
  956.             if (Fib->fib_DirEntryType > 0 )
  957.                 result = FALSE;     /* It's a directory */
  958.             else{
  959.                 d = &Fib->fib_Date;
  960.                 date->ds_Days = d->ds_Days;
  961.                 date->ds_Minute = d->ds_Minute;
  962.                 date->ds_Tick = d->ds_Tick;
  963.                 result = TRUE;
  964.             }
  965.         FreeMem((void *)Fib,(long)sizeof(struct FileInfoBlock));
  966.     }
  967.  
  968.     UnLock(FLock);
  969. exit1:
  970.     if (! result ) {
  971.         date->ds_Days = 0;
  972.         date->ds_Minute = 0;
  973.         date->ds_Tick = 0;
  974.     }
  975.     return result;
  976. }
  977.  
  978.  
  979. /*---------------------------------------------------------------------*/
  980. /*  SetFileDate: datestamp the given file with the given date.           */
  981. /*---------------------------------------------------------------------*/
  982.  
  983. #define ACTION_SETDATE_MODE 34L     /* Set creation date on file */
  984.  
  985. BOOL
  986. SetFileDate( name, date )
  987.     char *name; struct DateStamp *date;
  988. {
  989.     struct MsgPort       *task;        /* for process id handler */
  990.     ULONG    arg[4];                 /* array of arguments      */
  991.     int nameleng;
  992.     char   *bstr, *strcpy();                /* of file to be set      */
  993.     long    rc;
  994.     char   *strchr();
  995.     int    strlen();
  996.  
  997.     rc = 0;
  998.  
  999.     nameleng = strlen(name);
  1000.     if (!(bstr = (char *)AllocMem((long) (nameleng + 2),MEMF_PUBLIC)))
  1001.         goto exit2;
  1002.  
  1003.     if (!(task = (struct MsgPort *)DeviceProc(name )))
  1004.         goto exit1;
  1005.  
  1006.  /* Dos Packet needs the filename in Bstring format */
  1007.  
  1008.     (void) strcpy(bstr+1, name );
  1009.     *bstr = nameleng;
  1010.  
  1011.     arg[0]= (ULONG)NULL;
  1012.     arg[1]= (ULONG)IoErr();                 /* lock on parent director set by
  1013.                                        DeviceProc() */
  1014.     arg[2]= (ULONG)bstr >> 2;
  1015.     arg[3]= (ULONG)date;
  1016.     rc = sendpkt(task,ACTION_SETDATE_MODE,arg,4L );
  1017.  
  1018. exit1: if (bstr )
  1019.         FreeMem((void *)bstr, (long) (nameleng + 2));
  1020. exit2: if (rc == DOSTRUE )
  1021.         return TRUE;
  1022.     else
  1023.         return FALSE;
  1024. }
  1025.  
  1026.  
  1027.  
  1028.  
  1029. /* Copy the last modified date from one file to another.
  1030.  * Called with:
  1031.  *        from:        name of source file
  1032.  *        to:            name of destination file
  1033.  * Returns:
  1034.  *        0 => success, 1 => failure
  1035.  * Note:
  1036.  *        Dynamic memory allocation of the DateStamp struction is
  1037.  *        necessary to insure longword alignment.
  1038.  */
  1039.  
  1040. BOOL
  1041. CopyFileDate(from,to)
  1042.     char *from, *to;
  1043. {
  1044.     struct DateStamp *date;
  1045.     int status = 1;             /* default is fail code */
  1046.  
  1047.     if (date = (struct DateStamp *)
  1048.         AllocMem((long) sizeof(struct DateStamp), MEMF_PUBLIC)) {
  1049.         if (GetFileDate(from,date))
  1050.             if (SetFileDate(to,date))
  1051.                 status = 0;
  1052.         FreeMem(date, (long) sizeof(struct DateStamp));
  1053.     }
  1054.     return status;
  1055. }
  1056.  
  1057.  
  1058. /****************************************************************************/
  1059. /* Function:
  1060.  *        CopyFileAttr - Copy File Attributes
  1061.  *
  1062.  * Called with:
  1063.  *        srcName:        source file name
  1064.  *        dstName:        destination file name
  1065.  *
  1066.  * Returns:
  1067.  *        status where 0 => success
  1068.  *
  1069.  * Description:
  1070.  *        CopyFileAttr is used by file copying functions to assign the
  1071.  *        attributes of the source file to the destination file.
  1072.  */
  1073. int
  1074. CopyFileAttr(srcName, dstName)
  1075.     char *srcName, *dstName;
  1076. {
  1077.     struct Lock *srcLock = NULL;
  1078.     struct FileInfoBlock *srcFIB = NULL;
  1079.     int status = 0;
  1080.  
  1081.     if (! (srcFIB = AllocMem( (long) sizeof(*srcFIB), MEMF_FAST) ) ) {
  1082. nomem:
  1083.         status = ERROR_NO_FREE_STORE;
  1084.         goto done;
  1085.     }
  1086.  
  1087.     if (! (srcLock = (struct Lock *) Lock(srcName, ACCESS_READ) ) ) {
  1088. err:
  1089.         status = IoErr();
  1090.         goto done;
  1091.     }
  1092.  
  1093.     if (!Examine(srcLock, srcFIB)) goto err;
  1094.     SetFileDate(dstName, &srcFIB->fib_Date);
  1095.     if (srcFIB->fib_Comment[0])
  1096.         SetComment(dstName, srcFIB->fib_Comment);
  1097.     SetProtection(dstName, srcFIB->fib_Protection);
  1098.  
  1099. done:
  1100.     if (srcLock) UnLock(srcLock);
  1101.     if (srcFIB) FreeMem(srcFIB, (long) sizeof(*srcFIB));
  1102.     return status;
  1103. }
  1104.  
  1105.  
  1106. LONG
  1107. sendpkt(id,type,args,nargs)
  1108.     struct MsgPort *id;                /* process indentifier ... (handler's
  1109.                                        message port ) */
  1110.     LONG type,                        /* packet type ... (what you want
  1111.                                        handler to do )   */
  1112.     args[],                         /* a pointer to argument list */
  1113.     nargs;                            /* number of arguments in list    */
  1114. {
  1115.  
  1116.     struct MsgPort       *replyport;
  1117.     struct StandardPacket  *packet;
  1118.  
  1119.     LONG count,*pargs,res1=NULL;
  1120.  
  1121.     if (!(replyport = (struct MsgPort   *) CreatePort(NULL,NULL)))
  1122.         return(NULL);
  1123.  
  1124.     packet = (struct StandardPacket *)
  1125.            AllocMem((LONG)sizeof(*packet),MEMF_PUBLIC|MEMF_CLEAR);
  1126.  
  1127.     if (packet) {
  1128.         packet->sp_Msg.mn_Node.ln_Name = &(packet->sp_Pkt);/* link packet */
  1129.         packet->sp_Pkt.dp_Link = &(packet->sp_Msg);/* to message    */
  1130.         packet->sp_Pkt.dp_Port = replyport;/* set-up reply port   */
  1131.         packet->sp_Pkt.dp_Type = type;/* what to do... */
  1132.  
  1133.     /* move all the arguments to the packet */
  1134.         pargs = &(packet->sp_Pkt.dp_Arg1);/* address of first argument */
  1135.         for (count=0; (count < nargs) && (count < 7); count++)
  1136.             pargs[count] = args[count];
  1137.  
  1138.         PutMsg(id,packet);                      /* send packet */
  1139.         WaitPort(replyport);            /* wait for packet to come back */
  1140.         GetMsg(replyport);                      /* pull message */
  1141.  
  1142.         res1 = packet->sp_Pkt.dp_Res1;/* get result */
  1143.         FreeMem(packet,(LONG)sizeof(*packet));
  1144.  
  1145.     }
  1146.     DeletePort(replyport);
  1147.     return(res1);
  1148. }
  1149.  
  1150.