contrib/libarchive/libarchive/test/test_tar_filenames.c

   1 /*-
   2  * Copyright (c) 2003-2007 Tim Kientzle
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
  15  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  16  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  17  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
  18  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  19  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  20  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  21  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25 #include "test.h"
  26 __FBSDID("$FreeBSD$");
  27
  28 /*
  29  * Exercise various lengths of filenames in tar archives,
  30  * especially around the magic sizes where ustar breaks
  31  * filenames into prefix/suffix.
  32  */
  33
  34 static void
  35 test_filename(const char *prefix, int dlen, int flen)
  36 {
  37         char buff[8192];
  38         char filename[400];
  39         char dirname[400];
  40         struct archive_entry *ae;
  41         struct archive *a;
  42         size_t used;
  43         char *p;
  44         int i;
  45
  46         p = filename;
  47         if (prefix) {
  48                 strcpy(filename, prefix);
  49                 p += strlen(p);
  50         }
  51         if (dlen > 0) {
  52                 for (i = 0; i < dlen; i++)
  53                         *p++ = 'a';
  54                 *p++ = '/';
  55         }
  56         for (i = 0; i < flen; i++)
  57                 *p++ = 'b';
  58         *p = '\0';
  59
  60         strcpy(dirname, filename);
  61
  62         /* Create a new archive in memory. */
  63         assert((a = archive_write_new()) != NULL);
  64         assertA(0 == archive_write_set_format_pax_restricted(a));
  65         assertA(0 == archive_write_add_filter_none(a));
  66         assertA(0 == archive_write_set_bytes_per_block(a,0));
  67         assertA(0 == archive_write_open_memory(a, buff, sizeof(buff), &used));
  68
  69         /*
  70          * Write a file to it.
  71          */
  72         assert((ae = archive_entry_new()) != NULL);
  73         archive_entry_copy_pathname(ae, filename);
  74         archive_entry_set_mode(ae, S_IFREG | 0755);
  75         failure("Pathname %d/%d", dlen, flen);
  76         assertA(0 == archive_write_header(a, ae));
  77         archive_entry_free(ae);
  78
  79         /*
  80          * Write a dir to it (without trailing '/').
  81          */
  82         assert((ae = archive_entry_new()) != NULL);
  83         archive_entry_copy_pathname(ae, dirname);
  84         archive_entry_set_mode(ae, S_IFDIR | 0755);
  85         failure("Dirname %d/%d", dlen, flen);
  86         assertA(0 == archive_write_header(a, ae));
  87         archive_entry_free(ae);
  88
  89         /* Tar adds a '/' to directory names. */
  90         strcat(dirname, "/");
  91
  92         /*
  93          * Write a dir to it (with trailing '/').
  94          */
  95         assert((ae = archive_entry_new()) != NULL);
  96         archive_entry_copy_pathname(ae, dirname);
  97         archive_entry_set_mode(ae, S_IFDIR | 0755);
  98         failure("Dirname %d/%d", dlen, flen);
  99         assertA(0 == archive_write_header(a, ae));
 100         archive_entry_free(ae);
 101
 102         /* Close out the archive. */
 103         assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
 104         assertEqualInt(ARCHIVE_OK, archive_write_free(a));
 105
 106         /*
 107          * Now, read the data back.
 108          */
 109         assert((a = archive_read_new()) != NULL);
 110         assertA(0 == archive_read_support_format_all(a));
 111         assertA(0 == archive_read_support_filter_all(a));
 112         assertA(0 == archive_read_open_memory(a, buff, used));
 113
 114         /* Read the file and check the filename. */
 115         assertA(0 == archive_read_next_header(a, &ae));
 116         assertEqualString(filename, archive_entry_pathname(ae));
 117         assertEqualInt((S_IFREG | 0755), archive_entry_mode(ae));
 118
 119         /*
 120          * Read the two dirs and check the names.
 121          *
 122          * Both dirs should read back with the same name, since
 123          * tar should add a trailing '/' to any dir that doesn't
 124          * already have one.  We only report the first such failure
 125          * here.
 126          */
 127         assertA(0 == archive_read_next_header(a, &ae));
 128         assertEqualString(dirname, archive_entry_pathname(ae));
 129         assert((S_IFDIR | 0755) == archive_entry_mode(ae));
 130
 131         assertA(0 == archive_read_next_header(a, &ae));
 132         assertEqualString(dirname, archive_entry_pathname(ae));
 133         assert((S_IFDIR | 0755) == archive_entry_mode(ae));
 134
 135         /* Verify the end of the archive. */
 136         assert(1 == archive_read_next_header(a, &ae));
 137         assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
 138         assertEqualInt(ARCHIVE_OK, archive_read_free(a));
 139 }
 140
 141 DEFINE_TEST(test_tar_filenames)
 142 {
 143         int dlen, flen;
 144
 145         /* Repeat the following for a variety of dir/file lengths. */
 146         for (dlen = 45; dlen < 55; dlen++) {
 147                 for (flen = 45; flen < 55; flen++) {
 148                         test_filename(NULL, dlen, flen);
 149                         test_filename("/", dlen, flen);
 150                 }
 151         }
 152
 153         for (dlen = 0; dlen < 140; dlen += 10) {
 154                 for (flen = 98; flen < 102; flen++) {
 155                         test_filename(NULL, dlen, flen);
 156                         test_filename("/", dlen, flen);
 157                 }
 158         }
 159
 160         for (dlen = 140; dlen < 160; dlen++) {
 161                 for (flen = 95; flen < 105; flen++) {
 162                         test_filename(NULL, dlen, flen);
 163                         test_filename("/", dlen, flen);
 164                 }
 165         }
 166 }