1 /*- 2 * Copyright (c) 2009 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 time_t mtime; 35 uid_t uid; 36 gid_t gid; 37 } entries[] = { 38 { "./COPYING", S_IFREG | 0644, 1231975636, 1001, 1001 }, 39 { "./Makefile", S_IFREG | 0644, 1233041050, 1001, 1001 }, 40 { "./NEWS", S_IFREG | 0644, 1231975636, 1001, 1001 }, 41 { "./PROJECTS", S_IFREG | 0644, 1231975636, 1001, 1001 }, 42 { "./README", S_IFREG | 0644, 1231975636, 1001, 1001 }, 43 { "./subdir", S_IFDIR | 0755, 1233504586, 1001, 1001 }, 44 { "./subdir/README", S_IFREG | 0664, 1231975636, 1002, 1001 }, 45 { "./subdir/config", S_IFREG | 0664, 1232266273, 1003, 1003 }, 46 { "./subdir2", S_IFDIR | 0755, 1233504586, 1001, 1001 }, 47 { "./subdir3", S_IFDIR | 0755, 1233504586, 1001, 1001 }, 48 { "./subdir3/mtree", S_IFREG | 0664, 1232266273, 1003, 1003 }, 49 { NULL, 0, 0, 0, 0 } 50 }; 51 static struct { 52 const char *path; 53 mode_t mode; 54 time_t mtime; 55 uid_t uid; 56 gid_t gid; 57 } entries2[] = { 58 { "COPYING", S_IFREG | 0644, 1231975636, 1001, 1001 }, 59 { "Makefile", S_IFREG | 0644, 1233041050, 1001, 1001 }, 60 { "NEWS", S_IFREG | 0644, 1231975636, 1001, 1001 }, 61 { "PROJECTS", S_IFREG | 0644, 1231975636, 1001, 1001 }, 62 { "README", S_IFREG | 0644, 1231975636, 1001, 1001 }, 63 { "subdir", S_IFDIR | 0755, 1233504586, 1001, 1001 }, 64 { "subdir/README", S_IFREG | 0664, 1231975636, 1002, 1001 }, 65 { "subdir/config", S_IFREG | 0664, 1232266273, 1003, 1003 }, 66 { "subdir2", S_IFDIR | 0755, 1233504586, 1001, 1001 }, 67 { "subdir3", S_IFDIR | 0755, 1233504586, 1001, 1001 }, 68 { "subdir3/mtree", S_IFREG | 0664, 1232266273, 1003, 1003 }, 69 { NULL, 0, 0, 0, 0 } 70 }; 71 72 static void 73 test_write_format_mtree_sub(int use_set, int dironly) 74 { 75 struct archive_entry *ae; 76 struct archive* a; 77 size_t used; 78 int i; 79 80 /* Create a mtree format archive. */ 81 assert((a = archive_write_new()) != NULL); 82 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_mtree(a)); 83 if (use_set) 84 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_option(a, NULL, "use-set", "1")); 85 if (dironly) 86 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_option(a, NULL, "dironly", "1")); 87 assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, sizeof(buff)-1, &used)); 88 89 /* Write entries */ 90 for (i = 0; entries[i].path != NULL; i++) { 91 assert((ae = archive_entry_new()) != NULL); 92 archive_entry_set_mtime(ae, entries[i].mtime, 0); 93 assert(entries[i].mtime == archive_entry_mtime(ae)); 94 archive_entry_set_mode(ae, entries[i].mode); 95 assert(entries[i].mode == archive_entry_mode(ae)); 96 archive_entry_set_uid(ae, entries[i].uid); 97 assert(entries[i].uid == archive_entry_uid(ae)); 98 archive_entry_set_gid(ae, entries[i].gid); 99 assert(entries[i].gid == archive_entry_gid(ae)); 100 archive_entry_copy_pathname(ae, entries[i].path); 101 if ((entries[i].mode & AE_IFMT) != S_IFDIR) 102 archive_entry_set_size(ae, 8); 103 assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae)); 104 if ((entries[i].mode & AE_IFMT) != S_IFDIR) 105 assertEqualIntA(a, 8, 106 archive_write_data(a, "Hello012", 15)); 107 archive_entry_free(ae); 108 } 109 assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a)); 110 assertEqualInt(ARCHIVE_OK, archive_write_free(a)); 111 112 if (use_set) { 113 const char *p; 114 115 buff[used] = '\0'; 116 assert(NULL != (p = strstr(buff, "\n/set "))); 117 if (p != NULL) { 118 char *r; 119 const char *o; 120 p++; 121 r = strchr(p, '\n'); 122 if (r != NULL) 123 *r = '\0'; 124 if (dironly) 125 o = "/set type=dir uid=1001 gid=1001 mode=755"; 126 else 127 o = "/set type=file uid=1001 gid=1001 mode=644"; 128 assertEqualString(o, p); 129 if (r != NULL) 130 *r = '\n'; 131 } 132 } 133 134 /* 135 * Read the data and check it. 136 */ 137 assert((a = archive_read_new()) != NULL); 138 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a)); 139 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a)); 140 assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used)); 141 142 /* Read entries */ 143 for (i = 0; entries[i].path != NULL; i++) { 144 if (dironly && (entries[i].mode & AE_IFMT) != S_IFDIR) 145 continue; 146 assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae)); 147 assertEqualInt(entries[i].mtime, archive_entry_mtime(ae)); 148 assertEqualInt(entries[i].mode, archive_entry_mode(ae)); 149 assertEqualInt(entries[i].uid, archive_entry_uid(ae)); 150 assertEqualInt(entries[i].gid, archive_entry_gid(ae)); 151 assertEqualString(entries[i].path, archive_entry_pathname(ae)); 152 if ((entries[i].mode & AE_IFMT) != S_IFDIR) 153 assertEqualInt(8, archive_entry_size(ae)); 154 } 155 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a)); 156 assertEqualInt(ARCHIVE_OK, archive_read_free(a)); 157 } 158 159 static void 160 test_write_format_mtree_sub2(int use_set, int dironly) 161 { 162 struct archive_entry *ae; 163 struct archive* a; 164 size_t used; 165 int i; 166 char str[32]; 167 168 /* Create a mtree format archive. */ 169 assert((a = archive_write_new()) != NULL); 170 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_mtree(a)); 171 if (use_set) 172 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_option(a, NULL, "use-set", "1")); 173 if (dironly) 174 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_option(a, NULL, "dironly", "1")); 175 assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, sizeof(buff)-1, &used)); 176 177 /* Write entries2 */ 178 for (i = 0; entries2[i].path != NULL; i++) { 179 assert((ae = archive_entry_new()) != NULL); 180 archive_entry_set_mtime(ae, entries2[i].mtime, 0); 181 assert(entries2[i].mtime == archive_entry_mtime(ae)); 182 archive_entry_set_mode(ae, entries2[i].mode); 183 assert(entries2[i].mode == archive_entry_mode(ae)); 184 archive_entry_set_uid(ae, entries2[i].uid); 185 assert(entries2[i].uid == archive_entry_uid(ae)); 186 archive_entry_set_gid(ae, entries2[i].gid); 187 assert(entries2[i].gid == archive_entry_gid(ae)); 188 archive_entry_copy_pathname(ae, entries2[i].path); 189 if ((entries2[i].mode & AE_IFMT) != S_IFDIR) 190 archive_entry_set_size(ae, 8); 191 assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae)); 192 if ((entries2[i].mode & AE_IFMT) != S_IFDIR) 193 assertEqualIntA(a, 8, 194 archive_write_data(a, "Hello012", 15)); 195 archive_entry_free(ae); 196 } 197 assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a)); 198 assertEqualInt(ARCHIVE_OK, archive_write_free(a)); 199 200 if (use_set) { 201 const char *p; 202 203 buff[used] = '\0'; 204 assert(NULL != (p = strstr(buff, "\n/set "))); 205 if (p != NULL) { 206 char *r; 207 const char *o; 208 p++; 209 r = strchr(p, '\n'); 210 if (r != NULL) 211 *r = '\0'; 212 if (dironly) 213 o = "/set type=dir uid=1001 gid=1001 mode=755"; 214 else 215 o = "/set type=file uid=1001 gid=1001 mode=644"; 216 assertEqualString(o, p); 217 if (r != NULL) 218 *r = '\n'; 219 } 220 } 221 222 /* 223 * Read the data and check it. 224 */ 225 assert((a = archive_read_new()) != NULL); 226 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a)); 227 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a)); 228 assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used)); 229 230 /* Read entries2 */ 231 memset(str, 0, sizeof(str)); 232 strcpy(str, "./"); 233 for (i = 0; entries2[i].path != NULL; i++) { 234 if (dironly && (entries2[i].mode & AE_IFMT) != S_IFDIR) 235 continue; 236 assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae)); 237 assertEqualInt(entries2[i].mtime, archive_entry_mtime(ae)); 238 assertEqualInt(entries2[i].mode, archive_entry_mode(ae)); 239 assertEqualInt(entries2[i].uid, archive_entry_uid(ae)); 240 assertEqualInt(entries2[i].gid, archive_entry_gid(ae)); 241 strcpy(str + 2, entries2[i].path); 242 assertEqualString(str, archive_entry_pathname(ae)); 243 if ((entries2[i].mode & AE_IFMT) != S_IFDIR) 244 assertEqualInt(8, archive_entry_size(ae)); 245 } 246 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a)); 247 assertEqualInt(ARCHIVE_OK, archive_read_free(a)); 248 } 249 250 DEFINE_TEST(test_write_format_mtree) 251 { 252 /* Default setting */ 253 test_write_format_mtree_sub(0, 0); 254 /* Directory only */ 255 test_write_format_mtree_sub(0, 1); 256 /* Use /set keyword */ 257 test_write_format_mtree_sub(1, 0); 258 /* Use /set keyword with directory only */ 259 test_write_format_mtree_sub(1, 1); 260 } 261 262 DEFINE_TEST(test_write_format_mtree_no_leading_dotslash) 263 { 264 /* Default setting */ 265 test_write_format_mtree_sub2(0, 0); 266 /* Directory only */ 267 test_write_format_mtree_sub2(0, 1); 268 /* Use /set keyword */ 269 test_write_format_mtree_sub2(1, 0); 270 /* Use /set keyword with directory only */ 271 test_write_format_mtree_sub2(1, 1); 272 } 273