X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=tests%2Flibpspp%2Fzip-test.c;h=3556926d6c2c01b77c078eb0d6a8fcd4c406df80;hb=refs%2Fheads%2Fctables15;hp=eded13e8bf67dccefeab4a547776716c96807c13;hpb=681f6120a7998b83ccd439068d8c4090de10b9b5;p=pspp diff --git a/tests/libpspp/zip-test.c b/tests/libpspp/zip-test.c index eded13e8bf..3556926d6c 100644 --- a/tests/libpspp/zip-test.c +++ b/tests/libpspp/zip-test.c @@ -30,7 +30,6 @@ #include #include -#include "xalloc.h" /* Exit with a failure code. (Place a breakpoint on this function while debugging.) */ @@ -43,52 +42,53 @@ check_die (void) int main (int argc, char **argv) { - if ( argc < 4) + if (argc < 4) { fprintf (stderr, "Usage zip-test: {r|w} archive file0 file1 ... filen\n"); check_die (); } - if ( 0 == strcmp ("w", argv[1])) + 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 (); + FILE *fp = fopen (argv[i], "rb"); + if (!fp) check_die (); zip_writer_add (zw, fp, argv[i]); } zip_writer_close (zw); } - else if ( 0 == strcmp ("r", argv[1])) + 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); - if ( NULL == zr) + struct zip_reader *zr; + char *error = zip_reader_create (argv[2], &zr); + if (error) { - fprintf (stderr, "Could not create zip reader: %s\n", ds_cstr (&str)); + fprintf (stderr, "Could not create zip reader: %s\n", error); check_die (); } for (i = 3; i < argc; ++i) { int x = 0; - struct zip_member *zm ; - FILE *fp = fopen (argv[i], "w"); - if ( NULL == fp) + FILE *fp = fopen (argv[i], "wb"); + if (NULL == fp) { int e = errno; fprintf (stderr, "Could not create file %s: %s\n", argv[i], strerror(e)); check_die (); } - zm = zip_member_open (zr, argv[i]); - if ( NULL == zm) + + struct zip_member *zm ; + char *error = zip_member_open (zr, argv[i], &zm); + if (error) { fprintf (stderr, "Could not open zip member %s from archive: %s\n", - argv[i], ds_cstr (&str)); + argv[i], error); check_die (); } @@ -96,16 +96,20 @@ main (int argc, char **argv) { fwrite (buf, x, 1, fp); } + error = zip_member_steal_error (zm); + zip_member_finish (zm); fclose (fp); - if ( x < 0) + + assert ((error != NULL) == (x < 0)); + if (x < 0) { - fprintf (stderr, "Unzip failed: %s\n", ds_cstr (&str)); + fprintf (stderr, "Unzip failed: %s\n", error); check_die (); } } - zip_reader_destroy (zr); + zip_reader_unref (zr); } - else + else exit (1); return 0;