Implemented a zip-writer to correspond to zip-reader

[pspp] / tests / libpspp / zip-test.c

diff --git a/tests/libpspp/zip-test.c b/tests/libpspp/zip-test.c
new file mode 100644 (file)
index 0000000..7a2fcb1
--- /dev/null
+++ b/tests/libpspp/zip-test.c
@@ -0,0 +1,95 @@
+/* PSPP - a program for statistical analysis.
+   Copyright (C) 2011 Free Software Foundation, Inc.
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>. */
+
+
+/* A simple program to zip or unzip a file */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <string.h>
+#include <stdio.h>
+#include "libpspp/assertion.h"
+#include <libpspp/compiler.h>
+#include <libpspp/zip-writer.h>
+#include <libpspp/zip-reader.h>
+#include <libpspp/str.h>
+
+#include "xalloc.h"
+\f
+/* Exit with a failure code.
+   (Place a breakpoint on this function while debugging.) */
+static void
+check_die (void)
+{
+  exit (EXIT_FAILURE);
+}
+
+int
+main (int argc, char **argv)
+{
+  if ( argc < 4)
+    {
+      fprintf (stderr, "Usage zip-test: {r|w} archive file0 file1 ... filen\n");
+      check_die ();
+    }
+
+  if ( 0 == strcmp ("w", argv[1]))
+    {
+      int i;
+      struct zip_writer *zw = zip_writer_create (argv[2]);
+      for (i = 3; i < argc; ++i)
+       {
+         FILE *fp = fopen (argv[i], "r");
+         if (!fp ) check_die ();
+         zip_writer_add (zw, fp, argv[i]);
+       }
+      zip_writer_close (zw);
+    }
+  else if ( 0  == strcmp ("r", argv[1]))
+    {
+      const int BUFSIZE=256;
+      char buf[BUFSIZE];
+      int i;
+      struct string str;
+      struct zip_reader *zr = zip_reader_create (argv[2], &str);
+      for (i = 3; i < argc; ++i)
+       {
+         int x = 0;
+         struct zip_member *zm = zip_member_open (zr, argv[i]);
+         FILE *fp = fopen (argv[i], "w");
+
+         while ((x = zip_member_read (zm, buf, BUFSIZE)) > 0)
+           {
+             fwrite (buf, x, 1, fp);
+           }
+         fclose (fp);
+         if ( x < 0)
+           {
+             fprintf (stderr, "Unzip failed: %s", ds_cstr (&str));
+             check_die ();
+           }
+         
+         zip_member_unref (zm);
+       }
+      zip_reader_destroy (zr);
+    }
+  else 
+    exit (1);
+
+  return 0;
+}