home *** CD-ROM | disk | FTP | other *** search
/ Liren Large Software Subsidy 7 / 07.iso / c / c019 / 2.ddi / SOURCE / ZIPNOTE.C < prev    next >
Encoding:
C/C++ Source or Header  |  1993-09-10  |  10.2 KB  |  367 lines
  1. /*
  2.  
  3.  Copyright (C) 1990-1993 Mark Adler, Richard B. Wales, Jean-loup Gailly,
  4.  Kai Uwe Rommel and Igor Mandrichenko.
  5.  Permission is granted to any individual or institution to use, copy, or
  6.  redistribute this software so long as all of the original files are included,
  7.  that it is not sold for profit, and that this copyright notice is retained.
  8.  
  9. */
  10.  
  11. /*
  12.  *  zipnote.c by Mark Adler.
  13.  */
  14.  
  15. #define UTIL
  16. #include "revision.h"
  17. #include "zip.h"
  18. #include <signal.h>
  19.  
  20.  
  21. /* Character to mark zip entry names in the comment file */
  22. #define MARK '@'
  23. #define MARKE " (comment above this line)"
  24. #define MARKZ " (zip file comment below this line)"
  25.  
  26. /* Temporary zip file name and file pointer */
  27. local char *tempzip;
  28. local FILE *tempzf;
  29.  
  30.  
  31. /* Local functions */
  32. #ifdef PROTO
  33.    local void handler(int);
  34.    local void license(void);
  35.    local void help(void);
  36.    local void putclean(char *, int);
  37.    local int catalloc(char * far *, char *);
  38.    int main(int, char **);
  39. #endif /* PROTO */
  40.  
  41.  
  42.  
  43. void err(c, h)
  44. int c;                  /* error code from the ZE_ class */
  45. char *h;                /* message about how it happened */
  46. /* Issue a message for the error, clean up files and memory, and exit. */
  47. {
  48.   if (PERR(c))
  49.     perror("zipnote error");
  50.   fprintf(stderr, "zipnote error: %s (%s)\n", errors[c-1], h);
  51.   if (tempzf != NULL)
  52.     fclose(tempzf);
  53.   if (tempzip != NULL)
  54.   {
  55.     destroy(tempzip);
  56.     free((voidp *)tempzip);
  57.   }
  58.   if (zipfile != NULL)
  59.     free((voidp *)zipfile);
  60. #ifdef VMS
  61.   exit(0);
  62. #else /* !VMS */
  63.   exit(c);
  64. #endif /* ?VMS */
  65. }
  66.  
  67.  
  68. local void handler(s)
  69. int s;                  /* signal number (ignored) */
  70. /* Upon getting a user interrupt, abort cleanly using err(). */
  71. {
  72. #ifndef MSDOS
  73.   putc('\n', stderr);
  74. #endif /* !MSDOS */
  75.   err(ZE_ABORT, "aborting");
  76.   s++;                                  /* keep some compilers happy */
  77. }
  78.  
  79.  
  80. void warn(a, b)
  81. char *a, *b;            /* message strings juxtaposed in output */
  82. /* Print a warning message to stderr and return. */
  83. {
  84.   fprintf(stderr, "zipnote warning: %s%s\n", a, b);
  85. }
  86.  
  87.  
  88. local void license()
  89. /* Print license information to stdout. */
  90. {
  91.   extent i;             /* counter for copyright array */
  92.  
  93.   for (i = 0; i < sizeof(copyright)/sizeof(char *); i++) {
  94.     printf(copyright[i], "zipnote");
  95.     putchar('\n');
  96.   }
  97.   for (i = 0; i < sizeof(disclaimer)/sizeof(char *); i++)
  98.     puts(disclaimer[i]);
  99. }
  100.  
  101.  
  102. local void help()
  103. /* Print help (along with license info) to stdout. */
  104. {
  105.   extent i;             /* counter for help array */
  106.  
  107.   /* help array */
  108.   static char *text[] = {
  109. "",
  110. "ZipNote %s (%s)",
  111. "Usage:  zipnote [-w] [-b path] zipfile",
  112. "  the default action is to write the comments in zipfile to stdout",
  113. "  -w   write the zipfile comments from stdin",
  114. "  -b   use \"path\" for the temporary zip file",
  115. "  -h   show this help               -L   show software license",
  116. "",
  117. "Example:",
  118. #ifdef VMS
  119. "     define/user sys$output foo.tmp",
  120. "     zipnote foo.zip",
  121. "     edit foo.tmp",
  122. "     ... then you edit the comments, save, and exit ...",
  123. "     define/user sys$input foo.tmp",
  124. "     zipnote -w foo.zip"
  125. #else /* !VMS */
  126. "     zipnote foo.zip > foo.tmp",
  127. "     ed foo.tmp",
  128. "     ... then you edit the comments, save, and exit ...",
  129. "     zipnote -w foo.zip < foo.tmp"
  130. #endif /* ?VMS */
  131.   };
  132.  
  133.   for (i = 0; i < sizeof(copyright)/sizeof(char *); i++) {
  134.     printf(copyright[i], "zipnote");
  135.     putchar('\n');
  136.   }
  137.   for (i = 0; i < sizeof(text)/sizeof(char *); i++)
  138.   {
  139.     printf(text[i], VERSION, REVDATE);
  140.     putchar('\n');
  141.   }
  142. }
  143.  
  144.  
  145. local void putclean(s, n)
  146. char *s;                /* string to write to stdout */
  147. int n;                  /* length of string */
  148. /* Write the string s to stdout, filtering out control characters that are
  149.    not tab or newline (mainly to remove carriage returns), and prefix MARK's
  150.    and backslashes with a backslash.  Also, terminate with a newline if
  151.    needed. */
  152. {
  153.   int c;                /* next character in string */
  154.   int e;                /* last character written */
  155.  
  156.   e = '\n';                     /* if empty, write nothing */
  157.   while (n--)
  158.   {
  159.     c = *(uch *)s++;
  160.     if (c == MARK || c == '\\')
  161.       putchar('\\');
  162.     if (c >= ' ' || c == '\t' || c == '\n')
  163.       { e=c; putchar(e); }
  164.   }
  165.   if (e != '\n')
  166.     putchar('\n');
  167. }
  168.  
  169.  
  170. local int catalloc(a, s)
  171. char * far *a;          /* pointer to a pointer to a malloc'ed string */
  172. char *s;                /* string to concatenate on a */
  173. /* Concatentate the string s to the malloc'ed string pointed to by a.
  174.    Preprocess s by removing backslash escape characters. */
  175. {
  176.   char *p;              /* temporary pointer */
  177.   char *q;              /* temporary pointer */
  178.  
  179.   for (p = q = s; *q; *p++ = *q++)
  180.     if (*q == '\\' && *(q+1))
  181.       q++;
  182.   *p = 0;
  183.   if ((p = malloc(strlen(*a) + strlen(s) + 3)) == NULL)
  184.     return ZE_MEM;
  185.   strcat(strcat(strcpy(p, *a), **a ? "\r\n" : ""), s);
  186.   free((voidp *)*a);
  187.   *a = p;
  188.   return ZE_OK;
  189. }
  190.  
  191.  
  192. int main(argc, argv)
  193. int argc;               /* number of tokens in command line */
  194. char **argv;            /* command line tokens */
  195. /* Write the comments in the zipfile to stdout, or read them from stdin. */
  196. {
  197.   char a[FNMAX+1];      /* input line buffer */
  198.   ulg c;                /* start of central directory */
  199.   int k;                /* next argument type */
  200.   char *q;              /* steps through option arguments */
  201.   int r;                /* arg counter, temporary variable */
  202.   ulg s;                /* length of central directory */
  203.   int t;                /* attributes of zip file */
  204.   int w;                /* true if updating zip file from stdin */
  205.   FILE *x, *y;          /* input and output zip files */
  206.   struct zlist far *z;  /* steps through zfiles linked list */
  207.  
  208.  
  209.   /* If no args, show help */
  210.   if (argc == 1)
  211.   {
  212.     help();
  213.     exit(0);
  214.   }
  215.  
  216.   init_upper();           /* build case map table */
  217.  
  218.   /* Go through args */
  219.   zipfile = tempzip = NULL;
  220.   tempzf = NULL;
  221.   signal(SIGINT, handler);
  222. #ifdef SIGTERM              /* AMIGA has no SIGTERM */
  223.   signal(SIGTERM, handler);
  224. #endif
  225.   k = w = 0;
  226.   for (r = 1; r < argc; r++)
  227.     if (*argv[r] == '-')
  228.       if (argv[r][1])
  229.         for (q = argv[r]+1; *q; q++)
  230.           switch(*q)
  231.           {
  232.             case 'b':   /* Specify path for temporary file */
  233.               if (k)
  234.                 err(ZE_PARMS, "use -b before zip file name");
  235.               else
  236.                 k = 1;          /* Next non-option is path */
  237.               break;
  238.             case 'h':   /* Show help */
  239.               help();  exit(0);
  240.             case 'l':  case 'L':  /* Show copyright and disclaimer */
  241.               license();  exit(0);
  242.             case 'w':
  243.               w = 1;  break;
  244.             default:
  245.               err(ZE_PARMS, "unknown option");
  246.           }
  247.       else
  248.         err(ZE_PARMS, "zip file cannot be stdin");
  249.     else
  250.       if (k == 0)
  251.         if (zipfile == NULL)
  252.         {
  253.           if ((zipfile = ziptyp(argv[r])) == NULL)
  254.             err(ZE_MEM, "was processing arguments");
  255.         }
  256.         else
  257.           err(ZE_PARMS, "can only specify one zip file");
  258.       else
  259.       {
  260.         tempath = argv[r];
  261.         k = 0;
  262.       }
  263.   if (zipfile == NULL)
  264.     err(ZE_PARMS, "need to specify zip file");
  265.  
  266.   /* Read zip file */
  267.   if ((r = readzipfile()) != ZE_OK)
  268.     err(r, zipfile);
  269.   if (zfiles == NULL)
  270.     err(ZE_NAME, zipfile);
  271.  
  272.   /* Put comments to stdout, if not -w */
  273.   if (!w)
  274.   {
  275.     for (z = zfiles; z != NULL; z = z->nxt)
  276.     {
  277.       printf("%c %s\n", MARK, z->zname);
  278.       putclean(z->comment, z->com);
  279.       printf("%c%s\n", MARK, MARKE);
  280.     }
  281.     printf("%c%s\n", MARK, MARKZ);
  282.     putclean(zcomment, zcomlen);
  283.     exit(ZE_OK);
  284.   }
  285.  
  286.   /* If updating comments, make sure zip file is writeable */
  287.   if ((x = fopen(zipfile, "a")) == NULL)
  288.     err(ZE_CREAT, zipfile);
  289.   fclose(x);
  290.   t = getfileattr(zipfile);
  291.  
  292.   /* Process stdin, replacing comments */
  293.   z = zfiles;
  294.   while (gets(a) != NULL && (a[0] != MARK || strcmp(a + 1, MARKZ)))
  295.   {                                     /* while input and not file comment */
  296.     if (a[0] != MARK || a[1] != ' ')    /* better be "@ name" */
  297.       err(ZE_NOTE, "unexpected input");
  298.     while (z != NULL && strcmp(a + 2, z->zname))
  299.       z = z->nxt;                       /* allow missing entries in order */
  300.     if (z == NULL)
  301.       err(ZE_NOTE, "unknown entry name");
  302.     if (z->com)                         /* change zip entry comment */
  303.       free((voidp *)z->comment);
  304.     z->comment = malloc(1);  *(z->comment) = 0;
  305.     while (gets(a) != NULL && *a != MARK)
  306.       if ((r = catalloc(&(z->comment), a)) != ZE_OK)
  307.         err(r, "was building new comments");
  308.     z->com = strlen(z->comment);
  309.     z = z->nxt;                         /* point to next entry */
  310.   }
  311.   if (a != NULL)                        /* change zip file comment */
  312.   {
  313.     zcomment = malloc(1);  *zcomment = 0;
  314.     while (gets(a) != NULL)
  315.       if ((r = catalloc(&zcomment, a)) != ZE_OK)
  316.         err(r, "was building new comments");
  317.     zcomlen = strlen(zcomment);
  318.   }
  319.  
  320.   /* Open output zip file for writing */
  321.   if ((tempzf = y = fopen(tempzip = tempname(zipfile), FOPW)) == NULL)
  322.     err(ZE_TEMP, tempzip);
  323.  
  324.   /* Open input zip file again, copy preamble if any */
  325.   if ((x = fopen(zipfile, FOPR)) == NULL)
  326.     err(ZE_NAME, zipfile);
  327.   if (zipbeg && (r = fcopy(x, y, zipbeg)) != ZE_OK)
  328.     err(r, r == ZE_TEMP ? tempzip : zipfile);
  329.  
  330.   /* Go through local entries, copying them over as is */
  331.   for (z = zfiles; z != NULL; z = z->nxt)
  332.     if ((r = zipcopy(z, x, y)) != ZE_OK)
  333.       err(r, "was copying an entry");
  334.   fclose(x);
  335.  
  336.   /* Write central directory and end of central directory with new comments */
  337.   if ((c = ftell(y)) == -1L)    /* get start of central */
  338.     err(ZE_TEMP, tempzip);
  339.   for (z = zfiles; z != NULL; z = z->nxt)
  340.     if ((r = putcentral(z, y)) != ZE_OK)
  341.       err(r, tempzip);
  342.   if ((s = ftell(y)) == -1L)    /* get end of central */
  343.     err(ZE_TEMP, tempzip);
  344.   s -= c;                       /* compute length of central */
  345.   if ((r = putend((int)zcount, s, c, zcomlen, zcomment, y)) != ZE_OK)
  346.     err(r, tempzip);
  347.   tempzf = NULL;
  348.   if (fclose(y))
  349.     err(ZE_TEMP, tempzip);
  350.   if ((r = replace(zipfile, tempzip)) != ZE_OK)
  351.   {
  352.     warn("new zip file left as: ", tempzip);
  353.     free((voidp *)tempzip);
  354.     tempzip = NULL;
  355.     err(r, "was replacing the original zip file");
  356.   }
  357.   free((voidp *)tempzip);
  358.   tempzip = NULL;
  359.   setfileattr(zipfile, t);
  360.   free((voidp *)zipfile);
  361.   zipfile = NULL;
  362.  
  363.   /* Done! */
  364.   exit(ZE_OK);
  365.   return 0; /* avoid warning */
  366. }
  367.