X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=tests%2Flibpspp%2Fzip-test.c;h=3556926d6c2c01b77c078eb0d6a8fcd4c406df80;hb=1e9875fc6bd073f970e28de5ef49e82717705858;hp=e132cb93da5299983a35438bf4d7da00b4d39949;hpb=7d683abd904f855c4dc5614a326dc6b2146853c9;p=pspp diff --git a/tests/libpspp/zip-test.c b/tests/libpspp/zip-test.c index e132cb93da..3556926d6c 100644 --- a/tests/libpspp/zip-test.c +++ b/tests/libpspp/zip-test.c @@ -42,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 (); } @@ -95,15 +96,18 @@ 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 exit (1);