home *** CD-ROM | disk | FTP | other *** search
/ Windows NT Super Tune-Up Kit / PIE-WindowsNTSuperTuneUpKit-1997.iso / COMPRESS / NT_PKZIP / ZIPUP.C < prev   
C/C++ Source or Header  |  1992-03-26  |  10KB  |  426 lines
  1. /*
  2.  
  3.  Copyright (C) 1990,1991 Mark Adler, Richard B. Wales, and Jean-loup Gailly.
  4.  Permission is granted to any individual or institution to use, copy, or
  5.  redistribute this software so long as all of the original files are included
  6.  unmodified, that it is not sold for profit, and that this copyright notice
  7.  is retained.
  8.  
  9. */
  10.  
  11. /*
  12.  *  zipup.c by Mark Adler.
  13.  */
  14.  
  15. #define NOCPYRT         /* this is not a main module */
  16. #include "zip.h"
  17. #include "revision.h"
  18.  
  19. /* Use the raw functions for MSDOS and Unix to save on buffer space.
  20.    They're not used for VMS since it doesn't work (raw is weird on VMS).
  21.    (This sort of stuff belongs in fileio.c, but oh well.) */
  22. #ifdef VMS
  23.    typedef FILE *ftype;
  24. #  define fhow FOPR
  25. #  define fbad NULL
  26. #  define zopen(n,p) fopen(n,p)
  27. #  define zread(f,b,n) fread(b,1,n,f)
  28. #  define zclose(f) fclose(f)
  29. #  define zerr(f) ferror(f)
  30. #  define zrew(f) rewind(f)
  31. #else /* !VMS */
  32. #  ifdef MSDOS
  33. #    include <io.h>
  34. #    include <fcntl.h>
  35. #    define fhow (O_RDONLY|O_BINARY)
  36. #  else /* !MSDOS */
  37.      int open OF((char *, int));
  38.      int read OF((int, char *, int));
  39.      int close OF((int));
  40.      int lseek OF((int,long,int));
  41. #    define fhow 0
  42. #  endif /* ?MSDOS */
  43.    typedef int ftype;
  44. #  define fbad (-1)
  45. #  define zopen(n,p) open(n,p)
  46. #  define zread(f,b,n) read(f,b,n)
  47. #  define zclose(f) close(f)
  48. #  define zerr(f) (k<0)
  49. #  define zrew(f) lseek(f,0L,0)
  50. #endif /* ?VMS */
  51.  
  52.  
  53. /* Local functions */
  54. #ifdef PROTO
  55. #  ifndef UTIL
  56.      local int suffixes(char *, char *);
  57. #  endif /* !UTIL */
  58. #endif /* PROTO */
  59.  
  60.  
  61. /* Note: a zip "entry" includes a local header (which includes the file
  62.    name), an encryption header if encrypting, and the compressed data. */
  63.  
  64.  
  65. int zipcopy(z, x, y)
  66. struct zlist far *z;    /* zip entry to copy */
  67. FILE *x, *y;            /* source and destination files */
  68. /* Copy the zip entry described by *z from file *x to file *y.  Return an
  69.    error code in the ZE_ class.  */
  70. {
  71.   ulg n;                /* holds local header offset */
  72.  
  73.   if (fseek(x, z->off, SEEK_SET))
  74.     return ferror(x) ? ZE_READ : ZE_EOF;
  75.   if ((n = ftell(y)) == -1L)
  76.     return ZE_TEMP;
  77.   z->off = n;
  78.   n = 4 + LOCHEAD + (long)z->nam + (long)z->ext + z->siz;
  79.   return fcopy(x, y, n);
  80. }
  81.  
  82.  
  83. #ifndef UTIL
  84.  
  85. int percent(n, m)
  86. ulg n, m;               /* n is the original size, m is the new size */
  87. /* Return the percentage compression from n to m using only integer
  88.    operations */
  89. {
  90.   if (n > 0xffffffL)            /* If n >= 16M */
  91.   {                             /*  then divide n and m by 256 */
  92.     n += 0x80;  n >>= 8;
  93.     m += 0x80;  m >>= 8;
  94.   }
  95.   return n ? (int)(1 + (200 * (n - m))/n) >> 1 : 0;
  96. }
  97.  
  98.  
  99. local int suffixes(a, s)
  100. char *a;                /* name to check suffix of */
  101. char *s;                /* list of suffixes separated by : or ; */
  102. /* Return true if a ends in any of the suffixes in the list s. */
  103. {
  104.   int m;                /* true if suffix matches so far */
  105.   char *p;              /* pointer into special */
  106.   char *q;              /* pointer into name a */
  107.  
  108.   m = 1;
  109.   q = a + strlen(a) - 1;
  110.   for (p = s + strlen(s) - 1; p >= s; p--)
  111.     if (*p == ':' || *p == ';')
  112.       if (m)
  113.         return 1;
  114.       else
  115.       {
  116.         m = 1;
  117.         q = a + strlen(a) - 1;
  118.       }
  119.     else
  120. #ifdef OS2
  121.     {
  122.       m = m && q >= a && tolower(*p) == tolower(*q);
  123.       q--;
  124.     }
  125. #else /* !OS2 */
  126.       m = m && q >= a && *p == *q--;
  127. #endif /* ?OS2 */
  128.   return m;
  129. }
  130.  
  131.  
  132. int zipup(z, y)
  133. struct zlist far *z;    /* zip entry to compress */
  134. FILE *y;                /* output file */
  135. /* Compress the file z->name into the zip entry described by *z and write
  136.    it to the file *y.  Determine the best method (store, shrink, or implode)
  137.    and use it.  Encrypt if requested.  Return an error code in the
  138.    ZE_ class. */
  139. {
  140.   ulg a;                /* attributes returned by filetime() */
  141.   char *b;              /* malloc'ed file buffer */
  142.   ulg c;                /* crc on uncompressed file data */
  143.   ftype f;              /* file to compress */
  144.   uch g;                /* flags returned by implode */
  145.   int h;                /* compression method chosen (how) */
  146. #ifndef NOIMPLODE
  147.   ulg i;                /* size of imploded data */
  148. #endif /* !NOIMPLODE */
  149.   extent k;             /* result of zread */
  150.   int l;                /* true if this file is a symbolic link */
  151.   int m;                /* method for this entry */
  152.   ulg n;                /* size of uncompressed file */
  153.   long q;               /* size returned by filetime */
  154.   int r;                /* temporary variable */
  155.   ulg s;                /* size of shrunk or compressed data */
  156.  
  157.   /* Open file to zip up */
  158.   if ((z->tim = filetime(z->name, &a, &q)) == 0 || q < 0)
  159.     return ZE_OPEN;
  160.   l = issymlnk(a);
  161.   if (l)
  162.     f = fbad;
  163.   else if ((f = zopen(z->name, fhow)) == fbad)
  164.     return ZE_OPEN;
  165.  
  166.   /* Select method based on the suffix and the global method */
  167.   m = special != NULL && suffixes(z->name, special) ? STORE : method;
  168.  
  169.   /* Don't bother with shrink for "large" files (320 is what PKZIP uses--the
  170.      number used here must be <= BSZ) */
  171.   if (m == BEST && q > 512)
  172.     m = IMPLODE;
  173.   if (q == 0)
  174.     m = STORE;
  175.  
  176.   /* Make first pass on the file, computing the crc and length, and running
  177.      shrink and implode on it. */
  178.   n = 0;
  179.   c = updcrc((char *)NULL, 0);
  180.  
  181.   if ((b = malloc(BSZ)) == NULL)
  182.     return ZE_MEM;
  183.   if (m == BEST || m == SHRINK)
  184.     if ((r = shr_setup()) != ZE_OK)
  185.       return r;
  186.     else
  187.       shract = 1;
  188.   while ((k = l ? rdsymlnk(z->name, b, BSZ) : zread(f, b, BSZ)) > 0)
  189.   {
  190.     n += k;
  191.     c = updcrc(b, k);
  192. #ifdef MINIX
  193.     if (l)
  194.       q = k;
  195. #endif /* MINIX */
  196.     if ((m == BEST || m == SHRINK) && (r = shr_p1((uch *)b, k)) != ZE_OK)
  197.     {
  198.       free((voidp *)b);
  199.       shr_clear();
  200.       shract = 0;
  201.       return r;
  202.     }
  203.     if (m == BEST)              /* free up shrink data structures */
  204.     {
  205.       if ((!l && zread(f, b, BSZ) != 0)
  206. #ifndef VMS
  207.           || n != (ulg)q
  208. #endif /* !VMS */
  209.          )
  210.         return ZE_READ;
  211.       if ((r = shr_size(&s)) != ZE_OK)
  212.       {
  213.         shr_clear();
  214.         shract = 0;
  215.         return r;
  216.       }
  217.     }
  218. #ifndef NOIMPLODE
  219.     if (m == BEST || m == IMPLODE)
  220.     {
  221.       if (!impact)
  222.         if ((r = imp_setup(q, level)) != ZE_OK)
  223.           return r;
  224.         else
  225.           impact = 1;
  226.       if ((r = imp_p1(b, k)) != ZE_OK)
  227.       {
  228.         free((voidp *)b);
  229.         imp_clear();
  230.         impact = 0;
  231.         return r;
  232.       }
  233.     }
  234. #endif /* !NOIMPLODE */
  235.     if (m == BEST || l)
  236.       break;
  237.   }
  238.   free((voidp *)b);
  239.   if (!l && zerr(f))
  240.     return ZE_READ;
  241.   /* Determine the best method to use */
  242.   g = 0;
  243.   if (noisy && verbose)
  244.     printf(" (n=%lu)", n);
  245.   if (m == SHRINK && (r = shr_size(&s)) != ZE_OK)
  246.   {
  247.  
  248.     shr_clear();
  249.     shract = 0;
  250.     return r;
  251.   }
  252.   if (noisy && verbose && (m == BEST || m == SHRINK))
  253.     printf(" (s=%lu)", s);
  254. #ifndef NOIMPLODE
  255.   if ((m == BEST || m == IMPLODE) && (r = imp_size(&i, &g)) != ZE_OK)
  256.   {
  257.     imp_clear();
  258.     impact = 0;
  259.     return r;
  260.   }
  261.   if (noisy && verbose && (m == BEST || m == IMPLODE))
  262.     printf(" (i=%lu)", i);
  263.   if ((m == BEST || m == IMPLODE) && i < n && (m == IMPLODE || i < s))
  264.   {
  265.     if (noisy)
  266.     {
  267.       printf(" (imploded %d%%)", percent(n, i));
  268.       fflush(stdout);
  269.     }
  270.     h = IMPLODE;
  271.     s = i;
  272.     if (m == BEST)
  273.     {
  274.       shr_clear();
  275.       shract = 0;
  276.     }
  277.     if (!l)
  278.       zclose(f);
  279.   }
  280.   else
  281. #endif /* !NOIMPLODE */
  282.   if ((m == BEST || m == SHRINK) && s < n)
  283.   {
  284.     if (noisy)
  285.     {
  286.       printf(" (shrunk %d%%)", percent(n, s));
  287.       fflush(stdout);
  288.     }
  289.     h = SHRINK;
  290. #ifndef NOIMPLODE
  291.     if (m == BEST)
  292.     {
  293.       imp_clear();
  294.       impact = 0;
  295.     }
  296. #endif /* !NOIMPLODE */
  297.     if (!l)
  298.       zclose(f);
  299.   }
  300.   else
  301.   {
  302.     if (noisy)
  303.     {
  304.       printf(" (stored 0%%)");
  305.       fflush(stdout);
  306.     }
  307.     h = STORE;
  308.     s = n;
  309. #ifndef NOIMPLODE
  310.     if (m == BEST || m == IMPLODE)
  311.     {
  312.       imp_clear();
  313.       impact = 0;
  314.     }
  315. #endif /* !NOIMPLODE */
  316.     if (m == BEST || m == SHRINK)
  317.     {
  318.       shr_clear();
  319.       shract = 0;
  320.     }
  321.   }
  322.  
  323. #ifndef VMS
  324.   /* Check size (but not in VMS--variable record lengths mess it up) */
  325.   if (n != (ulg)q)
  326.     return ZE_READ;
  327. #endif /* !VMS */
  328.  
  329.   /* Fill in header information and write local header to zip file */
  330. #ifdef OS2
  331.   if ( dosify < 2 )
  332.     dosify = IsFileSystemFAT(z->name);
  333. #endif /* OS2 */
  334.  
  335.   /* (Assume ext, cext, com, and zname already filled in.) */
  336.   z->vem = dosify ? 11 :                /* Made under MSDOS by PKZIP 1.1 */
  337. #ifdef VMS
  338.                     0x200 + REVISION;   /* Made under VMS by this Zip */
  339. #else /* !VMS */
  340. #ifdef OS2
  341.                     0x600 + REVISION;   /* Made under OS/2 by this Zip */
  342. #else /* !OS2 */
  343.                     0x300 + REVISION;   /* Made under Unix by this Zip */
  344. #endif /* ?OS2 */
  345. #endif /* ?VMS */
  346.   z->ver = 10;                          /* Need PKUNZIP 1.0 */
  347.   z->flg = (ush)g;
  348.   if (key != NULL)
  349.     z->flg |= 1;
  350.   z->lflg = z->flg;
  351.   z->how = h;
  352.   z->crc = c;
  353.   z->siz = s;
  354.   if (key != NULL)
  355.     z->siz += 12;
  356.   z->len = n;
  357.   z->nam = strlen(z->zname);
  358.   z->dsk = 0;
  359.   z->att = 0;                           /* Assume they're all binary */
  360.   z->atx = dosify ? a & 0xff : a;       /* Attributes from filetime() */
  361.   if ((z->off = ftell(y)) == -1L)
  362.     return ZE_WRITE;
  363.   if ((r = putlocal(z, y)) != ZE_OK)
  364.     return r;
  365.  
  366.   /* Write stored, shrunk, or imploded file to zip file */
  367. #ifndef EXPORT
  368.   if (key != NULL)
  369.     crypthead(key, z->crc, y);
  370. #endif /* !EXPORT */
  371.   n = ftell(y);                         /* Save offset for logic check */
  372. #ifndef NOIMPLODE
  373.   if (h == IMPLODE)
  374.   {
  375.     if ((r = imp_p2(y)) != ZE_OK)
  376.       return r;
  377.     imp_clear();
  378.     impact = 0;
  379.   }
  380.   else
  381. #endif /* !NOIMPLODE */
  382.   if (h == SHRINK)
  383.   {
  384.     if ((r = shr_p2(y)) != ZE_OK)
  385.       return r;
  386.     shr_clear();
  387.     shract = 0;
  388.   }
  389.   else
  390.   {
  391.     if (!l)
  392.       zrew(f);
  393.     if ((b = malloc(BSZ)) == NULL)
  394.       return ZE_MEM;
  395.     while ((k = l ? rdsymlnk(z->name, b, BSZ) : zread(f, b, BSZ)) > 0)
  396.     {
  397.       if (zfwrite(b, 1, k, y) != k)
  398.       {
  399.         free((voidp *)b);
  400.         return ZE_TEMP;
  401.       }
  402.       if (l)
  403.         break;
  404.     }
  405.     free((voidp *)b);
  406.     if (!l && zerr(f))
  407.       return ZE_READ;
  408.     if (!l)
  409.       zclose(f);
  410.   }
  411.  
  412.   /* Check length of compressed data */
  413.   if (ftell(y) != n + s)
  414.     return ZE_LOGIC;
  415.  
  416.   /* Done--clean up and leave */
  417.   if (noisy)
  418.   {
  419.     putchar('\n');
  420.     fflush(stdout);
  421.   }
  422.   return ZE_OK;
  423. }
  424.  
  425. #endif /* !UTIL */
  426.