contrib/libarchive/libarchive/test/test_write_format_mtree_classic.c

   1 /*-
   2  * Copyright (c) 2012 Michihiro NAKAJIMA
   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  *    in this position and unchanged.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #include "test.h"
  28 __FBSDID("$FreeBSD$");
  29
  30 static char buff[4096];
  31 static struct {
  32         const char      *path;
  33         mode_t           mode;
  34         int              nlink;
  35         time_t           mtime;
  36         uid_t            uid;
  37         gid_t            gid;
  38 } entries[] = {
  39         { ".",                  S_IFDIR | 0755, 3, 1231975636, 1001, 1001 },
  40         { "./COPYING",          S_IFREG | 0644, 1, 1231975636, 1001, 1001 },
  41         { "./Makefile",         S_IFREG | 0644, 1, 1233041050, 1001, 1001 },
  42         { "./NEWS",             S_IFREG | 0644, 1, 1231975636, 1001, 1001 },
  43         { "./PROJECTS",         S_IFREG | 0644, 1, 1231975636, 1001, 1001 },
  44         { "./README",           S_IFREG | 0644, 1, 1231975636, 1001, 1001 },
  45         { "./subdir",           S_IFDIR | 0755, 3, 1233504586, 1001, 1001 },
  46         { "./subdir/README",    S_IFREG | 0664, 1, 1231975636, 1002, 1001 },
  47         { "./subdir/config",    S_IFREG | 0664, 1, 1232266273, 1003, 1003 },
  48         { "./subdir2",          S_IFDIR | 0755, 3, 1233504586, 1001, 1001 },
  49         { "./subdir3",          S_IFDIR | 0755, 3, 1233504586, 1001, 1001 },
  50         { "./subdir3/mtree",    S_IFREG | 0664, 2, 1232266273, 1003, 1003 },
  51         { NULL, 0, 0, 0, 0, 0 }
  52 };
  53
  54 static const char image [] = {
  55 "#mtree\n"
  56 "\n"
  57 "# .\n"
  58 "/set type=file uid=1001 gid=1001 mode=644\n"
  59 ".               time=1231975636.0 mode=755 type=dir\n"
  60 "    COPYING     time=1231975636.0 size=8\n"
  61 "    Makefile    time=1233041050.0 size=8\n"
  62 "    NEWS        time=1231975636.0 size=8\n"
  63 "    PROJECTS    time=1231975636.0 size=8\n"
  64 "    README      time=1231975636.0 size=8\n"
  65 "\n"
  66 "# ./subdir\n"
  67 "/set mode=664\n"
  68 "subdir          time=1233504586.0 mode=755 type=dir\n"
  69 "    README      time=1231975636.0 uid=1002 size=8\n"
  70 "    config      time=1232266273.0 gid=1003 uid=1003 size=8\n"
  71 "# ./subdir\n"
  72 "..\n"
  73 "\n"
  74 "\n"
  75 "# ./subdir2\n"
  76 "subdir2         time=1233504586.0 mode=755 type=dir\n"
  77 "# ./subdir2\n"
  78 "..\n"
  79 "\n"
  80 "\n"
  81 "# ./subdir3\n"
  82 "subdir3         time=1233504586.0 mode=755 type=dir\n"
  83 "    mtree       nlink=2 time=1232266273.0 gid=1003 uid=1003 size=8\n"
  84 "# ./subdir3\n"
  85 "..\n"
  86 "\n"
  87 "..\n\n"
  88 };
  89
  90 static const char image_dironly [] = {
  91 "#mtree\n"
  92 "# .\n"
  93 "/set type=dir uid=1001 gid=1001 mode=755\n"
  94 ".               time=1231975636.0\n"
  95 "# ./subdir\n"
  96 "subdir          time=1233504586.0\n"
  97 "# ./subdir\n"
  98 "..\n"
  99 "# ./subdir2\n"
 100 "subdir2         time=1233504586.0\n"
 101 "# ./subdir2\n"
 102 "..\n"
 103 "# ./subdir3\n"
 104 "subdir3         time=1233504586.0\n"
 105 "# ./subdir3\n"
 106 "..\n"
 107 "..\n"
 108 };
 109
 110 static void
 111 test_write_format_mtree_sub(int dironly)
 112 {
 113         struct archive_entry *ae;
 114         struct archive* a;
 115         size_t used;
 116         int i;
 117
 118         /* Create a mtree format archive. */
 119         assert((a = archive_write_new()) != NULL);
 120         assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_mtree_classic(a));
 121         if (dironly)
 122                 assertEqualIntA(a, ARCHIVE_OK,
 123                         archive_write_set_format_option(a, NULL, "dironly", "1"));
 124         assertEqualIntA(a, ARCHIVE_OK,
 125                 archive_write_open_memory(a, buff, sizeof(buff)-1, &used));
 126
 127         /* Write entries */
 128         for (i = 0; entries[i].path != NULL; i++) {
 129                 assert((ae = archive_entry_new()) != NULL);
 130                 archive_entry_set_nlink(ae, entries[i].nlink);
 131                 assertEqualInt(entries[i].nlink, archive_entry_nlink(ae));
 132                 archive_entry_set_mtime(ae, entries[i].mtime, 0);
 133                 assertEqualInt(entries[i].mtime, archive_entry_mtime(ae));
 134                 archive_entry_set_mode(ae, entries[i].mode);
 135                 assertEqualInt(entries[i].mode, archive_entry_mode(ae));
 136                 archive_entry_set_uid(ae, entries[i].uid);
 137                 assertEqualInt(entries[i].uid, archive_entry_uid(ae));
 138                 archive_entry_set_gid(ae, entries[i].gid);
 139                 assertEqualInt(entries[i].gid, archive_entry_gid(ae));
 140                 archive_entry_copy_pathname(ae, entries[i].path);
 141                 if ((entries[i].mode & AE_IFMT) != S_IFDIR)
 142                         archive_entry_set_size(ae, 8);
 143                 assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
 144                 if ((entries[i].mode & AE_IFMT) != S_IFDIR)
 145                         assertEqualIntA(a, 8,
 146                             archive_write_data(a, "Hello012", 15));
 147                 archive_entry_free(ae);
 148         }
 149         assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
 150         assertEqualInt(ARCHIVE_OK, archive_write_free(a));
 151
 152         buff[used] = '\0';
 153         if (dironly)
 154                 assertEqualString(buff, image_dironly);
 155         else
 156                 assertEqualString(buff, image);
 157
 158         /*
 159          * Read the data and check it.
 160          */
 161         assert((a = archive_read_new()) != NULL);
 162         assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
 163         assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a));
 164         assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used));
 165
 166         /* Read entries */
 167         for (i = 0; entries[i].path != NULL; i++) {
 168                 if (dironly && (entries[i].mode & AE_IFMT) != S_IFDIR)
 169                         continue;
 170                 assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
 171                 assertEqualInt(entries[i].mtime, archive_entry_mtime(ae));
 172                 assertEqualInt(entries[i].mode, archive_entry_mode(ae));
 173                 assertEqualInt(entries[i].uid, archive_entry_uid(ae));
 174                 assertEqualInt(entries[i].gid, archive_entry_gid(ae));
 175                 if (i > 0)
 176                         assertEqualString(entries[i].path + 2,
 177                                 archive_entry_pathname(ae));
 178                 else
 179                         assertEqualString(entries[i].path,
 180                                 archive_entry_pathname(ae));
 181                 if ((entries[i].mode & AE_IFMT) != S_IFDIR)
 182                         assertEqualInt(8, archive_entry_size(ae));
 183         }
 184         assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
 185         assertEqualInt(ARCHIVE_OK, archive_read_free(a));
 186 }
 187
 188 DEFINE_TEST(test_write_format_mtree_classic)
 189 {
 190         /* Generate classic format. */
 191         test_write_format_mtree_sub(0);
 192         /* Generate classic format and Write directory only. */
 193         test_write_format_mtree_sub(1);
 194 }