pintos-os.org Git - pspp/blob - tests/libpspp/zip-test.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2011 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation, either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
  16
  17
  18 /* A simple program to zip or unzip a file */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include <config.h>
  22 #endif
  23
  24 #include <string.h>
  25 #include <stdio.h>
  26 #include "libpspp/assertion.h"
  27 #include <libpspp/compiler.h>
  28 #include <libpspp/zip-writer.h>
  29 #include <libpspp/zip-reader.h>
  30 #include <libpspp/str.h>
  31
  32 #include <errno.h>
  33 \f
  34 /* Exit with a failure code.
  35    (Place a breakpoint on this function while debugging.) */
  36 static void
  37 check_die (void)
  38 {
  39   exit (EXIT_FAILURE);
  40 }
  41
  42 int
  43 main (int argc, char **argv)
  44 {
  45   if (argc < 4)
  46     {
  47       fprintf (stderr, "Usage zip-test: {r|w} archive file0 file1 ... filen\n");
  48       check_die ();
  49     }
  50
  51   if (0 == strcmp ("w", argv[1]))
  52     {
  53       int i;
  54       struct zip_writer *zw = zip_writer_create (argv[2]);
  55       for (i = 3; i < argc; ++i)
  56         {
  57           FILE *fp = fopen (argv[i], "rb");
  58           if (!fp) check_die ();
  59           zip_writer_add (zw, fp, argv[i]);
  60         }
  61       zip_writer_close (zw);
  62     }
  63   else if (0  == strcmp ("r", argv[1]))
  64     {
  65       const int BUFSIZE=256;
  66       char buf[BUFSIZE];
  67       int i;
  68       struct zip_reader *zr;
  69       char *error = zip_reader_create (argv[2], &zr);
  70       if (error)
  71         {
  72           fprintf (stderr, "Could not create zip reader: %s\n", error);
  73           check_die ();
  74         }
  75       for (i = 3; i < argc; ++i)
  76         {
  77           int x = 0;
  78           FILE *fp = fopen (argv[i], "wb");
  79           if (NULL == fp)
  80             {
  81               int e = errno;
  82               fprintf (stderr, "Could not create file %s: %s\n", argv[i], strerror(e));
  83               check_die ();
  84             }
  85
  86           struct zip_member *zm ;
  87           char *error = zip_member_open (zr, argv[i], &zm);
  88           if (error)
  89             {
  90               fprintf (stderr, "Could not open zip member %s from archive: %s\n",
  91                        argv[i], error);
  92               check_die ();
  93             }
  94
  95           while ((x = zip_member_read (zm, buf, BUFSIZE)) > 0)
  96             {
  97               fwrite (buf, x, 1, fp);
  98             }
  99           error = zip_member_steal_error (zm);
 100           zip_member_finish (zm);
 101           fclose (fp);
 102
 103           assert ((error != NULL) == (x < 0));
 104           if (x < 0)
 105             {
 106               fprintf (stderr, "Unzip failed: %s\n", error);
 107               check_die ();
 108             }
 109         }
 110       zip_reader_unref (zr);
 111     }
 112   else
 113     exit (1);
 114
 115   return 0;
 116 }